add grinding checks #84

NilFoundation · Nov 8, 2023 · 55f4c2a · 55f4c2a
1 parent cbeb5e2
commit 55f4c2a
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 10 deletions.
diff --git a/contracts/zkllvm/circuit1/commitment.sol b/contracts/zkllvm/circuit1/commitment.sol
@@ -453,8 +453,9 @@ library modular_commitment_scheme_circuit1 {
 
         bytes calldata proof_of_work = blob[blob.length - 4:];
         transcript.update_transcript(tr_state, proof_of_work);
-        transcript.get_integral_challenge_be(tr_state, 4);
-
+        uint256 p_o_w = transcript.get_integral_challenge_be(tr_state, 4);
+        if (p_o_w & 4294901760 != 0) return false;
+
 
             unchecked{
                 offset += 0x8 + r;
@@ -586,4 +587,4 @@ library modular_commitment_scheme_circuit1 {
         return true;
     }
 }        
-
+
diff --git a/contracts/zkllvm/circuit2/commitment.sol b/contracts/zkllvm/circuit2/commitment.sol
@@ -585,4 +585,4 @@ library modular_commitment_scheme_circuit2 {
         return true;
     }
 }        
-
+
diff --git a/contracts/zkllvm/circuit3/commitment.sol b/contracts/zkllvm/circuit3/commitment.sol
@@ -457,8 +457,9 @@ library modular_commitment_scheme_circuit3 {
 
         bytes calldata proof_of_work = blob[blob.length - 4:];
         transcript.update_transcript(tr_state, proof_of_work);
-        transcript.get_integral_challenge_be(tr_state, 4);
-
+        uint256 p_o_w = transcript.get_integral_challenge_be(tr_state, 4);
+        if (p_o_w & 4294901760 != 0) return false;
+
 
             unchecked{
                 offset += 0x8 + r;

diff --git a/contracts/zkllvm/circuit4/commitment.sol b/contracts/zkllvm/circuit4/commitment.sol
@@ -457,8 +457,9 @@ library modular_commitment_scheme_circuit4 {
 
         bytes calldata proof_of_work = blob[blob.length - 4:];
         transcript.update_transcript(tr_state, proof_of_work);
-        transcript.get_integral_challenge_be(tr_state, 4);
-
+        uint256 p_o_w = transcript.get_integral_challenge_be(tr_state, 4);
+        if (p_o_w & 4294901760 != 0) return false;
+
 
             unchecked{
                 offset += 0x8 + r;

diff --git a/contracts/zkllvm/circuit6/commitment.sol b/contracts/zkllvm/circuit6/commitment.sol
@@ -460,8 +460,9 @@ library modular_commitment_scheme_circuit6 {
 
         bytes calldata proof_of_work = blob[blob.length - 4:];
         transcript.update_transcript(tr_state, proof_of_work);
-        transcript.get_integral_challenge_be(tr_state, 4);
-
+        uint256 p_o_w = transcript.get_integral_challenge_be(tr_state, 4);
+        if (p_o_w & 4294901760 != 0) return false;
+
 
             unchecked{
                 offset += 0x8 + r;
Original file line number	Diff line number	Diff line change
Expand Up		@@ -585,4 +585,4 @@ library modular_commitment_scheme_circuit2 {
		return true;
		}
		}