Switch over to rustpkg

1 files changed, 451 insertions, 0 deletions

diff --git a/src/crypto/pkey.rs b/src/crypto/pkey.rs
new file mode 100644
index 00000000..7b5245b2
--- /dev/null
+++ b/src/crypto/pkey.rs
@@ -0,0 +1,451 @@
+use std::cast;
+use std::libc::{c_int, c_uint};
+use std::libc;
+use std::ptr;
+use std::vec;
+use hash::{HashType, MD5, SHA1, SHA224, SHA256, SHA384, SHA512};
+
+#[allow(non_camel_case_types)]
+pub type EVP_PKEY = *libc::c_void;
+
+#[allow(non_camel_case_types)]
+pub type RSA = *libc::c_void;
+
+mod libcrypto {
+    use super::*;
+    use std::libc::{c_char, c_int, c_uint};
+
+    #[link(name = "crypto")]
+    extern {
+        pub fn EVP_PKEY_new() -> *EVP_PKEY;
+        pub fn EVP_PKEY_free(k: *EVP_PKEY);
+        pub fn EVP_PKEY_assign(pkey: *EVP_PKEY, typ: c_int, key: *c_char) -> c_int;
+        pub fn EVP_PKEY_get1_RSA(k: *EVP_PKEY) -> *RSA;
+
+        pub fn i2d_PublicKey(k: *EVP_PKEY, buf: **mut u8) -> c_int;
+        pub fn d2i_PublicKey(t: c_int, k: **EVP_PKEY, buf: **u8, len: c_uint) -> *EVP_PKEY;
+        pub fn i2d_PrivateKey(k: *EVP_PKEY, buf: **mut u8) -> c_int;
+        pub fn d2i_PrivateKey(t: c_int, k: **EVP_PKEY, buf: **u8, len: c_uint) -> *EVP_PKEY;
+
+        pub fn RSA_generate_key(modsz: c_uint, e: c_uint, cb: *u8, cbarg: *u8) -> *RSA;
+        pub fn RSA_size(k: *RSA) -> c_uint;
+
+        pub fn RSA_public_encrypt(flen: c_uint, from: *u8, to: *mut u8, k: *RSA,
+                                  pad: c_int) -> c_int;
+        pub fn RSA_private_decrypt(flen: c_uint, from: *u8, to: *mut u8, k: *RSA,
+                                   pad: c_int) -> c_int;
+        pub fn RSA_sign(t: c_int, m: *u8, mlen: c_uint, sig: *mut u8, siglen: *mut c_uint,
+                        k: *RSA) -> c_int;
+        pub fn RSA_verify(t: c_int, m: *u8, mlen: c_uint, sig: *u8, siglen: c_uint,
+                          k: *RSA) -> c_int;
+    }
+}
+
+enum Parts {
+    Neither,
+    Public,
+    Both
+}
+
+/// Represents a role an asymmetric key might be appropriate for.
+pub enum Role {
+    Encrypt,
+    Decrypt,
+    Sign,
+    Verify
+}
+
+/// Type of encryption padding to use.
+pub enum EncryptionPadding {
+    OAEP,
+    PKCS1v15
+}
+
+fn openssl_padding_code(padding: EncryptionPadding) -> c_int {
+    match padding {
+        OAEP => 4,
+        PKCS1v15 => 1
+    }
+}
+
+fn openssl_hash_nid(hash: HashType) -> c_int {
+    match hash {
+        MD5    => 4,   // NID_md5,
+        SHA1   => 64,  // NID_sha1
+        SHA224 => 675, // NID_sha224
+        SHA256 => 672, // NID_sha256
+        SHA384 => 673, // NID_sha384
+        SHA512 => 674, // NID_sha512
+    }
+}
+
+pub struct PKey {
+    priv evp: *EVP_PKEY,
+    priv parts: Parts,
+}
+
+/// Represents a public key, optionally with a private key attached.
+impl PKey {
+    pub fn new() -> PKey {
+        PKey {
+            evp: unsafe { libcrypto::EVP_PKEY_new() },
+            parts: Neither,
+        }
+    }
+
+    fn _tostr(&self, f: extern "C" unsafe fn(*EVP_PKEY, **mut u8) -> c_int) -> ~[u8] {
+        unsafe {
+            let len = f(self.evp, ptr::null());
+            if len < 0 as c_int { return ~[]; }
+            let mut s = vec::from_elem(len as uint, 0u8);
+
+            let r = s.as_mut_buf(|buf, _| {
+                f(self.evp, &buf)
+            });
+
+            s.truncate(r as uint);
+            s
+        }
+    }
+
+    fn _fromstr(&mut self, s: &[u8], f: extern "C" unsafe fn(c_int, **EVP_PKEY, **u8, c_uint) -> *EVP_PKEY) {
+        s.as_imm_buf(|ps, len| {
+            let evp = ptr::null();
+            unsafe {
+                f(6 as c_int, &evp, &ps, len as c_uint);
+            }
+            self.evp = evp;
+        });
+    }
+
+    pub fn gen(&mut self, keysz: uint) {
+        unsafe {
+            let rsa = libcrypto::RSA_generate_key(
+                keysz as c_uint,
+                65537u as c_uint,
+                ptr::null(),
+                ptr::null()
+            );
+
+            // XXX: 6 == NID_rsaEncryption
+            libcrypto::EVP_PKEY_assign(
+                self.evp,
+                6 as c_int,
+                cast::transmute(rsa));
+
+            self.parts = Both;
+        }
+    }
+
+    /**
+     * Returns a serialized form of the public key, suitable for load_pub().
+     */
+    pub fn save_pub(&self) -> ~[u8] {
+        self._tostr(libcrypto::i2d_PublicKey)
+    }
+
+    /**
+     * Loads a serialized form of the public key, as produced by save_pub().
+     */
+    pub fn load_pub(&mut self, s: &[u8]) {
+        self._fromstr(s, libcrypto::d2i_PublicKey);
+        self.parts = Public;
+    }
+
+    /**
+     * Returns a serialized form of the public and private keys, suitable for
+     * load_priv().
+     */
+    pub fn save_priv(&self) -> ~[u8] {
+        self._tostr(libcrypto::i2d_PrivateKey)
+    }
+    /**
+     * Loads a serialized form of the public and private keys, as produced by
+     * save_priv().
+     */
+    pub fn load_priv(&mut self, s: &[u8]) {
+        self._fromstr(s, libcrypto::d2i_PrivateKey);
+        self.parts = Both;
+    }
+
+    /**
+     * Returns the size of the public key modulus.
+     */
+    pub fn size(&self) -> uint {
+        unsafe {
+            libcrypto::RSA_size(libcrypto::EVP_PKEY_get1_RSA(self.evp)) as uint
+        }
+    }
+
+    /**
+     * Returns whether this pkey object can perform the specified role.
+     */
+    pub fn can(&self, r: Role) -> bool {
+        match r {
+            Encrypt =>
+                match self.parts {
+                    Neither => false,
+                    _ => true,
+                },
+            Verify =>
+                match self.parts {
+                    Neither => false,
+                    _ => true,
+                },
+            Decrypt =>
+                match self.parts {
+                    Both => true,
+                    _ => false,
+                },
+            Sign =>
+                match self.parts {
+                    Both => true,
+                    _ => false,
+                },
+        }
+    }
+
+    /**
+     * Returns the maximum amount of data that can be encrypted by an encrypt()
+     * call.
+     */
+    pub fn max_data(&self) -> uint {
+        unsafe {
+            let rsa = libcrypto::EVP_PKEY_get1_RSA(self.evp);
+            let len = libcrypto::RSA_size(rsa);
+
+            // 41 comes from RSA_public_encrypt(3) for OAEP
+            len as uint - 41u
+        }
+    }
+
+    pub fn encrypt_with_padding(&self, s: &[u8], padding: EncryptionPadding) -> ~[u8] {
+        unsafe {
+            let rsa = libcrypto::EVP_PKEY_get1_RSA(self.evp);
+            let len = libcrypto::RSA_size(rsa);
+
+            assert!(s.len() < self.max_data());
+
+            let mut r = vec::from_elem(len as uint + 1u, 0u8);
+
+            let rv = r.as_mut_buf(|pr, _len| {
+                        s.as_imm_buf(|ps, s_len| {
+                            libcrypto::RSA_public_encrypt(
+                                s_len as c_uint,
+                                ps,
+                                pr,
+                                rsa,
+                                openssl_padding_code(padding)
+                            )
+                        })
+                     });
+            if rv < 0 as c_int {
+                ~[]
+            } else {
+                r.truncate(rv as uint);
+                r
+            }
+        }
+    }
+
+    pub fn decrypt_with_padding(&self, s: &[u8], padding: EncryptionPadding) -> ~[u8] {
+        unsafe {
+            let rsa = libcrypto::EVP_PKEY_get1_RSA(self.evp);
+            let len = libcrypto::RSA_size(rsa);
+
+            assert_eq!(s.len() as c_uint, libcrypto::RSA_size(rsa));
+
+            let mut r = vec::from_elem(len as uint + 1u, 0u8);
+
+            let rv = r.as_mut_buf(|pr, _len| {
+                        s.as_imm_buf(|ps, s_len| {
+                            libcrypto::RSA_private_decrypt(
+                                s_len as c_uint,
+                                ps,
+                                pr,
+                                rsa,
+                                openssl_padding_code(padding)
+                            )
+                        })
+                     });
+
+            if rv < 0 as c_int {
+                ~[]
+            } else {
+                r.truncate(rv as uint);
+                r
+            }
+        }
+    }
+
+    /**
+     * Encrypts data using OAEP padding, returning the encrypted data. The
+     * supplied data must not be larger than max_data().
+     */
+    pub fn encrypt(&self, s: &[u8]) -> ~[u8] { self.encrypt_with_padding(s, OAEP) }
+
+    /**
+     * Decrypts data, expecting OAEP padding, returning the decrypted data.
+     */
+    pub fn decrypt(&self, s: &[u8]) -> ~[u8] { self.decrypt_with_padding(s, OAEP) }
+
+    /**
+     * Signs data, using OpenSSL's default scheme and sha256. Unlike encrypt(),
+     * can process an arbitrary amount of data; returns the signature.
+     */
+    pub fn sign(&self, s: &[u8]) -> ~[u8] { self.sign_with_hash(s, SHA256) }
+
+    /**
+     * Verifies a signature s (using OpenSSL's default scheme and sha256) on a
+     * message m. Returns true if the signature is valid, and false otherwise.
+     */
+    pub fn verify(&self, m: &[u8], s: &[u8]) -> bool { self.verify_with_hash(m, s, SHA256) }
+
+    pub fn sign_with_hash(&self, s: &[u8], hash: HashType) -> ~[u8] {
+        unsafe {
+            let rsa = libcrypto::EVP_PKEY_get1_RSA(self.evp);
+            let mut len = libcrypto::RSA_size(rsa);
+            let mut r = vec::from_elem(len as uint + 1u, 0u8);
+
+            let rv = r.as_mut_buf(|pr, _len| {
+                        s.as_imm_buf(|ps, s_len| {
+                            libcrypto::RSA_sign(
+                                openssl_hash_nid(hash),
+                                ps,
+                                s_len as c_uint,
+                                pr,
+                                &mut len,
+                                rsa)
+                        })
+                     });
+
+            if rv < 0 as c_int {
+                ~[]
+            } else {
+                r.truncate(len as uint);
+                r
+            }
+        }
+    }
+
+    pub fn verify_with_hash(&self, m: &[u8], s: &[u8], hash: HashType) -> bool {
+        unsafe {
+            let rsa = libcrypto::EVP_PKEY_get1_RSA(self.evp);
+
+            m.as_imm_buf(|pm, m_len| {
+                s.as_imm_buf(|ps, s_len| {
+                    let rv = libcrypto::RSA_verify(
+                        openssl_hash_nid(hash),
+                        pm,
+                        m_len as c_uint,
+                        ps,
+                        s_len as c_uint,
+                        rsa
+                    );
+
+                    rv == 1 as c_int
+                })
+            })
+        }
+    }
+}
+
+impl Drop for PKey {
+    fn drop(&mut self) {
+        unsafe {
+            libcrypto::EVP_PKEY_free(self.evp);
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use hash::{MD5, SHA1};
+
+    #[test]
+    fn test_gen_pub() {
+        let mut k0 = PKey::new();
+        let mut k1 = PKey::new();
+        k0.gen(512u);
+        k1.load_pub(k0.save_pub());
+        assert!(k0.save_pub() == k1.save_pub());
+        assert!(k0.size() == k1.size());
+        assert!(k0.can(Encrypt));
+        assert!(k0.can(Decrypt));
+        assert!(k0.can(Verify));
+        assert!(k0.can(Sign));
+        assert!(k1.can(Encrypt));
+        assert!(!k1.can(Decrypt));
+        assert!(k1.can(Verify));
+        assert!(!k1.can(Sign));
+    }
+
+    #[test]
+    fn test_gen_priv() {
+        let mut k0 = PKey::new();
+        let mut k1 = PKey::new();
+        k0.gen(512u);
+        k1.load_priv(k0.save_priv());
+        assert!(k0.save_priv() == k1.save_priv());
+        assert!(k0.size() == k1.size());
+        assert!(k0.can(Encrypt));
+        assert!(k0.can(Decrypt));
+        assert!(k0.can(Verify));
+        assert!(k0.can(Sign));
+        assert!(k1.can(Encrypt));
+        assert!(k1.can(Decrypt));
+        assert!(k1.can(Verify));
+        assert!(k1.can(Sign));
+    }
+
+    #[test]
+    fn test_encrypt() {
+        let mut k0 = PKey::new();
+        let mut k1 = PKey::new();
+        let msg = ~[0xdeu8, 0xadu8, 0xd0u8, 0x0du8];
+        k0.gen(512u);
+        k1.load_pub(k0.save_pub());
+        let emsg = k1.encrypt(msg);
+        let dmsg = k0.decrypt(emsg);
+        assert!(msg == dmsg);
+    }
+
+    #[test]
+    fn test_encrypt_pkcs() {
+        let mut k0 = PKey::new();
+        let mut k1 = PKey::new();
+        let msg = ~[0xdeu8, 0xadu8, 0xd0u8, 0x0du8];
+        k0.gen(512u);
+        k1.load_pub(k0.save_pub());
+        let emsg = k1.encrypt_with_padding(msg, PKCS1v15);
+        let dmsg = k0.decrypt_with_padding(emsg, PKCS1v15);
+        assert!(msg == dmsg);
+    }
+
+    #[test]
+    fn test_sign() {
+        let mut k0 = PKey::new();
+        let mut k1 = PKey::new();
+        let msg = ~[0xdeu8, 0xadu8, 0xd0u8, 0x0du8];
+        k0.gen(512u);
+        k1.load_pub(k0.save_pub());
+        let sig = k0.sign(msg);
+        let rv = k1.verify(msg, sig);
+        assert!(rv == true);
+    }
+
+    #[test]
+    fn test_sign_hashes() {
+        let mut k0 = PKey::new();
+        let mut k1 = PKey::new();
+        let msg = ~[0xdeu8, 0xadu8, 0xd0u8, 0x0du8];
+        k0.gen(512u);
+        k1.load_pub(k0.save_pub());
+
+        let sig = k0.sign_with_hash(msg, MD5);
+
+        assert!(k1.verify_with_hash(msg, sig, MD5));
+        assert!(!k1.verify_with_hash(msg, sig, SHA1));
+    }
+}