X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Futil%2Fmessage_signing.rs;h=79572f31246a7ae53fa1885b13ee3cc282c68d50;hb=9f1ffab24c4870d0f9846d75a693f2a784648a60;hp=477cbcbdd9aa93493de6703e0d4d57cd81c135d2;hpb=32a0113fda05f6c7c8b7b12359c9ef82ab999e07;p=rust-lightning

diff --git a/lightning/src/util/message_signing.rs b/lightning/src/util/message_signing.rs
index 477cbcbd..79572f31 100644
--- a/lightning/src/util/message_signing.rs
+++ b/lightning/src/util/message_signing.rs
@@ -20,6 +20,7 @@
 //! <https://lightning.readthedocs.io/lightning-signmessage.7.html>
 //! <https://api.lightning.community/#signmessage>
 
+#[allow(unused)]
 use crate::prelude::*;
 use crate::util::base32;
 use bitcoin::hashes::{sha256d, Hash};
@@ -57,7 +58,7 @@ pub fn sign(msg: &[u8], sk: &SecretKey) -> Result<String, Error> {
 	let secp_ctx = Secp256k1::signing_only();
 	let msg_hash = sha256d::Hash::hash(&[LN_MESSAGE_PREFIX, msg].concat());
 
-	let sig = secp_ctx.sign_ecdsa_recoverable(&Message::from_slice(&msg_hash)?, sk);
+	let sig = secp_ctx.sign_ecdsa_recoverable(&Message::from_slice(msg_hash.as_byte_array())?, sk);
 	Ok(base32::Alphabet::ZBase32.encode(&sigrec_encode(sig)))
 }
 
@@ -69,7 +70,7 @@ pub fn recover_pk(msg: &[u8], sig: &str) ->  Result<PublicKey, Error> {
 	match base32::Alphabet::ZBase32.decode(&sig) {
 		Ok(sig_rec) => {
 			match sigrec_decode(sig_rec) {
-				Ok(sig) => secp_ctx.recover_ecdsa(&Message::from_slice(&msg_hash)?, &sig),
+				Ok(sig) => secp_ctx.recover_ecdsa(&Message::from_slice(msg_hash.as_byte_array())?, &sig),
 				Err(e) => Err(e)
 			}
 		},
@@ -90,13 +91,14 @@ pub fn verify(msg: &[u8], sig: &str, pk: &PublicKey) -> bool {
 mod test {
 	use core::str::FromStr;
 	use crate::util::message_signing::{sign, recover_pk, verify};
-	use bitcoin::secp256k1::ONE_KEY;
-	use bitcoin::secp256k1::{PublicKey, Secp256k1};
+	use bitcoin::secp256k1::constants::ONE;
+	use bitcoin::secp256k1::{PublicKey, SecretKey, Secp256k1};
 
 	#[test]
 	fn test_sign() {
 		let message = "test message";
-		let zbase32_sig = sign(message.as_bytes(), &ONE_KEY);
+		let one_key = SecretKey::from_slice(&ONE).unwrap();
+		let zbase32_sig = sign(message.as_bytes(), &one_key);
 
 		assert_eq!(zbase32_sig.unwrap(), "d9tibmnic9t5y41hg7hkakdcra94akas9ku3rmmj4ag9mritc8ok4p5qzefs78c9pqfhpuftqqzhydbdwfg7u6w6wdxcqpqn4sj4e73e")
 	}
@@ -104,17 +106,19 @@ mod test {
 	#[test]
 	fn test_recover_pk() {
 		let message = "test message";
+		let one_key = SecretKey::from_slice(&ONE).unwrap();
 		let sig = "d9tibmnic9t5y41hg7hkakdcra94akas9ku3rmmj4ag9mritc8ok4p5qzefs78c9pqfhpuftqqzhydbdwfg7u6w6wdxcqpqn4sj4e73e";
 		let pk = recover_pk(message.as_bytes(), sig);
 
-		assert_eq!(pk.unwrap(), PublicKey::from_secret_key(&Secp256k1::signing_only(), &ONE_KEY))
+		assert_eq!(pk.unwrap(), PublicKey::from_secret_key(&Secp256k1::signing_only(), &one_key))
 	}
 
 	#[test]
 	fn test_verify() {
 		let message = "another message";
-		let sig = sign(message.as_bytes(), &ONE_KEY).unwrap();
-		let pk = PublicKey::from_secret_key(&Secp256k1::signing_only(), &ONE_KEY);
+		let one_key = SecretKey::from_slice(&ONE).unwrap();
+		let sig = sign(message.as_bytes(), &one_key).unwrap();
+		let pk = PublicKey::from_secret_key(&Secp256k1::signing_only(), &one_key);
 
 		assert!(verify(message.as_bytes(), &sig, &pk))
 	}