mirror of
https://github.com/reactos/syzkaller.git
synced 2024-11-27 05:10:43 +00:00
80d43738f1
We will need a wrapper for target.SanitizeCall that will do more than just calling the target-provided function. To avoid confusion and potential mistakes, give the target function and prog function different names. Prog package will continue to call this "sanitize", which will include target's "neutralize" + more. Also refactor API a bit: we need a helper function that sanitizes the whole program because that's needed most of the time. Fixes #477 Fixes #502
36 lines
1.0 KiB
Go
36 lines
1.0 KiB
Go
// Copyright 2015 syzkaller project authors. All rights reserved.
|
|
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
|
|
|
|
package prog
|
|
|
|
import (
|
|
"math/rand"
|
|
)
|
|
|
|
// Generate generates a random program with ncalls calls.
|
|
// ct contains a set of allowed syscalls, if nil all syscalls are used.
|
|
func (target *Target) Generate(rs rand.Source, ncalls int, ct *ChoiceTable) *Prog {
|
|
p := &Prog{
|
|
Target: target,
|
|
}
|
|
r := newRand(target, rs)
|
|
s := newState(target, ct, nil)
|
|
for len(p.Calls) < ncalls {
|
|
calls := r.generateCall(s, p, len(p.Calls))
|
|
for _, c := range calls {
|
|
s.analyze(c)
|
|
p.Calls = append(p.Calls, c)
|
|
}
|
|
}
|
|
// For the last generated call we could get additional calls that create
|
|
// resources and overflow ncalls. Remove some of these calls.
|
|
// The resources in the last call will be replaced with the default values,
|
|
// which is exactly what we want.
|
|
for len(p.Calls) > ncalls {
|
|
p.removeCall(ncalls - 1)
|
|
}
|
|
p.sanitizeFix()
|
|
p.debugValidate()
|
|
return p
|
|
}
|