From 7fb497c435596a36c0fb0bd0ecae2a84793bcc70 Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@buetow.org>
Date: Mon, 18 May 2026 14:30:26 +0300
Subject: j6: account bytes for ret-classified syscalls

---
 internal/generate/retclassify_test.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'internal/generate/retclassify_test.go')

diff --git a/internal/generate/retclassify_test.go b/internal/generate/retclassify_test.go
index 3152005..9a75a15 100644
--- a/internal/generate/retclassify_test.go
+++ b/internal/generate/retclassify_test.go
@@ -57,3 +57,28 @@ func TestClassifyRetCaseInsensitive(t *testing.T) {
 		t.Errorf("ClassifyRet(sys_exit_READ) = %q, want READ_CLASSIFIED", got)
 	}
 }
+
+func TestPhaseAByteClassifiedSyscallsUseExistingRetClassifications(t *testing.T) {
+	tests := []struct {
+		name string
+		want RetClassification
+	}{
+		{"recvfrom", ReadClassified},
+		{"recvmsg", ReadClassified},
+		{"sendto", WriteClassified},
+		{"sendmsg", WriteClassified},
+		{"sendfile64", TransferClassified},
+		{"splice", TransferClassified},
+		{"tee", TransferClassified},
+		{"process_vm_readv", ReadClassified},
+		{"process_vm_writev", WriteClassified},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := ClassifyRet("sys_exit_" + tt.name); got != tt.want {
+				t.Fatalf("ClassifyRet(sys_exit_%s) = %q, want %q", tt.name, got, tt.want)
+			}
+		})
+	}
+}
-- 
cgit v1.2.3