1use crate::{
60 cbb_to_buffer, digest, parse_with_cbs, scoped, sealed, with_output_vec, Buffer, FfiSlice,
61 ForeignTypeRef, InvalidSignatureError,
62};
63use alloc::vec::Vec;
64use core::ptr::null_mut;
65
66pub struct PublicKey(*mut bssl_sys::RSA);
68
69impl PublicKey {
70 pub fn from_der_rsa_public_key(der: &[u8]) -> Option<Self> {
72 Some(PublicKey(parse_with_cbs(
73 der,
74 |ptr| unsafe { bssl_sys::RSA_free(ptr) },
76 |cbs| unsafe { bssl_sys::RSA_parse_public_key(cbs) },
78 )?))
79 }
80
81 pub fn to_der_rsa_public_key(&self) -> Buffer {
83 cbb_to_buffer(300, |cbb| unsafe {
84 assert_eq!(1, bssl_sys::RSA_marshal_public_key(cbb, self.0))
86 })
87 }
88
89 pub fn from_der_subject_public_key_info(spki: &[u8]) -> Option<Self> {
92 let mut pkey = scoped::EvpPkey::from_ptr(parse_with_cbs(
93 spki,
94 |pkey| unsafe { bssl_sys::EVP_PKEY_free(pkey) },
96 |cbs| unsafe { bssl_sys::EVP_parse_public_key(cbs) },
98 )?);
99 let rsa = unsafe { bssl_sys::EVP_PKEY_get1_RSA(pkey.as_ffi_ptr()) };
100 if !rsa.is_null() {
101 Some(PublicKey(rsa))
104 } else {
105 None
106 }
107 }
108
109 pub fn to_der_subject_public_key_info(&self) -> Buffer {
112 let mut pkey = scoped::EvpPkey::new();
113 assert_eq!(1, unsafe {
115 bssl_sys::EVP_PKEY_set1_RSA(pkey.as_ffi_ptr(), self.0)
116 });
117 cbb_to_buffer(384, |cbb| unsafe {
118 assert_eq!(1, bssl_sys::EVP_marshal_public_key(cbb, pkey.as_ffi_ptr()));
121 })
122 }
123
124 pub fn verify_pkcs1<Hash: digest::Algorithm>(
128 &self,
129 signed_msg: &[u8],
130 signature: &[u8],
131 ) -> Result<(), InvalidSignatureError> {
132 let digest = Hash::hash_to_vec(signed_msg);
133 let hash_nid = unsafe { bssl_sys::EVP_MD_nid(Hash::get_md(sealed::Sealed).as_ptr()) };
135 let result = unsafe {
136 bssl_sys::RSA_verify(
138 hash_nid,
139 digest.as_slice().as_ffi_ptr(),
140 digest.len(),
141 signature.as_ffi_ptr(),
142 signature.len(),
143 self.0,
144 )
145 };
146 if result == 1 {
147 Ok(())
148 } else {
149 Err(InvalidSignatureError)
150 }
151 }
152}
153
154unsafe impl Sync for PublicKey {}
162unsafe impl Send for PublicKey {}
163
164impl Drop for PublicKey {
165 fn drop(&mut self) {
166 unsafe { bssl_sys::RSA_free(self.0) }
168 }
169}
170
171#[cfg(feature = "std")]
172impl std::fmt::Debug for PublicKey {
173 fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
174 write!(f, "RSA PublicKey")
175 }
176}
177
178#[allow(missing_docs)]
180pub enum KeySize {
181 Rsa2048 = 2048,
182 Rsa3072 = 3072,
183 Rsa4096 = 4096,
184}
185
186pub struct PrivateKey(*mut bssl_sys::RSA);
188
189impl PrivateKey {
190 pub fn generate(size: KeySize) -> Self {
192 let e = scoped::Bignum::from_u64(bssl_sys::RSA_F4 as u64);
193 let ptr = unsafe { bssl_sys::RSA_new() };
194 assert!(!ptr.is_null());
195
196 let result = unsafe {
197 bssl_sys::RSA_generate_key_ex(ptr, size as core::ffi::c_int, e.as_ffi_ptr(), null_mut())
199 };
200 assert_eq!(1, result);
201 Self(ptr)
203 }
204
205 pub fn from_der_rsa_private_key(der: &[u8]) -> Option<Self> {
207 Some(PrivateKey(parse_with_cbs(
208 der,
209 |ptr| unsafe { bssl_sys::RSA_free(ptr) },
211 |cbs| unsafe { bssl_sys::RSA_parse_private_key(cbs) },
213 )?))
214 }
215
216 pub fn to_der_rsa_private_key(&self) -> Buffer {
218 cbb_to_buffer(512, |cbb| unsafe {
219 assert_eq!(1, bssl_sys::RSA_marshal_private_key(cbb, self.0))
221 })
222 }
223
224 pub fn from_der_private_key_info(der: &[u8]) -> Option<Self> {
226 let mut pkey = scoped::EvpPkey::from_ptr(parse_with_cbs(
227 der,
228 |pkey| unsafe { bssl_sys::EVP_PKEY_free(pkey) },
230 |cbs| unsafe { bssl_sys::EVP_parse_private_key(cbs) },
232 )?);
233 let rsa = unsafe { bssl_sys::EVP_PKEY_get1_RSA(pkey.as_ffi_ptr()) };
235 if rsa.is_null() {
236 return None;
237 }
238 Some(Self(rsa))
239 }
240
241 pub fn to_der_private_key_info(&self) -> Buffer {
243 let mut pkey = scoped::EvpPkey::new();
244 assert_eq!(1, unsafe {
245 bssl_sys::EVP_PKEY_set1_RSA(pkey.as_ffi_ptr(), self.0)
248 });
249 unsafe {
250 cbb_to_buffer(384, |cbb| {
251 assert_eq!(1, bssl_sys::EVP_marshal_private_key(cbb, pkey.as_ffi_ptr()));
253 })
254 }
255 }
256
257 pub fn sign_pkcs1<Hash: digest::Algorithm>(&self, to_be_signed: &[u8]) -> Vec<u8> {
261 let digest = Hash::hash_to_vec(to_be_signed);
262 let hash_nid = unsafe { bssl_sys::EVP_MD_nid(Hash::get_md(sealed::Sealed).as_ptr()) };
264 let max_output = unsafe { bssl_sys::RSA_size(self.0) } as usize;
265
266 unsafe {
267 with_output_vec(max_output, |out_buf| {
268 let mut out_len: core::ffi::c_uint = 0;
269 let result = bssl_sys::RSA_sign(
272 hash_nid,
273 digest.as_slice().as_ffi_ptr(),
274 digest.len(),
275 out_buf,
276 &mut out_len,
277 self.0,
278 );
279 assert_eq!(1, result);
282 let out_len = out_len as usize;
283 assert!(out_len <= max_output);
284 out_len
286 })
287 }
288 }
289
290 pub fn as_public(&self) -> PublicKey {
292 unsafe { bssl_sys::RSA_up_ref(self.0) };
295 PublicKey(self.0)
296 }
297}
298
299unsafe impl Sync for PrivateKey {}
307unsafe impl Send for PrivateKey {}
308
309impl Drop for PrivateKey {
310 fn drop(&mut self) {
311 unsafe { bssl_sys::RSA_free(self.0) }
313 }
314}
315
316#[cfg(test)]
317mod test {
318 use super::*;
319
320 #[test]
321 fn sign_and_verify() {
322 let key = PrivateKey::from_der_private_key_info(TEST_PKCS8_BYTES).unwrap();
323 let signed_msg = b"hello world";
324 let sig = key.sign_pkcs1::<digest::Sha256>(signed_msg);
325 assert!(key
326 .as_public()
327 .verify_pkcs1::<digest::Sha256>(signed_msg, &sig)
328 .is_ok());
329 }
330}
331
332#[doc(hidden)]
335pub const TEST_PKCS8_BYTES: &[u8] = b"\x30\x82\x04\xbd\x02\x01\x00\x30\x0d\x06\x09\x2a\x86\x48\x86\xf7\x0d\x01\x01\x01\x05\x00\x04\x82\x04\xa7\x30\x82\x04\xa3\x02\x01\x00\x02\x82\x01\x01\x00\x98\x3f\xf5\xc4\x89\xb7\x6f\x12\xc8\xb5\x82\xaa\x98\xe6\x75\x39\xc4\x44\x46\xa0\x45\x62\x42\x43\x21\x81\xa0\x53\x17\x47\xb3\xdc\xfc\x3b\x76\x03\xd6\xd4\xce\x5e\x9d\x22\xe5\xa3\x59\xa2\x47\x0c\xe4\x82\x33\x7a\x21\xa5\x61\x2a\x77\xa2\x6b\xfa\xa3\x45\x41\x50\xc2\xf7\x0d\xe1\xa6\x3a\x83\x5b\xe6\xb8\x1f\x24\x1e\x24\x89\xf8\x8d\xde\x5f\xf1\x50\x27\x0f\x2b\xbe\x58\xaa\x64\x67\xef\x22\x57\x1e\xf4\x3f\x2e\xba\x4b\x2f\xc3\x5e\x67\xcc\xc3\xf6\xdd\x6b\x31\x58\xb9\xbd\x7b\xf9\x23\xac\xf2\xa9\xb6\x8f\x88\x75\x0f\x73\xdf\xd2\x14\xaa\x41\x28\x5c\x9a\xd6\xc4\xab\x6f\xb0\x53\xb9\x0a\x2c\xfb\x56\x6e\x56\x94\xaa\x1a\x25\x29\x3b\x01\x0c\x7e\x44\x1b\xe1\x76\x12\x73\xc4\x16\x62\x64\x3d\xe6\xf7\x9f\x69\x3f\xc9\x3b\x75\xd6\x80\xee\x87\x68\x83\xde\x2d\x18\xe4\x26\xdd\x1a\x02\xd8\xd2\x1d\xb6\xf1\x71\xf5\x63\x62\x0c\xd7\x35\x21\xc6\x75\xb4\xd5\x0f\x89\x08\x17\x13\x24\x07\xc2\x7c\x73\xe2\x17\x00\x12\x8a\xc9\x39\xdb\xf0\xc8\x6f\x1f\xf7\x99\xed\x8c\x67\x9c\xf2\x30\x5c\xd0\xd0\x0d\xc1\x15\x07\xa3\x1d\xf5\xd4\x92\x82\xfd\x9c\x5a\x11\x69\x3b\x02\x03\x01\x00\x01\x02\x82\x01\x00\x44\xe1\x5a\xfd\x8a\x18\xd5\x45\xb8\x4c\x76\x4b\x5c\x55\x97\x5f\x85\x2e\x26\x8d\xc8\x16\x46\x48\x3c\xd6\x7a\x84\x5d\x19\xf1\x83\xdf\x11\xbf\xb8\xc8\xef\x0a\x56\xbf\xdc\xd3\xeb\xed\x57\x7f\xb1\x93\x88\x5c\x65\xba\xe7\x29\x68\x9f\x2b\x7a\x92\xb0\x5f\x5a\xc7\x81\x0d\x68\xd8\x57\xee\x4d\x13\xbc\xf4\x3c\x12\x89\x18\x9a\xdb\x3a\xc4\x0a\xc0\x10\x35\x3b\xa5\xdc\xbe\x1c\x88\xc4\x84\xea\x12\x64\x4c\xb8\x71\x19\x93\x7e\x8e\x73\x1d\x9f\x04\x61\xa1\x97\x27\x82\x2e\xb6\x4d\x6a\x4f\xfb\xa4\xe5\xa7\x54\x94\xb5\xf1\x41\xc8\xa4\x3d\xa1\xe6\x4a\xf0\xdb\xbb\xc2\x91\x26\x9a\x0f\xbf\xdd\x57\x1e\x83\x5c\x9a\x7b\x28\x53\x1d\x2d\x44\x91\x1f\x02\x81\x7b\x6f\xb5\xf7\x48\x7d\xa0\x12\x22\xdb\xbf\xd9\x04\x17\xe4\x97\xf2\xac\x32\xf8\x70\xfa\x75\xe3\x5a\xb0\xef\x1f\x2d\x24\xb9\x26\x83\x33\xe7\x3c\x3c\xfb\x0b\xd8\x70\x33\x76\xb1\x1c\x1d\x38\x06\x0a\xdb\xbd\xd2\x34\x5e\xe6\xb1\x6f\x5d\x8f\x18\xac\x94\xd2\x0d\xee\x39\x0b\xa3\xb4\xcf\xf1\xe1\x91\x30\xcb\xce\xa5\x2f\xa9\xcc\x4f\xee\xe4\xdd\xee\x8a\x77\x0e\xd1\xbd\xcc\xb0\x11\x55\x15\x5e\x99\xf1\x02\x81\x81\x00\xd1\x75\x33\xe4\x31\xc2\xfc\x09\x6c\xf6\x04\x97\xc7\xa3\xb1\x88\x36\x26\xd8\x4e\x86\x2d\xb8\x99\x68\x97\xd8\x0b\xc6\xc3\xe7\x58\x49\xc3\x41\xcd\xcd\x33\x09\xa0\x90\xb2\x77\xfa\xa3\xb6\x71\x09\x33\x43\x0a\x6a\xd8\xc3\x36\xaf\xa9\x11\x54\x64\x77\x82\xf4\xf1\xe0\x12\x5a\xb8\x9f\x5a\x04\xb3\x29\xd4\xc6\xba\x4c\xdc\x04\x97\xfb\xb6\x7e\x1b\x89\x09\x0c\x8a\xb8\x6c\x9f\x2b\x91\x0d\x34\x18\x39\xf3\x38\xf9\xe6\xed\x29\x48\x30\xe4\x3c\x09\x15\x33\xe0\xb8\x2f\xd8\xfa\xf2\x6d\x1f\xf1\xee\x02\xc2\xb4\xf9\xf4\x63\x4b\xa5\x02\x81\x81\x00\xba\x14\x89\xff\x65\xb5\xe6\x52\x45\x23\x37\x5e\x0c\x62\xde\xe9\x7f\xa9\x05\xee\x28\x0d\x91\xb1\x99\xd6\x8b\xf8\x58\x50\x8b\xb1\xee\x57\xbd\x2b\x7b\xf0\x25\x03\xeb\xbc\x87\x73\xc8\xbf\x57\x16\xda\x49\x7a\x79\x82\x25\x99\x46\x9c\xb3\xd2\xd5\xb0\xae\xec\xeb\xbc\xd2\x4b\xae\xd0\x0a\x54\xcd\xad\x44\x90\x74\x79\xa2\x34\x73\x8a\x3a\x6c\x0b\x13\x20\x5d\xa4\xcc\x7b\xb4\x64\xcf\x61\x6e\xdf\xc1\x8c\xd4\x84\x22\xf1\x19\x32\x6d\xf1\x6f\xe1\x1e\xa5\xf6\x20\x6c\xc6\xa8\x9c\x4d\x8d\x59\xdf\x90\x71\x67\x1a\x48\xa3\x4b\x5f\x02\x81\x81\x00\x97\x4d\x8f\x7f\x7e\x86\xb8\x23\x62\xe7\x50\x28\x07\xd9\x72\x4b\xcf\xba\x3d\xb4\x73\x6e\xa1\x93\x87\x9f\x70\x3c\x09\x87\xc8\x1c\xd9\xa3\xc7\x6c\x0f\x97\x97\x93\xba\x12\x81\x62\xb7\x51\xf9\xd3\x48\x89\x5c\x04\x14\xb2\xe7\x54\xfa\xce\xfe\xe4\x58\x04\x6c\x46\x30\xb3\x71\x7f\x3d\xf4\xfb\xc2\x24\x2c\x84\xa5\x5d\x11\xed\xeb\x8f\xb3\xa2\xe2\xe7\x19\x77\x4a\xd9\xaf\xf5\x46\xb6\x50\x10\x5a\x93\xb9\xe3\x65\x79\xef\xc5\x4b\x55\xad\xf8\xc4\x22\xe1\xc7\xa9\xa5\x3e\x9a\xff\xf5\xde\x06\x98\x04\xbc\x7b\x98\xb7\x75\xe6\xd5\x02\x81\x80\x0a\x7f\x38\x1d\xa9\x2e\x2e\xb4\xfb\x63\x76\x2f\x1f\x01\xc0\xd3\x69\x39\x2e\xb5\x75\x9a\xf6\x5a\x0f\x74\x93\xe6\xc9\x8c\x99\xa4\xca\xee\x36\x24\xaa\xd4\x2c\x32\x61\x6c\xfc\x33\x22\xe2\xf0\x55\xc0\xb0\x9e\x71\x16\x4f\x6a\xab\x1a\x11\xe6\xd5\xd9\x26\xb5\x04\xc3\x5d\x15\x99\xe1\xf0\x83\x42\x2b\x01\x10\x29\x11\xe7\x7d\x8f\xfa\xff\x3a\xb3\x11\x3c\x25\x2c\x33\xc0\xd2\xb7\x51\x1f\x8c\xf2\xa0\x67\x82\x61\x85\xdb\x15\xf1\xcb\x53\xf0\x5c\xc1\xae\xd9\x08\x91\x3a\x4f\xae\xa9\x8d\x4c\xc1\x98\xd3\x5c\xde\x95\xb4\x68\x7f\x02\x81\x80\x7d\x3e\x6b\x2c\x16\xe8\x17\x2c\x27\x9c\xc5\xc5\xfb\x30\x1a\xf7\x32\x53\x93\xfe\xc1\xa0\x5d\xac\x7d\x6f\xba\x1b\x56\x7e\x34\xf6\xa7\x91\x1f\x39\x84\x1c\x94\x58\x13\xe2\xb9\xec\xb6\x24\xfe\x76\x35\x1b\xcc\x4f\x8e\x0d\x88\x5b\x5a\x6f\xb6\xa2\x0b\xc3\xb6\x98\x2d\xca\xce\xce\x26\xb4\x36\x37\x42\xa4\xc0\xa9\x85\x57\x4b\x6b\xc2\xed\x14\x96\xe5\xbc\x2b\x83\x32\xe9\x83\x24\x7f\x85\x74\x09\x3c\xfa\x45\xfd\x21\xeb\xd8\xa3\x02\xd2\x70\x0a\x9a\x9d\x7d\xe4\x39\xc4\x59\xc8\x16\x6f\xce\xd5\x1d\xea\x91\x4d\x12\x78\xc3\x30";