Commit 36977c2a authored by Steven Fackler's avatar Steven Fackler
Browse files

Merge pull request #56 from vhbit/single-ffi

Single `ffi` module
parents 359043a7 02637ec7
Loading
Loading
Loading
Loading
+0 −23
Original line number Diff line number Diff line
pub mod ffi {
    #![allow(dead_code)]
    #![allow(non_camel_case_types)]
    use libc::{c_int, c_long, c_void};

    pub type ASN1_INTEGER = c_void;
    pub type ASN1_TIME = c_void;
    pub type ASN1_STRING = c_void;

    pub static MBSTRING_FLAG: c_int = 0x1000;
    pub static MBSTRING_UTF8: c_int = MBSTRING_FLAG;
    pub static MBSTRING_ASC:  c_int = MBSTRING_FLAG | 1;
    pub static MBSTRING_BMP:  c_int = MBSTRING_FLAG | 2;
    pub static MBSTRING_UNIV: c_int = MBSTRING_FLAG | 4;

    pub static V_ASN1_UTCTIME:         c_int = 23;
    pub static V_ASN1_GENERALIZEDTIME: c_int = 24;

    extern "C" {
        pub fn ASN1_STRING_type_new(ty: c_int) -> *mut ASN1_STRING;
        pub fn ASN1_INTEGER_set(dest: *mut ASN1_INTEGER, value: c_long) -> c_int;
    }
}
+1 −17
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ use std::io::{IoResult, IoError, OtherIoError};
use std::io::{Reader, Writer};
use std::ptr;

use ffi;
use ssl::error::{SslError};

pub struct MemBio {
@@ -84,20 +85,3 @@ impl Writer for MemBio {
        }
    }
}

pub mod ffi {
    #![allow(non_camel_case_types)]

    use libc::{c_int, c_void};

    pub type BIO = c_void;
    pub type BIO_METHOD = c_void;

    extern "C" {
        pub fn BIO_s_mem() -> *const BIO_METHOD;
        pub fn BIO_new(type_: *const BIO_METHOD) -> *mut BIO;
        pub fn BIO_free_all(a: *mut BIO);
        pub fn BIO_read(b: *mut BIO, buf: *mut c_void, len: c_int) -> c_int;
        pub fn BIO_write(b: *mut BIO, buf: *const c_void, len: c_int) -> c_int;
    }
}
+56 −135
Original line number Diff line number Diff line

use libc::{c_void, c_int, c_ulong, c_char};
use libc::{c_int, c_ulong};
use std::{fmt, ptr};
use std::c_str::CString;
use std::num::{One, Zero};

use ffi;
use ssl::error::SslError;

#[allow(dead_code)]
#[repr(C)]
struct BIGNUM {
    d: *mut c_void,
    top: c_int,
    dmax: c_int,
    neg: c_int,
    flags: c_int,
}

#[allow(non_camel_case_types)]
type BN_CTX = *mut c_void;

#[link(name = "crypto")]
extern {
    fn BN_new() -> *mut BIGNUM;
    fn BN_clear_free(bn: *mut BIGNUM);

    fn BN_CTX_new() -> *mut BN_CTX;
    fn BN_CTX_free(ctx: *mut BN_CTX);

    fn BN_set_word(bn: *mut BIGNUM, n: c_ulong) -> c_int;
    fn BN_set_negative(bn: *mut BIGNUM, n: c_int);
    fn BN_num_bits(bn: *mut BIGNUM) -> c_int;

    /* Arithmetic operations on BIGNUMs */
    fn BN_add(r: *mut BIGNUM, a: *mut BIGNUM, b: *mut BIGNUM) -> c_int;
    fn BN_sub(r: *mut BIGNUM, a: *mut BIGNUM, b: *mut BIGNUM) -> c_int;
    fn BN_mul(r: *mut BIGNUM, a: *mut BIGNUM, b: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_sqr(r: *mut BIGNUM, a: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_div(dv: *mut BIGNUM, rem: *mut BIGNUM, a: *mut BIGNUM, b: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_nnmod(rem: *mut BIGNUM, a: *mut BIGNUM, m: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_mod_add(r: *mut BIGNUM, a: *mut BIGNUM, b: *mut BIGNUM, m: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_mod_sub(r: *mut BIGNUM, a: *mut BIGNUM, b: *mut BIGNUM, m: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_mod_mul(r: *mut BIGNUM, a: *mut BIGNUM, b: *mut BIGNUM, m: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_mod_sqr(r: *mut BIGNUM, a: *mut BIGNUM, m: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_exp(r: *mut BIGNUM, a: *mut BIGNUM, p: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_mod_exp(r: *mut BIGNUM, a: *mut BIGNUM, p: *mut BIGNUM, m: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;
    fn BN_mod_inverse(r: *mut BIGNUM, a: *mut BIGNUM, n: *mut BIGNUM, ctx: *mut BN_CTX) -> *const BIGNUM;
    fn BN_gcd(r: *mut BIGNUM, a: *mut BIGNUM, b: *mut BIGNUM, ctx: *mut BN_CTX) -> c_int;

    /* Bit operations on BIGNUMs */
    fn BN_set_bit(a: *mut BIGNUM, n: c_int) -> c_int;
    fn BN_clear_bit(a: *mut BIGNUM, n: c_int) -> c_int;
    fn BN_is_bit_set(a: *mut BIGNUM, n: c_int) -> c_int;
    fn BN_mask_bits(a: *mut BIGNUM, n: c_int) -> c_int;
    fn BN_lshift(r: *mut BIGNUM, a: *mut BIGNUM, n: c_int) -> c_int;
    fn BN_lshift1(r: *mut BIGNUM, a: *mut BIGNUM) -> c_int;
    fn BN_rshift(r: *mut BIGNUM, a: *mut BIGNUM, n: c_int) -> c_int;
    fn BN_rshift1(r: *mut BIGNUM, a: *mut BIGNUM) -> c_int;

    /* Comparisons on BIGNUMs */
    fn BN_cmp(a: *mut BIGNUM, b: *mut BIGNUM) -> c_int;
    fn BN_ucmp(a: *mut BIGNUM, b: *mut BIGNUM) -> c_int;
    fn BN_is_zero(a: *mut BIGNUM) -> c_int;

    /* Prime handling */
    fn BN_generate_prime_ex(r: *mut BIGNUM, bits: c_int, safe: c_int, add: *mut BIGNUM, rem: *mut BIGNUM, cb: *const c_void) -> c_int;
    fn BN_is_prime_ex(p: *mut BIGNUM, checks: c_int, ctx: *mut BN_CTX, cb: *const c_void) -> c_int;
    fn BN_is_prime_fasttest_ex(p: *mut BIGNUM, checks: c_int, ctx: *mut BN_CTX, do_trial_division: c_int, cb: *const c_void) -> c_int;

    /* Random number handling */
    fn BN_rand(r: *mut BIGNUM, bits: c_int, top: c_int, bottom: c_int) -> c_int;
    fn BN_pseudo_rand(r: *mut BIGNUM, bits: c_int, top: c_int, bottom: c_int) -> c_int;
    fn BN_rand_range(r: *mut BIGNUM, range: *mut BIGNUM) -> c_int;
    fn BN_pseudo_rand_range(r: *mut BIGNUM, range: *mut BIGNUM) -> c_int;

    /* Conversion from/to binary representation */
    fn BN_bn2bin(a: *mut BIGNUM, to: *mut u8) -> c_int;
    fn BN_bin2bn(s: *const u8, size: c_int, ret: *mut BIGNUM) -> *mut BIGNUM;

    /* Conversion from/to string representation */
    fn BN_bn2dec(a: *mut BIGNUM) -> *const c_char;
    fn CRYPTO_free(buf: *const c_char);
}

pub struct BigNum(*mut BIGNUM);
pub struct BigNum(*mut ffi::BIGNUM);

#[repr(C)]
pub enum RNGProperty {
@@ -93,12 +17,12 @@ pub enum RNGProperty {

macro_rules! with_ctx(
    ($name:ident, $action:block) => ({
        let $name = BN_CTX_new();
        let $name = ffi::BN_CTX_new();
        if ($name).is_null() {
            Err(SslError::get())
        } else {
            let r = $action;
            BN_CTX_free($name);
            ffi::BN_CTX_free($name);
            r
        }
    });
@@ -125,7 +49,7 @@ macro_rules! with_bn_in_ctx(
        let tmp = BigNum::new();
        match tmp {
            Ok($name) => {
                let $ctx_name = BN_CTX_new();
                let $ctx_name = ffi::BN_CTX_new();
                if ($ctx_name).is_null() {
                    Err(SslError::get())
                } else {
@@ -135,7 +59,7 @@ macro_rules! with_bn_in_ctx(
                        } else {
                            Err(SslError::get())
                        };
                    BN_CTX_free($ctx_name);
                    ffi::BN_CTX_free($ctx_name);
                    r
                }
            },
@@ -147,7 +71,7 @@ macro_rules! with_bn_in_ctx(
impl BigNum {
    pub fn new() -> Result<BigNum, SslError> {
        unsafe {
            let v = BN_new();
            let v = ffi::BN_new();
            if v.is_null() {
                Err(SslError::get())
            } else {
@@ -158,8 +82,8 @@ impl BigNum {

    pub fn new_from(n: u64) -> Result<BigNum, SslError> {
        unsafe {
            let bn = BN_new();
            if bn.is_null() || BN_set_word(bn, n as c_ulong) == 0 {
            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))
@@ -169,8 +93,8 @@ impl BigNum {

    pub fn new_from_slice(n: &[u8]) -> Result<BigNum, SslError> {
        unsafe {
            let bn = BN_new();
            if bn.is_null() || BN_bin2bn(n.as_ptr(), n.len() as c_int, bn).is_null() {
            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))
@@ -180,61 +104,61 @@ impl BigNum {

    pub fn checked_sqr(&self) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_sqr(r.raw(), self.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_sqr(r.raw(), self.raw(), ctx) == 1 })
        }
    }

    pub fn checked_nnmod(&self, n: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_nnmod(r.raw(), self.raw(), n.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_nnmod(r.raw(), self.raw(), n.raw(), ctx) == 1 })
        }
    }

    pub fn checked_mod_add(&self, a: &BigNum, n: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_mod_add(r.raw(), self.raw(), a.raw(), n.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_mod_add(r.raw(), self.raw(), a.raw(), n.raw(), ctx) == 1 })
        }
    }

    pub fn checked_mod_sub(&self, a: &BigNum, n: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_mod_sub(r.raw(), self.raw(), a.raw(), n.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_mod_sub(r.raw(), self.raw(), a.raw(), n.raw(), ctx) == 1 })
        }
    }

    pub fn checked_mod_mul(&self, a: &BigNum, n: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_mod_mul(r.raw(), self.raw(), a.raw(), n.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_mod_mul(r.raw(), self.raw(), a.raw(), n.raw(), ctx) == 1 })
        }
    }

    pub fn checked_mod_sqr(&self, n: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_mod_sqr(r.raw(), self.raw(), n.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_mod_sqr(r.raw(), self.raw(), n.raw(), ctx) == 1 })
        }
    }

    pub fn checked_exp(&self, p: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_exp(r.raw(), self.raw(), p.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_exp(r.raw(), self.raw(), p.raw(), ctx) == 1 })
        }
    }

    pub fn checked_mod_exp(&self, p: &BigNum, n: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_mod_exp(r.raw(), self.raw(), p.raw(), n.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_mod_exp(r.raw(), self.raw(), p.raw(), n.raw(), ctx) == 1 })
        }
    }

    pub fn checked_mod_inv(&self, n: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { !BN_mod_inverse(r.raw(), self.raw(), n.raw(), ctx).is_null() })
            with_bn_in_ctx!(r, ctx, { !ffi::BN_mod_inverse(r.raw(), self.raw(), n.raw(), ctx).is_null() })
        }
    }

    pub fn checked_gcd(&self, a: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_gcd(r.raw(), self.raw(), a.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_gcd(r.raw(), self.raw(), a.raw(), ctx) == 1 })
        }
    }

@@ -244,7 +168,7 @@ impl BigNum {
                let add_arg = add.map(|a| a.raw()).unwrap_or(ptr::mut_null());
                let rem_arg = rem.map(|r| r.raw()).unwrap_or(ptr::mut_null());

                BN_generate_prime_ex(r.raw(), bits as c_int, safe as c_int, add_arg, rem_arg, ptr::null()) == 1
                ffi::BN_generate_prime_ex(r.raw(), bits as c_int, safe as c_int, add_arg, rem_arg, ptr::null()) == 1
            })
        }
    }
@@ -252,7 +176,7 @@ impl BigNum {
    pub fn is_prime(&self, checks: i32) -> Result<bool, SslError> {
        unsafe {
            with_ctx!(ctx, {
                Ok(BN_is_prime_ex(self.raw(), checks as c_int, ctx, ptr::null()) == 1)
                Ok(ffi::BN_is_prime_ex(self.raw(), checks as c_int, ctx, ptr::null()) == 1)
            })
        }
    }
@@ -260,38 +184,38 @@ impl BigNum {
    pub fn is_prime_fast(&self, checks: i32, do_trial_division: bool) -> Result<bool, SslError> {
        unsafe {
            with_ctx!(ctx, {
                Ok(BN_is_prime_fasttest_ex(self.raw(), checks as c_int, ctx, do_trial_division as c_int, ptr::null()) == 1)
                Ok(ffi::BN_is_prime_fasttest_ex(self.raw(), checks as c_int, ctx, do_trial_division as c_int, ptr::null()) == 1)
            })
        }
    }

    pub fn checked_new_random(bits: i32, prop: RNGProperty, odd: bool) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_rand(r.raw(), bits as c_int, prop as c_int, odd as c_int) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_rand(r.raw(), bits as c_int, prop as c_int, odd as c_int) == 1 })
        }
    }

    pub fn checked_new_pseudo_random(bits: i32, prop: RNGProperty, odd: bool) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_pseudo_rand(r.raw(), bits as c_int, prop as c_int, odd as c_int) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_pseudo_rand(r.raw(), bits as c_int, prop as c_int, odd as c_int) == 1 })
        }
    }

    pub fn checked_rand_in_range(&self) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_rand_range(r.raw(), self.raw()) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_rand_range(r.raw(), self.raw()) == 1 })
        }
    }

    pub fn checked_pseudo_rand_in_range(&self) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_pseudo_rand_range(r.raw(), self.raw()) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_pseudo_rand_range(r.raw(), self.raw()) == 1 })
        }
    }

    pub fn set_bit(&mut self, n: i32) -> Result<(), SslError> {
        unsafe {
            if BN_set_bit(self.raw(), n as c_int) == 1 {
            if ffi::BN_set_bit(self.raw(), n as c_int) == 1 {
                Ok(())
            } else {
                Err(SslError::get())
@@ -301,7 +225,7 @@ impl BigNum {

    pub fn clear_bit(&mut self, n: i32) -> Result<(), SslError> {
        unsafe {
            if BN_clear_bit(self.raw(), n as c_int) == 1 {
            if ffi::BN_clear_bit(self.raw(), n as c_int) == 1 {
                Ok(())
            } else {
                Err(SslError::get())
@@ -311,13 +235,13 @@ impl BigNum {

    pub fn is_bit_set(&self, n: i32) -> bool {
        unsafe {
            BN_is_bit_set(self.raw(), n as c_int) == 1
            ffi::BN_is_bit_set(self.raw(), n as c_int) == 1
        }
    }

    pub fn mask_bits(&mut self, n: i32) -> Result<(), SslError> {
        unsafe {
            if BN_mask_bits(self.raw(), n as c_int) == 1 {
            if ffi::BN_mask_bits(self.raw(), n as c_int) == 1 {
                Ok(())
            } else {
                Err(SslError::get())
@@ -327,67 +251,67 @@ impl BigNum {

    pub fn checked_shl1(&self) -> Result<BigNum, SslError> {
        unsafe {
            with_bn!(r, { BN_lshift1(r.raw(), self.raw()) == 1 })
            with_bn!(r, { ffi::BN_lshift1(r.raw(), self.raw()) == 1 })
        }
    }

    pub fn checked_shr1(&self) -> Result<BigNum, SslError> {
        unsafe {
            with_bn!(r, { BN_rshift1(r.raw(), self.raw()) == 1 })
            with_bn!(r, { ffi::BN_rshift1(r.raw(), self.raw()) == 1 })
        }
    }

    pub fn checked_add(&self, a: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn!(r, { BN_add(r.raw(), self.raw(), a.raw()) == 1 })
            with_bn!(r, { ffi::BN_add(r.raw(), self.raw(), a.raw()) == 1 })
        }
    }

    pub fn checked_sub(&self, a: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn!(r, { BN_sub(r.raw(), self.raw(), a.raw()) == 1 })
            with_bn!(r, { ffi::BN_sub(r.raw(), self.raw(), a.raw()) == 1 })
        }
    }

    pub fn checked_mul(&self, a: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_mul(r.raw(), self.raw(), a.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_mul(r.raw(), self.raw(), a.raw(), ctx) == 1 })
        }
    }

    pub fn checked_div(&self, a: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_div(r.raw(), ptr::mut_null(), self.raw(), a.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_div(r.raw(), ptr::mut_null(), self.raw(), a.raw(), ctx) == 1 })
        }
    }

    pub fn checked_mod(&self, a: &BigNum) -> Result<BigNum, SslError> {
        unsafe {
            with_bn_in_ctx!(r, ctx, { BN_div(ptr::mut_null(), r.raw(), self.raw(), a.raw(), ctx) == 1 })
            with_bn_in_ctx!(r, ctx, { ffi::BN_div(ptr::mut_null(), r.raw(), self.raw(), a.raw(), ctx) == 1 })
        }
    }

    pub fn checked_shl(&self, a: &i32) -> Result<BigNum, SslError> {
        unsafe {
            with_bn!(r, { BN_lshift(r.raw(), self.raw(), *a as c_int) == 1 })
            with_bn!(r, { ffi::BN_lshift(r.raw(), self.raw(), *a as c_int) == 1 })
        }
    }

    pub fn checked_shr(&self, a: &i32) -> Result<BigNum, SslError> {
        unsafe {
            with_bn!(r, { BN_rshift(r.raw(), self.raw(), *a as c_int) == 1 })
            with_bn!(r, { ffi::BN_rshift(r.raw(), self.raw(), *a as c_int) == 1 })
        }
    }

    pub fn negate(&mut self) {
        unsafe {
            BN_set_negative(self.raw(), !self.is_negative() as c_int)
            ffi::BN_set_negative(self.raw(), !self.is_negative() as c_int)
        }
    }

    pub fn abs_cmp(&self, oth: BigNum) -> Ordering {
        unsafe {
            let res = BN_ucmp(self.raw(), oth.raw()) as i32;
            let res = ffi::BN_ucmp(self.raw(), oth.raw()) as i32;
            if res < 0 {
                Less
            } else if res > 0 {
@@ -406,7 +330,7 @@ impl BigNum {

    pub fn num_bits(&self) -> i32 {
        unsafe {
            BN_num_bits(self.raw()) as i32
            ffi::BN_num_bits(self.raw()) as i32
        }
    }

@@ -414,7 +338,7 @@ impl BigNum {
        (self.num_bits() + 7) / 8
    }

    unsafe fn raw(&self) -> *mut BIGNUM {
    unsafe fn raw(&self) -> *mut ffi::BIGNUM {
        let BigNum(n) = *self;
        n
    }
@@ -423,7 +347,7 @@ impl BigNum {
        let size = self.num_bytes() as uint;
        let mut v = Vec::with_capacity(size);
        unsafe {
            BN_bn2bin(self.raw(), v.as_mut_ptr());
            ffi::BN_bn2bin(self.raw(), v.as_mut_ptr());
            v.set_len(size);
        }
        v
@@ -431,11 +355,11 @@ impl BigNum {

    pub fn to_dec_str(&self) -> String {
        unsafe {
            let buf = BN_bn2dec(self.raw());
            let buf = ffi::BN_bn2dec(self.raw());
            assert!(!buf.is_null());
            let c_str = CString::new(buf, false);
            let str = c_str.as_str().unwrap().to_string();
            CRYPTO_free(buf);
            ffi::CRYPTO_free(buf);
            str
        }
    }
@@ -459,7 +383,7 @@ impl Zero for BigNum {
    }
    fn is_zero(&self) -> bool {
        unsafe {
            BN_is_zero(self.raw()) == 1
            ffi::BN_is_zero(self.raw()) == 1
        }
    }
}
@@ -468,7 +392,7 @@ impl Eq for BigNum { }
impl PartialEq for BigNum {
    fn eq(&self, oth: &BigNum) -> bool {
        unsafe {
            BN_cmp(self.raw(), oth.raw()) == 0
            ffi::BN_cmp(self.raw(), oth.raw()) == 0
        }
    }
}
@@ -482,7 +406,7 @@ impl Ord for BigNum {
impl PartialOrd for BigNum {
    fn partial_cmp(&self, oth: &BigNum) -> Option<Ordering> {
        unsafe {
            let v = BN_cmp(self.raw(), oth.raw());
            let v = ffi::BN_cmp(self.raw(), oth.raw());
            let ret =
                if v == 0 {
                    Equal
@@ -500,18 +424,15 @@ impl Drop for BigNum {
    fn drop(&mut self) {
        unsafe {
            if !self.raw().is_null() {
                BN_clear_free(self.raw());
                ffi::BN_clear_free(self.raw());
            }
        }
    }
}

pub mod unchecked {
    use super::{BIGNUM, BigNum};

    extern {
        fn BN_dup(n: *mut BIGNUM) -> *mut BIGNUM;
    }
    use ffi;
    use super::{BigNum};

    impl Add<BigNum, BigNum> for BigNum {
        fn add(&self, oth: &BigNum) -> BigNum {
@@ -558,7 +479,7 @@ pub mod unchecked {
    impl Clone for BigNum {
        fn clone(&self) -> BigNum {
            unsafe {
                let r = BN_dup(self.raw());
                let r = ffi::BN_dup(self.raw());
                if r.is_null() {
                    fail!("Unexpected null pointer from BN_dup(..)")
                } else {
+17 −54
Original line number Diff line number Diff line
use libc;
use libc::c_uint;
use std::ptr;

use ffi;

pub enum HashType {
    MD5,
    SHA1,
@@ -12,71 +13,33 @@ pub enum HashType {
    RIPEMD160
}

#[allow(dead_code)]
#[allow(non_camel_case_types)]
#[repr(C)]
pub struct EVP_MD_CTX {
    digest: *mut EVP_MD,
    engine: *mut libc::c_void,
    flags: libc::c_ulong,
    md_data: *mut libc::c_void,
    pctx: *mut EVP_PKEY_CTX,
    update: *mut libc::c_void
}

#[allow(non_camel_case_types)]
#[repr(C)]
pub struct EVP_MD;

#[allow(non_camel_case_types)]
#[repr(C)]
pub struct EVP_PKEY_CTX;

#[link(name = "crypto")]
extern {
    fn EVP_MD_CTX_create() -> *mut EVP_MD_CTX;
    fn EVP_MD_CTX_destroy(ctx: *mut EVP_MD_CTX);

    fn EVP_md5() -> *const EVP_MD;
    fn EVP_sha1() -> *const EVP_MD;
    fn EVP_sha224() -> *const EVP_MD;
    fn EVP_sha256() -> *const EVP_MD;
    fn EVP_sha384() -> *const EVP_MD;
    fn EVP_sha512() -> *const EVP_MD;
    fn EVP_ripemd160() -> *const EVP_MD;

    fn EVP_DigestInit(ctx: *mut EVP_MD_CTX, typ: *const EVP_MD);
    fn EVP_DigestUpdate(ctx: *mut EVP_MD_CTX, data: *const u8, n: c_uint);
    fn EVP_DigestFinal(ctx: *mut EVP_MD_CTX, res: *mut u8, n: *mut u32);
}

pub fn evpmd(t: HashType) -> (*const EVP_MD, uint) {
pub fn evpmd(t: HashType) -> (*const ffi::EVP_MD, uint) {
    unsafe {
        match t {
            MD5 => (EVP_md5(), 16u),
            SHA1 => (EVP_sha1(), 20u),
            SHA224 => (EVP_sha224(), 28u),
            SHA256 => (EVP_sha256(), 32u),
            SHA384 => (EVP_sha384(), 48u),
            SHA512 => (EVP_sha512(), 64u),
            RIPEMD160 => (EVP_ripemd160(), 20u),
            MD5 => (ffi::EVP_md5(), 16u),
            SHA1 => (ffi::EVP_sha1(), 20u),
            SHA224 => (ffi::EVP_sha224(), 28u),
            SHA256 => (ffi::EVP_sha256(), 32u),
            SHA384 => (ffi::EVP_sha384(), 48u),
            SHA512 => (ffi::EVP_sha512(), 64u),
            RIPEMD160 => (ffi::EVP_ripemd160(), 20u),
        }
    }
}

#[allow(dead_code)]
pub struct Hasher {
    evp: *const EVP_MD,
    ctx: *mut EVP_MD_CTX,
    evp: *const ffi::EVP_MD,
    ctx: *mut ffi::EVP_MD_CTX,
    len: uint,
}

impl Hasher {
    pub fn new(ht: HashType) -> Hasher {
        let ctx = unsafe { EVP_MD_CTX_create() };
        let ctx = unsafe { ffi::EVP_MD_CTX_create() };
        let (evp, mdlen) = evpmd(ht);
        unsafe {
            EVP_DigestInit(ctx, evp);
            ffi::EVP_DigestInit(ctx, evp);
        }

        Hasher { evp: evp, ctx: ctx, len: mdlen }
@@ -85,7 +48,7 @@ impl Hasher {
    /// Update this hasher with more input bytes
    pub fn update(&self, data: &[u8]) {
        unsafe {
            EVP_DigestUpdate(self.ctx, data.as_ptr(), data.len() as c_uint)
            ffi::EVP_DigestUpdate(self.ctx, data.as_ptr(), data.len() as c_uint)
        }
    }

@@ -96,7 +59,7 @@ impl Hasher {
    pub fn final(&self) -> Vec<u8> {
        unsafe {
            let mut res = Vec::from_elem(self.len, 0u8);
            EVP_DigestFinal(self.ctx, res.as_mut_ptr(), ptr::null_mut());
            ffi::EVP_DigestFinal(self.ctx, res.as_mut_ptr(), ptr::null_mut());
            res
        }
    }
@@ -105,7 +68,7 @@ impl Hasher {
impl Drop for Hasher {
    fn drop(&mut self) {
        unsafe {
            EVP_MD_CTX_destroy(self.ctx);
            ffi::EVP_MD_CTX_destroy(self.ctx);
        }
    }
}
+12 −34
Original line number Diff line number Diff line
@@ -14,35 +14,13 @@
 * limitations under the License.
 */

use libc::{c_uchar, c_int, c_uint};
use crypto::hash;

#[allow(dead_code)]
#[allow(non_camel_case_types)]
#[repr(C)]
pub struct HMAC_CTX {
    md: *mut hash::EVP_MD,
    md_ctx: hash::EVP_MD_CTX,
    i_ctx: hash::EVP_MD_CTX,
    o_ctx: hash::EVP_MD_CTX,
    key_length: c_uint,
    key: [c_uchar, ..128]
}
use libc::{c_int, c_uint};

#[link(name = "crypto")]
extern {
    fn HMAC_CTX_init(ctx: *mut HMAC_CTX);
    fn HMAC_Init_ex(ctx: *mut HMAC_CTX, key: *const u8, keylen: c_int, md: *const hash::EVP_MD, imple: *const ENGINE);
    fn HMAC_Update(ctx: *mut HMAC_CTX, input: *const u8, len: c_uint);
    fn HMAC_Final(ctx: *mut HMAC_CTX, output: *mut u8, len: *mut c_uint);
}

#[allow(non_camel_case_types)]
#[repr(C)]
struct ENGINE;
use crypto::hash;
use ffi;

pub struct HMAC {
    ctx: HMAC_CTX,
    ctx: ffi::HMAC_CTX,
    len: uint,
}

@@ -51,10 +29,10 @@ pub fn HMAC(ht: hash::HashType, key: &[u8]) -> HMAC {
    unsafe {
        let (evp, mdlen) = hash::evpmd(ht);

        let mut ctx : HMAC_CTX = ::std::mem::uninitialized();
        let mut ctx : ffi::HMAC_CTX = ::std::mem::uninitialized();

        HMAC_CTX_init(&mut ctx);
        HMAC_Init_ex(&mut ctx,
        ffi::HMAC_CTX_init(&mut ctx);
        ffi::HMAC_Init_ex(&mut ctx,
                          key.as_ptr(),
                          key.len() as c_int,
                          evp, 0 as *const _);
@@ -66,7 +44,7 @@ pub fn HMAC(ht: hash::HashType, key: &[u8]) -> HMAC {
impl HMAC {
    pub fn update(&mut self, data: &[u8]) {
        unsafe {
            HMAC_Update(&mut self.ctx, data.as_ptr(), data.len() as c_uint)
            ffi::HMAC_Update(&mut self.ctx, data.as_ptr(), data.len() as c_uint)
        }
    }

@@ -74,7 +52,7 @@ impl HMAC {
        unsafe {
            let mut res = Vec::from_elem(self.len, 0u8);
            let mut outlen = 0;
            HMAC_Final(&mut self.ctx, res.as_mut_ptr(), &mut outlen);
            ffi::HMAC_Final(&mut self.ctx, res.as_mut_ptr(), &mut outlen);
            assert!(self.len == outlen as uint)
            res
        }
Loading