822 lines
22 KiB
C
822 lines
22 KiB
C
// SPDX-FileCopyrightText: 2009-2020 pancake <pancake@nopcode.org>
|
|
// SPDX-FileCopyrightText: 2009-2020 nibble <nibble.ds@gmail.com>
|
|
// SPDX-License-Identifier: LGPL-3.0-only
|
|
|
|
#include <rz_analysis.h>
|
|
#include <rz_util.h>
|
|
#include <rz_list.h>
|
|
#include <rz_util/rz_path.h>
|
|
#include <rz_arch.h>
|
|
#include <rz_lib.h>
|
|
|
|
/**
|
|
* \brief Returns the default size byte width of memory access operations.
|
|
* The size is just a best guess.
|
|
*
|
|
* \param analysis The current RzAnalysis in use.
|
|
*
|
|
* \return The default width of a memory access in bytes.
|
|
*/
|
|
RZ_API ut32 rz_analysis_guessed_mem_access_width(RZ_NONNULL const RzAnalysis *analysis) {
|
|
if (analysis->bits == 16 && RZ_STR_EQ(analysis->cur->arch, "arm")) {
|
|
// Thumb access is usually 4 bytes of memory by default.
|
|
return 4;
|
|
}
|
|
// Best guess for variable size.
|
|
return analysis->bits / 8;
|
|
}
|
|
|
|
RZ_API void rz_analysis_set_limits(RzAnalysis *analysis, ut64 from, ut64 to) {
|
|
free(analysis->limit);
|
|
analysis->limit = RZ_NEW0(RzAnalysisRange);
|
|
if (analysis->limit) {
|
|
analysis->limit->from = from;
|
|
analysis->limit->to = to;
|
|
}
|
|
}
|
|
|
|
RZ_API void rz_analysis_unset_limits(RzAnalysis *analysis) {
|
|
RZ_FREE(analysis->limit);
|
|
}
|
|
|
|
static void meta_unset_for(RzEvent *ev, int type, void *user, void *data) {
|
|
RzSpaces *s = (RzSpaces *)ev->user;
|
|
RzAnalysis *analysis = container_of(s, RzAnalysis, meta_spaces);
|
|
RzSpaceEvent *se = (RzSpaceEvent *)data;
|
|
rz_meta_space_unset_for(analysis, se->data.unset.space);
|
|
}
|
|
|
|
static void meta_count_for(RzEvent *ev, int type, void *user, void *data) {
|
|
RzSpaces *s = (RzSpaces *)ev->user;
|
|
RzAnalysis *analysis = container_of(s, RzAnalysis, meta_spaces);
|
|
RzSpaceEvent *se = (RzSpaceEvent *)data;
|
|
se->res = rz_meta_space_count_for(analysis, se->data.count.space);
|
|
}
|
|
|
|
void rz_analysis_hint_storage_init(RzAnalysis *a);
|
|
|
|
void rz_analysis_hint_storage_fini(RzAnalysis *a);
|
|
|
|
static void meta_item_free(void *item) {
|
|
if (!item) {
|
|
return;
|
|
}
|
|
RzAnalysisMetaItem *it = item;
|
|
free(it->str);
|
|
free(it);
|
|
}
|
|
|
|
static void global_kv_free(HtPPKv *kv) {
|
|
free(kv->key);
|
|
rz_analysis_var_global_free(kv->value);
|
|
}
|
|
|
|
RZ_API RzAnalysis *rz_analysis_new(void) {
|
|
RzAnalysis *analysis = RZ_NEW0(RzAnalysis);
|
|
if (!analysis) {
|
|
return NULL;
|
|
}
|
|
if (!rz_str_constpool_init(&analysis->constpool)) {
|
|
free(analysis);
|
|
return NULL;
|
|
}
|
|
analysis->esilinterstate = RZ_NEW0(RzAnalysisEsilInterState);
|
|
if (!analysis->esilinterstate) {
|
|
free(analysis);
|
|
return NULL;
|
|
}
|
|
analysis->bb_tree = NULL;
|
|
analysis->ht_addr_fun = ht_up_new0();
|
|
analysis->ht_name_fun = ht_pp_new0();
|
|
analysis->os = strdup(RZ_SYS_OS);
|
|
analysis->esil_goto_limit = RZ_ANALYSIS_ESIL_GOTO_LIMIT;
|
|
analysis->opt.nopskip = true; // skip nops in code analysis
|
|
analysis->opt.hpskip = false; // skip `mov reg,reg` and `lea reg,[reg]`
|
|
analysis->gp = 0LL;
|
|
analysis->sdb = sdb_new0();
|
|
analysis->cpp_abi = RZ_ANALYSIS_CPP_ABI_ITANIUM;
|
|
analysis->opt.depth = 32;
|
|
analysis->opt.noncode = false; // do not analyze data by default
|
|
rz_spaces_init(&analysis->meta_spaces, "CS");
|
|
rz_event_hook(analysis->meta_spaces.event, RZ_SPACE_EVENT_UNSET, meta_unset_for, NULL);
|
|
rz_event_hook(analysis->meta_spaces.event, RZ_SPACE_EVENT_COUNT, meta_count_for, NULL);
|
|
|
|
rz_analysis_hint_storage_init(analysis);
|
|
rz_interval_tree_init(&analysis->meta, meta_item_free);
|
|
analysis->typedb = rz_type_db_new();
|
|
analysis->sdb_fmts = sdb_ns(analysis->sdb, "spec", 1);
|
|
analysis->sdb_cc = sdb_ns(analysis->sdb, "cc", 1);
|
|
analysis->sdb_classes = sdb_ns(analysis->sdb, "classes", 1);
|
|
analysis->sdb_classes_attrs = sdb_ns(analysis->sdb_classes, "attrs", 1);
|
|
analysis->sdb_noret = sdb_ns(analysis->sdb, "noreturn", 1);
|
|
(void)rz_analysis_xrefs_init(analysis);
|
|
analysis->syscall = rz_syscall_new();
|
|
analysis->arch_target = rz_platform_target_new();
|
|
analysis->platform_target = rz_platform_target_index_new();
|
|
rz_io_bind_init(analysis->iob);
|
|
rz_flag_bind_init(analysis->flb);
|
|
analysis->reg = rz_reg_new();
|
|
analysis->last_disasm_reg = NULL;
|
|
analysis->lineswidth = 0;
|
|
analysis->fcns = rz_list_newf(rz_analysis_function_free);
|
|
analysis->leaddrs = NULL;
|
|
analysis->imports = rz_list_newf(free);
|
|
rz_analysis_set_bits(analysis, 32);
|
|
analysis->plugins = rz_list_new();
|
|
if (analysis->plugins) {
|
|
const size_t n_plugins = rz_arch_get_n_plugins();
|
|
for (size_t i = 0; i < n_plugins; i++) {
|
|
RzAnalysisPlugin *plugin = rz_arch_get_analysis_plugin(i);
|
|
if (!plugin) {
|
|
continue;
|
|
}
|
|
rz_analysis_plugin_add(analysis, plugin);
|
|
}
|
|
}
|
|
analysis->ht_global_var = ht_pp_new(NULL, global_kv_free, NULL);
|
|
analysis->global_var_tree = NULL;
|
|
analysis->il_vm = NULL;
|
|
analysis->hash = rz_hash_new();
|
|
analysis->debug_info = rz_analysis_debug_info_new();
|
|
analysis->cmpval = UT64_MAX;
|
|
analysis->lea_jmptbl_ip = UT64_MAX;
|
|
return analysis;
|
|
}
|
|
|
|
RZ_API void plugin_fini(RzAnalysis *analysis) {
|
|
RzAnalysisPlugin *p = analysis->cur;
|
|
if (p && p->fini && !p->fini(analysis->plugin_data)) {
|
|
RZ_LOG_ERROR("analysis plugin '%s' failed to terminate.\n", p->name);
|
|
}
|
|
analysis->plugin_data = NULL;
|
|
}
|
|
|
|
void __block_free_rb(RBNode *node, void *user);
|
|
|
|
RZ_API RzAnalysis *rz_analysis_free(RzAnalysis *a) {
|
|
if (!a) {
|
|
return NULL;
|
|
}
|
|
|
|
plugin_fini(a);
|
|
|
|
rz_hash_free(a->hash);
|
|
rz_analysis_il_vm_cleanup(a);
|
|
rz_list_free(a->fcns);
|
|
ht_up_free(a->ht_addr_fun);
|
|
ht_pp_free(a->ht_name_fun);
|
|
set_u_free(a->visited);
|
|
rz_analysis_hint_storage_fini(a);
|
|
rz_interval_tree_fini(&a->meta);
|
|
free(a->cpu);
|
|
free(a->os);
|
|
rz_rbtree_free(a->bb_tree, __block_free_rb, NULL);
|
|
rz_spaces_fini(&a->meta_spaces);
|
|
rz_syscall_free(a->syscall);
|
|
rz_platform_target_free(a->arch_target);
|
|
rz_platform_target_index_free(a->platform_target);
|
|
rz_reg_free(a->reg);
|
|
ht_up_free(a->ht_xrefs_from);
|
|
ht_up_free(a->ht_xrefs_to);
|
|
rz_list_free(a->leaddrs);
|
|
rz_type_db_free(a->typedb);
|
|
sdb_free(a->sdb);
|
|
if (a->esil) {
|
|
rz_analysis_esil_free(a->esil);
|
|
a->esil = NULL;
|
|
}
|
|
free(a->esilinterstate);
|
|
free(a->last_disasm_reg);
|
|
rz_list_free(a->imports);
|
|
rz_str_constpool_fini(&a->constpool);
|
|
ht_pp_free(a->ht_global_var);
|
|
rz_list_free(a->plugins);
|
|
rz_analysis_debug_info_free(a->debug_info);
|
|
free(a);
|
|
return NULL;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_plugin_add(RzAnalysis *analysis, RZ_NONNULL RzAnalysisPlugin *p) {
|
|
rz_return_val_if_fail(analysis && p, false);
|
|
RZ_PLUGIN_CHECK_AND_ADD(analysis->plugins, p, RzAnalysisPlugin);
|
|
return true;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_plugin_del(RzAnalysis *analysis, RZ_NONNULL RzAnalysisPlugin *p) {
|
|
rz_return_val_if_fail(analysis && p, false);
|
|
if (analysis->cur == p) {
|
|
plugin_fini(analysis);
|
|
analysis->cur = NULL;
|
|
}
|
|
return rz_list_delete_data(analysis->plugins, p);
|
|
}
|
|
|
|
RZ_API bool rz_analysis_use(RzAnalysis *analysis, const char *name) {
|
|
RzListIter *it;
|
|
RzAnalysisPlugin *h;
|
|
|
|
if (analysis) {
|
|
if (analysis->cur && !strcmp(analysis->cur->name, name)) {
|
|
return true;
|
|
}
|
|
rz_list_foreach (analysis->plugins, it, h) {
|
|
if (!h || !h->name || strcmp(h->name, name)) {
|
|
continue;
|
|
}
|
|
plugin_fini(analysis);
|
|
analysis->cur = h;
|
|
if (h->init && !h->init(&analysis->plugin_data)) {
|
|
RZ_LOG_ERROR("analysis plugin '%s' failed to initialize.\n", h->name);
|
|
return false;
|
|
}
|
|
rz_analysis_set_reg_profile(analysis);
|
|
if (analysis->il_vm) {
|
|
rz_analysis_il_vm_setup(analysis);
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
RZ_API char *rz_analysis_get_reg_profile(RzAnalysis *analysis) {
|
|
return (analysis && analysis->cur && analysis->cur->get_reg_profile)
|
|
? analysis->cur->get_reg_profile(analysis)
|
|
: NULL;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_set_reg_profile(RzAnalysis *analysis) {
|
|
bool ret = false;
|
|
char *p = rz_analysis_get_reg_profile(analysis);
|
|
if (p) {
|
|
rz_reg_set_profile_string(analysis->reg, p);
|
|
ret = true;
|
|
}
|
|
free(p);
|
|
return ret;
|
|
}
|
|
|
|
static bool analysis_set_os(RzAnalysis *analysis, const char *os) {
|
|
rz_return_val_if_fail(analysis, false);
|
|
if (!os || !*os) {
|
|
os = RZ_SYS_OS;
|
|
}
|
|
free(analysis->os);
|
|
analysis->os = strdup(os);
|
|
char *types_dir = rz_path_system(RZ_SDB_TYPES);
|
|
rz_type_db_set_os(analysis->typedb, os);
|
|
rz_type_db_reload(analysis->typedb, types_dir);
|
|
free(types_dir);
|
|
return true;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_set_triplet(RzAnalysis *analysis, const char *os, const char *arch, int bits) {
|
|
rz_return_val_if_fail(analysis, false);
|
|
if (!arch || !*arch) {
|
|
arch = analysis->cur ? analysis->cur->arch : RZ_SYS_ARCH;
|
|
}
|
|
if (bits < 1) {
|
|
bits = analysis->bits;
|
|
}
|
|
analysis_set_os(analysis, os);
|
|
rz_analysis_set_bits(analysis, bits);
|
|
return rz_analysis_use(analysis, arch);
|
|
}
|
|
|
|
RZ_API bool rz_analysis_set_os(RzAnalysis *analysis, const char *os) {
|
|
return rz_analysis_set_triplet(analysis, os, NULL, -1);
|
|
}
|
|
|
|
static bool is_arm_thumb_hack(RzAnalysis *analysis, int bits) {
|
|
if (!analysis || !analysis->cpu) {
|
|
return false;
|
|
}
|
|
if ((analysis->bits != bits) && !strcmp(analysis->cpu, "arm")) {
|
|
return (analysis->bits == 16 && bits == 32) || (analysis->bits == 32 && bits == 16);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_set_bits(RzAnalysis *analysis, int bits) {
|
|
switch (bits) {
|
|
case 8:
|
|
case 16:
|
|
case 27:
|
|
case 32:
|
|
case 64:
|
|
if (analysis->bits != bits) {
|
|
bool is_hack = is_arm_thumb_hack(analysis, bits);
|
|
analysis->bits = bits;
|
|
int v = rz_analysis_archinfo(analysis, RZ_ANALYSIS_ARCHINFO_TEXT_ALIGN);
|
|
analysis->pcalign = RZ_MAX(0, v);
|
|
rz_type_db_set_bits(analysis->typedb, bits);
|
|
rz_type_db_set_address_bits(analysis->typedb, rz_analysis_get_address_bits(analysis));
|
|
if (!is_hack) {
|
|
char *types_dir = rz_path_system(RZ_SDB_TYPES);
|
|
rz_type_db_reload(analysis->typedb, types_dir);
|
|
free(types_dir);
|
|
}
|
|
rz_analysis_set_reg_profile(analysis);
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* \brief The actual size of an address in bits.
|
|
*
|
|
* This may differ from analysis.bits in some cases such as arm thumb
|
|
* being identified as bits=16, but still using 32-bit addresses,
|
|
* or "8-bit" architectures like 6502 which still use 16-bit addresses.
|
|
*/
|
|
RZ_API int rz_analysis_get_address_bits(RzAnalysis *analysis) {
|
|
if (!analysis->cur || !analysis->cur->address_bits) {
|
|
return analysis->bits;
|
|
}
|
|
int r = analysis->cur->address_bits(analysis, analysis->bits);
|
|
return r > 0 ? r : analysis->bits;
|
|
}
|
|
|
|
RZ_API void rz_analysis_set_cpu(RzAnalysis *analysis, const char *cpu) {
|
|
if (RZ_STR_EQ(cpu, analysis->cpu)) {
|
|
return;
|
|
}
|
|
free(analysis->cpu);
|
|
analysis->cpu = cpu ? strdup(cpu) : NULL;
|
|
int v = rz_analysis_archinfo(analysis, RZ_ANALYSIS_ARCHINFO_TEXT_ALIGN);
|
|
if (v != -1) {
|
|
analysis->pcalign = v;
|
|
}
|
|
rz_analysis_set_reg_profile(analysis);
|
|
if (RZ_STR_EQ(cpu, analysis->typedb->target->cpu)) {
|
|
return;
|
|
}
|
|
|
|
rz_type_db_set_cpu(analysis->typedb, cpu);
|
|
char *types_dir = rz_path_system(RZ_SDB_TYPES);
|
|
rz_type_db_reload(analysis->typedb, types_dir);
|
|
free(types_dir);
|
|
}
|
|
|
|
RZ_API int rz_analysis_set_big_endian(RzAnalysis *analysis, int bigend) {
|
|
analysis->big_endian = bigend;
|
|
if (analysis->reg) {
|
|
analysis->reg->big_endian = bigend;
|
|
}
|
|
rz_type_db_set_endian(analysis->typedb, bigend);
|
|
return true;
|
|
}
|
|
|
|
RZ_API ut8 *rz_analysis_mask(RzAnalysis *analysis, ut32 size, const ut8 *data, ut64 at) {
|
|
RzAnalysisOp *op = NULL;
|
|
ut8 *ret = NULL;
|
|
int oplen = 0;
|
|
ut32 idx = 0;
|
|
|
|
if (!data) {
|
|
return NULL;
|
|
}
|
|
|
|
if (analysis->cur && analysis->cur->analysis_mask) {
|
|
return analysis->cur->analysis_mask(analysis, size, data, at);
|
|
}
|
|
|
|
if (!(op = rz_analysis_op_new())) {
|
|
return NULL;
|
|
}
|
|
|
|
if (!(ret = malloc(size))) {
|
|
rz_analysis_op_free(op);
|
|
return NULL;
|
|
}
|
|
|
|
memset(ret, 0xff, size);
|
|
|
|
while (idx < size) {
|
|
if ((oplen = rz_analysis_op(analysis, op, at, data + idx, size - idx, RZ_ANALYSIS_OP_MASK_BASIC)) < 1) {
|
|
break;
|
|
}
|
|
if ((op->ptr != UT64_MAX || op->jump != UT64_MAX) && op->nopcode != 0) {
|
|
memset(ret + idx + op->nopcode, 0, oplen - op->nopcode);
|
|
}
|
|
idx += oplen;
|
|
at += oplen;
|
|
rz_analysis_op_fini(op);
|
|
rz_analysis_op_init(op);
|
|
}
|
|
|
|
rz_analysis_op_free(op);
|
|
|
|
return ret;
|
|
}
|
|
|
|
RZ_API void rz_analysis_trace_bb(RzAnalysis *analysis, ut64 addr) {
|
|
RzAnalysisBlock *bbi;
|
|
RzAnalysisFunction *fcni;
|
|
RzListIter *iter2;
|
|
fcni = rz_analysis_get_fcn_in(analysis, addr, 0);
|
|
if (fcni) {
|
|
rz_list_foreach (fcni->bbs, iter2, bbi) {
|
|
if (addr >= bbi->addr && addr < (bbi->addr + bbi->size)) {
|
|
bbi->traced = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
RZ_API RzList /*<RzAnalysisFunction *>*/ *rz_analysis_get_fcns(RzAnalysis *analysis) {
|
|
// avoid received to free this thing
|
|
analysis->fcns->free = NULL;
|
|
return analysis->fcns;
|
|
}
|
|
|
|
RZ_API RzAnalysisOp *rz_analysis_op_hexstr(RzAnalysis *analysis, ut64 addr, const char *str) {
|
|
RzAnalysisOp *op = RZ_NEW0(RzAnalysisOp);
|
|
if (!op) {
|
|
return NULL;
|
|
}
|
|
ut8 *buf = calloc(1, strlen(str) + 1);
|
|
if (!buf) {
|
|
free(op);
|
|
return NULL;
|
|
}
|
|
int len = rz_hex_str2bin(str, buf);
|
|
rz_analysis_op(analysis, op, addr, buf, len, RZ_ANALYSIS_OP_MASK_BASIC);
|
|
free(buf);
|
|
return op;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_op_is_eob(RzAnalysisOp *op) {
|
|
if (op->eob) {
|
|
return true;
|
|
}
|
|
switch (op->type) {
|
|
case RZ_ANALYSIS_OP_TYPE_JMP:
|
|
case RZ_ANALYSIS_OP_TYPE_UJMP:
|
|
case RZ_ANALYSIS_OP_TYPE_RJMP:
|
|
case RZ_ANALYSIS_OP_TYPE_IJMP:
|
|
case RZ_ANALYSIS_OP_TYPE_IRJMP:
|
|
case RZ_ANALYSIS_OP_TYPE_CJMP:
|
|
case RZ_ANALYSIS_OP_TYPE_RET:
|
|
case RZ_ANALYSIS_OP_TYPE_TRAP:
|
|
return true;
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
RZ_API void rz_analysis_purge(RzAnalysis *analysis) {
|
|
rz_analysis_hint_clear(analysis);
|
|
rz_interval_tree_fini(&analysis->meta);
|
|
rz_interval_tree_init(&analysis->meta, meta_item_free);
|
|
rz_type_db_purge(analysis->typedb);
|
|
sdb_reset(analysis->sdb_classes);
|
|
sdb_reset(analysis->sdb_classes_attrs);
|
|
sdb_reset(analysis->sdb_cc);
|
|
sdb_reset(analysis->sdb_noret);
|
|
rz_list_free(analysis->fcns);
|
|
analysis->fcns = rz_list_newf(rz_analysis_function_free);
|
|
rz_analysis_purge_imports(analysis);
|
|
}
|
|
|
|
/**
|
|
* \brief Returns the queried information regarding the current architecture
|
|
*
|
|
* \param analysis The RzAnalysis object to use
|
|
* \param[in] query The architecture detail to query
|
|
*
|
|
* \return Negative when fails.
|
|
*/
|
|
RZ_API int rz_analysis_archinfo(RzAnalysis *analysis, RzAnalysisInfoType query) {
|
|
rz_return_val_if_fail(analysis && query < RZ_ANALYSIS_ARCHINFO_ENUM_SIZE, -1);
|
|
if (!analysis->cur || !analysis->cur->archinfo) {
|
|
switch (query) {
|
|
case RZ_ANALYSIS_ARCHINFO_MIN_OP_SIZE:
|
|
return 1;
|
|
case RZ_ANALYSIS_ARCHINFO_CAN_USE_POINTERS:
|
|
return true;
|
|
default:
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
int value = analysis->cur->archinfo(analysis, query);
|
|
switch (query) {
|
|
case RZ_ANALYSIS_ARCHINFO_MIN_OP_SIZE:
|
|
// Always consume at least 1 byte
|
|
return value > 0 ? value : 1;
|
|
case RZ_ANALYSIS_ARCHINFO_CAN_USE_POINTERS:
|
|
// When negative (i.e. error) we assume the architecture does use them.
|
|
return value < 0 ? true : value;
|
|
default:
|
|
return value;
|
|
}
|
|
return value;
|
|
}
|
|
|
|
static bool sdb_noret_addr_set(Sdb *db, ut64 addr, bool v, ut32 cas) {
|
|
char key[128];
|
|
rz_strf(key, "addr.%" PFMT64x ".noreturn", addr);
|
|
return sdb_bool_set(db, key, v, cas);
|
|
}
|
|
|
|
static bool sdb_noret_addr_get(Sdb *db, ut64 addr, ut32 *cas) {
|
|
char key[128];
|
|
rz_strf(key, "addr.%" PFMT64x ".noreturn", addr);
|
|
return sdb_bool_get(db, key, cas);
|
|
}
|
|
|
|
static int sdb_noret_addr_unset(Sdb *db, ut64 addr, ut32 cas) {
|
|
char key[128];
|
|
rz_strf(key, "addr.%" PFMT64x ".noreturn", addr);
|
|
return sdb_unset(db, key, cas);
|
|
}
|
|
|
|
static bool sdb_noret_func_set(Sdb *db, const char *name, bool v, ut32 cas) {
|
|
char *key = rz_str_newf("func.%s.noreturn", name);
|
|
if (!key) {
|
|
return false;
|
|
}
|
|
bool res = sdb_bool_set(db, key, v, cas);
|
|
free(key);
|
|
return res;
|
|
}
|
|
|
|
static bool sdb_noret_func_get(Sdb *db, const char *name, ut32 *cas) {
|
|
char *key = rz_str_newf("func.%s.noreturn", name);
|
|
if (!key) {
|
|
return false;
|
|
}
|
|
bool res = sdb_bool_get(db, key, cas);
|
|
free(key);
|
|
return res;
|
|
}
|
|
|
|
static int sdb_noret_func_unset(Sdb *db, const char *name, ut32 cas) {
|
|
char *key = rz_str_newf("func.%s.noreturn", name);
|
|
if (!key) {
|
|
return false;
|
|
}
|
|
int res = sdb_unset(db, key, cas);
|
|
free(key);
|
|
return res;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_noreturn_add(RzAnalysis *analysis, const char *name, ut64 addr) {
|
|
const char *tmp_name = NULL;
|
|
Sdb *NDB = analysis->sdb_noret;
|
|
char *fnl_name = NULL;
|
|
if (addr != UT64_MAX) {
|
|
if (sdb_noret_addr_set(NDB, addr, true, 0)) {
|
|
RzAnalysisFunction *fcn = rz_analysis_get_function_at(analysis, addr);
|
|
if (fcn) {
|
|
fcn->is_noreturn = true;
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
if (name && *name) {
|
|
tmp_name = name;
|
|
} else {
|
|
RzAnalysisFunction *fcn = rz_analysis_get_fcn_in(analysis, addr, -1);
|
|
RzFlagItem *fi = analysis->flb.get_at(analysis->flb.f, addr, false);
|
|
if (!fcn && !fi) {
|
|
RZ_LOG_ERROR("Cannot find function and flag at address 0x%" PFMT64x "\n", addr);
|
|
return false;
|
|
}
|
|
tmp_name = fcn ? fcn->name : fi->name;
|
|
if (fcn) {
|
|
fcn->is_noreturn = true;
|
|
}
|
|
}
|
|
if (rz_type_func_exist(analysis->typedb, tmp_name)) {
|
|
fnl_name = strdup(tmp_name);
|
|
} else if (!(fnl_name = rz_analysis_function_name_guess(analysis->typedb, (char *)tmp_name))) {
|
|
if (addr == UT64_MAX) {
|
|
if (name) {
|
|
sdb_noret_func_set(NDB, name, true, 0);
|
|
} else {
|
|
RZ_LOG_ERROR("Cannot find prototype for: %s\n", tmp_name);
|
|
}
|
|
} else {
|
|
RZ_LOG_ERROR("Cannot find prototype for: %s\n", tmp_name);
|
|
}
|
|
// return false;
|
|
}
|
|
if (fnl_name) {
|
|
sdb_noret_func_set(NDB, fnl_name, true, 0);
|
|
free(fnl_name);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_noreturn_drop(RzAnalysis *analysis, const char *expr) {
|
|
Sdb *NDB = analysis->sdb_noret;
|
|
expr = rz_str_trim_head_ro(expr);
|
|
const char *fcnname = NULL;
|
|
if (!strncmp(expr, "0x", 2)) {
|
|
ut64 n = rz_num_math(NULL, expr);
|
|
sdb_noret_addr_unset(NDB, n, 0);
|
|
RzAnalysisFunction *fcn = rz_analysis_get_fcn_in(analysis, n, -1);
|
|
if (!fcn) {
|
|
// eprintf ("can't find function at 0x%"PFMT64x"\n", n);
|
|
return false;
|
|
}
|
|
fcnname = fcn->name;
|
|
} else {
|
|
fcnname = expr;
|
|
}
|
|
sdb_noret_func_unset(NDB, fcnname, 0);
|
|
return false;
|
|
}
|
|
|
|
static bool rz_analysis_is_noreturn(RzAnalysis *analysis, const char *name) {
|
|
return rz_type_func_is_noreturn(analysis->typedb, name) ||
|
|
sdb_noret_func_get(analysis->sdb_noret, name, NULL);
|
|
}
|
|
|
|
static bool rz_analysis_noreturn_at_name(RzAnalysis *analysis, const char *name) {
|
|
if (rz_analysis_is_noreturn(analysis, name)) {
|
|
return true;
|
|
}
|
|
char *tmp = rz_analysis_function_name_guess(analysis->typedb, (char *)name);
|
|
if (tmp) {
|
|
if (rz_analysis_is_noreturn(analysis, tmp)) {
|
|
free(tmp);
|
|
return true;
|
|
}
|
|
free(tmp);
|
|
}
|
|
if (rz_str_startswith(name, "reloc.")) {
|
|
return rz_analysis_noreturn_at_name(analysis, name + 6);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_noreturn_at_addr(RzAnalysis *analysis, ut64 addr) {
|
|
return sdb_noret_addr_get(analysis->sdb_noret, addr, NULL);
|
|
}
|
|
|
|
static bool noreturn_recurse(RzAnalysis *analysis, ut64 addr) {
|
|
RzAnalysisOp op = { 0 };
|
|
ut8 bbuf[0x10] = { 0 };
|
|
ut64 recurse_addr = UT64_MAX;
|
|
if (!analysis->iob.read_at(analysis->iob.io, addr, bbuf, sizeof(bbuf))) {
|
|
RZ_LOG_ERROR("Cannot read buffer at 0x%" PFMT64x "\n", addr);
|
|
return false;
|
|
}
|
|
if (rz_analysis_op(analysis, &op, addr, bbuf, sizeof(bbuf), RZ_ANALYSIS_OP_MASK_BASIC | RZ_ANALYSIS_OP_MASK_VAL) < 1) {
|
|
return false;
|
|
}
|
|
switch (op.type & RZ_ANALYSIS_OP_TYPE_MASK) {
|
|
case RZ_ANALYSIS_OP_TYPE_JMP:
|
|
if (op.jump == UT64_MAX) {
|
|
recurse_addr = op.ptr;
|
|
} else {
|
|
recurse_addr = op.jump;
|
|
}
|
|
break;
|
|
case RZ_ANALYSIS_OP_TYPE_UCALL:
|
|
case RZ_ANALYSIS_OP_TYPE_RCALL:
|
|
case RZ_ANALYSIS_OP_TYPE_ICALL:
|
|
case RZ_ANALYSIS_OP_TYPE_IRCALL:
|
|
recurse_addr = op.ptr;
|
|
break;
|
|
case RZ_ANALYSIS_OP_TYPE_CCALL:
|
|
case RZ_ANALYSIS_OP_TYPE_CALL:
|
|
recurse_addr = op.jump;
|
|
break;
|
|
}
|
|
if (recurse_addr == UT64_MAX || recurse_addr == addr) {
|
|
return false;
|
|
}
|
|
return rz_analysis_noreturn_at(analysis, recurse_addr);
|
|
}
|
|
|
|
RZ_API bool rz_analysis_noreturn_at(RzAnalysis *analysis, ut64 addr) {
|
|
if (!addr || addr == UT64_MAX) {
|
|
return false;
|
|
}
|
|
if (rz_analysis_noreturn_at_addr(analysis, addr)) {
|
|
return true;
|
|
}
|
|
/* XXX this is very slow */
|
|
RzAnalysisFunction *f = rz_analysis_get_function_at(analysis, addr);
|
|
if (f) {
|
|
if (rz_analysis_noreturn_at_name(analysis, f->name)) {
|
|
return true;
|
|
}
|
|
}
|
|
RzFlagItem *fi = analysis->flag_get(analysis->flb.f, addr);
|
|
if (fi) {
|
|
if (rz_analysis_noreturn_at_name(analysis, fi->realname ? fi->realname : fi->name)) {
|
|
return true;
|
|
}
|
|
}
|
|
if (analysis->recursive_noreturn) {
|
|
return noreturn_recurse(analysis, addr);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
RZ_API RzList /*<char *>*/ *rz_analysis_noreturn_functions(RzAnalysis *analysis) {
|
|
rz_return_val_if_fail(analysis, NULL);
|
|
// At first we read all noreturn functions from the Types DB
|
|
RzList *noretl = rz_type_noreturn_function_names(analysis->typedb);
|
|
// Then we propagate all noreturn functions that were inferred by
|
|
// the analysis process
|
|
SdbKv *kv;
|
|
SdbListIter *iter;
|
|
SdbList *l = sdb_foreach_list(analysis->sdb_noret, true);
|
|
ls_foreach (l, iter, kv) {
|
|
const char *k = sdbkv_key(kv);
|
|
if (!strncmp(k, "func.", 5) && strstr(k, ".noreturn")) {
|
|
char *s = strdup(k + 5);
|
|
char *d = strchr(s, '.');
|
|
if (d) {
|
|
*d = 0;
|
|
}
|
|
rz_list_append(noretl, strdup(s));
|
|
free(s);
|
|
}
|
|
if (!strncmp(k, "addr.", 5)) {
|
|
char *off;
|
|
if (!(off = strdup(k + 5))) {
|
|
break;
|
|
}
|
|
char *ptr = strstr(off, ".noreturn");
|
|
if (ptr) {
|
|
*ptr = 0;
|
|
char *addr = rz_str_newf("0x%s", off);
|
|
rz_list_append(noretl, addr);
|
|
}
|
|
free(off);
|
|
}
|
|
}
|
|
ls_free(l);
|
|
return noretl;
|
|
}
|
|
|
|
RZ_API void rz_analysis_bind(RzAnalysis *analysis, RzAnalysisBind *b) {
|
|
if (b) {
|
|
b->analysis = analysis;
|
|
b->get_fcn_in = rz_analysis_get_fcn_in;
|
|
b->get_hint = rz_analysis_hint_get;
|
|
}
|
|
}
|
|
|
|
RZ_API RzList /*<RzSearchKeyword *>*/ *rz_analysis_preludes(RzAnalysis *analysis) {
|
|
if (analysis->cur && analysis->cur->preludes) {
|
|
return analysis->cur->preludes(analysis);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
RZ_API bool rz_analysis_is_prelude(RzAnalysis *analysis, const ut8 *data, int len) {
|
|
RzList *l = rz_analysis_preludes(analysis);
|
|
if (l) {
|
|
RzSearchKeyword *kw;
|
|
RzListIter *iter;
|
|
rz_list_foreach (l, iter, kw) {
|
|
int ks = kw->keyword_length;
|
|
if (len >= ks && !memcmp(data, kw->bin_keyword, ks)) {
|
|
rz_list_free(l);
|
|
return true;
|
|
}
|
|
}
|
|
rz_list_free(l);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
RZ_API void rz_analysis_add_import(RzAnalysis *analysis, const char *imp) {
|
|
RzListIter *it;
|
|
const char *eimp;
|
|
rz_list_foreach (analysis->imports, it, eimp) {
|
|
if (!strcmp(eimp, imp)) {
|
|
return;
|
|
}
|
|
}
|
|
char *cimp = strdup(imp);
|
|
if (!cimp) {
|
|
return;
|
|
}
|
|
rz_list_push(analysis->imports, cimp);
|
|
}
|
|
|
|
RZ_API void rz_analysis_remove_import(RzAnalysis *analysis, const char *imp) {
|
|
RzListIter *it;
|
|
const char *eimp;
|
|
rz_list_foreach (analysis->imports, it, eimp) {
|
|
if (!strcmp(eimp, imp)) {
|
|
rz_list_delete(analysis->imports, it);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
RZ_API void rz_analysis_purge_imports(RzAnalysis *analysis) {
|
|
rz_list_purge(analysis->imports);
|
|
}
|