simon/tiny_httpd
1
0
Fork 0
mirror of https://github.com/c-cube/tiny_httpd.git synced 2025-12-05 19:00:32 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(http_of_dir): better parsing of max-upload size

Browse source
This commit is contained in:
Simon Cruanes 2019-11-18 18:45:26 -06:00
parent 9c84f95e12
commit fba2af546c
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/bin/http_of_dir.ml
View file

@ -124,6 +124,13 @@ let serve ~config (dir:string) : _ result =
));
S.run server
let parse_size s : int =
try Scanf.sscanf s "%dM" (fun n -> n * 1_024 * 1_024)
with _ ->
try Scanf.sscanf s "%dk" (fun n -> n * 1_024)
with _ ->
try int_of_string s
with _ -> raise (Arg.Bad "invalid size (expected <int>[kM]?)")
let main () =
let config = default_config () in
@ -135,8 +142,8 @@ let main () =
"-p", Int (fun x -> config.port <- x), " alias to --port";
"--dir", Set_string dir_, " directory to serve (default: \".\")";
"--no-upload", Unit (fun () -> config.upload <- false), " disable file uploading";
"--max-upload", Int (fun i -> config.max_upload_size <- 1024 * 1024 * i),
"maximum size of files that can be uploaded, in MB";
"--max-upload", String (fun i -> config.max_upload_size <- parse_size i),
"maximum size of files that can be uploaded";
"--debug", Unit (fun () -> S._enable_debug true), " debug mode";
"--delete", Unit (fun () -> config.delete <- true), " enable `delete` on files";
"--no-delete", Unit (fun () -> config.delete <- false), " disable `delete` on files";