use bit_vec::BitVec;
use yasna::{ASN1Result, BERDecodable, BERReader, DERWriter, Tag};
use crate::{x509::GenericCertificate, DerWrite};
use super::{body::PkiBody, header::PkiHeader};
derive_sequence! {
PkiMessage<'a> {
header: [_] UNTAGGED REQUIRED: PkiHeader<'a>,
body: [_] UNTAGGED REQUIRED: PkiBody,
protection: [0] EXPLICIT OPTIONAL: Option<PkiProtection>,
extra_certs: [1] EXPLICIT OPTIONAL: Option<CmpCertificates>,
}
}
pub type PkiProtection = BitVec;
pub type CmpCertificate = GenericCertificate;
derive_sequence_of!{
CmpCertificate => CmpCertificates
}
derive_sequence! {
ProtectedPart<'a> {
header: PkiHeader<'a>,
body: PkiBody,
}
}
#[cfg(test)]
mod test {
use crate::{FromDer, ToDer};
use b64_ct::{FromBase64, ToBase64, STANDARD};
use super::*;
#[test]
fn test_ftx_pki_msg_decode_encode() {
let pki_msg_base64 = include_str!("../../tests/data/ftx_test_pki_msg.b64");
check_pki_msg_decode_encode(pki_msg_base64);
}
#[test]
fn test_openssl_pki_msg_decode_encode() {
let pki_msg_base64 = include_str!("../../tests/data/openssl_pki_msg.b64");
check_pki_msg_decode_encode(pki_msg_base64);
}
fn check_pki_msg_decode_encode(pki_msg_base64: &str) {
let pki_msg_data = pki_msg_base64.trim().from_base64().expect("base64 decode test pki msg");
let pki_msg = PkiMessage::from_der(&pki_msg_data).expect("DER decode test pki msg");
let pki_msg_back_to_der = pki_msg.to_der();
assert!(
pki_msg_data == pki_msg_back_to_der,
"{}",
pki_msg_back_to_der.to_base64(STANDARD)
);
let pki_msg2 = PkiMessage::from_der(&pki_msg_back_to_der).expect("DER decode re-encoded pki msg");
assert_eq!(pki_msg, pki_msg2);
}
}