De-enumify Padding

This commit is contained in:
Steven Fackler 2016-10-15 16:19:19 -07:00
parent ee18988584
commit 4ba5292a0a
1 changed files with 23 additions and 23 deletions

View File

@ -11,19 +11,19 @@ use crypto::util::{CallbackState, invoke_passwd_cb};
/// Type of encryption padding to use.
#[derive(Copy, Clone)]
pub enum Padding {
None,
OAEP,
PKCS1v15
}
pub struct Padding(c_int);
impl Padding {
fn openssl_padding_code(&self) -> c_int {
match *self {
Padding::None => ffi::RSA_NO_PADDING,
Padding::OAEP => ffi::RSA_PKCS1_OAEP_PADDING,
Padding::PKCS1v15 => ffi::RSA_PKCS1_PADDING
pub fn none() -> Padding {
Padding(ffi::RSA_NO_PADDING)
}
pub fn pkcs1() -> Padding {
Padding(ffi::RSA_PKCS1_PADDING)
}
pub fn pkcs1_oaep() -> Padding {
Padding(ffi::RSA_PKCS1_OAEP_PADDING)
}
}
@ -183,7 +183,7 @@ impl RSA {
from.as_ptr(),
to.as_mut_ptr(),
self.0,
padding.openssl_padding_code()));
padding.0));
to.truncate(enc_len as usize);
Ok(to)
}
@ -202,7 +202,7 @@ impl RSA {
from.as_ptr(),
to.as_mut_ptr(),
self.0,
padding.openssl_padding_code()));
padding.0));
assert!(enc_len as u32 == k_len);
Ok(to)
@ -221,7 +221,7 @@ impl RSA {
from.as_ptr(),
to.as_mut_ptr(),
self.0,
padding.openssl_padding_code()));
padding.0));
to.truncate(enc_len as usize);
Ok(to)
}
@ -239,7 +239,7 @@ impl RSA {
from.as_ptr(),
to.as_mut_ptr(),
self.0,
padding.openssl_padding_code()));
padding.0));
assert!(enc_len as u32 == k_len);
Ok(to)
@ -425,13 +425,13 @@ mod test {
let public_key = RSA::public_key_from_pem(key).unwrap();
let original_data: Vec<u8> = "This is test".to_string().into_bytes();
let result = public_key.public_encrypt(&original_data, Padding::PKCS1v15).unwrap();
let result = public_key.public_encrypt(&original_data, Padding::pkcs1()).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, Padding::PKCS1v15).unwrap();
let dec_result = private_key.private_decrypt(&result, Padding::pkcs1()).unwrap();
assert_eq!(dec_result, original_data);
}
@ -444,8 +444,8 @@ mod test {
let msg = vec!(0xdeu8, 0xadu8, 0xd0u8, 0x0du8);
let emsg = k0.private_encrypt(&msg, Padding::PKCS1v15).unwrap();
let dmsg = k1.public_decrypt(&emsg, Padding::PKCS1v15).unwrap();
let emsg = k0.private_encrypt(&msg, Padding::pkcs1()).unwrap();
let dmsg = k1.public_decrypt(&emsg, Padding::pkcs1()).unwrap();
assert!(msg == dmsg);
}
@ -457,8 +457,8 @@ mod test {
let msg = vec!(0xdeu8, 0xadu8, 0xd0u8, 0x0du8);
let emsg = k1.public_encrypt(&msg, Padding::OAEP).unwrap();
let dmsg = k0.private_decrypt(&emsg, Padding::OAEP).unwrap();
let emsg = k1.public_encrypt(&msg, Padding::pkcs1_oaep()).unwrap();
let dmsg = k0.private_decrypt(&emsg, Padding::pkcs1_oaep()).unwrap();
assert!(msg == dmsg);
}
@ -470,8 +470,8 @@ mod test {
let msg = vec!(0xdeu8, 0xadu8, 0xd0u8, 0x0du8);
let emsg = k1.public_encrypt(&msg, super::Padding::PKCS1v15).unwrap();
let dmsg = k0.private_decrypt(&emsg, super::Padding::PKCS1v15).unwrap();
let emsg = k1.public_encrypt(&msg, super::Padding::pkcs1()).unwrap();
let dmsg = k0.private_decrypt(&emsg, super::Padding::pkcs1()).unwrap();
assert!(msg == dmsg);
}