84 lines
2.5 KiB
C
84 lines
2.5 KiB
C
#include "unicorn/unicorn.h"
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
|
|
|
|
#define OK(x) {uc_err __err; if ((__err = x)) { fprintf(stderr, "%s", uc_strerror(__err)); assert(false); } }
|
|
static void test_vmovdqu(void)
|
|
{
|
|
uc_engine *uc;
|
|
|
|
int r_esi = 0x1234;
|
|
int r_edi = 0x7890;
|
|
|
|
uint64_t r_xmm0[2] = {0x08090a0b0c0d0e0f, 0x0001020304050607};
|
|
|
|
/* 128 bit at address esi (0x1234) this should not be read into xmm0 */
|
|
char mem_esi[] = { '\xE7', '\x1D', '\xA7', '\xE8', '\x88', '\xE4', '\x94', '\x40', '\x54', '\x74', '\x24', '\x97', '\x1F', '\x2E', '\xB6', '\x40' };
|
|
|
|
/* 128 bit at address edi (0x7890) this SHOULD be read into xmm0 */
|
|
char mem_edi[] = { '\xAD', '\xFA', '\x5C', '\x6D', '\x45', '\x4A', '\x93', '\x40', '\xD2', '\x00', '\xDE', '\x02', '\x89', '\xE8', '\x94', '\x40' };
|
|
|
|
/* vmovdqu xmm0, [edi] */
|
|
char code[] = { '\xC5', '\xFA', '\x6F', '\x07' };
|
|
|
|
/* initialize memory and run emulation */
|
|
OK(uc_open(UC_ARCH_X86, UC_MODE_32, &uc));
|
|
OK(uc_mem_map(uc, 0, 2 * 1024 * 1024, UC_PROT_ALL));
|
|
|
|
OK(uc_mem_write(uc, 0, code, sizeof(code) / sizeof(code[0])));
|
|
|
|
// initialize machine registers;
|
|
OK(uc_reg_write(uc, UC_X86_REG_XMM0, &r_xmm0));
|
|
|
|
OK(uc_reg_write(uc, UC_X86_REG_ESI, &r_esi));
|
|
OK(uc_reg_write(uc, UC_X86_REG_EDI, &r_edi));
|
|
OK(uc_mem_write(uc, r_esi, mem_esi, sizeof(mem_esi) / sizeof(mem_esi[0])));
|
|
OK(uc_mem_write(uc, r_edi, mem_edi, sizeof(mem_edi) / sizeof(mem_edi[0])));
|
|
|
|
OK(uc_emu_start(uc, 0, sizeof(code) / sizeof(code[0]), 0, 0));
|
|
|
|
/* Read xmm0 after emulation */
|
|
OK(uc_reg_read(uc, UC_X86_REG_XMM0, &r_xmm0));
|
|
|
|
|
|
assert(0x4094e88902de00d2 == r_xmm0[0] && 0x40934a456d5cfaad == r_xmm0[1]);
|
|
|
|
OK(uc_close(uc));
|
|
}
|
|
|
|
/* https://github.com/unicorn-engine/unicorn/issues/1656 */
|
|
static void test_vex_l(void)
|
|
{
|
|
uc_engine *uc;
|
|
uc_err err;
|
|
|
|
/* vmovdqu ymm1, [rcx] */
|
|
char code[] = { '\xC5', '\xFE', '\x6F', '\x09' };
|
|
|
|
/* initialize memory and run emulation */
|
|
OK(uc_open(UC_ARCH_X86, UC_MODE_64, &uc));
|
|
OK(uc_mem_map(uc, 0, 2 * 1024 * 1024, UC_PROT_ALL));
|
|
|
|
OK(uc_mem_write(uc, 0, code, sizeof(code) / sizeof(code[0])));
|
|
|
|
err = uc_emu_start(uc, 0, sizeof(code) / sizeof(code[0]), 0, 0);
|
|
if(err != UC_ERR_INSN_INVALID) {
|
|
fprintf(stderr, "%s", uc_strerror(err));
|
|
assert(false);
|
|
}
|
|
|
|
OK(uc_close(uc));
|
|
}
|
|
|
|
|
|
/* TODO: Add more vex prefixed instructions
|
|
Suggestions: vxorpd, vxorps, vandpd, ... */
|
|
int main(int argc, char **argv, char **envp)
|
|
{
|
|
test_vmovdqu();
|
|
test_vex_l();
|
|
return 0;
|
|
}
|
|
|