mirror of
https://github.com/radareorg/radare2.git
synced 2024-12-25 23:05:55 +00:00
209 lines
6.4 KiB
C
209 lines
6.4 KiB
C
/* radare - LGPL - Copyright 2015-2016 - pancake */
|
|
|
|
#include <r_core.h>
|
|
|
|
|
|
R_API int r_core_pseudo_code(RCore *core, const char *input) {
|
|
Sdb *db;
|
|
ut64 queuegoto = 0LL;
|
|
const char *blocktype = "else";
|
|
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, core->offset, R_ANAL_FCN_TYPE_NULL);
|
|
RConfigHold *hc = r_config_hold_new (core->config);
|
|
if (!hc) {
|
|
return false;
|
|
}
|
|
r_config_save_num (hc, "asm.pseudo", "asm.decode", "asm.lines", "asm.bytes", NULL);
|
|
r_config_save_num (hc, "asm.offset", "asm.flags", "asm.fcnlines", "asm.comments", NULL);
|
|
r_config_save_num (hc, "asm.functions", "asm.section", "asm.cmtcol", "asm.filter", NULL);
|
|
if (!fcn) {
|
|
eprintf ("Cannot find function in 0x%08"PFMT64x"\n", core->offset);
|
|
r_config_hold_free (hc);
|
|
return false;
|
|
}
|
|
r_config_set_i (core->config, "asm.pseudo", 1);
|
|
r_config_set_i (core->config, "asm.decode", 0);
|
|
r_config_set_i (core->config, "asm.filter", 1);
|
|
r_config_set_i (core->config, "asm.lines", 0);
|
|
r_config_set_i (core->config, "asm.bytes", 0);
|
|
r_config_set_i (core->config, "asm.offset", 0);
|
|
r_config_set_i (core->config, "asm.flags", 0);
|
|
r_config_set_i (core->config, "asm.fcnlines", 0);
|
|
r_config_set_i (core->config, "asm.comments", 1);
|
|
r_config_set_i (core->config, "asm.functions", 0);
|
|
r_config_set_i (core->config, "asm.tabs", 0);
|
|
r_config_set_i (core->config, "asm.section", 0);
|
|
r_config_set_i (core->config, "asm.cmtcol", 30);
|
|
|
|
db = sdb_new0 ();
|
|
|
|
/* */
|
|
// walk all basic blocks
|
|
// define depth level for each block
|
|
// use it for indentation
|
|
// asm.pseudo=true
|
|
// asm.decode=true
|
|
RAnalBlock *bb = r_list_first (fcn->bbs);
|
|
char indentstr[1024];
|
|
int n_bb = r_list_length (fcn->bbs);
|
|
r_cons_printf ("function %s () {", fcn->name);
|
|
int indent = 1;
|
|
int nindent = 1;
|
|
|
|
do {
|
|
#define I_TAB 4
|
|
#define K_MARK(x) sdb_fmt(0,"mark.%"PFMT64x,x)
|
|
#define K_ELSE(x) sdb_fmt(0,"else.%"PFMT64x,x)
|
|
#define K_INDENT(x) sdb_fmt(0,"loc.%"PFMT64x,x)
|
|
#define SET_INDENT(x) { memset (indentstr, ' ', x*I_TAB); indentstr [(x*I_TAB)-2] = 0; }
|
|
if (!bb) break;
|
|
r_cons_push ();
|
|
char *code = r_core_cmd_str (core, sdb_fmt(0,
|
|
"pDI %d @ 0x%08"PFMT64x"\n", bb->size, bb->addr));
|
|
r_cons_pop ();
|
|
memset (indentstr, ' ', indent*I_TAB);
|
|
indentstr [(indent*I_TAB)-2] = 0;
|
|
code = r_str_prefix_all (code, indentstr);
|
|
code[strlen(code)-1] = 0; // chop last newline
|
|
//r_cons_printf ("\n%s loc_0x%llx:\n", indentstr, bb->addr);
|
|
//if (nindent != indent) {
|
|
// r_cons_printf ("\n%s loc_0x%llx:\n", indentstr, bb->addr);
|
|
//}
|
|
if (!sdb_const_get (db, K_MARK(bb->addr), 0)) {
|
|
bool mustprint = !queuegoto || queuegoto != bb->addr;
|
|
if (mustprint) {
|
|
if (queuegoto) {
|
|
r_cons_printf ("\n%s goto loc_0x%llx", indentstr, queuegoto);
|
|
queuegoto = 0LL;
|
|
}
|
|
r_cons_printf ("\n%s loc_0x%llx:\n", indentstr, bb->addr);
|
|
indentstr[(indent * I_TAB) - 2] = 0;
|
|
r_cons_printf ("\n%s", code);
|
|
free (code);
|
|
sdb_num_set (db, K_MARK(bb->addr), 1, 0);
|
|
}
|
|
}
|
|
if (sdb_const_get (db, K_INDENT (bb->addr), 0)) {
|
|
// already analyzed, go pop and continue
|
|
// XXX check if cant pop
|
|
//eprintf ("%s// 0x%08llx already analyzed\n", indentstr, bb->addr);
|
|
ut64 addr = sdb_array_pop_num (db, "indent", NULL);
|
|
if (addr==UT64_MAX) {
|
|
int i;
|
|
nindent = 1;
|
|
for (i=indent; i!=nindent; i--) {
|
|
SET_INDENT (i);
|
|
r_cons_printf ("\n%s}", indentstr);
|
|
}
|
|
r_cons_printf ("\n%sreturn;\n", indentstr);
|
|
break;
|
|
}
|
|
if (sdb_num_get (db, K_ELSE(bb->addr), 0)) {
|
|
if (!strcmp (blocktype, "else")) {
|
|
r_cons_printf ("\n%s } %s {", indentstr, blocktype);
|
|
} else {
|
|
r_cons_printf ("\n%s } %s (?);", indentstr, blocktype);
|
|
}
|
|
} else {
|
|
r_cons_printf ("\n%s}", indentstr);
|
|
}
|
|
if (addr != bb->addr) {
|
|
queuegoto = addr;
|
|
//r_cons_printf ("\n%s goto loc_0x%llx", indentstr, addr);
|
|
}
|
|
bb = r_anal_bb_from_offset (core->anal, addr);
|
|
if (!bb) {
|
|
eprintf ("failed block\n");
|
|
break;
|
|
}
|
|
//eprintf ("next is %llx\n", addr);
|
|
nindent = sdb_num_get (db, K_INDENT(addr), NULL);
|
|
if (indent > nindent && !strcmp (blocktype, "else")) {
|
|
int i;
|
|
for (i=indent; i!=nindent; i--) {
|
|
SET_INDENT (i);
|
|
r_cons_printf ("\n%s }", indentstr);
|
|
}
|
|
}
|
|
indent = nindent;
|
|
} else {
|
|
sdb_set (db, K_INDENT(bb->addr), "passed", 0);
|
|
if (bb->jump != UT64_MAX) {
|
|
int swap = 1;
|
|
// TODO: determine which branch take first
|
|
ut64 jump = swap? bb->jump: bb->fail;
|
|
ut64 fail = swap? bb->fail: bb->jump;
|
|
// if its from another function chop it!
|
|
RAnalFunction *curfcn = r_anal_get_fcn_in (core->anal, jump, R_ANAL_FCN_TYPE_NULL);
|
|
if (curfcn != fcn) {
|
|
// chop that branch
|
|
r_cons_printf ("\n // chop\n");
|
|
break;
|
|
}
|
|
if (sdb_get (db, K_INDENT(jump), 0)) {
|
|
// already tracekd
|
|
if (!sdb_get (db, K_INDENT(fail), 0)) {
|
|
bb = r_anal_bb_from_offset (core->anal, fail);
|
|
}
|
|
} else {
|
|
bb = r_anal_bb_from_offset (core->anal, jump);
|
|
if (!bb) {
|
|
eprintf ("failed to retrieve blcok at 0x%"PFMT64x"\n", jump);
|
|
break;
|
|
}
|
|
if (fail != UT64_MAX) {
|
|
// do not push if already pushed
|
|
indent++;
|
|
if (sdb_get (db, K_INDENT(bb->fail), 0)) {
|
|
/* do nothing here */
|
|
eprintf ("BlockAlready 0x%"PFMT64x"\n", bb->addr);
|
|
} else {
|
|
// r_cons_printf (" { RADICAL %llx\n", bb->addr);
|
|
sdb_array_push_num (db, "indent", fail, 0);
|
|
sdb_num_set (db, K_INDENT(fail), indent, 0);
|
|
sdb_num_set (db, K_ELSE(fail), 1, 0);
|
|
r_cons_printf (" {");
|
|
}
|
|
} else {
|
|
r_cons_printf ("\n%s do", indentstr);
|
|
sdb_array_push_num (db, "indent", jump, 0);
|
|
sdb_num_set (db, K_INDENT(jump), indent, 0);
|
|
sdb_num_set (db, K_ELSE(jump), 1, 0);
|
|
if (jump <= bb->addr) {
|
|
blocktype = "while";
|
|
} else {
|
|
blocktype = "else";
|
|
}
|
|
r_cons_printf (" {");
|
|
indent++;
|
|
}
|
|
}
|
|
} else {
|
|
ut64 addr = sdb_array_pop_num (db, "indent", NULL);
|
|
if (addr == UT64_MAX) {
|
|
r_cons_printf ("\nbreak\n");
|
|
break;
|
|
}
|
|
bb = r_anal_bb_from_offset (core->anal, addr);
|
|
nindent = sdb_num_get (db, K_INDENT(addr), NULL);
|
|
if (indent>nindent) {
|
|
int i;
|
|
for (i = indent; i != nindent; i--) {
|
|
SET_INDENT (i);
|
|
r_cons_printf ("\n%s}", indentstr);
|
|
}
|
|
}
|
|
if (nindent != indent) {
|
|
r_cons_printf ("\n%s} else {\n", indentstr);
|
|
}
|
|
indent = nindent;
|
|
}
|
|
}
|
|
//n_bb --;
|
|
} while (n_bb > 0);
|
|
r_cons_printf ("}\n");
|
|
r_config_restore (hc);
|
|
r_config_hold_free (hc);
|
|
sdb_free (db);
|
|
return true;
|
|
}
|