diff options
Diffstat (limited to 'pkg/fuzzer')
| -rw-r--r-- | pkg/fuzzer/fuzzer.go | 8 | ||||
| -rw-r--r-- | pkg/fuzzer/job.go | 2 |
2 files changed, 9 insertions, 1 deletions
diff --git a/pkg/fuzzer/fuzzer.go b/pkg/fuzzer/fuzzer.go index 0c0119e71..fdfe95518 100644 --- a/pkg/fuzzer/fuzzer.go +++ b/pkg/fuzzer/fuzzer.go @@ -72,6 +72,13 @@ func NewFuzzer(ctx context.Context, cfg *Config, rnd *rand.Rand, return f } +func (fuzzer *Fuzzer) RecommendedCalls() int { + if fuzzer.Config.ModeKFuzzTest { + return prog.RecommendedCallsKFuzzTest + } + return prog.RecommendedCalls +} + type execQueues struct { triageCandidateQueue *queue.DynamicOrderer candidateQueue *queue.PlainQueue @@ -214,6 +221,7 @@ type Config struct { FetchRawCover bool NewInputFilter func(call string) bool PatchTest bool + ModeKFuzzTest bool } func (fuzzer *Fuzzer) triageProgCall(p *prog.Prog, info *flatrpc.CallInfo, call int, triage *map[int]*triageCall) { diff --git a/pkg/fuzzer/job.go b/pkg/fuzzer/job.go index 7f1e47bf6..bbac544f6 100644 --- a/pkg/fuzzer/job.go +++ b/pkg/fuzzer/job.go @@ -43,7 +43,7 @@ func (ji *JobInfo) ID() string { func genProgRequest(fuzzer *Fuzzer, rnd *rand.Rand) *queue.Request { p := fuzzer.target.Generate(rnd, - prog.RecommendedCalls, + fuzzer.RecommendedCalls(), fuzzer.ChoiceTable()) return &queue.Request{ Prog: p, |