Merge pull request #616 from Chia-Network/sk-sign-augment

CHIA-922: CHIA-938: Add signing and verification methods

crates/chia-bls/src/public_key.rs

@@ -313,6 +313,10 @@ impl PublicKey {
         Self::generator()
     }
 
+    pub fn verify(&self, signature: &crate::Signature, msg: &[u8]) -> bool {
+        crate::verify(signature, self, msg)
+    }
+
     pub fn pair(&self, other: &crate::Signature) -> crate::GTElement {
         other.pair(self)
     }

crates/chia-bls/src/secret_key.rs

@@ -242,8 +242,15 @@ impl SecretKey {
     #[classattr]
     pub const PRIVATE_KEY_SIZE: usize = 32;
 
-    pub fn sign_g2(&self, msg: &[u8]) -> crate::Signature {
-        crate::sign(self, msg)
+    pub fn sign(&self, msg: &[u8], final_pk: Option<PublicKey>) -> crate::Signature {
+        match final_pk {
+            Some(prefix) => {
+                let mut aug_msg = prefix.to_bytes().to_vec();
+                aug_msg.extend_from_slice(msg);
+                crate::sign_raw(self, aug_msg)
+            }
+            None => crate::sign(self, msg),
+        }
     }
 
     pub fn get_g1(&self) -> PublicKey {

wheel/generate_type_stubs.py

@@ -379,6 +379,7 @@ def __init__(
             "SIZE: ClassVar[int] = ...",
             "def __new__(cls) -> G1Element: ...",
             "def get_fingerprint(self) -> int: ...",
+            "def verify(self, signature: G2Element, msg: bytes) -> bool: ...",
             "def pair(self, other: G2Element) -> GTElement: ...",
             "@staticmethod",
             "def generator() -> G1Element: ...",
@@ -420,7 +421,7 @@ def __init__(
         [],
         [
             "PRIVATE_KEY_SIZE: ClassVar[int] = ...",
-            "def sign_g2(self, msg: bytes, dst: bytes) -> G2Element: ...",
+            "def sign(self, msg: bytes, final_pk: Optional[G1Element] = None) -> G2Element: ...",
             "def get_g1(self) -> G1Element: ...",
             "def __str__(self) -> str: ...",
             "def public_key(self) -> G1Element: ...",

wheel/python/chia_rs/chia_rs.pyi

@@ -117,6 +117,7 @@ class G1Element:
     SIZE: ClassVar[int] = ...
     def __new__(cls) -> G1Element: ...
     def get_fingerprint(self) -> int: ...
+    def verify(self, signature: G2Element, msg: bytes) -> bool: ...
     def pair(self, other: G2Element) -> GTElement: ...
     @staticmethod
     def generator() -> G1Element: ...
@@ -206,7 +207,7 @@ class GTElement:
 
 class PrivateKey:
     PRIVATE_KEY_SIZE: ClassVar[int] = ...
-    def sign_g2(self, msg: bytes, dst: bytes) -> G2Element: ...
+    def sign(self, msg: bytes, final_pk: Optional[G1Element] = None) -> G2Element: ...
     def get_g1(self) -> G1Element: ...
     def __str__(self) -> str: ...
     def public_key(self) -> G1Element: ...