From a939f6c8ea4e7d493aa282403e1f39a12604484d Mon Sep 17 00:00:00 2001 From: demo Date: Wed, 27 May 2026 12:18:56 -0400 Subject: feat: configure maxDepth from the command line Similar to maxURLs, a maxDepth of zero means no limit. --- main.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'main.go') diff --git a/main.go b/main.go index 5673441..311cece 100644 --- a/main.go +++ b/main.go @@ -16,6 +16,7 @@ func main() { maxConcurrency := flag.Int("c", 0, "Maximum number of concurrent queue pushes") startRawURL := flag.String("url", "", "Entry-point URL") maxURLs := flag.Int("max", 0, "Maximum number of URLs to collect (omitted or 0 means no limit)") + maxDepth := flag.Int("depth", 0, "Maximum URL depth (omitted or 0 means no limit)") flag.Parse() @@ -35,13 +36,17 @@ func main() { log.Fatalf("Invalid -max argument: %d", *maxURLs) } + if *maxDepth < 0 { + log.Fatalf("Invalid -depth argument: %d", *maxDepth) + } + startURL, err := url.Parse(*startRawURL) if err != nil { log.Fatal(err) } getLeakProfile(func() { - classic(*startURL, *maxConcurrency, *maxURLs, 1) + classic(*startURL, *maxConcurrency, *maxURLs, *maxDepth) }) } -- cgit v1.2.3