From 3164ac0214c991b993f7b8e0b1410024eb7a7986 Mon Sep 17 00:00:00 2001 From: Valerii Hiora Date: Tue, 14 Oct 2014 22:39:07 +0300 Subject: [PATCH] Cleaned up BigNum constructors --- src/bn/mod.rs | 36 +++++++++++------------------------- 1 file changed, 11 insertions(+), 25 deletions(-) diff --git a/src/bn/mod.rs b/src/bn/mod.rs index 510c9269..6311e89c 100644 --- a/src/bn/mod.rs +++ b/src/bn/mod.rs @@ -79,41 +79,27 @@ macro_rules! with_bn_in_ctx( ) impl BigNum { - // FIXME: squash 3 constructors into one pub fn new() -> Result { unsafe { ffi::init(); - let v = ffi::BN_new(); - if v.is_null() { - Err(SslError::get()) - } else { - Ok(BigNum(v)) - } + + let v = try_ssl_null!(ffi::BN_new()); + Ok(BigNum(v)) } } pub fn new_from(n: u64) -> Result { - unsafe { - ffi::init(); - let bn = ffi::BN_new(); - if bn.is_null() || ffi::BN_set_word(bn, n as c_ulong) == 0 { - Err(SslError::get()) - } else { - Ok(BigNum(bn)) - } - } + BigNum::new().and_then(|v| unsafe { + try_ssl!(ffi::BN_set_word(v.raw(), n as c_ulong)); + Ok(v) + }) } pub fn new_from_slice(n: &[u8]) -> Result { - unsafe { - ffi::init(); - let bn = ffi::BN_new(); - if bn.is_null() || ffi::BN_bin2bn(n.as_ptr(), n.len() as c_int, bn).is_null() { - Err(SslError::get()) - } else { - Ok(BigNum(bn)) - } - } + BigNum::new().and_then(|v| unsafe { + try_ssl_null!(ffi::BN_bin2bn(n.as_ptr(), n.len() as c_int, v.raw())); + Ok(v) + }) } pub fn checked_sqr(&self) -> Result {