From 6ef3e47010676e4e159edb346c219e8d30cabadc Mon Sep 17 00:00:00 2001
From: Dmitry Vyukov <dvyukov@google.com>
Date: Thu, 23 May 2024 10:13:32 +0200
Subject: syz-fuzzer: repair leak checking

Notify fuzzer from the manager when corpus triage has finished
to start leak checking.

Fixes #4728
---
 pkg/flatrpc/flatrpc.go | 83 ++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 74 insertions(+), 9 deletions(-)

(limited to 'pkg/flatrpc/flatrpc.go')

diff --git a/pkg/flatrpc/flatrpc.go b/pkg/flatrpc/flatrpc.go
index 2ef9b2595..89172feeb 100644
--- a/pkg/flatrpc/flatrpc.go
+++ b/pkg/flatrpc/flatrpc.go
@@ -89,21 +89,24 @@ func (v Feature) String() string {
 type HostMessagesRaw byte
 
 const (
-	HostMessagesRawNONE         HostMessagesRaw = 0
-	HostMessagesRawExecRequest  HostMessagesRaw = 1
-	HostMessagesRawSignalUpdate HostMessagesRaw = 2
+	HostMessagesRawNONE            HostMessagesRaw = 0
+	HostMessagesRawExecRequest     HostMessagesRaw = 1
+	HostMessagesRawSignalUpdate    HostMessagesRaw = 2
+	HostMessagesRawStartLeakChecks HostMessagesRaw = 3
 )
 
 var EnumNamesHostMessagesRaw = map[HostMessagesRaw]string{
-	HostMessagesRawNONE:         "NONE",
-	HostMessagesRawExecRequest:  "ExecRequest",
-	HostMessagesRawSignalUpdate: "SignalUpdate",
+	HostMessagesRawNONE:            "NONE",
+	HostMessagesRawExecRequest:     "ExecRequest",
+	HostMessagesRawSignalUpdate:    "SignalUpdate",
+	HostMessagesRawStartLeakChecks: "StartLeakChecks",
 }
 
 var EnumValuesHostMessagesRaw = map[string]HostMessagesRaw{
-	"NONE":         HostMessagesRawNONE,
-	"ExecRequest":  HostMessagesRawExecRequest,
-	"SignalUpdate": HostMessagesRawSignalUpdate,
+	"NONE":            HostMessagesRawNONE,
+	"ExecRequest":     HostMessagesRawExecRequest,
+	"SignalUpdate":    HostMessagesRawSignalUpdate,
+	"StartLeakChecks": HostMessagesRawStartLeakChecks,
 }
 
 func (v HostMessagesRaw) String() string {
@@ -127,6 +130,8 @@ func (t *HostMessagesRawT) Pack(builder *flatbuffers.Builder) flatbuffers.UOffse
 		return t.Value.(*ExecRequestRawT).Pack(builder)
 	case HostMessagesRawSignalUpdate:
 		return t.Value.(*SignalUpdateRawT).Pack(builder)
+	case HostMessagesRawStartLeakChecks:
+		return t.Value.(*StartLeakChecksRawT).Pack(builder)
 	}
 	return 0
 }
@@ -139,6 +144,9 @@ func (rcv HostMessagesRaw) UnPack(table flatbuffers.Table) *HostMessagesRawT {
 	case HostMessagesRawSignalUpdate:
 		x := SignalUpdateRaw{_tab: table}
 		return &HostMessagesRawT{Type: HostMessagesRawSignalUpdate, Value: x.UnPack()}
+	case HostMessagesRawStartLeakChecks:
+		x := StartLeakChecksRaw{_tab: table}
+		return &HostMessagesRawT{Type: HostMessagesRawStartLeakChecks, Value: x.UnPack()}
 	}
 	return nil
 }
@@ -2080,6 +2088,63 @@ func SignalUpdateRawEnd(builder *flatbuffers.Builder) flatbuffers.UOffsetT {
 	return builder.EndObject()
 }
 
+type StartLeakChecksRawT struct {
+}
+
+func (t *StartLeakChecksRawT) Pack(builder *flatbuffers.Builder) flatbuffers.UOffsetT {
+	if t == nil {
+		return 0
+	}
+	StartLeakChecksRawStart(builder)
+	return StartLeakChecksRawEnd(builder)
+}
+
+func (rcv *StartLeakChecksRaw) UnPackTo(t *StartLeakChecksRawT) {
+}
+
+func (rcv *StartLeakChecksRaw) UnPack() *StartLeakChecksRawT {
+	if rcv == nil {
+		return nil
+	}
+	t := &StartLeakChecksRawT{}
+	rcv.UnPackTo(t)
+	return t
+}
+
+type StartLeakChecksRaw struct {
+	_tab flatbuffers.Table
+}
+
+func GetRootAsStartLeakChecksRaw(buf []byte, offset flatbuffers.UOffsetT) *StartLeakChecksRaw {
+	n := flatbuffers.GetUOffsetT(buf[offset:])
+	x := &StartLeakChecksRaw{}
+	x.Init(buf, n+offset)
+	return x
+}
+
+func GetSizePrefixedRootAsStartLeakChecksRaw(buf []byte, offset flatbuffers.UOffsetT) *StartLeakChecksRaw {
+	n := flatbuffers.GetUOffsetT(buf[offset+flatbuffers.SizeUint32:])
+	x := &StartLeakChecksRaw{}
+	x.Init(buf, n+offset+flatbuffers.SizeUint32)
+	return x
+}
+
+func (rcv *StartLeakChecksRaw) Init(buf []byte, i flatbuffers.UOffsetT) {
+	rcv._tab.Bytes = buf
+	rcv._tab.Pos = i
+}
+
+func (rcv *StartLeakChecksRaw) Table() flatbuffers.Table {
+	return rcv._tab
+}
+
+func StartLeakChecksRawStart(builder *flatbuffers.Builder) {
+	builder.StartObject(0)
+}
+func StartLeakChecksRawEnd(builder *flatbuffers.Builder) flatbuffers.UOffsetT {
+	return builder.EndObject()
+}
+
 type ExecutingMessageRawT struct {
 	Id           int64 `json:"id"`
 	ProcId       int32 `json:"proc_id"`
-- 
cgit mrf-deployment