修复bug

go/middleware/security_probe_test.go

@@ -0,0 +1,92 @@
+package middleware
+
+import "testing"
+
+func TestShouldSuppressAccessLogMsg(t *testing.T) {
+	probeLine := "|      127.0.0.1| 404 |    221.723µs| nomatch| GET      /middleware/security_probe.go"
+	if !shouldSuppressAccessLogMsg(probeLine) {
+		t.Fatalf("probe access line should be suppressed")
+	}
+
+	rootLine := "|      127.0.0.1| 404 |    219.710µs| nomatch| GET      /"
+	if !shouldSuppressAccessLogMsg(rootLine) {
+		t.Fatalf("root 404 access line should be suppressed")
+	}
+
+	normalLine := "|      127.0.0.1| 200 |      1.649ms|   match| GET      /auth/login   r:/auth/login"
+	if shouldSuppressAccessLogMsg(normalLine) {
+		t.Fatalf("normal access line should not be suppressed")
+	}
+}
+
+func TestIsScannerProbe(t *testing.T) {
+	probes := []string{
+		"/config.js",
+		"/config.json",
+		"/config.yaml",
+		"/config.php",
+		"/config/config.js",
+		"/database.sql",
+		"/.env",
+		"/demo/.env",
+		"/sendgrid.env",
+		"/config.env",
+		"/.env.lithic",
+		"/credentials",
+		"/credentials.json",
+		"/credentials.ini",
+		"/secrets.json",
+		"/lithic.json",
+		"/sardine.json",
+		"/phpinfo",
+		"/phpinfo.php",
+		"/info.php",
+		"/wp-config.php",
+		"/wp-config.php.bak",
+		"/wp-content/debug.log",
+		"/wp-login.php",
+		"/vendor/composer/installed.json",
+		"/docker-compose.yml",
+		"/composer.json",
+		"/gitconfig",
+		"/runtime-config.js",
+		"/settings.json",
+		"/index.js",
+		"/config/aws.yml",
+		"/v1/.env",
+		"/demo/../.env",
+		"/middleware",
+		"/middleware/security_probe.go",
+		"/controllers/backend_reimburse.go",
+		"/routers",
+		"/routers/spa.go",
+		"/models/reimburse.go",
+		"/services/approval/engine.go",
+		"/scripts/uidmigrate/migrate_uid.go",
+		"/conf/app.conf",
+	}
+	for _, p := range probes {
+		if !IsScannerProbe(p) {
+			t.Errorf("IsScannerProbe(%q) = false, want true", p)
+		}
+	}
+
+	normal := []string{
+		"/",
+		"/favicon.ico",
+		"/backend/reimbursements",
+		"/platform/menu/1",
+		"/site/setting",
+		"/site/news/1",
+		"/uploads/logo.png",
+		"/themes/default/index.html",
+		"/static/js/app.js",
+		"/news",
+		"/page/about",
+	}
+	for _, p := range normal {
+		if IsScannerProbe(p) {
+			t.Errorf("IsScannerProbe(%q) = true, want false", p)
+		}
+	}
+}