122 lines
4.2 KiB
C
122 lines
4.2 KiB
C
// SPDX-FileCopyrightText: 2020 Aswin C (officialcjunior) <realc@protonmail.com>
|
|
// SPDX-License-Identifier: LGPL-3.0-only
|
|
|
|
#include <rz_core.h>
|
|
#include <rz_arch.h>
|
|
#include <rz_project.h>
|
|
|
|
#include "../unit/minunit.h"
|
|
#include "../unit/rz_arch_buffer.h"
|
|
|
|
bool test_cpu_profiles() {
|
|
// 1. Open the file
|
|
RzCore *core = rz_core_new();
|
|
mu_assert_notnull(core, "new RzCore instance");
|
|
ut64 loadaddr = 0;
|
|
const char *fpath = "bins/firmware/arduino_avr.bin";
|
|
RzCoreFile *file = rz_core_file_open(core, fpath, RZ_PERM_R, loadaddr);
|
|
mu_assert_notnull(file, "opening the firmware");
|
|
rz_core_bin_load(core, fpath, loadaddr);
|
|
|
|
const char *tempfile = rz_file_temp(".sdb");
|
|
rz_file_dump(tempfile, cpu_buffer, sizeof(cpu_buffer), false);
|
|
rz_arch_load_profile_sdb(core->analysis->arch_target, tempfile);
|
|
|
|
// 2. Analyse the file
|
|
rz_arch_profile_add_flag_every_io(core->analysis->arch_target->profile, core->flags);
|
|
|
|
RzFlagItem *item = rz_flag_get(core->flags, "DDRB");
|
|
mu_assert_eq(item->offset, 0x00000004, "Flag DDRB not found");
|
|
item = rz_flag_get(core->flags, "PORTB");
|
|
mu_assert_eq(item->offset, 0x00000005, "Flag PORTB not found");
|
|
|
|
// 3. Save into the project
|
|
if (!rz_file_is_directory(".tmp" RZ_SYS_DIR)) {
|
|
mu_assert_true(rz_sys_mkdir(".tmp/"), "create tmp directory");
|
|
}
|
|
RzProjectErr err = rz_project_save_file(core, ".tmp/cpu_profile.rzdb", true);
|
|
mu_assert_eq(err, RZ_PROJECT_ERR_SUCCESS, "project save err");
|
|
|
|
// 4. Close the file
|
|
rz_core_file_close(file);
|
|
rz_core_free(core);
|
|
|
|
// 5. Create a new core
|
|
core = rz_core_new();
|
|
mu_assert_notnull(core, "new RzCore instance");
|
|
|
|
// 6. Load the previously saved project
|
|
RzSerializeResultInfo *res = rz_serialize_result_info_new();
|
|
mu_assert_notnull(res, "result info new");
|
|
err = rz_project_load_file(core, ".tmp/cpu_profile.rzdb", true, res);
|
|
rz_serialize_result_info_free(res);
|
|
mu_assert_eq(err, RZ_PROJECT_ERR_SUCCESS, "project load err");
|
|
|
|
// 7. Check the values again
|
|
item = rz_flag_get(core->flags, "DDRB");
|
|
mu_assert_eq(item->offset, 0x00000004, "Flag DDRB not found");
|
|
item = rz_flag_get(core->flags, "PORTB");
|
|
mu_assert_eq(item->offset, 0x00000005, "Flag PORTB not found");
|
|
|
|
rz_core_free(core);
|
|
mu_end;
|
|
}
|
|
|
|
bool test_platform_profiles() {
|
|
// 1. Open the file
|
|
RzCore *core = rz_core_new();
|
|
mu_assert_notnull(core, "new RzCore instance");
|
|
ut64 loadaddr = 0;
|
|
const char *fpath = "bins/arm/elf/hello_world";
|
|
RzCoreFile *file = rz_core_file_open(core, fpath, RZ_PERM_R, loadaddr);
|
|
mu_assert_notnull(file, "opening the binary");
|
|
rz_core_bin_load(core, fpath, loadaddr);
|
|
|
|
const char *tempfile = rz_file_temp(".sdb");
|
|
rz_file_dump(tempfile, platform_buffer, sizeof(platform_buffer), false);
|
|
rz_arch_load_platform_sdb(core->analysis->platform_target, tempfile);
|
|
|
|
// 2. Analyse the file
|
|
rz_arch_platform_add_flags_comments(core);
|
|
|
|
RzFlagItem *item = rz_flag_get(core->flags, "AUX_MU_IER_REG");
|
|
mu_assert_eq(item->offset, 0x7e215044, "Flag AUX_MU_IER_REG not found");
|
|
const char *comment = rz_meta_get_string(core->analysis, RZ_META_TYPE_COMMENT, 0x7e804000);
|
|
mu_assert_streq(comment, "Broadcom Serial Controller 1 (BSC)", "Comment unequal!");
|
|
|
|
// 3. Save into the project
|
|
RzProjectErr err = rz_project_save_file(core, ".tmp/cpu_platform.rzdb", true);
|
|
mu_assert_eq(err, RZ_PROJECT_ERR_SUCCESS, "project save err");
|
|
|
|
// 4. Close the file
|
|
rz_core_file_close(file);
|
|
rz_core_free(core);
|
|
|
|
// 5. Create a new core
|
|
core = rz_core_new();
|
|
mu_assert_notnull(core, "new RzCore instance");
|
|
|
|
// 6. Load the previously saved project
|
|
RzSerializeResultInfo *res = rz_serialize_result_info_new();
|
|
mu_assert_notnull(res, "result info new");
|
|
err = rz_project_load_file(core, ".tmp/cpu_platform.rzdb", true, res);
|
|
rz_serialize_result_info_free(res);
|
|
mu_assert_eq(err, RZ_PROJECT_ERR_SUCCESS, "project load err");
|
|
|
|
// 7. Check the values again
|
|
comment = rz_meta_get_string(core->analysis, RZ_META_TYPE_COMMENT, 0x7e804000);
|
|
mu_assert_streq(comment, "Broadcom Serial Controller 1 (BSC)", "Comment not found");
|
|
item = rz_flag_get(core->flags, "AUX_MU_IER_REG");
|
|
mu_assert_eq(item->offset, 0x7e215044, "Flag AUX_MU_IER_REG not found");
|
|
|
|
rz_core_free(core);
|
|
mu_end;
|
|
}
|
|
|
|
int all_tests() {
|
|
mu_run_test(test_cpu_profiles);
|
|
mu_run_test(test_platform_profiles);
|
|
return tests_passed != tests_run;
|
|
}
|
|
|
|
mu_main(all_tests)
|