[litmus] Introduce variant checks for exs, eis and eos

Author: diaolo01

litmus/libdir/_aarch64/kvm-headers.h

@@ -25,6 +25,39 @@
 
 #define LITMUS_PAGE_SIZE PAGE_SIZE
 
+static inline int check_exs(const char *tname, int need_exs, int need_eis, int need_eos) {
+  uint64_t mmfr0;
+  asm volatile("mrs %0, ID_AA64MMFR0_EL1" : "=r" (mmfr0));
+  int exs = (mmfr0 >> 44) & 0xf;
+  if (need_exs && exs == 0) {
+    printf("Test %s, ExS not supported on this system\n", tname);
+    return 0;
+  }
+
+  uint64_t sctlr;
+  asm volatile("mrs %0, SCTLR_EL1" : "=r" (sctlr));
+  int eis = (sctlr >> 22) & 1;
+  int eos = (sctlr >> 11) & 1;
+  if (eis != need_eis || eos != need_eos) {
+    uint64_t new_sctlr = sctlr;
+    if (need_eis) new_sctlr |= 1ULL << 22;
+    else new_sctlr &= ~(1ULL << 22);
+    if (need_eos) new_sctlr |= 1ULL << 11;
+    else new_sctlr &= ~(1ULL << 11);
+    asm volatile("msr SCTLR_EL1, %0" :: "r" (new_sctlr));
+    asm volatile("isb");
+    asm volatile("mrs %0, SCTLR_EL1" : "=r" (sctlr));
+    eis = (sctlr >> 22) & 1;
+    eos = (sctlr >> 11) & 1;
+    if (eis != need_eis || eos != need_eos) {
+      printf("Test %s, cannot set ExS settings EIS=%d EOS=%d\n",
+             tname, need_eis, need_eos);
+      return 0;
+    }
+  }
+  return 1;
+}
+
 static inline pteval_t *litmus_tr_pte(void *p) {
   return follow_pte(mmu_idmap, (uintptr_t)p);
 }

litmus/preSi.ml

@@ -80,6 +80,11 @@ module Make
     if Cfg.variant Variant_litmus.ConstPacField && not (Cfg.variant Variant_litmus.Pac) then
       Warn.user_error "\"const-pac-field\" variant require \"pac\" variant"
 
+  let () =
+    if (Cfg.variant Variant_litmus.EIS || Cfg.variant Variant_litmus.EOS)
+       && not (Cfg.variant Variant_litmus.ExS) then
+      Warn.user_error "\"eis\"/\"eos\" variants require \"exs\" variant"
+
   module Insert =
       ObjUtil.Insert
         (struct
@@ -2329,6 +2334,15 @@ module Make
         end ;
         if Cfg.variant Variant_litmus.ConstPacField then
           O.fi "if (!check_const_pac_field_variant(%S)) return 0;" doc.Name.name;
+        if Cfg.is_kvm &&
+           (Cfg.variant Variant_litmus.ExS ||
+            Cfg.variant Variant_litmus.EIS ||
+            Cfg.variant Variant_litmus.EOS) then
+          O.fi "if (!check_exs(%S,%d,%d,%d)) return 0;"
+            doc.Name.name
+            (if Cfg.variant Variant_litmus.ExS then 1 else 0)
+            (if Cfg.variant Variant_litmus.EIS then 1 else 0)
+            (if Cfg.variant Variant_litmus.EOS then 1 else 0) ;
         if Cfg.is_kvm then begin
           match db with
           | None ->