//! Descriptions objects for different kinds of directory requests

//! that we can make.

use tor_llcrypto::pk::rsa::RsaIdentity;

use tor_netdoc::doc::authcert::AuthCertKeyIds;

use tor_netdoc::doc::microdesc::MdDigest;

use tor_netdoc::doc::netstatus::ConsensusFlavor;

#[cfg(feature = "routerdesc")]

use tor_netdoc::doc::routerdesc::RdDigest;

use tor_proto::circuit::ClientCirc;

#[cfg(feature = "hs-client")]

use tor_hscrypto::pk::HsBlindId;

/// Alias for a result with a `RequestError`.

type Result<T> = std::result::Result<T, crate::err::RequestError>;

use base64ct::{Base64Unpadded, Encoding as _};

use std::borrow::Cow;

use std::iter::FromIterator;

use std::time::{Duration, SystemTime};

use itertools::Itertools;

use crate::err::RequestError;

use crate::AnonymizedRequest;

/// Declare an inaccessible public type.

pub(crate) mod sealed {

    use super::{AnonymizedRequest, ClientCirc, Result};

    /// Sealed trait to help implement [`Requestable`](super::Requestable): not

    /// visible outside this crate, so we can change its methods however we like.

    pub trait RequestableInner: Send + Sync {

        /// Build an [`http::Request`] from this Requestable, if

        /// it is well-formed.

        //

        // TODO: This API is a bit troublesome in how it takes &self and

        // returns a Request<String>.  First, most Requestables don't actually have

        // a body to send, and for them having an empty String in their body is a

        // bit silly.  Second, taking a reference to self but returning an owned

        // String means that we will often have to clone an internal string owned by

        // this Requestable instance.

        fn make_request(&self) -> Result<http::Request<String>>;

        /// Return true if partial response bodies are potentially useful.

        ///

        /// This is true for request types where we're going to be downloading

        /// multiple documents, and we know how to parse out the ones we wanted

        /// if the answer is truncated.

        fn partial_response_body_ok(&self) -> bool;

        /// Return the maximum allowable response length we'll accept for this

        /// request.

        /// Return an error if there is some problem with the provided circuit that

        /// would keep it from being used for this request.

        /// Return a value to say whether this request must be anonymized.

        fn anonymized(&self) -> AnonymizedRequest;

    }

}

/// A request for an object that can be served over the Tor directory system.

pub trait Requestable: sealed::RequestableInner {

    /// Return a wrapper around this [`Requestable`] that implements `Debug`,

    /// and whose output shows the actual HTTP request that will be generated.

    ///

    /// The format is not guaranteed to  be stable.

}

impl<T: sealed::RequestableInner> Requestable for T {}

/// A wrapper to implement [`Requestable::debug_request`].

pub struct DisplayRequestable<'a, R: Requestable>(&'a R);

impl<'a, R: Requestable> std::fmt::Debug for DisplayRequestable<'a, R> {

}

/// How much clock skew do we allow in the distance between the directory

/// cache's clock and our own?

///

///  If we find more skew than this, we end the

/// request early, on the theory that the directory will not tell us any

/// information we'd accept.

#[derive(Clone, Debug)]

struct SkewLimit {

    /// We refuse to proceed if the directory says we are more fast than this.

    ///

    /// (This is equivalent to deciding that, from our perspective, the

    /// directory is at least this slow.)

    max_fast: Duration,

    /// We refuse to proceed if the directory says that we are more slow than

    /// this.

    ///

    /// (This is equivalent to deciding that, from our perspective, the

    /// directory is at least this fast.)

    max_slow: Duration,

}

/// A Request for a consensus directory.

#[derive(Debug, Clone)]

pub struct ConsensusRequest {

    /// What flavor of consensus are we asking for?  Right now, only

    /// "microdesc" and "ns" are supported.

    flavor: ConsensusFlavor,

    /// A list of the authority identities that we believe in.  We tell the

    /// directory cache only to give us a consensus if it is signed by enough

    /// of these authorities.

    authority_ids: Vec<RsaIdentity>,

    /// The publication time of the most recent consensus we have.  Used to

    /// generate an If-Modified-Since header so that we don't get a document

    /// we already have.

    last_consensus_published: Option<SystemTime>,

    /// A set of SHA3-256 digests of the _signed portion_ of consensuses we have.

    /// Used to declare what diffs we would accept.

    ///

    /// (Currently we don't send this, since we can't handle diffs.)

    last_consensus_sha3_256: Vec<[u8; 32]>,

    /// If present, the largest amount of clock skew to allow between ourself and a directory cache.

    skew_limit: Option<SkewLimit>,

}

impl ConsensusRequest {

    /// Create a new request for a consensus directory document.

    /// Add `id` to the list of authorities that this request should

    /// say we believe in.

    /// Add `d` to the list of consensus digests this request should

    /// say we already have.

    /// Set the publication time we should say we have for our last

    /// consensus to `when`.

    /// Return a slice of the consensus digests that we're saying we

    /// already have.

    /// Return an iterator of the authority identities that this request

    /// is saying we believe in.

    /// Return the date we're reporting for our most recent consensus.

    /// Tell the directory client that we should abort the request early if the

    /// directory's clock skew exceeds certain limits.

    ///

    /// The `max_fast` parameter is the most fast that we're willing to be with

    /// respect to the directory (or in other words, the most slow that we're

    /// willing to let the directory be with respect to us).

    ///

    /// The `max_slow` parameter is the most _slow_ that we're willing to be with

    /// respect to the directory ((or in other words, the most slow that we're

    /// willing to let the directory be with respect to us).

}

/// Convert a list of digests in some format to a string, for use in a request

///

/// The digests `DL` will be sorted, converted to strings with `EF`,

/// separated with `sep`, and returned as an fresh `String`.

///

/// If the digests list is empty, returns None instead.

//

// In principle this ought to be doable with much less allocating,

// starting with hex::encode etc.

impl Default for ConsensusRequest {

}

impl sealed::RequestableInner for ConsensusRequest {

        }

        // Without authorities, "../consensus-microdesc.z"

        // Possibly, add an if-modified-since header.

        // Possibly, add an X-Or-Diff-From-Consensus header.

        use tor_proto::ClockSkew::*;

        // This is the clock skew _according to the directory_.

            }

            }

        }

}

/// A request for one or more authority certificates.

#[derive(Debug, Clone, Default)]

pub struct AuthCertRequest {

    /// The identity/signing keys of the certificates we want.

    ids: Vec<AuthCertKeyIds>,

}

impl AuthCertRequest {

    /// Create a new request, asking for no authority certificates.

    /// Add `ids` to the list of certificates we're asking for.

    /// Return a list of the keys that we're asking for.

}

impl sealed::RequestableInner for AuthCertRequest {

}

impl FromIterator<AuthCertKeyIds> for AuthCertRequest {

}

/// A request for one or more microdescriptors

#[derive(Debug, Clone, Default)]

pub struct MicrodescRequest {

    /// The SHA256 digests of the microdescriptors we want.

    digests: Vec<MdDigest>,

}

impl MicrodescRequest {

    /// Construct a request for no microdescriptors.

    /// Add `d` to the list of microdescriptors we want to request.

    /// Return a list of the microdescriptor digests that we're asking for.

}

impl sealed::RequestableInner for MicrodescRequest {

}

impl FromIterator<MdDigest> for MicrodescRequest {

}

/// A request for one, many or all router descriptors.

#[derive(Debug, Clone)]

#[cfg(feature = "routerdesc")]

pub struct RouterDescRequest {

    /// The descriptors to request.

    requested_descriptors: RequestedDescs,

}

/// Tracks the different router descriptor types.

#[derive(Debug, Clone)]

#[cfg(feature = "routerdesc")]

enum RequestedDescs {

    /// If this is set, we just ask for all the descriptors.

    AllDescriptors,

    /// A list of digests to download.

    Digests(Vec<RdDigest>),

}

#[cfg(feature = "routerdesc")]

impl Default for RouterDescRequest {

}

#[cfg(feature = "routerdesc")]

impl RouterDescRequest {

    /// Construct a request for all router descriptors.

    /// Construct a new empty request.

}

#[cfg(feature = "routerdesc")]

impl sealed::RequestableInner for RouterDescRequest {

            }

        }

        }

        }

}

#[cfg(feature = "routerdesc")]

impl FromIterator<RdDigest> for RouterDescRequest {

}

/// A request for the descriptor of whatever relay we are making the request to

#[derive(Debug, Clone, Default)]

#[cfg(feature = "routerdesc")]

#[non_exhaustive]

pub struct RoutersOwnDescRequest {}

#[cfg(feature = "routerdesc")]

impl RoutersOwnDescRequest {

    /// Construct a new request.

}

#[cfg(feature = "routerdesc")]

impl sealed::RequestableInner for RoutersOwnDescRequest {

}

/// A request to download a hidden service descriptor

///

/// rend-spec-v3 2.2.6

#[derive(Debug, Clone)]

#[cfg(feature = "hs-client")]

pub struct HsDescDownloadRequest {

    /// What hidden service?

    hsid: HsBlindId,

    /// What's the largest acceptable response length?

    max_len: usize,

}

#[cfg(feature = "hs-client")]

impl HsDescDownloadRequest {

    /// Construct a request for a single onion service descriptor by its

    /// blinded ID.

        /// Default maximum length to use when we have no other information.

        const DEFAULT_HSDESC_MAX_LEN: usize = 50_000;

    /// Set the maximum acceptable response length.

}

#[cfg(feature = "hs-client")]

impl sealed::RequestableInner for HsDescDownloadRequest {

}

/// A request to upload a hidden service descriptor

///

/// rend-spec-v3 2.2.6

#[derive(Debug, Clone)]

#[cfg(feature = "hs-service")]

pub struct HsDescUploadRequest(String);

#[cfg(feature = "hs-service")]

impl HsDescUploadRequest {

    /// Construct a request for uploading a single onion service descriptor.

}

#[cfg(feature = "hs-service")]

impl sealed::RequestableInner for HsDescUploadRequest {

        /// The upload URI.

        const URI: &str = "/tor/hs/3/publish";

}

/// Encodings that all Tor clients support.

const UNIVERSAL_ENCODINGS: &str = "deflate, identity";

/// List all the encodings we accept

/// Add commonly used headers to the HTTP request.

///

/// (Right now, this is only Accept-Encoding.)

        AnonymizedRequest::Anonymized => {

            // In an anonymized request, we do not admit to supporting any

            // encoding besides those that are always available.

        }

    }

#[cfg(test)]

mod test {

    // @@ begin test lint list maintained by maint/add_warning @@

    #![allow(clippy::bool_assert_comparison)]

    #![allow(clippy::clone_on_copy)]

    #![allow(clippy::dbg_macro)]

    #![allow(clippy::mixed_attributes_style)]

    #![allow(clippy::print_stderr)]

    #![allow(clippy::print_stdout)]

    #![allow(clippy::single_char_pattern)]

    #![allow(clippy::unwrap_used)]

    #![allow(clippy::unchecked_duration_subtraction)]

    #![allow(clippy::useless_vec)]

    #![allow(clippy::needless_pass_by_value)]

    //! <!-- @@ end test lint list maintained by maint/add_warning @@ -->

    use super::sealed::RequestableInner;

    use super::*;

    #[test]

    fn test_md_request() -> Result<()> {

        let d1 = b"This is a testing digest. it isn";

        let d2 = b"'t actually SHA-256.............";

        let mut req = MicrodescRequest::default();

        req.push(*d1);

        assert!(!req.partial_response_body_ok());

        req.push(*d2);

        assert!(req.partial_response_body_ok());

        assert_eq!(req.max_response_len(), 16 << 10);

        let req = crate::util::encode_request(&req.make_request()?);

        assert_eq!(req,

                   format!("GET /tor/micro/d/J3QgYWN0dWFsbHkgU0hBLTI1Ni4uLi4uLi4uLi4uLi4-VGhpcyBpcyBhIHRlc3RpbmcgZGlnZXN0LiBpdCBpc24.z HTTP/1.0\r\naccept-encoding: {}\r\n\r\n", all_encodings()));

        // Try it with FromIterator, and use some accessors.

        let req2: MicrodescRequest = vec![*d1, *d2].into_iter().collect();

        let ds: Vec<_> = req2.digests().collect();

        assert_eq!(ds, vec![d1, d2]);

        let req2 = crate::util::encode_request(&req2.make_request()?);

        assert_eq!(req, req2);

        Ok(())

    }

    #[test]

    fn test_cert_request() -> Result<()> {

        let d1 = b"This is a testing dn";

        let d2 = b"'t actually SHA-256.";

        let key1 = AuthCertKeyIds {

            id_fingerprint: (*d1).into(),

            sk_fingerprint: (*d2).into(),

        };

        let d3 = b"blah blah blah 1 2 3";

        let d4 = b"I like pizza from Na";

        let key2 = AuthCertKeyIds {

            id_fingerprint: (*d3).into(),

            sk_fingerprint: (*d4).into(),

        };

        let mut req = AuthCertRequest::default();

        req.push(key1);

        assert!(!req.partial_response_body_ok());

        req.push(key2);

        assert!(req.partial_response_body_ok());

        assert_eq!(req.max_response_len(), 32 << 10);

        let keys: Vec<_> = req.keys().collect();

        assert_eq!(keys, vec![&key1, &key2]);

        let req = crate::util::encode_request(&req.make_request()?);

        assert_eq!(req,

                   format!("GET /tor/keys/fp-sk/5468697320697320612074657374696e6720646e-27742061637475616c6c79205348412d3235362e+626c616820626c616820626c6168203120322033-49206c696b652070697a7a612066726f6d204e61.z HTTP/1.0\r\naccept-encoding: {}\r\n\r\n", all_encodings()));

        let req2: AuthCertRequest = vec![key1, key2].into_iter().collect();

        let req2 = crate::util::encode_request(&req2.make_request()?);

        assert_eq!(req, req2);

        Ok(())

    }

    #[test]

    fn test_consensus_request() -> Result<()> {

        let d1 = RsaIdentity::from_bytes(

            &hex::decode("03479E93EBF3FF2C58C1C9DBF2DE9DE9C2801B3E").unwrap(),

        )

        .unwrap();

        let d2 = b"blah blah blah 12 blah blah blah";

        let d3 = SystemTime::now();

        let mut req = ConsensusRequest::default();

        let when = httpdate::fmt_http_date(d3);

        req.push_authority_id(d1);

        req.push_old_consensus_digest(*d2);

        req.set_last_consensus_date(d3);

        assert!(!req.partial_response_body_ok());

        assert_eq!(req.max_response_len(), (16 << 20) - 1);

        assert_eq!(req.old_consensus_digests().next(), Some(d2));

        assert_eq!(req.authority_ids().next(), Some(&d1));

        assert_eq!(req.last_consensus_date(), Some(d3));

        let req = crate::util::encode_request(&req.make_request()?);

        assert_eq!(req,

                   format!("GET /tor/status-vote/current/consensus-microdesc/03479e93ebf3ff2c58c1c9dbf2de9de9c2801b3e.z HTTP/1.0\r\naccept-encoding: {}\r\nif-modified-since: {}\r\nx-or-diff-from-consensus: 626c616820626c616820626c616820313220626c616820626c616820626c6168\r\n\r\n", all_encodings(), when));

        // Request without authorities

        let req = ConsensusRequest::default();

        let req = crate::util::encode_request(&req.make_request()?);

        assert_eq!(req,

                   format!("GET /tor/status-vote/current/consensus-microdesc.z HTTP/1.0\r\naccept-encoding: {}\r\n\r\n", all_encodings()));

        Ok(())

    }

    #[test]

    #[cfg(feature = "routerdesc")]

    fn test_rd_request_all() -> Result<()> {

        let req = RouterDescRequest::all();

        assert!(req.partial_response_body_ok());

        assert_eq!(req.max_response_len(), 1 << 26);

        let req = crate::util::encode_request(&req.make_request()?);

        assert_eq!(

            req,

            format!(

                "GET /tor/server/all.z HTTP/1.0\r\naccept-encoding: {}\r\n\r\n",

                all_encodings()

            )

        );

        Ok(())

    }

    #[test]

    #[cfg(feature = "routerdesc")]

    fn test_rd_request() -> Result<()> {

        let d1 = b"at some point I got ";

        let d2 = b"of writing in hex...";

        let mut req = RouterDescRequest::default();

        if let RequestedDescs::Digests(ref mut digests) = req.requested_descriptors {

            digests.push(*d1);

        }

        assert!(!req.partial_response_body_ok());

        if let RequestedDescs::Digests(ref mut digests) = req.requested_descriptors {

            digests.push(*d2);

        }

        assert!(req.partial_response_body_ok());

        assert_eq!(req.max_response_len(), 16 << 10);

        let req = crate::util::encode_request(&req.make_request()?);

        assert_eq!(req,

                   format!("GET /tor/server/d/617420736f6d6520706f696e74204920676f7420+6f662077726974696e6720696e206865782e2e2e.z HTTP/1.0\r\naccept-encoding: {}\r\n\r\n", all_encodings()));

        // Try it with FromIterator, and use some accessors.

        let req2: RouterDescRequest = vec![*d1, *d2].into_iter().collect();

        let ds: Vec<_> = match req2.requested_descriptors {

            RequestedDescs::Digests(ref digests) => digests.iter().collect(),

            RequestedDescs::AllDescriptors => Vec::new(),

        };

        assert_eq!(ds, vec![d1, d2]);

        let req2 = crate::util::encode_request(&req2.make_request()?);

        assert_eq!(req, req2);

        Ok(())

    }

    #[test]

    #[cfg(feature = "hs-client")]

    fn test_hs_desc_download_request() -> Result<()> {

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

        let hsid = [1, 2, 3, 4].iter().cycle().take(32).cloned().collect_vec();

        let hsid = Ed25519Identity::new(hsid[..].try_into().unwrap());

        let hsid = HsBlindId::from(hsid);

        let req = HsDescDownloadRequest::new(hsid);

        assert!(!req.partial_response_body_ok());

        assert_eq!(req.max_response_len(), 50 * 1000);

        let req = crate::util::encode_request(&req.make_request()?);

        assert_eq!(

            req,

            format!("GET /tor/hs/3/AQIDBAECAwQBAgMEAQIDBAECAwQBAgMEAQIDBAECAwQ HTTP/1.0\r\naccept-encoding: {}\r\n\r\n", UNIVERSAL_ENCODINGS)

        );

        Ok(())

    }

}