292 lines
12 KiB
C
292 lines
12 KiB
C
/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
|
|
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8 :vi│
|
|
╞══════════════════════════════════════════════════════════════════════════════╡
|
|
│ Copyright 2020 Justine Alexandra Roberts Tunney │
|
|
│ │
|
|
│ This program is free software; you can redistribute it and/or modify │
|
|
│ it under the terms of the GNU General Public License as published by │
|
|
│ the Free Software Foundation; version 2 of the License. │
|
|
│ │
|
|
│ This program is distributed in the hope that it will be useful, but │
|
|
│ WITHOUT ANY WARRANTY; without even the implied warranty of │
|
|
│ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU │
|
|
│ General Public License for more details. │
|
|
│ │
|
|
│ You should have received a copy of the GNU General Public License │
|
|
│ along with this program; if not, write to the Free Software │
|
|
│ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA │
|
|
│ 02110-1301 USA │
|
|
╚─────────────────────────────────────────────────────────────────────────────*/
|
|
#include "libc/rand/lcg.h"
|
|
#include "libc/rand/rand.h"
|
|
#include "libc/stdio/stdio.h"
|
|
#include "libc/testlib/ezbench.h"
|
|
#include "libc/testlib/testlib.h"
|
|
#include "tool/build/lib/dis.h"
|
|
#include "tool/build/lib/modrm.h"
|
|
|
|
char b1[64];
|
|
char b2[64];
|
|
struct Dis d[1];
|
|
|
|
TEST(DisInst, testInt3) {
|
|
uint8_t op[] = {0xcc};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("int3 ", b1);
|
|
}
|
|
|
|
TEST(DisInst, testImmMem_needsSuffix) {
|
|
uint8_t op[] = {0x80, 0x3c, 0x07, 0x00};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("cmpb $0,(%rdi,%rax)", b1);
|
|
}
|
|
|
|
TEST(DisInst, testImmReg_doesntNeedSuffix) {
|
|
uint8_t op[] = {0xb8, 0x08, 0x70, 0x40, 0x00};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("mov $0x407008,%eax", b1);
|
|
}
|
|
|
|
TEST(DisInst, testPuttingOnTheRiz) {
|
|
static uint8_t ops[][15] = {
|
|
{0x8d, 0b00110100, 0b00100110},
|
|
{0x67, 0x8d, 0b00110100, 0b11100110},
|
|
{0x67, 0x8d, 0b10110100, 0b11100101, 0, 0, 0, 0},
|
|
{0x8d, 0b00110100, 0b11100101, 0x37, 0x13, 0x03, 0},
|
|
{0x8d, 0b10110100, 0b11100101, 0, 0, 0, 0},
|
|
};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, ops[0], sizeof(ops[0])));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("lea (%rsi),%esi", b1);
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, ops[1], sizeof(ops[1])));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("lea (%esi,%eiz,8),%esi", b1);
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, ops[2], sizeof(ops[2])));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("lea 0(%ebp,%eiz,8),%esi", b1);
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, ops[3], sizeof(ops[3])));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("lea 0x31337,%esi", b1);
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, ops[4], sizeof(ops[4])));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("lea 0(%rbp,%riz,8),%esi", b1);
|
|
}
|
|
|
|
TEST(DisInst, testSibIndexOnly) {
|
|
uint8_t op[] = {76, 141, 4, 141, 0, 0, 0, 0};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("lea 0(,%rcx,4),%r8", b1);
|
|
}
|
|
|
|
TEST(DisInst, testRealMode) {
|
|
uint8_t op[] = {0x89, 0xe5};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("mov %sp,%bp", b1);
|
|
}
|
|
|
|
TEST(DisInst, testNop) {
|
|
uint8_t op[] = {0x66, 0x2e, 0x0f, 0x1f, 0x84, 0x00, 0x00, 0x00, 0x00, 0x00};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("nopw %cs:0(%rax,%rax)", b1);
|
|
}
|
|
|
|
TEST(DisInst, testPush) {
|
|
uint8_t op[] = {0x41, 0x5c};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
EXPECT_EQ(4, ModrmSrm(d->xedd->op.rde));
|
|
EXPECT_EQ(1, Rexb(d->xedd->op.rde));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("pop %r12", b1);
|
|
}
|
|
|
|
TEST(DisInst, testMovb) {
|
|
uint8_t op[] = {0x8a, 0x1e, 0x0c, 0x32};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("mov (%rsi),%bl", b1);
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("mov 0x320c,%bl", b1);
|
|
}
|
|
|
|
TEST(DisInst, testLes) {
|
|
uint8_t op[] = {0xc4, 0x3e, 0x16, 0x32};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("les 0x3216,%di", b1);
|
|
}
|
|
|
|
TEST(DisInst, testStosbLong) {
|
|
uint8_t op[] = {0xAA};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("stosb %al,(%rdi)", b1);
|
|
}
|
|
|
|
TEST(DisInst, testStosbReal) {
|
|
uint8_t op[] = {0xAA};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("stosb %al,(%di)", b1);
|
|
}
|
|
|
|
TEST(DisInst, testStosbLegacy) {
|
|
uint8_t op[] = {0xAA};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LEGACY_32);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("stosb %al,(%edi)", b1);
|
|
}
|
|
|
|
TEST(DisInst, testStosbLongAsz) {
|
|
uint8_t op[] = {0x67, 0xAA};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("stosb %al,(%edi)", b1);
|
|
}
|
|
|
|
TEST(DisInst, testAddLong) {
|
|
uint8_t op[] = {0x01, 0xff};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("add %edi,%edi", b1);
|
|
}
|
|
|
|
TEST(DisInst, testAddLegacy) {
|
|
uint8_t op[] = {0x01, 0xff};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LEGACY_32);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("add %edi,%edi", b1);
|
|
}
|
|
|
|
TEST(DisInst, testAddReal) {
|
|
uint8_t op[] = {0x01, 0xff};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("add %di,%di", b1);
|
|
}
|
|
|
|
TEST(DisInst, testAddLongOsz) {
|
|
uint8_t op[] = {0x66, 0x01, 0xff};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("add %di,%di", b1);
|
|
}
|
|
|
|
TEST(DisInst, testAddLegacyOsz) {
|
|
uint8_t op[] = {0x66, 0x01, 0xff};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LEGACY_32);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("add %di,%di", b1);
|
|
}
|
|
|
|
TEST(DisInst, testAddRealOsz) {
|
|
uint8_t op[] = {0x66, 0x01, 0xff};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("add %edi,%edi", b1);
|
|
}
|
|
|
|
TEST(DisInst, testFxam) {
|
|
uint8_t op[] = {0xd9, 0xe5};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
ASSERT_EQ(4, ModrmReg(d->xedd->op.rde));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("fxam ", b1);
|
|
}
|
|
|
|
TEST(DisInst, testOrImmCode16gcc) {
|
|
uint8_t op[] = {0x67, 0x81, 0x4c, 0x24, 0x0c, 0x00, 0x0c};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("orw $0xc00,12(%esp)", b1);
|
|
}
|
|
|
|
TEST(DisInst, testPause) {
|
|
uint8_t op[] = {0xf3, 0x90};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("pause ", b1);
|
|
}
|
|
|
|
TEST(DisInst, testJmpEw) {
|
|
uint8_t op[] = {0xff, 0xe0};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("jmp %ax", b1);
|
|
}
|
|
|
|
TEST(DisInst, testJmpEv16) {
|
|
uint8_t op[] = {0x66, 0xff, 0xe0};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("jmp %eax", b1);
|
|
}
|
|
|
|
TEST(DisInst, testJmpEv32) {
|
|
uint8_t op[] = {0xff, 0xe0};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LEGACY_32);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("jmp %eax", b1);
|
|
}
|
|
|
|
TEST(DisInst, testJmpEq) {
|
|
uint8_t op[] = {0x66, 0xff, 0xe0};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_LONG_64);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("jmp %rax", b1);
|
|
}
|
|
|
|
TEST(DisInst, testMovswSs) {
|
|
uint8_t op[] = {0x36, 0xA5};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("movs %ss:(%si),(%di)", b1);
|
|
}
|
|
|
|
TEST(DisInst, testRealModrm_sibOverlap_showsNoDisplacement) {
|
|
uint8_t op[] = {0x8b, 0b00100101};
|
|
xed_decoded_inst_zero_set_mode(d->xedd, XED_MACHINE_MODE_REAL);
|
|
ASSERT_EQ(0, xed_instruction_length_decode(d->xedd, op, sizeof(op)));
|
|
DisInst(d, b1, DisSpec(d->xedd, b2));
|
|
EXPECT_STREQ("mov (%di),%sp", b1);
|
|
}
|