wdes/security
1
0
Fork 0
Code Actions Packages Releases Activity

Add an ENV for SERVER_ADDRESS
All checks were successful
Gitea Actions Demo / Explore-Gitea-Actions (push) Successful in 6m51s
Details

Browse Source
This commit is contained in:
William Desportes
2024-06-23 11:15:59 +02:00
parent 2be2d1b451
commit 3f3e069aab
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
snow-scanner/src/main.rs
View File

@ -12,7 +12,7 @@ use sha2::Sha256;
use std::str::FromStr; use std::str::FromStr;
use std::sync::Mutex; use std::sync::Mutex;
use std::time::Duration; use std::time::Duration;
use std::{fmt, thread}; use std::{env, fmt, thread};
use uuid7::Uuid; use uuid7::Uuid;
use hickory_client::client::SyncClient; use hickory_client::client::SyncClient;
@ -369,7 +369,13 @@ fn get_dns_client() -> SyncClient<TcpClientConnection> {
} }
fn main() -> Result<()> { fn main() -> Result<()> {
println!("Now listening on localhost:8000"); let server_address: String = if let Ok(env) = env::var("SERVER_ADDRESS") {
env
} else {
"localhost:8000".to_string()
};
println!("Now listening on {}", server_address);
let conn = Mutex::new(get_connection()); let conn = Mutex::new(get_connection());
conn.lock() conn.lock()
@ -414,7 +420,7 @@ fn main() -> Result<()> {
thread::sleep(two_hundred_millis); thread::sleep(two_hundred_millis);
}); });
rouille::start_server("localhost:8000", move |request| { rouille::start_server(server_address, move |request| {
router!(request, router!(request,
(GET) (/) => { (GET) (/) => {
rouille::Response::html(FORM) rouille::Response::html(FORM)