projectdiscovery/httpx
1
1
Fork 0
mirror of https://github.com/projectdiscovery/httpx.git synced 2024-10-05 13:27:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

using std waitgroup for non-changing output

Browse Source
This commit is contained in:
Mzack9999 2024-06-12 14:57:04 +02:00
parent dece6fce63
commit 746b821388
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
runner/runner.go
View File

@ -22,6 +22,7 @@ import (
"sort"
"strconv"
"strings"
"sync"
"time"
"golang.org/x/exp/maps"
@ -692,12 +693,12 @@ func (r *Runner) RunEnumeration() {
}
// output routine
wgoutput, _ := syncutil.New(syncutil.WithSize(2))
wgoutput.Add()
var wgoutput sync.WaitGroup
output := make(chan Result)
nextStep := make(chan Result)
wgoutput.Add(1)
go func(output chan Result, nextSteps ...chan Result) {
defer wgoutput.Done()
@ -1077,7 +1078,7 @@ func (r *Runner) RunEnumeration() {
// HTML Summary
// - needs output of previous routine
// - separate goroutine due to incapability of go templates to render from file
wgoutput.Add()
wgoutput.Add(1)
go func(output chan Result) {
defer wgoutput.Done()
@ -1241,7 +1242,7 @@ func (r *Runner) Process(t string, wg *syncutil.AdaptiveWaitGroup, protocol stri
}
func (r *Runner) process(t string, wg *syncutil.AdaptiveWaitGroup, hp *httpx.HTTPX, protocol string, scanopts *ScanOptions, output chan Result) {
// attempts to set the workpool size to the number of threads
if r.options.Threads > 0 && wg.Size != r.options.Threads {
if err := wg.Resize(context.Background(), r.options.Threads); err != nil {
gologger.Error().Msgf("Could not resize workpool: %s\n", err)