jak-project/test/decompiler/test_AtomicOpBuilder.cpp
Tyler Wilding 60db0e5ef9
deps: update fmt to latest version (#3403)
This updates `fmt` to the latest version and moves to just being a copy
of their repo to make updating easier (no editing their cmake / figuring
out which files to minimally include).

The motivation for this is now that we switched to C++ 20, there were a
ton of deprecated function usages that is going away in future compiler
versions. This gets rid of all those warnings.
2024-03-05 22:11:52 -05:00

791 lines
34 KiB
C++

#include <regex>
#include "decompiler/Disasm/InstructionParser.h"
#include "decompiler/Function/Warnings.h"
#include "decompiler/IR2/AtomicOp.h"
#include "decompiler/analysis/atomic_op_builder.h"
#include "gtest/gtest.h"
#include "fmt/core.h"
#include "fmt/format.h"
using namespace decompiler;
std::regex labelRegex("^L\\d+:\\s*$");
// Auto indents / adds new-lines to a list of assembly lines
std::string assembly_from_list(std::vector<std::string> assemblyLines) {
std::string str = "";
for (std::string line : assemblyLines) {
if (std::regex_match(line, labelRegex)) {
str += fmt::format("{}\n", line);
} else {
str += fmt::format(" {}\n", line);
}
}
return str;
}
// jak 1 specific
void test_case(std::string assembly_lines,
std::vector<std::string> output_lines,
std::vector<std::vector<std::string>> write_regs,
std::vector<std::vector<std::string>> read_regs,
std::vector<std::vector<std::string>> clobbered_regs) {
InstructionParser parser;
// convert to Instructions:
// some MIPS instructions. Can be a sequence of instructions, possibly with labels.
ParsedProgram prg = parser.parse_program(assembly_lines);
// this verifies we can convert from a string to an instruction, and back to a string again.
// the instruction printer will add two leading spaces and a newline.
EXPECT_EQ(prg.print(), assembly_lines);
// next, set up a test environment for the conversion. The FunctionAtomicOps will hold
// the result of the conversion
FunctionAtomicOps container;
// treat the entire program as a single basic block, and convert!
DecompWarnings warnings;
convert_block_to_atomic_ops(0, prg.instructions.begin(), prg.instructions.end(), prg.labels,
&container, warnings, GameVersion::Jak1);
// count operations
EXPECT_EQ(container.ops.size(), output_lines.size());
// for now, we create an empty environment. The environment will be used in the future to
// rename register to variables, but for now, we just leave it empty and the printing will
// use register names
Env env;
// check the we get the right result:
for (size_t i = 0; i < container.ops.size(); i++) {
const auto& op = container.ops.at(i);
EXPECT_EQ(op->to_form(prg.labels, env).print(), output_lines.at(i));
// check that the registers read/written are identified for the operation
// check write registers
EXPECT_EQ(op->write_regs().size(), write_regs.at(i).size());
for (size_t j = 0; j < op->write_regs().size(); j++) {
const std::string expected_reg = op->write_regs().at(j).to_string();
// the ordering of the registers doesn't matter. It could happen to be in the same order
// as the opcode here, but it may not always be the case.
bool found = false;
for (const std::string& reg : write_regs.at(i)) {
// TODO - is there a potential bug here in the event that either list has duplicate
// registers?
if (reg == expected_reg) {
found = true;
break;
}
}
EXPECT_TRUE(found) << fmt::format("Unable to find expected WRITE register - {}",
expected_reg);
}
// check read registers
EXPECT_EQ(op->read_regs().size(), read_regs.at(i).size());
for (size_t j = 0; j < op->read_regs().size(); j++) {
const std::string expected_reg = op->read_regs().at(j).to_string();
// the ordering of the registers doesn't matter. It could happen to be in the same order
// as the opcode here, but it may not always be the case.
bool found = false;
for (const std::string& reg : read_regs.at(i)) {
// TODO - is there a potential bug here in the event that either list has duplicate
// registers?
if (reg == expected_reg) {
found = true;
break;
}
}
EXPECT_TRUE(found) << fmt::format("Unable to find expected READ register - {}", expected_reg);
}
// check clobbered registers
EXPECT_EQ(op->clobber_regs().size(), clobbered_regs.at(i).size());
for (size_t j = 0; j < op->clobber_regs().size(); j++) {
const std::string expected_reg = op->clobber_regs().at(j).to_string();
// the ordering of the registers doesn't matter. It could happen to be in the same order
// as the opcode here, but it may not always be the case.
bool found = false;
for (const std::string& reg : clobbered_regs.at(i)) {
// TODO - is there a potential bug here in the event that either list has duplicate
// registers?
if (reg == expected_reg) {
found = true;
break;
}
}
EXPECT_TRUE(found) << fmt::format("Unable to find expected CLOBBERED register - {}",
expected_reg);
}
}
}
TEST(DecompilerAtomicOpBuilder, RegUseDuplication) {
auto assembly =
assembly_from_list({"L100:", "sltiu a3, a0, 12", "beq a3, r0, L100", "or a3, s7, r0"});
InstructionParser parser;
ParsedProgram prg = parser.parse_program(assembly);
EXPECT_EQ(prg.print(), assembly);
FunctionAtomicOps container;
DecompWarnings warnings;
convert_block_to_atomic_ops(0, prg.instructions.begin(), prg.instructions.end(), prg.labels,
&container, warnings, GameVersion::Jak1);
ASSERT_EQ(1, container.ops.size());
auto& op = container.ops.at(0);
for (const auto& reg_group : {op->read_regs(), op->write_regs(), op->clobber_regs()}) {
std::unordered_set<Register, Register::hash> unique_regs;
for (auto& reg : reg_group) {
unique_regs.insert(reg);
}
EXPECT_EQ(unique_regs.size(), reg_group.size());
}
}
TEST(DecompilerAtomicOpBuilder, Example) {
test_case(assembly_from_list({"and v0, v1, a3", "and a1, a2, a2"}),
{"(set! v0 (logand v1 a3))", "(set! a1 (logand a2 a2))"}, {{"v0"}, {"a1"}},
{{"v1", "a3"}, {"a2", "a2"}}, {{}, {}});
}
TEST(DecompilerAtomicOpBuilder, ABSS) {
test_case(assembly_from_list({"abs.s f1, f2"}), {"(set! f1 (abs.s f2))"}, {{"f1"}}, {{"f2"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, ADDIU) {
test_case(assembly_from_list({"addiu a1, r0, 12"}), {"(set! a1 12)"}, {{"a1"}}, {{}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, ADDS) {
test_case(assembly_from_list({"add.s f1, f2, f3"}), {"(set! f1 (+.s f2 f3))"}, {{"f1"}},
{{"f2", "f3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, AND) {
test_case(assembly_from_list({"and a1, a2, a3"}), {"(set! a1 (logand a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, ANDI) {
test_case(assembly_from_list({"andi a1, a2, 1234"}), {"(set! a1 (logand a2 1234))"}, {{"a1"}},
{{"a2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BEQL_SLL) {
test_case(assembly_from_list({"L100:", "beql a0, a1, L100"}),
{"(bl! (= a0 a1) L100 (no-delay!))"}, {{}}, {{"a0", "a1"}}, {{}});
test_case(assembly_from_list({"L100:", "beql r0, r0, L100"}), {"(bl! #t L100 (no-delay!))"}, {{}},
{{}}, {{}});
test_case(assembly_from_list({"L100:", "beql a0, r0, L100"}),
{"(bl! (zero? a0) L100 (no-delay!))"}, {{}}, {{"a0"}}, {{}});
test_case(assembly_from_list({"L100:", "beql s7, a0, L100"}), {"(bl! (not a0) L100 (no-delay!))"},
{{}}, {{"a0"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BEQ_SLL) {
test_case(assembly_from_list({"L100:", "beq a0, a1, L100", "sll r0, r0, 0"}),
{"(b! (= a0 a1) L100 (nop!))"}, {{}}, {{"a0", "a1"}}, {{}});
test_case(assembly_from_list({"L100:", "beq r0, r0, L100", "sll r0, r0, 0"}),
{"(b! #t L100 (nop!))"}, {{}}, {{}}, {{}});
test_case(assembly_from_list({"L100:", "beq a0, r0, L100", "sll r0, r0, 0"}),
{"(b! (zero? a0) L100 (nop!))"}, {{}}, {{"a0"}}, {{}});
test_case(assembly_from_list({"L100:", "beq s7, a0, L100", "sll r0, r0, 0"}),
{"(b! (not a0) L100 (nop!))"}, {{}}, {{"a0"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BGEZL_SLL) {
test_case(assembly_from_list({"L100:", "bgezl a0, L100"}), {"(bl! (>=0.si a0) L100 (no-delay!))"},
{{}}, {{"a0"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BGTZL_SLL) {
test_case(assembly_from_list({"L100:", "bgtzl a0, L100"}), {"(bl! (>0.si a0) L100 (no-delay!))"},
{{}}, {{"a0"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BLTZL_SLL) {
test_case(assembly_from_list({"L100:", "bltzl a0, L100"}), {"(bl! (<0.si a0) L100 (no-delay!))"},
{{}}, {{"a0"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BNEL_SLL) {
test_case(assembly_from_list({"L100:", "bnel a1, a2, L100"}),
{"(bl! (!= a1 a2) L100 (no-delay!))"}, {{}}, {{"a1", "a2"}}, {{}});
test_case(assembly_from_list({"L100:", "bnel a1, r0, L100"}),
{"(bl! (nonzero? a1) L100 (no-delay!))"}, {{}}, {{"a1"}}, {{}});
test_case(assembly_from_list({"L100:", "bnel s7, a1, L100"}),
{"(bl! (truthy a1) L100 (no-delay!))"}, {{}}, {{"a1"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BNE_DADDIU) {
test_case(assembly_from_list({"L100:", "bne a1, a2, L100", "daddiu a3, s7, 8"}),
{"(b! (!= a1 a2) L100 (set! a3 #t))"}, {{"a3"}}, {{"a1", "a2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BNE_DSLLV) {
test_case(assembly_from_list({"L100:", "bne a1, a2, L100", "dsllv a3, t0, t1"}),
{"(b! (!= a1 a2) L100 (set! a3 (sll t0 t1)))"}, {{"a3"}}, {{"a1", "a2", "t0", "t1"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, BNE_DSUBU) {
test_case(assembly_from_list({"L100:", "bne a1, a2, L100", "dsubu a3, r0, t1"}),
{"(b! (!= a1 a2) L100 (set! a3 (- t1)))"}, {{"a3"}}, {{"a1", "a2", "t1"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BNE_LW) {
test_case(assembly_from_list({"L100:", "bne a1, a2, L100", "lw a3, pair(s7)"}),
{"(b! (!= a1 a2) L100 (set! a3 pair))"}, {{"a3"}}, {{"a1", "a2"}}, {{}});
test_case(assembly_from_list({"L100:", "bne a1, a2, L100", "lw a3, binteger(s7)"}),
{"(b! (!= a1 a2) L100 (set! a3 binteger))"}, {{"a3"}}, {{"a1", "a2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BNE_OR) {
test_case(assembly_from_list({"L100:", "bne a1, a2, L100", "or a3, s7, r0"}),
{"(b! (!= a1 a2) L100 (set! a3 #f))"}, {{"a3"}}, {{"a1", "a2"}}, {{}});
test_case(assembly_from_list({"L100:", "bne a1, a2, L100", "or a3, t0, r0"}),
{"(b! (!= a1 a2) L100 (set! a3 t0))"}, {{"a3"}}, {{"a1", "a2", "t0"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, BNE_SLL) {
test_case(assembly_from_list({"L100:", "bne a1, a2, L100", "sll r0, r0, 0"}),
{"(b! (!= a1 a2) L100 (nop!))"}, {{}}, {{"a1", "a2"}}, {{}});
test_case(assembly_from_list({"L100:", "bne a1, r0, L100", "sll r0, r0, 0"}),
{"(b! (nonzero? a1) L100 (nop!))"}, {{}}, {{"a1"}}, {{}});
test_case(assembly_from_list({"L100:", "bne s7, a1, L100", "sll r0, r0, 0"}),
{"(b! (truthy a1) L100 (nop!))"}, {{}}, {{"a1"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, CEQS_BC1F_SLL) {
test_case(assembly_from_list({"L100:", "c.eq.s f1, f2", "bc1f L100", "sll r0, r0, 0"}),
{"(b! (!=.s f1 f2) L100 (nop!))"}, {{}}, {{"f1", "f2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, CEQS_BC1T_SLL) {
test_case(assembly_from_list({"L100:", "c.eq.s f1, f2", "bc1t L100", "sll r0, r0, 0"}),
{"(b! (=.s f1 f2) L100 (nop!))"}, {{}}, {{"f1", "f2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, CLES_BC1F_SLL) {
test_case(assembly_from_list({"L100:", "c.le.s f1, f2", "bc1f L100", "sll r0, r0, 0"}),
{"(b! (>.s f1 f2) L100 (nop!))"}, {{}}, {{"f1", "f2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, CLES_BC1T_SLL) {
test_case(assembly_from_list({"L100:", "c.le.s f1, f2", "bc1t L100", "sll r0, r0, 0"}),
{"(b! (<=.s f1 f2) L100 (nop!))"}, {{}}, {{"f1", "f2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, CLTS_BC1F_SLL) {
test_case(assembly_from_list({"L100:", "c.lt.s f1, f2", "bc1f L100", "sll r0, r0, 0"}),
{"(b! (>=.s f1 f2) L100 (nop!))"}, {{}}, {{"f1", "f2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, CLTS_BC1T_SLL) {
test_case(assembly_from_list({"L100:", "c.lt.s f1, f2", "bc1t L100", "sll r0, r0, 0"}),
{"(b! (<.s f1 f2) L100 (nop!))"}, {{}}, {{"f1", "f2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, CVTSW) {
test_case(assembly_from_list({"cvt.s.w f1, f2"}), {"(set! f1 (i2f f2))"}, {{"f1"}}, {{"f2"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, CVTWS) {
test_case(assembly_from_list({"cvt.w.s f1, f2"}), {"(set! f1 (f2i f2))"}, {{"f1"}}, {{"f2"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DADDIU) {
test_case(assembly_from_list({"daddiu a1, s7, test"}), {"(set! a1 'test)"}, {{"a1"}}, {{}}, {{}});
test_case(assembly_from_list({"daddiu a1, s7, -10"}), {"(set! a1 '())"}, {{"a1"}}, {{}}, {{}});
test_case(assembly_from_list({"daddiu a1, s7, -32768"}), {"(set! a1 __START-OF-TABLE__)"},
{{"a1"}}, {{}}, {{}});
test_case(assembly_from_list({"daddiu a1, s7, 8"}), {"(set! a1 #t)"}, {{"a1"}}, {{}}, {{}});
test_case(assembly_from_list({"L123:", "daddiu a1, fp, L123"}), {"(set! a1 L123)"}, {{"a1"}},
{{}}, {{}});
test_case(assembly_from_list({"daddiu a1, a2, 1234"}), {"(set! a1 (+ a2 1234))"}, {{"a1"}},
{{"a2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DADDIU_MOVN) {
test_case(assembly_from_list({"daddiu a1, s7, 8", "movn a1, s7, a0"}), {"(set! a1 (zero? a0))"},
{{"a1"}}, {{"a0"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DADDIU_MOVZ) {
test_case(assembly_from_list({"daddiu a1, s7, 8", "movz a1, s7, a0"}),
{"(set! a1 (nonzero? a0))"}, {{"a1"}}, {{"a0"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DADDU) {
test_case(assembly_from_list({"daddu a1, a2, a3"}), {"(set! a1 (+ a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
test_case(assembly_from_list({"daddu v1, r0, v0"}), {"(set! v1 (+ v0 0))"}, {{"v1"}}, {{"v0"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DIVS) {
test_case(assembly_from_list({"div.s f1, f2, f3"}), {"(set! f1 (/.s f2 f3))"}, {{"f1"}},
{{"f2", "f3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DIVU_MFHI) {
test_case(assembly_from_list({"divu a1, a2", "mfhi a3"}), {"(set! a3 (mod.ui a1 a2))"}, {{"a3"}},
{{"a1", "a2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DIVU_MFLO) {
test_case(assembly_from_list({"divu a1, a2", "mflo a3"}), {"(set! a3 (/.ui a1 a2))"}, {{"a3"}},
{{"a1", "a2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DIV_MFHI) {
test_case(assembly_from_list({"div a1, a2", "mfhi a3"}), {"(set! a3 (mod.si a1 a2))"}, {{"a3"}},
{{"a1", "a2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DIV_MFLO) {
test_case(assembly_from_list({"div a1, a2", "mflo a3"}), {"(set! a3 (/.si a1 a2))"}, {{"a3"}},
{{"a1", "a2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DSLL) {
test_case(assembly_from_list({"dsll a2, a3, 3"}), {"(set! a2 (sll a3 3))"}, {{"a2"}}, {{"a3"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DSLL32) {
test_case(assembly_from_list({"dsll32 a2, a3, 3"}), {"(set! a2 (sll a3 35))"}, {{"a2"}}, {{"a3"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DSLL32_SLT_BEQ_SLL) {
test_case(assembly_from_list({"L100:", "dsll32 v1, s4, 30", "slt v1, v1, r0", "beq v1, r0, L100",
"sll r0, r0, 0"}),
{"(b! (not-pair? s4) L100 (nop!))"}, {{}}, {{"s4"}}, {{"v1"}});
}
TEST(DecompilerAtomicOpBuilder, DSLLV) {
test_case(assembly_from_list({"dsllv a1, a2, a3"}), {"(set! a1 (sll a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DSRA) {
test_case(assembly_from_list({"dsra a2, a3, 3"}), {"(set! a2 (sra a3 3))"}, {{"a2"}}, {{"a3"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DSRA32) {
test_case(assembly_from_list({"dsra32 a2, a3, 3"}), {"(set! a2 (sra a3 35))"}, {{"a2"}}, {{"a3"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DSRAV) {
test_case(assembly_from_list({"dsrav a1, a2, a3"}), {"(set! a1 (sra a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DSRL) {
test_case(assembly_from_list({"dsrl a2, a3, 3"}), {"(set! a2 (srl a3 3))"}, {{"a2"}}, {{"a3"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DSRL32) {
test_case(assembly_from_list({"dsrl32 a2, a3, 3"}), {"(set! a2 (srl a3 35))"}, {{"a2"}}, {{"a3"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DSRLV) {
test_case(assembly_from_list({"dsrlv a1, a2, a3"}), {"(set! a1 (srl a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, DSUBU) {
test_case(assembly_from_list({"dsubu a1, a2, a3"}), {"(set! a1 (- a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
test_case(assembly_from_list({"dsubu a1, r0, a3"}), {"(set! a1 (- a3))"}, {{"a1"}}, {{"a3"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, DSUBU_DADDIU_MOVN) {
test_case(assembly_from_list({"dsubu a0, a1, a2", "daddiu t0, s7, 8", "movn t0, s7, a0"}),
{"(set! t0 (= a1 a2))"}, {{"t0"}}, {{"a1", "a2"}}, {{"a0"}});
}
TEST(DecompilerAtomicOpBuilder, DSUBU_DADDIU_MOVZ) {
test_case(assembly_from_list({"dsubu a0, a1, a2", "daddiu t0, s7, 8", "movz t0, s7, a0"}),
{"(set! t0 (!= a1 a2))"}, {{"t0"}}, {{"a1", "a2"}}, {{"a0"}});
}
TEST(DecompilerAtomicOpBuilder, JALR_SLL) {
test_case(assembly_from_list({"jalr ra, t9", "sll v0, ra, 0"}), {"(call!)"}, {{"v0"}}, {{"t9"}},
{{"a0", "a1", "a2", "a3", "t0", "t1", "t2", "t3", "t4", "t5", "t6", "t7", "t8", "t9",
"at", "v1"}});
}
TEST(DecompilerAtomicOpBuilder, LB) {
test_case(assembly_from_list({"L123:", "lb a3, L123(fp)"}), {"(set! a3 (l.b L123))"}, {{"a3"}},
{{}}, {{}});
test_case(assembly_from_list({"lb a2, 0(a3)"}), {"(set! a2 (l.b a3))"}, {{"a2"}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"lb a2, 12(a3)"}), {"(set! a2 (l.b (+ a3 12)))"}, {{"a2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LBU) {
test_case(assembly_from_list({"L123:", "lbu a3, L123(fp)"}), {"(set! a3 (l.bu L123))"}, {{"a3"}},
{{}}, {{}});
test_case(assembly_from_list({"lbu a2, 0(a3)"}), {"(set! a2 (l.bu a3))"}, {{"a2"}}, {{"a3"}},
{{}});
test_case(assembly_from_list({"lbu a2, 12(a3)"}), {"(set! a2 (l.bu (+ a3 12)))"}, {{"a2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LD) {
test_case(assembly_from_list({"L123:", "ld a3, L123(fp)"}), {"(set! a3 (l.d L123))"}, {{"a3"}},
{{}}, {{}});
test_case(assembly_from_list({"ld a2, 0(a3)"}), {"(set! a2 (l.d a3))"}, {{"a2"}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"ld a2, 12(a3)"}), {"(set! a2 (l.d (+ a3 12)))"}, {{"a2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LH) {
test_case(assembly_from_list({"L123:", "lh a3, L123(fp)"}), {"(set! a3 (l.h L123))"}, {{"a3"}},
{{}}, {{}});
test_case(assembly_from_list({"lh a2, 0(a3)"}), {"(set! a2 (l.h a3))"}, {{"a2"}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"lh a2, 12(a3)"}), {"(set! a2 (l.h (+ a3 12)))"}, {{"a2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LHU) {
test_case(assembly_from_list({"L123:", "lhu a3, L123(fp)"}), {"(set! a3 (l.hu L123))"}, {{"a3"}},
{{}}, {{}});
test_case(assembly_from_list({"lhu a2, 0(a3)"}), {"(set! a2 (l.hu a3))"}, {{"a2"}}, {{"a3"}},
{{}});
test_case(assembly_from_list({"lhu a2, 12(a3)"}), {"(set! a2 (l.hu (+ a3 12)))"}, {{"a2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LUI) {
test_case(assembly_from_list({"lui a3, 2"}), {"(set! a3 #x20000)"}, {{"a3"}}, {{}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LUI_ORI) {
test_case(assembly_from_list({"L100:", "lui a0, 2", "ori a1, a0, 3"}), {"(set! a1 #x20003)"},
{{"a1"}}, {{}}, {{"a0"}});
test_case(assembly_from_list({"L100:", "lui a0, 2", "ori a0, a0, 3"}), {"(set! a0 #x20003)"},
{{"a0"}}, {{}}, {{}});
test_case(assembly_from_list({"L100:", "lui a0, L100", "ori a1, a0, L100"}), {"(set! a1 L100)"},
{{"a1"}}, {{}}, {{"a0"}});
test_case(assembly_from_list({"L100:", "lui a0, L100", "ori a0, a0, L100"}), {"(set! a0 L100)"},
{{"a0"}}, {{}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LUI_ORI_ADDU) {
test_case(assembly_from_list({"L100:", "lui a0, L100", "ori a1, a0, L100", "addu a1, fp, a1"}),
{"(set! a1 L100)"}, {{"a1"}}, {{}}, {{"a0"}});
test_case(assembly_from_list({"L100:", "lui a1, L100", "ori a1, a1, L100", "addu a1, fp, a1"}),
{"(set! a1 L100)"}, {{"a1"}}, {{}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LW) {
test_case(assembly_from_list({"lw r0, 2(r0)"}), {"(break!)"}, {{}}, {{}}, {{}});
test_case(assembly_from_list({"lw a2, test(s7)"}), {"(set! a2 test)"}, {{"a2"}}, {{}}, {{}});
test_case(assembly_from_list({"L123:", "lw a3, L123(fp)"}), {"(set! a3 (l.w L123))"}, {{"a3"}},
{{}}, {{}});
test_case(assembly_from_list({"lw a2, 0(a3)"}), {"(set! a2 (l.w a3))"}, {{"a2"}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"lw a2, 12(a3)"}), {"(set! a2 (l.w (+ a3 12)))"}, {{"a2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LWC1) {
test_case(assembly_from_list({"L123:", "lwc1 f3, L123(fp)"}), {"(set! f3 (l.f L123))"}, {{"f3"}},
{{}}, {{}});
test_case(assembly_from_list({"lwc1 f2, 0(a3)"}), {"(set! f2 (l.f a3))"}, {{"f2"}}, {{"a3"}},
{{}});
test_case(assembly_from_list({"lwc1 f2, 12(a3)"}), {"(set! f2 (l.f (+ a3 12)))"}, {{"f2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, LWU) {
test_case(assembly_from_list({"L123:", "lwu a3, L123(fp)"}), {"(set! a3 (l.wu L123))"}, {{"a3"}},
{{}}, {{}});
test_case(assembly_from_list({"lwu a2, 0(a3)"}), {"(set! a2 (l.wu a3))"}, {{"a2"}}, {{"a3"}},
{{}});
test_case(assembly_from_list({"lwu a2, 12(a3)"}), {"(set! a2 (l.wu (+ a3 12)))"}, {{"a2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MAXS) {
test_case(assembly_from_list({"max.s f1, f2, f3"}), {"(set! f1 (max.s f2 f3))"}, {{"f1"}},
{{"f2", "f3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MFC1) {
test_case(assembly_from_list({"mfc1 a1, f3"}), {"(set! a1 (fpr->gpr f3))"}, {{"a1"}}, {{"f3"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, MINS) {
test_case(assembly_from_list({"min.s f1, f2, f3"}), {"(set! f1 (min.s f2 f3))"}, {{"f1"}},
{{"f2", "f3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MOVN) {
test_case(assembly_from_list({"movn a1, s7, a2"}), {"(cmove-#f-nonzero a1 a2 a1)"}, {{"a1"}},
{{"a2", "a1"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MOVS) {
test_case(assembly_from_list({"mov.s f1, f2"}), {"(set! f1 f2)"}, {{"f1"}}, {{"f2"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MOVZ) {
test_case(assembly_from_list({"movz a1, s7, a2"}), {"(cmove-#f-zero a1 a2 a1)"}, {{"a1"}},
{{"a2", "a1"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MTC1) {
test_case(assembly_from_list({"mtc1 f3, a1"}), {"(set! f3 (gpr->fpr a1))"}, {{"f3"}}, {{"a1"}},
{{}});
test_case(assembly_from_list({"mtc1 f3, r0"}), {"(set! f3 0)"}, {{"f3"}}, {{}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MULS) {
test_case(assembly_from_list({"mul.s f1, f2, f3"}), {"(set! f1 (*.s f2 f3))"}, {{"f1"}},
{{"f2", "f3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MULT3) {
test_case(assembly_from_list({"mult3 a1, a2, a3"}), {"(set! a1 (*.si a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, MULTU3) {
test_case(assembly_from_list({"multu3 a1, a2, a3"}), {"(set! a1 (*.ui a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, NEGS) {
test_case(assembly_from_list({"neg.s f1, f2"}), {"(set! f1 (neg.s f2))"}, {{"f1"}}, {{"f2"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, NOR) {
test_case(assembly_from_list({"nor a1, a2, r0"}), {"(set! a1 (lognot a2))"}, {{"a1"}}, {{"a2"}},
{{}});
test_case(assembly_from_list({"nor a1, a2, a3"}), {"(set! a1 (lognor a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, OR) {
test_case(assembly_from_list({"or a1, a2, a3"}), {"(set! a1 (logior a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
test_case(assembly_from_list({"or a2, r0, r0"}), {"(set! a2 0)"}, {{"a2"}}, {{}}, {{}});
test_case(assembly_from_list({"or a1, s7, r0"}), {"(set! a1 #f)"}, {{"a1"}}, {{}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, ORI) {
test_case(assembly_from_list({"ori a2, r0, 1234"}), {"(set! a2 1234)"}, {{"a2"}}, {{}}, {{}});
test_case(assembly_from_list({"ori a2, r0, -1234"}), {"(set! a2 -1234)"}, {{"a2"}}, {{}}, {{}});
test_case(assembly_from_list({"ori a2, a3, -1234"}), {"(set! a2 (logior a3 -1234))"}, {{"a2"}},
{{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, SB) {
test_case(assembly_from_list({"sb a1, 2(a3)"}), {"(s.b! (+ a3 2) a1)"}, {{}}, {{"a1", "a3"}},
{{}});
test_case(assembly_from_list({"sb a1, 0(a3)"}), {"(s.b! a3 a1)"}, {{}}, {{"a1", "a3"}}, {{}});
test_case(assembly_from_list({"sb s7, 2(a3)"}), {"(s.b! (+ a3 2) #f)"}, {{}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"sb s7, 0(a3)"}), {"(s.b! a3 #f)"}, {{}}, {{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, SD) {
test_case(assembly_from_list({"sd a1, 2(a3)"}), {"(s.d! (+ a3 2) a1)"}, {{}}, {{"a1", "a3"}},
{{}});
test_case(assembly_from_list({"sd a1, 0(a3)"}), {"(s.d! a3 a1)"}, {{}}, {{"a1", "a3"}}, {{}});
test_case(assembly_from_list({"sd s7, 2(a3)"}), {"(s.d! (+ a3 2) #f)"}, {{}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"sd s7, 0(a3)"}), {"(s.d! a3 #f)"}, {{}}, {{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, SH) {
test_case(assembly_from_list({"sh a1, 2(a3)"}), {"(s.h! (+ a3 2) a1)"}, {{}}, {{"a1", "a3"}},
{{}});
test_case(assembly_from_list({"sh a1, 0(a3)"}), {"(s.h! a3 a1)"}, {{}}, {{"a1", "a3"}}, {{}});
test_case(assembly_from_list({"sh s7, 2(a3)"}), {"(s.h! (+ a3 2) #f)"}, {{}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"sh s7, 0(a3)"}), {"(s.h! a3 #f)"}, {{}}, {{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, SLL) {
test_case(assembly_from_list({"sll r0, r0, 0"}), {"(nop!)"}, {{}}, {{}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, SLTIU_BEQ_OR) {
test_case(assembly_from_list({"L100:", "sltiu a3, a0, 12", "beq a3, r0, L100", "or a3, s7, r0"}),
{"(b! (>=.ui a0 12) L100 (set! a3 #f))"}, {{"a3"}}, {{"a0"}}, {{}});
test_case(assembly_from_list({"L100:", "sltiu a3, a0, 12", "beq a3, r0, L100", "or a2, s7, r0"}),
{"(b! (>=.ui a0 12) L100 (set! a2 #f))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTIU_BNE_OR) {
test_case(assembly_from_list({"L100:", "sltiu a3, a0, 12", "bne a3, r0, L100", "or a3, s7, r0"}),
{"(b! (<.ui a0 12) L100 (set! a3 #f))"}, {{"a3"}}, {{"a0"}}, {{}});
test_case(assembly_from_list({"L100:", "sltiu a3, a0, 12", "bne a3, r0, L100", "or a2, s7, r0"}),
{"(b! (<.ui a0 12) L100 (set! a2 #f))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTIU_DADDIU_MOVN) {
test_case(assembly_from_list({"sltiu a3, a0, 12", "daddiu a2, s7, 8", "movn a2, s7, a3"}),
{"(set! a2 (>=.ui a0 12))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTIU_DADDIU_MOVZ) {
test_case(assembly_from_list({"sltiu a3, a0, 12", "daddiu a2, s7, 8", "movz a2, s7, a3"}),
{"(set! a2 (<.ui a0 12))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTI_BEQ_OR) {
test_case(assembly_from_list({"L100:", "slti a3, a0, 12", "beq a3, r0, L100", "or a3, s7, r0"}),
{"(b! (>=.si a0 12) L100 (set! a3 #f))"}, {{"a3"}}, {{"a0"}}, {{}});
test_case(assembly_from_list({"L100:", "slti a3, a0, 12", "beq a3, r0, L100", "or a2, s7, r0"}),
{"(b! (>=.si a0 12) L100 (set! a2 #f))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTI_BNE_OR) {
test_case(assembly_from_list({"L100:", "slti a3, a0, 12", "bne a3, r0, L100", "or a3, s7, r0"}),
{"(b! (<.si a0 12) L100 (set! a3 #f))"}, {{"a3"}}, {{"a0"}}, {{}});
test_case(assembly_from_list({"L100:", "slti a3, a0, 12", "bne a3, r0, L100", "or a2, s7, r0"}),
{"(b! (<.si a0 12) L100 (set! a2 #f))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTI_DADDIU_MOVN) {
test_case(assembly_from_list({"slti a3, a0, 12", "daddiu a2, s7, 8", "movn a2, s7, a3"}),
{"(set! a2 (>=.si a0 12))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTI_DADDIU_MOVZ) {
test_case(assembly_from_list({"slti a3, a0, 12", "daddiu a2, s7, 8", "movz a2, s7, a3"}),
{"(set! a2 (<.si a0 12))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTU_BEQ_OR) {
test_case(assembly_from_list({"L100:", "sltu a3, a0, a1", "beq a3, r0, L100", "or a3, s7, r0"}),
{"(b! (>=.ui a0 a1) L100 (set! a3 #f))"}, {{"a3"}}, {{"a0", "a1"}}, {{}});
test_case(assembly_from_list({"L100:", "sltu a3, a0, a1", "beq a3, r0, L100", "or a2, s7, r0"}),
{"(b! (>=.ui a0 a1) L100 (set! a2 #f))"}, {{"a2"}}, {{"a0", "a1"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTU_BNE_OR) {
test_case(assembly_from_list({"L100:", "sltu a3, a0, a1", "bne a3, r0, L100", "or a3, s7, r0"}),
{"(b! (<.ui a0 a1) L100 (set! a3 #f))"}, {{"a3"}}, {{"a0", "a1"}}, {{}});
test_case(assembly_from_list({"L100:", "sltu a3, a0, a1", "bne a3, r0, L100", "or a2, s7, r0"}),
{"(b! (<.ui a0 a1) L100 (set! a2 #f))"}, {{"a2"}}, {{"a0", "a1"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTU_DADDIU_MOVN) {
test_case(assembly_from_list({"sltu a3, a0, a1", "daddiu a2, s7, 8", "movn a2, s7, a3"}),
{"(set! a2 (>=.ui a0 a1))"}, {{"a2"}}, {{"a0", "a1"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTU_DADDIU_MOVZ) {
test_case(assembly_from_list({"sltu a3, a0, a1", "daddiu a2, s7, 8", "movz a2, s7, a3"}),
{"(set! a2 (<.ui a0 a1))"}, {{"a2"}}, {{"a0", "a1"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLTU_MOVN) {
test_case(assembly_from_list({"sltu a0, a1, a2", "movn a1, a2, a0"}),
{"(set! a1 (max.ui a1 a2))"}, {{"a1"}}, {{"a1", "a2"}}, {{"a0"}});
}
TEST(DecompilerAtomicOpBuilder, SLTU_MOVZ) {
test_case(assembly_from_list({"sltu a0, a1, a2", "movz a1, a2, a0"}),
{"(set! a1 (min.ui a1 a2))"}, {{"a1"}}, {{"a1", "a2"}}, {{"a0"}});
}
TEST(DecompilerAtomicOpBuilder, SLT_BEQ_OR) {
test_case(assembly_from_list({"L100:", "slt a3, a0, a1", "beq a3, r0, L100", "or a3, s7, r0"}),
{"(b! (>=.si a0 a1) L100 (set! a3 #f))"}, {{"a3"}}, {{"a0", "a1"}}, {{}});
test_case(assembly_from_list({"L100:", "slt a3, a0, a1", "beq a3, r0, L100", "or a2, s7, r0"}),
{"(b! (>=.si a0 a1) L100 (set! a2 #f))"}, {{"a2"}}, {{"a0", "a1"}}, {{"a3"}});
test_case(assembly_from_list({"L100:", "slt v1, s2, r0", "beq v1, r0, L100", "or v1, s7, r0"}),
{"(b! (>=0.si s2) L100 (set! v1 #f))"}, {{"v1"}}, {{"s2"}}, {{}});
test_case(assembly_from_list({"L100:", "slt v1, s2, r0", "beq v1, r0, L100", "or v0, s7, r0"}),
{"(b! (>=0.si s2) L100 (set! v0 #f))"}, {{"v0"}}, {{"s2"}}, {{"v1"}});
}
TEST(DecompilerAtomicOpBuilder, SLT_BNE_OR) {
test_case(assembly_from_list({"L100:", "slt a3, a0, a1", "bne a3, r0, L100", "or a3, s7, r0"}),
{"(b! (<.si a0 a1) L100 (set! a3 #f))"}, {{"a3"}}, {{"a0", "a1"}}, {{}});
test_case(assembly_from_list({"L100:", "slt a3, a0, a1", "bne a3, r0, L100", "or a2, s7, r0"}),
{"(b! (<.si a0 a1) L100 (set! a2 #f))"}, {{"a2"}}, {{"a0", "a1"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLT_DADDIU_MOVN) {
test_case(assembly_from_list({"slt a3, a0, a1", "daddiu a2, s7, 8", "movn a2, s7, a3"}),
{"(set! a2 (>=.si a0 a1))"}, {{"a2"}}, {{"a0", "a1"}}, {{"a3"}});
test_case(assembly_from_list({"slt a3, a0, r0", "daddiu a2, s7, 8", "movn a2, s7, a3"}),
{"(set! a2 (>=0.si a0))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLT_DADDIU_MOVZ) {
test_case(assembly_from_list({"slt a3, a0, a1", "daddiu a2, s7, 8", "movz a2, s7, a3"}),
{"(set! a2 (<.si a0 a1))"}, {{"a2"}}, {{"a0", "a1"}}, {{"a3"}});
test_case(assembly_from_list({"slt a3, a0, r0", "daddiu a2, s7, 8", "movz a2, s7, a3"}),
{"(set! a2 (<0.si a0))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
test_case(assembly_from_list({"slt a3, r0, a0", "daddiu a2, s7, 8", "movz a2, s7, a3"}),
{"(set! a2 (>0.si a0))"}, {{"a2"}}, {{"a0"}}, {{"a3"}});
}
TEST(DecompilerAtomicOpBuilder, SLT_MOVN) {
test_case(assembly_from_list({"slt a0, a1, a2", "movn a1, a2, a0"}), {"(set! a1 (max.si a1 a2))"},
{{"a1"}}, {{"a1", "a2"}}, {{"a0"}});
}
TEST(DecompilerAtomicOpBuilder, SLT_MOVZ) {
test_case(assembly_from_list({"slt a0, a1, a2", "movz a1, a2, a0"}), {"(set! a1 (min.si a1 a2))"},
{{"a1"}}, {{"a1", "a2"}}, {{"a0"}});
}
TEST(DecompilerAtomicOpBuilder, SQRTS) {
test_case(assembly_from_list({"sqrt.s f1, f2"}), {"(set! f1 (sqrt.s f2))"}, {{"f1"}}, {{"f2"}},
{{}});
}
TEST(DecompilerAtomicOpBuilder, SUBS) {
test_case(assembly_from_list({"sub.s f1, f2, f3"}), {"(set! f1 (-.s f2 f3))"}, {{"f1"}},
{{"f2", "f3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, SW) {
test_case(assembly_from_list({"sw a1, test(s7)"}), {"(s.w! test a1)"}, {{}}, {{"a1"}}, {{}});
test_case(assembly_from_list({"sw s7, test(s7)"}), {"(s.w! test #f)"}, {{}}, {{}}, {{}});
test_case(assembly_from_list({"sw a1, 2(a3)"}), {"(s.w! (+ a3 2) a1)"}, {{}}, {{"a1", "a3"}},
{{}});
test_case(assembly_from_list({"sw a1, 0(a3)"}), {"(s.w! a3 a1)"}, {{}}, {{"a1", "a3"}}, {{}});
test_case(assembly_from_list({"sw s7, 2(a3)"}), {"(s.w! (+ a3 2) #f)"}, {{}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"sw s7, 0(a3)"}), {"(s.w! a3 #f)"}, {{}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"sw r0, test(s7)"}), {"(s.w! test 0)"}, {{}}, {{}}, {{}});
test_case(assembly_from_list({"sw r0, 2(a3)"}), {"(s.w! (+ a3 2) 0)"}, {{}}, {{"a3"}}, {{}});
test_case(assembly_from_list({"sw r0, 0(a3)"}), {"(s.w! a3 0)"}, {{}}, {{"a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, SWC1) {
test_case(assembly_from_list({"swc1 f2, 2(a3)"}), {"(s.f! (+ a3 2) f2)"}, {{}}, {{"f2", "a3"}},
{{}});
test_case(assembly_from_list({"swc1 f2, 0(a3)"}), {"(s.f! a3 f2)"}, {{}}, {{"f2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, XOR) {
test_case(assembly_from_list({"xor a1, a2, a3"}), {"(set! a1 (logxor a2 a3))"}, {{"a1"}},
{{"a2", "a3"}}, {{}});
}
TEST(DecompilerAtomicOpBuilder, XORI) {
test_case(assembly_from_list({"xori a1, a2, 1234"}), {"(set! a1 (logxor a2 1234))"}, {{"a1"}},
{{"a2"}}, {{}});
}