From ba902366432db186e8ec80b653e84e972d857bbc Mon Sep 17 00:00:00 2001
From: Kyle Huey <khuey@kylehuey.com>
Date: Thu, 23 Jan 2025 13:46:42 -0800
Subject: [PATCH] Repeat ARCH_XCOMP_GET_PERM if necessary as the relevant bit
 may be newly enabled.

---
 src/test/x86/arch_prctl_xstate.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/test/x86/arch_prctl_xstate.c b/src/test/x86/arch_prctl_xstate.c
index 49864e762ce..a082683a3de 100644
--- a/src/test/x86/arch_prctl_xstate.c
+++ b/src/test/x86/arch_prctl_xstate.c
@@ -38,6 +38,11 @@ int main(void) {
   ret = syscall(SYS_arch_prctl, ARCH_REQ_XCOMP_PERM, ARCH_XCOMP_TILEDATA);
   if ((1 << ARCH_XCOMP_TILEDATA) & *features_perm) {
     test_assert(0 == ret);
+  } else if (0 == ret) {
+    ret = syscall(SYS_arch_prctl, ARCH_GET_XCOMP_PERM, features_perm);
+    test_assert(0 == ret);
+    VERIFY_GUARD(features_perm);
+    test_assert((1 << ARCH_XCOMP_TILEDATA) & *features_perm);
   } else {
     test_assert(-1 == ret && errno == EOPNOTSUPP);
   }