mirror of
https://github.com/radareorg/radare2.git
synced 2025-03-04 20:39:46 +00:00
fdbb87d70e
* New commands: aCb, aCm, aCmn, aCll, aClj, aCv, aCm, aC-, aCm- * Fix Segfaults and Leaks in RTTI classes recovery * Accept .?AU as prefix for msvc class name demangling * Classes recovery refactoring * Skip own type descriptor in msvc classes recovery * Cleanup msvc rtti address calculation * Recover vtable, classes, base class, virtual methods from MSVC RTTI * Add flags for vtables and methods * Add r_str_sanitize_sdb_key()
146 lines
3.1 KiB
C
146 lines
3.1 KiB
C
/* radare - LGPL - Copyright 2009-2018 - pancake, maijin, thestr4ng3r */
|
|
|
|
#include "r_anal.h"
|
|
|
|
R_API char *r_anal_rtti_demangle_class_name(RAnal *anal, const char *name) {
|
|
RVTableContext context;
|
|
r_anal_vtable_begin (anal, &context);
|
|
if (context.abi == R_ANAL_CPP_ABI_MSVC) {
|
|
return r_anal_rtti_msvc_demangle_class_name (name);
|
|
}
|
|
// TODO: implement class name demangling for itanium
|
|
return NULL;
|
|
}
|
|
|
|
R_API void r_anal_rtti_print_at_vtable(RAnal *anal, ut64 addr, int mode) {
|
|
bool use_json = mode == 'j';
|
|
if (use_json) {
|
|
r_cons_print ("[");
|
|
}
|
|
|
|
RVTableContext context;
|
|
r_anal_vtable_begin (anal, &context);
|
|
if (context.abi == R_ANAL_CPP_ABI_MSVC) {
|
|
r_anal_rtti_msvc_print_at_vtable (&context, addr, mode, false);
|
|
} else {
|
|
r_anal_rtti_itanium_print_at_vtable (&context, addr, mode);
|
|
}
|
|
|
|
if (use_json) {
|
|
r_cons_print ("]\n");
|
|
}
|
|
}
|
|
|
|
static void rtti_msvc_print_all(RVTableContext *context, int mode) {
|
|
bool use_json = mode == 'j';
|
|
if (use_json) {
|
|
r_cons_print ("[");
|
|
}
|
|
|
|
r_cons_break_push (NULL, NULL);
|
|
RList *vtables = r_anal_vtable_search (context);
|
|
RListIter *vtableIter;
|
|
RVTableInfo *table;
|
|
|
|
if (vtables) {
|
|
bool comma = false;
|
|
bool success = false;
|
|
r_list_foreach (vtables, vtableIter, table) {
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
|
|
if (use_json && success) {
|
|
r_cons_print (",");
|
|
comma = true;
|
|
}
|
|
success = r_anal_rtti_msvc_print_at_vtable (context, table->saddr, mode, true);
|
|
if (success) {
|
|
comma = false;
|
|
if (!use_json) {
|
|
r_cons_print ("\n");
|
|
}
|
|
}
|
|
}
|
|
if (use_json && !success && comma) {
|
|
// drop last comma if necessary
|
|
r_cons_drop (1);
|
|
}
|
|
}
|
|
r_list_free (vtables);
|
|
|
|
if (use_json) {
|
|
r_cons_print ("]\n");
|
|
}
|
|
|
|
r_cons_break_pop ();
|
|
}
|
|
|
|
static void rtti_itanium_print_all(RVTableContext *context, int mode) {
|
|
bool use_json = mode == 'j';
|
|
bool json_first = true;
|
|
if (use_json) {
|
|
r_cons_print ("[");
|
|
}
|
|
|
|
r_cons_break_push (NULL, NULL);
|
|
RList *vtables = r_anal_vtable_search (context);
|
|
RListIter *vtableIter;
|
|
RVTableInfo *table;
|
|
|
|
if (vtables) {
|
|
r_list_foreach (vtables, vtableIter, table) {
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
|
|
if (use_json) {
|
|
if (json_first) {
|
|
json_first = false;
|
|
} else {
|
|
r_cons_print (",");
|
|
}
|
|
}
|
|
r_anal_rtti_itanium_print_at_vtable (context, table->saddr, mode);
|
|
if (!use_json) {
|
|
r_cons_print ("\n");
|
|
}
|
|
}
|
|
}
|
|
r_list_free (vtables);
|
|
|
|
if (use_json) {
|
|
r_cons_print ("]\n");
|
|
}
|
|
|
|
r_cons_break_pop ();
|
|
}
|
|
|
|
R_API void r_anal_rtti_print_all(RAnal *anal, int mode) {
|
|
RVTableContext context;
|
|
r_anal_vtable_begin (anal, &context);
|
|
if (context.abi == R_ANAL_CPP_ABI_MSVC) {
|
|
rtti_msvc_print_all (&context, mode);
|
|
} else {
|
|
rtti_itanium_print_all (&context, mode);
|
|
}
|
|
}
|
|
|
|
R_API void r_anal_rtti_recover_all(RAnal *anal) {
|
|
RVTableContext context;
|
|
r_anal_vtable_begin (anal, &context);
|
|
|
|
r_cons_break_push (NULL, NULL);
|
|
RList *vtables = r_anal_vtable_search (&context);
|
|
if (vtables) {
|
|
if (context.abi == R_ANAL_CPP_ABI_MSVC) {
|
|
r_anal_rtti_msvc_recover_all (&context, vtables);
|
|
} else {
|
|
// TODO: recovery for itanium
|
|
}
|
|
}
|
|
r_list_free (vtables);
|
|
r_cons_break_pop ();
|
|
}
|
|
|