diff options
| author | mo khan <mo@mokhan.ca> | 2025-07-15 16:37:08 -0600 |
|---|---|---|
| committer | mo khan <mo@mokhan.ca> | 2025-07-17 16:30:22 -0600 |
| commit | 45df4d0d9b577fecee798d672695fe24ff57fb1b (patch) | |
| tree | 1b99bf645035b58e0d6db08c7a83521f41f7a75b /vendor/hyper-util/src/client | |
| parent | f94f79608393d4ab127db63cc41668445ef6b243 (diff) | |
feat: migrate from Cedar to SpiceDB authorization system
This is a major architectural change that replaces the Cedar policy-based
authorization system with SpiceDB's relation-based authorization.
Key changes:
- Migrate from Rust to Go implementation
- Replace Cedar policies with SpiceDB schema and relationships
- Switch from envoy `ext_authz` with Cedar to SpiceDB permission checks
- Update build system and dependencies for Go ecosystem
- Maintain Envoy integration for external authorization
This change enables more flexible permission modeling through SpiceDB's
Google Zanzibar inspired relation-based system, supporting complex
hierarchical permissions that were difficult to express in Cedar.
Breaking change: Existing Cedar policies and Rust-based configuration
will no longer work and need to be migrated to SpiceDB schema.
Diffstat (limited to 'vendor/hyper-util/src/client')
21 files changed, 0 insertions, 7679 deletions
diff --git a/vendor/hyper-util/src/client/client.rs b/vendor/hyper-util/src/client/client.rs deleted file mode 100644 index a9fb244a..00000000 --- a/vendor/hyper-util/src/client/client.rs +++ /dev/null @@ -1,132 +0,0 @@ -use hyper::{Request, Response}; -use tower::{Service, MakeService}; - -use super::connect::Connect; -use super::pool; - -pub struct Client<M> { - // Hi there. So, let's take a 0.14.x hyper::Client, and build up its layers - // here. We don't need to fully expose the layers to start with, but that - // is the end goal. - // - // Client = MakeSvcAsService< - // SetHost< - // Http1RequestTarget< - // DelayedRelease< - // ConnectingPool<C, P> - // > - // > - // > - // > - make_svc: M, -} - -// We might change this... :shrug: -type PoolKey = hyper::Uri; - -struct ConnectingPool<C, P> { - connector: C, - pool: P, -} - -struct PoolableSvc<S>(S); - -/// A marker to identify what version a pooled connection is. -#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)] -#[allow(dead_code)] -pub enum Ver { - Auto, - Http2, -} - -// ===== impl Client ===== - -impl<M, /*ReqBody, ResBody,*/ E> Client<M> -where - M: MakeService< - hyper::Uri, - Request<()>, - Response = Response<()>, - Error = E, - MakeError = E, - >, - //M: Service<hyper::Uri, Error = E>, - //M::Response: Service<Request<ReqBody>, Response = Response<ResBody>>, -{ - pub async fn request(&mut self, req: Request<()>) -> Result<Response<()>, E> { - let mut svc = self.make_svc.make_service(req.uri().clone()).await?; - svc.call(req).await - } -} - -impl<M, /*ReqBody, ResBody,*/ E> Client<M> -where - M: MakeService< - hyper::Uri, - Request<()>, - Response = Response<()>, - Error = E, - MakeError = E, - >, - //M: Service<hyper::Uri, Error = E>, - //M::Response: Service<Request<ReqBody>, Response = Response<ResBody>>, -{ - -} - -// ===== impl ConnectingPool ===== - -impl<C, P> ConnectingPool<C, P> -where - C: Connect, - C::_Svc: Unpin + Send + 'static, -{ - async fn connection_for(&self, target: PoolKey) -> Result<pool::Pooled<PoolableSvc<C::_Svc>, PoolKey>, ()> { - todo!() - } -} - -impl<S> pool::Poolable for PoolableSvc<S> -where - S: Unpin + Send + 'static, -{ - fn is_open(&self) -> bool { - /* - match self.tx { - PoolTx::Http1(ref tx) => tx.is_ready(), - #[cfg(feature = "http2")] - PoolTx::Http2(ref tx) => tx.is_ready(), - } - */ - true - } - - fn reserve(self) -> pool::Reservation<Self> { - /* - match self.tx { - PoolTx::Http1(tx) => Reservation::Unique(PoolClient { - conn_info: self.conn_info, - tx: PoolTx::Http1(tx), - }), - #[cfg(feature = "http2")] - PoolTx::Http2(tx) => { - let b = PoolClient { - conn_info: self.conn_info.clone(), - tx: PoolTx::Http2(tx.clone()), - }; - let a = PoolClient { - conn_info: self.conn_info, - tx: PoolTx::Http2(tx), - }; - Reservation::Shared(a, b) - } - } - */ - pool::Reservation::Unique(self) - } - - fn can_share(&self) -> bool { - false - //self.is_http2() - } -} diff --git a/vendor/hyper-util/src/client/legacy/client.rs b/vendor/hyper-util/src/client/legacy/client.rs deleted file mode 100644 index 9899d346..00000000 --- a/vendor/hyper-util/src/client/legacy/client.rs +++ /dev/null @@ -1,1690 +0,0 @@ -//! The legacy HTTP Client from 0.14.x -//! -//! This `Client` will eventually be deconstructed into more composable parts. -//! For now, to enable people to use hyper 1.0 quicker, this `Client` exists -//! in much the same way it did in hyper 0.14. - -use std::error::Error as StdError; -use std::fmt; -use std::future::Future; -use std::pin::Pin; -use std::task::{self, Poll}; -use std::time::Duration; - -use futures_util::future::{self, Either, FutureExt, TryFutureExt}; -use http::uri::Scheme; -use hyper::client::conn::TrySendError as ConnTrySendError; -use hyper::header::{HeaderValue, HOST}; -use hyper::rt::Timer; -use hyper::{body::Body, Method, Request, Response, Uri, Version}; -use tracing::{debug, trace, warn}; - -use super::connect::capture::CaptureConnectionExtension; -#[cfg(feature = "tokio")] -use super::connect::HttpConnector; -use super::connect::{Alpn, Connect, Connected, Connection}; -use super::pool::{self, Ver}; - -use crate::common::future::poll_fn; -use crate::common::{lazy as hyper_lazy, timer, Exec, Lazy, SyncWrapper}; - -type BoxSendFuture = Pin<Box<dyn Future<Output = ()> + Send>>; - -/// A Client to make outgoing HTTP requests. -/// -/// `Client` is cheap to clone and cloning is the recommended way to share a `Client`. The -/// underlying connection pool will be reused. -#[cfg_attr(docsrs, doc(cfg(any(feature = "http1", feature = "http2"))))] -pub struct Client<C, B> { - config: Config, - connector: C, - exec: Exec, - #[cfg(feature = "http1")] - h1_builder: hyper::client::conn::http1::Builder, - #[cfg(feature = "http2")] - h2_builder: hyper::client::conn::http2::Builder<Exec>, - pool: pool::Pool<PoolClient<B>, PoolKey>, -} - -#[derive(Clone, Copy, Debug)] -struct Config { - retry_canceled_requests: bool, - set_host: bool, - ver: Ver, -} - -/// Client errors -pub struct Error { - kind: ErrorKind, - source: Option<Box<dyn StdError + Send + Sync>>, - #[cfg(any(feature = "http1", feature = "http2"))] - connect_info: Option<Connected>, -} - -#[derive(Debug)] -enum ErrorKind { - Canceled, - ChannelClosed, - Connect, - UserUnsupportedRequestMethod, - UserUnsupportedVersion, - UserAbsoluteUriRequired, - SendRequest, -} - -macro_rules! e { - ($kind:ident) => { - Error { - kind: ErrorKind::$kind, - source: None, - connect_info: None, - } - }; - ($kind:ident, $src:expr) => { - Error { - kind: ErrorKind::$kind, - source: Some($src.into()), - connect_info: None, - } - }; -} - -// We might change this... :shrug: -type PoolKey = (http::uri::Scheme, http::uri::Authority); - -enum TrySendError<B> { - Retryable { - error: Error, - req: Request<B>, - connection_reused: bool, - }, - Nope(Error), -} - -/// A `Future` that will resolve to an HTTP Response. -/// -/// This is returned by `Client::request` (and `Client::get`). -#[must_use = "futures do nothing unless polled"] -pub struct ResponseFuture { - inner: SyncWrapper< - Pin<Box<dyn Future<Output = Result<Response<hyper::body::Incoming>, Error>> + Send>>, - >, -} - -// ===== impl Client ===== - -impl Client<(), ()> { - /// Create a builder to configure a new `Client`. - /// - /// # Example - /// - /// ``` - /// # #[cfg(feature = "tokio")] - /// # fn run () { - /// use std::time::Duration; - /// use hyper_util::client::legacy::Client; - /// use hyper_util::rt::{TokioExecutor, TokioTimer}; - /// - /// let client = Client::builder(TokioExecutor::new()) - /// .pool_timer(TokioTimer::new()) - /// .pool_idle_timeout(Duration::from_secs(30)) - /// .http2_only(true) - /// .build_http(); - /// # let infer: Client<_, http_body_util::Full<bytes::Bytes>> = client; - /// # drop(infer); - /// # } - /// # fn main() {} - /// ``` - pub fn builder<E>(executor: E) -> Builder - where - E: hyper::rt::Executor<BoxSendFuture> + Send + Sync + Clone + 'static, - { - Builder::new(executor) - } -} - -impl<C, B> Client<C, B> -where - C: Connect + Clone + Send + Sync + 'static, - B: Body + Send + 'static + Unpin, - B::Data: Send, - B::Error: Into<Box<dyn StdError + Send + Sync>>, -{ - /// Send a `GET` request to the supplied `Uri`. - /// - /// # Note - /// - /// This requires that the `Body` type have a `Default` implementation. - /// It *should* return an "empty" version of itself, such that - /// `Body::is_end_stream` is `true`. - /// - /// # Example - /// - /// ``` - /// # #[cfg(feature = "tokio")] - /// # fn run () { - /// use hyper::Uri; - /// use hyper_util::client::legacy::Client; - /// use hyper_util::rt::TokioExecutor; - /// use bytes::Bytes; - /// use http_body_util::Full; - /// - /// let client: Client<_, Full<Bytes>> = Client::builder(TokioExecutor::new()).build_http(); - /// - /// let future = client.get(Uri::from_static("http://httpbin.org/ip")); - /// # } - /// # fn main() {} - /// ``` - pub fn get(&self, uri: Uri) -> ResponseFuture - where - B: Default, - { - let body = B::default(); - if !body.is_end_stream() { - warn!("default Body used for get() does not return true for is_end_stream"); - } - - let mut req = Request::new(body); - *req.uri_mut() = uri; - self.request(req) - } - - /// Send a constructed `Request` using this `Client`. - /// - /// # Example - /// - /// ``` - /// # #[cfg(feature = "tokio")] - /// # fn run () { - /// use hyper::{Method, Request}; - /// use hyper_util::client::legacy::Client; - /// use http_body_util::Full; - /// use hyper_util::rt::TokioExecutor; - /// use bytes::Bytes; - /// - /// let client: Client<_, Full<Bytes>> = Client::builder(TokioExecutor::new()).build_http(); - /// - /// let req: Request<Full<Bytes>> = Request::builder() - /// .method(Method::POST) - /// .uri("http://httpbin.org/post") - /// .body(Full::from("Hallo!")) - /// .expect("request builder"); - /// - /// let future = client.request(req); - /// # } - /// # fn main() {} - /// ``` - pub fn request(&self, mut req: Request<B>) -> ResponseFuture { - let is_http_connect = req.method() == Method::CONNECT; - match req.version() { - Version::HTTP_11 => (), - Version::HTTP_10 => { - if is_http_connect { - warn!("CONNECT is not allowed for HTTP/1.0"); - return ResponseFuture::new(future::err(e!(UserUnsupportedRequestMethod))); - } - } - Version::HTTP_2 => (), - // completely unsupported HTTP version (like HTTP/0.9)! - other => return ResponseFuture::error_version(other), - }; - - let pool_key = match extract_domain(req.uri_mut(), is_http_connect) { - Ok(s) => s, - Err(err) => { - return ResponseFuture::new(future::err(err)); - } - }; - - ResponseFuture::new(self.clone().send_request(req, pool_key)) - } - - async fn send_request( - self, - mut req: Request<B>, - pool_key: PoolKey, - ) -> Result<Response<hyper::body::Incoming>, Error> { - let uri = req.uri().clone(); - - loop { - req = match self.try_send_request(req, pool_key.clone()).await { - Ok(resp) => return Ok(resp), - Err(TrySendError::Nope(err)) => return Err(err), - Err(TrySendError::Retryable { - mut req, - error, - connection_reused, - }) => { - if !self.config.retry_canceled_requests || !connection_reused { - // if client disabled, don't retry - // a fresh connection means we definitely can't retry - return Err(error); - } - - trace!( - "unstarted request canceled, trying again (reason={:?})", - error - ); - *req.uri_mut() = uri.clone(); - req - } - } - } - } - - async fn try_send_request( - &self, - mut req: Request<B>, - pool_key: PoolKey, - ) -> Result<Response<hyper::body::Incoming>, TrySendError<B>> { - let mut pooled = self - .connection_for(pool_key) - .await - // `connection_for` already retries checkout errors, so if - // it returns an error, there's not much else to retry - .map_err(TrySendError::Nope)?; - - if let Some(conn) = req.extensions_mut().get_mut::<CaptureConnectionExtension>() { - conn.set(&pooled.conn_info); - } - - if pooled.is_http1() { - if req.version() == Version::HTTP_2 { - warn!("Connection is HTTP/1, but request requires HTTP/2"); - return Err(TrySendError::Nope( - e!(UserUnsupportedVersion).with_connect_info(pooled.conn_info.clone()), - )); - } - - if self.config.set_host { - let uri = req.uri().clone(); - req.headers_mut().entry(HOST).or_insert_with(|| { - let hostname = uri.host().expect("authority implies host"); - if let Some(port) = get_non_default_port(&uri) { - let s = format!("{hostname}:{port}"); - HeaderValue::from_str(&s) - } else { - HeaderValue::from_str(hostname) - } - .expect("uri host is valid header value") - }); - } - - // CONNECT always sends authority-form, so check it first... - if req.method() == Method::CONNECT { - authority_form(req.uri_mut()); - } else if pooled.conn_info.is_proxied { - absolute_form(req.uri_mut()); - } else { - origin_form(req.uri_mut()); - } - } else if req.method() == Method::CONNECT && !pooled.is_http2() { - authority_form(req.uri_mut()); - } - - let mut res = match pooled.try_send_request(req).await { - Ok(res) => res, - Err(mut err) => { - return if let Some(req) = err.take_message() { - Err(TrySendError::Retryable { - connection_reused: pooled.is_reused(), - error: e!(Canceled, err.into_error()) - .with_connect_info(pooled.conn_info.clone()), - req, - }) - } else { - Err(TrySendError::Nope( - e!(SendRequest, err.into_error()) - .with_connect_info(pooled.conn_info.clone()), - )) - } - } - }; - - // If the Connector included 'extra' info, add to Response... - if let Some(extra) = &pooled.conn_info.extra { - extra.set(res.extensions_mut()); - } - - // If pooled is HTTP/2, we can toss this reference immediately. - // - // when pooled is dropped, it will try to insert back into the - // pool. To delay that, spawn a future that completes once the - // sender is ready again. - // - // This *should* only be once the related `Connection` has polled - // for a new request to start. - // - // It won't be ready if there is a body to stream. - if pooled.is_http2() || !pooled.is_pool_enabled() || pooled.is_ready() { - drop(pooled); - } else if !res.body().is_end_stream() { - //let (delayed_tx, delayed_rx) = oneshot::channel::<()>(); - //res.body_mut().delayed_eof(delayed_rx); - let on_idle = poll_fn(move |cx| pooled.poll_ready(cx)).map(move |_| { - // At this point, `pooled` is dropped, and had a chance - // to insert into the pool (if conn was idle) - //drop(delayed_tx); - }); - - self.exec.execute(on_idle); - } else { - // There's no body to delay, but the connection isn't - // ready yet. Only re-insert when it's ready - let on_idle = poll_fn(move |cx| pooled.poll_ready(cx)).map(|_| ()); - - self.exec.execute(on_idle); - } - - Ok(res) - } - - async fn connection_for( - &self, - pool_key: PoolKey, - ) -> Result<pool::Pooled<PoolClient<B>, PoolKey>, Error> { - loop { - match self.one_connection_for(pool_key.clone()).await { - Ok(pooled) => return Ok(pooled), - Err(ClientConnectError::Normal(err)) => return Err(err), - Err(ClientConnectError::CheckoutIsClosed(reason)) => { - if !self.config.retry_canceled_requests { - return Err(e!(Connect, reason)); - } - - trace!( - "unstarted request canceled, trying again (reason={:?})", - reason, - ); - continue; - } - }; - } - } - - async fn one_connection_for( - &self, - pool_key: PoolKey, - ) -> Result<pool::Pooled<PoolClient<B>, PoolKey>, ClientConnectError> { - // Return a single connection if pooling is not enabled - if !self.pool.is_enabled() { - return self - .connect_to(pool_key) - .await - .map_err(ClientConnectError::Normal); - } - - // This actually races 2 different futures to try to get a ready - // connection the fastest, and to reduce connection churn. - // - // - If the pool has an idle connection waiting, that's used - // immediately. - // - Otherwise, the Connector is asked to start connecting to - // the destination Uri. - // - Meanwhile, the pool Checkout is watching to see if any other - // request finishes and tries to insert an idle connection. - // - If a new connection is started, but the Checkout wins after - // (an idle connection became available first), the started - // connection future is spawned into the runtime to complete, - // and then be inserted into the pool as an idle connection. - let checkout = self.pool.checkout(pool_key.clone()); - let connect = self.connect_to(pool_key); - let is_ver_h2 = self.config.ver == Ver::Http2; - - // The order of the `select` is depended on below... - - match future::select(checkout, connect).await { - // Checkout won, connect future may have been started or not. - // - // If it has, let it finish and insert back into the pool, - // so as to not waste the socket... - Either::Left((Ok(checked_out), connecting)) => { - // This depends on the `select` above having the correct - // order, such that if the checkout future were ready - // immediately, the connect future will never have been - // started. - // - // If it *wasn't* ready yet, then the connect future will - // have been started... - if connecting.started() { - let bg = connecting - .map_err(|err| { - trace!("background connect error: {}", err); - }) - .map(|_pooled| { - // dropping here should just place it in - // the Pool for us... - }); - // An execute error here isn't important, we're just trying - // to prevent a waste of a socket... - self.exec.execute(bg); - } - Ok(checked_out) - } - // Connect won, checkout can just be dropped. - Either::Right((Ok(connected), _checkout)) => Ok(connected), - // Either checkout or connect could get canceled: - // - // 1. Connect is canceled if this is HTTP/2 and there is - // an outstanding HTTP/2 connecting task. - // 2. Checkout is canceled if the pool cannot deliver an - // idle connection reliably. - // - // In both cases, we should just wait for the other future. - Either::Left((Err(err), connecting)) => { - if err.is_canceled() { - connecting.await.map_err(ClientConnectError::Normal) - } else { - Err(ClientConnectError::Normal(e!(Connect, err))) - } - } - Either::Right((Err(err), checkout)) => { - if err.is_canceled() { - checkout.await.map_err(move |err| { - if is_ver_h2 && err.is_canceled() { - ClientConnectError::CheckoutIsClosed(err) - } else { - ClientConnectError::Normal(e!(Connect, err)) - } - }) - } else { - Err(ClientConnectError::Normal(err)) - } - } - } - } - - #[cfg(any(feature = "http1", feature = "http2"))] - fn connect_to( - &self, - pool_key: PoolKey, - ) -> impl Lazy<Output = Result<pool::Pooled<PoolClient<B>, PoolKey>, Error>> + Send + Unpin - { - let executor = self.exec.clone(); - let pool = self.pool.clone(); - #[cfg(feature = "http1")] - let h1_builder = self.h1_builder.clone(); - #[cfg(feature = "http2")] - let h2_builder = self.h2_builder.clone(); - let ver = self.config.ver; - let is_ver_h2 = ver == Ver::Http2; - let connector = self.connector.clone(); - let dst = domain_as_uri(pool_key.clone()); - hyper_lazy(move || { - // Try to take a "connecting lock". - // - // If the pool_key is for HTTP/2, and there is already a - // connection being established, then this can't take a - // second lock. The "connect_to" future is Canceled. - let connecting = match pool.connecting(&pool_key, ver) { - Some(lock) => lock, - None => { - let canceled = e!(Canceled); - // TODO - //crate::Error::new_canceled().with("HTTP/2 connection in progress"); - return Either::Right(future::err(canceled)); - } - }; - Either::Left( - connector - .connect(super::connect::sealed::Internal, dst) - .map_err(|src| e!(Connect, src)) - .and_then(move |io| { - let connected = io.connected(); - // If ALPN is h2 and we aren't http2_only already, - // then we need to convert our pool checkout into - // a single HTTP2 one. - let connecting = if connected.alpn == Alpn::H2 && !is_ver_h2 { - match connecting.alpn_h2(&pool) { - Some(lock) => { - trace!("ALPN negotiated h2, updating pool"); - lock - } - None => { - // Another connection has already upgraded, - // the pool checkout should finish up for us. - let canceled = e!(Canceled, "ALPN upgraded to HTTP/2"); - return Either::Right(future::err(canceled)); - } - } - } else { - connecting - }; - - #[cfg_attr(not(feature = "http2"), allow(unused))] - let is_h2 = is_ver_h2 || connected.alpn == Alpn::H2; - - Either::Left(Box::pin(async move { - let tx = if is_h2 { - #[cfg(feature = "http2")] { - let (mut tx, conn) = - h2_builder.handshake(io).await.map_err(Error::tx)?; - - trace!( - "http2 handshake complete, spawning background dispatcher task" - ); - executor.execute( - conn.map_err(|e| debug!("client connection error: {}", e)) - .map(|_| ()), - ); - - // Wait for 'conn' to ready up before we - // declare this tx as usable - tx.ready().await.map_err(Error::tx)?; - PoolTx::Http2(tx) - } - #[cfg(not(feature = "http2"))] - panic!("http2 feature is not enabled"); - } else { - #[cfg(feature = "http1")] { - // Perform the HTTP/1.1 handshake on the provided I/O stream. - // Uses the h1_builder to establish a connection, returning a sender (tx) for requests - // and a connection task (conn) that manages the connection lifecycle. - let (mut tx, conn) = - h1_builder.handshake(io).await.map_err(crate::client::legacy::client::Error::tx)?; - - // Log that the HTTP/1.1 handshake has completed successfully. - // This indicates the connection is established and ready for request processing. - trace!( - "http1 handshake complete, spawning background dispatcher task" - ); - // Create a oneshot channel to communicate errors from the connection task. - // err_tx sends errors from the connection task, and err_rx receives them - // to correlate connection failures with request readiness errors. - let (err_tx, err_rx) = tokio::sync::oneshot::channel(); - // Spawn the connection task in the background using the executor. - // The task manages the HTTP/1.1 connection, including upgrades (e.g., WebSocket). - // Errors are sent via err_tx to ensure they can be checked if the sender (tx) fails. - executor.execute( - conn.with_upgrades() - .map_err(|e| { - // Log the connection error at debug level for diagnostic purposes. - debug!("client connection error: {:?}", e); - // Log that the error is being sent to the error channel. - trace!("sending connection error to error channel"); - // Send the error via the oneshot channel, ignoring send failures - // (e.g., if the receiver is dropped, which is handled later). - let _ =err_tx.send(e); - }) - .map(|_| ()), - ); - // Log that the client is waiting for the connection to be ready. - // Readiness indicates the sender (tx) can accept a request without blocking. - trace!("waiting for connection to be ready"); - // Check if the sender is ready to accept a request. - // This ensures the connection is fully established before proceeding. - // aka: - // Wait for 'conn' to ready up before we - // declare this tx as usable - match tx.ready().await { - // If ready, the connection is usable for sending requests. - Ok(_) => { - // Log that the connection is ready for use. - trace!("connection is ready"); - // Drop the error receiver, as it’s no longer needed since the sender is ready. - // This prevents waiting for errors that won’t occur in a successful case. - drop(err_rx); - // Wrap the sender in PoolTx::Http1 for use in the connection pool. - PoolTx::Http1(tx) - } - // If the sender fails with a closed channel error, check for a specific connection error. - // This distinguishes between a vague ChannelClosed error and an actual connection failure. - Err(e) if e.is_closed() => { - // Log that the channel is closed, indicating a potential connection issue. - trace!("connection channel closed, checking for connection error"); - // Check the oneshot channel for a specific error from the connection task. - match err_rx.await { - // If an error was received, it’s a specific connection failure. - Ok(err) => { - // Log the specific connection error for diagnostics. - trace!("received connection error: {:?}", err); - // Return the error wrapped in Error::tx to propagate it. - return Err(crate::client::legacy::client::Error::tx(err)); - } - // If the error channel is closed, no specific error was sent. - // Fall back to the vague ChannelClosed error. - Err(_) => { - // Log that the error channel is closed, indicating no specific error. - trace!("error channel closed, returning the vague ChannelClosed error"); - // Return the original error wrapped in Error::tx. - return Err(crate::client::legacy::client::Error::tx(e)); - } - } - } - // For other errors (e.g., timeout, I/O issues), propagate them directly. - // These are not ChannelClosed errors and don’t require error channel checks. - Err(e) => { - // Log the specific readiness failure for diagnostics. - trace!("connection readiness failed: {:?}", e); - // Return the error wrapped in Error::tx to propagate it. - return Err(crate::client::legacy::client::Error::tx(e)); - } - } - } - #[cfg(not(feature = "http1"))] { - panic!("http1 feature is not enabled"); - } - }; - - Ok(pool.pooled( - connecting, - PoolClient { - conn_info: connected, - tx, - }, - )) - })) - }), - ) - }) - } -} - -impl<C, B> tower_service::Service<Request<B>> for Client<C, B> -where - C: Connect + Clone + Send + Sync + 'static, - B: Body + Send + 'static + Unpin, - B::Data: Send, - B::Error: Into<Box<dyn StdError + Send + Sync>>, -{ - type Response = Response<hyper::body::Incoming>; - type Error = Error; - type Future = ResponseFuture; - - fn poll_ready(&mut self, _: &mut task::Context<'_>) -> Poll<Result<(), Self::Error>> { - Poll::Ready(Ok(())) - } - - fn call(&mut self, req: Request<B>) -> Self::Future { - self.request(req) - } -} - -impl<C, B> tower_service::Service<Request<B>> for &'_ Client<C, B> -where - C: Connect + Clone + Send + Sync + 'static, - B: Body + Send + 'static + Unpin, - B::Data: Send, - B::Error: Into<Box<dyn StdError + Send + Sync>>, -{ - type Response = Response<hyper::body::Incoming>; - type Error = Error; - type Future = ResponseFuture; - - fn poll_ready(&mut self, _: &mut task::Context<'_>) -> Poll<Result<(), Self::Error>> { - Poll::Ready(Ok(())) - } - - fn call(&mut self, req: Request<B>) -> Self::Future { - self.request(req) - } -} - -impl<C: Clone, B> Clone for Client<C, B> { - fn clone(&self) -> Client<C, B> { - Client { - config: self.config, - exec: self.exec.clone(), - #[cfg(feature = "http1")] - h1_builder: self.h1_builder.clone(), - #[cfg(feature = "http2")] - h2_builder: self.h2_builder.clone(), - connector: self.connector.clone(), - pool: self.pool.clone(), - } - } -} - -impl<C, B> fmt::Debug for Client<C, B> { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.debug_struct("Client").finish() - } -} - -// ===== impl ResponseFuture ===== - -impl ResponseFuture { - fn new<F>(value: F) -> Self - where - F: Future<Output = Result<Response<hyper::body::Incoming>, Error>> + Send + 'static, - { - Self { - inner: SyncWrapper::new(Box::pin(value)), - } - } - - fn error_version(ver: Version) -> Self { - warn!("Request has unsupported version \"{:?}\"", ver); - ResponseFuture::new(Box::pin(future::err(e!(UserUnsupportedVersion)))) - } -} - -impl fmt::Debug for ResponseFuture { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.pad("Future<Response>") - } -} - -impl Future for ResponseFuture { - type Output = Result<Response<hyper::body::Incoming>, Error>; - - fn poll(mut self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Self::Output> { - self.inner.get_mut().as_mut().poll(cx) - } -} - -// ===== impl PoolClient ===== - -// FIXME: allow() required due to `impl Trait` leaking types to this lint -#[allow(missing_debug_implementations)] -struct PoolClient<B> { - conn_info: Connected, - tx: PoolTx<B>, -} - -enum PoolTx<B> { - #[cfg(feature = "http1")] - Http1(hyper::client::conn::http1::SendRequest<B>), - #[cfg(feature = "http2")] - Http2(hyper::client::conn::http2::SendRequest<B>), -} - -impl<B> PoolClient<B> { - fn poll_ready( - &mut self, - #[allow(unused_variables)] cx: &mut task::Context<'_>, - ) -> Poll<Result<(), Error>> { - match self.tx { - #[cfg(feature = "http1")] - PoolTx::Http1(ref mut tx) => tx.poll_ready(cx).map_err(Error::closed), - #[cfg(feature = "http2")] - PoolTx::Http2(_) => Poll::Ready(Ok(())), - } - } - - fn is_http1(&self) -> bool { - !self.is_http2() - } - - fn is_http2(&self) -> bool { - match self.tx { - #[cfg(feature = "http1")] - PoolTx::Http1(_) => false, - #[cfg(feature = "http2")] - PoolTx::Http2(_) => true, - } - } - - fn is_poisoned(&self) -> bool { - self.conn_info.poisoned.poisoned() - } - - fn is_ready(&self) -> bool { - match self.tx { - #[cfg(feature = "http1")] - PoolTx::Http1(ref tx) => tx.is_ready(), - #[cfg(feature = "http2")] - PoolTx::Http2(ref tx) => tx.is_ready(), - } - } -} - -impl<B: Body + 'static> PoolClient<B> { - fn try_send_request( - &mut self, - req: Request<B>, - ) -> impl Future<Output = Result<Response<hyper::body::Incoming>, ConnTrySendError<Request<B>>>> - where - B: Send, - { - #[cfg(all(feature = "http1", feature = "http2"))] - return match self.tx { - #[cfg(feature = "http1")] - PoolTx::Http1(ref mut tx) => Either::Left(tx.try_send_request(req)), - #[cfg(feature = "http2")] - PoolTx::Http2(ref mut tx) => Either::Right(tx.try_send_request(req)), - }; - - #[cfg(feature = "http1")] - #[cfg(not(feature = "http2"))] - return match self.tx { - #[cfg(feature = "http1")] - PoolTx::Http1(ref mut tx) => tx.try_send_request(req), - }; - - #[cfg(not(feature = "http1"))] - #[cfg(feature = "http2")] - return match self.tx { - #[cfg(feature = "http2")] - PoolTx::Http2(ref mut tx) => tx.try_send_request(req), - }; - } -} - -impl<B> pool::Poolable for PoolClient<B> -where - B: Send + 'static, -{ - fn is_open(&self) -> bool { - !self.is_poisoned() && self.is_ready() - } - - fn reserve(self) -> pool::Reservation<Self> { - match self.tx { - #[cfg(feature = "http1")] - PoolTx::Http1(tx) => pool::Reservation::Unique(PoolClient { - conn_info: self.conn_info, - tx: PoolTx::Http1(tx), - }), - #[cfg(feature = "http2")] - PoolTx::Http2(tx) => { - let b = PoolClient { - conn_info: self.conn_info.clone(), - tx: PoolTx::Http2(tx.clone()), - }; - let a = PoolClient { - conn_info: self.conn_info, - tx: PoolTx::Http2(tx), - }; - pool::Reservation::Shared(a, b) - } - } - } - - fn can_share(&self) -> bool { - self.is_http2() - } -} - -enum ClientConnectError { - Normal(Error), - CheckoutIsClosed(pool::Error), -} - -fn origin_form(uri: &mut Uri) { - let path = match uri.path_and_query() { - Some(path) if path.as_str() != "/" => { - let mut parts = ::http::uri::Parts::default(); - parts.path_and_query = Some(path.clone()); - Uri::from_parts(parts).expect("path is valid uri") - } - _none_or_just_slash => { - debug_assert!(Uri::default() == "/"); - Uri::default() - } - }; - *uri = path -} - -fn absolute_form(uri: &mut Uri) { - debug_assert!(uri.scheme().is_some(), "absolute_form needs a scheme"); - debug_assert!( - uri.authority().is_some(), - "absolute_form needs an authority" - ); - // If the URI is to HTTPS, and the connector claimed to be a proxy, - // then it *should* have tunneled, and so we don't want to send - // absolute-form in that case. - if uri.scheme() == Some(&Scheme::HTTPS) { - origin_form(uri); - } -} - -fn authority_form(uri: &mut Uri) { - if let Some(path) = uri.path_and_query() { - // `https://hyper.rs` would parse with `/` path, don't - // annoy people about that... - if path != "/" { - warn!("HTTP/1.1 CONNECT request stripping path: {:?}", path); - } - } - *uri = match uri.authority() { - Some(auth) => { - let mut parts = ::http::uri::Parts::default(); - parts.authority = Some(auth.clone()); - Uri::from_parts(parts).expect("authority is valid") - } - None => { - unreachable!("authority_form with relative uri"); - } - }; -} - -fn extract_domain(uri: &mut Uri, is_http_connect: bool) -> Result<PoolKey, Error> { - let uri_clone = uri.clone(); - match (uri_clone.scheme(), uri_clone.authority()) { - (Some(scheme), Some(auth)) => Ok((scheme.clone(), auth.clone())), - (None, Some(auth)) if is_http_connect => { - let scheme = match auth.port_u16() { - Some(443) => { - set_scheme(uri, Scheme::HTTPS); - Scheme::HTTPS - } - _ => { - set_scheme(uri, Scheme::HTTP); - Scheme::HTTP - } - }; - Ok((scheme, auth.clone())) - } - _ => { - debug!("Client requires absolute-form URIs, received: {:?}", uri); - Err(e!(UserAbsoluteUriRequired)) - } - } -} - -fn domain_as_uri((scheme, auth): PoolKey) -> Uri { - http::uri::Builder::new() - .scheme(scheme) - .authority(auth) - .path_and_query("/") - .build() - .expect("domain is valid Uri") -} - -fn set_scheme(uri: &mut Uri, scheme: Scheme) { - debug_assert!( - uri.scheme().is_none(), - "set_scheme expects no existing scheme" - ); - let old = std::mem::take(uri); - let mut parts: ::http::uri::Parts = old.into(); - parts.scheme = Some(scheme); - parts.path_and_query = Some("/".parse().expect("slash is a valid path")); - *uri = Uri::from_parts(parts).expect("scheme is valid"); -} - -fn get_non_default_port(uri: &Uri) -> Option<http::uri::Port<&str>> { - match (uri.port().map(|p| p.as_u16()), is_schema_secure(uri)) { - (Some(443), true) => None, - (Some(80), false) => None, - _ => uri.port(), - } -} - -fn is_schema_secure(uri: &Uri) -> bool { - uri.scheme_str() - .map(|scheme_str| matches!(scheme_str, "wss" | "https")) - .unwrap_or_default() -} - -/// A builder to configure a new [`Client`](Client). -/// -/// # Example -/// -/// ``` -/// # #[cfg(feature = "tokio")] -/// # fn run () { -/// use std::time::Duration; -/// use hyper_util::client::legacy::Client; -/// use hyper_util::rt::TokioExecutor; -/// -/// let client = Client::builder(TokioExecutor::new()) -/// .pool_idle_timeout(Duration::from_secs(30)) -/// .http2_only(true) -/// .build_http(); -/// # let infer: Client<_, http_body_util::Full<bytes::Bytes>> = client; -/// # drop(infer); -/// # } -/// # fn main() {} -/// ``` -#[cfg_attr(docsrs, doc(cfg(any(feature = "http1", feature = "http2"))))] -#[derive(Clone)] -pub struct Builder { - client_config: Config, - exec: Exec, - #[cfg(feature = "http1")] - h1_builder: hyper::client::conn::http1::Builder, - #[cfg(feature = "http2")] - h2_builder: hyper::client::conn::http2::Builder<Exec>, - pool_config: pool::Config, - pool_timer: Option<timer::Timer>, -} - -impl Builder { - /// Construct a new Builder. - pub fn new<E>(executor: E) -> Self - where - E: hyper::rt::Executor<BoxSendFuture> + Send + Sync + Clone + 'static, - { - let exec = Exec::new(executor); - Self { - client_config: Config { - retry_canceled_requests: true, - set_host: true, - ver: Ver::Auto, - }, - exec: exec.clone(), - #[cfg(feature = "http1")] - h1_builder: hyper::client::conn::http1::Builder::new(), - #[cfg(feature = "http2")] - h2_builder: hyper::client::conn::http2::Builder::new(exec), - pool_config: pool::Config { - idle_timeout: Some(Duration::from_secs(90)), - max_idle_per_host: usize::MAX, - }, - pool_timer: None, - } - } - /// Set an optional timeout for idle sockets being kept-alive. - /// A `Timer` is required for this to take effect. See `Builder::pool_timer` - /// - /// Pass `None` to disable timeout. - /// - /// Default is 90 seconds. - /// - /// # Example - /// - /// ``` - /// # #[cfg(feature = "tokio")] - /// # fn run () { - /// use std::time::Duration; - /// use hyper_util::client::legacy::Client; - /// use hyper_util::rt::{TokioExecutor, TokioTimer}; - /// - /// let client = Client::builder(TokioExecutor::new()) - /// .pool_idle_timeout(Duration::from_secs(30)) - /// .pool_timer(TokioTimer::new()) - /// .build_http(); - /// - /// # let infer: Client<_, http_body_util::Full<bytes::Bytes>> = client; - /// # } - /// # fn main() {} - /// ``` - pub fn pool_idle_timeout<D>(&mut self, val: D) -> &mut Self - where - D: Into<Option<Duration>>, - { - self.pool_config.idle_timeout = val.into(); - self - } - - #[doc(hidden)] - #[deprecated(note = "renamed to `pool_max_idle_per_host`")] - pub fn max_idle_per_host(&mut self, max_idle: usize) -> &mut Self { - self.pool_config.max_idle_per_host = max_idle; - self - } - - /// Sets the maximum idle connection per host allowed in the pool. - /// - /// Default is `usize::MAX` (no limit). - pub fn pool_max_idle_per_host(&mut self, max_idle: usize) -> &mut Self { - self.pool_config.max_idle_per_host = max_idle; - self - } - - // HTTP/1 options - - /// Sets the exact size of the read buffer to *always* use. - /// - /// Note that setting this option unsets the `http1_max_buf_size` option. - /// - /// Default is an adaptive read buffer. - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_read_buf_exact_size(&mut self, sz: usize) -> &mut Self { - self.h1_builder.read_buf_exact_size(Some(sz)); - self - } - - /// Set the maximum buffer size for the connection. - /// - /// Default is ~400kb. - /// - /// Note that setting this option unsets the `http1_read_exact_buf_size` option. - /// - /// # Panics - /// - /// The minimum value allowed is 8192. This method panics if the passed `max` is less than the minimum. - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_max_buf_size(&mut self, max: usize) -> &mut Self { - self.h1_builder.max_buf_size(max); - self - } - - /// Set whether HTTP/1 connections will accept spaces between header names - /// and the colon that follow them in responses. - /// - /// Newline codepoints (`\r` and `\n`) will be transformed to spaces when - /// parsing. - /// - /// You probably don't need this, here is what [RFC 7230 Section 3.2.4.] has - /// to say about it: - /// - /// > No whitespace is allowed between the header field-name and colon. In - /// > the past, differences in the handling of such whitespace have led to - /// > security vulnerabilities in request routing and response handling. A - /// > server MUST reject any received request message that contains - /// > whitespace between a header field-name and colon with a response code - /// > of 400 (Bad Request). A proxy MUST remove any such whitespace from a - /// > response message before forwarding the message downstream. - /// - /// Note that this setting does not affect HTTP/2. - /// - /// Default is false. - /// - /// [RFC 7230 Section 3.2.4.]: https://tools.ietf.org/html/rfc7230#section-3.2.4 - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_allow_spaces_after_header_name_in_responses(&mut self, val: bool) -> &mut Self { - self.h1_builder - .allow_spaces_after_header_name_in_responses(val); - self - } - - /// Set whether HTTP/1 connections will accept obsolete line folding for - /// header values. - /// - /// You probably don't need this, here is what [RFC 7230 Section 3.2.4.] has - /// to say about it: - /// - /// > A server that receives an obs-fold in a request message that is not - /// > within a message/http container MUST either reject the message by - /// > sending a 400 (Bad Request), preferably with a representation - /// > explaining that obsolete line folding is unacceptable, or replace - /// > each received obs-fold with one or more SP octets prior to - /// > interpreting the field value or forwarding the message downstream. - /// - /// > A proxy or gateway that receives an obs-fold in a response message - /// > that is not within a message/http container MUST either discard the - /// > message and replace it with a 502 (Bad Gateway) response, preferably - /// > with a representation explaining that unacceptable line folding was - /// > received, or replace each received obs-fold with one or more SP - /// > octets prior to interpreting the field value or forwarding the - /// > message downstream. - /// - /// > A user agent that receives an obs-fold in a response message that is - /// > not within a message/http container MUST replace each received - /// > obs-fold with one or more SP octets prior to interpreting the field - /// > value. - /// - /// Note that this setting does not affect HTTP/2. - /// - /// Default is false. - /// - /// [RFC 7230 Section 3.2.4.]: https://tools.ietf.org/html/rfc7230#section-3.2.4 - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_allow_obsolete_multiline_headers_in_responses(&mut self, val: bool) -> &mut Self { - self.h1_builder - .allow_obsolete_multiline_headers_in_responses(val); - self - } - - /// Sets whether invalid header lines should be silently ignored in HTTP/1 responses. - /// - /// This mimics the behaviour of major browsers. You probably don't want this. - /// You should only want this if you are implementing a proxy whose main - /// purpose is to sit in front of browsers whose users access arbitrary content - /// which may be malformed, and they expect everything that works without - /// the proxy to keep working with the proxy. - /// - /// This option will prevent Hyper's client from returning an error encountered - /// when parsing a header, except if the error was caused by the character NUL - /// (ASCII code 0), as Chrome specifically always reject those. - /// - /// The ignorable errors are: - /// * empty header names; - /// * characters that are not allowed in header names, except for `\0` and `\r`; - /// * when `allow_spaces_after_header_name_in_responses` is not enabled, - /// spaces and tabs between the header name and the colon; - /// * missing colon between header name and colon; - /// * characters that are not allowed in header values except for `\0` and `\r`. - /// - /// If an ignorable error is encountered, the parser tries to find the next - /// line in the input to resume parsing the rest of the headers. An error - /// will be emitted nonetheless if it finds `\0` or a lone `\r` while - /// looking for the next line. - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_ignore_invalid_headers_in_responses(&mut self, val: bool) -> &mut Builder { - self.h1_builder.ignore_invalid_headers_in_responses(val); - self - } - - /// Set whether HTTP/1 connections should try to use vectored writes, - /// or always flatten into a single buffer. - /// - /// Note that setting this to false may mean more copies of body data, - /// but may also improve performance when an IO transport doesn't - /// support vectored writes well, such as most TLS implementations. - /// - /// Setting this to true will force hyper to use queued strategy - /// which may eliminate unnecessary cloning on some TLS backends - /// - /// Default is `auto`. In this mode hyper will try to guess which - /// mode to use - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_writev(&mut self, enabled: bool) -> &mut Builder { - self.h1_builder.writev(enabled); - self - } - - /// Set whether HTTP/1 connections will write header names as title case at - /// the socket level. - /// - /// Note that this setting does not affect HTTP/2. - /// - /// Default is false. - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_title_case_headers(&mut self, val: bool) -> &mut Self { - self.h1_builder.title_case_headers(val); - self - } - - /// Set whether to support preserving original header cases. - /// - /// Currently, this will record the original cases received, and store them - /// in a private extension on the `Response`. It will also look for and use - /// such an extension in any provided `Request`. - /// - /// Since the relevant extension is still private, there is no way to - /// interact with the original cases. The only effect this can have now is - /// to forward the cases in a proxy-like fashion. - /// - /// Note that this setting does not affect HTTP/2. - /// - /// Default is false. - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_preserve_header_case(&mut self, val: bool) -> &mut Self { - self.h1_builder.preserve_header_case(val); - self - } - - /// Set the maximum number of headers. - /// - /// When a response is received, the parser will reserve a buffer to store headers for optimal - /// performance. - /// - /// If client receives more headers than the buffer size, the error "message header too large" - /// is returned. - /// - /// The headers is allocated on the stack by default, which has higher performance. After - /// setting this value, headers will be allocated in heap memory, that is, heap memory - /// allocation will occur for each response, and there will be a performance drop of about 5%. - /// - /// Note that this setting does not affect HTTP/2. - /// - /// Default is 100. - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http1_max_headers(&mut self, val: usize) -> &mut Self { - self.h1_builder.max_headers(val); - self - } - - /// Set whether HTTP/0.9 responses should be tolerated. - /// - /// Default is false. - #[cfg(feature = "http1")] - #[cfg_attr(docsrs, doc(cfg(feature = "http1")))] - pub fn http09_responses(&mut self, val: bool) -> &mut Self { - self.h1_builder.http09_responses(val); - self - } - - /// Set whether the connection **must** use HTTP/2. - /// - /// The destination must either allow HTTP2 Prior Knowledge, or the - /// `Connect` should be configured to do use ALPN to upgrade to `h2` - /// as part of the connection process. This will not make the `Client` - /// utilize ALPN by itself. - /// - /// Note that setting this to true prevents HTTP/1 from being allowed. - /// - /// Default is false. - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_only(&mut self, val: bool) -> &mut Self { - self.client_config.ver = if val { Ver::Http2 } else { Ver::Auto }; - self - } - - /// Configures the maximum number of pending reset streams allowed before a GOAWAY will be sent. - /// - /// This will default to the default value set by the [`h2` crate](https://crates.io/crates/h2). - /// As of v0.4.0, it is 20. - /// - /// See <https://github.com/hyperium/hyper/issues/2877> for more information. - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_max_pending_accept_reset_streams( - &mut self, - max: impl Into<Option<usize>>, - ) -> &mut Self { - self.h2_builder.max_pending_accept_reset_streams(max.into()); - self - } - - /// Sets the [`SETTINGS_INITIAL_WINDOW_SIZE`][spec] option for HTTP2 - /// stream-level flow control. - /// - /// Passing `None` will do nothing. - /// - /// If not set, hyper will use a default. - /// - /// [spec]: https://http2.github.io/http2-spec/#SETTINGS_INITIAL_WINDOW_SIZE - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_initial_stream_window_size(&mut self, sz: impl Into<Option<u32>>) -> &mut Self { - self.h2_builder.initial_stream_window_size(sz.into()); - self - } - - /// Sets the max connection-level flow control for HTTP2 - /// - /// Passing `None` will do nothing. - /// - /// If not set, hyper will use a default. - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_initial_connection_window_size( - &mut self, - sz: impl Into<Option<u32>>, - ) -> &mut Self { - self.h2_builder.initial_connection_window_size(sz.into()); - self - } - - /// Sets the initial maximum of locally initiated (send) streams. - /// - /// This value will be overwritten by the value included in the initial - /// SETTINGS frame received from the peer as part of a [connection preface]. - /// - /// Passing `None` will do nothing. - /// - /// If not set, hyper will use a default. - /// - /// [connection preface]: https://httpwg.org/specs/rfc9113.html#preface - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_initial_max_send_streams( - &mut self, - initial: impl Into<Option<usize>>, - ) -> &mut Self { - self.h2_builder.initial_max_send_streams(initial); - self - } - - /// Sets whether to use an adaptive flow control. - /// - /// Enabling this will override the limits set in - /// `http2_initial_stream_window_size` and - /// `http2_initial_connection_window_size`. - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_adaptive_window(&mut self, enabled: bool) -> &mut Self { - self.h2_builder.adaptive_window(enabled); - self - } - - /// Sets the maximum frame size to use for HTTP2. - /// - /// Passing `None` will do nothing. - /// - /// If not set, hyper will use a default. - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_max_frame_size(&mut self, sz: impl Into<Option<u32>>) -> &mut Self { - self.h2_builder.max_frame_size(sz); - self - } - - /// Sets the max size of received header frames for HTTP2. - /// - /// Default is currently 16KB, but can change. - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_max_header_list_size(&mut self, max: u32) -> &mut Self { - self.h2_builder.max_header_list_size(max); - self - } - - /// Sets an interval for HTTP2 Ping frames should be sent to keep a - /// connection alive. - /// - /// Pass `None` to disable HTTP2 keep-alive. - /// - /// Default is currently disabled. - /// - /// # Cargo Feature - /// - /// Requires the `tokio` cargo feature to be enabled. - #[cfg(feature = "tokio")] - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_keep_alive_interval( - &mut self, - interval: impl Into<Option<Duration>>, - ) -> &mut Self { - self.h2_builder.keep_alive_interval(interval); - self - } - - /// Sets a timeout for receiving an acknowledgement of the keep-alive ping. - /// - /// If the ping is not acknowledged within the timeout, the connection will - /// be closed. Does nothing if `http2_keep_alive_interval` is disabled. - /// - /// Default is 20 seconds. - /// - /// # Cargo Feature - /// - /// Requires the `tokio` cargo feature to be enabled. - #[cfg(feature = "tokio")] - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_keep_alive_timeout(&mut self, timeout: Duration) -> &mut Self { - self.h2_builder.keep_alive_timeout(timeout); - self - } - - /// Sets whether HTTP2 keep-alive should apply while the connection is idle. - /// - /// If disabled, keep-alive pings are only sent while there are open - /// request/responses streams. If enabled, pings are also sent when no - /// streams are active. Does nothing if `http2_keep_alive_interval` is - /// disabled. - /// - /// Default is `false`. - /// - /// # Cargo Feature - /// - /// Requires the `tokio` cargo feature to be enabled. - #[cfg(feature = "tokio")] - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_keep_alive_while_idle(&mut self, enabled: bool) -> &mut Self { - self.h2_builder.keep_alive_while_idle(enabled); - self - } - - /// Sets the maximum number of HTTP2 concurrent locally reset streams. - /// - /// See the documentation of [`h2::client::Builder::max_concurrent_reset_streams`] for more - /// details. - /// - /// The default value is determined by the `h2` crate. - /// - /// [`h2::client::Builder::max_concurrent_reset_streams`]: https://docs.rs/h2/client/struct.Builder.html#method.max_concurrent_reset_streams - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_max_concurrent_reset_streams(&mut self, max: usize) -> &mut Self { - self.h2_builder.max_concurrent_reset_streams(max); - self - } - - /// Provide a timer to be used for h2 - /// - /// See the documentation of [`h2::client::Builder::timer`] for more - /// details. - /// - /// [`h2::client::Builder::timer`]: https://docs.rs/h2/client/struct.Builder.html#method.timer - pub fn timer<M>(&mut self, timer: M) -> &mut Self - where - M: Timer + Send + Sync + 'static, - { - #[cfg(feature = "http2")] - self.h2_builder.timer(timer); - self - } - - /// Provide a timer to be used for timeouts and intervals in connection pools. - pub fn pool_timer<M>(&mut self, timer: M) -> &mut Self - where - M: Timer + Clone + Send + Sync + 'static, - { - self.pool_timer = Some(timer::Timer::new(timer.clone())); - self - } - - /// Set the maximum write buffer size for each HTTP/2 stream. - /// - /// Default is currently 1MB, but may change. - /// - /// # Panics - /// - /// The value must be no larger than `u32::MAX`. - #[cfg(feature = "http2")] - #[cfg_attr(docsrs, doc(cfg(feature = "http2")))] - pub fn http2_max_send_buf_size(&mut self, max: usize) -> &mut Self { - self.h2_builder.max_send_buf_size(max); - self - } - - /// Set whether to retry requests that get disrupted before ever starting - /// to write. - /// - /// This means a request that is queued, and gets given an idle, reused - /// connection, and then encounters an error immediately as the idle - /// connection was found to be unusable. - /// - /// When this is set to `false`, the related `ResponseFuture` would instead - /// resolve to an `Error::Cancel`. - /// - /// Default is `true`. - #[inline] - pub fn retry_canceled_requests(&mut self, val: bool) -> &mut Self { - self.client_config.retry_canceled_requests = val; - self - } - - /// Set whether to automatically add the `Host` header to requests. - /// - /// If true, and a request does not include a `Host` header, one will be - /// added automatically, derived from the authority of the `Uri`. - /// - /// Default is `true`. - #[inline] - pub fn set_host(&mut self, val: bool) -> &mut Self { - self.client_config.set_host = val; - self - } - - /// Build a client with this configuration and the default `HttpConnector`. - #[cfg(feature = "tokio")] - pub fn build_http<B>(&self) -> Client<HttpConnector, B> - where - B: Body + Send, - B::Data: Send, - { - let mut connector = HttpConnector::new(); - if self.pool_config.is_enabled() { - connector.set_keepalive(self.pool_config.idle_timeout); - } - self.build(connector) - } - - /// Combine the configuration of this builder with a connector to create a `Client`. - pub fn build<C, B>(&self, connector: C) -> Client<C, B> - where - C: Connect + Clone, - B: Body + Send, - B::Data: Send, - { - let exec = self.exec.clone(); - let timer = self.pool_timer.clone(); - Client { - config: self.client_config, - exec: exec.clone(), - #[cfg(feature = "http1")] - h1_builder: self.h1_builder.clone(), - #[cfg(feature = "http2")] - h2_builder: self.h2_builder.clone(), - connector, - pool: pool::Pool::new(self.pool_config, exec, timer), - } - } -} - -impl fmt::Debug for Builder { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.debug_struct("Builder") - .field("client_config", &self.client_config) - .field("pool_config", &self.pool_config) - .finish() - } -} - -// ==== impl Error ==== - -impl fmt::Debug for Error { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - let mut f = f.debug_tuple("hyper_util::client::legacy::Error"); - f.field(&self.kind); - if let Some(ref cause) = self.source { - f.field(cause); - } - f.finish() - } -} - -impl fmt::Display for Error { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - write!(f, "client error ({:?})", self.kind) - } -} - -impl StdError for Error { - fn source(&self) -> Option<&(dyn StdError + 'static)> { - self.source.as_ref().map(|e| &**e as _) - } -} - -impl Error { - /// Returns true if this was an error from `Connect`. - pub fn is_connect(&self) -> bool { - matches!(self.kind, ErrorKind::Connect) - } - - /// Returns the info of the client connection on which this error occurred. - #[cfg(any(feature = "http1", feature = "http2"))] - pub fn connect_info(&self) -> Option<&Connected> { - self.connect_info.as_ref() - } - - #[cfg(any(feature = "http1", feature = "http2"))] - fn with_connect_info(self, connect_info: Connected) -> Self { - Self { - connect_info: Some(connect_info), - ..self - } - } - fn is_canceled(&self) -> bool { - matches!(self.kind, ErrorKind::Canceled) - } - - fn tx(src: hyper::Error) -> Self { - e!(SendRequest, src) - } - - fn closed(src: hyper::Error) -> Self { - e!(ChannelClosed, src) - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/capture.rs b/vendor/hyper-util/src/client/legacy/connect/capture.rs deleted file mode 100644 index b31b6433..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/capture.rs +++ /dev/null @@ -1,187 +0,0 @@ -use std::{ops::Deref, sync::Arc}; - -use http::Request; -use tokio::sync::watch; - -use super::Connected; - -/// [`CaptureConnection`] allows callers to capture [`Connected`] information -/// -/// To capture a connection for a request, use [`capture_connection`]. -#[derive(Debug, Clone)] -pub struct CaptureConnection { - rx: watch::Receiver<Option<Connected>>, -} - -/// Capture the connection for a given request -/// -/// When making a request with Hyper, the underlying connection must implement the [`Connection`] trait. -/// [`capture_connection`] allows a caller to capture the returned [`Connected`] structure as soon -/// as the connection is established. -/// -/// [`Connection`]: crate::client::legacy::connect::Connection -/// -/// *Note*: If establishing a connection fails, [`CaptureConnection::connection_metadata`] will always return none. -/// -/// # Examples -/// -/// **Synchronous access**: -/// The [`CaptureConnection::connection_metadata`] method allows callers to check if a connection has been -/// established. This is ideal for situations where you are certain the connection has already -/// been established (e.g. after the response future has already completed). -/// ```rust -/// use hyper_util::client::legacy::connect::capture_connection; -/// let mut request = http::Request::builder() -/// .uri("http://foo.com") -/// .body(()) -/// .unwrap(); -/// -/// let captured_connection = capture_connection(&mut request); -/// // some time later after the request has been sent... -/// let connection_info = captured_connection.connection_metadata(); -/// println!("we are connected! {:?}", connection_info.as_ref()); -/// ``` -/// -/// **Asynchronous access**: -/// The [`CaptureConnection::wait_for_connection_metadata`] method returns a future resolves as soon as the -/// connection is available. -/// -/// ```rust -/// # #[cfg(feature = "tokio")] -/// # async fn example() { -/// use hyper_util::client::legacy::connect::capture_connection; -/// use hyper_util::client::legacy::Client; -/// use hyper_util::rt::TokioExecutor; -/// use bytes::Bytes; -/// use http_body_util::Empty; -/// let mut request = http::Request::builder() -/// .uri("http://foo.com") -/// .body(Empty::<Bytes>::new()) -/// .unwrap(); -/// -/// let mut captured = capture_connection(&mut request); -/// tokio::task::spawn(async move { -/// let connection_info = captured.wait_for_connection_metadata().await; -/// println!("we are connected! {:?}", connection_info.as_ref()); -/// }); -/// -/// let client = Client::builder(TokioExecutor::new()).build_http(); -/// client.request(request).await.expect("request failed"); -/// # } -/// ``` -pub fn capture_connection<B>(request: &mut Request<B>) -> CaptureConnection { - let (tx, rx) = CaptureConnection::new(); - request.extensions_mut().insert(tx); - rx -} - -/// TxSide for [`CaptureConnection`] -/// -/// This is inserted into `Extensions` to allow Hyper to back channel connection info -#[derive(Clone)] -pub(crate) struct CaptureConnectionExtension { - tx: Arc<watch::Sender<Option<Connected>>>, -} - -impl CaptureConnectionExtension { - pub(crate) fn set(&self, connected: &Connected) { - self.tx.send_replace(Some(connected.clone())); - } -} - -impl CaptureConnection { - /// Internal API to create the tx and rx half of [`CaptureConnection`] - pub(crate) fn new() -> (CaptureConnectionExtension, Self) { - let (tx, rx) = watch::channel(None); - ( - CaptureConnectionExtension { tx: Arc::new(tx) }, - CaptureConnection { rx }, - ) - } - - /// Retrieve the connection metadata, if available - pub fn connection_metadata(&self) -> impl Deref<Target = Option<Connected>> + '_ { - self.rx.borrow() - } - - /// Wait for the connection to be established - /// - /// If a connection was established, this will always return `Some(...)`. If the request never - /// successfully connected (e.g. DNS resolution failure), this method will never return. - pub async fn wait_for_connection_metadata( - &mut self, - ) -> impl Deref<Target = Option<Connected>> + '_ { - if self.rx.borrow().is_some() { - return self.rx.borrow(); - } - let _ = self.rx.changed().await; - self.rx.borrow() - } -} - -#[cfg(all(test, not(miri)))] -mod test { - use super::*; - - #[test] - fn test_sync_capture_connection() { - let (tx, rx) = CaptureConnection::new(); - assert!( - rx.connection_metadata().is_none(), - "connection has not been set" - ); - tx.set(&Connected::new().proxy(true)); - assert!(rx - .connection_metadata() - .as_ref() - .expect("connected should be set") - .is_proxied()); - - // ensure it can be called multiple times - assert!(rx - .connection_metadata() - .as_ref() - .expect("connected should be set") - .is_proxied()); - } - - #[tokio::test] - async fn async_capture_connection() { - let (tx, mut rx) = CaptureConnection::new(); - assert!( - rx.connection_metadata().is_none(), - "connection has not been set" - ); - let test_task = tokio::spawn(async move { - assert!(rx - .wait_for_connection_metadata() - .await - .as_ref() - .expect("connection should be set") - .is_proxied()); - // can be awaited multiple times - assert!( - rx.wait_for_connection_metadata().await.is_some(), - "should be awaitable multiple times" - ); - - assert!(rx.connection_metadata().is_some()); - }); - // can't be finished, we haven't set the connection yet - assert!(!test_task.is_finished()); - tx.set(&Connected::new().proxy(true)); - - assert!(test_task.await.is_ok()); - } - - #[tokio::test] - async fn capture_connection_sender_side_dropped() { - let (tx, mut rx) = CaptureConnection::new(); - assert!( - rx.connection_metadata().is_none(), - "connection has not been set" - ); - drop(tx); - assert!(rx.wait_for_connection_metadata().await.is_none()); - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/dns.rs b/vendor/hyper-util/src/client/legacy/connect/dns.rs deleted file mode 100644 index abeb2cca..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/dns.rs +++ /dev/null @@ -1,360 +0,0 @@ -//! DNS Resolution used by the `HttpConnector`. -//! -//! This module contains: -//! -//! - A [`GaiResolver`] that is the default resolver for the `HttpConnector`. -//! - The `Name` type used as an argument to custom resolvers. -//! -//! # Resolvers are `Service`s -//! -//! A resolver is just a -//! `Service<Name, Response = impl Iterator<Item = SocketAddr>>`. -//! -//! A simple resolver that ignores the name and always returns a specific -//! address: -//! -//! ```rust,ignore -//! use std::{convert::Infallible, iter, net::SocketAddr}; -//! -//! let resolver = tower::service_fn(|_name| async { -//! Ok::<_, Infallible>(iter::once(SocketAddr::from(([127, 0, 0, 1], 8080)))) -//! }); -//! ``` -use std::error::Error; -use std::future::Future; -use std::net::{Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6, ToSocketAddrs}; -use std::pin::Pin; -use std::str::FromStr; -use std::task::{self, Poll}; -use std::{fmt, io, vec}; - -use tokio::task::JoinHandle; -use tower_service::Service; - -pub(super) use self::sealed::Resolve; - -/// A domain name to resolve into IP addresses. -#[derive(Clone, Hash, Eq, PartialEq)] -pub struct Name { - host: Box<str>, -} - -/// A resolver using blocking `getaddrinfo` calls in a threadpool. -#[derive(Clone)] -pub struct GaiResolver { - _priv: (), -} - -/// An iterator of IP addresses returned from `getaddrinfo`. -pub struct GaiAddrs { - inner: SocketAddrs, -} - -/// A future to resolve a name returned by `GaiResolver`. -pub struct GaiFuture { - inner: JoinHandle<Result<SocketAddrs, io::Error>>, -} - -impl Name { - pub(super) fn new(host: Box<str>) -> Name { - Name { host } - } - - /// View the hostname as a string slice. - pub fn as_str(&self) -> &str { - &self.host - } -} - -impl fmt::Debug for Name { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - fmt::Debug::fmt(&self.host, f) - } -} - -impl fmt::Display for Name { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - fmt::Display::fmt(&self.host, f) - } -} - -impl FromStr for Name { - type Err = InvalidNameError; - - fn from_str(host: &str) -> Result<Self, Self::Err> { - // Possibly add validation later - Ok(Name::new(host.into())) - } -} - -/// Error indicating a given string was not a valid domain name. -#[derive(Debug)] -pub struct InvalidNameError(()); - -impl fmt::Display for InvalidNameError { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.write_str("Not a valid domain name") - } -} - -impl Error for InvalidNameError {} - -impl GaiResolver { - /// Construct a new `GaiResolver`. - pub fn new() -> Self { - GaiResolver { _priv: () } - } -} - -impl Service<Name> for GaiResolver { - type Response = GaiAddrs; - type Error = io::Error; - type Future = GaiFuture; - - fn poll_ready(&mut self, _cx: &mut task::Context<'_>) -> Poll<Result<(), io::Error>> { - Poll::Ready(Ok(())) - } - - fn call(&mut self, name: Name) -> Self::Future { - let blocking = tokio::task::spawn_blocking(move || { - (&*name.host, 0) - .to_socket_addrs() - .map(|i| SocketAddrs { iter: i }) - }); - - GaiFuture { inner: blocking } - } -} - -impl fmt::Debug for GaiResolver { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.pad("GaiResolver") - } -} - -impl Future for GaiFuture { - type Output = Result<GaiAddrs, io::Error>; - - fn poll(mut self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Self::Output> { - Pin::new(&mut self.inner).poll(cx).map(|res| match res { - Ok(Ok(addrs)) => Ok(GaiAddrs { inner: addrs }), - Ok(Err(err)) => Err(err), - Err(join_err) => { - if join_err.is_cancelled() { - Err(io::Error::new(io::ErrorKind::Interrupted, join_err)) - } else { - panic!("gai background task failed: {join_err:?}") - } - } - }) - } -} - -impl fmt::Debug for GaiFuture { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.pad("GaiFuture") - } -} - -impl Drop for GaiFuture { - fn drop(&mut self) { - self.inner.abort(); - } -} - -impl Iterator for GaiAddrs { - type Item = SocketAddr; - - fn next(&mut self) -> Option<Self::Item> { - self.inner.next() - } -} - -impl fmt::Debug for GaiAddrs { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.pad("GaiAddrs") - } -} - -pub(super) struct SocketAddrs { - iter: vec::IntoIter<SocketAddr>, -} - -impl SocketAddrs { - pub(super) fn new(addrs: Vec<SocketAddr>) -> Self { - SocketAddrs { - iter: addrs.into_iter(), - } - } - - pub(super) fn try_parse(host: &str, port: u16) -> Option<SocketAddrs> { - if let Ok(addr) = host.parse::<Ipv4Addr>() { - let addr = SocketAddrV4::new(addr, port); - return Some(SocketAddrs { - iter: vec![SocketAddr::V4(addr)].into_iter(), - }); - } - if let Ok(addr) = host.parse::<Ipv6Addr>() { - let addr = SocketAddrV6::new(addr, port, 0, 0); - return Some(SocketAddrs { - iter: vec![SocketAddr::V6(addr)].into_iter(), - }); - } - None - } - - #[inline] - fn filter(self, predicate: impl FnMut(&SocketAddr) -> bool) -> SocketAddrs { - SocketAddrs::new(self.iter.filter(predicate).collect()) - } - - pub(super) fn split_by_preference( - self, - local_addr_ipv4: Option<Ipv4Addr>, - local_addr_ipv6: Option<Ipv6Addr>, - ) -> (SocketAddrs, SocketAddrs) { - match (local_addr_ipv4, local_addr_ipv6) { - (Some(_), None) => (self.filter(SocketAddr::is_ipv4), SocketAddrs::new(vec![])), - (None, Some(_)) => (self.filter(SocketAddr::is_ipv6), SocketAddrs::new(vec![])), - _ => { - let preferring_v6 = self - .iter - .as_slice() - .first() - .map(SocketAddr::is_ipv6) - .unwrap_or(false); - - let (preferred, fallback) = self - .iter - .partition::<Vec<_>, _>(|addr| addr.is_ipv6() == preferring_v6); - - (SocketAddrs::new(preferred), SocketAddrs::new(fallback)) - } - } - } - - pub(super) fn is_empty(&self) -> bool { - self.iter.as_slice().is_empty() - } - - pub(super) fn len(&self) -> usize { - self.iter.as_slice().len() - } -} - -impl Iterator for SocketAddrs { - type Item = SocketAddr; - #[inline] - fn next(&mut self) -> Option<SocketAddr> { - self.iter.next() - } -} - -mod sealed { - use std::future::Future; - use std::task::{self, Poll}; - - use super::{Name, SocketAddr}; - use tower_service::Service; - - // "Trait alias" for `Service<Name, Response = Addrs>` - pub trait Resolve { - type Addrs: Iterator<Item = SocketAddr>; - type Error: Into<Box<dyn std::error::Error + Send + Sync>>; - type Future: Future<Output = Result<Self::Addrs, Self::Error>>; - - fn poll_ready(&mut self, cx: &mut task::Context<'_>) -> Poll<Result<(), Self::Error>>; - fn resolve(&mut self, name: Name) -> Self::Future; - } - - impl<S> Resolve for S - where - S: Service<Name>, - S::Response: Iterator<Item = SocketAddr>, - S::Error: Into<Box<dyn std::error::Error + Send + Sync>>, - { - type Addrs = S::Response; - type Error = S::Error; - type Future = S::Future; - - fn poll_ready(&mut self, cx: &mut task::Context<'_>) -> Poll<Result<(), Self::Error>> { - Service::poll_ready(self, cx) - } - - fn resolve(&mut self, name: Name) -> Self::Future { - Service::call(self, name) - } - } -} - -pub(super) async fn resolve<R>(resolver: &mut R, name: Name) -> Result<R::Addrs, R::Error> -where - R: Resolve, -{ - crate::common::future::poll_fn(|cx| resolver.poll_ready(cx)).await?; - resolver.resolve(name).await -} - -#[cfg(test)] -mod tests { - use super::*; - use std::net::{Ipv4Addr, Ipv6Addr}; - - #[test] - fn test_ip_addrs_split_by_preference() { - let ip_v4 = Ipv4Addr::new(127, 0, 0, 1); - let ip_v6 = Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1); - let v4_addr = (ip_v4, 80).into(); - let v6_addr = (ip_v6, 80).into(); - - let (mut preferred, mut fallback) = SocketAddrs { - iter: vec![v4_addr, v6_addr].into_iter(), - } - .split_by_preference(None, None); - assert!(preferred.next().unwrap().is_ipv4()); - assert!(fallback.next().unwrap().is_ipv6()); - - let (mut preferred, mut fallback) = SocketAddrs { - iter: vec![v6_addr, v4_addr].into_iter(), - } - .split_by_preference(None, None); - assert!(preferred.next().unwrap().is_ipv6()); - assert!(fallback.next().unwrap().is_ipv4()); - - let (mut preferred, mut fallback) = SocketAddrs { - iter: vec![v4_addr, v6_addr].into_iter(), - } - .split_by_preference(Some(ip_v4), Some(ip_v6)); - assert!(preferred.next().unwrap().is_ipv4()); - assert!(fallback.next().unwrap().is_ipv6()); - - let (mut preferred, mut fallback) = SocketAddrs { - iter: vec![v6_addr, v4_addr].into_iter(), - } - .split_by_preference(Some(ip_v4), Some(ip_v6)); - assert!(preferred.next().unwrap().is_ipv6()); - assert!(fallback.next().unwrap().is_ipv4()); - - let (mut preferred, fallback) = SocketAddrs { - iter: vec![v4_addr, v6_addr].into_iter(), - } - .split_by_preference(Some(ip_v4), None); - assert!(preferred.next().unwrap().is_ipv4()); - assert!(fallback.is_empty()); - - let (mut preferred, fallback) = SocketAddrs { - iter: vec![v4_addr, v6_addr].into_iter(), - } - .split_by_preference(None, Some(ip_v6)); - assert!(preferred.next().unwrap().is_ipv6()); - assert!(fallback.is_empty()); - } - - #[test] - fn test_name_from_str() { - const DOMAIN: &str = "test.example.com"; - let name = Name::from_str(DOMAIN).expect("Should be a valid domain"); - assert_eq!(name.as_str(), DOMAIN); - assert_eq!(name.to_string(), DOMAIN); - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/http.rs b/vendor/hyper-util/src/client/legacy/connect/http.rs deleted file mode 100644 index f19a78eb..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/http.rs +++ /dev/null @@ -1,1468 +0,0 @@ -use std::error::Error as StdError; -use std::fmt; -use std::future::Future; -use std::io; -use std::marker::PhantomData; -use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr}; -use std::pin::Pin; -use std::sync::Arc; -use std::task::{self, Poll}; -use std::time::Duration; - -use futures_core::ready; -use futures_util::future::Either; -use http::uri::{Scheme, Uri}; -use pin_project_lite::pin_project; -use socket2::TcpKeepalive; -use tokio::net::{TcpSocket, TcpStream}; -use tokio::time::Sleep; -use tracing::{debug, trace, warn}; - -use super::dns::{self, resolve, GaiResolver, Resolve}; -use super::{Connected, Connection}; -use crate::rt::TokioIo; - -/// A connector for the `http` scheme. -/// -/// Performs DNS resolution in a thread pool, and then connects over TCP. -/// -/// # Note -/// -/// Sets the [`HttpInfo`](HttpInfo) value on responses, which includes -/// transport information such as the remote socket address used. -#[derive(Clone)] -pub struct HttpConnector<R = GaiResolver> { - config: Arc<Config>, - resolver: R, -} - -/// Extra information about the transport when an HttpConnector is used. -/// -/// # Example -/// -/// ``` -/// # fn doc(res: http::Response<()>) { -/// use hyper_util::client::legacy::connect::HttpInfo; -/// -/// // res = http::Response -/// res -/// .extensions() -/// .get::<HttpInfo>() -/// .map(|info| { -/// println!("remote addr = {}", info.remote_addr()); -/// }); -/// # } -/// ``` -/// -/// # Note -/// -/// If a different connector is used besides [`HttpConnector`](HttpConnector), -/// this value will not exist in the extensions. Consult that specific -/// connector to see what "extra" information it might provide to responses. -#[derive(Clone, Debug)] -pub struct HttpInfo { - remote_addr: SocketAddr, - local_addr: SocketAddr, -} - -#[derive(Clone)] -struct Config { - connect_timeout: Option<Duration>, - enforce_http: bool, - happy_eyeballs_timeout: Option<Duration>, - tcp_keepalive_config: TcpKeepaliveConfig, - local_address_ipv4: Option<Ipv4Addr>, - local_address_ipv6: Option<Ipv6Addr>, - nodelay: bool, - reuse_address: bool, - send_buffer_size: Option<usize>, - recv_buffer_size: Option<usize>, - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - interface: Option<String>, - #[cfg(any( - target_os = "illumos", - target_os = "ios", - target_os = "macos", - target_os = "solaris", - target_os = "tvos", - target_os = "visionos", - target_os = "watchos", - ))] - interface: Option<std::ffi::CString>, - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - tcp_user_timeout: Option<Duration>, -} - -#[derive(Default, Debug, Clone, Copy)] -struct TcpKeepaliveConfig { - time: Option<Duration>, - interval: Option<Duration>, - retries: Option<u32>, -} - -impl TcpKeepaliveConfig { - /// Converts into a `socket2::TcpKeealive` if there is any keep alive configuration. - fn into_tcpkeepalive(self) -> Option<TcpKeepalive> { - let mut dirty = false; - let mut ka = TcpKeepalive::new(); - if let Some(time) = self.time { - ka = ka.with_time(time); - dirty = true - } - if let Some(interval) = self.interval { - ka = Self::ka_with_interval(ka, interval, &mut dirty) - }; - if let Some(retries) = self.retries { - ka = Self::ka_with_retries(ka, retries, &mut dirty) - }; - if dirty { - Some(ka) - } else { - None - } - } - - #[cfg( - // See https://docs.rs/socket2/0.5.8/src/socket2/lib.rs.html#511-525 - any( - target_os = "android", - target_os = "dragonfly", - target_os = "freebsd", - target_os = "fuchsia", - target_os = "illumos", - target_os = "ios", - target_os = "visionos", - target_os = "linux", - target_os = "macos", - target_os = "netbsd", - target_os = "tvos", - target_os = "watchos", - target_os = "windows", - ) - )] - fn ka_with_interval(ka: TcpKeepalive, interval: Duration, dirty: &mut bool) -> TcpKeepalive { - *dirty = true; - ka.with_interval(interval) - } - - #[cfg(not( - // See https://docs.rs/socket2/0.5.8/src/socket2/lib.rs.html#511-525 - any( - target_os = "android", - target_os = "dragonfly", - target_os = "freebsd", - target_os = "fuchsia", - target_os = "illumos", - target_os = "ios", - target_os = "visionos", - target_os = "linux", - target_os = "macos", - target_os = "netbsd", - target_os = "tvos", - target_os = "watchos", - target_os = "windows", - ) - ))] - fn ka_with_interval(ka: TcpKeepalive, _: Duration, _: &mut bool) -> TcpKeepalive { - ka // no-op as keepalive interval is not supported on this platform - } - - #[cfg( - // See https://docs.rs/socket2/0.5.8/src/socket2/lib.rs.html#557-570 - any( - target_os = "android", - target_os = "dragonfly", - target_os = "freebsd", - target_os = "fuchsia", - target_os = "illumos", - target_os = "ios", - target_os = "visionos", - target_os = "linux", - target_os = "macos", - target_os = "netbsd", - target_os = "tvos", - target_os = "watchos", - ) - )] - fn ka_with_retries(ka: TcpKeepalive, retries: u32, dirty: &mut bool) -> TcpKeepalive { - *dirty = true; - ka.with_retries(retries) - } - - #[cfg(not( - // See https://docs.rs/socket2/0.5.8/src/socket2/lib.rs.html#557-570 - any( - target_os = "android", - target_os = "dragonfly", - target_os = "freebsd", - target_os = "fuchsia", - target_os = "illumos", - target_os = "ios", - target_os = "visionos", - target_os = "linux", - target_os = "macos", - target_os = "netbsd", - target_os = "tvos", - target_os = "watchos", - ) - ))] - fn ka_with_retries(ka: TcpKeepalive, _: u32, _: &mut bool) -> TcpKeepalive { - ka // no-op as keepalive retries is not supported on this platform - } -} - -// ===== impl HttpConnector ===== - -impl HttpConnector { - /// Construct a new HttpConnector. - pub fn new() -> HttpConnector { - HttpConnector::new_with_resolver(GaiResolver::new()) - } -} - -impl<R> HttpConnector<R> { - /// Construct a new HttpConnector. - /// - /// Takes a [`Resolver`](crate::client::legacy::connect::dns#resolvers-are-services) to handle DNS lookups. - pub fn new_with_resolver(resolver: R) -> HttpConnector<R> { - HttpConnector { - config: Arc::new(Config { - connect_timeout: None, - enforce_http: true, - happy_eyeballs_timeout: Some(Duration::from_millis(300)), - tcp_keepalive_config: TcpKeepaliveConfig::default(), - local_address_ipv4: None, - local_address_ipv6: None, - nodelay: false, - reuse_address: false, - send_buffer_size: None, - recv_buffer_size: None, - #[cfg(any( - target_os = "android", - target_os = "fuchsia", - target_os = "illumos", - target_os = "ios", - target_os = "linux", - target_os = "macos", - target_os = "solaris", - target_os = "tvos", - target_os = "visionos", - target_os = "watchos", - ))] - interface: None, - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - tcp_user_timeout: None, - }), - resolver, - } - } - - /// Option to enforce all `Uri`s have the `http` scheme. - /// - /// Enabled by default. - #[inline] - pub fn enforce_http(&mut self, is_enforced: bool) { - self.config_mut().enforce_http = is_enforced; - } - - /// Set that all sockets have `SO_KEEPALIVE` set with the supplied duration - /// to remain idle before sending TCP keepalive probes. - /// - /// If `None`, keepalive is disabled. - /// - /// Default is `None`. - #[inline] - pub fn set_keepalive(&mut self, time: Option<Duration>) { - self.config_mut().tcp_keepalive_config.time = time; - } - - /// Set the duration between two successive TCP keepalive retransmissions, - /// if acknowledgement to the previous keepalive transmission is not received. - #[inline] - pub fn set_keepalive_interval(&mut self, interval: Option<Duration>) { - self.config_mut().tcp_keepalive_config.interval = interval; - } - - /// Set the number of retransmissions to be carried out before declaring that remote end is not available. - #[inline] - pub fn set_keepalive_retries(&mut self, retries: Option<u32>) { - self.config_mut().tcp_keepalive_config.retries = retries; - } - - /// Set that all sockets have `SO_NODELAY` set to the supplied value `nodelay`. - /// - /// Default is `false`. - #[inline] - pub fn set_nodelay(&mut self, nodelay: bool) { - self.config_mut().nodelay = nodelay; - } - - /// Sets the value of the SO_SNDBUF option on the socket. - #[inline] - pub fn set_send_buffer_size(&mut self, size: Option<usize>) { - self.config_mut().send_buffer_size = size; - } - - /// Sets the value of the SO_RCVBUF option on the socket. - #[inline] - pub fn set_recv_buffer_size(&mut self, size: Option<usize>) { - self.config_mut().recv_buffer_size = size; - } - - /// Set that all sockets are bound to the configured address before connection. - /// - /// If `None`, the sockets will not be bound. - /// - /// Default is `None`. - #[inline] - pub fn set_local_address(&mut self, addr: Option<IpAddr>) { - let (v4, v6) = match addr { - Some(IpAddr::V4(a)) => (Some(a), None), - Some(IpAddr::V6(a)) => (None, Some(a)), - _ => (None, None), - }; - - let cfg = self.config_mut(); - - cfg.local_address_ipv4 = v4; - cfg.local_address_ipv6 = v6; - } - - /// Set that all sockets are bound to the configured IPv4 or IPv6 address (depending on host's - /// preferences) before connection. - #[inline] - pub fn set_local_addresses(&mut self, addr_ipv4: Ipv4Addr, addr_ipv6: Ipv6Addr) { - let cfg = self.config_mut(); - - cfg.local_address_ipv4 = Some(addr_ipv4); - cfg.local_address_ipv6 = Some(addr_ipv6); - } - - /// Set the connect timeout. - /// - /// If a domain resolves to multiple IP addresses, the timeout will be - /// evenly divided across them. - /// - /// Default is `None`. - #[inline] - pub fn set_connect_timeout(&mut self, dur: Option<Duration>) { - self.config_mut().connect_timeout = dur; - } - - /// Set timeout for [RFC 6555 (Happy Eyeballs)][RFC 6555] algorithm. - /// - /// If hostname resolves to both IPv4 and IPv6 addresses and connection - /// cannot be established using preferred address family before timeout - /// elapses, then connector will in parallel attempt connection using other - /// address family. - /// - /// If `None`, parallel connection attempts are disabled. - /// - /// Default is 300 milliseconds. - /// - /// [RFC 6555]: https://tools.ietf.org/html/rfc6555 - #[inline] - pub fn set_happy_eyeballs_timeout(&mut self, dur: Option<Duration>) { - self.config_mut().happy_eyeballs_timeout = dur; - } - - /// Set that all socket have `SO_REUSEADDR` set to the supplied value `reuse_address`. - /// - /// Default is `false`. - #[inline] - pub fn set_reuse_address(&mut self, reuse_address: bool) -> &mut Self { - self.config_mut().reuse_address = reuse_address; - self - } - - /// Sets the name of the interface to bind sockets produced by this - /// connector. - /// - /// On Linux, this sets the `SO_BINDTODEVICE` option on this socket (see - /// [`man 7 socket`] for details). On macOS (and macOS-derived systems like - /// iOS), illumos, and Solaris, this will instead use the `IP_BOUND_IF` - /// socket option (see [`man 7p ip`]). - /// - /// If a socket is bound to an interface, only packets received from that particular - /// interface are processed by the socket. Note that this only works for some socket - /// types, particularly `AF_INET`` sockets. - /// - /// On Linux it can be used to specify a [VRF], but the binary needs - /// to either have `CAP_NET_RAW` or to be run as root. - /// - /// This function is only available on the following operating systems: - /// - Linux, including Android - /// - Fuchsia - /// - illumos and Solaris - /// - macOS, iOS, visionOS, watchOS, and tvOS - /// - /// [VRF]: https://www.kernel.org/doc/Documentation/networking/vrf.txt - /// [`man 7 socket`] https://man7.org/linux/man-pages/man7/socket.7.html - /// [`man 7p ip`]: https://docs.oracle.com/cd/E86824_01/html/E54777/ip-7p.html - #[cfg(any( - target_os = "android", - target_os = "fuchsia", - target_os = "illumos", - target_os = "ios", - target_os = "linux", - target_os = "macos", - target_os = "solaris", - target_os = "tvos", - target_os = "visionos", - target_os = "watchos", - ))] - #[inline] - pub fn set_interface<S: Into<String>>(&mut self, interface: S) -> &mut Self { - let interface = interface.into(); - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - { - self.config_mut().interface = Some(interface); - } - #[cfg(not(any(target_os = "android", target_os = "fuchsia", target_os = "linux")))] - { - let interface = std::ffi::CString::new(interface) - .expect("interface name should not have nulls in it"); - self.config_mut().interface = Some(interface); - } - self - } - - /// Sets the value of the TCP_USER_TIMEOUT option on the socket. - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - #[inline] - pub fn set_tcp_user_timeout(&mut self, time: Option<Duration>) { - self.config_mut().tcp_user_timeout = time; - } - - // private - - fn config_mut(&mut self) -> &mut Config { - // If the are HttpConnector clones, this will clone the inner - // config. So mutating the config won't ever affect previous - // clones. - Arc::make_mut(&mut self.config) - } -} - -static INVALID_NOT_HTTP: &str = "invalid URL, scheme is not http"; -static INVALID_MISSING_SCHEME: &str = "invalid URL, scheme is missing"; -static INVALID_MISSING_HOST: &str = "invalid URL, host is missing"; - -// R: Debug required for now to allow adding it to debug output later... -impl<R: fmt::Debug> fmt::Debug for HttpConnector<R> { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.debug_struct("HttpConnector").finish() - } -} - -impl<R> tower_service::Service<Uri> for HttpConnector<R> -where - R: Resolve + Clone + Send + Sync + 'static, - R::Future: Send, -{ - type Response = TokioIo<TcpStream>; - type Error = ConnectError; - type Future = HttpConnecting<R>; - - fn poll_ready(&mut self, cx: &mut task::Context<'_>) -> Poll<Result<(), Self::Error>> { - ready!(self.resolver.poll_ready(cx)).map_err(ConnectError::dns)?; - Poll::Ready(Ok(())) - } - - fn call(&mut self, dst: Uri) -> Self::Future { - let mut self_ = self.clone(); - HttpConnecting { - fut: Box::pin(async move { self_.call_async(dst).await }), - _marker: PhantomData, - } - } -} - -fn get_host_port<'u>(config: &Config, dst: &'u Uri) -> Result<(&'u str, u16), ConnectError> { - trace!( - "Http::connect; scheme={:?}, host={:?}, port={:?}", - dst.scheme(), - dst.host(), - dst.port(), - ); - - if config.enforce_http { - if dst.scheme() != Some(&Scheme::HTTP) { - return Err(ConnectError { - msg: INVALID_NOT_HTTP, - addr: None, - cause: None, - }); - } - } else if dst.scheme().is_none() { - return Err(ConnectError { - msg: INVALID_MISSING_SCHEME, - addr: None, - cause: None, - }); - } - - let host = match dst.host() { - Some(s) => s, - None => { - return Err(ConnectError { - msg: INVALID_MISSING_HOST, - addr: None, - cause: None, - }) - } - }; - let port = match dst.port() { - Some(port) => port.as_u16(), - None => { - if dst.scheme() == Some(&Scheme::HTTPS) { - 443 - } else { - 80 - } - } - }; - - Ok((host, port)) -} - -impl<R> HttpConnector<R> -where - R: Resolve, -{ - async fn call_async(&mut self, dst: Uri) -> Result<TokioIo<TcpStream>, ConnectError> { - let config = &self.config; - - let (host, port) = get_host_port(config, &dst)?; - let host = host.trim_start_matches('[').trim_end_matches(']'); - - // If the host is already an IP addr (v4 or v6), - // skip resolving the dns and start connecting right away. - let addrs = if let Some(addrs) = dns::SocketAddrs::try_parse(host, port) { - addrs - } else { - let addrs = resolve(&mut self.resolver, dns::Name::new(host.into())) - .await - .map_err(ConnectError::dns)?; - let addrs = addrs - .map(|mut addr| { - set_port(&mut addr, port, dst.port().is_some()); - - addr - }) - .collect(); - dns::SocketAddrs::new(addrs) - }; - - let c = ConnectingTcp::new(addrs, config); - - let sock = c.connect().await?; - - if let Err(e) = sock.set_nodelay(config.nodelay) { - warn!("tcp set_nodelay error: {}", e); - } - - Ok(TokioIo::new(sock)) - } -} - -impl Connection for TcpStream { - fn connected(&self) -> Connected { - let connected = Connected::new(); - if let (Ok(remote_addr), Ok(local_addr)) = (self.peer_addr(), self.local_addr()) { - connected.extra(HttpInfo { - remote_addr, - local_addr, - }) - } else { - connected - } - } -} - -#[cfg(unix)] -impl Connection for tokio::net::UnixStream { - fn connected(&self) -> Connected { - Connected::new() - } -} - -#[cfg(windows)] -impl Connection for tokio::net::windows::named_pipe::NamedPipeClient { - fn connected(&self) -> Connected { - Connected::new() - } -} - -// Implement `Connection` for generic `TokioIo<T>` so that external crates can -// implement their own `HttpConnector` with `TokioIo<CustomTcpStream>`. -impl<T> Connection for TokioIo<T> -where - T: Connection, -{ - fn connected(&self) -> Connected { - self.inner().connected() - } -} - -impl HttpInfo { - /// Get the remote address of the transport used. - pub fn remote_addr(&self) -> SocketAddr { - self.remote_addr - } - - /// Get the local address of the transport used. - pub fn local_addr(&self) -> SocketAddr { - self.local_addr - } -} - -pin_project! { - // Not publicly exported (so missing_docs doesn't trigger). - // - // We return this `Future` instead of the `Pin<Box<dyn Future>>` directly - // so that users don't rely on it fitting in a `Pin<Box<dyn Future>>` slot - // (and thus we can change the type in the future). - #[must_use = "futures do nothing unless polled"] - #[allow(missing_debug_implementations)] - pub struct HttpConnecting<R> { - #[pin] - fut: BoxConnecting, - _marker: PhantomData<R>, - } -} - -type ConnectResult = Result<TokioIo<TcpStream>, ConnectError>; -type BoxConnecting = Pin<Box<dyn Future<Output = ConnectResult> + Send>>; - -impl<R: Resolve> Future for HttpConnecting<R> { - type Output = ConnectResult; - - fn poll(self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Self::Output> { - self.project().fut.poll(cx) - } -} - -// Not publicly exported (so missing_docs doesn't trigger). -pub struct ConnectError { - msg: &'static str, - addr: Option<SocketAddr>, - cause: Option<Box<dyn StdError + Send + Sync>>, -} - -impl ConnectError { - fn new<E>(msg: &'static str, cause: E) -> ConnectError - where - E: Into<Box<dyn StdError + Send + Sync>>, - { - ConnectError { - msg, - addr: None, - cause: Some(cause.into()), - } - } - - fn dns<E>(cause: E) -> ConnectError - where - E: Into<Box<dyn StdError + Send + Sync>>, - { - ConnectError::new("dns error", cause) - } - - fn m<E>(msg: &'static str) -> impl FnOnce(E) -> ConnectError - where - E: Into<Box<dyn StdError + Send + Sync>>, - { - move |cause| ConnectError::new(msg, cause) - } -} - -impl fmt::Debug for ConnectError { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - let mut b = f.debug_tuple("ConnectError"); - b.field(&self.msg); - if let Some(ref addr) = self.addr { - b.field(addr); - } - if let Some(ref cause) = self.cause { - b.field(cause); - } - b.finish() - } -} - -impl fmt::Display for ConnectError { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.write_str(self.msg) - } -} - -impl StdError for ConnectError { - fn source(&self) -> Option<&(dyn StdError + 'static)> { - self.cause.as_ref().map(|e| &**e as _) - } -} - -struct ConnectingTcp<'a> { - preferred: ConnectingTcpRemote, - fallback: Option<ConnectingTcpFallback>, - config: &'a Config, -} - -impl<'a> ConnectingTcp<'a> { - fn new(remote_addrs: dns::SocketAddrs, config: &'a Config) -> Self { - if let Some(fallback_timeout) = config.happy_eyeballs_timeout { - let (preferred_addrs, fallback_addrs) = remote_addrs - .split_by_preference(config.local_address_ipv4, config.local_address_ipv6); - if fallback_addrs.is_empty() { - return ConnectingTcp { - preferred: ConnectingTcpRemote::new(preferred_addrs, config.connect_timeout), - fallback: None, - config, - }; - } - - ConnectingTcp { - preferred: ConnectingTcpRemote::new(preferred_addrs, config.connect_timeout), - fallback: Some(ConnectingTcpFallback { - delay: tokio::time::sleep(fallback_timeout), - remote: ConnectingTcpRemote::new(fallback_addrs, config.connect_timeout), - }), - config, - } - } else { - ConnectingTcp { - preferred: ConnectingTcpRemote::new(remote_addrs, config.connect_timeout), - fallback: None, - config, - } - } - } -} - -struct ConnectingTcpFallback { - delay: Sleep, - remote: ConnectingTcpRemote, -} - -struct ConnectingTcpRemote { - addrs: dns::SocketAddrs, - connect_timeout: Option<Duration>, -} - -impl ConnectingTcpRemote { - fn new(addrs: dns::SocketAddrs, connect_timeout: Option<Duration>) -> Self { - let connect_timeout = connect_timeout.and_then(|t| t.checked_div(addrs.len() as u32)); - - Self { - addrs, - connect_timeout, - } - } -} - -impl ConnectingTcpRemote { - async fn connect(&mut self, config: &Config) -> Result<TcpStream, ConnectError> { - let mut err = None; - for addr in &mut self.addrs { - debug!("connecting to {}", addr); - match connect(&addr, config, self.connect_timeout)?.await { - Ok(tcp) => { - debug!("connected to {}", addr); - return Ok(tcp); - } - Err(mut e) => { - trace!("connect error for {}: {:?}", addr, e); - e.addr = Some(addr); - // only return the first error, we assume it's the most relevant - if err.is_none() { - err = Some(e); - } - } - } - } - - match err { - Some(e) => Err(e), - None => Err(ConnectError::new( - "tcp connect error", - std::io::Error::new(std::io::ErrorKind::NotConnected, "Network unreachable"), - )), - } - } -} - -fn bind_local_address( - socket: &socket2::Socket, - dst_addr: &SocketAddr, - local_addr_ipv4: &Option<Ipv4Addr>, - local_addr_ipv6: &Option<Ipv6Addr>, -) -> io::Result<()> { - match (*dst_addr, local_addr_ipv4, local_addr_ipv6) { - (SocketAddr::V4(_), Some(addr), _) => { - socket.bind(&SocketAddr::new((*addr).into(), 0).into())?; - } - (SocketAddr::V6(_), _, Some(addr)) => { - socket.bind(&SocketAddr::new((*addr).into(), 0).into())?; - } - _ => { - if cfg!(windows) { - // Windows requires a socket be bound before calling connect - let any: SocketAddr = match *dst_addr { - SocketAddr::V4(_) => ([0, 0, 0, 0], 0).into(), - SocketAddr::V6(_) => ([0, 0, 0, 0, 0, 0, 0, 0], 0).into(), - }; - socket.bind(&any.into())?; - } - } - } - - Ok(()) -} - -fn connect( - addr: &SocketAddr, - config: &Config, - connect_timeout: Option<Duration>, -) -> Result<impl Future<Output = Result<TcpStream, ConnectError>>, ConnectError> { - // TODO(eliza): if Tokio's `TcpSocket` gains support for setting the - // keepalive timeout, it would be nice to use that instead of socket2, - // and avoid the unsafe `into_raw_fd`/`from_raw_fd` dance... - use socket2::{Domain, Protocol, Socket, Type}; - - let domain = Domain::for_address(*addr); - let socket = Socket::new(domain, Type::STREAM, Some(Protocol::TCP)) - .map_err(ConnectError::m("tcp open error"))?; - - // When constructing a Tokio `TcpSocket` from a raw fd/socket, the user is - // responsible for ensuring O_NONBLOCK is set. - socket - .set_nonblocking(true) - .map_err(ConnectError::m("tcp set_nonblocking error"))?; - - if let Some(tcp_keepalive) = &config.tcp_keepalive_config.into_tcpkeepalive() { - if let Err(e) = socket.set_tcp_keepalive(tcp_keepalive) { - warn!("tcp set_keepalive error: {}", e); - } - } - - // That this only works for some socket types, particularly AF_INET sockets. - #[cfg(any( - target_os = "android", - target_os = "fuchsia", - target_os = "illumos", - target_os = "ios", - target_os = "linux", - target_os = "macos", - target_os = "solaris", - target_os = "tvos", - target_os = "visionos", - target_os = "watchos", - ))] - if let Some(interface) = &config.interface { - // On Linux-like systems, set the interface to bind using - // `SO_BINDTODEVICE`. - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - socket - .bind_device(Some(interface.as_bytes())) - .map_err(ConnectError::m("tcp bind interface error"))?; - - // On macOS-like and Solaris-like systems, we instead use `IP_BOUND_IF`. - // This socket option desires an integer index for the interface, so we - // must first determine the index of the requested interface name using - // `if_nametoindex`. - #[cfg(any( - target_os = "illumos", - target_os = "ios", - target_os = "macos", - target_os = "solaris", - target_os = "tvos", - target_os = "visionos", - target_os = "watchos", - ))] - { - let idx = unsafe { libc::if_nametoindex(interface.as_ptr()) }; - let idx = std::num::NonZeroU32::new(idx).ok_or_else(|| { - // If the index is 0, check errno and return an I/O error. - ConnectError::new( - "error converting interface name to index", - io::Error::last_os_error(), - ) - })?; - // Different setsockopt calls are necessary depending on whether the - // address is IPv4 or IPv6. - match addr { - SocketAddr::V4(_) => socket.bind_device_by_index_v4(Some(idx)), - SocketAddr::V6(_) => socket.bind_device_by_index_v6(Some(idx)), - } - .map_err(ConnectError::m("tcp bind interface error"))?; - } - } - - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - if let Some(tcp_user_timeout) = &config.tcp_user_timeout { - if let Err(e) = socket.set_tcp_user_timeout(Some(*tcp_user_timeout)) { - warn!("tcp set_tcp_user_timeout error: {}", e); - } - } - - bind_local_address( - &socket, - addr, - &config.local_address_ipv4, - &config.local_address_ipv6, - ) - .map_err(ConnectError::m("tcp bind local error"))?; - - #[cfg(unix)] - let socket = unsafe { - // Safety: `from_raw_fd` is only safe to call if ownership of the raw - // file descriptor is transferred. Since we call `into_raw_fd` on the - // socket2 socket, it gives up ownership of the fd and will not close - // it, so this is safe. - use std::os::unix::io::{FromRawFd, IntoRawFd}; - TcpSocket::from_raw_fd(socket.into_raw_fd()) - }; - #[cfg(windows)] - let socket = unsafe { - // Safety: `from_raw_socket` is only safe to call if ownership of the raw - // Windows SOCKET is transferred. Since we call `into_raw_socket` on the - // socket2 socket, it gives up ownership of the SOCKET and will not close - // it, so this is safe. - use std::os::windows::io::{FromRawSocket, IntoRawSocket}; - TcpSocket::from_raw_socket(socket.into_raw_socket()) - }; - - if config.reuse_address { - if let Err(e) = socket.set_reuseaddr(true) { - warn!("tcp set_reuse_address error: {}", e); - } - } - - if let Some(size) = config.send_buffer_size { - if let Err(e) = socket.set_send_buffer_size(size.try_into().unwrap_or(u32::MAX)) { - warn!("tcp set_buffer_size error: {}", e); - } - } - - if let Some(size) = config.recv_buffer_size { - if let Err(e) = socket.set_recv_buffer_size(size.try_into().unwrap_or(u32::MAX)) { - warn!("tcp set_recv_buffer_size error: {}", e); - } - } - - let connect = socket.connect(*addr); - Ok(async move { - match connect_timeout { - Some(dur) => match tokio::time::timeout(dur, connect).await { - Ok(Ok(s)) => Ok(s), - Ok(Err(e)) => Err(e), - Err(e) => Err(io::Error::new(io::ErrorKind::TimedOut, e)), - }, - None => connect.await, - } - .map_err(ConnectError::m("tcp connect error")) - }) -} - -impl ConnectingTcp<'_> { - async fn connect(mut self) -> Result<TcpStream, ConnectError> { - match self.fallback { - None => self.preferred.connect(self.config).await, - Some(mut fallback) => { - let preferred_fut = self.preferred.connect(self.config); - futures_util::pin_mut!(preferred_fut); - - let fallback_fut = fallback.remote.connect(self.config); - futures_util::pin_mut!(fallback_fut); - - let fallback_delay = fallback.delay; - futures_util::pin_mut!(fallback_delay); - - let (result, future) = - match futures_util::future::select(preferred_fut, fallback_delay).await { - Either::Left((result, _fallback_delay)) => { - (result, Either::Right(fallback_fut)) - } - Either::Right(((), preferred_fut)) => { - // Delay is done, start polling both the preferred and the fallback - futures_util::future::select(preferred_fut, fallback_fut) - .await - .factor_first() - } - }; - - if result.is_err() { - // Fallback to the remaining future (could be preferred or fallback) - // if we get an error - future.await - } else { - result - } - } - } - } -} - -/// Respect explicit ports in the URI, if none, either -/// keep non `0` ports resolved from a custom dns resolver, -/// or use the default port for the scheme. -fn set_port(addr: &mut SocketAddr, host_port: u16, explicit: bool) { - if explicit || addr.port() == 0 { - addr.set_port(host_port) - }; -} - -#[cfg(test)] -mod tests { - use std::io; - use std::net::SocketAddr; - - use ::http::Uri; - - use crate::client::legacy::connect::http::TcpKeepaliveConfig; - - use super::super::sealed::{Connect, ConnectSvc}; - use super::{Config, ConnectError, HttpConnector}; - - use super::set_port; - - async fn connect<C>( - connector: C, - dst: Uri, - ) -> Result<<C::_Svc as ConnectSvc>::Connection, <C::_Svc as ConnectSvc>::Error> - where - C: Connect, - { - connector.connect(super::super::sealed::Internal, dst).await - } - - #[tokio::test] - #[cfg_attr(miri, ignore)] - async fn test_errors_enforce_http() { - let dst = "https://example.domain/foo/bar?baz".parse().unwrap(); - let connector = HttpConnector::new(); - - let err = connect(connector, dst).await.unwrap_err(); - assert_eq!(&*err.msg, super::INVALID_NOT_HTTP); - } - - #[cfg(any(target_os = "linux", target_os = "macos"))] - fn get_local_ips() -> (Option<std::net::Ipv4Addr>, Option<std::net::Ipv6Addr>) { - use std::net::{IpAddr, TcpListener}; - - let mut ip_v4 = None; - let mut ip_v6 = None; - - let ips = pnet_datalink::interfaces() - .into_iter() - .flat_map(|i| i.ips.into_iter().map(|n| n.ip())); - - for ip in ips { - match ip { - IpAddr::V4(ip) if TcpListener::bind((ip, 0)).is_ok() => ip_v4 = Some(ip), - IpAddr::V6(ip) if TcpListener::bind((ip, 0)).is_ok() => ip_v6 = Some(ip), - _ => (), - } - - if ip_v4.is_some() && ip_v6.is_some() { - break; - } - } - - (ip_v4, ip_v6) - } - - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - fn default_interface() -> Option<String> { - pnet_datalink::interfaces() - .iter() - .find(|e| e.is_up() && !e.is_loopback() && !e.ips.is_empty()) - .map(|e| e.name.clone()) - } - - #[tokio::test] - #[cfg_attr(miri, ignore)] - async fn test_errors_missing_scheme() { - let dst = "example.domain".parse().unwrap(); - let mut connector = HttpConnector::new(); - connector.enforce_http(false); - - let err = connect(connector, dst).await.unwrap_err(); - assert_eq!(&*err.msg, super::INVALID_MISSING_SCHEME); - } - - // NOTE: pnet crate that we use in this test doesn't compile on Windows - #[cfg(any(target_os = "linux", target_os = "macos"))] - #[cfg_attr(miri, ignore)] - #[tokio::test] - async fn local_address() { - use std::net::{IpAddr, TcpListener}; - - let (bind_ip_v4, bind_ip_v6) = get_local_ips(); - let server4 = TcpListener::bind("127.0.0.1:0").unwrap(); - let port = server4.local_addr().unwrap().port(); - let server6 = TcpListener::bind(format!("[::1]:{port}")).unwrap(); - - let assert_client_ip = |dst: String, server: TcpListener, expected_ip: IpAddr| async move { - let mut connector = HttpConnector::new(); - - match (bind_ip_v4, bind_ip_v6) { - (Some(v4), Some(v6)) => connector.set_local_addresses(v4, v6), - (Some(v4), None) => connector.set_local_address(Some(v4.into())), - (None, Some(v6)) => connector.set_local_address(Some(v6.into())), - _ => unreachable!(), - } - - connect(connector, dst.parse().unwrap()).await.unwrap(); - - let (_, client_addr) = server.accept().unwrap(); - - assert_eq!(client_addr.ip(), expected_ip); - }; - - if let Some(ip) = bind_ip_v4 { - assert_client_ip(format!("http://127.0.0.1:{port}"), server4, ip.into()).await; - } - - if let Some(ip) = bind_ip_v6 { - assert_client_ip(format!("http://[::1]:{port}"), server6, ip.into()).await; - } - } - - // NOTE: pnet crate that we use in this test doesn't compile on Windows - #[cfg(any(target_os = "android", target_os = "fuchsia", target_os = "linux"))] - #[tokio::test] - #[ignore = "setting `SO_BINDTODEVICE` requires the `CAP_NET_RAW` capability (works when running as root)"] - async fn interface() { - use socket2::{Domain, Protocol, Socket, Type}; - use std::net::TcpListener; - - let interface: Option<String> = default_interface(); - - let server4 = TcpListener::bind("127.0.0.1:0").unwrap(); - let port = server4.local_addr().unwrap().port(); - - let server6 = TcpListener::bind(format!("[::1]:{port}")).unwrap(); - - let assert_interface_name = - |dst: String, - server: TcpListener, - bind_iface: Option<String>, - expected_interface: Option<String>| async move { - let mut connector = HttpConnector::new(); - if let Some(iface) = bind_iface { - connector.set_interface(iface); - } - - connect(connector, dst.parse().unwrap()).await.unwrap(); - let domain = Domain::for_address(server.local_addr().unwrap()); - let socket = Socket::new(domain, Type::STREAM, Some(Protocol::TCP)).unwrap(); - - assert_eq!( - socket.device().unwrap().as_deref(), - expected_interface.as_deref().map(|val| val.as_bytes()) - ); - }; - - assert_interface_name( - format!("http://127.0.0.1:{port}"), - server4, - interface.clone(), - interface.clone(), - ) - .await; - assert_interface_name( - format!("http://[::1]:{port}"), - server6, - interface.clone(), - interface.clone(), - ) - .await; - } - - #[test] - #[ignore] // TODO - #[cfg_attr(not(feature = "__internal_happy_eyeballs_tests"), ignore)] - fn client_happy_eyeballs() { - use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, TcpListener}; - use std::time::{Duration, Instant}; - - use super::dns; - use super::ConnectingTcp; - - let server4 = TcpListener::bind("127.0.0.1:0").unwrap(); - let addr = server4.local_addr().unwrap(); - let _server6 = TcpListener::bind(format!("[::1]:{}", addr.port())).unwrap(); - let rt = tokio::runtime::Builder::new_current_thread() - .enable_all() - .build() - .unwrap(); - - let local_timeout = Duration::default(); - let unreachable_v4_timeout = measure_connect(unreachable_ipv4_addr()).1; - let unreachable_v6_timeout = measure_connect(unreachable_ipv6_addr()).1; - let fallback_timeout = std::cmp::max(unreachable_v4_timeout, unreachable_v6_timeout) - + Duration::from_millis(250); - - let scenarios = &[ - // Fast primary, without fallback. - (&[local_ipv4_addr()][..], 4, local_timeout, false), - (&[local_ipv6_addr()][..], 6, local_timeout, false), - // Fast primary, with (unused) fallback. - ( - &[local_ipv4_addr(), local_ipv6_addr()][..], - 4, - local_timeout, - false, - ), - ( - &[local_ipv6_addr(), local_ipv4_addr()][..], - 6, - local_timeout, - false, - ), - // Unreachable + fast primary, without fallback. - ( - &[unreachable_ipv4_addr(), local_ipv4_addr()][..], - 4, - unreachable_v4_timeout, - false, - ), - ( - &[unreachable_ipv6_addr(), local_ipv6_addr()][..], - 6, - unreachable_v6_timeout, - false, - ), - // Unreachable + fast primary, with (unused) fallback. - ( - &[ - unreachable_ipv4_addr(), - local_ipv4_addr(), - local_ipv6_addr(), - ][..], - 4, - unreachable_v4_timeout, - false, - ), - ( - &[ - unreachable_ipv6_addr(), - local_ipv6_addr(), - local_ipv4_addr(), - ][..], - 6, - unreachable_v6_timeout, - true, - ), - // Slow primary, with (used) fallback. - ( - &[slow_ipv4_addr(), local_ipv4_addr(), local_ipv6_addr()][..], - 6, - fallback_timeout, - false, - ), - ( - &[slow_ipv6_addr(), local_ipv6_addr(), local_ipv4_addr()][..], - 4, - fallback_timeout, - true, - ), - // Slow primary, with (used) unreachable + fast fallback. - ( - &[slow_ipv4_addr(), unreachable_ipv6_addr(), local_ipv6_addr()][..], - 6, - fallback_timeout + unreachable_v6_timeout, - false, - ), - ( - &[slow_ipv6_addr(), unreachable_ipv4_addr(), local_ipv4_addr()][..], - 4, - fallback_timeout + unreachable_v4_timeout, - true, - ), - ]; - - // Scenarios for IPv6 -> IPv4 fallback require that host can access IPv6 network. - // Otherwise, connection to "slow" IPv6 address will error-out immediately. - let ipv6_accessible = measure_connect(slow_ipv6_addr()).0; - - for &(hosts, family, timeout, needs_ipv6_access) in scenarios { - if needs_ipv6_access && !ipv6_accessible { - continue; - } - - let (start, stream) = rt - .block_on(async move { - let addrs = hosts - .iter() - .map(|host| (*host, addr.port()).into()) - .collect(); - let cfg = Config { - local_address_ipv4: None, - local_address_ipv6: None, - connect_timeout: None, - tcp_keepalive_config: TcpKeepaliveConfig::default(), - happy_eyeballs_timeout: Some(fallback_timeout), - nodelay: false, - reuse_address: false, - enforce_http: false, - send_buffer_size: None, - recv_buffer_size: None, - #[cfg(any( - target_os = "android", - target_os = "fuchsia", - target_os = "linux" - ))] - interface: None, - #[cfg(any( - target_os = "illumos", - target_os = "ios", - target_os = "macos", - target_os = "solaris", - target_os = "tvos", - target_os = "visionos", - target_os = "watchos", - ))] - interface: None, - #[cfg(any( - target_os = "android", - target_os = "fuchsia", - target_os = "linux" - ))] - tcp_user_timeout: None, - }; - let connecting_tcp = ConnectingTcp::new(dns::SocketAddrs::new(addrs), &cfg); - let start = Instant::now(); - Ok::<_, ConnectError>((start, ConnectingTcp::connect(connecting_tcp).await?)) - }) - .unwrap(); - let res = if stream.peer_addr().unwrap().is_ipv4() { - 4 - } else { - 6 - }; - let duration = start.elapsed(); - - // Allow actual duration to be +/- 150ms off. - let min_duration = if timeout >= Duration::from_millis(150) { - timeout - Duration::from_millis(150) - } else { - Duration::default() - }; - let max_duration = timeout + Duration::from_millis(150); - - assert_eq!(res, family); - assert!(duration >= min_duration); - assert!(duration <= max_duration); - } - - fn local_ipv4_addr() -> IpAddr { - Ipv4Addr::new(127, 0, 0, 1).into() - } - - fn local_ipv6_addr() -> IpAddr { - Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1).into() - } - - fn unreachable_ipv4_addr() -> IpAddr { - Ipv4Addr::new(127, 0, 0, 2).into() - } - - fn unreachable_ipv6_addr() -> IpAddr { - Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 2).into() - } - - fn slow_ipv4_addr() -> IpAddr { - // RFC 6890 reserved IPv4 address. - Ipv4Addr::new(198, 18, 0, 25).into() - } - - fn slow_ipv6_addr() -> IpAddr { - // RFC 6890 reserved IPv6 address. - Ipv6Addr::new(2001, 2, 0, 0, 0, 0, 0, 254).into() - } - - fn measure_connect(addr: IpAddr) -> (bool, Duration) { - let start = Instant::now(); - let result = - std::net::TcpStream::connect_timeout(&(addr, 80).into(), Duration::from_secs(1)); - - let reachable = result.is_ok() || result.unwrap_err().kind() == io::ErrorKind::TimedOut; - let duration = start.elapsed(); - (reachable, duration) - } - } - - use std::time::Duration; - - #[test] - fn no_tcp_keepalive_config() { - assert!(TcpKeepaliveConfig::default().into_tcpkeepalive().is_none()); - } - - #[test] - fn tcp_keepalive_time_config() { - let kac = TcpKeepaliveConfig { - time: Some(Duration::from_secs(60)), - ..Default::default() - }; - if let Some(tcp_keepalive) = kac.into_tcpkeepalive() { - assert!(format!("{tcp_keepalive:?}").contains("time: Some(60s)")); - } else { - panic!("test failed"); - } - } - - #[cfg(not(any(target_os = "openbsd", target_os = "redox", target_os = "solaris")))] - #[test] - fn tcp_keepalive_interval_config() { - let kac = TcpKeepaliveConfig { - interval: Some(Duration::from_secs(1)), - ..Default::default() - }; - if let Some(tcp_keepalive) = kac.into_tcpkeepalive() { - assert!(format!("{tcp_keepalive:?}").contains("interval: Some(1s)")); - } else { - panic!("test failed"); - } - } - - #[cfg(not(any( - target_os = "openbsd", - target_os = "redox", - target_os = "solaris", - target_os = "windows" - )))] - #[test] - fn tcp_keepalive_retries_config() { - let kac = TcpKeepaliveConfig { - retries: Some(3), - ..Default::default() - }; - if let Some(tcp_keepalive) = kac.into_tcpkeepalive() { - assert!(format!("{tcp_keepalive:?}").contains("retries: Some(3)")); - } else { - panic!("test failed"); - } - } - - #[test] - fn test_set_port() { - // Respect explicit ports no matter what the resolved port is. - let mut addr = SocketAddr::from(([0, 0, 0, 0], 6881)); - set_port(&mut addr, 42, true); - assert_eq!(addr.port(), 42); - - // Ignore default host port, and use the socket port instead. - let mut addr = SocketAddr::from(([0, 0, 0, 0], 6881)); - set_port(&mut addr, 443, false); - assert_eq!(addr.port(), 6881); - - // Use the default port if the resolved port is `0`. - let mut addr = SocketAddr::from(([0, 0, 0, 0], 0)); - set_port(&mut addr, 443, false); - assert_eq!(addr.port(), 443); - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/mod.rs b/vendor/hyper-util/src/client/legacy/connect/mod.rs deleted file mode 100644 index 90a97679..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/mod.rs +++ /dev/null @@ -1,444 +0,0 @@ -//! Connectors used by the `Client`. -//! -//! This module contains: -//! -//! - A default [`HttpConnector`][] that does DNS resolution and establishes -//! connections over TCP. -//! - Types to build custom connectors. -//! -//! # Connectors -//! -//! A "connector" is a [`Service`][] that takes a [`Uri`][] destination, and -//! its `Response` is some type implementing [`Read`][], [`Write`][], -//! and [`Connection`][]. -//! -//! ## Custom Connectors -//! -//! A simple connector that ignores the `Uri` destination and always returns -//! a TCP connection to the same address could be written like this: -//! -//! ```rust,ignore -//! let connector = tower::service_fn(|_dst| async { -//! tokio::net::TcpStream::connect("127.0.0.1:1337") -//! }) -//! ``` -//! -//! Or, fully written out: -//! -//! ``` -//! use std::{future::Future, net::SocketAddr, pin::Pin, task::{self, Poll}}; -//! use http::Uri; -//! use tokio::net::TcpStream; -//! use tower_service::Service; -//! -//! #[derive(Clone)] -//! struct LocalConnector; -//! -//! impl Service<Uri> for LocalConnector { -//! type Response = TcpStream; -//! type Error = std::io::Error; -//! // We can't "name" an `async` generated future. -//! type Future = Pin<Box< -//! dyn Future<Output = Result<Self::Response, Self::Error>> + Send -//! >>; -//! -//! fn poll_ready(&mut self, _: &mut task::Context<'_>) -> Poll<Result<(), Self::Error>> { -//! // This connector is always ready, but others might not be. -//! Poll::Ready(Ok(())) -//! } -//! -//! fn call(&mut self, _: Uri) -> Self::Future { -//! Box::pin(TcpStream::connect(SocketAddr::from(([127, 0, 0, 1], 1337)))) -//! } -//! } -//! ``` -//! -//! It's worth noting that for `TcpStream`s, the [`HttpConnector`][] is a -//! better starting place to extend from. -//! -//! [`HttpConnector`]: HttpConnector -//! [`Service`]: tower_service::Service -//! [`Uri`]: ::http::Uri -//! [`Read`]: hyper::rt::Read -//! [`Write`]: hyper::rt::Write -//! [`Connection`]: Connection -use std::{ - fmt::{self, Formatter}, - sync::{ - atomic::{AtomicBool, Ordering}, - Arc, - }, -}; - -use ::http::Extensions; - -#[cfg(feature = "tokio")] -pub use self::http::{HttpConnector, HttpInfo}; - -#[cfg(feature = "tokio")] -pub mod dns; -#[cfg(feature = "tokio")] -mod http; - -pub mod proxy; - -pub(crate) mod capture; -pub use capture::{capture_connection, CaptureConnection}; - -pub use self::sealed::Connect; - -/// Describes a type returned by a connector. -pub trait Connection { - /// Return metadata describing the connection. - fn connected(&self) -> Connected; -} - -/// Extra information about the connected transport. -/// -/// This can be used to inform recipients about things like if ALPN -/// was used, or if connected to an HTTP proxy. -#[derive(Debug)] -pub struct Connected { - pub(super) alpn: Alpn, - pub(super) is_proxied: bool, - pub(super) extra: Option<Extra>, - pub(super) poisoned: PoisonPill, -} - -#[derive(Clone)] -pub(crate) struct PoisonPill { - poisoned: Arc<AtomicBool>, -} - -impl fmt::Debug for PoisonPill { - fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result { - // print the address of the pill—this makes debugging issues much easier - write!( - f, - "PoisonPill@{:p} {{ poisoned: {} }}", - self.poisoned, - self.poisoned.load(Ordering::Relaxed) - ) - } -} - -impl PoisonPill { - pub(crate) fn healthy() -> Self { - Self { - poisoned: Arc::new(AtomicBool::new(false)), - } - } - pub(crate) fn poison(&self) { - self.poisoned.store(true, Ordering::Relaxed) - } - - pub(crate) fn poisoned(&self) -> bool { - self.poisoned.load(Ordering::Relaxed) - } -} - -pub(super) struct Extra(Box<dyn ExtraInner>); - -#[derive(Clone, Copy, Debug, PartialEq)] -pub(super) enum Alpn { - H2, - None, -} - -impl Connected { - /// Create new `Connected` type with empty metadata. - pub fn new() -> Connected { - Connected { - alpn: Alpn::None, - is_proxied: false, - extra: None, - poisoned: PoisonPill::healthy(), - } - } - - /// Set whether the connected transport is to an HTTP proxy. - /// - /// This setting will affect if HTTP/1 requests written on the transport - /// will have the request-target in absolute-form or origin-form: - /// - /// - When `proxy(false)`: - /// - /// ```http - /// GET /guide HTTP/1.1 - /// ``` - /// - /// - When `proxy(true)`: - /// - /// ```http - /// GET http://hyper.rs/guide HTTP/1.1 - /// ``` - /// - /// Default is `false`. - pub fn proxy(mut self, is_proxied: bool) -> Connected { - self.is_proxied = is_proxied; - self - } - - /// Determines if the connected transport is to an HTTP proxy. - pub fn is_proxied(&self) -> bool { - self.is_proxied - } - - /// Set extra connection information to be set in the extensions of every `Response`. - pub fn extra<T: Clone + Send + Sync + 'static>(mut self, extra: T) -> Connected { - if let Some(prev) = self.extra { - self.extra = Some(Extra(Box::new(ExtraChain(prev.0, extra)))); - } else { - self.extra = Some(Extra(Box::new(ExtraEnvelope(extra)))); - } - self - } - - /// Copies the extra connection information into an `Extensions` map. - pub fn get_extras(&self, extensions: &mut Extensions) { - if let Some(extra) = &self.extra { - extra.set(extensions); - } - } - - /// Set that the connected transport negotiated HTTP/2 as its next protocol. - pub fn negotiated_h2(mut self) -> Connected { - self.alpn = Alpn::H2; - self - } - - /// Determines if the connected transport negotiated HTTP/2 as its next protocol. - pub fn is_negotiated_h2(&self) -> bool { - self.alpn == Alpn::H2 - } - - /// Poison this connection - /// - /// A poisoned connection will not be reused for subsequent requests by the pool - pub fn poison(&self) { - self.poisoned.poison(); - tracing::debug!( - poison_pill = ?self.poisoned, "connection was poisoned. this connection will not be reused for subsequent requests" - ); - } - - // Don't public expose that `Connected` is `Clone`, unsure if we want to - // keep that contract... - pub(super) fn clone(&self) -> Connected { - Connected { - alpn: self.alpn, - is_proxied: self.is_proxied, - extra: self.extra.clone(), - poisoned: self.poisoned.clone(), - } - } -} - -// ===== impl Extra ===== - -impl Extra { - pub(super) fn set(&self, res: &mut Extensions) { - self.0.set(res); - } -} - -impl Clone for Extra { - fn clone(&self) -> Extra { - Extra(self.0.clone_box()) - } -} - -impl fmt::Debug for Extra { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.debug_struct("Extra").finish() - } -} - -trait ExtraInner: Send + Sync { - fn clone_box(&self) -> Box<dyn ExtraInner>; - fn set(&self, res: &mut Extensions); -} - -// This indirection allows the `Connected` to have a type-erased "extra" value, -// while that type still knows its inner extra type. This allows the correct -// TypeId to be used when inserting into `res.extensions_mut()`. -#[derive(Clone)] -struct ExtraEnvelope<T>(T); - -impl<T> ExtraInner for ExtraEnvelope<T> -where - T: Clone + Send + Sync + 'static, -{ - fn clone_box(&self) -> Box<dyn ExtraInner> { - Box::new(self.clone()) - } - - fn set(&self, res: &mut Extensions) { - res.insert(self.0.clone()); - } -} - -struct ExtraChain<T>(Box<dyn ExtraInner>, T); - -impl<T: Clone> Clone for ExtraChain<T> { - fn clone(&self) -> Self { - ExtraChain(self.0.clone_box(), self.1.clone()) - } -} - -impl<T> ExtraInner for ExtraChain<T> -where - T: Clone + Send + Sync + 'static, -{ - fn clone_box(&self) -> Box<dyn ExtraInner> { - Box::new(self.clone()) - } - - fn set(&self, res: &mut Extensions) { - self.0.set(res); - res.insert(self.1.clone()); - } -} - -pub(super) mod sealed { - use std::error::Error as StdError; - use std::future::Future; - - use ::http::Uri; - use hyper::rt::{Read, Write}; - - use super::Connection; - - /// Connect to a destination, returning an IO transport. - /// - /// A connector receives a [`Uri`](::http::Uri) and returns a `Future` of the - /// ready connection. - /// - /// # Trait Alias - /// - /// This is really just an *alias* for the `tower::Service` trait, with - /// additional bounds set for convenience *inside* hyper. You don't actually - /// implement this trait, but `tower::Service<Uri>` instead. - // The `Sized` bound is to prevent creating `dyn Connect`, since they cannot - // fit the `Connect` bounds because of the blanket impl for `Service`. - pub trait Connect: Sealed + Sized { - #[doc(hidden)] - type _Svc: ConnectSvc; - #[doc(hidden)] - fn connect(self, internal_only: Internal, dst: Uri) -> <Self::_Svc as ConnectSvc>::Future; - } - - pub trait ConnectSvc { - type Connection: Read + Write + Connection + Unpin + Send + 'static; - type Error: Into<Box<dyn StdError + Send + Sync>>; - type Future: Future<Output = Result<Self::Connection, Self::Error>> + Unpin + Send + 'static; - - fn connect(self, internal_only: Internal, dst: Uri) -> Self::Future; - } - - impl<S, T> Connect for S - where - S: tower_service::Service<Uri, Response = T> + Send + 'static, - S::Error: Into<Box<dyn StdError + Send + Sync>>, - S::Future: Unpin + Send, - T: Read + Write + Connection + Unpin + Send + 'static, - { - type _Svc = S; - - fn connect(self, _: Internal, dst: Uri) -> crate::service::Oneshot<S, Uri> { - crate::service::Oneshot::new(self, dst) - } - } - - impl<S, T> ConnectSvc for S - where - S: tower_service::Service<Uri, Response = T> + Send + 'static, - S::Error: Into<Box<dyn StdError + Send + Sync>>, - S::Future: Unpin + Send, - T: Read + Write + Connection + Unpin + Send + 'static, - { - type Connection = T; - type Error = S::Error; - type Future = crate::service::Oneshot<S, Uri>; - - fn connect(self, _: Internal, dst: Uri) -> Self::Future { - crate::service::Oneshot::new(self, dst) - } - } - - impl<S, T> Sealed for S - where - S: tower_service::Service<Uri, Response = T> + Send, - S::Error: Into<Box<dyn StdError + Send + Sync>>, - S::Future: Unpin + Send, - T: Read + Write + Connection + Unpin + Send + 'static, - { - } - - pub trait Sealed {} - #[allow(missing_debug_implementations)] - pub struct Internal; -} - -#[cfg(test)] -mod tests { - use super::Connected; - - #[derive(Clone, Debug, PartialEq)] - struct Ex1(usize); - - #[derive(Clone, Debug, PartialEq)] - struct Ex2(&'static str); - - #[derive(Clone, Debug, PartialEq)] - struct Ex3(&'static str); - - #[test] - fn test_connected_extra() { - let c1 = Connected::new().extra(Ex1(41)); - - let mut ex = ::http::Extensions::new(); - - assert_eq!(ex.get::<Ex1>(), None); - - c1.extra.as_ref().expect("c1 extra").set(&mut ex); - - assert_eq!(ex.get::<Ex1>(), Some(&Ex1(41))); - } - - #[test] - fn test_connected_extra_chain() { - // If a user composes connectors and at each stage, there's "extra" - // info to attach, it shouldn't override the previous extras. - - let c1 = Connected::new() - .extra(Ex1(45)) - .extra(Ex2("zoom")) - .extra(Ex3("pew pew")); - - let mut ex1 = ::http::Extensions::new(); - - assert_eq!(ex1.get::<Ex1>(), None); - assert_eq!(ex1.get::<Ex2>(), None); - assert_eq!(ex1.get::<Ex3>(), None); - - c1.extra.as_ref().expect("c1 extra").set(&mut ex1); - - assert_eq!(ex1.get::<Ex1>(), Some(&Ex1(45))); - assert_eq!(ex1.get::<Ex2>(), Some(&Ex2("zoom"))); - assert_eq!(ex1.get::<Ex3>(), Some(&Ex3("pew pew"))); - - // Just like extensions, inserting the same type overrides previous type. - let c2 = Connected::new() - .extra(Ex1(33)) - .extra(Ex2("hiccup")) - .extra(Ex1(99)); - - let mut ex2 = ::http::Extensions::new(); - - c2.extra.as_ref().expect("c2 extra").set(&mut ex2); - - assert_eq!(ex2.get::<Ex1>(), Some(&Ex1(99))); - assert_eq!(ex2.get::<Ex2>(), Some(&Ex2("hiccup"))); - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/mod.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/mod.rs deleted file mode 100644 index 56ca3291..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/mod.rs +++ /dev/null @@ -1,6 +0,0 @@ -//! Proxy helpers -mod socks; -mod tunnel; - -pub use self::socks::{SocksV4, SocksV5}; -pub use self::tunnel::Tunnel; diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/mod.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/socks/mod.rs deleted file mode 100644 index d6077b94..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/mod.rs +++ /dev/null @@ -1,121 +0,0 @@ -mod v5; -pub use v5::{SocksV5, SocksV5Error}; - -mod v4; -pub use v4::{SocksV4, SocksV4Error}; - -use bytes::BytesMut; - -use hyper::rt::Read; - -#[derive(Debug)] -pub enum SocksError<C> { - Inner(C), - Io(std::io::Error), - - DnsFailure, - MissingHost, - MissingPort, - - V4(SocksV4Error), - V5(SocksV5Error), - - Parsing(ParsingError), - Serialize(SerializeError), -} - -#[derive(Debug)] -pub enum ParsingError { - Incomplete, - WouldOverflow, - Other, -} - -#[derive(Debug)] -pub enum SerializeError { - WouldOverflow, -} - -async fn read_message<T, M, C>(mut conn: &mut T, buf: &mut BytesMut) -> Result<M, SocksError<C>> -where - T: Read + Unpin, - M: for<'a> TryFrom<&'a mut BytesMut, Error = ParsingError>, -{ - let mut tmp = [0; 513]; - - loop { - let n = crate::rt::read(&mut conn, &mut tmp).await?; - buf.extend_from_slice(&tmp[..n]); - - match M::try_from(buf) { - Err(ParsingError::Incomplete) => { - if n == 0 { - if buf.spare_capacity_mut().is_empty() { - return Err(SocksError::Parsing(ParsingError::WouldOverflow)); - } else { - return Err(std::io::Error::new( - std::io::ErrorKind::UnexpectedEof, - "unexpected eof", - ) - .into()); - } - } - } - Err(err) => return Err(err.into()), - Ok(res) => return Ok(res), - } - } -} - -impl<C> std::fmt::Display for SocksError<C> { - fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { - f.write_str("SOCKS error: ")?; - - match self { - Self::Inner(_) => f.write_str("failed to create underlying connection"), - Self::Io(_) => f.write_str("io error during SOCKS handshake"), - - Self::DnsFailure => f.write_str("could not resolve to acceptable address type"), - Self::MissingHost => f.write_str("missing destination host"), - Self::MissingPort => f.write_str("missing destination port"), - - Self::Parsing(_) => f.write_str("failed parsing server response"), - Self::Serialize(_) => f.write_str("failed serialize request"), - - Self::V4(e) => e.fmt(f), - Self::V5(e) => e.fmt(f), - } - } -} - -impl<C: std::fmt::Debug + std::fmt::Display> std::error::Error for SocksError<C> {} - -impl<C> From<std::io::Error> for SocksError<C> { - fn from(err: std::io::Error) -> Self { - Self::Io(err) - } -} - -impl<C> From<ParsingError> for SocksError<C> { - fn from(err: ParsingError) -> Self { - Self::Parsing(err) - } -} - -impl<C> From<SerializeError> for SocksError<C> { - fn from(err: SerializeError) -> Self { - Self::Serialize(err) - } -} - -impl<C> From<SocksV4Error> for SocksError<C> { - fn from(err: SocksV4Error) -> Self { - Self::V4(err) - } -} - -impl<C> From<SocksV5Error> for SocksError<C> { - fn from(err: SocksV5Error) -> Self { - Self::V5(err) - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/errors.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/errors.rs deleted file mode 100644 index 5fdbd05c..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/errors.rs +++ /dev/null @@ -1,22 +0,0 @@ -use super::Status; - -#[derive(Debug)] -pub enum SocksV4Error { - IpV6, - Command(Status), -} - -impl From<Status> for SocksV4Error { - fn from(err: Status) -> Self { - Self::Command(err) - } -} - -impl std::fmt::Display for SocksV4Error { - fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { - match self { - Self::IpV6 => f.write_str("IPV6 is not supported"), - Self::Command(status) => status.fmt(f), - } - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/messages.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/messages.rs deleted file mode 100644 index bec8d081..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/messages.rs +++ /dev/null @@ -1,131 +0,0 @@ -use super::super::{ParsingError, SerializeError}; - -use bytes::{Buf, BufMut, BytesMut}; -use std::net::SocketAddrV4; - -/// +-----+-----+----+----+----+----+----+----+-------------+------+------------+------+ -/// | VN | CD | DSTPORT | DSTIP | USERID | NULL | DOMAIN | NULL | -/// +-----+-----+----+----+----+----+----+----+-------------+------+------------+------+ -/// | 1 | 1 | 2 | 4 | Variable | 1 | Variable | 1 | -/// +-----+-----+----+----+----+----+----+----+-------------+------+------------+------+ -/// ^^^^^^^^^^^^^^^^^^^^^ -/// optional: only do IP is 0.0.0.X -#[derive(Debug)] -pub struct Request<'a>(pub &'a Address); - -/// +-----+-----+----+----+----+----+----+----+ -/// | VN | CD | DSTPORT | DSTIP | -/// +-----+-----+----+----+----+----+----+----+ -/// | 1 | 1 | 2 | 4 | -/// +-----+-----+----+----+----+----+----+----+ -/// ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ -/// ignore: only for SOCKSv4 BIND -#[derive(Debug)] -pub struct Response(pub Status); - -#[derive(Debug)] -pub enum Address { - Socket(SocketAddrV4), - Domain(String, u16), -} - -#[derive(Debug, PartialEq)] -pub enum Status { - Success = 90, - Failed = 91, - IdentFailure = 92, - IdentMismatch = 93, -} - -impl Request<'_> { - pub fn write_to_buf<B: BufMut>(&self, mut buf: B) -> Result<usize, SerializeError> { - match self.0 { - Address::Socket(socket) => { - if buf.remaining_mut() < 10 { - return Err(SerializeError::WouldOverflow); - } - - buf.put_u8(0x04); // Version - buf.put_u8(0x01); // CONNECT - - buf.put_u16(socket.port()); // Port - buf.put_slice(&socket.ip().octets()); // IP - - buf.put_u8(0x00); // USERID - buf.put_u8(0x00); // NULL - - Ok(10) - } - - Address::Domain(domain, port) => { - if buf.remaining_mut() < 10 + domain.len() + 1 { - return Err(SerializeError::WouldOverflow); - } - - buf.put_u8(0x04); // Version - buf.put_u8(0x01); // CONNECT - - buf.put_u16(*port); // IP - buf.put_slice(&[0x00, 0x00, 0x00, 0xFF]); // Invalid IP - - buf.put_u8(0x00); // USERID - buf.put_u8(0x00); // NULL - - buf.put_slice(domain.as_bytes()); // Domain - buf.put_u8(0x00); // NULL - - Ok(10 + domain.len() + 1) - } - } - } -} - -impl TryFrom<&mut BytesMut> for Response { - type Error = ParsingError; - - fn try_from(buf: &mut BytesMut) -> Result<Self, Self::Error> { - if buf.remaining() < 8 { - return Err(ParsingError::Incomplete); - } - - if buf.get_u8() != 0x00 { - return Err(ParsingError::Other); - } - - let status = buf.get_u8().try_into()?; - let _addr = { - let port = buf.get_u16(); - let mut ip = [0; 4]; - buf.copy_to_slice(&mut ip); - - SocketAddrV4::new(ip.into(), port) - }; - - Ok(Self(status)) - } -} - -impl TryFrom<u8> for Status { - type Error = ParsingError; - - fn try_from(byte: u8) -> Result<Self, Self::Error> { - Ok(match byte { - 90 => Self::Success, - 91 => Self::Failed, - 92 => Self::IdentFailure, - 93 => Self::IdentMismatch, - _ => return Err(ParsingError::Other), - }) - } -} - -impl std::fmt::Display for Status { - fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { - f.write_str(match self { - Self::Success => "success", - Self::Failed => "server failed to execute command", - Self::IdentFailure => "server ident service failed", - Self::IdentMismatch => "server ident service did not recognise client identifier", - }) - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/mod.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/mod.rs deleted file mode 100644 index bee7e6dc..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v4/mod.rs +++ /dev/null @@ -1,183 +0,0 @@ -mod errors; -pub use errors::*; - -mod messages; -use messages::*; - -use std::future::Future; -use std::pin::Pin; -use std::task::{Context, Poll}; - -use std::net::{IpAddr, SocketAddr, SocketAddrV4, ToSocketAddrs}; - -use http::Uri; -use hyper::rt::{Read, Write}; -use tower_service::Service; - -use bytes::BytesMut; - -use pin_project_lite::pin_project; - -/// Tunnel Proxy via SOCKSv4 -/// -/// This is a connector that can be used by the `legacy::Client`. It wraps -/// another connector, and after getting an underlying connection, it established -/// a TCP tunnel over it using SOCKSv4. -#[derive(Debug, Clone)] -pub struct SocksV4<C> { - inner: C, - config: SocksConfig, -} - -#[derive(Debug, Clone)] -struct SocksConfig { - proxy: Uri, - local_dns: bool, -} - -pin_project! { - // Not publicly exported (so missing_docs doesn't trigger). - // - // We return this `Future` instead of the `Pin<Box<dyn Future>>` directly - // so that users don't rely on it fitting in a `Pin<Box<dyn Future>>` slot - // (and thus we can change the type in the future). - #[must_use = "futures do nothing unless polled"] - #[allow(missing_debug_implementations)] - pub struct Handshaking<F, T, E> { - #[pin] - fut: BoxHandshaking<T, E>, - _marker: std::marker::PhantomData<F> - } -} - -type BoxHandshaking<T, E> = Pin<Box<dyn Future<Output = Result<T, super::SocksError<E>>> + Send>>; - -impl<C> SocksV4<C> { - /// Create a new SOCKSv4 handshake service - /// - /// Wraps an underlying connector and stores the address of a tunneling - /// proxying server. - /// - /// A `SocksV4` can then be called with any destination. The `dst` passed to - /// `call` will not be used to create the underlying connection, but will - /// be used in a SOCKS handshake with the proxy destination. - pub fn new(proxy_dst: Uri, connector: C) -> Self { - Self { - inner: connector, - config: SocksConfig::new(proxy_dst), - } - } - - /// Resolve domain names locally on the client, rather than on the proxy server. - /// - /// Disabled by default as local resolution of domain names can be detected as a - /// DNS leak. - pub fn local_dns(mut self, local_dns: bool) -> Self { - self.config.local_dns = local_dns; - self - } -} - -impl SocksConfig { - pub fn new(proxy: Uri) -> Self { - Self { - proxy, - local_dns: false, - } - } - - async fn execute<T, E>( - self, - mut conn: T, - host: String, - port: u16, - ) -> Result<T, super::SocksError<E>> - where - T: Read + Write + Unpin, - { - let address = match host.parse::<IpAddr>() { - Ok(IpAddr::V6(_)) => return Err(SocksV4Error::IpV6.into()), - Ok(IpAddr::V4(ip)) => Address::Socket(SocketAddrV4::new(ip, port)), - Err(_) => { - if self.local_dns { - (host, port) - .to_socket_addrs()? - .find_map(|s| { - if let SocketAddr::V4(v4) = s { - Some(Address::Socket(v4)) - } else { - None - } - }) - .ok_or(super::SocksError::DnsFailure)? - } else { - Address::Domain(host, port) - } - } - }; - - let mut send_buf = BytesMut::with_capacity(1024); - let mut recv_buf = BytesMut::with_capacity(1024); - - // Send Request - let req = Request(&address); - let n = req.write_to_buf(&mut send_buf)?; - crate::rt::write_all(&mut conn, &send_buf[..n]).await?; - - // Read Response - let res: Response = super::read_message(&mut conn, &mut recv_buf).await?; - if res.0 == Status::Success { - Ok(conn) - } else { - Err(SocksV4Error::Command(res.0).into()) - } - } -} - -impl<C> Service<Uri> for SocksV4<C> -where - C: Service<Uri>, - C::Future: Send + 'static, - C::Response: Read + Write + Unpin + Send + 'static, - C::Error: Send + 'static, -{ - type Response = C::Response; - type Error = super::SocksError<C::Error>; - type Future = Handshaking<C::Future, C::Response, C::Error>; - - fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> { - self.inner.poll_ready(cx).map_err(super::SocksError::Inner) - } - - fn call(&mut self, dst: Uri) -> Self::Future { - let config = self.config.clone(); - let connecting = self.inner.call(config.proxy.clone()); - - let fut = async move { - let port = dst.port().map(|p| p.as_u16()).unwrap_or(443); - let host = dst - .host() - .ok_or(super::SocksError::MissingHost)? - .to_string(); - - let conn = connecting.await.map_err(super::SocksError::Inner)?; - config.execute(conn, host, port).await - }; - - Handshaking { - fut: Box::pin(fut), - _marker: Default::default(), - } - } -} - -impl<F, T, E> Future for Handshaking<F, T, E> -where - F: Future<Output = Result<T, E>>, -{ - type Output = Result<T, super::SocksError<E>>; - - fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> { - self.project().fut.poll(cx) - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/errors.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/errors.rs deleted file mode 100644 index 06b1a9a8..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/errors.rs +++ /dev/null @@ -1,47 +0,0 @@ -use super::Status; - -#[derive(Debug)] -pub enum SocksV5Error { - HostTooLong, - Auth(AuthError), - Command(Status), -} - -#[derive(Debug)] -pub enum AuthError { - Unsupported, - MethodMismatch, - Failed, -} - -impl From<Status> for SocksV5Error { - fn from(err: Status) -> Self { - Self::Command(err) - } -} - -impl From<AuthError> for SocksV5Error { - fn from(err: AuthError) -> Self { - Self::Auth(err) - } -} - -impl std::fmt::Display for SocksV5Error { - fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { - match self { - Self::HostTooLong => f.write_str("host address is more than 255 characters"), - Self::Command(e) => e.fmt(f), - Self::Auth(e) => e.fmt(f), - } - } -} - -impl std::fmt::Display for AuthError { - fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { - f.write_str(match self { - Self::Unsupported => "server does not support user/pass authentication", - Self::MethodMismatch => "server implements authentication incorrectly", - Self::Failed => "credentials not accepted", - }) - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/messages.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/messages.rs deleted file mode 100644 index ddf93538..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/messages.rs +++ /dev/null @@ -1,347 +0,0 @@ -use super::super::{ParsingError, SerializeError}; - -use bytes::{Buf, BufMut, BytesMut}; -use std::net::SocketAddr; - -/// +----+----------+----------+ -/// |VER | NMETHODS | METHODS | -/// +----+----------+----------+ -/// | 1 | 1 | 1 to 255 | -/// +----+----------+----------+ -#[derive(Debug)] -pub struct NegotiationReq<'a>(pub &'a AuthMethod); - -/// +----+--------+ -/// |VER | METHOD | -/// +----+--------+ -/// | 1 | 1 | -/// +----+--------+ -#[derive(Debug)] -pub struct NegotiationRes(pub AuthMethod); - -/// +----+------+----------+------+----------+ -/// |VER | ULEN | UNAME | PLEN | PASSWD | -/// +----+------+----------+------+----------+ -/// | 1 | 1 | 1 to 255 | 1 | 1 to 255 | -/// +----+------+----------+------+----------+ -#[derive(Debug)] -pub struct AuthenticationReq<'a>(pub &'a str, pub &'a str); - -/// +----+--------+ -/// |VER | STATUS | -/// +----+--------+ -/// | 1 | 1 | -/// +----+--------+ -#[derive(Debug)] -pub struct AuthenticationRes(pub bool); - -/// +----+-----+-------+------+----------+----------+ -/// |VER | CMD | RSV | ATYP | DST.ADDR | DST.PORT | -/// +----+-----+-------+------+----------+----------+ -/// | 1 | 1 | X'00' | 1 | Variable | 2 | -/// +----+-----+-------+------+----------+----------+ -#[derive(Debug)] -pub struct ProxyReq<'a>(pub &'a Address); - -/// +----+-----+-------+------+----------+----------+ -/// |VER | REP | RSV | ATYP | BND.ADDR | BND.PORT | -/// +----+-----+-------+------+----------+----------+ -/// | 1 | 1 | X'00' | 1 | Variable | 2 | -/// +----+-----+-------+------+----------+----------+ -#[derive(Debug)] -pub struct ProxyRes(pub Status); - -#[repr(u8)] -#[derive(Debug, Copy, Clone, PartialEq)] -pub enum AuthMethod { - NoAuth = 0x00, - UserPass = 0x02, - NoneAcceptable = 0xFF, -} - -#[derive(Debug)] -pub enum Address { - Socket(SocketAddr), - Domain(String, u16), -} - -#[derive(Debug, Copy, Clone, PartialEq)] -pub enum Status { - Success, - GeneralServerFailure, - ConnectionNotAllowed, - NetworkUnreachable, - HostUnreachable, - ConnectionRefused, - TtlExpired, - CommandNotSupported, - AddressTypeNotSupported, -} - -impl NegotiationReq<'_> { - pub fn write_to_buf(&self, buf: &mut BytesMut) -> Result<usize, SerializeError> { - if buf.capacity() - buf.len() < 3 { - return Err(SerializeError::WouldOverflow); - } - - buf.put_u8(0x05); // Version - buf.put_u8(0x01); // Number of authentication methods - buf.put_u8(*self.0 as u8); // Authentication method - - Ok(3) - } -} - -impl TryFrom<&mut BytesMut> for NegotiationRes { - type Error = ParsingError; - - fn try_from(buf: &mut BytesMut) -> Result<Self, ParsingError> { - if buf.remaining() < 2 { - return Err(ParsingError::Incomplete); - } - - if buf.get_u8() != 0x05 { - return Err(ParsingError::Other); - } - - let method = buf.get_u8().try_into()?; - Ok(Self(method)) - } -} - -impl AuthenticationReq<'_> { - pub fn write_to_buf(&self, buf: &mut BytesMut) -> Result<usize, SerializeError> { - if buf.capacity() - buf.len() < 3 + self.0.len() + self.1.len() { - return Err(SerializeError::WouldOverflow); - } - - buf.put_u8(0x01); // Version - - buf.put_u8(self.0.len() as u8); // Username length (guarenteed to be 255 or less) - buf.put_slice(self.0.as_bytes()); // Username - - buf.put_u8(self.1.len() as u8); // Password length (guarenteed to be 255 or less) - buf.put_slice(self.1.as_bytes()); // Password - - Ok(3 + self.0.len() + self.1.len()) - } -} - -impl TryFrom<&mut BytesMut> for AuthenticationRes { - type Error = ParsingError; - - fn try_from(buf: &mut BytesMut) -> Result<Self, ParsingError> { - if buf.remaining() < 2 { - return Err(ParsingError::Incomplete); - } - - if buf.get_u8() != 0x01 { - return Err(ParsingError::Other); - } - - if buf.get_u8() == 0 { - Ok(Self(true)) - } else { - Ok(Self(false)) - } - } -} - -impl ProxyReq<'_> { - pub fn write_to_buf(&self, buf: &mut BytesMut) -> Result<usize, SerializeError> { - let addr_len = match self.0 { - Address::Socket(SocketAddr::V4(_)) => 1 + 4 + 2, - Address::Socket(SocketAddr::V6(_)) => 1 + 16 + 2, - Address::Domain(ref domain, _) => 1 + 1 + domain.len() + 2, - }; - - if buf.capacity() - buf.len() < 3 + addr_len { - return Err(SerializeError::WouldOverflow); - } - - buf.put_u8(0x05); // Version - buf.put_u8(0x01); // TCP tunneling command - buf.put_u8(0x00); // Reserved - let _ = self.0.write_to_buf(buf); // Address - - Ok(3 + addr_len) - } -} - -impl TryFrom<&mut BytesMut> for ProxyRes { - type Error = ParsingError; - - fn try_from(buf: &mut BytesMut) -> Result<Self, ParsingError> { - if buf.remaining() < 2 { - return Err(ParsingError::Incomplete); - } - - // VER - if buf.get_u8() != 0x05 { - return Err(ParsingError::Other); - } - - // REP - let status = buf.get_u8().try_into()?; - - // RSV - if buf.get_u8() != 0x00 { - return Err(ParsingError::Other); - } - - // ATYP + ADDR - Address::try_from(buf)?; - - Ok(Self(status)) - } -} - -impl Address { - pub fn write_to_buf(&self, buf: &mut BytesMut) -> Result<usize, SerializeError> { - match self { - Self::Socket(SocketAddr::V4(v4)) => { - if buf.capacity() - buf.len() < 1 + 4 + 2 { - return Err(SerializeError::WouldOverflow); - } - - buf.put_u8(0x01); - buf.put_slice(&v4.ip().octets()); - buf.put_u16(v4.port()); // Network Order/BigEndian for port - - Ok(7) - } - - Self::Socket(SocketAddr::V6(v6)) => { - if buf.capacity() - buf.len() < 1 + 16 + 2 { - return Err(SerializeError::WouldOverflow); - } - - buf.put_u8(0x04); - buf.put_slice(&v6.ip().octets()); - buf.put_u16(v6.port()); // Network Order/BigEndian for port - - Ok(19) - } - - Self::Domain(domain, port) => { - if buf.capacity() - buf.len() < 1 + 1 + domain.len() + 2 { - return Err(SerializeError::WouldOverflow); - } - - buf.put_u8(0x03); - buf.put_u8(domain.len() as u8); // Guarenteed to be less than 255 - buf.put_slice(domain.as_bytes()); - buf.put_u16(*port); - - Ok(4 + domain.len()) - } - } - } -} - -impl TryFrom<&mut BytesMut> for Address { - type Error = ParsingError; - - fn try_from(buf: &mut BytesMut) -> Result<Self, Self::Error> { - if buf.remaining() < 2 { - return Err(ParsingError::Incomplete); - } - - Ok(match buf.get_u8() { - 0x01 => { - let mut ip = [0; 4]; - - if buf.remaining() < 6 { - return Err(ParsingError::Incomplete); - } - - buf.copy_to_slice(&mut ip); - let port = buf.get_u16(); - - Self::Socket(SocketAddr::new(ip.into(), port)) - } - - 0x03 => { - let len = buf.get_u8(); - - if len == 0 { - return Err(ParsingError::Other); - } else if buf.remaining() < (len as usize) + 2 { - return Err(ParsingError::Incomplete); - } - - let domain = std::str::from_utf8(&buf[..len as usize]) - .map_err(|_| ParsingError::Other)? - .to_string(); - - let port = buf.get_u16(); - - Self::Domain(domain, port) - } - - 0x04 => { - let mut ip = [0; 16]; - - if buf.remaining() < 6 { - return Err(ParsingError::Incomplete); - } - buf.copy_to_slice(&mut ip); - let port = buf.get_u16(); - - Self::Socket(SocketAddr::new(ip.into(), port)) - } - - _ => return Err(ParsingError::Other), - }) - } -} - -impl TryFrom<u8> for Status { - type Error = ParsingError; - - fn try_from(byte: u8) -> Result<Self, Self::Error> { - Ok(match byte { - 0x00 => Self::Success, - - 0x01 => Self::GeneralServerFailure, - 0x02 => Self::ConnectionNotAllowed, - 0x03 => Self::NetworkUnreachable, - 0x04 => Self::HostUnreachable, - 0x05 => Self::ConnectionRefused, - 0x06 => Self::TtlExpired, - 0x07 => Self::CommandNotSupported, - 0x08 => Self::AddressTypeNotSupported, - _ => return Err(ParsingError::Other), - }) - } -} - -impl TryFrom<u8> for AuthMethod { - type Error = ParsingError; - - fn try_from(byte: u8) -> Result<Self, Self::Error> { - Ok(match byte { - 0x00 => Self::NoAuth, - 0x02 => Self::UserPass, - 0xFF => Self::NoneAcceptable, - - _ => return Err(ParsingError::Other), - }) - } -} - -impl std::fmt::Display for Status { - fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { - f.write_str(match self { - Self::Success => "success", - Self::GeneralServerFailure => "general server failure", - Self::ConnectionNotAllowed => "connection not allowed", - Self::NetworkUnreachable => "network unreachable", - Self::HostUnreachable => "host unreachable", - Self::ConnectionRefused => "connection refused", - Self::TtlExpired => "ttl expired", - Self::CommandNotSupported => "command not supported", - Self::AddressTypeNotSupported => "address type not supported", - }) - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/mod.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/mod.rs deleted file mode 100644 index caf2446b..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/socks/v5/mod.rs +++ /dev/null @@ -1,313 +0,0 @@ -mod errors; -pub use errors::*; - -mod messages; -use messages::*; - -use std::future::Future; -use std::pin::Pin; -use std::task::{Context, Poll}; - -use std::net::{IpAddr, SocketAddr, ToSocketAddrs}; - -use http::Uri; -use hyper::rt::{Read, Write}; -use tower_service::Service; - -use bytes::BytesMut; - -use pin_project_lite::pin_project; - -/// Tunnel Proxy via SOCKSv5 -/// -/// This is a connector that can be used by the `legacy::Client`. It wraps -/// another connector, and after getting an underlying connection, it established -/// a TCP tunnel over it using SOCKSv5. -#[derive(Debug, Clone)] -pub struct SocksV5<C> { - inner: C, - config: SocksConfig, -} - -#[derive(Debug, Clone)] -pub struct SocksConfig { - proxy: Uri, - proxy_auth: Option<(String, String)>, - - local_dns: bool, - optimistic: bool, -} - -#[derive(Debug)] -enum State { - SendingNegReq, - ReadingNegRes, - SendingAuthReq, - ReadingAuthRes, - SendingProxyReq, - ReadingProxyRes, -} - -pin_project! { - // Not publicly exported (so missing_docs doesn't trigger). - // - // We return this `Future` instead of the `Pin<Box<dyn Future>>` directly - // so that users don't rely on it fitting in a `Pin<Box<dyn Future>>` slot - // (and thus we can change the type in the future). - #[must_use = "futures do nothing unless polled"] - #[allow(missing_debug_implementations)] - pub struct Handshaking<F, T, E> { - #[pin] - fut: BoxHandshaking<T, E>, - _marker: std::marker::PhantomData<F> - } -} - -type BoxHandshaking<T, E> = Pin<Box<dyn Future<Output = Result<T, super::SocksError<E>>> + Send>>; - -impl<C> SocksV5<C> { - /// Create a new SOCKSv5 handshake service. - /// - /// Wraps an underlying connector and stores the address of a tunneling - /// proxying server. - /// - /// A `SocksV5` can then be called with any destination. The `dst` passed to - /// `call` will not be used to create the underlying connection, but will - /// be used in a SOCKS handshake with the proxy destination. - pub fn new(proxy_dst: Uri, connector: C) -> Self { - Self { - inner: connector, - config: SocksConfig::new(proxy_dst), - } - } - - /// Use User/Pass authentication method during handshake. - /// - /// Username and Password must be maximum of 255 characters each. - /// 0 length strings are allowed despite RFC prohibiting it. This is done so that - /// for compatablity with server implementations that require it for IP authentication. - pub fn with_auth(mut self, user: String, pass: String) -> Self { - self.config.proxy_auth = Some((user, pass)); - self - } - - /// Resolve domain names locally on the client, rather than on the proxy server. - /// - /// Disabled by default as local resolution of domain names can be detected as a - /// DNS leak. - pub fn local_dns(mut self, local_dns: bool) -> Self { - self.config.local_dns = local_dns; - self - } - - /// Send all messages of the handshake optmistically (without waiting for server response). - /// - /// Typical SOCKS handshake with auithentication takes 3 round trips. Optimistic sending - /// can reduce round trip times and dramatically increase speed of handshake at the cost of - /// reduced portability; many server implementations do not support optimistic sending as it - /// is not defined in the RFC (RFC 1928). - /// - /// Recommended to ensure connector works correctly without optimistic sending before trying - /// with optimistic sending. - pub fn send_optimistically(mut self, optimistic: bool) -> Self { - self.config.optimistic = optimistic; - self - } -} - -impl SocksConfig { - fn new(proxy: Uri) -> Self { - Self { - proxy, - proxy_auth: None, - - local_dns: false, - optimistic: false, - } - } - - async fn execute<T, E>( - self, - mut conn: T, - host: String, - port: u16, - ) -> Result<T, super::SocksError<E>> - where - T: Read + Write + Unpin, - { - let address = match host.parse::<IpAddr>() { - Ok(ip) => Address::Socket(SocketAddr::new(ip, port)), - Err(_) if host.len() <= 255 => { - if self.local_dns { - let socket = (host, port) - .to_socket_addrs()? - .next() - .ok_or(super::SocksError::DnsFailure)?; - - Address::Socket(socket) - } else { - Address::Domain(host, port) - } - } - Err(_) => return Err(SocksV5Error::HostTooLong.into()), - }; - - let method = if self.proxy_auth.is_some() { - AuthMethod::UserPass - } else { - AuthMethod::NoAuth - }; - - let mut recv_buf = BytesMut::with_capacity(513); // Max length of valid recievable message is 513 from Auth Request - let mut send_buf = BytesMut::with_capacity(262); // Max length of valid sendable message is 262 from Auth Response - let mut state = State::SendingNegReq; - - loop { - match state { - State::SendingNegReq => { - let req = NegotiationReq(&method); - - let start = send_buf.len(); - req.write_to_buf(&mut send_buf)?; - crate::rt::write_all(&mut conn, &send_buf[start..]).await?; - - if self.optimistic { - if method == AuthMethod::UserPass { - state = State::SendingAuthReq; - } else { - state = State::SendingProxyReq; - } - } else { - state = State::ReadingNegRes; - } - } - - State::ReadingNegRes => { - let res: NegotiationRes = super::read_message(&mut conn, &mut recv_buf).await?; - - if res.0 == AuthMethod::NoneAcceptable { - return Err(SocksV5Error::Auth(AuthError::Unsupported).into()); - } - - if res.0 != method { - return Err(SocksV5Error::Auth(AuthError::MethodMismatch).into()); - } - - if self.optimistic { - if res.0 == AuthMethod::UserPass { - state = State::ReadingAuthRes; - } else { - state = State::ReadingProxyRes; - } - } else if res.0 == AuthMethod::UserPass { - state = State::SendingAuthReq; - } else { - state = State::SendingProxyReq; - } - } - - State::SendingAuthReq => { - let (user, pass) = self.proxy_auth.as_ref().unwrap(); - let req = AuthenticationReq(user, pass); - - let start = send_buf.len(); - req.write_to_buf(&mut send_buf)?; - crate::rt::write_all(&mut conn, &send_buf[start..]).await?; - - if self.optimistic { - state = State::SendingProxyReq; - } else { - state = State::ReadingAuthRes; - } - } - - State::ReadingAuthRes => { - let res: AuthenticationRes = - super::read_message(&mut conn, &mut recv_buf).await?; - - if !res.0 { - return Err(SocksV5Error::Auth(AuthError::Failed).into()); - } - - if self.optimistic { - state = State::ReadingProxyRes; - } else { - state = State::SendingProxyReq; - } - } - - State::SendingProxyReq => { - let req = ProxyReq(&address); - - let start = send_buf.len(); - req.write_to_buf(&mut send_buf)?; - crate::rt::write_all(&mut conn, &send_buf[start..]).await?; - - if self.optimistic { - state = State::ReadingNegRes; - } else { - state = State::ReadingProxyRes; - } - } - - State::ReadingProxyRes => { - let res: ProxyRes = super::read_message(&mut conn, &mut recv_buf).await?; - - if res.0 == Status::Success { - return Ok(conn); - } else { - return Err(SocksV5Error::Command(res.0).into()); - } - } - } - } - } -} - -impl<C> Service<Uri> for SocksV5<C> -where - C: Service<Uri>, - C::Future: Send + 'static, - C::Response: Read + Write + Unpin + Send + 'static, - C::Error: Send + 'static, -{ - type Response = C::Response; - type Error = super::SocksError<C::Error>; - type Future = Handshaking<C::Future, C::Response, C::Error>; - - fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> { - self.inner.poll_ready(cx).map_err(super::SocksError::Inner) - } - - fn call(&mut self, dst: Uri) -> Self::Future { - let config = self.config.clone(); - let connecting = self.inner.call(config.proxy.clone()); - - let fut = async move { - let port = dst.port().map(|p| p.as_u16()).unwrap_or(443); - let host = dst - .host() - .ok_or(super::SocksError::MissingHost)? - .to_string(); - - let conn = connecting.await.map_err(super::SocksError::Inner)?; - config.execute(conn, host, port).await - }; - - Handshaking { - fut: Box::pin(fut), - _marker: Default::default(), - } - } -} - -impl<F, T, E> Future for Handshaking<F, T, E> -where - F: Future<Output = Result<T, E>>, -{ - type Output = Result<T, super::SocksError<E>>; - - fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> { - self.project().fut.poll(cx) - } -} diff --git a/vendor/hyper-util/src/client/legacy/connect/proxy/tunnel.rs b/vendor/hyper-util/src/client/legacy/connect/proxy/tunnel.rs deleted file mode 100644 index ad948596..00000000 --- a/vendor/hyper-util/src/client/legacy/connect/proxy/tunnel.rs +++ /dev/null @@ -1,258 +0,0 @@ -use std::error::Error as StdError; -use std::future::Future; -use std::marker::{PhantomData, Unpin}; -use std::pin::Pin; -use std::task::{self, Poll}; - -use futures_core::ready; -use http::{HeaderMap, HeaderValue, Uri}; -use hyper::rt::{Read, Write}; -use pin_project_lite::pin_project; -use tower_service::Service; - -/// Tunnel Proxy via HTTP CONNECT -/// -/// This is a connector that can be used by the `legacy::Client`. It wraps -/// another connector, and after getting an underlying connection, it creates -/// an HTTP CONNECT tunnel over it. -#[derive(Debug)] -pub struct Tunnel<C> { - headers: Headers, - inner: C, - proxy_dst: Uri, -} - -#[derive(Clone, Debug)] -enum Headers { - Empty, - Auth(HeaderValue), - Extra(HeaderMap), -} - -#[derive(Debug)] -pub enum TunnelError { - ConnectFailed(Box<dyn StdError + Send + Sync>), - Io(std::io::Error), - MissingHost, - ProxyAuthRequired, - ProxyHeadersTooLong, - TunnelUnexpectedEof, - TunnelUnsuccessful, -} - -pin_project! { - // Not publicly exported (so missing_docs doesn't trigger). - // - // We return this `Future` instead of the `Pin<Box<dyn Future>>` directly - // so that users don't rely on it fitting in a `Pin<Box<dyn Future>>` slot - // (and thus we can change the type in the future). - #[must_use = "futures do nothing unless polled"] - #[allow(missing_debug_implementations)] - pub struct Tunneling<F, T> { - #[pin] - fut: BoxTunneling<T>, - _marker: PhantomData<F>, - } -} - -type BoxTunneling<T> = Pin<Box<dyn Future<Output = Result<T, TunnelError>> + Send>>; - -impl<C> Tunnel<C> { - /// Create a new Tunnel service. - /// - /// This wraps an underlying connector, and stores the address of a - /// tunneling proxy server. - /// - /// A `Tunnel` can then be called with any destination. The `dst` passed to - /// `call` will not be used to create the underlying connection, but will - /// be used in an HTTP CONNECT request sent to the proxy destination. - pub fn new(proxy_dst: Uri, connector: C) -> Self { - Self { - headers: Headers::Empty, - inner: connector, - proxy_dst, - } - } - - /// Add `proxy-authorization` header value to the CONNECT request. - pub fn with_auth(mut self, mut auth: HeaderValue) -> Self { - // just in case the user forgot - auth.set_sensitive(true); - match self.headers { - Headers::Empty => { - self.headers = Headers::Auth(auth); - } - Headers::Auth(ref mut existing) => { - *existing = auth; - } - Headers::Extra(ref mut extra) => { - extra.insert(http::header::PROXY_AUTHORIZATION, auth); - } - } - - self - } - - /// Add extra headers to be sent with the CONNECT request. - /// - /// If existing headers have been set, these will be merged. - pub fn with_headers(mut self, mut headers: HeaderMap) -> Self { - match self.headers { - Headers::Empty => { - self.headers = Headers::Extra(headers); - } - Headers::Auth(auth) => { - headers - .entry(http::header::PROXY_AUTHORIZATION) - .or_insert(auth); - self.headers = Headers::Extra(headers); - } - Headers::Extra(ref mut extra) => { - extra.extend(headers); - } - } - - self - } -} - -impl<C> Service<Uri> for Tunnel<C> -where - C: Service<Uri>, - C::Future: Send + 'static, - C::Response: Read + Write + Unpin + Send + 'static, - C::Error: Into<Box<dyn StdError + Send + Sync>>, -{ - type Response = C::Response; - type Error = TunnelError; - type Future = Tunneling<C::Future, C::Response>; - - fn poll_ready(&mut self, cx: &mut task::Context<'_>) -> Poll<Result<(), Self::Error>> { - ready!(self.inner.poll_ready(cx)).map_err(|e| TunnelError::ConnectFailed(e.into()))?; - Poll::Ready(Ok(())) - } - - fn call(&mut self, dst: Uri) -> Self::Future { - let connecting = self.inner.call(self.proxy_dst.clone()); - let headers = self.headers.clone(); - - Tunneling { - fut: Box::pin(async move { - let conn = connecting - .await - .map_err(|e| TunnelError::ConnectFailed(e.into()))?; - tunnel( - conn, - dst.host().ok_or(TunnelError::MissingHost)?, - dst.port().map(|p| p.as_u16()).unwrap_or(443), - &headers, - ) - .await - }), - _marker: PhantomData, - } - } -} - -impl<F, T, E> Future for Tunneling<F, T> -where - F: Future<Output = Result<T, E>>, -{ - type Output = Result<T, TunnelError>; - - fn poll(self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Self::Output> { - self.project().fut.poll(cx) - } -} - -async fn tunnel<T>(mut conn: T, host: &str, port: u16, headers: &Headers) -> Result<T, TunnelError> -where - T: Read + Write + Unpin, -{ - let mut buf = format!( - "\ - CONNECT {host}:{port} HTTP/1.1\r\n\ - Host: {host}:{port}\r\n\ - " - ) - .into_bytes(); - - match headers { - Headers::Auth(auth) => { - buf.extend_from_slice(b"Proxy-Authorization: "); - buf.extend_from_slice(auth.as_bytes()); - buf.extend_from_slice(b"\r\n"); - } - Headers::Extra(extra) => { - for (name, value) in extra { - buf.extend_from_slice(name.as_str().as_bytes()); - buf.extend_from_slice(b": "); - buf.extend_from_slice(value.as_bytes()); - buf.extend_from_slice(b"\r\n"); - } - } - Headers::Empty => (), - } - - // headers end - buf.extend_from_slice(b"\r\n"); - - crate::rt::write_all(&mut conn, &buf) - .await - .map_err(TunnelError::Io)?; - - let mut buf = [0; 8192]; - let mut pos = 0; - - loop { - let n = crate::rt::read(&mut conn, &mut buf[pos..]) - .await - .map_err(TunnelError::Io)?; - - if n == 0 { - return Err(TunnelError::TunnelUnexpectedEof); - } - pos += n; - - let recvd = &buf[..pos]; - if recvd.starts_with(b"HTTP/1.1 200") || recvd.starts_with(b"HTTP/1.0 200") { - if recvd.ends_with(b"\r\n\r\n") { - return Ok(conn); - } - if pos == buf.len() { - return Err(TunnelError::ProxyHeadersTooLong); - } - // else read more - } else if recvd.starts_with(b"HTTP/1.1 407") { - return Err(TunnelError::ProxyAuthRequired); - } else { - return Err(TunnelError::TunnelUnsuccessful); - } - } -} - -impl std::fmt::Display for TunnelError { - fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { - f.write_str("tunnel error: ")?; - - f.write_str(match self { - TunnelError::MissingHost => "missing destination host", - TunnelError::ProxyAuthRequired => "proxy authorization required", - TunnelError::ProxyHeadersTooLong => "proxy response headers too long", - TunnelError::TunnelUnexpectedEof => "unexpected end of file", - TunnelError::TunnelUnsuccessful => "unsuccessful", - TunnelError::ConnectFailed(_) => "failed to create underlying connection", - TunnelError::Io(_) => "io error establishing tunnel", - }) - } -} - -impl std::error::Error for TunnelError { - fn source(&self) -> Option<&(dyn std::error::Error + 'static)> { - match self { - TunnelError::Io(ref e) => Some(e), - TunnelError::ConnectFailed(ref e) => Some(&**e), - _ => None, - } - } -} diff --git a/vendor/hyper-util/src/client/legacy/mod.rs b/vendor/hyper-util/src/client/legacy/mod.rs deleted file mode 100644 index 1649ae7e..00000000 --- a/vendor/hyper-util/src/client/legacy/mod.rs +++ /dev/null @@ -1,10 +0,0 @@ -#[cfg(any(feature = "http1", feature = "http2"))] -mod client; -#[cfg(any(feature = "http1", feature = "http2"))] -pub use client::{Builder, Client, Error, ResponseFuture}; - -pub mod connect; -#[doc(hidden)] -// Publicly available, but just for legacy purposes. A better pool will be -// designed. -pub mod pool; diff --git a/vendor/hyper-util/src/client/legacy/pool.rs b/vendor/hyper-util/src/client/legacy/pool.rs deleted file mode 100644 index 727f54b2..00000000 --- a/vendor/hyper-util/src/client/legacy/pool.rs +++ /dev/null @@ -1,1093 +0,0 @@ -#![allow(dead_code)] - -use std::collections::{HashMap, HashSet, VecDeque}; -use std::convert::Infallible; -use std::error::Error as StdError; -use std::fmt::{self, Debug}; -use std::future::Future; -use std::hash::Hash; -use std::ops::{Deref, DerefMut}; -use std::pin::Pin; -use std::sync::{Arc, Mutex, Weak}; -use std::task::{self, Poll}; - -use std::time::{Duration, Instant}; - -use futures_channel::oneshot; -use futures_core::ready; -use tracing::{debug, trace}; - -use hyper::rt::Sleep; -use hyper::rt::Timer as _; - -use crate::common::{exec, exec::Exec, timer::Timer}; - -// FIXME: allow() required due to `impl Trait` leaking types to this lint -#[allow(missing_debug_implementations)] -pub struct Pool<T, K: Key> { - // If the pool is disabled, this is None. - inner: Option<Arc<Mutex<PoolInner<T, K>>>>, -} - -// Before using a pooled connection, make sure the sender is not dead. -// -// This is a trait to allow the `client::pool::tests` to work for `i32`. -// -// See https://github.com/hyperium/hyper/issues/1429 -pub trait Poolable: Unpin + Send + Sized + 'static { - fn is_open(&self) -> bool; - /// Reserve this connection. - /// - /// Allows for HTTP/2 to return a shared reservation. - fn reserve(self) -> Reservation<Self>; - fn can_share(&self) -> bool; -} - -pub trait Key: Eq + Hash + Clone + Debug + Unpin + Send + 'static {} - -impl<T> Key for T where T: Eq + Hash + Clone + Debug + Unpin + Send + 'static {} - -/// A marker to identify what version a pooled connection is. -#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)] -#[allow(dead_code)] -pub enum Ver { - Auto, - Http2, -} - -/// When checking out a pooled connection, it might be that the connection -/// only supports a single reservation, or it might be usable for many. -/// -/// Specifically, HTTP/1 requires a unique reservation, but HTTP/2 can be -/// used for multiple requests. -// FIXME: allow() required due to `impl Trait` leaking types to this lint -#[allow(missing_debug_implementations)] -pub enum Reservation<T> { - /// This connection could be used multiple times, the first one will be - /// reinserted into the `idle` pool, and the second will be given to - /// the `Checkout`. - #[cfg(feature = "http2")] - Shared(T, T), - /// This connection requires unique access. It will be returned after - /// use is complete. - Unique(T), -} - -/// Simple type alias in case the key type needs to be adjusted. -// pub type Key = (http::uri::Scheme, http::uri::Authority); //Arc<String>; - -struct PoolInner<T, K: Eq + Hash> { - // A flag that a connection is being established, and the connection - // should be shared. This prevents making multiple HTTP/2 connections - // to the same host. - connecting: HashSet<K>, - // These are internal Conns sitting in the event loop in the KeepAlive - // state, waiting to receive a new Request to send on the socket. - idle: HashMap<K, Vec<Idle<T>>>, - max_idle_per_host: usize, - // These are outstanding Checkouts that are waiting for a socket to be - // able to send a Request one. This is used when "racing" for a new - // connection. - // - // The Client starts 2 tasks, 1 to connect a new socket, and 1 to wait - // for the Pool to receive an idle Conn. When a Conn becomes idle, - // this list is checked for any parked Checkouts, and tries to notify - // them that the Conn could be used instead of waiting for a brand new - // connection. - waiters: HashMap<K, VecDeque<oneshot::Sender<T>>>, - // A oneshot channel is used to allow the interval to be notified when - // the Pool completely drops. That way, the interval can cancel immediately. - idle_interval_ref: Option<oneshot::Sender<Infallible>>, - exec: Exec, - timer: Option<Timer>, - timeout: Option<Duration>, -} - -// This is because `Weak::new()` *allocates* space for `T`, even if it -// doesn't need it! -struct WeakOpt<T>(Option<Weak<T>>); - -#[derive(Clone, Copy, Debug)] -pub struct Config { - pub idle_timeout: Option<Duration>, - pub max_idle_per_host: usize, -} - -impl Config { - pub fn is_enabled(&self) -> bool { - self.max_idle_per_host > 0 - } -} - -impl<T, K: Key> Pool<T, K> { - pub fn new<E, M>(config: Config, executor: E, timer: Option<M>) -> Pool<T, K> - where - E: hyper::rt::Executor<exec::BoxSendFuture> + Send + Sync + Clone + 'static, - M: hyper::rt::Timer + Send + Sync + Clone + 'static, - { - let exec = Exec::new(executor); - let timer = timer.map(|t| Timer::new(t)); - let inner = if config.is_enabled() { - Some(Arc::new(Mutex::new(PoolInner { - connecting: HashSet::new(), - idle: HashMap::new(), - idle_interval_ref: None, - max_idle_per_host: config.max_idle_per_host, - waiters: HashMap::new(), - exec, - timer, - timeout: config.idle_timeout, - }))) - } else { - None - }; - - Pool { inner } - } - - pub(crate) fn is_enabled(&self) -> bool { - self.inner.is_some() - } - - #[cfg(test)] - pub(super) fn no_timer(&self) { - // Prevent an actual interval from being created for this pool... - { - let mut inner = self.inner.as_ref().unwrap().lock().unwrap(); - assert!(inner.idle_interval_ref.is_none(), "timer already spawned"); - let (tx, _) = oneshot::channel(); - inner.idle_interval_ref = Some(tx); - } - } -} - -impl<T: Poolable, K: Key> Pool<T, K> { - /// Returns a `Checkout` which is a future that resolves if an idle - /// connection becomes available. - pub fn checkout(&self, key: K) -> Checkout<T, K> { - Checkout { - key, - pool: self.clone(), - waiter: None, - } - } - - /// Ensure that there is only ever 1 connecting task for HTTP/2 - /// connections. This does nothing for HTTP/1. - pub fn connecting(&self, key: &K, ver: Ver) -> Option<Connecting<T, K>> { - if ver == Ver::Http2 { - if let Some(ref enabled) = self.inner { - let mut inner = enabled.lock().unwrap(); - return if inner.connecting.insert(key.clone()) { - let connecting = Connecting { - key: key.clone(), - pool: WeakOpt::downgrade(enabled), - }; - Some(connecting) - } else { - trace!("HTTP/2 connecting already in progress for {:?}", key); - None - }; - } - } - - // else - Some(Connecting { - key: key.clone(), - // in HTTP/1's case, there is never a lock, so we don't - // need to do anything in Drop. - pool: WeakOpt::none(), - }) - } - - #[cfg(test)] - fn locked(&self) -> std::sync::MutexGuard<'_, PoolInner<T, K>> { - self.inner.as_ref().expect("enabled").lock().expect("lock") - } - - /* Used in client/tests.rs... - #[cfg(test)] - pub(super) fn h1_key(&self, s: &str) -> Key { - Arc::new(s.to_string()) - } - - #[cfg(test)] - pub(super) fn idle_count(&self, key: &Key) -> usize { - self - .locked() - .idle - .get(key) - .map(|list| list.len()) - .unwrap_or(0) - } - */ - - pub fn pooled( - &self, - #[cfg_attr(not(feature = "http2"), allow(unused_mut))] mut connecting: Connecting<T, K>, - value: T, - ) -> Pooled<T, K> { - let (value, pool_ref) = if let Some(ref enabled) = self.inner { - match value.reserve() { - #[cfg(feature = "http2")] - Reservation::Shared(to_insert, to_return) => { - let mut inner = enabled.lock().unwrap(); - inner.put(connecting.key.clone(), to_insert, enabled); - // Do this here instead of Drop for Connecting because we - // already have a lock, no need to lock the mutex twice. - inner.connected(&connecting.key); - // prevent the Drop of Connecting from repeating inner.connected() - connecting.pool = WeakOpt::none(); - - // Shared reservations don't need a reference to the pool, - // since the pool always keeps a copy. - (to_return, WeakOpt::none()) - } - Reservation::Unique(value) => { - // Unique reservations must take a reference to the pool - // since they hope to reinsert once the reservation is - // completed - (value, WeakOpt::downgrade(enabled)) - } - } - } else { - // If pool is not enabled, skip all the things... - - // The Connecting should have had no pool ref - debug_assert!(connecting.pool.upgrade().is_none()); - - (value, WeakOpt::none()) - }; - Pooled { - key: connecting.key.clone(), - is_reused: false, - pool: pool_ref, - value: Some(value), - } - } - - fn reuse(&self, key: &K, value: T) -> Pooled<T, K> { - debug!("reuse idle connection for {:?}", key); - // TODO: unhack this - // In Pool::pooled(), which is used for inserting brand new connections, - // there's some code that adjusts the pool reference taken depending - // on if the Reservation can be shared or is unique. By the time - // reuse() is called, the reservation has already been made, and - // we just have the final value, without knowledge of if this is - // unique or shared. So, the hack is to just assume Ver::Http2 means - // shared... :( - let mut pool_ref = WeakOpt::none(); - if !value.can_share() { - if let Some(ref enabled) = self.inner { - pool_ref = WeakOpt::downgrade(enabled); - } - } - - Pooled { - is_reused: true, - key: key.clone(), - pool: pool_ref, - value: Some(value), - } - } -} - -/// Pop off this list, looking for a usable connection that hasn't expired. -struct IdlePopper<'a, T, K> { - key: &'a K, - list: &'a mut Vec<Idle<T>>, -} - -impl<'a, T: Poolable + 'a, K: Debug> IdlePopper<'a, T, K> { - fn pop(self, expiration: &Expiration) -> Option<Idle<T>> { - while let Some(entry) = self.list.pop() { - // If the connection has been closed, or is older than our idle - // timeout, simply drop it and keep looking... - if !entry.value.is_open() { - trace!("removing closed connection for {:?}", self.key); - continue; - } - // TODO: Actually, since the `idle` list is pushed to the end always, - // that would imply that if *this* entry is expired, then anything - // "earlier" in the list would *have* to be expired also... Right? - // - // In that case, we could just break out of the loop and drop the - // whole list... - if expiration.expires(entry.idle_at) { - trace!("removing expired connection for {:?}", self.key); - continue; - } - - let value = match entry.value.reserve() { - #[cfg(feature = "http2")] - Reservation::Shared(to_reinsert, to_checkout) => { - self.list.push(Idle { - idle_at: Instant::now(), - value: to_reinsert, - }); - to_checkout - } - Reservation::Unique(unique) => unique, - }; - - return Some(Idle { - idle_at: entry.idle_at, - value, - }); - } - - None - } -} - -impl<T: Poolable, K: Key> PoolInner<T, K> { - fn put(&mut self, key: K, value: T, __pool_ref: &Arc<Mutex<PoolInner<T, K>>>) { - if value.can_share() && self.idle.contains_key(&key) { - trace!("put; existing idle HTTP/2 connection for {:?}", key); - return; - } - trace!("put; add idle connection for {:?}", key); - let mut remove_waiters = false; - let mut value = Some(value); - if let Some(waiters) = self.waiters.get_mut(&key) { - while let Some(tx) = waiters.pop_front() { - if !tx.is_canceled() { - let reserved = value.take().expect("value already sent"); - let reserved = match reserved.reserve() { - #[cfg(feature = "http2")] - Reservation::Shared(to_keep, to_send) => { - value = Some(to_keep); - to_send - } - Reservation::Unique(uniq) => uniq, - }; - match tx.send(reserved) { - Ok(()) => { - if value.is_none() { - break; - } else { - continue; - } - } - Err(e) => { - value = Some(e); - } - } - } - - trace!("put; removing canceled waiter for {:?}", key); - } - remove_waiters = waiters.is_empty(); - } - if remove_waiters { - self.waiters.remove(&key); - } - - match value { - Some(value) => { - // borrow-check scope... - { - let idle_list = self.idle.entry(key.clone()).or_default(); - if self.max_idle_per_host <= idle_list.len() { - trace!("max idle per host for {:?}, dropping connection", key); - return; - } - - debug!("pooling idle connection for {:?}", key); - idle_list.push(Idle { - value, - idle_at: Instant::now(), - }); - } - - self.spawn_idle_interval(__pool_ref); - } - None => trace!("put; found waiter for {:?}", key), - } - } - - /// A `Connecting` task is complete. Not necessarily successfully, - /// but the lock is going away, so clean up. - fn connected(&mut self, key: &K) { - let existed = self.connecting.remove(key); - debug_assert!(existed, "Connecting dropped, key not in pool.connecting"); - // cancel any waiters. if there are any, it's because - // this Connecting task didn't complete successfully. - // those waiters would never receive a connection. - self.waiters.remove(key); - } - - fn spawn_idle_interval(&mut self, pool_ref: &Arc<Mutex<PoolInner<T, K>>>) { - if self.idle_interval_ref.is_some() { - return; - } - let dur = if let Some(dur) = self.timeout { - dur - } else { - return; - }; - let timer = if let Some(timer) = self.timer.clone() { - timer - } else { - return; - }; - let (tx, rx) = oneshot::channel(); - self.idle_interval_ref = Some(tx); - - let interval = IdleTask { - timer: timer.clone(), - duration: dur, - deadline: Instant::now(), - fut: timer.sleep_until(Instant::now()), // ready at first tick - pool: WeakOpt::downgrade(pool_ref), - pool_drop_notifier: rx, - }; - - self.exec.execute(interval); - } -} - -impl<T, K: Eq + Hash> PoolInner<T, K> { - /// Any `FutureResponse`s that were created will have made a `Checkout`, - /// and possibly inserted into the pool that it is waiting for an idle - /// connection. If a user ever dropped that future, we need to clean out - /// those parked senders. - fn clean_waiters(&mut self, key: &K) { - let mut remove_waiters = false; - if let Some(waiters) = self.waiters.get_mut(key) { - waiters.retain(|tx| !tx.is_canceled()); - remove_waiters = waiters.is_empty(); - } - if remove_waiters { - self.waiters.remove(key); - } - } -} - -impl<T: Poolable, K: Key> PoolInner<T, K> { - /// This should *only* be called by the IdleTask - fn clear_expired(&mut self) { - let dur = self.timeout.expect("interval assumes timeout"); - - let now = Instant::now(); - //self.last_idle_check_at = now; - - self.idle.retain(|key, values| { - values.retain(|entry| { - if !entry.value.is_open() { - trace!("idle interval evicting closed for {:?}", key); - return false; - } - - // Avoid `Instant::sub` to avoid issues like rust-lang/rust#86470. - if now.saturating_duration_since(entry.idle_at) > dur { - trace!("idle interval evicting expired for {:?}", key); - return false; - } - - // Otherwise, keep this value... - true - }); - - // returning false evicts this key/val - !values.is_empty() - }); - } -} - -impl<T, K: Key> Clone for Pool<T, K> { - fn clone(&self) -> Pool<T, K> { - Pool { - inner: self.inner.clone(), - } - } -} - -/// A wrapped poolable value that tries to reinsert to the Pool on Drop. -// Note: The bounds `T: Poolable` is needed for the Drop impl. -pub struct Pooled<T: Poolable, K: Key> { - value: Option<T>, - is_reused: bool, - key: K, - pool: WeakOpt<Mutex<PoolInner<T, K>>>, -} - -impl<T: Poolable, K: Key> Pooled<T, K> { - pub fn is_reused(&self) -> bool { - self.is_reused - } - - pub fn is_pool_enabled(&self) -> bool { - self.pool.0.is_some() - } - - fn as_ref(&self) -> &T { - self.value.as_ref().expect("not dropped") - } - - fn as_mut(&mut self) -> &mut T { - self.value.as_mut().expect("not dropped") - } -} - -impl<T: Poolable, K: Key> Deref for Pooled<T, K> { - type Target = T; - fn deref(&self) -> &T { - self.as_ref() - } -} - -impl<T: Poolable, K: Key> DerefMut for Pooled<T, K> { - fn deref_mut(&mut self) -> &mut T { - self.as_mut() - } -} - -impl<T: Poolable, K: Key> Drop for Pooled<T, K> { - fn drop(&mut self) { - if let Some(value) = self.value.take() { - if !value.is_open() { - // If we *already* know the connection is done here, - // it shouldn't be re-inserted back into the pool. - return; - } - - if let Some(pool) = self.pool.upgrade() { - if let Ok(mut inner) = pool.lock() { - inner.put(self.key.clone(), value, &pool); - } - } else if !value.can_share() { - trace!("pool dropped, dropping pooled ({:?})", self.key); - } - // Ver::Http2 is already in the Pool (or dead), so we wouldn't - // have an actual reference to the Pool. - } - } -} - -impl<T: Poolable, K: Key> fmt::Debug for Pooled<T, K> { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.debug_struct("Pooled").field("key", &self.key).finish() - } -} - -struct Idle<T> { - idle_at: Instant, - value: T, -} - -// FIXME: allow() required due to `impl Trait` leaking types to this lint -#[allow(missing_debug_implementations)] -pub struct Checkout<T, K: Key> { - key: K, - pool: Pool<T, K>, - waiter: Option<oneshot::Receiver<T>>, -} - -#[derive(Debug)] -#[non_exhaustive] -pub enum Error { - PoolDisabled, - CheckoutNoLongerWanted, - CheckedOutClosedValue, -} - -impl Error { - pub(super) fn is_canceled(&self) -> bool { - matches!(self, Error::CheckedOutClosedValue) - } -} - -impl fmt::Display for Error { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.write_str(match self { - Error::PoolDisabled => "pool is disabled", - Error::CheckedOutClosedValue => "checked out connection was closed", - Error::CheckoutNoLongerWanted => "request was canceled", - }) - } -} - -impl StdError for Error {} - -impl<T: Poolable, K: Key> Checkout<T, K> { - fn poll_waiter( - &mut self, - cx: &mut task::Context<'_>, - ) -> Poll<Option<Result<Pooled<T, K>, Error>>> { - if let Some(mut rx) = self.waiter.take() { - match Pin::new(&mut rx).poll(cx) { - Poll::Ready(Ok(value)) => { - if value.is_open() { - Poll::Ready(Some(Ok(self.pool.reuse(&self.key, value)))) - } else { - Poll::Ready(Some(Err(Error::CheckedOutClosedValue))) - } - } - Poll::Pending => { - self.waiter = Some(rx); - Poll::Pending - } - Poll::Ready(Err(_canceled)) => { - Poll::Ready(Some(Err(Error::CheckoutNoLongerWanted))) - } - } - } else { - Poll::Ready(None) - } - } - - fn checkout(&mut self, cx: &mut task::Context<'_>) -> Option<Pooled<T, K>> { - let entry = { - let mut inner = self.pool.inner.as_ref()?.lock().unwrap(); - let expiration = Expiration::new(inner.timeout); - let maybe_entry = inner.idle.get_mut(&self.key).and_then(|list| { - trace!("take? {:?}: expiration = {:?}", self.key, expiration.0); - // A block to end the mutable borrow on list, - // so the map below can check is_empty() - { - let popper = IdlePopper { - key: &self.key, - list, - }; - popper.pop(&expiration) - } - .map(|e| (e, list.is_empty())) - }); - - let (entry, empty) = if let Some((e, empty)) = maybe_entry { - (Some(e), empty) - } else { - // No entry found means nuke the list for sure. - (None, true) - }; - if empty { - //TODO: This could be done with the HashMap::entry API instead. - inner.idle.remove(&self.key); - } - - if entry.is_none() && self.waiter.is_none() { - let (tx, mut rx) = oneshot::channel(); - trace!("checkout waiting for idle connection: {:?}", self.key); - inner - .waiters - .entry(self.key.clone()) - .or_insert_with(VecDeque::new) - .push_back(tx); - - // register the waker with this oneshot - assert!(Pin::new(&mut rx).poll(cx).is_pending()); - self.waiter = Some(rx); - } - - entry - }; - - entry.map(|e| self.pool.reuse(&self.key, e.value)) - } -} - -impl<T: Poolable, K: Key> Future for Checkout<T, K> { - type Output = Result<Pooled<T, K>, Error>; - - fn poll(mut self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Self::Output> { - if let Some(pooled) = ready!(self.poll_waiter(cx)?) { - return Poll::Ready(Ok(pooled)); - } - - if let Some(pooled) = self.checkout(cx) { - Poll::Ready(Ok(pooled)) - } else if !self.pool.is_enabled() { - Poll::Ready(Err(Error::PoolDisabled)) - } else { - // There's a new waiter, already registered in self.checkout() - debug_assert!(self.waiter.is_some()); - Poll::Pending - } - } -} - -impl<T, K: Key> Drop for Checkout<T, K> { - fn drop(&mut self) { - if self.waiter.take().is_some() { - trace!("checkout dropped for {:?}", self.key); - if let Some(Ok(mut inner)) = self.pool.inner.as_ref().map(|i| i.lock()) { - inner.clean_waiters(&self.key); - } - } - } -} - -// FIXME: allow() required due to `impl Trait` leaking types to this lint -#[allow(missing_debug_implementations)] -pub struct Connecting<T: Poolable, K: Key> { - key: K, - pool: WeakOpt<Mutex<PoolInner<T, K>>>, -} - -impl<T: Poolable, K: Key> Connecting<T, K> { - pub fn alpn_h2(self, pool: &Pool<T, K>) -> Option<Self> { - debug_assert!( - self.pool.0.is_none(), - "Connecting::alpn_h2 but already Http2" - ); - - pool.connecting(&self.key, Ver::Http2) - } -} - -impl<T: Poolable, K: Key> Drop for Connecting<T, K> { - fn drop(&mut self) { - if let Some(pool) = self.pool.upgrade() { - // No need to panic on drop, that could abort! - if let Ok(mut inner) = pool.lock() { - inner.connected(&self.key); - } - } - } -} - -struct Expiration(Option<Duration>); - -impl Expiration { - fn new(dur: Option<Duration>) -> Expiration { - Expiration(dur) - } - - fn expires(&self, instant: Instant) -> bool { - match self.0 { - // Avoid `Instant::elapsed` to avoid issues like rust-lang/rust#86470. - Some(timeout) => Instant::now().saturating_duration_since(instant) > timeout, - None => false, - } - } -} - -pin_project_lite::pin_project! { - struct IdleTask<T, K: Key> { - timer: Timer, - duration: Duration, - deadline: Instant, - fut: Pin<Box<dyn Sleep>>, - pool: WeakOpt<Mutex<PoolInner<T, K>>>, - // This allows the IdleTask to be notified as soon as the entire - // Pool is fully dropped, and shutdown. This channel is never sent on, - // but Err(Canceled) will be received when the Pool is dropped. - #[pin] - pool_drop_notifier: oneshot::Receiver<Infallible>, - } -} - -impl<T: Poolable + 'static, K: Key> Future for IdleTask<T, K> { - type Output = (); - - fn poll(self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Self::Output> { - let mut this = self.project(); - loop { - match this.pool_drop_notifier.as_mut().poll(cx) { - Poll::Ready(Ok(n)) => match n {}, - Poll::Pending => (), - Poll::Ready(Err(_canceled)) => { - trace!("pool closed, canceling idle interval"); - return Poll::Ready(()); - } - } - - ready!(Pin::new(&mut this.fut).poll(cx)); - // Set this task to run after the next deadline - // If the poll missed the deadline by a lot, set the deadline - // from the current time instead - *this.deadline += *this.duration; - if *this.deadline < Instant::now() - Duration::from_millis(5) { - *this.deadline = Instant::now() + *this.duration; - } - *this.fut = this.timer.sleep_until(*this.deadline); - - if let Some(inner) = this.pool.upgrade() { - if let Ok(mut inner) = inner.lock() { - trace!("idle interval checking for expired"); - inner.clear_expired(); - continue; - } - } - return Poll::Ready(()); - } - } -} - -impl<T> WeakOpt<T> { - fn none() -> Self { - WeakOpt(None) - } - - fn downgrade(arc: &Arc<T>) -> Self { - WeakOpt(Some(Arc::downgrade(arc))) - } - - fn upgrade(&self) -> Option<Arc<T>> { - self.0.as_ref().and_then(Weak::upgrade) - } -} - -#[cfg(all(test, not(miri)))] -mod tests { - use std::fmt::Debug; - use std::future::Future; - use std::hash::Hash; - use std::pin::Pin; - use std::task::{self, Poll}; - use std::time::Duration; - - use super::{Connecting, Key, Pool, Poolable, Reservation, WeakOpt}; - use crate::rt::{TokioExecutor, TokioTimer}; - - use crate::common::timer; - - #[derive(Clone, Debug, PartialEq, Eq, Hash)] - struct KeyImpl(http::uri::Scheme, http::uri::Authority); - - type KeyTuple = (http::uri::Scheme, http::uri::Authority); - - /// Test unique reservations. - #[derive(Debug, PartialEq, Eq)] - struct Uniq<T>(T); - - impl<T: Send + 'static + Unpin> Poolable for Uniq<T> { - fn is_open(&self) -> bool { - true - } - - fn reserve(self) -> Reservation<Self> { - Reservation::Unique(self) - } - - fn can_share(&self) -> bool { - false - } - } - - fn c<T: Poolable, K: Key>(key: K) -> Connecting<T, K> { - Connecting { - key, - pool: WeakOpt::none(), - } - } - - fn host_key(s: &str) -> KeyImpl { - KeyImpl(http::uri::Scheme::HTTP, s.parse().expect("host key")) - } - - fn pool_no_timer<T, K: Key>() -> Pool<T, K> { - pool_max_idle_no_timer(usize::MAX) - } - - fn pool_max_idle_no_timer<T, K: Key>(max_idle: usize) -> Pool<T, K> { - let pool = Pool::new( - super::Config { - idle_timeout: Some(Duration::from_millis(100)), - max_idle_per_host: max_idle, - }, - TokioExecutor::new(), - Option::<timer::Timer>::None, - ); - pool.no_timer(); - pool - } - - #[tokio::test] - async fn test_pool_checkout_smoke() { - let pool = pool_no_timer(); - let key = host_key("foo"); - let pooled = pool.pooled(c(key.clone()), Uniq(41)); - - drop(pooled); - - match pool.checkout(key).await { - Ok(pooled) => assert_eq!(*pooled, Uniq(41)), - Err(_) => panic!("not ready"), - }; - } - - /// Helper to check if the future is ready after polling once. - struct PollOnce<'a, F>(&'a mut F); - - impl<F, T, U> Future for PollOnce<'_, F> - where - F: Future<Output = Result<T, U>> + Unpin, - { - type Output = Option<()>; - - fn poll(mut self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Self::Output> { - match Pin::new(&mut self.0).poll(cx) { - Poll::Ready(Ok(_)) => Poll::Ready(Some(())), - Poll::Ready(Err(_)) => Poll::Ready(Some(())), - Poll::Pending => Poll::Ready(None), - } - } - } - - #[tokio::test] - async fn test_pool_checkout_returns_none_if_expired() { - let pool = pool_no_timer(); - let key = host_key("foo"); - let pooled = pool.pooled(c(key.clone()), Uniq(41)); - - drop(pooled); - tokio::time::sleep(pool.locked().timeout.unwrap()).await; - let mut checkout = pool.checkout(key); - let poll_once = PollOnce(&mut checkout); - let is_not_ready = poll_once.await.is_none(); - assert!(is_not_ready); - } - - #[tokio::test] - async fn test_pool_checkout_removes_expired() { - let pool = pool_no_timer(); - let key = host_key("foo"); - - pool.pooled(c(key.clone()), Uniq(41)); - pool.pooled(c(key.clone()), Uniq(5)); - pool.pooled(c(key.clone()), Uniq(99)); - - assert_eq!( - pool.locked().idle.get(&key).map(|entries| entries.len()), - Some(3) - ); - tokio::time::sleep(pool.locked().timeout.unwrap()).await; - - let mut checkout = pool.checkout(key.clone()); - let poll_once = PollOnce(&mut checkout); - // checkout.await should clean out the expired - poll_once.await; - assert!(!pool.locked().idle.contains_key(&key)); - } - - #[test] - fn test_pool_max_idle_per_host() { - let pool = pool_max_idle_no_timer(2); - let key = host_key("foo"); - - pool.pooled(c(key.clone()), Uniq(41)); - pool.pooled(c(key.clone()), Uniq(5)); - pool.pooled(c(key.clone()), Uniq(99)); - - // pooled and dropped 3, max_idle should only allow 2 - assert_eq!( - pool.locked().idle.get(&key).map(|entries| entries.len()), - Some(2) - ); - } - - #[tokio::test] - async fn test_pool_timer_removes_expired() { - let pool = Pool::new( - super::Config { - idle_timeout: Some(Duration::from_millis(10)), - max_idle_per_host: usize::MAX, - }, - TokioExecutor::new(), - Some(TokioTimer::new()), - ); - - let key = host_key("foo"); - - pool.pooled(c(key.clone()), Uniq(41)); - pool.pooled(c(key.clone()), Uniq(5)); - pool.pooled(c(key.clone()), Uniq(99)); - - assert_eq!( - pool.locked().idle.get(&key).map(|entries| entries.len()), - Some(3) - ); - - // Let the timer tick passed the expiration... - tokio::time::sleep(Duration::from_millis(30)).await; - // Yield so the Interval can reap... - tokio::task::yield_now().await; - - assert!(!pool.locked().idle.contains_key(&key)); - } - - #[tokio::test] - async fn test_pool_checkout_task_unparked() { - use futures_util::future::join; - use futures_util::FutureExt; - - let pool = pool_no_timer(); - let key = host_key("foo"); - let pooled = pool.pooled(c(key.clone()), Uniq(41)); - - let checkout = join(pool.checkout(key), async { - // the checkout future will park first, - // and then this lazy future will be polled, which will insert - // the pooled back into the pool - // - // this test makes sure that doing so will unpark the checkout - drop(pooled); - }) - .map(|(entry, _)| entry); - - assert_eq!(*checkout.await.unwrap(), Uniq(41)); - } - - #[tokio::test] - async fn test_pool_checkout_drop_cleans_up_waiters() { - let pool = pool_no_timer::<Uniq<i32>, KeyImpl>(); - let key = host_key("foo"); - - let mut checkout1 = pool.checkout(key.clone()); - let mut checkout2 = pool.checkout(key.clone()); - - let poll_once1 = PollOnce(&mut checkout1); - let poll_once2 = PollOnce(&mut checkout2); - - // first poll needed to get into Pool's parked - poll_once1.await; - assert_eq!(pool.locked().waiters.get(&key).unwrap().len(), 1); - poll_once2.await; - assert_eq!(pool.locked().waiters.get(&key).unwrap().len(), 2); - - // on drop, clean up Pool - drop(checkout1); - assert_eq!(pool.locked().waiters.get(&key).unwrap().len(), 1); - - drop(checkout2); - assert!(!pool.locked().waiters.contains_key(&key)); - } - - #[derive(Debug)] - struct CanClose { - #[allow(unused)] - val: i32, - closed: bool, - } - - impl Poolable for CanClose { - fn is_open(&self) -> bool { - !self.closed - } - - fn reserve(self) -> Reservation<Self> { - Reservation::Unique(self) - } - - fn can_share(&self) -> bool { - false - } - } - - #[test] - fn pooled_drop_if_closed_doesnt_reinsert() { - let pool = pool_no_timer(); - let key = host_key("foo"); - pool.pooled( - c(key.clone()), - CanClose { - val: 57, - closed: true, - }, - ); - - assert!(!pool.locked().idle.contains_key(&key)); - } -} diff --git a/vendor/hyper-util/src/client/mod.rs b/vendor/hyper-util/src/client/mod.rs deleted file mode 100644 index 0d896030..00000000 --- a/vendor/hyper-util/src/client/mod.rs +++ /dev/null @@ -1,8 +0,0 @@ -//! HTTP client utilities - -/// Legacy implementations of `connect` module and `Client` -#[cfg(feature = "client-legacy")] -pub mod legacy; - -#[cfg(feature = "client-proxy")] -pub mod proxy; diff --git a/vendor/hyper-util/src/client/proxy/matcher.rs b/vendor/hyper-util/src/client/proxy/matcher.rs deleted file mode 100644 index fd563bca..00000000 --- a/vendor/hyper-util/src/client/proxy/matcher.rs +++ /dev/null @@ -1,848 +0,0 @@ -//! Proxy matchers -//! -//! This module contains different matchers to configure rules for when a proxy -//! should be used, and if so, with what arguments. -//! -//! A [`Matcher`] can be constructed either using environment variables, or -//! a [`Matcher::builder()`]. -//! -//! Once constructed, the `Matcher` can be asked if it intercepts a `Uri` by -//! calling [`Matcher::intercept()`]. -//! -//! An [`Intercept`] includes the destination for the proxy, and any parsed -//! authentication to be used. - -use std::fmt; -use std::net::IpAddr; - -use http::header::HeaderValue; -use ipnet::IpNet; -use percent_encoding::percent_decode_str; - -#[cfg(docsrs)] -pub use self::builder::IntoValue; -#[cfg(not(docsrs))] -use self::builder::IntoValue; - -/// A proxy matcher, usually built from environment variables. -pub struct Matcher { - http: Option<Intercept>, - https: Option<Intercept>, - no: NoProxy, -} - -/// A matched proxy, -/// -/// This is returned by a matcher if a proxy should be used. -#[derive(Clone)] -pub struct Intercept { - uri: http::Uri, - auth: Auth, -} - -/// A builder to create a [`Matcher`]. -/// -/// Construct with [`Matcher::builder()`]. -#[derive(Default)] -pub struct Builder { - is_cgi: bool, - all: String, - http: String, - https: String, - no: String, -} - -#[derive(Clone)] -enum Auth { - Empty, - Basic(http::header::HeaderValue), - Raw(String, String), -} - -/// A filter for proxy matchers. -/// -/// This type is based off the `NO_PROXY` rules used by curl. -#[derive(Clone, Debug, Default)] -struct NoProxy { - ips: IpMatcher, - domains: DomainMatcher, -} - -#[derive(Clone, Debug, Default)] -struct DomainMatcher(Vec<String>); - -#[derive(Clone, Debug, Default)] -struct IpMatcher(Vec<Ip>); - -#[derive(Clone, Debug)] -enum Ip { - Address(IpAddr), - Network(IpNet), -} - -// ===== impl Matcher ===== - -impl Matcher { - /// Create a matcher reading the current environment variables. - /// - /// This checks for values in the following variables, treating them the - /// same as curl does: - /// - /// - `ALL_PROXY`/`all_proxy` - /// - `HTTPS_PROXY`/`https_proxy` - /// - `HTTP_PROXY`/`http_proxy` - /// - `NO_PROXY`/`no_proxy` - pub fn from_env() -> Self { - Builder::from_env().build() - } - - /// Create a matcher from the environment or system. - /// - /// This checks the same environment variables as `from_env()`, and if not - /// set, checks the system configuration for values for the OS. - /// - /// This constructor is always available, but if the `client-proxy-system` - /// feature is enabled, it will check more configuration. Use this - /// constructor if you want to allow users to optionally enable more, or - /// use `from_env` if you do not want the values to change based on an - /// enabled feature. - pub fn from_system() -> Self { - Builder::from_system().build() - } - - /// Start a builder to configure a matcher. - pub fn builder() -> Builder { - Builder::default() - } - - /// Check if the destination should be intercepted by a proxy. - /// - /// If the proxy rules match the destination, a new `Uri` will be returned - /// to connect to. - pub fn intercept(&self, dst: &http::Uri) -> Option<Intercept> { - // TODO(perf): don't need to check `no` if below doesn't match... - if self.no.contains(dst.host()?) { - return None; - } - - match dst.scheme_str() { - Some("http") => self.http.clone(), - Some("https") => self.https.clone(), - _ => None, - } - } -} - -impl fmt::Debug for Matcher { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - let mut b = f.debug_struct("Matcher"); - - if let Some(ref http) = self.http { - b.field("http", http); - } - - if let Some(ref https) = self.https { - b.field("https", https); - } - - if !self.no.is_empty() { - b.field("no", &self.no); - } - b.finish() - } -} - -// ===== impl Intercept ===== - -impl Intercept { - /// Get the `http::Uri` for the target proxy. - pub fn uri(&self) -> &http::Uri { - &self.uri - } - - /// Get any configured basic authorization. - /// - /// This should usually be used with a `Proxy-Authorization` header, to - /// send in Basic format. - /// - /// # Example - /// - /// ```rust - /// # use hyper_util::client::proxy::matcher::Matcher; - /// # let uri = http::Uri::from_static("https://hyper.rs"); - /// let m = Matcher::builder() - /// .all("https://Aladdin:opensesame@localhost:8887") - /// .build(); - /// - /// let proxy = m.intercept(&uri).expect("example"); - /// let auth = proxy.basic_auth().expect("example"); - /// assert_eq!(auth, "Basic QWxhZGRpbjpvcGVuc2VzYW1l"); - /// ``` - pub fn basic_auth(&self) -> Option<&HeaderValue> { - if let Auth::Basic(ref val) = self.auth { - Some(val) - } else { - None - } - } - - /// Get any configured raw authorization. - /// - /// If not detected as another scheme, this is the username and password - /// that should be sent with whatever protocol the proxy handshake uses. - /// - /// # Example - /// - /// ```rust - /// # use hyper_util::client::proxy::matcher::Matcher; - /// # let uri = http::Uri::from_static("https://hyper.rs"); - /// let m = Matcher::builder() - /// .all("socks5h://Aladdin:opensesame@localhost:8887") - /// .build(); - /// - /// let proxy = m.intercept(&uri).expect("example"); - /// let auth = proxy.raw_auth().expect("example"); - /// assert_eq!(auth, ("Aladdin", "opensesame")); - /// ``` - pub fn raw_auth(&self) -> Option<(&str, &str)> { - if let Auth::Raw(ref u, ref p) = self.auth { - Some((u.as_str(), p.as_str())) - } else { - None - } - } -} - -impl fmt::Debug for Intercept { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.debug_struct("Intercept") - .field("uri", &self.uri) - // dont output auth, its sensitive - .finish() - } -} - -// ===== impl Builder ===== - -impl Builder { - fn from_env() -> Self { - Builder { - is_cgi: std::env::var_os("REQUEST_METHOD").is_some(), - all: get_first_env(&["ALL_PROXY", "all_proxy"]), - http: get_first_env(&["HTTP_PROXY", "http_proxy"]), - https: get_first_env(&["HTTPS_PROXY", "https_proxy"]), - no: get_first_env(&["NO_PROXY", "no_proxy"]), - } - } - - fn from_system() -> Self { - #[allow(unused_mut)] - let mut builder = Self::from_env(); - - #[cfg(all(feature = "client-proxy-system", target_os = "macos"))] - mac::with_system(&mut builder); - - #[cfg(all(feature = "client-proxy-system", windows))] - win::with_system(&mut builder); - - builder - } - - /// Set the target proxy for all destinations. - pub fn all<S>(mut self, val: S) -> Self - where - S: IntoValue, - { - self.all = val.into_value(); - self - } - - /// Set the target proxy for HTTP destinations. - pub fn http<S>(mut self, val: S) -> Self - where - S: IntoValue, - { - self.http = val.into_value(); - self - } - - /// Set the target proxy for HTTPS destinations. - pub fn https<S>(mut self, val: S) -> Self - where - S: IntoValue, - { - self.https = val.into_value(); - self - } - - /// Set the "no" proxy filter. - /// - /// The rules are as follows: - /// * Entries are expected to be comma-separated (whitespace between entries is ignored) - /// * IP addresses (both IPv4 and IPv6) are allowed, as are optional subnet masks (by adding /size, - /// for example "`192.168.1.0/24`"). - /// * An entry "`*`" matches all hostnames (this is the only wildcard allowed) - /// * Any other entry is considered a domain name (and may contain a leading dot, for example `google.com` - /// and `.google.com` are equivalent) and would match both that domain AND all subdomains. - /// - /// For example, if `"NO_PROXY=google.com, 192.168.1.0/24"` was set, all of the following would match - /// (and therefore would bypass the proxy): - /// * `http://google.com/` - /// * `http://www.google.com/` - /// * `http://192.168.1.42/` - /// - /// The URL `http://notgoogle.com/` would not match. - pub fn no<S>(mut self, val: S) -> Self - where - S: IntoValue, - { - self.no = val.into_value(); - self - } - - /// Construct a [`Matcher`] using the configured values. - pub fn build(self) -> Matcher { - if self.is_cgi { - return Matcher { - http: None, - https: None, - no: NoProxy::empty(), - }; - } - - let all = parse_env_uri(&self.all); - - Matcher { - http: parse_env_uri(&self.http).or_else(|| all.clone()), - https: parse_env_uri(&self.https).or(all), - no: NoProxy::from_string(&self.no), - } - } -} - -fn get_first_env(names: &[&str]) -> String { - for name in names { - if let Ok(val) = std::env::var(name) { - return val; - } - } - - String::new() -} - -fn parse_env_uri(val: &str) -> Option<Intercept> { - let uri = val.parse::<http::Uri>().ok()?; - let mut builder = http::Uri::builder(); - let mut is_httpish = false; - let mut auth = Auth::Empty; - - builder = builder.scheme(match uri.scheme() { - Some(s) => { - if s == &http::uri::Scheme::HTTP || s == &http::uri::Scheme::HTTPS { - is_httpish = true; - s.clone() - } else if s.as_str() == "socks5" || s.as_str() == "socks5h" { - s.clone() - } else { - // can't use this proxy scheme - return None; - } - } - // if no scheme provided, assume they meant 'http' - None => { - is_httpish = true; - http::uri::Scheme::HTTP - } - }); - - let authority = uri.authority()?; - - if let Some((userinfo, host_port)) = authority.as_str().split_once('@') { - let (user, pass) = userinfo.split_once(':')?; - let user = percent_decode_str(user).decode_utf8_lossy(); - let pass = percent_decode_str(pass).decode_utf8_lossy(); - if is_httpish { - auth = Auth::Basic(encode_basic_auth(&user, Some(&pass))); - } else { - auth = Auth::Raw(user.into(), pass.into()); - } - builder = builder.authority(host_port); - } else { - builder = builder.authority(authority.clone()); - } - - // removing any path, but we MUST specify one or the builder errors - builder = builder.path_and_query("/"); - - let dst = builder.build().ok()?; - - Some(Intercept { uri: dst, auth }) -} - -fn encode_basic_auth(user: &str, pass: Option<&str>) -> HeaderValue { - use base64::prelude::BASE64_STANDARD; - use base64::write::EncoderWriter; - use std::io::Write; - - let mut buf = b"Basic ".to_vec(); - { - let mut encoder = EncoderWriter::new(&mut buf, &BASE64_STANDARD); - let _ = write!(encoder, "{user}:"); - if let Some(password) = pass { - let _ = write!(encoder, "{password}"); - } - } - let mut header = HeaderValue::from_bytes(&buf).expect("base64 is always valid HeaderValue"); - header.set_sensitive(true); - header -} - -impl NoProxy { - /* - fn from_env() -> NoProxy { - let raw = std::env::var("NO_PROXY") - .or_else(|_| std::env::var("no_proxy")) - .unwrap_or_default(); - - Self::from_string(&raw) - } - */ - - fn empty() -> NoProxy { - NoProxy { - ips: IpMatcher(Vec::new()), - domains: DomainMatcher(Vec::new()), - } - } - - /// Returns a new no-proxy configuration based on a `no_proxy` string (or `None` if no variables - /// are set) - /// The rules are as follows: - /// * The environment variable `NO_PROXY` is checked, if it is not set, `no_proxy` is checked - /// * If neither environment variable is set, `None` is returned - /// * Entries are expected to be comma-separated (whitespace between entries is ignored) - /// * IP addresses (both IPv4 and IPv6) are allowed, as are optional subnet masks (by adding /size, - /// for example "`192.168.1.0/24`"). - /// * An entry "`*`" matches all hostnames (this is the only wildcard allowed) - /// * Any other entry is considered a domain name (and may contain a leading dot, for example `google.com` - /// and `.google.com` are equivalent) and would match both that domain AND all subdomains. - /// - /// For example, if `"NO_PROXY=google.com, 192.168.1.0/24"` was set, all of the following would match - /// (and therefore would bypass the proxy): - /// * `http://google.com/` - /// * `http://www.google.com/` - /// * `http://192.168.1.42/` - /// - /// The URL `http://notgoogle.com/` would not match. - pub fn from_string(no_proxy_list: &str) -> Self { - let mut ips = Vec::new(); - let mut domains = Vec::new(); - let parts = no_proxy_list.split(',').map(str::trim); - for part in parts { - match part.parse::<IpNet>() { - // If we can parse an IP net or address, then use it, otherwise, assume it is a domain - Ok(ip) => ips.push(Ip::Network(ip)), - Err(_) => match part.parse::<IpAddr>() { - Ok(addr) => ips.push(Ip::Address(addr)), - Err(_) => { - if !part.trim().is_empty() { - domains.push(part.to_owned()) - } - } - }, - } - } - NoProxy { - ips: IpMatcher(ips), - domains: DomainMatcher(domains), - } - } - - /// Return true if this matches the host (domain or IP). - pub fn contains(&self, host: &str) -> bool { - // According to RFC3986, raw IPv6 hosts will be wrapped in []. So we need to strip those off - // the end in order to parse correctly - let host = if host.starts_with('[') { - let x: &[_] = &['[', ']']; - host.trim_matches(x) - } else { - host - }; - match host.parse::<IpAddr>() { - // If we can parse an IP addr, then use it, otherwise, assume it is a domain - Ok(ip) => self.ips.contains(ip), - Err(_) => self.domains.contains(host), - } - } - - fn is_empty(&self) -> bool { - self.ips.0.is_empty() && self.domains.0.is_empty() - } -} - -impl IpMatcher { - fn contains(&self, addr: IpAddr) -> bool { - for ip in &self.0 { - match ip { - Ip::Address(address) => { - if &addr == address { - return true; - } - } - Ip::Network(net) => { - if net.contains(&addr) { - return true; - } - } - } - } - false - } -} - -impl DomainMatcher { - // The following links may be useful to understand the origin of these rules: - // * https://curl.se/libcurl/c/CURLOPT_NOPROXY.html - // * https://github.com/curl/curl/issues/1208 - fn contains(&self, domain: &str) -> bool { - let domain_len = domain.len(); - for d in &self.0 { - if d == domain || d.strip_prefix('.') == Some(domain) { - return true; - } else if domain.ends_with(d) { - if d.starts_with('.') { - // If the first character of d is a dot, that means the first character of domain - // must also be a dot, so we are looking at a subdomain of d and that matches - return true; - } else if domain.as_bytes().get(domain_len - d.len() - 1) == Some(&b'.') { - // Given that d is a prefix of domain, if the prior character in domain is a dot - // then that means we must be matching a subdomain of d, and that matches - return true; - } - } else if d == "*" { - return true; - } - } - false - } -} - -mod builder { - /// A type that can used as a `Builder` value. - /// - /// Private and sealed, only visible in docs. - pub trait IntoValue { - #[doc(hidden)] - fn into_value(self) -> String; - } - - impl IntoValue for String { - #[doc(hidden)] - fn into_value(self) -> String { - self - } - } - - impl IntoValue for &String { - #[doc(hidden)] - fn into_value(self) -> String { - self.into() - } - } - - impl IntoValue for &str { - #[doc(hidden)] - fn into_value(self) -> String { - self.into() - } - } -} - -#[cfg(feature = "client-proxy-system")] -#[cfg(target_os = "macos")] -mod mac { - use system_configuration::core_foundation::base::{CFType, TCFType, TCFTypeRef}; - use system_configuration::core_foundation::dictionary::CFDictionary; - use system_configuration::core_foundation::number::CFNumber; - use system_configuration::core_foundation::string::{CFString, CFStringRef}; - use system_configuration::dynamic_store::SCDynamicStoreBuilder; - use system_configuration::sys::schema_definitions::{ - kSCPropNetProxiesHTTPEnable, kSCPropNetProxiesHTTPPort, kSCPropNetProxiesHTTPProxy, - kSCPropNetProxiesHTTPSEnable, kSCPropNetProxiesHTTPSPort, kSCPropNetProxiesHTTPSProxy, - }; - - pub(super) fn with_system(builder: &mut super::Builder) { - let store = SCDynamicStoreBuilder::new("").build(); - - let proxies_map = if let Some(proxies_map) = store.get_proxies() { - proxies_map - } else { - return; - }; - - if builder.http.is_empty() { - let http_proxy_config = parse_setting_from_dynamic_store( - &proxies_map, - unsafe { kSCPropNetProxiesHTTPEnable }, - unsafe { kSCPropNetProxiesHTTPProxy }, - unsafe { kSCPropNetProxiesHTTPPort }, - ); - if let Some(http) = http_proxy_config { - builder.http = http; - } - } - - if builder.https.is_empty() { - let https_proxy_config = parse_setting_from_dynamic_store( - &proxies_map, - unsafe { kSCPropNetProxiesHTTPSEnable }, - unsafe { kSCPropNetProxiesHTTPSProxy }, - unsafe { kSCPropNetProxiesHTTPSPort }, - ); - - if let Some(https) = https_proxy_config { - builder.https = https; - } - } - } - - fn parse_setting_from_dynamic_store( - proxies_map: &CFDictionary<CFString, CFType>, - enabled_key: CFStringRef, - host_key: CFStringRef, - port_key: CFStringRef, - ) -> Option<String> { - let proxy_enabled = proxies_map - .find(enabled_key) - .and_then(|flag| flag.downcast::<CFNumber>()) - .and_then(|flag| flag.to_i32()) - .unwrap_or(0) - == 1; - - if proxy_enabled { - let proxy_host = proxies_map - .find(host_key) - .and_then(|host| host.downcast::<CFString>()) - .map(|host| host.to_string()); - let proxy_port = proxies_map - .find(port_key) - .and_then(|port| port.downcast::<CFNumber>()) - .and_then(|port| port.to_i32()); - - return match (proxy_host, proxy_port) { - (Some(proxy_host), Some(proxy_port)) => Some(format!("{proxy_host}:{proxy_port}")), - (Some(proxy_host), None) => Some(proxy_host), - (None, Some(_)) => None, - (None, None) => None, - }; - } - - None - } -} - -#[cfg(feature = "client-proxy-system")] -#[cfg(windows)] -mod win { - pub(super) fn with_system(builder: &mut super::Builder) { - let settings = if let Ok(settings) = windows_registry::CURRENT_USER - .open("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings") - { - settings - } else { - return; - }; - - if settings.get_u32("ProxyEnable").unwrap_or(0) == 0 { - return; - } - - if let Ok(val) = settings.get_string("ProxyServer") { - if builder.http.is_empty() { - builder.http = val.clone(); - } - if builder.https.is_empty() { - builder.https = val; - } - } - - if builder.no.is_empty() { - if let Ok(val) = settings.get_string("ProxyOverride") { - builder.no = val - .split(';') - .map(|s| s.trim()) - .collect::<Vec<&str>>() - .join(",") - .replace("*.", ""); - } - } - } -} - -#[cfg(test)] -mod tests { - use super::*; - - #[test] - fn test_domain_matcher() { - let domains = vec![".foo.bar".into(), "bar.foo".into()]; - let matcher = DomainMatcher(domains); - - // domains match with leading `.` - assert!(matcher.contains("foo.bar")); - // subdomains match with leading `.` - assert!(matcher.contains("www.foo.bar")); - - // domains match with no leading `.` - assert!(matcher.contains("bar.foo")); - // subdomains match with no leading `.` - assert!(matcher.contains("www.bar.foo")); - - // non-subdomain string prefixes don't match - assert!(!matcher.contains("notfoo.bar")); - assert!(!matcher.contains("notbar.foo")); - } - - #[test] - fn test_no_proxy_wildcard() { - let no_proxy = NoProxy::from_string("*"); - assert!(no_proxy.contains("any.where")); - } - - #[test] - fn test_no_proxy_ip_ranges() { - let no_proxy = - NoProxy::from_string(".foo.bar, bar.baz,10.42.1.1/24,::1,10.124.7.8,2001::/17"); - - let should_not_match = [ - // random url, not in no_proxy - "hyper.rs", - // make sure that random non-subdomain string prefixes don't match - "notfoo.bar", - // make sure that random non-subdomain string prefixes don't match - "notbar.baz", - // ipv4 address out of range - "10.43.1.1", - // ipv4 address out of range - "10.124.7.7", - // ipv6 address out of range - "[ffff:db8:a0b:12f0::1]", - // ipv6 address out of range - "[2005:db8:a0b:12f0::1]", - ]; - - for host in &should_not_match { - assert!(!no_proxy.contains(host), "should not contain {host:?}"); - } - - let should_match = [ - // make sure subdomains (with leading .) match - "hello.foo.bar", - // make sure exact matches (without leading .) match (also makes sure spaces between entries work) - "bar.baz", - // make sure subdomains (without leading . in no_proxy) match - "foo.bar.baz", - // make sure subdomains (without leading . in no_proxy) match - this differs from cURL - "foo.bar", - // ipv4 address match within range - "10.42.1.100", - // ipv6 address exact match - "[::1]", - // ipv6 address match within range - "[2001:db8:a0b:12f0::1]", - // ipv4 address exact match - "10.124.7.8", - ]; - - for host in &should_match { - assert!(no_proxy.contains(host), "should contain {host:?}"); - } - } - - macro_rules! p { - ($($n:ident = $v:expr,)*) => ({Builder { - $($n: $v.into(),)* - ..Builder::default() - }.build()}); - } - - fn intercept(p: &Matcher, u: &str) -> Intercept { - p.intercept(&u.parse().unwrap()).unwrap() - } - - #[test] - fn test_all_proxy() { - let p = p! { - all = "http://om.nom", - }; - - assert_eq!("http://om.nom", intercept(&p, "http://example.com").uri()); - - assert_eq!("http://om.nom", intercept(&p, "https://example.com").uri()); - } - - #[test] - fn test_specific_overrides_all() { - let p = p! { - all = "http://no.pe", - http = "http://y.ep", - }; - - assert_eq!("http://no.pe", intercept(&p, "https://example.com").uri()); - - // the http rule is "more specific" than the all rule - assert_eq!("http://y.ep", intercept(&p, "http://example.com").uri()); - } - - #[test] - fn test_parse_no_scheme_defaults_to_http() { - let p = p! { - https = "y.ep", - http = "127.0.0.1:8887", - }; - - assert_eq!(intercept(&p, "https://example.local").uri(), "http://y.ep"); - assert_eq!( - intercept(&p, "http://example.local").uri(), - "http://127.0.0.1:8887" - ); - } - - #[test] - fn test_parse_http_auth() { - let p = p! { - all = "http://Aladdin:opensesame@y.ep", - }; - - let proxy = intercept(&p, "https://example.local"); - assert_eq!(proxy.uri(), "http://y.ep"); - assert_eq!( - proxy.basic_auth().expect("basic_auth"), - "Basic QWxhZGRpbjpvcGVuc2VzYW1l" - ); - } - - #[test] - fn test_parse_http_auth_without_scheme() { - let p = p! { - all = "Aladdin:opensesame@y.ep", - }; - - let proxy = intercept(&p, "https://example.local"); - assert_eq!(proxy.uri(), "http://y.ep"); - assert_eq!( - proxy.basic_auth().expect("basic_auth"), - "Basic QWxhZGRpbjpvcGVuc2VzYW1l" - ); - } - - #[test] - fn test_dont_parse_http_when_is_cgi() { - let mut builder = Matcher::builder(); - builder.is_cgi = true; - builder.http = "http://never.gonna.let.you.go".into(); - let m = builder.build(); - - assert!(m.intercept(&"http://rick.roll".parse().unwrap()).is_none()); - } -} diff --git a/vendor/hyper-util/src/client/proxy/mod.rs b/vendor/hyper-util/src/client/proxy/mod.rs deleted file mode 100644 index 59c8e46d..00000000 --- a/vendor/hyper-util/src/client/proxy/mod.rs +++ /dev/null @@ -1,3 +0,0 @@ -//! Proxy utilities - -pub mod matcher; diff --git a/vendor/hyper-util/src/client/service.rs b/vendor/hyper-util/src/client/service.rs deleted file mode 100644 index 580fb105..00000000 --- a/vendor/hyper-util/src/client/service.rs +++ /dev/null @@ -1,8 +0,0 @@ -struct ConnectingPool<C, P> { - connector: C, - pool: P, -} - -struct PoolableSvc<S>(S); - - |
