static void showcursor(RCore *core, int x) {
if (core && core->vmode) {
r_cons_show_cursor (x);
r_cons_enable_mouse (!!!x);
} else r_cons_enable_mouse (R_FALSE);
r_cons_flush ();
}
// XXX no control for max length here?!?!
R_API int r_cons_fgets(char *buf, int len, int argc, const char **argv) {
#define RETURN(x) { ret=x; goto beach; }
RCons *cons = r_cons_singleton ();
int ret = 0, color = cons->pal.input && *cons->pal.input;
if (cons->echo) {
r_cons_set_raw (0);
r_cons_show_cursor (1);
}
#if 0
int mouse = r_cons_enable_mouse (false);
r_cons_enable_mouse (false);
r_cons_flush ();
#endif
if (cons->user_fgets) {
RETURN (cons->user_fgets (buf, len));
}
printf ("%s", cons->line->prompt);
fflush (stdout);
*buf = '\0';
fflush (cons->fdin);
if (color) {
const char *p = cons->pal.input;
int len = p? strlen (p): 0;
if (len>0)
fwrite (p, len, 1, stdout);
fflush (stdout);
}
if (!fgets (buf, len, cons->fdin)) {
if (color) {
printf (Color_RESET);
fflush (stdout);
}
RETURN (-1);
}
if (feof (cons->fdin)) {
if (color) {
printf (Color_RESET);
}
RETURN (-2);
}
buf[strlen (buf)-1] = '\0';
if (color) printf (Color_RESET);
ret = strlen (buf);
beach:
#if __UNIX__
if (errno == EINTR) {
ret = 0;
}
#endif
//r_cons_enable_mouse (mouse);
return ret;
}
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 ();
}
R_API int r_cons_controlz(int ch) {
#if __UNIX__
if (ch == 0x1a) {
r_cons_show_cursor (1);
r_cons_enable_mouse (0);
r_sys_stop ();
return 0;
}
#endif
return ch;
}
static int parseMouseEvent() {
int ch = r_cons_readchar ();
/* Skip the x/y coordinates */
#if USE_CLICK
int x = r_cons_readchar() - 33;
int y = r_cons_readchar() - 33;
#else
(void) r_cons_readchar ();
(void) r_cons_readchar ();
#endif
#if USE_CLICK
if (ch == 35) {
/* handle click */
#define CLICK_DEBUG 1
#if CLICK_DEBUG
r_cons_gotoxy (0, 0);
r_cons_printf ("Click at %d %d\n", x, y);
r_cons_flush ();
#endif
RCons *cons = r_cons_singleton ();
if (cons->onclick) {
cons->onclick (cons->data, x, y);
}
r_cons_enable_mouse (false);
(void)r_cons_readchar ();
return 0;
}
#endif
if (ch != 0x20 && ch >= 64 + 32) {
/* Grab wheel events only */
I->mouse_event = 1;
return "kj"[(ch - (64 + 32))&1];
}
// temporary disable the mouse wheel to allow select
r_cons_enable_mouse (false);
(void)r_cons_readchar ();
return 0;
}
// XXX no control for max length here?!?!
R_API int r_cons_fgets(char *buf, int len, int argc, const char **argv) {
#define RETURN(x) { ret=x; goto beach; }
RCons *cons = r_cons_singleton ();
int ret = 0, color = cons->pal.input && *cons->pal.input;
#if 0
int mouse = r_cons_enable_mouse (R_FALSE);
r_cons_enable_mouse (R_FALSE);
r_cons_flush ();
#endif
if (cons->user_fgets) {
RETURN (cons->user_fgets (buf, len));
}
*buf = '\0';
fflush (cons->fdin);
if (color) {
const char *p = cons->pal.input;
int len = p? strlen (p): 0;
if (len>0)
fwrite (p, len, 1, stdout);
fflush (stdout);
}
if (fgets (buf, len, cons->fdin) == NULL) {
if (color) {
printf (Color_RESET);
fflush (stdout);
}
RETURN (-1);
}
if (feof (cons->fdin)) {
if (color) printf (Color_RESET);
RETURN (-2);
}
buf[strlen (buf)-1] = '\0';
if (color) printf (Color_RESET);
ret = strlen (buf);
beach:
//r_cons_enable_mouse (mouse);
return ret;
}
R_API int r_core_visual_cmd(RCore *core, int ch) {
RAsmOp op;
ut64 offset = core->offset;
char buf[4096];
int i, ret, offscreen, cols = core->print->cols, delta = 0;
ch = r_cons_arrow_to_hjkl (ch);
ch = visual_nkey (core, ch);
if (ch<2) return 1;
// do we need hotkeys for data references? not only calls?
if (ch>='0'&& ch<='9') {
ut64 off = core->asmqjmps[ch-'0'];
if (off != UT64_MAX) {
int delta = R_ABS ((st64)off-(st64)offset);
r_io_sundo_push (core->io, offset);
if (curset && delta<100) {
cursor = delta;
} else {
r_core_visual_seek_animation (core, off);
//r_core_seek (core, off, 1);
}
r_core_block_read (core, 1);
}
} else
switch (ch) {
case 0x0d:
{
r_cons_enable_mouse (R_TRUE);
RAnalOp *op = r_core_anal_op (core, core->offset+cursor);
if (op) {
if (op->type == R_ANAL_OP_TYPE_JMP ||
op->type == R_ANAL_OP_TYPE_CJMP ||
op->type == R_ANAL_OP_TYPE_CALL) {
r_io_sundo_push (core->io, offset);
r_core_visual_seek_animation(core, op->jump);
}
}
r_anal_op_free (op);
}
break;
case 90: // shift+tab
if (!strcmp (printfmt[0], "x"))
printfmt[0] = "pxa";
else printfmt[0] = "x";
break;
case 9: // tab
{ // XXX: unify diff mode detection
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->rwx & 2)) {
r_cons_printf ("\nFile has been opened in read-only mode. Use -w flag\n");
r_cons_any_key ();
return R_TRUE;
}
r_cons_printf ("Enter assembler opcodes separated with ';':\n");
showcursor (core, R_TRUE);
r_cons_flush ();
r_cons_set_raw (R_FALSE);
strcpy (buf, "wa ");
r_line_set_prompt (":> ");
if (r_cons_fgets (buf+3, 1000, 0, NULL) <0) buf[0]='\0';
if (*buf) {
if (curset) r_core_seek (core, core->offset + cursor, 0);
r_core_cmd (core, buf, R_TRUE);
if (curset) r_core_seek (core, core->offset - cursor, 1);
}
showcursor (core, R_FALSE);
r_cons_set_raw (R_TRUE);
break;
case '!':
r_cons_2048();
break;
case 'o':
visual_offset (core);
break;
case 'A':
{ int oc = curset;
ut64 off = curset? core->offset+cursor : core->offset;
curset = 0;
r_core_visual_asm (core, off);
curset = oc;
}
break;
case 'c':
setcursor (core, curset?0:1);
break;
case 'C':
color = color? 0: 1;
r_config_set_i (core->config, "scr.color", color);
break;
case 'd':
r_core_visual_define (core);
break;
case 'D':
setdiff (core);
break;
case 'f':
{
int range, min, max;
char name[256], *n;
r_line_set_prompt ("flag name: ");
showcursor (core, R_TRUE);
if (r_cons_fgets (name, sizeof (name), 0, NULL) >=0 && *name) {
n = r_str_chop (name);
if (*name=='-') {
if (*n) r_flag_unset (core->flags, n+1, NULL);
} else {
if (ocursor != -1) {
min = R_MIN (cursor, ocursor);
max = R_MAX (cursor, ocursor);
} else {
min = max = cursor;
}
range = max-min+1;
if (range<1) range = 1;
if (*n) r_flag_set (core->flags, n,
core->offset + min, range, 1);
}
}
}
showcursor (core, R_FALSE);
break;
case 'F':
r_flag_unset_i (core->flags, core->offset + cursor, NULL);
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 (core->file && !(core->file->rwx & 2)) {
r_cons_printf ("\nFile has been opened in read-only mode. Use -w flag\n");
r_cons_any_key ();
return R_TRUE;
}
showcursor (core, R_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-cursor);
r_core_cmd (core, buf, 0);
free (p);
break;
}
delta = (ocursor!=-1)? R_MIN (cursor, ocursor): cursor;
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 (ocursor != -1) {
int bs = R_ABS (cursor-ocursor)+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 (curset) r_core_seek (core, core->offset + delta, 0);
r_core_cmd (core, buf, 1);
if (curset) r_core_seek (core, offset, 1);
r_cons_set_raw (1);
showcursor (core, R_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;
case 'M':
r_core_visual_mounts (core);
break;
case 't':
r_core_visual_trackflags (core);
break;
case 'x':
{
int count = 0;
RList *xrefs = NULL;
RAnalRef *refi;
RListIter *iter;
RAnalFunction *fun;
if ((xrefs = r_anal_xref_get (core->anal, core->offset))) {
r_cons_gotoxy (1, 1);
r_cons_printf ("[GOTO XREF]> \n");
if (r_list_empty (xrefs)) {
r_cons_printf ("\tNo XREF found at 0x%"PFMT64x"\n", core->offset);
r_cons_any_key ();
r_cons_clear00 ();
} else {
r_list_foreach (xrefs, iter, refi) {
fun = r_anal_fcn_find (core->anal, refi->addr, R_ANAL_FCN_TYPE_NULL);
r_cons_printf (" [%i] 0x%08"PFMT64x" %s XREF 0x%08"PFMT64x" (%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?fun->name:"unk");
if (++count > 9) break;
}
}
} else xrefs = NULL;
R_API int r_cons_arrow_to_hjkl(int ch) {
I->mouse_event = 0;
/* emacs */
switch ((ut8)ch) {
case 0xc3:
r_cons_readchar();
ch='K';
break; // emacs repag (alt + v)
case 0x16:
ch='J';
break; // emacs avpag (ctrl + v)
case 0x10:
ch='k';
break; // emacs up (ctrl + p)
case 0x0e:
ch='j';
break; // emacs down (ctrl + n)
case 0x06:
ch='l';
break; // emacs right (ctrl + f)
case 0x02:
ch='h';
break; // emacs left (ctrl + b)
}
if (ch != 0x1b)
return ch;
ch = r_cons_readchar ();
if (!ch) return 0;
switch (ch) {
case 0x1b:
ch = 'q'; // XXX: must be 0x1b (R_CONS_KEY_ESC)
break;
case 0x4f: // function keys from f1 to f4
ch = r_cons_readchar ();
ch = 0xf1 + (ch&0xf);
break;
case '[': // function keys (2)
ch = r_cons_readchar ();
switch (ch) {
case '[':
ch = r_cons_readchar ();
switch (ch) {
case '2':
ch = R_CONS_KEY_F11;
break;
case 'A':
ch = R_CONS_KEY_F1;
break;
case 'B':
ch = R_CONS_KEY_F2;
break;
case 'C':
ch = R_CONS_KEY_F3;
break;
case 'D':
ch = R_CONS_KEY_F4;
break;
}
break;
case '2':
ch = r_cons_readchar ();
switch (ch) {
case 0x7e:
ch = R_CONS_KEY_F12;
break;
default:
r_cons_readchar ();
switch (ch) {
case '0':
ch = R_CONS_KEY_F9;
break;
case '1':
ch = R_CONS_KEY_F10;
break;
case '3':
ch = R_CONS_KEY_F11;
break;
}
break;
}
break;
case '1':
ch = r_cons_readchar ();
switch (ch) {
case ':': // arrow+shift
ch = r_cons_readchar ();
ch = r_cons_readchar ();
switch (ch) {
case 'A':
ch = 'K';
break;
case 'B':
ch = 'J';
break;
case 'C':
ch = 'L';
break;
case 'D':
ch = 'H';
break;
}
break;
/*
case '1': ch = R_CONS_KEY_F1; break;
case '2': ch = R_CONS_KEY_F2; break;
case '3': ch = R_CONS_KEY_F3; break;
case '4': ch = R_CONS_KEY_F4; break;
*/
case '5':
r_cons_readchar ();
ch = 0xf5;
break;
case '6':
r_cons_readchar ();
ch = 0xf7;
break;
case '7':
r_cons_readchar ();
ch = 0xf6;
break;
case '8':
r_cons_readchar ();
ch = 0xf7;
break;
case '9':
r_cons_readchar ();
ch = 0xf8;
break;
} // F9-F12 not yet supported!!
break;
case '5':
ch='K';
break; // repag
case '6':
ch='J';
break; // avpag
case 'A':
ch='k';
break; // up
case 'B':
ch='j';
break; // down
case 'C':
ch='l';
break; // right
case 'D':
ch='h';
break; // left
case 'M': // Mouse events
ch = r_cons_readchar ();
/* Skip the x/y coordinates */
(void)r_cons_readchar();
(void)r_cons_readchar();
if (ch==0x20) {
// click
r_cons_enable_mouse (R_FALSE);
ch = 0;
//r_cons_enable_mouse (R_TRUE);
} else if (ch >= 64 + 32) {
/* Grab wheel events only */
I->mouse_event = 1;
ch = "kj"[(ch - (64 + 32))&1];
} else {
// temporary disable the mouse wheel to allow select
r_cons_enable_mouse (R_FALSE);
(void)r_cons_readchar ();
ch = 0;
}
}
break;
}
return ch;
}
R_API int r_cons_arrow_to_hjkl(int ch) {
if (ch==0x1b) {
#if 0
//__UNIX__
if (!is_fd_ready (0))
return 0;
#endif
ch = r_cons_readchar ();
if (!ch) return 0;
switch (ch) {
case 0x1b:
ch = 'q'; // XXX: must be 0x1b (R_CONS_KEY_ESC)
break;
case 0x4f: // function keys from f1 to f4
ch = r_cons_readchar ();
ch = 0xf1 + (ch&0xf);
break;
case 0:
case '[': // function keys (2)
ch = r_cons_readchar ();
switch (ch) {
case '[':
ch = r_cons_readchar ();
switch (ch) {
case '2': ch = R_CONS_KEY_F11; break;
case 'A': ch = R_CONS_KEY_F1; break;
case 'B': ch = R_CONS_KEY_F2; break;
case 'C': ch = R_CONS_KEY_F3; break;
case 'D': ch = R_CONS_KEY_F4; break;
}
break;
case '2':
ch = r_cons_readchar ();
switch (ch) {
case 0x7e:
ch = R_CONS_KEY_F12;
break;
default:
r_cons_readchar ();
switch (ch) {
case '0': ch = R_CONS_KEY_F9; break;
case '1': ch = R_CONS_KEY_F10; break;
case '3': ch = R_CONS_KEY_F11; break;
}
break;
}
break;
case '1':
ch = r_cons_readchar ();
switch (ch) {
case ':': // arrow+shift
ch = r_cons_readchar ();
ch = r_cons_readchar ();
switch (ch) {
case 'A': ch = 'K'; break;
case 'B': ch = 'J'; break;
case 'C': ch = 'L'; break;
case 'D': ch = 'H'; break;
}
break;
/*
case '1': ch = R_CONS_KEY_F1; break;
case '2': ch = R_CONS_KEY_F2; break;
case '3': ch = R_CONS_KEY_F3; break;
case '4': ch = R_CONS_KEY_F4; break;
*/
case '5':
r_cons_readchar ();
ch = 0xf5;
break;
case '6':
r_cons_readchar ();
ch = 0xf7;
break;
case '7':
r_cons_readchar ();
ch = 0xf6;
break;
case '8':
r_cons_readchar ();
ch = 0xf7;
break;
case '9':
r_cons_readchar ();
ch = 0xf8;
break;
} // F9-F12 not yet supported!!
break;
case '5': ch='K'; break; // repag
case '6': ch='J'; break; // avpag
case 'A': ch='k'; break; // up
case 'B': ch='j'; break; // down
case 'C': ch='l'; break; // right
case 'D': ch='h'; break; // left
case 'M': // Mouse events
ch = r_cons_readchar ();
/* Skip the x/y coordinates */
(void)r_cons_readchar();
(void)r_cons_readchar();
if (ch==0x20) {
// click
r_cons_enable_mouse (R_FALSE);
ch = 0;
//r_cons_enable_mouse (R_TRUE);
} else
if (ch >= 64 + 32) {
/* Grab wheel events only */
ch = "kj"[(ch - (64 + 32))&1];
} else {
// temporary disable the mouse wheel to allow select
r_cons_enable_mouse (R_FALSE);
(void)r_cons_readchar ();
ch = 0;
}
}
break;
}
}
return ch;
}
R_API int r_core_visual_panels(RCore *core) {
#define OS_INIT() ostack.size = 0; ostack.panels[0] = 0;
#define OS_PUSH(x) if (ostack.size<LIMIT) {ostack.panels[++ostack.size]=x;}
#define OS_POP() ((ostack.size>0)? ostack.panels[--ostack.size]:0)
int okey, key, wheel;
int w, h;
int asm_comments = 0;
int asm_bytes = 0;
n_panels = 0;
panels = NULL;
callgraph = 0;
_core = core;
OS_INIT();
w = r_cons_get_size (&h);
can = r_cons_canvas_new (w, h);
can->linemode = 1;
can->color = r_config_get_i (core->config, "scr.color");
// disable colors in disasm because canvas doesnt supports ansi text yet
r_config_set_i (core->config, "scr.color", 0);
//can->color = 0;
if (!can) {
eprintf ("Cannot create RCons.canvas context\n");
return R_FALSE;
}
n_panels = bbPanels (core, &panels);
if (!panels) {
r_config_set_i (core->config, "scr.color", can->color);
free (can);
return R_FALSE;
}
reloadPanels (core);
asm_comments = r_config_get_i (core->config, "asm.comments");
r_config_set_i (core->config, "asm.comments", 0);
asm_bytes = r_config_get_i (core->config, "asm.bytes");
r_config_set_i (core->config, "asm.bytes", 0);
repeat:
core->cons->event_data = core;
core->cons->event_resize = \
(RConsEvent)r_core_panels_refresh;
w = r_cons_get_size (&h);
Layout_run (panels);
r_core_panels_refresh (core);
wheel = r_config_get_i (core->config, "scr.wheel");
if (wheel)
r_cons_enable_mouse (R_TRUE);
// r_core_graph_inputhandle()
okey = r_cons_readchar ();
key = r_cons_arrow_to_hjkl (okey);
switch (key) {
case ' ':
case '\n':
if (menu_y) {
const char *action = menus_sub[menu_x][menu_y-1];
eprintf ("ACTION %s\n", action);
if (strstr (action, "New")) {
int i;
// add new panel for testing
for (i=0; panels[i].text; i++) {
// find last panel
}
panels[i].text = strdup ("Test");
panels[i].cmd = r_str_newf ("pxW $r-2");
panels[i].addr = core->offset;
panels[i].type = PANEL_TYPE_FRAME;
i++;
n_panels++;
panels[i].text = NULL;
}
if (strstr (action, "Quit")) {
goto beach;
}
}
break;
case '?':
r_cons_clear00 ();
r_cons_printf ("Visual Ascii Art Panels:\n"
" ! run r2048 game\n"
" . - center graph to the current node\n"
" : - run r2 command in prompt\n"
" hl - toggle scr.color\n"
" HL - move vertical column split\n"
" JK - select prev/next panels\n"
" jk - scroll/select menu\n"
" q - quit, back to visual mode\n"
);
r_cons_flush ();
r_cons_any_key (NULL);
break;
case ':':
core->vmode = R_FALSE;
r_core_visual_prompt_input (core);
core->vmode = R_TRUE;
break;
case 'C':
can->color = !!!can->color; //WTF
//r_config_swap (core->config, "scr.color");
// refresh graph
// reloadPanels (core);
break;
case '!':
r_cons_2048 ();
break;
case 'j':
if (panels[curnode].type == PANEL_TYPE_FLOAT) {
if (menus_sub[menu_x][menu_y])
menu_y ++;
}
break;
case 'k':
if (panels[curnode].type == PANEL_TYPE_FLOAT) {
menu_y --;
if (menu_y<0)
menu_y = 0;
}
break;
case 'J':
curnode++;
if (!panels[curnode].text) {
curnode--;
}
break;
case 'K':
curnode--;
if (curnode<0)
curnode = 0;
break;
case 'H':
COLW += 4;
break;
case 'L':
COLW -= 4;
if (COLW<0)
COLW=0;
break;
case 'h':
if (menu_x) {
menu_x --;
menu_y = menu_y?1:0;
}
break;
case 'l':
if (menus[menu_x + 1]) {
menu_x ++;
menu_y = menu_y?1:0;
}
break;
case 'q':
case -1: // EOF
goto beach;
case 27: // ESC
if (r_cons_readchar () == 91) {
if (r_cons_readchar () == 90) {
}
}
break;
default:
//eprintf ("Key %d\n", key);
//sleep (1);
break;
}
goto repeat;
beach:
free (panels);
r_config_set_i (core->config, "scr.color", can->color);
free (can);
r_config_set_i (core->config, "asm.comments", asm_comments);
r_config_set_i (core->config, "asm.bytes", asm_bytes);
return R_TRUE;
}
