From f569417878c2e8ee7cc727d7bdd8ac94fe5f7e44 Mon Sep 17 00:00:00 2001
From: Quentin DUCASSE <quentin.ducasse@ensta-bretagne.org>
Date: Wed, 4 May 2022 17:18:47 +0200
Subject: [PATCH] Equivalent tests for riscv

---
 tests/unit/test_riscv.c | 87 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)

diff --git a/tests/unit/test_riscv.c b/tests/unit/test_riscv.c
index 3fa50f60..a8a97a36 100644
--- a/tests/unit/test_riscv.c
+++ b/tests/unit/test_riscv.c
@@ -537,6 +537,91 @@ static void test_riscv64_mmio_map(void)
     OK(uc_close(uc));
 }
 
+
+static void test_riscv_correct_address_in_small_jump_hook_callback(uc_engine *uc, int type, uint64_t address, int size, int64_t value, void *user_data)
+{
+  // Check registers
+  uint64_t r_x5 = 0x0;
+  uint64_t r_pc = 0x0;
+  OK(uc_reg_read(uc, UC_RISCV_REG_X5, &r_x5));
+  OK(uc_reg_read(uc, UC_RISCV_REG_PC, &r_pc));
+  TEST_CHECK(r_x5 == 0x7F00);
+  TEST_CHECK(r_pc == 0x7F00);
+
+  // Check address
+  // printf("%lx\n", address);
+  TEST_CHECK(address == 0x7F00);
+}
+
+static void test_riscv_correct_address_in_small_jump_hook(void)
+{
+    uc_engine *uc;
+    // li 0x7F00, x5  >  lui t0, 8; addiw t0, t0, -256;
+    // jr x5
+    char code[] = "\xb7\x82\x00\x00\x9b\x82\x02\xf0\x67\x80\x02\x00";
+
+    uint64_t r_x5 = 0x0;
+    uint64_t r_pc = 0x0;
+    uc_hook hook;
+
+    uc_common_setup(&uc, UC_ARCH_RISCV, UC_MODE_RISCV64, code, sizeof(code) - 1);
+    OK(uc_hook_add(uc, &hook, UC_HOOK_MEM_UNMAPPED, test_riscv_correct_address_in_small_jump_hook_callback, NULL, 1, 0));
+
+    uc_assert_err(
+        UC_ERR_FETCH_UNMAPPED,
+        uc_emu_start(uc, code_start, code_start + sizeof(code) - 1, 0, 0));
+
+    OK(uc_reg_read(uc, UC_RISCV_REG_X5, &r_x5));
+    OK(uc_reg_read(uc, UC_RISCV_REG_PC, &r_pc));
+    TEST_CHECK(r_x5 == 0x7F00);
+    TEST_CHECK(r_pc == 0x7F00);
+
+    OK(uc_close(uc));
+}
+
+static void test_riscv_correct_address_in_long_jump_hook_callback(uc_engine *uc, int type, uint64_t address, int size, int64_t value, void *user_data)
+{
+  // Check registers
+  uint64_t r_x5 = 0x0;
+  uint64_t r_pc = 0x0;
+  OK(uc_reg_read(uc, UC_RISCV_REG_X5, &r_x5));
+  OK(uc_reg_read(uc, UC_RISCV_REG_PC, &r_pc));
+  TEST_CHECK(r_x5 == 0x7FFFFFFFFFFFFF00);
+  TEST_CHECK(r_pc == 0x7FFFFFFFFFFFFF00);
+
+  // Check address
+  // printf("%lx\n", address);
+  TEST_CHECK(address == 0x7FFFFFFFFFFFFF00);
+}
+
+static void test_riscv_correct_address_in_long_jump_hook(void)
+{
+    uc_engine *uc;
+    // li 0x7FFFFFFFFFFFFF00, x5  >  addi t0, zero, -1; slli t0, t0, 63; addi t0, t0, -256;
+    // jr x5
+    char code[] = "\x93\x02\xf0\xff\x93\x92\xf2\x03\x93\x82\x02\xf0\x67\x80\x02\x00";
+
+    uint64_t r_x5 = 0x0;
+    uint64_t r_pc = 0x0;
+    uc_hook hook;
+
+    uc_common_setup(&uc, UC_ARCH_RISCV, UC_MODE_RISCV64, code, sizeof(code) - 1);
+    OK(uc_hook_add(uc, &hook, UC_HOOK_MEM_UNMAPPED, test_riscv_correct_address_in_long_jump_hook_callback, NULL, 1, 0));
+
+    uc_assert_err(
+        UC_ERR_FETCH_UNMAPPED,
+        uc_emu_start(uc, code_start, code_start + sizeof(code) - 1, 0, 0));
+
+    OK(uc_reg_read(uc, UC_RISCV_REG_X5, &r_x5));
+    OK(uc_reg_read(uc, UC_RISCV_REG_PC, &r_pc));
+    TEST_CHECK(r_x5 == 0x7FFFFFFFFFFFFF00);
+    TEST_CHECK(r_pc == 0x7FFFFFFFFFFFFF00);
+
+    OK(uc_close(uc));
+}
+
+
+
 TEST_LIST = {
     {"test_riscv32_nop", test_riscv32_nop},
     {"test_riscv64_nop", test_riscv64_nop},
@@ -556,4 +641,6 @@ TEST_LIST = {
     {"test_riscv32_map", test_riscv32_map},
     {"test_riscv64_code_patching", test_riscv64_code_patching},
     {"test_riscv64_code_patching_count", test_riscv64_code_patching_count},
+    {"test_riscv_correct_address_in_small_jump_hook", test_riscv_correct_address_in_small_jump_hook},
+    {"test_riscv_correct_address_in_long_jump_hook", test_riscv_correct_address_in_long_jump_hook},
     {NULL, NULL}};