check available cycles and accepted cycles

zkemail · Oct 28, 2024 · f792e6d · f792e6d
1 parent b3e5e7e
commit f792e6d
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/src/dns_client/src/lib.rs b/src/dns_client/src/lib.rs
@@ -44,7 +44,10 @@ pub async fn get_dkim_public_key(
             return Err("Insufficient cycles".to_string());
         }
     }
-    ic_cdk::api::call::msg_cycles_accept128(available_cycles);
+    let accepted_cycles = ic_cdk::api::call::msg_cycles_accept128(available_cycles);
+    if available_cycles != accepted_cycles {
+        return Err("Fail to accept all available cycles".to_string());
+    }
     // Verify selector and domain
     if !Regex::new(SELECTOR_REGEX).unwrap().is_match(&selector) {
         return Err("Invalid domain".to_string());

diff --git a/src/ic_dns_oracle_backend/src/lib.rs b/src/ic_dns_oracle_backend/src/lib.rs
@@ -275,7 +275,10 @@ pub async fn sign_dkim_public_key(
         }
     }
     // Accept all available cycles.
-    ic_cdk::api::call::msg_cycles_accept128(available_cycles);
+    let accepted_cycles = ic_cdk::api::call::msg_cycles_accept128(available_cycles);
+    if available_cycles != accepted_cycles {
+        return Err("Fail to accept all available cycles".to_string());
+    }
     write_log(
         "sign_dkim_public_key",
         &format!(
@@ -469,7 +472,10 @@ pub async fn revoke_dkim_public_key(
         }
     }
     // Accept all available cycles.
-    ic_cdk::api::call::msg_cycles_accept128(available_cycles);
+    let accepted_cycles = ic_cdk::api::call::msg_cycles_accept128(available_cycles);
+    if available_cycles != accepted_cycles {
+        return Err("Fail to accept all available cycles".to_string());
+    }
     write_log(
         "revoke_dkim_public_key",
         &format!(

diff --git a/src/poseidon/src/lib.rs b/src/poseidon/src/lib.rs
@@ -25,7 +25,10 @@ pub fn public_key_hash(public_key_hex: String) -> Result<String, String> {
         }
     }
     // Accept all available cycles.
-    ic_cdk::api::call::msg_cycles_accept128(available_cycles);
+    let accepted_cycles = ic_cdk::api::call::msg_cycles_accept128(available_cycles);
+    if available_cycles != accepted_cycles {
+        return Err("Fail to accept all available cycles".to_string());
+    }
     _public_key_hash(public_key_hex)
 }