radare2/libr/core/visual.c

2113 lines
54 KiB
C
Raw Normal View History

/* radare - LGPL - Copyright 2009-2016 - pancake */
#include <r_core.h>
2015-10-14 21:50:16 +00:00
#include <string.h>
#define NPF 7
static int obs = 0;
static int blocksize = 0;
static int autoblocksize = 1;
static void r_core_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[] = {
2015-04-10 23:05:51 +00:00
"x", "pd $r",
debugfmt_default,
"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;
r_core_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 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);
r_th_start (th, 1);
r_cons_break (NULL, NULL);
r_cons_any_key (NULL);
eprintf ("^C \n");
2015-09-14 10:35:38 +00:00
core->cons->breaked = true;
r_th_wait (th);
r_cons_break_end ();
}
}
#endif
2013-10-10 23:24:30 +00:00
static void showcursor(RCore *core, int x) {
if (core && core->vmode) {
2013-10-10 23:24:30 +00:00
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 {
2015-09-14 10:35:38 +00:00
r_cons_enable_mouse (false);
}
2015-09-14 10:35:38 +00:00
} else r_cons_enable_mouse (false);
r_cons_flush ();
2013-10-10 23:24:30 +00:00
}
static int color = 1;
static int debug = 1;
static int zoom = 0;
2015-08-31 11:57:39 +00:00
R_API int r_core_visual_hud(RCore *core) {
2014-08-29 22:16:46 +00:00
const char *c = r_config_get (core->config, "hud.path");
2013-02-07 08:41:05 +00:00
const char *f = R2_LIBDIR"/radare2/"R2_VERSION"/hud/main";
char *homehud = r_str_home (R2_HOMEDIR"/hud");
char *res = NULL;
char *p = 0;
2013-02-07 08:41:05 +00:00
2015-09-14 10:35:38 +00:00
showcursor (core, true);
2014-08-29 22:16:46 +00:00
if (c && *c && r_file_exists (c))
res = r_cons_hud_file (c);
if (!res && homehud)
res = r_cons_hud_file (homehud);
2014-08-29 22:16:46 +00:00
if (!res && r_file_exists (f))
res = r_cons_hud_file (f);
if (!res)
r_cons_message ("Cannot find hud file");
r_cons_clear ();
if (res) {
p = strchr (res, '\t');
r_cons_printf ("%s\n", res);
r_cons_flush ();
if (p) r_core_cmd0 (core, p+1);
free (res);
}
2015-09-14 10:35:38 +00:00
showcursor (core, false);
r_cons_flush ();
2014-05-03 00:21:02 +00:00
free (homehud);
return (int)(size_t)p;
}
static int visual_help() {
r_cons_clear00 ();
return r_cons_less_str (
2014-04-07 01:13:06 +00:00
"Visual mode help:\n"
" ? show this help or enter the userfriendly hud\n"
2014-06-25 01:07:54 +00:00
" & 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"
" /*+-[] change block size, [] = resize hex.cols\n"
" >||< seek aligned to block size\n"
2014-08-29 23:11:46 +00:00
" a/A (a)ssemble code, visual (A)ssembler\n"
" b toggle breakpoint\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"
2014-08-29 23:11:46 +00:00
" 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"
2014-08-29 23:11:46 +00:00
" 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"
2014-08-29 23:11:46 +00:00
" uU undo/redo seek\n"
" v visual code analysis menu\n"
" V (V)iew graph using cmd.graph (agv?)\n"
" wW seek cursor to next/prev word\n"
2014-08-05 17:49:33 +00:00
" 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"
2014-04-07 01:13:06 +00:00
"Function Keys: (See 'e key.'), defaults to:\n"
" F2 toggle breakpoint\n"
" F4 run to cursor\n"
2014-04-07 01:13:06 +00:00
" 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) {
2016-02-27 11:38:02 +00:00
if (!buf || buflen < 1)
return;
*buf = 0;
r_line_set_prompt (p);
2015-09-14 10:35:38 +00:00
showcursor (NULL, true);
r_cons_fgets (buf, buflen, 0, NULL);
2015-09-14 10:35:38 +00:00
showcursor (NULL, false);
}
2014-08-05 17:49:33 +00:00
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);
2015-11-19 11:23:10 +00:00
r_cons_reset_colors ();
r_cons_printf ("\nPress <enter> to return to Visual mode.\n");
2015-09-14 10:35:38 +00:00
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);
}
2014-08-05 17:49:33 +00:00
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) {
2014-08-05 17:49:33 +00:00
r_core_seek (core, addr, 1);
r_core_block_size (core, bsze);
}
}
2015-09-14 10:35:38 +00:00
r_cons_show_cursor (false);
core->vmode = true;
2014-08-05 17:49:33 +00:00
}
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
2015-09-14 10:35:38 +00:00
showcursor (core, true);
r_cons_fgets (buf, sizeof (buf), 0, NULL);
if (!strcmp (buf, "q")) {
2015-09-14 10:35:38 +00:00
ret = false;
} else if (*buf) {
r_line_hist_add (buf);
r_core_cmd (core, buf, 0);
r_cons_flush ();
2015-09-14 10:35:38 +00:00
ret = true;
} else {
2015-09-14 10:35:38 +00:00
ret = false;
//r_cons_any_key (NULL);
r_cons_clear00 ();
2015-09-14 10:35:38 +00:00
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) {
2013-08-26 23:12:58 +00:00
int flags = core->print->flags; // wtf
if (core->print->cur_enabled) flags |= R_PRINT_FLAGS_CURSOR;
2013-08-26 23:12:58 +00:00
else flags &= ~(R_PRINT_FLAGS_CURSOR);
2013-10-10 23:24:30 +00:00
core->print->cur_enabled = cur;
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;
2015-09-14 10:35:38 +00:00
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;
2015-09-14 10:35:38 +00:00
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;
2015-09-14 10:35:38 +00:00
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;
2015-09-14 10:35:38 +00:00
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) {
2015-11-19 11:23:10 +00:00
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;
// 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 - 1);
if (bb) {
ut64 res = r_anal_bb_opaddr_at (bb, addr - 1);
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;
2013-12-06 04:18:57 +00:00
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;
}
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);
}
}
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) {
2014-09-22 22:40:35 +00:00
int ret = 0;
#if FCN_OLD
char ch;
int count = 0;
RList *xrefs = NULL;
RAnalRef *refi;
RListIter *iter;
RAnalFunction *fun;
2016-02-16 00:20:25 +00:00
int skip = 0;
int idx = 0;
char cstr[32];
2016-02-16 00:20:25 +00:00
repeat:
if ((xrefs = r_anal_xref_get (core->anal, core->offset))) {
2016-02-16 00:20:25 +00:00
r_cons_clear00 ();
r_cons_gotoxy (1, 1);
2016-02-16 00:20:25 +00:00
r_cons_printf ("[GOTO XREF]> 0x%08"PFMT64x"\n", core->offset);
if (r_list_empty (xrefs)) {
r_cons_printf ("\tNo XREF found at 0x%"PFMT64x"\n", core->offset);
r_cons_any_key (NULL);
r_cons_clear00 ();
} else {
2016-02-16 00:20:25 +00:00
int lines;
(void)r_cons_get_size (&lines);
idx = 0;
count = 0;
lines -= 3;
r_list_foreach (xrefs, iter, refi) {
2016-02-16 00:20:25 +00:00
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 (++count >= lines) {
r_cons_printf ("...\n");
break;
}
}
idx++;
}
}
} else xrefs = NULL;
2014-08-17 18:30:28 +00:00
if (!xrefs || !r_list_length (xrefs)) {
r_list_free (xrefs);
return 0;
2014-08-17 18:30:28 +00:00
}
r_cons_flush ();
ch = r_cons_readchar ();
2016-02-16 00:20:25 +00:00
if (ch == 'j') {
skip++;
goto repeat;
} else if (ch == 'k') {
skip--;
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);
2014-09-22 22:40:35 +00:00
#else
eprintf ("TODO: sdbize xrefs here\n");
#endif
return ret;
}
R_API int r_core_visual_xrefs_X (RCore *core) {
2014-09-22 22:40:35 +00:00
int ret = 0;
#if FCN_OLD
char ch;
int count = 0;
RAnalRef *refi;
RListIter *iter;
RAnalFunction *fun;
fun = r_anal_get_fcn_in (core->anal, core->offset, 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", core->offset);
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;
}
}
}
2014-09-22 22:40:35 +00:00
#else
eprintf ("TODO: sdbize this\n");
#endif
return ret;
}
2015-04-07 14:25:14 +00:00
#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);
}
2015-04-07 14:25:14 +00:00
#endif
2015-11-18 16:02:42 +00:00
// 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) {
2015-11-25 11:49:02 +00:00
char *abspath = r_file_abspath (dir);
/* use prjdir as base directory for comma-ent files */
2015-11-24 17:26:10 +00:00
cwd = r_str_newf ("%s"R_SYS_DIR"%s.d", abspath, prj);
free (abspath);
} else {
/* use cwd as base directory for comma-ent files */
2015-11-18 16:02:42 +00:00
cwd = r_sys_getdir ();
}
2015-11-18 16:02:42 +00:00
return cwd;
}
static void visual_comma(RCore *core) {
ut64 addr = core->offset + (core->print->cur_enabled ? core->print->cur : 0);
2015-11-18 16:02:42 +00:00
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);
2015-11-18 22:32:18 +00:00
fn = r_cons_input ("<comment-file> ");
showcursor (core, false);
if (fn && *fn) {
cmtfile = strdup (fn);
if (!comment || !*comment) {
2015-11-18 16:02:42 +00:00
comment = r_str_newf (",(%s)", fn);
r_meta_set_string (core->anal, R_META_TYPE_COMMENT, addr, comment);
} else {
// append filename in current comment
2015-11-18 16:02:42 +00:00
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 {
2015-11-18 16:02:42 +00:00
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) {
if (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;
}
sz = r_asm_disassemble (core->assembler, &op,
core->block + next_roff, 32);
if (sz < 1) 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;
int row;
ut32 roff, prev_roff;
if (PIDX == 2) {
if (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);
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->screen_bounds > 1) {
bool off_is_visible = core->offset < core->screen_bounds;
bool cur_is_visible = core->offset + p->cur < core->screen_bounds;
bool is_close = core->offset + p->cur < core->screen_bounds + 32;
if (!cur_is_visible && !is_close) {
// when the cursor is not visible and it's far from the
// last visible byte, just seek there.
r_core_seek (core, core->offset + p->cur, 1);
p->cur = 0;
p->ocur = -1;
} 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 (core, core->offset + sz, 1);
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;
}
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;
2015-04-07 14:25:14 +00:00
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?
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);
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, 1);
}
} else
switch (ch) {
2015-04-07 14:25:14 +00:00
#if __WINDOWS__ && !__CYGWIN__
case 0xf5:
SetWindow(81,25);
break;
case 0xcf5:
SetWindow(81,40);
break;
2015-04-07 14:25:14 +00:00
#endif
2015-11-19 11:23:10 +00:00
case 0x0d: // "enter" "\\n" "newline"
{
RAnalOp *op;
2014-09-03 21:55:09 +00:00
int wheel = r_config_get_i (core->config, "scr.wheel");
if (wheel)
2015-09-14 10:35:38 +00:00
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) {
2014-09-23 11:32:51 +00:00
int delta = R_ABS ((st64)op->jump-(st64)offset);
if ( op->jump < core->offset || op->jump >= core->screen_bounds) {
2014-09-23 11:32:51 +00:00
r_io_sundo_push (core->io, offset);
r_core_visual_seek_animation (core, op->jump);
core->print->cur = 0;
2014-09-23 11:32:51 +00:00
} else {
core->print->cur = delta;
2014-09-23 11:32:51 +00:00
}
} else {
r_io_sundo_push (core->io, offset);
r_core_visual_seek_animation (core, op->jump);
2014-09-23 11:32:51 +00:00
}
}
}
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);
2015-09-14 10:35:38 +00:00
return true;
}
r_cons_printf ("Enter assembler opcodes separated with ';':\n");
2015-09-14 10:35:38 +00:00
showcursor (core, true);
r_cons_flush ();
2015-09-14 10:35:38 +00:00
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);
}
2015-09-14 10:35:38 +00:00
r_core_cmd (core, buf, true);
if (core->print->cur_enabled) {
int t = core->offset - core->print->cur;
r_core_seek (core, t, 1);
}
}
2015-09-14 10:35:38 +00:00
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");
2015-09-14 10:35:38 +00:00
if (wheel) r_cons_enable_mouse (false);
r_core_visual_define (core);
2015-09-14 10:35:38 +00:00
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: ");
2015-09-14 10:35:38 +00:00
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);
}
} }
2015-09-14 10:35:38 +00:00
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 == 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);
2015-09-14 10:35:38 +00:00
return true;
}
2015-09-14 10:35:38 +00:00
showcursor (core, true);
r_cons_flush ();
r_cons_set_raw (0);
if (ch=='I') {
strcpy (buf, "wow ");
r_line_set_prompt ("insert hexpair block: ");
2014-04-26 00:12:09 +00:00
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);
2015-09-14 10:35:38 +00:00
showcursor (core, false);
break;
case 'R':
r_core_cmd0 (core, "ecr");
break;
case 'e':
r_core_visual_config (core);
break;
case 'E':
r_core_visual_colors (core);
break;
2011-07-11 20:52:05 +00:00
case 'M':
r_core_visual_mounts (core);
break;
2014-06-25 01:07:54 +00:00
case '&':
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;
}
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;
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;
}
ocolor = r_config_get_i (core->config, "scr.color");
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);
break;
case 'G':
ret = 0;
2015-06-09 12:27:12 +00:00
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)
2015-11-23 15:24:36 +00:00
- 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)
2015-11-23 15:24:36 +00:00
- core->blocksize + 2 * scols, 1);
}
} else {
ret = -1;
}
if (ret != -1)
r_io_sundo_push (core->io, core->offset);
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 = wheelspeed;
if (times<1) times = 1;
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 + f->size;
} 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->screen_bounds > 1 && core->screen_bounds >= core->offset) {
r_core_seek (core, core->screen_bounds, 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 (core, core->offset - cols, 1);
2015-12-18 18:53:54 +00:00
}
}
}
break;
case 'K':
if (core->print->cur_enabled) {
cursor_prevrow (core, true);
} else {
if (core->screen_bounds >= core->offset) {
int delta = (core->screen_bounds - core->offset);
if (core->offset >= delta)
r_core_seek (core, core->offset - delta, 1);
2015-04-15 22:25:32 +00:00
else
r_core_seek (core, 0, 1);
} else {
ut64 at = (core->offset>obs)?core->offset-obs:0;
2015-04-15 22:25:32 +00:00
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 {
2015-11-23 15:24:36 +00:00
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);
2011-12-02 01:13:49 +00:00
r_cons_clear00 ();
} else r_core_yank_paste (core, core->offset+core->print->cur, 0);
break;
case '-':
if (core->print->cur_enabled) {
int cur = core->print->cur;
if (cur>=core->blocksize)
cur = core->print->cur-1;
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) {
int cur = core->print->cur;
if (cur>=core->blocksize)
cur = core->print->cur-1;
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 (!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);
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);
break;
case '.':
r_io_sundo_push (core->io, core->offset);
if (core->print->cur_enabled) {
r_core_seek (core, core->offset+core->print->cur, 1);
core->print->cur = 0;
} else {
2015-10-31 01:20:25 +00:00
ut64 addr = r_debug_reg_get (core->dbg, "PC");
2014-10-22 22:07:30 +00:00
if (addr) {
r_core_seek (core, addr, 1);
2015-10-31 01:20:25 +00:00
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 ':':
2014-08-05 17:49:33 +00:00
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");
2015-09-14 10:35:38 +00:00
showcursor (core, true);
r_cons_flush ();
2015-09-14 10:35:38 +00:00
r_cons_set_raw (false);
r_line_set_prompt ("comment: ");
2015-12-22 22:09:23 +00:00
strcpy (buf, "\"CC ");
i = strlen (buf);
2015-12-31 15:28:16 +00:00
if (r_cons_fgets (buf + i, sizeof (buf) - i - 1, 0, NULL) > 0) {
2015-12-22 22:09:23 +00:00
ut64 addr, orig;
addr = orig = core->offset;
if (core->print->cur_enabled) {
addr += core->print->cur;
r_core_seek (core, addr, 0);
2014-05-28 15:49:32 +00:00
r_core_cmdf (core, "s 0x%"PFMT64x, addr);
}
2015-12-31 15:28:16 +00:00
if (!strcmp (buf + i, "-")) {
2015-12-22 22:09:23 +00:00
strcpy (buf, "CC-");
} else {
switch (buf[i]) {
case '-':
2015-12-31 15:28:16 +00:00
memcpy (buf, "\"CC-", 4);
2015-12-22 22:09:23 +00:00
break;
case '!':
2015-12-31 15:28:16 +00:00
memcpy (buf, "\"CC!", 4);
2015-12-22 22:09:23 +00:00
break;
default:
memcpy (buf, "\"CC ", 4);
break;
}
strcat (buf, "\"");
}
r_core_cmd (core, buf, 1);
if (core->print->cur_enabled) r_core_seek (core, orig, 1);
}
2015-09-14 10:35:38 +00:00
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;
2014-08-29 23:11:46 +00:00
case 'O':
r_core_cmd0 (core, "e!asm.esil");
break;
case 'u':
{
ut64 off = r_io_sundo (core->io, core->offset);
if (off != UT64_MAX) {
r_core_visual_seek_animation (core, off);
} else {
eprintf ("Cannot undo\n");
}
}
break;
case 'U':
{
ut64 off = r_io_sundo_redo (core->io);
if (off != UT64_MAX) {
r_core_visual_seek_animation (core, off);
}
}
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);
2015-09-14 10:35:38 +00:00
return false;
}
r_core_block_read (core, 0);
2015-09-14 10:35:38 +00:00
return true;
}
2011-06-04 01:23:12 +00:00
R_API void r_core_visual_title (RCore *core, int color) {
2014-10-22 22:07:30 +00:00
static ut64 oldpc = 0;
const char *BEGIN = core->cons->pal.prompt;
const char *filename;
2015-02-06 18:24:26 +00:00
char pos[512], foo[512], bar[512], pcs[32];
2014-10-22 22:07:30 +00:00
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->screen_bounds > 1) {
// estimate new blocksize with the size of the last
// printed instructions
int new_sz = core->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;
}
2014-10-22 22:07:30 +00:00
if (r_config_get_i (core->config, "cfg.debug")) {
2015-10-31 01:20:25 +00:00
ut64 curpc = r_debug_reg_get (core->dbg, "PC");
if (curpc && curpc != UT64_MAX && curpc != oldpc) {
2014-10-22 22:07:30 +00:00
// check dbg.follow here
int follow = (int)(st64)r_config_get_i (core->config, "dbg.follow");
2014-10-22 22:07:30 +00:00
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);
2014-10-22 22:07:30 +00:00
}
oldpc = curpc;
}
}
2014-09-09 15:01:04 +00:00
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
2015-02-06 18:24:26 +00:00
{
ut64 sz = r_io_size (core->io);
ut64 pa = r_io_section_vaddr_to_maddr_try (core->io, core->offset);
2015-02-06 18:24:26 +00:00
if (sz == UT64_MAX) {
pcs[0] = 0;
} else {
2015-02-07 23:30:25 +00:00
if (!sz || pa>sz) {
2015-02-06 18:24:26 +00:00
pc = 0;
} else {
pc = ( pa * 100 ) / sz;
}
sprintf (pcs, "%d%% ", pc);
}
}
if (core->print->cur_enabled)
2015-02-06 18:24:26 +00:00
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
2015-02-06 18:24:26 +00:00
snprintf (foo, sizeof (foo), "[0x%08"PFMT64x" %s%d %s]> %s %s\n",
core->offset, pcs, core->blocksize, filename, bar, pos);
2015-02-12 15:41:02 +00:00
r_cons_printf ("%s", foo);
if (color) r_cons_strcat (Color_RESET);
}
2015-11-23 16:09:46 +00:00
static int r_core_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);
}
}
2015-11-23 16:09:46 +00:00
return w;
}
static void r_core_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);
2015-11-23 16:09:46 +00:00
core->cons->blankline = true;
w = r_core_visual_responsive (core);
if (autoblocksize) {
r_cons_gotoxy (0, 0);
2014-09-17 08:36:38 +00:00
} else {
r_cons_clear ();
}
2015-11-23 16:09:46 +00:00
r_cons_flush ();
r_cons_print_clear ();
//core->curtab = 0;
vi = r_config_get (core->config, "cmd.cprompt");
if (vi && *vi) {
// XXX: slow
2015-09-14 10:35:38 +00:00
core->cons->blankline = false;
r_cons_clear00 ();
2012-11-30 15:12:54 +00:00
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);
2012-11-30 15:12:54 +00:00
} else {
vi = r_config_get (core->config, "cmd.vprompt");
if (vi) r_core_cmd (core, vi, 0);
2012-11-30 15:12:54 +00:00
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->screen_bounds = 0;
r_core_cmd (core, vcmd, 0);
} else {
core->screen_bounds = 1LL;
r_core_cmd0 (core, zoom? "pz": printfmt[PIDX]);
}
if (core->screen_bounds != 1LL) {
r_cons_printf ("[0x%08"PFMT64x"..0x%08"PFMT64x"]\n",
core->offset, core->screen_bounds);
}
blocksize = core->num->value? core->num->value : core->blocksize;
2012-11-30 15:12:54 +00:00
/* this is why there's flickering */
if (core->print->vflush) {
r_cons_visual_flush ();
} else {
r_cons_reset ();
}
2015-09-14 10:35:38 +00:00
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;
}
2013-08-26 23:12:58 +00:00
obs = core->blocksize;
2015-09-14 10:35:38 +00:00
//r_cons_set_cup (true);
2015-09-14 10:35:38 +00:00
core->vmode = false;
while (*input) {
if (!r_core_visual_cmd (core, input[0]))
return 0;
input++;
}
2015-09-14 10:35:38 +00:00
core->vmode = true;
// disable tee in cons
teefile = r_cons_singleton ()->teefile;
r_cons_singleton ()->teefile = "";
core->print->flags |= R_PRINT_FLAGS_ADDRMOD;
do {
skip = false;
if (core->print->cur_enabled) {
// update the cursor when it's not visible anymore
skip |= fix_cursor (core);
}
if (core->printidx == 2) {
static char debugstr[512];
const char *cmdvhex = r_config_get (core->config, "cmd.stack");
const int ref = r_config_get_i (core->config, "dbg.slow");
const int pxa = r_config_get_i (core->config, "stack.anotated"); // stack.anotated
2014-10-03 18:05:33 +00:00
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");
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 {
2014-10-03 18:05:33 +00:00
const char *pxw;
if (ref) {
pxw = "pxr";
} else if (bytes) {
2014-10-03 18:05:33 +00:00
pxw = "px";
} else {
switch (core->assembler->bits) {
case 64: pxw = "pxq"; break;
case 32: pxw = "pxw"; break;
default: pxw = "px"; break;
}
}
2015-11-23 15:31:31 +00:00
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");
2015-09-14 10:35:38 +00:00
r_cons_show_cursor (false);
2015-11-23 15:31:31 +00:00
if (wheel) r_cons_enable_mouse (true);
core->cons->event_data = core;
core->cons->event_resize = (RConsEvent)r_core_visual_refresh;
2013-08-26 23:12:58 +00:00
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");
2015-08-28 16:29:31 +00:00
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;
r_core_visual_refresh (core);
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));
2015-09-14 10:35:38 +00:00
r_cons_enable_mouse (false);
if (color)
r_cons_printf (Color_RESET);
r_config_set_i (core->config, "scr.color", color);
2015-09-14 10:35:38 +00:00
core->print->cur_enabled = false;
if (autoblocksize)
r_core_block_size (core, obs);
r_cons_singleton ()->teefile = teefile;
2015-09-14 10:35:38 +00:00
r_cons_set_cup (false);
2014-09-17 08:36:38 +00:00
r_cons_clear00 ();
2015-09-14 10:35:38 +00:00
core->vmode = false;
core->cons->event_resize = NULL;
core->cons->event_data = NULL;
2015-09-14 10:35:38 +00:00
r_cons_show_cursor (true);
return 0;
}