radare2/libr/core/cmd_seek.c
2019-02-03 09:06:45 -06:00

798 lines
20 KiB
C

/* radare - LGPL - Copyright 2009-2018 - pancake */
#include "r_types.h"
#include "r_config.h"
#include "r_cons.h"
#include "r_core.h"
#include "r_debug.h"
#include "r_io.h"
static const char *help_msg_s[] = {
"Usage: s", "", " # Help for the seek commands. See ?$? to see all variables",
"s", "", "Print current address",
"s:", "pad", "Print current address with N padded zeros (defaults to 8)",
"s", " addr", "Seek to address",
"s-", "", "Undo seek",
"s-*", "", "Reset undo seek history",
"s-", " n", "Seek n bytes backward",
"s--", "[n]", "Seek blocksize bytes backward (/=n)",
"s+", "", "Redo seek",
"s+", " n", "Seek n bytes forward",
"s++", "[n]", "Seek blocksize bytes forward (/=n)",
"s[j*=!]", "", "List undo seek history (JSON, =list, *r2, !=names, s==)",
"s/", " DATA", "Search for next occurrence of 'DATA'",
"s/x", " 9091", "Search for next occurrence of \\x90\\x91",
"s.", "hexoff", "Seek honoring a base from core->offset",
"sa", " [[+-]a] [asz]", "Seek asz (or bsize) aligned to addr",
"sb", "", "Seek aligned to bb start",
"sC", "[?] string", "Seek to comment matching given string",
"sf", "", "Seek to next function (f->addr+f->size)",
"sf", " function", "Seek to address of specified function",
"sf.", "", "Seek to the beginning of current function",
"sg/sG", "", "Seek begin (sg) or end (sG) of section or file",
"sl", "[?] [+-]line", "Seek to line",
"sn/sp", " ([nkey])", "Seek to next/prev location, as specified by scr.nkey",
"so", " [N]", "Seek to N next opcode(s)",
"sr", " pc", "Seek to register",
"ss", "", "Seek silently (without adding an entry to the seek history)",
// "sp [page] seek page N (page = block)",
NULL
};
static const char *help_msg_sC[] = {
"Usage:", "sC", "Comment grep",
"sC", "*", "List all comments",
"sC", " str", "Seek to the first comment matching 'str'",
NULL
};
static const char *help_msg_sl[] = {
"Usage:", "sl+ or sl- or slc", "",
"sl", " [line]", "Seek to absolute line",
"sl", "[+-][line]", "Seek to relative line",
"slc", "", "Clear line cache",
"sll", "", "Show total number of lines",
NULL
};
static const char *help_msg_ss[] = {
"Usage: ss", "", " # Seek silently (not recorded in the seek history)",
"s?", "", "Works with all s subcommands",
NULL
};
static void cmd_seek_init(RCore *core) {
DEFINE_CMD_DESCRIPTOR (core, s);
DEFINE_CMD_DESCRIPTOR (core, sC);
DEFINE_CMD_DESCRIPTOR (core, sl);
DEFINE_CMD_DESCRIPTOR (core, ss);
}
static void __init_seek_line(RCore *core) {
ut64 from, to;
r_config_bump (core->config, "lines.to");
from = r_config_get_i (core->config, "lines.from");
to = r_config_get_i (core->config, "lines.to");
if (r_core_lines_initcache (core, from, to) == -1) {
eprintf ("ERROR: \"lines.from\" and \"lines.to\" must be set\n");
}
}
static void printPadded(RCore *core, int pad) {
if (pad < 1) {
pad = 8;
}
char *fmt = r_str_newf ("0x%%0%d" PFMT64x, pad);
char *off = r_str_newf (fmt, core->offset);
r_cons_printf ("%s\n", off);
free (off);
free (fmt);
}
static void __get_current_line(RCore *core) {
if (core->print->lines_cache_sz > 0) {
int curr = r_util_lines_getline (core->print->lines_cache, core->print->lines_cache_sz, core->offset);
r_cons_printf ("%d\n", curr);
}
}
static void __seek_line_absolute(RCore *core, int numline) {
if (numline < 1 || numline > core->print->lines_cache_sz - 1) {
eprintf ("ERROR: Line must be between 1 and %d\n", core->print->lines_cache_sz - 1);
} else {
r_core_seek (core, core->print->lines_cache[numline - 1], 1);
}
}
static void __seek_line_relative(RCore *core, int numlines) {
int curr = r_util_lines_getline (core->print->lines_cache, core->print->lines_cache_sz, core->offset);
if (numlines > 0 && curr + numlines >= core->print->lines_cache_sz - 1) {
eprintf ("ERROR: Line must be < %d\n", core->print->lines_cache_sz - 1);
} else if (numlines < 0 && curr + numlines < 1) {
eprintf ("ERROR: Line must be > 1\n");
} else {
r_core_seek (core, core->print->lines_cache[curr + numlines - 1], 1);
}
}
static void __clean_lines_cache(RCore *core) {
core->print->lines_cache_sz = -1;
R_FREE (core->print->lines_cache);
}
R_API int r_core_lines_currline(RCore *core) { // make priv8 again
int imin = 0;
int imax = core->print->lines_cache_sz;
int imid = 0;
while (imin <= imax) {
imid = imin + ((imax - imin) / 2);
if (core->print->lines_cache[imid] == core->offset) {
return imid;
} else if (core->print->lines_cache[imid] < core->offset) {
imin = imid + 1;
} else {
imax = imid - 1;
}
}
return imin;
}
R_API int r_core_lines_initcache(RCore *core, ut64 start_addr, ut64 end_addr) {
int i, line_count;
int bsz = core->blocksize;
char *buf;
ut64 off = start_addr;
ut64 baddr;
if (start_addr == UT64_MAX || end_addr == UT64_MAX) {
return -1;
}
free (core->print->lines_cache);
core->print->lines_cache = R_NEWS0 (ut64, bsz);
if (!core->print->lines_cache) {
return -1;
}
baddr = r_config_get_i (core->config, "bin.baddr");
line_count = start_addr? 0: 1;
core->print->lines_cache[0] = start_addr? 0: baddr;
buf = malloc (bsz);
if (!buf) {
return -1;
}
r_cons_break_push (NULL, NULL);
while (off < end_addr) {
if (r_cons_is_breaked ()) {
break;
}
r_io_read_at (core->io, off, (ut8 *) buf, bsz);
for (i = 0; i < bsz; i++) {
if (buf[i] == '\n') {
core->print->lines_cache[line_count] = start_addr? off + i + 1: off + i + 1 + baddr;
line_count++;
if (line_count % bsz == 0) {
ut64 *tmp = realloc (core->print->lines_cache,
(line_count + bsz) * sizeof (ut64));
if (tmp) {
core->print->lines_cache = tmp;
} else {
R_FREE (core->print->lines_cache);
goto beach;
}
}
}
}
off += bsz;
}
free (buf);
r_cons_break_pop ();
return line_count;
beach:
free (buf);
r_cons_break_pop ();
return -1;
}
static void seek_to_register(RCore *core, const char *input, bool is_silent) {
ut64 off;
if (core->io->debug) {
off = r_debug_reg_get (core->dbg, input);
if (!is_silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
r_core_seek (core, off, 1);
} else {
RReg *orig = core->dbg->reg;
core->dbg->reg = core->anal->reg;
off = r_debug_reg_get (core->dbg, input);
core->dbg->reg = orig;
if (!is_silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
r_core_seek (core, off, 1);
}
}
static int cmd_seek_opcode_backward(RCore *core, int numinstr) {
int i, val = 0;
// N previous instructions
ut64 addr = core->offset;
int ret = 0;
if (r_core_prevop_addr (core, core->offset, numinstr, &addr)) {
ret = core->offset - addr;
} else {
#if 0
// core_asm_bwdis_len is buggy as hell we should kill it. seems like prevop_addr
// works as expected, because is the one used from visual
ret = r_core_asm_bwdis_len (core, &instr_len, &addr, numinstr);
#endif
addr = core->offset;
const int mininstrsize = r_anal_archinfo (core->anal, R_ANAL_ARCHINFO_MIN_OP_SIZE);
for (i = 0; i < numinstr; i++) {
ut64 prev_addr = r_core_prevop_addr_force (core, addr, 1);
if (prev_addr == UT64_MAX) {
prev_addr = addr - mininstrsize;
}
if (prev_addr == UT64_MAX || prev_addr >= core->offset) {
break;
}
RAsmOp op = {0};
r_core_seek (core, prev_addr, 1);
r_asm_disassemble (core->assembler, &op, core->block, 32);
if (op.size < mininstrsize) {
op.size = mininstrsize;
}
val += op.size;
addr = prev_addr;
}
}
r_core_seek (core, addr, true);
val += ret;
return val;
}
static int cmd_seek_opcode_forward (RCore *core, int n) {
// N forward instructions
int i, ret, val = 0;
for (val = i = 0; i < n; i++) {
RAnalOp op;
ret = r_anal_op (core->anal, &op, core->offset, core->block,
core->blocksize, R_ANAL_OP_MASK_BASIC);
if (ret < 1) {
ret = 1;
}
r_core_seek_delta (core, ret);
r_anal_op_fini (&op);
val += ret;
}
return val;
}
static void cmd_seek_opcode(RCore *core, const char *input) {
if (input[0] == '?') {
eprintf ("Usage: so [-][n]\n");
return;
}
if (!strcmp (input, "-")) {
input = "-1";
}
int n = r_num_math (core->num, input);
if (n == 0) {
n = 1;
}
int val = (n < 0)
? cmd_seek_opcode_backward (core, -n)
: cmd_seek_opcode_forward (core, n);
core->num->value = val;
}
static int cmd_seek(void *data, const char *input) {
RCore *core = (RCore *) data;
char *cmd, *p;
ut64 off = core->offset;
if (!*input) {
r_cons_printf ("0x%"PFMT64x "\n", core->offset);
return 0;
}
char *ptr;
if ((ptr = strstr (input, "+.")) != NULL) {
char *dup = strdup (input);
dup[ptr - input] = '\x00';
off = r_num_math (core->num, dup + 1);
core->offset = off;
free (dup);
}
const char *inputnum = strchr (input, ' ');
{
const char *u_num = inputnum? inputnum + 1: input + 1;
off = r_num_math (core->num, u_num);
if (*u_num == '-') {
off = -(st64)off;
}
}
#if 1
// int sign = 1;
if (input[0] == ' ') {
switch (input[1]) {
case '-':
// sign = -1;
/* pass thru */
case '+':
input++;
break;
}
}
#endif
bool silent = false;
if (*input == 's') {
silent = true;
input++;
if (*input == '?') {
r_core_cmd_help (core, help_msg_ss);
return 0;
}
}
switch (*input) {
case 'r': // "sr"
if (input[1] && input[2]) {
seek_to_register (core, input + 2, silent);
} else {
eprintf ("|Usage| 'sr PC' seek to program counter register\n");
}
break;
case 'C': // "sC"
if (input[1] == '*') { // "sC*"
r_core_cmd0 (core, "C*~^\"CC");
} else if (input[1] == ' ') {
typedef struct {
ut64 addr;
char *str;
} MetaCallback;
int count = 0;
MetaCallback cb = {
0, NULL
};
ut64 addr;
char key[128];
const char *val, *comma;
char *list = sdb_get (core->anal->sdb_meta, "meta.C", 0);
char *str, *next, *cur = list;
if (list) {
for (;;) {
cur = sdb_anext (cur, &next);
addr = sdb_atoi (cur);
snprintf (key, sizeof (key) - 1, "meta.C.0x%"PFMT64x, addr);
val = sdb_const_get (core->anal->sdb_meta, key, 0);
if (val) {
comma = strchr (val, ',');
if (comma) {
str = (char *) sdb_decode (comma + 1, 0);
if (strstr (str, input + 2)) {
r_cons_printf ("0x%08"PFMT64x " %s\n", addr, str);
count++;
cb.addr = addr;
free (cb.str);
cb.str = str;
} else {
free (str);
}
}
} else {
eprintf ("sdb_const_get key not found '%s'\n", key);
}
if (!next) {
break;
}
cur = next;
}
}
switch (count) {
case 0:
eprintf ("No matching comments\n");
break;
case 1:
off = cb.addr;
if (!silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
r_core_seek (core, off, 1);
r_core_block_read (core);
break;
default:
eprintf ("Too many results\n");
break;
}
free (cb.str);
} else {
r_core_cmd_help (core, help_msg_sC);
}
break;
case ' ': // "s "
{
ut64 addr = r_num_math (core->num, input + 1);
if (core->num->nc.errors) {
if (r_cons_singleton ()->is_interactive) {
eprintf ("Cannot seek to unknown address '%s'\n", core->num->nc.calc_buf);
}
break;
}
if (!silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
r_core_seek (core, addr, 1);
r_core_block_read (core);
}
break;
case '/': // "s/"
{
const char *pfx = r_config_get (core->config, "search.prefix");
const ut64 saved_from = r_config_get_i (core->config, "search.from");
const ut64 saved_maxhits = r_config_get_i (core->config, "search.maxhits");
// kwidx cfg var is ignored
int kwidx = core->search->n_kws; // (int)r_config_get_i (core->config, "search.kwidx")-1;
if (kwidx < 0) {
kwidx = 0;
}
switch (input[1]) {
case ' ':
case 'v':
case 'V':
case 'w':
case 'W':
case 'z':
case 'm':
case 'c':
case 'A':
case 'e':
case 'E':
case 'i':
case 'R':
case 'r':
case '/':
case 'x':
r_config_set_i (core->config, "search.from", core->offset + 1);
r_config_set_i (core->config, "search.maxhits", 1);
r_core_cmdf (core, "s+1; %s; s-1; s %s%d_0; f-%s%d_0",
input, pfx, kwidx, pfx, kwidx, pfx, kwidx);
r_config_set_i (core->config, "search.from", saved_from);
r_config_set_i (core->config, "search.maxhits", saved_maxhits);
break;
case '?':
eprintf ("Usage: s/.. arg.\n");
r_cons_printf ("/?\n");
break;
default:
eprintf ("unknown search method\n");
break;
}
}
break;
case '.': // "s." "s.."
for (input++; *input == '.'; input++) {
;
}
r_core_seek_base (core, input);
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
break;
case 'j': // "sj"
{
RList /*<ut64 *>*/ *addrs = r_list_newf (free);
RList /*<char *>*/ *names = r_list_newf (free);
RList *list = r_io_sundo_list (core->io, '!');
ut64 lsz = 0;
ut64 i;
RListIter *iter;
RIOUndos *undo;
if (list) {
r_list_foreach (list, iter, undo) {
char *name = NULL;
RFlagItem *f = r_flag_get_at (core->flags, undo->off, true);
if (f) {
if (f->offset != undo->off) {
name = r_str_newf ("%s+%d", f->name,
(int)(undo->off- f->offset));
} else {
name = strdup (f->name);
}
}
if (!name) {
name = strdup ("");
}
ut64 *val = malloc (sizeof (ut64));
if (!val) {
free (name);
break;
}
*val = undo->off;
r_list_append (addrs, val);
r_list_append (names, strdup (name));
lsz++;
free (name);
}
r_list_free (list);
}
r_cons_printf ("[");
for (i = 0; i < lsz; ++i) {
ut64 *addr = r_list_get_n (addrs, i);
const char *name = r_list_get_n (names, i);
// XXX(should the "name" field be optional? That might make
// a bit more sense.
r_cons_printf ("{\"offset\":%"PFMT64d",\"symbol\":\"%s\"}", *addr, name);
if (i != lsz - 1) {
r_cons_printf (",");
}
}
r_cons_printf ("]\n");
r_list_free (addrs);
r_list_free (names);
}
break;
case '*': // "s*"
case '=': // "s="
case '!': // "s!"
{
char mode = input[0];
if (input[1] == '=') {
mode = 0;
}
RList *list = r_io_sundo_list (core->io, mode);
if (list) {
RListIter *iter;
RIOUndos *undo;
r_list_foreach (list, iter, undo) {
char *name = NULL;
RFlagItem *f = r_flag_get_at (core->flags, undo->off, true);
if (f) {
if (f->offset != undo->off) {
name = r_str_newf ("%s + %d\n", f->name,
(int)(undo->off - f->offset));
} else {
name = strdup (f->name);
}
}
if (mode) {
r_cons_printf ("0x%"PFMT64x" %s\n", undo->off, name? name: "");
} else {
if (!name) {
name = r_str_newf ("0x%"PFMT64x, undo->off);
}
r_cons_printf ("%s%s", name, iter->n? " > ":"");
}
free (name);
}
r_list_free (list);
if (!mode) {
r_cons_newline ();
}
}
}
break;
case '+': // "s+"
if (input[1] != '\0') {
int delta = off;
if (input[1] == '+') {
delta = core->blocksize;
int mult = r_num_math (core->num, input + 2);
if (mult > 0) {
delta /= mult;
}
}
// int delta = (input[1] == '+')? core->blocksize: off;
if (!silent) {
r_io_sundo_push (core->io, core->offset,
r_print_get_cursor (core->print));
}
r_core_seek_delta (core, delta);
r_core_block_read (core);
} else {
RIOUndos *undo = r_io_sundo_redo (core->io);
if (undo) {
r_core_seek (core, undo->off, 0);
r_core_block_read (core);
}
}
break;
case '-': // "s-"
switch (input[1]) {
case '*': // "s-*"
r_io_sundo_reset (core->io);
break;
case 0: // "s-"
{
RIOUndos *undo = r_io_sundo (core->io, core->offset);
if (undo) {
r_core_seek (core, undo->off, 0);
r_core_block_read (core);
}
}
break;
case '-': // "s--"
default:
{
int delta = -off;
if (input[1] == '-') {
delta = -core->blocksize;
int mult = r_num_math (core->num, input + 2);
if (mult > 0) {
delta /= mult;
}
}
if (!silent) {
r_io_sundo_push (core->io, core->offset,
r_print_get_cursor (core->print));
}
r_core_seek_delta (core, delta);
r_core_block_read (core);
}
break;
}
break;
case 'n': // "sn"
{
if (!silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
const char *nkey = (input[1] == ' ')
? input + 2
: r_config_get (core->config, "scr.nkey");
r_core_seek_next (core, nkey);
}
break;
case 'p': // "sp"
{
if (!silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
const char *nkey = (input[1] == ' ')
? input + 2
: r_config_get (core->config, "scr.nkey");
r_core_seek_previous (core, nkey);
}
break;
case 'a': // "sa"
off = core->blocksize;
if (input[1] && input[2]) {
cmd = strdup (input);
p = strchr (cmd + 2, ' ');
if (p) {
off = r_num_math (core->num, p + 1);;
*p = '\0';
}
cmd[0] = 's';
// perform real seek if provided
r_cmd_call (core->rcmd, cmd);
free (cmd);
}
if (!silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
r_core_seek_align (core, off, 0);
break;
case 'b': // "sb"
if (off == 0) {
off = core->offset;
}
if (!silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
r_core_anal_bb_seek (core, off);
break;
case 'f': { // "sf"
RAnalFunction *fcn;
switch (input[1]) {
case '\0': // "sf"
fcn = r_anal_get_fcn_in (core->anal, core->offset, 0);
if (fcn) {
r_core_seek (core, fcn->addr + r_anal_fcn_size (fcn), 1);
}
break;
case ' ': // "sf "
fcn = r_anal_fcn_find_name (core->anal, input + 2);
if (fcn) {
r_core_seek (core, fcn->addr, 1);
}
break;
case '.': // "sf."
fcn = r_anal_get_fcn_in (core->anal, core->offset, 0);
if (fcn) {
r_core_seek (core, fcn->addr, 1);
}
break;
}
break;
}
case 'o': // "so"
cmd_seek_opcode (core, input + 1);
break;
case 'g': // "sg"
{
RIOMap *map = r_io_map_get (core->io, core->offset);
if (map) {
r_core_seek (core, map->itv.addr, 1);
} else {
r_core_seek (core, 0, 1);
}
}
break;
case 'G': // "sG"
{
if (!core->file) {
break;
}
RIOMap *map = r_io_map_get (core->io, core->offset);
// XXX: this +2 is a hack. must fix gap between sections
if (map) {
r_core_seek (core, map->itv.addr + map->itv.size + 2, 1);
} else {
r_core_seek (core, r_io_fd_size (core->io, core->file->fd), 1);
}
}
break;
case 'l': // "sl"
{
int sl_arg = r_num_math (core->num, input + 1);
switch (input[1]) {
case '\0': // "sl"
if (!core->print->lines_cache) {
__init_seek_line (core);
}
__get_current_line (core);
break;
case ' ': // "sl "
if (!core->print->lines_cache) {
__init_seek_line (core);
}
__seek_line_absolute (core, sl_arg);
break;
case '+': // "sl+"
case '-': // "sl-"
if (!core->print->lines_cache) {
__init_seek_line (core);
}
__seek_line_relative (core, sl_arg);
break;
case 'c': // "slc"
__clean_lines_cache (core);
break;
case 'l': // "sll"
if (!core->print->lines_cache) {
__init_seek_line (core);
}
eprintf ("%d lines\n", core->print->lines_cache_sz - 1);
break;
case '?': // "sl?"
r_core_cmd_help (core, help_msg_sl);
break;
}
}
break;
case ':': // "s:"
printPadded (core, atoi (input + 1));
break;
case '?': // "s?"
r_core_cmd_help (core, help_msg_s);
break;
default:
{
ut64 n = r_num_math (core->num, input);
if (n) {
if (!silent) {
r_io_sundo_push (core->io, core->offset, r_print_get_cursor (core->print));
}
r_core_seek (core, n, 1);
r_core_block_read (core);
}
}
break;
}
return 0;
}