Add check for invalid enum values (#104)

projectdiscovery · Feb 16, 2023 · 74f5b75 · 74f5b75
1 parent 47f28bd
commit 74f5b75
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/goflags.go b/goflags.go
@@ -482,11 +482,16 @@ func (flagSet *FlagSet) EnumVar(field *string, long string, defaultValue EnumVar
 
 // EnumVarP adds a enum flag with a shortname and longname
 func (flagSet *FlagSet) EnumVarP(field *string, long, short string, defaultValue EnumVariable, usage string, allowedTypes AllowdTypes) *FlagData {
+	var hasDefaultValue bool
 	for k, v := range allowedTypes {
 		if v == defaultValue {
+			hasDefaultValue = true
 			*field = k
 		}
 	}
+	if !hasDefaultValue {
+		panic("undefined default value")
+	}
 	flagData := &FlagData{
 		usage:        usage,
 		long:         long,