bugfix

common/runner/runner.go

@@ -519,13 +519,15 @@ func (r *Runner) writeResult(f *os.File, result HttpResult) {
 				fpString += ":" + fp.Version
 			}
 		}
-		r.Options.Callback(CallbackScanResult{
-			TargetURL:       result.URL,
-			StatusCode:      result.StatusCode,
-			Title:           result.Title,
-			Fingerprint:     fpString,
-			Vulnerabilities: vuls,
-		})
+		if r.Options.Callback != nil {
+			r.Options.Callback(CallbackScanResult{
+				TargetURL:       result.URL,
+				StatusCode:      result.StatusCode,
+				Title:           result.Title,
+				Fingerprint:     fpString,
+				Vulnerabilities: vuls,
+			})
+		}
 	}
 	if len(result.Advisories) > 0 {
 		fmt.Println("\n存在漏洞:")

common/websocket/websocket.go

@@ -61,9 +61,6 @@ func (s *WSServer) handleMessages(conn *websocket.Conn) {
 	for {
 		_, message, err := conn.ReadMessage()
 		if err != nil {
-			if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
-				gologger.Errorf("读取消息错误: %v\n", err)
-			}
 			break
 		}
 		var scanReq ScanRequest
@@ -94,7 +91,10 @@ func (s *WSServer) handleScanRequest(conn *websocket.Conn, req *ScanRequest) {
 	case "netscan":
 		opts.Target = req.Target
 	}
+	mu := sync.Mutex{}
 	processFunc := func(data interface{}) {
+		mu.Lock()
+		defer mu.Unlock()
 		switch v := data.(type) {
 		case runner.CallbackScanResult:
 			s.SendMessage(conn, WSMsgTypeScanResult, v)