1use crate::der::{self, DerIterator, FromDer, CONSTRUCTED, CONTEXT_SPECIFIC};
16use crate::error::{DerTypeId, Error};
17use crate::subject_name::GeneralName;
18
19pub(crate) struct Extension<'a> {
20 pub(crate) critical: bool,
21 pub(crate) id: untrusted::Input<'a>,
22 pub(crate) value: untrusted::Input<'a>,
23}
24
25impl<'a> Extension<'a> {
26 pub(crate) fn unsupported(&self) -> Result<(), Error> {
27 match self.critical {
28 true => Err(Error::UnsupportedCriticalExtension),
29 false => Ok(()),
30 }
31 }
32}
33
34impl<'a> FromDer<'a> for Extension<'a> {
35 fn from_der(reader: &mut untrusted::Reader<'a>) -> Result<Self, Error> {
36 let id = der::expect_tag(reader, der::Tag::OID)?;
37 let critical = bool::from_der(reader)?;
38 let value = der::expect_tag(reader, der::Tag::OctetString)?;
39 Ok(Extension {
40 id,
41 critical,
42 value,
43 })
44 }
45
46 const TYPE_ID: DerTypeId = DerTypeId::Extension;
47}
48
49pub(crate) fn set_extension_once<T>(
50 destination: &mut Option<T>,
51 parser: impl Fn() -> Result<T, Error>,
52) -> Result<(), Error> {
53 match destination {
54 Some(..) => Err(Error::ExtensionValueInvalid),
57 None => {
58 *destination = Some(parser()?);
59 Ok(())
60 }
61 }
62}
63
64pub(crate) fn remember_extension(
65 extension: &Extension<'_>,
66 mut handler: impl FnMut(u8) -> Result<(), Error>,
67) -> Result<(), Error> {
68 static ID_CE: [u8; 2] = oid![2, 5, 29];
71
72 if extension.id.len() != ID_CE.len() + 1
73 || !extension.id.as_slice_less_safe().starts_with(&ID_CE)
74 {
75 return extension.unsupported();
76 }
77
78 let last_octet = *extension.id.as_slice_less_safe().last().unwrap();
80 handler(last_octet)
81}
82
83pub(crate) enum DistributionPointName<'a> {
88 NameRelativeToCrlIssuer,
90 FullName(DerIterator<'a, GeneralName<'a>>),
92}
93
94impl<'a> FromDer<'a> for DistributionPointName<'a> {
95 fn from_der(reader: &mut untrusted::Reader<'a>) -> Result<Self, Error> {
96 const FULL_NAME_TAG: u8 = CONTEXT_SPECIFIC | CONSTRUCTED;
100 const NAME_RELATIVE_TO_CRL_ISSUER_TAG: u8 = CONTEXT_SPECIFIC | CONSTRUCTED | 1;
101
102 let (tag, value) = der::read_tag_and_get_value(reader)?;
103 match tag {
104 FULL_NAME_TAG => Ok(DistributionPointName::FullName(DerIterator::new(value))),
105 NAME_RELATIVE_TO_CRL_ISSUER_TAG => Ok(DistributionPointName::NameRelativeToCrlIssuer),
106 _ => Err(Error::BadDer),
107 }
108 }
109
110 const TYPE_ID: DerTypeId = DerTypeId::DistributionPointName;
111}