From dafafaf4188f34ac385cec93ec41c77d61303c85 Mon Sep 17 00:00:00 2001 From: Kavin <20838718+FireMasterK@users.noreply.github.com> Date: Wed, 2 Nov 2022 20:43:17 +0000 Subject: [PATCH] Add option to disable ipv6. --- src/main.rs | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/src/main.rs b/src/main.rs index 5317b07..f059522 100644 --- a/src/main.rs +++ b/src/main.rs @@ -34,7 +34,19 @@ lazy_static!( ); lazy_static!( - static ref CLIENT: Client = Client::new(); + static ref CLIENT: Client = { + let builder = Client::builder() + .user_agent("Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0"); + + if env::var("IPV4_ONLY").is_ok() { + builder + .local_address(Some("0.0.0.0".parse().unwrap())) + .build() + .unwrap() + } else { + builder.build().unwrap() + } + }; ); const ALLOWED_DOMAINS: [&str; 7] = [ @@ -121,8 +133,6 @@ async fn index(req: HttpRequest) -> Result> { let request_headers = request.headers_mut(); - request_headers.insert("User-Agent", "Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0".parse().unwrap()); - for (key, value) in req.headers() { if is_header_allowed(key.as_str()) { request_headers.insert(key.clone(), value.clone());