proxy_http 0.1.9

Async HTTP proxy server in Rust. / Rust 异步 HTTP 代理服务器。
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use std::sync::Arc;

use hyper::{Request, body::Incoming, service::service_fn, upgrade::Upgraded};
use log::error;
use proxy_fetch::Fetch;

use crate::proxy;

pub async fn upgrade(host: String, port: u16, upgraded: Upgraded, fetch: Arc<Fetch>) {
  let base = if port == 443 || port == 80 {
    format!("https://{host}")
  } else {
    format!("https://{host}:{port}")
  };

  let service = service_fn(move |req: Request<Incoming>| {
    let fetch = Arc::clone(&fetch);
    let base = base.clone();

    async move {
      let url = format!(
        "{}{}",
        base,
        req
          .uri()
          .path_and_query()
          .map(|i| i.as_str())
          .unwrap_or_default()
      );

      proxy(url, req, fetch).await
    }
  });

  // 直接使用 Upgraded，它已经实现了 hyper::rt::Read 和 hyper::rt::Write
  // Use Upgraded directly, it already implements hyper::rt::Read and hyper::rt::Write
  if let Err(err) =
    hyper_util::server::conn::auto::Builder::new(hyper_util::rt::TokioExecutor::new())
      .serve_connection(upgraded, service)
      .await
  {
    error!("Error serving tunneled connection: {:?}", err);
  }
}