From f2239b6c9f0efbacb37e39584cfa218b17c4fc31 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BD=B1=E8=88=9E=E8=80=85?= <shadow1ng@qq.com>
Date: Sun, 3 Jul 2022 23:48:06 +0800
Subject: [PATCH] =?UTF-8?q?=E5=87=8F=E5=B0=91pocinfo=E7=BB=93=E6=9E=84?=
 =?UTF-8?q?=E4=BD=93=E5=A4=A7=E5=B0=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Plugins/webtitle.go   |  2 +-
 WebScan/WebScan.go    |  6 +++---
 WebScan/lib/client.go |  2 +-
 common/config.go      | 22 ++++++++++------------
 common/flag.go        |  4 ++--
 5 files changed, 17 insertions(+), 19 deletions(-)

diff --git a/Plugins/webtitle.go b/Plugins/webtitle.go
index 39de424..856f7d3 100644
--- a/Plugins/webtitle.go
+++ b/Plugins/webtitle.go
@@ -105,7 +105,7 @@ func geturl(info *common.HostInfo, flag int, CheckData []WebScan.CheckDatas) (er
 	req.Header.Set("User-agent", "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/28.0.1468.0 Safari/537.36")
 	req.Header.Set("Accept", "*/*")
 	req.Header.Set("Accept-Language", "zh-CN,zh;q=0.9")
-	req.Header.Set("Cookie", common.Pocinfo.Cookie)
+	req.Header.Set("Cookie", common.Cookie)
 	//if common.Pocinfo.Cookie != "" {
 	//	req.Header.Set("Cookie", "rememberMe=1;"+common.Pocinfo.Cookie)
 	//} else {
diff --git a/WebScan/WebScan.go b/WebScan/WebScan.go
index d6eac26..9a2e873 100644
--- a/WebScan/WebScan.go
+++ b/WebScan/WebScan.go
@@ -41,11 +41,11 @@ func Execute(PocInfo common.PocInfo) {
 		return
 	}
 	req.Header.Set("User-agent", "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/28.0.1468.0 Safari/537.36")
-	if PocInfo.Cookie != "" {
-		req.Header.Set("Cookie", PocInfo.Cookie)
+	if common.Cookie != "" {
+		req.Header.Set("Cookie", common.Cookie)
 	}
 	pocs := filterPoc(PocInfo.PocName)
-	lib.CheckMultiPoc(req, pocs, PocInfo.Num)
+	lib.CheckMultiPoc(req, pocs, common.PocNum)
 }
 
 func initpoc() {
diff --git a/WebScan/lib/client.go b/WebScan/lib/client.go
index 4ccb37a..90538a9 100644
--- a/WebScan/lib/client.go
+++ b/WebScan/lib/client.go
@@ -23,7 +23,7 @@ var (
 
 func Inithttp(PocInfo common.PocInfo) {
 	//PocInfo.Proxy = "http://127.0.0.1:8080"
-	err := InitHttpClient(PocInfo.Num, common.Proxy, time.Duration(common.WebTimeout)*time.Second)
+	err := InitHttpClient(common.PocNum, common.Proxy, time.Duration(common.WebTimeout)*time.Second)
 	if err != nil {
 		log.Fatal(err)
 	}
diff --git a/common/config.go b/common/config.go
index a04e06b..233db35 100644
--- a/common/config.go
+++ b/common/config.go
@@ -52,17 +52,8 @@ type HostInfo struct {
 }
 
 type PocInfo struct {
-	Num        int
-	Rate       int
-	PocName    string
-	PocDir     string
-	Target     string
-	TargetFile string
-	RawFile    string
-	Cookie     string
-	ForceSSL   bool
-	ApiKey     string
-	CeyeDomain string
+	Target  string
+	PocName string
 }
 
 var (
@@ -103,5 +94,12 @@ var (
 	BruteThread   int
 	LiveTop       int
 	Socks5Proxy   string
-	PocFull       bool
+)
+
+var (
+	PocNum     int
+	PocFull    bool
+	CeyeDomain string
+	ApiKey     string
+	Cookie     string
 )
diff --git a/common/flag.go b/common/flag.go
index 498b02b..4923614 100644
--- a/common/flag.go
+++ b/common/flag.go
@@ -57,9 +57,9 @@ func Flag(Info *HostInfo) {
 	flag.StringVar(&Pocinfo.PocName, "pocname", "", "use the pocs these contain pocname, -pocname weblogic")
 	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")
-	flag.StringVar(&Pocinfo.Cookie, "cookie", "", "set poc cookie,-cookie rememberMe=login")
+	flag.StringVar(&Cookie, "cookie", "", "set poc cookie,-cookie rememberMe=login")
 	flag.Int64Var(&WebTimeout, "wt", 5, "Set web timeout")
-	flag.IntVar(&Pocinfo.Num, "num", 20, "poc rate")
+	flag.IntVar(&PocNum, "num", 20, "poc rate")
 	flag.StringVar(&SC, "sc", "", "ms17 shellcode,as -sc add")
 	flag.Parse()
 }