radare2/libr/core/visual.c
2016-10-17 13:11:46 +02:00

2380 lines
60 KiB
C

/* radare - LGPL - Copyright 2009-2016 - pancake */
#include <r_core.h>
#include <string.h>
#define NPF 7
static int obs = 0;
static int blocksize = 0;
static int autoblocksize = 1;
static void visual_refresh(RCore *core);
#define PIDX (R_ABS(core->printidx%NPF))
#define debugfmt_default "?0;f tmp;sr SP;pxw 64;?1;dr=;?1;s-;s tmp;f-tmp;pd $r"
static const char *printfmt[] = {
"x", "pd $r",
debugfmt_default,
"pCd $r-1", "pxw", "pc", "pxA", "pxa"
};
#undef USE_THREADS
#define USE_THREADS 1
#if USE_THREADS
static int visual_repeat_thread(RThread *th) {
RCore *core = th->user;
int i = 0;
for (;;) {
if (core->cons->breaked) {
break;
}
visual_refresh (core);
r_cons_flush ();
r_cons_gotoxy (0, 0);
r_cons_printf ("[@%d] ", i++);
r_cons_flush ();
r_sys_sleep (1);
}
r_th_kill (th, 1);
return 0;
}
static void toggle_bits(RCore *core) {
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, core->offset, R_ANAL_FCN_TYPE_NULL);
if (fcn) {
int bits = fcn->bits? fcn->bits: core->assembler->bits;
switch (bits) {
case 16: bits = 32; break;
case 32: bits = 64; break;
default: bits = 16; break;
}
fcn->bits = bits;
return;
}
switch (core->assembler->bits) {
case 8:
r_config_set_i (core->config, "asm.bits", 16);
if (core->assembler->bits != 16) {
r_config_set_i (core->config, "asm.bits", 32);
if (core->assembler->bits != 32) {
r_config_set_i (core->config, "asm.bits", 64);
}
}
break;
case 16:
r_config_set_i (core->config, "asm.bits", 32);
if (core->assembler->bits != 32) {
r_config_set_i (core->config, "asm.bits", 64);
if (core->assembler->bits != 64) {
r_config_set_i (core->config, "asm.bits", 8);
}
}
break;
case 32:
r_config_set_i (core->config, "asm.bits", 64);
if (core->assembler->bits != 64) {
r_config_set_i (core->config, "asm.bits", 8);
if (core->assembler->bits != 8) {
r_config_set_i (core->config, "asm.bits", 16);
}
}
break;
case 64:
r_config_set_i (core->config, "asm.bits", 8);
if (core->assembler->bits != 8) {
r_config_set_i (core->config, "asm.bits", 16);
if (core->assembler->bits != 16) {
r_config_set_i (core->config, "asm.bits", 32);
}
}
break;
}
}
static void visual_repeat(RCore *core) {
int atport = r_config_get_i (core->config, "scr.atport");
if (atport) {
#if __UNIX__ && !__APPLE__
int port = r_config_get_i (core->config, "http.port");
if (!r_core_rtr_http (core, '&', NULL)) {
const char *xterm = r_config_get (core->config, "cmd.xterm");
// TODO: this must be configurable
r_sys_cmdf ("%s 'r2 -C http://localhost:%d/cmd/V;sleep 1' &", xterm, port);
//xterm -bg black -fg gray -e 'r2 -C http://localhost:%d/cmd/;sleep 1' &", port);
} else {
r_cons_any_key (NULL);
}
#else
eprintf ("Unsupported on this platform\n");
r_cons_any_key (NULL);
#endif
} else {
RThread *th = r_th_new (visual_repeat_thread, core, 0);
if (!th) return;
r_th_start (th, 1);
r_cons_break (NULL, NULL);
r_cons_any_key (NULL);
eprintf ("^C \n");
core->cons->breaked = true;
r_th_wait (th);
r_cons_break_end ();
}
}
#endif
static void showcursor(RCore *core, int x) {
if (core && core->vmode) {
r_cons_show_cursor (x);
if (x) {
// TODO: cache this
int wheel = r_config_get_i (core->config, "scr.wheel");
if (wheel) r_cons_enable_mouse (true);
else r_cons_enable_mouse (false);
} else {
r_cons_enable_mouse (false);
}
} else r_cons_enable_mouse (false);
r_cons_flush ();
}
static int color = 1;
static int debug = 1;
static int zoom = 0;
R_API int r_core_visual_hud(RCore *core) {
const char *c = r_config_get (core->config, "hud.path");
const char *f = R2_LIBDIR"/radare2/"R2_VERSION"/hud/main";
int use_color = core->print->flags & R_PRINT_FLAGS_COLOR;
char *homehud = r_str_home (R2_HOMEDIR"/hud");
char *res = NULL;
char *p = 0;
showcursor (core, true);
if (c && *c && r_file_exists (c))
res = r_cons_hud_file (c, use_color);
if (!res && homehud)
res = r_cons_hud_file (homehud, use_color);
if (!res && r_file_exists (f))
res = r_cons_hud_file (f, use_color);
if (!res)
r_cons_message ("Cannot find hud file");
r_cons_clear ();
if (res) {
p = strchr (res, '\t');
r_cons_println (res);
r_cons_flush ();
if (p) r_core_cmd0 (core, p+1);
free (res);
}
showcursor (core, false);
r_cons_flush ();
free (homehud);
return (int)(size_t)p;
}
static int visual_help() {
r_cons_clear00 ();
return r_cons_less_str (
"Visual mode help:\n"
" ? show this help or enter the userfriendly hud\n"
" & rotate asm.bits between supported 8, 16, 32, 64\n"
" % in cursor mode finds matching pair, otherwise toggle autoblocksz\n"
" @ set cmd.vprompt to run commands before the visual prompt\n"
" ! enter into the visual panels mode\n"
" _ enter the flag/comment/functions/.. hud (same as VF_)\n"
" = set cmd.vprompt (top row)\n"
" | set cmd.cprompt (right column)\n"
" . seek to program counter\n"
" / in cursor mode search in current block\n"
" :cmd run radare command\n"
" ;[-]cmt add/remove comment\n"
" ,file add a link to the text file\n"
" /*+-[] change block size, [] = resize hex.cols\n"
" >||< seek aligned to block size\n"
" a/A (a)ssemble code, visual (A)ssembler\n"
" b toggle breakpoint\n"
" B enumerate and inspect classes\n"
" c/C toggle (c)ursor and (C)olors\n"
" d[f?] define function, data, code, ..\n"
" D enter visual diff mode (set diff.from/to)\n"
" e edit eval configuration variables\n"
" f/F set/unset or browse flags. f- to unset, F to browse, ..\n"
" gG go seek to begin and end of file (0-$s)\n"
" hjkl move around (or HJKL) (left-down-up-right)\n"
" i insert hex or string (in hexdump) use tab to toggle\n"
" mK/'K mark/go to Key (any key)\n"
" M walk the mounted filesystems\n"
" n/N seek next/prev function/flag/hit (scr.nkey)\n"
" o go/seek to given offset\n"
" O toggle asm.esil\n"
" p/P rotate print modes (hex, disasm, debug, words, buf)\n"
" q back to radare shell\n"
" r browse anal info and comments\n"
" R randomize color palette (ecr)\n"
" sS step / step over\n"
" T enter textlog chat console (TT)\n"
" uU undo/redo seek\n"
" v visual function/vars code analysis menu\n"
" V (V)iew graph using cmd.graph (agv?)\n"
" wW seek cursor to next/prev word\n"
" xX show xrefs/refs of current function from/to data/code\n"
" yY copy and paste selection\n"
" z fold/unfold comments in disassembly\n"
" Z toggle zoom mode\n"
" Enter follow address of jump/call\n"
"Function Keys: (See 'e key.'), defaults to:\n"
" F2 toggle breakpoint\n"
" F4 run to cursor\n"
" F7 single step\n"
" F8 step over\n"
" F9 continue\n",
"?");
r_cons_flush ();
r_cons_clear00 ();
}
static void prompt_read (const char *p, char *buf, int buflen) {
if (!buf || buflen < 1)
return;
*buf = 0;
r_line_set_prompt (p);
showcursor (NULL, true);
r_cons_fgets (buf, buflen, 0, NULL);
showcursor (NULL, false);
}
R_API void r_core_visual_prompt_input (RCore *core) {
int ret;
ut64 addr = core->offset;
ut64 bsze = core->blocksize;
int h;
(void)r_cons_get_size (&h);
r_cons_gotoxy (0, h - 2);
r_cons_reset_colors ();
r_cons_printf ("\nPress <enter> to return to Visual mode.\n");
r_cons_show_cursor (true);
core->vmode = false;
ut64 newaddr = addr;
if (core->print->cur_enabled) {
if (core->print->ocur != -1) {
int newsz = core->print->cur - core->print->ocur;
newaddr = core->offset + core->print->ocur;
r_core_block_size (core, newsz);
} else newaddr = core->offset + core->print->cur;
r_core_seek (core, newaddr, 1);
}
do {
ret = r_core_visual_prompt (core);
if (core->offset != newaddr) {
// do not restore seek anymore
newaddr = addr;
}
} while (ret);
if (core->print->cur_enabled) {
if (addr != newaddr) {
r_core_seek (core, addr, 1);
r_core_block_size (core, bsze);
}
}
r_cons_show_cursor (false);
core->vmode = true;
}
R_API int r_core_visual_prompt (RCore *core) {
char buf[1024];
int ret;
#if __UNIX__
r_line_set_prompt (Color_RESET":> ");
#else
r_line_set_prompt (":> ");
#endif
showcursor (core, true);
r_cons_fgets (buf, sizeof (buf), 0, NULL);
if (!strcmp (buf, "q")) {
ret = false;
} else if (*buf) {
r_line_hist_add (buf);
r_core_cmd (core, buf, 0);
r_cons_flush ();
ret = true;
} else {
ret = false;
//r_cons_any_key (NULL);
r_cons_clear00 ();
showcursor (core, false);
}
return ret;
}
static int visual_nkey(RCore *core, int ch) {
const char *cmd;
ut64 oseek = UT64_MAX;
if (core->print->ocur == -1) {
oseek = core->offset;
r_core_seek (core, core->offset + core->print->cur, 0);
}
switch (ch) {
case R_CONS_KEY_F1:
cmd = r_config_get (core->config, "key.f1");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
else ch = '?';
break;
case R_CONS_KEY_F2:
cmd = r_config_get (core->config, "key.f2");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
break;
case R_CONS_KEY_F3:
cmd = r_config_get (core->config, "key.f3");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
break;
case R_CONS_KEY_F4:
cmd = r_config_get (core->config, "key.f4");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
else {
if (core->print->cur_enabled) {
// dcu 0xaddr
r_core_cmdf (core, "dcu 0x%08"PFMT64x, core->offset + core->print->cur);
core->print->cur_enabled = 0;
}
}
break;
case R_CONS_KEY_F5:
cmd = r_config_get (core->config, "key.f5");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
break;
case R_CONS_KEY_F6:
cmd = r_config_get (core->config, "key.f6");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
break;
case R_CONS_KEY_F7:
cmd = r_config_get (core->config, "key.f7");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
else ch = 's';
break;
case R_CONS_KEY_F8:
cmd = r_config_get (core->config, "key.f8");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
break;
case R_CONS_KEY_F9:
cmd = r_config_get (core->config, "key.f9");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
else r_core_cmd0 (core, "dc");
break;
case R_CONS_KEY_F10:
cmd = r_config_get (core->config, "key.f10");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
break;
case R_CONS_KEY_F11:
cmd = r_config_get (core->config, "key.f11");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
break;
case R_CONS_KEY_F12:
cmd = r_config_get (core->config, "key.f12");
if (cmd && *cmd) ch = r_core_cmd0 (core, cmd);
break;
}
if (oseek != UT64_MAX) {
r_core_seek (core, oseek, 0);
}
return ch;
}
static void setcursor (RCore *core, bool cur) {
int flags = core->print->flags; // wtf
if (core->print->cur_enabled) flags |= R_PRINT_FLAGS_CURSOR;
else flags &= ~(R_PRINT_FLAGS_CURSOR);
core->print->cur_enabled = cur;
if (core->print->cur == -1) {
core->print->cur = 0;
}
r_print_set_flags (core->print, flags);
core->print->col = core->print->cur_enabled ? 1 : 0;
}
static void setdiff (RCore *core) {
char from[64], to[64];
prompt_read ("diff from: ", from, sizeof (from));
r_config_set (core->config, "diff.from", from);
prompt_read ("diff to: ", to, sizeof (to));
r_config_set (core->config, "diff.to", to);
}
static void findPair (RCore *core) {
ut8 buf[256];
int i, len, d = core->print->cur + 1;
int delta = 0;
const ut8 *p, *q = NULL;
const char *keys = "{}[]()<>";
ut8 ch = core->block[core->print->cur];
p = (const ut8*)strchr (keys, ch);
if (p) {
delta = (size_t)(p-(const ut8*)keys);
ch = (delta%2 && p != (const ut8*)keys)? p[-1]: p[1];
}
len = 1;
buf[0] = ch;
if (p && (delta%2)) {
for (i = d-1; i>=0; i--) {
if (core->block[i] == ch) {
q = core->block + i;
break;
}
}
} else {
q = r_mem_mem (core->block+d, core->blocksize-d,
(const ut8*)buf, len);
if (!q) {
q = r_mem_mem (core->block, R_MIN (core->blocksize, d),
(const ut8*)buf, len);
}
}
if (q) {
core->print->cur = (int)(size_t)(q-core->block);
core->print->ocur = -1;
showcursor (core, true);
}
}
static void findNextWord (RCore *core) {
int i, d = core->print->cur_enabled ? core->print->cur : 0;
for (i = d+1; i<core->blocksize; i++) {
switch (core->block[i]) {
case ' ':
case '.':
case '\t':
case '\n':
if (core->print->cur_enabled) {
core->print->cur = i + 1;
core->print->ocur = -1;
showcursor (core, true);
} else {
r_core_seek (core, core->offset + i + 1, 1);
}
return;
}
}
}
static int isSpace (char ch) {
switch (ch) {
case ' ':
case '.':
case ',':
case '\t':
case '\n':
return 1;
}
return 0;
}
static void findPrevWord (RCore *core) {
int i = core->print->cur_enabled ? core->print->cur : 0;
while (i>1) {
if (isSpace (core->block[i]))
i--;
else if (isSpace (core->block[i-1]))
i-=2;
else break;
}
for (; i>=0; i--) {
if (isSpace (core->block[i])) {
if (core->print->cur_enabled) {
core->print->cur = i + 1;
core->print->ocur = -1;
showcursor (core, true);
} else {
// r_core_seek (core, core->offset + i + 1, 1);
}
break;
}
}
}
// TODO: integrate in '/' command with search.inblock ?
static void visual_search (RCore *core) {
const ut8 *p;
int len, d = core->print->cur;
char str[128], buf[258];
r_line_set_prompt ("search byte/string in block: ");
r_cons_fgets (str, sizeof (str), 0, NULL);
len = r_hex_str2bin (str, (ut8*)buf);
if (*str=='"') {
char *e = strncpy (buf, str+1, sizeof (buf)-1);
if (e) { --e; if (*e=='"') *e=0; }
len = strlen (buf);
} else
if (len<1) {
strncpy (buf, str, sizeof (buf)-1);
len = strlen (str);
}
p = r_mem_mem (core->block+d, core->blocksize-d,
(const ut8*)buf, len);
if (p) {
core->print->cur = (int)(size_t)(p-core->block);
if (len>1) {
core->print->ocur = core->print->cur+len-1;
} else core->print->ocur = -1;
showcursor (core, true);
eprintf ("FOUND IN %d\n", core->print->cur);
r_cons_any_key (NULL);
} else {
eprintf ("Cannot find bytes\n");
r_cons_any_key (NULL);
r_cons_clear00 ();
}
}
R_API void r_core_visual_show_char (RCore *core, char ch) {
if (r_config_get_i (core->config, "scr.feedback")<2)
return;
if (!IS_PRINTABLE (ch))
return;
r_cons_gotoxy (1, 2);
r_cons_printf (".---.\n");
r_cons_printf ("| %c |\n", ch);
r_cons_printf ("'---'\n");
r_cons_flush ();
r_sys_sleep (1);
}
R_API void r_core_visual_seek_animation (RCore *core, ut64 addr) {
r_core_seek (core, addr, 1);
if (r_config_get_i (core->config, "scr.feedback")<1)
return;
if (core->offset == addr)
return;
r_cons_gotoxy (1, 2);
if (addr>core->offset) {
r_cons_printf (".----.\n");
r_cons_printf ("| \\/ |\n");
r_cons_printf ("'----'\n");
} else {
r_cons_printf (".----.\n");
r_cons_printf ("| /\\ |\n");
r_cons_printf ("'----'\n");
}
r_cons_flush();
r_sys_usleep (90000);
}
static void setprintmode (RCore *core, int n) {
RAsmOp op;
if (n>0) {
core->printidx = R_ABS ((core->printidx+1)%NPF);
} else {
if (core->printidx)
core->printidx--;
else core->printidx = NPF-1;
}
switch (core->printidx) {
case 0:
core->inc = 16;
break;
case 1:
case 2:
core->inc = r_asm_disassemble (core->assembler,
&op, core->block, 32);
break;
case 5: // "pxA"
core->inc = 256;
break;
}
}
#define OPDELTA 32
static ut64 prevop_addr (RCore *core, ut64 addr) {
ut8 buf[OPDELTA * 2];
ut64 target, base;
RAnalBlock *bb;
RAnalOp op;
int len, ret, i;
int minop = r_anal_archinfo (core->anal, R_ANAL_ARCHINFO_MIN_OP_SIZE);
int maxop = r_anal_archinfo (core->anal, R_ANAL_ARCHINFO_MAX_OP_SIZE);
if (minop == maxop) {
if (minop == -1) {
return addr - 4;
}
return addr - minop;
}
// let's see if we can use anal info to get the previous instruction
// TODO: look in the current basicblock, then in the current function
// and search in all functions only as a last chance, to try to speed
// up the process.
bb = r_anal_bb_from_offset (core->anal, addr - minop);
if (bb) {
ut64 res = r_anal_bb_opaddr_at (bb, addr - minop);
if (res != UT64_MAX) {
return res;
}
}
// if we anal info didn't help then fallback to the dumb solution.
target = addr;
base = target - OPDELTA;
r_core_read_at (core, base, buf, sizeof (buf));
for (i = 0; i < sizeof (buf); i++) {
ret = r_anal_op (core->anal, &op, base + i,
buf + i, sizeof (buf) - i);
if (!ret) continue;
len = op.size;
r_anal_op_fini (&op); // XXX
if (len < 1) continue;
if (target == base + i + len) {
return base + i;
}
i += len - 1;
}
return target - 4;
}
// Returns true if we can use analysis to find the previous operation address,
// sets prev_addr to the value of the instruction numinstrs back.
// If we can't use the anal, then set prev_addr to UT64_MAX and return false;
R_API bool r_core_prevop_addr (RCore* core, ut64 start_addr, int numinstrs, ut64* prev_addr) {
RAnalBlock* bb;
int i;
// Check that we're in a bb, otherwise this prevop stuff won't work.
bb = r_anal_bb_from_offset (core->anal, start_addr);
if (bb) {
if (r_anal_bb_opaddr_at (bb, start_addr) != UT64_MAX) {
// Do some anal looping.
for (i = 0; i < numinstrs; ++i) {
*prev_addr = prevop_addr (core, start_addr);
start_addr = *prev_addr;
}
return true;
}
}
// Dang! not in a bb, return false and fallback to other methods.
*prev_addr = UT64_MAX;
return false;
}
static void reset_print_cur(RPrint *p) {
p->cur = 0;
p->ocur = -1;
}
static void visual_offset(RCore *core) {
char buf[256];
r_line_set_prompt ("[offset]> ");
strcpy (buf, "s ");
if (r_cons_fgets (buf+2, sizeof (buf)-3, 0, NULL) >0) {
if (buf[2]=='.')buf[1]='.';
r_core_cmd0 (core, buf);
reset_print_cur (core->print);
}
}
static int prevopsz(RCore *core, ut64 addr) {
ut64 prev_addr = prevop_addr (core, addr);
return addr - prev_addr;
}
R_API int r_core_visual_xrefs_x (RCore *core) {
int ret = 0;
#if FCN_OLD
char ch;
int count = 0;
RList *xrefs = NULL;
RAnalRef *refi;
RListIter *iter;
RAnalFunction *fun;
int skip = 0;
int idx = 0;
char cstr[32];
ut64 addr = core->offset;
if (core->print->cur_enabled) {
addr += core->print->cur;
}
repeat:
if ((xrefs = r_anal_xref_get (core->anal, addr))) {
r_cons_clear00 ();
r_cons_gotoxy (1, 1);
r_cons_printf ("[GOTO XREF]> 0x%08"PFMT64x"\n", addr);
if (r_list_empty (xrefs)) {
r_cons_printf ("No XREF found at 0x%"PFMT64x"\n", addr);
r_cons_any_key (NULL);
r_cons_clear00 ();
} else {
int lines;
(void)r_cons_get_size (&lines);
idx = 0;
count = 0;
lines -= 3;
r_list_foreach (xrefs, iter, refi) {
if (idx >= skip) {
if (count > 9) {
strcpy (cstr, "?");
} else {
snprintf (cstr, sizeof (cstr), "%d", count);
}
fun = r_anal_get_fcn_in (core->anal, refi->addr, R_ANAL_FCN_TYPE_NULL);
#if 0
r_cons_printf (" %d [%s] 0x%08"PFMT64x" %s XREF 0x%08"PFMT64x" (%s) \n",
idx, cstr, refi->at,
refi->type==R_ANAL_REF_TYPE_CODE?"CODE (JMP)":
refi->type==R_ANAL_REF_TYPE_CALL?"CODE (CALL)":"DATA", refi->addr,
fun?fun->name:"unk");
#endif
r_cons_printf (" %d [%s] 0x%08"PFMT64x" %s XREF (%s)\n",
idx, cstr, refi->addr,
refi->type==R_ANAL_REF_TYPE_CODE?"CODE (JMP)":
refi->type==R_ANAL_REF_TYPE_CALL?"CODE (CALL)":"DATA",
fun?fun->name:"unk");
if (idx == skip) {
r_core_cmdf (core, "pd 20 @ 0x%08"PFMT64x, refi->addr);
r_cons_column (60);
}
if (++count >= lines) {
r_cons_printf ("...\n");
break;
}
}
idx++;
}
}
} else {
xrefs = NULL;
}
if (!xrefs || !r_list_length (xrefs)) {
r_list_free (xrefs);
return 0;
}
r_cons_flush ();
ch = r_cons_readchar ();
if (ch == 'j') {
skip++;
goto repeat;
} else if (ch == 'k') {
skip--;
if (skip < 0) {
skip = 0;
}
goto repeat;
} else if (ch == ' ' || ch == '\n' || ch == '\r') {
refi = r_list_get_n (xrefs, skip);
if (refi) {
r_core_cmdf (core, "s 0x%"PFMT64x, refi->addr);
ret = 1;
}
} else if (ch >= '0' && ch <= '9') {
refi = r_list_get_n (xrefs, ch - 0x30);
if (refi) {
r_core_cmdf (core, "s 0x%"PFMT64x, refi->addr);
ret = 1;
}
}
r_list_free (xrefs);
#else
eprintf ("TODO: sdbize xrefs here\n");
#endif
return ret;
}
R_API int r_core_visual_xrefs_X (RCore *core) {
int ret = 0;
#if FCN_OLD
char ch;
int count = 0;
RAnalRef *refi;
RListIter *iter;
RAnalFunction *fun;
ut64 addr = core->offset;
if (core->print->cur_enabled) {
addr += core->print->cur;
}
fun = r_anal_get_fcn_in (core->anal, addr, R_ANAL_FCN_TYPE_NULL);
if (fun) {
r_cons_clear00 ();
r_cons_gotoxy (1, 1);
r_cons_printf ("[GOTO REF]> \n");
if (r_list_empty (fun->refs)) {
r_cons_printf ("\tNo REF found at 0x%"PFMT64x"\n", addr);
r_cons_any_key (NULL);
r_cons_clear00 ();
} else {
r_list_foreach (fun->refs, iter, refi) {
RFlagItem *f = r_flag_get_at (core->flags, refi->addr);
if (f) {
eprintf ("%s\n", f->name);
}
r_cons_printf (" [%i] 0x%08"PFMT64x" %s XREF 0x%08"PFMT64x" (%s)(%s) \n", count,
refi->at,
refi->type==R_ANAL_REF_TYPE_CODE?"CODE (JMP)":
refi->type==R_ANAL_REF_TYPE_CALL?"CODE (CALL)":"DATA", refi->addr, fun->name, f?f->name:"");
if (++count > 9) break;
}
}
}
r_cons_flush ();
if (!count)
return 0;
ch = r_cons_readchar ();
if (fun && fun->refs) {
if (ch >= '0' && ch <= '9') {
refi = r_list_get_n (fun->refs, ch-0x30);
if (refi) {
r_core_cmdf (core, "s 0x%"PFMT64x, refi->addr);
ret = 1;
}
}
}
#else
eprintf ("TODO: sdbize this\n");
#endif
return ret;
}
#if __WINDOWS__ && !__CYGWIN__
void SetWindow(int Width, int Height) {
COORD coord;
coord.X = Width;
coord.Y = Height;
SMALL_RECT Rect;
Rect.Top = 0;
Rect.Left = 0;
Rect.Bottom = Height - 1;
Rect.Right = Width - 1;
HANDLE Handle = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleScreenBufferSize(Handle, coord);
SetConsoleWindowInfo(Handle, TRUE, &Rect);
}
#endif
// unnecesarily public
char *getcommapath(RCore *core) {
char *cwd;
const char *dir = r_config_get (core->config, "dir.projects");
const char *prj = r_config_get (core->config, "file.project");
if (dir && *dir && prj && *prj) {
char *abspath = r_file_abspath (dir);
/* use prjdir as base directory for comma-ent files */
cwd = r_str_newf ("%s"R_SYS_DIR"%s.d", abspath, prj);
free (abspath);
} else {
/* use cwd as base directory for comma-ent files */
cwd = r_sys_getdir ();
}
return cwd;
}
static void visual_comma(RCore *core) {
ut64 addr = core->offset + (core->print->cur_enabled ? core->print->cur : 0);
char *comment, *cwd, *cmtfile;
comment = r_meta_get_string (core->anal, R_META_TYPE_COMMENT, addr);
cmtfile = r_str_between (comment, ",(", ")");
cwd = getcommapath (core);
if (!cmtfile) {
char *fn;
showcursor (core, true);
fn = r_cons_input ("<comment-file> ");
showcursor (core, false);
if (fn && *fn) {
cmtfile = strdup (fn);
if (!comment || !*comment) {
comment = r_str_newf (",(%s)", fn);
r_meta_set_string (core->anal, R_META_TYPE_COMMENT, addr, comment);
} else {
// append filename in current comment
char *nc = r_str_newf ("%s ,(%s)", comment, fn);
r_meta_set_string (core->anal, R_META_TYPE_COMMENT, addr, nc);
free (nc);
}
}
free (fn);
}
if (cmtfile) {
char *cwf = r_str_newf ("%s"R_SYS_DIR"%s", cwd, cmtfile);
char *odata = r_file_slurp (cwf, NULL);
char *data = r_core_editor (core, NULL, odata);
r_file_dump (cwf, (const ut8*)data, -1, 0);
free (data);
free (odata);
free (cwf);
} else {
eprintf ("No commafile found.\n");
}
free (comment);
}
static bool isDisasmPrint(int mode) {
return (mode == 1 || mode == 2);
}
static void cursor_ocur(RCore *core, bool use_ocur) {
RPrint *p = core->print;
if (use_ocur && p->ocur == -1) {
p->ocur = p->cur;
} else if (!use_ocur) {
p->ocur = -1;
}
}
static void cursor_nextrow(RCore *core, bool use_ocur) {
RPrint *p = core->print;
ut32 roff, next_roff;
int row, sz, delta;
RAsmOp op;
cursor_ocur (core, use_ocur);
if (PIDX == 2 && core->seltab == 1) {
const int cols = core->dbg->regcols;
p->cur += cols > 0 ? cols: 3;
return;
}
if (p->row_offsets != NULL) {
// FIXME: cache the current row
row = r_print_row_at_off (p, p->cur);
roff = r_print_rowoff (p, row);
if (roff == -1) {
p->cur++;
return;
}
next_roff = r_print_rowoff (p, row + 1);
if (next_roff == -1) {
p->cur++;
return;
}
if (next_roff + 32 < core->blocksize) {
sz = r_asm_disassemble (core->assembler, &op,
core->block + next_roff, 32);
if (sz < 1) sz = 1;
} else {
sz = 1;
}
delta = p->cur - roff;
p->cur = next_roff + R_MIN (delta, sz - 1);
} else {
p->cur += R_MAX (1, p->cols);
}
}
static void cursor_prevrow(RCore *core, bool use_ocur) {
RPrint *p = core->print;
ut32 roff, prev_roff;
int row;
if (PIDX == 2 && core->seltab == 1) {
const int cols = core->dbg->regcols;
p->cur -= cols>0? cols: 4;
return;
}
cursor_ocur (core, use_ocur);
if (p->row_offsets != NULL) {
int delta, prev_sz;
// FIXME: cache the current row
row = r_print_row_at_off (p, p->cur);
roff = r_print_rowoff (p, row);
if (roff == UT32_MAX) {
p->cur--;
return;
}
prev_roff = row > 0 ? r_print_rowoff (p, row - 1) : UT32_MAX;
delta = p->cur - roff;
if (prev_roff == UT32_MAX) {
ut64 prev_addr = prevop_addr (core, core->offset + roff);
if (prev_addr > core->offset) {
prev_roff = 0;
prev_sz = 1;
} else {
RAsmOp op;
prev_roff = 0;
r_core_seek (core, prev_addr, 1);
prev_sz = r_asm_disassemble (core->assembler, &op,
core->block, 32);
}
} else {
prev_sz = roff - prev_roff;
}
p->cur = prev_roff + R_MIN (delta, prev_sz - 1);
} else {
p->cur -= p->cols;
}
}
static void cursor_left(RCore *core, bool use_ocur) {
if (PIDX == 2) {
if (core->seltab == 1) {
core->print->cur--;
return;
}
}
cursor_ocur (core, use_ocur);
core->print->cur--;
}
static void cursor_right(RCore *core, bool use_ocur) {
if (PIDX == 2) {
if (core->seltab == 1) {
core->print->cur++;
return;
}
}
cursor_ocur (core, use_ocur);
core->print->cur++;
}
static bool fix_cursor(RCore *core) {
RPrint *p = core->print;
int offscreen = (core->cons->rows - 3) * p->cols;
bool res = false;
if (!core->print->cur_enabled) {
return false;
}
if (core->print->screen_bounds > 1) {
bool off_is_visible = core->offset < core->print->screen_bounds;
bool cur_is_visible = core->offset + p->cur < core->print->screen_bounds;
bool is_close = core->offset + p->cur < core->print->screen_bounds + 32;
if ((!cur_is_visible && !is_close) || (!cur_is_visible && p->cur == 0)) {
// when the cursor is not visible and it's far from the
// last visible byte, just seek there.
r_core_seek_delta (core, p->cur);
reset_print_cur (p);
} else if ((!cur_is_visible && is_close) || !off_is_visible) {
RAsmOp op;
int sz = r_asm_disassemble (core->assembler,
&op, core->block, 32);
if (sz < 1) sz = 1;
r_core_seek_delta (core, sz);
p->cur = R_MAX (p->cur - sz, 0);
if (p->ocur != -1) p->ocur = R_MAX (p->ocur - sz, 0);
res |= off_is_visible;
}
} else if (core->print->cur >= offscreen) {
r_core_seek (core, core->offset + p->cols, 1);
p->cur -= p->cols;
if (p->ocur != -1) p->ocur -= p->cols;
}
if (p->cur < 0) {
int sz = p->cols;
if (isDisasmPrint (core->printidx)) {
sz = prevopsz (core, core->offset + p->cur);
if (sz < 1) sz = 1;
}
r_core_seek_delta (core, -sz);
p->cur += sz;
if (p->ocur != -1) p->ocur += sz;
}
return res;
}
static bool __ime = false;
static int __nib = -1;
static bool insert_mode_enabled(RCore *core) {
if (!__ime) {
return false;
}
char ch = r_cons_readchar ();
char arrows = r_cons_arrow_to_hjkl (ch);
switch (ch) {
case 127:
core->print->cur = R_MAX (0, core->print->cur - 1);
return true;
case 9: // tab
core->print->col = core->print->col==1? 2: 1;
break;
}
if (ch != 'h' && arrows == 'h') {
core->print->cur = R_MAX (0, core->print->cur - 1);
return true;
} else if (ch != 'l' && arrows == 'l') {
core->print->cur = core->print->cur + 1;
return true;
} else if (ch != 'j' && arrows == 'j') {
cursor_nextrow (core, false);
return true;
} else if (ch != 'k' && arrows == 'k') {
cursor_prevrow (core, false);
return true;
}
if (core->print->col == 2) {
/* ascii column */
if (IS_PRINTABLE (ch)) {
r_core_cmdf (core, "\"w %c\" @ $$+%d", ch, core->print->cur);
core->print->cur ++;
}
return true;
}
ch = arrows;
/* hex column */
switch (ch) {
case '0':
case '1':
case '2':
case '3':
case '4':
case '5':
case '6':
case '7':
case '8':
case '9':
case 'a':
case 'b':
case 'c':
case 'd':
case 'e':
case 'f':
if (__nib != -1) {
r_core_cmdf (core, "wx %c%c @ $$+%d", __nib, ch, core->print->cur);
core->print->cur ++;
__nib = -1;
} else {
r_core_cmdf (core, "wx %c. @ $$+%d", ch, core->print->cur);
__nib = ch;
}
break;
case 'h':
core->print->cur = R_MAX (0, core->print->cur - 1);
break;
case 'l':
core->print->cur = core->print->cur + 1;
break;
case 'j':
cursor_nextrow (core, false);
break;
case 'k':
cursor_prevrow (core, false);
break;
case 'q':
__ime = false;
break;
}
return true;
}
R_API int r_core_visual_cmd(RCore *core, int ch) {
RAsmOp op;
ut64 offset = core->offset;
char buf[4096];
const char *key_s;
int i, ret, cols = core->print->cols, delta = 0;
int wheelspeed;
ch = r_cons_arrow_to_hjkl (ch);
ch = visual_nkey (core, ch);
if (ch<2) return 1;
if (r_cons_singleton()->mouse_event) {
wheelspeed = r_config_get_i (core->config, "scr.wheelspeed");
} else {
wheelspeed = 1;
}
// do we need hotkeys for data references? not only calls?
// '0' is handled to seek at the beginning of the function
if (ch > '0' && ch <= '9') {
char chbuf[2];
ut64 off;
chbuf[0] = ch;
chbuf[1] = '\0';
off = r_core_get_asmqjmps (core, chbuf);
if (off != UT64_MAX) {
int delta = R_ABS ((st64)off-(st64)offset);
r_io_sundo_push (core->io, offset, r_print_get_cursor (core->print));
if (core->print->cur_enabled && delta<100) {
core->print->cur = delta;
} else {
r_core_visual_seek_animation (core, off);
core->print->cur = 0;
}
r_core_block_read (core);
}
} else
switch (ch) {
#if __WINDOWS__ && !__CYGWIN__
case 0xf5:
SetWindow(81,25);
break;
case 0xcf5:
SetWindow(81,40);
break;
#endif
case 0x0d: // "enter" "\\n" "newline"
{
RAnalOp *op;
int wheel = r_config_get_i (core->config, "scr.wheel");
if (wheel)
r_cons_enable_mouse (true);
do {
op = r_core_anal_op (core, core->offset + core->print->cur);
if (op) {
if (op->type == R_ANAL_OP_TYPE_JMP ||
op->type == R_ANAL_OP_TYPE_CJMP ||
op->type == R_ANAL_OP_TYPE_CALL ||
op->type == R_ANAL_OP_TYPE_CCALL) {
if (core->print->cur_enabled) {
int delta = R_ABS ((st64)op->jump-(st64)offset);
if ( op->jump < core->offset || op->jump >= core->print->screen_bounds) {
r_io_sundo_push (core->io, offset, r_print_get_cursor (core->print));
r_core_visual_seek_animation (core, op->jump);
core->print->cur = 0;
} else {
r_io_sundo_push (core->io, offset, r_print_get_cursor (core->print));
core->print->cur = delta;
}
} else {
r_io_sundo_push (core->io, offset, 0);
r_core_visual_seek_animation (core, op->jump);
}
}
}
r_anal_op_free (op);
} while (--wheelspeed>0);
}
break;
case 9: // tab
core->curtab = 0;
if (core->printidx == 2) {
core->print->cur = 0;
core->seltab ++;
if (core->seltab>2) {
core->seltab = 0;
}
} else {
core->seltab = 0;
ut64 f = r_config_get_i (core->config, "diff.from");
ut64 t = r_config_get_i (core->config, "diff.to");
if (f == t && f == 0) {
core->print->col = core->print->col==1? 2: 1;
} else {
ut64 delta = offset - f;
r_core_seek (core, t+delta, 1);
r_config_set_i (core->config, "diff.from", t);
r_config_set_i (core->config, "diff.to", f);
}
}
break;
case 'a':
if (core->file && core->file->desc && !(core->file->desc->flags & 2)) {
r_cons_printf ("\nFile has been opened in read-only mode. Use -w flag\n");
r_cons_any_key (NULL);
return true;
}
r_cons_printf ("Enter assembler opcodes separated with ';':\n");
showcursor (core, true);
r_cons_flush ();
r_cons_set_raw (false);
strcpy (buf, "wa ");
r_line_set_prompt (":> ");
if (r_cons_fgets (buf+3, 1000, 0, NULL) <0) buf[0]='\0';
if (*buf) {
if (core->print->cur_enabled) {
int t = core->offset + core->print->cur;
r_core_seek (core, t, 0);
}
r_core_cmd (core, buf, true);
if (core->print->cur_enabled) {
int t = core->offset - core->print->cur;
r_core_seek (core, t, 1);
}
}
showcursor (core, false);
r_cons_set_raw (true);
break;
case '=':
{ // TODO: edit
const char *buf = NULL;
#define I core->cons
const char *cmd = r_config_get (core->config, "cmd.vprompt");
r_line_set_prompt ("cmd.vprompt> ");
I->line->contents = strdup (cmd);
buf = r_line_readline ();
// if (r_cons_fgets (buf, sizeof (buf)-4, 0, NULL) <0) buf[0]='\0';
I->line->contents = NULL;
r_config_set (core->config, "cmd.vprompt", buf);
}
break;
case '|':
{ // TODO: edit
const char *buf = NULL;
#define I core->cons
const char *cmd = r_config_get (core->config, "cmd.cprompt");
r_line_set_prompt ("cmd.cprompt> ");
I->line->contents = strdup (cmd);
buf = r_line_readline ();
// if (r_cons_fgets (buf, sizeof (buf)-4, 0, NULL) <0) buf[0]='\0';
I->line->contents = NULL;
r_config_set (core->config, "cmd.cprompt", buf);
}
break;
case '!':
r_core_visual_panels (core);
break;
case 'o':
visual_offset (core);
break;
case 'A':
{
int oc = core->print->cur_enabled;
ut64 off = oc ? core->offset + core->print->cur : core->offset;
core->print->cur_enabled = 0;
r_core_visual_asm (core, off);
core->print->cur_enabled = oc;
}
break;
case 'c':
setcursor (core, !core->print->cur_enabled);
break;
case '@':
visual_repeat (core);
break;
case 'C':
color = color? 0: 1;
r_config_set_i (core->config, "scr.color", color);
break;
case 'd':
{
int wheel = r_config_get_i (core->config, "scr.wheel");
if (wheel) r_cons_enable_mouse (false);
r_core_visual_define (core);
if (wheel) r_cons_enable_mouse (true);
}
break;
case 'D':
setdiff (core);
break;
case 'f':
{
int range, min, max;
char name[256], *n;
r_line_set_prompt ("flag name: ");
showcursor (core, true);
if (r_cons_fgets (name, sizeof (name), 0, NULL) >=0 && *name) {
n = r_str_chop (name);
if (core->print->ocur != -1) {
min = R_MIN (core->print->cur, core->print->ocur);
max = R_MAX (core->print->cur, core->print->ocur);
} else {
min = max = core->print->cur;
}
range = max-min+1;
if (!strcmp (n, "-")) {
r_flag_unset_off (core->flags, core->offset + core->print->cur);
} else if (*n=='.') {
if (n[1]=='-') {
//unset
r_core_cmdf (core, "f.-%s@0x%"PFMT64x, n+1, core->offset+min);
} else {
r_core_cmdf (core, "f.%s@0x%"PFMT64x, n+1, core->offset+min);
}
} else if (*n=='-') {
if (*n) r_flag_unset_name (core->flags, n+1);
} else {
if (range<1) range = 1;
if (*n) r_flag_set (core->flags, n,
core->offset + min, range);
}
} }
showcursor (core, false);
break;
case ',':
visual_comma (core);
break;
case 'T':
if (r_sandbox_enable (0)) {
eprintf ("sandbox not enabled\n");
} else {
if (r_config_get_i (core->config, "scr.interactive")) {
r_core_cmd0 (core, "TT");
}
}
break;
case 'n':
r_core_seek_next (core, r_config_get (core->config, "scr.nkey"));
break;
case 'N':
r_core_seek_previous (core, r_config_get (core->config, "scr.nkey"));
break;
case 'i':
case 'I':
if (PIDX == 0) {
if (core->print->ocur == -1) {
__ime = true;
core->print->cur_enabled = true;
return true;
}
}
if (PIDX == 2 && core->seltab == 1) {
char buf[128];
prompt_read ("new-reg-value> ", buf, sizeof (buf));
if (*buf) {
const char *creg = core->dbg->creg;
if (creg) {
r_core_cmdf (core, "dr %s = %s\n", creg, buf);
}
}
return true;
}
if (core->file && core->file->desc &&!(core->file->desc->flags & 2)) {
r_cons_printf ("\nFile has been opened in read-only mode. Use -w flag\n");
r_cons_any_key (NULL);
return true;
}
showcursor (core, true);
r_cons_flush ();
r_cons_set_raw (0);
if (ch=='I') {
strcpy (buf, "wow ");
r_line_set_prompt ("insert hexpair block: ");
if (r_cons_fgets (buf + 4, sizeof (buf) - 5, 0, NULL) < 0) {
buf[0]='\0';
}
char *p = strdup (buf);
int cur = core->print->cur;
if (cur >= core->blocksize) {
cur = core->print->cur - 1;
}
snprintf (buf, sizeof (buf), "%s @ $$0!%i", p,
core->blocksize-core->print->cur);
r_core_cmd (core, buf, 0);
free (p);
break;
}
delta = (core->print->ocur!=-1)? R_MIN (core->print->cur, core->print->ocur): core->print->cur;
if (core->print->col==2) {
strcpy (buf, "\"w ");
r_line_set_prompt ("insert string: ");
if (r_cons_fgets (buf + 3, sizeof (buf)-4, 0, NULL) < 0) {
buf[0]='\0';
}
strcat (buf, "\"");
} else {
r_line_set_prompt ("insert hex: ");
if (core->print->ocur != -1) {
int bs = R_ABS (core->print->cur-core->print->ocur)+1;
core->blocksize = bs;
strcpy (buf, "wow ");
} else {
strcpy (buf, "wx ");
}
if (r_cons_fgets (buf+strlen (buf), sizeof (buf)-strlen (buf), 0, NULL) < 0) {
buf[0]='\0';
}
}
if (core->print->cur_enabled) {
r_core_seek (core, core->offset + delta, 0);
}
r_core_cmd (core, buf, 1);
if (core->print->cur_enabled) {
r_core_seek (core, offset, 1);
}
r_cons_set_raw (1);
showcursor (core, false);
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 'e':
r_core_visual_config (core);
break;
case 'E':
r_core_visual_colors (core);
break;
case 'M':
if (!r_list_empty (core->fs->roots)) {
r_core_visual_mounts (core);
}
break;
case '&':
toggle_bits (core);
break;
case 't':
r_core_visual_types (core);
break;
case 'B':
r_core_visual_classes (core);
break;
case 'F':
r_core_visual_trackflags (core);
break;
case 'x':
r_core_visual_xrefs_x (core);
break;
case 'X':
r_core_visual_xrefs_X (core);
break;
case 'r':
r_core_visual_comments (core);
break;
case ' ':
case 'V':
if (r_config_get_i (core->config, "graph.web")) {
r_core_cmd0 (core, "agv $$");
} else {
RAnalFunction *fun = r_anal_get_fcn_in (core->anal, core->offset, R_ANAL_FCN_TYPE_NULL);
int ocolor = r_config_get_i (core->config, "scr.color");
if (!fun) {
r_cons_message ("Not in a function. Type 'df' to define it here");
break;
} else if (r_list_empty (fun->bbs)) {
r_cons_message ("No basic blocks in this function. You may want to use 'afb+'.");
break;
}
reset_print_cur (core->print);
r_core_visual_graph (core, NULL, NULL, true);
r_config_set_i (core->config, "scr.color", ocolor);
}
break;
case 'v':
r_core_visual_anal (core);
break;
case 'g':
if (core->io->va) {
ut64 offset = r_io_section_get_vaddr (core->io, 0);
if (offset == -1) {
offset = 0;
}
r_core_seek (core, offset, 1);
} else {
r_core_seek (core, 0, 1);
}
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
break;
case 'G':
ret = 0;
int scols = r_config_get_i (core->config, "hex.cols");
if (core->file) {
if (core->io->va) {
ut64 offset = r_io_section_get_vaddr (core->io, 0);
if (offset == UT64_MAX) {
offset = r_io_desc_size (core->io,
core->file->desc)
- core->blocksize + 2*scols;
ret = r_core_seek (core, offset, 1);
} else {
offset += r_io_desc_size (core->io,
core->file->desc)
- core->blocksize + 2 * scols;
ret = r_core_seek (core, offset, 1);
}
} else {
ret = r_core_seek (core,
r_io_desc_size (core->io, core->file->desc)
- core->blocksize + 2 * scols, 1);
}
} else {
ret = -1;
}
if (ret != -1) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
break;
case 'h':
if (core->print->cur_enabled) {
cursor_left (core, false);
} else {
r_core_seek_delta (core, -1);
}
break;
case 'H':
if (core->print->cur_enabled) {
cursor_left (core, true);
} else {
r_core_seek_delta (core, -2);
}
break;
case 'l':
if (core->print->cur_enabled) {
cursor_right (core, false);
} else {
r_core_seek_delta (core, 1);
}
break;
case 'L':
if (core->print->cur_enabled) {
cursor_right (core, true);
} else {
r_core_seek_delta (core, 2);
}
break;
case 'j':
if (core->print->cur_enabled) {
cursor_nextrow (core, false);
} else {
if (r_config_get_i (core->config, "scr.wheelnkey")) {
r_core_cmd0 (core, "sn");
} else {
int times = R_MAX (1, wheelspeed);
while (times--) {
if (isDisasmPrint (core->printidx)) {
RAnalFunction *f = NULL;
if (true) {
f = r_anal_get_fcn_in (core->anal, core->offset, 0);
}
if (f && f->folded) {
cols = core->offset - f->addr + r_anal_fcn_size (f);
} else {
r_asm_set_pc (core->assembler, core->offset);
cols = r_asm_disassemble (core->assembler,
&op, core->block, 32);
}
if (cols < 1) {
cols = op.size;
}
if (cols < 1) {
cols = 1;
}
}
r_core_seek (core, core->offset + cols, 1);
}
}
}
break;
case 'J':
if (core->print->cur_enabled) {
cursor_nextrow (core, true);
} else {
if (core->print->screen_bounds > 1 && core->print->screen_bounds >= core->offset) {
ut64 addr = core->print->screen_bounds;
if (core->print->screen_bounds == core->offset) {
addr += r_asm_disassemble (core->assembler, &op, core->block, 32);
}
r_core_seek (core, addr, 1);
} else {
r_core_seek (core, core->offset + obs, 1);
}
}
break;
case 'k':
if (core->print->cur_enabled) {
cursor_prevrow (core, false);
} else {
if (r_config_get_i (core->config, "scr.wheelnkey")) {
r_core_cmd0 (core, "sp");
} else {
int times = wheelspeed;
if (times<1) times = 1;
while (times--) {
if (isDisasmPrint (core->printidx)) {
RAnalFunction *f = r_anal_get_fcn_in (core->anal, core->offset, R_ANAL_FCN_TYPE_NULL);
if (f && f->folded) {
cols = core->offset - f->addr; // + f->size;
if (cols<1) {
cols = 4;
}
} else {
cols = prevopsz (core, core->offset);
}
}
r_core_seek_delta (core, -cols);
}
}
}
break;
case 'K':
if (core->print->cur_enabled) {
cursor_prevrow (core, true);
} else {
if (core->print->screen_bounds > 1 && core->print->screen_bounds > core->offset) {
int delta = (core->print->screen_bounds - core->offset);
if (core->offset >= delta)
r_core_seek (core, core->offset - delta, 1);
else
r_core_seek (core, 0, 1);
} else {
ut64 at = (core->offset>obs)?core->offset-obs:0;
if (core->offset >obs)
r_core_seek (core, at, 1);
else
r_core_seek (core, 0, 1);
}
}
break;
case '[':
{
int scrcols = r_config_get_i (core->config, "hex.cols");
if (scrcols>2)
r_config_set_i (core->config, "hex.cols", scrcols-2);
}
break;
case ']':
{
int scrcols = r_config_get_i (core->config, "hex.cols");
r_config_set_i (core->config, "hex.cols", scrcols+2);
}
break;
#if 0
case 'I':
r_core_cmd (core, "dsp", 0);
r_core_cmd (core, ".dr*", 0);
break;
#endif
case 's':
key_s = r_config_get (core->config, "key.s");
if (key_s && *key_s) {
r_core_cmd0 (core, key_s);
} else {
if (r_config_get_i (core->config, "cfg.debug")) {
if (core->print->cur_enabled) {
// dcu 0xaddr
r_core_cmdf (core, "dcu 0x%08"PFMT64x, core->offset + core->print->cur);
core->print->cur_enabled = 0;
} else {
r_core_cmd (core, "ds", 0);
r_core_cmd (core, ".dr*", 0);
}
} else {
r_core_cmd (core, "aes", 0);
r_core_cmd (core, ".ar*", 0);
}
}
break;
case 'S':
key_s = r_config_get (core->config, "key.S");
if (key_s && *key_s) {
r_core_cmd0 (core, key_s);
} else {
if (r_config_get_i (core->config, "cfg.debug")) {
if (core->print->cur_enabled) {
r_core_cmd (core, "dcr", 0);
core->print->cur_enabled = 0;
} else {
r_core_cmd (core, "dso", 0);
r_core_cmd (core, ".dr*", 0);
}
} else {
r_core_cmd (core, "aeso", 0);
r_core_cmd (core, ".ar*", 0);
}
}
break;
case 'p':
setprintmode (core, 1);
break;
case 'P':
setprintmode (core, -1);
break;
case '%':
if (core->print->cur_enabled) {
findPair (core);
} else {
/* do nothing? */
autoblocksize = !autoblocksize;
if (autoblocksize)
obs = core->blocksize;
else r_core_block_size (core, obs);
r_cons_clear ();
}
break;
case 'w':
findNextWord (core);
break;
case 'W':
findPrevWord (core);
//r_core_cmd0 (core, "=H");
break;
case 'm':
r_core_visual_mark (core, r_cons_readchar ());
break;
case '\'':
r_core_visual_mark_seek (core, r_cons_readchar ());
break;
case 'y':
if (core->print->ocur==-1) r_core_yank (core, core->offset+core->print->cur, 1);
else r_core_yank (core, core->offset+((core->print->ocur<core->print->cur)?
core->print->ocur:core->print->cur), R_ABS (core->print->cur-core->print->ocur)+1);
break;
case 'Y':
if (!core->yank_buf) {
r_cons_strcat ("Cannot paste, clipboard is empty.\n");
r_cons_flush ();
r_cons_any_key (NULL);
r_cons_clear00 ();
} else r_core_yank_paste (core, core->offset+core->print->cur, 0);
break;
case '0':
{
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, core->offset, R_ANAL_FCN_TYPE_NULL);
if (fcn) {
r_core_seek (core, fcn->addr, 1);
}
}
break;
case '-':
if (core->print->cur_enabled) {
if (core->print->ocur == -1) {
sprintf (buf, "wos 01 @ $$+%i!1",core->print->cur);
} else {
sprintf (buf, "wos 01 @ $$+%i!%i", core->print->cur < core->print->ocur
? core->print->cur
: core->print->ocur,
R_ABS (core->print->ocur - core->print->cur) + 1);
}
r_core_cmd (core, buf, 0);
} else {
if (!autoblocksize) {
r_core_block_size (core, core->blocksize-1);
}
}
break;
case '+':
if (core->print->cur_enabled) {
if (core->print->ocur == -1) {
sprintf (buf, "woa 01 @ $$+%i!1", core->print->cur);
} else {
sprintf (buf, "woa 01 @ $$+%i!%i", core->print->cur < core->print->ocur
? core->print->cur
: core->print->ocur,
R_ABS (core->print->ocur - core->print->cur) + 1);
}
r_core_cmd (core, buf, 0);
} else {
if (!autoblocksize)
r_core_block_size (core, core->blocksize+1);
}
break;
case '/':
if (core->print->cur_enabled) {
visual_search (core);
} else {
if (autoblocksize) {
r_core_cmd0 (core, "?i highlight;e scr.highlight=`?y`");
} else {
r_core_block_size (core, core->blocksize-cols);
}
}
break;
case '*':
if (core->print->cur_enabled) {
r_core_cmdf (core, "dr PC=0x%08"PFMT64x, core->offset + core->print->cur);
} else if (!autoblocksize) {
r_core_block_size (core, core->blocksize+cols);
}
break;
case '>':
r_core_seek_align (core, core->blocksize, 1);
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
break;
case '<':
r_core_seek_align (core, core->blocksize, -1);
r_core_seek_align (core, core->blocksize, -1);
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
break;
case '.':
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
if (core->print->cur_enabled) {
r_core_seek (core, core->offset+core->print->cur, 1);
core->print->cur = 0;
} else {
ut64 addr = r_debug_reg_get (core->dbg, "PC");
if (addr) {
r_core_seek (core, addr, 1);
r_core_cmdf (core, "ar `arn PC`=0x%"PFMT64x, addr);
} else {
r_core_seek (core, r_num_get (core->num, "entry0"), 1);
//r_core_cmd (core, "s entry0", 0);
}
}
break;
#if 0
case 'n': r_core_seek_delta (core, core->blocksize); break;
case 'N': r_core_seek_delta (core, 0-(int)core->blocksize); break;
#endif
case ':':
r_core_visual_prompt_input (core);
break;
case '_':
r_core_visual_hudstuff (core);
break;
case ';':
r_cons_printf ("Enter a comment: ('-' to remove, '!' to use $EDITOR)\n");
showcursor (core, true);
r_cons_flush ();
r_cons_set_raw (false);
r_line_set_prompt ("comment: ");
strcpy (buf, "\"CC ");
i = strlen (buf);
if (r_cons_fgets (buf + i, sizeof (buf) - i - 1, 0, NULL) > 0) {
ut64 addr, orig;
addr = orig = core->offset;
if (core->print->cur_enabled) {
addr += core->print->cur;
r_core_seek (core, addr, 0);
r_core_cmdf (core, "s 0x%"PFMT64x, addr);
}
if (!strcmp (buf + i, "-")) {
strcpy (buf, "CC-");
} else {
switch (buf[i]) {
case '-':
memcpy (buf, "\"CC-\x00", 5);
break;
case '!':
memcpy (buf, "\"CC!\x00", 5);
break;
default:
memcpy (buf, "\"CC ", 4);
break;
}
strcat (buf, "\"");
}
if (buf[3] == ' ') {
// have to escape any quotes.
int j, len = strlen (buf);
char* duped = strdup (buf);
i = 4, j=4;
for (i=4, j=4; i < len; ++i,++j) {
char c = duped[i];
if (c == '"' && i != (len - 1)) {
buf[j] = '\\';
++j;
buf[j] = '"';
} else {
buf[j] = c;
}
}
free (duped);
}
r_core_cmd (core, buf, 1);
if (core->print->cur_enabled) r_core_seek (core, orig, 1);
}
r_cons_set_raw (true);
showcursor (core, false);
break;
case 'b':
{
ut64 addr = core->print->cur_enabled? core->offset + core->print->cur : core->offset;
RBreakpointItem *bp = r_bp_get_at (core->dbg->bp, addr);
if (bp) {
r_bp_del (core->dbg->bp, addr);
} else {
r_bp_add_sw (core->dbg->bp, addr, 1, R_BP_PROT_EXEC);
}
}
break;
case 'O':
r_core_cmd0 (core, "e!asm.esil");
break;
case 'u':
{
RIOUndos *undo = r_io_sundo (core->io, core->offset);
if (undo) {
r_core_visual_seek_animation (core, undo->off);
core->print->cur = undo->cursor;
} else {
eprintf ("Cannot undo\n");
}
}
break;
case 'U':
{
RIOUndos *undo = r_io_sundo_redo (core->io);
if (undo) {
r_core_visual_seek_animation (core, undo->off);
reset_print_cur (core->print);
}
}
break;
case 'z':
{
RAnalFunction *fcn;
if (core->print->cur_enabled) {
fcn = r_anal_get_fcn_in (core->anal,
core->offset+core->print->cur, R_ANAL_FCN_TYPE_NULL);
} else {
fcn = r_anal_get_fcn_in (core->anal,
core->offset, R_ANAL_FCN_TYPE_NULL);
}
if (fcn) {
fcn->folded = !fcn->folded;
} else {
r_config_toggle (core->config, "asm.cmtfold");
}
}
break;
case 'Z':
if (zoom && core->print->cur) {
ut64 from = r_config_get_i (core->config, "zoom.from");
ut64 to = r_config_get_i (core->config, "zoom.to");
r_core_seek (core, from + ((to-from)/core->blocksize)*core->print->cur, 1);
}
zoom = !zoom;
break;
case '?':
if (visual_help ()=='?')
r_core_visual_hud (core);
break;
case 0x1b:
case 'q':
case 'Q':
setcursor (core, false);
return false;
}
r_core_block_read (core);
return true;
}
R_API void r_core_visual_title (RCore *core, int color) {
static ut64 oldpc = 0;
const char *BEGIN = core->cons->pal.prompt;
const char *filename;
char pos[512], foo[512], bar[512], pcs[32];
if (!oldpc) oldpc = core->offset;
/* automatic block size */
int pc, hexcols = r_config_get_i (core->config, "hex.cols");
if (autoblocksize)
switch (core->printidx) {
case 0: // x"
case 6: // pxa
r_core_block_size (core, core->cons->rows * hexcols);
break;
case 3: // XXX pw
r_core_block_size (core, core->cons->rows * hexcols);
break;
case 4: // XXX pc
r_core_block_size (core, core->cons->rows * 5);
break;
case 1: // pd
case 2: // pd+dbg
{
int bsize = core->cons->rows * 5;
if (core->print->screen_bounds > 1) {
// estimate new blocksize with the size of the last
// printed instructions
int new_sz = core->print->screen_bounds - core->offset + 32;
if (new_sz > bsize) bsize = new_sz;
}
r_core_block_size (core, bsize);
break;
}
case 5: // pxA
r_core_block_size (core, hexcols * core->cons->rows * 8);
break;
}
if (r_config_get_i (core->config, "scr.zoneflags")) {
r_core_cmd (core, "fz:", 0);
}
if (r_config_get_i (core->config, "cfg.debug")) {
ut64 curpc = r_debug_reg_get (core->dbg, "PC");
if (curpc && curpc != UT64_MAX && curpc != oldpc) {
// check dbg.follow here
int follow = (int)(st64)r_config_get_i (core->config, "dbg.follow");
if (follow > 0) {
if ((curpc<core->offset) || (curpc> (core->offset+follow))) {
r_core_seek (core, curpc, 1);
}
} else if (follow < 0) {
r_core_seek (core, curpc + follow, 1);
}
oldpc = curpc;
}
}
filename = (core->file && core->file->desc && core->file->desc->name)? core->file->desc->name: "";
{ /* get flag with delta */
ut64 addr = core->offset + (core->print->cur_enabled? core->print->cur: 0);
RFlagItem *f = r_flag_get_at (core->flags, addr);
if (f) {
if (f->offset == addr || !f->offset)
snprintf (pos, sizeof (pos), "@ %s", f->name);
else snprintf (pos, sizeof (pos), "@ %s+%d # 0x%"PFMT64x,
f->name, (int)(addr-f->offset), addr);
} else pos[0] = 0;
}
if (core->print->cur<0) core->print->cur = 0;
if (color) r_cons_strcat (BEGIN);
strncpy (bar, printfmt[PIDX], sizeof (bar)-1);
bar[sizeof (bar)-1] = 0; // '\0'-terminate bar
bar[10] = '.'; // chop cmdfmt
bar[11] = '.'; // chop cmdfmt
bar[12] = 0; // chop cmdfmt
{
ut64 sz = r_io_size (core->io);
ut64 pa = r_io_section_vaddr_to_maddr_try (core->io, core->offset);
if (sz == UT64_MAX) {
pcs[0] = 0;
} else {
if (!sz || pa>sz) {
pc = 0;
} else {
pc = ( pa * 100 ) / sz;
}
sprintf (pcs, "%d%% ", pc);
}
}
if (__ime) {
snprintf (foo, sizeof (foo), "[0x%08"PFMT64x" + %d> * INSERT MODE *\n",
core->offset, core->print->cur);
} else {
if (core->print->cur_enabled) {
snprintf (foo, sizeof (foo), "[0x%08"PFMT64x" %s%d (0x%x:%d=%d)]> %s %s\n",
core->offset, pcs, core->blocksize,
core->print->cur, core->print->ocur, core->print->ocur==-1?
1:R_ABS (core->print->cur-core->print->ocur)+1,
bar, pos);
} else {
snprintf (foo, sizeof (foo), "[0x%08"PFMT64x" %s%d %s]> %s %s\n",
core->offset, pcs, core->blocksize, filename, bar, pos);
}
}
r_cons_print (foo);
if (color) {
r_cons_strcat (Color_RESET);
}
}
static int visual_responsive(RCore *core) {
int h, w = r_cons_get_size (&h);
if (r_config_get_i (core->config, "scr.responsive")) {
if (w<110) {
r_config_set_i (core->config, "asm.cmtright", 0);
} else {
r_config_set_i (core->config, "asm.cmtright", 1);
}
if (w<68) {
r_config_set_i (core->config, "hex.cols", w/5.2);
} else {
r_config_set_i (core->config, "hex.cols", 16);
}
if (w<25) {
r_config_set_i (core->config, "asm.offset", 0);
} else {
r_config_set_i (core->config, "asm.offset", 1);
}
if (w>80) {
r_config_set_i (core->config, "asm.lineswidth", 14);
r_config_set_i (core->config, "asm.lineswidth", w-(w/1.2));
r_config_set_i (core->config, "asm.cmtcol", w-(w/2.5));
} else {
r_config_set_i (core->config, "asm.lineswidth", 7);
}
if (w<70) {
r_config_set_i (core->config, "asm.lineswidth", 1);
r_config_set_i (core->config, "asm.bytes", 0);
} else {
r_config_set_i (core->config, "asm.bytes", 1);
}
}
return w;
}
static void visual_refresh(RCore *core) {
int w;
const char *vi, *vcmd;
if (!core) return;
r_print_set_cursor (core->print, core->print->cur_enabled, core->print->ocur, core->print->cur);
core->cons->blankline = true;
w = visual_responsive (core);
if (autoblocksize) {
r_cons_gotoxy (0, 0);
} else {
r_cons_clear ();
}
r_cons_flush ();
r_cons_print_clear ();
vi = r_config_get (core->config, "cmd.cprompt");
if (vi && *vi) {
// XXX: slow
core->cons->blankline = false;
r_cons_clear00 ();
r_cons_flush ();
{
int hc = r_config_get_i (core->config, "hex.cols");
int nw = 12 + 4 + hc + (hc*3);
if (nw>w) {
// do not show column contents
} else {
r_cons_printf ("[cmd.cprompt=%s]\n", vi);
r_core_cmd0 (core, vi);
r_cons_column (nw);
r_cons_flush ();
}
}
r_cons_gotoxy (0, 0);
r_core_visual_title (core, color);
vi = r_config_get (core->config, "cmd.vprompt");
if (vi) r_core_cmd (core, vi, 0);
} else {
vi = r_config_get (core->config, "cmd.vprompt");
if (vi) r_core_cmd (core, vi, 0);
r_core_visual_title (core, color);
}
vcmd = r_config_get (core->config, "cmd.visual");
if (vcmd && *vcmd) {
// disable screen bounds when it's a user-defined command
// because it can cause some issues
core->print->screen_bounds = 0;
r_core_cmd (core, vcmd, 0);
} else {
core->print->screen_bounds = 1LL;
r_core_cmd0 (core, zoom ? "pz" : printfmt[PIDX]);
}
if (core->print->screen_bounds != 1LL) {
r_cons_printf ("[0x%08"PFMT64x"..0x%08"PFMT64x"]\n",
core->offset, core->print->screen_bounds);
}
blocksize = core->num->value? core->num->value : core->blocksize;
/* this is why there's flickering */
if (core->print->vflush) {
r_cons_visual_flush ();
} else {
r_cons_reset ();
}
core->cons->blankline = true;
core->curtab = 0; // which command are we focusing
//core->seltab = 0; // user selected tab
}
R_API int r_core_visual(RCore *core, const char *input) {
const char *cmdprompt, *teefile;
ut64 scrseek;
int wheel, flags, ch;
bool skip;
if (r_cons_get_size (&ch) < 1 || ch < 1) {
eprintf ("Cannot create Visual context. Use scr.fix_{columns|rows}\n");
return 0;
}
obs = core->blocksize;
//r_cons_set_cup (true);
core->vmode = false;
/* honor vim */
if (!strncmp (input, "im", 2)) {
char *cmd = r_str_newf ("!v%s", input);
int ret = r_core_cmd0 (core, cmd);
free (cmd);
return ret;
}
while (*input) {
if (!r_core_visual_cmd (core, input[0])) {
return 0;
}
input++;
}
core->vmode = true;
// disable tee in cons
teefile = r_cons_singleton ()->teefile;
r_cons_singleton ()->teefile = "";
core->print->flags |= R_PRINT_FLAGS_ADDRMOD;
do {
dodo:
// update the cursor when it's not visible anymore
skip = fix_cursor (core);
if (core->printidx == 2) {
static char debugstr[512];
const int ref = r_config_get_i (core->config, "dbg.slow");
const int pxa = r_config_get_i (core->config, "stack.anotated"); // stack.anotated
const int size = r_config_get_i (core->config, "stack.size");
const int delta = r_config_get_i (core->config, "stack.delta");
const int bytes = r_config_get_i (core->config, "stack.bytes");
const char *cmdvhex = r_config_get (core->config, "cmd.stack");
if (cmdvhex && *cmdvhex) {
snprintf (debugstr, sizeof (debugstr),
"?0;f tmp;sr SP;%s;?1;%s;?1;s-;"
"s tmp;f-tmp;pd $r", cmdvhex,
ref? "drr": "dr=");
debugstr[sizeof (debugstr)-1]=0;
} else {
const char *pxw;
if (ref) {
pxw = "pxr";
} else if (bytes) {
pxw = "px";
} else {
switch (core->assembler->bits) {
case 64: pxw = "pxq"; break;
case 32: pxw = "pxw"; break;
default: pxw = "px"; break;
}
}
snprintf (debugstr, sizeof (debugstr),
"?0;f tmp;sr SP;%s %d@$$-%d;"
"?1;%s;s-;"
"?1;s tmp;f-tmp;pd $r",
pxa? "pxa": pxw, size, delta,
ref? "drr": "dr=");
}
printfmt[2] = debugstr;
}
wheel = r_config_get_i (core->config, "scr.wheel");
r_cons_show_cursor (false);
if (wheel) r_cons_enable_mouse (true);
core->cons->event_data = core;
core->cons->event_resize = (RConsEvent)visual_refresh;
flags = core->print->flags;
color = r_config_get_i (core->config, "scr.color");
if (color) flags |= R_PRINT_FLAGS_COLOR;
debug = r_config_get_i (core->config, "cfg.debug");
flags |= R_PRINT_FLAGS_ADDRMOD | R_PRINT_FLAGS_HEADER;
r_print_set_flags (core->print, core->print->flags);
scrseek = r_num_math (core->num,
r_config_get (core->config, "scr.seek"));
if (scrseek != 0LL) {
r_core_seek (core, scrseek, 1);
}
if (debug) {
r_core_cmd (core, ".dr*", 0);
}
cmdprompt = r_config_get (core->config, "cmd.vprompt");
if (cmdprompt && *cmdprompt) {
r_core_cmd (core, cmdprompt, 0);
}
core->print->vflush = !skip;
visual_refresh (core);
if (insert_mode_enabled (core)) {
goto dodo;
}
if (!skip) {
ch = r_cons_readchar ();
r_core_visual_show_char (core, ch);
if (ch == -1 || ch == 4) break; // error or eof
}
} while (skip || r_core_visual_cmd (core, ch));
r_cons_enable_mouse (false);
if (color) {
r_cons_printf (Color_RESET);
}
r_config_set_i (core->config, "scr.color", color);
core->print->cur_enabled = false;
if (autoblocksize) {
r_core_block_size (core, obs);
}
r_cons_singleton ()->teefile = teefile;
r_cons_set_cup (false);
r_cons_clear00 ();
core->vmode = false;
core->cons->event_resize = NULL;
core->cons->event_data = NULL;
r_cons_show_cursor (true);
return 0;
}