Simplify addPayloadFlags switch

Author: terrorbyte

cli/commandline.go

@@ -852,48 +852,12 @@ func addPayloadFlags(conf *config.Config) {
 		count := map[payload.Types]int{}
 		for i, supported := range conf.SupportedPayloads {
 			switch supported.Types {
-			case payload.LinuxCommand:
+			case payload.LinuxCommand, payload.WindowsCommand, payload.MacCommand, payload.GenericCommand:
 				_, exists := conf.StringFlagsMap["command"]
 				if !exists {
 					conf.CreateStringFlag("command", "", "Command to use for the exploit, an empty string will use the exploit default.")
 				}
-			case payload.WindowsCommand:
-				_, exists := conf.StringFlagsMap["command"]
-				if !exists {
-					conf.CreateStringFlag("command", "", "Command to use for the exploit, an empty string will use the exploit default.")
-				}
-			case payload.MacCommand:
-				_, exists := conf.StringFlagsMap["command"]
-				if !exists {
-					conf.CreateStringFlag("command", "", "Command to use for the exploit, an empty string will use the exploit default.")
-				}
-			case payload.GenericCommand:
-				_, exists := conf.StringFlagsMap["command"]
-				if !exists {
-					conf.CreateStringFlag("command", "", "Command to use for the exploit, an empty string will use the exploit default.")
-				}
-			case payload.LinuxELF:
-				_, exists := conf.StringFlagsMap["payload"]
-				if !exists {
-					conf.CreateStringFlag("payload", "", "Path to load custom payload from, an empty string will use the exploit default.")
-				}
-			case payload.LinuxSO:
-				_, exists := conf.StringFlagsMap["payload"]
-				if !exists {
-					conf.CreateStringFlag("payload", "", "Path to load custom payload from, an empty string will use the exploit default.")
-				}
-			case payload.WindowsEXE:
-				_, exists := conf.StringFlagsMap["payload"]
-				if !exists {
-					conf.CreateStringFlag("payload", "", "Path to load custom payload from, an empty string will use the exploit default.")
-				}
-
-			case payload.WindowsDLL:
-				_, exists := conf.StringFlagsMap["payload"]
-				if !exists {
-					conf.CreateStringFlag("payload", "", "Path to load custom payload from, an empty string will use the exploit default.")
-				}
-			case payload.Webshell:
+			case payload.LinuxELF, payload.LinuxSO, payload.WindowsEXE, payload.WindowsDLL, payload.Webshell:
 				_, exists := conf.StringFlagsMap["payload"]
 				if !exists {
 					conf.CreateStringFlag("payload", "", "Path to load custom payload from, an empty string will use the exploit default.")