From f37414e7d7170aa4f0078f993b60e7bb9f0eb58b Mon Sep 17 00:00:00 2001
From: M09Ic <m09ician@gmail.com>
Date: Wed, 4 Jan 2023 13:52:03 +0800
Subject: [PATCH] =?UTF-8?q?probe=E4=B8=AD=E6=B7=BB=E5=8A=A0source.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/baseline.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/pkg/baseline.go b/pkg/baseline.go
index c5c7639..b7c1d7c 100644
--- a/pkg/baseline.go
+++ b/pkg/baseline.go
@@ -13,6 +13,29 @@ import (
 	"strings"
 )
 
+func GetSourceName(s int) string {
+	switch s {
+	case 1:
+		return "check"
+	case 2:
+		return "index"
+	case 3:
+		return "random"
+	case 4:
+		return "redirect"
+	case 5:
+		return "crawl"
+	case 6:
+		return "active"
+	case 7:
+		return "word"
+	case 8:
+		return "waf"
+	default:
+		return "unknown"
+	}
+}
+
 func NewBaseline(u, host string, resp *ihttp.Response) *Baseline {
 	bl := &Baseline{
 		UrlString: u,
@@ -216,6 +239,8 @@ func (bl *Baseline) Get(key string) string {
 		return strconv.Itoa(bl.Status)
 	case "spend":
 		return strconv.Itoa(int(bl.Spended))
+	case "source":
+		return GetSourceName(bl.Source)
 	case "extract":
 		return bl.Extracteds.String()
 	case "frame", "framework":