Commit d0676bf3 authored by Edoardo Morandi's avatar Edoardo Morandi
Browse files

Add encrypt_len and decrypt_len

parent b2410923

openssl/src/encrypt.rs

+44 −0
Original line number Diff line number Diff line
@@ -117,6 +117,8 @@ impl<'a> Encrypter<'a> { 


    /// Performs public key encryption.

    ///

    /// In order to know the size needed for the output buffer, use [`encrypt_len`](Encrypter::encrypt_len).

    ///

    /// This corresponds to [`EVP_PKEY_encrypt`].

    ///

    /// [`EVP_PKEY_encrypt`]: https://www.openssl.org/docs/manmaster/man3/EVP_PKEY_encrypt.html
@@ -134,6 +136,26 @@ impl<'a> Encrypter<'a> { 


        Ok(written)

    }



    /// Gets the size of the buffer needed to encrypt the input data.

    ///

    /// This corresponds to [`EVP_PKEY_encrypt`] called with a null pointer as output argument.

    ///

    /// [`EVP_PKEY_encrypt`]: https://www.openssl.org/docs/manmaster/man3/EVP_PKEY_encrypt.html

    pub fn encrypt_len(&self, from: &[u8]) -> Result<usize, ErrorStack> {

        let mut written = 0;

        unsafe {

            cvt(ffi::EVP_PKEY_encrypt(

                self.pctx,

                ptr::null_mut(),

                &mut written,

                from.as_ptr(),

                from.len(),

            ))?;

        }



        Ok(written)

    }

}

pub struct Decrypter<'a> {

    pctx: *mut ffi::EVP_PKEY_CTX,
@@ -245,6 +267,8 @@ impl<'a> Decrypter<'a> { 


    /// Performs public key decryption.

    ///

    /// In order to know the size needed for the output buffer, use [`decrypt_len`](Decrypter::decrypt_len).

    ///

    /// This corresponds to [`EVP_PKEY_decrypt`].

    ///

    /// [`EVP_PKEY_decrypt`]: https://www.openssl.org/docs/manmaster/man3/EVP_PKEY_decrypt.html
@@ -262,6 +286,26 @@ impl<'a> Decrypter<'a> { 


        Ok(written)

    }



    /// Gets the size of the buffer needed to decrypt the input data.

    ///

    /// This corresponds to [`EVP_PKEY_decrypt`] called with a null pointer as output argument.

    ///

    /// [`EVP_PKEY_decrypt`]: https://www.openssl.org/docs/manmaster/man3/EVP_PKEY_decrypt.html

    pub fn decrypt_len(&self, from: &[u8]) -> Result<usize, ErrorStack> {

        let mut written = 0;

        unsafe {

            cvt(ffi::EVP_PKEY_decrypt(

                self.pctx,

                ptr::null_mut(),

                &mut written,

                from.as_ptr(),

                from.len(),

            ))?;

        }



        Ok(written)

    }

}



#[cfg(test)]