pierre/PagesServer
1
0
Fork 0
Code Issues Pull Requests 9 Projects Releases Wiki Activity

Prevent empty archive extraction

Browse Source
This commit is contained in:
Pierre HUBERT 2022-03-27 14:29:11 +02:00
parent c100d87145
commit d85cce8264
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/main.rs
View File

@ -5,7 +5,7 @@ use actix_files::{Files, NamedFile};
use actix_multipart::Multipart; use actix_multipart::Multipart;
use actix_web::{App, Error, HttpRequest, HttpResponse, HttpServer, web}; use actix_web::{App, Error, HttpRequest, HttpResponse, HttpServer, web};
use actix_web::dev::{fn_service, ServiceRequest, ServiceResponse}; use actix_web::dev::{fn_service, ServiceRequest, ServiceResponse};
use actix_web::error::{ErrorInternalServerError, ErrorUnauthorized}; use actix_web::error::{ErrorBadRequest, ErrorInternalServerError, ErrorUnauthorized};
use actix_web::middleware::Logger; use actix_web::middleware::Logger;
use bytes::BufMut; use bytes::BufMut;
use clap::Parser; use clap::Parser;
@ -178,6 +178,11 @@ async fn replace_files(args: web::Data<Args>, req: HttpRequest, mut payload: Mul
} }
} }
// Check if at least a file was sent
if new_files.is_empty() {
return Err(ErrorBadRequest("No file to extract!"));
}
// Delete all current files in storage // Delete all current files in storage
for entry in std::fs::read_dir(args.storage_path())? { for entry in std::fs::read_dir(args.storage_path())? {
let entry = entry?; let entry = entry?;