From 3b10fcd9470d0e2aa0d70cb68f285b8f69c966af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=87=BB=E5=8A=BF=E5=AE=89=E5=85=A8=E5=9B=A2=E9=98=9F?= <54792624+JishiTeam-J1wa@users.noreply.github.com> Date: Tue, 24 Feb 2026 14:32:59 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E7=BB=9F=E4=B8=80=E6=8E=A7?= =?UTF-8?q?=E5=88=B6=E5=8F=B0=E8=BE=93=E5=87=BA=E9=A3=8E=E6=A0=BC=EF=BC=8C?= =?UTF-8?q?=E4=BD=BF=E7=94=A8=E7=BB=9F=E4=B8=80=E7=9A=84=E6=97=A5=E5=BF=97?= =?UTF-8?q?=E5=87=BD=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 将 Plugins/MS17010-Exp.go 中的 fmt.Println 替换为 Common.LogError - 将 WebScan/lib/Client.go 中的 POC 加载错误输出替换为 Common.LogError - 将 WebScan/lib/Eval.go 中的调试输出替换为 Common.LogDebug 所有控制台输出现在通过统一的日志系统,确保格式一致性和可维护性。 Co-Authored-By: Claude --- Plugins/MS17010-Exp.go | 2 +- WebScan/lib/Client.go | 12 ++++++------ WebScan/lib/Eval.go | 2 +- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/Plugins/MS17010-Exp.go b/Plugins/MS17010-Exp.go index b48f329b..52d0ad88 100644 --- a/Plugins/MS17010-Exp.go +++ b/Plugins/MS17010-Exp.go @@ -70,7 +70,7 @@ func MS17010EXP(info *Common.HostInfo) { // 验证shellcode有效性 if len(sc) < 20 { - fmt.Println("无效的Shellcode") + Common.LogError("无效的Shellcode") return } diff --git a/WebScan/lib/Client.go b/WebScan/lib/Client.go index 73c22e83..4bc05295 100644 --- a/WebScan/lib/Client.go +++ b/WebScan/lib/Client.go @@ -256,7 +256,7 @@ func LoadMultiPoc(Pocs embed.FS, pocname string) []*Poc { if p, err := LoadPoc(f, Pocs); err == nil { pocs = append(pocs, p) } else { - fmt.Printf("POC加载失败 %s: %v\n", f, err) + Common.LogError(fmt.Sprintf("POC加载失败 %s: %v", f, err)) } } return pocs @@ -268,14 +268,14 @@ func LoadPoc(fileName string, Pocs embed.FS) (*Poc, error) { // 读取POC文件内容 yamlFile, err := Pocs.ReadFile("pocs/" + fileName) if err != nil { - fmt.Printf("POC文件读取失败 %s: %v\n", fileName, err) + Common.LogError(fmt.Sprintf("POC文件读取失败 %s: %v", fileName, err)) return nil, err } // 解析YAML内容 err = yaml.Unmarshal(yamlFile, p) if err != nil { - fmt.Printf("POC解析失败 %s: %v\n", fileName, err) + Common.LogError(fmt.Sprintf("POC解析失败 %s: %v", fileName, err)) return nil, err } return p, err @@ -285,7 +285,7 @@ func LoadPoc(fileName string, Pocs embed.FS) (*Poc, error) { func SelectPoc(Pocs embed.FS, pocname string) []string { entries, err := Pocs.ReadDir("pocs") if err != nil { - fmt.Printf("读取POC目录失败: %v\n", err) + Common.LogError(fmt.Sprintf("读取POC目录失败: %v", err)) } var foundFiles []string @@ -304,14 +304,14 @@ func LoadPocbyPath(fileName string) (*Poc, error) { // 读取POC文件内容 data, err := os.ReadFile(fileName) if err != nil { - fmt.Printf("POC文件读取失败 %s: %v\n", fileName, err) + Common.LogError(fmt.Sprintf("POC文件读取失败 %s: %v", fileName, err)) return nil, err } // 解析YAML内容 err = yaml.Unmarshal(data, p) if err != nil { - fmt.Printf("POC解析失败 %s: %v\n", fileName, err) + Common.LogError(fmt.Sprintf("POC解析失败 %s: %v", fileName, err)) return nil, err } return p, err diff --git a/WebScan/lib/Eval.go b/WebScan/lib/Eval.go index 79406a41..59038688 100644 --- a/WebScan/lib/Eval.go +++ b/WebScan/lib/Eval.go @@ -620,7 +620,7 @@ func reverseCheck(r *Reverse, timeout int64) bool { isOK := bytes.Contains(resp.Body, []byte(`"message": "OK"`)) if hasData && isOK { - fmt.Println(apiURL) + Common.LogDebug(apiURL) return true } return false