use std::future::Future;
use std::sync::Arc;
use crate::op::{MessageFinalizer, MessageVerifier};
#[cfg(feature = "dnssec")]
#[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
use crate::proto::rr::dnssec::tsig::TSigner;
#[cfg(feature = "dnssec")]
#[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
use crate::proto::rr::dnssec::SigSigner;
use crate::proto::{error::ProtoError, xfer::DnsRequestSender};
use crate::proto::error::ProtoResult;
use crate::proto::op::Message;
use crate::proto::rr::Record;
#[allow(missing_copy_implementations)]
pub enum Signer {
#[cfg(feature = "dnssec")]
#[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
Sig0(Box<SigSigner>),
#[cfg(feature = "dnssec")]
#[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
TSIG(TSigner),
}
#[cfg(feature = "dnssec")]
#[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
impl From<SigSigner> for Signer {
fn from(s: SigSigner) -> Self {
Self::Sig0(Box::new(s))
}
}
#[cfg(feature = "dnssec")]
#[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
impl From<TSigner> for Signer {
fn from(s: TSigner) -> Self {
Self::TSIG(s)
}
}
impl MessageFinalizer for Signer {
#[allow(unreachable_patterns, unused_variables)]
fn finalize_message(
&self,
message: &Message,
time: u32,
) -> ProtoResult<(Vec<Record>, Option<MessageVerifier>)> {
match self {
#[cfg(feature = "dnssec")]
#[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
Self::Sig0(s0) => s0.finalize_message(message, time),
#[cfg(feature = "dnssec")]
#[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
Self::TSIG(tsig) => tsig.finalize_message(message, time),
_ => unreachable!("the feature `dnssec` is required for Message signing"),
}
}
}
pub trait ClientConnection: 'static + Sized + Send + Sync + Unpin {
type Sender: DnsRequestSender;
type SenderFuture: Future<Output = Result<Self::Sender, ProtoError>> + 'static + Send + Unpin;
fn new_stream(&self, signer: Option<Arc<Signer>>) -> Self::SenderFuture;
}