//! Relay responder channel.

//!

//! Code related to the relay channel opened as a responder. The handshake code is responsible for

//! creating an [`MaybeVerifiableRelayResponderChannel`] when accepting an inbound connection.

//!

//! It can then be used to get a fully working channel.

use digest::Digest;

use futures::{AsyncRead, AsyncWrite};

use safelog::{MaybeSensitive, Sensitive};

use std::{net::IpAddr, ops::Deref, sync::Arc};

use subtle::ConstantTimeEq;

use tracing::instrument;

use tor_cell::chancell::msg;

use tor_linkspec::{HasRelayIds, OwnedChanTarget, RelayIds};

use tor_llcrypto as ll;

use tor_llcrypto::pk::ed25519::Ed25519Identity;

use tor_rtcompat::{CertifiedConn, CoarseTimeProvider, Runtime, SleepProvider, StreamOps};

use web_time_compat::{SystemTime, SystemTimeExt};

use crate::{

    ClockSkew, Error, RelayChannelAuthMaterial, Result,

    channel::{

        Channel, ChannelMode, ClogDigest, Reactor, SlogDigest,

        circmap::CircIdRange,

        handshake::{UnverifiedChannel, VerifiedChannel},

    },

    peer::{PeerAddr, PeerInfo},

    relay::CreateRequestHandler,

    relay::channel::ChannelAuthenticationData,

};

/// An enum combining both the possibility of a verifiable (relay) or non verifiable channel

/// (client/bridge).

#[allow(clippy::exhaustive_enums)]

pub enum MaybeVerifiableRelayResponderChannel<

    T: AsyncRead + AsyncWrite + CertifiedConn + StreamOps + Send + Unpin + 'static,

    S: CoarseTimeProvider + SleepProvider,

> {

    /// Verifiable channel (relay).

    Verifiable(UnverifiedResponderRelayChannel<T, S>),

    /// Non verifiable channel (client/bridge).

    NonVerifiable(NonVerifiableResponderRelayChannel<T, S>),

}

/// A channel that can NOT be verified. This is solely either a client or bridge on the other end.

///

/// This can only be built if no [`msg::Authenticate`] was ever received.

pub struct NonVerifiableResponderRelayChannel<

    T: AsyncRead + AsyncWrite + CertifiedConn + StreamOps + Send + Unpin + 'static,

    S: CoarseTimeProvider + SleepProvider,

> {

    /// The common unverified channel that both client and relays use.

    pub(crate) inner: UnverifiedChannel<T, S>,

    /// The netinfo cell received from the initiator.

    pub(crate) netinfo_cell: msg::Netinfo,

    /// Our advertised addresses.

    pub(crate) my_addrs: Vec<IpAddr>,

    /// The peer address which is sensitive considering it is either client or bridge.

    pub(crate) peer_addr: Sensitive<PeerAddr>,

    /// Provided to each new channel so that they can handle CREATE* requests.

    pub(crate) create_request_handler: Arc<CreateRequestHandler>,

    /// Our Ed25519 identity.

    ///

    /// Needed for ntor handshakes.

    pub(crate) our_ed25519_id: Ed25519Identity,

}

/// A verifiable relay responder channel that is currently unverified. This can only be a relay on

/// the other end.

///

/// The verify() and then finish() functions are to be used to get a final Channel/Reactor.

pub struct UnverifiedResponderRelayChannel<

    T: AsyncRead + AsyncWrite + CertifiedConn + StreamOps + Send + Unpin + 'static,

    S: CoarseTimeProvider + SleepProvider,

> {

    /// The common unverified channel that both client and relays use.

    pub(crate) inner: UnverifiedChannel<T, S>,

    /// AUTHENTICATE cell received from the initiator.

    pub(crate) auth_cell: msg::Authenticate,

    /// The netinfo cell received from the initiator.

    pub(crate) netinfo_cell: msg::Netinfo,

    /// The [`msg::Certs`] cell received from the initiator.

    pub(crate) certs_cell: msg::Certs,

    /// Our authentication key material.

    pub(crate) auth_material: Arc<RelayChannelAuthMaterial>,

    /// Our advertised addresses.

    pub(crate) my_addrs: Vec<IpAddr>,

    /// The peer address which we know is a relay.

    pub(crate) peer_addr: PeerAddr,

    /// The CLOG digest.

    pub(crate) clog_digest: ClogDigest,

    /// The SLOG digest.

    pub(crate) slog_digest: SlogDigest,

    /// Provided to each new channel so that they can handle CREATE* requests.

    pub(crate) create_request_handler: Arc<CreateRequestHandler>,

}

/// A verified relay responder channel.

///

/// Only finish() remains to transform this into a fully usable [`crate::channel::Channel`] and

/// [`crate::channel::Reactor`].

pub struct VerifiedResponderRelayChannel<

    T: AsyncRead + AsyncWrite + CertifiedConn + StreamOps + Send + Unpin + 'static,

    S: CoarseTimeProvider + SleepProvider,

> {

    /// The common unverified channel that both client and relays use.

    inner: VerifiedChannel<T, S>,

    /// The netinfo cell that we got from the relay. Canonicity decision.

    netinfo_cell: msg::Netinfo,

    /// Our advertised addresses.

    my_addrs: Vec<IpAddr>,

    /// The peer address which we know is a relay.

    peer_addr: PeerAddr,

    /// Provided to each new channel so that they can handle CREATE* requests.

    create_request_handler: Arc<CreateRequestHandler>,

    /// Our Ed25519 identity.

    ///

    /// Needed for ntor handshakes.

    our_ed25519_id: Ed25519Identity,

}

impl<T, S> UnverifiedResponderRelayChannel<T, S>

where

    T: AsyncRead + AsyncWrite + CertifiedConn + StreamOps + Send + Unpin + 'static,

    S: CoarseTimeProvider + SleepProvider,

{

    /// Validate the certificates and keys in the relay's handshake.

    ///

    /// 'peer_target_no_ids' is the peer, without identities as we are accepting a connection and thus

    /// don't have expectations on any identity, that we want to make sure we're connecting to.

    ///

    /// 'our_tls_cert' is the x.509 certificate that we presented during the TLS handshake.

    ///

    /// 'now' is the time at which to check that certificates are valid.  `None` means to use the

    /// current time. It can be used for testing to override the current view of the time.

    ///

    /// This is a separate function because it's likely to be somewhat CPU-intensive.

    #[instrument(skip_all, level = "trace")]

        // Get these object out as we consume "self" in the inner check().

        // We are a relay responder. We have received a CERTS cell and we need to verify these

        // certs:

        //

        //   Relay Identities:

        //      IDENTITY_V_SIGNING_CERT (CertType 4)

        //      RSA_ID_X509             (CertType 2)

        //      RSA_ID_V_IDENTITY       (CertType 7)

        //

        //   Connection Cert:

        //      SIGNING_V_LINK_AUTH     (CertType 6)

        //

        // Validating the relay identities first so we can make sure we are talking to the relay

        // (peer) we wanted. Then, check the AUTHENTICATE cell.

        //

        // The end result is a verified channel (not authenticated yet) which guarantee that we are

        // talking to the right relay that we wanted. We validate so we can prove these:

        //

        // - IDENTITY_V_SIGNING proves that KP_relaysign_ed speaks on behalf of KP_relayid_ed

        // - SIGNING_V_LINK_AUTH proves that KP_link_ed speaks on behalf of KP_relaysign_ed

        // - The AUTHENTICATE cell proves that the TLS session's key material is known by the

        //   owner of KP_link_ed

        // - Therefore, we have a chain from:

        //   KS_relayid_ed → KP_relaysign_ed → KP_link_ed → AUTHENTICATE cell → the channel itself.

        //

        // As for legacy certs, they prove nothing but we can extract keys:

        //

        // - RSA_ID_X509 proves nothing; we just extract its subject key as KP_relayid_rsa.

        // - RSA_ID_V_IDENTITY proves that KP_relayid_ed speaks on behalf of KP_relayid_rsa.

        // - Therefore we have a chain from:

        //   KP_relayid_rsa → KS_relayid_ed → KP_relaysign_ed → KP_link_ed → AUTHENTICATE cell →

        //   the channel itself.

        // Check the relay identities in the CERTS cell.

        // Next, verify the LINK_AUTH cert (CertType 6).

        // By building the ChannelAuthenticationData, we are certain that the authentication type

        // of the initiator is supported by us.

        // CRITICAL: This if is what authenticates a channel on the responder side. We compare

        // what we expected to what we received.

        // This equality is in constant-time to avoid timing attack oracle.

        {

        // CRITICAL: Verify the signature of the AUTHENTICATE cell with the peer KP_link_ed.

        // Transform our inner into a verified channel now that we are verified.

        // This part is very important as we now flag that we are verified and thus authenticated.

        //

        // At this point, the underlying cell handler is in the Handshake state. Setting the

        // channel type here as authenticated means that once the handler transition to the Open

        // state, it will carry this authenticated flag leading to the message filter of the

        // channel codec to adapt its restricted message sets (meaning R2R only).

        //

        // After this call, it is considered a R2R channel.

    /// Return the clock skew of this channel.

}

impl<T, S> VerifiedResponderRelayChannel<T, S>

where

    T: AsyncRead + AsyncWrite + CertifiedConn + StreamOps + Send + Unpin + 'static,

    S: CoarseTimeProvider + SleepProvider,

{

    /// Finish the handhshake which will create an open channel and reactor.

    ///

    /// The resulting channel is considered, by Tor protocol standard, an authenticated relay

    /// channel on which circuits can be opened.

    #[instrument(skip_all, level = "trace")]

        // Relay<->Relay channels are NOT sensitive as we need their info in the log.

        let peer_info = MaybeSensitive::not_sensitive(PeerInfo::new(

            self.peer_addr,

            self.inner.relay_ids().clone(),

        ));

        let channel_mode = ChannelMode::Relay {

            circ_id_range: CircIdRange::Low,

            our_ed25519_id: self.our_ed25519_id,

            create_request_handler: self.create_request_handler,

        };

        self.inner

            .finish(&self.netinfo_cell, &self.my_addrs, peer_info, channel_mode)

            .await

}

impl<T, S> NonVerifiableResponderRelayChannel<T, S>

where

    T: AsyncRead + AsyncWrite + CertifiedConn + StreamOps + Send + Unpin + 'static,

    S: CoarseTimeProvider + SleepProvider,

{

    /// Finish the handhshake which will create an open channel and reactor.

    ///

    /// The resulting channel is considered, by Tor protocol standard, a client/bridge relay

    /// channel meaning not authenticated. Circuit can be opened on it.

    #[instrument(skip_all, level = "trace")]

    {

        // This is either a client or a bridge so very sensitive.

        ));

        // Non verifiable responder channel, we simply finalize our underlying channel and we are

        // done. We are connected to a client or bridge.

}