延时关闭nucleiRunner，确保结果都发送到服务器 2022-10-05

GhostTroops · Oct 5, 2022 · 0c8be1b · 0c8be1b
1 parent d42caac
commit 0c8be1b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/engine/engine.go b/engine/engine.go
@@ -111,7 +111,7 @@ func init() {
 					})
 				default:
 					util.DoDelayClear()
-					util.DoSleep()
+					//util.DoSleep()
 				}
 			}
 		}()

diff --git a/projectdiscovery/nuclei_Yaml/nuclei_yaml.go b/projectdiscovery/nuclei_Yaml/nuclei_yaml.go
@@ -147,7 +147,8 @@ func RunNuclei(buf *bytes.Buffer, xx chan bool, oOpts *map[string]interface{}, o
 	}
 
 	readConfig(options)
-	options.Targets = strings.Split(strings.TrimSpace(buf.String()), "\n")
+	s001 := strings.TrimSpace(buf.String())
+	options.Targets = strings.Split(s001, "\n")
 	log.Printf("options.Targets = %+v", options.Targets)
 	/////////////////////////////////////
 	options.Verbose = false
@@ -187,7 +188,10 @@ func RunNuclei(buf *bytes.Buffer, xx chan bool, oOpts *map[string]interface{}, o
 		outNuclei <- nil
 		return
 	}
-	defer nucleiRunner.Close()
+	util.RegDelayCbk(s001, func() {
+		nucleiRunner.Close2()
+		nucleiRunner.Close()
+	}, nil, 0, 50)
 	//data, _ := json.Marshal(options)
 	//log.Printf("%+v", string(data))
 	outNuclei <- nucleiRunner
@@ -198,6 +202,7 @@ func RunNuclei(buf *bytes.Buffer, xx chan bool, oOpts *map[string]interface{}, o
 			gologger.Fatal().Msgf("Could not run nuclei: %s\n", err)
 		}
 	}
+
 }
 func readConfig(options *types.Options) {
 	pwd, _ := os.Getwd()