63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
// Copyright 2023 The go-fuzz-headers Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package gofuzzheaders
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
)
|
|
|
|
type Continue struct {
|
|
F *ConsumeFuzzer
|
|
}
|
|
|
|
func (f *ConsumeFuzzer) AddFuncs(fuzzFuncs []interface{}) {
|
|
for i := range fuzzFuncs {
|
|
v := reflect.ValueOf(fuzzFuncs[i])
|
|
if v.Kind() != reflect.Func {
|
|
panic("Need only funcs!")
|
|
}
|
|
t := v.Type()
|
|
if t.NumIn() != 2 || t.NumOut() != 1 {
|
|
fmt.Println(t.NumIn(), t.NumOut())
|
|
|
|
panic("Need 2 in and 1 out params. In must be the type. Out must be an error")
|
|
}
|
|
argT := t.In(0)
|
|
switch argT.Kind() {
|
|
case reflect.Ptr, reflect.Map:
|
|
default:
|
|
panic("fuzzFunc must take pointer or map type")
|
|
}
|
|
if t.In(1) != reflect.TypeOf(Continue{}) {
|
|
panic("fuzzFunc's second parameter must be type Continue")
|
|
}
|
|
f.Funcs[argT] = v
|
|
}
|
|
}
|
|
|
|
func (f *ConsumeFuzzer) GenerateWithCustom(targetStruct interface{}) error {
|
|
e := reflect.ValueOf(targetStruct).Elem()
|
|
return f.fuzzStruct(e, true)
|
|
}
|
|
|
|
func (c Continue) GenerateStruct(targetStruct interface{}) error {
|
|
return c.F.GenerateStruct(targetStruct)
|
|
}
|
|
|
|
func (c Continue) GenerateStructWithCustom(targetStruct interface{}) error {
|
|
return c.F.GenerateWithCustom(targetStruct)
|
|
}
|