radare2/libr/core/cmd_flag.c

598 lines
16 KiB
C

/* radare - LGPL - Copyright 2009-2015 - pancake */
static void flagbars(RCore *core) {
int total = 0;
int cols = r_cons_get_size (NULL);
RListIter *iter;
RFlagItem *flag;
r_list_foreach (core->flags->flags, iter, flag) {
total += flag->offset;
}
if (!total) // avoid a division by zero
return;
cols -= 15;
r_cons_printf ("Total: %d\n", total);
r_list_foreach (core->flags->flags, iter, flag) {
ut32 pbar_val = flag->offset>0 ? flag->offset : 1;
r_cons_printf ("%10s %.8"PFMT64d, flag->name, flag->offset);
r_print_progressbar (core->print,
(pbar_val*100)/total, cols);
r_cons_newline ();
}
}
static void flagbars_dos(RCore *core) {
int total = 0;
int cols = r_cons_get_size (NULL);
RListIter *iter;
RFlagItem *flag;
r_list_foreach (core->flags->flags, iter, flag) {
total = R_MAX(total,flag->offset);
}
if (!total) // avoid a division by zero
return;
cols-=15;
r_cons_printf ("Total: %d\n", total);
r_list_foreach (core->flags->flags, iter, flag) {
ut32 pbar_val = flag->offset>0 ? flag->offset : 1;
r_cons_printf ("%10s %.8"PFMT64d, flag->name, flag->offset);
r_print_progressbar (core->print,
(pbar_val*100)/total, cols);
r_cons_newline ();
}
}
static int cmd_flag(void *data, const char *input) {
static int flagenum = 0;
RCore *core = (RCore *)data;
ut64 off = core->offset;
char *ptr, *str = NULL;
RFlagItem *item;
char *name = NULL;
st64 base;
// TODO: off+=cursor
if (*input)
str = strdup (input+1);
rep:
switch (*input) {
case 'e':
switch (input[1]) {
case ' ':
ptr = r_str_newf ("%s.%d", input+2, flagenum);
(void)r_flag_set (core->flags, ptr,
core->offset, 1, 0);
flagenum++;
free (ptr);
break;
case '-':
flagenum = 0;
break;
default:
eprintf ("|Usage: fe[-| name] @@= 1 2 3 4\n");
break;
}
break;
case '=': // "f="
switch (input[1]) {
case '=':
flagbars_dos (core);
break;
case '?':
eprintf ("Usage: f= or f== to display flag bars\n");
break;
default:
flagbars (core);
break;
}
break;
case 'a':
if (input[1]==' '){
RFlagItem *fi;
R_FREE (str);
str = strdup (input+2);
ptr = strchr (str, '=');
if (!ptr)
ptr = strchr (str, ' ');
if (ptr) *ptr++ = 0;
name = (char *)r_str_chop_ro (str);
ptr = (char *)r_str_chop_ro (ptr);
fi = r_flag_get (core->flags, name);
if (!fi)
fi = r_flag_set (core->flags, name,
core->offset, 1, 0);
if (fi) {
r_flag_item_set_alias (fi, ptr);
} else {
eprintf ("Cannot find flag '%s'\n", name);
}
} else {
eprintf ("Usage: fa flagname flagalias\n");
}
break;
case 'V': // visual marks
switch(input[1]) {
case '-':
r_core_visual_mark_reset (core);
break;
case ' ':
{
const char *arg = strchr (input+2, ' ');
ut64 addr = arg? r_num_math (core->num, arg): core->offset;
r_core_visual_mark_set (core, atoi (input+1), addr);
}
break;
case '?':
eprintf ("Usage: fV[*-] [nkey] [offset]\n");
eprintf ("Dump/Restore visual marks (mK/'K)\n");
break;
default:
r_core_visual_mark_dump (core);
break;
}
break;
case 'm':
r_flag_move (core->flags, core->offset, r_num_math (core->num, input+1));
break;
case '2':
r_flag_get_i2 (core->flags, r_num_math (core->num, input+1));
break;
case 'R':
{
if (*str == '\0'){
eprintf ("Usage: fR [from] [to] ([mask])\n");
eprintf ("Example to relocate PIE flags on debugger:\n"
" > fR entry0 `dm~:1[1]`\n");
break;
}
char *p = strchr (str+1, ' ');
ut64 from, to, mask = 0xffff;
int ret;
if (p) {
char *q = strchr (p+1, ' ');
*p = 0;
if (q) {
*q = 0;
mask = r_num_math (core->num, q+1);
}
from = r_num_math (core->num, str+1);
to = r_num_math (core->num, p+1);
ret = r_flag_relocate (core->flags, from, mask, to);
eprintf ("Relocated %d flags\n", ret);
} else {
eprintf ("Usage: fR [from] [to] ([mask])\n");
eprintf ("Example to relocate PIE flags on debugger:\n"
" > fR entry0 `dm~:1[1]`\n");
}
}
break;
case 'b':
switch (input[1]) {
case ' ':
free(str);
str = strdup (input+2);
ptr = strchr (str, ' ');
if (ptr) {
RListIter *iter;
RFlagItem *flag;
RFlag *f = core->flags;
*ptr = 0;
base = r_num_math (core->num, str);
r_list_foreach (f->flags, iter, flag) {
if (r_str_glob (flag->name, ptr+1))
flag->offset += base;
}
} else core->flags->base = r_num_math (core->num, input+1);
free (str);
str = NULL;
break;
case '\0':
r_cons_printf ("%"PFMT64d" 0x%"PFMT64x"\n",
core->flags->base,
core->flags->base);
break;
default:
eprintf ("Usage: fb [addr] [[flags*]]\n");
break;
}
break;
case '+':
case ' ': {
char* eq = strchr (str, '=');
char* s = strchr (str, ' ');
char* s2 = NULL;
ut32 bsze = 1; //core->blocksize;
if (eq) {
// TODO: add support for '=' char in flag comments
*eq = 0;
off = r_num_math (core->num, eq+1);
}
if (s) {
*s = '\0';
s2 = strchr (s+1, ' ');
if (s2) {
*s2 = '\0';
if (s2[1]&&s2[2])
off = r_num_math (core->num, s2+1);
}
bsze = r_num_math (core->num, s+1);
}
if (*str == '.') {
input++;
goto rep;
#if 0
eprintf ("WTF 'f .xxx' adds a variable to the function? ?!!?(%s)\n");
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, off, 0);
if (fcn) r_anal_var_add (core->anal, fcn->addr, 0, off, 'v', "int", 4, str+1);
else eprintf ("Cannot find function at 0x%08"PFMT64x"\n", off);
#endif
} else r_flag_set (core->flags, str, off, bsze, (*input=='+'));
}
break;
case '-':
if (input[1]=='-') {
r_flag_unset_all (core->flags);
} else if (input[1]) {
const char *flagname = input+1;
while (*flagname==' ') flagname++;
if (*flagname=='.') {
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, off, 0);
if (fcn) eprintf ("TODO: local_del_name has been deprecated\n");
//;r_anal_fcn_local_del_name (core->anal, fcn, flagname+1);
else eprintf ("Cannot find function at 0x%08"PFMT64x"\n", off);
} else {
if (strchr (flagname, '*'))
r_flag_unset_glob (core->flags, flagname);
else r_flag_unset (core->flags, flagname, NULL);
}
} else r_flag_unset_i (core->flags, off, NULL);
break;
case '.':
if (input[1]) {
if (input[1] == '*') {
if (input[2] == '*') {
r_anal_fcn_labels (core->anal, NULL, 1);
} else {
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, off, 0);
if (fcn) r_anal_fcn_labels (core->anal, fcn, 1);
else eprintf ("Cannot find function at 0x%08"PFMT64x"\n", off);
}
} else {
const char *name = input+((input[2]==' ')? 2:1);
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, off, 0);
if (fcn) {
if (*name=='-') {
r_anal_fcn_label_del (core->anal, fcn, name+1, off);
} else {
r_anal_fcn_label_set (core->anal, fcn, name, off);
}
} else eprintf ("Cannot find function at 0x%08"PFMT64x"\n", off);
}
} else {
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, off, 0);
if (fcn) r_anal_fcn_labels (core->anal, fcn, 0);
else eprintf ("Cannot find function at 0x%08"PFMT64x"\n", off);
}
break;
case 'l': // "fl"
if (input[1] == ' ') {
char *p, *arg = strdup (input+2);
r_str_trim_head_tail (arg);
p = strchr (arg, ' ');
if (p) {
*p++ = 0;
item = r_flag_get_i (core->flags,
r_num_math (core->num, str));
if (item)
item->size = r_num_math (core->num, p);
} else {
item = r_flag_get_i (core->flags,
r_num_math (core->num, str));
if (item)
r_cons_printf ("0x%08"PFMT64x"\n", item->size);
}
free (arg);
} else {
item = r_flag_get_i (core->flags, core->offset);
if (item)
r_cons_printf ("0x%08"PFMT64x"\n", item->size);
}
break;
#if 0
case 'd':
if (input[1] == ' ') {
char cmd[128];
RFlagItem *item = r_flag_get_i (core->flags,
r_num_math (core->num, input+2));
if (item) {
r_cons_printf ("0x%08"PFMT64x"\n", item->offset);
snprintf (cmd, sizeof (cmd), "pD@%"PFMT64d":%"PFMT64d,
item->offset, item->size);
r_core_cmd0 (core, cmd);
}
} else eprintf ("Missing arguments\n");
break;
#endif
case 'x':
if (input[1] == ' ') {
char cmd[128];
RFlagItem *item = r_flag_get_i (core->flags,
r_num_math (core->num, input+2));
if (item) {
r_cons_printf ("0x%08"PFMT64x"\n", item->offset);
snprintf (cmd, sizeof (cmd), "px@%"PFMT64d":%"PFMT64d,
item->offset, item->size);
r_core_cmd0 (core, cmd);
}
} else eprintf ("Missing arguments\n");
break;
case 'S':
r_flag_sort (core->flags, (input[1]=='n'));
break;
case 's':
switch (input[1]) {
case '?':
{
const char *help_msg[] = {
"Usage: fs","[*] [+-][flagspace|addr]", " # Manage flagspaces",
"fs","","display flagspaces",
"fs"," *","select all flagspaces",
"fs"," flagspace","select flagspace or create if it doesn't exist",
"fs","-flagspace","remove flagspace",
"fs","-*","remove all flagspaces",
"fs","+foo","push previous flagspace and set",
"fs","-","pop to the previous flagspace",
"fs","-.","remove the current flagspace",
"fsm"," [addr]","move flags at given address to the current flagspace",
"fsr"," newname","rename selected flagspace",
NULL};
r_core_cmd_help (core, help_msg);
}
break;
case '+':
r_flag_space_push (core->flags, input+2);
break;
case 'r':
if (input[2]==' ')
r_flag_space_rename (core->flags, NULL, input+2);
else eprintf ("Usage: fsr [newname]\n");
break;
case '-':
switch (input[2]) {
case '*':
r_flag_space_unset (core->flags, NULL);
break;
case '.':
{
const char *curfs = r_flag_space_cur (core->flags);
r_flag_space_unset (core->flags, curfs);
}
break;
case 0:
r_flag_space_pop (core->flags);
break;
default:
r_flag_space_unset (core->flags, input+2);
break;
}
break;
case 'j':
case '\0':
case '*':
r_flag_space_list (core->flags, input[1]);
break;
case ' ':
r_flag_space_set (core->flags, input+2);
break;
case 'm':
{ RFlagItem *f;
ut64 off = core->offset;
if (input[2] == ' ')
off = r_num_math (core->num, input+2);
f = r_flag_get_i (core->flags, off);
if (f) {
f->space = core->flags->space_idx;
} else eprintf ("Cannot find any flag at 0x%"PFMT64x".\n", off);
}
break;
default: {
int i, j = 0;
for (i=0; i<R_FLAG_SPACES_MAX; i++) {
if (core->flags->spaces[i])
r_cons_printf ("%02d %c %s\n", j++,
(i==core->flags->space_idx)?'*':' ',
core->flags->spaces[i]);
}
} break;
}
break;
case 'g':
r_core_cmd0 (core, "V");
break;
case 'c':
if (input[1]=='?') {
const char *help_msg[] = {
"Usage: fc", "<flagname> [color]", " # List colors with 'ecs'",
"fc", " flagname", "Get current color for given flagname",
"fc", " flagname color", "Set color to a flag",
NULL
};
r_core_cmd_help (core, help_msg);
} else {
RFlagItem *fi;
const char *ret;
char *arg = r_str_chop (strdup (input+2));
char *color = strchr (arg, ' ');
if (color && color[1])
*color++ = 0;
fi = r_flag_get (core->flags, arg);
if (fi) {
ret = r_flag_color (core->flags, fi, color);
if (!color && ret)
r_cons_printf ("%s\n", ret);
} else {
eprintf ("Unknown flag '%s'\n", arg);
}
free (arg);
}
break;
case 'C':
if (input[1]==' ') {
RFlagItem *item;
char *q, *p = strdup (input+2);
q = strchr (p, ' ');
if (q) {
*q = 0;
item = r_flag_get (core->flags, p);
if (item) {
r_flag_item_set_comment (item, q+1);
} else eprintf ("Cannot find flag with name '%s'\n", p);
} else {
item = r_flag_get_i (core->flags, r_num_math (core->num, p));
if (item && item->comment) {
r_cons_printf ("%s\n", item->comment);
} else eprintf ("Cannot find item\n");
}
free (p);
} else eprintf ("Usage: fC [name] [comment]\n");
break;
case 'o':
{ // TODO: use file.fortunes // can be dangerous in sandbox mode
char *fortunes_tips = R2_PREFIX"/share/doc/radare2/fortunes.tips";
char *fortunes_fun = R2_PREFIX"/share/doc/radare2/fortunes.fun";
char *fortunes_nsfw = R2_PREFIX"/share/doc/radare2/fortunes.nsfw";
char *types = (char *)r_config_get(core->config, "cfg.fortunetype");
char *line = NULL, *templine = NULL;
int i = 0;
if (strstr(types, "tips")) {
templine = r_file_slurp_random_line_count (fortunes_tips, &i);
line = templine;
}
if (strstr(types, "fun")) {
templine = r_file_slurp_random_line_count (fortunes_fun, &i);
if (templine) {
free (line);
line = templine;
}
}
if (strstr(types, "nsfw")) {
templine = r_file_slurp_random_line_count (fortunes_nsfw, &i);
if (templine) {
free (line);
line = templine;
}
}
if (line) {
r_cons_printf (" -- %s\n", line);
free (line);
}
}
break;
case 'r':
if (input[1]==' ' && input[2]) {
char *old, *new;
RFlagItem *item;
old = str+1;
new = strchr (old, ' ');
if (new) {
*new = 0;
new++;
item = r_flag_get (core->flags, old);
if (!item && !strncmp (old, "fcn.", 4)) {
item = r_flag_get (core->flags, old+4);
}
} else {
new = old;
item = r_flag_get_i (core->flags, core->offset);
}
if (item) {
if (!r_flag_rename (core->flags, item, new))
eprintf ("Invalid name\n");
} else {
eprintf ("Cannot find flag (%s)\n", old);
}
}
break;
case 'n': // "fn"
case '*': // "f*"
case '\0':
case 'j': // "fj"
r_flag_list (core->flags, *input, input[0]? input+1:"");
break;
case 'd': // "fd"
{
ut64 addr = 0;
RFlagItem *f = NULL;
switch (input[1]) {
case '?':
eprintf ("Usage: fd [offset|flag|expression]\n");
if (str)
free (str);
return R_FALSE;
case '\0':
addr = core->offset;
break;
default:
addr = r_num_math (core->num, input+2);
break;
}
f = r_flag_get_at (core->flags, addr);
if (f) {
if (f->offset != addr) {
r_cons_printf ("%s + %d\n", f->name,
(int)(addr - f->offset));
} else {
r_cons_printf ("%s\n", f->name);
}
}
}
break;
case '?':
{
const char *help_msg[] = {
"Usage: f","[?] [flagname]", " # Manage offset-name flags",
"f","","list flags (will only list flags from selected flagspaces)",
"f."," [*[*]]","list local per-function flags (*) as r2 commands",
"f.","blah=$$+12","set local function label named 'blah'",
"f*","","list flags in r commands",
"f"," name 12 @ 33","set flag 'name' with length 12 at offset 33",
"f"," name = 33","alias for 'f name @ 33' or 'f name 1 33'",
"f"," name 12 33 [cmt]","same as above + optional comment",
"f-",".blah@fcn.foo","delete local label from function at current seek (also f.-)",
"f--","","delete all flags and flagspaces (deinit)",
"f+","name 12 @ 33","like above but creates new one if doesnt exist",
"f-","name","remove flag 'name'",
"f-","@addr","remove flag at address expression",
"f."," fname","list all local labels for the given function",
"fa"," [name] [alias]","alias a flag to evaluate an expression",
"fb"," [addr]","set base address for new flags",
"fb"," [addr] [flag*]","move flags matching 'flag' to relative addr",
"fc"," [name] [color]","set color for given flag",
"fC"," [name] [cmt]","set comment for given flag",
"fd"," addr","return flag+delta",
"fe-","","resets the enumerator counter",
"fe"," [name]","create flag name.#num# enumerated flag. See fe?",
"fg","","bring visual mode to foreground",
"fj","","list flags in JSON format",
"fl"," [flag] [size]","show or set flag length (size)",
"fm"," addr","move flag at current offset to new address",
"fn","","list flags displaying the real name (demangled)",
"fo","","show fortunes",
//" fc [name] [cmt] ; set execution command for a specific flag"
"fr"," [old] [[new]]","rename flag (if no new flag current seek one is used)",
"fR"," [f] [t] [m]","relocate all flags matching f&~m 'f'rom, 't'o, 'm'ask",
"fs"," ?+-*","manage flagspaces",
"fS","[on]","sort flags by offset or name",
"fx","[d]","show hexdump (or disasm) of flag:flagsize",
NULL};
r_core_cmd_help (core, help_msg);
break;
}
}
if (str)
free (str);
return 0;
}