perf: 优化Flag.go的代码,添加注释,规范输出

This commit is contained in:
ZacharyZcR 2024-12-18 22:25:22 +08:00
parent 1d0676e508
commit 66125a3a2d

View File

@ -19,55 +19,76 @@ func Banner() {
func Flag(Info *Config.HostInfo) { func Flag(Info *Config.HostInfo) {
Banner() Banner()
flag.StringVar(&Info.Host, "h", "", "IP address of the host you want to scan,for example: 192.168.11.11 | 192.168.11.11-255 | 192.168.11.11,192.168.11.12")
flag.StringVar(&NoHosts, "hn", "", "the hosts no scan,as: -hn 192.168.1.1/24") // 目标配置
flag.StringVar(&Ports, "p", DefaultPorts, "Select a port,for example: 22 | 1-65535 | 22,80,3306") flag.StringVar(&Info.Host, "h", "", "目标主机IP例如: 192.168.11.11 | 192.168.11.11-255 | 192.168.11.11,192.168.11.12")
flag.StringVar(&PortAdd, "pa", "", "add port base DefaultPorts,-pa 3389") flag.StringVar(&NoHosts, "hn", "", "排除的主机范围,例如: -hn 192.168.1.1/24")
flag.StringVar(&UserAdd, "usera", "", "add a user base DefaultUsers,-usera user") flag.StringVar(&Ports, "p", DefaultPorts, "端口配置,例如: 22 | 1-65535 | 22,80,3306")
flag.StringVar(&PassAdd, "pwda", "", "add a password base DefaultPasses,-pwda password") flag.StringVar(&PortAdd, "pa", "", "在默认端口基础上添加端口,-pa 3389")
flag.StringVar(&NoPorts, "pn", "", "the ports no scan,as: -pn 445") flag.StringVar(&NoPorts, "pn", "", "排除的端口,例如: -pn 445")
flag.StringVar(&Command, "c", "", "exec command (ssh|wmiexec)")
flag.StringVar(&SshKey, "sshkey", "", "sshkey file (id_rsa)") // 认证配置
flag.StringVar(&Domain, "domain", "", "smb domain") flag.StringVar(&UserAdd, "usera", "", "在默认用户列表基础上添加用户,-usera user")
flag.StringVar(&Username, "user", "", "username") flag.StringVar(&PassAdd, "pwda", "", "在默认密码列表基础上添加密码,-pwda password")
flag.StringVar(&Password, "pwd", "", "password") flag.StringVar(&Username, "user", "", "用户名")
flag.Int64Var(&Timeout, "time", 3, "Set timeout") flag.StringVar(&Password, "pwd", "", "密码")
flag.StringVar(&Scantype, "m", "all", "Select scan type ,as: -m ssh") flag.StringVar(&Domain, "domain", "", "域名(用于SMB)")
flag.StringVar(&Path, "path", "", "fcgi、smb romote file path") flag.StringVar(&SshKey, "sshkey", "", "SSH密钥文件(id_rsa)")
flag.IntVar(&Threads, "t", 600, "Thread nums")
flag.IntVar(&LiveTop, "top", 10, "show live len top") // 扫描配置
flag.StringVar(&HostFile, "hf", "", "host file, -hf ip.txt") flag.StringVar(&Scantype, "m", "all", "扫描类型,例如: -m ssh")
flag.StringVar(&Userfile, "userf", "", "username file") flag.IntVar(&Threads, "t", 600, "线程数量")
flag.StringVar(&Passfile, "pwdf", "", "password file") flag.Int64Var(&Timeout, "time", 3, "超时时间(秒)")
flag.StringVar(&Hashfile, "hashf", "", "hash file") flag.IntVar(&LiveTop, "top", 10, "显示存活主机数量")
flag.StringVar(&PortFile, "portf", "", "Port File") flag.BoolVar(&NoPing, "np", false, "禁用存活探测")
flag.StringVar(&PocPath, "pocpath", "", "poc file path") flag.BoolVar(&Ping, "ping", false, "使用ping替代ICMP")
flag.StringVar(&RedisFile, "rf", "", "redis file to write sshkey file (as: -rf id_rsa.pub)") flag.StringVar(&Command, "c", "", "执行命令(支持ssh|wmiexec)")
flag.StringVar(&RedisShell, "rs", "", "redis shell to write cron file (as: -rs 192.168.1.1:6666)")
flag.BoolVar(&NoPoc, "nopoc", false, "not to scan web vul") // 文件配置
flag.BoolVar(&IsBrute, "nobr", false, "not to Brute password") flag.StringVar(&HostFile, "hf", "", "主机列表文件")
flag.IntVar(&BruteThread, "br", 1, "Brute threads") flag.StringVar(&Userfile, "userf", "", "用户名字典")
flag.BoolVar(&NoPing, "np", false, "not to ping") flag.StringVar(&Passfile, "pwdf", "", "密码字典")
flag.BoolVar(&Ping, "ping", false, "using ping replace icmp") flag.StringVar(&Hashfile, "hashf", "", "Hash字典")
flag.StringVar(&Outputfile, "o", "result.txt", "Outputfile") flag.StringVar(&PortFile, "portf", "", "端口列表文件")
flag.BoolVar(&TmpSave, "no", false, "not to save output log")
flag.Int64Var(&WaitTime, "debug", 60, "every time to LogErr") // Web配置
flag.BoolVar(&Silent, "silent", false, "silent scan") flag.StringVar(&URL, "u", "", "目标URL")
flag.BoolVar(&Nocolor, "nocolor", false, "no color") flag.StringVar(&UrlFile, "uf", "", "URL列表文件")
flag.BoolVar(&PocFull, "full", false, "poc full scan,as: shiro 100 key") flag.StringVar(&Cookie, "cookie", "", "设置Cookie")
flag.StringVar(&URL, "u", "", "url") flag.Int64Var(&WebTimeout, "wt", 5, "Web请求超时时间")
flag.StringVar(&UrlFile, "uf", "", "urlfile") flag.StringVar(&Proxy, "proxy", "", "设置HTTP代理")
flag.StringVar(&Pocinfo.PocName, "pocname", "", "use the pocs these contain pocname, -pocname weblogic") flag.StringVar(&Socks5Proxy, "socks5", "", "设置Socks5代理(将用于TCP连接,超时设置将失效)")
flag.StringVar(&Proxy, "proxy", "", "set poc proxy, -proxy http://127.0.0.1:8080")
flag.StringVar(&Socks5Proxy, "socks5", "", "set socks5 proxy, will be used in tcp connection, timeout setting will not work") // POC配置
flag.StringVar(&Cookie, "cookie", "", "set poc cookie,-cookie rememberMe=login") flag.StringVar(&PocPath, "pocpath", "", "POC文件路径")
flag.Int64Var(&WebTimeout, "wt", 5, "Set web timeout") flag.StringVar(&Pocinfo.PocName, "pocname", "", "使用包含指定名称的POC,例如: -pocname weblogic")
flag.BoolVar(&DnsLog, "dns", false, "using dnslog poc") flag.BoolVar(&NoPoc, "nopoc", false, "禁用Web漏洞扫描")
flag.IntVar(&PocNum, "num", 20, "poc rate") flag.BoolVar(&PocFull, "full", false, "完整POC扫描,如:shiro 100个key")
flag.StringVar(&SC, "sc", "", "ms17 shellcode,as -sc add") flag.BoolVar(&DnsLog, "dns", false, "启用dnslog验证")
flag.BoolVar(&IsWmi, "wmi", false, "start wmi") flag.IntVar(&PocNum, "num", 20, "POC并发数")
flag.StringVar(&Hash, "hash", "", "hash")
flag.BoolVar(&Noredistest, "noredis", false, "no redis sec test") // Redis利用配置
flag.BoolVar(&JsonOutput, "json", false, "json output") flag.StringVar(&RedisFile, "rf", "", "Redis写入SSH公钥文件")
flag.StringVar(&RedisShell, "rs", "", "Redis写入计划任务")
flag.BoolVar(&Noredistest, "noredis", false, "禁用Redis安全检测")
// 暴力破解配置
flag.BoolVar(&IsBrute, "nobr", false, "禁用密码爆破")
flag.IntVar(&BruteThread, "br", 1, "密码爆破线程数")
// 其他配置
flag.StringVar(&Path, "path", "", "FCG/SMB远程文件路径")
flag.StringVar(&Hash, "hash", "", "Hash值")
flag.StringVar(&SC, "sc", "", "MS17漏洞shellcode")
flag.BoolVar(&IsWmi, "wmi", false, "启用WMI")
// 输出配置
flag.StringVar(&Outputfile, "o", "result.txt", "结果输出文件")
flag.BoolVar(&TmpSave, "no", false, "禁用结果保存")
flag.BoolVar(&Silent, "silent", false, "静默扫描模式")
flag.BoolVar(&Nocolor, "nocolor", false, "禁用彩色输出")
flag.BoolVar(&JsonOutput, "json", false, "JSON格式输出")
flag.Int64Var(&WaitTime, "debug", 60, "错误日志输出间隔")
flag.Parse() flag.Parse()
} }