trust-dns-client 0.23.2

Trust-DNS is a safe and secure DNS library. This is the Client library with DNSSEC support. DNSSEC with NSEC validation for negative records, is complete. The client supports dynamic DNS with SIG0 authenticated requests, implementing easy to use high level funtions. Trust-DNS is based on the Tokio and Futures libraries, which means it should be easily integrated into other software that also use those libraries.
Documentation
// Copyright (C) 2016 Benjamin Fry <benjaminfry@me.com>
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

//! Trait for client connections
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;

/// List of currently supported signers
#[allow(missing_copy_implementations)]
pub enum Signer {
    /// A Sig0 based signer
    #[cfg(feature = "dnssec")]
    #[cfg_attr(docsrs, doc(cfg(feature = "dnssec")))]
    Sig0(Box<SigSigner>),
    /// A TSIG based signer
    #[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"),
        }
    }
}

/// Trait for client connections
pub trait ClientConnection: 'static + Sized + Send + Sync + Unpin {
    /// The associated DNS RequestSender type.
    type Sender: DnsRequestSender;
    /// A future that resolves to the RequestSender
    type SenderFuture: Future<Output = Result<Self::Sender, ProtoError>> + 'static + Send + Unpin;

    /// Construct a new stream for use in the Client
    fn new_stream(&self, signer: Option<Arc<Signer>>) -> Self::SenderFuture;
}