diff --git a/openssl/src/crypto/rsa.rs b/openssl/src/crypto/rsa.rs index 3ba063cd..da2e1c79 100644 --- a/openssl/src/crypto/rsa.rs +++ b/openssl/src/crypto/rsa.rs @@ -11,6 +11,24 @@ use HashTypeInternals; use crypto::hash; use crypto::util::{CallbackState, invoke_passwd_cb}; +/// Type of encryption padding to use. +#[derive(Copy, Clone)] +pub enum EncryptionPadding { + NoPadding, + OAEP, + PKCS1v15 +} + +impl EncryptionPadding { + fn openssl_padding_code(&self) -> c_int { + match self { + &EncryptionPadding::NoPadding => ffi::RSA_NO_PADDING, + &EncryptionPadding::OAEP => ffi::RSA_PKCS1_OAEP_PADDING, + &EncryptionPadding::PKCS1v15 => ffi::RSA_PKCS1_PADDING + } + } +} + pub struct RSA(*mut ffi::RSA); impl Drop for RSA { @@ -162,7 +180,83 @@ impl RSA { } } - pub fn sign(&self, hash: hash::Type, message: &[u8]) -> Result, ErrorStack> { + /** + * Decrypts data with the private key, using provided padding, returning the decrypted data. + */ + pub fn private_decrypt(&self, from: &[u8], padding: EncryptionPadding) -> Result, ErrorStack> { + assert!(self.d().is_some(), "private components missing"); + let k_len = self.size().expect("RSA missing an n"); + let mut to: Vec = vec![0; k_len as usize]; + + unsafe { + let enc_len = try_ssl_returns_size!(ffi::RSA_private_decrypt(from.len() as i32, + from.as_ptr(), + to.as_mut_ptr(), + self.0, + padding.openssl_padding_code())); + to.truncate(enc_len as usize); + Ok(to) + } + } + + /** + * Encrypts data with the private key, using provided padding, returning the encrypted data. + */ + pub fn private_encrypt(&self, from: &[u8], padding: EncryptionPadding) -> Result, ErrorStack> { + assert!(self.d().is_some(), "private components missing"); + let k_len = self.size().expect("RSA missing an n"); + let mut to:Vec = vec![0; k_len as usize]; + + unsafe { + let enc_len = try_ssl_returns_size!(ffi::RSA_private_encrypt(from.len() as c_int, + from.as_ptr(), + to.as_mut_ptr(), + self.0, + padding.openssl_padding_code())); + assert!(enc_len as u32 == k_len); + + Ok(to) + } + } + + /** + * Decrypts data with the public key, using provided padding, returning the decrypted data. + */ + pub fn public_decrypt(&self, from: &[u8], padding: EncryptionPadding) -> Result, ErrorStack> { + let k_len = self.size().expect("RSA missing an n"); + let mut to: Vec = vec![0; k_len as usize]; + + unsafe { + let enc_len = try_ssl_returns_size!(ffi::RSA_public_decrypt(from.len() as i32, + from.as_ptr(), + to.as_mut_ptr(), + self.0, + padding.openssl_padding_code())); + to.truncate(enc_len as usize); + Ok(to) + } + } + + /** + * Encrypts data with the public key, using provided padding, returning the encrypted data. + */ + pub fn public_encrypt(&self, from: &[u8], padding: EncryptionPadding) -> Result, ErrorStack> { + let k_len = self.size().expect("RSA missing an n"); + let mut to:Vec = vec![0; k_len as usize]; + + unsafe { + let enc_len = try_ssl_returns_size!(ffi::RSA_public_encrypt(from.len() as c_int, + from.as_ptr(), + to.as_mut_ptr(), + self.0, + padding.openssl_padding_code())); + assert!(enc_len as u32 == k_len); + + Ok(to) + } + } + + pub fn sign(&self, hash: hash::Type, message: &[u8]) -> Result, ErrorStack> { assert!(self.d().is_some(), "private components missing"); let k_len = self.size().expect("RSA missing an n"); let mut sig = vec![0; k_len as usize]; @@ -335,4 +429,61 @@ mod test { assert!(password_queried); } + + #[test] + pub fn test_public_encrypt_private_decrypt_with_padding() { + let key = include_bytes!("../../test/rsa.pem.pub"); + let public_key = RSA::public_key_from_pem(key).unwrap(); + + let original_data: Vec = "This is test".to_string().into_bytes(); + let result = public_key.public_encrypt(&original_data, EncryptionPadding::PKCS1v15).unwrap(); + + assert_eq!(result.len(), 256); + + let pkey = include_bytes!("../../test/rsa.pem"); + let private_key = RSA::private_key_from_pem(pkey).unwrap(); + let dec_result = private_key.private_decrypt(&result, EncryptionPadding::PKCS1v15).unwrap(); + + assert_eq!(dec_result, original_data); + } + + #[test] + fn test_private_encrypt() { + let mut k0 = super::RSA::generate(512).unwrap(); + let k0pkey = k0.public_key_to_pem().unwrap(); + let mut k1 = super::RSA::public_key_from_pem(&k0pkey).unwrap(); + + let msg = vec!(0xdeu8, 0xadu8, 0xd0u8, 0x0du8); + + let emsg = k0.private_encrypt(&msg, EncryptionPadding::PKCS1v15).unwrap(); + let dmsg = k1.public_decrypt(&emsg, EncryptionPadding::PKCS1v15).unwrap(); + assert!(msg == dmsg); + } + + #[test] + fn test_public_encrypt() { + let mut k0 = super::RSA::generate(512).unwrap(); + let k0pkey = k0.public_key_to_pem().unwrap(); + let mut k1 = super::RSA::public_key_from_pem(&k0pkey).unwrap(); + + let msg = vec!(0xdeu8, 0xadu8, 0xd0u8, 0x0du8); + + let emsg = k1.public_encrypt(&msg, EncryptionPadding::OAEP).unwrap(); + let dmsg = k0.private_decrypt(&emsg, EncryptionPadding::OAEP).unwrap(); + assert!(msg == dmsg); + } + + #[test] + fn test_public_encrypt_pkcs() { + let mut k0 = super::RSA::generate(512).unwrap(); + let k0pkey = k0.public_key_to_pem().unwrap(); + let mut k1 = super::RSA::public_key_from_pem(&k0pkey).unwrap(); + + let msg = vec!(0xdeu8, 0xadu8, 0xd0u8, 0x0du8); + + let emsg = k1.public_encrypt(&msg, super::EncryptionPadding::PKCS1v15).unwrap(); + let dmsg = k0.private_decrypt(&emsg, super::EncryptionPadding::PKCS1v15).unwrap(); + assert!(msg == dmsg); + } + } diff --git a/openssl/src/macros.rs b/openssl/src/macros.rs index 35221f1c..e2d9cae5 100644 --- a/openssl/src/macros.rs +++ b/openssl/src/macros.rs @@ -33,6 +33,17 @@ macro_rules! try_ssl_null{ }) } +/// Shortcut return with SSL error if last error result is -1 +/// (default for size) +macro_rules! try_ssl_returns_size{ + ($e:expr) => ( + if $e == -1 { + return Err(::error::ErrorStack::get().into()) + } else { + $e + } + ) +} /// Lifts current SSL error code into Result<(), Error> /// if expression is true @@ -57,3 +68,15 @@ macro_rules! lift_ssl_if{ macro_rules! lift_ssl { ($e:expr) => (lift_ssl_if!($e == 0)) } + +/// Lifts current SSL error code into Result<(), Error> +/// if SSL returned -1 (default size error indication) +macro_rules! lift_ssl_returns_size { + ($e:expr) => ( { + if $e == -1 { + Err(::error::ErrorStack::get().into()) + } else { + Ok($e) + } + }) +}