allow use of custom nonce in ecdsa_sign

ludbb · Mar 29, 2016 · 0adbf67 · 0adbf67
1 parent b4475d9
commit 0adbf67
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/secp256k1/__init__.py b/secp256k1/__init__.py
@@ -380,12 +380,15 @@ def tweak_mul(self, scalar):
         """
         return _tweak_private(self, lib.secp256k1_ec_privkey_tweak_mul, scalar)
 
-    def ecdsa_sign(self, msg, raw=False, digest=hashlib.sha256):
+    def ecdsa_sign(self, msg, raw=False, digest=hashlib.sha256, custom_nonce=None):
         msg32 = _hash32(msg, raw, digest)
         raw_sig = ffi.new('secp256k1_ecdsa_signature *')
-
+        nonce_fn = ffi.NULL
+        nonce_data = ffi.NULL
+        if custom_nonce:
+            nonce_fn, nonce_data = custom_nonce
         signed = lib.secp256k1_ecdsa_sign(
-            self.ctx, raw_sig, msg32, self.private_key, ffi.NULL, ffi.NULL)
+            self.ctx, raw_sig, msg32, self.private_key, nonce_fn, nonce_data)
         assert signed == 1
 
         return raw_sig