better import for libsecp256k1

stratum-mining · May 12, 2024 · 14f090d · 14f090d
1 parent a803f1a
commit 14f090d
Show file tree

Hide file tree

Showing 12 changed files with 1,218 additions and 55 deletions.
diff --git a/benches/Cargo.lock b/benches/Cargo.lock
diff --git a/common/Cargo.lock b/common/Cargo.lock
diff --git a/common/Cargo.toml b/common/Cargo.toml
@@ -8,3 +8,4 @@ repository = "https://github.com/stratum-mining/stratum"
 
 [dependencies]
 bitcoin = {version="0.29.1",optional=true}
+secp256k1 = { version = "0.28.2", default-features = false, features =["alloc","rand","rand-std"] }
diff --git a/common/src/lib.rs b/common/src/lib.rs
@@ -4,3 +4,4 @@
 //! and manage the shared dependencies and utils across stratum crates.
 #[cfg(feature = "bitcoin")]
 pub use bitcoin;
+pub use secp256k1;
diff --git a/protocols/Cargo.lock b/protocols/Cargo.lock
diff --git a/roles/Cargo.lock b/roles/Cargo.lock
diff --git a/roles/jd-server/Cargo.toml b/roles/jd-server/Cargo.toml
@@ -30,6 +30,5 @@ serde_json = { version = "1.0", default-features = false, features = ["alloc","r
 serde = { version = "1.0.89", features = ["derive", "alloc"], default-features = false }
 hashbrown = { version = "0.11", default-features = false, features = ["ahash", "serde"] }
 key-utils = { version = "^1.0.0", path = "../../utils/key-utils" }
-secp256k1 = { version = "0.28.2", default-features = false, features =["alloc","rand","rand-std"] }
 rpc_sv2 = { version = "1.0.0", path = "../roles-utils/rpc" }
 hex = "0.4.3"
diff --git a/roles/jd-server/src/lib/job_declarator/mod.rs b/roles/jd-server/src/lib/job_declarator/mod.rs
@@ -4,7 +4,7 @@ use async_channel::{Receiver, Sender};
 use binary_sv2::{B0255, U256};
 use codec_sv2::{Frame, HandshakeRole, Responder};
 use error_handling::handle_result;
-use key_utils::{Secp256k1PublicKey, Secp256k1SecretKey};
+use key_utils::{Secp256k1PublicKey, Secp256k1SecretKey, SignatureService};
 use network_helpers_sv2::noise_connection_tokio::Connection;
 use nohash_hasher::BuildNoHashHasher;
 use roles_logic_sv2::{
@@ -14,7 +14,6 @@ use roles_logic_sv2::{
     parsers::{JobDeclaration, PoolMessages as JdsMessages},
     utils::{Id, Mutex},
 };
-use secp256k1::{Keypair, Message as SecpMessage, Secp256k1};
 use std::{collections::HashMap, convert::TryInto, sync::Arc};
 use tokio::{net::TcpListener, time::Duration};
 use tracing::{debug, error, info};
@@ -395,15 +394,9 @@ pub fn signed_token(
     _pub_key: &Secp256k1PublicKey,
     prv_key: &Secp256k1SecretKey,
 ) -> B0255<'static> {
-    let secp = Secp256k1::signing_only();
+    let secp = SignatureService::default();
 
-    // Create the SecretKey and PublicKey instances
-    let secret_key = prv_key.0;
-    let kp = Keypair::from_secret_key(&secp, &secret_key);
-
-    let message: Vec<u8> = tx_hash_list_hash.to_vec();
-
-    let signature = secp.sign_schnorr(&SecpMessage::from_digest_slice(&message).unwrap(), &kp);
+    let signature = secp.sign(tx_hash_list_hash.to_vec(), prv_key.0);
 
     // Sign message
     signature.as_ref().to_vec().try_into().unwrap()

diff --git a/roles/pool/Cargo.toml b/roles/pool/Cargo.toml
@@ -30,7 +30,6 @@ async-recursion = "1.0.0"
 error_handling = { version = "1.0.0", path = "../../utils/error-handling" }
 nohash-hasher = "0.2.0"
 key-utils = { version = "^1.0.0", path = "../../utils/key-utils" }
-secp256k1 = { version = "0.28.2", default-features = false, features =["alloc","rand","rand-std"] }
 
 [dev-dependencies]
 hex = "0.4.3"