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

Add STATIC_DATA_DIR ENV for storing static files

Browse Source
This commit is contained in:
William Desportes
2024-06-27 23:01:37 +02:00
parent dee1968e36
commit 0d797713a6
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -333,6 +333,7 @@ fn handle_report(conn: &Mutex<Connection>, request: &Request) -> Response {
fn handle_list_scanners( fn handle_list_scanners(
conn: &Mutex<Connection>, conn: &Mutex<Connection>,
static_data_dir: &str,
scanner_name: Scanners, scanner_name: Scanners,
request: &Request, request: &Request,
) -> Response { ) -> Response {
@ -342,7 +343,7 @@ fn handle_list_scanners(
// located. // located.
// In order to avoid potential security threats, `match_assets` will never return any // In order to avoid potential security threats, `match_assets` will never return any
// file outside of this directory even if the URL is for example `/../../foo.txt`. // file outside of this directory even if the URL is for example `/../../foo.txt`.
let response = rouille::match_assets(&request, "../data/"); let response = rouille::match_assets(&request, static_data_dir);
if response.is_success() { if response.is_success() {
return response; return response;
@ -505,6 +506,11 @@ fn main() -> Result<()> {
"./snow-scanner.sqlite".to_string() "./snow-scanner.sqlite".to_string()
}; };
let static_data_dir: String = match env::var("STATIC_DATA_DIR") {
Ok(val) => val,
Err(_) => "../data/".to_string(),
};
println!("Database will be saved at: {}", db_file); println!("Database will be saved at: {}", db_file);
let conn = Mutex::new(get_connection(db_file.as_str())); let conn = Mutex::new(get_connection(db_file.as_str()));
@ -607,7 +613,7 @@ fn main() -> Result<()> {
}, },
(GET) (/scanners/{scanner_name: Scanners}) => { (GET) (/scanners/{scanner_name: Scanners}) => {
handle_list_scanners(&conn, scanner_name, &request) handle_list_scanners(&conn, &static_data_dir, scanner_name, &request)
}, },
(GET) (/{api_key: String}/scanners/{scanner_name: String}) => { (GET) (/{api_key: String}/scanners/{scanner_name: String}) => {
let mut mac = HmacSha256::new_from_slice(b"my secret and secure key") let mut mac = HmacSha256::new_from_slice(b"my secret and secure key")