ftp默认列出前5个目录

common/Parse.go

@@ -94,7 +94,11 @@ func ParseInput(Info *HostInfo) {
 		os.Exit(0)
 	}
 	if Info.Outputfile != "" {
-		Outputfile = Info.Outputfile
+		if !strings.Contains(Outputfile, "/") && !strings.Contains(Outputfile, `\`) {
+			Outputfile = getpath() + Info.Outputfile
+		} else {
+			Outputfile = Info.Outputfile
+		}
 	}
 	if Info.IsSave == true {
 		IsSave = false
@@ -116,6 +120,10 @@ func ParseScantype(Info *HostInfo) {
 			switch Info.Scantype {
 			case "webtitle":
 				Info.Ports = "80,81,443,7001,8000,8080,8089,9200"
+			case "ms17010":
+				Info.Ports = "445"
+			case "cve20200796":
+				Info.Ports = "445"
 			case "portscan":
 			default:
 				port, _ := PORTList[Info.Scantype]
@@ -132,3 +140,18 @@ func CheckErr(text string, err error) {
 		os.Exit(0)
 	}
 }
+
+func getpath() string {
+	filename := os.Args[0]
+	var path string
+	if strings.Contains(filename, "/") {
+		tmp := strings.Split(filename, `/`)
+		tmp[len(tmp)-1] = ``
+		path = strings.Join(tmp, `/`)
+	} else if strings.Contains(filename, `\`) {
+		tmp := strings.Split(filename, `\`)
+		tmp[len(tmp)-1] = ``
+		path = strings.Join(tmp, `\`)
+	}
+	return path
+}

common/config.go

@@ -29,6 +29,7 @@ var PORTList = map[string]int{
 	"findnet":     135,
 	"all":         0,
 	"portscan":    0,
+	"icmp":        0,
 }
 
 var PortlistBack = map[string]int{
@@ -48,9 +49,11 @@ var PortlistBack = map[string]int{
 	"findnet":     135,
 	"all":         0,
 	"portscan":    0,
+	"icmp":        0,
 }
 
-var Outputfile = "result.txt"
+var Outputfile = getpath() + "result.txt"
+
 var IsSave = true
 
 var DefaultPorts = "21,22,80,81,135,443,445,1433,3306,5432,6379,7001,8000,8080,8089,9200,11211,27017"

common/flag.go

@@ -23,7 +23,7 @@ func Flag(Info *HostInfo) {
 	flag.StringVar(&Info.Ports, "p", DefaultPorts, "Select a port,for example: 22 | 1-65535 | 22,80,3306")
 	flag.StringVar(&Info.Command, "c", "", "exec command (ssh)")
 	flag.IntVar(&Info.Threads, "t", 200, "Thread nums")
-	flag.IntVar(&Info.IcmpThreads, "it", 11000, "Icmp Threads nums")
+	flag.IntVar(&Info.IcmpThreads, "it", 3300, "Icmp Threads nums")
 	flag.BoolVar(&Info.Isping, "np", false, "not to ping")
 	flag.BoolVar(&Info.Ping, "ping", false, "using ping replace icmp")
 	flag.BoolVar(&Info.IsSave, "no", false, "not to save output log")