b5fee3471c
* Hide RzAnalysis structure to force the usage of the C APIs. * Fix rop code & test * Fix linter * Fix rz_analysis_free to return void
267 lines
10 KiB
C
267 lines
10 KiB
C
// SPDX-FileCopyrightText: 2021 Basstorm <basstorm@nyist.edu.cn>
|
|
// SPDX-License-Identifier: LGPL-3.0-only
|
|
|
|
#include <rz_analysis.h>
|
|
#include <rz_core.h>
|
|
#include "test_config.h"
|
|
#include "../unit/minunit.h"
|
|
|
|
bool test_rz_analysis_global_var() {
|
|
RzCore *core = rz_core_new();
|
|
RzAnalysis *analysis = core->analysis;
|
|
RzTypeDB *typedb = rz_analysis_get_type_db(analysis);
|
|
rz_type_db_init(typedb, TEST_BUILD_TYPES_DIR, NULL, 0, NULL);
|
|
|
|
RzAnalysisVarGlobal *glob = rz_analysis_var_global_new("foo", 0x1337);
|
|
mu_assert_notnull(glob, "create a global variable");
|
|
mu_assert_streq(glob->name, "foo", "global var name");
|
|
mu_assert_eq(glob->addr, 0x1337, "global var address");
|
|
mu_assert_null(glob->analysis, "global var analysis");
|
|
RzTypeParser *parser = rz_type_parser_new();
|
|
mu_assert_notnull(parser, "create type parser");
|
|
char *errmsg = NULL;
|
|
RzType *typ = rz_type_parse_string_single(parser, "int", &errmsg);
|
|
mu_assert_notnull(typ, "parsed type");
|
|
rz_analysis_var_global_set_type(glob, typ);
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
RzFlagItem *flag = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_null(flag, "no flag yet");
|
|
|
|
bool added = rz_analysis_var_global_add(analysis, glob);
|
|
mu_assert_true(added, "add global var");
|
|
flag = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_notnull(flag, "global var flag_item");
|
|
mu_assert_eq(flag->offset, glob->addr, "flag item addr");
|
|
mu_assert_streq(flag->name, "foo", "flag item name");
|
|
mu_assert_streq(flag->space->name, RZ_FLAGS_FS_GLOBALS, "flag space");
|
|
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byaddr_at(analysis, 0x1337);
|
|
mu_assert_notnull(glob, "get global var by addr");
|
|
mu_assert_streq(glob->name, "foo", "global var name");
|
|
mu_assert_eq(glob->addr, 0x1337, "global var address");
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byname(analysis, "foo");
|
|
mu_assert_notnull(glob, "get global var by addr");
|
|
mu_assert_streq(glob->name, "foo", "global var name");
|
|
mu_assert_eq(glob->addr, 0x1337, "global var address");
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byaddr_in(analysis, 0x1339); // test RBTree
|
|
mu_assert_notnull(glob, "get global var by addr");
|
|
mu_assert_streq(glob->name, "foo", "global var name");
|
|
mu_assert_eq(glob->addr, 0x1337, "global var address");
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
RzFlagItem *flag_exists = rz_flag_get(core->flags, glob->name);
|
|
mu_assert_notnull(flag_exists, "flag not found");
|
|
|
|
bool rename = rz_analysis_var_global_rename(analysis, "foo", "bar");
|
|
mu_assert_true(rename, "rename global var");
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byname(analysis, "bar");
|
|
mu_assert_notnull(glob, "get global var by addr");
|
|
mu_assert_streq(glob->name, "bar", "global var name");
|
|
mu_assert_streq(flag->name, "bar", "global flag_item name");
|
|
RzFlagItem *flag2 = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_ptreq(flag2, flag, "still same flag");
|
|
|
|
bool deleted = rz_analysis_var_global_delete_byaddr_at(analysis, 0x1337);
|
|
mu_assert_true(deleted, "delete global var");
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byaddr_at(analysis, 0x1337);
|
|
mu_assert_null(glob, "get deleted global var");
|
|
RzFlagItem *flag_deleted = rz_flag_get_i(core->flags, 0x1337);
|
|
mu_assert_null(flag_deleted, "get deleted flag");
|
|
|
|
// re add
|
|
glob = rz_analysis_var_global_new("foo", 0x1337);
|
|
mu_assert_notnull(glob, "create a global variable");
|
|
mu_assert_streq(glob->name, "foo", "global var name");
|
|
mu_assert_eq(glob->addr, 0x1337, "global var address");
|
|
mu_assert_null(glob->analysis, "global var flags");
|
|
errmsg = NULL;
|
|
typ = rz_type_parse_string_single(parser, "int", &errmsg);
|
|
mu_assert_notnull(typ, "parsed type");
|
|
rz_analysis_var_global_set_type(glob, typ);
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
added = rz_analysis_var_global_add(analysis, glob);
|
|
mu_assert_true(added, "add global var");
|
|
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byaddr_at(analysis, 0x1337);
|
|
mu_assert_notnull(glob, "get readded global var");
|
|
|
|
flag_exists = rz_flag_get(core->flags, glob->name);
|
|
mu_assert_notnull(flag_exists, "flag not found");
|
|
|
|
deleted = rz_analysis_var_global_delete_byaddr_in(analysis, 0x133A); // test RBTree again
|
|
mu_assert_true(deleted, "delete global var");
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byaddr_in(analysis, 0x133A);
|
|
mu_assert_null(glob, "get deleted global var");
|
|
|
|
// re add
|
|
glob = rz_analysis_var_global_new("bar", 0x114514);
|
|
mu_assert_notnull(glob, "create a global variable");
|
|
mu_assert_streq(glob->name, "bar", "global var name");
|
|
mu_assert_eq(glob->addr, 0x114514, "global var address");
|
|
mu_assert_null(glob->analysis, "global var flags");
|
|
typ = rz_type_parse_string_single(parser, "int", &errmsg);
|
|
mu_assert_notnull(typ, "parsed type");
|
|
rz_analysis_var_global_set_type(glob, typ);
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
added = rz_analysis_var_global_add(analysis, glob);
|
|
mu_assert_true(added, "add global var");
|
|
flag = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_notnull(flag, "global var flag_item");
|
|
mu_assert_eq(flag->offset, glob->addr, "flag item addr");
|
|
mu_assert_streq(flag->name, "bar", "flag item name");
|
|
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byname(analysis, "bar");
|
|
mu_assert_notnull(glob, "get readded global var");
|
|
|
|
flag_exists = rz_flag_get(core->flags, glob->name);
|
|
mu_assert_notnull(flag_exists, "flag not found");
|
|
|
|
deleted = rz_analysis_var_global_delete_byname(analysis, "bar");
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byname(analysis, "bar");
|
|
mu_assert_null(glob, "get deleted global var");
|
|
|
|
// create global
|
|
typ = rz_type_parse_string_single(parser, "int", &errmsg);
|
|
mu_assert_notnull(typ, "parsed type");
|
|
mu_assert_true(rz_analysis_var_global_create(analysis, "crab", typ, 0x125418),
|
|
"create global var");
|
|
|
|
glob = rz_analysis_var_global_get_byname(analysis, "crab");
|
|
mu_assert_notnull(glob, "create a global variable");
|
|
mu_assert_streq(glob->name, "crab", "global var name");
|
|
mu_assert_eq(glob->addr, 0x125418, "global var address");
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
flag = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_notnull(flag, "global var flag_item");
|
|
mu_assert_eq(flag->offset, glob->addr, "flag item addr");
|
|
mu_assert_streq(flag->name, "crab", "flag item name");
|
|
|
|
rz_type_parser_free(parser);
|
|
rz_core_free(core);
|
|
mu_end;
|
|
}
|
|
|
|
bool test_flag_confusion_space_name() {
|
|
RzCore *core = rz_core_new();
|
|
RzAnalysis *analysis = core->analysis;
|
|
|
|
rz_flag_space_set(core->flags, "mire");
|
|
RzAnalysisVarGlobal *glob = rz_analysis_var_global_new("foo", 0x1337);
|
|
RzTypeParser *parser = rz_type_parser_new();
|
|
mu_assert_notnull(parser, "create type parser");
|
|
RzType *typ = rz_type_parse_string_single(parser, "int", NULL);
|
|
rz_analysis_var_global_set_type(glob, typ);
|
|
rz_analysis_var_global_add(analysis, glob);
|
|
RzFlagItem *fi = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_notnull(fi, "global var flag_item");
|
|
mu_assert_eq(fi->offset, glob->addr, "flag item addr");
|
|
mu_assert_streq(fi->name, "foo", "flag item name");
|
|
mu_assert_streq(fi->space->name, RZ_FLAGS_FS_GLOBALS, "flag space");
|
|
|
|
rz_flag_space_set(core->flags, "ulu-mulu");
|
|
RzFlagItem *fii = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_ptreq(fii, fi, "unaffected by space change");
|
|
|
|
rz_flag_rename(core->flags, fi, "bar");
|
|
fi = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_null(fi, "flag lost");
|
|
|
|
rz_type_parser_free(parser);
|
|
rz_core_free(core);
|
|
mu_end;
|
|
}
|
|
|
|
bool test_flag_confusion_addr() {
|
|
RzCore *core = rz_core_new();
|
|
RzAnalysis *analysis = core->analysis;
|
|
|
|
RzAnalysisVarGlobal *glob = rz_analysis_var_global_new("foo", 0x1337);
|
|
RzTypeParser *parser = rz_type_parser_new();
|
|
mu_assert_notnull(parser, "create type parser");
|
|
RzType *typ = rz_type_parse_string_single(parser, "int", NULL);
|
|
rz_analysis_var_global_set_type(glob, typ);
|
|
rz_analysis_var_global_add(analysis, glob);
|
|
RzFlagItem *fi = rz_analysis_var_global_get_flag_item(glob);
|
|
|
|
rz_flag_set(core->flags, fi->name, 0x31337, fi->size);
|
|
fi = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_null(fi, "flag lost");
|
|
|
|
rz_type_parser_free(parser);
|
|
rz_core_free(core);
|
|
mu_end;
|
|
}
|
|
|
|
bool test_flag_confusion_delete() {
|
|
RzCore *core = rz_core_new();
|
|
RzAnalysis *analysis = core->analysis;
|
|
RzTypeDB *typedb = rz_analysis_get_type_db(analysis);
|
|
rz_type_db_init(typedb, TEST_BUILD_TYPES_DIR, NULL, 0, NULL);
|
|
|
|
RzAnalysisVarGlobal *glob = rz_analysis_var_global_new("foo", 0x1337);
|
|
RzTypeParser *parser = rz_type_parser_new();
|
|
mu_assert_notnull(parser, "create type parser");
|
|
RzType *typ = rz_type_parse_string_single(parser, "int", NULL);
|
|
rz_analysis_var_global_set_type(glob, typ);
|
|
rz_analysis_var_global_add(analysis, glob);
|
|
RzFlagItem *fi = rz_analysis_var_global_get_flag_item(glob);
|
|
|
|
rz_flag_unset(core->flags, fi);
|
|
fi = rz_analysis_var_global_get_flag_item(glob);
|
|
mu_assert_null(fi, "flag lost");
|
|
|
|
rz_analysis_var_global_rename(analysis, "foo", "bar");
|
|
mu_assert_streq(glob->name, "bar", "rename without flag");
|
|
mu_assert_eq(glob->addr, 0x1337, "addr");
|
|
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byaddr_at(analysis, 0x1337);
|
|
mu_assert_notnull(glob, "get global var by addr");
|
|
mu_assert_streq(glob->name, "bar", "global var name");
|
|
mu_assert_eq(glob->addr, 0x1337, "global var address");
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byname(analysis, "bar");
|
|
mu_assert_notnull(glob, "get global var by addr");
|
|
mu_assert_streq(glob->name, "bar", "global var name");
|
|
mu_assert_eq(glob->addr, 0x1337, "global var address");
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
glob = NULL;
|
|
glob = rz_analysis_var_global_get_byaddr_in(analysis, 0x1339); // test RBTree
|
|
mu_assert_notnull(glob, "get global var by addr");
|
|
mu_assert_streq(glob->name, "bar", "global var name");
|
|
mu_assert_eq(glob->addr, 0x1337, "global var address");
|
|
mu_assert_streq(glob->type->identifier.name, "int", "global var type");
|
|
|
|
rz_type_parser_free(parser);
|
|
rz_core_free(core);
|
|
mu_end;
|
|
}
|
|
|
|
int all_tests() {
|
|
mu_run_test(test_rz_analysis_global_var);
|
|
mu_run_test(test_flag_confusion_space_name);
|
|
mu_run_test(test_flag_confusion_addr);
|
|
mu_run_test(test_flag_confusion_delete);
|
|
return tests_passed != tests_run;
|
|
}
|
|
|
|
mu_main(all_tests)
|