Merge remote-tracking branch 'rust-bitcoin/master'

* rust-bitcoin/master:
  Fix no-std raw test, after removal of lang items
  Fix broken benchmarks
  Disable emscripten tests until they work again https://github.com/rust-lang/rust/issues/66916 https://github.com/rustwasm/team/issues/291
  Add constant of the prime of the curve field.
  Simplify callback logic to returning raw coordinates
  Removed no longer used dont_replace_c_symbols feature
  Fix wrong feature name external-symbols
  Fix missing return c_int in NonceFn
This commit is contained in:
Andronik Ordian 2020-04-09 15:13:43 +02:00
commit b22b9e5709
No known key found for this signature in database
GPG Key ID: C66F3C680DE0E6ED
9 changed files with 85 additions and 162 deletions

View File

@ -1,7 +1,7 @@
language: rust language: rust
cache: # cache:
directories: # directories:
- cargo_web # - cargo_web
rust: rust:
- stable - stable
@ -45,12 +45,12 @@ script:
- cargo run --example generate_keys --features=rand - cargo run --example generate_keys --features=rand
- if [ ${TRAVIS_RUST_VERSION} == "stable" ]; then cargo doc --verbose --features="rand,serde,recovery,endomorphism"; fi - if [ ${TRAVIS_RUST_VERSION} == "stable" ]; then cargo doc --verbose --features="rand,serde,recovery,endomorphism"; fi
- if [ ${TRAVIS_RUST_VERSION} == "nightly" ]; then cargo test --verbose --benches --features=unstable; fi - if [ ${TRAVIS_RUST_VERSION} == "nightly" ]; then cargo test --verbose --benches --features=unstable; fi
- if [ ${TRAVIS_RUST_VERSION} == "nightly" -a "$TRAVIS_OS_NAME" = "linux" ]; then - if [ ${TRAVIS_RUST_VERSION} == "nightly" -a "$TRAVIS_OS_NAME" = "linux" ]; then
cd no_std_test && cd no_std_test &&
cargo run --release | grep -q "Verified Successfully"; cargo run --release | grep -q "Verified Successfully";
fi fi
- if [ ${TRAVIS_RUST_VERSION} == "stable" -a "$TRAVIS_OS_NAME" = "linux" ]; then - #if [ ${TRAVIS_RUST_VERSION} == "stable" -a "$TRAVIS_OS_NAME" = "linux" ]; then
CARGO_TARGET_DIR=cargo_web cargo install --verbose --force cargo-web && #CARGO_TARGET_DIR=cargo_web cargo install --verbose --force cargo-web &&
cargo web build --verbose --target=asmjs-unknown-emscripten && #cargo web build --verbose --target=asmjs-unknown-emscripten &&
cargo web test --verbose --target=asmjs-unknown-emscripten; #cargo web test --verbose --target=asmjs-unknown-emscripten;
fi #fi

View File

@ -22,7 +22,7 @@ name = "secp256k1"
path = "src/lib.rs" path = "src/lib.rs"
[features] [features]
unstable = [] unstable = ["recovery", "rand-std"]
default = ["std"] default = ["std"]
std = ["secp256k1-sys/std"] std = ["secp256k1-sys/std"]
rand-std = ["rand/std"] rand-std = ["rand/std"]

View File

@ -14,13 +14,13 @@
//! # secp256k1 no-std test. //! # secp256k1 no-std test.
//! This binary is a short smallest rust code to produce a working binary *without libstd*. //! This binary is a short smallest rust code to produce a working binary *without libstd*.
//! This gives us 2 things: //! This gives us 2 things:
//! 1. Test that the parts of the code that should work in a no-std enviroment actually work. //! 1. Test that the parts of the code that should work in a no-std enviroment actually work.
//! 2. Test that we don't accidentally import libstd into `secp256k1`. //! 2. Test that we don't accidentally import libstd into `secp256k1`.
//! //!
//! The first is tested using the following command `cargo run --release | grep -q "Verified Successfully"`. //! The first is tested using the following command `cargo run --release | grep -q "Verified Successfully"`.
//! (Making sure that it successfully printed that. i.e. it didn't abort before that). //! (Making sure that it successfully printed that. i.e. it didn't abort before that).
//! //!
//! The second is tested by the fact that it compiles. if we accidentally link against libstd we should see the following error: //! The second is tested by the fact that it compiles. if we accidentally link against libstd we should see the following error:
//! `error[E0152]: duplicate lang item found`. //! `error[E0152]: duplicate lang item found`.
//! Example: //! Example:
@ -33,11 +33,11 @@
//! | //! |
//! = note: first defined in crate `panic_unwind` (which `std` depends on). //! = note: first defined in crate `panic_unwind` (which `std` depends on).
//! ``` //! ```
//! //!
//! Notes: //! Notes:
//! * Requires `panic=abort` and `--release` to not depend on libunwind(which is provided usually by libstd) https://github.com/rust-lang/rust/issues/47493 //! * Requires `panic=abort` and `--release` to not depend on libunwind(which is provided usually by libstd) https://github.com/rust-lang/rust/issues/47493
//! * Requires linking with `libc` for calling `printf`. //! * Requires linking with `libc` for calling `printf`.
//! //!
#![feature(lang_items)] #![feature(lang_items)]
#![feature(start)] #![feature(start)]
@ -52,10 +52,10 @@ use core::fmt::{self, write, Write};
use core::intrinsics; use core::intrinsics;
use core::panic::PanicInfo; use core::panic::PanicInfo;
use secp256k1::ecdh::SharedSecret;
use secp256k1::rand::{self, RngCore}; use secp256k1::rand::{self, RngCore};
use secp256k1::serde::Serialize; use secp256k1::serde::Serialize;
use secp256k1::*; use secp256k1::*;
use secp256k1::ecdh::SharedSecret;
use serde_cbor::de; use serde_cbor::de;
use serde_cbor::ser::SliceWrite; use serde_cbor::ser::SliceWrite;
@ -105,30 +105,17 @@ fn start(_argc: isize, _argv: *const *const u8) -> isize {
let _ = SharedSecret::new(&public_key, &secret_key); let _ = SharedSecret::new(&public_key, &secret_key);
let mut x_arr = [0u8; 32]; let mut x_arr = [0u8; 32];
let y_arr = unsafe { SharedSecret::new_with_hash_no_panic(&public_key, &secret_key, |x,y| { let y_arr = SharedSecret::new_with_hash(&public_key, &secret_key, |x,y| {
x_arr = x; x_arr = x;
y.into() y.into()
})}.unwrap(); });
assert_ne!(x_arr, [0u8; 32]); assert_ne!(x_arr, [0u8; 32]);
assert_ne!(&y_arr[..], &[0u8; 32][..]); assert_ne!(&y_arr[..], &[0u8; 32][..]);
unsafe { libc::printf("Verified Successfully!\n\0".as_ptr() as _) }; unsafe { libc::printf("Verified Successfully!\n\0".as_ptr() as _) };
0 0
} }
// These functions are used by the compiler, but not
// for a bare-bones hello world. These are normally
// provided by libstd.
#[lang = "eh_personality"]
#[no_mangle]
pub extern "C" fn rust_eh_personality() {}
// This function may be needed based on the compilation target.
#[lang = "eh_unwind_resume"]
#[no_mangle]
pub extern "C" fn rust_eh_unwind_resume() {}
const MAX_PRINT: usize = 511; const MAX_PRINT: usize = 511;
struct Print { struct Print {
loc: usize, loc: usize,

View File

@ -63,7 +63,6 @@ fn main() {
} else { } else {
base_config.define("ECMULT_WINDOW_SIZE", Some("15")); // This is the default in the configure file (`auto`) base_config.define("ECMULT_WINDOW_SIZE", Some("15")); // This is the default in the configure file (`auto`)
} }
#[cfg(not(feature = "dont_replace_c_symbols"))]
base_config.define("USE_EXTERNAL_DEFAULT_CALLBACKS", Some("1")); base_config.define("USE_EXTERNAL_DEFAULT_CALLBACKS", Some("1"));
#[cfg(feature = "endomorphism")] #[cfg(feature = "endomorphism")]
base_config.define("USE_ENDOMORPHISM", Some("1")); base_config.define("USE_ENDOMORPHISM", Some("1"));

View File

@ -63,7 +63,7 @@ pub type NonceFn = unsafe extern "C" fn(nonce32: *mut c_uchar,
algo16: *const c_uchar, algo16: *const c_uchar,
data: *mut c_void, data: *mut c_void,
attempt: c_uint, attempt: c_uint,
); ) -> c_int;
/// Hash function to use to post-process an ECDH point to get /// Hash function to use to post-process an ECDH point to get
/// a shared secret. /// a shared secret.
@ -295,7 +295,7 @@ extern "C" {
// Returns: a newly created context object. // Returns: a newly created context object.
// In: flags: which parts of the context to initialize. // In: flags: which parts of the context to initialize.
#[no_mangle] #[no_mangle]
#[cfg(all(feature = "std", not(feature = "external_symbols")))] #[cfg(all(feature = "std", not(feature = "external-symbols")))]
pub unsafe extern "C" fn rustsecp256k1_v0_1_1_context_create(flags: c_uint) -> *mut Context { pub unsafe extern "C" fn rustsecp256k1_v0_1_1_context_create(flags: c_uint) -> *mut Context {
use std::mem; use std::mem;
assert!(mem::align_of::<usize>() >= mem::align_of::<u8>()); assert!(mem::align_of::<usize>() >= mem::align_of::<u8>());
@ -312,7 +312,7 @@ pub unsafe extern "C" fn rustsecp256k1_v0_1_1_context_create(flags: c_uint) -> *
secp256k1_context_preallocated_create(ptr as *mut c_void, flags) secp256k1_context_preallocated_create(ptr as *mut c_void, flags)
} }
#[cfg(all(feature = "std", not(feature = "external_symbols")))] #[cfg(all(feature = "std", not(feature = "external-symbols")))]
pub unsafe fn secp256k1_context_create(flags: c_uint) -> *mut Context { pub unsafe fn secp256k1_context_create(flags: c_uint) -> *mut Context {
rustsecp256k1_v0_1_1_context_create(flags) rustsecp256k1_v0_1_1_context_create(flags)
} }
@ -324,7 +324,7 @@ pub unsafe fn secp256k1_context_create(flags: c_uint) -> *mut Context {
/// The pointer shouldn't be used after passing to this function, consider it as passing it to `free()`. /// The pointer shouldn't be used after passing to this function, consider it as passing it to `free()`.
/// ///
#[no_mangle] #[no_mangle]
#[cfg(all(feature = "std", not(feature = "external_symbols")))] #[cfg(all(feature = "std", not(feature = "external-symbols")))]
pub unsafe extern "C" fn rustsecp256k1_v0_1_1_context_destroy(ctx: *mut Context) { pub unsafe extern "C" fn rustsecp256k1_v0_1_1_context_destroy(ctx: *mut Context) {
secp256k1_context_preallocated_destroy(ctx); secp256k1_context_preallocated_destroy(ctx);
let ctx: *mut usize = ctx as *mut usize; let ctx: *mut usize = ctx as *mut usize;
@ -335,7 +335,7 @@ pub unsafe extern "C" fn rustsecp256k1_v0_1_1_context_destroy(ctx: *mut Context)
let _ = Box::from_raw(slice as *mut [usize]); let _ = Box::from_raw(slice as *mut [usize]);
} }
#[cfg(all(feature = "std", not(feature = "external_symbols")))] #[cfg(all(feature = "std", not(feature = "external-symbols")))]
pub unsafe fn secp256k1_context_destroy(ctx: *mut Context) { pub unsafe fn secp256k1_context_destroy(ctx: *mut Context) {
rustsecp256k1_v0_1_1_context_destroy(ctx) rustsecp256k1_v0_1_1_context_destroy(ctx)
} }
@ -360,7 +360,7 @@ pub unsafe fn secp256k1_context_destroy(ctx: *mut Context) {
/// See also secp256k1_default_error_callback_fn. /// See also secp256k1_default_error_callback_fn.
/// ///
#[no_mangle] #[no_mangle]
#[cfg(not(feature = "external_symbols"))] #[cfg(not(feature = "external-symbols"))]
pub unsafe extern "C" fn rustsecp256k1_v0_1_1_default_illegal_callback_fn(message: *const c_char, _data: *mut c_void) { pub unsafe extern "C" fn rustsecp256k1_v0_1_1_default_illegal_callback_fn(message: *const c_char, _data: *mut c_void) {
use core::str; use core::str;
let msg_slice = slice::from_raw_parts(message as *const u8, strlen(message)); let msg_slice = slice::from_raw_parts(message as *const u8, strlen(message));
@ -383,7 +383,7 @@ pub unsafe extern "C" fn rustsecp256k1_v0_1_1_default_illegal_callback_fn(messag
/// See also secp256k1_default_illegal_callback_fn. /// See also secp256k1_default_illegal_callback_fn.
/// ///
#[no_mangle] #[no_mangle]
#[cfg(not(feature = "external_symbols"))] #[cfg(not(feature = "external-symbols"))]
pub unsafe extern "C" fn rustsecp256k1_v0_1_1_default_error_callback_fn(message: *const c_char, _data: *mut c_void) { pub unsafe extern "C" fn rustsecp256k1_v0_1_1_default_error_callback_fn(message: *const c_char, _data: *mut c_void) {
use core::str; use core::str;
let msg_slice = slice::from_raw_parts(message as *const u8, strlen(message)); let msg_slice = slice::from_raw_parts(message as *const u8, strlen(message));
@ -491,13 +491,6 @@ mod fuzz_dummy {
1 1
} }
// TODO secp256k1_context_set_illegal_callback
// TODO secp256k1_context_set_error_callback
// (Actually, I don't really want these exposed; if either of these
// are ever triggered it indicates a bug in rust-secp256k1, since
// one goal is to use Rust's type system to eliminate all possible
// bad inputs.)
// Pubkeys // Pubkeys
/// Parse 33/65 byte pubkey into PublicKey, losing compressed information /// Parse 33/65 byte pubkey into PublicKey, losing compressed information
pub unsafe fn secp256k1_ec_pubkey_parse(cx: *const Context, pk: *mut PublicKey, pub unsafe fn secp256k1_ec_pubkey_parse(cx: *const Context, pk: *mut PublicKey,

View File

@ -34,6 +34,14 @@ pub const MAX_SIGNATURE_SIZE: usize = 72;
/// The maximum size of a compact signature /// The maximum size of a compact signature
pub const COMPACT_SIGNATURE_SIZE: usize = 64; pub const COMPACT_SIGNATURE_SIZE: usize = 64;
/// The Prime for the secp256k1 field element.
pub const FIELD_SIZE: [u8; 32] = [
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xfe, 0xff, 0xff, 0xfc, 0x2f
];
/// The order of the secp256k1 curve /// The order of the secp256k1 curve
pub const CURVE_ORDER: [u8; 32] = [ pub const CURVE_ORDER: [u8; 32] = [
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,

View File

@ -22,7 +22,6 @@ use core::ops::{FnMut, Deref};
use key::{SecretKey, PublicKey}; use key::{SecretKey, PublicKey};
use ffi::{self, CPtr}; use ffi::{self, CPtr};
use secp256k1_sys::types::{c_int, c_uchar, c_void}; use secp256k1_sys::types::{c_int, c_uchar, c_void};
use Error;
/// A tag used for recovering the public key from a compact signature /// A tag used for recovering the public key from a compact signature
#[derive(Copy, Clone)] #[derive(Copy, Clone)]
@ -89,39 +88,12 @@ impl Deref for SharedSecret {
} }
unsafe fn callback_logic<F>(output: *mut c_uchar, x: *const c_uchar, y: *const c_uchar, data: *mut c_void) -> c_int unsafe extern "C" fn c_callback(output: *mut c_uchar, x: *const c_uchar, y: *const c_uchar, _data: *mut c_void) -> c_int {
where F: FnMut([u8; 32], [u8; 32]) -> SharedSecret { ptr::copy_nonoverlapping(x, output, 32);
let callback: &mut F = &mut *(data as *mut F); ptr::copy_nonoverlapping(y, output.offset(32), 32);
1
let mut x_arr = [0; 32];
let mut y_arr = [0; 32];
ptr::copy_nonoverlapping(x, x_arr.as_mut_ptr(), 32);
ptr::copy_nonoverlapping(y, y_arr.as_mut_ptr(), 32);
let secret = callback(x_arr, y_arr);
ptr::copy_nonoverlapping(secret.as_ptr(), output as *mut u8, secret.len());
secret.len() as c_int
} }
#[cfg(feature = "std")]
unsafe extern "C" fn hash_callback_catch_unwind<F>(output: *mut c_uchar, x: *const c_uchar, y: *const c_uchar, data: *mut c_void) -> c_int
where F: FnMut([u8; 32], [u8; 32]) -> SharedSecret {
let res = ::std::panic::catch_unwind(||callback_logic::<F>(output, x, y, data));
if let Ok(len) = res {
len
} else {
-1
}
}
unsafe extern "C" fn hash_callback_unsafe<F>(output: *mut c_uchar, x: *const c_uchar, y: *const c_uchar, data: *mut c_void) -> c_int
where F: FnMut([u8; 32], [u8; 32]) -> SharedSecret {
callback_logic::<F>(output, x, y, data)
}
impl SharedSecret { impl SharedSecret {
/// Creates a new shared secret from a pubkey and secret key /// Creates a new shared secret from a pubkey and secret key
#[inline] #[inline]
@ -137,35 +109,17 @@ impl SharedSecret {
ptr::null_mut(), ptr::null_mut(),
) )
}; };
debug_assert_eq!(res, 1); // The default `secp256k1_ecdh_hash_function_default` should always return 1. // The default `secp256k1_ecdh_hash_function_default` should always return 1.
// and the scalar was verified to be valid(0 > scalar > group_order) via the type system
debug_assert_eq!(res, 1);
ss.set_len(32); // The default hash function is SHA256, which is 32 bytes long. ss.set_len(32); // The default hash function is SHA256, which is 32 bytes long.
ss ss
} }
fn new_with_callback_internal<F>(point: &PublicKey, scalar: &SecretKey, mut closure: F, callback: ffi::EcdhHashFn) -> Result<SharedSecret, Error>
where F: FnMut([u8; 32], [u8; 32]) -> SharedSecret {
let mut ss = SharedSecret::empty();
let res = unsafe {
ffi::secp256k1_ecdh(
ffi::secp256k1_context_no_precomp,
ss.get_data_mut_ptr(),
point.as_ptr(),
scalar.as_ptr(),
callback,
&mut closure as *mut F as *mut c_void,
)
};
if res == -1 {
return Err(Error::CallbackPanicked);
}
debug_assert!(res >= 16); // 128 bit is the minimum for a secure hash function and the minimum we let users.
ss.set_len(res as usize);
Ok(ss)
}
/// Creates a new shared secret from a pubkey and secret key with applied custom hash function /// Creates a new shared secret from a pubkey and secret key with applied custom hash function
/// The custom hash function must be in the form of `fn(x: [u8;32], y: [u8;32]) -> SharedSecret`
/// `SharedSecret` can be easily created via the `From` impl from arrays.
/// # Examples /// # Examples
/// ``` /// ```
/// # use secp256k1::ecdh::SharedSecret; /// # use secp256k1::ecdh::SharedSecret;
@ -182,43 +136,29 @@ impl SharedSecret {
/// }); /// });
/// ///
/// ``` /// ```
#[cfg(feature = "std")] pub fn new_with_hash<F>(point: &PublicKey, scalar: &SecretKey, mut hash_function: F) -> SharedSecret
pub fn new_with_hash<F>(point: &PublicKey, scalar: &SecretKey, hash_function: F) -> Result<SharedSecret, Error>
where F: FnMut([u8; 32], [u8; 32]) -> SharedSecret { where F: FnMut([u8; 32], [u8; 32]) -> SharedSecret {
Self::new_with_callback_internal(point, scalar, hash_function, hash_callback_catch_unwind::<F>) let mut xy = [0u8; 64];
}
/// Creates a new shared secret from a pubkey and secret key with applied custom hash function let res = unsafe {
/// Note that this function is the same as [`new_with_hash`] ffi::secp256k1_ecdh(
/// ffi::secp256k1_context_no_precomp,
/// # Safety xy.as_mut_ptr(),
/// The function doesn't wrap the callback with [`catch_unwind`] point.as_ptr(),
/// so if the callback panics it will panic through an FFI boundray which is [`Undefined Behavior`] scalar.as_ptr(),
/// If possible you should use [`new_with_hash`] which does wrap the callback with [`catch_unwind`] so is safe to use. c_callback,
/// ptr::null_mut(),
/// [`catch_unwind`]: https://doc.rust-lang.org/std/panic/fn.catch_unwind.html )
/// [`Undefined Behavior`]: https://doc.rust-lang.org/nomicon/ffi.html#ffi-and-panics };
/// [`new_with_hash`]: #method.new_with_hash // Our callback *always* returns 1.
/// # Examples // and the scalar was verified to be valid(0 > scalar > group_order) via the type system
/// ``` debug_assert_eq!(res, 1);
/// # use secp256k1::ecdh::SharedSecret;
/// # use secp256k1::{Secp256k1, PublicKey, SecretKey}; let mut x = [0u8; 32];
/// # fn sha2(_a: &[u8], _b: &[u8]) -> [u8; 32] {[0u8; 32]} let mut y = [0u8; 32];
/// # let secp = Secp256k1::signing_only(); x.copy_from_slice(&xy[..32]);
/// # let secret_key = SecretKey::from_slice(&[3u8; 32]).unwrap(); y.copy_from_slice(&xy[32..]);
/// # let secret_key2 = SecretKey::from_slice(&[7u8; 32]).unwrap(); hash_function(x, y)
/// # let public_key = PublicKey::from_secret_key(&secp, &secret_key2);
//
/// let secret = unsafe { SharedSecret::new_with_hash_no_panic(&public_key, &secret_key, |x,y| {
/// let hash: [u8; 32] = sha2(&x,&y);
/// hash.into()
/// })};
///
///
/// ```
pub unsafe fn new_with_hash_no_panic<F>(point: &PublicKey, scalar: &SecretKey, hash_function: F) -> Result<SharedSecret, Error>
where F: FnMut([u8; 32], [u8; 32]) -> SharedSecret {
Self::new_with_callback_internal(point, scalar, hash_function, hash_callback_unsafe::<F>)
} }
} }
@ -248,9 +188,9 @@ mod tests {
let (sk1, pk1) = s.generate_keypair(&mut thread_rng()); let (sk1, pk1) = s.generate_keypair(&mut thread_rng());
let (sk2, pk2) = s.generate_keypair(&mut thread_rng()); let (sk2, pk2) = s.generate_keypair(&mut thread_rng());
let sec1 = SharedSecret::new_with_hash(&pk1, &sk2, |x,_| x.into()).unwrap(); let sec1 = SharedSecret::new_with_hash(&pk1, &sk2, |x,_| x.into());
let sec2 = SharedSecret::new_with_hash(&pk2, &sk1, |x,_| x.into()).unwrap(); let sec2 = SharedSecret::new_with_hash(&pk2, &sk1, |x,_| x.into());
let sec_odd = SharedSecret::new_with_hash(&pk1, &sk1, |x,_| x.into()).unwrap(); let sec_odd = SharedSecret::new_with_hash(&pk1, &sk1, |x,_| x.into());
assert_eq!(sec1, sec2); assert_eq!(sec1, sec2);
assert_ne!(sec_odd, sec2); assert_ne!(sec_odd, sec2);
} }
@ -262,32 +202,29 @@ mod tests {
let expect_result: [u8; 64] = [123; 64]; let expect_result: [u8; 64] = [123; 64];
let mut x_out = [0u8; 32]; let mut x_out = [0u8; 32];
let mut y_out = [0u8; 32]; let mut y_out = [0u8; 32];
let result = SharedSecret::new_with_hash(&pk1, &sk1, | x, y | { let result = SharedSecret::new_with_hash(&pk1, &sk1, |x, y| {
x_out = x; x_out = x;
y_out = y; y_out = y;
expect_result.into() expect_result.into()
}).unwrap(); });
let result_unsafe = unsafe {SharedSecret::new_with_hash_no_panic(&pk1, &sk1, | x, y | {
x_out = x;
y_out = y;
expect_result.into()
}).unwrap()};
assert_eq!(&expect_result[..], &result[..]); assert_eq!(&expect_result[..], &result[..]);
assert_eq!(result, result_unsafe);
assert_ne!(x_out, [0u8; 32]); assert_ne!(x_out, [0u8; 32]);
assert_ne!(y_out, [0u8; 32]); assert_ne!(y_out, [0u8; 32]);
} }
#[test] #[test]
fn ecdh_with_hash_callback_panic() { fn test_c_callback() {
let s = Secp256k1::signing_only(); let x = [5u8; 32];
let (sk1, pk1) = s.generate_keypair(&mut thread_rng()); let y = [7u8; 32];
let mut res = [0u8; 48]; let mut output = [0u8; 64];
let result = SharedSecret::new_with_hash(&pk1, &sk1, | x, _ | { let res = unsafe { super::c_callback(output.as_mut_ptr(), x.as_ptr(), y.as_ptr(), ::ptr::null_mut()) };
res.copy_from_slice(&x); // res.len() != x.len(). this will panic. assert_eq!(res, 1);
res.into() let mut new_x = [0u8; 32];
}); let mut new_y = [0u8; 32];
assert_eq!(result, Err(Error::CallbackPanicked)); new_x.copy_from_slice(&output[..32]);
new_y.copy_from_slice(&output[32..]);
assert_eq!(x, new_x);
assert_eq!(y, new_y);
} }
} }
@ -304,7 +241,6 @@ mod benches {
let s = Secp256k1::signing_only(); let s = Secp256k1::signing_only();
let (sk, pk) = s.generate_keypair(&mut thread_rng()); let (sk, pk) = s.generate_keypair(&mut thread_rng());
let s = Secp256k1::new();
bh.iter( || { bh.iter( || {
let res = SharedSecret::new(&pk, &sk); let res = SharedSecret::new(&pk, &sk);
black_box(res); black_box(res);

View File

@ -496,8 +496,6 @@ pub enum Error {
InvalidTweak, InvalidTweak,
/// Didn't pass enough memory to context creation with preallocated memory /// Didn't pass enough memory to context creation with preallocated memory
NotEnoughMemory, NotEnoughMemory,
/// The callback has panicked.
CallbackPanicked,
} }
impl Error { impl Error {
@ -511,7 +509,6 @@ impl Error {
Error::InvalidRecoveryId => "secp: bad recovery id", Error::InvalidRecoveryId => "secp: bad recovery id",
Error::InvalidTweak => "secp: bad tweak", Error::InvalidTweak => "secp: bad tweak",
Error::NotEnoughMemory => "secp: not enough memory allocated", Error::NotEnoughMemory => "secp: not enough memory allocated",
Error::CallbackPanicked => "secp: a callback passed has panicked",
} }
} }
} }
@ -734,7 +731,6 @@ mod tests {
#[test] #[test]
#[cfg(not(feature = "dont_replace_c_symbols"))]
fn test_manual_create_destroy() { fn test_manual_create_destroy() {
let ctx_full = unsafe { ffi::secp256k1_context_create(AllPreallocated::FLAGS) }; let ctx_full = unsafe { ffi::secp256k1_context_create(AllPreallocated::FLAGS) };
let ctx_sign = unsafe { ffi::secp256k1_context_create(SignOnlyPreallocated::FLAGS) }; let ctx_sign = unsafe { ffi::secp256k1_context_create(SignOnlyPreallocated::FLAGS) };

View File

@ -363,6 +363,10 @@ mod tests {
#[cfg(all(test, feature = "unstable"))] #[cfg(all(test, feature = "unstable"))]
mod benches { mod benches {
use rand::{thread_rng, RngCore};
use test::{Bencher, black_box};
use super::{Message, Secp256k1};
#[bench] #[bench]
pub fn bench_recover(bh: &mut Bencher) { pub fn bench_recover(bh: &mut Bencher) {
let s = Secp256k1::new(); let s = Secp256k1::new();