fix: check runtime in _submit

waterfall-mkt · Dec 19, 2023 · 76a3b6c · 76a3b6c
1 parent 5a5e8f1
commit 76a3b6c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/CurtaGolf.sol b/src/CurtaGolf.sol
@@ -191,17 +191,17 @@ 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 solution 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.
+        if (!purityChecker.check(target.code, getAllowedOpcodes[_courseId])) {
+            revert PollutedSolution();
+        }
+
         // Run solution and mint NFT if it beats the leading score.
         uint32 gasUsed = courseData.course.run(target, block.prevrandao);
         if (courseData.gasUsed == 0 || gasUsed < courseData.gasUsed) {