#pragma once

#include <sys/uio.h>
#include <capstone/capstone.h>

struct archinfo {
    unsigned long progmctr;
    unsigned long stackptr;
    unsigned long bp_insn;
    unsigned long bp_mask;
    unsigned long bp_adjust;
    int cs_arch;
    int cs_mode;
    unsigned cs_call;
    unsigned wordsize;
};

extern void architecture_info(struct archinfo *ai, const struct iovec *regs);

/* Architecture Definitions */
#if defined(__x86_64__) || defined(i386) || defined(__i386__)

typedef union {
    struct user_regs_64 {
        unsigned long long int r15, r14, r13, r12, rbp, rbx, r11, r10, r9, r8,
                      rax, rcx, rdx, rsi, rdi, orig_rax, rip, cs, eflags, rsp,
                      ss, fs_base, gs_base, ds, es, fs, gs;
    } x86_64;

    struct user_regs_32 {
        unsigned int ebx, ecx, edx, esi, edi, ebp, eax, xds, xes, xfs, xgs,
                     orig_eax, eip, xcs, eflags, esp, xss;
    } x86_32;
} user_regs_t;

#define ARCH_X86

#define PROGMCTR_64             x86_64.rip
#define STACKPTR_64             x86_64.rsp
#define BREAKPOINT_INSN_64      0xccul
#define BREAKPOINT_MASK_64      0xfful
#define BREAKPOINT_ADJS_64      0x1
#define CAPSTONE_ARCH_64        CS_ARCH_X86
#define CAPSTONE_MODE_64        CS_MODE_64
#define CAPSTONE_CALL_64        X86_INS_CALL
#define WORDSIZE_64             8

#define PROGMCTR_32             x86_32.eip
#define STACKPTR_32             x86_32.esp
#define BREAKPOINT_INSN_32      0xccul
#define BREAKPOINT_MASK_32      0xfful
#define BREAKPOINT_ADJS_32      0x1
#define CAPSTONE_ARCH_32        CS_ARCH_X86
#define CAPSTONE_MODE_32        CS_MODE_32
#define CAPSTONE_CALL_32        X86_INS_CALL
#define WORDSIZE_32             4

#elif defined(__aarch64__) || defined(_M_ARM64)

typedef union {
    struct user_regs_64 {
        unsigned long long regs[31];
        unsigned long long sp, pc, pstate;
    } arm64;

    struct user_regs_32 {
        unsigned int x;
    } arm32;
} user_regs_t;

#define ARCH_AARCH64

#define PROGMCTR_64             arm64.pc
#define STACKPTR_64             arm64.sp
#define BREAKPOINT_INSN_64      0xd4200000ul
#define BREAKPOINT_MASK_64      0xfffffffful
#define BREAKPOINT_ADJS_64      0x0
#define CAPSTONE_ARCH_64        CS_ARCH_ARM64
#define CAPSTONE_MODE_64        CS_MODE_ARM
#define CAPSTONE_CALL_64        ARM64_INS_BL
#define WORDSIZE_64             8

#define PROGMCTR_32             arm32.x
#define STACKPTR_32             arm32.x
#define BREAKPOINT_INSN_32      0
#define BREAKPOINT_MASK_32      0
#define BREAKPOINT_ADJS_32      0
#define CAPSTONE_ARCH_32        0
#define CAPSTONE_MODE_32        0
#define CAPSTONE_CALL_32        0
#define WORDSIZE_32             4

#else
#error Detected architecture is not supported!
#endif