2021-10-26 20:46:40 +02:00
|
|
|
//! Run with
|
|
|
|
|
//!
|
|
|
|
|
//! ```not_rust
|
|
|
|
|
//! $ cargo run -p example-http-proxy
|
|
|
|
|
//! ```
|
|
|
|
|
//!
|
|
|
|
|
//! In another terminal:
|
|
|
|
|
//!
|
|
|
|
|
//! ```not_rust
|
|
|
|
|
//! $ curl -v -x "127.0.0.1:3000" https://tokio.rs
|
|
|
|
|
//! ```
|
|
|
|
|
//!
|
2021-11-04 12:48:37 +01:00
|
|
|
//! Example is based on <https://github.com/hyperium/hyper/blob/master/examples/http_proxy.rs>
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// TODO
|
|
|
|
|
fn main() {
|
|
|
|
|
eprint!("this example has not yet been updated to hyper 1.0");
|
|
|
|
|
}
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// use axum::{
|
|
|
|
|
// body::Body,
|
|
|
|
|
// extract::Request,
|
|
|
|
|
// http::{Method, StatusCode},
|
|
|
|
|
// response::{IntoResponse, Response},
|
|
|
|
|
// routing::get,
|
|
|
|
|
// Router,
|
|
|
|
|
// };
|
|
|
|
|
// use hyper::upgrade::Upgraded;
|
|
|
|
|
// use std::net::SocketAddr;
|
|
|
|
|
// use tokio::net::TcpStream;
|
|
|
|
|
// use tower::{make::Shared, ServiceExt};
|
|
|
|
|
// use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// #[tokio::main]
|
|
|
|
|
// async fn main() {
|
|
|
|
|
// tracing_subscriber::registry()
|
|
|
|
|
// .with(
|
|
|
|
|
// tracing_subscriber::EnvFilter::try_from_default_env()
|
|
|
|
|
// .unwrap_or_else(|_| "example_http_proxy=trace,tower_http=debug".into()),
|
|
|
|
|
// )
|
|
|
|
|
// .with(tracing_subscriber::fmt::layer())
|
|
|
|
|
// .init();
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// let router_svc = Router::new().route("/", get(|| async { "Hello, World!" }));
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// let service = tower::service_fn(move |req: Request<_>| {
|
|
|
|
|
// let router_svc = router_svc.clone();
|
|
|
|
|
// let req = req.map(Body::new);
|
|
|
|
|
// async move {
|
|
|
|
|
// if req.method() == Method::CONNECT {
|
|
|
|
|
// proxy(req).await
|
|
|
|
|
// } else {
|
|
|
|
|
// router_svc.oneshot(req).await.map_err(|err| match err {})
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
// });
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
|
|
|
|
|
// tracing::debug!("listening on {}", addr);
|
|
|
|
|
// hyper::Server::bind(&addr)
|
|
|
|
|
// .http1_preserve_header_case(true)
|
|
|
|
|
// .http1_title_case_headers(true)
|
|
|
|
|
// .serve(Shared::new(service))
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
// }
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// async fn proxy(req: Request) -> Result<Response, hyper::Error> {
|
|
|
|
|
// tracing::trace!(?req);
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// if let Some(host_addr) = req.uri().authority().map(|auth| auth.to_string()) {
|
|
|
|
|
// tokio::task::spawn(async move {
|
|
|
|
|
// match hyper::upgrade::on(req).await {
|
|
|
|
|
// Ok(upgraded) => {
|
|
|
|
|
// if let Err(e) = tunnel(upgraded, host_addr).await {
|
|
|
|
|
// tracing::warn!("server io error: {}", e);
|
|
|
|
|
// };
|
|
|
|
|
// }
|
|
|
|
|
// Err(e) => tracing::warn!("upgrade error: {}", e),
|
|
|
|
|
// }
|
|
|
|
|
// });
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// Ok(Response::new(Body::empty()))
|
|
|
|
|
// } else {
|
|
|
|
|
// tracing::warn!("CONNECT host is not socket addr: {:?}", req.uri());
|
|
|
|
|
// Ok((
|
|
|
|
|
// StatusCode::BAD_REQUEST,
|
|
|
|
|
// "CONNECT must be to a socket address",
|
|
|
|
|
// )
|
|
|
|
|
// .into_response())
|
|
|
|
|
// }
|
|
|
|
|
// }
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// async fn tunnel(mut upgraded: Upgraded, addr: String) -> std::io::Result<()> {
|
|
|
|
|
// let mut server = TcpStream::connect(addr).await?;
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// let (from_client, from_server) =
|
|
|
|
|
// tokio::io::copy_bidirectional(&mut upgraded, &mut server).await?;
|
2021-10-26 20:46:40 +02:00
|
|
|
|
2023-11-23 12:03:03 +01:00
|
|
|
// tracing::debug!(
|
|
|
|
|
// "client wrote {} bytes and received {} bytes",
|
|
|
|
|
// from_client,
|
|
|
|
|
// from_server
|
|
|
|
|
// );
|
|
|
|
|
|
|
|
|
|
// Ok(())
|
|
|
|
|
// }
|