Clean up x509 name entries

This commit is contained in:
Steven Fackler 2016-10-31 22:43:05 -07:00
parent 50d8cca640
commit dc4098bdd8
5 changed files with 90 additions and 54 deletions

View File

@ -1660,8 +1660,11 @@ extern {
pub fn X509_NAME_free(x: *mut X509_NAME); pub fn X509_NAME_free(x: *mut X509_NAME);
pub fn X509_NAME_add_entry_by_txt(x: *mut X509_NAME, field: *const c_char, ty: c_int, bytes: *const c_uchar, len: c_int, loc: c_int, set: c_int) -> c_int; pub fn X509_NAME_add_entry_by_txt(x: *mut X509_NAME, field: *const c_char, ty: c_int, bytes: *const c_uchar, len: c_int, loc: c_int, set: c_int) -> c_int;
pub fn X509_NAME_get_index_by_NID(n: *mut X509_NAME, nid: c_int, last_pos: c_int) ->c_int; pub fn X509_NAME_get_index_by_NID(n: *mut X509_NAME, nid: c_int, last_pos: c_int) -> c_int;
pub fn X509_NAME_ENTRY_free(x: *mut X509_NAME_ENTRY);
pub fn ASN1_STRING_free(x: *mut ASN1_STRING);
pub fn ASN1_STRING_length(x: *const ASN1_STRING) -> c_int; pub fn ASN1_STRING_length(x: *const ASN1_STRING) -> c_int;
pub fn X509_STORE_CTX_free(ctx: *mut X509_STORE_CTX); pub fn X509_STORE_CTX_free(ctx: *mut X509_STORE_CTX);

View File

@ -1,9 +1,11 @@
use libc::c_long; use libc::c_long;
use std::{ptr, fmt}; use std::{ptr, fmt};
use std::slice;
use ffi; use ffi;
use {cvt, cvt_p}; use {cvt, cvt_p};
use bio::MemBio; use bio::MemBio;
use crypto::CryptoString;
use error::ErrorStack; use error::ErrorStack;
use types::{OpenSslType, Ref}; use types::{OpenSslType, Ref};
@ -35,3 +37,29 @@ impl Asn1Time {
Asn1Time::from_period(days as c_long * 60 * 60 * 24) Asn1Time::from_period(days as c_long * 60 * 60 * 24)
} }
} }
type_!(Asn1String, ffi::ASN1_STRING, ffi::ASN1_STRING_free);
impl Ref<Asn1String> {
pub fn as_utf8(&self) -> Result<CryptoString, ErrorStack> {
unsafe {
let mut ptr = ptr::null_mut();
let len = ffi::ASN1_STRING_to_UTF8(&mut ptr, self.as_ptr());
if len < 0 {
return Err(ErrorStack::get());
}
Ok(CryptoString::from_raw_parts(ptr, len as usize))
}
}
pub fn as_slice(&self) -> &[u8] {
unsafe { slice::from_raw_parts(ffi::ASN1_STRING_data(self.as_ptr()), self.len()) }
}
pub fn len(&self) -> usize {
unsafe {
ffi::ASN1_STRING_length(self.as_ptr()) as usize
}
}
}

View File

@ -253,6 +253,7 @@ fn setup_verify(ssl: &mut Ssl, domain: &str) -> Result<(), ErrorStack> {
#[cfg(not(any(ossl102, ossl110)))] #[cfg(not(any(ossl102, ossl110)))]
mod verify { mod verify {
use std::net::IpAddr; use std::net::IpAddr;
use std::str;
use nid; use nid;
use x509::{X509StoreContext, X509, GeneralNames, X509Name}; use x509::{X509StoreContext, X509, GeneralNames, X509Name};
@ -305,7 +306,12 @@ mod verify {
} }
fn verify_subject_name(domain: &str, subject_name: &Ref<X509Name>) -> bool { fn verify_subject_name(domain: &str, subject_name: &Ref<X509Name>) -> bool {
if let Some(pattern) = subject_name.text_by_nid(nid::COMMONNAME) { if let Some(pattern) = subject_name.entries_by_nid(nid::COMMONNAME).next() {
let pattern = match str::from_utf8(pattern.data().as_slice()) {
Ok(pattern) => pattern,
Err(_) => return false,
};
// Unlike with SANs, IP addresses in the subject name don't have a // Unlike with SANs, IP addresses in the subject name don't have a
// different encoding. We need to pass this down to matches_dns to // different encoding. We need to pass this down to matches_dns to
// disallow wildcard matches with bogus patterns like *.0.0.1 // disallow wildcard matches with bogus patterns like *.0.0.1

View File

@ -12,9 +12,8 @@ use std::slice;
use std::str; use std::str;
use {cvt, cvt_p}; use {cvt, cvt_p};
use asn1::Asn1Time; use asn1::{Asn1String, Asn1Time};
use bio::{MemBio, MemBioSlice}; use bio::{MemBio, MemBioSlice};
use crypto::CryptoString;
use hash::MessageDigest; use hash::MessageDigest;
use pkey::PKey; use pkey::PKey;
use rand::rand_bytes; use rand::rand_bytes;
@ -473,33 +472,49 @@ impl Borrow<Ref<X509>> for X509 {
type_!(X509Name, ffi::X509_NAME, ffi::X509_NAME_free); type_!(X509Name, ffi::X509_NAME, ffi::X509_NAME_free);
impl Ref<X509Name> { impl Ref<X509Name> {
pub fn text_by_nid(&self, nid: Nid) -> Option<CryptoString> { pub fn entries_by_nid<'a>(&'a self, nid: Nid) -> X509NameEntries<'a> {
X509NameEntries {
name: self,
nid: nid,
loc: -1,
}
}
}
pub struct X509NameEntries<'a> {
name: &'a Ref<X509Name>,
nid: Nid,
loc: c_int,
}
impl<'a> Iterator for X509NameEntries<'a> {
type Item = &'a Ref<X509NameEntry>;
fn next(&mut self) -> Option<&'a Ref<X509NameEntry>> {
unsafe { unsafe {
let loc = ffi::X509_NAME_get_index_by_NID(self.as_ptr(), nid.as_raw(), -1); self.loc = ffi::X509_NAME_get_index_by_NID(self.name.as_ptr(),
if loc == -1 { self.nid.as_raw(),
self.loc);
if self.loc == -1 {
return None; return None;
} }
let ne = ffi::X509_NAME_get_entry(self.as_ptr(), loc); let entry = ffi::X509_NAME_get_entry(self.name.as_ptr(), self.loc);
if ne.is_null() { assert!(!entry.is_null());
return None;
Some(Ref::from_ptr(entry))
} }
let asn1_str = ffi::X509_NAME_ENTRY_get_data(ne);
if asn1_str.is_null() {
return None;
} }
}
let mut str_from_asn1: *mut u8 = ptr::null_mut(); type_!(X509NameEntry, ffi::X509_NAME_ENTRY, ffi::X509_NAME_ENTRY_free);
let len = ffi::ASN1_STRING_to_UTF8(&mut str_from_asn1, asn1_str);
if len < 0 { impl Ref<X509NameEntry> {
return None; pub fn data(&self) -> &Ref<Asn1String> {
} unsafe {
let data = ffi::X509_NAME_ENTRY_get_data(self.as_ptr());
assert!(!str_from_asn1.is_null()); Ref::from_ptr(data)
Some(CryptoString::from_raw_parts(str_from_asn1, len as usize))
} }
} }
} }

View File

@ -113,58 +113,42 @@ fn test_save_der() {
#[test] #[test]
fn test_subject_read_cn() { fn test_subject_read_cn() {
let cert = include_bytes!("../../test/cert.pem"); let cert = include_bytes!("../../test/cert.pem");
let cert = X509::from_pem(cert).ok().expect("Failed to load PEM"); let cert = X509::from_pem(cert).unwrap();
let subject = cert.subject_name(); let subject = cert.subject_name();
let cn = match subject.text_by_nid(nid::COMMONNAME) { let cn = subject.entries_by_nid(nid::COMMONNAME).next().unwrap();
Some(x) => x, assert_eq!(cn.data().as_slice(), b"foobar.com")
None => panic!("Failed to read CN from cert"),
};
assert_eq!(&cn as &str, "foobar.com")
} }
#[test] #[test]
fn test_nid_values() { fn test_nid_values() {
let cert = include_bytes!("../../test/nid_test_cert.pem"); let cert = include_bytes!("../../test/nid_test_cert.pem");
let cert = X509::from_pem(cert).ok().expect("Failed to load PEM"); let cert = X509::from_pem(cert).unwrap();
let subject = cert.subject_name(); let subject = cert.subject_name();
let cn = match subject.text_by_nid(nid::COMMONNAME) { let cn = subject.entries_by_nid(nid::COMMONNAME).next().unwrap();
Some(x) => x, assert_eq!(cn.data().as_slice(), b"example.com");
None => panic!("Failed to read CN from cert"),
};
assert_eq!(&cn as &str, "example.com");
let email = match subject.text_by_nid(nid::PKCS9_EMAILADDRESS) { let email = subject.entries_by_nid(nid::PKCS9_EMAILADDRESS).next().unwrap();
Some(x) => x, assert_eq!(email.data().as_slice(), b"test@example.com");
None => panic!("Failed to read subject email address from cert"),
};
assert_eq!(&email as &str, "test@example.com");
let friendly = match subject.text_by_nid(nid::FRIENDLYNAME) { let friendly = subject.entries_by_nid(nid::FRIENDLYNAME).next().unwrap();
Some(x) => x, assert_eq!(&*friendly.data().as_utf8().unwrap(), "Example");
None => panic!("Failed to read subject friendly name from cert"),
};
assert_eq!(&friendly as &str, "Example");
} }
#[test] #[test]
fn test_nid_uid_value() { fn test_nid_uid_value() {
let cert = include_bytes!("../../test/nid_uid_test_cert.pem"); let cert = include_bytes!("../../test/nid_uid_test_cert.pem");
let cert = X509::from_pem(cert).ok().expect("Failed to load PEM"); let cert = X509::from_pem(cert).unwrap();
let subject = cert.subject_name(); let subject = cert.subject_name();
let cn = match subject.text_by_nid(nid::USERID) { let cn = subject.entries_by_nid(nid::USERID).next().unwrap();
Some(x) => x, assert_eq!(cn.data().as_slice(), b"this is the userId");
None => panic!("Failed to read UID from cert"),
};
assert_eq!(&cn as &str, "this is the userId");
} }
#[test] #[test]
fn test_subject_alt_name() { fn test_subject_alt_name() {
let cert = include_bytes!("../../test/alt_name_cert.pem"); let cert = include_bytes!("../../test/alt_name_cert.pem");
let cert = X509::from_pem(cert).ok().expect("Failed to load PEM"); let cert = X509::from_pem(cert).unwrap();
let subject_alt_names = cert.subject_alt_names().unwrap(); let subject_alt_names = cert.subject_alt_names().unwrap();
assert_eq!(3, subject_alt_names.len()); assert_eq!(3, subject_alt_names.len());