t7x/deps/udis86/tests/libcheck.c
2023-12-06 17:43:39 -05:00

233 lines
7.1 KiB
C

/* udis86 - tests/libcheck.c
*
* Copyright (c) 2013 Vivek Thampi
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without modification,
* are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
* ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include <stdio.h>
#include <udis86.h>
#include <string.h>
unsigned int testcase_check_count;
unsigned int testcase_check_fails;
#define TEST_DECL(name) \
const char * __testcase_name = name \
#define TEST_CASE() \
do { \
volatile int __c = ++ testcase_check_count; \
if (0) __c += 1; \
do
#define TEST_CASE_SET_FAIL() \
do { \
testcase_check_fails++; \
printf("Testcase %s: failure at line %d\n", __testcase_name, __LINE__); \
} while (0)
#define TEST_CASE_REPORT_ACTUAL(v) \
printf("Testcase %s: actual = %d\n", __testcase_name, (v))
#define TEST_CASE_REPORT_EXPECTED(v) \
printf("Testcase %s: expected = %d\n", __testcase_name, (v))
#define TEST_CASE_END() \
while (0); \
} while (0)
#define TEST_CHECK(cond) \
TEST_CASE() { \
int eval = (cond); \
if (!eval) { \
TEST_CASE_SET_FAIL(); \
} \
} TEST_CASE_END()
#define TEST_CHECK_INT(expr, val) \
TEST_CASE() { \
int eval = (expr); \
int val2 = (val); \
if (eval != val2) { \
TEST_CASE_SET_FAIL(); \
TEST_CASE_REPORT_EXPECTED(val2); \
TEST_CASE_REPORT_ACTUAL(eval); \
} \
} TEST_CASE_END()
#define TEST_CHECK_OP_REG(o, n, r) \
TEST_CHECK(ud_insn_opr(o, n)->type == UD_OP_REG && \
ud_insn_opr(o, n)->base == (r))
static int
input_callback(ud_t *u)
{
int *n = (int *) ud_get_user_opaque_data(u);
if (*n == 0) {
return UD_EOI;
}
--*n;
return 0x90;
}
static void
check_input(ud_t *ud_obj)
{
TEST_DECL("check_input");
const uint8_t code[] = { 0x89, 0xc8 }; /* mov eax, ecx */
int i;
/* truncate buffer */
ud_set_mode(ud_obj, 32);
for (i = 0; i < 5; ++i) {
ud_set_input_buffer(ud_obj, code, (sizeof code) - 1);
TEST_CHECK(ud_disassemble(ud_obj) == 1);
TEST_CHECK(ud_insn_len(ud_obj) == 1);
TEST_CHECK(ud_obj->mnemonic == UD_Iinvalid);
}
/* input skip on buffer */
{
const uint8_t code[] = { 0x89, 0xc8, /* mov eax, ecx*/
0x90 }; /* nop */
ud_set_input_buffer(ud_obj, code, (sizeof code));
ud_input_skip(ud_obj, 2);
TEST_CHECK_INT(ud_disassemble(ud_obj), 1);
TEST_CHECK_INT(ud_obj->mnemonic, UD_Inop);
ud_set_input_buffer(ud_obj, code, (sizeof code));
ud_input_skip(ud_obj, 0);
TEST_CHECK_INT(ud_disassemble(ud_obj), 2);
TEST_CHECK_INT(ud_obj->mnemonic, UD_Imov);
TEST_CHECK(ud_insn_ptr(ud_obj)[0] == 0x89);
TEST_CHECK(ud_insn_ptr(ud_obj)[1] == 0xc8);
/* bad skip */
ud_set_input_buffer(ud_obj, code, (sizeof code));
ud_input_skip(ud_obj, 3);
TEST_CHECK_INT(ud_disassemble(ud_obj), 0);
ud_input_skip(ud_obj, 1);
TEST_CHECK_INT(ud_disassemble(ud_obj), 0);
ud_set_input_buffer(ud_obj, code, (sizeof code));
ud_input_skip(ud_obj, 0);
TEST_CHECK_INT(ud_disassemble(ud_obj), 2);
ud_input_skip(ud_obj, 1000);
TEST_CHECK_INT(ud_disassemble(ud_obj), 0);
}
/* input hook test */
{
int n;
ud_set_user_opaque_data(ud_obj, (void *) &n);
ud_set_input_hook(ud_obj, &input_callback);
n = 0;
TEST_CHECK(ud_disassemble(ud_obj) == 0);
n = 1;
ud_set_input_hook(ud_obj, &input_callback);
TEST_CHECK_INT(ud_disassemble(ud_obj), 1);
TEST_CHECK(ud_insn_ptr(ud_obj)[0] == 0x90);
TEST_CHECK_INT(ud_obj->mnemonic, UD_Inop);
n = 2;
ud_set_input_hook(ud_obj, &input_callback);
ud_input_skip(ud_obj, 1);
TEST_CHECK(ud_disassemble(ud_obj) == 1);
TEST_CHECK(ud_obj->mnemonic == UD_Inop);
TEST_CHECK(ud_disassemble(ud_obj) == 0);
TEST_CHECK(ud_insn_len(ud_obj) == 0);
TEST_CHECK(ud_obj->mnemonic == UD_Iinvalid);
n = 1;
ud_input_skip(ud_obj, 2);
TEST_CHECK_INT(ud_disassemble(ud_obj), 0);
TEST_CHECK(ud_input_end(ud_obj));
}
/* a known buffer overrun test case (used to be bufoverrun.c) */
{
const uint8_t code[] = { 0xf0, 0x66, 0x36, 0x67, 0x65, 0x66,
0xf3, 0x67, 0xda };
ud_set_mode(ud_obj, 16);
ud_set_input_buffer(ud_obj, code, sizeof code);
TEST_CHECK(ud_disassemble(ud_obj) > 0);
}
}
static void
check_mode(ud_t *ud_obj)
{
TEST_DECL("check_mode");
const uint8_t code[] = { 0x89, 0xc8 }; /* mov eax, ecx */
ud_set_input_buffer(ud_obj, code, sizeof code);
ud_set_mode(ud_obj, 32);
TEST_CHECK(ud_disassemble(ud_obj) == 2);
TEST_CHECK_OP_REG(ud_obj, 0, UD_R_EAX);
TEST_CHECK_OP_REG(ud_obj, 1, UD_R_ECX);
}
static void
check_disasm(ud_t *ud_obj)
{
TEST_DECL("check_mode");
const uint8_t code[] = { 0x89, 0xc8, /* mov eax, ecx */
0x90 }; /* nop */
ud_set_input_buffer(ud_obj, code, sizeof code);
ud_set_mode(ud_obj, 32);
ud_set_pc(ud_obj, 0x100);
TEST_CHECK(ud_disassemble(ud_obj) == 2);
TEST_CHECK(ud_insn_off(ud_obj) == 0x100);
TEST_CHECK(ud_insn_ptr(ud_obj)[0] == 0x89);
TEST_CHECK(ud_insn_ptr(ud_obj)[1] == 0xc8);
TEST_CHECK(ud_insn_mnemonic(ud_obj) == UD_Imov);
TEST_CHECK(strcmp(ud_lookup_mnemonic(UD_Imov), "mov") == 0);
TEST_CHECK(ud_disassemble(ud_obj) == 1);
TEST_CHECK(ud_insn_off(ud_obj) == 0x102);
TEST_CHECK(ud_insn_ptr(ud_obj)[0] == 0x90);
TEST_CHECK(ud_insn_mnemonic(ud_obj) == UD_Inop);
TEST_CHECK(strcmp(ud_lookup_mnemonic(UD_Inop), "nop") == 0);
}
int
main(void)
{
ud_t ud_obj;
ud_init(&ud_obj);
ud_set_syntax(&ud_obj, UD_SYN_INTEL);
check_input(&ud_obj);
check_mode(&ud_obj);
check_disasm(&ud_obj);
if (testcase_check_fails > 0) {
printf("libcheck result: %d checks, %d failures\n",
testcase_check_count, testcase_check_fails);
return 1;
}
return 0;
}
/* vim: set ts=2 sw=2 expandtab: */