realloc/frostfs-api-go
1
0
Fork 0
forked from TrueCloudLab/frostfs-api-go
Code Issues Pull requests Projects Releases Packages Wiki Activity

[#77] protogen: Initial implementation

Browse source 
Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
This commit is contained in:
Evgenii Stratonikov 2024-04-25 15:13:10 +03:00
parent a28ceb251a
commit adb7c602d7
115 changed files with 36376 additions and 20397 deletions
Download patch file Download diff file Expand all files Collapse all files

69
util/protogen/internalgengo/fuzz.go Normal file
View file

@ -0,0 +1,69 @@
package internalgengo
import (
"google.golang.org/protobuf/compiler/protogen"
)
var testingPackage = protogen.GoImportPath("testing")
func GenerateFuzzTests(gen *protogen.Plugin, file *protogen.File) {
{
filename := file.GeneratedFilenamePrefix + "_frostfs_fuzz.go"
g := gen.NewGeneratedFile(filename, file.GoImportPath)
g.P("//go:build gofuzz")
g.P("// +build gofuzz")
g.P("// Code generated by protoc-gen-go-frostfs. DO NOT EDIT.")
g.P()
g.P("package ", file.GoPackageName)
g.P()
for _, msg := range file.Messages {
emitFuzzWrappers(g, msg)
}
}
{
filename := file.GeneratedFilenamePrefix + "_frostfs_test.go"
g := gen.NewGeneratedFile(filename, file.GoImportPath)
g.P("//go:build gofuzz")
g.P("// +build gofuzz")
g.P("// Code generated by protoc-gen-go-frostfs. DO NOT EDIT.")
g.P()
g.P("package ", file.GoPackageName)
g.P()
for _, msg := range file.Messages {
emitFuzzTests(g, msg)
}
}
}
func emitFuzzWrappers(g *protogen.GeneratedFile, msg *protogen.Message) {
g.P("func DoFuzzProto", msg.GoIdent.GoName, "(data []byte) int {")
g.P("msg := new(", msg.GoIdent.GoName, ")")
g.P("if err := msg.UnmarshalProtobuf(data); err != nil { return 0 }")
g.P("_ = msg.MarshalProtobuf(nil)")
g.P("return 1")
g.P("}")
g.P("func DoFuzzJSON", msg.GoIdent.GoName, "(data []byte) int {")
g.P("msg := new(", msg.GoIdent.GoName, ")")
g.P("if err := msg.UnmarshalJSON(data); err != nil { return 0 }")
g.P("_, err := msg.MarshalJSON()")
g.P("if err != nil { panic(err) }")
g.P("return 1")
g.P("}")
}
func emitFuzzTests(g *protogen.GeneratedFile, msg *protogen.Message) {
g.P("func FuzzProto", msg.GoIdent.GoName, "(f *", testingPackage.Ident("F"), ") {")
g.P("f.Fuzz(func(t *", testingPackage.Ident("T"), ", data []byte) {")
g.P("DoFuzzProto", msg.GoIdent.GoName, "(data)")
g.P("})}")
g.P("func FuzzJSON", msg.GoIdent.GoName, "(f *", testingPackage.Ident("F"), ") {")
g.P("f.Fuzz(func(t *", testingPackage.Ident("T"), ", data []byte) {")
g.P("DoFuzzJSON", msg.GoIdent.GoName, "(data)")
g.P("})}")
}