fix: fetch input

ac0d3r · ac0d3r · commit 061be540e7ff · 2025-07-12T11:50:15.000+08:00
diff --git a/mutator/helpers.go b/mutator/helpers.go
@@ -18,11 +18,17 @@ func (m *Mutator) getFuzzedInput(set string) string {
 }
 
 func (m *Mutator) fetchInput(set string) string {
+	m.mux.RLock()
+	defer m.mux.RUnlock()
+
 	k := m.r.Intn(len(m.inputSets[set]))
 	return m.inputSets[set][k]
 }
 
 func (m *Mutator) addCorpus(set, value string) {
+	m.mux.Lock()
+	defer m.mux.Unlock()
+
 	if set == "" || value == "" {
 		return
 	}
diff --git a/mutator/mutator.go b/mutator/mutator.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"math/rand"
 	"strings"
+	"sync"
 	"time"
 )
 
@@ -40,6 +41,7 @@ func NewMutator(inp, app string, runs uint, fnName string, ignoreCrashes bool, i
 }
 
 type Mutator struct {
+	mux           sync.RWMutex
 	fuzzIdx       int
 	runs          uint
 	baseURL       string

Original file line number	Diff line number	Diff line change
`@@ -18,11 +18,17 @@ func (m *Mutator) getFuzzedInput(set string) string {`
`18`	`18`	`}`
`19`	`19`
`20`	`20`	`func (m *Mutator) fetchInput(set string) string {`
	`21`	`+ m.mux.RLock()`
	`22`	`+ defer m.mux.RUnlock()`
	`23`	`+`
`21`	`24`	`k := m.r.Intn(len(m.inputSets[set]))`
`22`	`25`	`return m.inputSets[set][k]`
`23`	`26`	`}`
`24`	`27`
`25`	`28`	`func (m *Mutator) addCorpus(set, value string) {`
	`29`	`+ m.mux.Lock()`
	`30`	`+ defer m.mux.Unlock()`
	`31`	`+`
`26`	`32`	`if set == "" \|\| value == "" {`
`27`	`33`	`return`
`28`	`34`	`}`
Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,7 @@ import (`
`4`	`4`	`"fmt"`
`5`	`5`	`"math/rand"`
`6`	`6`	`"strings"`
	`7`	`+ "sync"`
`7`	`8`	`"time"`
`8`	`9`	`)`
`9`	`10`
`@@ -40,6 +41,7 @@ func NewMutator(inp, app string, runs uint, fnName string, ignoreCrashes bool, i`
`40`	`41`	`}`
`41`	`42`
`42`	`43`	`type Mutator struct {`
	`44`	`+ mux sync.RWMutex`
`43`	`45`	`fuzzIdx int`
`44`	`46`	`runs uint`
`45`	`47`	`baseURL string`