Add support for AXFLAG and XAFLAG instructions.

Change-Id: I95867fa8adb6eeefe49dc5e0f56ed51b4d060a1d
diff --git a/src/aarch64/assembler-aarch64.cc b/src/aarch64/assembler-aarch64.cc
index 060681b..04bd226 100644
--- a/src/aarch64/assembler-aarch64.cc
+++ b/src/aarch64/assembler-aarch64.cc
@@ -2549,6 +2549,18 @@
 }
 
 
+void Assembler::axflag() {
+  VIXL_ASSERT(CPUHas(CPUFeatures::kAXFlag));
+  Emit(AXFLAG);
+}
+
+
+void Assembler::xaflag() {
+  VIXL_ASSERT(CPUHas(CPUFeatures::kAXFlag));
+  Emit(XAFLAG);
+}
+
+
 void Assembler::clrex(int imm4) { Emit(CLREX | CRm(imm4)); }
 
 
diff --git a/src/aarch64/assembler-aarch64.h b/src/aarch64/assembler-aarch64.h
index d67824d..309b49a 100644
--- a/src/aarch64/assembler-aarch64.h
+++ b/src/aarch64/assembler-aarch64.h
@@ -2126,6 +2126,14 @@
   // Invert carry flag [Armv8.4].
   void cfinv();
 
+  // Convert floating-point condition flags from alternative format to Arm
+  // format [Armv8.5].
+  void xaflag();
+
+  // Convert floating-point condition flags from Arm format to alternative
+  // format [Armv8.5].
+  void axflag();
+
   // System instruction.
   void sys(int op1, int crn, int crm, int op2, const Register& xt = xzr);
 
diff --git a/src/aarch64/constants-aarch64.h b/src/aarch64/constants-aarch64.h
index 3f6dc58..16f2715 100644
--- a/src/aarch64/constants-aarch64.h
+++ b/src/aarch64/constants-aarch64.h
@@ -779,7 +779,9 @@
   SystemPStateFixed = 0xD5004000,
   SystemPStateFMask = 0xFFF8F000,
   SystemPStateMask  = 0xFFFFF0FF,
-  CFINV             = SystemPStateFixed | 0x0000001F
+  CFINV             = SystemPStateFixed | 0x0000001F,
+  XAFLAG            = SystemPStateFixed | 0x0000003F,
+  AXFLAG            = SystemPStateFixed | 0x0000005F
 };
 
 enum SystemHintOp {
diff --git a/src/aarch64/cpu-features-auditor-aarch64.cc b/src/aarch64/cpu-features-auditor-aarch64.cc
index c0cd5a9..d3639d3 100644
--- a/src/aarch64/cpu-features-auditor-aarch64.cc
+++ b/src/aarch64/cpu-features-auditor-aarch64.cc
@@ -1086,6 +1086,10 @@
       case CFINV:
         scope.Record(CPUFeatures::kFlagM);
         break;
+      case AXFLAG:
+      case XAFLAG:
+        scope.Record(CPUFeatures::kAXFlag);
+        break;
     }
   }
 }
diff --git a/src/aarch64/disasm-aarch64.cc b/src/aarch64/disasm-aarch64.cc
index e976603..2161043 100644
--- a/src/aarch64/disasm-aarch64.cc
+++ b/src/aarch64/disasm-aarch64.cc
@@ -2069,6 +2069,14 @@
         mnemonic = "cfinv";
         form = NULL;
         break;
+      case AXFLAG:
+        mnemonic = "axflag";
+        form = NULL;
+        break;
+      case XAFLAG:
+        mnemonic = "xaflag";
+        form = NULL;
+        break;
     }
   } else if (instr->Mask(SystemPAuthFMask) == SystemPAuthFixed) {
     switch (instr->Mask(SystemPAuthMask)) {
diff --git a/src/aarch64/macro-assembler-aarch64.h b/src/aarch64/macro-assembler-aarch64.h
index a1de4dd..8051bd1 100644
--- a/src/aarch64/macro-assembler-aarch64.h
+++ b/src/aarch64/macro-assembler-aarch64.h
@@ -2007,6 +2007,16 @@
     SingleEmissionCheckScope guard(this);
     cfinv();
   }
+  void Axflag() {
+    VIXL_ASSERT(allow_macro_instructions_);
+    SingleEmissionCheckScope guard(this);
+    axflag();
+  }
+  void Xaflag() {
+    VIXL_ASSERT(allow_macro_instructions_);
+    SingleEmissionCheckScope guard(this);
+    xaflag();
+  }
   void Sys(int op1, int crn, int crm, int op2, const Register& rt = xzr) {
     VIXL_ASSERT(allow_macro_instructions_);
     SingleEmissionCheckScope guard(this);
diff --git a/src/aarch64/simulator-aarch64.cc b/src/aarch64/simulator-aarch64.cc
index c1d32a9..1c1d52e 100644
--- a/src/aarch64/simulator-aarch64.cc
+++ b/src/aarch64/simulator-aarch64.cc
@@ -3732,6 +3732,24 @@
       case CFINV:
         ReadNzcv().SetC(!ReadC());
         break;
+      case AXFLAG:
+        ReadNzcv().SetN(0);
+        ReadNzcv().SetZ(ReadNzcv().GetZ() | ReadNzcv().GetV());
+        ReadNzcv().SetC(ReadNzcv().GetC() & ~ReadNzcv().GetV());
+        ReadNzcv().SetV(0);
+        break;
+      case XAFLAG: {
+        // Can't set the flags in place due to the logical dependencies.
+        uint32_t n = (~ReadNzcv().GetC() & ~ReadNzcv().GetZ()) & 1;
+        uint32_t z = ReadNzcv().GetZ() & ReadNzcv().GetC();
+        uint32_t c = ReadNzcv().GetC() | ReadNzcv().GetZ();
+        uint32_t v = ~ReadNzcv().GetC() & ReadNzcv().GetZ();
+        ReadNzcv().SetN(n);
+        ReadNzcv().SetZ(z);
+        ReadNzcv().SetC(c);
+        ReadNzcv().SetV(v);
+        break;
+      }
     }
   } else if (instr->Mask(SystemPAuthFMask) == SystemPAuthFixed) {
     // Check BType allows PACI[AB]SP instructions.
diff --git a/src/cpu-features.h b/src/cpu-features.h
index 88b6914..fffef44 100644
--- a/src/cpu-features.h
+++ b/src/cpu-features.h
@@ -96,7 +96,9 @@
   /* Data-independent timing (for selected instructions).                   */ \
   V(kDIT,                 "DIT",                    "dit")                     \
   /* Branch target identification.                                          */ \
-  V(kBTI,                 "BTI",                    NULL)
+  V(kBTI,                 "BTI",                    NULL)                      \
+  /* Flag manipulation instructions: {AX,XA}FLAG                            */ \
+  V(kAXFlag,              "AXFlag",                 NULL)
 // clang-format on