1 files changed, 38 insertions, 0 deletions
diff --git a/cmd/ioworkload/scenario_security_test.go b/cmd/ioworkload/scenario_security_test.go
new file mode 100644
index 0000000..f1b6152
--- /dev/null
+++ b/cmd/ioworkload/scenario_security_test.go
@@ -0,0 +1,38 @@
+package main
+
+import "testing"
+
+func TestSecuritySyscallNumbers(t *testing.T) {
+	for _, tc := range []struct {
+		name    string
+		arch    string
+		wantErr bool
+		addKey  uintptr
+		ptrace  uintptr
+		perf    uintptr
+	}{
+		{name: "amd64", arch: "amd64", addKey: 248, ptrace: 101, perf: 298},
+		{name: "arm64", arch: "arm64", addKey: 217, ptrace: 117, perf: 241},
+		{name: "unsupported", arch: "riscv64", wantErr: true},
+	} {
+		got, err := securitySyscallNumbers(tc.arch)
+		if tc.wantErr {
+			if err == nil {
+				t.Fatalf("%s: expected error", tc.name)
+			}
+			continue
+		}
+		if err != nil {
+			t.Fatalf("%s: unexpected error: %v", tc.name, err)
+		}
+		if got.addKey != tc.addKey || got.ptrace != tc.ptrace || got.perfEventOpen != tc.perf {
+			t.Fatalf(
+				"%s: unexpected numbers add_key=%d ptrace=%d perf_event_open=%d",
+				tc.name,
+				got.addKey,
+				got.ptrace,
+				got.perfEventOpen,
+			)
+		}
+	}
+}