Add support for building proofs using a local recursive resolver
[dnssec-prover] / src / lib.rs
1 //! The DNS provides a single, global, hierarchical namespace with (when DNSSEC is used)
2 //! cryptographic guarantees on all of its data.
3 //!
4 //! This makes it incredibly powerful for resolving human-readable names into arbitrary, secured
5 //! data.
6 //!
7 //! Unlike TLS, this cryptographic security provides transferable proofs which can convince an
8 //! offline device, using simple cryptographic primitives and a single root trusted key, of the
9 //! validity of DNS data.
10 //!
11 //! This crate implements the creation and validation of such proofs, using the format from RFC
12 //! 9102 to create transferable proofs of DNS entries.
13 //!
14 //! It is no-std (but requires `alloc`) and seeks to have minimal dependencies and a reasonably
15 //! conservative MSRV policy, allowing it to be used in as many places as possible.
16
17 #![deny(missing_docs)]
18
19 #![cfg_attr(not(feature = "std"), no_std)]
20 extern crate alloc;
21
22 use alloc::vec::Vec;
23 use alloc::vec;
24
25 use ring::signature;
26
27 pub mod rr;
28 use rr::*;
29
30 mod ser;
31 use ser::{bytes_to_rsa_pk, parse_rr, write_name};
32
33 #[cfg(feature = "std")]
34 pub mod query;
35
36 /// Gets the trusted root anchors
37 ///
38 /// These are available at <https://data.iana.org/root-anchors/root-anchors.xml>
39 pub fn root_hints() -> Vec<DS> {
40         #[allow(unused_mut)]
41         let mut res = vec![DS {
42                 name: ".".try_into().unwrap(), key_tag: 19036, alg: 8, digest_type: 2,
43                 digest: hex_lit::hex!("49AAC11D7B6F6446702E54A1607371607A1A41855200FD2CE1CDDE32F24E8FB5").to_vec(),
44         }, DS {
45                 name: ".".try_into().unwrap(), key_tag: 20326, alg: 8, digest_type: 2,
46                 digest: hex_lit::hex!("E06D44B80B8F1D39A95C0B0D7C65D08458E880409BBC683457104237C7F8EC8D").to_vec(),
47         }];
48         // In tests, add the trust anchor from RFC 9102
49         #[cfg(test)]
50         res.push(DS {
51                 name: ".".try_into().unwrap(), key_tag: 47005, alg: 13, digest_type: 2,
52                 digest: hex_lit::hex!("2eb6e9f2480126691594d649a5a613de3052e37861634641bb568746f2ffc4d4").to_vec(),
53         });
54         res
55 }
56
57 /// Parse a stream of [`RR`]s from the format described in [RFC 9102](https://www.rfc-editor.org/rfc/rfc9102.html).
58 ///
59 /// Note that this is only the series of `AuthenticationChain` records, and does not read the
60 /// `ExtSupportLifetime` field at the start of a `DnssecChainExtension`.
61 pub fn parse_rr_stream(mut inp: &[u8]) -> Result<Vec<RR>, ()> {
62         let mut res = Vec::with_capacity(32);
63         while !inp.is_empty() {
64                 res.push(parse_rr(&mut inp)?);
65         }
66         Ok(res)
67 }
68
69 /// Writes the given resource record in its wire encoding to the given `Vec`.
70 ///
71 /// An [RFC 9102](https://www.rfc-editor.org/rfc/rfc9102.html) `AuthenticationChain` is simply a
72 /// series of such records with no additional bytes in between.
73 pub fn write_rr<RR: Record>(rr: &RR, ttl: u32, out: &mut Vec<u8>) {
74         write_name(out, rr.name());
75         out.extend_from_slice(&rr.ty().to_be_bytes());
76         out.extend_from_slice(&1u16.to_be_bytes()); // The INternet class
77         out.extend_from_slice(&ttl.to_be_bytes());
78         rr.write_u16_len_prefixed_data(out);
79 }
80
81 #[derive(Debug, PartialEq)]
82 /// An error when validating DNSSEC signatures or other data
83 pub enum ValidationError {
84         /// An algorithm used in signing was not supported.
85         ///
86         /// In general DNS usage the resulting data should be used anyway, as we were able to verify
87         /// that a zone wished to use the unsupported algorithm.
88         ///
89         /// However, in cases where signing is mandatory, this can be treated as an error.
90         UnsupportedAlgorithm,
91         /// The provided data was invalid or signatures did not validate.
92         Invalid,
93 }
94
95 // TODO: return the validity period
96 fn verify_rrsig<'a, RR: Record, Keys>(sig: &RRSig, dnskeys: Keys, mut records: Vec<&RR>)
97 -> Result<(), ValidationError>
98 where Keys: IntoIterator<Item = &'a DnsKey> {
99         for record in records.iter() {
100                 if sig.ty != record.ty() { return Err(ValidationError::Invalid); }
101         }
102         for dnskey in dnskeys.into_iter() {
103                 if dnskey.key_tag() == sig.key_tag {
104                         // Protocol must be 3, otherwise its not DNSSEC
105                         if dnskey.protocol != 3 { continue; }
106                         // The ZONE flag must be set if we're going to validate RRs with this key.
107                         if dnskey.flags & 0b1_0000_0000 == 0 { continue; }
108                         if dnskey.alg != sig.alg { continue; }
109
110                         // TODO: Check orig_ttl somehow?
111
112                         let mut signed_data = Vec::with_capacity(2048);
113                         signed_data.extend_from_slice(&sig.ty.to_be_bytes());
114                         signed_data.extend_from_slice(&sig.alg.to_be_bytes());
115                         signed_data.extend_from_slice(&sig.labels.to_be_bytes()); // Check this somehow?
116                         signed_data.extend_from_slice(&sig.orig_ttl.to_be_bytes());
117                         signed_data.extend_from_slice(&sig.expiration.to_be_bytes()); // Return this and inception
118                         signed_data.extend_from_slice(&sig.inception.to_be_bytes());
119                         signed_data.extend_from_slice(&sig.key_tag.to_be_bytes());
120                         write_name(&mut signed_data, &sig.key_name);
121
122                         records.sort();
123
124                         for record in records.iter() {
125                                 let periods = record.name().as_str().chars().filter(|c| *c == '.').count();
126                                 let labels = sig.labels.into();
127                                 if periods != 1 && periods != labels {
128                                         if periods < labels { return Err(ValidationError::Invalid); }
129                                         let signed_name = record.name().as_str().splitn(periods - labels + 1, ".").last();
130                                         debug_assert!(signed_name.is_some());
131                                         if let Some(name) = signed_name {
132                                                 signed_data.extend_from_slice(b"\x01*");
133                                                 write_name(&mut signed_data, name);
134                                         } else { return Err(ValidationError::Invalid); }
135                                 } else {
136                                         write_name(&mut signed_data, record.name());
137                                 }
138                                 signed_data.extend_from_slice(&record.ty().to_be_bytes());
139                                 signed_data.extend_from_slice(&1u16.to_be_bytes()); // The INternet class
140                                 signed_data.extend_from_slice(&sig.orig_ttl.to_be_bytes());
141                                 record.write_u16_len_prefixed_data(&mut signed_data);
142                         }
143
144                         match sig.alg {
145                                 8|10 => {
146                                         let alg = if sig.alg == 8 {
147                                                 &signature::RSA_PKCS1_1024_8192_SHA256_FOR_LEGACY_USE_ONLY
148                                         } else {
149                                                 &signature::RSA_PKCS1_1024_8192_SHA512_FOR_LEGACY_USE_ONLY
150                                         };
151                                         bytes_to_rsa_pk(&dnskey.pubkey).map_err(|_| ValidationError::Invalid)?
152                                                 .verify(alg, &signed_data, &sig.signature)
153                                                 .map_err(|_| ValidationError::Invalid)?;
154                                 },
155                                 13|14 => {
156                                         let alg = if sig.alg == 13 {
157                                                 &signature::ECDSA_P256_SHA256_FIXED
158                                         } else {
159                                                 &signature::ECDSA_P384_SHA384_FIXED
160                                         };
161
162                                         // Add 0x4 identifier to the ECDSA pubkey as expected by ring.
163                                         let mut key = Vec::with_capacity(dnskey.pubkey.len() + 1);
164                                         key.push(0x4);
165                                         key.extend_from_slice(&dnskey.pubkey);
166
167                                         signature::UnparsedPublicKey::new(alg, &key)
168                                                 .verify(&signed_data, &sig.signature)
169                                                 .map_err(|_| ValidationError::Invalid)?;
170                                 },
171                                 15 => {
172                                         signature::UnparsedPublicKey::new(&signature::ED25519, &dnskey.pubkey)
173                                                 .verify(&signed_data, &sig.signature)
174                                                 .map_err(|_| ValidationError::Invalid)?;
175                                 },
176                                 _ => return Err(ValidationError::UnsupportedAlgorithm),
177                         }
178
179                         return Ok(());
180                 }
181         }
182         Err(ValidationError::Invalid)
183 }
184
185 fn verify_dnskey_rrsig<'a, T, I>(sig: &RRSig, dses: T, records: Vec<&DnsKey>)
186 -> Result<(), ValidationError>
187 where T: IntoIterator<IntoIter = I>, I: Iterator<Item = &'a DS> + Clone {
188         let mut validated_dnskeys = Vec::with_capacity(records.len());
189         let dses = dses.into_iter();
190
191         let mut had_known_digest_type = false;
192         let mut had_ds = false;
193         for ds in dses.clone() {
194                 had_ds = true;
195                 if ds.digest_type == 2 || ds.digest_type == 4 {
196                         had_known_digest_type = true;
197                         break;
198                 }
199         }
200         if !had_ds { return Err(ValidationError::Invalid); }
201         if !had_known_digest_type { return Err(ValidationError::UnsupportedAlgorithm); }
202
203         for dnskey in records.iter() {
204                 for ds in dses.clone() {
205                         if ds.digest_type != 2 && ds.digest_type != 4 { continue; }
206                         if ds.alg != dnskey.alg { continue; }
207                         if dnskey.key_tag() == ds.key_tag {
208                                 let alg = match ds.digest_type {
209                                         2 => &ring::digest::SHA256,
210                                         4 => &ring::digest::SHA384,
211                                         _ => continue,
212                                 };
213                                 let mut ctx = ring::digest::Context::new(alg);
214                                 write_name(&mut ctx, &dnskey.name);
215                                 ctx.update(&dnskey.flags.to_be_bytes());
216                                 ctx.update(&dnskey.protocol.to_be_bytes());
217                                 ctx.update(&dnskey.alg.to_be_bytes());
218                                 ctx.update(&dnskey.pubkey);
219                                 let hash = ctx.finish();
220                                 if hash.as_ref() == &ds.digest {
221                                         validated_dnskeys.push(*dnskey);
222                                         break;
223                                 }
224                         }
225                 }
226         }
227         verify_rrsig(sig, validated_dnskeys.iter().map(|k| *k), records)
228 }
229
230 /// Verifies the given set of resource records.
231 ///
232 /// Given a set of arbitrary records, this attempts to validate DNSSEC data from the [`root_hints`]
233 /// through to any supported non-DNSSEC record types.
234 ///
235 /// All records which could be validated are returned, though if an error is found validating any
236 /// contained record, only `Err` will be returned.
237 pub fn verify_rr_stream<'a>(inp: &'a [RR]) -> Result<Vec<&'a RR>, ValidationError> {
238         let mut zone = ".";
239         let mut res = Vec::new();
240         let mut pending_ds_sets = Vec::with_capacity(1);
241         'next_zone: while zone == "." || !pending_ds_sets.is_empty() {
242                 let mut found_unsupported_alg = false;
243                 let next_ds_set;
244                 if let Some((next_zone, ds_set)) = pending_ds_sets.pop() {
245                         next_ds_set = Some(ds_set);
246                         zone = next_zone;
247                 } else {
248                         debug_assert_eq!(zone, ".");
249                         next_ds_set = None;
250                 }
251
252                 for rrsig in inp.iter()
253                         .filter_map(|rr| if let RR::RRSig(sig) = rr { Some(sig) } else { None })
254                         .filter(|rrsig| rrsig.name.as_str() == zone && rrsig.ty == DnsKey::TYPE)
255                 {
256                         let dnskeys = inp.iter()
257                                 .filter_map(|rr| if let RR::DnsKey(dnskey) = rr { Some(dnskey) } else { None })
258                                 .filter(move |dnskey| dnskey.name.as_str() == zone);
259                         let dnskeys_verified = if zone == "." {
260                                 verify_dnskey_rrsig(rrsig, &root_hints(), dnskeys.clone().collect())
261                         } else {
262                                 debug_assert!(next_ds_set.is_some());
263                                 if next_ds_set.is_none() { break 'next_zone; }
264                                 verify_dnskey_rrsig(rrsig, next_ds_set.clone().unwrap(), dnskeys.clone().collect())
265                         };
266                         if dnskeys_verified.is_ok() {
267                                 for rrsig in inp.iter()
268                                         .filter_map(|rr| if let RR::RRSig(sig) = rr { Some(sig) } else { None })
269                                         .filter(move |rrsig| rrsig.key_name.as_str() == zone && rrsig.name.as_str() != zone)
270                                 {
271                                         if !rrsig.name.ends_with(zone) { return Err(ValidationError::Invalid); }
272                                         let signed_records = inp.iter()
273                                                 .filter(|rr| rr.name() == &rrsig.name && rr.ty() == rrsig.ty);
274                                         verify_rrsig(rrsig, dnskeys.clone(), signed_records.clone().collect())?;
275                                         match rrsig.ty {
276                                                 // RRSigs shouldn't cover child `DnsKey`s or other `RRSig`s
277                                                 RRSig::TYPE|DnsKey::TYPE => return Err(ValidationError::Invalid),
278                                                 DS::TYPE => {
279                                                         if !pending_ds_sets.iter().any(|(pending_zone, _)| pending_zone == &rrsig.name.as_str()) {
280                                                                 pending_ds_sets.push((
281                                                                         &rrsig.name,
282                                                                         signed_records.filter_map(|rr|
283                                                                                 if let RR::DS(ds) = rr { Some(ds) }
284                                                                                 else { debug_assert!(false, "We already filtered by type"); None })
285                                                                 ));
286                                                         }
287                                                 },
288                                                 _ => {
289                                                         for record in signed_records {
290                                                                 if !res.contains(&record) { res.push(record); }
291                                                         }
292                                                 },
293                                         }
294                                 }
295                                 continue 'next_zone;
296                         } else if dnskeys_verified == Err(ValidationError::UnsupportedAlgorithm) {
297                                 // There may be redundant signatures by different keys, where one we don't supprt
298                                 // and another we do. Ignore ones we don't support, but if there are no more,
299                                 // return UnsupportedAlgorithm
300                                 found_unsupported_alg = true;
301                         } else {
302                                 // We don't explicitly handle invalid signatures here, instead we move on to the
303                                 // next RRSig (if there is one) and return `Invalid` if no `RRSig`s match.
304                         }
305                 }
306                 // No RRSigs were able to verify our DnsKey set
307                 if found_unsupported_alg {
308                         return Err(ValidationError::UnsupportedAlgorithm);
309                 } else {
310                         return Err(ValidationError::Invalid);
311                 }
312         }
313         if res.is_empty() { Err(ValidationError::Invalid) }
314         else { Ok(res) }
315 }
316
317 #[cfg(test)]
318 mod tests {
319         #![allow(deprecated)]
320
321         use super::*;
322
323         use alloc::borrow::ToOwned;
324
325         use hex_conservative::FromHex;
326         use rand::seq::SliceRandom;
327
328         fn root_dnskey() -> (Vec<DnsKey>, Vec<RR>) {
329                 let dnskeys = vec![DnsKey {
330                         name: ".".try_into().unwrap(), flags: 256, protocol: 3, alg: 8,
331                         pubkey: base64::decode("AwEAAentCcIEndLh2QSK+pHFq/PkKCwioxt75d7qNOUuTPMo0Fcte/NbwDPbocvbZ/eNb5RV/xQdapaJASQ/oDLsqzD0H1+JkHNuuKc2JLtpMxg4glSE4CnRXT2CnFTW5IwOREL+zeqZHy68OXy5ngW5KALbevRYRg/q2qFezRtCSQ0knmyPwgFsghVYLKwi116oxwEU5yZ6W7npWMxt5Z+Qs8diPNWrS5aXLgJtrWUGIIuFfuZwXYziGRP/z3o1EfMo9zZU19KLopkoLXX7Ls/diCXdSEdJXTtFA8w0/OKQviuJebfKscoElCTswukVZ1VX5gbaFEo2xWhHJ9Uo63wYaTk=").unwrap(),
332                 }, DnsKey {
333                         name: ".".try_into().unwrap(), flags: 257, protocol: 3, alg: 8,
334                         pubkey: base64::decode("AwEAAaz/tAm8yTn4Mfeh5eyI96WSVexTBAvkMgJzkKTOiW1vkIbzxeF3+/4RgWOq7HrxRixHlFlExOLAJr5emLvN7SWXgnLh4+B5xQlNVz8Og8kvArMtNROxVQuCaSnIDdD5LKyWbRd2n9WGe2R8PzgCmr3EgVLrjyBxWezF0jLHwVN8efS3rCj/EWgvIWgb9tarpVUDK/b58Da+sqqls3eNbuv7pr+eoZG+SrDK6nWeL3c6H5Apxz7LjVc1uTIdsIXxuOLYA4/ilBmSVIzuDWfdRUfhHdY6+cn8HFRm+2hM8AnXGXws9555KrUB5qihylGa8subX2Nn6UwNR1AkUTV74bU=").unwrap(),
335                 }];
336                 let dnskey_rrsig = RRSig {
337                         name: ".".try_into().unwrap(), ty: DnsKey::TYPE, alg: 8, labels: 0, orig_ttl: 172800,
338                         expiration: 1708473600, inception: 1706659200, key_tag: 20326, key_name: ".".try_into().unwrap(),
339                         signature: base64::decode("ZO8LbjtwAiVkkBzOnGbiI/3ilGUPmmJpagsLSBVbIZRG6o/8a+hUZpIPTvk5ERZ1rAW4x0YxKAU8qtaHQpKIp3qYA6u97DYytVD7RdtXKHmGYAvR6QbD5eVTkCw1Sz705rJxbwt6+YM5OBweSUAy5Glo6JSQPDQwRDwj/bV2fLRhJbvfsBgxqaXJA0SaE/ceyvK8gB2NIaguTJNrztr2TENrHxi86OKOuHYDHthOW0TFoPfr19qj/P2eEC6dYniTVovUwHT7e+Hqrb05dJF4mI4ZjaIb5mFf8i5RehT1aRlnb3CLiwJ01bEjrRBo3xUn5I3PkCnglHhx3EvkO73OzA==").unwrap(),
340                 };
341                 let root_hints = root_hints();
342                 verify_dnskey_rrsig(&dnskey_rrsig, &root_hints, dnskeys.iter().collect()).unwrap();
343                 let rrs = vec![dnskeys[0].clone().into(), dnskeys[1].clone().into(), dnskey_rrsig.into()];
344                 (dnskeys, rrs)
345         }
346
347         fn com_dnskey() -> (Vec<DnsKey>, Vec<RR>) {
348                 let root_dnskeys = root_dnskey().0;
349                 let mut com_ds = vec![DS {
350                         name: "com.".try_into().unwrap(), key_tag: 19718, alg: 13, digest_type: 2,
351                         digest: Vec::from_hex("8ACBB0CD28F41250A80A491389424D341522D946B0DA0C0291F2D3D771D7805A").unwrap(),
352                 }];
353                 let ds_rrsig = RRSig {
354                         name: "com.".try_into().unwrap(), ty: DS::TYPE, alg: 8, labels: 1, orig_ttl: 86400,
355                         expiration: 1708189200, inception: 1707062400, key_tag: 30903, key_name: ".".try_into().unwrap(),
356                         signature: base64::decode("vwMOBBwqRBdlmGZB+0FKfyMSignEtpYW9sD4TzPW2E+wdbF7O7epR5cmKmvcv0RUJdM0dGC/QmhCfgf/yqw1Xp7TpmPaYzaruW70hjGXZJO2nY3G6stUVe4S7lM2CzHL7nbbpaB5B+iSu6Ua9dZ+nyKrxfB7855HBLCLrHrkMGxWQiEPTallXXS8tEM1Y2XrsuzAQu2vZ2D2ClhFspFbPwwOdw+G6+NsZ8PnIfTkCj6DuKcgbdxjmGaYmw/6hVt9OU3kGCOBaJaEy4LrD8Kwzfu4S7axMwTKP4y4c5Y/E4k/mVAW0cuUtv549HaDfD2V0CvW1bDl6PqRkOiVsqM/lA==").unwrap(),
357                 };
358                 verify_rrsig(&ds_rrsig, &root_dnskeys, com_ds.iter().collect()).unwrap();
359                 let dnskeys = vec![DnsKey {
360                         name: "com.".try_into().unwrap(), flags: 256, protocol: 3, alg: 13,
361                         pubkey: base64::decode("5i9qjJgyH+9MBz7VO269/srLQB/xRRllyUoVq8oLBZshPe4CGzDSFGnXAM3L/QPzB9ULpJuuy7jcxmBZ5Ebo7A==").unwrap(),
362                 }, DnsKey {
363                         name: "com.".try_into().unwrap(), flags: 257, protocol: 3, alg: 13,
364                         pubkey: base64::decode("tx8EZRAd2+K/DJRV0S+hbBzaRPS/G6JVNBitHzqpsGlz8huE61Ms9ANe6NSDLKJtiTBqfTJWDAywEp1FCsEINQ==").unwrap(),
365                 }];
366                 let dnskey_rrsig = RRSig {
367                         name: "com.".try_into().unwrap(), ty: DnsKey::TYPE, alg: 13, labels: 1, orig_ttl: 86400,
368                         expiration: 1707750155, inception: 1706453855, key_tag: 19718, key_name: "com.".try_into().unwrap(),
369                         signature: base64::decode("ZFGChM7QfJt0QSqVWerWnG5pMjpL1pXyJAmuHe8dHI/olmaNCxm+mqNHv9i3AploFY6JoNtiHmeBiC6zuFj/ZQ==").unwrap(),
370                 };
371                 verify_dnskey_rrsig(&dnskey_rrsig, &com_ds, dnskeys.iter().collect()).unwrap();
372                 let rrs = vec![com_ds.pop().unwrap().into(), ds_rrsig.into(),
373                         dnskeys[0].clone().into(), dnskeys[1].clone().into(), dnskey_rrsig.into()];
374                 (dnskeys, rrs)
375         }
376
377         fn mattcorallo_dnskey() -> (Vec<DnsKey>, Vec<RR>) {
378                 let com_dnskeys = com_dnskey().0;
379                 let mut mattcorallo_ds = vec![DS {
380                         name: "mattcorallo.com.".try_into().unwrap(), key_tag: 25630, alg: 13, digest_type: 2,
381                         digest: Vec::from_hex("DC608CA62BE89B3B9DB1593F9A59930D24FBA79D486E19C88A7792711EC00735").unwrap(),
382                 }];
383                 let ds_rrsig = RRSig {
384                         name: "mattcorallo.com.".try_into().unwrap(), ty: DS::TYPE, alg: 13, labels: 2, orig_ttl: 86400,
385                         expiration: 1707631252, inception: 1707022252, key_tag: 4534, key_name: "com.".try_into().unwrap(),
386                         signature: base64::decode("M7Fk+CjfLz6hRsY5iSuw5bwc2OqlS3XtKH8FDs7lcbhEiR63n+DzOF0I8L+3k06SXFnE89uuofQECzWmAyef6Q==").unwrap(),
387                 };
388                 verify_rrsig(&ds_rrsig, &com_dnskeys, mattcorallo_ds.iter().collect()).unwrap();
389                 let dnskeys = vec![DnsKey {
390                         name: "mattcorallo.com.".try_into().unwrap(), flags: 257, protocol: 3, alg: 13,
391                         pubkey: base64::decode("8BP51Etiu4V6cHvGCYqwNqCip4pvHChjEgkgG4zpdDvO9YRcTGuV/p71hAUut2/qEdxqXfUOT/082BJ/Z089DA==").unwrap(),
392                 }, DnsKey {
393                         name: "mattcorallo.com.".try_into().unwrap(), flags: 256, protocol: 3, alg: 13,
394                         pubkey: base64::decode("AhUlQ8qk7413R0m4zKfTDHb/FQRlKag+ncGXxNxT+qTzSZTb9E5IGjo9VCEp6+IMqqpkd4GrXpN9AzDvlcU9Ig==").unwrap(),
395                 }];
396                 let dnskey_rrsig = RRSig {
397                         name: "mattcorallo.com.".try_into().unwrap(), ty: DnsKey::TYPE, alg: 13, labels: 2, orig_ttl: 604800,
398                         expiration: 1708278650, inception: 1707063650, key_tag: 25630, key_name: "mattcorallo.com.".try_into().unwrap(),
399                         signature: base64::decode("nyVDwG+la8d5dyWgB7m+H3BQwCvTWLQ/kAqNruMzdLmn9B3VC9u/rvM/ortEu0WPbA1FZWJbRKpF1Ohkj3ltNw==").unwrap(),
400                 };
401                 verify_dnskey_rrsig(&dnskey_rrsig, &mattcorallo_ds, dnskeys.iter().collect()).unwrap();
402                 let rrs = vec![mattcorallo_ds.pop().unwrap().into(), ds_rrsig.into(),
403                         dnskeys[0].clone().into(), dnskeys[1].clone().into(), dnskey_rrsig.into()];
404                 (dnskeys, rrs)
405         }
406
407         fn mattcorallo_txt_record() -> (Txt, RRSig) {
408                 let txt_resp = Txt {
409                         name: "matt.user._bitcoin-payment.mattcorallo.com.".try_into().unwrap(),
410                         data: "bitcoin:?b12=lno1qsgqmqvgm96frzdg8m0gc6nzeqffvzsqzrxqy32afmr3jn9ggkwg3egfwch2hy0l6jut6vfd8vpsc3h89l6u3dm4q2d6nuamav3w27xvdmv3lpgklhg7l5teypqz9l53hj7zvuaenh34xqsz2sa967yzqkylfu9xtcd5ymcmfp32h083e805y7jfd236w9afhavqqvl8uyma7x77yun4ehe9pnhu2gekjguexmxpqjcr2j822xr7q34p078gzslf9wpwz5y57alxu99s0z2ql0kfqvwhzycqq45ehh58xnfpuek80hw6spvwrvttjrrq9pphh0dpydh06qqspp5uq4gpyt6n9mwexde44qv7lstzzq60nr40ff38u27un6y53aypmx0p4qruk2tf9mjwqlhxak4znvna5y".to_owned().into_bytes(),
411                 };
412                 let txt_rrsig = RRSig {
413                         name: "matt.user._bitcoin-payment.mattcorallo.com.".try_into().unwrap(),
414                         ty: Txt::TYPE, alg: 13, labels: 5, orig_ttl: 3600, expiration: 1708123318,
415                         inception: 1706908318, key_tag: 47959, key_name: "mattcorallo.com.".try_into().unwrap(),
416                         signature: base64::decode("mgU6iwyMWO0w9nj2Gmt1+RmaIJIU3KO7DWVZiCD1bmU9e9zNefXCtnWOC2HtwjUsn/QYkWluvuSfYpBrt1IjpQ==").unwrap(),
417                 };
418                 (txt_resp, txt_rrsig)
419         }
420
421         fn matcorallo_dnskey() -> (Vec<DnsKey>, Vec<RR>) {
422                 let com_dnskeys = com_dnskey().0;
423                 let mut matcorallo_ds = vec![DS {
424                         name: "matcorallo.com.".try_into().unwrap(), key_tag: 24930, alg: 13, digest_type: 2,
425                         digest: Vec::from_hex("693E990CBB1CE1095E387092D3C04BCE907C008891F32A88D41D3ECB129E5E23").unwrap(),
426                 }];
427                 let ds_rrsig = RRSig {
428                         name: "matcorallo.com.".try_into().unwrap(), ty: DS::TYPE, alg: 13, labels: 2, orig_ttl: 86400,
429                         expiration: 1707628636, inception: 1707019636, key_tag: 4534, key_name: "com.".try_into().unwrap(),
430                         signature: base64::decode("l9b+DhtnJSIzR6y4Bwx+0L9kep77UNCBoTg74RTSL6oMrQd8w4OobHxzwDyXqnLfyxVP18V+AnQp4DdJ2nUW1g==").unwrap(),
431                 };
432                 verify_rrsig(&ds_rrsig, &com_dnskeys, matcorallo_ds.iter().collect()).unwrap();
433                 let dnskeys = vec![DnsKey {
434                         name: "matcorallo.com.".try_into().unwrap(), flags: 257, protocol: 3, alg: 13,
435                         pubkey: base64::decode("pfO3ow3SrKhLS7AMEi3b5W9P28nCOB9vryxfSXhqMcXFP1x9V4xAt0/JLr0zNodsqRD/8d9Yhu4Wf3hnSlaavw==").unwrap(),
436                 }, DnsKey {
437                         name: "matcorallo.com.".try_into().unwrap(), flags: 256, protocol: 3, alg: 13,
438                         pubkey: base64::decode("OO6LQTV1mnRsFgn6YQoyeo/SDqS3eajfVv8WGQVnuSYO/bTS9St1tJiox2fgU6wRWDU3chhjz1Pj0unKUAQKig==").unwrap(),
439                 }];
440                 let dnskey_rrsig = RRSig {
441                         name: "matcorallo.com.".try_into().unwrap(), ty: DnsKey::TYPE, alg: 13, labels: 2, orig_ttl: 604800,
442                         expiration: 1708309135, inception: 1707094135, key_tag: 24930, key_name: "matcorallo.com.".try_into().unwrap(),
443                         signature: base64::decode("2MKg3bTn9zf4ThwCoKRFadqD6l1D6SuLksRieKxFC0QQnzUOCRgZSK2/IlT0DMEoM0+mGrJZo7UG79UILMGUyg==").unwrap(),
444                 };
445                 verify_dnskey_rrsig(&dnskey_rrsig, &matcorallo_ds, dnskeys.iter().collect()).unwrap();
446                 let rrs = vec![matcorallo_ds.pop().unwrap().into(), ds_rrsig.into(),
447                         dnskeys[0].clone().into(), dnskeys[1].clone().into(), dnskey_rrsig.into()];
448                 (dnskeys, rrs)
449         }
450
451         fn matcorallo_txt_record() -> (Txt, RRSig) {
452                 let txt_resp = Txt {
453                         name: "txt_test.matcorallo.com.".try_into().unwrap(),
454                         data: "dnssec_prover_test".to_owned().into_bytes(),
455                 };
456                 let txt_rrsig = RRSig {
457                         name: "txt_test.matcorallo.com.".try_into().unwrap(),
458                         ty: Txt::TYPE, alg: 13, labels: 3, orig_ttl: 30, expiration: 1708319203,
459                         inception: 1707104203, key_tag: 34530, key_name: "matcorallo.com.".try_into().unwrap(),
460                         signature: base64::decode("4vaE5Jex2VvIT39JpuMNT7Ds7O0OfzTik5f8WcRRxO0IJnGAO16syAsNUkNkNqsMYknnjHDF0lI4agszgzdpsw==").unwrap(),
461                 };
462                 (txt_resp, txt_rrsig)
463         }
464
465         fn matcorallo_cname_record() -> (CName, RRSig) {
466                 let cname_resp = CName {
467                         name: "cname_test.matcorallo.com.".try_into().unwrap(),
468                         canonical_name: "txt_test.matcorallo.com.".try_into().unwrap(),
469                 };
470                 let cname_rrsig = RRSig {
471                         name: "cname_test.matcorallo.com.".try_into().unwrap(),
472                         ty: CName::TYPE, alg: 13, labels: 3, orig_ttl: 30, expiration: 1708319203,
473                         inception: 1707104203, key_tag: 34530, key_name: "matcorallo.com.".try_into().unwrap(),
474                         signature: base64::decode("5HIrmEotbVb95umE6SX3NrPboKsthdcY8b7DdaYQZzm0Nj5m2VgcfOmEPJYS8o1xE4GvGGF4sdfSy3Uw7TibBg==").unwrap(),
475                 };
476                 (cname_resp, cname_rrsig)
477         }
478
479         fn matcorallo_wildcard_record() -> (Txt, RRSig) {
480                 let txt_resp = Txt {
481                         name: "test.wildcard_test.matcorallo.com.".try_into().unwrap(),
482                         data: "wildcard_test".to_owned().into_bytes(),
483                 };
484                 let txt_rrsig = RRSig {
485                         name: "test.wildcard_test.matcorallo.com.".try_into().unwrap(),
486                         ty: Txt::TYPE, alg: 13, labels: 3, orig_ttl: 30, expiration: 1708321778,
487                         inception: 1707106778, key_tag: 34530, key_name: "matcorallo.com.".try_into().unwrap(),
488                         signature: base64::decode("vdnXunPY4CnbW/BL8VOOR9o33+dqyKA/4h+u5VM7NjB30Shp8L8gL5UwE0k7TKRNgHC8j3TqEPEmNMIHz87Z4Q==").unwrap(),
489                 };
490                 (txt_resp, txt_rrsig)
491         }
492
493         fn matcorallo_cname_wildcard_record() -> (CName, RRSig, Txt, RRSig) {
494                 let cname_resp = CName {
495                         name: "test.cname_wildcard_test.matcorallo.com.".try_into().unwrap(),
496                         canonical_name: "cname.wildcard_test.matcorallo.com.".try_into().unwrap(),
497                 };
498                 let txt_resp = Txt {
499                         name: "cname.wildcard_test.matcorallo.com.".try_into().unwrap(),
500                         data: "wildcard_test".to_owned().into_bytes(),
501                 };
502                 let cname_rrsig = RRSig {
503                         name: "test.cname_wildcard_test.matcorallo.com.".try_into().unwrap(),
504                         ty: CName::TYPE, alg: 13, labels: 3, orig_ttl: 30, expiration: 1708322050,
505                         inception: 1707107050, key_tag: 34530, key_name: "matcorallo.com.".try_into().unwrap(),
506                         signature: base64::decode("JfJuSemF5dtQYxEw6eKL4IRP8BaDt6FtbtdpZ6HjODTDflhKQRhBEbwT7kwceKPAq18q5sWHFV1bMTqE/F3WLw==").unwrap(),
507                 };
508                 let txt_rrsig = RRSig {
509                         name: "cname.wildcard_test.matcorallo.com.".try_into().unwrap(),
510                         ty: Txt::TYPE, alg: 13, labels: 3, orig_ttl: 30, expiration: 1708321778,
511                         inception: 1707106778, key_tag: 34530, key_name: "matcorallo.com.".try_into().unwrap(),
512                         signature: base64::decode("vdnXunPY4CnbW/BL8VOOR9o33+dqyKA/4h+u5VM7NjB30Shp8L8gL5UwE0k7TKRNgHC8j3TqEPEmNMIHz87Z4Q==").unwrap(),
513                 };
514                 (cname_resp, cname_rrsig, txt_resp, txt_rrsig)
515         }
516
517         #[test]
518         fn check_txt_record_a() {
519                 let dnskeys = mattcorallo_dnskey().0;
520                 let (txt, txt_rrsig) = mattcorallo_txt_record();
521                 let txt_resp = [txt];
522                 verify_rrsig(&txt_rrsig, &dnskeys, txt_resp.iter().collect()).unwrap();
523         }
524
525         #[test]
526         fn check_single_txt_proof() {
527                 let mut rr_stream = Vec::new();
528                 for rr in root_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
529                 for rr in com_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
530                 for rr in mattcorallo_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
531                 let (txt, txt_rrsig) = mattcorallo_txt_record();
532                 for rr in [RR::Txt(txt), RR::RRSig(txt_rrsig)] { write_rr(&rr, 1, &mut rr_stream); }
533
534                 let mut rrs = parse_rr_stream(&rr_stream).unwrap();
535                 rrs.shuffle(&mut rand::rngs::OsRng);
536                 let verified_rrs = verify_rr_stream(&rrs).unwrap();
537                 assert_eq!(verified_rrs.len(), 1);
538                 if let RR::Txt(txt) = &verified_rrs[0] {
539                         assert_eq!(txt.name.as_str(), "matt.user._bitcoin-payment.mattcorallo.com.");
540                         assert_eq!(txt.data, b"bitcoin:?b12=lno1qsgqmqvgm96frzdg8m0gc6nzeqffvzsqzrxqy32afmr3jn9ggkwg3egfwch2hy0l6jut6vfd8vpsc3h89l6u3dm4q2d6nuamav3w27xvdmv3lpgklhg7l5teypqz9l53hj7zvuaenh34xqsz2sa967yzqkylfu9xtcd5ymcmfp32h083e805y7jfd236w9afhavqqvl8uyma7x77yun4ehe9pnhu2gekjguexmxpqjcr2j822xr7q34p078gzslf9wpwz5y57alxu99s0z2ql0kfqvwhzycqq45ehh58xnfpuek80hw6spvwrvttjrrq9pphh0dpydh06qqspp5uq4gpyt6n9mwexde44qv7lstzzq60nr40ff38u27un6y53aypmx0p4qruk2tf9mjwqlhxak4znvna5y");
541                 } else { panic!(); }
542         }
543
544         #[test]
545         fn check_txt_record_b() {
546                 let dnskeys = matcorallo_dnskey().0;
547                 let (txt, txt_rrsig) = matcorallo_txt_record();
548                 let txt_resp = [txt];
549                 verify_rrsig(&txt_rrsig, &dnskeys, txt_resp.iter().collect()).unwrap();
550         }
551
552         #[test]
553         fn check_cname_record() {
554                 let dnskeys = matcorallo_dnskey().0;
555                 let (cname, cname_rrsig) = matcorallo_cname_record();
556                 let cname_resp = [cname];
557                 verify_rrsig(&cname_rrsig, &dnskeys, cname_resp.iter().collect()).unwrap();
558         }
559
560         #[test]
561         fn check_multi_zone_proof() {
562                 let mut rr_stream = Vec::new();
563                 for rr in root_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
564                 for rr in com_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
565                 for rr in mattcorallo_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
566                 let (txt, txt_rrsig) = mattcorallo_txt_record();
567                 for rr in [RR::Txt(txt), RR::RRSig(txt_rrsig)] { write_rr(&rr, 1, &mut rr_stream); }
568                 for rr in matcorallo_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
569                 let (txt, txt_rrsig) = matcorallo_txt_record();
570                 for rr in [RR::Txt(txt), RR::RRSig(txt_rrsig)] { write_rr(&rr, 1, &mut rr_stream); }
571                 let (cname, cname_rrsig) = matcorallo_cname_record();
572                 for rr in [RR::CName(cname), RR::RRSig(cname_rrsig)] { write_rr(&rr, 1, &mut rr_stream); }
573
574                 let mut rrs = parse_rr_stream(&rr_stream).unwrap();
575                 rrs.shuffle(&mut rand::rngs::OsRng);
576                 let mut verified_rrs = verify_rr_stream(&rrs).unwrap();
577                 verified_rrs.sort();
578                 assert_eq!(verified_rrs.len(), 3);
579                 if let RR::Txt(txt) = &verified_rrs[0] {
580                         assert_eq!(txt.name.as_str(), "matt.user._bitcoin-payment.mattcorallo.com.");
581                         assert_eq!(txt.data, b"bitcoin:?b12=lno1qsgqmqvgm96frzdg8m0gc6nzeqffvzsqzrxqy32afmr3jn9ggkwg3egfwch2hy0l6jut6vfd8vpsc3h89l6u3dm4q2d6nuamav3w27xvdmv3lpgklhg7l5teypqz9l53hj7zvuaenh34xqsz2sa967yzqkylfu9xtcd5ymcmfp32h083e805y7jfd236w9afhavqqvl8uyma7x77yun4ehe9pnhu2gekjguexmxpqjcr2j822xr7q34p078gzslf9wpwz5y57alxu99s0z2ql0kfqvwhzycqq45ehh58xnfpuek80hw6spvwrvttjrrq9pphh0dpydh06qqspp5uq4gpyt6n9mwexde44qv7lstzzq60nr40ff38u27un6y53aypmx0p4qruk2tf9mjwqlhxak4znvna5y");
582                 } else { panic!(); }
583                 if let RR::Txt(txt) = &verified_rrs[1] {
584                         assert_eq!(txt.name.as_str(), "txt_test.matcorallo.com.");
585                         assert_eq!(txt.data, b"dnssec_prover_test");
586                 } else { panic!(); }
587                 if let RR::CName(cname) = &verified_rrs[2] {
588                         assert_eq!(cname.name.as_str(), "cname_test.matcorallo.com.");
589                         assert_eq!(cname.canonical_name.as_str(), "txt_test.matcorallo.com.");
590                 } else { panic!(); }
591         }
592
593         #[test]
594         fn check_wildcard_record() {
595                 let dnskeys = matcorallo_dnskey().0;
596                 let (txt, txt_rrsig) = matcorallo_wildcard_record();
597                 let txt_resp = [txt];
598                 verify_rrsig(&txt_rrsig, &dnskeys, txt_resp.iter().collect()).unwrap();
599         }
600
601         #[test]
602         fn check_wildcard_proof() {
603                 let mut rr_stream = Vec::new();
604                 for rr in root_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
605                 for rr in com_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
606                 for rr in matcorallo_dnskey().1 { write_rr(&rr, 1, &mut rr_stream); }
607                 let (cname, cname_rrsig, txt, txt_rrsig) = matcorallo_cname_wildcard_record();
608                 for rr in [RR::CName(cname), RR::RRSig(cname_rrsig)] { write_rr(&rr, 1, &mut rr_stream); }
609                 for rr in [RR::Txt(txt), RR::RRSig(txt_rrsig)] { write_rr(&rr, 1, &mut rr_stream); }
610
611                 let mut rrs = parse_rr_stream(&rr_stream).unwrap();
612                 rrs.shuffle(&mut rand::rngs::OsRng);
613                 let mut verified_rrs = verify_rr_stream(&rrs).unwrap();
614                 verified_rrs.sort();
615                 assert_eq!(verified_rrs.len(), 2);
616                 if let RR::Txt(txt) = &verified_rrs[0] {
617                         assert_eq!(txt.name.as_str(), "cname.wildcard_test.matcorallo.com.");
618                         assert_eq!(txt.data, b"wildcard_test");
619                 } else { panic!(); }
620                 if let RR::CName(cname) = &verified_rrs[1] {
621                         assert_eq!(cname.name.as_str(), "test.cname_wildcard_test.matcorallo.com.");
622                         assert_eq!(cname.canonical_name.as_str(), "cname.wildcard_test.matcorallo.com.");
623                 } else { panic!(); }
624         }
625
626         #[test]
627         fn rfc9102_parse_test() {
628                 // Note that this is the `AuthenticationChain` field only, and ignores the
629                 // `ExtSupportLifetime` field (stripping the top two 0 bytes from the front).
630 let rfc9102_test_vector = Vec::from_hex("045f343433045f74637003777777076578616d706c6503636f6d000034000100000e1000230301018bd1da95272f7fa4ffb24137fc0ed03aae67e5c4d8b3c50734e1050a7920b922045f343433045f74637003777777076578616d706c6503636f6d00002e000100000e10005f00340d0500000e105fc6d9005bfdda80074e076578616d706c6503636f6d00ce1d3adeb7dc7cee656d61cfb472c5977c8c9caeae9b765155c518fb107b6a1fe0355fbaaf753c192832fa621fa73a8b85ed79d374117387598fcc812e1ef3fb076578616d706c6503636f6d000030000100000e1000440101030d2670355e0c894d9cfea6c5af6eb7d458b57a50ba88272512d8241d8541fd54adf96ec956789a51ceb971094b3bb3f4ec49f64c686595be5b2e89e8799c7717cc076578616d706c6503636f6d00002e000100000e10005f00300d0200000e105fc6d9005bfdda80074e076578616d706c6503636f6d004628383075b8e34b743a209b27ae148d110d4e1a246138a91083249cb4a12a2d9bc4c2d7ab5eb3afb9f5d1037e4d5da8339c162a9298e9be180741a8ca74accc076578616d706c6503636f6d00002b00010002a3000024074e0d02e9b533a049798e900b5c29c90cd25a986e8a44f319ac3cd302bafc08f5b81e16076578616d706c6503636f6d00002e00010002a3000057002b0d020002a3005fc6d9005bfdda80861703636f6d00a203e704a6facbeb13fc9384fdd6de6b50de5659271f38ce81498684e6363172d47e2319fdb4a22a58a231edc2f1ff4fb2811a1807be72cb5241aa26fdaee03903636f6d00003000010002a30000440100030dec8204e43a25f2348c52a1d3bce3a265aa5d11b43dc2a471162ff341c49db9f50a2e1a41caf2e9cd20104ea0968f7511219f0bdc56b68012cc3995336751900b03636f6d00003000010002a30000440101030d45b91c3bef7a5d99a7a7c8d822e33896bc80a777a04234a605a4a8880ec7efa4e6d112c73cd3d4c65564fa74347c873723cc5f643370f166b43dedff836400ff03636f6d00003000010002a30000440101030db3373b6e22e8e49e0e1e591a9f5bd9ac5e1a0f86187fe34703f180a9d36c958f71c4af48ce0ebc5c792a724e11b43895937ee53404268129476eb1aed323939003636f6d00002e00010002a300005700300d010002a3005fc6d9005bfdda8049f303636f6d0018a948eb23d44f80abc99238fcb43c5a18debe57004f7343593f6deb6ed71e04654a433f7aa1972130d9bd921c73dcf63fcf665f2f05a0aaebafb059dc12c96503636f6d00002e00010002a300005700300d010002a3005fc6d9005bfdda80708903636f6d006170e6959bd9ed6e575837b6f580bd99dbd24a44682b0a359626a246b1812f5f9096b75e157e77848f068ae0085e1a609fc19298c33b736863fbccd4d81f5eb203636f6d00002b000100015180002449f30d0220f7a9db42d0e2042fbbb9f9ea015941202f9eabb94487e658c188e7bcb5211503636f6d00002b000100015180002470890d02ad66b3276f796223aa45eda773e92c6d98e70643bbde681db342a9e5cf2bb38003636f6d00002e0001000151800053002b0d01000151805fc6d9005bfdda807cae00122e276d45d9e9816f7922ad6ea2e73e82d26fce0a4b718625f314531ac92f8ae82418df9b898f989d32e80bc4deaba7c4a7c8f172adb57ced7fb5e77a784b0700003000010001518000440100030dccacfe0c25a4340fefba17a254f706aac1f8d14f38299025acc448ca8ce3f561f37fc3ec169fe847c8fcbe68e358ff7c71bb5ee1df0dbe518bc736d4ce8dfe1400003000010001518000440100030df303196789731ddc8a6787eff24cacfeddd032582f11a75bb1bcaa5ab321c1d7525c2658191aec01b3e98ab7915b16d571dd55b4eae51417110cc4cdd11d171100003000010001518000440101030dcaf5fe54d4d48f16621afb6bd3ad2155bacf57d1faad5bac42d17d948c421736d9389c4c4011666ea95cf17725bd0fa00ce5e714e4ec82cfdfacc9b1c863ad4600002e000100015180005300300d00000151805fc6d9005bfdda80b79d00de7a6740eeecba4bda1e5c2dd4899b2c965893f3786ce747f41e50d9de8c0a72df82560dfb48d714de3283ae99a49c0fcb50d3aaadb1a3fc62ee3a8a0988b6be").unwrap();
631
632                 let mut rrs = parse_rr_stream(&rfc9102_test_vector).unwrap();
633                 rrs.shuffle(&mut rand::rngs::OsRng);
634                 let verified_rrs = verify_rr_stream(&rrs).unwrap();
635                 assert_eq!(verified_rrs.len(), 1);
636                 if let RR::TLSA(tlsa) = &verified_rrs[0] {
637                         assert_eq!(tlsa.cert_usage, 3);
638                         assert_eq!(tlsa.selector, 1);
639                         assert_eq!(tlsa.data_ty, 1);
640                         assert_eq!(tlsa.data, Vec::from_hex("8bd1da95272f7fa4ffb24137fc0ed03aae67e5c4d8b3c50734e1050a7920b922").unwrap());
641                 } else { panic!(); }
642         }
643 }