a07663a212
* Add integration test and a fix Coverity scan issue * Return `NULL` if `target->platforms` is `NULL` in `rz_arch_platform_target_new` * Move the call of the API which adds the flags/comments for both CPU and platform profiles inside `rz_core_analysis_all()`. * Use `rz_file_is_directory` instead of `rz_file_exists` for the integration tests * Use `RZ_SYS_DIR` on the path while checking for `.tmp` folder * Add a comma at the end of the list on \`meson.build\`
122 lines
4.4 KiB
C
122 lines
4.4 KiB
C
#include <rz_core.h>
|
|
#include <rz_arch.h>
|
|
#include <rz_project.h>
|
|
|
|
#include "../unit/minunit.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);
|
|
|
|
rz_config_set(core->config, "asm.cpu", "ATTiny48");
|
|
const char *asmcpu = rz_config_get(core->config, "asm.cpu");
|
|
const char *dir_prefix = rz_config_get(core->config, "dir.prefix");
|
|
const char *asmarch = rz_config_get(core->config, "asm.arch");
|
|
rz_arch_profiles_init(core->analysis->arch_target, asmcpu, asmarch, dir_prefix);
|
|
|
|
// 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");
|
|
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);
|
|
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);
|
|
|
|
rz_config_set(core->config, "asm.cpu", "arm1176");
|
|
rz_config_set(core->config, "asm.platform", "bcm2835");
|
|
const char *asmcpu = rz_config_get(core->config, "asm.cpu");
|
|
const char *dir_prefix = rz_config_get(core->config, "dir.prefix");
|
|
const char *asmarch = rz_config_get(core->config, "asm.arch");
|
|
const char *asmplatform = rz_config_get(core->config, "asm.platform");
|
|
rz_arch_platform_init(core->analysis->platform_target, asmarch, asmcpu, asmplatform, dir_prefix);
|
|
|
|
// 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");
|
|
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);
|
|
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)
|