add browserPath flag

Author: testtesttest704

cmd/root.go

@@ -21,14 +21,16 @@ var (
 	threads    int
 	configPath string
 	outputFile string
+	browserPath string
 )
 
 func init() {
 	rootCmd.Flags().StringVarP(&singleURL, "url", "u", "", "Single target URL to scan (e.g. https://example.com)")
 	rootCmd.Flags().StringVarP(&listFile, "list", "l", "", "File containing target URLs (one per line)")
 	rootCmd.Flags().IntVarP(&threads, "threads", "t", 20, "Number of concurrent threads for scanning")
-	rootCmd.Flags().StringVar(&configPath, "config", "config/config.yaml", "Path to config file (e.g. config.yaml)")
+	rootCmd.Flags().StringVarP(&configPath, "config", "c", "config/config.yaml", "Path to config file (e.g. config.yaml)")
 	rootCmd.Flags().StringVarP(&outputFile, "output", "o", "", "Output file (supports .txt, .csv, .json)")
+	rootCmd.Flags().StringVarP(&browserPath, "browser", "b", "", "Path to Chrome/Chromium executable (optional). If not set, will use Rod's default.")
 }
 
 var rootCmd = &cobra.Command{
@@ -57,7 +59,7 @@ var rootCmd = &cobra.Command{
 		var toParse []string // 所有捕获的链接放入 toParse
 
 		// 	2.1 收集加载某个目标 url 后（使用无头浏览器），默认加载的所有其他链接（js等）并放入 toParse
-		err := crawler.CollectLinks(urls, threads, uniqueLinks, &toParse)
+		err := crawler.CollectLinks(urls, threads, uniqueLinks, &toParse, browserPath)
 		if err != nil {
 			log.Fatalf("[!] Error collecting links: %v", err)
 		}

internal/crawler/crawler.go

@@ -1,7 +1,7 @@
 package crawler
 
 import (
-	"SecureJS/internal/utils" // 你自己项目里的包路径，如有不同需改
+	"SecureJS/internal/utils"
 	"fmt"
 	"log"
 	"strings"
@@ -12,8 +12,6 @@ import (
 	"github.com/go-rod/rod/lib/launcher"
 	"github.com/go-rod/rod/lib/proto"
 
-	// 确保版本更新到包含 `Inject` 的版本
-	//"github.com/go-rod/stealth"
 )
 
 type CrawlResult struct {
@@ -25,15 +23,21 @@ type CrawlResult struct {
 // -----------------------------------------------------------
 // 并发爬取多个链接
 // -----------------------------------------------------------
-func crawlAll(urls []string, concurrency int) ([]*CrawlResult, error) {
+func crawlAll(urls []string, concurrency int, browserPath string) ([]*CrawlResult, error) {
 	if len(urls) == 0 {
 		return nil, fmt.Errorf("no URLs provided")
 	}
 	if concurrency <= 0 {
 		concurrency = 1
 	}
 
-	chromePath := launcher.NewBrowser().MustGet()
+	var chromePath string
+	if browserPath != "" {
+		chromePath = browserPath
+	} else {
+		chromePath = launcher.NewBrowser().MustGet()
+	}
+
 	u := launcher.New().
 		Bin(chromePath).
 		Headless(true). // 调试时可设置为 false
@@ -181,8 +185,8 @@ func tryFetchOneURL(browser *rod.Browser, url string, timeout time.Duration) (*C
 // -----------------------------------------------------------
 // 对外的接口，用于收集
 // -----------------------------------------------------------
-func CollectLinks(urls []string, threads int, uniqueLinks map[string]struct{}, toParse *[]string) error {
-	results, err := crawlAll(urls, threads)
+func CollectLinks(urls []string, threads int, uniqueLinks map[string]struct{}, toParse *[]string, browserPath string) error {
+	results, err := crawlAll(urls, threads, browserPath)
 	if err != nil {
 		return fmt.Errorf("failed to crawl: %v", err)
 	}

main.go

@@ -1,12 +1,23 @@
 package main
 
 import (
-	"SecureJS/cmd"
-	"log"
+    "fmt"
+    "os"
+    "runtime/debug"
+
+    "SecureJS/cmd"
 )
 
 func main() {
-	if err := cmd.Execute(); err != nil {
-		log.Fatal(err)
-	}
-}
+    debug.SetTraceback("none")
+    
+    // 2) 在最顶层拦截 panic
+    defer func() {
+        if r := recover(); r != nil {
+            fmt.Fprintf(os.Stderr, "[!] An error occurred: %v\n", r)
+            os.Exit(1)
+        }
+    }()
+
+    cmd.Execute()
+}