新增output导出功能

MornS0 · Feb 19, 2021 · f6b77d1 · f6b77d1
1 parent 67a5211
commit f6b77d1
Show file tree

Hide file tree

Showing 5 changed files with 37 additions and 15 deletions.
diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/app/params/check.go b/app/params/check.go
@@ -1,11 +1,32 @@
 package params
 
 import (
+	"../../lib/misc"
 	"fmt"
 	"os"
 	"regexp"
 )
 
+var OutPutFile *os.File
+
+func initOutPutFile(path string) *os.File {
+	if misc.FileIsExist(path) {
+		f, err := os.Open(path)
+		if err != nil {
+			fmt.Printf("[-]%s", err.Error())
+			os.Exit(0)
+		}
+		return f
+	} else {
+		f, err := os.Create(path)
+		if err != nil {
+			fmt.Printf("[-]%s", err.Error())
+			os.Exit(0)
+		}
+		return f
+	}
+}
+
 func checkParams() {
 	//判断冲突参数
 	if params.port != "" && params.top != 0 {
@@ -28,6 +49,8 @@ func checkParams() {
 		}
 		if params.output != "" {
 			//验证output参数
+			f := initOutPutFile(params.output)
+			OutPutFile = f
 		}
 		if params.proxy != "" {
 			if !checkProxyParam(params.proxy) {

diff --git a/lib/port/banner.go b/lib/port/banner.go
@@ -159,6 +159,9 @@ func makeResultInfo(res portInfo) string {
 		Banner = res.Banner[i : i+30]
 	}
 	res.Info = fmt.Sprintf("\r[+]%s\t%s\t%s\t%s\n", res.Url, res.Title, Banner, strings.Join(infoArr, ","))
+	if params.OutPutFile != nil {
+		_, _ = params.OutPutFile.WriteString(fmt.Sprintf("\r[+]%s\t%s\t%s\t%s", res.Url, res.Title, Banner, strings.Join(infoArr, ",")))
+	}
 	return res.Info
 }
 

diff --git a/lib/shttp/shttp.go b/lib/shttp/shttp.go
@@ -62,7 +62,7 @@ var headerKeys = map[string]string{
 	"Entry1":                           "不知道",
 	"Accept-Encoding":                  "不知道",
 	"X-Amz-Request-Id":                 "不知道",
-	"Cfl_asynch":                       "浙江大华dh650平台设备",
+	"Cfl_asynch":                       "keyword:浙江大华dh650平台设备",
 }
 
 var newHeaderKeys *os.File