radare2/libr/core/vmenus.c
2020-01-19 20:37:29 +08:00

4239 lines
100 KiB
C

/* radare - LGPL - Copyright 2009-2019 - pancake */
#include <r_core.h>
#include <r_util.h>
#include <string.h>
#define MAX_FORMAT 3
enum {
R_BYTE_DATA = 1,
R_WORD_DATA = 2,
R_DWORD_DATA = 4,
R_QWORD_DATA = 8
};
enum {
SORT_NONE,
SORT_NAME,
SORT_OFFSET
};
typedef struct {
RCore *core;
int t_idx;
int t_ctr;
const char *type;
char *curname;
char *curfmt;
const char *optword;
} RCoreVisualTypes;
// TODO: move this helper into r_cons
static char *prompt(const char *str, const char *txt) {
char cmd[1024];
char *res = NULL;
char *oprompt = strdup (r_cons_singleton ()->line->prompt);
r_cons_show_cursor (true);
if (txt && *txt) {
free (r_cons_singleton ()->line->contents);
r_cons_singleton ()->line->contents = strdup (txt);
} else {
R_FREE (r_cons_singleton ()->line->contents);
}
*cmd = '\0';
r_line_set_prompt (str);
if (r_cons_fgets (cmd, sizeof (cmd) - 1, 0, NULL) < 0) {
*cmd = '\0';
}
//line[strlen(line)-1]='\0';
if (*cmd) {
res = strdup (cmd);
}
r_line_set_prompt (oprompt);
free (oprompt);
R_FREE (r_cons_singleton ()->line->contents);
return res;
}
static inline char *getformat (RCoreVisualTypes *vt, const char *k) {
return sdb_get (vt->core->anal->sdb_types,
sdb_fmt ("type.%s", k), 0);
}
static char *colorize_asm_string(RCore *core, const char *buf_asm, int optype, ut64 addr) {
char *tmp, *spacer = NULL;
char *source = (char*)buf_asm;
bool use_color = core->print->flags & R_PRINT_FLAGS_COLOR;
const char *color_num = core->cons->context->pal.num;
const char *color_reg = core->cons->context->pal.reg;
RAnalFunction* fcn = r_anal_get_fcn_in (core->anal, addr, R_ANAL_FCN_TYPE_NULL);
if (!use_color) {
return strdup (source);
}
// workaround dummy colorizer in case of paired commands (tms320 & friends)
spacer = strstr (source, "||");
if (spacer) {
char *s1 = r_str_ndup (source, spacer - source);
char *s2 = strdup (spacer + 2);
char *scol1 = r_print_colorize_opcode (core->print, s1, color_reg, color_num, false, fcn ? fcn->addr : 0);
char *scol2 = r_print_colorize_opcode (core->print, s2, color_reg, color_num, false, fcn ? fcn->addr : 0);
char *source = r_str_newf ("%s||%s", r_str_get2 (scol1), r_str_get2 (scol2));
free (scol1);
free (scol2);
free (s1);
free (s2);
return source;
}
char *res = strdup ("");
res = r_str_append (res, r_print_color_op_type (core->print, optype));
tmp = r_print_colorize_opcode (core->print, source, color_reg, color_num, false, fcn ? fcn->addr : 0);
res = r_str_append (res, tmp);
free (tmp);
return res;
}
static int rotate_nibble (const ut8 b, int dir) {
if (dir > 0) {
bool high = b >> 7;
return (b << 1) | high;
}
bool lower = b & 1;
return (b >> 1) | (lower << 7);
}
static int wordpos(const char *esil, int n) {
const char *w = esil;
if (n < 1) {
n = 0;
}
while (w && n--) {
const char *nw = strchr (w + 1, ',');
if (!nw) {
return strlen (esil);
}
w = nw;
}
if (!w && n > 0) {
return strlen (esil);
}
return (size_t)(w - esil);
}
static void showreg(RAnalEsil *esil, const char *rn, const char *desc) {
ut64 nm = 0;
int sz = 0;
r_cons_printf ("%s 0x%08"PFMT64x" (%d) ; %s\n", rn, nm, sz, desc);
}
R_API bool r_core_visual_esil(RCore *core) {
const int nbits = sizeof (ut64) * 8;
int analopType;
char *word = NULL;
int x = 0;
RAsmOp asmop;
RAnalOp analop;
ut8 buf[sizeof (ut64)];
unsigned int addrsize = r_config_get_i (core->config, "esil.addr.size");
if (core->blocksize < sizeof (ut64)) {
return false;
}
memcpy (buf, core->block, sizeof (ut64));
RAnalEsil *esil = r_anal_esil_new (20, 0, addrsize);
esil->anal = core->anal;
r_anal_esil_set_pc (esil, core->offset);
for (;;) {
r_cons_clear00 ();
// bool use_color = core->print->flags & R_PRINT_FLAGS_COLOR;
(void) r_asm_disassemble (core->assembler, &asmop, buf, sizeof (ut64));
analop.type = -1;
(void)r_anal_op (core->anal, &analop, core->offset, buf, sizeof (ut64), R_ANAL_OP_MASK_ESIL);
analopType = analop.type & R_ANAL_OP_TYPE_MASK;
r_cons_printf ("r2's esil debugger:\n\n");
r_cons_printf ("pos: %d\n", x);
{
char *op_hex = r_asm_op_get_hex (&asmop);
char *res = r_print_hexpair (core->print, op_hex, -1);
r_cons_printf ("hex: %s\n"Color_RESET, res);
free (res);
free (op_hex);
}
{
char *op = colorize_asm_string (core, r_asm_op_get_asm (&asmop), analopType, core->offset);
r_cons_printf (Color_RESET"asm: %s\n"Color_RESET, op);
free (op);
}
{
const char *expr = r_strbuf_get (&analop.esil);
r_cons_printf (Color_RESET"esil: %s\n"Color_RESET, expr);
int wp = wordpos (expr, x);
char *pas = strdup (r_str_pad (' ', wp ? wp + 1: 0));
int wp2 = wordpos (expr, x + 1);
free (word);
word = r_str_ndup (expr + (wp?(wp+1):0), (wp2 - wp) - (wp?1:0));
if (wp == wp2) {
// x --;
eprintf ("Done\n");
x = 0;
r_sys_sleep (1);
free (pas);
continue;
}
const char *pad = r_str_pad ('-', wp2 - ((wp > 0)? wp + 1: 0));
r_cons_printf (Color_RESET" %s%s\n"Color_RESET, pas, pad);
free (pas);
// free (pad);
}
r_cons_printf ("esil regs:\n");
showreg (esil, "$$", "address");
showreg (esil, "$z", "zero");
showreg (esil, "$b", "borrow");
showreg (esil, "$c", "carry");
showreg (esil, "$o", "overflow");
showreg (esil, "$p", "parity");
showreg (esil, "$r", "regsize");
showreg (esil, "$s", "sign");
showreg (esil, "$d", "delay");
showreg (esil, "$j", "jump");
r_cons_printf ("regs:\n");
char *r = r_core_cmd_str (core, "dr=");
if (r) {
r_cons_printf ("%s", r);
free (r);
}
r_cons_printf ("esil stack:\n");
r_anal_esil_dumpstack (esil);
r_anal_op_fini (&analop);
r_cons_newline ();
r_cons_visual_flush ();
int ch = r_cons_readchar ();
if (ch == -1 || ch == 4) {
break;
}
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case 'Q':
case 'q':
goto beach;
case 's':
eprintf ("step ((%s))\n", word);
r_sys_usleep (500);
x = R_MIN (x + 1, nbits - 1);
r_anal_esil_runword (esil, word);
break;
case 'S':
eprintf ("esil step over :D\n");
r_sys_usleep (500);
break;
case 'r':
case 'h':
x = 0; //R_MAX (x - 1, 0);
break;
case '?':
r_cons_clear00 ();
r_cons_printf (
"Vd1?: Visual Bit Editor Help:\n\n"
" q - quit the bit editor\n"
" h/r - reset / go back (reinitialize esil state)\n"
" s - esil step in\n"
" j/k - toggle bit value (same as space key)\n"
" : - enter command\n");
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':': // TODO: move this into a separate helper function
{
char cmd[1024];
r_cons_show_cursor (true);
r_cons_set_raw (0);
*cmd = 0;
r_line_set_prompt (":> ");
if (r_cons_fgets (cmd, sizeof (cmd)-1, 0, NULL) < 0) {
cmd[0] = '\0';
}
r_core_cmd_task_sync (core, cmd, 1);
r_cons_set_raw (1);
r_cons_show_cursor (false);
if (cmd[0]) {
r_cons_any_key (NULL);
}
r_cons_clear ();
}
break;
}
}
beach:
r_anal_esil_free (esil);
free (word);
return true;
}
R_API bool r_core_visual_bit_editor(RCore *core) {
const int nbits = sizeof (ut64) * 8;
bool colorBits = false;
int analopType;
int i, j, x = 0;
RAsmOp asmop;
RAnalOp analop;
ut8 buf[sizeof (ut64)];
if (core->blocksize < sizeof (ut64)) {
return false;
}
int cur = 0;
if (core->print->cur != -1) {
cur = core->print->cur;
}
memcpy (buf, core->block + cur, sizeof (ut64));
for (;;) {
r_cons_clear00 ();
bool use_color = core->print->flags & R_PRINT_FLAGS_COLOR;
(void) r_asm_disassemble (core->assembler, &asmop, buf, sizeof (ut64));
analop.type = -1;
(void)r_anal_op (core->anal, &analop, core->offset, buf, sizeof (ut64), R_ANAL_OP_MASK_ESIL);
analopType = analop.type & R_ANAL_OP_TYPE_MASK;
r_cons_printf ("r2's bit editor:\n\n");
r_cons_printf ("offset: 0x%08"PFMT64x"\n"Color_RESET, core->offset + cur);
{
char *op_hex = r_asm_op_get_hex (&asmop);
char *res = r_print_hexpair (core->print, op_hex, -1);
r_cons_printf ("hex: %s\n"Color_RESET, res);
free (res);
free (op_hex);
}
r_cons_printf ("len: %d\n", asmop.size);
{
ut32 word = (x % 32);
r_cons_printf ("shift: >> %d << %d\n", word, (asmop.size * 8) - word - 1);
}
{
char *op = colorize_asm_string (core, r_asm_op_get_asm (&asmop), analopType, core->offset);
r_cons_printf (Color_RESET"asm: %s\n"Color_RESET, op);
free (op);
}
r_cons_printf (Color_RESET"esl: %s\n"Color_RESET, r_strbuf_get (&analop.esil));
r_anal_op_fini (&analop);
r_cons_printf ("chr:");
for (i = 0; i < 8; i++) {
const ut8 *byte = buf + i;
char ch = IS_PRINTABLE (*byte)? *byte: '?';
if (i == 4) {
r_cons_printf (" |");
}
if (use_color) {
r_cons_printf (" %5s'%s%c"Color_RESET"'", " ", core->cons->context->pal.btext, ch);
} else {
r_cons_printf (" %5s'%c'", " ", ch);
}
}
r_cons_printf ("\ndec:");
for (i = 0; i < 8; i++) {
const ut8 *byte = buf + i;
if (i == 4) {
r_cons_printf (" |");
}
r_cons_printf (" %8d", *byte);
}
r_cons_printf ("\nhex:");
for (i = 0; i < 8; i++) {
const ut8 *byte = buf + i;
if (i == 4) {
r_cons_printf (" |");
}
r_cons_printf (" 0x%02x", *byte);
}
r_cons_printf ("\nbit: ");
if (use_color) {
r_cons_print (core->cons->context->pal.b0x7f);
colorBits = true;
}
for (i = 0; i < 8; i++) {
ut8 *byte = buf + i;
if (i == 4) {
r_cons_printf ("| ");
}
if (colorBits && i >= asmop.size) {
r_cons_print (Color_RESET);
colorBits = false;
}
for (j = 0; j < 8; j++) {
bool bit = R_BIT_CHK (byte, 7 - j);
r_cons_printf ("%d", bit? 1: 0);
}
r_cons_print (" ");
}
r_cons_newline ();
char str_pos[128];
memset (str_pos, '-', nbits + 9);
int pos = x;
if (pos > 31) {
pos += 2;
}
str_pos[pos + (x / 8)] = '^';
str_pos[nbits + 9] = 0;
str_pos[8] = ' ';
str_pos[17] = ' ';
str_pos[26] = ' ';
str_pos[35] = ' ';
str_pos[36] = ' ';
str_pos[37] = ' ';
str_pos[46] = ' ';
str_pos[55] = ' ';
str_pos[64] = ' ';
r_cons_printf ("pos: %s\n", str_pos);
r_cons_newline ();
r_cons_visual_flush ();
int ch = r_cons_readchar ();
if (ch == -1 || ch == 4) {
break;
}
if (ch != 10) {
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
}
switch (ch) {
case 'Q':
case 'q':
{
char *op_hex = r_asm_op_get_hex (&asmop);
char *res = r_print_hexpair (core->print, op_hex, -1);
r_core_cmdf (core, "wx %02x%02x%02x%02x", buf[0], buf[1], buf[2], buf[3]);
free (res);
free (op_hex);
}
return false;
case 'H':
{
int y = R_MAX (x - 8, 0);
x = y - y%8;
}
break;
case 'L':
case 9:
{
int y = R_MIN (x + 8, nbits - 8);
x = y - y%8;
}
break;
case 'j':
case 'k':
case 10:
case ' ':
//togglebit();
{
const int nbyte = x / 8;
const int nbit = 7 - (x - (nbyte * 8));
ut8 *byte = buf + nbyte;
bool bit = R_BIT_CHK (byte, nbit);
if (bit) {
R_BIT_UNSET (byte, nbit);
} else {
R_BIT_SET (byte, nbit);
}
}
break;
case '>':
buf[x/8] = rotate_nibble (buf [(x / 8)], -1);
break;
case '<':
buf[x/8] = rotate_nibble (buf [(x / 8)], 1);
break;
case 'i':
{
r_line_set_prompt ("> ");
const char *line = r_line_readline ();
ut64 num = r_num_math (core->num, line);
if (num || (!num && *line == '0')) {
buf[x/8] = num;
}
}
break;
case 'R':
if (r_config_get_i (core->config, "scr.randpal")) {
r_core_cmd0 (core, "ecr");
} else {
r_core_cmd0 (core, "ecn");
}
break;
case '+':
buf[(x/8)]++;
break;
case '-':
buf[(x/8)]--;
break;
case 'h':
x = R_MAX (x - 1, 0);
break;
case 'l':
x = R_MIN (x + 1, nbits - 1);
break;
case '?':
r_cons_clear00 ();
r_cons_printf (
"Vd1?: Visual Bit Editor Help:\n\n"
" q - quit the bit editor\n"
" R - randomize color palette\n"
" j/k - toggle bit value (same as space key)\n"
" h/l - select next/previous bit\n"
" +/- - increment or decrement byte value\n"
" </> - rotate left/right byte value\n"
" i - insert numeric value of byte\n"
" : - enter command\n");
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':': // TODO: move this into a separate helper function
{
char cmd[1024];
r_cons_show_cursor (true);
r_cons_set_raw (0);
cmd[0]='\0';
r_line_set_prompt (":> ");
if (r_cons_fgets (cmd, sizeof (cmd) - 1, 0, NULL) < 0) {
cmd[0] = '\0';
}
r_core_cmd (core, cmd, 1);
r_cons_set_raw (1);
r_cons_show_cursor (false);
if (cmd[0]) {
r_cons_any_key (NULL);
}
r_cons_clear ();
}
break;
}
}
return true;
}
// belongs to r_core_visual_types
static int sdbforcb (void *p, const char *k, const char *v) {
const char *pre = " ";
RCoreVisualTypes *vt = (RCoreVisualTypes*)p;
bool use_color = vt->core->print->flags & R_PRINT_FLAGS_COLOR;
char *color_sel = vt->core->cons->context->pal.prompt;
if (vt->optword) {
if (!strcmp (vt->type, "struct")) {
char *s = r_str_newf ("struct.%s.", vt->optword);
/* enum */
if (!strncmp (s, k, strlen (s))) {
if (vt->t_idx == vt->t_ctr) {
free (vt->curname);
vt->curname = strdup (k);
free (vt->curfmt);
vt->curfmt = strdup (v);
pre = ">";
}
if (use_color && *pre == '>') {
r_cons_printf ("%s %s %s %s %10s\n", color_sel,
Color_RESET, pre, k+strlen (s), v);
} else {
r_cons_printf (" %s %s %10s\n",
pre, k + strlen (s), v);
}
vt->t_ctr ++;
}
free (s);
} else {
char *s = r_str_newf ("%s.", vt->optword);
/* enum */
if (!strncmp (s, k, strlen (s))) {
if (!strstr (k, ".0x")) {
if (vt->t_idx == vt->t_ctr) {
free (vt->curname);
vt->curname = strdup (v);
free (vt->curfmt);
vt->curfmt = strdup (v);
pre = ">";
}
if (use_color && *pre=='>') {
r_cons_printf ("%s %s %s %s\n", color_sel,
Color_RESET, pre, k, v);
} else {
r_cons_printf (" %s %s %s\n",
pre, k, v);
}
vt->t_ctr ++;
}
}
free (s);
}
} else if (!strcmp (v, vt->type)) {
if (!strcmp (vt->type, "type")) {
char *fmt = getformat (vt, k);
if (vt->t_idx == vt->t_ctr) {
free (vt->curname);
vt->curname = strdup (k);
free (vt->curfmt);
vt->curfmt = strdup (fmt);
pre = ">";
}
if (use_color && *pre=='>') {
r_cons_printf ("%s %s pf %3s %s\n"Color_RESET,
color_sel, pre, fmt, k);
} else {
r_cons_printf (" %s pf %3s %s\n",
pre, fmt, k);
}
free (fmt);
} else {
if (vt->t_idx == vt->t_ctr) {
free (vt->curname);
vt->curname = strdup (k);
free (vt->curfmt);
vt->curfmt = strdup (v);
pre = ">";
}
if (use_color && *pre == '>') {
r_cons_printf ("%s %s %s\n"Color_RESET, color_sel,
(vt->t_idx == vt->t_ctr)?
">":" ", k);
} else {
r_cons_printf (" %s %s\n",
(vt->t_idx == vt->t_ctr)?
">":" ", k);
}
}
vt->t_ctr ++;
}
return 1;
}
R_API int r_core_visual_types(RCore *core) {
RCoreVisualTypes vt = {core, 0, 0};
int i, ch;
int _option = 0;
int option = 0;
char *txt;
char cmd[1024];
int menu = 0;
int h_opt = 0;
char *optword = NULL;
const char *opts[] = {
"type",
"enum",
"struct",
"func",
"union",
"cc",
NULL
};
bool use_color = core->print->flags & R_PRINT_FLAGS_COLOR;
if (r_flag_space_is_empty (core->flags)) {
menu = 1;
}
for (;;) {
r_cons_clear00 ();
for (i = 0; opts[i]; i++) {
if (use_color) {
if (h_opt == i) {
r_cons_printf ("%s[%s]%s ", core->cons->context->pal.call,
opts[i], Color_RESET);
} else {
r_cons_printf ("%s%s%s ", core->cons->context->pal.other,
opts[i], Color_RESET);
}
} else {
r_cons_printf (h_opt == i ? "[%s] " : " %s ", opts[i]);
}
}
r_cons_newline ();
if (optword) {
r_cons_printf (">> %s\n", optword);
}
if (!strcmp (opts[h_opt], "cc")) {
// XXX TODO: make this work (select with cursor, to delete, or add a new one with 'i', etc)
r_core_cmdf (core, "tfcl");
} else {
vt.t_idx = option;
vt.t_ctr = 0;
vt.type = opts[h_opt];
vt.optword = optword;
sdb_foreach (core->anal->sdb_types, sdbforcb, &vt);
}
r_cons_visual_flush ();
ch = r_cons_readchar ();
if (ch == -1 || ch == 4) {
return false;
}
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case 'h':
h_opt--;
if (h_opt < 0) {
h_opt = 0;
}
option = 0;
R_FREE (optword);
break;
case 'l':
h_opt++;
option = 0;
if (!opts[h_opt]) {
h_opt--;
}
R_FREE (optword);
break;
case 'o':
{
char *file = prompt ("Filename: ", NULL);
if (file) {
r_core_cmdf (core, "\"to %s\"", file);
free (file);
}
}
break;
case 'j':
if (++option >= vt.t_ctr) {
option = vt.t_ctr - 1;
}
break;
case 'J':
option += 10;
if (option >= vt.t_ctr) {
option = vt.t_ctr-1;
}
break;
case 'k':
if (--option < 0) {
option = 0;
}
break;
case 'K':
option -= 10;
if (option < 0) {
option = 0;
}
break;
case 'b':
r_core_cmdf (core, "tl %s", vt.curname);
break;
case -1: // EOF
case 'Q':
case 'q':
if (optword) {
R_FREE (optword);
break;
}
if (menu <= 0) {
return true;
}
menu--;
option = _option;
if (menu==0) {
// if no flagspaces, just quit
if (r_flag_space_is_empty (core->flags)) {
return true;
}
}
break;
case 'a':
{
txt = prompt ("add C type: ", NULL);
if (txt) {
r_core_cmdf (core, "\"td %s\"", txt);
free (txt);
}
}
break;
case 'd':
r_core_cmdf (core, "t- %s", vt.curname);
break;
case '-':
r_core_cmd0 (core, "to -");
break;
case ' ':
case '\r':
case '\n':
case 'e':
if (optword) {
/* TODO: edit field */
} else {
switch (h_opt) {
case 0: // type
/* TODO: do something with this data */
prompt ("name: ", vt.curname);
prompt ("pf: ", vt.curfmt);
break;
case 1: // enum
case 2: // struct
free (optword);
if (vt.curname) {
optword = strdup (vt.curname);
} else {
optword = NULL;
}
break;
default:
break;
}
}
break;
case '?':
r_cons_clear00 ();
r_cons_printf (
"Vt?: Visual Types Help:\n\n"
" q - quit menu\n"
" j/k - down/up keys\n"
" h/l - left-right\n"
" a - add new type (C syntax)\n"
" b - bind type to current offset\n"
" d - delete current type\n"
" e - edit current type\n"
" o - open .h include file\n"
" - - Open cfg.editor to load types\n"
" : - enter command\n");
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':':
r_cons_show_cursor (true);
r_cons_set_raw (0);
cmd[0] = '\0';
r_line_set_prompt (":> ");
if (r_cons_fgets (cmd, sizeof (cmd) - 1, 0, NULL) < 0) {
cmd[0]='\0';
}
r_core_cmd (core, cmd, 1);
r_cons_set_raw (1);
r_cons_show_cursor (false);
if (cmd[0]) {
r_cons_any_key (NULL);
}
r_cons_clear ();
continue;
}
}
return true;
}
static int cmtcb(void *usr, const char *k, const char *v) {
if (!strncmp (k, "meta.C.", 7)) {
RList *list = (RList*)usr;
char *msg, *comma = strchr (v, ',');
if (comma) {
comma = strchr (comma + 1, ',');
if (comma) {
msg = (char *)sdb_decode (comma + 1, NULL);
if (msg) {
msg = r_str_replace (msg, "\n", "", true);
r_list_append (list, r_str_newf ("%s %s", k+7, msg));
free (msg);
}
}
}
}
return 1;
}
R_API bool r_core_visual_hudclasses(RCore *core) {
RListIter *iter, *iter2;
RBinClass *c;
RBinField *f;
RBinSymbol *m;
ut64 addr;
char *res;
RList *list = r_list_new ();
if (!list) {
return false;
}
list->free = free;
RList *classes = r_bin_get_classes (core->bin);
r_list_foreach (classes, iter, c) {
r_list_foreach (c->fields, iter2, f) {
r_list_append (list, r_str_newf ("0x%08"PFMT64x" %s %s",
f->vaddr, c->name, f->name));
}
r_list_foreach (c->methods, iter2, m) {
const char *name = m->dname? m->dname: m->name;
r_list_append (list, r_str_newf ("0x%08"PFMT64x" %s %s",
m->vaddr, c->name, name));
}
}
res = r_cons_hud (list, NULL);
if (res) {
char *p = strchr (res, ' ');
if (p) {
*p = 0;
}
addr = r_num_get (NULL, res);
r_core_seek (core, addr, true);
free (res);
}
r_list_free (list);
return res? true: false;
}
static bool hudstuff_append(RFlagItem *fi, void *user) {
RList *list = (RList *)user;
char *s = r_str_newf ("0x%08"PFMT64x" %s", fi->offset, fi->name);
if (s) {
r_list_append (list, s);
}
return true;
}
R_API bool r_core_visual_hudstuff(RCore *core) {
ut64 addr;
char *res;
RList *list = r_list_new ();
if (!list) {
return false;
}
list->free = free;
r_flag_foreach (core->flags, hudstuff_append, list);
sdb_foreach (core->anal->sdb_meta, cmtcb, list);
res = r_cons_hud (list, NULL);
if (res) {
char *p = strchr (res, ' ');
if (p) {
*p = 0;
}
addr = r_num_get (NULL, res);
r_core_seek (core, addr, true);
free (res);
}
r_list_free (list);
return res? true: false;
}
static bool r_core_visual_config_hud(RCore *core) {
RListIter *iter;
RConfigNode *bt;
RList *list = r_list_new ();
if (!list) {
return false;
}
char *res;
list->free = free;
r_list_foreach (core->config->nodes, iter, bt) {
r_list_append (list, r_str_newf ("%s %s", bt->name, bt->value));
}
res = r_cons_hud (list, NULL);
if (res) {
const char *oldvalue = NULL;
char cmd[512];
char *p = strchr (res, ' ');
if (p) {
*p = 0;
}
oldvalue = r_config_get (core->config, res);
r_cons_show_cursor (true);
r_cons_set_raw (0);
cmd[0] = '\0';
eprintf ("set new value for %s (old=%s)\n", res, oldvalue);
r_line_set_prompt (":> ");
if (r_cons_fgets (cmd, sizeof (cmd) - 1, 0, NULL) < 0) {
cmd[0] = '\0';
}
r_config_set (core->config, res, cmd);
r_cons_set_raw (1);
r_cons_show_cursor (false);
}
r_list_free (list);
return true;
}
// TODO: skip N first elements
// TODO: show only N elements of the list
// TODO: wrap index when out of boundaries
// TODO: Add support to show class fields too
static void *show_class(RCore *core, int mode, int *idx, RBinClass *_c, const char *grep, RList *list) {
bool show_color = r_config_get_i (core->config, "scr.color");
RListIter *iter;
RBinClass *c, *cur = NULL;
RBinSymbol *m, *mur = NULL;
RBinField *f, *fur = NULL;
int i = 0;
int skip = *idx - 10;
switch (mode) {
case 'c':
r_cons_printf ("[hjkl_/Cfm]> classes:\n\n");
r_list_foreach (list, iter, c) {
if (grep) {
if (!r_str_casestr (c->name, grep)) {
i++;
continue;
}
} else {
if (*idx > 10) {
skip--;
if (skip > 0) {
i++;
continue;
}
}
}
if (show_color) {
if (i == *idx) {
const char *clr = Color_BLUE;
r_cons_printf (Color_GREEN ">>" Color_RESET " %02d %s0x%08"
PFMT64x Color_YELLOW " %s\n" Color_RESET,
i, clr, c->addr, c->name);
} else {
r_cons_printf ("- %02d %s0x%08"PFMT64x Color_RESET" %s\n",
i, core->cons->context->pal.offset, c->addr, c->name);
}
} else {
r_cons_printf ("%s %02d 0x%08"PFMT64x" %s\n",
(i==*idx)?">>":"- ", i, c->addr, c->name);
}
if (i++ == *idx) {
cur = c;
}
}
if (!cur) {
*idx = i - 1;
if (r_list_empty (list)) {
return NULL;
}
r_cons_clear00 ();
return show_class (core, mode, idx, _c, grep, list);
}
return cur;
case 'f':
// show fields
r_cons_printf ("[hjkl_/cFm]> fields of %s:\n\n", _c->name);
r_list_foreach (_c->fields, iter, f) {
const char *name = f->name;
if (grep) {
if (!r_str_casestr (name, grep)) {
i++;
continue;
}
} else {
if (*idx > 10) {
skip--;
if (skip > 0) {
i++;
continue;
}
}
}
char *mflags = strdup ("");
if (r_str_startswith (name, _c->name)) {
name += strlen (_c->name);
}
if (show_color) {
if (i == *idx) {
const char *clr = Color_BLUE;
r_cons_printf (Color_GREEN ">>" Color_RESET " %02d %s0x%08"
PFMT64x Color_YELLOW " %s %s\n" Color_RESET,
i, clr, f->vaddr, mflags, name);
} else {
r_cons_printf ("- %02d %s0x%08"PFMT64x Color_RESET" %s %s\n",
i, core->cons->context->pal.offset, f->vaddr, mflags, name);
}
} else {
r_cons_printf ("%s %02d 0x%08"PFMT64x" %s %s\n",
(i==*idx)? ">>": "- ", i, f->vaddr, mflags, name);
}
R_FREE (mflags);
if (i++ == *idx) {
fur = f;
}
}
if (!fur) {
*idx = i - 1;
if (r_list_empty (_c->fields)) {
return NULL;
}
r_cons_clear00 ();
return show_class (core, mode, idx, _c, grep, list);
}
return fur;
break;
case 'm':
// show methods
if (!_c) {
eprintf ("No class selected.\n");
return mur;
}
r_cons_printf ("[hjkl_/cfM]> methods of %s\n\n", _c->name);
r_list_foreach (_c->methods, iter, m) {
const char *name = m->dname? m->dname: m->name;
char *mflags;
if (grep) {
if (!r_str_casestr (name, grep)) {
i++;
continue;
}
} else {
if (*idx > 10) {
skip--;
if (skip > 0) {
i++;
continue;
}
}
}
mflags = r_core_bin_method_flags_str (m->method_flags, 0);
if (show_color) {
if (r_str_startswith (name, _c->name)) {
name += strlen (_c->name);
}
if (i == *idx) {
const char *clr = Color_BLUE;
r_cons_printf (Color_GREEN ">>" Color_RESET " %02d %s0x%08"
PFMT64x Color_YELLOW " %s %s\n" Color_RESET,
i, clr, m->vaddr, mflags, name);
} else {
r_cons_printf ("- %02d %s0x%08"PFMT64x Color_RESET" %s %s\n",
i, core->cons->context->pal.offset, m->vaddr, mflags, name);
}
} else {
r_cons_printf ("%s %02d 0x%08"PFMT64x" %s %s\n",
(i==*idx)? ">>": "- ", i, m->vaddr, mflags, name);
}
R_FREE (mflags);
if (i++ == *idx) {
mur = m;
}
}
if (!mur) {
*idx = i - 1;
if (r_list_empty (_c->methods)) {
return NULL;
}
r_cons_clear00 ();
return show_class (core, mode, idx, _c, grep, list);
}
return mur;
}
return NULL;
}
R_API int r_core_visual_classes(RCore *core) {
int ch, index = 0;
char cmd[1024];
int mode = 'c';
RBinClass *cur = NULL;
RBinSymbol *mur = NULL;
RBinField *fur = NULL;
void *ptr;
int oldcur = 0;
char *grep = NULL;
bool grepmode = false;
RList *list = r_bin_get_classes (core->bin);
if (r_list_empty (list)) {
r_cons_message ("No Classes");
return false;
}
for (;;) {
int cols;
r_cons_clear00 ();
if (grepmode) {
r_cons_printf ("Grep: %s\n", grep? grep: "");
}
ptr = show_class (core, mode, &index, cur, grep, list);
switch (mode) {
case 'f':
fur = (RBinField*)ptr;
break;
case 'm':
mur = (RBinSymbol*)ptr;
break;
case 'c':
cur = (RBinClass*)ptr;
break;
}
/* update terminal size */
(void) r_cons_get_size (&cols);
r_cons_visual_flush ();
ch = r_cons_readchar ();
if (ch == -1 || ch == 4) {
R_FREE (grep);
return false;
}
if (grepmode) {
switch (ch) {
case 127:
if (grep) {
int len = strlen (grep);
if (len < 1) {
grepmode = false;
} else {
grep[len - 1] = 0;
}
}
break;
case ' ':
case '\r':
case '\n':
R_FREE (grep);
grepmode = false;
break;
default:
grep = grep
? r_str_appendf (grep, "%c", ch)
: r_str_newf ("%c", ch);
break;
}
continue;
}
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case 'C':
r_config_toggle (core->config, "scr.color");
break;
case '_':
if (r_core_visual_hudclasses (core)) {
return true;
}
break;
case 'J': index += 10; break;
case 'j': index++; break;
case 'k':
if (--index < 0) {
index = 0;
}
break;
case 'K':
index -= 10;
if (index < 0) {
index = 0;
}
break;
case 'g':
index = 0;
break;
case 'G':
index = r_list_length (list) - 1;
break;
case 'i':
{
char *num = prompt ("Index:", NULL);
if (num) {
index = atoi (num);
free (num);
}
}
break;
case 'p':
if (mode == 'm' && mur) {
r_core_seek (core, mur->vaddr, true);
r_core_cmd0 (core, "af;pdf~..");
}
break;
case 'm': // methods
mode = 'm';
break;
case 'f': // fields
mode = 'f';
break;
case 'h':
case 127: // backspace
case 'b': // back
case 'Q':
case 'c':
case 'q':
if (mode == 'c') {
return true;
}
mode = 'c';
index = oldcur;
break;
case '/':
grepmode = true;
break;
case 'l':
case ' ':
case '\r':
case '\n':
if (mur && mode == 'm') {
r_core_seek (core, mur->vaddr, true);
return true;
}
if (fur) {
r_core_seek (core, fur->vaddr, true);
return true;
}
if (cur) {
oldcur = index;
index = 0;
mode = 'm';
}
break;
case '?':
r_cons_clear00 ();
r_cons_printf (
"\nVF: Visual Classes help:\n\n"
" q - quit menu\n"
" j/k - down/up keys\n"
" h/b - go back\n"
" g/G - go first/last item\n"
" i - specify index"
" / - grep mode\n"
" C - toggle colors\n"
" f - show class fields\n"
" m - show class methods\n"
" l/' ' - accept current selection\n"
" p - preview method disasm with less\n"
" : - enter command\n");
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':':
r_cons_show_cursor (true);
r_cons_set_raw (0);
cmd[0] = '\0';
r_line_set_prompt (":> ");
if (r_cons_fgets (cmd, sizeof (cmd)-1, 0, NULL) < 0) {
cmd[0]='\0';
}
//line[strlen(line)-1]='\0';
r_core_cmd (core, cmd, 1);
r_cons_set_raw (1);
r_cons_show_cursor (false);
if (cmd[0]) {
r_cons_any_key (NULL);
}
//cons_gotoxy(0,0);
r_cons_clear ();
break;
}
}
return true;
}
static int flag_name_sort(const void *a, const void *b) {
const RFlagItem *fa = (const RFlagItem *)a;
const RFlagItem *fb = (const RFlagItem *)b;
return strcmp (fa->name, fb->name);
}
static int flag_offset_sort(const void *a, const void *b) {
const RFlagItem *fa = (const RFlagItem *)a;
const RFlagItem *fb = (const RFlagItem *)b;
if (fa->offset < fb->offset) {
return -1;
}
if (fa->offset > fb->offset) {
return 1;
}
return 0;
}
static void sort_flags(RList *l, int sort) {
switch (sort) {
case SORT_NAME:
r_list_sort (l, flag_name_sort);
break;
case SORT_OFFSET:
r_list_sort (l, flag_offset_sort);
break;
case SORT_NONE:
default:
break;
}
}
// TODO: remove this statement, should be a separate .o
static char *print_rop(void *_core, void *_item, bool selected) {
char *line = _item;
// TODO: trim if too long
return r_str_newf ("%c %s\n", selected?'>':' ', line);
}
R_API int r_core_visual_view_rop(RCore *core) {
RListIter *iter;
const int rows = 7;
int cur = 0;
r_line_set_prompt ("rop regexp: ");
const char *line = r_line_readline ();
int scr_h, scr_w = r_cons_get_size (&scr_h);
if (!line || !*line) {
return false;
}
// maybe store in RCore, so we can save it in project and use it outside visual
eprintf ("Searching ROP gadgets...\n");
char *ropstr = r_core_cmd_strf (core, "\"/Rl %s\" @e:scr.color=0", line);
RList *rops = r_str_split_list (ropstr, "\n", 0);
int delta = 0;
bool show_color = core->print->flags & R_PRINT_FLAGS_COLOR;
bool forceaddr = false;
ut64 addr = UT64_MAX;
char *cursearch = strdup (line);
while (true) {
r_cons_clear00 ();
r_cons_printf ("[0x%08"PFMT64x"]-[visual-r2rop] %s (see pdp command)\n",
(addr == UT64_MAX)? 0: addr + delta, cursearch);
// compute chain
RStrBuf *sb = r_strbuf_new ("");
char *msg;
r_list_foreach (core->ropchain, iter, msg) {
if (core->assembler->bits == 64) {
ut64 n = r_num_get (NULL, msg);
n = r_read_be64 (&n);
r_strbuf_appendf (sb, "%016"PFMT64x, n);
} else {
ut32 n = r_num_get (NULL, msg);
n = r_read_be32 (&n);
r_strbuf_appendf (sb, "%08x", n);
}
}
char *chainstr = r_strbuf_drain (sb);
char *wlist = r_str_widget_list (core, rops, rows, cur, print_rop);
r_cons_printf ("%s", wlist);
free (wlist);
char *curline = r_str_dup (NULL, r_str_trim_ro (r_str_widget_list (
core, rops, rows, cur, print_rop)));
if (curline) {
char *sp = strchr (curline, ' ');
if (sp) {
*sp = 0;
if (!forceaddr) {
addr = r_num_math (NULL, curline);
}
*sp = ' ';
}
if (addr != UT64_MAX) {
r_cons_printf ("Gadget:");
// get comment
char *output = r_core_cmd_strf (core, "piu 10 @ 0x%08"PFMT64x, addr + delta);
if (output) {
r_cons_strcat_at (output, 0, 10, scr_w, 10);
free (output);
}
}
}
int count = 0;
r_cons_flush ();
r_cons_gotoxy (0, 20);
r_cons_printf ("ROPChain:\n %s\n", chainstr? chainstr: "");
r_list_foreach (core->ropchain, iter, msg) {
int extra = strlen (chainstr) / scr_w;
r_cons_gotoxy (0, extra + 22 + count);
r_cons_strcat (msg);
char *cmt = r_meta_get_string (core->anal, R_META_TYPE_COMMENT, r_num_get (NULL, msg));
if (cmt) {
r_cons_strcat (cmt);
free (cmt);
}
count ++;
}
r_cons_flush ();
int ch = r_cons_readchar ();
if (ch == -1 || ch == 4) {
free (curline);
free (cursearch);
R_FREE (chainstr);
return false;
}
#define NEWTYPE(x,y) r_mem_dup (&(y), sizeof (x));
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case 127:
free (r_list_pop (core->ropchain));
break;
case '?':
r_cons_clear00 ();
r_cons_printf ("[r2rop-visual] Help\n"
" jk - select next/prev rop gadget\n"
" JK - scroll next/prev page from list\n"
" hl - increase/decrease delta offset in disasm\n"
" \\n - enter key or dot will add the current offset into the chain\n"
" i - enter a number to be pushed into the chain\n"
" : - run r2 command\n"
" ; - add comment in current offset\n"
" <- - backspace - delete last gadget from the chain\n"
" / - highlight given word\n"
" y - yank current rop chain into the clipboard (y?)\n"
" o - seek to given offset\n"
" r - run /R again\n"
" ? - show this help message\n"
" q - quit this view\n"
);
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':': // TODO: move this into a separate helper function
r_cons_show_cursor (true);
r_cons_set_raw (0);
while (true) {
char cmd[1024];
cmd[0] = '\0';
r_line_set_prompt (":> ");
if (r_cons_fgets (cmd, sizeof (cmd) - 1, 0, NULL) < 0) {
cmd[0] = '\0';
}
if (!*cmd || *cmd == 'q') {
break;
}
ut64 oseek = core->offset;
r_core_seek (core, addr + delta, 0);
r_core_cmd (core, cmd, 1);
r_core_seek (core, oseek, 0);
r_cons_flush ();
}
r_cons_set_raw (1);
r_cons_show_cursor (false);
break;
case 'y':
r_core_cmdf (core, "yfx %s", chainstr);
break;
case 'o':
{
r_line_set_prompt ("offset: ");
const char *line = r_line_readline ();
if (line && *line) {
ut64 off = r_num_math (core->num, line);
r_core_seek (core, off, 1);
addr = off;
forceaddr = true;
delta = 0;
}
}
break;
case 'r':
{
r_line_set_prompt ("rop regexp: ");
const char *line = r_line_readline ();
if (line && *line) {
free (cursearch);
delta = 0;
addr = UT64_MAX;
cur = 0;
cursearch = strdup (line);
free (ropstr);
ropstr = r_core_cmd_strf (core, "\"/Rl %s\" @e:scr.color=0", line);
r_list_free (rops);
rops = r_str_split_list (ropstr, "\n", 0);
}
}
break;
case '/':
r_core_cmd0 (core, "?i highlight;e scr.highlight=`yp`");
break;
case 'i':
{
r_line_set_prompt ("insert value: ");
const char *line = r_line_readline ();
if (line && *line) {
ut64 n = r_num_math (core->num, line);
r_list_push (core->ropchain, r_str_newf ("0x%08"PFMT64x, n));
}
}
break;
case ';':
{
r_line_set_prompt ("comment: ");
const char *line = r_line_readline ();
if (line && *line) {
// XXX code injection bug here
r_core_cmdf (core, "CC %s @ 0x%08"PFMT64x, line, addr + delta);
}
}
break;
case '.':
case '\n':
case '\r':
if (curline && *curline) {
char *line = r_core_cmd_strf (core, "piuq@0x%08"PFMT64x, addr + delta);
r_str_replace_char (line, '\n', ';');
if (show_color) {
// XXX parsing fails to read this ansi-offset
// const char *offsetColor = r_cons_singleton ()->context->pal.offset; // TODO etooslow. must cache
// r_list_push (core->ropchain, r_str_newf ("%s0x%08"PFMT64x""Color_RESET" %s", offsetColor, addr + delta, line));
r_list_push (core->ropchain, r_str_newf ("0x%08"PFMT64x" %s", addr + delta, line));
} else {
r_list_push (core->ropchain, r_str_newf ("0x%08"PFMT64x" %s", addr + delta, line));
}
free (line);
}
break;
case 'h':
delta--;
break;
case 'l':
delta++;
break;
case 'J':
cur+=10;
forceaddr = false;
delta = 0;
break;
case 'K':
delta = 0;
forceaddr = false;
if (cur > 10) {
cur-=10;
} else {
cur = 0;
}
break;
case '0':
delta = 0;
cur = 0;
break;
case 'j':
delta = 0;
cur++;
forceaddr = false;
break;
case 'k':
delta = 0;
forceaddr = false;
if (cur > 0) {
cur--;
} else {
cur = 0;
}
break;
case 'q':
free (curline);
free (cursearch);
R_FREE (chainstr);
return true;
}
R_FREE (chainstr);
free (curline);
}
free (cursearch);
return false;
}
R_API int r_core_visual_trackflags(RCore *core) {
const char *fs = NULL, *fs2 = NULL;
int hit, i, j, ch;
int _option = 0;
int option = 0;
char cmd[1024];
int format = 0;
int delta = 7;
int menu = 0;
int sort = SORT_NONE;
if (r_flag_space_is_empty (core->flags)) {
menu = 1;
}
for (;;) {
bool hasColor = r_config_get_i (core->config, "scr.color");
r_cons_clear00 ();
if (menu) {
r_cons_printf ("Flags in flagspace '%s'. Press '?' for help.\n\n",
r_flag_space_cur_name (core->flags));
hit = 0;
i = j = 0;
RList *l = r_flag_all_list (core->flags, true);
RListIter *iter;
RFlagItem *fi;
sort_flags (l, sort);
r_list_foreach (l, iter, fi) {
if (option == i) {
fs2 = fi->name;
hit = 1;
}
if ((i>=option-delta) && ((i<option+delta)||((option<delta)&&(i<(delta<<1))))) {
bool cur = option == i;
if (cur && hasColor) {
r_cons_printf (Color_INVERT);
}
r_cons_printf (" %c %03d 0x%08"PFMT64x" %4"PFMT64d" %s\n",
cur?'>':' ', i, fi->offset, fi->size, fi->name);
if (cur && hasColor) {
r_cons_printf (Color_RESET);
}
j++;
}
i++;
}
r_list_free (l);
if (!hit && i > 0) {
option = i - 1;
continue;
}
if (fs2) {
int cols, rows = r_cons_get_size (&cols);
//int rows = 20;
rows -= 12;
r_cons_printf ("\n Selected: %s\n\n", fs2);
// Honor MAX_FORMATS here
switch (format) {
case 0: snprintf (cmd, sizeof (cmd), "px %d @ %s!64", rows*16, fs2); core->printidx = 0; break;
case 1: snprintf (cmd, sizeof (cmd), "pd %d @ %s!64", rows, fs2); core->printidx = 1; break;
case 2: snprintf (cmd, sizeof (cmd), "ps @ %s!64", fs2); core->printidx = 5; break;
case 3: strcpy (cmd, "f="); break;
default: format = 0; continue;
}
if (*cmd) {
r_core_cmd (core, cmd, 0);
}
} else {
r_cons_printf ("(no flags)\n");
}
} else {
r_cons_printf ("Flag spaces:\n\n");
hit = 0;
RSpaceIter it;
const RSpace *s, *cur = r_flag_space_cur (core->flags);
int i = 0;
r_flag_space_foreach (core->flags, it, s) {
if (option == i) {
fs = s->name;
hit = 1;
}
if ((i >= option - delta) && ((i < option + delta) ||
((option < delta) && (i < (delta << 1))))) {
r_cons_printf (" %c %c %s\n",
(option == i)? '>': ' ',
(s == cur)? '*': ' ',
s->name);
}
i++;
}
if (option == i) {
fs = "*";
hit = 1;
}
r_cons_printf (" %c %c %s\n", (option == i)? '>': ' ',
!cur? '*': ' ', "*");
i++;
if (!hit && i > 0) {
option = i - 1;
continue;
}
}
r_cons_visual_flush ();
ch = r_cons_readchar ();
if (ch == -1 || ch == 4) {
return false;
}
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case 'C':
r_config_toggle (core->config, "scr.color");
break;
case '_':
if (r_core_visual_hudstuff (core)) {
return true;
}
break;
case 'J': option += 10; break;
case 'o': sort = SORT_OFFSET; break;
case 'n': sort = SORT_NAME; break;
case 'j': option++; break;
case 'k':
if (--option < 0) {
option = 0;
}
break;
case 'K': option-=10;
if (option < 0) {
option = 0;
}
break;
case 'h':
case 'b': // back
case 'Q':
case 'q':
if (menu <= 0) {
return true;
}
menu--;
option = _option;
if (menu == 0) {
r_flag_space_set (core->flags, NULL);
// if no flagspaces, just quit
if (r_flag_space_is_empty (core->flags)) {
return true;
}
}
break;
case 'a':
switch (menu) {
case 0: // new flag space
r_cons_show_cursor (true);
r_line_set_prompt ("add flagspace: ");
strcpy (cmd, "fs ");
if (r_cons_fgets (cmd+3, sizeof (cmd)-4, 0, NULL) > 0) {
r_core_cmd (core, cmd, 0);
r_cons_set_raw (1);
r_cons_show_cursor (false);
}
break;
case 1: // new flag
r_cons_show_cursor (true);
r_line_set_prompt ("add flag: ");
strcpy (cmd, "f ");
if (r_cons_fgets (cmd+2, sizeof (cmd)-3, 0, NULL) > 0) {
r_core_cmd (core, cmd, 0);
r_cons_set_raw (1);
r_cons_show_cursor (false);
}
break;
}
break;
case 'd':
r_flag_unset_name (core->flags, fs2);
break;
case 'e':
/* TODO: prompt for addr, size, name */
eprintf ("TODO\n");
r_sys_sleep (1);
break;
case '*':
r_core_block_size (core, core->blocksize+16);
break;
case '/':
r_core_block_size (core, core->blocksize-16);
break;
case '+':
if (menu == 1) {
r_core_cmdf (core, "f %s=%s+1", fs2, fs2);
} else {
r_core_block_size (core, core->blocksize + 1);
}
break;
case '-':
if (menu == 1) {
r_core_cmdf (core, "f %s=%s-1", fs2, fs2);
} else {
r_core_block_size (core, core->blocksize-1);
}
break;
case 'r': // "Vtr"
if (menu == 1) {
int len;
r_cons_show_cursor (true);
r_cons_set_raw (0);
// TODO: use r_flag_rename or wtf?..fr doesnt uses this..
snprintf (cmd, sizeof (cmd), "fr %s ", fs2);
len = strlen (cmd);
eprintf ("Rename flag '%s' as:\n", fs2);
r_line_set_prompt (":> ");
if (r_cons_fgets (cmd + len, sizeof (cmd) - len - 1, 0, NULL) < 0) {
cmd[0] = '\0';
}
r_core_cmd (core, cmd, 0);
r_cons_set_raw (1);
r_cons_show_cursor (false);
}
break;
case 'R':
if (menu == 1) {
char line[1024];
r_cons_show_cursor (true);
r_cons_set_raw (0);
eprintf ("Rename function '%s' as:\n", fs2);
r_line_set_prompt (":> ");
if (r_cons_fgets (line, sizeof (line), 0, NULL) < 0) {
cmd[0] = '\0';
}
int res = snprintf (cmd, sizeof (cmd), "afr %s %s", line, fs2);
if (res < sizeof (cmd)) {
r_core_cmd (core, cmd, 0);
}
r_cons_set_raw (1);
r_cons_show_cursor (false);
}
break;
case 'P':
if (--format < 0) {
format = MAX_FORMAT;
}
break;
// = (format<=0)? MAX_FORMAT: format-1; break;
case 'p': format++; break;
case 'l':
case ' ':
case '\r':
case '\n':
if (menu == 1) {
sprintf (cmd, "s %s", fs2);
r_core_cmd (core, cmd, 0);
return true;
}
r_flag_space_set (core->flags, fs);
menu = 1;
_option = option;
option = 0;
break;
case '?':
r_cons_clear00 ();
r_cons_printf (
"\nVF: Visual Flags help:\n\n"
" q - quit menu\n"
" j/k - line down/up keys\n"
" J/K - page down/up keys\n"
" h/b - go back\n"
" C - toggle colors\n"
" l/' ' - accept current selection\n"
" a/d/e - add/delete/edit flag\n"
" +/- - increase/decrease block size\n"
" o - sort flags by offset\n"
" r/R - rename flag / Rename function\n"
" n - sort flags by name\n"
" p/P - rotate print format\n"
" _ - hud for flags and comments\n"
" : - enter command\n");
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':':
r_cons_show_cursor (true);
r_cons_set_raw (0);
*cmd = 0;
r_line_set_prompt (":> ");
if (r_cons_fgets (cmd, sizeof (cmd) - 1, 0, NULL) <0) {
*cmd = 0;
}
cmd[sizeof (cmd) - 1] = 0;
r_core_cmd_task_sync (core, cmd, 1);
r_cons_set_raw (1);
r_cons_show_cursor (false);
if (*cmd) {
r_cons_any_key (NULL);
}
//cons_gotoxy(0,0);
r_cons_clear ();
continue;
}
}
return true;
}
static bool meta_deserialize(RAnal *a, RAnalMetaItem *it, const char *k, const char *v) {
if (strlen (k) < 8) {
return false;
}
if (memcmp (k + 6, ".0x", 3)) {
return false;
}
return r_meta_deserialize_val (a, it, k[5], sdb_atoi (k + 7), v);
}
static int meta_enumerate_cb(void *user, const char *k, const char *v) {
RAnalMetaUserItem *ui = user;
RList *list = ui->user;
RAnalMetaItem *it = R_NEW0 (RAnalMetaItem);
if (!it) {
return 0;
}
if (!meta_deserialize (ui->anal, it, k, v)) {
free (it);
goto beach;
}
if (!it->str) {
free (it);
goto beach;
}
r_list_append (list, it);
beach:
return 1;
}
R_API int r_core_visual_comments (RCore *core) {
char *str;
char cmd[512], *p = NULL;
int ch, option = 0;
int format = 0, i = 0;
ut64 addr, from = 0, size = 0;
for (;;) {
r_cons_clear00 ();
r_cons_strcat ("Comments:\n");
RList *items = r_list_newf (free);
RAnalMetaItem *item;
RListIter *iter;
r_meta_list_cb (core->anal, R_META_TYPE_COMMENT, 0, meta_enumerate_cb, items, UT64_MAX);
i = 0;
r_list_foreach (items, iter, item) {
str = item->str;
addr = item->from;
if (option==i) {
from = addr;
size = 1; // XXX: remove this thing size for comments is useless d->size;
free (p);
p = str;
r_cons_printf (" > %s\n", str);
} else {
r_cons_printf (" %s\n", str);
free (str);
}
i ++;
}
if (!i) {
if (--option < 0) {
r_cons_any_key ("No comments");
break;
}
continue;
}
r_cons_newline ();
switch (format) {
case 0: sprintf (cmd, "px @ 0x%"PFMT64x":64", from); core->printidx = 0; break;
case 1: sprintf (cmd, "pd 12 @ 0x%"PFMT64x":64", from); core->printidx = 1; break;
case 2: sprintf (cmd, "ps @ 0x%"PFMT64x":64", from); core->printidx = 5; break;
default: format = 0; continue;
}
if (*cmd) {
r_core_cmd (core, cmd, 0);
}
r_cons_visual_flush ();
ch = r_cons_readchar ();
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case 'a':
//TODO
break;
case 'e':
//TODO
break;
case 'd':
if (p) {
r_meta_del (core->anal, R_META_TYPE_ANY, from, size);
}
break;
case 'P':
if (--format < 0) {
format = MAX_FORMAT;
}
break;
case 'p':
format++;
break;
case 'J':
option += 10;
break;
case 'j':
option++;
break;
case 'k':
if (--option < 0) {
option = 0;
}
break;
case 'K':
option -= 10;
if (option < 0) {
option = 0;
}
break;
case 'l':
case ' ':
case '\r':
case '\n':
r_core_cmdf (core, "s 0x%"PFMT64x, from);
R_FREE (p);
return true;
case 'Q':
case 'q':
R_FREE (p);
return true;
case '?':
case 'h':
r_cons_clear00 ();
r_cons_printf (
"\nVT: Visual Comments/Anal help:\n\n"
" q - quit menu\n"
" j/k - down/up keys\n"
" h/b - go back\n"
" l/' ' - accept current selection\n"
" a/d/e - add/delete/edit comment/anal symbol\n"
" p/P - rotate print format\n");
r_cons_flush ();
r_cons_any_key (NULL);
break;
}
R_FREE (p);
}
return true;
}
static void config_visual_hit_i(RCore *core, const char *name, int delta) {
struct r_config_node_t *node;
node = r_config_node_get (core->config, name);
if (node && r_config_node_is_int (node)) {
int hitDelta = r_config_get_i (core->config, name) + delta;
(void) r_config_set_i (core->config, name, hitDelta);
}
}
/* Visually activate the config variable */
static void config_visual_hit(RCore *core, const char *name, int editor) {
char buf[1024];
RConfigNode *node;
if (!(node = r_config_node_get (core->config, name))) {
return;
}
if (r_config_node_is_bool (node)) {
r_config_set_i (core->config, name, node->i_value? 0:1);
} else {
// XXX: must use config_set () to run callbacks!
if (editor) {
char * buf = r_core_editor (core, NULL, node->value);
node->value = r_str_dup (node->value, buf);
free (buf);
} else {
// FGETS AND SO
r_cons_printf ("New value (old=%s): \n", node->value);
r_cons_show_cursor (true);
r_cons_flush ();
r_cons_set_raw (0);
r_line_set_prompt (":> ");
r_cons_fgets (buf, sizeof (buf)-1, 0, 0);
r_cons_set_raw (1);
r_cons_show_cursor (false);
r_config_set (core->config, name, buf);
//node->value = r_str_dup (node->value, buf);
}
}
}
R_API void r_core_visual_config(RCore *core) {
char *fs = NULL, *fs2 = NULL, *desc = NULL;
int i, j, ch, hit, show;
int option, _option = 0;
RListIter *iter;
RConfigNode *bt;
char old[1024];
int delta = 9;
int menu = 0;
old[0]='\0';
option = 0;
for (;;) {
r_cons_clear00 ();
r_cons_get_size (&delta);
delta /= 4;
switch (menu) {
case 0: // flag space
r_cons_printf ("[EvalSpace]\n\n");
hit = j = i = 0;
r_list_foreach (core->config->nodes, iter, bt) {
if (option == i) {
fs = bt->name;
}
if (!old[0]) {
r_str_ccpy (old, bt->name, '.');
show = 1;
} else if (r_str_ccmp (old, bt->name, '.')) {
r_str_ccpy (old, bt->name, '.');
show = 1;
} else {
show = 0;
}
if (show) {
if (option == i) {
hit = 1;
}
if ( (i >=option-delta) && ((i<option+delta)||((option<delta)&&(i<(delta<<1))))) {
r_cons_printf (" %c %s\n", (option == i)?'>':' ', old);
j++;
}
i++;
}
}
if (!hit && j > 0) {
option--;
continue;
}
r_cons_printf ("\n Sel:%s \n\n", fs);
break;
case 1: // flag selection
r_cons_printf ("[EvalSpace < Variables: %s]\n\n", fs);
hit = 0;
j = i = 0;
// TODO: cut -d '.' -f 1 | sort | uniq !!!
r_list_foreach (core->config->nodes, iter, bt) {
if (!r_str_ccmp (bt->name, fs, '.')) {
if (option==i) {
fs2 = bt->name;
desc = bt->desc;
hit = 1;
}
if ( (i>=option-delta) && ((i<option+delta)||((option<delta)&&(i<(delta<<1))))) {
// TODO: Better align
r_cons_printf (" %c %s = %s\n", (option==i)?'>':' ', bt->name, bt->value);
j++;
}
i++;
}
}
if (!hit && j>0) {
option = i-1;
continue;
}
if (fs2 != NULL) {
// TODO: Break long lines.
r_cons_printf ("\n Selected: %s (%s)\n\n",
fs2, desc);
}
}
if (fs && !strncmp (fs, "asm.", 4)) {
r_core_cmd (core, "pd $r", 0);
}
r_cons_visual_flush ();
ch = r_cons_readchar ();
if (ch == 4 || ch == -1) {
return;
}
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case 'j': option++; break;
case 'k': option = (option<=0)? 0: option-1; break;
case 'J': option+=4; break;
case 'K': option = (option<=3)? 0: option-4; break;
case 'h':
case 'b': // back
menu = 0;
option = _option;
break;
case '_':
r_core_visual_config_hud (core);
break;
case 'Q':
case 'q':
if (menu <= 0) {
return;
}
menu--;
option = _option;
break;
case '$':
r_core_cmd0 (core, "?$");
r_cons_any_key (NULL);
break;
case '*':
case '+':
fs2 ? config_visual_hit_i (core, fs2, +1) : 0;
continue;
case '/':
case '-':
fs2 ? config_visual_hit_i (core, fs2, -1) : 0;
continue;
case 'l':
case 'E': // edit value
case 'e': // edit value
case ' ':
case '\r':
case '\n': // never happens
if (menu == 1) {
fs2 ? config_visual_hit (core, fs2, (ch=='E')) : 0;
} else {
menu = 1;
_option = option;
option = 0;
}
break;
case '?':
r_cons_clear00 ();
r_cons_printf ("\nVe: Visual Eval help:\n\n"
" q - quit menu\n"
" j/k - down/up keys\n"
" h/b - go back\n"
" $ - same as ?$ - show values of vars\n"
" e/' ' - edit/toggle current variable\n"
" E - edit variable with 'cfg.editor' (vi?)\n"
" +/- - increase/decrease numeric value (* and /, too)\n"
" : - enter command\n");
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':':
r_cons_show_cursor (true);
r_cons_set_raw (0);
{
char *cmd = prompt (":> ", NULL);
r_core_cmd (core, cmd, 1);
free (cmd);
}
r_cons_set_raw (1);
r_cons_show_cursor (false);
r_cons_any_key (NULL);
r_cons_clear00 ();
continue;
}
}
}
R_API void r_core_visual_mounts(RCore *core) {
RList *list = NULL;
RFSRoot *fsroot = NULL;
RListIter *iter;
RFSFile *file;
RFSPartition *part;
int i, ch, option, mode, partition, dir, delta = 7;
char *str, path[4096], buf[1024], *root = NULL;
const char *n, *p;
dir = partition = option = mode = 0;
for (;;) {
/* Clear */
r_cons_clear00 ();
/* Show */
if (mode == 0) {
if (list) {
r_list_free (list);
list = NULL;
}
r_cons_printf ("Press '/' to navigate the root filesystem.\nPartitions:\n\n");
n = r_fs_partition_type_get (partition);
list = r_fs_partitions (core->fs, n, 0);
i = 0;
if (list) {
r_list_foreach (list, iter, part) {
if ((option-delta <= i) && (i <= option+delta)) {
if (option == i) {
r_cons_printf (" > ");
} else {
r_cons_printf (" ");
}
r_cons_printf ("%d %02x 0x%010"PFMT64x" 0x%010"PFMT64x"\n",
part->number, part->type,
part->start, part->start+part->length);
}
i++;
}
r_list_free (list);
list = NULL;
} else {
r_cons_printf ("Cannot read partition\n");
}
} else if (mode == 1) {
r_cons_printf ("Types:\n\n");
for (i=0;;i++) {
n = r_fs_partition_type_get (i);
if (!n) {
break;
}
r_cons_printf ("%s%s\n", (i==partition)?" > ":" ", n);
}
} else if (mode == 3) {
i = 0;
r_cons_printf ("Mountpoints:\n\n");
r_list_foreach (core->fs->roots, iter, fsroot) {
if (fsroot && (option-delta <= i) && (i <= option+delta)) {
r_cons_printf ("%s %s\n", (option == i)?" > ":" ",
fsroot->path);
}
i++;
}
} else {
if (root) {
list = r_fs_dir (core->fs, path);
if (list) {
r_cons_printf ("%s:\n\n", path);
i = 0;
r_list_foreach (list, iter, file) {
if ((dir-delta <= i) && (i <= dir+delta)) {
r_cons_printf ("%s%c %s\n", (dir == i)?" > ":" ",
file->type, file->name);
}
i++;
}
r_cons_printf ("\n");
r_list_free (list);
list = NULL;
} else {
r_cons_printf ("Cannot open '%s' directory\n", root);
}
} else {
r_cons_printf ("Root undefined\n");
}
}
if (mode==2) {
r_str_trim_path (path);
str = path + strlen (path);
strncat (path, "/", sizeof (path)-strlen (path)-1);
list = r_fs_dir (core->fs, path);
file = r_list_get_n (list, dir);
if (file && file->type != 'd') {
r_core_cmdf (core, "px @ 0x%" PFMT64x "!64", file->off);
}
r_list_free (list);
list = NULL;
*str='\0';
}
r_cons_flush ();
/* Ask for option */
ch = r_cons_readchar ();
if (ch==-1||ch==4){
free (root);
return;
}
ch = r_cons_arrow_to_hjkl (ch);
switch (ch) {
case '/':
R_FREE (root);
root = strdup ("/");
strncpy (path, root, sizeof (path)-1);
mode = 2;
break;
case 'l':
case '\r':
case '\n':
if (mode == 0) {
n = r_fs_partition_type_get (partition);
list = r_fs_partitions (core->fs, n, 0);
if (!list) {
r_cons_printf ("Unknown partition\n");
r_cons_any_key (NULL);
r_cons_flush ();
break;
}
part = r_list_get_n (list, option);
if (!part) {
r_cons_printf ("Unknown partition\n");
r_cons_any_key (NULL);
r_cons_flush ();
break;
}
p = r_fs_partition_type (n, part->type);
if (p) {
if (r_fs_mount (core->fs, p, "/root", part->start)) {
free (root);
root = strdup ("/root");
strncpy (path, root, sizeof (path)-1);
mode = 2;
} else {
r_cons_printf ("Cannot mount partition\n");
r_cons_flush ();
r_cons_any_key (NULL);
}
} else {
r_cons_printf ("Unknown partition type\n");
r_cons_flush ();
r_cons_any_key (NULL);
}
r_list_free (list);
list = NULL;
} else if (mode == 2){
r_str_trim_path (path);
strncat (path, "/", sizeof (path)-strlen (path)-1);
list = r_fs_dir (core->fs, path);
file = r_list_get_n (list, dir);
if (file) {
if (file->type == 'd') {
strncat (path, file->name, sizeof (path)-strlen (path)-1);
r_str_trim_path (path);
if (root && strncmp (root, path, strlen (root) - 1)) {
strncpy (path, root, sizeof (path) - 1);
}
} else {
r_core_cmdf (core, "s 0x%"PFMT64x, file->off);
r_fs_umount (core->fs, root);
free (root);
return;
}
} else {
r_cons_printf ("Unknown file\n");
r_cons_flush ();
r_cons_any_key (NULL);
}
} else if (mode == 3) {
fsroot = r_list_get_n (core->fs->roots, option);
if (fsroot) {
root = strdup (fsroot->path);
strncpy (path, root, sizeof (path)-1);
}
mode = 2;
}
dir = partition = option = 0;
break;
case 'k':
if (mode == 0 || mode == 3) {
if (option > 0) {
option--;
}
} else if (mode == 1) {
if (partition > 0) {
partition--;
}
} else {
if (dir > 0) {
dir--;
}
}
break;
case 'j':
if (mode == 0) {
n = r_fs_partition_type_get (partition);
list = r_fs_partitions (core->fs, n, 0);
if (option < r_list_length (list) - 1) {
option++;
}
} else if (mode == 1) {
if (partition < r_fs_partition_get_size () - 1) {
partition++;
}
} else if (mode == 3) {
if (option < r_list_length (core->fs->roots) - 1) {
option++;
}
} else {
list = r_fs_dir (core->fs, path);
if (dir < r_list_length (list) - 1) {
dir++;
}
}
break;
case 't':
mode = 1;
break;
case 'h':
if (mode == 2) {
if (!root) {
mode = 0;
} else
if (strcmp (path, root)) {
strcat (path, "/..");
r_str_trim_path (path);
} else {
r_fs_umount (core->fs, root);
mode = 0;
}
} else if (mode == 1) {
mode = 0;
} else {
return;
}
break;
case 'Q':
case 'q':
if (mode == 2 && root) {
r_fs_umount (core->fs, root);
mode = 0;
} else {
return;
}
break;
case 'g':
if (mode == 2) {
r_str_trim_path (path);
str = path + strlen (path);
strncat (path, "/", sizeof (path)-strlen (path)-1);
list = r_fs_dir (core->fs, path);
file = r_list_get_n (list, dir);
if (file && root) {
strncat (path, file->name, sizeof (path)-strlen (path)-1);
r_str_trim_path (path);
if (strncmp (root, path, strlen (root) - 1)) {
strncpy (path, root, sizeof (path) - 1);
}
file = r_fs_open (core->fs, path, false);
if (file) {
r_fs_read (core->fs, file, 0, file->size);
r_cons_show_cursor (true);
r_cons_set_raw (0);
r_line_set_prompt ("Dump path (ej: /tmp/file): ");
r_cons_fgets (buf, sizeof (buf)-1, 0, 0);
r_cons_set_raw (1);
r_cons_show_cursor (false);
r_file_dump (buf, file->data, file->size, 0);
r_fs_close (core->fs, file);
r_cons_printf ("Done\n");
} else {
r_cons_printf ("Cannot dump file\n");
}
} else {
r_cons_printf ("Cannot dump file\n");
}
r_cons_flush ();
r_cons_any_key (NULL);
*str='\0';
}
break;
case 'm':
mode = 3;
option = 0;
break;
case '?':
r_cons_clear00 ();
r_cons_printf ("\nVM: Visual Mount points help:\n\n");
r_cons_printf (" q - go back or quit menu\n");
r_cons_printf (" j/k - down/up keys\n");
r_cons_printf (" h/l - forward/go keys\n");
r_cons_printf (" t - choose partition type\n");
r_cons_printf (" g - dump file\n");
r_cons_printf (" m - show mountpoints\n");
r_cons_printf (" : - enter command\n");
r_cons_printf (" ? - show this help\n");
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':':
r_cons_show_cursor (true);
r_cons_set_raw (0);
r_line_set_prompt (":> ");
r_cons_fgets (buf, sizeof (buf)-1, 0, 0);
r_cons_set_raw (1);
r_cons_show_cursor (false);
r_core_cmd (core, buf, 1);
r_cons_any_key (NULL);
break;
}
}
}
#if 0
static void var_index_show(RAnal *anal, RAnalFunction *fcn, ut64 addr, int idx) {
int i = 0;
RAnalVar *v;
RAnalVarAccess *x;
RListIter *iter, *iter2;
int window ;
// Adjust the windows size automaticaly
(void)r_cons_get_size (&window);
window-=5; // Size of printed things
int wdelta = (idx>5)?idx-5:0;
if (!fcn) return;
r_list_foreach(fcn->vars, iter, v) {
if (addr == 0 || (addr >= v->addr && addr <= v->eaddr)) {
if (i>=wdelta) {
if (i>window+wdelta) {
r_cons_printf("...\n");
break;
}
if (idx == i) r_cons_printf (" * ");
else r_cons_printf (" ");
#if 0
if (v->type->type == R_ANAL_TYPE_ARRAY) {
eprintf ("TODO: support for arrays\n");
r_cons_printf ("0x%08llx - 0x%08llx scope=%s type=%s name=%s delta=%d array=%d\n",
v->addr, v->eaddr, r_anal_var_scope_to_str (anal, v->scope),
r_anal_type_to_str (anal, v->type, ""),
v->name, v->delta, v->type->custom.a->count);
} else
#endif
{
char *s = r_anal_type_to_str (anal, v->type);
if (!s) s = strdup ("<unk>");
r_cons_printf ("0x%08llx - 0x%08llx scope=%d type=%s name=%s delta=%d\n",
v->addr, v->eaddr, v->scope, s, v->name, v->delta);
free (s);
}
r_list_foreach (v->accesses, iter2, x) {
r_cons_printf (" 0x%08llx %s\n", x->addr, x->set?"set":"get");
}
}
i++;
}
}
}
#endif
// helper
static void function_rename(RCore *core, ut64 addr, const char *name) {
RListIter *iter;
RAnalFunction *fcn;
r_list_foreach (core->anal->fcns, iter, fcn) {
if (fcn->addr == addr) {
r_flag_unset_name (core->flags, fcn->name);
free (fcn->name);
fcn->name = strdup (name);
r_flag_set (core->flags, name, addr, r_anal_function_size_from_entry (fcn));
break;
}
}
}
static void variable_rename (RCore *core, ut64 addr, int vindex, const char *name) {
RAnalFunction* fcn = r_anal_get_fcn_in (core->anal, addr, R_ANAL_FCN_TYPE_NULL);
ut64 a_tmp = core->offset;
int i = 0;
RListIter *iter;
RList *list = r_anal_var_all_list (core->anal, fcn);
RAnalVar* var;
r_list_foreach (list, iter, var) {
if (i == vindex) {
r_core_seek (core, addr, false);
r_core_cmd_strf (core, "afvn %s %s", name, var->name);
r_core_seek (core, a_tmp, false);
break;
}
++i;
}
r_list_free (list);
}
static void variable_set_type (RCore *core, ut64 addr, int vindex, const char *type) {
RAnalFunction* fcn = r_anal_get_fcn_in (core->anal, addr, R_ANAL_FCN_TYPE_NULL);
RList *list = r_anal_var_all_list (core->anal, fcn);
RListIter *iter;
RAnalVar* var;
r_list_foreach (list, iter, var) {
if (vindex == 0) {
r_anal_var_retype (core->anal, fcn->addr,
R_ANAL_VAR_SCOPE_LOCAL, -1, var->kind, type, -1, var->isarg, var->name);
break;
}
vindex--;
}
r_list_free (list);
}
// In visual mode, display function list
static ut64 var_functions_show(RCore *core, int idx, int show, int cols) {
int wdelta = (idx > 5)? idx - 5: 0;
char *var_functions;
ut64 seek = core->offset;
ut64 addr = core->offset;
RAnalFunction *fcn;
int window, i = 0, print_full_func;
RListIter *iter;
// Adjust the windows size automaticaly
(void)r_cons_get_size (&window);
window -= 8; // Size of printed things
bool color = r_config_get_i (core->config, "scr.color");
const char *color_addr = core->cons->context->pal.offset;
const char *color_fcn = core->cons->context->pal.fname;
r_list_foreach (core->anal->fcns, iter, fcn) {
print_full_func = true;
if (i >= wdelta) {
if (i > window + wdelta - 1) {
r_cons_printf ("...\n");
break;
}
if (idx == i) {
addr = fcn->addr;
}
if (show) {
char *tmp;
if (color) {
var_functions = r_str_newf ("%c%c %s0x%08"PFMT64x"" Color_RESET" %4d %s%s"Color_RESET"",
(seek == fcn->addr)?'>':' ',
(idx==i)?'*':' ',
color_addr, fcn->addr, r_anal_function_realsize (fcn),
color_fcn, fcn->name);
} else {
var_functions = r_str_newf ("%c%c 0x%08"PFMT64x" %4d %s",
(seek == fcn->addr)?'>':' ',
(idx==i)?'*':' ',
fcn->addr, r_anal_function_realsize (fcn), fcn->name);
}
if (var_functions) {
if (!r_cons_singleton ()->show_vals) {
int fun_len = r_str_ansi_len (var_functions);
int columns = fun_len > cols ? cols - 2 : cols;
tmp = r_str_ansi_crop (var_functions, 0, 0, columns, window);
if (r_str_ansi_len (tmp) < fun_len) {
r_cons_printf ("%s..%s\n", tmp, Color_RESET);
print_full_func = false;
}
r_free (tmp);
}
if (print_full_func) {
r_cons_println (var_functions);
}
r_free (var_functions);
}
}
}
i++;
}
return addr;
}
// In visual mode, display the variables.
static ut64 var_variables_show(RCore* core, int idx, int *vindex, int show, int cols) {
int i = 0;
const ut64 addr = var_functions_show (core, idx, 0, cols);
RAnalFunction* fcn = r_anal_get_fcn_in (core->anal, addr, R_ANAL_FCN_TYPE_NULL);
int window;
int wdelta = (idx > 5) ? idx - 5 : 0;
RListIter *iter;
RList *list = r_anal_var_all_list (core->anal, fcn);
RAnalVar* var;
// Adjust the window size automatically.
(void)r_cons_get_size (&window);
window -= 8; // Size of printed things.
// A new line so this looks reasonable.
r_cons_newline ();
int llen = r_list_length (list);
if (*vindex >= llen) {
*vindex = llen - 1;
}
r_list_foreach (list, iter, var) {
if (i >= wdelta) {
if (i > window + wdelta) {
r_cons_printf ("...\n");
break;
}
if (show) {
switch (var->kind & 0xff) {
case 'r':
{
RRegItem *r = r_reg_index_get (core->anal->reg, var->delta);
if (!r) {
eprintf ("Register not found");
break;
}
r_cons_printf ("%sarg %s %s @ %s\n",
i == *vindex ? "* ":" ",
var->type, var->name,
r->name);
}
break;
case 'b':
r_cons_printf ("%s%s %s %s @ %s%s0x%x\n",
i == *vindex ? "* ":" ",
var->delta < 0? "var": "arg",
var->type, var->name,
core->anal->reg->name[R_REG_NAME_BP],
(var->kind == 'v')?"-":"+",
var->delta);
break;
case 's':
r_cons_printf ("%s%s %s %s @ %s%s0x%x\n",
i == *vindex ? "* ":" ",
var->delta < 0? "var": "arg",
var->type, var->name,
core->anal->reg->name[R_REG_NAME_BP],
(var->kind == 'v')?"-":"+",
var->delta);
break;
}
}
}
++i;
}
r_list_free (list);
return addr;
}
static int level = 0;
static st64 delta = 0;
static int option = 0;
static int variable_option = 0;
static int printMode = 0;
static bool selectPanel = false;
#define lastPrintMode 6
static const char *cmd, *printCmds[lastPrintMode] = {
"pdf", "pd $r", "afi", "pdsf", "pdc", "pdr"
};
static void r_core_visual_anal_refresh_column (RCore *core, int colpos) {
const ut64 addr = (level != 0 && level != 1)
? core->offset
: var_functions_show (core, option, 0, colpos);
// RAnalFunction* fcn = r_anal_get_fcn_in(core->anal, addr, R_ANAL_FCN_TYPE_NULL);
int h, w = r_cons_get_size (&h);
// int sz = (fcn)? R_MIN (r_anal_fcn_size (fcn), h * 15) : 16; // max instr is 15 bytes.
if (printMode > 0 && printMode < lastPrintMode) {
cmd = printCmds[printMode];
} else {
cmd = printCmds[printMode = 0];
}
char *cmdf = r_str_newf ("%s @ 0x%"PFMT64x, cmd, addr + delta);
if (!cmdf) {
return;
}
char *output = r_core_cmd_str (core, cmdf);
if (output) {
// 'h - 2' because we have two new lines in r_cons_printf
char *out = r_str_ansi_crop (output, 0, 0, w - colpos, h - 2);
r_cons_printf ("\n%s\n", out);
free (out);
R_FREE (output);
}
free (cmdf);
}
static const char *help_fun_visual[] = {
"(a)", "analyze ", "(-)", "delete ", "(x)", "xrefs ", "(X)", "refs j/k next/prev\n",
"(r)", "rename ", "(c)", "calls ", "(d)", "definetab column (_) hud\n",
"(d)", "define ", "(v)", "vars ", "(?)"," help ", "(:)", "shell " ,"(q)", "quit\n",
"(s)", "edit function signature. \n\n",
NULL
};
static const char *help_var_visual[] = {
"(a)", "add " ,"(x)", "xrefs ", "(r)", "rename\n",
"(t)", "type ", "(g)", "go ", "(-)" ,"delete\n",
"(q)", "quit ", "(s)", "signature\n\n",
NULL
};
static const char *help_vv_visual[] = {
"j,k", "select next/prev item or scroll if tab pressed",
"J,K", "scroll next/prev page \"\"",
"h,q", "go back, quit",
"p,P", "switch next/prev print mode",
"v", "view selected function arguments and variables",
"x,X", "see xrefs to the selected function",
"tab", "toggle disasm column selection (to scroll in code)",
"!", "run 'afls' to sort all functions by address",
".", "seek to current function address",
":", "run r2 commands",
"_", "hud mode. same as: s $(afl~...)",
"enter", "enter function view (variables), xrefs",
NULL
};
static const char *help_vv_actions_visual[] = {
" functions:", "Add, Modify, Delete, Xrefs Calls Vars",
" variables:", "Add, Modify, Delete",
NULL
};
static void r_core_vmenu_append_help (RStrBuf *p, const char **help) {
int i;
RConsContext *cons_ctx = r_cons_singleton ()->context;
const char *pal_args_color = cons_ctx->color_mode ? cons_ctx->pal.args : "",
*pal_help_color = cons_ctx->color_mode ? cons_ctx->pal.help : "",
*pal_reset = cons_ctx->color_mode ? cons_ctx->pal.reset : "";
for (i = 0; help[i]; i += 2) {
r_strbuf_appendf (p, "%s%s %s%s%s",
pal_args_color, help[i],
pal_help_color, help[i + 1], pal_reset);
}
}
static ut64 r_core_visual_anal_refresh (RCore *core) {
if (!core) {
return 0LL;
}
ut64 addr;
RStrBuf *buf;
char old[1024];
bool color = r_config_get_i (core->config, "scr.color");
int h, cols = r_cons_get_size (&h);
old[0] = '\0';
addr = core->offset;
cols -= 50;
if (cols > 60) {
cols = 60;
}
r_cons_clear00 ();
r_core_visual_anal_refresh_column (core, cols);
if (cols > 30) {
r_cons_column (cols);
}
switch (level) {
// Show functions list help in visual mode
case 0:
buf = r_strbuf_new ("");
if (color) {
r_cons_strcat (core->cons->context->pal.prompt);
}
if (selectPanel) {
r_cons_printf ("-- functions -----------------[ %s ]-->>", printCmds[printMode]);
} else {
r_cons_printf ("-[ functions ]----------------- %s ---", printCmds[printMode]);
}
if (color) {
r_cons_strcat ("\n" Color_RESET);
}
r_core_vmenu_append_help (buf, help_fun_visual);
r_cons_printf ("%s", r_strbuf_drain (buf));
addr = var_functions_show (core, option, 1, cols);
break;
case 1:
buf = r_strbuf_new ("");
if (color) {
r_cons_strcat (core->cons->context->pal.prompt);
}
r_cons_printf ("-[ variables ]----- 0x%08"PFMT64x"", addr);
if (color) {
r_cons_strcat ("\n" Color_RESET);
}
r_core_vmenu_append_help (buf, help_var_visual);
char *drained = r_strbuf_drain (buf);
r_cons_printf ("%s", drained);
addr = var_variables_show (core, option, &variable_option, 1, cols);
free (drained);
// var_index_show (core->anal, fcn, addr, option);
break;
case 2:
r_cons_printf ("Press 'q' to quit call refs\n");
if (color) {
r_cons_strcat (core->cons->context->pal.prompt);
}
r_cons_printf ("-[ calls ]----------------------- 0x%08"PFMT64x" (TODO)\n", addr);
if (color) {
r_cons_strcat ("\n" Color_RESET);
}
// TODO: filter only the callrefs. but we cant grep here
sprintf (old, "afi @ 0x%08"PFMT64x, addr);
char *output = r_core_cmd_str (core, old);
if (output) {
// 'h - 2' because we have two new lines in r_cons_printf
if (!r_cons_singleton ()->show_vals) {
char *out = r_str_ansi_crop(output, 0, 0, cols, h - 2);
r_cons_printf ("\n%s\n", out);
free (out);
R_FREE (output);
} else {
r_cons_printf ("\n%s\n", output);
R_FREE (output);
}
}
break;
default:
// assert
break;
}
r_cons_flush ();
return addr;
}
static void r_core_visual_anal_refresh_oneshot (RCore *core) {
r_core_task_enqueue_oneshot (&core->tasks, (RCoreTaskOneShot) r_core_visual_anal_refresh, core);
}
static void r_core_visual_debugtraces_help(RCore *core) {
r_cons_clear00 ();
r_cons_printf (
"vbd: Visual Browse Debugtraces:\n\n"
" q - quit the bit editor\n"
" Q - Quit (jump into the disasm view)\n"
" j/k - Select next/previous trace\n"
" : - enter command\n");
r_cons_flush ();
r_cons_any_key (NULL);
}
R_API void r_core_visual_debugtraces(RCore *core, const char *input) {
int i, delta = 0;
for (;;) {
char *trace_addr_str = r_core_cmd_strf (core, "dtdq %d", delta);
ut64 trace_addr = r_num_get (NULL, trace_addr_str);
free (trace_addr_str);
r_cons_printf ("[0x%08"PFMT64x"]> %d dbg.trace\n", trace_addr, delta);
for (i = 0; i < delta; i++) {
r_core_cmdf (core, ".dte %d", i);
}
r_core_cmd0 (core, "x 64@r:SP");
r_core_cmd0 (core, "dri");
// limit by rows here
//int rows = r_cons_get_size (NULL);
r_core_cmdf (core, "dtd %d", delta);
r_cons_visual_flush ();
char ch ;
if (input && *input) {
ch = *input;
input++;
} else {
ch = r_cons_readchar ();
}
if (ch == 4 || ch == -1) {
if (level == 0) {
goto beach;
}
level--;
continue;
}
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case 'Q': // tab
{
ut64 oseek = core->offset;
core->vmode = false;
r_core_seek (core, trace_addr, true);
r_core_visual (core, "");
r_core_seek (core, oseek, true);
}
break;
case 'q':
goto beach;
break;
case ']':
r_config_set_i (core->config, "hex.cols", r_config_get_i (core->config, "hex.cols") + 1);
break;
case '[':
r_config_set_i (core->config, "hex.cols", r_config_get_i (core->config, "hex.cols") - 1);
break;
case 'J':
delta += 10;
break;
case 'K':
delta -= 10;
if (delta < 0) {
delta = 0;
}
case 'j':
delta++;
break;
case 'k':
delta--;
if (delta < 0) {
delta = 0;
}
break;
case ':':
r_core_visual_prompt (core);
r_cons_any_key (NULL);
break;
case '?':
r_core_visual_debugtraces_help (core);
break;
}
}
beach:
;
}
static char *__prompt(const char *msg, void *p) {
char res[128];
r_cons_show_cursor (true);
r_cons_set_raw (false);
r_line_set_prompt (msg);
res[0] =0;
if (!r_cons_fgets (res, sizeof (res), 0, NULL)) {
res[0] = 0;
}
return strdup (res);
}
static void addVar(RCore *core, int ch, const char *msg) {
char *src = __prompt (msg, NULL);
char *name = __prompt ("Variable Name: ", NULL);
char *type = __prompt ("Type of Variable (int32_t): ", NULL);
char *cmd = r_str_newf ("afv%c %s %s %s", ch, src, name, type);
r_str_trim (cmd);
r_core_cmd0 (core, cmd);
free(cmd);
free (src);
free (name);
free (type);
}
/* Like emenu but for real */
R_API void r_core_visual_anal(RCore *core, const char *input) {
char old[218];
int nfcns, ch, _option = 0;
RConsEvent olde = core->cons->event_resize;
void *olde_user = core->cons->event_data;
ut64 addr = core->offset;
core->cons->event_resize = NULL; // avoid running old event with new data
core->cons->event_data = core;
core->cons->event_resize = (RConsEvent) r_core_visual_anal_refresh_oneshot;
level = 0;
int asmbytes = r_config_get_i (core->config, "asm.bytes");
r_config_set_i (core->config, "asm.bytes", 0);
for (;;) {
nfcns = r_list_length (core->anal->fcns);
addr = r_core_visual_anal_refresh (core);
if (input && *input) {
ch = *input;
input++;
} else {
ch = r_cons_readchar ();
}
if (ch == 4 || ch == -1) {
if (level == 0) {
goto beach;
}
level--;
continue;
}
ch = r_cons_arrow_to_hjkl (ch); // get ESC+char, return 'hjkl' char
switch (ch) {
case '[':
r_cons_singleton ()->show_vals = true;
break;
case ']':
r_cons_singleton ()->show_vals = false;
break;
case '?':
r_cons_clear00 ();
RStrBuf *buf = r_strbuf_new ("");
r_cons_println ("|Usage: vv");
r_core_visual_append_help (buf, "Actions supported", help_vv_actions_visual);
r_core_visual_append_help (buf, "Keys", help_vv_visual);
r_cons_printf ("%s", r_strbuf_drain (buf));
r_cons_flush ();
r_cons_any_key (NULL);
break;
case 9:
selectPanel = !selectPanel;
if (!selectPanel) {
delta = 0;
printMode = 0;
}
break;
case ':':
{
ut64 orig = core->offset;
r_core_seek (core, addr, 0);
while (r_core_visual_prompt (core));
r_core_seek (core, orig, 0);
}
continue;
case '/':
r_core_cmd0 (core, "?i highlight;e scr.highlight=`yp`");
break;
case 'a':
switch (level) {
case 0:
r_core_cmd0 (core, "af-$$;af"); // reanalize
break;
case 1:
{
eprintf ("Select variable source ('r'egister, 's'tackptr or 'b'aseptr): ");
int type = r_cons_readchar ();
switch (type) {
case 'r':
addVar (core, type, "Source Register Name: ");
break;
case 's':
addVar (core, type, "BP Relative Delta: ");
break;
case 'b':
addVar (core, type, "SP Relative Delta: ");
break;
}
}
break;
}
break;
case 'r':
{
switch (level) {
case 1:
r_cons_show_cursor (true);
r_cons_set_raw (false);
r_line_set_prompt ("New name: ");
if (r_cons_fgets (old, sizeof (old), 0, NULL)) {
if (*old) {
//old[strlen (old)-1] = 0;
variable_rename (core, addr, variable_option, old);
}
}
break;
default:
r_line_set_prompt ("New name: ");
if (r_cons_fgets (old, sizeof (old), 0, NULL)) {
if (*old) {
//old[strlen (old)-1] = 0;
function_rename (core, addr, old);
}
}
break;
}
r_cons_set_raw (true);
r_cons_show_cursor (false);
}
break;
case 't':
if (level == 1) {
r_cons_show_cursor (true);
r_cons_set_raw (false);
r_line_set_prompt ("New type: ");
if (r_cons_fgets (old, sizeof (old), 0, NULL)) {
if (*old) {
//old[strlen (old)-1] = 0;
variable_set_type (core, addr, variable_option, old);
}
}
r_cons_set_raw (true);
r_cons_show_cursor (false);
}
break;
case '.':
delta = 0;
break;
case 'R':
r_core_cmd0 (core, "ecn");
break;
case 'p':
printMode ++;
break;
case 'P':
if (printMode == 0) {
printMode = lastPrintMode;
} else {
printMode --;
}
break;
case 'd':
r_core_visual_define(core, "", 0);
break;
case '-':
switch (level) {
case 0:
r_core_cmdf (core, "af-0x%"PFMT64x, addr);
break;
}
break;
case 'x':
r_core_visual_refs (core, false, true);
break;
case 'X':
r_core_visual_refs (core, true, true);
break;
case 's':
r_core_cmdf (core, "afs!@0x%08"PFMT64x, addr);
break;
case 'c':
level = 2;
break;
case 'v':
level = 1;
variable_option = 0;
break;
case '_':
{
r_core_cmd0 (core, "s $(afl~...)");
int n = 0;
RListIter *iter;
RAnalFunction *fcn;
r_list_foreach (core->anal->fcns, iter, fcn) {
if (fcn->addr == core->offset) {
option = n;
break;
}
n ++;
}
}
break;
case 'j':
if (selectPanel) {
printMode = 1;
delta += 16;
} else {
delta = 0;
switch (level) {
case 1:
variable_option++;
break;
default:
option++;
if (option >= nfcns) {
--option;
}
break;
}
}
break;
case '!':
// TODO: use aflsn/aflsb/aflss/...
{
static int sortMode = 0;
const char *sortModes[4] = { "aflsa", "aflss", "aflsb", "aflsn" };
r_core_cmd0 (core, sortModes[sortMode%4]);
sortMode++;
}
break;
case 'k':
if (selectPanel) {
printMode = 1;
delta -= 16;
} else {
delta = 0;
switch (level) {
case 1:
variable_option = (variable_option<=0)? 0: variable_option-1;
break;
default:
option = (option<=0)? 0: option-1;
break;
}
}
break;
case 'J':
if (selectPanel) {
printMode = 1;
delta += 40;
} else {
int rows = 0;
r_cons_get_size (&rows);
option += (rows - 5);
if (option >= nfcns) {
option = nfcns - 1;
}
}
break;
case 'K':
if (selectPanel) {
printMode = 1;
delta -= 40;
} else {
int rows = 0;
r_cons_get_size (&rows);
option -= (rows - 5);
if (option < 0) {
option = 0;
}
}
break;
case 'g': {
r_core_visual_showcursor (core, true);
r_core_visual_offset (core); // change the seek to selected offset
RListIter *iter; // change the current option to selected seek
RAnalFunction *fcn;
int i = 0;
r_list_foreach (core->anal->fcns, iter, fcn) {
if (core->offset == fcn->addr){
option = i;
}
i++;
}
r_core_visual_showcursor (core, false);
}
break;
case 'G':
r_core_seek (core, addr, SEEK_SET);
goto beach;
case ' ':
case '\r':
case '\n':
level = 0;
r_core_seek (core, addr, SEEK_SET);
goto beach;
break;
case 'l':
level = 1;
_option = option;
break;
case 'h':
case 'b': // back
level = 0;
option = _option;
break;
case 'Q':
case 'q':
if (level == 0) {
goto beach;
}
level--;
break;
}
}
beach:
core->cons->event_resize = NULL; // avoid running old event with new data
core->cons->event_data = olde_user;
core->cons->event_resize = olde;
level = 0;
r_config_set_i (core->config, "asm.bytes", asmbytes);
}
struct seek_flag_offset_t {
ut64 offset;
ut64 *next;
bool is_next;
};
static bool seek_flag_offset(RFlagItem *fi, void *user) {
struct seek_flag_offset_t *u = (struct seek_flag_offset_t *)user;
if (u->is_next) {
if (fi->offset < *u->next && fi->offset > u->offset) {
*u->next = fi->offset;
}
} else {
if (fi->offset > *u->next && fi->offset < u->offset) {
*u->next = fi->offset;
}
}
return true;
}
R_API void r_core_seek_next(RCore *core, const char *type) {
RListIter *iter;
ut64 next = UT64_MAX;
if (strstr (type, "opc")) {
RAnalOp aop;
if (r_anal_op (core->anal, &aop, core->offset, core->block, core->blocksize, R_ANAL_OP_MASK_BASIC)) {
next = core->offset + aop.size;
} else {
eprintf ("Invalid opcode\n");
}
} else if (strstr (type, "fun")) {
RAnalFunction *fcni;
r_list_foreach (core->anal->fcns, iter, fcni) {
if (fcni->addr < next && fcni->addr > core->offset) {
next = fcni->addr;
}
}
} else if (strstr (type, "hit")) {
const char *pfx = r_config_get (core->config, "search.prefix");
struct seek_flag_offset_t u = { .offset = core->offset, .next = &next, .is_next = true };
r_flag_foreach_prefix (core->flags, pfx, -1, seek_flag_offset, &u);
} else { // flags
struct seek_flag_offset_t u = { .offset = core->offset, .next = &next, .is_next = true };
r_flag_foreach (core->flags, seek_flag_offset, &u);
}
if (next != UT64_MAX) {
r_core_seek (core, next, 1);
}
}
R_API void r_core_seek_previous (RCore *core, const char *type) {
RListIter *iter;
ut64 next = 0;
if (strstr (type, "opc")) {
eprintf ("TODO: r_core_seek_previous (opc)\n");
} else
if (strstr (type, "fun")) {
RAnalFunction *fcni;
r_list_foreach (core->anal->fcns, iter, fcni) {
if (fcni->addr > next && fcni->addr < core->offset) {
next = fcni->addr;
}
}
} else
if (strstr (type, "hit")) {
const char *pfx = r_config_get (core->config, "search.prefix");
struct seek_flag_offset_t u = { .offset = core->offset, .next = &next, .is_next = false };
r_flag_foreach_prefix (core->flags, pfx, -1, seek_flag_offset, &u);
} else { // flags
struct seek_flag_offset_t u = { .offset = core->offset, .next = &next, .is_next = false };
r_flag_foreach (core->flags, seek_flag_offset, &u);
}
if (next != 0) {
r_core_seek (core, next, 1);
}
}
//define the data at offset according to the type (byte, word...) n times
static void define_data_ntimes (RCore *core, ut64 off, int times, int type) {
int i = 0;
r_meta_cleanup (core->anal, off, off + core->blocksize);
if (times < 0) {
times = 1;
}
for (i = 0; i < times; i++, off += type) {
r_meta_add (core->anal, R_META_TYPE_DATA, off, off + type, "");
}
}
static bool isDisasmPrint(int mode) {
return (mode == 1 || mode == 2);
}
static void handleHints(RCore *core) {
//TODO extend for more anal hints
int i = 0;
char ch[64] = R_EMPTY;
const char *lines[] = {"[dh]- Define anal hint:"
," b [16,32,64] set bits hint"
, NULL};
for (i = 0; lines[i]; i++) {
r_cons_fill_line ();
r_cons_printf ("\r%s\n", lines[i]);
}
r_cons_flush ();
r_line_set_prompt ("anal hint: ");
if (r_cons_fgets (ch, sizeof (ch) - 1, 0, NULL) > 0) {
switch (ch[0]) {
case 'b':
{
int bits = atoi (r_str_trim_head_tail (ch + 1));
if (bits == 8 || bits == 16 || bits == 32 || bits == 64) {
r_anal_hint_set_bits (core->anal, core->offset, bits);
}
}
break;
default:
break;
}
}
}
R_API void r_core_visual_define(RCore *core, const char *args, int distance) {
int plen = core->blocksize;
ut64 off = core->offset;
int i, h = 0, n, ch, ntotal = 0;
ut8 *p = core->block;
int rep = -1;
char *name;
int delta = 0;
if (core->print->cur_enabled) {
int cur = core->print->cur;
if (core->print->ocur != -1) {
plen = R_ABS (core->print->cur - core->print->ocur) + 1;
if (core->print->ocur<cur) {
cur = core->print->ocur;
}
}
off += cur;
p += cur;
}
(void) r_cons_get_size (&h);
h -= 19;
if (h < 0) {
h = 0;
r_cons_clear00 ();
} else {
r_cons_gotoxy (0, 3);
}
const char *lines[] = { ""
,"[Vd]- Define current block as:"
," $ define flag size"
," 1 edit bits"
," a assembly"
," b as byte (1 byte)"
," B define half word (16 bit, 2 byte size)"
," c as code (unset any data / string / format) in here"
," C define flag color (fc)"
," d set as data"
," e end of function"
," f analyze function"
," F format"
," h define hint (for half-word, see 'B')"
," i (ahi) immediate base (b(in), o(ct), d(ec), h(ex), s(tr))"
," I (ahi1) immediate base (b(in), o(ct), d(ec), h(ex), s(tr))"
," j merge down (join this and next functions)"
," k merge up (join this and previous function)"
," h define anal hint"
," m manpage for current call"
," n rename flag used at cursor"
," o opcode string"
," r rename function"
," R find references /r"
," s set string"
," S set strings in current block"
," t set opcode type via aht hints (call, nop, jump, ...)"
," u undefine metadata here"
," v rename variable at offset that matches some hex digits"
," x find xrefs to current address (./r)"
," w set as 32bit word"
," W set as 64bit word"
," q quit menu"
," z zone flag"
, NULL};
for (i = 0; lines[i]; i++) {
r_cons_fill_line ();
r_cons_printf ("\r%s\n", lines[i]);
}
r_cons_flush ();
int wordsize = 0;
// get ESC+char, return 'hjkl' char
repeat:
if (*args) {
ch = *args;
args++;
} else {
ch = r_cons_arrow_to_hjkl (r_cons_readchar ());
}
onemoretime:
wordsize = 4;
switch (ch) {
case 'F':
{
char cmd[128];
r_cons_show_cursor (true);
r_core_cmd0 (core, "pf?");
r_cons_flush ();
r_line_set_prompt ("format: ");
strcpy (cmd, "Cf 0 ");
if (r_cons_fgets (cmd + 5, sizeof (cmd) - 6, 0, NULL) > 0) {
r_core_cmdf (core, "%s @ 0x%08"PFMT64x, cmd, off);
r_cons_set_raw (1);
r_cons_show_cursor (false);
}
}
break;
case '1':
r_core_visual_bit_editor (core);
break;
case 't':
case 'o':
{
char str[128];
r_cons_show_cursor (true);
r_line_set_prompt (ch=='t'?"type: ": "opstr: ");
if (r_cons_fgets (str, sizeof (str), 0, NULL) > 0) {
r_core_cmdf (core, "ah%c %s @ 0x%"PFMT64x, ch, str, off);
}
}
break;
case 'x':
r_core_cmd0 (core, "/r $$");
break;
case 'i':
{
char str[128];
r_cons_show_cursor (true);
r_line_set_prompt ("immbase: ");
if (r_cons_fgets (str, sizeof (str), 0, NULL) > 0) {
r_core_cmdf (core, "ahi %s @ 0x%"PFMT64x, str, off);
}
}
break;
case 'I':
{
char str[128];
r_cons_show_cursor (true);
r_line_set_prompt ("immbase: ");
if (r_cons_fgets (str, sizeof (str), 0, NULL) > 0) {
r_core_cmdf (core, "ahi1 %s @ 0x%"PFMT64x, str, off);
}
}
break;
case 'a':
r_core_visual_asm (core, off);
break;
case 'b':
if (plen != core->blocksize) {
rep = plen / 2;
}
define_data_ntimes (core, off, rep, R_BYTE_DATA);
wordsize = 1;
break;
case 'B': // "VdB"
if (plen != core->blocksize) {
rep = plen / 2;
}
define_data_ntimes (core, off, rep, R_WORD_DATA);
wordsize = 2;
break;
case 'w':
if (plen != core->blocksize) {
rep = plen / 4;
}
define_data_ntimes (core, off, rep, R_DWORD_DATA);
wordsize = 4;
break;
case 'W':
if (plen != core->blocksize) {
rep = plen / 8;
}
define_data_ntimes (core, off, rep, R_QWORD_DATA);
wordsize = 8;
break;
case 'm':
{
char *man = NULL;
/* check for manpage */
RAnalOp *op = r_core_anal_op (core, off, R_ANAL_OP_MASK_BASIC);
if (op) {
if (op->jump != UT64_MAX) {
RFlagItem *item = r_flag_get_i (core->flags, op->jump);
if (item) {
const char *ptr = r_str_lchr (item->name, '.');
if (ptr) {
man = strdup (ptr + 1);
}
}
}
r_anal_op_free (op);
}
if (man) {
char *p = strstr (man, "INODE");
if (p) {
*p = 0;
}
r_cons_clear ();
r_cons_flush ();
r_sys_cmdf ("man %s", man);
free (man);
}
r_cons_any_key (NULL);
}
break;
case 'n':
{
RAnalOp op;
char *q = NULL;
ut64 tgt_addr = UT64_MAX;
if (!isDisasmPrint (core->printidx)) {
break;
}
// TODO: get the aligned instruction even if the cursor is in the middle of it.
r_anal_op (core->anal, &op, off,
core->block + off - core->offset, 32, R_ANAL_OP_MASK_BASIC);
tgt_addr = op.jump != UT64_MAX ? op.jump : op.ptr;
if (op.var) {
// q = r_str_newf ("?i Rename variable %s to;afvn %s `yp`", op.var->name, op.var->name);
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, off, 0);
if (fcn) {
RAnalVar *bar = r_anal_var_get_byname (core->anal, fcn->addr, op.var->name);
if (bar) {
char *newname = r_cons_input (sdb_fmt ("New variable name for '%s': ", bar->name));
if (newname && *newname) {
r_anal_var_rename (core->anal, fcn->addr, bar->scope,
bar->kind, bar->name, newname, true);
free (newname);
}
} else {
eprintf ("Cannot find variable\n");
r_sys_sleep (1);
}
} else {
eprintf ("Cannot find function\n");
r_sys_sleep (1);
}
} else if (tgt_addr != UT64_MAX) {
RAnalFunction *fcn = r_anal_get_function_at (core->anal, tgt_addr);
RFlagItem *f = r_flag_get_i (core->flags, tgt_addr);
if (fcn) {
q = r_str_newf ("?i Rename function %s to;afn `yp` 0x%"PFMT64x,
fcn->name, tgt_addr);
} else if (f) {
q = r_str_newf ("?i Rename flag %s to;fr %s `yp`",
f->name, f->name);
} else {
q = r_str_newf ("?i Create flag at 0x%"PFMT64x" named;f `yp` @ 0x%"PFMT64x,
tgt_addr, tgt_addr);
}
}
if (q) {
r_core_cmd0 (core, q);
free (q);
}
r_anal_op_fini (&op);
break;
}
case 'C':
{
RFlagItem *item = r_flag_get_i (core->flags, off);
if (item) {
char cmd[128];
r_cons_show_cursor (true);
r_cons_flush ();
r_line_set_prompt ("color: ");
if (r_cons_fgets (cmd, sizeof (cmd)-1, 0, NULL) > 0) {
r_flag_item_set_color (item, cmd);
r_cons_set_raw (1);
r_cons_show_cursor (false);
}
} else {
eprintf ("Sorry. No flag here\n");
r_cons_any_key (NULL);
}
}
break;
case '$':
{
RFlagItem *item = r_flag_get_i (core->flags, off);
if (item) {
char cmd[128];
r_cons_printf ("Current flag size is: %d\n", item->size);
r_cons_show_cursor (true);
r_cons_flush ();
r_line_set_prompt ("new size: ");
if (r_cons_fgets (cmd, sizeof (cmd)-1, 0, NULL) > 0) {
item->size = r_num_math (core->num, cmd);
r_cons_set_raw (1);
r_cons_show_cursor (false);
}
} else {
eprintf ("Sorry. No flag here\n");
r_cons_any_key (NULL);
}
}
break;
case 'e':
// set function size
{
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, off, 0);
if (!fcn) {
fcn = r_anal_get_fcn_in (core->anal, core->offset, 0);
}
if (fcn) {
RAnalOp op;
ut64 size;
if (r_anal_op (core->anal, &op, off, core->block+delta,
core->blocksize-delta, R_ANAL_OP_MASK_BASIC)) {
size = off - fcn->addr + op.size;
r_anal_fcn_resize (core->anal, fcn, size);
}
}
}
break;
case 'j':
r_core_cmdf (core, "afm $$+$F @0x%08"PFMT64x, off);
break;
case 'k':
eprintf ("TODO: merge up\n");
r_cons_any_key (NULL);
break;
// very weak and incomplete
case 'h': // "Vdh"
handleHints (core);
break;
case 'r': // "Vdr"
r_core_cmdf (core, "?i new function name;afn `yp` @ 0x%08"PFMT64x, off);
break;
case 'z': // "Vdz"
r_core_cmdf (core, "?i zone name;fz `yp` @ 0x%08"PFMT64x, off);
break;
case 'R': // "VdR"
eprintf ("Finding references to 0x%08"PFMT64x" ...\n", off);
r_core_cmdf (core, "./r 0x%08"PFMT64x" @ $S", off);
break;
case 'S':
{
int i, j;
bool is_wide = false;
do {
n = r_str_nlen_w ((const char *)p + ntotal,
plen - ntotal) + 1;
if (n < 2) {
break;
}
name = malloc (n + 10);
strcpy (name, "str.");
for (i = 0, j = 0; i < n; i++, j++) {
name[4 + i] = p[j + ntotal];
if (!p[j + ntotal]) {
break;
}
if (!p[j + 1 + ntotal]) {
//check if is still wide
if (j + 3 + ntotal < n) {
if (p[j + 3]) {
break;
}
}
is_wide = true;
j++;
}
}
name[4 + n] = '\0';
if (is_wide) {
r_meta_add (core->anal, R_META_TYPE_STRING,
off + ntotal, off + (n * 2) + ntotal,
(const char *)name + 4);
} else {
r_meta_add (core->anal, R_META_TYPE_STRING,
off + ntotal, off + n + ntotal,
(const char *)name + 4);
}
r_name_filter (name, n + 10);
r_flag_set (core->flags, name, off + ntotal, n);
free (name);
if (is_wide) {
ntotal += n * 2 - 1;
} else {
ntotal += n;
}
} while (ntotal < plen);
wordsize = ntotal;
}
break;
case 's':
{
int i, j;
bool is_wide = false;
if (core->print->ocur != -1) {
n = plen;
} else {
n = r_str_nlen_w ((const char*)p, plen) + 1;
}
name = malloc (n + 10);
if (!name) {
break;
}
strcpy (name, "str.");
for (i = 0, j = 0; i < n; i++, j++) {
name[4 + i] = p[j];
if (!p[j + 1]) {
break;
}
if (!p[j + 1]) {
if (j + 3 < n) {
if (p[j + 3]) {
break;
}
}
is_wide = true;
j++;
}
}
name[4 + n] = '\0';
//handle wide strings
//memcpy (name + 4, (const char *)p, n);
if (is_wide) {
r_meta_add (core->anal, R_META_TYPE_STRING, off,
off + (n * 2), (const char *)name + 4);
} else {
r_meta_add (core->anal, R_META_TYPE_STRING, off,
off + n, (const char *)name + 4);
}
r_name_filter (name, n + 10);
r_flag_set (core->flags, name, off, n);
wordsize = n;
free (name);
}
break;
case 'd': // TODO: check
r_meta_cleanup (core->anal, off, off+plen);
r_meta_add (core->anal, R_META_TYPE_DATA, off, off+plen, "");
break;
case 'c': // TODO: check
r_meta_cleanup (core->anal, off, off + plen);
r_meta_add (core->anal, R_META_TYPE_CODE, off, off + plen, "");
break;
case 'u':
r_core_anal_undefine (core, off);
break;
case 'f':
{
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, core->offset, 0);
if (fcn) {
r_anal_fcn_resize (core->anal, fcn, core->offset - fcn->addr);
}
r_cons_break_push (NULL, NULL);
r_core_cmdf (core, "af @ 0x%08" PFMT64x, off); // required for thumb autodetection
r_cons_break_pop ();
}
break;
case 'v':
{
ut64 N;
char *endptr = NULL;
char *end_off = r_cons_input ("Last hexadecimal digits of instruction: ");
if (end_off) {
N = strtoull (end_off, &endptr, 16);
}
if (!end_off || end_off == endptr) {
eprintf ("Invalid numeric input\n");
r_cons_any_key (NULL);
free (end_off);
break;
}
free (end_off);
ut64 incr = 0x10;
ut64 tmp_N = N >> 4;
while (tmp_N > 0) {
tmp_N = tmp_N >> 4;
incr = incr << 4;
}
ut64 mask = incr - 1;
ut64 start_off = (off & ~mask) ^ N;
if ((off & mask) > N) {
if (start_off > incr) {
start_off -= incr;
} else {
start_off = N;
}
}
ut64 try_off;
bool found = false;
RAnalOp *op = NULL;
for (try_off = start_off; try_off < start_off + incr*16; try_off += incr) {
r_anal_op_free (op);
op = r_core_anal_op (core, try_off, R_ANAL_OP_MASK_ALL);
if (!op) {
break;
}
if (op->var) {
found = true;
break;
}
}
if (found) {
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, off, 0);
if (fcn) {
RAnalVar *bar = r_anal_var_get_byname (core->anal, fcn->addr, op->var->name);
if (bar) {
char *newname = r_cons_input (sdb_fmt ("New variable name for '%s': ", bar->name));
if (newname && *newname) {
r_anal_var_rename (core->anal, fcn->addr, bar->scope,
bar->kind, bar->name, newname, true);
free (newname);
}
} else {
eprintf ("Cannot find variable\n");
r_cons_any_key (NULL);
}
} else {
eprintf ("Cannot find function\n");
r_cons_any_key (NULL);
}
} else {
eprintf ("Cannot find instruction with a variable\n");
r_cons_any_key (NULL);
}
r_anal_op_free (op);
break;
}
case 'Q':
case 'q':
default:
if (IS_DIGIT (ch)) {
if (rep < 0) {
rep = 0;
}
rep = rep * 10 + atoi ((char *)&ch);
goto repeat;
}
break;
}
if (distance > 0) {
distance--;
off += wordsize;
goto onemoretime;
}
}
R_API void r_core_visual_colors(RCore *core) {
char *color = calloc (1, 64), cstr[32];
char preview_cmd[128] = "pd $r";
int ch, opt = 0, oopt = -1;
bool truecolor = r_cons_singleton ()->context->color_mode == COLOR_MODE_16M;
char *rgb_xxx_fmt = truecolor ? "rgb:%2.2x%2.2x%2.2x ":"rgb:%x%x%x ";
const char *k;
RColor rcolor;
r_cons_show_cursor (false);
rcolor = r_cons_pal_get_i (opt);
for (;;) {
r_cons_clear ();
r_cons_gotoxy (0, 0);
k = r_cons_pal_get_name (opt);
if (!k) {
opt = 0;
k = r_cons_pal_get_name (opt);
}
if (!truecolor) {
rcolor.r &= 0xf;
rcolor.g &= 0xf;
rcolor.b &= 0xf;
rcolor.r2 &= 0xf;
rcolor.g2 &= 0xf;
rcolor.b2 &= 0xf;
} else {
rcolor.r &= 0xff;
rcolor.g &= 0xff;
rcolor.b &= 0xff;
rcolor.r2 &= 0xff;
rcolor.g2 &= 0xff;
rcolor.b2 &= 0xff;
}
sprintf (color, rgb_xxx_fmt, rcolor.r, rcolor.g, rcolor.b);
if (rcolor.r2 || rcolor.g2 || rcolor.b2) {
color = r_str_appendf (color, rgb_xxx_fmt, rcolor.r2, rcolor.g2, rcolor.b2);
rcolor.a = ALPHA_FGBG;
} else {
rcolor.a = ALPHA_FG;
}
r_cons_rgb_str (cstr, sizeof (cstr), &rcolor);
char *esc = strchr (cstr + 1, '\x1b');
char *curtheme = r_core_get_theme ();
r_cons_printf ("# Use '.' to randomize current color and ':' to randomize palette\n");
r_cons_printf ("# Press '"Color_RED"rR"Color_GREEN"gG"Color_BLUE"bB"Color_RESET
"' or '"Color_BGRED"eE"Color_BGGREEN"fF"Color_BGBLUE"vV"Color_RESET
"' to change foreground/background color\n");
r_cons_printf ("# Export colorscheme with command 'ec* > filename'\n");
r_cons_printf ("# Preview command: '%s' - Press 'c' to change it\n", preview_cmd);
r_cons_printf ("# Selected colorscheme : %s - Use 'hl' or left/right arrow keys to change colorscheme\n", curtheme ? curtheme : "default");
r_cons_printf ("# Selected element: %s - Use 'jk' or up/down arrow keys to change element\n", k);
r_cons_printf ("# ec %s %s # %d (\\x1b%.*s)",
k, color, atoi (cstr+7), esc ? esc - cstr - 1 : strlen (cstr + 1), cstr+1);
if (esc) {
r_cons_printf (" (\\x1b%s)", esc + 1);
}
r_cons_newline ();
r_core_cmdf (core, "ec %s %s", k, color);
char * res = r_core_cmd_str (core, preview_cmd);
int h, w = r_cons_get_size (&h);
char *body = r_str_ansi_crop (res, 0, 0, w, h - 8);
if (body) {
r_cons_printf ("\n%s", body);
}
r_cons_flush ();
ch = r_cons_readchar ();
ch = r_cons_arrow_to_hjkl (ch);
switch (ch) {
#define CASE_RGB(x,X,y) \
case x:if ((y) > 0x00) { (y)--; } break;\
case X:if ((y) < 0xff) { (y)++; } break;
CASE_RGB ('R','r',rcolor.r);
CASE_RGB ('G','g',rcolor.g);
CASE_RGB ('B','b',rcolor.b);
CASE_RGB ('E','e',rcolor.r2);
CASE_RGB ('F','f',rcolor.g2);
CASE_RGB ('V','v',rcolor.b2);
case 'Q':
case 'q':
free (body);
free (color);
return;
case 'k':
opt--;
break;
case 'j':
opt++;
break;
case 'l':
r_core_cmd0 (core, "ecn");
oopt = -1;
break;
case 'h':
r_core_cmd0 (core, "ecp");
oopt = -1;
break;
case 'K':
opt = 0;
break;
case 'J':
opt = r_cons_pal_len () - 1;
break;
case ':':
r_cons_pal_random ();
break;
case '.':
rcolor.r = r_num_rand (0xff);
rcolor.g = r_num_rand (0xff);
rcolor.b = r_num_rand (0xff);
break;
case 'c':
r_line_set_prompt ("Preview command> ");
r_cons_show_cursor (true);
r_cons_fgets (preview_cmd, sizeof preview_cmd, 0, NULL);
r_cons_show_cursor (false);
}
if (opt != oopt) {
rcolor = r_cons_pal_get_i (opt);
oopt = opt;
}
free (body);
}
}