use {
crate::rfc4519::{
OID_COMMON_NAME, OID_COUNTRY_NAME, OID_LOCALITY_NAME, OID_ORGANIZATIONAL_UNIT_NAME,
OID_ORGANIZATION_NAME, OID_STATE_PROVINCE_NAME,
},
bcder::{
decode::{BytesSource, Constructed, DecodeError, Source},
encode,
encode::{PrimitiveContent, Values},
string::{Ia5String, PrintableString, Utf8String},
Captured, Mode, OctetString, Oid, Tag,
},
std::{
fmt::{Debug, Display, Formatter},
io::Write,
ops::{Deref, DerefMut},
str::FromStr,
},
};
pub type GeneralNames = Vec<GeneralName>;
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum GeneralName {
OtherName(AnotherName),
Rfc822Name(Ia5String),
DnsName(Ia5String),
X400Address(OrAddress),
DirectoryName(Name),
EdiPartyName(EdiPartyName),
UniformResourceIdentifier(Ia5String),
IpAddress(OctetString),
RegisteredId(Oid),
}
impl GeneralName {
pub fn take_from<S: Source>(cons: &mut Constructed<S>) -> Result<Self, DecodeError<S::Error>> {
if let Some(name) =
cons.take_opt_constructed_if(Tag::CTX_0, |cons| AnotherName::take_from(cons))?
{
Ok(Self::OtherName(name))
} else if let Some(name) =
cons.take_opt_constructed_if(Tag::CTX_1, |cons| Ia5String::take_from(cons))?
{
Ok(Self::Rfc822Name(name))
} else if let Some(name) =
cons.take_opt_constructed_if(Tag::CTX_2, |cons| Ia5String::take_from(cons))?
{
Ok(Self::DnsName(name))
} else if let Some(name) =
cons.take_opt_constructed_if(Tag::CTX_3, |cons| OrAddress::take_from(cons))?
{
Ok(Self::X400Address(name))
} else if let Some(name) =
cons.take_opt_constructed_if(Tag::CTX_4, |cons| Name::take_from(cons))?
{
Ok(Self::DirectoryName(name))
} else if let Some(name) =
cons.take_opt_constructed_if(Tag::CTX_5, |cons| EdiPartyName::take_from(cons))?
{
Ok(Self::EdiPartyName(name))
} else if let Some(name) =
cons.take_opt_constructed_if(Tag::CTX_6, |cons| Ia5String::take_from(cons))?
{
Ok(Self::UniformResourceIdentifier(name))
} else if let Some(name) =
cons.take_opt_constructed_if(Tag::ctx(7), |cons| OctetString::take_from(cons))?
{
Ok(Self::IpAddress(name))
} else if let Some(name) =
cons.take_opt_constructed_if(Tag::ctx(8), |cons| Oid::take_from(cons))?
{
Ok(Self::RegisteredId(name))
} else {
Err(cons.content_err("unexpected GeneralName variant"))
}
}
pub fn encode_ref(&self) -> impl Values + '_ {
match self {
Self::OtherName(name) => (
Some(name.explicit(Tag::CTX_0)),
None,
None,
None,
None,
None,
None,
None,
),
Self::Rfc822Name(name) => (
None,
Some(name.encode_ref_as(Tag::CTX_1)),
None,
None,
None,
None,
None,
None,
),
Self::DnsName(name) => (
None,
None,
Some(name.encode_ref_as(Tag::CTX_2)),
None,
None,
None,
None,
None,
),
Self::X400Address(_name) => {
unimplemented!()
}
Self::DirectoryName(name) => (
None,
None,
None,
Some(name.encode_ref_as(Tag::CTX_4)),
None,
None,
None,
None,
),
Self::EdiPartyName(name) => (
None,
None,
None,
None,
Some(name.encode_ref_as(Tag::CTX_5)),
None,
None,
None,
),
Self::UniformResourceIdentifier(name) => (
None,
None,
None,
None,
None,
Some(name.encode_ref_as(Tag::CTX_6)),
None,
None,
),
Self::IpAddress(name) => (
None,
None,
None,
None,
None,
None,
Some(name.encode_ref_as(Tag::ctx(7))),
None,
),
Self::RegisteredId(name) => (
None,
None,
None,
None,
None,
None,
None,
Some(name.encode_ref_as(Tag::ctx(8))),
),
}
}
}
#[derive(Clone, Debug)]
pub struct AnotherName {
pub type_id: Oid,
pub value: Captured,
}
impl PartialEq for AnotherName {
fn eq(&self, other: &Self) -> bool {
self.type_id == other.type_id && self.value.as_slice() == other.value.as_slice()
}
}
impl Eq for AnotherName {}
impl AnotherName {
pub fn take_from<S: Source>(cons: &mut Constructed<S>) -> Result<Self, DecodeError<S::Error>> {
cons.take_sequence(|cons| {
let type_id = Oid::take_from(cons)?;
let value = cons.take_constructed_if(Tag::CTX_0, |cons| cons.capture_all())?;
Ok(Self { type_id, value })
})
}
}
impl Values for AnotherName {
fn encoded_len(&self, mode: Mode) -> usize {
encode::sequence((self.type_id.encode_ref(), &self.value)).encoded_len(mode)
}
fn write_encoded<W: Write>(&self, mode: Mode, target: &mut W) -> Result<(), std::io::Error> {
encode::sequence((self.type_id.encode_ref(), &self.value)).write_encoded(mode, target)
}
}
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct EdiPartyName {
pub name_assigner: Option<DirectoryString>,
pub party_name: DirectoryString,
}
impl EdiPartyName {
pub fn take_from<S: Source>(cons: &mut Constructed<S>) -> Result<Self, DecodeError<S::Error>> {
cons.take_sequence(|cons| {
let name_assigner =
cons.take_opt_constructed_if(Tag::CTX_0, |cons| DirectoryString::take_from(cons))?;
let party_name =
cons.take_constructed_if(Tag::CTX_1, |cons| DirectoryString::take_from(cons))?;
Ok(Self {
name_assigner,
party_name,
})
})
}
pub fn encode_ref(&self) -> impl Values + '_ {
encode::sequence((
self.name_assigner
.as_ref()
.map(|name_assigner| name_assigner.encode_ref()),
self.party_name.encode_ref(),
))
}
pub fn encode_ref_as(&self, tag: Tag) -> impl Values + '_ {
encode::sequence_as(
tag,
(
self.name_assigner
.as_ref()
.map(|name_assigner| name_assigner.encode_ref()),
self.party_name.encode_ref(),
),
)
}
}
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum DirectoryString {
TeletexString,
PrintableString(PrintableString),
UniversalString,
Utf8String(Utf8String),
BmpString,
}
impl DirectoryString {
pub fn take_from<S: Source>(cons: &mut Constructed<S>) -> Result<Self, DecodeError<S::Error>> {
cons.take_value(|tag, content| {
if tag == Tag::PRINTABLE_STRING {
Ok(Self::PrintableString(PrintableString::from_content(
content,
)?))
} else if tag == Tag::UTF8_STRING {
Ok(Self::Utf8String(Utf8String::from_content(content)?))
} else {
Err(content
.content_err("only decoding of PrintableString and UTF8String is implemented"))
}
})
}
pub fn encode_ref(&self) -> impl Values + '_ {
match self {
Self::PrintableString(ps) => (Some(ps.encode_ref()), None),
Self::Utf8String(s) => (None, Some(s.encode_ref())),
_ => unimplemented!(),
}
}
}
impl Display for DirectoryString {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
let str = match self {
Self::PrintableString(s) => s.to_string(),
Self::Utf8String(s) => s.to_string(),
_ => unimplemented!(),
};
write!(f, "{}", str)
}
}
impl Values for DirectoryString {
fn encoded_len(&self, mode: Mode) -> usize {
self.encode_ref().encoded_len(mode)
}
fn write_encoded<W: Write>(&self, mode: Mode, target: &mut W) -> Result<(), std::io::Error> {
self.encode_ref().write_encoded(mode, target)
}
}
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum Name {
RdnSequence(RdnSequence),
}
impl Name {
pub fn take_from<S: Source>(cons: &mut Constructed<S>) -> Result<Self, DecodeError<S::Error>> {
Ok(Self::RdnSequence(RdnSequence::take_from(cons)?))
}
pub fn encode_ref(&self) -> impl Values + '_ {
match self {
Self::RdnSequence(seq) => seq.encode_ref(),
}
}
pub fn encode_ref_as(&self, tag: Tag) -> impl Values + '_ {
match self {
Self::RdnSequence(seq) => seq.encode_ref_as(tag),
}
}
pub fn iter_rdn(&self) -> impl Iterator<Item = &RelativeDistinguishedName> {
self.0.iter()
}
pub fn iter_attributes(&self) -> impl Iterator<Item = &AttributeTypeAndValue> {
self.0.iter().flat_map(|rdn| rdn.iter())
}
pub fn iter_mut_attributes(&mut self) -> impl Iterator<Item = &mut AttributeTypeAndValue> {
self.0.iter_mut().flat_map(|rdn| rdn.iter_mut())
}
pub fn iter_by_oid(&self, oid: Oid) -> impl Iterator<Item = &AttributeTypeAndValue> {
self.iter_attributes().filter(move |atv| atv.typ == oid)
}
pub fn iter_mut_by_oid(
&mut self,
oid: Oid,
) -> impl Iterator<Item = &mut AttributeTypeAndValue> {
self.iter_mut_attributes().filter(move |atv| atv.typ == oid)
}
pub fn iter_common_name(&self) -> impl Iterator<Item = &AttributeTypeAndValue> {
self.iter_by_oid(Oid(OID_COMMON_NAME.as_ref().into()))
}
pub fn iter_country(&self) -> impl Iterator<Item = &AttributeTypeAndValue> {
self.iter_by_oid(Oid(OID_COUNTRY_NAME.as_ref().into()))
}
pub fn iter_locality(&self) -> impl Iterator<Item = &AttributeTypeAndValue> {
self.iter_by_oid(Oid(OID_LOCALITY_NAME.as_ref().into()))
}
pub fn iter_state_province(&self) -> impl Iterator<Item = &AttributeTypeAndValue> {
self.iter_by_oid(Oid(OID_STATE_PROVINCE_NAME.as_ref().into()))
}
pub fn iter_organization(&self) -> impl Iterator<Item = &AttributeTypeAndValue> {
self.iter_by_oid(Oid(OID_ORGANIZATION_NAME.as_ref().into()))
}
pub fn iter_organizational_unit(&self) -> impl Iterator<Item = &AttributeTypeAndValue> {
self.iter_by_oid(Oid(OID_ORGANIZATIONAL_UNIT_NAME.as_ref().into()))
}
pub fn find_attribute(&self, oid: Oid) -> Option<&AttributeTypeAndValue> {
self.iter_by_oid(oid).next()
}
pub fn find_first_attribute_string(
&self,
oid: Oid,
) -> Result<Option<String>, DecodeError<<BytesSource as Source>::Error>> {
if let Some(atv) = self.find_attribute(oid) {
Ok(Some(atv.to_string()?))
} else {
Ok(None)
}
}
pub fn user_friendly_str(&self) -> Result<String, DecodeError<<BytesSource as Source>::Error>> {
let mut fields = vec![];
for cn in self.iter_common_name() {
fields.push(format!("CN={}", cn.to_string()?));
}
for ou in self.iter_organizational_unit() {
fields.push(format!("OU={}", ou.to_string()?));
}
for o in self.iter_organization() {
fields.push(format!("O={}", o.to_string()?));
}
for l in self.iter_locality() {
fields.push(format!("L={}", l.to_string()?));
}
for state in self.iter_state_province() {
fields.push(format!("S={}", state.to_string()?));
}
for c in self.iter_country() {
fields.push(format!("C={}", c.to_string()?));
}
Ok(fields.join(", "))
}
pub fn append_printable_string(
&mut self,
oid: Oid,
value: &str,
) -> Result<(), bcder::string::CharSetError> {
let mut rdn = RelativeDistinguishedName::default();
rdn.push(AttributeTypeAndValue::new_printable_string(oid, value)?);
self.0.push(rdn);
Ok(())
}
pub fn append_utf8_string(
&mut self,
oid: Oid,
value: &str,
) -> Result<(), bcder::string::CharSetError> {
let mut rdn = RelativeDistinguishedName::default();
rdn.push(AttributeTypeAndValue::new_utf8_string(oid, value)?);
self.0.push(rdn);
Ok(())
}
pub fn append_common_name_utf8_string(
&mut self,
value: &str,
) -> Result<(), bcder::string::CharSetError> {
self.append_utf8_string(Oid(OID_COMMON_NAME.as_ref().into()), value)
}
pub fn append_country_utf8_string(
&mut self,
value: &str,
) -> Result<(), bcder::string::CharSetError> {
self.append_utf8_string(Oid(OID_COUNTRY_NAME.as_ref().into()), value)
}
pub fn append_organization_utf8_string(
&mut self,
value: &str,
) -> Result<(), bcder::string::CharSetError> {
self.append_utf8_string(Oid(OID_ORGANIZATION_NAME.as_ref().into()), value)
}
pub fn append_organizational_unit_utf8_string(
&mut self,
value: &str,
) -> Result<(), bcder::string::CharSetError> {
self.append_utf8_string(Oid(OID_ORGANIZATIONAL_UNIT_NAME.as_ref().into()), value)
}
}
impl Default for Name {
fn default() -> Self {
Self::RdnSequence(RdnSequence::default())
}
}
impl Deref for Name {
type Target = RdnSequence;
fn deref(&self) -> &Self::Target {
match self {
Self::RdnSequence(seq) => seq,
}
}
}
impl DerefMut for Name {
fn deref_mut(&mut self) -> &mut Self::Target {
match self {
Self::RdnSequence(seq) => seq,
}
}
}
#[derive(Clone, Debug, Default, Eq, PartialEq)]
pub struct RdnSequence(Vec<RelativeDistinguishedName>);
impl Deref for RdnSequence {
type Target = Vec<RelativeDistinguishedName>;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl DerefMut for RdnSequence {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.0
}
}
impl RdnSequence {
pub fn take_from<S: Source>(cons: &mut Constructed<S>) -> Result<Self, DecodeError<S::Error>> {
cons.take_sequence(|cons| {
let mut values = Vec::new();
while let Some(value) = RelativeDistinguishedName::take_opt_from(cons)? {
values.push(value);
}
Ok(Self(values))
})
}
pub fn encode_ref(&self) -> impl Values + '_ {
encode::sequence(&self.0)
}
pub fn encode_ref_as(&self, tag: Tag) -> impl Values + '_ {
encode::sequence_as(tag, &self.0)
}
}
pub type DistinguishedName = RdnSequence;
#[derive(Clone, Debug, Default, Eq, PartialEq)]
pub struct RelativeDistinguishedName(Vec<AttributeTypeAndValue>);
impl Deref for RelativeDistinguishedName {
type Target = Vec<AttributeTypeAndValue>;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl DerefMut for RelativeDistinguishedName {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.0
}
}
impl RelativeDistinguishedName {
pub fn take_opt_from<S: Source>(
cons: &mut Constructed<S>,
) -> Result<Option<Self>, DecodeError<S::Error>> {
cons.take_opt_set(|cons| {
let mut values = Vec::new();
while let Some(value) = AttributeTypeAndValue::take_opt_from(cons)? {
values.push(value);
}
Ok(Self(values))
})
}
pub fn encode_ref(&self) -> impl Values + '_ {
encode::set(&self.0)
}
}
impl Values for RelativeDistinguishedName {
fn encoded_len(&self, mode: Mode) -> usize {
self.encode_ref().encoded_len(mode)
}
fn write_encoded<W: Write>(&self, mode: Mode, target: &mut W) -> Result<(), std::io::Error> {
self.encode_ref().write_encoded(mode, target)
}
}
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct OrAddress {}
impl OrAddress {
pub fn take_from<S: Source>(cons: &mut Constructed<S>) -> Result<Self, DecodeError<S::Error>> {
Err(cons.content_err("parsing of OrAddress not implemented"))
}
}
#[derive(Clone)]
pub struct AttributeTypeAndValue {
pub typ: AttributeType,
pub value: AttributeValue,
}
impl Debug for AttributeTypeAndValue {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
let mut s = f.debug_struct("AttributeTypeAndValue");
s.field("type", &format_args!("{}", self.typ));
s.field("value", &self.value);
s.finish()
}
}
impl AttributeTypeAndValue {
pub fn take_opt_from<S: Source>(
cons: &mut Constructed<S>,
) -> Result<Option<Self>, DecodeError<S::Error>> {
cons.take_opt_sequence(|cons| {
let typ = AttributeType::take_from(cons)?;
let value = cons.capture_all()?;
Ok(Self {
typ,
value: value.into(),
})
})
}
pub fn encode_ref(&self) -> impl Values + '_ {
encode::sequence((self.typ.encode_ref(), self.value.deref()))
}
pub fn to_string(&self) -> Result<String, DecodeError<<BytesSource as Source>::Error>> {
self.value.to_string()
}
pub fn new_printable_string(oid: Oid, s: &str) -> Result<Self, bcder::string::CharSetError> {
Ok(Self {
typ: oid,
value: AttributeValue::new_printable_string(s)?,
})
}
pub fn new_utf8_string(oid: Oid, s: &str) -> Result<Self, bcder::string::CharSetError> {
Ok(Self {
typ: oid,
value: AttributeValue::new_utf8_string(s)?,
})
}
pub fn set_utf8_string_value(&mut self, s: &str) -> Result<(), bcder::string::CharSetError> {
self.value.set_utf8_string_value(s)
}
}
impl PartialEq for AttributeTypeAndValue {
fn eq(&self, other: &Self) -> bool {
self.typ == other.typ && self.value.as_slice() == other.value.as_slice()
}
}
impl Eq for AttributeTypeAndValue {}
impl Values for AttributeTypeAndValue {
fn encoded_len(&self, mode: Mode) -> usize {
self.encode_ref().encoded_len(mode)
}
fn write_encoded<W: Write>(&self, mode: Mode, target: &mut W) -> Result<(), std::io::Error> {
self.encode_ref().write_encoded(mode, target)
}
}
pub type AttributeType = Oid;
#[derive(Clone)]
pub struct AttributeValue(Captured);
impl Debug for AttributeValue {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
f.write_fmt(format_args!("{}", hex::encode(self.0.as_slice())))
}
}
impl AttributeValue {
pub fn new_printable_string(s: &str) -> Result<Self, bcder::string::CharSetError> {
let mut slf = Self(Captured::empty(Mode::Der));
slf.set_printable_string_value(s)?;
Ok(slf)
}
pub fn new_utf8_string(s: &str) -> Result<Self, bcder::string::CharSetError> {
let mut slf = Self(Captured::empty(Mode::Der));
slf.set_utf8_string_value(s)?;
Ok(slf)
}
pub fn to_string(&self) -> Result<String, DecodeError<<BytesSource as Source>::Error>> {
self.0.clone().decode(|cons| {
if let Some(s) = cons.take_opt_value_if(Tag::NUMERIC_STRING, |content| {
bcder::NumericString::from_content(content)
})? {
Ok(s.to_string())
} else if let Some(s) = cons.take_opt_value_if(Tag::PRINTABLE_STRING, |content| {
bcder::PrintableString::from_content(content)
})? {
Ok(s.to_string())
} else if let Some(s) = cons.take_opt_value_if(Tag::UTF8_STRING, |content| {
bcder::Utf8String::from_content(content)
})? {
Ok(s.to_string())
} else if let Some(s) = cons.take_opt_value_if(Tag::IA5_STRING, |content| {
bcder::Ia5String::from_content(content)
})? {
Ok(s.to_string())
} else {
Ok(DirectoryString::take_from(cons)?.to_string())
}
})
}
pub fn set_printable_string_value(
&mut self,
value: &str,
) -> Result<(), bcder::string::CharSetError> {
let ps = DirectoryString::PrintableString(PrintableString::from_str(value)?);
let captured = bcder::Captured::from_values(Mode::Der, ps);
self.0 = captured;
Ok(())
}
pub fn set_utf8_string_value(
&mut self,
value: &str,
) -> Result<(), bcder::string::CharSetError> {
let ds = DirectoryString::Utf8String(Utf8String::from_str(value)?);
let captured = bcder::Captured::from_values(Mode::Der, ds);
self.0 = captured;
Ok(())
}
}
impl Deref for AttributeValue {
type Target = Captured;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl From<Captured> for AttributeValue {
fn from(v: Captured) -> Self {
Self(v)
}
}