fix katana extension filter

yhy0 · Mar 18, 2024 · 41da6a2 · 41da6a2
1 parent bd14f78
commit 41da6a2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/crawler/katana.go b/crawler/katana.go
@@ -17,9 +17,9 @@ import (
   @desc: //TODO
 **/
 
-// 默认过滤的后缀名
-var extensionFilter = []string{
-    ".css", ".png", ".gif", ".jpg", ".mp4", ".mp3", ".mng", ".pct", ".bmp", ".jpeg", ".pst", ".psp", ".ttf",
+// ExtensionFilter 默认过滤的后缀名
+var ExtensionFilter = []string{
+    ".png", ".gif", ".jpg", ".mp4", ".mp3", ".mng", ".pct", ".bmp", ".jpeg", ".pst", ".psp", ".ttf",
     ".tif", ".tiff", ".ai", ".drw", ".wma", ".ogg", ".wav", ".ra", ".aac", ".mid", ".au", ".aiff",
     ".dxf", ".eps", ".ps", ".svg", ".3gp", ".asf", ".asx", ".avi", ".mov", ".mpg", ".qt", ".rm",
     ".wmv", ".m4a", ".bin", ".xls", ".xlsx", ".ppt", ".pptx", ".doc", ".docx", ".odt", ".ods", ".odg",
@@ -43,7 +43,7 @@ func Katana(target string, headless bool, show bool, out func(result output.Resu
         OnResult:        out,
         Headless:        headless,
         Proxy:           conf.GlobalConfig.Http.Proxy,
-        ExtensionFilter: extensionFilter,
+        ExtensionFilter: ExtensionFilter,
     }
     if options.Headless {
         options.ShowBrowser = show

diff --git a/pkg/mode/active.go b/pkg/mode/active.go
@@ -19,6 +19,7 @@ import (
     "github.com/yhy0/Jie/scan/gadget/waf"
     "github.com/yhy0/logging"
     "net/url"
+    "path"
     "regexp"
     "strings"
     "time"
@@ -135,6 +136,12 @@ func Katana(target string, waf []string, t *task.Task, fingerprint []string) []s
             logging.Logger.Errorln(err)
             return
         }
+        extension := strings.ToLower(path.Ext(parseUrl.Path))
+
+        if util.InSlice(crawler.ExtensionFilter, extension) {
+            return
+        }
+
         logging.Logger.Infof("Katana: [%s] %v %v", result.Request.Method, result.Request.URL, result.Request.Body)
         i++