Limit file upload size

This commit is contained in:
Bel LaPointe
2020-07-24 23:57:24 -06:00
parent c47850647e
commit 8741db40b4
2 changed files with 8 additions and 5 deletions

View File

@@ -16,6 +16,7 @@ type Config struct {
FileRoot string
Auth bool
AuthLifetime time.Duration
MaxFileSize int64
}
func New() Config {
@@ -35,6 +36,7 @@ func New() Config {
as.Append(args.STRING, "drivertype", "database driver to use", "boltdb")
as.Append(args.BOOL, "auth", "check for authorized access", false)
as.Append(args.DURATION, "authlifetime", "duration auth is valid for", time.Hour)
as.Append(args.INT, "max-file-size", "max file size for uploads in bytes", 50*(1<<20))
if err := as.Parse(); err != nil {
panic(err)
@@ -49,5 +51,6 @@ func New() Config {
DriverType: as.GetString("drivertype"),
Auth: as.GetBool("auth"),
AuthLifetime: as.GetDuration("authlifetime"),
MaxFileSize: int64(as.GetInt("max-file-size")),
}
}