#[cfg(test)]
use quickcheck::{Arbitrary, Gen};
use crate::{
cert::prelude::*,
Error,
Fingerprint,
Result,
types::{
PublicKeyAlgorithm,
},
};
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct RevocationKey {
pk_algo: PublicKeyAlgorithm,
fp: Fingerprint,
sensitive: bool,
unknown: u8,
}
assert_send_and_sync!(RevocationKey);
impl From<&Cert> for RevocationKey {
fn from(cert: &Cert) -> Self {
RevocationKey::new(cert.primary_key().pk_algo(),
cert.fingerprint(),
false)
}
}
impl RevocationKey {
pub fn new(pk_algo: PublicKeyAlgorithm, fp: Fingerprint, sensitive: bool)
-> Self
{
RevocationKey {
pk_algo, fp, sensitive, unknown: 0,
}
}
pub fn from_bits(pk_algo: PublicKeyAlgorithm, fp: Fingerprint, class: u8)
-> Result<Self> {
if class & REVOCATION_KEY_FLAG_MUST_BE_SET == 0 {
return Err(Error::InvalidArgument(
"Most significant bit of class must be set".into()).into());
}
let sensitive = class & REVOCATION_KEY_FLAG_SENSITIVE > 0;
let unknown = class & REVOCATION_KEY_MASK_UNKNOWN;
Ok(RevocationKey {
pk_algo, fp, sensitive, unknown,
})
}
pub fn class(&self) -> u8 {
REVOCATION_KEY_FLAG_MUST_BE_SET
| if self.sensitive() {
REVOCATION_KEY_FLAG_SENSITIVE
} else {
0
}
| self.unknown
}
pub fn revoker(&self) -> (PublicKeyAlgorithm, &Fingerprint) {
(self.pk_algo, &self.fp)
}
pub fn set_revoker(&mut self, pk_algo: PublicKeyAlgorithm, fp: Fingerprint)
-> (PublicKeyAlgorithm, Fingerprint) {
let pk_algo = std::mem::replace(&mut self.pk_algo, pk_algo);
let fp = std::mem::replace(&mut self.fp, fp);
(pk_algo, fp)
}
pub fn sensitive(&self) -> bool {
self.sensitive
}
pub fn set_sensitive(mut self, v: bool) -> Self {
self.sensitive = v;
self
}
}
const REVOCATION_KEY_FLAG_MUST_BE_SET: u8 = 0x80;
const REVOCATION_KEY_FLAG_SENSITIVE: u8 = 0x40;
const REVOCATION_KEY_MASK_UNKNOWN: u8 = ! (REVOCATION_KEY_FLAG_MUST_BE_SET
| REVOCATION_KEY_FLAG_SENSITIVE);
#[cfg(test)]
impl Arbitrary for RevocationKey {
fn arbitrary<G: Gen>(g: &mut G) -> Self {
RevocationKey {
pk_algo: Arbitrary::arbitrary(g),
fp: Arbitrary::arbitrary(g),
sensitive: Arbitrary::arbitrary(g),
unknown: u8::arbitrary(g) & REVOCATION_KEY_MASK_UNKNOWN,
}
}
}