Commit b6719de9 authored by Steven Fackler's avatar Steven Fackler
Browse files

Rename EncryptionPadding to Padding

parent b8e58653
Loading
Loading
Loading
Loading
+17 −17
Original line number Diff line number Diff line
@@ -13,18 +13,18 @@ use crypto::util::{CallbackState, invoke_passwd_cb};

/// Type of encryption padding to use.
#[derive(Copy, Clone)]
pub enum EncryptionPadding {
pub enum Padding {
    NoPadding,
    OAEP,
    PKCS1v15
}

impl EncryptionPadding {
impl Padding {
    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
            &Padding::NoPadding => ffi::RSA_NO_PADDING,
            &Padding::OAEP => ffi::RSA_PKCS1_OAEP_PADDING,
            &Padding::PKCS1v15 => ffi::RSA_PKCS1_PADDING
        }
    }
}
@@ -183,7 +183,7 @@ impl RSA {
    /**
     * Decrypts data with the private key, using provided padding, returning the decrypted data.
     */
    pub fn private_decrypt(&self, from: &[u8], padding: EncryptionPadding) -> Result<Vec<u8>, ErrorStack> {
    pub fn private_decrypt(&self, from: &[u8], padding: Padding) -> Result<Vec<u8>, ErrorStack> {
        assert!(self.d().is_some(), "private components missing");
        let k_len = self.size().expect("RSA missing an n");
        let mut to: Vec<u8> = vec![0; k_len as usize];
@@ -202,7 +202,7 @@ impl RSA {
    /**
     * Encrypts data with the private key, using provided padding, returning the encrypted data.
     */
    pub fn private_encrypt(&self, from: &[u8], padding: EncryptionPadding) -> Result<Vec<u8>, ErrorStack> {
    pub fn private_encrypt(&self, from: &[u8], padding: Padding) -> Result<Vec<u8>, ErrorStack> {
        assert!(self.d().is_some(), "private components missing");
        let k_len = self.size().expect("RSA missing an n");
        let mut to:Vec<u8> = vec![0; k_len as usize];
@@ -222,7 +222,7 @@ impl RSA {
    /**
     * Decrypts data with the public key, using provided padding, returning the decrypted data.
     */
    pub fn public_decrypt(&self, from: &[u8], padding: EncryptionPadding) -> Result<Vec<u8>, ErrorStack> {
    pub fn public_decrypt(&self, from: &[u8], padding: Padding) -> Result<Vec<u8>, ErrorStack> {
        let k_len = self.size().expect("RSA missing an n");
        let mut to: Vec<u8> = vec![0; k_len as usize];

@@ -240,7 +240,7 @@ impl RSA {
    /**
     * Encrypts data with the public key, using provided padding, returning the encrypted data.
     */
    pub fn public_encrypt(&self, from: &[u8], padding: EncryptionPadding) -> Result<Vec<u8>, ErrorStack> {
    pub fn public_encrypt(&self, from: &[u8], padding: Padding) -> Result<Vec<u8>, ErrorStack> {
        let k_len = self.size().expect("RSA missing an n");
        let mut to:Vec<u8> = vec![0; k_len as usize];

@@ -436,13 +436,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, EncryptionPadding::PKCS1v15).unwrap();
        let result = public_key.public_encrypt(&original_data, Padding::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();
        let dec_result = private_key.private_decrypt(&result, Padding::PKCS1v15).unwrap();

       assert_eq!(dec_result, original_data);
    }
@@ -455,8 +455,8 @@ mod test {

       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();
       let emsg = k0.private_encrypt(&msg, Padding::PKCS1v15).unwrap();
       let dmsg = k1.public_decrypt(&emsg, Padding::PKCS1v15).unwrap();
       assert!(msg == dmsg);
   }

@@ -468,8 +468,8 @@ mod test {

       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();
       let emsg = k1.public_encrypt(&msg, Padding::OAEP).unwrap();
       let dmsg = k0.private_decrypt(&emsg, Padding::OAEP).unwrap();
       assert!(msg == dmsg);
   }

@@ -481,8 +481,8 @@ mod test {

       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();
       let emsg = k1.public_encrypt(&msg, super::Padding::PKCS1v15).unwrap();
       let dmsg = k0.private_decrypt(&emsg, super::Padding::PKCS1v15).unwrap();
       assert!(msg == dmsg);
   }