From d04ac3a54895a36998e3f1da27c2ce33f7a80c5a Mon Sep 17 00:00:00 2001
From: Dmitry Vyukov <dvyukov@google.com>
Date: Mon, 3 Apr 2023 07:45:09 +0200
Subject: sys/linux: add syz_pkey_set syscalls

The syscall sets PKRU register which is part of protection keys (pkey).
---
 pkg/csource/generated.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'pkg/csource')

diff --git a/pkg/csource/generated.go b/pkg/csource/generated.go
index 552cc9191..204d6d2ee 100644
--- a/pkg/csource/generated.go
+++ b/pkg/csource/generated.go
@@ -11378,6 +11378,25 @@ static long syz_clone3(volatile long a0, volatile long a1)
 
 #endif
 
+#if SYZ_EXECUTOR || __NR_syz_pkey_set
+static long syz_pkey_set(volatile long pkey, volatile long val)
+{
+#if GOARCH_amd64 || GOARCH_386
+	uint32 eax = 0;
+	uint32 ecx = 0;
+	asm volatile("rdpkru"
+		     : "=a"(eax)
+		     : "c"(ecx)
+		     : "edx");
+	eax &= ~(3 << ((pkey % 16) * 2));
+	eax |= (val & 3) << ((pkey % 16) * 2);
+	uint32 edx = 0;
+	asm volatile("wrpkru" ::"a"(eax), "c"(ecx), "d"(edx));
+#endif
+	return 0;
+}
+#endif
+
 #elif GOOS_test
 
 #include <stdlib.h>
-- 
cgit mrf-deployment