encoder: add compact fuzz tests and fix bugs related to unhandled event types

pkg/encoder/encoder.go

@@ -226,6 +226,10 @@ func HumanStackTrace(response *tetragon.GetEventsResponse, colorer *Colorer) str
 
 func (p *CompactEncoder) EventToString(response *tetragon.GetEventsResponse) (string, error) {
 	switch response.Event.(type) {
+	case *tetragon.GetEventsResponse_Test:
+		// This shouldn't normally be reachable since Tetragon won't generate Test
+		// events outside of specific unit tests.
+		return "TEST EVENT", nil
 	case *tetragon.GetEventsResponse_ProcessExec:
 		exec := response.GetProcessExec()
 		if exec.Process == nil {
@@ -491,6 +495,14 @@ func (p *CompactEncoder) EventToString(response *tetragon.GetEventsResponse) (st
 			event := p.Colorer.Blue.Sprintf("⁉️ %-7s", "tracepoint")
 			return CapTrailorPrinter(fmt.Sprintf("%s %s %s %s", event, processInfo, tp.Subsys, tp.Event), caps), nil
 		}
+	case *tetragon.GetEventsResponse_ProcessUprobe:
+		uprobe := response.GetProcessUprobe()
+		if uprobe.Process == nil {
+			return "", ErrMissingProcessInfo
+		}
+		processInfo, caps := p.Colorer.ProcessInfo(response.NodeName, uprobe.Process)
+		event := p.Colorer.Blue.Sprintf("🕵️ %-7s", "uprobe")
+		return CapTrailorPrinter(fmt.Sprintf("%s %s %s %s", event, processInfo, uprobe.Path, uprobe.Symbol), caps), nil
 	case *tetragon.GetEventsResponse_ProcessLsm:
 		lsm := response.GetProcessLsm()
 		if lsm.Process == nil {
@@ -501,7 +513,7 @@ func (p *CompactEncoder) EventToString(response *tetragon.GetEventsResponse) (st
 		return CapTrailorPrinter(fmt.Sprintf("%s %s %s", event, processInfo, lsm.FunctionName), caps), nil
 	}
 
-	return "", ErrUnknownEventType
+	return "", fmt.Errorf("%w: %s", ErrUnknownEventType, response.EventType())
 }
 
 func rawSyscallEnter(tp *tetragon.ProcessTracepoint) string {

pkg/encoder/encoder_test.go

@@ -14,6 +14,7 @@ import (
 	"google.golang.org/protobuf/types/known/timestamppb"
 
 	"github.com/cilium/tetragon/api/v1/tetragon"
+	"github.com/cilium/tetragon/api/v1/tetragon/codegen/helpers"
 	"github.com/sryoya/protorand"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -535,3 +536,51 @@ func FuzzProtojsonCompatibility(f *testing.F) {
 		assert.True(t, proto.Equal(msg, msgProtojson))
 	})
 }
+
+func FuzzCompactEncoder(f *testing.F) {
+	for _, n := range []int64{
+		1337,
+		78776406,
+		56343416,
+		68876713,
+		51156281,
+		45544244,
+		4011756,
+	} {
+		for _, cm := range []uint8{0, 1, 2} {
+			for _, ts := range []bool{true, false} {
+				for _, st := range []bool{true, false} {
+					f.Add(n, cm, ts, st)
+				}
+			}
+		}
+	}
+	f.Fuzz(func(t *testing.T, seed int64, colorMode uint8, timestamps bool, stackTraces bool) {
+		var cm ColorMode
+		switch colorMode % 3 {
+		case 0:
+			cm = "never"
+		case 1:
+			cm = "always"
+		case 2:
+			cm = "auto"
+		default:
+			panic("unreachable")
+		}
+
+		pr := protorand.New()
+		pr.Seed(seed)
+		ev := &tetragon.GetEventsResponse{}
+		msg, err := pr.Gen(ev)
+		require.NoError(t, err)
+
+		if helpers.ResponseGetProcess(msg.(*tetragon.GetEventsResponse)) == nil {
+			t.Skipf("Empty process")
+		}
+
+		var buf1 bytes.Buffer
+		compactEncoder := NewCompactEncoder(&buf1, cm, timestamps, stackTraces)
+		err = compactEncoder.Encode(msg)
+		require.NoError(t, err)
+	})
+}

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/0056013c3120f707

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776374)
+byte('\x02')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/0238502285390cc4

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776519)
+byte('.')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/04f1c2566953755d

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343604)
+byte('\u0096')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/055fd67d3ece8673

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343515)
+byte('\x02')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/08682a07e807ad1b

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(51156230)
+byte('\u0086')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/09fe04e6a5cab965

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(45544323)
+byte('L')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/0b70fd5bed47bd65

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(45544350)
+byte('\x1e')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/0cea5849dcabadfd

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1284)
+byte('\x01')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/0dae6e0b6fde4082

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(51156231)
+byte('\x11')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/0e7e958184345eee

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(51156190)
+byte('_')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1087b9781be2e4ea

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(45544205)
+byte('U')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/12a3fb2618cc4d03

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(4011786)
+byte('\x01')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/13f8752cc7e42b7a

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776489)
+byte('\x00')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/15283e4b6a01657c

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(68876694)
+byte('5')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1548dbec3d38cbde

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776485)
+byte('\x02')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1598fcf0a64ffd33

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1560)
+byte('b')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/159d27214c561d4c

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(68876791)
+byte('Z')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/17087f2e8f9a7e6e

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776456)
+byte('\x00')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/19ab4da44947a65b

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1575)
+byte('b')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1a8727b3253382af

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1420)
+byte('\x02')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1accae4e7ad75314

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(45544220)
+byte('\x02')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1dd69ec3c8ed980e

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(4011791)
+byte('\x01')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1e95f134b44feb48

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(51156231)
+byte('\x01')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1f1272a7af3505a9

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(68876745)
+byte('\x00')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/1fa36d480803e0f3

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(4011706)
+byte('\x00')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/2027fe3f62620ff5

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343414)
+byte('#')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/2030dd818e3a6825

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343363)
+byte('V')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/205274079ed3cdfd

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776601)
+byte('\x00')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/21d9c9a88bfb95f5

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(51156303)
+byte('\x00')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/21e9fdbb2814e74c

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776406)
+byte('.')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/22317988c9fde4fa

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776323)
+byte('\x00')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/22560513d314e134

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776435)
+byte('\x05')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/22abc7249c26edc2

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(4011620)
+byte('\r')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/22fe5a9be17f3d74

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(51156230)
+byte('\x00')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/231a779422264629

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776385)
+byte('[')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/242eb17689170eb9

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1351)
+byte('\x01')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/275bebb2fd3d432f

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(68876842)
+byte('\x16')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/28b8124895f33d20

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1306)
+byte('\x01')
+bool(true)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/2a9551f78e55a0dd

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(78776334)
+byte('\x0f')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/2ba02468746d677b

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1371)
+byte('7')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/2bb329139f4e3d67

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(45544228)
+byte('/')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/2c27a5892baabc36

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(51156359)
+byte('\x02')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/2ca049db44842a1a

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343431)
+byte('$')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/2e4d893aff6a2fe4

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343330)
+byte('Y')
+bool(true)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/32cc668bb7c4bb3b

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343506)
+byte('\x00')
+bool(false)
+bool(true)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/32f80625bc3a80f7

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343604)
+byte('\u0096')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/3342220a67146a72

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1291)
+byte('\x02')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/33931c073ee2202b

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(68876708)
+byte('\x00')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/38e57ed4dc3d4bc9

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(1275)
+byte('\x00')
+bool(false)
+bool(false)

pkg/encoder/testdata/fuzz/FuzzCompactEncoder/3b010f74131ce1e6

@@ -0,0 +1,5 @@
+go test fuzz v1
+int64(56343514)
+byte('\x00')
+bool(true)
+bool(true)