jak-project/test/test_zydis.cpp
2020-10-31 14:31:13 -04:00

27 lines
953 B
C++

#include "gtest/gtest.h"
#include "common/common_types.h"
#include "third-party/zydis/include/Zydis/Zydis.h"
TEST(Zydis, Basic) {
// int3, return
u8 code[] = {0xcc, 0xc3};
ZydisDecoder decoder;
ZydisDecoderInit(&decoder, ZYDIS_MACHINE_MODE_LONG_64, ZYDIS_ADDRESS_WIDTH_64);
ZydisFormatter formatter;
ZydisFormatterInit(&formatter, ZYDIS_FORMATTER_STYLE_INTEL);
ZydisDecodedInstruction instr;
// should get first instruction
EXPECT_TRUE(ZYAN_SUCCESS(ZydisDecoderDecodeBuffer(&decoder, code, 2, &instr)));
char result[256];
ZydisFormatterFormatInstruction(&formatter, &instr, result, 256, 0);
EXPECT_EQ(std::string("int3"), result);
EXPECT_EQ(1, instr.length);
// should get second instruction
EXPECT_TRUE(ZYAN_SUCCESS(ZydisDecoderDecodeBuffer(&decoder, code + 1, 1, &instr)));
ZydisFormatterFormatInstruction(&formatter, &instr, result, 256, 0);
EXPECT_EQ(std::string("ret"), result);
EXPECT_EQ(1, instr.length);
}