Allow 0-length labels.

arkworks-rs · Sep 8, 2023 · 049a978 · 049a978
1 parent a29ce38
commit 049a978
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/plugins/mod.rs b/src/plugins/mod.rs
@@ -4,8 +4,8 @@ pub mod arkworks;
 #[cfg(feature = "dalek")]
 pub mod dalek;
 
-#[cfg(feature="zkcrypto")]
+#[cfg(feature = "zkcrypto")]
 pub mod zkcrypto;
 
-#[cfg(all(test, feature="arkworks", feature="dalek", feature="zkcrypto"))]
+#[cfg(all(test, feature = "arkworks", feature = "dalek", feature = "zkcrypto"))]
 mod tests;
diff --git a/src/safe.rs b/src/safe.rs
@@ -70,15 +70,15 @@ impl<H: DuplexHash> IOPattern<H> {
     pub fn absorb(self, count: usize, label: &'static str) -> Self {
         assert!(count > 0, "Count must be positive");
         assert!(!label.contains(SEP_BYTE));
-        assert!(!label[..1].parse::<u8>().is_ok());
+        assert!(label.len() == 0 || !label[..1].parse::<u8>().is_ok());
 
         Self::from_string(self.io + SEP_BYTE + &format!("A{}", count) + label)
     }
 
     pub fn squeeze(self, count: usize, label: &'static str) -> Self {
         assert!(count > 0, "Count must be positive");
         assert!(!label.contains(SEP_BYTE));
-        assert!(!label[..1].parse::<u8>().is_ok());
+        assert!(label.len() == 0 || !label[..1].parse::<u8>().is_ok());
 
         Self::from_string(self.io + SEP_BYTE + &format!("S{}", count) + label)
     }