Stefan Bühler
3 years ago
9 changed files with 513 additions and 26 deletions
@ -0,0 +1,9 @@ |
|||
use crate::ser::{packet::DnsPacketData, text::DnsTextData}; |
|||
|
|||
#[dnsbox_derive::native_flags(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
/// Flags for the CAA RR
|
|||
pub enum CaaFlags { |
|||
/// Issuer Critical Flag
|
|||
CRITICAL = 0x01, // bit "7"
|
|||
} |
|||
@ -0,0 +1,125 @@ |
|||
use crate::ser::{packet::DnsPacketData, text::DnsTextData}; |
|||
|
|||
// https://www.iana.org/assignments/dns-sec-alg-numbers/dns-sec-alg-numbers.xhtml
|
|||
#[dnsbox_derive::native_enum(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
pub enum DnsSecAlgorithm { |
|||
/// Delete DS
|
|||
// [RFC4034][RFC4398][RFC8078]
|
|||
DELETE = 0, |
|||
/// RSA/MD5 (deprecated, see 5)
|
|||
// [RFC3110][RFC4034]
|
|||
RSAMD5 = 1, |
|||
/// Diffie-Hellman
|
|||
// [RFC2539][proposed standard]
|
|||
DH = 2, |
|||
/// DSA/SHA1
|
|||
// [RFC3755][proposed standard][RFC2536][proposed standard][Federal Information Processing Standards Publication (FIPS PUB) 186, Digital Signature Standard, 18 May 1994.][Federal Information Processing Standards Publication (FIPS PUB) 180-1, Secure Hash Standard, 17 April 1995. (Supersedes FIPS PUB 180 dated 11 May 1993.)]
|
|||
DSA = 3, |
|||
|
|||
// Reserved: 4 [RFC6725]
|
|||
|
|||
/// RSA/SHA-1
|
|||
// [RFC3110][RFC4034]
|
|||
RSASHA1 = 5, |
|||
/// DSA-NSEC3-SHA1
|
|||
// [RFC5155][proposed standard]
|
|||
DSA_NSEC3_SHA1 = 6, |
|||
/// RSASHA1-NSEC3-SHA1
|
|||
// [RFC5155][proposed standard]
|
|||
RSASHA1_NSEC3_SHA1 = 7, |
|||
/// RSA/SHA-256
|
|||
// [RFC5702][proposed standard]
|
|||
RSASHA256 = 8, |
|||
|
|||
// Reserved: 9 [RFC6725]
|
|||
|
|||
/// RSA/SHA-512
|
|||
// [RFC5702][proposed standard]
|
|||
RSASHA512 = 10, |
|||
|
|||
// Reserved: 11 [RFC6725]
|
|||
|
|||
/// GOST R 34.10-2001
|
|||
// [RFC5933][standards track]
|
|||
ECC_GOST = 12, |
|||
/// ECDSA Curve P-256 with SHA-256
|
|||
// [RFC6605][standards track]
|
|||
ECDSAP256SHA256 = 13, |
|||
/// ECDSA Curve P-384 with SHA-384
|
|||
// [RFC6605][standards track]
|
|||
ECDSAP384SHA384 = 14, |
|||
/// Ed25519
|
|||
// [RFC8080][standards track]
|
|||
ED25519 = 15, |
|||
/// Ed448
|
|||
// [RFC8080][standards track]
|
|||
ED448 = 16, |
|||
/// Reserved for Indirect Keys
|
|||
// [RFC4034][proposed standard]
|
|||
INDIRECT = 252, |
|||
/// private algorithm
|
|||
// [RFC4034]
|
|||
PRIVATEDNS = 253, |
|||
/// private algorithm OID
|
|||
// [RFC4034]
|
|||
PRIVATEOID = 254, |
|||
|
|||
// Reserved: 255 [RFC4034][proposed standard]
|
|||
} |
|||
|
|||
#[dnsbox_derive::native_flags(u16)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
/// Flags for the DNSKEY RR
|
|||
pub enum DnskeyFlags { |
|||
ZONE_KEY = 0x0100, // bit "7"
|
|||
/// secure entry point, SEP
|
|||
SECURE_ENTRY_POINT = 0x0001, // bit "15"
|
|||
} |
|||
|
|||
#[dnsbox_derive::native_enum(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
/// Protocol for the DNSKEY RR (only DNSSEC(3) is valid for DNSKEY)
|
|||
pub enum DnskeyProtocol { |
|||
// reserved: 0x00
|
|||
TLS = 0x01, |
|||
EMAIL = 0x02, |
|||
DNSSEC = 0x03, |
|||
IPSEC = 0x04, |
|||
ALL = 0xff, |
|||
} |
|||
|
|||
// https://www.iana.org/assignments/ds-rr-types/ds-rr-types.xhtml
|
|||
#[dnsbox_derive::native_enum(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
pub enum DnsSecDigestAlgorithm { |
|||
// Reserved: 0 [RFC3658]
|
|||
SHA1 = 0x01, // [RFC3658]
|
|||
SHA256 = 0x02, // [RFC4509]
|
|||
GOST_R_34_11_94 = 0x03, // [RFC5933]
|
|||
SHA384 = 0x04, // [RFC6605]
|
|||
} |
|||
|
|||
// https://www.iana.org/assignments/dnssec-nsec3-parameters/dnssec-nsec3-parameters.xhtml
|
|||
#[dnsbox_derive::native_flags(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
/// Flags for the NSEC3 RR
|
|||
pub enum Nsec3Flags { |
|||
OPT_OUT = 0x01, // bit "7"
|
|||
} |
|||
|
|||
// https://www.iana.org/assignments/dnssec-nsec3-parameters/dnssec-nsec3-parameters.xhtml
|
|||
#[dnsbox_derive::native_flags(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
/// Flags for the NSEC3PARAM RR
|
|||
pub enum Nsec3ParamFlags { |
|||
// reserved: Nsec3Flags::OPT_OUT bit "7" (0x01)
|
|||
} |
|||
|
|||
// https://www.iana.org/assignments/dnssec-nsec3-parameters/dnssec-nsec3-parameters.xhtml
|
|||
#[dnsbox_derive::native_enum(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
pub enum Nsec3Algorithm { |
|||
// Reserved: 0 [RFC5155]
|
|||
SHA1 = 0x01, // [RFC5155]
|
|||
} |
|||
@ -0,0 +1,25 @@ |
|||
use crate::ser::{packet::DnsPacketData, text::DnsTextData}; |
|||
|
|||
// https://www.iana.org/assignments/dns-sshfp-rr-parameters/dns-sshfp-rr-parameters.xml
|
|||
|
|||
#[dnsbox_derive::native_enum(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
pub enum SshFpAlgorithm { |
|||
// Reserved: 0
|
|||
|
|||
// [RFC4255]
|
|||
RSA = 1, |
|||
// [RFC4255]
|
|||
DSA = 2, |
|||
// [RFC6594]
|
|||
ECDSA = 3, |
|||
// [RFC7479]
|
|||
ED25519 = 4, |
|||
} |
|||
|
|||
#[dnsbox_derive::native_enum(u8)] |
|||
#[derive(DnsPacketData, DnsTextData)] |
|||
pub enum SshFpType { |
|||
SHA1 = 1, |
|||
SHA256 = 2, |
|||
} |
|||
@ -0,0 +1,88 @@ |
|||
use proc_macro2::TokenStream; |
|||
|
|||
pub fn attribute_native_enum( |
|||
native_type: TokenStream, |
|||
structure: synstructure::Structure, |
|||
) -> TokenStream { |
|||
let ast = structure.ast(); |
|||
let in_attrs = ast.attrs.iter().map(|a| quote!{#a}).collect::<TokenStream>(); |
|||
let in_vis = &ast.vis; |
|||
let name = &ast.ident; |
|||
|
|||
let known_name = syn::Ident::new(&format!("{}Known", name), proc_macro2::Span::call_site()); |
|||
|
|||
let enumdata = match &ast.data { |
|||
syn::Data::Enum(de) => de, |
|||
_ => panic!("not an enum"), |
|||
}; |
|||
let known_enum; |
|||
|
|||
{ |
|||
let variants = &enumdata.variants; |
|||
let doc_str = format!("Known enum variants of [`{}`]\n\n[`{}`]: struct.{}.html\n", name, name, name); |
|||
known_enum = quote! { |
|||
#[doc = #doc_str] |
|||
#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)] |
|||
#[repr(#native_type)] |
|||
#[allow(non_camel_case_types)] |
|||
#in_vis enum #known_name { |
|||
#variants |
|||
} |
|||
} |
|||
} |
|||
|
|||
let mut consts = TokenStream::new(); |
|||
let mut convert = TokenStream::new(); |
|||
for variant in &enumdata.variants { |
|||
let variant_attrs = variant.attrs.iter().map(|a| quote!{#a}).collect::<TokenStream>(); |
|||
let disc_name = &variant.ident; |
|||
let disc = variant.discriminant.as_ref().map(|(_, d)| quote!{#d}).unwrap_or_else(|| quote!{ |
|||
#known_name::#disc_name as #native_type |
|||
}); |
|||
consts.extend(quote! { |
|||
#variant_attrs |
|||
pub const #disc_name: Self = #name(#disc); |
|||
}); |
|||
convert.extend(quote! { |
|||
Self::#disc_name => #known_name::#disc_name, |
|||
}); |
|||
} |
|||
|
|||
quote! { |
|||
#known_enum |
|||
|
|||
#in_attrs |
|||
#[derive(Clone, Copy, PartialEq, Eq)] |
|||
#in_vis struct #name(pub #native_type); |
|||
|
|||
impl #name { |
|||
#consts |
|||
|
|||
/// Try converting to known enum values
|
|||
pub fn into_known(self) -> Option<#known_name> { |
|||
Some( |
|||
#[allow(unreachable_patterns)] |
|||
match self { |
|||
#convert |
|||
_ => return None, |
|||
} |
|||
) |
|||
} |
|||
} |
|||
|
|||
impl core::fmt::Debug for #name { |
|||
fn fmt(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result { |
|||
match self.into_known() { |
|||
Some(v) => v.fmt(f), |
|||
None => f.debug_tuple(stringify!(#name)).field(&self.0).finish(), |
|||
} |
|||
} |
|||
} |
|||
|
|||
impl From<#known_name> for #name { |
|||
fn from(v: #known_name) -> Self { |
|||
#name(v as #native_type) |
|||
} |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,136 @@ |
|||
use proc_macro2::TokenStream; |
|||
|
|||
pub fn attribute_native_flags( |
|||
native_type: TokenStream, |
|||
structure: synstructure::Structure, |
|||
) -> TokenStream { |
|||
let ast = structure.ast(); |
|||
let in_attrs = ast.attrs.iter().map(|a| quote!{#a}).collect::<TokenStream>(); |
|||
let in_vis = &ast.vis; |
|||
let name = &ast.ident; |
|||
|
|||
let hidden_impl_mod = syn::Ident::new(&format!("_{}_hidden_native_flags", name), proc_macro2::Span::call_site()); |
|||
|
|||
let enumdata = match &ast.data { |
|||
syn::Data::Enum(de) => de, |
|||
_ => panic!("not an enum"), |
|||
}; |
|||
|
|||
let mut consts = TokenStream::new(); |
|||
let mut known_mask = TokenStream::new(); |
|||
let mut dbg = TokenStream::new(); |
|||
for variant in &enumdata.variants { |
|||
let variant_attrs = variant.attrs.iter().map(|a| quote!{#a}).collect::<TokenStream>(); |
|||
let disc_name = &variant.ident; |
|||
let disc = variant.discriminant.as_ref().map(|(_, d)| d).expect("all variants need explicit bitmask discriminants"); |
|||
consts.extend(quote! { |
|||
#variant_attrs |
|||
pub const #disc_name: Flag = Flag { mask: #disc }; |
|||
}); |
|||
known_mask.extend(quote!{ |
|||
| #disc |
|||
}); |
|||
dbg.extend(quote! { |
|||
if self & Self::#disc_name { |
|||
if !first { f.write_str(" | ")?; } |
|||
first = false; |
|||
f.write_str(stringify!(#disc_name))?; |
|||
} |
|||
}); |
|||
} |
|||
|
|||
let bitops = quote! { |
|||
impl core::ops::BitAnd<Flag> for super::#name { |
|||
type Output = bool; |
|||
|
|||
fn bitand(self, rhs: Flag) -> Self::Output { |
|||
rhs.mask == self.0 & rhs.mask |
|||
} |
|||
} |
|||
|
|||
impl core::ops::BitAnd<Flag> for &super::#name { |
|||
type Output = bool; |
|||
|
|||
fn bitand(self, rhs: Flag) -> Self::Output { |
|||
rhs.mask == self.0 & rhs.mask |
|||
} |
|||
} |
|||
|
|||
impl core::ops::BitOr for Flag { |
|||
type Output = super::#name; |
|||
|
|||
fn bitor(self, rhs: Flag) -> Self::Output { |
|||
super::#name(self.mask | rhs.mask) |
|||
} |
|||
} |
|||
|
|||
impl core::ops::BitOr<Flag> for super::#name { |
|||
type Output = super::#name; |
|||
|
|||
fn bitor(self, rhs: Flag) -> Self::Output { |
|||
super::#name(self.0 | rhs.mask) |
|||
} |
|||
} |
|||
|
|||
impl core::ops::BitOr<super::#name> for Flag { |
|||
type Output = super::#name; |
|||
|
|||
fn bitor(self, rhs: super::#name) -> Self::Output { |
|||
super::#name(self.mask | rhs.0) |
|||
} |
|||
} |
|||
}; |
|||
|
|||
quote! { |
|||
#in_attrs |
|||
///
|
|||
/// The `Flag` constants can be used to check if a flag is set
|
|||
/// by using bitwise and: `flags & flag`.
|
|||
///
|
|||
/// They can be converted to and combined with bitwise or to set
|
|||
/// flags. Their bitmask is available as public struct member
|
|||
/// `mask`.
|
|||
#[derive(Clone, Copy, PartialEq, Eq)] |
|||
#in_vis struct #name(pub #native_type); |
|||
|
|||
mod #hidden_impl_mod { |
|||
#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)] |
|||
pub struct Flag { |
|||
pub mask: #native_type, |
|||
} |
|||
|
|||
const known_mask: #native_type = 0 #known_mask; |
|||
|
|||
impl super::#name { |
|||
pub fn new_flag(mask: #native_type) -> Flag { |
|||
Flag { mask } |
|||
} |
|||
|
|||
#consts |
|||
} |
|||
|
|||
#bitops |
|||
|
|||
impl From<Flag> for super::#name { |
|||
fn from(v: Flag) -> Self { |
|||
Self(v.mask) |
|||
} |
|||
} |
|||
|
|||
impl core::fmt::Debug for super::#name { |
|||
fn fmt(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result { |
|||
let mut first = true; |
|||
let unknown_bits = self.0 & !known_mask; |
|||
#dbg |
|||
if unknown_bits != 0 { |
|||
if !first { f.write_str(" | ")?; } |
|||
write!(f, "unknown(0x{:x})", unknown_bits)?; |
|||
} else if first { |
|||
f.write_str("(empty)")?; |
|||
} |
|||
Ok(()) |
|||
} |
|||
} |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue