refactor: 输出格式重构,去掉所有插件的多线程,因为多线程会导致结果不准确,加入进度条

This commit is contained in:
ZacharyZcR
2025-01-01 07:18:36 +08:00
parent 277ea5d332
commit ceede3cd68
49 changed files with 1448 additions and 2911 deletions
+1 -1
View File
@@ -58,7 +58,7 @@ func InfoCheck(Url string, CheckData *[]CheckDatas) []string {
// 输出结果
if len(matchedInfos) > 0 {
result := fmt.Sprintf("[+] 发现指纹 目标: %-25v 指纹: %s", Url, matchedInfos)
result := fmt.Sprintf("发现指纹 目标: %-25v 指纹: %s", Url, matchedInfos)
Common.LogSuccess(result)
return matchedInfos
}
+2 -2
View File
@@ -83,7 +83,7 @@ func initpoc() {
}
} else {
// 从指定目录加载POC
Common.LogSuccess(fmt.Sprintf("[*] 从目录加载POC: %s", Common.PocPath))
Common.LogSuccess(fmt.Sprintf("从目录加载POC: %s", Common.PocPath))
err := filepath.Walk(Common.PocPath, func(path string, info os.FileInfo, err error) error {
if err != nil || info == nil {
return err
@@ -98,7 +98,7 @@ func initpoc() {
})
if err != nil {
Common.LogError(fmt.Sprintf("[-] 加载外部POC失败: %v", err))
Common.LogError(fmt.Sprintf("加载外部POC失败: %v", err))
}
}
}
+6 -6
View File
@@ -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)
fmt.Printf("POC加载失败 %s: %v\n", 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)
fmt.Printf("POC文件读取失败 %s: %v\n", fileName, err)
return nil, err
}
// 解析YAML内容
err = yaml.Unmarshal(yamlFile, p)
if err != nil {
fmt.Printf("[-] POC解析失败 %s: %v\n", fileName, err)
fmt.Printf("POC解析失败 %s: %v\n", 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)
fmt.Printf("读取POC目录失败: %v\n", 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)
fmt.Printf("POC文件读取失败 %s: %v\n", fileName, err)
return nil, err
}
// 解析YAML内容
err = yaml.Unmarshal(data, p)
if err != nil {
fmt.Printf("[-] POC解析失败 %s: %v\n", fileName, err)
fmt.Printf("POC解析失败 %s: %v\n", fileName, err)
return nil, err
}
return p, err