1
0
mirror of https://github.com/actix/examples synced 2025-06-26 17:17:42 +02:00

move http-proxy to other

This commit is contained in:
Rob Ede
2022-02-02 02:03:54 +00:00
parent a99c4ca546
commit ffb20a1b20
4 changed files with 8 additions and 8 deletions

View File

@ -1,13 +0,0 @@
[package]
name = "http-proxy"
version = "1.0.0"
edition = "2021"
[dependencies]
actix-web = { version = "4.0.0-rc.1", features = ["openssl"] }
awc = "3.0.0-beta.19"
clap = { version = "3", features = ["derive"] }
env_logger = "0.9"
log = "0.4"
url = "2.2"

View File

@ -1,12 +0,0 @@
## HTTP naïve proxy example
This is a relatively simple HTTP proxy, forwarding HTTP requests to another HTTP server, including
request body, headers, and streaming uploads.
To start:
``` shell
cargo run -- <listen addr> <listen port> <forward addr> <forward port>
# example:
cargo run -- 127.0.0.1 3333 127.0.0.1 8080
```

View File

@ -1,90 +0,0 @@
use std::net::ToSocketAddrs;
use actix_web::{
error, middleware, web, App, Error, HttpRequest, HttpResponse, HttpServer,
};
use awc::Client;
use clap::StructOpt;
use url::Url;
async fn forward(
req: HttpRequest,
payload: web::Payload,
url: web::Data<Url>,
client: web::Data<Client>,
) -> Result<HttpResponse, Error> {
let mut new_url = url.get_ref().clone();
new_url.set_path(req.uri().path());
new_url.set_query(req.uri().query());
// TODO: This forwarded implementation is incomplete as it only handles the unofficial
// X-Forwarded-For header but not the official Forwarded one.
let forwarded_req = client
.request_from(new_url.as_str(), req.head())
.no_decompress();
let forwarded_req = match req.head().peer_addr {
Some(addr) => {
forwarded_req.insert_header(("x-forwarded-for", format!("{}", addr.ip())))
}
None => forwarded_req,
};
let res = forwarded_req
.send_stream(payload)
.await
.map_err(error::ErrorInternalServerError)?;
let mut client_resp = HttpResponse::build(res.status());
// Remove `Connection` as per
// https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Connection#Directives
for (header_name, header_value) in
res.headers().iter().filter(|(h, _)| *h != "connection")
{
client_resp.insert_header((header_name.clone(), header_value.clone()));
}
Ok(client_resp.streaming(res))
}
#[derive(clap::Parser, Debug)]
struct CliArguments {
listen_addr: String,
listen_port: u16,
forward_addr: String,
forward_port: u16,
}
#[actix_web::main]
async fn main() -> std::io::Result<()> {
env_logger::init_from_env(env_logger::Env::new().default_filter_or("info"));
let args = CliArguments::parse();
let forward_socket_addr = (args.forward_addr, args.forward_port)
.to_socket_addrs()?
.next()
.expect("given forwarding address was not valid");
let forward_url = format!("http://{}", forward_socket_addr);
let forward_url = Url::parse(&forward_url).unwrap();
log::info!(
"starting HTTP serer at http://{}:{}",
&args.listen_addr,
args.listen_port
);
log::info!("forwarding to {forward_url}");
HttpServer::new(move || {
App::new()
.app_data(web::Data::new(Client::default()))
.app_data(web::Data::new(forward_url.clone()))
.wrap(middleware::Logger::default())
.default_service(web::to(forward))
})
.bind((args.listen_addr, args.listen_port))?
.workers(2)
.run()
.await
}