/src/capstonenext/arch/AArch64/AArch64Mapping.c

Source (jump to first uncovered line)
/* Capstone Disassembly Engine */
/* By Nguyen Anh Quynh <aquynh@gmail.com>, 2013-2019 */

#ifdef CAPSTONE_HAS_AARCH64

#include <stdio.h> // debug
#include <string.h>

#include "capstone/aarch64.h"

#include "../../cs_simple_types.h"
#include "../../Mapping.h"
#include "../../MathExtras.h"
#include "../../utils.h"

#include "AArch64AddressingModes.h"
#include "AArch64BaseInfo.h"
#include "AArch64DisassemblerExtension.h"
#include "AArch64Linkage.h"
#include "AArch64Mapping.h"

#define CHAR(c) #c[0]

static float aarch64_exact_fp_to_fp(aarch64_exactfpimm exact)
{
  switch (exact) {
  default:
    CS_ASSERT(0 && "Not handled.");
    return 999.0;
  case AARCH64_EXACTFPIMM_HALF:
    return 0.5;
  case AARCH64_EXACTFPIMM_ONE:
    return 1.0;
  case AARCH64_EXACTFPIMM_TWO:
    return 2.0;
  case AARCH64_EXACTFPIMM_ZERO:
    return 0.0;
  }
}

#ifndef CAPSTONE_DIET
static const aarch64_reg aarch64_flag_regs[] = {
  AARCH64_REG_NZCV,
};

static const aarch64_sysreg aarch64_flag_sys_regs[] = {
  AARCH64_SYSREG_NZCV, AARCH64_SYSREG_PMOVSCLR_EL0,
  AARCH64_SYSREG_PMOVSSET_EL0, AARCH64_SYSREG_SPMOVSCLR_EL0,
  AARCH64_SYSREG_SPMOVSSET_EL0
};
#endif // CAPSTONE_DIET

static AArch64Layout_VectorLayout sme_reg_to_vas(aarch64_reg reg)
{
  switch (reg) {
  default:
    return AARCH64LAYOUT_INVALID;
  case AARCH64_REG_ZAB0:
    return AARCH64LAYOUT_VL_B;
  case AARCH64_REG_ZAH0:
  case AARCH64_REG_ZAH1:
    return AARCH64LAYOUT_VL_H;
  case AARCH64_REG_ZAS0:
  case AARCH64_REG_ZAS1:
  case AARCH64_REG_ZAS2:
  case AARCH64_REG_ZAS3:
    return AARCH64LAYOUT_VL_S;
  case AARCH64_REG_ZAD0:
  case AARCH64_REG_ZAD1:
  case AARCH64_REG_ZAD2:
  case AARCH64_REG_ZAD3:
  case AARCH64_REG_ZAD4:
  case AARCH64_REG_ZAD5:
  case AARCH64_REG_ZAD6:
  case AARCH64_REG_ZAD7:
    return AARCH64LAYOUT_VL_D;
  case AARCH64_REG_ZAQ0:
  case AARCH64_REG_ZAQ1:
  case AARCH64_REG_ZAQ2:
  case AARCH64_REG_ZAQ3:
  case AARCH64_REG_ZAQ4:
  case AARCH64_REG_ZAQ5:
  case AARCH64_REG_ZAQ6:
  case AARCH64_REG_ZAQ7:
  case AARCH64_REG_ZAQ8:
  case AARCH64_REG_ZAQ9:
  case AARCH64_REG_ZAQ10:
  case AARCH64_REG_ZAQ11:
  case AARCH64_REG_ZAQ12:
  case AARCH64_REG_ZAQ13:
  case AARCH64_REG_ZAQ14:
  case AARCH64_REG_ZAQ15:
    return AARCH64LAYOUT_VL_Q;
  case AARCH64_REG_ZA:
    return AARCH64LAYOUT_VL_COMPLETE;
  }
}

void AArch64_init_mri(MCRegisterInfo *MRI)
{
  MCRegisterInfo_InitMCRegisterInfo(
    MRI, AArch64RegDesc, AARCH64_REG_ENDING, 0, 0,
    AArch64MCRegisterClasses, ARR_SIZE(AArch64MCRegisterClasses), 0,
    0, AArch64RegDiffLists, 0, AArch64SubRegIdxLists,
    ARR_SIZE(AArch64SubRegIdxLists), 0);
}

/// Sets up a new SME matrix operand at the currently active detail operand.
static void setup_sme_operand(MCInst *MI)
{
  if (!detail_is_set(MI))
    return;

  AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_SME;
  AArch64_get_detail_op(MI, 0)->sme.type = AARCH64_SME_OP_INVALID;
  AArch64_get_detail_op(MI, 0)->sme.tile = AARCH64_REG_INVALID;
  AArch64_get_detail_op(MI, 0)->sme.slice_reg = AARCH64_REG_INVALID;
  AArch64_get_detail_op(MI, 0)->sme.slice_offset.imm =
    AARCH64_SLICE_IMM_INVALID;
  AArch64_get_detail_op(MI, 0)->sme.slice_offset.imm_range.first =
    AARCH64_SLICE_IMM_RANGE_INVALID;
  AArch64_get_detail_op(MI, 0)->sme.slice_offset.imm_range.offset =
    AARCH64_SLICE_IMM_RANGE_INVALID;
}

static void setup_pred_operand(MCInst *MI)
{
  if (!detail_is_set(MI))
    return;

  AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_PRED;
  AArch64_get_detail_op(MI, 0)->pred.imm_index = -1;
}

const insn_map aarch64_insns[] = {
#include "AArch64GenCSMappingInsn.inc"
};

static const name_map insn_alias_mnem_map[] = {
#include "AArch64GenCSAliasMnemMap.inc"
  { AARCH64_INS_ALIAS_CFP, "cfp" },
  { AARCH64_INS_ALIAS_DVP, "dvp" },
  { AARCH64_INS_ALIAS_COSP, "cosp" },
  { AARCH64_INS_ALIAS_CPP, "cpp" },
  { AARCH64_INS_ALIAS_IC, "ic" },
  { AARCH64_INS_ALIAS_DC, "dc" },
  { AARCH64_INS_ALIAS_AT, "at" },
  { AARCH64_INS_ALIAS_TLBI, "tlbi" },
  { AARCH64_INS_ALIAS_TLBIP, "tlbip" },
  { AARCH64_INS_ALIAS_RPRFM, "rprfm" },
  { AARCH64_INS_ALIAS_LSL, "lsl" },
  { AARCH64_INS_ALIAS_SBFX, "sbfx" },
  { AARCH64_INS_ALIAS_UBFX, "ubfx" },
  { AARCH64_INS_ALIAS_SBFIZ, "sbfiz" },
  { AARCH64_INS_ALIAS_UBFIZ, "ubfiz" },
  { AARCH64_INS_ALIAS_BFC, "bfc" },
  { AARCH64_INS_ALIAS_BFI, "bfi" },
  { AARCH64_INS_ALIAS_BFXIL, "bfxil" },
  { AARCH64_INS_ALIAS_END, NULL },
};

static const char *get_custom_reg_alias(unsigned reg)
{
  switch (reg) {
  case AARCH64_REG_X29:
    return "fp";
  case AARCH64_REG_X30:
    return "lr";
  }
  return NULL;
}

/// Very annoyingly LLVM hard codes the vector layout post-fixes into the asm string.
/// In this function we check for these cases and add the vectorlayout/arrangement
/// specifier.
void AArch64_add_vas(MCInst *MI, const SStream *OS)
{
  if (!detail_is_set(MI)) {
    return;
  }

  if (AArch64_get_detail(MI)->op_count == 0) {
    return;
  }
  if (MCInst_getOpcode(MI) == AArch64_MUL53HI ||
      MCInst_getOpcode(MI) == AArch64_MUL53LO) {
    // Proprietary Apple instrucions.
    AArch64_get_detail(MI)->operands[0].vas = AARCH64LAYOUT_VL_2D;
    AArch64_get_detail(MI)->operands[1].vas = AARCH64LAYOUT_VL_2D;
    return;
  }

  // Search for r".[0-9]{1,2}[bhsdq]\W"
  // with poor mans regex
  const char *vl_ptr = strchr(OS->buffer, '.');
  while (vl_ptr) {
    // Number after dot?
    unsigned num = 0;
    if (strchr("1248", vl_ptr[1])) {
      num = atoi(vl_ptr + 1);
      vl_ptr = num > 9 ? vl_ptr + 3 : vl_ptr + 2;
    } else {
      vl_ptr++;
    }

    // Layout letter
    char letter = '\0';
    if (strchr("bhsdq", vl_ptr[0])) {
      letter = vl_ptr[0];
    }
    if (!letter) {
      goto next_dot_continue;
    }

    AArch64Layout_VectorLayout vl = AARCH64LAYOUT_INVALID;
    switch (letter) {
    default:
      CS_ASSERT_RET(0 && "Unhandled vector layout letter.");
      return;
    case 'b':
      vl = AARCH64LAYOUT_VL_B;
      break;
    case 'h':
      vl = AARCH64LAYOUT_VL_H;
      break;
    case 's':
      vl = AARCH64LAYOUT_VL_S;
      break;
    case 'd':
      vl = AARCH64LAYOUT_VL_D;
      break;
    case 'q':
      vl = AARCH64LAYOUT_VL_Q;
      break;
    }
    vl |= (num << 8);

    // Determine op index by searching for trailing commata after op string
    uint32_t op_idx = 0;
    const char *comma_ptr = strchr(OS->buffer, ',');
    ;
    while (comma_ptr && comma_ptr < vl_ptr) {
      ++op_idx;
      comma_ptr = strchr(comma_ptr + 1, ',');
    }
    if (!comma_ptr) {
      // Last op doesn't have a trailing commata.
      op_idx = AArch64_get_detail(MI)->op_count - 1;
    }
    if (op_idx >= AArch64_get_detail(MI)->op_count) {
      // A memory operand with a commata in [base, dist]
      op_idx = AArch64_get_detail(MI)->op_count - 1;
    }

    // Search for the operand this one belongs to.
    cs_aarch64_op *op = &AArch64_get_detail(MI)->operands[op_idx];
    if ((op->type != AARCH64_OP_REG &&
         op->type != AARCH64_OP_SME) ||
        op->vas != AARCH64LAYOUT_INVALID) {
      goto next_dot_continue;
    }
    op->vas = vl;

next_dot_continue:
    vl_ptr = strchr(vl_ptr + 1, '.');
  }
}

const char *AArch64_reg_name(csh handle, unsigned int reg)
{
  int syntax_opt = ((cs_struct *)(uintptr_t)handle)->syntax;
  const char *alias = get_custom_reg_alias(reg);
  if ((syntax_opt & CS_OPT_SYNTAX_CS_REG_ALIAS) && alias)
    return alias;

  if (((cs_struct *)(uintptr_t)handle)->syntax &
      CS_OPT_SYNTAX_NOREGNAME) {
    return AArch64_LLVM_getRegisterName(reg, AArch64_NoRegAltName);
  }
  // TODO Add options for the other register names
  return AArch64_LLVM_getRegisterName(reg, AArch64_NoRegAltName);
}

void AArch64_setup_op(cs_aarch64_op *op)
{
  memset(op, 0, sizeof(cs_aarch64_op));
  op->type = AARCH64_OP_INVALID;
  op->vector_index = -1;
}

void AArch64_init_cs_detail(MCInst *MI)
{
  if (detail_is_set(MI)) {
    memset(get_detail(MI), 0,
           offsetof(cs_detail, aarch64) + sizeof(cs_aarch64));
    for (int i = 0; i < ARR_SIZE(AArch64_get_detail(MI)->operands);
         i++)
      AArch64_setup_op(&AArch64_get_detail(MI)->operands[i]);
    AArch64_get_detail(MI)->cc = AArch64CC_Invalid;
  }
}

/// Unfortunately, the AARCH64 definitions do not indicate in any way
/// (exception are the instruction identifiers), if memory accesses
/// is post- or pre-indexed.
/// So the only generic way to determine, if the memory access is in
/// post-indexed addressing mode, is by search for "<membase>], #<memdisp>" in
/// @p OS.
/// Searching the asm string to determine such a property is enormously ugly
/// and wastes resources.
/// Sorry, I know and do feel bad about it. But for now it works.
static bool AArch64_check_post_index_am(const MCInst *MI, const SStream *OS)
{
  if (AArch64_get_detail(MI)->post_index) {
    return true;
  }
  cs_aarch64_op *memop = NULL;
  for (int i = 0; i < AArch64_get_detail(MI)->op_count; ++i) {
    if (AArch64_get_detail(MI)->operands[i].type & CS_OP_MEM) {
      memop = &AArch64_get_detail(MI)->operands[i];
      break;
    }
  }
  if (!memop)
    return false;
  if (memop->mem.base == AARCH64_REG_INVALID) {
    // Load/Store from/to label. Has no register base.
    return false;
  }
  const char *membase = AArch64_LLVM_getRegisterName(
    memop->mem.base, AArch64_NoRegAltName);
  int64_t memdisp = memop->mem.disp;
  SStream pattern = { 0 };
  SStream_concat(&pattern, membase);
  SStream_concat(&pattern, "], ");
  printInt32Bang(&pattern, memdisp);
  return strstr(OS->buffer, pattern.buffer) != NULL;
}

static void AArch64_check_updates_flags(MCInst *MI)
{
#ifndef CAPSTONE_DIET
  if (!detail_is_set(MI))
    return;
  cs_detail *detail = get_detail(MI);
  // Implicitly written registers
  for (int i = 0; i < detail->regs_write_count; ++i) {
    if (detail->regs_write[i] == 0)
      break;
    for (int j = 0; j < ARR_SIZE(aarch64_flag_regs); ++j) {
      if (detail->regs_write[i] == aarch64_flag_regs[j]) {
        detail->aarch64.update_flags = true;
        return;
      }
    }
  }
  for (int i = 0; i < detail->aarch64.op_count; ++i) {
    if (detail->aarch64.operands[i].type == AARCH64_OP_SYSREG &&
        detail->aarch64.operands[i].sysop.sub_type ==
          AARCH64_OP_REG_MSR) {
      for (int j = 0; j < ARR_SIZE(aarch64_flag_sys_regs);
           ++j)
        if (detail->aarch64.operands[i]
              .sysop.reg.sysreg ==
            aarch64_flag_sys_regs[j]) {
          detail->aarch64.update_flags = true;
          return;
        }
    } else if (detail->aarch64.operands[i].type == AARCH64_OP_REG &&
         detail->aarch64.operands[i].access & CS_AC_WRITE) {
      for (int j = 0; j < ARR_SIZE(aarch64_flag_regs); ++j)
        if (detail->aarch64.operands[i].reg ==
            aarch64_flag_regs[j]) {
          detail->aarch64.update_flags = true;
          return;
        }
    }
  }
#endif // CAPSTONE_DIET
}

static aarch64_shifter id_to_shifter(unsigned Opcode)
{
  switch (Opcode) {
  default:
    return AARCH64_SFT_INVALID;
  case AArch64_RORVXr:
  case AArch64_RORVWr:
    return AARCH64_SFT_ROR_REG;
  case AArch64_LSRVXr:
  case AArch64_LSRVWr:
    return AARCH64_SFT_LSR_REG;
  case AArch64_LSLVXr:
  case AArch64_LSLVWr:
    return AARCH64_SFT_LSL_REG;
  case AArch64_ASRVXr:
  case AArch64_ASRVWr:
    return AARCH64_SFT_ASR_REG;
  }
}

static void add_non_alias_details(MCInst *MI)
{
  unsigned Opcode = MCInst_getOpcode(MI);
  switch (Opcode) {
  default:
    break;
  case AArch64_RORVXr:
  case AArch64_RORVWr:
  case AArch64_LSRVXr:
  case AArch64_LSRVWr:
  case AArch64_LSLVXr:
  case AArch64_LSLVWr:
  case AArch64_ASRVXr:
  case AArch64_ASRVWr:
    if (AArch64_get_detail(MI)->op_count != 3) {
      return;
    }
    CS_ASSERT_RET(AArch64_get_detail_op(MI, -1)->type ==
            AARCH64_OP_REG);

    // The shift by register instructions don't set the shift value properly.
    // Correct it here.
    uint64_t shift = AArch64_get_detail_op(MI, -1)->reg;
    cs_aarch64_op *op1 = AArch64_get_detail_op(MI, -2);
    op1->shift.type = id_to_shifter(Opcode);
    op1->shift.value = shift;
    AArch64_dec_op_count(MI);
    break;
  case AArch64_FCMPDri:
  case AArch64_FCMPEDri:
  case AArch64_FCMPEHri:
  case AArch64_FCMPESri:
  case AArch64_FCMPHri:
  case AArch64_FCMPSri:
    AArch64_insert_detail_op_reg_at(MI, -1, AARCH64_REG_XZR,
            CS_AC_READ);
    break;
  case AArch64_CMEQv16i8rz:
  case AArch64_CMEQv1i64rz:
  case AArch64_CMEQv2i32rz:
  case AArch64_CMEQv2i64rz:
  case AArch64_CMEQv4i16rz:
  case AArch64_CMEQv4i32rz:
  case AArch64_CMEQv8i16rz:
  case AArch64_CMEQv8i8rz:
  case AArch64_CMGEv16i8rz:
  case AArch64_CMGEv1i64rz:
  case AArch64_CMGEv2i32rz:
  case AArch64_CMGEv2i64rz:
  case AArch64_CMGEv4i16rz:
  case AArch64_CMGEv4i32rz:
  case AArch64_CMGEv8i16rz:
  case AArch64_CMGEv8i8rz:
  case AArch64_CMGTv16i8rz:
  case AArch64_CMGTv1i64rz:
  case AArch64_CMGTv2i32rz:
  case AArch64_CMGTv2i64rz:
  case AArch64_CMGTv4i16rz:
  case AArch64_CMGTv4i32rz:
  case AArch64_CMGTv8i16rz:
  case AArch64_CMGTv8i8rz:
  case AArch64_CMLEv16i8rz:
  case AArch64_CMLEv1i64rz:
  case AArch64_CMLEv2i32rz:
  case AArch64_CMLEv2i64rz:
  case AArch64_CMLEv4i16rz:
  case AArch64_CMLEv4i32rz:
  case AArch64_CMLEv8i16rz:
  case AArch64_CMLEv8i8rz:
  case AArch64_CMLTv16i8rz:
  case AArch64_CMLTv1i64rz:
  case AArch64_CMLTv2i32rz:
  case AArch64_CMLTv2i64rz:
  case AArch64_CMLTv4i16rz:
  case AArch64_CMLTv4i32rz:
  case AArch64_CMLTv8i16rz:
  case AArch64_CMLTv8i8rz:
    AArch64_insert_detail_op_imm_at(MI, -1, 0);
    break;
  case AArch64_FCMEQ_PPzZ0_D:
  case AArch64_FCMEQ_PPzZ0_H:
  case AArch64_FCMEQ_PPzZ0_S:
  case AArch64_FCMEQv1i16rz:
  case AArch64_FCMEQv1i32rz:
  case AArch64_FCMEQv1i64rz:
  case AArch64_FCMEQv2i32rz:
  case AArch64_FCMEQv2i64rz:
  case AArch64_FCMEQv4i16rz:
  case AArch64_FCMEQv4i32rz:
  case AArch64_FCMEQv8i16rz:
  case AArch64_FCMGE_PPzZ0_D:
  case AArch64_FCMGE_PPzZ0_H:
  case AArch64_FCMGE_PPzZ0_S:
  case AArch64_FCMGEv1i16rz:
  case AArch64_FCMGEv1i32rz:
  case AArch64_FCMGEv1i64rz:
  case AArch64_FCMGEv2i32rz:
  case AArch64_FCMGEv2i64rz:
  case AArch64_FCMGEv4i16rz:
  case AArch64_FCMGEv4i32rz:
  case AArch64_FCMGEv8i16rz:
  case AArch64_FCMGT_PPzZ0_D:
  case AArch64_FCMGT_PPzZ0_H:
  case AArch64_FCMGT_PPzZ0_S:
  case AArch64_FCMGTv1i16rz:
  case AArch64_FCMGTv1i32rz:
  case AArch64_FCMGTv1i64rz:
  case AArch64_FCMGTv2i32rz:
  case AArch64_FCMGTv2i64rz:
  case AArch64_FCMGTv4i16rz:
  case AArch64_FCMGTv4i32rz:
  case AArch64_FCMGTv8i16rz:
  case AArch64_FCMLE_PPzZ0_D:
  case AArch64_FCMLE_PPzZ0_H:
  case AArch64_FCMLE_PPzZ0_S:
  case AArch64_FCMLEv1i16rz:
  case AArch64_FCMLEv1i32rz:
  case AArch64_FCMLEv1i64rz:
  case AArch64_FCMLEv2i32rz:
  case AArch64_FCMLEv2i64rz:
  case AArch64_FCMLEv4i16rz:
  case AArch64_FCMLEv4i32rz:
  case AArch64_FCMLEv8i16rz:
  case AArch64_FCMLT_PPzZ0_D:
  case AArch64_FCMLT_PPzZ0_H:
  case AArch64_FCMLT_PPzZ0_S:
  case AArch64_FCMLTv1i16rz:
  case AArch64_FCMLTv1i32rz:
  case AArch64_FCMLTv1i64rz:
  case AArch64_FCMLTv2i32rz:
  case AArch64_FCMLTv2i64rz:
  case AArch64_FCMLTv4i16rz:
  case AArch64_FCMLTv4i32rz:
  case AArch64_FCMLTv8i16rz:
  case AArch64_FCMNE_PPzZ0_D:
  case AArch64_FCMNE_PPzZ0_H:
  case AArch64_FCMNE_PPzZ0_S: {
    aarch64_sysop sysop = { 0 };
    sysop.imm.exactfpimm = AARCH64_EXACTFPIMM_ZERO;
    sysop.sub_type = AARCH64_OP_EXACTFPIMM;
    AArch64_insert_detail_op_sys(MI, -1, sysop, AARCH64_OP_SYSIMM);
    break;
  }
  }
}

#define ADD_ZA0_S \
  { \
    aarch64_op_sme za0_op = { \
      .type = AARCH64_SME_OP_TILE, \
      .tile = AARCH64_REG_ZAS0, \
      .slice_reg = AARCH64_REG_INVALID, \
      .slice_offset = { -1 }, \
      .has_range_offset = false, \
      .is_vertical = false, \
    }; \
    AArch64_insert_detail_op_sme(MI, -1, za0_op); \
    AArch64_get_detail_op(MI, -1)->vas = AARCH64LAYOUT_VL_S; \
    AArch64_get_detail_op(MI, -1)->access = CS_AC_WRITE; \
  }
#define ADD_ZA1_S \
  { \
    aarch64_op_sme za1_op = { \
      .type = AARCH64_SME_OP_TILE, \
      .tile = AARCH64_REG_ZAS1, \
      .slice_reg = AARCH64_REG_INVALID, \
      .slice_offset = { -1 }, \
      .has_range_offset = false, \
      .is_vertical = false, \
    }; \
    AArch64_insert_detail_op_sme(MI, -1, za1_op); \
    AArch64_get_detail_op(MI, -1)->vas = AARCH64LAYOUT_VL_S; \
    AArch64_get_detail_op(MI, -1)->access = CS_AC_WRITE; \
  }
#define ADD_ZA2_S \
  { \
    aarch64_op_sme za2_op = { \
      .type = AARCH64_SME_OP_TILE, \
      .tile = AARCH64_REG_ZAS2, \
      .slice_reg = AARCH64_REG_INVALID, \
      .slice_offset = { -1 }, \
      .has_range_offset = false, \
      .is_vertical = false, \
    }; \
    AArch64_insert_detail_op_sme(MI, -1, za2_op); \
    AArch64_get_detail_op(MI, -1)->vas = AARCH64LAYOUT_VL_S; \
    AArch64_get_detail_op(MI, -1)->access = CS_AC_WRITE; \
  }
#define ADD_ZA3_S \
  { \
    aarch64_op_sme za3_op = { \
      .type = AARCH64_SME_OP_TILE, \
      .tile = AARCH64_REG_ZAS3, \
      .slice_reg = AARCH64_REG_INVALID, \
      .slice_offset = { -1 }, \
      .has_range_offset = false, \
      .is_vertical = false, \
    }; \
    AArch64_insert_detail_op_sme(MI, -1, za3_op); \
    AArch64_get_detail_op(MI, -1)->vas = AARCH64LAYOUT_VL_S; \
    AArch64_get_detail_op(MI, -1)->access = CS_AC_WRITE; \
  }
#define ADD_ZA \
  { \
    aarch64_op_sme za_op = { \
      .type = AARCH64_SME_OP_TILE, \
      .tile = AARCH64_REG_ZA, \
      .slice_reg = AARCH64_REG_INVALID, \
      .slice_offset = { -1 }, \
      .has_range_offset = false, \
      .is_vertical = false, \
    }; \
    AArch64_insert_detail_op_sme(MI, -1, za_op); \
    AArch64_get_detail_op(MI, -1)->access = CS_AC_WRITE; \
  }

static void AArch64_add_not_defined_ops(MCInst *MI, const SStream *OS)
{
  if (!detail_is_set(MI))
    return;

  if (!MI->flat_insn->is_alias || !MI->flat_insn->usesAliasDetails) {
    add_non_alias_details(MI);
    return;
  }

  // Alias details
  switch (MI->flat_insn->alias_id) {
  default:
    return;
  case AARCH64_INS_ALIAS_ROR:
    if (AArch64_get_detail(MI)->op_count != 3) {
      return;
    }
    // The ROR alias doesn't set the shift value properly.
    // Correct it here.
    bool reg_shift = AArch64_get_detail_op(MI, -1)->type ==
         AARCH64_OP_REG;
    uint64_t shift = reg_shift ?
           AArch64_get_detail_op(MI, -1)->reg :
           AArch64_get_detail_op(MI, -1)->imm;
    cs_aarch64_op *op1 = AArch64_get_detail_op(MI, -2);
    op1->shift.type = reg_shift ? AARCH64_SFT_ROR_REG :
                AARCH64_SFT_ROR;
    op1->shift.value = shift;
    AArch64_dec_op_count(MI);
    break;
  case AARCH64_INS_ALIAS_FMOV:
    if (AArch64_get_detail_op(MI, -1)->type == AARCH64_OP_FP) {
      break;
    }
    AArch64_insert_detail_op_float_at(MI, -1, 0.0f, CS_AC_READ);
    break;
  case AARCH64_INS_ALIAS_LD1:
  case AARCH64_INS_ALIAS_LD1R:
  case AARCH64_INS_ALIAS_LD2:
  case AARCH64_INS_ALIAS_LD2R:
  case AARCH64_INS_ALIAS_LD3:
  case AARCH64_INS_ALIAS_LD3R:
  case AARCH64_INS_ALIAS_LD4:
  case AARCH64_INS_ALIAS_LD4R:
  case AARCH64_INS_ALIAS_ST1:
  case AARCH64_INS_ALIAS_ST2:
  case AARCH64_INS_ALIAS_ST3:
  case AARCH64_INS_ALIAS_ST4: {
    // Add post-index disp
    const char *disp_off = strrchr(OS->buffer, '#');
    if (!disp_off)
      return;
    unsigned disp = atoi(disp_off + 1);
    AArch64_get_detail_op(MI, -1)->type = AARCH64_OP_MEM;
    AArch64_get_detail_op(MI, -1)->mem.base =
      AArch64_get_detail_op(MI, -1)->reg;
    AArch64_get_detail_op(MI, -1)->mem.disp = disp;
    AArch64_get_detail(MI)->post_index = true;
    break;
  }
  case AARCH64_INS_ALIAS_GCSB:
    // TODO
    // Only CSYNC is defined in LLVM. So we need to add it.
    //     /* 2825 */ "gcsb dsync\0"
    break;
  case AARCH64_INS_ALIAS_SMSTART:
  case AARCH64_INS_ALIAS_SMSTOP: {
    const char *disp_off = NULL;
    disp_off = strstr(OS->buffer, "smstart\tza");
    if (disp_off) {
      aarch64_sysop sysop = { 0 };
      sysop.alias.svcr = AARCH64_SVCR_SVCRZA;
      sysop.sub_type = AARCH64_OP_SVCR;
      AArch64_insert_detail_op_sys(MI, -1, sysop,
                 AARCH64_OP_SYSALIAS);
      return;
    }
    disp_off = strstr(OS->buffer, "smstart\tsm");
    if (disp_off) {
      aarch64_sysop sysop = { 0 };
      sysop.alias.svcr = AARCH64_SVCR_SVCRSM;
      sysop.sub_type = AARCH64_OP_SVCR;
      AArch64_insert_detail_op_sys(MI, -1, sysop,
                 AARCH64_OP_SYSALIAS);
      return;
    }
    break;
  }
  case AARCH64_INS_ALIAS_ZERO: {
    // It is ugly, but the hard coded search patterns do it for now.
    const char *disp_off = NULL;

    disp_off = strstr(OS->buffer, "{za}");
    if (disp_off) {
      ADD_ZA;
      return;
    }
    disp_off = strstr(OS->buffer, "{za1.h}");
    if (disp_off) {
      aarch64_op_sme op = {
        .type = AARCH64_SME_OP_TILE,
        .tile = AARCH64_REG_ZAH1,
        .slice_reg = AARCH64_REG_INVALID,
        .slice_offset = { -1 },
        .has_range_offset = false,
        .is_vertical = false,
      };
      AArch64_insert_detail_op_sme(MI, -1, op);
      AArch64_get_detail_op(MI, -1)->vas = AARCH64LAYOUT_VL_H;
      AArch64_get_detail_op(MI, -1)->access = CS_AC_WRITE;
      return;
    }
    disp_off = strstr(OS->buffer, "{za0.h}");
    if (disp_off) {
      aarch64_op_sme op = {
        .type = AARCH64_SME_OP_TILE,
        .tile = AARCH64_REG_ZAH0,
        .slice_reg = AARCH64_REG_INVALID,
        .slice_offset = { -1 },
        .has_range_offset = false,
        .is_vertical = false,
      };
      AArch64_insert_detail_op_sme(MI, -1, op);
      AArch64_get_detail_op(MI, -1)->vas = AARCH64LAYOUT_VL_H;
      AArch64_get_detail_op(MI, -1)->access = CS_AC_WRITE;
      return;
    }
    disp_off = strstr(OS->buffer, "{za0.s}");
    if (disp_off) {
      ADD_ZA0_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za1.s}");
    if (disp_off) {
      ADD_ZA1_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za2.s}");
    if (disp_off) {
      ADD_ZA2_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za3.s}");
    if (disp_off) {
      ADD_ZA3_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za0.s,za1.s}");
    if (disp_off) {
      ADD_ZA0_S;
      ADD_ZA1_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za0.s,za3.s}");
    if (disp_off) {
      ADD_ZA0_S;
      ADD_ZA3_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za1.s,za2.s}");
    if (disp_off) {
      ADD_ZA1_S;
      ADD_ZA2_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za2.s,za3.s}");
    if (disp_off) {
      ADD_ZA2_S;
      ADD_ZA3_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za0.s,za1.s,za2.s}");
    if (disp_off) {
      ADD_ZA0_S;
      ADD_ZA1_S;
      ADD_ZA2_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za0.s,za1.s,za3.s}");
    if (disp_off) {
      ADD_ZA0_S;
      ADD_ZA1_S;
      ADD_ZA3_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za0.s,za2.s,za3.s}");
    if (disp_off) {
      ADD_ZA0_S;
      ADD_ZA2_S;
      ADD_ZA3_S;
      return;
    }
    disp_off = strstr(OS->buffer, "{za1.s,za2.s,za3.s}");
    if (disp_off) {
      ADD_ZA1_S;
      ADD_ZA2_S;
      ADD_ZA3_S;
      return;
    }
    break;
  }
  }
}

void AArch64_set_instr_map_data(MCInst *MI)
{
  map_cs_id(MI, aarch64_insns, ARR_SIZE(aarch64_insns));
  map_implicit_reads(MI, aarch64_insns);
  map_implicit_writes(MI, aarch64_insns);
  map_groups(MI, aarch64_insns);
}

bool AArch64_getInstruction(csh handle, const uint8_t *code, size_t code_len,
          MCInst *MI, uint16_t *size, uint64_t address,
          void *info)
{
  AArch64_init_cs_detail(MI);
  DecodeStatus Result = AArch64_LLVM_getInstruction(
    handle, code, code_len, MI, size, address, info);
  AArch64_set_instr_map_data(MI);
  if (Result == MCDisassembler_SoftFail) {
    MCInst_setSoftFail(MI);
  }
  return Result != MCDisassembler_Fail;
}

/// Patches the register names with Capstone specific alias.
/// Those are common alias for registers (e.g. r15 = pc)
/// which are not set in LLVM.
static void patch_cs_reg_alias(char *asm_str)
{
  bool skip_sub = false;
  char *x29 = strstr(asm_str, "x29");
  if (x29 > asm_str && strstr(asm_str, "0x29") == (x29 - 1)) {
    // Check for hex prefix
    skip_sub = true;
  }
  while (x29 && !skip_sub) {
    x29[0] = 'f';
    x29[1] = 'p';
    memmove(x29 + 2, x29 + 3, strlen(x29 + 3));
    asm_str[strlen(asm_str) - 1] = '\0';
    x29 = strstr(asm_str, "x29");
  }
  skip_sub = false;
  char *x30 = strstr(asm_str, "x30");
  if (x30 > asm_str && strstr(asm_str, "0x30") == (x30 - 1)) {
    // Check for hex prefix
    skip_sub = true;
  }
  while (x30 && !skip_sub) {
    x30[0] = 'l';
    x30[1] = 'r';
    memmove(x30 + 2, x30 + 3, strlen(x30 + 3));
    asm_str[strlen(asm_str) - 1] = '\0';
    x30 = strstr(asm_str, "x30");
  }
}

/// Adds group to the instruction which are not defined in LLVM.
static void AArch64_add_cs_groups(MCInst *MI)
{
  unsigned Opcode = MI->flat_insn->id;
  switch (Opcode) {
  default:
    return;
  case AARCH64_INS_SVC:
    add_group(MI, AARCH64_GRP_INT);
    break;
  case AARCH64_INS_SMC:
  case AARCH64_INS_MSR:
  case AARCH64_INS_MRS:
    add_group(MI, AARCH64_GRP_PRIVILEGE);
    break;
  case AARCH64_INS_RET:
  case AARCH64_INS_RETAA:
  case AARCH64_INS_RETAB:
    add_group(MI, AARCH64_GRP_RET);
    break;
  }
}

static void AArch64_correct_mem_access(MCInst *MI)
{
  if (!detail_is_set(MI))
    return;
  cs_ac_type access =
    aarch64_insns[MI->Opcode].suppl_info.aarch64.mem_acc;
  if (access == CS_AC_INVALID) {
    return;
  }
  for (int i = 0; i < AArch64_get_detail(MI)->op_count; ++i) {
    if (AArch64_get_detail_op(MI, -i)->type == AARCH64_OP_MEM) {
      AArch64_get_detail_op(MI, -i)->access = access;
      return;
    }
  }
}

void AArch64_printer(MCInst *MI, SStream *O, void * /* MCRegisterInfo* */ info)
{
  MCRegisterInfo *MRI = (MCRegisterInfo *)info;
  MI->MRI = MRI;
  MI->fillDetailOps = detail_is_set(MI);
  MI->flat_insn->usesAliasDetails = map_use_alias_details(MI);
  AArch64_LLVM_printInstruction(MI, O, info);
  if (detail_is_set(MI)) {
    if (AArch64_get_detail(MI)->is_doing_sme) {
      // Last operand still needs to be closed.
      AArch64_get_detail(MI)->is_doing_sme = false;
      AArch64_inc_op_count(MI);
    }
    AArch64_get_detail(MI)->post_index =
      AArch64_check_post_index_am(MI, O);
  }
  AArch64_check_updates_flags(MI);
  map_set_alias_id(MI, O, insn_alias_mnem_map,
       ARR_SIZE(insn_alias_mnem_map) - 1);
  int syntax_opt = MI->csh->syntax;
  if (syntax_opt & CS_OPT_SYNTAX_CS_REG_ALIAS)
    patch_cs_reg_alias(O->buffer);
  AArch64_add_not_defined_ops(MI, O);
  AArch64_add_cs_groups(MI);
  AArch64_add_vas(MI, O);
  AArch64_correct_mem_access(MI);
}

// given internal insn id, return public instruction info
void AArch64_get_insn_id(cs_struct *h, cs_insn *insn, unsigned int id)
{
  // Done after disassembly
  return;
}

static const char *const insn_name_maps[] = {
#include "AArch64GenCSMappingInsnName.inc"
};

const char *AArch64_insn_name(csh handle, unsigned int id)
{
#ifndef CAPSTONE_DIET
  if (id < AARCH64_INS_ALIAS_END && id > AARCH64_INS_ALIAS_BEGIN) {
    if (id - AARCH64_INS_ALIAS_BEGIN >=
        ARR_SIZE(insn_alias_mnem_map))
      return NULL;

    return insn_alias_mnem_map[id - AARCH64_INS_ALIAS_BEGIN - 1]
      .name;
  }
  if (id >= AARCH64_INS_ENDING)
    return NULL;

  if (id < ARR_SIZE(insn_name_maps))
    return insn_name_maps[id];

  // not found
  return NULL;
#else
  return NULL;
#endif
}

#ifndef CAPSTONE_DIET
static const name_map group_name_maps[] = {
  // generic groups
  { AARCH64_GRP_INVALID, NULL },
  { AARCH64_GRP_JUMP, "jump" },
  { AARCH64_GRP_CALL, "call" },
  { AARCH64_GRP_RET, "return" },
  { AARCH64_GRP_PRIVILEGE, "privilege" },
  { AARCH64_GRP_INT, "int" },
  { AARCH64_GRP_BRANCH_RELATIVE, "branch_relative" },

// architecture-specific groups
#include "AArch64GenCSFeatureName.inc"
};
#endif

const char *AArch64_group_name(csh handle, unsigned int id)
{
#ifndef CAPSTONE_DIET
  return id2name(group_name_maps, ARR_SIZE(group_name_maps), id);
#else
  return NULL;
#endif
}

// map instruction name to public instruction ID
aarch64_insn AArch64_map_insn(const char *name)
{
  unsigned int i;

  for (i = 1; i < ARR_SIZE(insn_name_maps); i++) {
    if (!strcmp(name, insn_name_maps[i]))
      return i;
  }

  // not found
  return AARCH64_INS_INVALID;
}

#ifndef CAPSTONE_DIET

static const map_insn_ops insn_operands[] = {
#include "AArch64GenCSMappingInsnOp.inc"
};

void AArch64_reg_access(const cs_insn *insn, cs_regs regs_read,
      uint8_t *regs_read_count, cs_regs regs_write,
      uint8_t *regs_write_count)
{
  uint8_t i;
  uint8_t read_count, write_count;
  cs_aarch64 *aarch64 = &(insn->detail->aarch64);

  read_count = insn->detail->regs_read_count;
  write_count = insn->detail->regs_write_count;

  // implicit registers
  memcpy(regs_read, insn->detail->regs_read,
         read_count * sizeof(insn->detail->regs_read[0]));
  memcpy(regs_write, insn->detail->regs_write,
         write_count * sizeof(insn->detail->regs_write[0]));

  // explicit registers
  for (i = 0; i < aarch64->op_count; i++) {
    cs_aarch64_op *op = &(aarch64->operands[i]);
    switch ((int)op->type) {
    case AARCH64_OP_REG:
      if ((op->access & CS_AC_READ) &&
          !arr_exist(regs_read, read_count, op->reg)) {
        regs_read[read_count] = (uint16_t)op->reg;
        read_count++;
      }
      if ((op->access & CS_AC_WRITE) &&
          !arr_exist(regs_write, write_count, op->reg)) {
        regs_write[write_count] = (uint16_t)op->reg;
        write_count++;
      }
      break;
    case AARCH64_OP_MEM:
      // registers appeared in memory references always being read
      if ((op->mem.base != AARCH64_REG_INVALID) &&
          !arr_exist(regs_read, read_count, op->mem.base)) {
        regs_read[read_count] = (uint16_t)op->mem.base;
        read_count++;
      }
      if ((op->mem.index != AARCH64_REG_INVALID) &&
          !arr_exist(regs_read, read_count, op->mem.index)) {
        regs_read[read_count] = (uint16_t)op->mem.index;
        read_count++;
      }
      if ((insn->detail->writeback) &&
          (op->mem.base != AARCH64_REG_INVALID) &&
          !arr_exist(regs_write, write_count, op->mem.base)) {
        regs_write[write_count] =
          (uint16_t)op->mem.base;
        write_count++;
      }
      break;
    case AARCH64_OP_SME:
      if ((op->access & CS_AC_READ) &&
          (op->sme.tile != AARCH64_REG_INVALID) &&
          !arr_exist(regs_read, read_count, op->sme.tile)) {
        regs_read[read_count] = (uint16_t)op->sme.tile;
        read_count++;
      }
      if ((op->access & CS_AC_WRITE) &&
          (op->sme.tile != AARCH64_REG_INVALID) &&
          !arr_exist(regs_write, write_count, op->sme.tile)) {
        regs_write[write_count] =
          (uint16_t)op->sme.tile;
        write_count++;
      }
      if ((op->sme.slice_reg != AARCH64_REG_INVALID) &&
          !arr_exist(regs_read, read_count,
               op->sme.slice_reg)) {
        regs_read[read_count] =
          (uint16_t)op->sme.slice_reg;
        read_count++;
      }
      break;
    case AARCH64_OP_PRED:
      if ((op->access & CS_AC_READ) &&
          (op->pred.reg != AARCH64_REG_INVALID) &&
          !arr_exist(regs_read, read_count, op->pred.reg)) {
        regs_read[read_count] = (uint16_t)op->pred.reg;
        read_count++;
      }
      if ((op->access & CS_AC_WRITE) &&
          (op->pred.reg != AARCH64_REG_INVALID) &&
          !arr_exist(regs_write, write_count, op->pred.reg)) {
        regs_write[write_count] =
          (uint16_t)op->pred.reg;
        write_count++;
      }
      if ((op->pred.vec_select != AARCH64_REG_INVALID) &&
          !arr_exist(regs_read, read_count,
               op->pred.vec_select)) {
        regs_read[read_count] =
          (uint16_t)op->pred.vec_select;
        read_count++;
      }
      break;
    default:
      break;
    }
    if (op->shift.type >= AARCH64_SFT_LSL_REG) {
      if (!arr_exist(regs_read, read_count,
               op->shift.value)) {
        regs_read[read_count] =
          (uint16_t)op->shift.value;
        read_count++;
      }
    }
  }

  switch (insn->alias_id) {
  default:
    break;
  case AARCH64_INS_ALIAS_RET:
    regs_read[read_count] = AARCH64_REG_X30;
    read_count++;
    break;
  }

  *regs_read_count = read_count;
  *regs_write_count = write_count;
}
#endif

static AArch64Layout_VectorLayout get_vl_by_suffix(const char suffix)
{
  switch (suffix) {
  default:
    return AARCH64LAYOUT_INVALID;
  case 'b':
  case 'B':
    return AARCH64LAYOUT_VL_B;
  case 'h':
  case 'H':
    return AARCH64LAYOUT_VL_H;
  case 's':
  case 'S':
    return AARCH64LAYOUT_VL_S;
  case 'd':
  case 'D':
    return AARCH64LAYOUT_VL_D;
  case 'q':
  case 'Q':
    return AARCH64LAYOUT_VL_Q;
  }
}

static unsigned get_vec_list_num_regs(MCInst *MI, unsigned Reg)
{
  // Work out how many registers there are in the list (if there is an actual
  // list).
  unsigned NumRegs = 1;
  if (MCRegisterClass_contains(
        MCRegisterInfo_getRegClass(MI->MRI, AArch64_DDRegClassID),
        Reg) ||
      MCRegisterClass_contains(
        MCRegisterInfo_getRegClass(MI->MRI, AArch64_ZPR2RegClassID),
        Reg) ||
      MCRegisterClass_contains(
        MCRegisterInfo_getRegClass(MI->MRI, AArch64_QQRegClassID),
        Reg) ||
      MCRegisterClass_contains(
        MCRegisterInfo_getRegClass(MI->MRI, AArch64_PPR2RegClassID),
        Reg) ||
      MCRegisterClass_contains(
        MCRegisterInfo_getRegClass(MI->MRI,
                 AArch64_ZPR2StridedRegClassID),
        Reg))
    NumRegs = 2;
  else if (MCRegisterClass_contains(
       MCRegisterInfo_getRegClass(MI->MRI,
                AArch64_DDDRegClassID),
       Reg) ||
     MCRegisterClass_contains(
       MCRegisterInfo_getRegClass(MI->MRI,
                AArch64_ZPR3RegClassID),
       Reg) ||
     MCRegisterClass_contains(
       MCRegisterInfo_getRegClass(MI->MRI,
                AArch64_QQQRegClassID),
       Reg))
    NumRegs = 3;
  else if (MCRegisterClass_contains(
       MCRegisterInfo_getRegClass(MI->MRI,
                AArch64_DDDDRegClassID),
       Reg) ||
     MCRegisterClass_contains(
       MCRegisterInfo_getRegClass(MI->MRI,
                AArch64_ZPR4RegClassID),
       Reg) ||
     MCRegisterClass_contains(
       MCRegisterInfo_getRegClass(MI->MRI,
                AArch64_QQQQRegClassID),
       Reg) ||
     MCRegisterClass_contains(
       MCRegisterInfo_getRegClass(
         MI->MRI, AArch64_ZPR4StridedRegClassID),
       Reg))
    NumRegs = 4;
  return NumRegs;
}

static unsigned get_vec_list_stride(MCInst *MI, unsigned Reg)
{
  unsigned Stride = 1;
  if (MCRegisterClass_contains(
        MCRegisterInfo_getRegClass(MI->MRI,
                 AArch64_ZPR2StridedRegClassID),
        Reg))
    Stride = 8;
  else if (MCRegisterClass_contains(
       MCRegisterInfo_getRegClass(
         MI->MRI, AArch64_ZPR4StridedRegClassID),
       Reg))
    Stride = 4;
  return Stride;
}

static unsigned get_vec_list_first_reg(MCInst *MI, unsigned RegL)
{
  unsigned Reg = RegL;
  // Now forget about the list and find out what the first register is.
  if (MCRegisterInfo_getSubReg(MI->MRI, RegL, AArch64_dsub0))
    Reg = MCRegisterInfo_getSubReg(MI->MRI, RegL, AArch64_dsub0);
  else if (MCRegisterInfo_getSubReg(MI->MRI, RegL, AArch64_qsub0))
    Reg = MCRegisterInfo_getSubReg(MI->MRI, RegL, AArch64_qsub0);
  else if (MCRegisterInfo_getSubReg(MI->MRI, RegL, AArch64_zsub0))
    Reg = MCRegisterInfo_getSubReg(MI->MRI, RegL, AArch64_zsub0);
  else if (MCRegisterInfo_getSubReg(MI->MRI, RegL, AArch64_psub0))
    Reg = MCRegisterInfo_getSubReg(MI->MRI, RegL, AArch64_psub0);

  // If it's a D-reg, we need to promote it to the equivalent Q-reg before
  // printing (otherwise getRegisterName fails).
  if (MCRegisterClass_contains(MCRegisterInfo_getRegClass(
               MI->MRI, AArch64_FPR64RegClassID),
             Reg)) {
    const MCRegisterClass *FPR128RC = MCRegisterInfo_getRegClass(
      MI->MRI, AArch64_FPR128RegClassID);
    Reg = MCRegisterInfo_getMatchingSuperReg(
      MI->MRI, Reg, AArch64_dsub, FPR128RC);
  }
  return Reg;
}

static bool is_vector_reg(unsigned Reg)
{
  if ((Reg >= AArch64_Q0) && (Reg <= AArch64_Q31))
    return true;
  else if ((Reg >= AArch64_Z0) && (Reg <= AArch64_Z31))
    return true;
  else if ((Reg >= AArch64_P0) && (Reg <= AArch64_P15))
    return true;
  return false;
}

static unsigned getNextVectorRegister(unsigned Reg, unsigned Stride /* = 1 */)
{
  while (Stride--) {
    if (!is_vector_reg(Reg)) {
      CS_ASSERT(0 && "Vector register expected!");
      return 0;
    }
    // Vector lists can wrap around.
    else if (Reg == AArch64_Q31)
      Reg = AArch64_Q0;
    // Vector lists can wrap around.
    else if (Reg == AArch64_Z31)
      Reg = AArch64_Z0;
    // Vector lists can wrap around.
    else if (Reg == AArch64_P15)
      Reg = AArch64_P0;
    else
      // Assume ordered registers
      ++Reg;
  }
  return Reg;
}

static aarch64_extender llvm_to_cs_ext(AArch64_AM_ShiftExtendType ExtType)
{
  switch (ExtType) {
  default:
    return AARCH64_EXT_INVALID;
  case AArch64_AM_UXTB:
    return AARCH64_EXT_UXTB;
  case AArch64_AM_UXTH:
    return AARCH64_EXT_UXTH;
  case AArch64_AM_UXTW:
    return AARCH64_EXT_UXTW;
  case AArch64_AM_UXTX:
    return AARCH64_EXT_UXTX;
  case AArch64_AM_SXTB:
    return AARCH64_EXT_SXTB;
  case AArch64_AM_SXTH:
    return AARCH64_EXT_SXTH;
  case AArch64_AM_SXTW:
    return AARCH64_EXT_SXTW;
  case AArch64_AM_SXTX:
    return AARCH64_EXT_SXTX;
  }
}

static aarch64_shifter llvm_to_cs_shift(AArch64_AM_ShiftExtendType ShiftExtType)
{
  switch (ShiftExtType) {
  default:
    return AARCH64_SFT_INVALID;
  case AArch64_AM_LSL:
    return AARCH64_SFT_LSL;
  case AArch64_AM_LSR:
    return AARCH64_SFT_LSR;
  case AArch64_AM_ASR:
    return AARCH64_SFT_ASR;
  case AArch64_AM_ROR:
    return AARCH64_SFT_ROR;
  case AArch64_AM_MSL:
    return AARCH64_SFT_MSL;
  }
}

/// Initializes or finishes a memory operand of Capstone (depending on \p
/// status). A memory operand in Capstone can be assembled by two LLVM operands.
/// E.g. the base register and the immediate disponent.
void AArch64_set_mem_access(MCInst *MI, bool status)
{
  if (!detail_is_set(MI))
    return;
  set_doing_mem(MI, status);
  if (status) {
    if (AArch64_get_detail(MI)->op_count > 0 &&
        AArch64_get_detail_op(MI, -1)->type == AARCH64_OP_MEM &&
        AArch64_get_detail_op(MI, -1)->mem.index ==
          AARCH64_REG_INVALID &&
        AArch64_get_detail_op(MI, -1)->mem.disp == 0) {
      // Previous memory operand not done yet. Select it.
      AArch64_dec_op_count(MI);
      return;
    }

    // Init a new one.
    AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_MEM;
    AArch64_get_detail_op(MI, 0)->mem.base = AARCH64_REG_INVALID;
    AArch64_get_detail_op(MI, 0)->mem.index = AARCH64_REG_INVALID;
    AArch64_get_detail_op(MI, 0)->mem.disp = 0;

#ifndef CAPSTONE_DIET
    uint8_t access =
      map_get_op_access(MI, AArch64_get_detail(MI)->op_count);
    AArch64_get_detail_op(MI, 0)->access = access;
#endif
  } else {
    // done, select the next operand slot
    AArch64_inc_op_count(MI);
  }
}

/// Common prefix for all AArch64_add_cs_detail_* functions
static bool add_cs_detail_begin(MCInst *MI, unsigned op_num)
{
  if (!detail_is_set(MI) || !map_fill_detail_ops(MI))
    return false;

  if (AArch64_get_detail(MI)->is_doing_sme) {
    // Unset the flag if there is no bound operand anymore.
    if (!(map_get_op_type(MI, op_num) & CS_OP_BOUND)) {
      AArch64_get_detail(MI)->is_doing_sme = false;
      AArch64_inc_op_count(MI);
    }
  }
  return true;
}

/// Fills cs_detail with the data of the operand.
/// This function handles operands which's original printer function has no
/// specialities.
void AArch64_add_cs_detail_0(MCInst *MI, aarch64_op_group op_group,
           unsigned OpNum)
{
  if (!add_cs_detail_begin(MI, OpNum))
    return;

  // Fill cs_detail
  switch (op_group) {
  default:
    printf("ERROR: Operand group %d not handled!\n", op_group);
    CS_ASSERT_RET(0);
  case AArch64_OP_GROUP_Operand: {
    cs_op_type primary_op_type = map_get_op_type(MI, OpNum) &
               ~(CS_OP_MEM | CS_OP_BOUND);
    switch (primary_op_type) {
    default:
      printf("Unhandled operand type 0x%x\n",
             primary_op_type);
      CS_ASSERT_RET(0);
    case AARCH64_OP_REG:
      AArch64_set_detail_op_reg(MI, OpNum,
              MCInst_getOpVal(MI, OpNum));
      break;
    case AARCH64_OP_IMM:
      AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
              MCInst_getOpVal(MI, OpNum));
      break;
    case AARCH64_OP_FP: {
      // printOperand does not handle FP operands. But sometimes
      // is used to print FP operands as normal immediate.
      AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_IMM;
      AArch64_get_detail_op(MI, 0)->imm =
        MCInst_getOpVal(MI, OpNum);
      AArch64_get_detail_op(MI, 0)->access =
        map_get_op_access(MI, OpNum);
      AArch64_inc_op_count(MI);
      break;
    }
    }
    break;
  }
  case AArch64_OP_GROUP_AddSubImm: {
    unsigned Val = (MCInst_getOpVal(MI, OpNum) & 0xfff);
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM, Val);
    // Shift is added in printShifter()
    break;
  }
  case AArch64_OP_GROUP_AdrLabel: {
    if (MCOperand_isImm(MCInst_getOperand(MI, OpNum))) {
      int64_t Offset = MCInst_getOpVal(MI, OpNum);
      AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
              (MI->address & -4) + Offset);
    } else {
      // Expression
      AArch64_set_detail_op_imm(
        MI, OpNum, AARCH64_OP_IMM,
        MCOperand_isImm(MCInst_getOperand(MI, OpNum)));
    }
    break;
  }
  case AArch64_OP_GROUP_AdrpLabel: {
    if (MCOperand_isImm(MCInst_getOperand(MI, OpNum))) {
      int64_t Offset = MCInst_getOpVal(MI, OpNum) * 4096;
      AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
              (MI->address & -4096) +
                Offset);
    } else {
      // Expression
      AArch64_set_detail_op_imm(
        MI, OpNum, AARCH64_OP_IMM,
        MCOperand_isImm(MCInst_getOperand(MI, OpNum)));
    }
    break;
  }
  case AArch64_OP_GROUP_AdrAdrpLabel: {
    if (!MCOperand_isImm(MCInst_getOperand(MI, OpNum))) {
      // Expression
      AArch64_set_detail_op_imm(
        MI, OpNum, AARCH64_OP_IMM,
        MCOperand_isImm(MCInst_getOperand(MI, OpNum)));
      break;
    }
    int64_t Offset = MCInst_getOpVal(MI, OpNum);
    uint64_t Address = MI->address;
    if (MCInst_getOpcode(MI) == AArch64_ADRP) {
      Offset = Offset * 4096;
      Address = Address & -4096;
    }
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
            Address + Offset);
    break;
  }
  case AArch64_OP_GROUP_AlignedLabel: {
    if (MCOperand_isImm(MCInst_getOperand(MI, OpNum))) {
      int64_t Offset = MCInst_getOpVal(MI, OpNum) * 4;
      AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
              MI->address + Offset);
    } else {
      // Expression
      AArch64_set_detail_op_imm(
        MI, OpNum, AARCH64_OP_IMM,
        MCOperand_isImm(MCInst_getOperand(MI, OpNum)));
    }
    break;
  }
  case AArch64_OP_GROUP_AMNoIndex: {
    AArch64_set_detail_op_mem(MI, OpNum,
            MCInst_getOpVal(MI, OpNum));
    break;
  }
  case AArch64_OP_GROUP_ArithExtend: {
    unsigned Val = MCInst_getOpVal(MI, OpNum);
    AArch64_AM_ShiftExtendType ExtType =
      AArch64_AM_getArithExtendType(Val);
    unsigned ShiftVal = AArch64_AM_getArithShiftValue(Val);

    AArch64_get_detail_op(MI, -1)->ext = llvm_to_cs_ext(ExtType);
    AArch64_get_detail_op(MI, -1)->shift.value = ShiftVal;
    AArch64_get_detail_op(MI, -1)->shift.type = AARCH64_SFT_LSL;
    break;
  }
  case AArch64_OP_GROUP_BarriernXSOption: {
    unsigned Val = MCInst_getOpVal(MI, OpNum);
    aarch64_sysop sysop = { 0 };
    const AArch64DBnXS_DBnXS *DB =
      AArch64DBnXS_lookupDBnXSByEncoding(Val);
    if (DB)
      sysop.imm.dbnxs = (aarch64_dbnxs)DB->SysImm.dbnxs;
    else
      sysop.imm.raw_val = Val;
    sysop.sub_type = AARCH64_OP_DBNXS;
    AArch64_set_detail_op_sys(MI, OpNum, sysop, AARCH64_OP_SYSIMM);
    break;
  }
  case AArch64_OP_GROUP_AppleSysBarrierOption: {
    // Proprietary stuff. We just add the
    // immediate here.
    unsigned Val = MCOperand_getImm(MCInst_getOperand(MI, OpNum));
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM, Val);
    break;
  }
  case AArch64_OP_GROUP_BarrierOption: {
    unsigned Val = MCOperand_getImm(MCInst_getOperand(MI, OpNum));
    unsigned Opcode = MCInst_getOpcode(MI);
    aarch64_sysop sysop = { 0 };

    if (Opcode == AArch64_ISB) {
      const AArch64ISB_ISB *ISB =
        AArch64ISB_lookupISBByEncoding(Val);
      if (ISB)
        sysop.alias.isb =
          (aarch64_isb)ISB->SysAlias.isb;
      else
        sysop.alias.raw_val = Val;
      sysop.sub_type = AARCH64_OP_ISB;
      AArch64_set_detail_op_sys(MI, OpNum, sysop,
              AARCH64_OP_SYSALIAS);
    } else if (Opcode == AArch64_TSB) {
      const AArch64TSB_TSB *TSB =
        AArch64TSB_lookupTSBByEncoding(Val);
      if (TSB)
        sysop.alias.tsb =
          (aarch64_tsb)TSB->SysAlias.tsb;
      else
        sysop.alias.raw_val = Val;
      sysop.sub_type = AARCH64_OP_TSB;
      AArch64_set_detail_op_sys(MI, OpNum, sysop,
              AARCH64_OP_SYSALIAS);
    } else {
      const AArch64DB_DB *DB =
        AArch64DB_lookupDBByEncoding(Val);
      if (DB)
        sysop.alias.db = (aarch64_db)DB->SysAlias.db;
      else
        sysop.alias.raw_val = Val;
      sysop.sub_type = AARCH64_OP_DB;
      AArch64_set_detail_op_sys(MI, OpNum, sysop,
              AARCH64_OP_SYSALIAS);
    }
    break;
  }
  case AArch64_OP_GROUP_BTIHintOp: {
    aarch64_sysop sysop = { 0 };
    unsigned btihintop = MCInst_getOpVal(MI, OpNum) ^ 32;
    const AArch64BTIHint_BTI *BTI =
      AArch64BTIHint_lookupBTIByEncoding(btihintop);
    if (BTI)
      sysop.alias.bti = (aarch64_bti)BTI->SysAlias.bti;
    else
      sysop.alias.raw_val = btihintop;
    sysop.sub_type = AARCH64_OP_BTI;
    AArch64_set_detail_op_sys(MI, OpNum, sysop,
            AARCH64_OP_SYSALIAS);
    break;
  }
  case AArch64_OP_GROUP_CondCode: {
    AArch64_get_detail(MI)->cc = MCInst_getOpVal(MI, OpNum);
    break;
  }
  case AArch64_OP_GROUP_ExtendedRegister: {
    AArch64_set_detail_op_reg(MI, OpNum,
            MCInst_getOpVal(MI, OpNum));
    break;
  }
  case AArch64_OP_GROUP_FPImmOperand: {
    MCOperand *MO = MCInst_getOperand(MI, (OpNum));
    float FPImm =
      MCOperand_isDFPImm(MO) ?
        BitsToDouble(MCOperand_getImm(MO)) :
        AArch64_AM_getFPImmFloat(MCOperand_getImm(MO));
    AArch64_set_detail_op_float(MI, OpNum, FPImm);
    break;
  }
  case AArch64_OP_GROUP_GPR64as32: {
    unsigned Reg = MCInst_getOpVal(MI, OpNum);
    AArch64_set_detail_op_reg(MI, OpNum, getWRegFromXReg(Reg));
    break;
  }
  case AArch64_OP_GROUP_GPR64x8: {
    unsigned Reg = MCInst_getOpVal(MI, (OpNum));
    Reg = MCRegisterInfo_getSubReg(MI->MRI, Reg, AArch64_x8sub_0);
    AArch64_set_detail_op_reg(MI, OpNum, Reg);
    break;
  }
  case AArch64_OP_GROUP_Imm:
  case AArch64_OP_GROUP_ImmHex:
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
            MCInst_getOpVal(MI, OpNum));
    break;
  case AArch64_OP_GROUP_ImplicitlyTypedVectorList:
    // The TypedVectorList implements the logic of implicitly typed operand.
    AArch64_add_cs_detail_2(
      MI, AArch64_OP_GROUP_TypedVectorList_0_b, OpNum, 0, 0);
    break;
  case AArch64_OP_GROUP_InverseCondCode: {
    AArch64CC_CondCode CC = (AArch64CC_CondCode)MCOperand_getImm(
      MCInst_getOperand(MI, (OpNum)));
    AArch64_get_detail(MI)->cc = AArch64CC_getInvertedCondCode(CC);
    break;
  }
  case AArch64_OP_GROUP_MatrixTile: {
    const char *RegName = AArch64_LLVM_getRegisterName(
      MCInst_getOpVal(MI, OpNum), AArch64_NoRegAltName);
    const char *Dot = strstr(RegName, ".");
    AArch64Layout_VectorLayout vas = AARCH64LAYOUT_INVALID;
    if (!Dot) {
      // The matrix dimensions are machine dependent.
      // Currently we do not support differentiation of machines.
      // So we just indicate the use of the complete matrix.
      vas = sme_reg_to_vas(MCInst_getOpVal(MI, OpNum));
    } else
      vas = get_vl_by_suffix(Dot[1]);
    AArch64_set_detail_op_sme(MI, OpNum, AARCH64_SME_MATRIX_TILE,
            vas);
    break;
  }
  case AArch64_OP_GROUP_MatrixTileList: {
    unsigned MaxRegs = 8;
    unsigned RegMask = MCInst_getOpVal(MI, (OpNum));

    for (unsigned I = 0; I < MaxRegs; ++I) {
      unsigned Reg = RegMask & (1 << I);
      if (Reg == 0)
        continue;
      AArch64_get_detail_op(MI, 0)->is_list_member = true;
      AArch64_set_detail_op_sme(MI, OpNum,
              AARCH64_SME_MATRIX_TILE_LIST,
              AARCH64LAYOUT_VL_D,
              (int)(AARCH64_REG_ZAD0 + I));
      AArch64_inc_op_count(MI);
    }
    AArch64_get_detail(MI)->is_doing_sme = false;
    break;
  }
  case AArch64_OP_GROUP_MRSSystemRegister:
  case AArch64_OP_GROUP_MSRSystemRegister: {
    unsigned Val = MCInst_getOpVal(MI, OpNum);
    const AArch64SysReg_SysReg *Reg =
      AArch64SysReg_lookupSysRegByEncoding(Val);
    bool Read = (op_group == AArch64_OP_GROUP_MRSSystemRegister) ?
            true :
            false;

    bool isValidSysReg =
      (Reg && (Read ? Reg->Readable : Reg->Writeable) &&
       AArch64_testFeatureList(MI->csh->mode,
             Reg->FeaturesRequired));

    if (Reg && !isValidSysReg)
      Reg = AArch64SysReg_lookupSysRegByName(Reg->AltName);
    aarch64_sysop sysop = { 0 };
    // If Reg is NULL it is a generic system register.
    if (Reg)
      sysop.reg.sysreg = (aarch64_sysreg)Reg->SysReg.sysreg;
    else {
      sysop.reg.raw_val = Val;
    }
    aarch64_op_type type =
      (op_group == AArch64_OP_GROUP_MRSSystemRegister) ?
        AARCH64_OP_REG_MRS :
        AARCH64_OP_REG_MSR;
    sysop.sub_type = type;
    AArch64_set_detail_op_sys(MI, OpNum, sysop, AARCH64_OP_SYSREG);
    break;
  }
  case AArch64_OP_GROUP_PSBHintOp: {
    unsigned psbhintop = MCInst_getOpVal(MI, OpNum);
    const AArch64PSBHint_PSB *PSB =
      AArch64PSBHint_lookupPSBByEncoding(psbhintop);
    aarch64_sysop sysop = { 0 };
    if (PSB)
      sysop.alias.psb = (aarch64_psb)PSB->SysAlias.psb;
    else
      sysop.alias.raw_val = psbhintop;
    sysop.sub_type = AARCH64_OP_PSB;
    AArch64_set_detail_op_sys(MI, OpNum, sysop,
            AARCH64_OP_SYSALIAS);
    break;
  }
  case AArch64_OP_GROUP_RPRFMOperand: {
    unsigned prfop = MCInst_getOpVal(MI, OpNum);
    const AArch64PRFM_PRFM *PRFM =
      AArch64PRFM_lookupPRFMByEncoding(prfop);
    aarch64_sysop sysop = { 0 };
    if (PRFM)
      sysop.alias.prfm = (aarch64_prfm)PRFM->SysAlias.prfm;
    else
      sysop.alias.raw_val = prfop;
    sysop.sub_type = AARCH64_OP_PRFM;
    AArch64_set_detail_op_sys(MI, OpNum, sysop,
            AARCH64_OP_SYSALIAS);
    break;
  }
  case AArch64_OP_GROUP_ShiftedRegister: {
    AArch64_set_detail_op_reg(MI, OpNum,
            MCInst_getOpVal(MI, OpNum));
    // Shift part is handled in printShifter()
    break;
  }
  case AArch64_OP_GROUP_Shifter: {
    unsigned Val = MCInst_getOpVal(MI, OpNum);
    AArch64_AM_ShiftExtendType ShExtType =
      AArch64_AM_getShiftType(Val);
    AArch64_get_detail_op(MI, -1)->ext = llvm_to_cs_ext(ShExtType);
    AArch64_get_detail_op(MI, -1)->shift.type =
      llvm_to_cs_shift(ShExtType);
    AArch64_get_detail_op(MI, -1)->shift.value =
      AArch64_AM_getShiftValue(Val);
    break;
  }
  case AArch64_OP_GROUP_SIMDType10Operand: {
    unsigned RawVal = MCInst_getOpVal(MI, OpNum);
    uint64_t Val = AArch64_AM_decodeAdvSIMDModImmType10(RawVal);
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM, Val);
    break;
  }
  case AArch64_OP_GROUP_SVCROp: {
    unsigned svcrop = MCInst_getOpVal(MI, OpNum);
    const AArch64SVCR_SVCR *SVCR =
      AArch64SVCR_lookupSVCRByEncoding(svcrop);
    aarch64_sysop sysop = { 0 };
    if (SVCR)
      sysop.alias.svcr = (aarch64_svcr)SVCR->SysAlias.svcr;
    else
      sysop.alias.raw_val = svcrop;
    sysop.sub_type = AARCH64_OP_SVCR;
    AArch64_set_detail_op_sys(MI, OpNum, sysop,
            AARCH64_OP_SYSALIAS);
    break;
  }
  case AArch64_OP_GROUP_SVEPattern: {
    unsigned Val = MCInst_getOpVal(MI, OpNum);
    const AArch64SVEPredPattern_SVEPREDPAT *Pat =
      AArch64SVEPredPattern_lookupSVEPREDPATByEncoding(Val);
    if (!Pat) {
      AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
              Val);
      break;
    }
    aarch64_sysop sysop = { 0 };
    sysop.alias = Pat->SysAlias;
    sysop.sub_type = AARCH64_OP_SVEPREDPAT;
    AArch64_set_detail_op_sys(MI, OpNum, sysop,
            AARCH64_OP_SYSALIAS);
    break;
  }
  case AArch64_OP_GROUP_SVEVecLenSpecifier: {
    unsigned Val = MCInst_getOpVal(MI, OpNum);
    // Pattern has only 1 bit
    if (Val > 1)
      CS_ASSERT_RET(0 && "Invalid vector length specifier");
    const AArch64SVEVecLenSpecifier_SVEVECLENSPECIFIER *Pat =
      AArch64SVEVecLenSpecifier_lookupSVEVECLENSPECIFIERByEncoding(
        Val);
    if (!Pat)
      break;
    aarch64_sysop sysop = { 0 };
    sysop.alias = Pat->SysAlias;
    sysop.sub_type = AARCH64_OP_SVEVECLENSPECIFIER;
    AArch64_set_detail_op_sys(MI, OpNum, sysop,
            AARCH64_OP_SYSALIAS);
    break;
  }
  case AArch64_OP_GROUP_SysCROperand: {
    uint64_t cimm = MCInst_getOpVal(MI, OpNum);
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_CIMM, cimm);
    break;
  }
  case AArch64_OP_GROUP_SyspXzrPair: {
    unsigned Reg = MCInst_getOpVal(MI, OpNum);
    AArch64_set_detail_op_reg(MI, OpNum, Reg);
    AArch64_set_detail_op_reg(MI, OpNum, Reg);
    break;
  }
  case AArch64_OP_GROUP_SystemPStateField: {
    unsigned Val = MCInst_getOpVal(MI, OpNum);

    aarch64_sysop sysop = { 0 };
    const AArch64PState_PStateImm0_15 *PStateImm15 =
      AArch64PState_lookupPStateImm0_15ByEncoding(Val);
    const AArch64PState_PStateImm0_1 *PStateImm1 =
      AArch64PState_lookupPStateImm0_1ByEncoding(Val);
    if (PStateImm15 &&
        AArch64_testFeatureList(MI->csh->mode,
              PStateImm15->FeaturesRequired)) {
      sysop.alias = PStateImm15->SysAlias;
      sysop.sub_type = AARCH64_OP_PSTATEIMM0_15;
      AArch64_set_detail_op_sys(MI, OpNum, sysop,
              AARCH64_OP_SYSALIAS);
    } else if (PStateImm1 &&
         AArch64_testFeatureList(
           MI->csh->mode,
           PStateImm1->FeaturesRequired)) {
      sysop.alias = PStateImm1->SysAlias;
      sysop.sub_type = AARCH64_OP_PSTATEIMM0_1;
      AArch64_set_detail_op_sys(MI, OpNum, sysop,
              AARCH64_OP_SYSALIAS);
    } else {
      AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
              Val);
    }
    break;
  }
  case AArch64_OP_GROUP_VRegOperand: {
    unsigned Reg = MCInst_getOpVal(MI, OpNum);
    AArch64_get_detail_op(MI, 0)->is_vreg = true;
    AArch64_set_detail_op_reg(MI, OpNum, Reg);
    break;
  }
  }
}

/// Fills cs_detail with the data of the operand.
/// This function handles operands which original printer function is a template
/// with one argument.
void AArch64_add_cs_detail_1(MCInst *MI, aarch64_op_group op_group,
           unsigned OpNum, uint64_t temp_arg_0)
{
  if (!add_cs_detail_begin(MI, OpNum))
    return;
  switch (op_group) {
  default:
    printf("ERROR: Operand group %d not handled!\n", op_group);
    CS_ASSERT_RET(0);
  case AArch64_OP_GROUP_GPRSeqPairsClassOperand_32:
  case AArch64_OP_GROUP_GPRSeqPairsClassOperand_64: {
    unsigned size = temp_arg_0;
    unsigned Reg = MCInst_getOpVal(MI, (OpNum));

    unsigned Sube = (size == 32) ? AArch64_sube32 : AArch64_sube64;
    unsigned Subo = (size == 32) ? AArch64_subo32 : AArch64_subo64;

    unsigned Even = MCRegisterInfo_getSubReg(MI->MRI, Reg, Sube);
    unsigned Odd = MCRegisterInfo_getSubReg(MI->MRI, Reg, Subo);
    AArch64_set_detail_op_reg(MI, OpNum, Even);
    AArch64_set_detail_op_reg(MI, OpNum, Odd);
    break;
  }
  case AArch64_OP_GROUP_Imm8OptLsl_int16_t:
  case AArch64_OP_GROUP_Imm8OptLsl_int32_t:
  case AArch64_OP_GROUP_Imm8OptLsl_int64_t:
  case AArch64_OP_GROUP_Imm8OptLsl_int8_t:
  case AArch64_OP_GROUP_Imm8OptLsl_uint16_t:
  case AArch64_OP_GROUP_Imm8OptLsl_uint32_t:
  case AArch64_OP_GROUP_Imm8OptLsl_uint64_t:
  case AArch64_OP_GROUP_Imm8OptLsl_uint8_t: {
    unsigned UnscaledVal = MCInst_getOpVal(MI, (OpNum));
    unsigned Shift = MCInst_getOpVal(MI, (OpNum + 1));

    if ((UnscaledVal == 0) &&
        (AArch64_AM_getShiftValue(Shift) != 0)) {
      AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
              UnscaledVal);
      // Shift is handled in printShifter()
      break;
    }

#define SCALE_SET(T) \
  do { \
    T Val; \
    if (CHAR(T) == 'i') /* Signed */ \
      Val = (int8_t)UnscaledVal * \
            (1 << AArch64_AM_getShiftValue(Shift)); \
    else \
      Val = (uint8_t)UnscaledVal * \
            (1 << AArch64_AM_getShiftValue(Shift)); \
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM, Val); \
  } while (0)

    switch (op_group) {
    default:
      CS_ASSERT_RET(
        0 &&
        "Operand group for Imm8OptLsl not handled.");
    case AArch64_OP_GROUP_Imm8OptLsl_int16_t: {
      SCALE_SET(int16_t);
      break;
    }
    case AArch64_OP_GROUP_Imm8OptLsl_int32_t: {
      SCALE_SET(int32_t);
      break;
    }
    case AArch64_OP_GROUP_Imm8OptLsl_int64_t: {
      SCALE_SET(int64_t);
      break;
    }
    case AArch64_OP_GROUP_Imm8OptLsl_int8_t: {
      SCALE_SET(int8_t);
      break;
    }
    case AArch64_OP_GROUP_Imm8OptLsl_uint16_t: {
      SCALE_SET(uint16_t);
      break;
    }
    case AArch64_OP_GROUP_Imm8OptLsl_uint32_t: {
      SCALE_SET(uint32_t);
      break;
    }
    case AArch64_OP_GROUP_Imm8OptLsl_uint64_t: {
      SCALE_SET(uint64_t);
      break;
    }
    case AArch64_OP_GROUP_Imm8OptLsl_uint8_t: {
      SCALE_SET(uint8_t);
      break;
    }
    }
    break;
  }
  case AArch64_OP_GROUP_ImmScale_16:
  case AArch64_OP_GROUP_ImmScale_2:
  case AArch64_OP_GROUP_ImmScale_3:
  case AArch64_OP_GROUP_ImmScale_32:
  case AArch64_OP_GROUP_ImmScale_4:
  case AArch64_OP_GROUP_ImmScale_8: {
    unsigned Scale = temp_arg_0;
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
            Scale * MCInst_getOpVal(MI, OpNum));
    break;
  }
  case AArch64_OP_GROUP_LogicalImm_int16_t:
  case AArch64_OP_GROUP_LogicalImm_int32_t:
  case AArch64_OP_GROUP_LogicalImm_int64_t:
  case AArch64_OP_GROUP_LogicalImm_int8_t: {
    unsigned TypeSize = temp_arg_0;
    uint64_t Val = AArch64_AM_decodeLogicalImmediate(
      MCInst_getOpVal(MI, OpNum), 8 * TypeSize);
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM, Val);
    break;
  }
  case AArch64_OP_GROUP_Matrix_0:
  case AArch64_OP_GROUP_Matrix_16:
  case AArch64_OP_GROUP_Matrix_32:
  case AArch64_OP_GROUP_Matrix_64: {
    unsigned EltSize = temp_arg_0;
    AArch64_set_detail_op_sme(MI, OpNum, AARCH64_SME_MATRIX_TILE,
            (AArch64Layout_VectorLayout)EltSize);
    break;
  }
  case AArch64_OP_GROUP_MatrixIndex_0:
  case AArch64_OP_GROUP_MatrixIndex_1:
  case AArch64_OP_GROUP_MatrixIndex_8: {
    unsigned scale = temp_arg_0;
    if (AArch64_get_detail_op(MI, 0)->type == AARCH64_OP_SME) {
      // The index is part of an SME matrix
      AArch64_set_detail_op_sme(
        MI, OpNum, AARCH64_SME_MATRIX_SLICE_OFF,
        AARCH64LAYOUT_INVALID,
        (uint32_t)(MCInst_getOpVal(MI, OpNum) * scale));
    } else if (AArch64_get_detail_op(MI, 0)->type ==
         AARCH64_OP_PRED) {
      // The index is part of a predicate
      AArch64_set_detail_op_pred(MI, OpNum);
    } else {
      // The index is used for an SVE2 instruction.
      AArch64_set_detail_op_imm(
        MI, OpNum, AARCH64_OP_IMM,
        scale * MCInst_getOpVal(MI, OpNum));
    }
    break;
  }
  case AArch64_OP_GROUP_MatrixTileVector_0:
  case AArch64_OP_GROUP_MatrixTileVector_1: {
    bool isVertical = temp_arg_0;
    const char *RegName = AArch64_LLVM_getRegisterName(
      MCInst_getOpVal(MI, OpNum), AArch64_NoRegAltName);
    const char *Dot = strstr(RegName, ".");
    AArch64Layout_VectorLayout vas = AARCH64LAYOUT_INVALID;
    if (!Dot) {
      // The matrix dimensions are machine dependent.
      // Currently we do not support differentiation of machines.
      // So we just indicate the use of the complete matrix.
      vas = sme_reg_to_vas(MCInst_getOpVal(MI, OpNum));
    } else
      vas = get_vl_by_suffix(Dot[1]);
    setup_sme_operand(MI);
    AArch64_set_detail_op_sme(MI, OpNum, AARCH64_SME_MATRIX_TILE,
            vas);
    AArch64_get_detail_op(MI, 0)->sme.is_vertical = isVertical;
    break;
  }
  case AArch64_OP_GROUP_PostIncOperand_1:
  case AArch64_OP_GROUP_PostIncOperand_12:
  case AArch64_OP_GROUP_PostIncOperand_16:
  case AArch64_OP_GROUP_PostIncOperand_2:
  case AArch64_OP_GROUP_PostIncOperand_24:
  case AArch64_OP_GROUP_PostIncOperand_3:
  case AArch64_OP_GROUP_PostIncOperand_32:
  case AArch64_OP_GROUP_PostIncOperand_4:
  case AArch64_OP_GROUP_PostIncOperand_48:
  case AArch64_OP_GROUP_PostIncOperand_6:
  case AArch64_OP_GROUP_PostIncOperand_64:
  case AArch64_OP_GROUP_PostIncOperand_8: {
    uint64_t Imm = temp_arg_0;
    unsigned Reg = MCInst_getOpVal(MI, OpNum);
    if (Reg == AArch64_XZR) {
      AArch64_get_detail_op(MI, -1)->mem.disp = Imm;
      AArch64_get_detail(MI)->post_index = true;
      AArch64_inc_op_count(MI);
    } else
      AArch64_set_detail_op_reg(MI, OpNum, Reg);
    break;
  }
  case AArch64_OP_GROUP_PredicateAsCounter_0:
  case AArch64_OP_GROUP_PredicateAsCounter_16:
  case AArch64_OP_GROUP_PredicateAsCounter_32:
  case AArch64_OP_GROUP_PredicateAsCounter_64:
  case AArch64_OP_GROUP_PredicateAsCounter_8: {
    unsigned EltSize = temp_arg_0;
    AArch64_get_detail_op(MI, 0)->vas = EltSize;
    AArch64_set_detail_op_reg(MI, OpNum,
            MCInst_getOpVal(MI, OpNum));
    break;
  }
  case AArch64_OP_GROUP_PrefetchOp_0:
  case AArch64_OP_GROUP_PrefetchOp_1: {
    bool IsSVEPrefetch = (bool)temp_arg_0;
    unsigned prfop = MCInst_getOpVal(MI, (OpNum));
    aarch64_sysop sysop = { 0 };
    if (IsSVEPrefetch) {
      const AArch64SVEPRFM_SVEPRFM *PRFM =
        AArch64SVEPRFM_lookupSVEPRFMByEncoding(prfop);
      if (PRFM) {
        sysop.alias = PRFM->SysAlias;
        sysop.sub_type = AARCH64_OP_SVEPRFM;
        AArch64_set_detail_op_sys(MI, OpNum, sysop,
                AARCH64_OP_SYSALIAS);
        break;
      }
    } else {
      const AArch64PRFM_PRFM *PRFM =
        AArch64PRFM_lookupPRFMByEncoding(prfop);
      if (PRFM &&
          AArch64_testFeatureList(MI->csh->mode,
                PRFM->FeaturesRequired)) {
        sysop.alias = PRFM->SysAlias;
        sysop.sub_type = AARCH64_OP_PRFM;
        AArch64_set_detail_op_sys(MI, OpNum, sysop,
                AARCH64_OP_SYSALIAS);
        break;
      }
    }
    AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_IMM;
    AArch64_get_detail_op(MI, 0)->imm = prfop;
    AArch64_get_detail_op(MI, 0)->access =
      map_get_op_access(MI, OpNum);
    AArch64_inc_op_count(MI);
    break;
  }
  case AArch64_OP_GROUP_SImm_16:
  case AArch64_OP_GROUP_SImm_8: {
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
            MCInst_getOpVal(MI, OpNum));
    break;
  }
  case AArch64_OP_GROUP_SVELogicalImm_int16_t:
  case AArch64_OP_GROUP_SVELogicalImm_int32_t:
  case AArch64_OP_GROUP_SVELogicalImm_int64_t: {
    // General issue here that we do not save the operand type
    // for each operand. So we choose the largest type.
    uint64_t Val = MCInst_getOpVal(MI, OpNum);
    uint64_t DecodedVal =
      AArch64_AM_decodeLogicalImmediate(Val, 64);
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
            DecodedVal);
    break;
  }
  case AArch64_OP_GROUP_SVERegOp_0:
  case AArch64_OP_GROUP_SVERegOp_b:
  case AArch64_OP_GROUP_SVERegOp_d:
  case AArch64_OP_GROUP_SVERegOp_h:
  case AArch64_OP_GROUP_SVERegOp_q:
  case AArch64_OP_GROUP_SVERegOp_s: {
    char Suffix = (char)temp_arg_0;
    AArch64_get_detail_op(MI, 0)->vas = get_vl_by_suffix(Suffix);
    AArch64_set_detail_op_reg(MI, OpNum,
            MCInst_getOpVal(MI, OpNum));
    break;
  }
  case AArch64_OP_GROUP_UImm12Offset_1:
  case AArch64_OP_GROUP_UImm12Offset_16:
  case AArch64_OP_GROUP_UImm12Offset_2:
  case AArch64_OP_GROUP_UImm12Offset_4:
  case AArch64_OP_GROUP_UImm12Offset_8: {
    // Otherwise it is an expression. For which we only add the immediate
    unsigned Scale = MCOperand_isImm(MCInst_getOperand(MI, OpNum)) ?
           temp_arg_0 :
           1;
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM,
            Scale * MCInst_getOpVal(MI, OpNum));
    break;
  }
  case AArch64_OP_GROUP_VectorIndex_1:
  case AArch64_OP_GROUP_VectorIndex_8: {
    CS_ASSERT_RET(AArch64_get_detail(MI)->op_count > 0);
    unsigned Scale = temp_arg_0;
    unsigned VIndex = Scale * MCInst_getOpVal(MI, OpNum);
    // The index can either be for one operand, or for each operand of a list.
    if (!AArch64_get_detail_op(MI, -1)->is_list_member) {
      AArch64_get_detail_op(MI, -1)->vector_index = VIndex;
      break;
    }
    for (int i = AArch64_get_detail(MI)->op_count - 1; i >= 0;
         --i) {
      if (!AArch64_get_detail(MI)->operands[i].is_list_member)
        break;
      AArch64_get_detail(MI)->operands[i].vector_index =
        VIndex;
    }
    break;
  }
  case AArch64_OP_GROUP_ZPRasFPR_128:
  case AArch64_OP_GROUP_ZPRasFPR_16:
  case AArch64_OP_GROUP_ZPRasFPR_32:
  case AArch64_OP_GROUP_ZPRasFPR_64:
  case AArch64_OP_GROUP_ZPRasFPR_8: {
    unsigned Base = AArch64_NoRegister;
    unsigned Width = temp_arg_0;
    switch (Width) {
    case 8:
      Base = AArch64_B0;
      break;
    case 16:
      Base = AArch64_H0;
      break;
    case 32:
      Base = AArch64_S0;
      break;
    case 64:
      Base = AArch64_D0;
      break;
    case 128:
      Base = AArch64_Q0;
      break;
    default:
      CS_ASSERT_RET(0 && "Unsupported width");
    }
    unsigned Reg = MCInst_getOpVal(MI, (OpNum));
    AArch64_set_detail_op_reg(MI, OpNum, Reg - AArch64_Z0 + Base);
    break;
  }
  }
}

/// Fills cs_detail with the data of the operand.
/// This function handles operands which original printer function is a template
/// with two arguments.
void AArch64_add_cs_detail_2(MCInst *MI, aarch64_op_group op_group,
           unsigned OpNum, uint64_t temp_arg_0,
           uint64_t temp_arg_1)
{
  if (!add_cs_detail_begin(MI, OpNum))
    return;
  switch (op_group) {
  default:
    printf("ERROR: Operand group %d not handled!\n", op_group);
    CS_ASSERT_RET(0);
  case AArch64_OP_GROUP_ComplexRotationOp_180_90:
  case AArch64_OP_GROUP_ComplexRotationOp_90_0: {
    unsigned Angle = temp_arg_0;
    unsigned Remainder = temp_arg_1;
    unsigned Imm = (MCInst_getOpVal(MI, OpNum) * Angle) + Remainder;
    AArch64_set_detail_op_imm(MI, OpNum, AARCH64_OP_IMM, Imm);
    break;
  }
  case AArch64_OP_GROUP_ExactFPImm_AArch64ExactFPImm_half_AArch64ExactFPImm_one:
  case AArch64_OP_GROUP_ExactFPImm_AArch64ExactFPImm_half_AArch64ExactFPImm_two:
  case AArch64_OP_GROUP_ExactFPImm_AArch64ExactFPImm_zero_AArch64ExactFPImm_one: {
    aarch64_exactfpimm ImmIs0 = temp_arg_0;
    aarch64_exactfpimm ImmIs1 = temp_arg_1;
    const AArch64ExactFPImm_ExactFPImm *Imm0Desc =
      AArch64ExactFPImm_lookupExactFPImmByEnum(ImmIs0);
    const AArch64ExactFPImm_ExactFPImm *Imm1Desc =
      AArch64ExactFPImm_lookupExactFPImmByEnum(ImmIs1);
    unsigned Val = MCInst_getOpVal(MI, (OpNum));
    aarch64_sysop sysop = { 0 };
    sysop.imm = Val ? Imm1Desc->SysImm : Imm0Desc->SysImm;
    sysop.sub_type = AARCH64_OP_EXACTFPIMM;
    AArch64_set_detail_op_sys(MI, OpNum, sysop, AARCH64_OP_SYSIMM);
    break;
  }
  case AArch64_OP_GROUP_ImmRangeScale_2_1:
  case AArch64_OP_GROUP_ImmRangeScale_4_3: {
    uint64_t Scale = temp_arg_0;
    uint64_t Offset = temp_arg_1;
    unsigned FirstImm = Scale * MCInst_getOpVal(MI, (OpNum));
    AArch64_set_detail_op_imm_range(MI, OpNum, FirstImm,
            FirstImm + Offset);
    break;
  }
  case AArch64_OP_GROUP_MemExtend_w_128:
  case AArch64_OP_GROUP_MemExtend_w_16:
  case AArch64_OP_GROUP_MemExtend_w_32:
  case AArch64_OP_GROUP_MemExtend_w_64:
  case AArch64_OP_GROUP_MemExtend_w_8:
  case AArch64_OP_GROUP_MemExtend_x_128:
  case AArch64_OP_GROUP_MemExtend_x_16:
  case AArch64_OP_GROUP_MemExtend_x_32:
  case AArch64_OP_GROUP_MemExtend_x_64:
  case AArch64_OP_GROUP_MemExtend_x_8: {
    char SrcRegKind = (char)temp_arg_0;
    unsigned ExtWidth = temp_arg_1;
    bool SignExtend = MCInst_getOpVal(MI, OpNum);
    bool DoShift = MCInst_getOpVal(MI, OpNum + 1);
    AArch64_set_detail_shift_ext(MI, OpNum, SignExtend, DoShift,
               ExtWidth, SrcRegKind);
    break;
  }
  case AArch64_OP_GROUP_TypedVectorList_0_b:
  case AArch64_OP_GROUP_TypedVectorList_0_d:
  case AArch64_OP_GROUP_TypedVectorList_0_h:
  case AArch64_OP_GROUP_TypedVectorList_0_q:
  case AArch64_OP_GROUP_TypedVectorList_0_s:
  case AArch64_OP_GROUP_TypedVectorList_0_0:
  case AArch64_OP_GROUP_TypedVectorList_16_b:
  case AArch64_OP_GROUP_TypedVectorList_1_d:
  case AArch64_OP_GROUP_TypedVectorList_2_d:
  case AArch64_OP_GROUP_TypedVectorList_2_s:
  case AArch64_OP_GROUP_TypedVectorList_4_h:
  case AArch64_OP_GROUP_TypedVectorList_4_s:
  case AArch64_OP_GROUP_TypedVectorList_8_b:
  case AArch64_OP_GROUP_TypedVectorList_8_h: {
    uint8_t NumLanes = (uint8_t)temp_arg_0;
    char LaneKind = (char)temp_arg_1;
    uint16_t Pair = ((NumLanes << 8) | LaneKind);

    AArch64Layout_VectorLayout vas = AARCH64LAYOUT_INVALID;
    switch (Pair) {
    default:
      printf("Typed vector list with NumLanes = %d and LaneKind = %c not handled.\n",
             NumLanes, LaneKind);
      CS_ASSERT_RET(0);
    case ((8 << 8) | 'b'):
      vas = AARCH64LAYOUT_VL_8B;
      break;
    case ((4 << 8) | 'h'):
      vas = AARCH64LAYOUT_VL_4H;
      break;
    case ((2 << 8) | 's'):
      vas = AARCH64LAYOUT_VL_2S;
      break;
    case ((1 << 8) | 'd'):
      vas = AARCH64LAYOUT_VL_1D;
      break;
    case ((16 << 8) | 'b'):
      vas = AARCH64LAYOUT_VL_16B;
      break;
    case ((8 << 8) | 'h'):
      vas = AARCH64LAYOUT_VL_8H;
      break;
    case ((4 << 8) | 's'):
      vas = AARCH64LAYOUT_VL_4S;
      break;
    case ((2 << 8) | 'd'):
      vas = AARCH64LAYOUT_VL_2D;
      break;
    case 'b':
      vas = AARCH64LAYOUT_VL_B;
      break;
    case 'h':
      vas = AARCH64LAYOUT_VL_H;
      break;
    case 's':
      vas = AARCH64LAYOUT_VL_S;
      break;
    case 'd':
      vas = AARCH64LAYOUT_VL_D;
      break;
    case 'q':
      vas = AARCH64LAYOUT_VL_Q;
      break;
    case '0':
      // Implicitly Typed register
      break;
    }

    unsigned Reg = MCOperand_getReg(MCInst_getOperand(MI, OpNum));
    unsigned NumRegs = get_vec_list_num_regs(MI, Reg);
    unsigned Stride = get_vec_list_stride(MI, Reg);
    Reg = get_vec_list_first_reg(MI, Reg);

    if ((MCRegisterClass_contains(
           MCRegisterInfo_getRegClass(MI->MRI,
              AArch64_ZPRRegClassID),
           Reg) ||
         MCRegisterClass_contains(
           MCRegisterInfo_getRegClass(MI->MRI,
              AArch64_PPRRegClassID),
           Reg)) &&
        NumRegs > 1 && Stride == 1 &&
        Reg < getNextVectorRegister(Reg, NumRegs - 1)) {
      AArch64_get_detail_op(MI, 0)->is_list_member = true;
      AArch64_get_detail_op(MI, 0)->vas = vas;
      AArch64_set_detail_op_reg(MI, OpNum, Reg);
      if (NumRegs > 1) {
        // Add all registers of the list to the details.
        for (size_t i = 0; i < NumRegs - 1; ++i) {
          AArch64_get_detail_op(MI, 0)
            ->is_list_member = true;
          AArch64_get_detail_op(MI, 0)->vas = vas;
          AArch64_set_detail_op_reg(
            MI, OpNum,
            getNextVectorRegister(Reg + i,
                      1));
        }
      }
    } else {
      for (unsigned i = 0; i < NumRegs;
           ++i, Reg = getNextVectorRegister(Reg, Stride)) {
        if (!(MCRegisterClass_contains(
                MCRegisterInfo_getRegClass(
                  MI->MRI,
                  AArch64_ZPRRegClassID),
                Reg) ||
              MCRegisterClass_contains(
                MCRegisterInfo_getRegClass(
                  MI->MRI,
                  AArch64_PPRRegClassID),
                Reg))) {
          AArch64_get_detail_op(MI, 0)->is_vreg =
            true;
        }
        AArch64_get_detail_op(MI, 0)->is_list_member =
          true;
        AArch64_get_detail_op(MI, 0)->vas = vas;
        AArch64_set_detail_op_reg(MI, OpNum, Reg);
      }
    }
  }
  }
}

/// Fills cs_detail with the data of the operand.
/// This function handles operands which original printer function is a template
/// with four arguments.
void AArch64_add_cs_detail_4(MCInst *MI, aarch64_op_group op_group,
           unsigned OpNum, uint64_t temp_arg_0,
           uint64_t temp_arg_1, uint64_t temp_arg_2,
           uint64_t temp_arg_3)
{
  if (!add_cs_detail_begin(MI, OpNum))
    return;
  switch (op_group) {
  default:
    printf("ERROR: Operand group %d not handled!\n", op_group);
    CS_ASSERT_RET(0);
  case AArch64_OP_GROUP_RegWithShiftExtend_0_128_x_0:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_16_w_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_16_w_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_16_x_0:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_16_x_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_16_x_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_32_w_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_32_w_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_32_x_0:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_32_x_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_32_x_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_64_w_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_64_w_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_64_x_0:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_64_x_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_64_x_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_8_w_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_8_w_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_8_x_0:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_8_x_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_0_8_x_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_1_16_w_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_1_16_w_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_1_32_w_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_1_32_w_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_1_64_w_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_1_64_w_s:
  case AArch64_OP_GROUP_RegWithShiftExtend_1_8_w_d:
  case AArch64_OP_GROUP_RegWithShiftExtend_1_8_w_s: {
    // signed (s) and unsigned (u) extend
    bool SignExtend = (bool)temp_arg_0;
    // Extend width
    int ExtWidth = (int)temp_arg_1;
    // w = word, x = doubleword
    char SrcRegKind = (char)temp_arg_2;
    // Vector register element/arrangement specifier:
    // B = 8bit, H = 16bit, S = 32bit, D = 64bit, Q = 128bit
    // No suffix = complete register
    // According to: ARM Reference manual supplement, doc number: DDI 0584
    char Suffix = (char)temp_arg_3;

    // Register will be added in printOperand() afterwards. Here we only handle
    // shift and extend.
    AArch64_get_detail_op(MI, -1)->vas = get_vl_by_suffix(Suffix);

    bool DoShift = ExtWidth != 8;
    if (!(SignExtend || DoShift || SrcRegKind == 'w'))
      return;

    AArch64_set_detail_shift_ext(MI, OpNum, SignExtend, DoShift,
               ExtWidth, SrcRegKind);
    break;
  }
  }
}

/// Adds a register AArch64 operand at position OpNum and increases the op_count by
/// one.
void AArch64_set_detail_op_reg(MCInst *MI, unsigned OpNum, aarch64_reg Reg)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  if (Reg == AARCH64_REG_ZA ||
      (Reg >= AARCH64_REG_ZAB0 && Reg < AARCH64_REG_ZT0)) {
    // A tile register should be treated as SME operand.
    AArch64_set_detail_op_sme(MI, OpNum, AARCH64_SME_MATRIX_TILE,
            sme_reg_to_vas(Reg));
    return;
  } else if (((Reg >= AARCH64_REG_P0) && (Reg <= AARCH64_REG_P15)) ||
       ((Reg >= AARCH64_REG_PN0) && (Reg <= AARCH64_REG_PN15))) {
    // SME/SVE predicate register.
    AArch64_set_detail_op_pred(MI, OpNum);
    return;
  } else if (AArch64_get_detail(MI)->is_doing_sme) {
    CS_ASSERT_RET(map_get_op_type(MI, OpNum) & CS_OP_BOUND);
    if (AArch64_get_detail_op(MI, 0)->type == AARCH64_OP_SME) {
      AArch64_set_detail_op_sme(MI, OpNum,
              AARCH64_SME_MATRIX_SLICE_REG,
              AARCH64LAYOUT_INVALID);
    } else if (AArch64_get_detail_op(MI, 0)->type ==
         AARCH64_OP_PRED) {
      AArch64_set_detail_op_pred(MI, OpNum);
    } else {
      CS_ASSERT_RET(0 && "Unkown SME/SVE operand type");
    }
    return;
  }
  if (map_get_op_type(MI, OpNum) & CS_OP_MEM) {
    AArch64_set_detail_op_mem(MI, OpNum, Reg);
    return;
  }

  CS_ASSERT_RET(!(map_get_op_type(MI, OpNum) & CS_OP_BOUND));
  CS_ASSERT_RET(!(map_get_op_type(MI, OpNum) & CS_OP_MEM));
  CS_ASSERT_RET(map_get_op_type(MI, OpNum) == CS_OP_REG);

  AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_REG;
  AArch64_get_detail_op(MI, 0)->reg = Reg;
  AArch64_get_detail_op(MI, 0)->access = map_get_op_access(MI, OpNum);
  AArch64_inc_op_count(MI);
}

/// Check if the previous operand is a memory operand
/// with only the base register set AND if this base register
/// is write-back.
/// This indicates the following immediate is a post-indexed
/// memory offset.
static bool prev_is_membase_wb(MCInst *MI)
{
  return AArch64_get_detail(MI)->op_count > 0 &&
         AArch64_get_detail_op(MI, -1)->type == AARCH64_OP_MEM &&
         AArch64_get_detail_op(MI, -1)->mem.disp == 0 &&
         get_detail(MI)->writeback;
}

/// Adds an immediate AArch64 operand at position OpNum and increases the op_count
/// by one.
void AArch64_set_detail_op_imm(MCInst *MI, unsigned OpNum,
             aarch64_op_type ImmType, int64_t Imm)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  if (AArch64_get_detail(MI)->is_doing_sme) {
    CS_ASSERT_RET(map_get_op_type(MI, OpNum) & CS_OP_BOUND);
    if (AArch64_get_detail_op(MI, 0)->type == AARCH64_OP_SME) {
      AArch64_set_detail_op_sme(MI, OpNum,
              AARCH64_SME_MATRIX_SLICE_OFF,
              AARCH64LAYOUT_INVALID,
              (uint32_t)1);
    } else if (AArch64_get_detail_op(MI, 0)->type ==
         AARCH64_OP_PRED) {
      AArch64_set_detail_op_pred(MI, OpNum);
    } else {
      CS_ASSERT_RET(0 && "Unkown SME operand type");
    }
    return;
  }
  if (map_get_op_type(MI, OpNum) & CS_OP_MEM || prev_is_membase_wb(MI)) {
    AArch64_set_detail_op_mem(MI, OpNum, Imm);
    return;
  }

  CS_ASSERT_RET(!(map_get_op_type(MI, OpNum) & CS_OP_MEM));
  CS_ASSERT_RET((map_get_op_type(MI, OpNum) & ~CS_OP_BOUND) == CS_OP_IMM);
  CS_ASSERT_RET(ImmType == AARCH64_OP_IMM || ImmType == AARCH64_OP_CIMM);

  AArch64_get_detail_op(MI, 0)->type = ImmType;
  AArch64_get_detail_op(MI, 0)->imm = Imm;
  AArch64_get_detail_op(MI, 0)->access = map_get_op_access(MI, OpNum);
  AArch64_inc_op_count(MI);
}

void AArch64_set_detail_op_imm_range(MCInst *MI, unsigned OpNum,
             uint32_t FirstImm, uint32_t Offset)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  if (AArch64_get_detail(MI)->is_doing_sme) {
    CS_ASSERT_RET(map_get_op_type(MI, OpNum) & CS_OP_BOUND);
    if (AArch64_get_detail_op(MI, 0)->type == AARCH64_OP_SME) {
      AArch64_set_detail_op_sme(
        MI, OpNum, AARCH64_SME_MATRIX_SLICE_OFF_RANGE,
        AARCH64LAYOUT_INVALID, (uint32_t)FirstImm,
        (uint32_t)Offset);
    } else if (AArch64_get_detail_op(MI, 0)->type ==
         AARCH64_OP_PRED) {
      CS_ASSERT_RET(0 &&
              "Unkown SME predicate imm range type");
    } else {
      CS_ASSERT_RET(0 && "Unkown SME operand type");
    }
    return;
  }

  CS_ASSERT_RET(!(map_get_op_type(MI, OpNum) & CS_OP_MEM));
  CS_ASSERT_RET(map_get_op_type(MI, OpNum) == CS_OP_IMM);

  AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_IMM_RANGE;
  AArch64_get_detail_op(MI, 0)->imm_range.first = FirstImm;
  AArch64_get_detail_op(MI, 0)->imm_range.offset = Offset;
  AArch64_get_detail_op(MI, 0)->access = map_get_op_access(MI, OpNum);
  AArch64_inc_op_count(MI);
}

/// Adds a memory AARCH64 operand at position OpNum. op_count is *not* increased by
/// one. This is done by set_mem_access().
void AArch64_set_detail_op_mem(MCInst *MI, unsigned OpNum, uint64_t Val)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  AArch64_set_mem_access(MI, true);

  cs_op_type secondary_type = map_get_op_type(MI, OpNum) & ~CS_OP_MEM;
  switch (secondary_type) {
  default:
    CS_ASSERT_RET(0 && "Secondary type not supported yet.");
  case CS_OP_REG: {
    bool is_index_reg = AArch64_get_detail_op(MI, 0)->mem.base !=
            AARCH64_REG_INVALID;
    if (is_index_reg)
      AArch64_get_detail_op(MI, 0)->mem.index = Val;
    else {
      AArch64_get_detail_op(MI, 0)->mem.base = Val;
    }

    if (MCInst_opIsTying(MI, OpNum)) {
      // Especially base registers can be writeback registers.
      // For this they tie an MC operand which has write
      // access. But this one is never processed in the printer
      // (because it is never emitted). Therefor it is never
      // added to the modified list.
      // Here we check for this case and add the memory register
      // to the modified list.
      map_add_implicit_write(MI, MCInst_getOpVal(MI, OpNum));
    }
    break;
  }
  case CS_OP_IMM: {
    AArch64_get_detail_op(MI, 0)->mem.disp = Val;
    break;
  }
  }

  AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_MEM;
  AArch64_get_detail_op(MI, 0)->access = map_get_op_access(MI, OpNum);
  AArch64_set_mem_access(MI, false);
}

/// Adds the shift and sign extend info to the previous operand.
/// op_count is *not* incremented by one.
void AArch64_set_detail_shift_ext(MCInst *MI, unsigned OpNum, bool SignExtend,
          bool DoShift, unsigned ExtWidth,
          char SrcRegKind)
{
  bool IsLSL = !SignExtend && SrcRegKind == 'x';
  if (IsLSL)
    AArch64_get_detail_op(MI, -1)->shift.type = AARCH64_SFT_LSL;
  else {
    aarch64_extender ext = SignExtend ? AARCH64_EXT_SXTB :
                AARCH64_EXT_UXTB;
    switch (SrcRegKind) {
    default:
      CS_ASSERT_RET(0 && "Extender not handled\n");
    case 'b':
      ext += 0;
      break;
    case 'h':
      ext += 1;
      break;
    case 'w':
      ext += 2;
      break;
    case 'x':
      ext += 3;
      break;
    }
    AArch64_get_detail_op(MI, -1)->ext = ext;
  }
  if (DoShift || IsLSL) {
    unsigned ShiftAmount = DoShift ? Log2_32(ExtWidth / 8) : 0;
    AArch64_get_detail_op(MI, -1)->shift.type = AARCH64_SFT_LSL;
    AArch64_get_detail_op(MI, -1)->shift.value = ShiftAmount;
  }
}

/// Transforms the immediate of the operand to a float and stores it.
/// Increments the op_counter by one.
void AArch64_set_detail_op_float(MCInst *MI, unsigned OpNum, float Val)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_FP;
  AArch64_get_detail_op(MI, 0)->fp = Val;
  AArch64_get_detail_op(MI, 0)->access = map_get_op_access(MI, OpNum);
  AArch64_inc_op_count(MI);
}

/// Adds a the system operand and increases the op_count by
/// one.
void AArch64_set_detail_op_sys(MCInst *MI, unsigned OpNum, aarch64_sysop sys_op,
             aarch64_op_type type)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  AArch64_get_detail_op(MI, 0)->type = type;
  AArch64_get_detail_op(MI, 0)->sysop = sys_op;
  if (sys_op.sub_type == AARCH64_OP_EXACTFPIMM) {
    AArch64_get_detail_op(MI, 0)->fp =
      aarch64_exact_fp_to_fp(sys_op.imm.exactfpimm);
  }
  AArch64_inc_op_count(MI);
}

void AArch64_set_detail_op_pred(MCInst *MI, unsigned OpNum)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  if (AArch64_get_detail_op(MI, 0)->type == AARCH64_OP_INVALID) {
    setup_pred_operand(MI);
  }
  aarch64_op_pred *p = &AArch64_get_detail_op(MI, 0)->pred;
  if (p->reg == AARCH64_REG_INVALID) {
    p->reg = MCInst_getOpVal(MI, OpNum);
    AArch64_get_detail_op(MI, 0)->access =
      map_get_op_access(MI, OpNum);
    AArch64_get_detail(MI)->is_doing_sme = true;
    return;
  } else if (p->vec_select == AARCH64_REG_INVALID) {
    p->vec_select = MCInst_getOpVal(MI, OpNum);
    return;
  } else if (p->imm_index == -1) {
    p->imm_index = MCInst_getOpVal(MI, OpNum);
    return;
  }
  CS_ASSERT_RET(0 && "Should not be reached.");
}

/// Adds a SME matrix component to a SME operand.
void AArch64_set_detail_op_sme(MCInst *MI, unsigned OpNum,
             aarch64_sme_op_part part,
             AArch64Layout_VectorLayout vas, ...)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  AArch64_get_detail_op(MI, 0)->type = AARCH64_OP_SME;
  switch (part) {
  default:
    printf("Unhandled SME operand part %d\n", part);
    CS_ASSERT_RET(0);
  case AARCH64_SME_MATRIX_TILE_LIST: {
    setup_sme_operand(MI);
    va_list args;
    va_start(args, vas);
    // NOLINTBEGIN(clang-analyzer-valist.Uninitialized)
    int Tile = va_arg(args, int);
    // NOLINTEND(clang-analyzer-valist.Uninitialized)
    va_end(args);
    AArch64_get_detail_op(MI, 0)->sme.type = AARCH64_SME_OP_TILE;
    AArch64_get_detail_op(MI, 0)->sme.tile = Tile;
    AArch64_get_detail_op(MI, 0)->vas = vas;
    AArch64_get_detail_op(MI, 0)->access =
      map_get_op_access(MI, OpNum);
    AArch64_get_detail(MI)->is_doing_sme = true;
    break;
  }
  case AARCH64_SME_MATRIX_TILE:
    CS_ASSERT_RET(map_get_op_type(MI, OpNum) == CS_OP_REG);

    setup_sme_operand(MI);
    AArch64_get_detail_op(MI, 0)->sme.type = AARCH64_SME_OP_TILE;
    AArch64_get_detail_op(MI, 0)->sme.tile =
      MCInst_getOpVal(MI, OpNum);
    AArch64_get_detail_op(MI, 0)->vas = vas;
    AArch64_get_detail_op(MI, 0)->access =
      map_get_op_access(MI, OpNum);
    AArch64_get_detail(MI)->is_doing_sme = true;
    break;
  case AARCH64_SME_MATRIX_SLICE_REG:
    CS_ASSERT_RET((map_get_op_type(MI, OpNum) &
             ~(CS_OP_MEM | CS_OP_BOUND)) == CS_OP_REG);
    CS_ASSERT_RET(AArch64_get_detail_op(MI, 0)->type ==
            AARCH64_OP_SME);

    // SME operand already present. Add the slice to it.
    AArch64_get_detail_op(MI, 0)->sme.type =
      AARCH64_SME_OP_TILE_VEC;
    AArch64_get_detail_op(MI, 0)->sme.slice_reg =
      MCInst_getOpVal(MI, OpNum);
    break;
  case AARCH64_SME_MATRIX_SLICE_OFF: {
    CS_ASSERT_RET((map_get_op_type(MI, OpNum) &
             ~(CS_OP_MEM | CS_OP_BOUND)) == CS_OP_IMM);
    // Because we took care of the slice register before, the op at -1 must be a SME operand.
    CS_ASSERT_RET(AArch64_get_detail_op(MI, 0)->type ==
            AARCH64_OP_SME);
    CS_ASSERT_RET(
      AArch64_get_detail_op(MI, 0)->sme.slice_offset.imm ==
      AARCH64_SLICE_IMM_INVALID);
    va_list args;
    va_start(args, vas);
    // NOLINTBEGIN(clang-analyzer-valist.Uninitialized)
    uint16_t offset = va_arg(args, uint32_t);
    // NOLINTEND(clang-analyzer-valist.Uninitialized)
    va_end(args);
    AArch64_get_detail_op(MI, 0)->sme.slice_offset.imm = offset;
    break;
  }
  case AARCH64_SME_MATRIX_SLICE_OFF_RANGE: {
    va_list args;
    va_start(args, vas);
    // NOLINTBEGIN(clang-analyzer-valist.Uninitialized)
    uint8_t First = va_arg(args, uint32_t);
    uint8_t Offset = va_arg(args, uint32_t);
    // NOLINTEND(clang-analyzer-valist.Uninitialized)
    va_end(args);
    AArch64_get_detail_op(MI, 0)->sme.slice_offset.imm_range.first =
      First;
    AArch64_get_detail_op(MI, 0)->sme.slice_offset.imm_range.offset =
      Offset;
    AArch64_get_detail_op(MI, 0)->sme.has_range_offset = true;
    break;
  }
  }
}

static void insert_op(MCInst *MI, unsigned index, cs_aarch64_op op)
{
  if (!detail_is_set(MI)) {
    return;
  }

  AArch64_check_safe_inc(MI);
  cs_aarch64_op *ops = AArch64_get_detail(MI)->operands;
  int i = AArch64_get_detail(MI)->op_count;
  if (index == -1) {
    ops[i] = op;
    AArch64_inc_op_count(MI);
    return;
  }
  for (; i > 0 && i > index; --i) {
    ops[i] = ops[i - 1];
  }
  ops[index] = op;
  AArch64_inc_op_count(MI);
}

/// Inserts a float to the detail operands at @index.
/// If @index == -1, it pushes the operand to the end of the ops array.
/// Already present operands are moved.
void AArch64_insert_detail_op_float_at(MCInst *MI, unsigned index, double val,
               cs_ac_type access)
{
  if (!detail_is_set(MI))
    return;

  AArch64_check_safe_inc(MI);

  cs_aarch64_op op;
  AArch64_setup_op(&op);
  op.type = AARCH64_OP_FP;
  op.fp = val;
  op.access = access;

  insert_op(MI, index, op);
}

/// Inserts a register to the detail operands at @index.
/// If @index == -1, it pushes the operand to the end of the ops array.
/// Already present operands are moved.
void AArch64_insert_detail_op_reg_at(MCInst *MI, unsigned index,
             aarch64_reg Reg, cs_ac_type access)
{
  if (!detail_is_set(MI))
    return;

  AArch64_check_safe_inc(MI);

  cs_aarch64_op op;
  AArch64_setup_op(&op);
  op.type = AARCH64_OP_REG;
  op.reg = Reg;
  op.access = access;

  insert_op(MI, index, op);
}

/// Inserts a immediate to the detail operands at @index.
/// If @index == -1, it pushes the operand to the end of the ops array.
/// Already present operands are moved.
void AArch64_insert_detail_op_imm_at(MCInst *MI, unsigned index, int64_t Imm)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  cs_aarch64_op op;
  AArch64_setup_op(&op);
  op.type = AARCH64_OP_IMM;
  op.imm = Imm;
  op.access = CS_AC_READ;

  insert_op(MI, index, op);
}

void AArch64_insert_detail_op_sys(MCInst *MI, unsigned index,
          aarch64_sysop sys_op, aarch64_op_type type)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  cs_aarch64_op op;
  AArch64_setup_op(&op);
  op.type = type;
  op.sysop = sys_op;
  if (op.sysop.sub_type == AARCH64_OP_EXACTFPIMM) {
    op.fp = aarch64_exact_fp_to_fp(op.sysop.imm.exactfpimm);
  }
  insert_op(MI, index, op);
}

void AArch64_insert_detail_op_sme(MCInst *MI, unsigned index,
          aarch64_op_sme sme_op)
{
  if (!detail_is_set(MI))
    return;
  AArch64_check_safe_inc(MI);

  cs_aarch64_op op;
  AArch64_setup_op(&op);
  op.type = AARCH64_OP_SME;
  op.sme = sme_op;
  insert_op(MI, index, op);
}

#endif

Coverage Report

Created: 2025-08-29 06:29