radare2/libr/core/anal.c
pancake e36ef11025 * Initial working basic block conditional decompilation
- Only 'test' opcode for x86 is currently supported (just for testing)
  - analysis backend generates RAnalValue's for each argument
  - compiles a cmp+cjmp into a RAnalCond class
  - de/serializes the RAnalCond into an evaluable string
  - Make anal api more stable
* Fix a memory leak in anal_bb
* Reassign anal->reg into dbg->reg
  - Replicate dbg->reg into dbg->anal->reg
  - Such nasty cascade assignation..
2010-06-16 21:44:19 +02:00

397 lines
12 KiB
C

/* radare - LGPL - Copyright 2009-2010 nibble<.ds@gmail.com> */
#include <r_types.h>
#include <r_list.h>
#include <r_flags.h>
#include <r_core.h>
static char *r_core_anal_graph_label(RCore *core, struct r_anal_bb_t *bb, int opts) {
struct r_anal_aop_t *aopi;
RListIter *iter;
char cmd[1024], file[1024], *cmdstr = NULL, *filestr = NULL, *str = NULL;
int i, j, line = 0, oline = 0, idx = 0;
if (opts & R_CORE_ANAL_GRAPHLINES) {
r_list_foreach (bb->aops, iter, aopi) {
r_bin_meta_get_line (core->bin, aopi->addr, file, 1023, &line);
if (line != 0 && line != oline && strcmp (file, "??")) {
filestr = r_file_slurp_line (file, line, 0);
if (filestr) {
cmdstr = realloc (cmdstr, idx + strlen (filestr) + 3);
cmdstr[idx] = 0;
strcat (cmdstr, filestr);
strcat (cmdstr, "\\l");
idx+=strlen (filestr);
free (filestr);
}
}
oline = line;
}
} else if (opts & R_CORE_ANAL_GRAPHBODY) {
snprintf (cmd, sizeof (cmd), "pD %"PFMT64d" @ 0x%08"PFMT64x"", bb->size, bb->addr);
cmdstr = r_core_cmd_str (core, cmd);
}
if (cmdstr) {
if (!(str = malloc(strlen(cmdstr)*2)))
return NULL;
for(i=j=0;cmdstr[i];i++,j++) {
switch(cmdstr[i]) {
case 0x1b:
/* skip ansi chars */
for(i++;cmdstr[i]&&cmdstr[i]!='m'&&cmdstr[i]!='H'&&cmdstr[i]!='J';i++);
j--;
break;
case '"':
str[j]='\\';
str[++j]='"';
break;
case '\n':
case '\r':
str[j]='\\';
str[++j]='l';
break;
default:
str[j]=cmdstr[i];
}
}
str[j]='\0';
free (cmdstr);
}
return str;
}
static void r_core_anal_graph_nodes(RCore *core, RList *pbb, ut64 addr, int opts) {
struct r_anal_bb_t *bbi, *bbc;
RListIter *iter;
char *str;
/* In partial graphs test if the bb is already printed */
if (pbb)
r_list_foreach (pbb, iter, bbi)
if (addr == bbi->addr)
return;
r_list_foreach (core->anal->bbs, iter, bbi) {
if (addr == 0 || addr == bbi->addr) {
if (pbb) { /* Copy BB and append to the list of printed bbs */
bbc = R_NEW (RAnalBlock);
if (bbc) {
memcpy (bbc, bbi, sizeof (RAnalBlock));
bbc->aops = NULL;
bbc->fingerprint = NULL;
bbc->cond = NULL;
r_list_append (pbb, bbc);
}
}
if (bbi->jump != -1) {
r_cons_printf ("\t\"0x%08"PFMT64x"\" -> \"0x%08"PFMT64x"\" [color=\"%s\"];\n", bbi->addr, bbi->jump,
bbi->fail != -1 ? "green" : "blue");
r_cons_flush ();
if (addr != 0) r_core_anal_graph_nodes (core, pbb, bbi->jump, opts);
}
if (bbi->fail != -1) {
r_cons_printf ("\t\"0x%08"PFMT64x"\" -> \"0x%08"PFMT64x"\" [color=\"red\"];\n", bbi->addr, bbi->fail);
r_cons_flush ();
if (addr != 0) r_core_anal_graph_nodes (core, pbb, bbi->fail, opts);
}
if ((str = r_core_anal_graph_label (core, bbi, opts))) {
if (opts & R_CORE_ANAL_GRAPHDIFF) {
r_cons_printf (" \"0x%08"PFMT64x"\" [color=%s,label=\"%s\"]\n", bbi->addr,
bbi->diff==R_ANAL_DIFF_MATCH?"green":
bbi->diff==R_ANAL_DIFF_UNMATCH?"red":"lightgray",str);
} else {
r_cons_printf (" \"0x%08"PFMT64x"\" [label=\"%s\"]\n", bbi->addr, str);
}
r_cons_flush ();
free (str);
}
}
}
}
R_API int r_core_anal_bb(RCore *core, ut64 at, int depth, int head) {
struct r_anal_bb_t *bb, *bbi;
RListIter *iter;
ut64 jump, fail;
ut8 *buf;
int ret = R_ANAL_RET_NEW, buflen, bblen = 0;
int split = r_config_get_i (core->config, "anal.split");
if (depth < 0)
return R_FALSE;
if (!(bb = r_anal_bb_new()))
return R_FALSE;
if (split)
ret = r_anal_bb_split (core->anal, bb, core->anal->bbs, at);
else {
r_list_foreach (core->anal->bbs, iter, bbi)
if (at == bbi->addr)
ret = R_ANAL_RET_DUP;
}
if (ret == R_ANAL_RET_DUP) { /* Dupped bb */
r_anal_bb_free (bb);
return R_FALSE;
} else if (ret == R_ANAL_RET_NEW) { /* New bb */
if (!(buf = malloc (core->blocksize)))
return R_FALSE;
do {
if ((buflen = r_io_read_at (core->io, at+bblen, buf, core->blocksize)) != core->blocksize)
return R_FALSE;
bblen = r_anal_bb (core->anal, bb, at+bblen, buf, buflen, head);
if (bblen == R_ANAL_RET_ERROR) { /* Error analyzing bb */
r_anal_bb_free (bb);
return R_FALSE;
} else if (bblen == R_ANAL_RET_END) { /* bb analysis complete */
if (split)
ret = r_anal_bb_overlap (core->anal, bb, core->anal->bbs);
if (ret == R_ANAL_RET_NEW) {
r_list_append (core->anal->bbs, bb);
fail = bb->fail;
jump = bb->jump;
if (fail != -1)
r_core_anal_bb (core, fail, depth-1, R_FALSE);
if (jump != -1)
r_core_anal_bb (core, jump, depth-1, R_FALSE);
}
}
} while (bblen != R_ANAL_RET_END);
free (buf);
}
return R_TRUE;
}
R_API int r_core_anal_bb_list(RCore *core, int rad) {
struct r_anal_bb_t *bbi;
RListIter *iter;
r_list_foreach (core->anal->bbs, iter, bbi) {
if (rad) {
r_cons_printf ("ab+ 0x%08"PFMT64x" %04"PFMT64d" ", bbi->addr, bbi->size);
r_cons_printf ("0x%08"PFMT64x" ", bbi->jump);
r_cons_printf ("0x%08"PFMT64x" ", bbi->fail);
if (bbi->type != R_ANAL_BB_TYPE_NULL) {
if ((bbi->type & R_ANAL_BB_TYPE_BODY))
r_cons_printf ("b");
if ((bbi->type & R_ANAL_BB_TYPE_FOOT))
r_cons_printf ("f");
if ((bbi->type & R_ANAL_BB_TYPE_HEAD))
r_cons_printf ("h");
if ((bbi->type & R_ANAL_BB_TYPE_LAST))
r_cons_printf ("l");
} else r_cons_printf ("n");
if ((bbi->diff == R_ANAL_DIFF_MATCH))
r_cons_printf (" m");
else if ((bbi->diff == R_ANAL_DIFF_UNMATCH))
r_cons_printf (" u");
else r_cons_printf (" n");
r_cons_printf ("\n");
} else {
r_cons_printf ("[0x%08"PFMT64x"] size=%04"PFMT64d, bbi->addr, bbi->size);
if (bbi->jump != -1)
r_cons_printf (" jump=0x%08"PFMT64x, bbi->jump);
if (bbi->fail != -1)
r_cons_printf (" fail=0x%08"PFMT64x, bbi->fail);
r_cons_printf (" type=");
if (bbi->type == R_ANAL_BB_TYPE_NULL)
r_cons_printf ("null ");
else {
if ((bbi->type & R_ANAL_BB_TYPE_BODY))
r_cons_printf ("body ");
if ((bbi->type & R_ANAL_BB_TYPE_FOOT))
r_cons_printf ("foot ");
if ((bbi->type & R_ANAL_BB_TYPE_HEAD))
r_cons_printf ("head ");
if ((bbi->type & R_ANAL_BB_TYPE_LAST))
r_cons_printf ("last ");
}
r_cons_printf ("diff=");
if ((bbi->diff == R_ANAL_DIFF_MATCH))
r_cons_printf ("match");
else if ((bbi->diff == R_ANAL_DIFF_UNMATCH))
r_cons_printf ("unmatch");
else r_cons_printf ("new");
r_cons_printf (" cond=\"%s\"\n", bbi->cond?
r_anal_cond_to_string (bbi->cond):"none");
}
}
r_cons_flush ();
return R_TRUE;
}
R_API int r_core_anal_bb_seek(RCore *core, ut64 addr) {
struct r_anal_bb_t *bbi;
RListIter *iter;
r_list_foreach (core->anal->bbs, iter, bbi)
if (addr >= bbi->addr && addr < bbi->addr+bbi->size)
return r_core_seek (core, bbi->addr, R_FALSE);
return r_core_seek (core, addr, R_FALSE);
}
R_API int r_core_anal_fcn(RCore *core, ut64 at, ut64 from, int depth) {
RAnalFcn *fcn, *fcni;
struct r_anal_ref_t *refi;
RListIter *iter, *iter2;
char *flagname;
RAnalRef *ref;
ut8 *buf;
int buflen, fcnlen = 0;
if (depth < 0)
return R_FALSE;
r_list_foreach (core->anal->fcns, iter, fcni)
if ((at >= fcni->addr && at < fcni->addr+fcni->size) ||
(at == fcni->addr && fcni->size == 0)) {
if (from != -1) {
r_list_foreach (fcni->xrefs, iter2, refi) {
if (from == refi->addr)
return R_FALSE;
}
if (!(ref = r_anal_ref_new ())) {
eprintf ("Error: new (xref)\n");
return R_ANAL_RET_ERROR;
}
ref->addr = from;
r_list_append (fcni->xrefs, ref);
}
return R_FALSE;
}
if (!(fcn = r_anal_fcn_new()))
return R_FALSE;
if (!(buf = malloc (core->blocksize)))
return R_FALSE;
do {
if ((buflen = r_io_read_at (core->io, at+fcnlen, buf, core->blocksize)) != core->blocksize)
return R_FALSE;
fcnlen = r_anal_fcn (core->anal, fcn, at+fcnlen, buf, buflen);
if (fcnlen == R_ANAL_RET_ERROR) { /* Error analyzing function */
eprintf ("Unknown opcode at 0x%08"PFMT64x"\n", at+fcnlen);
r_anal_fcn_free (fcn);
return R_FALSE;
} else if (fcnlen == R_ANAL_RET_END) { /* function analysis complete */
fcn->name = r_str_dup_printf ("fcn_%08"PFMT64x"", at);
/* Add flag */
flagname = r_str_dup_printf ("fcn.%s", fcn->name);
r_flag_space_set (core->flags, "functions");
r_flag_set (core->flags, flagname, at, fcn->size, 0);
free (flagname);
r_list_append (core->anal->fcns, fcn);
r_list_foreach (fcn->refs, iter, refi) {
if (refi->addr != -1)
r_core_anal_fcn (core, refi->addr, at, depth-1);
}
}
} while (fcnlen != R_ANAL_RET_END);
free (buf);
return R_TRUE;
}
R_API int r_core_anal_fcn_clean(RCore *core, ut64 addr) {
RAnalFcn *fcni;
RListIter *iter;
if (addr == 0) {
r_list_destroy (core->anal->fcns);
if (!(core->anal->fcns = r_anal_fcn_list_new ()))
return R_FALSE;
} else r_list_foreach (core->anal->fcns, iter, fcni)
if (addr >= fcni->addr && addr < fcni->addr+fcni->size)
r_list_unlink (core->anal->fcns, fcni);
return R_TRUE;
}
R_API void r_core_anal_refs(RCore *core, ut64 addr, int gv) {
RListIter *iter, *iter2;
RAnalRef *fcnr;
RAnalFcn *fcni;
if (gv) r_cons_printf ("digraph code {\n"
"\tgraph [bgcolor=white];\n"
"\tnode [color=lightgray, style=filled shape=box"
" fontname=\"Courier\" fontsize=\"8\"];\n");
r_list_foreach (core->anal->fcns, iter, fcni) {
if (addr != 0 && addr != fcni->addr)
continue;
if (!gv) r_cons_printf ("0x%08"PFMT64x"\n", fcni->addr);
r_list_foreach (fcni->refs, iter2, fcnr) {
// TODO: display only code or data refs?
RFlagItem *flag = r_flag_get_i (core->flags, fcnr->addr);
if (gv) r_cons_printf ("\t\"0x%08"PFMT64x"\" -> \"0x%08"PFMT64x"\" "
"[label=\"%s\" color=\"%s\"];\n",
fcni->addr, fcnr->addr, flag?flag->name:"",
(fcnr->type==R_ANAL_REF_TYPE_CODE)?"green":"red");
else r_cons_printf (" - 0x%08"PFMT64x" (%c)\n", fcnr->addr, fcnr->type);
}
}
r_cons_printf ("}\n");
}
R_API int r_core_anal_fcn_list(RCore *core, int rad) {
RAnalFcn *fcni;
struct r_anal_ref_t *refi;
struct r_anal_var_t *vari;
RListIter *iter, *iter2;
r_list_foreach (core->anal->fcns, iter, fcni)
if (!rad) {
r_cons_printf ("[0x%08"PFMT64x"] size=%"PFMT64d" name=%s",
fcni->addr, fcni->size, fcni->name);
r_cons_printf (" diff=%s", fcni->diff=='m'?"match": fcni->diff=='u'?"unmatch": "new");
r_cons_printf ("\n refs: ");
r_list_foreach (fcni->refs, iter2, refi) {
r_cons_printf ("0x%08"PFMT64x"(%c) ", refi->addr, refi->type);
}
r_cons_printf ("\n xrefs: ");
r_list_foreach (fcni->xrefs, iter2, refi) {
r_cons_printf ("0x%08"PFMT64x"(%c) ", refi->addr, refi->type);
}
r_cons_printf ("\n vars:\n");
r_list_foreach (fcni->vars, iter2, vari) {
r_cons_printf (" %-10s delta=0x%02x type=%s\n", vari->name, vari->delta,
r_anal_var_type_to_str (core->anal, vari->type));
}
r_cons_newline ();
} else r_cons_printf ("af+ 0x%08"PFMT64x" %"PFMT64d" %s (%c)\n",
fcni->addr, fcni->size, fcni->name, fcni->diff?fcni->diff:'n');
r_cons_flush ();
return R_TRUE;
}
R_API int r_core_anal_graph(RCore *core, ut64 addr, int opts) {
RList *pbb = NULL;
int reflines = r_config_get_i (core->config, "asm.lines");
int bytes = r_config_get_i (core->config, "asm.bytes");
int dwarf = r_config_get_i (core->config, "asm.dwarf");
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.dwarf", 0);
r_cons_printf ("digraph code {\n"
"\tgraph [bgcolor=white];\n"
"\tnode [color=lightgray, style=filled shape=box"
" fontname=\"Courier\" fontsize=\"8\"];\n");
r_cons_flush ();
if (addr != 0) pbb = r_anal_bb_list_new (); /* In partial graphs define printed bb list */
r_core_anal_graph_nodes (core, pbb, addr, opts);
if (pbb) r_list_free (pbb);
r_cons_printf ("}\n");
r_cons_flush ();
r_config_set_i (core->config, "asm.lines", reflines);
r_config_set_i (core->config, "asm.bytes", bytes);
r_config_set_i (core->config, "asm.dwarf", dwarf);
return R_TRUE;
}
R_API int r_core_anal_graph_fcn(RCore *core, char *fname, int opts) {
RListIter *iter;
RAnalFcn *fcni;
r_list_foreach (core->anal->fcns, iter, fcni)
if (!strcmp (fname, fcni->name))
return r_core_anal_graph (core, fcni->addr, opts);
return R_FALSE;
}