radare2/libr/core/vmenus.c

3843 lines
90 KiB
C

/* radare - LGPL - Copyright 2009-2018 - 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 *res = r_print_hexpair (core->print, r_asm_op_get_hex (&asmop), -1);
r_cons_printf ("hex: %s\n"Color_RESET, res);
free (res);
}
{
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;
}
static bool edit_bits (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 *res = r_print_hexpair (core->print, r_asm_op_get_hex (&asmop), -1);
r_cons_printf ("hex: %s\n"Color_RESET, res);
free (res);
}
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 *res = r_print_hexpair (core->print, r_asm_op_get_hex (&asmop), -1);
r_core_cmdf (core, "wx %02x%02x%02x%02x", buf[0], buf[1], buf[2], buf[3]);
free (res);
}
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\n", color_sel,
Color_RESET, pre, k+strlen (s), v);
} else {
r_cons_printf (" %s %s %s\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;
int i = 0;
int skip = idx - 10;
switch (mode) {
case 'c':
r_cons_printf ("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;
}
}
return cur;
case 'f':
// show fields
r_cons_printf ("Fields:\n\n");
break;
case 'm':
// show methods
if (!_c) {
eprintf ("No class defined\n");
return mur;
}
r_cons_printf ("MethodsFor: %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 (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;
}
}
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;
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 '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 'h':
case 127: // backspace
case 'b': // back
case 'Q':
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 (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"
" 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");
int delta = 0;
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\n", 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 *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 ("Current Gadget:\n");
// get comment
char *output = r_core_cmd_strf (core, "pd 10 @ 0x%08"PFMT64x, addr + delta);
r_cons_strcat_at (output, 0, 11, scr_w, 10);
free (output);
}
}
int count = 0;
r_cons_flush ();
r_cons_gotoxy (0, 22);
r_cons_printf ("ROPChain:\n");
if (chainstr) {
r_cons_printf ("%s\n", chainstr);
}
r_list_foreach (core->ropchain, iter, msg) {
int extra = strlen (chainstr) / scr_w;
r_cons_gotoxy (0, extra + 24 + 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
{
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;
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");
}
}
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, "pi 4 @ 0x%08"PFMT64x, addr + delta);
r_str_replace_char (line, '\n', ';');
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 && ((node->flags & CN_INT) || (node->flags & CN_OFFT))) {
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 (node->flags & CN_BOOL) {
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 '/':
if (root) R_FREE (root);
root = strdup ("/");
strncpy (path, root, sizeof (path)-1);
R_FREE (root);
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);
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_fcn_size (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);
}
// In visual mode, display function list
static ut64 var_functions_show(RCore *core, int idx, int show) {
int wdelta = (idx > 5)? idx - 5: 0;
ut64 seek = core->offset;
ut64 addr = core->offset;
RAnalFunction *fcn;
int window, i = 0;
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) {
if (i >= wdelta) {
if (i> window+wdelta) {
r_cons_printf ("...\n");
break;
}
if (idx == i) {
addr = fcn->addr;
}
if (show) {
if (color) {
r_cons_printf ("%c%c %s0x%08"PFMT64x"" Color_RESET" %4d %s%s"Color_RESET"\n",
(seek == fcn->addr)?'>':' ',
(idx==i)?'*':' ',
color_addr, fcn->addr, r_anal_fcn_realsize (fcn),
color_fcn, fcn->name);
} else {
r_cons_printf ("%c%c 0x%08"PFMT64x" %4d %s\n",
(seek == fcn->addr)?'>':' ',
(idx==i)?'*':' ',
fcn->addr, r_anal_fcn_realsize (fcn), fcn->name);
}
}
}
i++;
}
return addr;
}
// In visual mode, display the variables.
static ut64 var_variables_show(RCore* core, int idx, int *vindex, int show) {
int i = 0;
const ut64 addr = var_functions_show (core, idx, 0);
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) {
r_cons_printf ("%s%s %s %s @ %s%s0x%x\n",
i == *vindex ? "* ":" ",
var->kind == 'v'?"var":"arg",
var->type, var->name,
core->anal->reg->name[R_REG_NAME_BP],
(var->kind == 'v')?"-":"+",
var->delta);
}
}
++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);
// 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 ut64 r_core_visual_anal_refresh (RCore *core) {
if (!core) {
return 0LL;
}
ut64 addr;
char old[1024];
bool color = r_config_get_i (core->config, "scr.color");
int cols = r_cons_get_size (NULL);
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:
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_cons_printf (
"(a) add (x) xrefs (q) quit (jk) next/prev\n"
"(r) rename (c) calls (g) go (tab) column\n"
"(d) delete (v) vars (?) help (:) enter cmd\n");
addr = var_functions_show (core, option, 1);
break;
case 1:
r_cons_printf (
"-[ variables ]----- 0x%08"PFMT64x"\n"
"(a) add (x)xrefs \n"
"(r) rename (g)go \n"
"(d) delete (q)quit \n", addr);
addr = var_variables_show (core, option, &variable_option, 1);
// var_index_show (core->anal, fcn, addr, option);
break;
case 2:
r_cons_printf ("Press 'q' to quit call refs\n");
r_cons_printf ("-[ calls ]----------------------- 0x%08"PFMT64x" (TODO)\n", addr);
// TODO: filter only the callrefs. but we cant grep here
sprintf (old, "afi @ 0x%08"PFMT64x, addr);
r_core_cmd0 (core, old);
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, (RCoreTaskOneShot) r_core_visual_anal_refresh, core);
}
/* 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_clear00 ();
r_cons_printf (
"Usage: vv\n"
"Actions supported:\n"
" functions: Add, Modify, Delete, Xrefs Calls Vars\n"
" variables: Add, Modify, Delete\n"
"Keys:\n"
" j,k select next/prev item or scroll if tab pressed\n"
" J,K scroll next/prev page \"\"\n"
" h,q go back, quit\n"
" p,P switch next/prev print mode\n"
" v view selected function arguments and variables\n"
" x,X see xrefs to the selected function\n"
" tab toggle disasm column selection (to scroll in code)\n"
" ! run 'afls' to sort all functions by address\n"
" . seek to current function address\n"
" : run r2 commands\n"
" _ hud mode. same as: s $(afl~...)\n"
" enter enter function view (variables), xrefs\n"
);
r_cons_flush ();
r_cons_any_key (NULL);
break;
case 9:
selectPanel = !selectPanel;
if (!selectPanel) {
delta = 0;
printMode = 0;
}
break;
case ':':
r_core_visual_prompt (core);
r_cons_any_key (NULL);
continue;
case 'a':
switch (level) {
case 0:
eprintf ("TODO: Add new function manually\n");
/*
r_cons_show_cursor (true);
r_cons_set_raw (false);
r_line_set_prompt ("Address: ");
if (!r_cons_fgets (old, sizeof (old), 0, NULL)) break;
old[strlen (old)-1] = 0;
if (!*old) break;
addr = r_num_math (core->num, old);
r_line_set_prompt ("Size: ");
if (!r_cons_fgets (old, sizeof (old), 0, NULL)) break;
old[strlen (old)-1] = 0;
if (!*old) break;
size = r_num_math (core->num, old);
r_line_set_prompt ("Name: ");
if (!r_cons_fgets (old, sizeof (old), 0, NULL)) break;
old[strlen (old)-1] = 0;
r_flag_set (core->flags, old, addr, 0, 0);
//XXX sprintf(cmd, "CF %lld @ 0x%08llx", size, addr);
// XXX r_core_cmd0(core, cmd);
r_cons_set_raw (true);
r_cons_show_cursor (false);
*/
break;
case 1:
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 '.':
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':
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 '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 '!':
r_core_cmd0 (core, "afls");
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_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"
," b set as byte"
," B set as short word (2 bytes)"
," c set as code"
," C define flag color (fc)"
," d set as data"
," e end of function"
," f analyze function"
," F format"
," 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':
edit_bits (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 'b':
if (plen != core->blocksize) {
rep = plen / 2;
}
define_data_ntimes (core, off, rep, R_BYTE_DATA);
wordsize = 1;
break;
case 'B':
if (plen != core->blocksize) {
rep = plen;
}
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_fcn_at (core->anal, tgt_addr, R_ANAL_FCN_TYPE_NULL);
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_color (core->flags, 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;
case 'h': // "Vdh"
handleHints (core);
//r_core_cmdf (core, "?i highlight;e scr.highlight=`yp` @ 0x%08"PFMT64x, off);
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':
{
int funsize = 0;
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);
}
//int depth = r_config_get_i (core->config, "anal.depth");
if (core->print->cur_enabled) {
if (core->print->ocur != -1) {
funsize = 1 + R_ABS (core->print->cur - core->print->ocur);
}
//depth = 0;
}
r_cons_break_push (NULL, NULL);
r_core_cmdf (core, "af @ 0x%08" PFMT64x, off); // required for thumb autodetection
r_cons_break_pop ();
if (funsize) {
RAnalFunction *f = r_anal_get_fcn_in (core->anal, off, -1);
r_anal_fcn_set_size (core->anal, f, funsize);
}
}
break;
case 'v':
{
RAnalOp op;
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;
if ((off & mask) <= N) {
start_off = (off & ~mask) ^ N;
} else {
start_off = ((off & ~mask) ^ incr) ^ N;
}
ut64 try_off;
bool found = false;
for (try_off = start_off; try_off < start_off + incr*16; try_off += incr) {
r_anal_op (core->anal, &op, try_off,
core->block + try_off - core->offset, 32, R_ANAL_OP_MASK_ALL);
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_fini (&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 == 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);
}
}