diff options
| -rw-r--r-- | classic.go | 6 | ||||
| -rw-r--r-- | main.go | 2 |
2 files changed, 6 insertions, 2 deletions
@@ -8,7 +8,7 @@ import ( "sync" ) -func classic(startURL url.URL, maxConcurrency, maxURLs int) { +func classic(startURL url.URL, maxConcurrency, maxURLs, maxDepth int) { worklist := make(chan []packet) var numPendingSends int @@ -40,6 +40,10 @@ loop: break loop } + if p.depth == maxDepth { + continue + } + numPendingSends++ wg.Go(func() { sema <- struct{}{} @@ -41,7 +41,7 @@ func main() { } getLeakProfile(func() { - classic(*startURL, *maxConcurrency, *maxURLs) + classic(*startURL, *maxConcurrency, *maxURLs, 1) }) } |
