mirror of
https://github.com/radareorg/radare2.git
synced 2024-12-11 23:16:05 +00:00
e55da9853f
Tests fixed: db/cmd/feat_grep multi-command single-line grep db/cmd/cmd_zignature af ; af ; zfs libc-v10.sig - newshell db/cmd/cmd_zignature af ; af ; zfs libc-v7.sig - newshell New test failed: db/tools/rax2 C array {0x33,33,23,32} Signed-off-by: Anthoine Bourgeois <anthoine.bourgeois@gmail.com>
5610 lines
138 KiB
C
5610 lines
138 KiB
C
/* radare - LGPL - Copyright 2009-2021 - nibble, pancake */
|
|
|
|
#define INTERACTIVE_MAX_REP 1024
|
|
|
|
#include <r_core.h>
|
|
#include <r_anal.h>
|
|
#include <r_cons.h>
|
|
#include <r_cmd.h>
|
|
#include <stdint.h>
|
|
#include <sys/types.h>
|
|
#include <ctype.h>
|
|
#include <stdarg.h>
|
|
#include "cmd_helps.h"
|
|
#if __UNIX__
|
|
#include <sys/utsname.h>
|
|
#include <pwd.h>
|
|
#endif
|
|
|
|
// NOTE: this should be in sync with SPECIAL_CHARACTERS in
|
|
// radare2-shell-parser grammar, except for ", ' and
|
|
// whitespaces, because we let cmd_substitution_arg create
|
|
// new arguments
|
|
static const char *SPECIAL_CHARS_REGULAR = "@;~$#|`\"'()<>";
|
|
#if 0
|
|
static const char *SPECIAL_CHARS_PF = "@;~$#|`\"'<>";
|
|
static const char *SPECIAL_CHARS_DOUBLE_QUOTED = "\"";
|
|
static const char *SPECIAL_CHARS_SINGLE_QUOTED = "'";
|
|
|
|
static char *escape_special_chars(char *s, const char *special_chars) {
|
|
size_t s_len = strlen (s);
|
|
char *d = R_NEWS (char, s_len * 2 + 1);
|
|
int i, j = 0;
|
|
for (i = 0; i < s_len; i++) {
|
|
if (strchr (special_chars, s[i])) {
|
|
d[j++] = '\\';
|
|
}
|
|
d[j++] = s[i];
|
|
}
|
|
d[j++] = '\0';
|
|
free (s);
|
|
return d;
|
|
}
|
|
#endif
|
|
|
|
|
|
static RCmdDescriptor *cmd_descriptor(const char *cmd, const char *help[]) {
|
|
RCmdDescriptor *d = R_NEW0 (RCmdDescriptor);
|
|
if (d) {
|
|
d->cmd = cmd;
|
|
d->help_msg = help;
|
|
}
|
|
return d;
|
|
}
|
|
|
|
#define DEFINE_CMD_DESCRIPTOR(core, cmd_) \
|
|
{ \
|
|
RCmdDescriptor *d = cmd_descriptor (#cmd_, help_msg_##cmd_); \
|
|
if (d) { \
|
|
r_list_append ((core)->cmd_descriptors, d); \
|
|
} \
|
|
}
|
|
|
|
#define DEFINE_CMD_DESCRIPTOR_WITH_DETAIL(core, cmd_) \
|
|
{ \
|
|
RCmdDescriptor *d = cmd_descriptor (#cmd_, help_msg##cmd_); \
|
|
if (d) { \
|
|
d->help_detail = help_detail_##cmd_; \
|
|
r_list_append ((core)->cmd_descriptors, d); \
|
|
} \
|
|
}
|
|
|
|
#define DEFINE_CMD_DESCRIPTOR_WITH_DETAIL2(core, cmd_) \
|
|
{ \
|
|
RCmdDescriptor *d = cmd_descriptor (#cmd_, help_msg_##cmd_); \
|
|
if (d) { \
|
|
d->help_detail = help_detail_##cmd_; \
|
|
d->help_detail2 = help_detail2_##cmd_; \
|
|
r_list_append ((core)->cmd_descriptors, d); \
|
|
} \
|
|
}
|
|
|
|
#define DEFINE_CMD_DESCRIPTOR_SPECIAL(core, cmd_, named_cmd) \
|
|
{ \
|
|
RCmdDescriptor *d = R_NEW0 (RCmdDescriptor); \
|
|
if (d) { \
|
|
d->cmd = #cmd_; \
|
|
d->help_msg = help_msg_##named_cmd; \
|
|
r_list_append ((core)->cmd_descriptors, d); \
|
|
} \
|
|
}
|
|
|
|
static int r_core_cmd_subst_i(RCore *core, char *cmd, char* colon, bool *tmpseek);
|
|
|
|
static int bb_cmpaddr(const void *_a, const void *_b) {
|
|
const RAnalBlock *a = _a, *b = _b;
|
|
return a->addr > b->addr ? 1 : (a->addr < b->addr ? -1 : 0);
|
|
}
|
|
|
|
static void cmd_debug_reg(RCore *core, const char *str);
|
|
|
|
#include "cmd_quit.c"
|
|
#include "cmd_hash.c"
|
|
#include "cmd_debug.c"
|
|
#include "cmd_log.c"
|
|
#include "cmd_flag.c"
|
|
#include "cmd_zign.c"
|
|
#include "cmd_project.c"
|
|
#include "cmd_write.c"
|
|
#include "cmd_cmp.c"
|
|
#include "cmd_eval.c"
|
|
#include "cmd_type.c"
|
|
#include "cmd_anal.c"
|
|
#include "cmd_open.c"
|
|
#include "cmd_meta.c"
|
|
#include "cmd_egg.c"
|
|
#include "cmd_info.c"
|
|
#include "cmd_macro.c"
|
|
#include "cmd_magic.c"
|
|
#include "cmd_mount.c"
|
|
#include "cmd_seek.c"
|
|
#include "cmd_search.c" // defines incDigitBuffer... used by cmd_print
|
|
#include "cmd_print.c"
|
|
#include "cmd_help.c"
|
|
#include "cmd_colon.c"
|
|
|
|
static const char *help_msg_dollar[] = {
|
|
"Usage:", "$alias[=cmd] [args...]", "Alias commands and strings (See ?$? for help on $variables)",
|
|
"$", "", "list all defined aliases",
|
|
"$*", "", "list all the aliases as r2 commands in base64",
|
|
"$**", "", "same as above, but using plain text",
|
|
"$", "foo:=123", "alias for 'f foo=123'",
|
|
"$", "foo-=4", "alias for 'f foo-=4'",
|
|
"$", "foo+=4", "alias for 'f foo+=4'",
|
|
"$", "foo", "alias for 's foo' (note that command aliases can override flag resolution)",
|
|
"$", "dis=base64:AAA==", "alias this base64 encoded text to be printed when $dis is called",
|
|
"$", "dis=$hello world", "alias this text to be printed when $dis is called",
|
|
"$", "dis=-", "open cfg.editor to set the new value for dis alias",
|
|
"$", "dis=af;pdf", "create command - analyze to show function",
|
|
"$", "test=#!pipe node /tmp/test.js", "create command - rlangpipe script",
|
|
"$", "dis=", "undefine alias",
|
|
"$", "dis", "execute the previously defined alias",
|
|
"$", "dis?", "show commands aliased by $dis",
|
|
"$", "dis?n", "show commands aliased by $dis, without a new line",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_star[] = {
|
|
"Usage:", "*<addr>[=[0x]value]", "Pointer read/write data/values",
|
|
"*", "entry0=cc", "write trap in entrypoint",
|
|
"*", "entry0+10=0x804800", "write value in delta address",
|
|
"*", "entry0", "read byte at given address",
|
|
"*", "/", "end multiline comment. (use '/*' to start mulitiline comment",
|
|
"TODO: last command should honor asm.bits", "", "",
|
|
NULL
|
|
};
|
|
|
|
static const char *cmd_table_help[] = {
|
|
"Usage:", ",[,.-/*jhr] [file]", "# load table data",
|
|
",", "", "display table",
|
|
", ", "[table-query]", "filter and print table. See ,? for more details",
|
|
",.", " file.csv", "load table from CSV file (comma dot)",
|
|
",,", "", "print table in csv format (comma comma)",
|
|
",-", "", "reset table",
|
|
",/", "?", "query/filter current table (non-destructive)",
|
|
",*", ">$foo", "print table as r2 commands",
|
|
",j", "", "print table in json format",
|
|
",h", " xxd foo bar cow", "define header column names and types",
|
|
",r", " 1 2 foo", "adds a row using the given format string",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_dot[] = {
|
|
"Usage:", ".[r2cmd] | [file] | [!command] | [(macro)]", "# define macro or interpret r2, r_lang,\n"
|
|
" cparse, d, es6, exe, go, js, lsp, pl, py, rb, sh, vala or zig file",
|
|
".", "", "repeat last command backward",
|
|
".", "r2cmd", "interpret the output of the command as r2 commands",
|
|
"..", " [file]", "run the output of the execution of a script as r2 commands",
|
|
"...", "", "repeat last command forward (same as \\n)",
|
|
".:", "8080", "listen for commands on given tcp port",
|
|
".--", "", "terminate tcp server for remote commands",
|
|
".", " foo.r2", "interpret script",
|
|
".-", "", "open cfg.editor and interpret tmp file",
|
|
".*", " file ...", "same as #!pipe open cfg.editor and interpret tmp file",
|
|
".!", "rabin -ri $FILE", "interpret output of command",
|
|
".", "(foo 1 2 3)", "run macro 'foo' with args 1, 2, 3",
|
|
"./", " ELF", "interpret output of command /m ELF as r. commands",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_equal[] = {
|
|
"Usage:", " =[:!+-=ghH] [...]", " # connect with other instances of r2",
|
|
"\nremote commands:", "", "",
|
|
"=", "", "list all open connections",
|
|
"=<", "[fd] cmd", "send output of local command to remote fd", // XXX may not be a special char
|
|
"=", "[fd] cmd", "exec cmd at remote 'fd' (last open is default one)",
|
|
"=!", " cmd", "run command via r_io_system",
|
|
"=+", " [proto://]host:port", "connect to remote host:port (*rap://, raps://, tcp://, udp://, http://)",
|
|
"=-", "[fd]", "remove all hosts or host 'fd'",
|
|
"==", "[fd]", "open remote session with host 'fd', 'q' to quit",
|
|
"=!=", "", "disable remote cmd mode",
|
|
"!=!", "", "enable remote cmd mode",
|
|
"\nservers:","","",
|
|
".:", "9000", "start the tcp server (echo x|nc ::1 9090 or curl ::1:9090/cmd/x)",
|
|
"=:", "port", "start the rap server (o rap://9999)",
|
|
"=g", "[?]", "start the gdbserver",
|
|
"=h", "[?]", "start the http webserver",
|
|
"=H", "[?]", "start the http webserver (and launch the web browser)",
|
|
"\nother:","","",
|
|
"=&", ":port", "start rap server in background (same as '&_=h')",
|
|
"=", ":host:port cmd", "run 'cmd' command on remote server",
|
|
"\nexamples:","","",
|
|
"=+", "tcp://localhost:9090/", "connect to: r2 -c.:9090 ./bin",
|
|
// "=+", "udp://localhost:9090/", "connect to: r2 -c.:9090 ./bin",
|
|
"=+", "rap://localhost:9090/", "connect to: r2 rap://:9090",
|
|
"=+", "http://localhost:9090/cmd/", "connect to: r2 -c'=h 9090' bin",
|
|
"o ", "rap://:9090/", "start the rap server on tcp port 9090",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_equalh[] = {
|
|
"Usage:", " =[hH] [...]", " # http server",
|
|
"http server:", "", "",
|
|
"=h", " port", "listen for http connections (r2 -qc=H /bin/ls)",
|
|
"=h-", "", "stop background webserver",
|
|
"=h--", "", "stop foreground webserver",
|
|
"=h*", "", "restart current webserver",
|
|
"=h&", " port", "start http server in background",
|
|
"=H", " port", "launch browser and listen for http",
|
|
"=H&", " port", "launch browser and listen for http in background",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_equalg[] = {
|
|
"Usage:", " =[g] [...]", " # gdb server",
|
|
"gdbserver:", "", "",
|
|
"=g", " port file [args]", "listen on 'port' debugging 'file' using gdbserver",
|
|
"=g!", " port file [args]", "same as above, but debug protocol messages (like gdbserver --remote-debug)",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_b[] = {
|
|
"Usage:", "b[f] [arg]\n", "Get/Set block size",
|
|
"b", " 33", "set block size to 33",
|
|
"b", " eip+4", "numeric argument can be an expression",
|
|
"b", "", "display current block size",
|
|
"b", "+3", "increase blocksize by 3",
|
|
"b", "-16", "decrease blocksize by 16",
|
|
"b*", "", "display current block size in r2 command",
|
|
"bf", " foo", "set block size to flag size",
|
|
"bj", "", "display block size information in JSON",
|
|
"bm", " 1M", "set max block size",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_k[] = {
|
|
"Usage:", "k[s] [key[=value]]", "Sdb Query",
|
|
"k", " anal/**", "list namespaces under anal",
|
|
"k", " anal/meta/*", "list kv from anal > meta namespaces",
|
|
"k", " anal/meta/meta.0x80404", "get value for meta.0x80404 key",
|
|
"k", " foo", "show value",
|
|
"k", " foo=bar", "set value",
|
|
"k", "", "list keys",
|
|
"kd", " [file.sdb] [ns]", "dump namespace to disk",
|
|
"kj", "", "List all namespaces and sdb databases in JSON format",
|
|
"ko", " [file.sdb] [ns]", "open file into namespace",
|
|
"ks", " [ns]", "enter the sdb query shell",
|
|
//"kl", " ha.sdb", "load keyvalue from ha.sdb",
|
|
//"ks", " ha.sdb", "save keyvalue to ha.sdb",
|
|
NULL,
|
|
};
|
|
|
|
static const char *help_msg_r[] = {
|
|
"Usage:", "r[+-][ size]", "Resize file",
|
|
"r", "", "display file size",
|
|
"rj", "", "display the file size in JSON format",
|
|
"r", " size", "expand or truncate file to given size",
|
|
"r-", "num", "remove num bytes, move following data down",
|
|
"r+", "num", "insert num bytes, move following data up",
|
|
"rb", "oldbase @ newbase", "rebase all flags, bin.info, breakpoints and analysis",
|
|
"rm" ," [file]", "remove file",
|
|
"rh" ,"", "show size in human format",
|
|
"r2" ," [file]", "launch r2 (same for rax2, rasm2, ...)",
|
|
"reset" ,"", "reset console settings (clear --hard)",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_u[] = {
|
|
"Usage:", "u", "uname or undo write/seek",
|
|
"u", "", "show system uname",
|
|
"uw", "", "alias for wc (requires: e io.cache=true)",
|
|
"us", "", "alias for s- (seek history)",
|
|
"uc", "[?]", "undo core commands (uc?, ucl, uc*, ..)",
|
|
"uid", "", "display numeric user id",
|
|
"uniq", "", "filter rows to avoid duplicates",
|
|
"uname", "", "uname - show system information",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_uc[] = {
|
|
"Usage:", "uc [cmd] [revert-cmd]", "undo core commands",
|
|
"uc", "", "list all core undos",
|
|
"uc*", "", "list all core undos as r2 commands",
|
|
"uc-", "", "undo last action",
|
|
"uc.", "", "list all reverts in current",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_y[] = {
|
|
"Usage:", "y[ptxy] [len] [[@]addr]", " # See wd? for memcpy, same as 'yf'.",
|
|
"y!", "", "open cfg.editor to edit the clipboard",
|
|
"y", " 16 @ 0x200", "copy 16 bytes into clipboard from 0x200",
|
|
"y", " 16 0x200", "copy 16 bytes into clipboard from 0x200",
|
|
"y", " 16", "copy 16 bytes into clipboard",
|
|
"y", "", "show yank buffer information (origin len bytes)",
|
|
"y*", "", "print in r2 commands what's been yanked",
|
|
"yf", " 64 0x200", "copy file 64 bytes from 0x200 from file",
|
|
"yfa", " file copy", "copy all bytes from file (opens w/ io)",
|
|
"yfx", " 10203040", "yank from hexpairs (same as ywx)",
|
|
"yj", "", "print in JSON commands what's been yanked",
|
|
"yp", "", "print contents of clipboard",
|
|
"yq", "", "print contents of clipboard in hexpairs",
|
|
"ys", "", "print contents of clipboard as string",
|
|
"yt", " 64 0x200", "copy 64 bytes from current seek to 0x200",
|
|
"ytf", " file", "dump the clipboard to given file",
|
|
"yw", " hello world", "yank from string",
|
|
"ywx", " 10203040", "yank from hexpairs (same as yfx)",
|
|
"yx", "", "print contents of clipboard in hexadecimal",
|
|
"yy", " @ 0x3344", "paste contents of clipboard to 0x3344",
|
|
"yy", " 0x3344", "paste contents of clipboard to 0x3344",
|
|
"yy", "", "paste contents of clipboard at current seek",
|
|
"yz", " [len]", "copy nul-terminated string (up to blocksize) into clipboard",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_triple_exclamation[] = {
|
|
"Usage:", "!!![-*][cmd] [arg|$type...]", " # user-defined autocompletion for commands",
|
|
"!!!", "", "list all autocompletions",
|
|
"!!!?", "", "show this help",
|
|
"!!!", "-*", "remove all user-defined autocompletions",
|
|
"!!!", "-\\*", "remove autocompletions matching this glob expression",
|
|
"!!!", "-foo", "remove autocompletion named 'foo'",
|
|
"!!!", "foo", "add 'foo' for autocompletion",
|
|
"!!!", "bar $flag", "add 'bar' for autocompletion with $flag as argument",
|
|
"Types:", "", "",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_vertical_bar[] = {
|
|
"Usage:", "[cmd] | [program|H|T|.|]", "",
|
|
"", "[cmd] |?", "show this help",
|
|
"", "[cmd] |", "disable scr.html and scr.color",
|
|
"", "[cmd] |H", "enable scr.html, respect scr.color",
|
|
"", "[cmd] |T", "use scr.tts to speak out the stdout",
|
|
"", "[cmd] | [program]", "pipe output of command to program",
|
|
"", "[cmd] |.", "alias for .[cmd]",
|
|
NULL
|
|
};
|
|
|
|
static const char *help_msg_v[] = {
|
|
"Usage:", "v[*i]", "",
|
|
"v", "", "open visual panels",
|
|
"v", " test", "load saved layout with name test",
|
|
"v=", " test", "save current layout with name test",
|
|
"vi", " test", "open the file test in 'cfg.editor'",
|
|
NULL
|
|
};
|
|
|
|
R_API void r_core_cmd_help(const RCore *core, const char *help[]) {
|
|
r_cons_cmd_help (help, core->print->flags & R_PRINT_FLAGS_COLOR);
|
|
}
|
|
|
|
struct duplicate_flag_t {
|
|
RList *ret;
|
|
const char *word;
|
|
};
|
|
|
|
static bool duplicate_flag(RFlagItem *flag, void *u) {
|
|
struct duplicate_flag_t *user = (struct duplicate_flag_t *)u;
|
|
/* filter per flag spaces */
|
|
if (r_str_glob (flag->name, user->word)) {
|
|
RFlagItem *cloned_item = r_flag_item_clone (flag);
|
|
if (!cloned_item) {
|
|
return false;
|
|
}
|
|
r_list_append (user->ret, cloned_item);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool foreach_newline(RCore *core) {
|
|
bool nl = r_config_get_i (core->config, "scr.loopnl");
|
|
if (nl) {
|
|
r_cons_newline ();
|
|
}
|
|
return r_cons_is_breaked ();
|
|
}
|
|
|
|
static void recursive_help_go(RCore *core, int detail, RCmdDescriptor *desc) {
|
|
int i;
|
|
if (desc->help_msg) {
|
|
r_core_cmd_help (core, desc->help_msg);
|
|
}
|
|
if (detail >= 1) {
|
|
if (desc->help_detail) {
|
|
r_core_cmd_help (core, desc->help_detail);
|
|
}
|
|
if (detail >= 2 && desc->help_detail2) {
|
|
r_core_cmd_help (core, desc->help_detail2);
|
|
}
|
|
}
|
|
for (i = 32; i < R_ARRAY_SIZE (desc->sub); i++) {
|
|
if (desc->sub[i]) {
|
|
recursive_help_go (core, detail, desc->sub[i]);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void recursive_help(RCore *core, int detail, const char *cmd_prefix) {
|
|
const ut8 *p;
|
|
RCmdDescriptor *desc = &core->root_cmd_descriptor;
|
|
for (p = (const ut8 *)cmd_prefix; *p && *p < R_ARRAY_SIZE (desc->sub); p++) {
|
|
if (!(desc = desc->sub[*p])) {
|
|
return;
|
|
}
|
|
}
|
|
recursive_help_go (core, detail, desc);
|
|
}
|
|
|
|
static bool lastcmd_repeat(RCore *core, int next) {
|
|
int res = -1;
|
|
// Fix for backtickbug px`~`
|
|
if (!core->lastcmd || core->cons->context->cmd_depth < 1) {
|
|
return false;
|
|
}
|
|
switch (*core->lastcmd) {
|
|
case '.':
|
|
if (core->lastcmd[1] == '(') { // macro call
|
|
res = r_core_cmd0 (core, core->lastcmd);
|
|
}
|
|
break;
|
|
case 'd': // debug
|
|
res = r_core_cmd0 (core, core->lastcmd);
|
|
switch (core->lastcmd[1]) {
|
|
case 's':
|
|
case 'c':
|
|
r_core_cmd0 (core, "sr PC;pd 1");
|
|
}
|
|
break;
|
|
case 'p': // print
|
|
case 'x':
|
|
case '$':
|
|
if (!strncmp (core->lastcmd, "pd", 2)) {
|
|
if (core->lastcmd[2]== ' ') {
|
|
r_core_cmdf (core, "so %s", core->lastcmd + 3);
|
|
} else {
|
|
r_core_cmd0 (core, "so `pi~?`");
|
|
}
|
|
} else {
|
|
if (next) {
|
|
r_core_seek (core, core->offset + core->blocksize, true);
|
|
} else {
|
|
if (core->blocksize > core->offset) {
|
|
r_core_seek (core, 0, true);
|
|
} else {
|
|
r_core_seek (core, core->offset - core->blocksize, true);
|
|
}
|
|
}
|
|
}
|
|
res = r_core_cmd0 (core, core->lastcmd);
|
|
break;
|
|
}
|
|
return res != -1;
|
|
}
|
|
|
|
static int r_core_cmd_nullcallback(void *data) {
|
|
RCore *core = (RCore*) data;
|
|
if (core->cons->context->breaked) {
|
|
core->cons->context->breaked = false;
|
|
return 0;
|
|
}
|
|
if (!core->cmdrepeat) {
|
|
return 0;
|
|
}
|
|
lastcmd_repeat (core, true);
|
|
return 1;
|
|
}
|
|
|
|
static int cmd_uname(void *data, const char *input) { // "uniq"
|
|
RSysInfo *si = r_sys_info();
|
|
if (si) {
|
|
r_cons_printf ("%s", si->sysname);
|
|
if (strstr (input, "-r")) {
|
|
r_cons_printf (" %s", si->release);
|
|
}
|
|
r_cons_newline ();
|
|
r_sys_info_free (si);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_uniq(void *data, const char *input) { // "uniq"
|
|
RCore *core = (RCore *)data;
|
|
const char *arg = strchr (input, ' ');
|
|
if (arg) {
|
|
arg = r_str_trim_head_ro (arg + 1);
|
|
}
|
|
switch (*input) {
|
|
case '?': // "uniq?"
|
|
eprintf ("Usage: uniq # uniq to list unique strings in file\n");
|
|
break;
|
|
default: // "uniq"
|
|
if (!arg) {
|
|
arg = "";
|
|
}
|
|
if (r_fs_check (core->fs, arg)) {
|
|
r_core_cmdf (core, "md %s", arg);
|
|
} else {
|
|
char *res = r_syscmd_uniq (arg);
|
|
if (res) {
|
|
r_cons_print (res);
|
|
free (res);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_head (void *data, const char *_input) { // "head"
|
|
RCore *core = (RCore *)data;
|
|
int lines = 5;
|
|
char *input = strdup (_input);
|
|
char *arg = strchr (input, ' ');
|
|
char *tmp, *count;
|
|
if (arg) {
|
|
arg = (char *)r_str_trim_head_ro (arg + 1); // contains "count filename"
|
|
count = strchr (arg, ' ');
|
|
if (count) {
|
|
*count = 0; // split the count and file name
|
|
tmp = (char *)r_str_trim_head_ro (count + 1);
|
|
lines = atoi (arg);
|
|
arg = tmp;
|
|
}
|
|
}
|
|
switch (*input) {
|
|
case '?': // "head?"
|
|
eprintf ("Usage: head [file] # to list first n lines in file\n");
|
|
break;
|
|
default: // "head"
|
|
if (!arg) {
|
|
arg = "";
|
|
}
|
|
if (r_fs_check (core->fs, arg)) {
|
|
r_core_cmdf (core, "md %s", arg);
|
|
} else {
|
|
char *res = r_syscmd_head (arg, lines);
|
|
if (res) {
|
|
r_cons_print (res);
|
|
free (res);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
free (input);
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_undo(void *data, const char *input) {
|
|
RCore *core = (RCore *)data;
|
|
switch (input[0]) {
|
|
case 'c': // "uc"
|
|
switch (input[1]) {
|
|
case ' ': {
|
|
char *cmd = strdup (input + 2);
|
|
char *rcmd = strchr (cmd, ',');
|
|
if (rcmd) {
|
|
*rcmd++ = 0;
|
|
RCoreUndo *undo = r_core_undo_new (core->offset, cmd, rcmd);
|
|
r_core_undo_push (core, undo);
|
|
} else {
|
|
eprintf ("Usage: uc [cmd] [revert-cmd]");
|
|
}
|
|
free (cmd);
|
|
}
|
|
break;
|
|
case '?': // "uc?"
|
|
r_core_cmd_help (core, help_msg_uc);
|
|
break;
|
|
case '.': { // "uc."
|
|
RCoreUndoCondition cond = {
|
|
.addr = core->offset,
|
|
.minstamp = 0,
|
|
.glob = NULL
|
|
};
|
|
r_core_undo_print (core, 1, &cond);
|
|
break;
|
|
}
|
|
case '*': // "uc*"
|
|
r_core_undo_print (core, 1, NULL);
|
|
break;
|
|
case '-': // "uc-"
|
|
r_core_undo_pop (core);
|
|
break;
|
|
default:
|
|
r_core_undo_print (core, 0, NULL);
|
|
break;
|
|
}
|
|
return 1;
|
|
case 'i': // "ui"
|
|
r_cons_printf ("%d\n", r_sys_uid ());
|
|
return 1;
|
|
case 's': // "us"
|
|
r_core_cmdf (data, "s-%s", input + 1);
|
|
return 1;
|
|
case 'w': // "uw"
|
|
r_core_cmdf (data, "wc%s", input + 1);
|
|
return 1;
|
|
case 'n': // "un"
|
|
if (input[1] == 'a') { // "uname"
|
|
(void)cmd_uname (core, input);
|
|
} else if (input[1] == 'i' && input[2] == 'q') {
|
|
(void)cmd_uniq (core, input);
|
|
}
|
|
return 1;
|
|
default:
|
|
case '?': // "u?"
|
|
r_core_cmd_help (data, help_msg_u);
|
|
return 1;
|
|
}
|
|
#if __UNIX__
|
|
struct utsname un;
|
|
uname (&un);
|
|
r_cons_printf ("%s %s %s %s\n", un.sysname,
|
|
un.nodename, un.release, un.machine);
|
|
#elif __WINDOWS__
|
|
r_cons_printf ("windows\n");
|
|
#else
|
|
r_cons_printf ("unknown\n");
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_alias(void *data, const char *input) {
|
|
RCore *core = (RCore *)data;
|
|
if (*input == '?') {
|
|
r_core_cmd_help (core, help_msg_dollar);
|
|
return 0;
|
|
}
|
|
int i = strlen (input);
|
|
char *buf = malloc (i + 2);
|
|
if (!buf) {
|
|
return 0;
|
|
}
|
|
*buf = '$'; // prefix aliases with a dollar
|
|
memcpy (buf + 1, input, i + 1);
|
|
char *q = strchr (buf, ' ');
|
|
char *def = strchr (buf, '=');
|
|
char *desc = strchr (buf, '?');
|
|
char *nonl = strchr (buf, 'n');
|
|
|
|
int defmode = 0;
|
|
if (def && def > buf) {
|
|
char *prev = def - 1;
|
|
switch (*prev) {
|
|
case ':':
|
|
defmode = *prev;
|
|
*prev = 0;
|
|
break;
|
|
case '+':
|
|
defmode = *prev;
|
|
*prev = 0;
|
|
break;
|
|
case '-':
|
|
defmode = *prev;
|
|
*prev = 0;
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* create alias */
|
|
if ((def && q && (def < q)) || (def && !q)) {
|
|
*def++ = 0;
|
|
size_t len = strlen (def);
|
|
if (defmode) {
|
|
ut64 at = r_num_math (core->num, def);
|
|
switch (defmode) {
|
|
case ':':
|
|
r_flag_set (core->flags, buf + 1, at, 1);
|
|
return 1;
|
|
case '+':
|
|
at = r_num_get (core->num, buf + 1) + at;
|
|
r_flag_set (core->flags, buf + 1, at, 1);
|
|
return 1;
|
|
case '-':
|
|
at = r_num_get (core->num, buf + 1) - at;
|
|
r_flag_set (core->flags, buf + 1, at, 1);
|
|
return 1;
|
|
}
|
|
}
|
|
/* Remove quotes */
|
|
if (len > 0 && (def[0] == '\'') && (def[len - 1] == '\'')) {
|
|
def[len - 1] = 0x00;
|
|
def++;
|
|
}
|
|
if (!q || (q && q > def)) {
|
|
if (*def) {
|
|
if (!strcmp (def, "-")) {
|
|
const char *v = r_cmd_alias_get (core->rcmd, buf, 0);
|
|
char *n = r_cons_editor (NULL, v);
|
|
if (n) {
|
|
r_cmd_alias_set (core->rcmd, buf, n, 0);
|
|
free (n);
|
|
}
|
|
} else {
|
|
r_cmd_alias_set (core->rcmd, buf, def, 0);
|
|
}
|
|
} else {
|
|
r_cmd_alias_del (core->rcmd, buf);
|
|
}
|
|
}
|
|
/* Show command for alias */
|
|
} else if (desc && !q) {
|
|
*desc = 0;
|
|
const char *v = r_cmd_alias_get (core->rcmd, buf, 0);
|
|
if (v) {
|
|
if (nonl == desc + 1) {
|
|
r_cons_print (v);
|
|
} else {
|
|
r_cons_println (v);
|
|
}
|
|
free (buf);
|
|
return 1;
|
|
} else {
|
|
eprintf ("unknown key '%s'\n", buf);
|
|
}
|
|
} else if (buf[1] == '*') {
|
|
/* Show aliases */
|
|
int i, count = 0;
|
|
char **keys = r_cmd_alias_keys (core->rcmd, &count);
|
|
for (i = 0; i < count; i++) {
|
|
const char *v = r_cmd_alias_get (core->rcmd, keys[i], 0);
|
|
char *q = r_base64_encode_dyn (v, -1);
|
|
if (buf[2] == '*') {
|
|
r_cons_printf ("%s=%s\n", keys[i], v);
|
|
} else {
|
|
r_cons_printf ("%s=base64:%s\n", keys[i], q);
|
|
}
|
|
free (q);
|
|
}
|
|
} else if (!buf[1]) {
|
|
int i, count = 0;
|
|
char **keys = r_cmd_alias_keys (core->rcmd, &count);
|
|
for (i = 0; i < count; i++) {
|
|
r_cons_println (keys[i]);
|
|
}
|
|
} else {
|
|
/* Execute alias */
|
|
if (q) {
|
|
*q = 0;
|
|
}
|
|
const char *v = r_cmd_alias_get (core->rcmd, buf, 0);
|
|
if (v) {
|
|
if (*v == '$') {
|
|
r_cons_strcat (v + 1);
|
|
r_cons_newline ();
|
|
} else if (q) {
|
|
char *out = r_str_newf ("%s %s", v, q + 1);
|
|
r_core_cmd0 (core, out);
|
|
free (out);
|
|
} else {
|
|
r_core_cmd0 (core, v);
|
|
}
|
|
} else {
|
|
ut64 at = r_num_get (core->num, buf + 1);
|
|
if (at != UT64_MAX) {
|
|
r_core_seek (core, at, true);
|
|
} else {
|
|
eprintf ("Unknown alias '%s'\n", buf + 1);
|
|
}
|
|
}
|
|
}
|
|
free (buf);
|
|
return 0;
|
|
}
|
|
|
|
static int getArg(char ch, int def) {
|
|
switch (ch) {
|
|
case '&':
|
|
case '-':
|
|
return ch;
|
|
}
|
|
return def;
|
|
}
|
|
|
|
// wtf dupe for local vs remote?
|
|
static void aliascmd(RCore *core, const char *str) {
|
|
switch (str[0]) {
|
|
case '\0': // "=$"
|
|
r_core_cmd0 (core, "$");
|
|
break;
|
|
case '-': // "=$-"
|
|
if (str[1]) {
|
|
r_cmd_alias_del (core->rcmd, str + 2);
|
|
} else {
|
|
r_cmd_alias_del (core->rcmd, NULL);
|
|
// r_cmd_alias_reset (core->rcmd);
|
|
}
|
|
break;
|
|
case '?': // "=$?"
|
|
eprintf ("Usage: =$[-][remotecmd] # remote command alias\n");
|
|
eprintf (" =$dr # makes 'dr' alias for =!dr\n");
|
|
eprintf (" =$-dr # unset 'dr' alias\n");
|
|
break;
|
|
default:
|
|
r_cmd_alias_set (core->rcmd, str, "", 1);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void cmd_remote(RCore *core, const char *input, bool retry) {
|
|
if (!*input) {
|
|
return;
|
|
}
|
|
if (*input == '?') {
|
|
r_cons_printf ("Usage: =r localhost:9999\n");
|
|
return;
|
|
}
|
|
char *host = strdup (input);
|
|
char *port = strchr (host, ':');
|
|
if (port) {
|
|
*port++ = 0;
|
|
}
|
|
RSocket *s = r_socket_new (false);
|
|
repeat:
|
|
if (r_socket_connect (s, host, port, R_SOCKET_PROTO_TCP, 1500)) {
|
|
char buf[1024];
|
|
void *bed = r_cons_sleep_begin ();
|
|
r_cons_break_push (NULL, NULL);
|
|
for (;;) {
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
r_socket_printf (s, "[0x%08"PFMT64x"]> ", core->offset);
|
|
r_socket_flush (s);
|
|
memset (buf, 0, sizeof (buf));
|
|
r_socket_block_time (s, true, 99999, 0);
|
|
if (r_socket_read (s, (ut8*)buf, sizeof (buf) - 1) < 1) {
|
|
break;
|
|
}
|
|
if (*buf == 'q') {
|
|
break;
|
|
}
|
|
const bool orig = r_config_get_b (core->config, "scr.interactive");
|
|
r_config_set_b (core->config, "scr.interactive", false);
|
|
char *res = r_core_cmd_str (core, buf);
|
|
r_config_set_b (core->config, "scr.interactive", orig);
|
|
r_socket_printf (s, "%s\n", res);
|
|
r_socket_flush (s);
|
|
free (res);
|
|
}
|
|
r_cons_break_pop ();
|
|
r_cons_sleep_end (bed);
|
|
} else {
|
|
if (retry) {
|
|
r_sys_sleep (1);
|
|
goto repeat;
|
|
}
|
|
}
|
|
r_socket_close (s);
|
|
r_socket_free (s);
|
|
free (host);
|
|
}
|
|
|
|
static int cmd_rap(void *data, const char *input) {
|
|
RCore *core = (RCore *)data;
|
|
switch (*input) {
|
|
case '\0': // "="
|
|
r_core_rtr_list (core);
|
|
break;
|
|
case 'r': // "=r"
|
|
cmd_remote (core, r_str_trim_head_ro (input + 1), false);
|
|
break;
|
|
case 'R': // "=R"
|
|
cmd_remote (core, r_str_trim_head_ro (input + 1), true);
|
|
break;
|
|
case 'j': // "=j"
|
|
eprintf ("TODO: list connections in json\n");
|
|
break;
|
|
case '!': // "=!"
|
|
if (input[1] == 'q') {
|
|
R_FREE (core->cmdremote);
|
|
} else if (input[1] == '=') { // =!=0 or =!= for iosystem
|
|
R_FREE (core->cmdremote);
|
|
core->cmdremote = r_str_trim_dup (input + 2);
|
|
} else {
|
|
char *res = r_io_system (core->io, input + 1);
|
|
if (res) {
|
|
r_cons_printf ("%s\n", res);
|
|
free (res);
|
|
}
|
|
}
|
|
break;
|
|
case '$': // "=$"
|
|
// XXX deprecate?
|
|
aliascmd (core, input + 1);
|
|
break;
|
|
case '+': // "=+"
|
|
r_core_rtr_add (core, input + 1);
|
|
break;
|
|
case '-': // "=-"
|
|
r_core_rtr_remove (core, input + 1);
|
|
break;
|
|
//case ':': r_core_rtr_cmds (core, input + 1); break;
|
|
case '<': // "=<"
|
|
r_core_rtr_pushout (core, input + 1);
|
|
break;
|
|
case '=': // "=="
|
|
r_core_rtr_session (core, input + 1);
|
|
break;
|
|
case 'g': // "=g"
|
|
if (input[1] == '?') {
|
|
r_core_cmd_help (core, help_msg_equalg);
|
|
} else {
|
|
r_core_rtr_gdb (core, getArg (input[1], 'g'), input + 1);
|
|
}
|
|
break;
|
|
case 'h': // "=h"
|
|
if (input[1] == '?') {
|
|
r_core_cmd_help (core, help_msg_equalh);
|
|
} else {
|
|
r_core_rtr_http (core, getArg (input[1], 'h'), 'h', input + 1);
|
|
}
|
|
break;
|
|
case 'H': // "=H"
|
|
if (input[1] == '?') {
|
|
r_core_cmd_help (core, help_msg_equalh);
|
|
} else {
|
|
const char *arg = r_str_trim_head_ro (input + 1);
|
|
r_core_rtr_http (core, getArg (input[1], 'H'), 'H', arg);
|
|
}
|
|
break;
|
|
case '?': // "=?"
|
|
r_core_cmd_help (core, help_msg_equal);
|
|
break;
|
|
default:
|
|
r_core_rtr_cmd (core, input);
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_rap_run(void *data, const char *input) {
|
|
RCore *core = (RCore *)data;
|
|
char *res = r_io_system (core->io, input);
|
|
if (res) {
|
|
int ret = atoi (res);
|
|
free (res);
|
|
return ret;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static int cmd_yank(void *data, const char *input) {
|
|
ut64 n;
|
|
RCore *core = (RCore *)data;
|
|
switch (input[0]) {
|
|
case ' ': // "y "
|
|
r_core_yank (core, core->offset, r_num_math (core->num, input + 1));
|
|
break;
|
|
case 'l': // "yl"
|
|
core->num->value = r_buf_size (core->yank_buf);
|
|
break;
|
|
case 'y': // "yy"
|
|
while (input[1] == ' ') {
|
|
input++;
|
|
}
|
|
n = input[1]? r_num_math (core->num, input + 1): core->offset;
|
|
r_core_yank_paste (core, n, 0);
|
|
break;
|
|
case 'x': // "yx"
|
|
r_core_yank_hexdump (core, r_num_math (core->num, input + 1));
|
|
break;
|
|
case 'z': // "yz"
|
|
r_core_yank_string (core, core->offset, r_num_math (core->num, input + 1));
|
|
break;
|
|
case 'w': // "yw" ... we have yf which makes more sense than 'w'
|
|
switch (input[1]) {
|
|
case ' ':
|
|
r_core_yank_set (core, 0, (const ut8*)input + 2, strlen (input + 2));
|
|
break;
|
|
case 'x':
|
|
if (input[2] == ' ') {
|
|
char *out = strdup (input + 3);
|
|
int len = r_hex_str2bin (input + 3, (ut8*)out);
|
|
if (len > 0) {
|
|
r_core_yank_set (core, core->offset, (const ut8*)out, len);
|
|
} else {
|
|
eprintf ("Invalid length\n");
|
|
}
|
|
free (out);
|
|
} else {
|
|
eprintf ("Usage: ywx [hexpairs]\n");
|
|
}
|
|
// r_core_yank_write_hex (core, input + 2);
|
|
break;
|
|
default:
|
|
eprintf ("Usage: ywx [hexpairs]\n");
|
|
break;
|
|
}
|
|
break;
|
|
case 'p': // "yp"
|
|
r_core_yank_cat (core, r_num_math (core->num, input + 1));
|
|
break;
|
|
case 's': // "ys"
|
|
r_core_yank_cat_string (core, r_num_math (core->num, input + 1));
|
|
break;
|
|
case 't': // "wt"
|
|
if (input[1] == 'f') { // "wtf"
|
|
ut64 tmpsz;
|
|
const char *file = r_str_trim_head_ro (input + 2);
|
|
const ut8 *tmp = r_buf_data (core->yank_buf, &tmpsz);
|
|
if (!r_file_dump (file, tmp, tmpsz, false)) {
|
|
eprintf ("Cannot dump to '%s'\n", file);
|
|
}
|
|
} else if (input[1] == ' ') {
|
|
r_core_yank_to (core, input + 1);
|
|
} else {
|
|
eprintf ("Usage: wt[f] [arg] ..\n");
|
|
}
|
|
break;
|
|
case 'f': // "yf"
|
|
switch (input[1]) {
|
|
case ' ': // "yf"
|
|
r_core_yank_file_ex (core, input + 1);
|
|
break;
|
|
case 'x': // "yfx"
|
|
r_core_yank_hexpair (core, input + 2);
|
|
break;
|
|
case 'a': // "yfa"
|
|
r_core_yank_file_all (core, input + 2);
|
|
break;
|
|
default:
|
|
eprintf ("Usage: yf[xa] [arg]\n");
|
|
eprintf ("yf [file] - copy blocksize from file into the clipboard\n");
|
|
eprintf ("yfa [path] - yank the whole file\n");
|
|
eprintf ("yfx [hexpair] - yank from hexpair string\n");
|
|
break;
|
|
}
|
|
break;
|
|
case '!': // "y!"
|
|
{
|
|
char *sig = r_core_cmd_str (core, "y*");
|
|
if (!sig || !*sig) {
|
|
free (sig);
|
|
sig = strdup ("wx 10203040");
|
|
}
|
|
char *data = r_core_editor (core, NULL, sig);
|
|
(void) strtok (data, ";\n");
|
|
r_core_cmdf (core, "y%s", data);
|
|
free (sig);
|
|
free (data);
|
|
}
|
|
break;
|
|
case '*': // "y*"
|
|
case 'j': // "yj"
|
|
case 'q': // "yq"
|
|
case '\0': // "y"
|
|
r_core_yank_dump (core, 0, input[0]);
|
|
break;
|
|
default:
|
|
r_core_cmd_help (core, help_msg_y);
|
|
break;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static int lang_run_file(RCore *core, RLang *lang, const char *file) {
|
|
r_core_sysenv_begin (core, NULL);
|
|
return r_lang_run_file (core->lang, file);
|
|
}
|
|
|
|
static char *langFromHashbang(RCore *core, const char *file) {
|
|
int fd = r_sandbox_open (file, O_RDONLY, 0);
|
|
if (fd != -1) {
|
|
char firstLine[128] = {0};
|
|
int len = r_sandbox_read (fd, (ut8*)firstLine, sizeof (firstLine) - 1);
|
|
firstLine[len] = 0;
|
|
if (!strncmp (firstLine, "#!/", 3)) {
|
|
// I CAN HAS A HASHBANG
|
|
char *nl = strchr (firstLine, '\n');
|
|
if (nl) {
|
|
*nl = 0;
|
|
}
|
|
nl = strchr (firstLine, ' ');
|
|
if (nl) {
|
|
*nl = 0;
|
|
}
|
|
nl = strdup (firstLine + 2);
|
|
r_sandbox_close (fd);
|
|
return nl;
|
|
}
|
|
r_sandbox_close (fd);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
R_API bool r_core_run_script(RCore *core, const char *file) {
|
|
bool ret = false;
|
|
RListIter *iter;
|
|
RLangPlugin *p;
|
|
char *name;
|
|
|
|
r_list_foreach (core->scriptstack, iter, name) {
|
|
if (!strcmp (file, name)) {
|
|
eprintf ("Warning: ignored nested source: %s\n", file);
|
|
return false;
|
|
}
|
|
}
|
|
r_list_push (core->scriptstack, strdup (file));
|
|
|
|
if (!strcmp (file, "-")) {
|
|
char *out = r_core_editor (core, NULL, NULL);
|
|
if (out) {
|
|
ret = r_core_cmd_lines (core, out);
|
|
free (out);
|
|
}
|
|
} else if (r_str_endswith (file, ".html")) {
|
|
const bool httpSandbox = r_config_get_i (core->config, "http.sandbox");
|
|
char *httpIndex = strdup (r_config_get (core->config, "http.index"));
|
|
r_config_set_b (core->config, "http.sandbox", false);
|
|
char *absfile = r_file_abspath (file);
|
|
r_config_set (core->config, "http.index", absfile);
|
|
free (absfile);
|
|
r_core_cmdf (core, "=H");
|
|
r_config_set_b (core->config, "http.sandbox", httpSandbox);
|
|
r_config_set (core->config, "http.index", httpIndex);
|
|
free (httpIndex);
|
|
ret = true;
|
|
} else if (r_str_endswith (file, ".c")) {
|
|
r_core_cmd_strf (core, "#!c %s", file);
|
|
ret = true;
|
|
} else if (r_file_is_c (file)) {
|
|
const char *dir = r_config_get (core->config, "dir.types");
|
|
char *out = r_parse_c_file (core->anal, file, dir, NULL);
|
|
if (out) {
|
|
r_cons_strcat (out);
|
|
sdb_query_lines (core->anal->sdb_types, out);
|
|
free (out);
|
|
}
|
|
ret = out != NULL;
|
|
} else {
|
|
p = r_lang_get_by_extension (core->lang, file);
|
|
if (p) {
|
|
r_lang_use (core->lang, p->name);
|
|
ret = lang_run_file (core, core->lang, file);
|
|
} else {
|
|
// XXX this is an ugly hack, we need to use execve here and specify args properly
|
|
#if __WINDOWS__
|
|
#define cmdstr(x) r_str_newf (x" %s", file);
|
|
#else
|
|
#define cmdstr(x) r_str_newf (x" '%s'", file);
|
|
#endif
|
|
const char *p = r_str_lchr (file, '.');
|
|
if (p) {
|
|
const char *ext = p + 1;
|
|
/* TODO: handle this inside r_lang_pipe with new APIs */
|
|
if (!strcmp (ext, "js")) {
|
|
char *cmd = cmdstr ("node");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "exe")) {
|
|
#if __WINDOWS__
|
|
char *cmd = r_str_newf ("%s", file);
|
|
#else
|
|
char *cmd = cmdstr ("wine");
|
|
#endif
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "zig")) {
|
|
char *cmd = cmdstr ("zig run");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "d")) {
|
|
char *cmd = cmdstr ("dmd -run");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "lsp")) {
|
|
char *cmd = cmdstr ("newlisp -n");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "go")) {
|
|
char *cmd = cmdstr ("go run");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "es6")) {
|
|
char *cmd = cmdstr ("babel-node");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "rb")) {
|
|
char *cmd = cmdstr ("ruby");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "vala")) {
|
|
r_lang_use (core->lang, "vala");
|
|
lang_run_file (core, core->lang, file);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "sh")) {
|
|
char *shell = r_sys_getenv ("SHELL");
|
|
if (!shell) {
|
|
shell = strdup ("sh");
|
|
}
|
|
if (shell) {
|
|
r_lang_use (core->lang, "pipe");
|
|
char *cmd = r_str_newf ("%s '%s'", shell, file);
|
|
if (cmd) {
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
}
|
|
free (shell);
|
|
}
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "pl")) {
|
|
char *cmd = cmdstr ("perl");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
} else if (!strcmp (ext, "py")) {
|
|
char *cmd = cmdstr ("python");
|
|
r_lang_use (core->lang, "pipe");
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (cmd);
|
|
ret = 1;
|
|
}
|
|
} else {
|
|
char *abspath = r_file_path (file);
|
|
char *lang = langFromHashbang (core, file);
|
|
if (lang) {
|
|
r_lang_use (core->lang, "pipe");
|
|
char *cmd = r_str_newf ("%s '%s'", lang, file);
|
|
lang_run_file (core, core->lang, cmd);
|
|
free (lang);
|
|
free (cmd);
|
|
ret = 1;
|
|
}
|
|
free (abspath);
|
|
}
|
|
if (!ret) {
|
|
ret = r_core_cmd_file (core, file);
|
|
}
|
|
}
|
|
}
|
|
free (r_list_pop (core->scriptstack));
|
|
return ret;
|
|
}
|
|
|
|
static int cmd_lsr(RCore *core, const char *input) {
|
|
const char *arg;
|
|
char *path;
|
|
RList *files;
|
|
RListIter *iter;
|
|
if (R_STR_ISEMPTY (input)) {
|
|
arg = ".";
|
|
} else {
|
|
arg = input;
|
|
}
|
|
files = r_file_lsrf (arg);
|
|
if (!files) {
|
|
eprintf ("Failed to read directories\n");
|
|
return 0;
|
|
}
|
|
r_list_foreach (files, iter, path) {
|
|
r_cons_println (path);
|
|
}
|
|
r_list_free (files);
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_ls(void *data, const char *input) { // "ls"
|
|
RCore *core = (RCore *)data;
|
|
const char *arg = strchr (input, ' ');
|
|
if (arg) {
|
|
arg = r_str_trim_head_ro (arg + 1);
|
|
}
|
|
switch (*input) {
|
|
case 'r':
|
|
cmd_lsr (core, arg);
|
|
break;
|
|
case '?': // "l?"
|
|
eprintf ("Usage: l[es] # ls to list files, le[ss] to less a file\n");
|
|
break;
|
|
case 'e': // "le"
|
|
if (arg) {
|
|
r_core_cmdf (core, "cat %s~..", arg);
|
|
} else {
|
|
eprintf ("Usage: less [file]\n");
|
|
}
|
|
break;
|
|
default: // "ls"
|
|
if (!arg) {
|
|
arg = "";
|
|
}
|
|
if (r_fs_check (core->fs, arg)) {
|
|
r_core_cmdf (core, "md %s", arg);
|
|
} else {
|
|
char *res = r_syscmd_ls (arg);
|
|
if (res) {
|
|
r_cons_print (res);
|
|
free (res);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_join(void *data, const char *input) { // "join"
|
|
RCore *core = (RCore *)data;
|
|
char *tmp = strdup (input);
|
|
const char *arg1 = strchr (tmp, ' ');
|
|
if (!arg1) {
|
|
goto beach;
|
|
}
|
|
arg1 = r_str_trim_head_ro (arg1);
|
|
if (!arg1) {
|
|
goto beach;
|
|
}
|
|
char *end = strchr (arg1, ' ');
|
|
if (!end) {
|
|
goto beach;
|
|
}
|
|
*end = '\0';
|
|
const char *arg2 = end+1;
|
|
if (!arg2) {
|
|
goto beach;
|
|
}
|
|
arg2 = r_str_trim_head_ro (arg2);
|
|
switch (*input) {
|
|
case '?': // "join?"
|
|
goto beach;
|
|
default: // "join"
|
|
if (!arg1) {
|
|
arg1 = "";
|
|
}
|
|
if (!arg2) {
|
|
arg2 = "";
|
|
}
|
|
if (!r_fs_check (core->fs, arg1) && !r_fs_check (core->fs, arg2)) {
|
|
char *res = r_syscmd_join (arg1, arg2);
|
|
if (res) {
|
|
r_cons_print (res);
|
|
free (res);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
free (tmp);
|
|
return 0;
|
|
beach:
|
|
eprintf ("Usage: join [file1] [file2] # join the contents of the two files\n");
|
|
free (tmp);
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_stdin(void *data, const char *input) {
|
|
RCore *core = (RCore *)data;
|
|
if (input[0] == '?') {
|
|
r_cons_printf ("Usage: '-' '.-' '. -' do the same\n");
|
|
return false;
|
|
}
|
|
return r_core_run_script (core, "-");
|
|
}
|
|
|
|
static void load_table_json(RCore *core, RTable *t, char *data) {
|
|
// parse json file and iterate over all the entries
|
|
// RTableRow *row = r_table_row_new (items);
|
|
// r_list_append (t->rows, row);
|
|
eprintf ("TODO: Loading tables from JSON is not yet implemented\n");
|
|
}
|
|
|
|
static const char *get_type_string(const char *s) {
|
|
if (!strncmp (s, "0x", 2)) {
|
|
return "x";
|
|
}
|
|
if (*s == '0' || atoi (s)) {
|
|
return "d";
|
|
}
|
|
return "s";
|
|
}
|
|
|
|
static void load_table_csv(RCore *core, RTable *t, RList *lines) {
|
|
RListIter *iter;
|
|
char *line;
|
|
int row = 0;
|
|
|
|
RList *cols = NULL;
|
|
r_list_foreach (lines, iter, line) {
|
|
char *word;
|
|
RListIter *iter2;
|
|
RList *words = r_str_split_list (line, ",", 0);
|
|
if (r_list_length (words) > 0) {
|
|
switch (row) {
|
|
case 0:
|
|
cols = words;
|
|
words = NULL;
|
|
break;
|
|
case 1:
|
|
{
|
|
RStrBuf *b = r_strbuf_new (",h ");
|
|
RStrBuf *args = r_strbuf_new ("");
|
|
r_list_foreach (words, iter2, word) {
|
|
const char *type = get_type_string (word);
|
|
r_strbuf_append (b, type);
|
|
}
|
|
r_list_foreach (cols, iter2, word) {
|
|
r_strbuf_append (b , " ");
|
|
r_strbuf_append (b, word);
|
|
}
|
|
r_core_cmd0 (core, r_strbuf_get (b));
|
|
r_strbuf_free (args);
|
|
r_strbuf_free (b);
|
|
}
|
|
/* fallthrough */
|
|
default:
|
|
{
|
|
RStrBuf *b = r_strbuf_new (",r ");
|
|
r_list_foreach (words, iter2, word) {
|
|
r_strbuf_append (b, " ");
|
|
r_strbuf_append (b, word);
|
|
}
|
|
r_core_cmd0 (core, r_strbuf_get (b));
|
|
r_strbuf_free (b);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
r_list_free (words);
|
|
row++;
|
|
}
|
|
}
|
|
|
|
static void load_table_asciiart(RCore *core, RTable *t, RList *lines) {
|
|
RListIter *iter;
|
|
char *line;
|
|
const char *separator = "|";
|
|
int ncols = 0;
|
|
bool expect_header = false;
|
|
bool expect_rows = false;
|
|
r_list_foreach (lines, iter, line) {
|
|
if (!expect_rows) {
|
|
if (r_str_startswith (line, ".--")) {
|
|
expect_header = true;
|
|
separator = "|";
|
|
continue;
|
|
}
|
|
if (r_str_startswith (line, "┌")) {
|
|
expect_header = true;
|
|
separator = "│";
|
|
continue;
|
|
}
|
|
if (r_str_startswith (line, ")-")) {
|
|
expect_rows = true;
|
|
separator = "|";
|
|
continue;
|
|
}
|
|
if (r_str_startswith (line, "│─")) {
|
|
expect_rows = true;
|
|
separator = "│";
|
|
continue;
|
|
}
|
|
}
|
|
|
|
RTableColumnType *typeString = r_table_type ("string");
|
|
RTableColumnType *typeNumber = r_table_type ("number");
|
|
if (expect_header) {
|
|
char *arg;
|
|
RList *args = r_str_split_list (line + strlen (separator), separator, 0);
|
|
RListIter *iter2;
|
|
ncols = 0;
|
|
if (r_list_length (t->cols) > 0) {
|
|
eprintf ("Warning: Not re-adding headers. Use ,- to reset the table.\n");
|
|
continue;
|
|
}
|
|
r_list_foreach (args, iter2, arg) {
|
|
char *s = strchr (arg, ' ');
|
|
char *ss = r_str_trim_dup (s? s + 1: arg);
|
|
if (!*ss) {
|
|
free (ss);
|
|
continue;
|
|
}
|
|
r_table_add_column (t, typeString, ss, 0);
|
|
ncols ++;
|
|
}
|
|
expect_header = false;
|
|
} else if (expect_rows) {
|
|
char *arg;
|
|
size_t line_len = strlen (line);
|
|
size_t separator_len = strlen (separator);
|
|
size_t pos = (line_len < separator_len)? line_len: separator_len;
|
|
RList *args = r_str_split_list (line + pos, separator, 0);
|
|
RList *items = r_list_newf (free);
|
|
RListIter *iter2;
|
|
if (r_list_length (args) < ncols) {
|
|
// dowarn?
|
|
continue;
|
|
}
|
|
r_list_foreach (args, iter2, arg) {
|
|
char *ss = r_str_trim_dup (arg);
|
|
if (!*ss) {
|
|
free (ss);
|
|
continue;
|
|
}
|
|
if (isdigit ((unsigned char)*ss)) {
|
|
int col = r_list_length (items);
|
|
RTableColumn *c = r_list_get_n (t->cols, col);
|
|
if (c) {
|
|
c->type = typeNumber;
|
|
}
|
|
}
|
|
r_list_append (items, ss);
|
|
}
|
|
RTableRow *row = r_table_row_new (items);
|
|
r_list_append (t->rows, row);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void load_table(RCore *core, RTable *t, char *data) {
|
|
r_return_if_fail (core && t && data);
|
|
if (*data == '[') {
|
|
load_table_json (core, t, data);
|
|
} else {
|
|
RList *lines = r_str_split_list (data, "\n", 0);
|
|
if (strchr (data, ',')) {
|
|
load_table_csv (core, t, lines);
|
|
} else {
|
|
load_table_asciiart (core, t, lines);
|
|
}
|
|
r_list_free (lines);
|
|
}
|
|
free (data);
|
|
}
|
|
|
|
static void display_table(char *ts) {
|
|
if (ts) {
|
|
r_cons_printf ("%s\n", ts);
|
|
free (ts);
|
|
}
|
|
}
|
|
|
|
static void cmd_table_header(RCore *core, char *s) {
|
|
RList *list = r_str_split_list (s, " ", 0); // owns *s
|
|
RListIter *iter;
|
|
char *format = r_list_pop_head (list);
|
|
if (!format) {
|
|
return;
|
|
}
|
|
if (!core->table) {
|
|
core->table = r_core_table (core, "header");
|
|
}
|
|
size_t i = 0;
|
|
r_list_foreach (list, iter, s) {
|
|
const char type_char = format[i];
|
|
if (!type_char) {
|
|
break;
|
|
}
|
|
const char *type_name = (type_char == 's')
|
|
? "string": "number";
|
|
RTableColumnType *typeString = r_table_type (type_name);
|
|
r_table_add_column (core->table, typeString, s, 0);
|
|
i++;
|
|
}
|
|
r_list_free (list);
|
|
free (format);
|
|
}
|
|
|
|
static bool display_table_filter(RCore *core, const char *input) {
|
|
r_return_val_if_fail (core && input, false);
|
|
if (!core->table) {
|
|
return false;
|
|
}
|
|
int skip = (*input == ' ')? 1: (*input&&input[1])? 2: 0;
|
|
if (skip) {
|
|
const char *q = r_str_trim_head_ro (input + skip);
|
|
return r_table_query (core->table, q);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static int cmd_table(void *data, const char *input) {
|
|
RCore *core = (RCore*)data;
|
|
if (!core->table) {
|
|
core->table = r_table_new ("table");
|
|
}
|
|
switch (*input) {
|
|
case 'h': // table header columns
|
|
case 'c': // table columns
|
|
cmd_table_header (core, r_str_trim_dup (input + 1));
|
|
break;
|
|
case 'r': // add row
|
|
{
|
|
if (!core->table) {
|
|
core->table = r_table_new ("table");
|
|
}
|
|
char *args = r_str_trim_dup (input + 1);
|
|
if (*args) {
|
|
RList *list = r_str_split_list (args, " ", 0);
|
|
if (list) {
|
|
r_table_add_row_list (core->table, list);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
case '-':
|
|
r_table_free (core->table);
|
|
core->table = r_table_new ("table");
|
|
break;
|
|
case '/':
|
|
// query here
|
|
{
|
|
RTable *ot = r_table_clone (core->table);
|
|
if (display_table_filter (core, input)) {
|
|
display_table (r_table_tostring (core->table));
|
|
}
|
|
r_table_free (core->table);
|
|
core->table = ot;
|
|
}
|
|
break;
|
|
case '.': // ",."
|
|
if (R_STR_ISEMPTY (input + 1)) {
|
|
eprintf ("Usage: ,. [file | $alias]\n");
|
|
} else {
|
|
const char *file = r_str_trim_head_ro (input + 1);
|
|
if (*file == '$') {
|
|
const char *file_data = r_cmd_alias_get (core->rcmd, file, 1);
|
|
if (file_data) {
|
|
load_table (core, core->table, strdup (file_data + 1));
|
|
}
|
|
} else {
|
|
char *file_data = r_file_slurp (file, NULL);
|
|
if (file_data) {
|
|
load_table (core, core->table, file_data);
|
|
} else {
|
|
eprintf ("Cannot open file.\n");
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
case ' ':
|
|
if (display_table_filter (core, input)) {
|
|
display_table (r_table_tostring (core->table));
|
|
}
|
|
break;
|
|
case ',':
|
|
if (display_table_filter (core, input)) {
|
|
display_table (r_table_tocsv (core->table));
|
|
}
|
|
break;
|
|
case '*':
|
|
if (display_table_filter (core, input)) {
|
|
display_table (r_table_tor2cmds (core->table));
|
|
}
|
|
break;
|
|
case 'j':
|
|
if (display_table_filter (core, input)) {
|
|
display_table (r_table_tojson (core->table));
|
|
}
|
|
break;
|
|
case 0:
|
|
if (core->table) {
|
|
display_table (r_table_tofancystring (core->table));
|
|
}
|
|
break;
|
|
case '?':
|
|
r_core_cmd_help (core, cmd_table_help);
|
|
r_cons_printf ("%s\n", r_table_help ());
|
|
break;
|
|
default:
|
|
r_core_cmd_help (core, cmd_table_help);
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_interpret(void *data, const char *input) {
|
|
char *str, *ptr, *eol, *rbuf, *filter, *inp;
|
|
const char *host, *port, *cmd;
|
|
RCore *core = (RCore *)data;
|
|
|
|
if (!strcmp (input, "?")) {
|
|
r_core_cmd_help (core, help_msg_dot);
|
|
return 0;
|
|
}
|
|
switch (*input) {
|
|
case '\0': // "."
|
|
lastcmd_repeat (core, 0);
|
|
break;
|
|
case ':': // ".:"
|
|
if ((ptr = strchr (input + 1, ' '))) {
|
|
/* .:port cmd */
|
|
/* .:host:port cmd */
|
|
cmd = ptr + 1;
|
|
*ptr = 0;
|
|
eol = strchr (input + 1, ':');
|
|
if (eol) {
|
|
*eol = 0;
|
|
host = input + 1;
|
|
port = eol + 1;
|
|
} else {
|
|
host = "localhost";
|
|
port = input + ((input[1] == ':')? 2: 1);
|
|
}
|
|
rbuf = r_core_rtr_cmds_query (core, host, port, cmd);
|
|
if (rbuf) {
|
|
r_cons_print (rbuf);
|
|
free (rbuf);
|
|
}
|
|
} else {
|
|
r_core_rtr_cmds (core, input + 1);
|
|
}
|
|
break;
|
|
case '.': // ".." same as \n
|
|
if (input[1] == '.') { // "..." run the last command repeated
|
|
// same as \n with e cmd.repeat=true
|
|
lastcmd_repeat (core, 1);
|
|
} else if (input[1]) {
|
|
char *str = r_core_cmd_str_pipe (core, r_str_trim_head_ro (input));
|
|
if (str) {
|
|
r_core_cmd (core, str, 0);
|
|
free (str);
|
|
}
|
|
} else {
|
|
eprintf ("Usage: .. ([file])\n");
|
|
}
|
|
break;
|
|
case '*': // ".*"
|
|
{
|
|
const char *a = r_str_trim_head_ro (input + 1);
|
|
char *s = strdup (a);
|
|
char *sp = strchr (s, ' ');
|
|
if (sp) {
|
|
*sp = 0;
|
|
}
|
|
if (R_STR_ISNOTEMPTY (s)) {
|
|
r_core_run_script (core, s);
|
|
}
|
|
free (s);
|
|
}
|
|
break;
|
|
case '-': // ".-"
|
|
if (input[1] == '?') {
|
|
r_cons_printf ("Usage: '-' '.-' '. -' do the same\n");
|
|
} else {
|
|
r_core_run_script (core, "-");
|
|
}
|
|
break;
|
|
case ' ': // ". "
|
|
{
|
|
const char *script_file = r_str_trim_head_ro (input + 1);
|
|
if (*script_file == '$') {
|
|
const char *oldText = r_cmd_alias_get (core->rcmd, script_file, 1);
|
|
r_core_cmd0 (core, oldText); // script_file);
|
|
} else {
|
|
if (!r_core_run_script (core, script_file)) {
|
|
eprintf ("Cannot find script '%s'\n", script_file);
|
|
core->num->value = 1;
|
|
} else {
|
|
core->num->value = 0;
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
case '!': // ".!"
|
|
/* from command */
|
|
r_core_cmd_command (core, input + 1);
|
|
break;
|
|
case '(': // ".("
|
|
r_cmd_macro_call (&core->rcmd->macro, input + 1);
|
|
break;
|
|
default:
|
|
if (*input >= 0 && *input <= 9) {
|
|
eprintf ("|ERROR| No .[0..9] to avoid infinite loops\n");
|
|
break;
|
|
}
|
|
inp = strdup (input);
|
|
filter = strchr (inp, '~');
|
|
if (filter) {
|
|
*filter = 0;
|
|
}
|
|
int tmp_html = r_cons_singleton ()->is_html;
|
|
r_cons_singleton ()->is_html = false;
|
|
ptr = str = r_core_cmd_str (core, inp);
|
|
r_cons_singleton ()->is_html = tmp_html;
|
|
|
|
if (filter) {
|
|
*filter = '~';
|
|
}
|
|
r_cons_break_push (NULL, NULL);
|
|
if (ptr) {
|
|
for (;;) {
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
eol = strchr (ptr, '\n');
|
|
if (eol) {
|
|
*eol = '\0';
|
|
}
|
|
if (*ptr) {
|
|
char *p = r_str_append (strdup (ptr), filter);
|
|
r_core_cmd0 (core, p);
|
|
free (p);
|
|
}
|
|
if (!eol) {
|
|
break;
|
|
}
|
|
ptr = eol + 1;
|
|
}
|
|
}
|
|
r_cons_break_pop ();
|
|
free (str);
|
|
free (inp);
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static bool callback_foreach_kv(void *user, const char *k, const char *v) {
|
|
r_cons_printf ("%s=%s\n", k, v);
|
|
return true;
|
|
}
|
|
|
|
R_API int r_line_hist_sdb_up(RLine *line) {
|
|
if (!line->sdbshell_hist_iter || !line->sdbshell_hist_iter->n) {
|
|
return false;
|
|
}
|
|
line->sdbshell_hist_iter = line->sdbshell_hist_iter->n;
|
|
strncpy (line->buffer.data, line->sdbshell_hist_iter->data, R_LINE_BUFSIZE - 1);
|
|
line->buffer.index = line->buffer.length = strlen (line->buffer.data);
|
|
return true;
|
|
}
|
|
|
|
R_API int r_line_hist_sdb_down(RLine *line) {
|
|
if (!line->sdbshell_hist_iter || !line->sdbshell_hist_iter->p) {
|
|
return false;
|
|
}
|
|
line->sdbshell_hist_iter = line->sdbshell_hist_iter->p;
|
|
strncpy (line->buffer.data, line->sdbshell_hist_iter->data, R_LINE_BUFSIZE - 1);
|
|
line->buffer.index = line->buffer.length = strlen (line->buffer.data);
|
|
return true;
|
|
}
|
|
|
|
static int cmd_kuery(void *data, const char *input) {
|
|
char buf[1024], *out;
|
|
RCore *core = (RCore*)data;
|
|
const char *sp, *p = "[sdb]> ";
|
|
Sdb *s = core->sdb;
|
|
|
|
char *cur_pos = NULL, *cur_cmd = NULL, *next_cmd = NULL;
|
|
char *temp_pos = NULL, *temp_cmd = NULL;
|
|
|
|
switch (input[0]) {
|
|
case 'j':
|
|
out = sdb_querys (s, NULL, 0, "anal/**");
|
|
if (!out) {
|
|
r_cons_println ("No Output from sdb");
|
|
break;
|
|
}
|
|
PJ * pj = pj_new ();
|
|
if (!pj) {
|
|
free (out);
|
|
break;
|
|
}
|
|
pj_o (pj);
|
|
pj_ko (pj, "anal");
|
|
pj_ka (pj, "cur_cmd");
|
|
|
|
while (*out) {
|
|
cur_pos = strchr (out, '\n');
|
|
if (!cur_pos) {
|
|
break;
|
|
}
|
|
cur_cmd = r_str_ndup (out, cur_pos - out);
|
|
pj_s (pj, cur_cmd);
|
|
|
|
free (next_cmd);
|
|
next_cmd = r_str_newf ("anal/%s/*", cur_cmd);
|
|
char *query_result = sdb_querys (s, NULL, 0, next_cmd);
|
|
|
|
if (!query_result) {
|
|
out = cur_pos + 1;
|
|
continue;
|
|
}
|
|
|
|
char *temp = query_result;
|
|
while (*temp) {
|
|
temp_pos = strchr (temp, '\n');
|
|
if (!temp_pos) {
|
|
break;
|
|
}
|
|
temp_cmd = r_str_ndup (temp, temp_pos - temp);
|
|
pj_s (pj, temp_cmd);
|
|
temp = temp_pos + 1;
|
|
}
|
|
out = cur_pos + 1;
|
|
free (query_result);
|
|
}
|
|
pj_end (pj);
|
|
pj_end (pj);
|
|
pj_end (pj);
|
|
r_cons_println (pj_string (pj));
|
|
pj_free (pj);
|
|
R_FREE (next_cmd);
|
|
free (next_cmd);
|
|
free (cur_cmd);
|
|
break;
|
|
case ' ':
|
|
if (s) {
|
|
out = sdb_querys (s, NULL, 0, input + 1);
|
|
if (out) {
|
|
r_cons_print (out);
|
|
}
|
|
R_FREE (out);
|
|
}
|
|
break;
|
|
//case 's': r_pair_save (s, input + 3); break;
|
|
//case 'l': r_pair_load (sdb, input + 3); break;
|
|
case '\0':
|
|
sdb_foreach (s, callback_foreach_kv, NULL);
|
|
break;
|
|
// TODO: add command to list all namespaces // sdb_ns_foreach ?
|
|
case 's': // "ks"
|
|
if (core->http_up) {
|
|
return false;
|
|
}
|
|
if (!r_cons_is_interactive ()) {
|
|
return false;
|
|
}
|
|
if (input[1] == ' ') {
|
|
char *n, *o, *p = strdup (input + 2);
|
|
// TODO: slash split here? or inside sdb_ns ?
|
|
for (n = o = p; n; o = n) {
|
|
n = strchr (o, '/'); // SDB_NS_SEPARATOR NAMESPACE
|
|
if (n) {
|
|
*n++ = 0;
|
|
}
|
|
s = sdb_ns (s, o, 1);
|
|
}
|
|
free (p);
|
|
}
|
|
if (!s) {
|
|
s = core->sdb;
|
|
}
|
|
RLine *line = core->cons->line;
|
|
if (!line->sdbshell_hist) {
|
|
line->sdbshell_hist = r_list_newf (free);
|
|
r_list_append (line->sdbshell_hist, r_str_new ("\0"));
|
|
}
|
|
RList *sdb_hist = line->sdbshell_hist;
|
|
r_line_set_hist_callback (line, &r_line_hist_sdb_up, &r_line_hist_sdb_down);
|
|
for (;;) {
|
|
r_line_set_prompt (p);
|
|
if (r_cons_fgets (buf, sizeof (buf), 0, NULL) < 1) {
|
|
break;
|
|
}
|
|
if (!*buf) {
|
|
break;
|
|
}
|
|
if (sdb_hist) {
|
|
if ((r_list_length (sdb_hist) == 1) || (r_list_length (sdb_hist) > 1 && strcmp (r_list_get_n (sdb_hist, 1), buf))) {
|
|
r_list_insert (sdb_hist, 1, strdup (buf));
|
|
}
|
|
line->sdbshell_hist_iter = sdb_hist->head;
|
|
}
|
|
out = sdb_querys (s, NULL, 0, buf);
|
|
if (out) {
|
|
r_cons_println (out);
|
|
r_cons_flush ();
|
|
}
|
|
}
|
|
r_line_set_hist_callback (core->cons->line, &r_line_hist_cmd_up, &r_line_hist_cmd_down);
|
|
break;
|
|
case 'o': // "ko"
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("This command is disabled in sandbox mode\n");
|
|
return 0;
|
|
}
|
|
if (input[1] == ' ') {
|
|
char *fn = strdup (input + 2);
|
|
if (!fn) {
|
|
eprintf("Unable to allocate memory\n");
|
|
return 0;
|
|
}
|
|
char *ns = strchr (fn, ' ');
|
|
if (ns) {
|
|
Sdb *db;
|
|
*ns++ = 0;
|
|
if (r_file_exists (fn)) {
|
|
db = sdb_ns_path (core->sdb, ns, 1);
|
|
if (db) {
|
|
Sdb *newdb = sdb_new (NULL, fn, 0);
|
|
if (newdb) {
|
|
sdb_drain (db, newdb);
|
|
} else {
|
|
eprintf ("Cannot open sdb '%s'\n", fn);
|
|
}
|
|
} else {
|
|
eprintf ("Cannot find sdb '%s'\n", ns);
|
|
}
|
|
} else {
|
|
eprintf ("Cannot open file\n");
|
|
}
|
|
} else {
|
|
eprintf ("Missing sdb namespace\n");
|
|
}
|
|
free (fn);
|
|
} else {
|
|
eprintf ("Usage: ko [file] [namespace]\n");
|
|
}
|
|
break;
|
|
case 'd': // "kd"
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("This command is disabled in sandbox mode\n");
|
|
return 0;
|
|
}
|
|
if (input[1] == ' ') {
|
|
char *fn = strdup (input + 2);
|
|
char *ns = strchr (fn, ' ');
|
|
if (ns) {
|
|
*ns++ = 0;
|
|
Sdb *db = sdb_ns_path (core->sdb, ns, 0);
|
|
if (db) {
|
|
sdb_file (db, fn);
|
|
sdb_sync (db);
|
|
} else {
|
|
eprintf ("Cannot find sdb '%s'\n", ns);
|
|
}
|
|
} else {
|
|
eprintf ("Missing sdb namespace\n");
|
|
}
|
|
free (fn);
|
|
} else {
|
|
eprintf ("Usage: kd [file] [namespace]\n");
|
|
}
|
|
break;
|
|
case '?':
|
|
r_core_cmd_help (core, help_msg_k);
|
|
break;
|
|
}
|
|
|
|
if (input[0] == '\0') {
|
|
/* nothing more to do, the command has been parsed. */
|
|
return 0;
|
|
}
|
|
|
|
sp = strchr (input + 1, ' ');
|
|
if (sp) {
|
|
char *inp = strdup (input);
|
|
inp [(size_t)(sp - input)] = 0;
|
|
s = sdb_ns (core->sdb, inp + 1, 1);
|
|
out = sdb_querys (s, NULL, 0, sp + 1);
|
|
if (out) {
|
|
r_cons_println (out);
|
|
free (out);
|
|
}
|
|
free (inp);
|
|
return 0;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_bsize(void *data, const char *input) {
|
|
ut64 n;
|
|
RFlagItem *flag;
|
|
RCore *core = (RCore *)data;
|
|
switch (input[0]) {
|
|
case 'm': // "bm"
|
|
n = r_num_math (core->num, input + 1);
|
|
if (n > 1) {
|
|
core->blocksize_max = n;
|
|
} else {
|
|
r_cons_printf ("0x%x\n", (ut32)core->blocksize_max);
|
|
}
|
|
break;
|
|
case '+': // "b+"
|
|
n = r_num_math (core->num, input + 1);
|
|
r_core_block_size (core, core->blocksize + n);
|
|
break;
|
|
case '-': // "b-"
|
|
n = r_num_math (core->num, input + 1);
|
|
r_core_block_size (core, core->blocksize - n);
|
|
break;
|
|
case 'f': // "bf"
|
|
if (input[1] == ' ') {
|
|
flag = r_flag_get (core->flags, input + 2);
|
|
if (flag) {
|
|
r_core_block_size (core, flag->size);
|
|
} else {
|
|
eprintf ("bf: cannot find flag named '%s'\n", input + 2);
|
|
}
|
|
} else {
|
|
eprintf ("Usage: bf [flagname]\n");
|
|
}
|
|
break;
|
|
case 'j': { // "bj"
|
|
PJ * pj = pj_new ();
|
|
if (!pj) {
|
|
break;
|
|
}
|
|
pj_o (pj);
|
|
pj_ki (pj, "blocksize", core->blocksize);
|
|
pj_ki (pj, "blocksize_limit", core->blocksize_max);
|
|
pj_end (pj);
|
|
r_cons_println (pj_string (pj));
|
|
pj_free (pj);
|
|
break;
|
|
}
|
|
case '*': // "b*"
|
|
r_cons_printf ("b 0x%x\n", core->blocksize);
|
|
break;
|
|
case '\0': // "b"
|
|
r_cons_printf ("0x%x\n", core->blocksize);
|
|
break;
|
|
case ' ':
|
|
r_core_block_size (core, r_num_math (core->num, input));
|
|
break;
|
|
default:
|
|
case '?': // "b?"
|
|
r_core_cmd_help (core, help_msg_b);
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int __runMain(RMainCallback cb, const char *arg) {
|
|
char *a = r_str_trim_dup (arg);
|
|
int argc = 0;
|
|
char **args = r_str_argv (a, &argc);
|
|
int res = cb (argc, (const char **)args);
|
|
free (args);
|
|
free (a);
|
|
return res;
|
|
}
|
|
|
|
static bool cmd_r2cmd(RCore *core, const char *_input) {
|
|
char *input = r_str_newf ("r%s", _input);
|
|
int rc = 0;
|
|
if (r_str_startswith (input, "rax2")) {
|
|
rc = __runMain (core->r_main_rax2, input);
|
|
} else if (r_str_startswith (input, "r2")) {
|
|
r_sys_cmdf ("%s", input);
|
|
// rc = __runMain (core->r_main_radare2, input);
|
|
} else if (r_str_startswith (input, "radare2")) {
|
|
r_sys_cmdf ("%s", input);
|
|
// rc = __runMain (core->r_main_radare2, input);
|
|
} else if (r_str_startswith (input, "rasm2")) {
|
|
r_sys_cmdf ("%s", input);
|
|
// rc = __runMain (core->r_main_rasm2, input);
|
|
} else if (r_str_startswith (input, "rabin2")) {
|
|
r_sys_cmdf ("%s", input);
|
|
// rc = __runMain (core->r_main_rabin2, input);
|
|
} else if (r_str_startswith (input, "ragg2")) {
|
|
r_sys_cmdf ("%s", input);
|
|
// rc = __runMain (core->r_main_ragg2, input);
|
|
} else if (r_str_startswith (input, "r2pm")) {
|
|
r_sys_cmdf ("%s", input);
|
|
// rc = __runMain (core->r_main_r2pm, input);
|
|
} else if (r_str_startswith (input, "radiff2")) {
|
|
rc = __runMain (core->r_main_radiff2, input);
|
|
} else {
|
|
const char *r2cmds[] = {
|
|
"rax2", "r2pm", "rasm2", "rabin2", "rahash2", "rafind2", "rarun2", "ragg2", "radare2", "r2", NULL
|
|
};
|
|
int i;
|
|
for (i = 0; r2cmds[i]; i++) {
|
|
if (r_str_startswith (input, r2cmds[i])) {
|
|
free (input);
|
|
return true;
|
|
}
|
|
}
|
|
free (input);
|
|
return false;
|
|
}
|
|
free (input);
|
|
core->num->value = rc;
|
|
return true;
|
|
}
|
|
|
|
static int cmd_rebase(RCore *core, const char *input) {
|
|
ut64 addr = r_num_math (core->num, input);
|
|
if (!addr) {
|
|
r_cons_printf ("Usage: rb oldbase @ newbase\n");
|
|
return 0;
|
|
}
|
|
// old base = addr
|
|
// new base = core->offset
|
|
r_debug_bp_rebase (core->dbg, addr, core->offset);
|
|
r_bin_set_baddr (core->bin, core->offset);
|
|
r_flag_move (core->flags, addr, core->offset);
|
|
r_core_cmd0 (core, ".is*");
|
|
r_core_cmd0 (core, ".iM*");
|
|
r_core_cmd0 (core, ".ii*");
|
|
r_core_cmd0 (core, ".iz*");
|
|
// TODO: r_anal_move :??
|
|
// TODO: differentiate analysis by map ranges (associated with files or memory maps)
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_resize(void *data, const char *input) {
|
|
RCore *core = (RCore *)data;
|
|
ut64 newsize = 0;
|
|
st64 delta = 0;
|
|
int grow, ret;
|
|
|
|
if (cmd_r2cmd (core, input)) {
|
|
return true;
|
|
}
|
|
|
|
ut64 oldsize = (core->io->desc) ? r_io_fd_size (core->io, core->io->desc->fd): 0;
|
|
switch (*input) {
|
|
case 'a': // "r..."
|
|
if (r_str_startswith (input, "adare2")) {
|
|
__runMain (core->r_main_radare2, input - 1);
|
|
}
|
|
return true;
|
|
case 'b': // "rb" rebase
|
|
return cmd_rebase (core, input + 1);
|
|
case '2': // "r2" // XXX should be handled already in cmd_r2cmd()
|
|
// TODO: use argv[0] instead of 'radare2'
|
|
// TODO: { char **argv = { "r2", NULL }; r_main_radare2 (1, argv); }
|
|
r_sys_cmdf ("radare%s", input);
|
|
return true;
|
|
case 'm': // "rm"
|
|
if (input[1] == ' ') {
|
|
const char *file = r_str_trim_head_ro (input + 2);
|
|
if (*file == '$') {
|
|
r_cmd_alias_del (core->rcmd, file);
|
|
} else {
|
|
r_file_rm (file);
|
|
}
|
|
} else {
|
|
eprintf ("Usage: rm [file] # removes a file\n");
|
|
}
|
|
return true;
|
|
case '\0':
|
|
if (core->io->desc) {
|
|
if (oldsize != -1) {
|
|
r_cons_printf ("%"PFMT64d"\n", oldsize);
|
|
}
|
|
}
|
|
return true;
|
|
case 'j': { // "rj"
|
|
PJ * pj = pj_new ();
|
|
pj_o (pj);
|
|
if (oldsize != -1) {
|
|
pj_kn (pj, "size", oldsize);
|
|
}
|
|
pj_end (pj);
|
|
char *s = pj_drain (pj);
|
|
r_cons_println (s);
|
|
free (s);
|
|
return true;
|
|
}
|
|
case 'h':
|
|
if (core->io->desc) {
|
|
if (oldsize != -1) {
|
|
char humansz[8];
|
|
r_num_units (humansz, sizeof (humansz), oldsize);
|
|
r_cons_printf ("%s\n", humansz);
|
|
}
|
|
}
|
|
return true;
|
|
case '+': // "r+"
|
|
case '-': // "r-"
|
|
delta = (st64)r_num_math (core->num, input);
|
|
newsize = oldsize + delta;
|
|
break;
|
|
case ' ': // "r "
|
|
newsize = r_num_math (core->num, input + 1);
|
|
if (newsize == 0) {
|
|
if (input[1] == '0') {
|
|
eprintf ("Invalid size\n");
|
|
}
|
|
return false;
|
|
}
|
|
break;
|
|
case 'e':
|
|
{
|
|
int rc = write (1, Color_RESET_TERMINAL, strlen (Color_RESET_TERMINAL));
|
|
if (rc == -1) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
case '?': // "r?"
|
|
default:
|
|
r_core_cmd_help (core, help_msg_r);
|
|
return true;
|
|
}
|
|
|
|
grow = (newsize > oldsize);
|
|
if (grow) {
|
|
ret = r_io_resize (core->io, newsize);
|
|
if (ret < 1) {
|
|
eprintf ("r_io_resize: cannot resize\n");
|
|
}
|
|
}
|
|
if (delta && core->offset < newsize) {
|
|
r_io_shift (core->io, core->offset, grow?newsize:oldsize, delta);
|
|
}
|
|
if (!grow) {
|
|
ret = r_io_resize (core->io, newsize);
|
|
if (ret < 1) {
|
|
eprintf ("r_io_resize: cannot resize\n");
|
|
}
|
|
}
|
|
if (newsize < core->offset+core->blocksize || oldsize < core->offset + core->blocksize) {
|
|
r_core_block_read (core);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static int cmd_panels(void *data, const char *input) {
|
|
RCore *core = (RCore*) data;
|
|
if (core->vmode) {
|
|
return false;
|
|
}
|
|
if (*input == '?') {
|
|
r_core_cmd_help (core, help_msg_v);
|
|
return false;
|
|
}
|
|
if (!r_cons_is_interactive ()) {
|
|
eprintf ("Panel mode requires scr.interactive=true.\n");
|
|
return false;
|
|
}
|
|
if (*input == ' ') {
|
|
if (core->panels) {
|
|
r_core_panels_load (core, input + 1);
|
|
}
|
|
r_config_set (core->config, "scr.layout", input + 1);
|
|
return true;
|
|
}
|
|
if (*input == '=') {
|
|
r_core_panels_save (core, input + 1);
|
|
r_config_set (core->config, "scr.layout", input + 1);
|
|
return true;
|
|
}
|
|
if (*input == 'i') {
|
|
char *sp = strchr (input, ' ');
|
|
if (sp) {
|
|
char *r = r_core_editor (core, sp + 1, NULL);
|
|
if (r) {
|
|
free (r);
|
|
} else {
|
|
eprintf ("Cannot open file (%s)\n", sp + 1);
|
|
}
|
|
}
|
|
////r_sys_cmdf ("v%s", input);
|
|
return false;
|
|
}
|
|
r_core_panels_root (core, core->panels_root);
|
|
return true;
|
|
}
|
|
|
|
static int cmd_visual(void *data, const char *input) {
|
|
RCore *core = (RCore*) data;
|
|
if (core->http_up) {
|
|
return false;
|
|
}
|
|
if (!r_cons_is_interactive ()) {
|
|
eprintf ("Visual mode requires scr.interactive=true.\n");
|
|
return false;
|
|
}
|
|
return r_core_visual ((RCore *)data, input);
|
|
}
|
|
|
|
static int cmd_pipein(void *user, const char *input) {
|
|
char *buf = strdup (input);
|
|
int len = r_str_unescape (buf);
|
|
r_cons_readpush (buf, len);
|
|
free (buf);
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_tasks(void *data, const char *input) {
|
|
RCore *core = (RCore*) data;
|
|
switch (input[0]) {
|
|
case '\0': // "&"
|
|
case 'j': // "&j"
|
|
r_core_task_list (core, *input);
|
|
break;
|
|
case 'b': { // "&b"
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("This command is disabled in sandbox mode\n");
|
|
return 0;
|
|
}
|
|
int tid = r_num_math (core->num, input + 1);
|
|
if (tid) {
|
|
r_core_task_break (&core->tasks, tid);
|
|
}
|
|
break;
|
|
}
|
|
case '&': { // "&&"
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("This command is disabled in sandbox mode\n");
|
|
return 0;
|
|
}
|
|
int tid = r_num_math (core->num, input + 1);
|
|
r_core_task_join (&core->tasks, core->tasks.current_task, tid ? tid : -1);
|
|
break;
|
|
}
|
|
case '=': { // "&="
|
|
// r_core_task_list (core, '=');
|
|
int tid = r_num_math (core->num, input + 1);
|
|
if (tid) {
|
|
RCoreTask *task = r_core_task_get_incref (&core->tasks, tid);
|
|
if (task) {
|
|
if (task->res) {
|
|
r_cons_println (task->res);
|
|
}
|
|
r_core_task_decref (task);
|
|
} else {
|
|
eprintf ("Cannot find task\n");
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
case '-': // "&-"
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("This command is disabled in sandbox mode\n");
|
|
return 0;
|
|
}
|
|
if (input[1] == '*') {
|
|
r_core_task_del_all_done (&core->tasks);
|
|
} else {
|
|
r_core_task_del (&core->tasks, r_num_math (core->num, input + 1));
|
|
}
|
|
break;
|
|
case '?': // "&?"
|
|
default:
|
|
helpCmdTasks (core);
|
|
break;
|
|
case ' ': // "& "
|
|
case '_': // "&_"
|
|
case 't': { // "&t"
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("This command is disabled in sandbox mode\n");
|
|
return 0;
|
|
}
|
|
RCoreTask *task = r_core_task_new (core, true, input + 1, NULL, core);
|
|
if (!task) {
|
|
break;
|
|
}
|
|
task->transient = input[0] == 't';
|
|
r_core_task_enqueue (&core->tasks, task);
|
|
break;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_pointer(void *data, const char *input) {
|
|
RCore *core = (RCore*) data;
|
|
int ret = true;
|
|
char *str, *eq;
|
|
input = r_str_trim_head_ro (input);
|
|
while (*input == ' ') {
|
|
input++;
|
|
}
|
|
if (!*input || *input == '?') {
|
|
r_core_cmd_help (core, help_msg_star);
|
|
return ret;
|
|
}
|
|
str = strdup (input);
|
|
eq = strchr (str, '=');
|
|
if (eq) {
|
|
*eq++ = 0;
|
|
if (!strncmp (eq, "0x", 2)) {
|
|
ret = r_core_cmdf (core, "wv %s@%s", eq, str);
|
|
} else {
|
|
ret = r_core_cmdf (core, "wx %s@%s", eq, str);
|
|
}
|
|
} else {
|
|
ret = r_core_cmdf (core, "?v [%s]", input);
|
|
}
|
|
free (str);
|
|
return ret;
|
|
}
|
|
|
|
static int cmd_env(void *data, const char *input) {
|
|
RCore *core = (RCore*)data;
|
|
int ret = true;
|
|
switch (*input) {
|
|
case '?':
|
|
cmd_help_percent (core);
|
|
break;
|
|
default:
|
|
ret = r_core_cmdf (core, "env %s", input);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static struct autocomplete_flag_map_t {
|
|
const char* name;
|
|
const char* desc;
|
|
int type;
|
|
} autocomplete_flags [] = {
|
|
{ "$dflt", "default autocomplete flag", R_CORE_AUTOCMPLT_DFLT },
|
|
{ "$flag", "shows known flag hints", R_CORE_AUTOCMPLT_FLAG },
|
|
{ "$flsp", "shows known flag-spaces hints", R_CORE_AUTOCMPLT_FLSP },
|
|
{ "$seek", "shows the seek hints", R_CORE_AUTOCMPLT_SEEK },
|
|
{ "$fcn", "shows the functions hints", R_CORE_AUTOCMPLT_FCN },
|
|
{ "$vars", "autocomplete function varnames", R_CORE_AUTOCMPLT_VARS },
|
|
{ "$zign", "shows known zignatures hints", R_CORE_AUTOCMPLT_ZIGN },
|
|
{ "$eval", "shows known evals hints", R_CORE_AUTOCMPLT_EVAL },
|
|
{ "$prjt", "shows known projects hints", R_CORE_AUTOCMPLT_PRJT },
|
|
{ "$mins", NULL, R_CORE_AUTOCMPLT_MINS },
|
|
{ "$brkp", "shows known breakpoints hints", R_CORE_AUTOCMPLT_BRKP },
|
|
{ "$macro", NULL, R_CORE_AUTOCMPLT_MACR },
|
|
{ "$file", "hints file paths", R_CORE_AUTOCMPLT_FILE },
|
|
{ "$thme", "shows known themes hints", R_CORE_AUTOCMPLT_THME },
|
|
{ "$optn", "allows the selection for multiple options", R_CORE_AUTOCMPLT_OPTN },
|
|
{ "$ms", "shows mount hints", R_CORE_AUTOCMPLT_MS},
|
|
{ "$sdb", "shows sdb hints", R_CORE_AUTOCMPLT_SDB},
|
|
{ NULL, NULL, 0 }
|
|
};
|
|
|
|
static inline void print_dict(RCoreAutocomplete* a, int sub) {
|
|
if (!a) {
|
|
return;
|
|
}
|
|
int i, j;
|
|
const char* name = "unknown";
|
|
for (i = 0; i < a->n_subcmds; i++) {
|
|
RCoreAutocomplete* b = a->subcmds[i];
|
|
if (b->locked) {
|
|
continue;
|
|
}
|
|
for (j = 0; j < R_CORE_AUTOCMPLT_END; j++) {
|
|
if (b->type == autocomplete_flags[j].type) {
|
|
name = autocomplete_flags[j].name;
|
|
break;
|
|
}
|
|
}
|
|
eprintf ("[%3d] %s: '%s'\n", sub, name, b->cmd);
|
|
print_dict (a->subcmds[i], sub + 1);
|
|
}
|
|
}
|
|
|
|
static int autocomplete_type(const char* strflag) {
|
|
int i;
|
|
for (i = 0; i < R_CORE_AUTOCMPLT_END; i++) {
|
|
if (autocomplete_flags[i].desc && !strncmp (strflag, autocomplete_flags[i].name, 5)) {
|
|
return autocomplete_flags[i].type;
|
|
}
|
|
}
|
|
eprintf ("Invalid flag '%s'\n", strflag);
|
|
return R_CORE_AUTOCMPLT_END;
|
|
}
|
|
|
|
static void cmd_autocomplete_help(RCore *core) {
|
|
r_core_cmd_help (core, help_msg_triple_exclamation);
|
|
// non-zero-cost survival without iterators 101
|
|
const char **help = calloc (R_CORE_AUTOCMPLT_END + 1, 3 * sizeof (char *));
|
|
int i;
|
|
size_t n;
|
|
for (i = 0, n = 0; i < R_CORE_AUTOCMPLT_END; i++) {
|
|
if (autocomplete_flags[i].desc) {
|
|
// highlight "$" as cmd and the rest of the name as args
|
|
help[n + 0] = "$";
|
|
help[n + 1] = autocomplete_flags[i].name + 1;
|
|
help[n + 2] = autocomplete_flags[i].desc;
|
|
n += 3;
|
|
}
|
|
}
|
|
r_core_cmd_help (core, help);
|
|
free (help);
|
|
}
|
|
|
|
static void cmd_autocomplete(RCore *core, const char *input) {
|
|
RCoreAutocomplete* b = core->autocomplete;
|
|
input = r_str_trim_head_ro (input);
|
|
char arg[256];
|
|
if (!*input) {
|
|
print_dict (core->autocomplete, 0);
|
|
return;
|
|
}
|
|
if (*input == '?') {
|
|
cmd_autocomplete_help (core);
|
|
return;
|
|
}
|
|
if (*input == '-') {
|
|
const char *arg = input + 1;
|
|
if (!*input) {
|
|
eprintf ("Use !!!-* or !!!-<cmd>\n");
|
|
return;
|
|
}
|
|
r_core_autocomplete_remove (b, arg);
|
|
return;
|
|
}
|
|
while (b) {
|
|
const char* end = r_str_trim_head_wp (input);
|
|
if (!end) {
|
|
break;
|
|
}
|
|
if ((end - input) >= sizeof (arg)) {
|
|
// wtf?
|
|
eprintf ("Exceeded the max arg length (255).\n");
|
|
return;
|
|
}
|
|
if (end == input) {
|
|
break;
|
|
}
|
|
memcpy (arg, input, end - input);
|
|
arg[end - input] = 0;
|
|
RCoreAutocomplete* a = r_core_autocomplete_find (b, arg, true);
|
|
input = r_str_trim_head_ro (end);
|
|
if (input && *input && !a) {
|
|
if (b->type == R_CORE_AUTOCMPLT_DFLT && !(b = r_core_autocomplete_add (b, arg, R_CORE_AUTOCMPLT_DFLT, false))) {
|
|
eprintf ("ENOMEM\n");
|
|
return;
|
|
} else if (b->type != R_CORE_AUTOCMPLT_DFLT) {
|
|
eprintf ("Cannot add autocomplete to '%s'. type not $dflt\n", b->cmd);
|
|
return;
|
|
}
|
|
} else if ((!input || !*input) && !a) {
|
|
if (arg[0] == '$') {
|
|
int type = autocomplete_type (arg);
|
|
if (type != R_CORE_AUTOCMPLT_END && !b->locked && !b->n_subcmds) {
|
|
b->type = type;
|
|
} else if (b->locked || b->n_subcmds) {
|
|
if (!b->cmd) {
|
|
return;
|
|
}
|
|
eprintf ("Changing type of '%s' is forbidden.\n", b->cmd);
|
|
}
|
|
} else {
|
|
if (!r_core_autocomplete_add (b, arg, R_CORE_AUTOCMPLT_DFLT, false)) {
|
|
eprintf ("ENOMEM\n");
|
|
return;
|
|
}
|
|
}
|
|
return;
|
|
} else if ((!input || !*input) && a) {
|
|
// eprintf ("Cannot add '%s'. Already exists.\n", arg);
|
|
return;
|
|
} else {
|
|
b = a;
|
|
}
|
|
}
|
|
eprintf ("Invalid usage of !!!\n");
|
|
}
|
|
|
|
static int cmd_last(void *data, const char *input) {
|
|
switch (*input) {
|
|
case 0:
|
|
r_cons_last ();
|
|
break;
|
|
default:
|
|
eprintf ("Usage: _ print last output\n");
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_system(void *data, const char *input) {
|
|
RCore *core = (RCore*)data;
|
|
ut64 n;
|
|
int ret = 0;
|
|
switch (*input) {
|
|
case '-': //!-
|
|
if (input[1]) {
|
|
r_line_hist_free();
|
|
r_line_hist_save (R2_HOME_HISTORY);
|
|
} else {
|
|
r_line_hist_free();
|
|
}
|
|
break;
|
|
case '=': //!=
|
|
if (input[1] == '?') {
|
|
r_cons_printf ("Usage: !=[!] - enable/disable remote commands\n");
|
|
} else {
|
|
if (!r_sandbox_enable (0)) {
|
|
R_FREE (core->cmdremote);
|
|
}
|
|
}
|
|
break;
|
|
case '!': //!!
|
|
if (input[1] == '!') { // !!! & !!!-
|
|
cmd_autocomplete (core, input + 2);
|
|
} else if (input[1] == '?') {
|
|
cmd_help_exclamation (core);
|
|
} else if (input[1] == '*') {
|
|
char *cmd = r_str_trim_dup (input + 1);
|
|
(void)r_core_cmdf (core, "\"#!pipe %s\"", cmd);
|
|
free (cmd);
|
|
} else {
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("This command is disabled in sandbox mode\n");
|
|
return 0;
|
|
}
|
|
if (input[1]) {
|
|
int olen;
|
|
char *out = NULL;
|
|
char *cmd = r_core_sysenv_begin (core, input);
|
|
if (cmd) {
|
|
void *bed = r_cons_sleep_begin ();
|
|
ret = r_sys_cmd_str_full (cmd + 1, NULL, 0, &out, &olen, NULL);
|
|
r_cons_sleep_end (bed);
|
|
r_core_sysenv_end (core, input);
|
|
r_cons_write (out, olen);
|
|
free (out);
|
|
free (cmd);
|
|
} //else eprintf ("Error setting up system environment\n");
|
|
} else {
|
|
eprintf ("History saved to "R2_HOME_HISTORY"\n");
|
|
r_line_hist_save (R2_HOME_HISTORY);
|
|
}
|
|
}
|
|
break;
|
|
case '\0':
|
|
r_line_hist_list ();
|
|
break;
|
|
case '?': //!?
|
|
cmd_help_exclamation (core);
|
|
break;
|
|
case '*':
|
|
// TODO: use the api
|
|
{
|
|
char *cmd = r_str_trim_dup (input + 1);
|
|
cmd = r_str_replace (cmd, " ", "\\ ", true);
|
|
cmd = r_str_replace (cmd, "\\ ", " ", false);
|
|
cmd = r_str_replace (cmd, "\"", "'", false);
|
|
ret = r_core_cmdf (core, "\"#!pipe %s\"", cmd);
|
|
free (cmd);
|
|
}
|
|
break;
|
|
default:
|
|
n = atoi (input);
|
|
if (*input == '0' || n > 0) {
|
|
const char *cmd = r_line_hist_get (n);
|
|
if (cmd) {
|
|
r_core_cmd0 (core, cmd);
|
|
}
|
|
//else eprintf ("Error setting up system environment\n");
|
|
} else {
|
|
char *cmd = r_core_sysenv_begin (core, input);
|
|
if (cmd) {
|
|
void *bed = r_cons_sleep_begin ();
|
|
ret = r_sys_cmd (cmd);
|
|
r_cons_sleep_end (bed);
|
|
r_core_sysenv_end (core, input);
|
|
free (cmd);
|
|
} else {
|
|
eprintf ("Error setting up system environment\n");
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static char *unescape_special_chars(const char *s, const char *special_chars) {
|
|
char *dst = R_NEWS (char, strlen (s) + 1);
|
|
int i, j = 0;
|
|
|
|
for (i = 0; s[i]; i++) {
|
|
if (s[i] != '\\' || !strchr (special_chars, s[i + 1])) {
|
|
dst[j++] = s[i];
|
|
continue;
|
|
}
|
|
dst[j++] = s[i + 1];
|
|
i++;
|
|
}
|
|
dst[j++] = '\0';
|
|
return dst;
|
|
}
|
|
|
|
#if __WINDOWS__
|
|
#include <tchar.h>
|
|
#define __CLOSE_DUPPED_PIPES() \
|
|
close (1); \
|
|
close (fd_out); \
|
|
fd_out = -1;
|
|
|
|
static void r_w32_cmd_pipe(RCore *core, char *radare_cmd, char *shell_cmd) {
|
|
STARTUPINFO si = {0};
|
|
PROCESS_INFORMATION pi = {0};
|
|
SECURITY_ATTRIBUTES sa;
|
|
HANDLE pipe[2] = {NULL, NULL};
|
|
int fd_out = -1, cons_out = -1;
|
|
char *_shell_cmd = NULL;
|
|
LPTSTR _shell_cmd_ = NULL;
|
|
DWORD mode;
|
|
TCHAR *systemdir = NULL;
|
|
GetConsoleMode (GetStdHandle (STD_OUTPUT_HANDLE), &mode);
|
|
|
|
sa.nLength = sizeof (SECURITY_ATTRIBUTES);
|
|
sa.bInheritHandle = TRUE;
|
|
sa.lpSecurityDescriptor = NULL;
|
|
if (!CreatePipe (&pipe[0], &pipe[1], &sa, 0)) {
|
|
r_sys_perror ("r_w32_cmd_pipe/CreatePipe");
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
if (!SetHandleInformation (pipe[1], HANDLE_FLAG_INHERIT, 0)) {
|
|
r_sys_perror ("r_w32_cmd_pipe/SetHandleInformation");
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
si.hStdError = GetStdHandle (STD_ERROR_HANDLE);
|
|
si.hStdOutput = GetStdHandle (STD_OUTPUT_HANDLE);
|
|
si.hStdInput = pipe[0];
|
|
si.dwFlags |= STARTF_USESTDHANDLES;
|
|
si.cb = sizeof (si);
|
|
_shell_cmd = shell_cmd;
|
|
while (*_shell_cmd && isspace ((ut8)*_shell_cmd)) {
|
|
_shell_cmd++;
|
|
}
|
|
char *tmp = r_str_newf ("/Q /c \"%s\"", _shell_cmd);
|
|
if (!tmp) {
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
_shell_cmd = tmp;
|
|
_shell_cmd_ = r_sys_conv_utf8_to_win (_shell_cmd);
|
|
free (tmp);
|
|
if (!_shell_cmd_) {
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
systemdir = calloc (MAX_PATH, sizeof (TCHAR));
|
|
if (!systemdir) {
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
int ret = GetSystemDirectory (systemdir, MAX_PATH);
|
|
if (!ret) {
|
|
r_sys_perror ("r_w32_cmd_pipe/systemdir");
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
_tcscat_s (systemdir, MAX_PATH, TEXT("\\cmd.exe"));
|
|
// exec windows process
|
|
if (!CreateProcess (systemdir, _shell_cmd_, NULL, NULL, TRUE, 0, NULL, NULL, &si, &pi)) {
|
|
r_sys_perror ("r_w32_cmd_pipe/CreateProcess");
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
fd_out = _open_osfhandle ((intptr_t)pipe[1], _O_WRONLY|_O_TEXT);
|
|
if (fd_out == -1) {
|
|
perror ("_open_osfhandle");
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
cons_out = dup (1);
|
|
dup2 (fd_out, 1);
|
|
// exec radare command
|
|
r_core_cmd (core, radare_cmd, 0);
|
|
|
|
HANDLE th = CreateThread (NULL, 0,(LPTHREAD_START_ROUTINE) r_cons_flush, NULL, 0, NULL);
|
|
if (!th) {
|
|
__CLOSE_DUPPED_PIPES ();
|
|
goto err_r_w32_cmd_pipe;
|
|
}
|
|
while (true) {
|
|
int ret = WaitForSingleObject (th, 50);
|
|
if (!ret) {
|
|
// Successfully written everything to pipe
|
|
__CLOSE_DUPPED_PIPES ();
|
|
WaitForSingleObject (pi.hProcess, INFINITE);
|
|
break;
|
|
}
|
|
ret = WaitForSingleObject (pi.hProcess, 50);
|
|
if (!ret) {
|
|
// Process exited before we finished writing to pipe
|
|
DWORD exit;
|
|
if (GetExitCodeThread (th, &exit) && exit == STILL_ACTIVE) {
|
|
CancelSynchronousIo (th);
|
|
}
|
|
WaitForSingleObject (th, INFINITE);
|
|
__CLOSE_DUPPED_PIPES ();
|
|
break;
|
|
}
|
|
}
|
|
CloseHandle (th);
|
|
err_r_w32_cmd_pipe:
|
|
if (pi.hProcess) {
|
|
CloseHandle (pi.hProcess);
|
|
}
|
|
if (pi.hThread) {
|
|
CloseHandle (pi.hThread);
|
|
}
|
|
if (pipe[0]) {
|
|
CloseHandle (pipe[0]);
|
|
}
|
|
if (fd_out != -1) {
|
|
close (fd_out);
|
|
}
|
|
if (cons_out != -1) {
|
|
dup2 (cons_out, 1);
|
|
close (cons_out);
|
|
}
|
|
free (systemdir);
|
|
free (_shell_cmd_);
|
|
SetConsoleMode (GetStdHandle (STD_OUTPUT_HANDLE), mode);
|
|
}
|
|
#undef __CLOSE_DUPPED_PIPES
|
|
#endif
|
|
|
|
R_API int r_core_cmd_pipe(RCore *core, char *radare_cmd, char *shell_cmd) {
|
|
#if __UNIX__
|
|
int stdout_fd, fds[2];
|
|
int child;
|
|
#endif
|
|
int si, olen, ret = -1, pipecolor = -1;
|
|
char *str, *out = NULL;
|
|
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("Pipes are not allowed in sandbox mode\n");
|
|
return -1;
|
|
}
|
|
si = r_cons_is_interactive ();
|
|
r_config_set_b (core->config, "scr.interactive", false);
|
|
if (!r_config_get_i (core->config, "scr.color.pipe")) {
|
|
pipecolor = r_config_get_i (core->config, "scr.color");
|
|
r_config_set_i (core->config, "scr.color", COLOR_MODE_DISABLED);
|
|
}
|
|
if (*shell_cmd=='!') {
|
|
r_cons_grep_parsecmd (shell_cmd, "\"");
|
|
olen = 0;
|
|
out = NULL;
|
|
// TODO: implement foo
|
|
str = r_core_cmd_str (core, radare_cmd);
|
|
r_sys_cmd_str_full (shell_cmd + 1, str, -1, &out, &olen, NULL);
|
|
free (str);
|
|
r_cons_write (out, olen);
|
|
free (out);
|
|
ret = 0;
|
|
}
|
|
#if __UNIX__
|
|
r_str_trim_head (radare_cmd);
|
|
r_str_trim_head (shell_cmd);
|
|
|
|
r_sys_signal (SIGPIPE, SIG_IGN);
|
|
stdout_fd = dup (1);
|
|
if (stdout_fd != -1) {
|
|
if (pipe (fds) == 0) {
|
|
child = r_sys_fork ();
|
|
if (child == -1) {
|
|
eprintf ("Cannot fork\n");
|
|
} else if (child) {
|
|
dup2 (fds[1], 1);
|
|
close (fds[1]);
|
|
close (fds[0]);
|
|
r_core_cmd (core, radare_cmd, 0);
|
|
r_cons_flush ();
|
|
close (1);
|
|
wait (&ret);
|
|
dup2 (stdout_fd, 1);
|
|
} else {
|
|
close (fds[1]);
|
|
dup2 (fds[0], 0);
|
|
//dup2 (1, 2); // stderr goes to stdout
|
|
r_sandbox_system (shell_cmd, 0);
|
|
}
|
|
} else {
|
|
eprintf ("r_core_cmd_pipe: Could not pipe\n");
|
|
}
|
|
close (stdout_fd);
|
|
}
|
|
#elif __WINDOWS__
|
|
r_w32_cmd_pipe (core, radare_cmd, shell_cmd);
|
|
#else
|
|
#ifdef _MSC_VER
|
|
#pragma message ("r_core_cmd_pipe UNIMPLEMENTED FOR THIS PLATFORM")
|
|
#else
|
|
#warning r_core_cmd_pipe UNIMPLEMENTED FOR THIS PLATFORM
|
|
#endif
|
|
eprintf ("r_core_cmd_pipe: unimplemented for this platform\n");
|
|
#endif
|
|
if (pipecolor != -1) {
|
|
r_config_set_i (core->config, "scr.color", pipecolor);
|
|
}
|
|
r_config_set_i (core->config, "scr.interactive", si);
|
|
return ret;
|
|
}
|
|
|
|
static char *parse_tmp_evals(RCore *core, const char *str) {
|
|
char *s = strdup (str);
|
|
int i, argc = r_str_split (s, ',');
|
|
char *res = strdup ("");
|
|
if (!s || !res) {
|
|
free (s);
|
|
free (res);
|
|
return NULL;
|
|
}
|
|
for (i = 0; i < argc; i++) {
|
|
char *eq, *kv = (char *)r_str_word_get0 (s, i);
|
|
if (!kv) {
|
|
break;
|
|
}
|
|
eq = strchr (kv, '=');
|
|
if (eq) {
|
|
*eq = 0;
|
|
const char *ov = r_config_get (core->config, kv);
|
|
if (!ov) {
|
|
continue;
|
|
}
|
|
char *cmd = r_str_newf ("e %s=%s;", kv, ov);
|
|
if (!cmd) {
|
|
free (s);
|
|
free (res);
|
|
return NULL;
|
|
}
|
|
res = r_str_prepend (res, cmd);
|
|
free (cmd);
|
|
r_config_set (core->config, kv, eq + 1);
|
|
*eq = '=';
|
|
} else {
|
|
eprintf ("Missing '=' in e: expression (%s)\n", kv);
|
|
}
|
|
}
|
|
free (s);
|
|
return res;
|
|
}
|
|
|
|
static bool is_macro_command(const char *ptr) {
|
|
ptr = r_str_trim_head_ro (ptr);
|
|
while (IS_DIGIT (*ptr)) {
|
|
ptr++;
|
|
}
|
|
return *ptr == '(';
|
|
}
|
|
|
|
static char *find_ch_after_macro(char *ptr, char ch) {
|
|
int depth = 0;
|
|
while (*ptr) {
|
|
if (depth == 0 && *ptr == ch) {
|
|
return ptr;
|
|
}
|
|
if (*ptr == '(') {
|
|
depth++;
|
|
} else if (*ptr == ')') {
|
|
depth--;
|
|
}
|
|
ptr++;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
static int r_core_cmd_subst(RCore *core, char *cmd) {
|
|
ut64 rep = strtoull (cmd, NULL, 10);
|
|
int ret = 0, orep;
|
|
char *colon = NULL, *icmd = NULL;
|
|
bool tmpseek = false;
|
|
bool original_tmpseek = core->tmpseek;
|
|
|
|
if (r_str_startswith (cmd, "GET /cmd/")) {
|
|
memmove (cmd, cmd + 9, strlen (cmd + 9) + 1);
|
|
char *http = strstr (cmd, "HTTP");
|
|
if (http) {
|
|
*http = 0;
|
|
http--;
|
|
if (*http == ' ') {
|
|
*http = 0;
|
|
}
|
|
}
|
|
r_cons_printf ("HTTP/1.0 %d %s\r\n%s"
|
|
"Connection: close\r\nContent-Length: %d\r\n\r\n",
|
|
200, "OK", "", -1);
|
|
return r_core_cmd0 (core, cmd);
|
|
}
|
|
|
|
/* must store a local orig_offset because there can be
|
|
* nested call of this function */
|
|
ut64 orig_offset = core->offset;
|
|
icmd = strdup (cmd);
|
|
if (!icmd) {
|
|
goto beach;
|
|
}
|
|
|
|
if (core->max_cmd_depth - core->cons->context->cmd_depth == 1) {
|
|
core->prompt_offset = core->offset;
|
|
}
|
|
cmd = (char *)r_str_trim_head_ro (icmd);
|
|
r_str_trim_tail (cmd);
|
|
// lines starting with # are ignored (never reach cmd_hash()), except #! and #?
|
|
if (!*cmd) {
|
|
if (core->cmdrepeat > 0) {
|
|
lastcmd_repeat (core, true);
|
|
ret = r_core_cmd_nullcallback (core);
|
|
}
|
|
goto beach;
|
|
}
|
|
if (!icmd || (cmd[0] == '#' && cmd[1] != '!' && cmd[1] != '?')) {
|
|
goto beach;
|
|
}
|
|
if (*cmd) {
|
|
int inside_quote = 0;
|
|
char *hash;
|
|
for (hash = cmd; *hash; hash++) {
|
|
if (*hash == '\\') {
|
|
hash++;
|
|
if (*hash == '#' || *hash == '"') {
|
|
continue;
|
|
} else if (!*hash) {
|
|
break;
|
|
}
|
|
}
|
|
if (*hash == '"') {
|
|
inside_quote ^= 1;
|
|
}
|
|
if (*hash == '#' && hash != cmd && !inside_quote) {
|
|
break;
|
|
}
|
|
}
|
|
if (hash && *hash && !inside_quote) {
|
|
*hash = 0;
|
|
r_str_trim_tail (cmd);
|
|
}
|
|
}
|
|
if (*cmd != '"') {
|
|
if (!strchr (cmd, '\'')) { // allow | awk '{foo;bar}' // ignore ; if there's a single quote
|
|
if (is_macro_command (cmd)) {
|
|
colon = find_ch_after_macro (cmd, ';');
|
|
} else {
|
|
colon = strchr (cmd, ';');
|
|
}
|
|
if (colon) {
|
|
*colon = 0;
|
|
}
|
|
}
|
|
} else {
|
|
colon = NULL;
|
|
}
|
|
if (rep > 0) {
|
|
while (IS_DIGIT (*cmd)) {
|
|
cmd++;
|
|
}
|
|
// do not repeat null cmd
|
|
if (!*cmd) {
|
|
goto beach;
|
|
}
|
|
}
|
|
if (rep < 1) {
|
|
rep = 1;
|
|
}
|
|
// XXX if output is a pipe then we don't want to be interactive
|
|
if (rep > 1 && r_sandbox_enable (0)) {
|
|
eprintf ("Command repeat sugar disabled in sandbox mode (%s)\n", cmd);
|
|
goto beach;
|
|
} else {
|
|
if (rep > INTERACTIVE_MAX_REP) {
|
|
if (r_cons_is_interactive ()) {
|
|
if (!r_cons_yesno ('n', "Are you sure to repeat this %"PFMT64d" times? (y/N)", rep)) {
|
|
goto beach;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// TODO: store in core->cmdtimes to speedup ?
|
|
const char *cmdrep = r_str_get (core->cmdtimes);
|
|
orep = rep;
|
|
|
|
r_cons_break_push (NULL, NULL);
|
|
|
|
int ocur_enabled = core->print && core->print->cur_enabled;
|
|
while (rep-- && *cmd) {
|
|
if (core->print) {
|
|
core->print->cur_enabled = false;
|
|
if (ocur_enabled && core->seltab >= 0) {
|
|
if (core->seltab == core->curtab) {
|
|
core->print->cur_enabled = true;
|
|
}
|
|
}
|
|
}
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
char *cr = strdup (cmdrep);
|
|
core->break_loop = false;
|
|
ret = r_core_cmd_subst_i (core, cmd, colon, (rep == orep - 1) ? &tmpseek : NULL);
|
|
if (*cmd == 's') {
|
|
// do not restore tmpseek if the command executed is the 's'eek
|
|
tmpseek = false;
|
|
}
|
|
if (ret && *cmd == 'q') {
|
|
free (cr);
|
|
goto beach;
|
|
}
|
|
if (core->break_loop) {
|
|
free (cr);
|
|
break;
|
|
}
|
|
if (colon) {
|
|
r_cons_flush ();
|
|
}
|
|
if (cr && *cr && orep > 1) {
|
|
// XXX: do not flush here, we need r_cons_push () and r_cons_pop()
|
|
r_cons_flush ();
|
|
// XXX: we must import register flags in C
|
|
(void)r_core_cmd0 (core, ".dr*");
|
|
(void)r_core_cmd0 (core, cr);
|
|
}
|
|
free (cr);
|
|
}
|
|
|
|
r_cons_break_pop ();
|
|
|
|
if (tmpseek) {
|
|
r_core_seek (core, orig_offset, true);
|
|
core->tmpseek = original_tmpseek;
|
|
}
|
|
if (core->print) {
|
|
core->print->cur_enabled = ocur_enabled;
|
|
}
|
|
if (colon && colon[1]) {
|
|
for (++colon; *colon == ';'; colon++) {
|
|
;
|
|
}
|
|
r_core_cmd_subst (core, colon);
|
|
} else {
|
|
if (!*icmd) {
|
|
r_core_cmd_nullcallback (core);
|
|
}
|
|
}
|
|
beach:
|
|
free (icmd);
|
|
return ret;
|
|
}
|
|
|
|
static char *find_eoq(char *p) {
|
|
for (; *p; p++) {
|
|
if (*p == '"') {
|
|
break;
|
|
}
|
|
if (*p == '\\' && p[1] == '"') {
|
|
p++;
|
|
}
|
|
}
|
|
return p;
|
|
}
|
|
|
|
static char* findSeparator(char *p) {
|
|
char *q = strchr (p, '+');
|
|
if (q) {
|
|
return q;
|
|
}
|
|
return strchr (p, '-');
|
|
}
|
|
|
|
static void tmpenvs_free(void *item) {
|
|
r_sys_setenv (item, NULL);
|
|
free (item);
|
|
}
|
|
|
|
static bool set_tmp_arch(RCore *core, char *arch, char **tmparch) {
|
|
r_return_val_if_fail (tmparch, false);
|
|
*tmparch = strdup (r_config_get (core->config, "asm.arch"));
|
|
r_config_set (core->config, "asm.arch", arch);
|
|
core->fixedarch = true;
|
|
return true;
|
|
}
|
|
|
|
static bool set_tmp_bits(RCore *core, int bits, char **tmpbits, int *cmd_ignbithints) {
|
|
r_return_val_if_fail (tmpbits, false);
|
|
*tmpbits = strdup (r_config_get (core->config, "asm.bits"));
|
|
r_config_set_i (core->config, "asm.bits", bits);
|
|
core->fixedbits = true;
|
|
// XXX: why?
|
|
*cmd_ignbithints = r_config_get_i (core->config, "anal.ignbithints");
|
|
r_config_set_b (core->config, "anal.ignbithints", true);
|
|
return true;
|
|
}
|
|
|
|
static int r_core_cmd_subst_i(RCore *core, char *cmd, char *colon, bool *tmpseek) {
|
|
RList *tmpenvs = r_list_newf (tmpenvs_free);
|
|
const char *quotestr = "`";
|
|
const char *tick = NULL;
|
|
char *ptr, *ptr2, *str;
|
|
char *arroba = NULL;
|
|
char *grep = NULL;
|
|
RIODesc *tmpdesc = NULL;
|
|
int pamode = !core->io->va;
|
|
int i, ret = 0, pipefd;
|
|
bool usemyblock = false;
|
|
int scr_html = -1;
|
|
int scr_color = -1;
|
|
bool eos = false;
|
|
bool haveQuote = false;
|
|
bool oldfixedarch = core->fixedarch;
|
|
bool oldfixedbits = core->fixedbits;
|
|
bool cmd_tmpseek = false;
|
|
ut64 tmpbsz = core->blocksize;
|
|
int cmd_ignbithints = -1;
|
|
|
|
if (!cmd) {
|
|
r_list_free (tmpenvs);
|
|
return 0;
|
|
}
|
|
r_str_trim (cmd);
|
|
|
|
char *$0 = strstr (cmd, "$(");
|
|
if ($0) {
|
|
char *$1 = strchr ($0 + 2, ')');
|
|
if ($1) {
|
|
*$0 = '`';
|
|
*$1 = '`';
|
|
memmove ($0 + 1, $0 + 2, strlen ($0 + 2) + 1);
|
|
} else {
|
|
eprintf ("Unterminated $() block\n");
|
|
}
|
|
}
|
|
|
|
/* quoted / raw command */
|
|
switch (*cmd) {
|
|
case '.':
|
|
if (cmd[1] == '"') { /* interpret */
|
|
r_list_free (tmpenvs);
|
|
return r_cmd_call (core->rcmd, cmd);
|
|
}
|
|
break;
|
|
case '"':
|
|
for (; *cmd; ) {
|
|
int pipefd = -1;
|
|
ut64 oseek = UT64_MAX;
|
|
char *line, *p;
|
|
haveQuote = *cmd == '"';
|
|
if (haveQuote) {
|
|
cmd++;
|
|
p = *cmd ? find_eoq (cmd) : NULL;
|
|
if (!p || !*p) {
|
|
eprintf ("Missing \" in (%s).", cmd);
|
|
r_list_free (tmpenvs);
|
|
return false;
|
|
}
|
|
*p++ = 0;
|
|
if (!*p) {
|
|
eos = true;
|
|
}
|
|
} else {
|
|
char *sc = strchr (cmd, ';');
|
|
if (sc) {
|
|
*sc = 0;
|
|
}
|
|
r_core_cmd0 (core, cmd);
|
|
if (!sc) {
|
|
break;
|
|
}
|
|
cmd = sc + 1;
|
|
continue;
|
|
}
|
|
char op0 = 0;
|
|
if (*p) {
|
|
// workaround :D
|
|
if (p[0] == '@') {
|
|
p--;
|
|
}
|
|
while (p[1] == ';' || IS_WHITESPACE (p[1])) {
|
|
p++;
|
|
}
|
|
if (p[1] == '@' || (p[1] && p[2] == '@')) {
|
|
char *q = strchr (p + 1, '"');
|
|
if (q) {
|
|
op0 = *q;
|
|
*q = 0;
|
|
}
|
|
haveQuote = q != NULL;
|
|
oseek = core->offset;
|
|
r_core_seek (core, r_num_math (core->num, p + 2), true);
|
|
if (q) {
|
|
*p = '"';
|
|
p = q;
|
|
} else {
|
|
p = strchr (p + 1, ';');
|
|
}
|
|
}
|
|
if (p && *p && p[1] == '>') {
|
|
str = p + 2;
|
|
while (*str == '>') {
|
|
str++;
|
|
}
|
|
str = (char *)r_str_trim_head_ro (str);
|
|
r_cons_flush ();
|
|
const bool append = p[2] == '>';
|
|
pipefd = r_cons_pipe_open (str, 1, append);
|
|
}
|
|
}
|
|
line = strdup (cmd);
|
|
line = r_str_replace (line, "\\\"", "\"", true);
|
|
if (p && *p && p[1] == '|') {
|
|
str = (char *)r_str_trim_head_ro (p + 2);
|
|
r_core_cmd_pipe (core, cmd, str);
|
|
} else {
|
|
r_cmd_call (core->rcmd, line);
|
|
}
|
|
free (line);
|
|
if (oseek != UT64_MAX) {
|
|
r_core_seek (core, oseek, true);
|
|
}
|
|
if (pipefd != -1) {
|
|
r_cons_flush ();
|
|
r_cons_pipe_close (pipefd);
|
|
}
|
|
if (!p) {
|
|
break;
|
|
}
|
|
if (eos) {
|
|
break;
|
|
}
|
|
if (haveQuote) {
|
|
if (*p == ';') {
|
|
cmd = p + 1;
|
|
} else {
|
|
if (*p == '"') {
|
|
cmd = p;
|
|
} else {
|
|
*p = op0;
|
|
cmd = p;
|
|
}
|
|
}
|
|
} else {
|
|
cmd = p + 1;
|
|
}
|
|
}
|
|
r_list_free (tmpenvs);
|
|
return true;
|
|
case '(':
|
|
if (cmd[1] != '*' && !strstr (cmd, ")()")) {
|
|
r_list_free (tmpenvs);
|
|
return r_cmd_call (core->rcmd, cmd);
|
|
}
|
|
break;
|
|
case '?':
|
|
if (cmd[1] == '>') {
|
|
r_core_cmd_help (core, help_msg_greater_sign);
|
|
r_list_free (tmpenvs);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
// TODO must honor `
|
|
/* comments */
|
|
if (*cmd != '#') {
|
|
ptr = (char *)r_str_firstbut (cmd, '#', "`\""); // TODO: use quotestr here
|
|
if (ptr && (ptr[1] == ' ' || ptr[1] == '\t')) {
|
|
*ptr = '\0';
|
|
}
|
|
}
|
|
|
|
/* multiple commands */
|
|
// TODO: must honor " and ` boundaries
|
|
//ptr = strrchr (cmd, ';');
|
|
if (*cmd != '#') {
|
|
if (is_macro_command (cmd)) {
|
|
ptr = find_ch_after_macro (cmd, ';');
|
|
} else {
|
|
ptr = (char *)r_str_lastbut (cmd, ';', quotestr);
|
|
}
|
|
if (colon && ptr) {
|
|
int ret ;
|
|
*ptr = '\0';
|
|
if (r_core_cmd_subst (core, cmd) == -1) {
|
|
r_list_free (tmpenvs);
|
|
return -1;
|
|
}
|
|
cmd = ptr + 1;
|
|
ret = r_core_cmd_subst (core, cmd);
|
|
*ptr = ';';
|
|
r_list_free (tmpenvs);
|
|
return ret;
|
|
//r_cons_flush ();
|
|
}
|
|
}
|
|
|
|
// TODO must honor " and `
|
|
/* pipe console to shell process */
|
|
//ptr = strchr (cmd, '|');
|
|
ptr = (char *)r_str_lastbut (cmd, '|', quotestr);
|
|
if (ptr) {
|
|
if (ptr > cmd) {
|
|
char *ch = ptr - 1;
|
|
if (*ch == '\\') {
|
|
memmove (ch, ptr, strlen (ptr) + 1);
|
|
goto escape_pipe;
|
|
}
|
|
}
|
|
char *ptr2 = strchr (cmd, '`');
|
|
if (!ptr2 || (ptr2 && ptr2 > ptr)) {
|
|
if (!tick || (tick && tick > ptr)) {
|
|
*ptr = '\0';
|
|
cmd = r_str_trim_nc (cmd);
|
|
if (!strcmp (ptr + 1, "?")) { // "|?"
|
|
r_core_cmd_help (core, help_msg_vertical_bar);
|
|
r_list_free (tmpenvs);
|
|
return ret;
|
|
} else if (!strncmp (ptr + 1, "H", 1)) { // "|H"
|
|
scr_html = r_config_get_b (core->config, "scr.html");
|
|
r_config_set_b (core->config, "scr.html", true);
|
|
} else if (!strcmp (ptr + 1, "T")) { // "|T"
|
|
scr_color = r_config_get_i (core->config, "scr.color");
|
|
r_config_set_i (core->config, "scr.color", COLOR_MODE_DISABLED);
|
|
core->cons->use_tts = true;
|
|
} else if (!strcmp (ptr + 1, ".")) { // "|."
|
|
ret = *cmd ? r_core_cmdf (core, ".%s", cmd) : 0;
|
|
r_list_free (tmpenvs);
|
|
return ret;
|
|
} else if (ptr[1]) { // "| grep .."
|
|
int value = core->num->value;
|
|
if (*cmd) {
|
|
r_core_cmd_pipe (core, cmd, ptr + 1);
|
|
} else {
|
|
char *res = r_io_system (core->io, ptr + 1);
|
|
if (res) {
|
|
r_cons_printf ("%s\n", res);
|
|
free (res);
|
|
}
|
|
}
|
|
core->num->value = value;
|
|
r_list_free (tmpenvs);
|
|
return 0;
|
|
} else { // "|"
|
|
scr_html = r_config_get_b (core->config, "scr.html");
|
|
r_config_set_b (core->config, "scr.html", false);
|
|
scr_color = r_config_get_i (core->config, "scr.color");
|
|
r_config_set_i (core->config, "scr.color", COLOR_MODE_DISABLED);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
escape_pipe:
|
|
|
|
// TODO must honor " and `
|
|
/* bool conditions */
|
|
ptr = (char *)r_str_lastbut (cmd, '&', quotestr);
|
|
//ptr = strchr (cmd, '&');
|
|
while (ptr && *ptr && ptr[1] == '&') {
|
|
*ptr = '\0';
|
|
ret = r_cmd_call (core->rcmd, cmd);
|
|
if (ret == -1) {
|
|
eprintf ("command error(%s)\n", cmd);
|
|
if (scr_html != -1) {
|
|
r_config_set_b (core->config, "scr.html", scr_html);
|
|
}
|
|
if (scr_color != -1) {
|
|
r_config_set_i (core->config, "scr.color", scr_color);
|
|
}
|
|
r_list_free (tmpenvs);
|
|
return ret;
|
|
}
|
|
for (cmd = ptr + 2; cmd && *cmd == ' '; cmd++) {
|
|
;
|
|
}
|
|
ptr = strchr (cmd, '&');
|
|
}
|
|
|
|
ptr = strstr (cmd, "?*");
|
|
if (ptr && (ptr == cmd || ptr[-1] != '~')) {
|
|
ptr[0] = 0;
|
|
if (*cmd != '#') {
|
|
int detail = 0;
|
|
if (cmd < ptr && ptr[-1] == '?') {
|
|
detail++;
|
|
if (cmd < ptr - 1 && ptr[-2] == '?') {
|
|
detail++;
|
|
}
|
|
}
|
|
r_cons_break_push (NULL, NULL);
|
|
recursive_help (core, detail, cmd);
|
|
r_cons_break_pop ();
|
|
r_cons_grep_parsecmd (ptr + 2, "`");
|
|
if (scr_html != -1) {
|
|
r_config_set_b (core->config, "scr.html", scr_html);
|
|
}
|
|
if (scr_color != -1) {
|
|
r_config_set_i (core->config, "scr.color", scr_color);
|
|
}
|
|
r_list_free (tmpenvs);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
/* pipe console to file */
|
|
ptr = (char *)r_str_firstbut (cmd, '>', "\"");
|
|
// TODO honor `
|
|
if (ptr) {
|
|
if (ptr > cmd) {
|
|
char *ch = ptr - 1;
|
|
if (*ch == '\\') {
|
|
memmove (ch, ptr, strlen (ptr) + 1);
|
|
goto escape_redir;
|
|
}
|
|
}
|
|
if (ptr[0] && ptr[1] == '?') {
|
|
r_core_cmd_help (core, help_msg_greater_sign);
|
|
r_list_free (tmpenvs);
|
|
return true;
|
|
}
|
|
int fdn = 1;
|
|
int pipecolor = r_config_get_i (core->config, "scr.color.pipe");
|
|
int use_editor = false;
|
|
int ocolor = r_config_get_i (core->config, "scr.color");
|
|
*ptr = '\0';
|
|
str = ptr + 1 + (ptr[1] == '>');
|
|
r_str_trim (str);
|
|
if (!*str) {
|
|
eprintf ("No output?\n");
|
|
goto next2;
|
|
}
|
|
/* r_cons_flush() handles interactive output (to the terminal)
|
|
* differently (e.g. asking about too long output). This conflicts
|
|
* with piping to a file. Disable it while piping. */
|
|
if (ptr > (cmd + 1) && IS_WHITECHAR (ptr[-2])) {
|
|
char *fdnum = ptr - 1;
|
|
if (*fdnum == 'H') { // "H>"
|
|
scr_html = r_config_get_i (core->config, "scr.html");
|
|
r_config_set_i (core->config, "scr.html", true);
|
|
pipecolor = true;
|
|
*fdnum = 0;
|
|
} else {
|
|
if (IS_DIGIT (*fdnum)) {
|
|
fdn = *fdnum - '0';
|
|
}
|
|
*fdnum = 0;
|
|
}
|
|
}
|
|
r_cons_set_interactive (false);
|
|
if (!strcmp (str, "-")) {
|
|
use_editor = true;
|
|
str = r_file_temp ("dumpedit");
|
|
r_config_set_i (core->config, "scr.color", COLOR_MODE_DISABLED);
|
|
}
|
|
const bool appendResult = (ptr[1] == '>');
|
|
if (*str == '$') {
|
|
// pipe to alias variable
|
|
// register output of command as an alias
|
|
char *o = r_core_cmd_str (core, cmd);
|
|
if (appendResult) {
|
|
const char *oldText = r_cmd_alias_get (core->rcmd, str, 1);
|
|
if (oldText) {
|
|
char *two = r_str_newf ("%s%s", oldText, o);
|
|
if (two) {
|
|
r_cmd_alias_set (core->rcmd, str, two, 1);
|
|
free (two);
|
|
}
|
|
} else {
|
|
char *n = r_str_newf ("$%s", o);
|
|
r_cmd_alias_set (core->rcmd, str, n, 1);
|
|
free (n);
|
|
}
|
|
} else {
|
|
char *n = r_str_newf ("$%s", o);
|
|
r_cmd_alias_set (core->rcmd, str, n, 1);
|
|
free (n);
|
|
}
|
|
ret = 0;
|
|
free (o);
|
|
} else if (fdn > 0) {
|
|
// pipe to file (or append)
|
|
pipefd = r_cons_pipe_open (str, fdn, appendResult);
|
|
if (pipefd != -1) {
|
|
if (!pipecolor) {
|
|
r_config_set_i (core->config, "scr.color", COLOR_MODE_DISABLED);
|
|
}
|
|
ret = r_core_cmd_subst (core, cmd);
|
|
r_cons_flush ();
|
|
r_cons_pipe_close (pipefd);
|
|
}
|
|
}
|
|
r_cons_set_last_interactive ();
|
|
if (!pipecolor) {
|
|
r_config_set_i (core->config, "scr.color", ocolor);
|
|
}
|
|
if (use_editor) {
|
|
const char *editor = r_config_get (core->config, "cfg.editor");
|
|
if (editor && *editor) {
|
|
r_sys_cmdf ("%s '%s'", editor, str);
|
|
r_file_rm (str);
|
|
} else {
|
|
eprintf ("No cfg.editor configured\n");
|
|
}
|
|
r_config_set_i (core->config, "scr.color", ocolor);
|
|
free (str);
|
|
}
|
|
if (scr_html != -1) {
|
|
r_config_set_i (core->config, "scr.html", scr_html);
|
|
}
|
|
if (scr_color != -1) {
|
|
r_config_set_i (core->config, "scr.color", scr_color);
|
|
}
|
|
core->cons->use_tts = false;
|
|
r_list_free (tmpenvs);
|
|
return ret;
|
|
}
|
|
escape_redir:
|
|
next2:
|
|
/* sub commands */
|
|
ptr = strchr (cmd, '`');
|
|
if (ptr) {
|
|
if (ptr > cmd) {
|
|
char *ch = ptr - 1;
|
|
if (*ch == '\\') {
|
|
memmove (ch, ptr, strlen (ptr) + 1);
|
|
goto escape_backtick;
|
|
}
|
|
}
|
|
bool empty = false;
|
|
int oneline = 1;
|
|
if (ptr[1] == '`') {
|
|
memmove (ptr, ptr + 1, strlen (ptr));
|
|
oneline = 0;
|
|
empty = true;
|
|
}
|
|
ptr2 = strchr (ptr + 1, '`');
|
|
if (empty) {
|
|
/* do nothing */
|
|
} else if (!ptr2) {
|
|
eprintf ("parse: Missing backtick in expression.\n");
|
|
goto fail;
|
|
} else {
|
|
int value = core->num->value;
|
|
*ptr = '\0';
|
|
*ptr2 = '\0';
|
|
if (ptr[1] == '!') {
|
|
str = r_core_cmd_str_pipe (core, ptr + 1);
|
|
} else {
|
|
// Color disabled when doing backticks ?e `pi 1`
|
|
int ocolor = r_config_get_i (core->config, "scr.color");
|
|
r_config_set_i (core->config, "scr.color", 0);
|
|
core->cmd_in_backticks = true;
|
|
str = r_core_cmd_str (core, ptr + 1);
|
|
core->cmd_in_backticks = false;
|
|
r_config_set_i (core->config, "scr.color", ocolor);
|
|
}
|
|
if (!str) {
|
|
goto fail;
|
|
}
|
|
// ignore contents if first char is pipe or comment
|
|
if (*str == '|' || *str == '*') {
|
|
eprintf ("r_core_cmd_subst_i: invalid backticked command\n");
|
|
free (str);
|
|
goto fail;
|
|
}
|
|
if (oneline && str) {
|
|
for (i = 0; str[i]; i++) {
|
|
if (str[i] == '\n') {
|
|
str[i] = ' ';
|
|
}
|
|
}
|
|
}
|
|
str = r_str_append (str, ptr2 + 1);
|
|
cmd = r_str_append (strdup (cmd), str);
|
|
core->num->value = value;
|
|
ret = r_core_cmd_subst (core, cmd);
|
|
free (cmd);
|
|
if (scr_html != -1) {
|
|
r_config_set_i (core->config, "scr.html", scr_html);
|
|
}
|
|
free (str);
|
|
r_list_free (tmpenvs);
|
|
return ret;
|
|
}
|
|
}
|
|
escape_backtick:
|
|
// TODO must honor " and `
|
|
if (*cmd != '"' && *cmd) {
|
|
const char *s = strstr (cmd, "~?");
|
|
if (s) {
|
|
bool showHelp = false;
|
|
if (cmd == s) {
|
|
// ~?
|
|
// ~??
|
|
showHelp = true;
|
|
} else {
|
|
// pd~?
|
|
// pd~??
|
|
if (!strcmp (s, "~??")) {
|
|
showHelp = true;
|
|
}
|
|
}
|
|
if (showHelp) {
|
|
r_cons_grep_help ();
|
|
r_list_free (tmpenvs);
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
if (*cmd != '.') {
|
|
grep = r_cons_grep_strip (cmd, quotestr);
|
|
}
|
|
|
|
/* temporary seek commands */
|
|
// if (*cmd != '(' && *cmd != '"')
|
|
if (*cmd != '"') {
|
|
ptr = strchr (cmd, '@');
|
|
if (ptr == cmd + 1 && *cmd == '?') {
|
|
ptr = NULL;
|
|
}
|
|
} else {
|
|
ptr = NULL;
|
|
}
|
|
|
|
cmd_tmpseek = core->tmpseek = ptr != NULL;
|
|
int rc = 0;
|
|
if (ptr) {
|
|
char *f, *ptr2 = strchr (ptr + 1, '!');
|
|
ut64 addr = core->offset;
|
|
bool addr_is_set = false;
|
|
char *tmpbits = NULL;
|
|
const char *offstr = NULL;
|
|
bool is_bits_set = false;
|
|
bool is_arch_set = false;
|
|
char *tmpeval = NULL;
|
|
char *tmpasm = NULL;
|
|
bool flgspc_changed = false;
|
|
int tmpfd = -1;
|
|
size_t sz;
|
|
int len;
|
|
ut8 *buf;
|
|
|
|
*ptr++ = '\0';
|
|
repeat_arroba:
|
|
arroba = (ptr[0] && ptr[1] && ptr[2])?
|
|
strchr (ptr + 2, '@'): NULL;
|
|
if (arroba) {
|
|
*arroba = 0;
|
|
}
|
|
|
|
for (; *ptr == ' '; ptr++) {
|
|
//nothing to see here
|
|
}
|
|
if (*ptr && ptr[1] == ':') {
|
|
/* do nothing here */
|
|
} else {
|
|
ptr--;
|
|
}
|
|
|
|
r_str_trim_tail (ptr);
|
|
|
|
if (ptr[1] == '?') {
|
|
r_core_cmd_help (core, help_msg_at);
|
|
} else if (ptr[1] == '%') { // "@%"
|
|
char *k = strdup (ptr + 2);
|
|
char *v = strchr (k, '=');
|
|
if (v) {
|
|
*v++ = 0;
|
|
r_sys_setenv (k, v);
|
|
r_list_append (tmpenvs, k);
|
|
} else {
|
|
free (k);
|
|
}
|
|
} else if (ptr[1] == '.') { // "@."
|
|
if (ptr[2] == '.') { // "@.."
|
|
if (ptr[3] == '.') { // "@..."
|
|
ut64 addr = r_num_tail (core->num, core->offset, ptr + 4);
|
|
r_core_block_size (core, R_ABS ((st64)addr - (st64)core->offset));
|
|
goto fuji;
|
|
} else {
|
|
addr = r_num_tail (core->num, core->offset, ptr + 3);
|
|
r_core_seek (core, addr, true);
|
|
cmd_tmpseek = core->tmpseek = true;
|
|
goto fuji;
|
|
}
|
|
} else {
|
|
// WAT DU
|
|
eprintf ("TODO: what do you expect for @. import offset from file maybe?\n");
|
|
}
|
|
} else if (ptr[0] && ptr[1] == ':' && ptr[2]) {
|
|
switch (ptr[0]) {
|
|
case 'F': // "@F:" // temporary flag space
|
|
flgspc_changed = r_flag_space_push (core->flags, ptr + 2);
|
|
break;
|
|
case 'B': // "@B:#" // seek to the last instruction in current bb
|
|
{
|
|
int index = (int)r_num_math (core->num, ptr + 2);
|
|
RAnalBlock *bb = r_anal_bb_from_offset (core->anal, core->offset);
|
|
if (bb) {
|
|
// handle negative indices
|
|
if (index < 0) {
|
|
index = bb->ninstr + index;
|
|
}
|
|
|
|
if (index >= 0 && index < bb->ninstr) {
|
|
ut16 inst_off = r_anal_bb_offset_inst (bb, index);
|
|
r_core_seek (core, bb->addr + inst_off, true);
|
|
cmd_tmpseek = core->tmpseek = true;
|
|
} else {
|
|
eprintf ("The current basic block has %d instructions\n", bb->ninstr);
|
|
}
|
|
} else {
|
|
eprintf ("Can't find a basic block for 0x%08"PFMT64x"\n", core->offset);
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
case 'f': // "@f:" // slurp file in block
|
|
f = r_file_slurp (ptr + 2, &sz);
|
|
if (f) {
|
|
{
|
|
RBuffer *b = r_buf_new_with_bytes ((const ut8*)f, (ut64)sz);
|
|
RIODesc *d = r_io_open_buffer (core->io, b, R_PERM_RWX, 0);
|
|
if (d) {
|
|
if (tmpdesc) {
|
|
r_io_desc_close (tmpdesc);
|
|
}
|
|
tmpdesc = d;
|
|
if (pamode) {
|
|
r_config_set_i (core->config, "io.va", 1);
|
|
}
|
|
r_io_map_new (core->io, d->fd, d->perm, 0, core->offset, r_buf_size (b));
|
|
}
|
|
}
|
|
} else {
|
|
eprintf ("cannot open '%s'\n", ptr + 3);
|
|
}
|
|
break;
|
|
case 'r': // "@r:" // regname
|
|
if (ptr[1] == ':') {
|
|
ut64 regval;
|
|
char *mander = strdup (ptr + 2);
|
|
char *sep = findSeparator (mander);
|
|
if (sep) {
|
|
char ch = *sep;
|
|
*sep = 0;
|
|
regval = r_debug_reg_get (core->dbg, mander);
|
|
*sep = ch;
|
|
char *numexpr = r_str_newf ("0x%"PFMT64x"%s", regval, sep);
|
|
regval = r_num_math (core->num, numexpr);
|
|
free (numexpr);
|
|
} else {
|
|
regval = r_debug_reg_get (core->dbg, ptr + 2);
|
|
}
|
|
r_core_seek (core, regval, true);
|
|
cmd_tmpseek = core->tmpseek = true;
|
|
free (mander);
|
|
}
|
|
break;
|
|
case 'b': // "@b:" // bits
|
|
is_bits_set = set_tmp_bits (core, r_num_math (core->num, ptr + 2), &tmpbits, &cmd_ignbithints);
|
|
break;
|
|
case 'i': // "@i:"
|
|
{
|
|
ut64 addr = r_num_math (core->num, ptr + 2);
|
|
if (addr) {
|
|
r_core_cmdf (core, "so %s", ptr + 2);
|
|
cmd_tmpseek = core->tmpseek = true;
|
|
}
|
|
}
|
|
break;
|
|
case 'e': // "@e:"
|
|
{
|
|
char *cmd = parse_tmp_evals (core, ptr + 2);
|
|
if (!tmpeval) {
|
|
tmpeval = cmd;
|
|
} else {
|
|
tmpeval = r_str_prepend (tmpeval, cmd);
|
|
free (cmd);
|
|
}
|
|
}
|
|
break;
|
|
case 'v': // "@v:" // value (honors asm.bits and cfg.bigendian)
|
|
if (ptr[1] == ':') {
|
|
ut8 buf[8] = {0};
|
|
ut64 v = r_num_math (core->num, ptr + 2);
|
|
int be = r_config_get_i (core->config, "cfg.bigendian");
|
|
int bi = r_config_get_i (core->config, "asm.bits");
|
|
if (bi == 64) {
|
|
r_write_ble64 (buf, v, be);
|
|
len = 8;
|
|
} else {
|
|
r_write_ble32 (buf, v, be);
|
|
len = 4;
|
|
}
|
|
tmpfd = r_io_fd_get_current(core->io);
|
|
r_core_block_size (core, R_ABS (len));
|
|
RBuffer *b = r_buf_new_with_bytes (buf, len);
|
|
RIODesc *d = r_io_open_buffer (core->io, b, R_PERM_RWX, 0);
|
|
if (d) {
|
|
if (tmpdesc) {
|
|
r_io_desc_close (tmpdesc);
|
|
}
|
|
tmpdesc = d;
|
|
if (pamode) {
|
|
r_config_set_i (core->config, "io.va", 1);
|
|
}
|
|
r_io_map_new (core->io, d->fd, d->perm, 0, core->offset, r_buf_size (b));
|
|
r_core_block_size (core, len);
|
|
r_core_block_read (core);
|
|
}
|
|
} else {
|
|
eprintf ("Invalid @v: syntax\n");
|
|
}
|
|
break;
|
|
case 'x': // "@x:" // hexpairs
|
|
if (ptr[1] == ':') {
|
|
buf = malloc (strlen (ptr + 2) + 1);
|
|
if (buf) {
|
|
len = r_hex_str2bin (ptr + 2, buf);
|
|
r_core_block_size (core, R_ABS (len));
|
|
if (len > 0) {
|
|
RBuffer *b = r_buf_new_with_bytes (buf, len);
|
|
RIODesc *d = r_io_open_buffer (core->io, b, R_PERM_RWX, 0);
|
|
if (d) {
|
|
if (tmpdesc) {
|
|
r_io_desc_close (tmpdesc);
|
|
}
|
|
tmpdesc = d;
|
|
if (pamode) {
|
|
r_config_set_i (core->config, "io.va", 1);
|
|
}
|
|
r_io_map_new (core->io, d->fd, d->perm, 0, core->offset, r_buf_size (b));
|
|
r_core_block_size (core, len);
|
|
r_core_block_read (core);
|
|
}
|
|
} else {
|
|
eprintf ("Error: Invalid hexpairs for @x:\n");
|
|
}
|
|
free (buf);
|
|
} else {
|
|
eprintf ("cannot allocate\n");
|
|
}
|
|
} else {
|
|
eprintf ("Invalid @x: syntax\n");
|
|
}
|
|
break;
|
|
case 'k': // "@k"
|
|
{
|
|
char *out = sdb_querys (core->sdb, NULL, 0, ptr + ((ptr[1])? 2: 1));
|
|
if (out) {
|
|
r_core_seek (core, r_num_math (core->num, out), true);
|
|
free (out);
|
|
usemyblock = true;
|
|
}
|
|
}
|
|
break;
|
|
case 'o': // "@o:3"
|
|
if (ptr[1] == ':') {
|
|
tmpfd = core->io->desc ? core->io->desc->fd : -1;
|
|
r_io_use_fd (core->io, atoi (ptr + 2));
|
|
}
|
|
break;
|
|
case 'a': // "@a:"
|
|
if (ptr[1] == ':') {
|
|
char *q = strchr (ptr + 2, ':');
|
|
if (q) {
|
|
*q++ = 0;
|
|
int bits = r_num_math (core->num, q);
|
|
is_bits_set = set_tmp_bits (core, bits, &tmpbits, &cmd_ignbithints);
|
|
}
|
|
is_arch_set = set_tmp_arch (core, ptr + 2, &tmpasm);
|
|
} else {
|
|
eprintf ("Usage: pd 10 @a:arm:32\n");
|
|
}
|
|
break;
|
|
case 's': // "@s:" // wtf syntax
|
|
{
|
|
len = strlen (ptr + 2);
|
|
r_core_block_size (core, len);
|
|
const ut8 *buf = (const ut8*)r_str_trim_head_ro (ptr + 2);
|
|
|
|
if (len > 0) {
|
|
RBuffer *b = r_buf_new_with_bytes (buf, len);
|
|
RIODesc *d = r_io_open_buffer (core->io, b, R_PERM_RWX, 0);
|
|
if (!core->io->va) {
|
|
r_config_set_i (core->config, "io.va", 1);
|
|
}
|
|
if (d) {
|
|
if (tmpdesc) {
|
|
r_io_desc_close (tmpdesc);
|
|
}
|
|
tmpdesc = d;
|
|
if (pamode) {
|
|
r_config_set_i (core->config, "io.va", 1);
|
|
}
|
|
r_io_map_new (core->io, d->fd, d->perm, 0, core->offset, r_buf_size (b));
|
|
r_core_block_size (core, len);
|
|
// r_core_block_read (core);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
goto ignore;
|
|
}
|
|
*ptr = '@';
|
|
/* trim whitespaces before the @ */
|
|
/* Fixes pd @x:9090 */
|
|
char *trim = ptr - 2;
|
|
while (trim > cmd) {
|
|
if (!IS_WHITESPACE (*trim)) {
|
|
break;
|
|
}
|
|
*trim = 0;
|
|
trim--;
|
|
}
|
|
goto next_arroba;
|
|
}
|
|
ignore:
|
|
r_str_trim_head (ptr + 1);
|
|
cmd = r_str_trim_nc (cmd);
|
|
if (ptr2) {
|
|
if (strlen (ptr + 1) == 13 && strlen (ptr2 + 1) == 6 &&
|
|
!memcmp (ptr + 1, "0x", 2) &&
|
|
!memcmp (ptr2 + 1, "0x", 2)) {
|
|
/* 0xXXXX:0xYYYY */
|
|
} else if (strlen (ptr + 1) == 9 && strlen (ptr2 + 1) == 4) {
|
|
/* XXXX:YYYY */
|
|
} else {
|
|
*ptr2 = '\0';
|
|
if (!ptr2[1]) {
|
|
goto fail;
|
|
}
|
|
r_core_block_size (
|
|
core, r_num_math (core->num, ptr2 + 1));
|
|
}
|
|
}
|
|
|
|
r_str_trim_head (ptr + 1);
|
|
offstr = ptr + 1;
|
|
|
|
addr = (*offstr == '{')? core->offset: r_num_math (core->num, offstr);
|
|
addr_is_set = true;
|
|
|
|
if (isalpha ((ut8)ptr[1]) && !addr) {
|
|
if (!r_flag_get (core->flags, ptr + 1)) {
|
|
eprintf ("Invalid address (%s)\n", ptr + 1);
|
|
goto fail;
|
|
}
|
|
} else {
|
|
char ch = *offstr;
|
|
if (ch == '-' || ch == '+') {
|
|
addr = core->offset + addr;
|
|
}
|
|
}
|
|
// remap thhe tmpdesc if any
|
|
if (addr) {
|
|
RIODesc *d = tmpdesc;
|
|
if (d) {
|
|
r_io_map_new (core->io, d->fd, d->perm, 0, addr, r_io_desc_size (d));
|
|
}
|
|
}
|
|
next_arroba:
|
|
if (arroba) {
|
|
ptr = arroba + 1;
|
|
*arroba = '@';
|
|
arroba = NULL;
|
|
goto repeat_arroba;
|
|
}
|
|
core->fixedblock = !!tmpdesc;
|
|
if (core->fixedblock) {
|
|
r_core_block_read (core);
|
|
}
|
|
if (ptr[1] == '@') {
|
|
if (ptr[2] == '@') {
|
|
char *rule = ptr + 3;
|
|
while (*rule && *rule == ' ') {
|
|
rule++;
|
|
}
|
|
ret = r_core_cmd_foreach3 (core, cmd, rule);
|
|
} else {
|
|
ret = r_core_cmd_foreach (core, cmd, ptr + 2);
|
|
}
|
|
} else {
|
|
bool tmpseek = false;
|
|
const char *fromvars[] = { "anal.from", "diff.from", "graph.from", "search.from", "zoom.from", NULL };
|
|
const char *tovars[] = { "anal.to", "diff.to", "graph.to", "search.to", "zoom.to", NULL };
|
|
ut64 curfrom[R_ARRAY_SIZE (fromvars) - 1], curto[R_ARRAY_SIZE (tovars) - 1];
|
|
|
|
// "@{A B}"
|
|
if (ptr[1] == '{') {
|
|
char *range = ptr + 2;
|
|
char *p = strchr (range, ' ');
|
|
if (!p) {
|
|
eprintf ("Usage: / ABCD @{0x1000 0x3000}\n");
|
|
eprintf ("Run command and define the following vars:\n");
|
|
eprintf (" (anal|diff|graph|search|zoom).{from,to}\n");
|
|
free (tmpeval);
|
|
free (tmpasm);
|
|
free (tmpbits);
|
|
goto fail;
|
|
}
|
|
char *arg = p + 1;
|
|
int arg_len = strlen (arg);
|
|
if (arg_len > 0) {
|
|
arg[arg_len - 1] = 0;
|
|
}
|
|
*p = '\x00';
|
|
ut64 from = r_num_math (core->num, range);
|
|
ut64 to = r_num_math (core->num, arg);
|
|
// save current ranges
|
|
for (i = 0; fromvars[i]; i++) {
|
|
curfrom[i] = r_config_get_i (core->config, fromvars[i]);
|
|
}
|
|
for (i = 0; tovars[i]; i++) {
|
|
curto[i] = r_config_get_i (core->config, tovars[i]);
|
|
}
|
|
// set new ranges
|
|
for (i = 0; fromvars[i]; i++) {
|
|
r_config_set_i (core->config, fromvars[i], from);
|
|
}
|
|
for (i = 0; tovars[i]; i++) {
|
|
r_config_set_i (core->config, tovars[i], to);
|
|
}
|
|
tmpseek = true;
|
|
}
|
|
if (usemyblock) {
|
|
if (addr_is_set) {
|
|
core->offset = addr;
|
|
}
|
|
ret = r_cmd_call (core->rcmd, r_str_trim_head_ro (cmd));
|
|
} else {
|
|
if (addr_is_set) {
|
|
if (ptr[1]) {
|
|
r_core_seek (core, addr, true);
|
|
r_core_block_read (core);
|
|
}
|
|
}
|
|
ret = r_cmd_call (core->rcmd, r_str_trim_head_ro (cmd));
|
|
|
|
}
|
|
if (tmpseek) {
|
|
// restore ranges
|
|
for (i = 0; fromvars[i]; i++) {
|
|
r_config_set_i (core->config, fromvars[i], curfrom[i]);
|
|
}
|
|
for (i = 0; tovars[i]; i++) {
|
|
r_config_set_i (core->config, tovars[i], curto[i]);
|
|
}
|
|
}
|
|
}
|
|
if (ptr2) {
|
|
*ptr2 = '!';
|
|
r_core_block_size (core, tmpbsz);
|
|
}
|
|
if (is_arch_set) {
|
|
core->fixedarch = oldfixedarch;
|
|
r_config_set (core->config, "asm.arch", tmpasm);
|
|
R_FREE (tmpasm);
|
|
}
|
|
if (tmpdesc) {
|
|
if (pamode) {
|
|
r_config_set_i (core->config, "io.va", 0);
|
|
}
|
|
r_io_desc_close (tmpdesc);
|
|
tmpdesc = NULL;
|
|
}
|
|
if (tmpfd != -1) {
|
|
// TODO: reuse tmpfd instead of
|
|
r_io_use_fd (core->io, tmpfd);
|
|
tmpfd = -1;
|
|
}
|
|
if (is_bits_set) {
|
|
r_config_set (core->config, "asm.bits", tmpbits);
|
|
core->fixedbits = oldfixedbits;
|
|
}
|
|
if (tmpbsz != core->blocksize) {
|
|
r_core_block_size (core, tmpbsz);
|
|
}
|
|
if (tmpeval) {
|
|
r_core_cmd0 (core, tmpeval);
|
|
R_FREE (tmpeval);
|
|
}
|
|
if (flgspc_changed) {
|
|
r_flag_space_pop (core->flags);
|
|
}
|
|
*ptr = '@';
|
|
rc = ret;
|
|
goto beach;
|
|
}
|
|
fuji:
|
|
if (cmd) {
|
|
r_str_trim_head (cmd);
|
|
rc = r_cmd_call (core->rcmd, cmd);
|
|
} else {
|
|
rc = false;
|
|
}
|
|
beach:
|
|
if (grep) {
|
|
char *old_grep = grep;
|
|
grep = unescape_special_chars (old_grep, SPECIAL_CHARS_REGULAR);
|
|
free (old_grep);
|
|
}
|
|
r_cons_grep_process (grep);
|
|
if (scr_html != -1) {
|
|
r_cons_flush ();
|
|
r_config_set_i (core->config, "scr.html", scr_html);
|
|
}
|
|
if (scr_color != -1) {
|
|
r_config_set_i (core->config, "scr.color", scr_color);
|
|
}
|
|
r_list_free (tmpenvs);
|
|
if (tmpdesc) {
|
|
r_io_desc_close (tmpdesc);
|
|
tmpdesc = NULL;
|
|
}
|
|
core->fixedarch = oldfixedarch;
|
|
core->fixedbits = oldfixedbits;
|
|
if (tmpseek) {
|
|
*tmpseek = cmd_tmpseek;
|
|
}
|
|
if (cmd_ignbithints != -1) {
|
|
r_config_set_i (core->config, "anal.ignbithints", cmd_ignbithints);
|
|
}
|
|
return rc;
|
|
fail:
|
|
rc = -1;
|
|
goto beach;
|
|
}
|
|
|
|
struct exec_command_t {
|
|
RCore *core;
|
|
const char *cmd;
|
|
};
|
|
|
|
static bool copy_into_flagitem_list(RFlagItem *flg, void *u) {
|
|
RFlagItem *fi = r_mem_dup (flg, sizeof (RFlagItem));
|
|
r_list_append (u, fi);
|
|
return true;
|
|
}
|
|
|
|
static void foreach_pairs(RCore *core, const char *cmd, const char *each) {
|
|
const char *arg;
|
|
int pair = 0;
|
|
for (arg = each ; ; ) {
|
|
char *next = strchr (arg, ' ');
|
|
if (next) {
|
|
*next = 0;
|
|
}
|
|
if (arg && *arg) {
|
|
ut64 n = r_num_get (NULL, arg);
|
|
if (pair % 2) {
|
|
r_core_block_size (core, n);
|
|
r_core_cmd0 (core, cmd);
|
|
} else {
|
|
r_core_seek (core, n, true);
|
|
}
|
|
pair++;
|
|
}
|
|
if (!next) {
|
|
break;
|
|
}
|
|
arg = next + 1;
|
|
}
|
|
}
|
|
|
|
R_API int r_core_cmd_foreach3(RCore *core, const char *cmd, char *each) { // "@@@"
|
|
RDebug *dbg = core->dbg;
|
|
RList *list, *head;
|
|
RListIter *iter;
|
|
int i;
|
|
const char *filter = NULL;
|
|
|
|
if (each[0] && each[1] == ':') {
|
|
filter = each + 2;
|
|
}
|
|
|
|
switch (each[0]) {
|
|
case '=':
|
|
foreach_pairs (core, cmd, each + 1);
|
|
break;
|
|
case '?':
|
|
r_core_cmd_help (core, help_msg_at_at_at);
|
|
break;
|
|
case 'c':
|
|
if (filter) {
|
|
char *arg = r_core_cmd_str (core, filter);
|
|
foreach_pairs (core, cmd, arg);
|
|
free (arg);
|
|
} else {
|
|
eprintf ("Usage: @@@c:command # same as @@@=`command`\n");
|
|
}
|
|
break;
|
|
case 'C': {
|
|
char *glob = filter ? r_str_trim_dup (filter): NULL;
|
|
RIntervalTreeIter it;
|
|
RAnalMetaItem *meta;
|
|
r_interval_tree_foreach (&core->anal->meta, it, meta) {
|
|
if (meta->type != R_META_TYPE_COMMENT) {
|
|
continue;
|
|
}
|
|
if (!glob || (meta->str && r_str_glob (meta->str, glob))) {
|
|
r_core_seek (core, r_interval_tree_iter_get (&it)->start, true);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
free (glob);
|
|
break;
|
|
}
|
|
case 'm':
|
|
{
|
|
int fd = r_io_fd_get_current (core->io);
|
|
// only iterate maps of current fd
|
|
RList *maps = r_io_map_get_by_fd (core->io, fd);
|
|
RIOMap *map;
|
|
if (maps) {
|
|
RListIter *iter;
|
|
r_list_foreach (maps, iter, map) {
|
|
r_core_seek (core, r_io_map_begin (map), true);
|
|
r_core_block_size (core, r_io_map_size (map));
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_list_free (maps);
|
|
}
|
|
}
|
|
break;
|
|
case 'M': // @@@M
|
|
if (dbg && dbg->h && dbg->maps) {
|
|
RDebugMap *map;
|
|
r_list_foreach (dbg->maps, iter, map) {
|
|
r_core_seek (core, map->addr, true);
|
|
//r_core_block_size (core, map->size);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
case 't':
|
|
// iterate over all threads
|
|
if (dbg && dbg->h && dbg->h->threads) {
|
|
int origpid = dbg->pid;
|
|
RDebugPid *p;
|
|
list = dbg->h->threads (dbg, dbg->pid);
|
|
if (!list) {
|
|
return false;
|
|
}
|
|
r_list_foreach (list, iter, p) {
|
|
r_core_cmdf (core, "dp %d", p->pid);
|
|
r_cons_printf ("PID %d\n", p->pid);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_core_cmdf (core, "dp %d", origpid);
|
|
r_list_free (list);
|
|
}
|
|
break;
|
|
case 'r': // @@@r
|
|
{
|
|
ut64 offorig = core->offset;
|
|
for (i = 0; i < R_REG_TYPE_LAST; i++) {
|
|
RRegItem *item;
|
|
ut64 value;
|
|
head = r_reg_get_list (core->dbg->reg, i);
|
|
if (!head) {
|
|
continue;
|
|
}
|
|
RList *list = r_list_newf (free);
|
|
r_list_foreach (head, iter, item) {
|
|
if (item->size != core->anal->bits) {
|
|
continue;
|
|
}
|
|
if (item->type != i) {
|
|
continue;
|
|
}
|
|
r_list_append (list, strdup (item->name));
|
|
}
|
|
const char *item_name;
|
|
r_list_foreach (list, iter, item_name) {
|
|
value = r_reg_getv (core->dbg->reg, item_name);
|
|
r_core_seek (core, value, true);
|
|
r_cons_printf ("%s: ", item_name);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_list_free (list);
|
|
}
|
|
r_core_seek (core, offorig, true);
|
|
}
|
|
break;
|
|
case 'i': // @@@i
|
|
{
|
|
RBinImport *imp;
|
|
ut64 offorig = core->offset;
|
|
list = r_bin_get_imports (core->bin);
|
|
RList *lost = r_list_newf (free);
|
|
r_list_foreach (list, iter, imp) {
|
|
char *impflag = r_str_newf ("sym.imp.%s", imp->name);
|
|
ut64 addr = r_num_math (core->num, impflag);
|
|
ut64 *n = R_NEW (ut64);
|
|
*n = addr;
|
|
r_list_append (lost, n);
|
|
free (impflag);
|
|
}
|
|
ut64 *naddr;
|
|
r_list_foreach (lost, iter, naddr) {
|
|
ut64 addr = *naddr;
|
|
if (addr && addr != UT64_MAX) {
|
|
r_core_seek (core, addr, true);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
r_core_seek (core, offorig, true);
|
|
r_list_free (lost);
|
|
}
|
|
break;
|
|
case 'S': // "@@@S"
|
|
{
|
|
RBinObject *obj = r_bin_cur_object (core->bin);
|
|
if (obj) {
|
|
ut64 offorig = core->offset;
|
|
ut64 bszorig = core->blocksize;
|
|
RBinSection *sec;
|
|
RListIter *iter;
|
|
r_list_foreach (obj->sections, iter, sec) {
|
|
r_core_seek (core, sec->vaddr, true);
|
|
r_core_block_size (core, sec->vsize);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_core_block_size (core, bszorig);
|
|
r_core_seek (core, offorig, true);
|
|
}
|
|
}
|
|
#if ATTIC
|
|
if (each[1] == 'S') {
|
|
RListIter *it;
|
|
RBinSection *sec;
|
|
RBinObject *obj = r_bin_cur_object (core->bin);
|
|
int cbsz = core->blocksize;
|
|
r_list_foreach (obj->sections, it, sec){
|
|
ut64 addr = sec->vaddr;
|
|
ut64 size = sec->vsize;
|
|
// TODO:
|
|
//if (R_BIN_SCN_EXECUTABLE & sec->perm) {
|
|
// continue;
|
|
//}
|
|
r_core_seek_size (core, addr, size);
|
|
r_core_cmd (core, cmd, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_core_block_size (core, cbsz);
|
|
}
|
|
#endif
|
|
break;
|
|
case 's':
|
|
if (each[1] == 't') { // strings
|
|
list = r_bin_get_strings (core->bin);
|
|
if (list) {
|
|
ut64 offorig = core->offset;
|
|
ut64 obs = core->blocksize;
|
|
RBinString *s;
|
|
RList *lost = r_list_newf (free);
|
|
r_list_foreach (list, iter, s) {
|
|
RBinString *bs = r_mem_dup (s, sizeof (RBinString));
|
|
r_list_append (lost, bs);
|
|
}
|
|
r_list_foreach (lost, iter, s) {
|
|
r_core_block_size (core, s->size);
|
|
r_core_seek (core, s->vaddr, true);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_core_block_size (core, obs);
|
|
r_core_seek (core, offorig, true);
|
|
r_list_free (lost);
|
|
}
|
|
} else {
|
|
// symbols
|
|
RBinSymbol *sym;
|
|
ut64 offorig = core->offset;
|
|
ut64 obs = core->blocksize;
|
|
list = r_bin_get_symbols (core->bin);
|
|
r_cons_break_push (NULL, NULL);
|
|
RList *lost = r_list_newf (free);
|
|
r_list_foreach (list, iter, sym) {
|
|
RBinSymbol *bs = r_mem_dup (sym, sizeof (RBinSymbol));
|
|
r_list_append (lost, bs);
|
|
}
|
|
r_list_foreach (lost, iter, sym) {
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
r_core_block_size (core, sym->size);
|
|
r_core_seek (core, sym->vaddr, true);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_cons_break_pop ();
|
|
r_list_free (lost);
|
|
r_core_block_size (core, obs);
|
|
r_core_seek (core, offorig, true);
|
|
}
|
|
break;
|
|
case 'f': // flags
|
|
{
|
|
// TODO: honor ^C
|
|
char *glob = filter? r_str_trim_dup (filter): NULL;
|
|
ut64 off = core->offset;
|
|
ut64 obs = core->blocksize;
|
|
RList *flags = r_list_newf (free);
|
|
r_flag_foreach_glob (core->flags, glob, copy_into_flagitem_list, flags);
|
|
RListIter *iter;
|
|
RFlagItem *f;
|
|
r_list_foreach (flags, iter, f) {
|
|
r_core_block_size (core, f->size);
|
|
r_core_seek (core, f->offset, true);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_core_seek (core, off, false);
|
|
r_core_block_size (core, obs);
|
|
free (glob);
|
|
}
|
|
break;
|
|
case 'F': // functions
|
|
{
|
|
ut64 obs = core->blocksize;
|
|
ut64 offorig = core->offset;
|
|
RAnalFunction *fcn;
|
|
list = core->anal->fcns;
|
|
r_cons_break_push (NULL, NULL);
|
|
r_list_foreach (list, iter, fcn) {
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
if (!filter || r_str_glob (fcn->name, filter)) {
|
|
r_core_seek (core, fcn->addr, true);
|
|
r_core_block_size (core, r_anal_function_linear_size (fcn));
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
r_cons_break_pop ();
|
|
r_core_block_size (core, obs);
|
|
r_core_seek (core, offorig, true);
|
|
}
|
|
break;
|
|
case 'b':
|
|
{
|
|
RAnalFunction *fcn = r_anal_get_fcn_in (core->anal, core->offset, 0);
|
|
ut64 offorig = core->offset;
|
|
ut64 obs = core->blocksize;
|
|
if (fcn) {
|
|
RListIter *iter;
|
|
RAnalBlock *bb;
|
|
r_list_foreach (fcn->bbs, iter, bb) {
|
|
r_core_seek (core, bb->addr, true);
|
|
r_core_block_size (core, bb->size);
|
|
r_core_cmd0 (core, cmd);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_core_block_size (core, obs);
|
|
r_core_seek (core, offorig, true);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static void foreachOffset(RCore *core, const char *_cmd, const char *each) {
|
|
char *cmd = strdup (_cmd);
|
|
char *nextLine = NULL;
|
|
ut64 addr;
|
|
/* foreach list of items */
|
|
while (each) {
|
|
// skip spaces
|
|
while (*each == ' ') {
|
|
each++;
|
|
}
|
|
// stahp if empty string
|
|
if (!*each) {
|
|
break;
|
|
}
|
|
// find newline
|
|
char *nl = strchr (each, '\n');
|
|
if (nl) {
|
|
*nl = 0;
|
|
nextLine = nl + 1;
|
|
} else {
|
|
nextLine = NULL;
|
|
}
|
|
// chop comment in line
|
|
nl = strchr (each, '#');
|
|
if (nl) {
|
|
*nl = 0;
|
|
}
|
|
// space separated numbers
|
|
while (each && *each) {
|
|
// find spaces
|
|
while (*each == ' ') {
|
|
each++;
|
|
}
|
|
char *str = strchr (each, ' ');
|
|
if (str) {
|
|
*str = '\0';
|
|
addr = r_num_math (core->num, each);
|
|
*str = ' ';
|
|
each = str + 1;
|
|
} else {
|
|
if (!*each) {
|
|
break;
|
|
}
|
|
addr = r_num_math (core->num, each);
|
|
each = NULL;
|
|
}
|
|
r_core_seek (core, addr, true);
|
|
r_core_cmd (core, cmd, 0);
|
|
foreach_newline (core);
|
|
r_cons_flush ();
|
|
}
|
|
each = nextLine;
|
|
}
|
|
free (cmd);
|
|
}
|
|
|
|
R_API int r_core_cmd_foreach(RCore *core, const char *cmd, char *each) {
|
|
int i, j;
|
|
char ch;
|
|
char *word = NULL;
|
|
char *str, *ostr = NULL;
|
|
RListIter *iter;
|
|
RFlagItem *flag;
|
|
ut64 oseek, addr;
|
|
|
|
for (; *cmd == ' '; cmd++) {
|
|
;
|
|
}
|
|
|
|
oseek = core->offset;
|
|
ostr = str = strdup (each);
|
|
r_cons_break_push (NULL, NULL); //pop on return
|
|
switch (each[0]) {
|
|
case '/': // "@@/"
|
|
{
|
|
char *cmdhit = strdup (r_config_get (core->config, "cmd.hit"));
|
|
r_config_set (core->config, "cmd.hit", cmd);
|
|
r_core_cmd0 (core, each);
|
|
r_config_set (core->config, "cmd.hit", cmdhit);
|
|
free (cmdhit);
|
|
}
|
|
free (ostr);
|
|
return 0;
|
|
case '?': // "@@?"
|
|
r_core_cmd_help (core, help_msg_at_at);
|
|
break;
|
|
case 'b': // "@@b" - function basic blocks
|
|
{
|
|
RListIter *iter;
|
|
RAnalBlock *bb;
|
|
RAnalFunction *fcn = r_anal_get_function_at (core->anal, core->offset);
|
|
int bs = core->blocksize;
|
|
if (fcn) {
|
|
r_list_sort (fcn->bbs, bb_cmp);
|
|
r_list_foreach (fcn->bbs, iter, bb) {
|
|
r_core_block_size (core, bb->size);
|
|
r_core_seek (core, bb->addr, true);
|
|
r_core_cmd (core, cmd, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
r_core_block_size (core, bs);
|
|
goto out_finish;
|
|
}
|
|
break;
|
|
case 's': // "@@s" - sequence
|
|
{
|
|
char *str = each + 1;
|
|
if (*str == ':' || *str == ' ') {
|
|
str++;
|
|
}
|
|
int count = r_str_split (str, ' ');
|
|
if (count == 3) {
|
|
ut64 cur;
|
|
ut64 from = r_num_math (core->num, r_str_word_get0 (str, 0));
|
|
ut64 to = r_num_math (core->num, r_str_word_get0 (str, 1));
|
|
ut64 step = r_num_math (core->num, r_str_word_get0 (str, 2));
|
|
for (cur = from; cur <= to; cur += step) {
|
|
(void) r_core_seek (core, cur, true);
|
|
r_core_cmd (core, cmd, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
} else {
|
|
eprintf ("Usage: cmd @@s:from to step\n");
|
|
}
|
|
goto out_finish;
|
|
}
|
|
break;
|
|
case 'i': // "@@i" - function instructions
|
|
{
|
|
RListIter *iter;
|
|
RAnalBlock *bb;
|
|
int i;
|
|
RAnalFunction *fcn = r_anal_get_function_at (core->anal, core->offset);
|
|
if (fcn) {
|
|
r_list_sort (fcn->bbs, bb_cmp);
|
|
r_list_foreach (fcn->bbs, iter, bb) {
|
|
for (i = 0; i < bb->op_pos_size; i++) {
|
|
ut64 addr = bb->addr + bb->op_pos[i];
|
|
r_core_seek (core, addr, true);
|
|
r_core_cmd (core, cmd, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
goto out_finish;
|
|
}
|
|
break;
|
|
case 'f': // "@@f"
|
|
if (each[1] == ':') {
|
|
RAnalFunction *fcn;
|
|
RListIter *iter;
|
|
if (core->anal) {
|
|
r_list_foreach (core->anal->fcns, iter, fcn) {
|
|
if (each[2] && strstr (fcn->name, each + 2)) {
|
|
r_core_seek (core, fcn->addr, true);
|
|
r_core_cmd (core, cmd, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
goto out_finish;
|
|
} else {
|
|
RAnalFunction *fcn;
|
|
RListIter *iter;
|
|
if (core->anal) {
|
|
RConsGrep grep = core->cons->context->grep;
|
|
r_list_foreach (core->anal->fcns, iter, fcn) {
|
|
char *buf;
|
|
r_core_seek (core, fcn->addr, true);
|
|
r_cons_push ();
|
|
r_core_cmd (core, cmd, 0);
|
|
buf = (char *)r_cons_get_buffer ();
|
|
if (buf) {
|
|
buf = strdup (buf);
|
|
}
|
|
r_cons_pop ();
|
|
r_cons_strcat (buf);
|
|
free (buf);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
core->cons->context->grep = grep;
|
|
}
|
|
goto out_finish;
|
|
}
|
|
break;
|
|
case 't': // "@@t"
|
|
{
|
|
RDebugPid *p;
|
|
int pid = core->dbg->pid;
|
|
if (core->dbg->h && core->dbg->h->pids) {
|
|
RList *list = core->dbg->h->pids (core->dbg, R_MAX (0, pid));
|
|
r_list_foreach (list, iter, p) {
|
|
r_cons_printf ("# PID %d\n", p->pid);
|
|
r_debug_select (core->dbg, p->pid, p->pid);
|
|
r_core_cmd (core, cmd, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
}
|
|
r_list_free (list);
|
|
}
|
|
r_debug_select (core->dbg, pid, pid);
|
|
goto out_finish;
|
|
}
|
|
break;
|
|
case 'c': // "@@c:"
|
|
if (each[1] == ':') {
|
|
char *arg = r_core_cmd_str (core, each + 2);
|
|
if (arg) {
|
|
foreachOffset (core, cmd, arg);
|
|
free (arg);
|
|
}
|
|
}
|
|
break;
|
|
case '=': // "@@="
|
|
foreachOffset (core, cmd, str + 1);
|
|
break;
|
|
case 'd': // "@@d"
|
|
if (each[1] == 'b' && each[2] == 't') {
|
|
ut64 oseek = core->offset;
|
|
RDebugFrame *frame;
|
|
RListIter *iter;
|
|
RList *list;
|
|
list = r_debug_frames (core->dbg, UT64_MAX);
|
|
i = 0;
|
|
r_list_foreach (list, iter, frame) {
|
|
switch (each[3]) {
|
|
case 'b':
|
|
r_core_seek (core, frame->bp, true);
|
|
break;
|
|
case 's':
|
|
r_core_seek (core, frame->sp, true);
|
|
break;
|
|
default:
|
|
case 'a':
|
|
r_core_seek (core, frame->addr, true);
|
|
break;
|
|
}
|
|
r_core_cmd (core, cmd, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
i++;
|
|
}
|
|
r_core_seek (core, oseek, false);
|
|
r_list_free (list);
|
|
} else {
|
|
eprintf("Invalid for-each statement. Use @@=dbt[abs]\n");
|
|
}
|
|
break;
|
|
case 'k': // "@@k"
|
|
/* foreach list of items */
|
|
{
|
|
char *out = sdb_querys (core->sdb, NULL, 0, str + ((str[1])? 2: 1));
|
|
if (out) {
|
|
each = out;
|
|
do {
|
|
while (*each == ' ') {
|
|
each++;
|
|
}
|
|
if (!*each) {
|
|
break;
|
|
}
|
|
str = strchr (each, ' ');
|
|
if (str) {
|
|
*str = '\0';
|
|
addr = r_num_math (core->num, each);
|
|
*str = ' ';
|
|
} else {
|
|
addr = r_num_math (core->num, each);
|
|
}
|
|
//eprintf ("; 0x%08"PFMT64x":\n", addr);
|
|
each = str + 1;
|
|
r_core_seek (core, addr, true);
|
|
r_core_cmd (core, cmd, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
r_cons_flush ();
|
|
} while (str != NULL);
|
|
free (out);
|
|
}
|
|
}
|
|
break;
|
|
case '.': // "@@."
|
|
if (each[1] == '(') {
|
|
char cmd2[1024];
|
|
// XXX what's this 999 ?
|
|
i = 0;
|
|
for (core->rcmd->macro.counter = 0; i < 999; core->rcmd->macro.counter++) {
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
r_cmd_macro_call (&core->rcmd->macro, each + 2);
|
|
if (!core->rcmd->macro.brk_value) {
|
|
break;
|
|
}
|
|
addr = core->rcmd->macro._brk_value;
|
|
sprintf (cmd2, "%s @ 0x%08"PFMT64x"", cmd, addr);
|
|
eprintf ("0x%08"PFMT64x" (%s)\n", addr, cmd2);
|
|
r_core_seek (core, addr, true);
|
|
r_core_cmd (core, cmd2, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
i++;
|
|
}
|
|
} else {
|
|
char buf[1024];
|
|
char cmd2[1024];
|
|
FILE *fd = r_sandbox_fopen (each + 1, "r");
|
|
if (fd) {
|
|
core->rcmd->macro.counter = 0;
|
|
while (!feof (fd)) {
|
|
buf[0] = '\0';
|
|
if (!fgets (buf, sizeof (buf), fd)) {
|
|
break;
|
|
}
|
|
addr = r_num_math (core->num, buf);
|
|
eprintf ("0x%08"PFMT64x": %s\n", addr, cmd);
|
|
sprintf (cmd2, "%s @ 0x%08"PFMT64x"", cmd, addr);
|
|
r_core_seek (core, addr, true); // XXX
|
|
r_core_cmd (core, cmd2, 0);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
core->rcmd->macro.counter++;
|
|
}
|
|
fclose (fd);
|
|
} else {
|
|
eprintf ("cannot open file '%s' to read offsets\n", each + 1);
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
core->rcmd->macro.counter = 0;
|
|
for (; *each == ' '; each++) {
|
|
;
|
|
}
|
|
i = 0;
|
|
while (str[i]) {
|
|
j = i;
|
|
for (; str[j] && str[j] == ' '; j++) {
|
|
; // skip spaces
|
|
}
|
|
for (i = j; str[i] && str[i] != ' '; i++) {
|
|
; // find EOS
|
|
}
|
|
ch = str[i];
|
|
str[i] = '\0';
|
|
word = strdup (str + j);
|
|
if (!word) {
|
|
break;
|
|
}
|
|
str[i] = ch;
|
|
{
|
|
const RSpace *flagspace = r_flag_space_cur (core->flags);
|
|
RList *match_flag_items = r_list_newf ((RListFree)r_flag_item_free);
|
|
if (!match_flag_items) {
|
|
break;
|
|
}
|
|
|
|
/* duplicate flags that match word, to be sure
|
|
the command is going to be executed on flags
|
|
values at the moment the command is called
|
|
(without side effects) */
|
|
struct duplicate_flag_t u = {
|
|
.ret = match_flag_items,
|
|
.word = word,
|
|
};
|
|
r_flag_foreach_space (core->flags, flagspace, duplicate_flag, &u);
|
|
|
|
/* for all flags that match */
|
|
r_list_foreach (match_flag_items, iter, flag) {
|
|
if (r_cons_is_breaked ()) {
|
|
break;
|
|
}
|
|
|
|
char *buf = NULL;
|
|
const char *tmp = NULL;
|
|
r_core_seek (core, flag->offset, true);
|
|
r_cons_push ();
|
|
r_core_cmd (core, cmd, 0);
|
|
tmp = r_cons_get_buffer ();
|
|
buf = tmp? strdup (tmp): NULL;
|
|
r_cons_pop ();
|
|
r_cons_strcat (buf);
|
|
free (buf);
|
|
if (foreach_newline (core)) {
|
|
break;
|
|
}
|
|
r_core_task_yield (&core->tasks);
|
|
}
|
|
|
|
r_list_free (match_flag_items);
|
|
core->rcmd->macro.counter++ ;
|
|
R_FREE (word);
|
|
}
|
|
}
|
|
}
|
|
r_cons_break_pop ();
|
|
// XXX: use r_core_seek here
|
|
core->offset = oseek;
|
|
|
|
free (word);
|
|
free (ostr);
|
|
return true;
|
|
out_finish:
|
|
free (ostr);
|
|
r_cons_break_pop ();
|
|
return false;
|
|
}
|
|
|
|
static int run_cmd_depth(RCore *core, char *cmd) {
|
|
char *rcmd;
|
|
int ret = false;
|
|
|
|
if (core->cons->context->cmd_depth < 1) {
|
|
eprintf ("r_core_cmd: That was too deep (%s)...\n", cmd);
|
|
return false;
|
|
}
|
|
core->cons->context->cmd_depth--;
|
|
for (rcmd = cmd;;) {
|
|
char *ptr = strchr (rcmd, '\n');
|
|
if (ptr) {
|
|
*ptr = '\0';
|
|
}
|
|
ret = r_core_cmd_subst (core, rcmd);
|
|
if (ret == -1) {
|
|
r_cons_eprintf ("|ERROR| Invalid command '%s' (0x%02x)\n", rcmd, *rcmd);
|
|
break;
|
|
}
|
|
if (!ptr) {
|
|
break;
|
|
}
|
|
rcmd = ptr + 1;
|
|
}
|
|
core->cons->context->cmd_depth++;
|
|
return ret;
|
|
}
|
|
|
|
R_API int r_core_cmd(RCore *core, const char *cstr, bool log) {
|
|
int ret = false;
|
|
size_t i;
|
|
if (core->cmdfilter) {
|
|
const char *invalid_chars = ";|>`@";
|
|
for (i = 0; invalid_chars[i]; i++) {
|
|
if (strchr (cstr, invalid_chars[i])) {
|
|
ret = true;
|
|
goto beach;
|
|
}
|
|
}
|
|
if (strncmp (cstr, core->cmdfilter, strlen (core->cmdfilter))) {
|
|
ret = true;
|
|
goto beach;
|
|
}
|
|
}
|
|
if (core->cmdremote) {
|
|
if (*cstr == 'q') {
|
|
R_FREE (core->cmdremote);
|
|
goto beach; // false
|
|
} else if (*cstr != '=' && strncmp (cstr, "!=", 2)) {
|
|
if (core->cmdremote[0]) {
|
|
char *s = r_str_newf ("%s %s", core->cmdremote, cstr);
|
|
r_core_rtr_cmd (core, s);
|
|
free (s);
|
|
} else {
|
|
char *res = r_io_system (core->io, cstr);
|
|
if (res) {
|
|
r_cons_printf ("%s\n", res);
|
|
free (res);
|
|
}
|
|
}
|
|
if (log) {
|
|
r_line_hist_add (cstr);
|
|
}
|
|
goto beach; // false
|
|
}
|
|
}
|
|
|
|
if (!cstr || (*cstr == '|' && cstr[1] != '?')) {
|
|
// raw comment syntax
|
|
goto beach; // false;
|
|
}
|
|
if (!strncmp (cstr, "/*", 2)) {
|
|
if (r_sandbox_enable (0)) {
|
|
eprintf ("This command is disabled in sandbox mode\n");
|
|
goto beach; // false
|
|
}
|
|
core->incomment = true;
|
|
} else if (!strncmp (cstr, "*/", 2)) {
|
|
core->incomment = false;
|
|
goto beach; // false
|
|
}
|
|
if (core->incomment) {
|
|
goto beach; // false
|
|
}
|
|
if (log && (*cstr && (*cstr != '.' || !strncmp (cstr, ".(", 2)))) {
|
|
free (core->lastcmd);
|
|
core->lastcmd = strdup (cstr);
|
|
}
|
|
|
|
char *cmd = malloc (strlen (cstr) + 4096);
|
|
if (!cmd) {
|
|
goto beach;
|
|
}
|
|
r_str_cpy (cmd, cstr);
|
|
if (log) {
|
|
r_line_hist_add (cstr);
|
|
}
|
|
|
|
ret = run_cmd_depth (core, cmd);
|
|
free (cmd);
|
|
beach:
|
|
return ret;
|
|
}
|
|
|
|
R_API int r_core_cmd_lines(RCore *core, const char *lines) {
|
|
if (core->use_newshell) {
|
|
// handle newshell
|
|
}
|
|
int r, ret = true;
|
|
char *nl, *data, *odata;
|
|
|
|
if (!lines || !*lines) {
|
|
return true;
|
|
}
|
|
data = odata = strdup (lines);
|
|
if (!odata) {
|
|
return false;
|
|
}
|
|
size_t line_count = r_str_char_count(lines, '\n');
|
|
|
|
#if __UNIX__
|
|
const bool istty = r_cons_isatty ();
|
|
#else
|
|
const bool istty = true;
|
|
#endif
|
|
const bool show_progress_bar = core->print->enable_progressbar && r_config_get_i (core->config, "scr.interactive") && r_config_get_i (core->config, "scr.progressbar") && istty;
|
|
size_t current_line = 0;
|
|
nl = strchr (odata, '\n');
|
|
if (nl) {
|
|
r_cons_break_push (NULL, NULL);
|
|
do {
|
|
if (show_progress_bar) {
|
|
r_print_progressbar_with_count (core->print, current_line++, line_count, 80, true);
|
|
}
|
|
if (r_cons_is_breaked ()) {
|
|
free (odata);
|
|
r_cons_break_pop ();
|
|
return ret;
|
|
}
|
|
*nl = '\0';
|
|
r = r_core_cmd (core, data, 0);
|
|
if (r < 0) {
|
|
data = nl + 1;
|
|
ret = -1;
|
|
break;
|
|
}
|
|
r_cons_flush ();
|
|
if (data[0] == 'q') {
|
|
if (data[1] == '!') {
|
|
ret = -1;
|
|
} else {
|
|
eprintf ("'q': quit ignored. Use 'q!'\n");
|
|
}
|
|
data = nl + 1;
|
|
break;
|
|
}
|
|
data = nl + 1;
|
|
r_core_task_yield (&core->tasks);
|
|
} while ((nl = strchr (data, '\n')));
|
|
r_cons_break_pop ();
|
|
if (show_progress_bar) {
|
|
r_print_progressbar_with_count (core->print, line_count, line_count, 80, true);
|
|
r_cons_newline ();
|
|
}
|
|
}
|
|
if (ret >= 0 && data && *data) {
|
|
r_core_cmd (core, data, 0);
|
|
r_cons_flush ();
|
|
r_core_task_yield (&core->tasks);
|
|
}
|
|
free (odata);
|
|
return ret;
|
|
}
|
|
|
|
R_API int r_core_cmd_file(RCore *core, const char *file) {
|
|
char *data = r_file_abspath (file);
|
|
if (!data) {
|
|
return false;
|
|
}
|
|
char *odata = r_file_slurp (data, NULL);
|
|
free (data);
|
|
if (!odata) {
|
|
return false;
|
|
}
|
|
if (!r_core_cmd_lines (core, odata)) {
|
|
eprintf ("Failed to run script '%s'\n", file);
|
|
free (odata);
|
|
return false;
|
|
}
|
|
free (odata);
|
|
return true;
|
|
}
|
|
|
|
R_API int r_core_cmd_command(RCore *core, const char *command) {
|
|
int ret, len;
|
|
char *buf, *rcmd, *ptr;
|
|
char *cmd = r_core_sysenv_begin (core, command);
|
|
rcmd = ptr = buf = r_sys_cmd_str (cmd, 0, &len);
|
|
if (!buf) {
|
|
free (cmd);
|
|
return -1;
|
|
}
|
|
ret = r_core_cmd (core, rcmd, 0);
|
|
r_core_sysenv_end (core, command);
|
|
free (buf);
|
|
return ret;
|
|
}
|
|
|
|
//TODO: Fix disasm loop is mandatory
|
|
R_API char *r_core_disassemble_instr(RCore *core, ut64 addr, int l) {
|
|
char *cmd, *ret = NULL;
|
|
cmd = r_str_newf ("pd %i @ 0x%08"PFMT64x, l, addr);
|
|
if (cmd) {
|
|
ret = r_core_cmd_str (core, cmd);
|
|
free (cmd);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
R_API char *r_core_disassemble_bytes(RCore *core, ut64 addr, int b) {
|
|
char *cmd, *ret = NULL;
|
|
cmd = r_str_newf ("pD %i @ 0x%08"PFMT64x, b, addr);
|
|
if (cmd) {
|
|
ret = r_core_cmd_str (core, cmd);
|
|
free (cmd);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
R_API int r_core_cmd_buffer(RCore *core, const char *buf) {
|
|
char *ptr, *optr, *str = strdup (buf);
|
|
if (!str) {
|
|
return false;
|
|
}
|
|
optr = str;
|
|
ptr = strchr (str, '\n');
|
|
while (ptr) {
|
|
*ptr = '\0';
|
|
r_core_cmd (core, optr, 0);
|
|
optr = ptr + 1;
|
|
ptr = strchr (str, '\n');
|
|
}
|
|
r_core_cmd (core, optr, 0);
|
|
free (str);
|
|
return true;
|
|
}
|
|
|
|
R_API int r_core_cmdf(RCore *core, const char *fmt, ...) {
|
|
char string[4096];
|
|
int ret;
|
|
va_list ap;
|
|
va_start (ap, fmt);
|
|
vsnprintf (string, sizeof (string), fmt, ap);
|
|
ret = r_core_cmd (core, string, 0);
|
|
va_end (ap);
|
|
return ret;
|
|
}
|
|
|
|
R_API int r_core_cmd0(RCore *core, const char *cmd) {
|
|
return r_core_cmd (core, cmd, 0);
|
|
}
|
|
|
|
R_API int r_core_flush(RCore *core, const char *cmd) {
|
|
int ret = r_core_cmd (core, cmd, 0);
|
|
r_cons_flush ();
|
|
return ret;
|
|
}
|
|
|
|
R_API char *r_core_cmd_str_pipe(RCore *core, const char *cmd) {
|
|
char *tmp = NULL;
|
|
char *p = (*cmd != '"')? strchr (cmd, '|'): NULL;
|
|
if (!p && *cmd != '!' && *cmd != '.') {
|
|
return r_core_cmd_str (core, cmd);
|
|
}
|
|
r_cons_reset ();
|
|
r_sandbox_disable (true);
|
|
if (r_file_mkstemp ("cmd", &tmp) != -1) {
|
|
int pipefd = r_cons_pipe_open (tmp, 1, 0);
|
|
if (pipefd == -1) {
|
|
r_file_rm (tmp);
|
|
r_sandbox_disable (false);
|
|
free (tmp);
|
|
return r_core_cmd_str (core, cmd);
|
|
}
|
|
char *_cmd = strdup (cmd);
|
|
if (core->use_newshell) {
|
|
r_core_cmd (core, _cmd, 0);
|
|
} else {
|
|
r_core_cmd_subst (core, _cmd);
|
|
}
|
|
r_cons_flush ();
|
|
r_cons_pipe_close (pipefd);
|
|
if (r_file_exists (tmp)) {
|
|
char *s = r_file_slurp (tmp, NULL);
|
|
r_file_rm (tmp);
|
|
r_sandbox_disable (false);
|
|
free (tmp);
|
|
free (_cmd);
|
|
return s? s: strdup ("");
|
|
}
|
|
eprintf ("slurp %s fails\n", tmp);
|
|
r_file_rm (tmp);
|
|
free (tmp);
|
|
free (_cmd);
|
|
r_sandbox_disable (false);
|
|
return r_core_cmd_str (core, cmd);
|
|
}
|
|
r_sandbox_disable (0);
|
|
return NULL;
|
|
}
|
|
|
|
R_API char *r_core_cmd_strf(RCore *core, const char *fmt, ...) {
|
|
char string[4096];
|
|
va_list ap;
|
|
va_start (ap, fmt);
|
|
vsnprintf (string, sizeof (string), fmt, ap);
|
|
char *ret = r_core_cmd_str (core, string);
|
|
va_end (ap);
|
|
return ret;
|
|
}
|
|
|
|
/* return: pointer to a buffer with the output of the command */
|
|
R_API char *r_core_cmd_str(RCore *core, const char *cmd) {
|
|
r_cons_push ();
|
|
if (r_core_cmd (core, cmd, 0) == -1) {
|
|
//eprintf ("Invalid command: %s\n", cmd);
|
|
return NULL;
|
|
}
|
|
r_cons_filter ();
|
|
const char *static_str = r_cons_get_buffer ();
|
|
char *retstr = strdup (r_str_get (static_str));
|
|
r_cons_pop ();
|
|
r_cons_echo (NULL);
|
|
return retstr;
|
|
}
|
|
|
|
/* run cmd in the main task synchronously */
|
|
R_API int r_core_cmd_task_sync(RCore *core, const char *cmd, bool log) {
|
|
RCoreTask *task = core->tasks.main_task;
|
|
char *s = strdup (cmd);
|
|
if (!s) {
|
|
return 0;
|
|
}
|
|
task->cmd = s;
|
|
task->cmd_log = log;
|
|
task->state = R_CORE_TASK_STATE_BEFORE_START;
|
|
int res = r_core_task_run_sync (&core->tasks, task);
|
|
free (s);
|
|
return res;
|
|
}
|
|
|
|
static int cmd_ox(void *data, const char *input) {
|
|
return r_core_cmdf ((RCore*)data, "s 0%s", input);
|
|
}
|
|
|
|
static int compare_cmd_descriptor_name(const void *a, const void *b) {
|
|
return strcmp (((RCmdDescriptor *)a)->cmd, ((RCmdDescriptor *)b)->cmd);
|
|
}
|
|
|
|
static void cmd_descriptor_init(RCore *core) {
|
|
const ut8 *p;
|
|
RListIter *iter;
|
|
RCmdDescriptor *x, *y;
|
|
int n = core->cmd_descriptors->length;
|
|
r_list_sort (core->cmd_descriptors, compare_cmd_descriptor_name);
|
|
r_list_foreach (core->cmd_descriptors, iter, y) {
|
|
if (--n < 0) {
|
|
break;
|
|
}
|
|
x = &core->root_cmd_descriptor;
|
|
for (p = (const ut8 *)y->cmd; *p; p++) {
|
|
if (!x->sub[*p]) {
|
|
if (p[1]) {
|
|
RCmdDescriptor *d = R_NEW0 (RCmdDescriptor);
|
|
r_list_append (core->cmd_descriptors, d);
|
|
x->sub[*p] = d;
|
|
} else {
|
|
x->sub[*p] = y;
|
|
}
|
|
} else if (!p[1]) {
|
|
eprintf ("Command '%s' is duplicated, please check\n", y->cmd);
|
|
}
|
|
x = x->sub[*p];
|
|
}
|
|
}
|
|
}
|
|
|
|
static int core_cmd0_wrapper(void *core, const char *cmd) {
|
|
return r_core_cmd0 ((RCore *)core, cmd);
|
|
}
|
|
|
|
R_API void r_core_cmd_init(RCore *core) {
|
|
struct {
|
|
const char *cmd;
|
|
const char *description;
|
|
RCmdCb cb;
|
|
void (*descriptor_init)(RCore *core, RCmdDesc *parent);
|
|
const RCmdDescHelp *help;
|
|
const RCmdDescHelp *group_help;
|
|
RCmdDescType type;
|
|
RCmdArgvCb argv_cb;
|
|
} cmds[] = {
|
|
{"!", "run system command", cmd_system, NULL, &system_help},
|
|
{"_", "print last output", cmd_last, NULL, &underscore_help},
|
|
{"#", "calculate hash", cmd_hash, NULL, &hash_help},
|
|
{"$", "alias", cmd_alias, NULL, &alias_help},
|
|
{"%", "short version of 'env' command", cmd_env, NULL, &env_help},
|
|
{"&", "tasks", cmd_tasks, NULL, &tasks_help},
|
|
{"(", "macro", cmd_macro, cmd_macro_init, ¯o_help},
|
|
{"*", "pointer read/write", cmd_pointer, NULL, &pointer_help},
|
|
{"-", "open cfg.editor and run script", cmd_stdin, NULL, &stdin_help},
|
|
{".", "interpret", cmd_interpret, NULL, &interpret_help},
|
|
{",", "create and manipulate tables", cmd_table, NULL, &table_help},
|
|
{"/", "search kw, pattern aes", cmd_search, cmd_search_init, &search_help},
|
|
{"=", "io pipe", cmd_rap, NULL, &rap_help},
|
|
{"?", "help message", cmd_help, cmd_help_init, &help_help},
|
|
{"\\","alias for =!", cmd_rap_run, NULL, &rap_run_help},
|
|
{"'", "alias for =!", cmd_rap_run, NULL, &rap_run_help},
|
|
{"0", "alias for s 0x", cmd_ox, NULL, &zero_help},
|
|
{"a", "analysis", cmd_anal, cmd_anal_init, &anal_help},
|
|
{"b", "change block size", cmd_bsize, NULL, &b_help},
|
|
{"c", "compare memory", cmd_cmp, cmd_cmp_init, &c_help},
|
|
{"C", "code metadata", cmd_meta, cmd_meta_init, &C_help},
|
|
{"d", "debugger operations", cmd_debug, cmd_debug_init, &d_help},
|
|
{"e", "evaluate configuration variable", cmd_eval, cmd_eval_init, &e_help},
|
|
{"f", "get/set flags", cmd_flag, cmd_flag_init, &f_help},
|
|
{"g", "egg manipulation", cmd_egg, cmd_egg_init, &g_help},
|
|
{"i", "get file info", cmd_info, cmd_info_init, &i_help},
|
|
{"k", "perform sdb query", cmd_kuery, NULL, &k_help},
|
|
{"l", "list files and directories", cmd_ls, NULL, &l_help},
|
|
{"j", "join the contents of the two files", cmd_join, NULL, &j_help},
|
|
{"h", "show the top n number of line in file", cmd_head, NULL, &h_help},
|
|
{"L", "manage dynamically loaded plugins", cmd_plugins, NULL, &L_help},
|
|
{"m", "mount filesystem", cmd_mount, cmd_mount_init, &m_help},
|
|
{"o", "open or map file", cmd_open, cmd_open_init, &o_help},
|
|
{"p", "print current block", cmd_print, cmd_print_init, &p_help},
|
|
{"P", "project", cmd_project, cmd_project_init, &P_help},
|
|
{"q", "exit program session", cmd_quit, cmd_quit_init, &q_help},
|
|
{"Q", "alias for q!", cmd_Quit, NULL, &Q_help},
|
|
{":", "long commands starting with :", cmd_colon, NULL, &colon_help},
|
|
{"r", "change file size", cmd_resize, NULL, &r_help},
|
|
{"s", "seek to an offset", cmd_seek, cmd_seek_init, &s_help},
|
|
{"t", "type information (cparse)", cmd_type, cmd_type_init, &t_help},
|
|
{"T", "Text log utility", cmd_log, cmd_log_init, &T_help},
|
|
{"u", "uname/undo", cmd_undo, NULL, &u_help},
|
|
{"<", "pipe into RCons.readChar", cmd_pipein, NULL, &pipein_help},
|
|
{"V", "enter visual mode", cmd_visual, NULL, &V_help},
|
|
{"v", "enter visual mode", cmd_panels, NULL, &v_help},
|
|
{"w", "write bytes", cmd_write, cmd_write_init, &w_help, &w_group_help, R_CMD_DESC_TYPE_GROUP, w_handler},
|
|
{"x", "alias for px", cmd_hexdump, NULL, &x_help},
|
|
{"y", "yank bytes", cmd_yank, NULL, &y_help},
|
|
{"z", "zignatures", cmd_zign, cmd_zign_init, &z_help},
|
|
};
|
|
|
|
core->rcmd = r_cmd_new ();
|
|
core->rcmd->macro.user = core;
|
|
core->rcmd->macro.num = core->num;
|
|
core->rcmd->macro.cmd = core_cmd0_wrapper;
|
|
core->rcmd->nullcallback = r_core_cmd_nullcallback;
|
|
core->rcmd->macro.cb_printf = (PrintfCallback)r_cons_printf;
|
|
r_cmd_set_data (core->rcmd, core);
|
|
core->cmd_descriptors = r_list_newf (free);
|
|
|
|
RCmdDesc *root = r_cmd_get_root (core->rcmd);
|
|
size_t i;
|
|
for (i = 0; i < R_ARRAY_SIZE (cmds); i++) {
|
|
r_cmd_add (core->rcmd, cmds[i].cmd, cmds[i].cb);
|
|
|
|
RCmdDesc *cd = NULL;
|
|
switch (cmds[i].type) {
|
|
case R_CMD_DESC_TYPE_OLDINPUT:
|
|
cd = r_cmd_desc_oldinput_new (core->rcmd, root, cmds[i].cmd, cmds[i].cb, cmds[i].help);
|
|
break;
|
|
case R_CMD_DESC_TYPE_ARGV:
|
|
cd = r_cmd_desc_argv_new (core->rcmd, root, cmds[i].cmd, cmds[i].argv_cb, cmds[i].help);
|
|
break;
|
|
case R_CMD_DESC_TYPE_INNER:
|
|
cd = r_cmd_desc_inner_new (core->rcmd, root, cmds[i].cmd, cmds[i].help);
|
|
break;
|
|
case R_CMD_DESC_TYPE_GROUP:
|
|
cd = r_cmd_desc_group_new (core->rcmd, root, cmds[i].cmd, cmds[i].argv_cb, cmds[i].help, cmds[i].group_help);
|
|
break;
|
|
}
|
|
if (cd && cmds[i].descriptor_init) {
|
|
cmds[i].descriptor_init (core, cd);
|
|
}
|
|
}
|
|
DEFINE_CMD_DESCRIPTOR_SPECIAL (core, $, dollar);
|
|
DEFINE_CMD_DESCRIPTOR_SPECIAL (core, %, percent);
|
|
DEFINE_CMD_DESCRIPTOR_SPECIAL (core, *, star);
|
|
DEFINE_CMD_DESCRIPTOR_SPECIAL (core, ., dot);
|
|
DEFINE_CMD_DESCRIPTOR_SPECIAL (core, =, equal);
|
|
|
|
DEFINE_CMD_DESCRIPTOR (core, b);
|
|
DEFINE_CMD_DESCRIPTOR (core, k);
|
|
DEFINE_CMD_DESCRIPTOR (core, r);
|
|
DEFINE_CMD_DESCRIPTOR (core, u);
|
|
DEFINE_CMD_DESCRIPTOR (core, y);
|
|
cmd_descriptor_init (core);
|
|
}
|