fix: add back initcode check

waterfall-mkt · Dec 23, 2023 · 65b085e · 65b085e
1 parent 6f279bb
commit 65b085e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/CurtaGolf.sol b/src/CurtaGolf.sol
@@ -191,13 +191,18 @@ contract CurtaGolf is ICurtaGolf, KingERC721, Owned {
         // Revert if the course does not exist.
         if (address(courseData.course) == address(0)) revert CourseDoesNotExist(_courseId);
 
+        // Revert if the initcode contains invalid opcodes.
+        if (!purityChecker.check(_solution, getAllowedOpcodes[_courseId])) {
+            revert PollutedSolution();
+        }
+
         // Deploy the solution.
         address target;
         assembly {
             target := create(0, add(_solution, 0x20), mload(_solution))
         }
 
-        // Revert if the solution contains invalid opcodes.
+        // Revert if the runtime contains invalid opcodes.
         if (!purityChecker.check(target.code, getAllowedOpcodes[_courseId])) {
             revert PollutedSolution();
         }