Merge pull request #60 from vhbit/cert-gen-cleanup (40c69488) · Commits · Public Repositories / Rust Openssl

src/asn1/mod.rs

+47 −0

Original line number	Diff line number	Diff line
		use libc::{c_long};
		use std::ptr;

		use ffi;
		use ssl::error::{SslError};


		pub struct Asn1Time {
		handle: *mut ffi::ASN1_TIME,
		owned: bool
		}

		impl Asn1Time {
		/// Wraps existing ASN1_TIME and takes ownership
		pub fn new(handle: *mut ffi::ASN1_TIME) -> Asn1Time {
		Asn1Time {
		handle: handle,
		owned: true
		}
		}

		fn new_with_period(period: u64) -> Result<Asn1Time, SslError> {
		let handle = unsafe {
		try_ssl_null!(ffi::X509_gmtime_adj(ptr::null_mut(),
		period as c_long))
		};
		Ok(Asn1Time::new(handle))
		}

		/// Creates a new time on specified interval in days from now
		pub fn days_from_now(days: uint) -> Result<Asn1Time, SslError> {
		Asn1Time::new_with_period(days as u64 * 60 * 60 * 24)
		}

		/// Returns raw handle
		pub unsafe fn get_handle(&self) -> *mut ffi::ASN1_TIME {
		return self.handle
		}
		}

		impl Drop for Asn1Time {
		fn drop(&mut self) {
		if self.owned {
		unsafe { ffi::ASN1_TIME_free(self.handle) };
		}
		}
		}

src/ffi.rs

+2 −0

Original line number	Diff line number	Diff line
		@@ -192,6 +192,7 @@ pub unsafe fn BN_is_zero(a: *mut BIGNUM) -> c_int { bn_is_zero(a) }
		extern "C" {
		pub fn ASN1_INTEGER_set(dest: *mut ASN1_INTEGER, value: c_long) -> c_int;
		pub fn ASN1_STRING_type_new(ty: c_int) -> *mut ASN1_STRING;
		pub fn ASN1_TIME_free(tm: *mut ASN1_TIME);

		pub fn BIO_free_all(a: *mut BIO);
		pub fn BIO_new(type_: const BIO_METHOD) -> mut BIO;
		@@ -402,6 +403,7 @@ extern "C" {

		pub fn X509_add_ext(x: mut X509, ext: mut X509_EXTENSION, loc: c_int) -> c_int;
		pub fn X509_digest(x: mut X509, digest: const EVP_MD, buf: mut c_char, len: mut c_uint) -> c_int;
		pub fn X509_free(x: *mut X509);
		pub fn X509_get_serialNumber(x: mut X509) -> mut ASN1_INTEGER;
		pub fn X509_get_subject_name(x: mut X509) -> mut X509_NAME;
		pub fn X509_gmtime_adj(time: mut ASN1_TIME, adj: c_long) -> mut ASN1_TIME;

src/lib.rs

+2 −2

Original line number	Diff line number	Diff line
		#![feature(struct_variant, macro_rules)]
		#![feature(struct_variant, macro_rules, unsafe_destructor)]
		#![crate_name="openssl"]
		#![crate_type="rlib"]
		#![crate_type="dylib"]
		@@ -11,7 +11,7 @@ extern crate sync;

		mod macros;

		mod asn1;
		pub mod asn1;
		pub mod bn;
		pub mod bio;
		pub mod crypto;

src/macros.rs

+5 −1

Original line number	Diff line number	Diff line
		@@ -26,7 +26,11 @@ macro_rules! try_ssl{

		/// Shortcut return with SSL if got a null result
		macro_rules! try_ssl_null{
		($e:expr) => (try_ssl_if!($e == ptr::null_mut()))
		($e:expr) => ({
		let t = $e;
		try_ssl_if!(t == ptr::null_mut());
		t
		})
		}

src/x509/mod.rs

+47 −28

Original line number	Diff line number	Diff line
		@@ -2,6 +2,7 @@ use libc::{c_int, c_long, c_uint};
		use std::mem;
		use std::ptr;

		use asn1::{Asn1Time};
		use bio::{MemBio};
		use crypto::hash::{HashType, evpmd, SHA1};
		use crypto::pkey::{PKey};
		@@ -39,7 +40,7 @@ impl X509StoreContext {
		if ptr.is_null() {
		None
		} else {
		Some(X509 { ctx: Some(self), x509: ptr })
		Some(X509 { ctx: Some(self), handle: ptr, owned: false })
		}
		}
		}
		@@ -184,16 +185,21 @@ impl X509Generator {

		fn add_extension(x509: *mut ffi::X509, extension: c_int, value: &str) -> Result<(), SslError> {
		unsafe {
		// FIXME: RAII
		let mut ctx: ffi::X509V3_CTX = mem::zeroed();
		ffi::X509V3_set_ctx(&mut ctx, x509, x509,
		ptr::null_mut(), ptr::null_mut(), 0);
		let ext = value.with_c_str(\|value\|
		ffi::X509V3_EXT_conf_nid(ptr::null_mut(), mem::transmute(&ctx), extension, mem::transmute(value)));
		try_ssl_null!(ext);
		try_ssl!(ffi::X509_add_ext(x509, ext, -1));
		ffi::X509V3_EXT_conf_nid(ptr::null_mut(),
		mem::transmute(&ctx),
		extension,
		mem::transmute(value)));

		let mut success = false;
		if ext != ptr::null_mut() {
		success = ffi::X509_add_ext(x509, ext, -1) != 0;
		ffi::X509_EXTENSION_free(ext);
		Ok(())
		}
		lift_ssl_if!(!success)
		}
		}

		@@ -222,44 +228,47 @@ impl X509Generator {
		let mut p_key = PKey::new();
		p_key.gen(self.bits);

		// FIXME: all allocated resources should be correctly
		// dropped in case of failure
		unsafe {
		let x509 = ffi::X509_new();
		try_ssl_null!(x509);
		try_ssl!(ffi::X509_set_version(x509, 2));
		try_ssl!(ffi::ASN1_INTEGER_set(ffi::X509_get_serialNumber(x509), X509Generator::random_serial()));

		let not_before = ffi::X509_gmtime_adj(ptr::null_mut(), 0);
		try_ssl_null!(not_before);
		let x509 = X509 { handle: x509, ctx: None, owned: true};

		try_ssl!(ffi::X509_set_version(x509.handle, 2));
		try_ssl!(ffi::ASN1_INTEGER_set(ffi::X509_get_serialNumber(x509.handle), X509Generator::random_serial()));

		let not_after = ffi::X509_gmtime_adj(ptr::null_mut(), 606024*self.days as i64);
		try_ssl_null!(not_after);
		let not_before = try!(Asn1Time::days_from_now(0));
		let not_after = try!(Asn1Time::days_from_now(self.days));

		try_ssl!(ffi::X509_set_notBefore(x509, mem::transmute(not_before)));
		try_ssl!(ffi::X509_set_notAfter(x509, mem::transmute(not_after)));
		try_ssl!(ffi::X509_set_notBefore(x509.handle, mem::transmute(not_before.get_handle())));
		// If prev line succeded - ownership should go to cert
		mem::forget(not_before);

		try_ssl!(ffi::X509_set_pubkey(x509, p_key.get_handle()));
		try_ssl!(ffi::X509_set_notAfter(x509.handle, mem::transmute(not_after.get_handle())));
		// If prev line succeded - ownership should go to cert
		mem::forget(not_after);

		let name = ffi::X509_get_subject_name(x509);
		try_ssl!(ffi::X509_set_pubkey(x509.handle, p_key.get_handle()));

		let name = ffi::X509_get_subject_name(x509.handle);
		try_ssl_null!(name);

		try!(X509Generator::add_name(name, "CN", self.CN.as_slice()));
		ffi::X509_set_issuer_name(x509, name);
		ffi::X509_set_issuer_name(x509.handle, name);

		if self.key_usage.len() > 0 {
		try!(X509Generator::add_extension(x509, ffi::NID_key_usage,
		try!(X509Generator::add_extension(x509.handle, ffi::NID_key_usage,
		self.key_usage.to_str().as_slice()));
		}

		if self.ext_key_usage.len() > 0 {
		try!(X509Generator::add_extension(x509, ffi::NID_ext_key_usage,
		try!(X509Generator::add_extension(x509.handle, ffi::NID_ext_key_usage,
		self.ext_key_usage.to_str().as_slice()));
		}

		let (hash_fn, _) = evpmd(self.hash_type);
		try_ssl!(ffi::X509_sign(x509, p_key.get_handle(), hash_fn));
		Ok((X509 { x509: x509, ctx: None }, p_key))
		try_ssl!(ffi::X509_sign(x509.handle, p_key.get_handle(), hash_fn));
		Ok((x509, p_key))
		}
		}
		}
		@@ -268,12 +277,13 @@ impl X509Generator {
		/// A public key certificate
		pub struct X509<'ctx> {
		ctx: Option<&'ctx X509StoreContext>,
		x509: *mut ffi::X509
		handle: *mut ffi::X509,
		owned: bool
		}

		impl<'ctx> X509<'ctx> {
		pub fn subject_name<'a>(&'a self) -> X509Name<'a> {
		let name = unsafe { ffi::X509_get_subject_name(self.x509) };
		let name = unsafe { ffi::X509_get_subject_name(self.handle) };
		X509Name { x509: self, name: name }
		}

		@@ -283,7 +293,7 @@ impl<'ctx> X509<'ctx> {
		let v: Vec<u8> = Vec::from_elem(len, 0);
		let act_len: c_uint = 0;
		let res = unsafe {
		ffi::X509_digest(self.x509, evp, mem::transmute(v.as_ptr()),
		ffi::X509_digest(self.handle, evp, mem::transmute(v.as_ptr()),
		mem::transmute(&act_len))
		};

		@@ -305,13 +315,22 @@ impl<'ctx> X509<'ctx> {
		let mut mem_bio = try!(MemBio::new());
		unsafe {
		try_ssl!(ffi::PEM_write_bio_X509(mem_bio.get_handle(),
		self.x509));
		self.handle));
		}
		let buf = try!(mem_bio.read_to_end().map_err(StreamError));
		writer.write(buf.as_slice()).map_err(StreamError)
		}
		}

		#[unsafe_destructor]
		impl<'ctx> Drop for X509<'ctx> {
		fn drop(&mut self) {
		if self.owned {
		unsafe { ffi::X509_free(self.handle) };
		}
		}
		}

		#[allow(dead_code)]
		pub struct X509Name<'x> {
		x509: &'x X509<'x>,