mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-02-02 20:43:32 +00:00
23d333c090
This change includes many bug fixes, for the NT polyfills, strings, memory, boot, and math libraries which were discovered by adding more tools for recreational programming, such as PC emulation. Lemon has also been vendored because it works so well at parsing languages.
271 lines
11 KiB
C
271 lines
11 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/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("or $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);
|
|
}
|