运行时的异常处理

Author: SummerSec

cmd/commons/core/runner.go

@@ -106,6 +106,11 @@ func Start(u string, hashmap map[string]interface{}, i int, c chan int) {
 	}()
 	attack.Sevice(target, hashmap)
 
+	defer func() {
+		if errs := recover(); errs != nil {
+			log.Debug("Runner panic: ", errs)
+		}
+	}()
 	// 放到最后，不然无法生效
 	c <- i + 1
 

cmd/test/checkerr.go

@@ -0,0 +1,24 @@
+package main
+
+import "fmt"
+
+func getErr(int1, int2 int) error {
+	defer func() {
+		if err := recover(); err != nil {
+			fmt.Println(err)
+		}
+	}()
+	_ = int1 / int2
+	return nil
+}
+
+func main() {
+	defer func() {
+		if err := recover(); err != nil {
+			fmt.Println(err)
+		}
+	}()
+	_ = getErr(1, 0)
+	fmt.Println("Hello, playground")
+
+}