radare2/libr/core/cmd.c
pancake c673ede6b7 * Bug identified in cons with r_core_cmd_str identified
- We have to grep stuff when inserting it, not when flushing
* Fix 'pd' command (initialize base address
* Initial import of the foreach @@ syntax sugar
  - Basic flag support, allows @@= for lists and @@.() for macros
* Add 'g' and 'G' keys in visual
* Fix issues in the macro engine and usage
  - Proper identification of dupped commands
  - Split macro name and arguments in internal structures
  - Chop last ')' to make () and (,) macros work happily
* Fix build of r_sign and a segfault in _free() method
2009-03-20 21:05:12 +00:00

1466 lines
38 KiB
C

/* radare - LGPL - Copyright 2009 pancake<nopcode.org> */
#include "r_core.h"
#include "r_flags.h"
#include "r_hash.h"
#include "r_asm.h"
#include "r_anal.h"
/* Ugly */
#include "../asm/arch/x86/bea/BeaEngine.h"
#include <stdarg.h>
int r_core_write_op(struct r_core_t *core, const char *arg, char op);
static int cmd_print(void *data, const char *input);
static int cmd_iopipe(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case '\0':
r_lib_list(&core->lib);
r_io_handle_list(&core->io);
break;
default:
r_io_system(&core->io, core->file->fd, input);
break;
}
return R_TRUE;
}
static int cmd_yank(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case ' ':
r_core_yank(core, core->seek, atoi(input+1));
break;
case 'y':
r_core_yank_paste(core, r_num_math(&core->num, input+2), 0);
break;
case '\0':
if (core->yank == NULL) {
fprintf(stderr, "No buffer yanked already\n");
} else {
int i;
r_cons_printf("0x%08llx %d ",
core->yank_off, core->yank_len);
for(i=0;i<core->yank_len;i++)
r_cons_printf("%02x", core->yank[i]);
r_cons_newline();
}
break;
default:
r_cons_printf(
"Usage: y[y] [len] [[@]addr]\n"
" y ; show yank buffer information (srcoff len bytes)\n"
" y 16 ; copy 16 bytes into clipboard\n"
" y 16 0x200 ; copy 16 bytes into clipboard from 0x200\n"
" y 16 @ 0x200 ; copy 16 bytes into clipboard from 0x200\n"
" yy 0x3344 ; paste clipboard\n");
break;
}
return R_TRUE;
}
static int cmd_quit(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case '?':
fprintf(stderr,
"Usage: q[!] [retvalue]\n"
" q ; quit program\n"
" q! ; force quit (no questions)\n"
" q 1 ; quit with return value 1\n"
" q a-b ; quit with return value a-b\n");
break;
case '\0':
case ' ':
case '!':
default:
exit(r_num_math(&core->num, input+1));
break;
}
return 0;
}
static int cmd_interpret(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case ' ':
/* interpret file */
r_core_cmd_file(core, input+1);
break;
case '!':
/* from command */
fprintf(stderr, "TODO\n");
break;
case '(':
//fprintf(stderr, "macro call (%s)\n", input+1);
r_macro_call(&core->macro, input+1);
break;
case '?':
fprintf(stderr,
"Usage: . [file] | [!command] | [(macro)]\n"
" . foo.rs ; interpret r script\n"
" .!rabin -ri $FILE ; interpret output of command\n"
" .(foo 1 2 3) ; run macro 'foo' with args 1, 2, 3\n"
" ./m ELF ; interpret output of command /m ELF as r. commands\n");
break;
default:
{
char *str = r_core_cmd_str(core, input);
char *ptr = str;
while(1) {
char *eol = strchr(ptr, '\n');
if (eol) eol[0]='\0';
r_core_cmd(core, ptr, 0);
if (!eol) break;
ptr = eol+1;
}
free(str);
}
break;
}
return 0;
}
static int cmd_seek(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
if (input[0]=='\0') {
r_cons_printf("0x%llx\n", core->seek);
} else {
int idelta = (input[1]==' ')?2:1;
u64 off = r_num_math(&core->num, input+idelta);
if (input[0]==' ' && (input[1]=='+'||input[1]=='-'))
input = input+1;
switch(input[0]) {
case ' ':
r_core_seek(core, off);
break;
case '+':
if (input[1]=='+') r_core_seek_delta(core, core->blocksize);
else r_core_seek_delta(core, off);
break;
case '-':
if (input[1]=='-') r_core_seek_delta(core, -core->blocksize);
else r_core_seek_delta(core, -off);
break;
case 'a':
{
u64 alsz = core->blocksize;
if (input[1]&&input[2]) {
char *cmd, *p;
cmd = strdup(input);
p = strchr(cmd+2, ' ');
if (p) {
alsz = r_num_math(&core->num, p+1);;
*p='\0';
}
cmd[0]='s';
// perform real seek if provided
r_cmd_call(&core->cmd, cmd);
free(cmd);
}
r_core_seek_align(core, alsz, 0);
}
break;
case '?':
fprintf(stderr,
"Usage: s[+-] [addr]\n"
" s 0x320 ; seek to this address\n"
" s++ ; seek blocksize bytes forward\n"
" s-- ; seek blocksize bytes backward\n"
" s+ 512 ; seek 512 bytes forward\n"
" s- 512 ; seek 512 bytes backward\n"
" sa [[+-]a] [asz] ; seek asz (or bsize) aligned to addr\n");
break;
}
}
return 0;
}
static int cmd_help(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
u64 n;
switch(input[0]) {
case ' ':
n = r_num_math(&core->num, input+1);
r_cons_printf("%lld 0x%llx\n", n,n);
break;
case '=':
r_num_math(&core->num, input+1);
break;
case '+':
if (input[1]) {
if (core->num.value & U64_GT0)
r_core_cmd(core, input+1, 0);
} else r_cons_printf("0x%llx\n", core->num.value);
break;
case '-':
if (input[1]) {
if (core->num.value & U64_LT0)
r_core_cmd(core, input+1, 0);
} else r_cons_printf("0x%llx\n", core->num.value);
break;
case '!': // ??
if (input[1]) {
if (core->num.value != U64_MIN)
r_core_cmd(core, input+1, 0);
} else r_cons_printf("0x%llx\n", core->num.value);
break;
case 'z':
for(input=input+1;input[0]==' ';input=input+1);
core->num.value = strlen(input);
break;
case 't':
{
struct r_prof_t prof;
r_prof_start(&prof);
r_core_cmd(core, input+1, 0);
r_prof_end(&prof);
core->num.value = (u64)prof.result;
eprintf("%lf\n", prof.result);
}
break;
case '?': // ???
if (input[1]=='?') {
fprintf(stderr,
"Usage: ?[?[?]] expression\n"
" ? eip-0x804800 ; calculate result for this math expr\n"
" ?= eip-0x804800 ; same as above without user feedback\n"
" ?? [cmd] ; ? == 0 run command when math matches\n"
" ?z str ; returns the length of string (0 if null)\n"
" ?t cmd ; returns the time to run a command\n"
" ?! [cmd] ; ? != 0\n"
" ?+ [cmd] ; ? > 0\n"
" ?- [cmd] ; ? < 0\n"
" ??? ; show this help\n");
} else
if (input[1]) {
if (core->num.value == U64_MIN)
r_core_cmd(core, input+1, 0);
} else r_cons_printf("0x%llx\n", core->num.value);
break;
case '\0':
default:
r_cons_printf(
"Usage:\n"
" a ; perform analysis of code\n"
" b [bsz] ; get or change block size\n"
" C[CFf..] ; Code metadata management\n"
" d[hrscb] ; debugger commands\n"
" e [a[=b]] ; list/get/set config evaluable vars\n"
" f [name][sz][at] ; set flag at current address\n"
" s [addr] ; seek to address\n"
" i [file] ; get info about opened file\n"
" p?[len] ; print current block with format and length\n"
" V[vcmds] ; enter visual mode (vcmds=visualvisual keystrokes)\n"
" w[mode] [arg] ; multiple write operations\n"
" x [len] ; alias for 'px' (print hexadecimal\n"
" y [len] [off] ; yank/paste bytes from/to memory\n"
" ? [expr] ; help or evaluate math expression\n"
" /[xmp/] ; search for bytes, regexps, patterns, ..\n"
" |[cmd] ; run this command thru the io pipe (no args=list)\n"
" #[algo] [len] ; calculate hash checksum of current block\n"
" .[ file|!cmd|cmd|(macro)] ; interpret as radare cmds\n"
" (macro arg0 arg1) ; define scripting macros\n"
" q [ret] ; quit program with a return value\n"
"Append '?' to any char command to get detailed help\n");
break;
}
return 0;
}
static int cmd_bsize(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case '\0':
r_cons_printf("0x%x\n", core->blocksize);
break;
default:
//input = r_str_clean(input);
r_core_block_size(core, r_num_math(NULL, input));
break;
}
return 0;
}
static int cmd_hexdump(void *data, const char *input)
{
return cmd_print(data, input-1);
}
static int cmd_info(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case '?':
break;
case '*':
break;
default:
r_cons_printf("URI: %s\n", core->file->uri);
r_cons_printf("blocksize: 0x%x\n", core->blocksize);
}
return 0;
}
static int cmd_print(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
int l, len = core->blocksize;
u32 tbs = core->blocksize;
int show_offset = r_config_get_i(&core->config, "asm.offset");
int show_bytes = r_config_get_i(&core->config, "asm.bytes");
if (input[0] && input[1]) {
l = (int) r_num_get(&core->num, input+2);
if (l>0) len = l;
if (l>tbs) {
r_core_block_size(core, l);
len = l;
}
}
switch(input[0]) {
case 'd':
{
/* XXX hardcoded */
int ret, idx;
u8 *buf = core->block;
int pseudo = (int)r_config_get_i(&core->config, "asm.pseudo");
char str[128];
struct r_asm_aop_t aop;
r_asm_set_pc(&core->assembler, core->seek);
for(idx=ret=0; idx < len; idx+=ret) {
r_asm_set_pc(&core->assembler, core->assembler.pc + ret);
ret = r_asm_disassemble(&core->assembler, &aop, buf+idx, len-idx);
if (ret <1) {
ret = 1;
fprintf(stderr, "** invalid opcode at 0x%08llx **\n", core->assembler.pc + ret);
}
if (show_offset) r_cons_printf("0x%08llx ", core->seek + idx);
if (show_bytes) r_cons_printf("%14s ", aop.buf_hex);
if (pseudo) {
r_parse_parse(&core->parser, aop.buf_asm, str);
r_cons_printf("%s\n", str);
} else r_cons_printf("%s\n", aop.buf_asm);
}
}
break;
case 's':
r_print_string(&core->print, core->seek, core->block, len, 0, 1, 0); //, 78, 1);
break;
case 'S':
r_print_string(&core->print, core->seek, core->block, len, 1, 1, 0); //, 78, 1);
break;
case 'u':
r_print_string(&core->print, core->seek, core->block, len, 0, 1, 1); //, 78, 1);
break;
case 'U':
r_print_string(&core->print, core->seek, core->block, len, 1, 1, 1); //, 78, 1);
break;
case 'c':
r_print_code(&core->print, core->seek, core->block, len); //, 78, 1);
break;
case 'r':
r_print_raw(&core->print, core->block, len);
break;
case 'o':
r_print_hexdump(&core->print, core->seek, core->block, len, 8, 1); //, 78, !(input[1]=='-'));
break;
case 'x':
r_print_hexdump(&core->print, core->seek, core->block, len, 16, 1); //, 78, !(input[1]=='-'));
break;
case '8':
r_print_bytes(&core->print, core->block, len, "%02x");
break;
default:
r_cons_printf("Usage: p[8] [len]\n"
" p8 [len] 8bit hexpair list of bytes\n"
" px [len] hexdump of N bytes\n"
" po [len] octal dump of N bytes\n"
" pc [len] output C format\n"
" ps [len] print string\n"
" pS [len] print wide string\n"
" pd [len] disassemble N bytes\n"
" pr [len] print N raw bytes\n"
" pu [len] print N url encoded bytes\n"
" pU [len] print N wide url encoded bytes\n");
break;
}
if (tbs != core->blocksize)
r_core_block_size(core, tbs);
return 0;
}
static int cmd_flag(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
int len = strlen(input)+1;
char *str = alloca(len);
memcpy(str, input+1, len);
switch(input[0]) {
case '+':
r_flag_set(&core->flags, str, core->seek, core->blocksize, 1);
break;
case ' ': {
char *s = NULL, *s2 = NULL;
u64 seek = core->seek;
u32 bsze = core->blocksize;
s = strchr(str, ' ');
if (s) {
*s = '\0';
s2 = strchr(s+1, ' ');
if (s2) {
*s2 = '\0';
seek = r_num_math(&core->num, s2+1);
}
bsze = r_num_math(&core->num, s+1);
}
r_flag_set(&core->flags, str, seek, bsze, 0);
if (s) *s=' ';
if (s2) *s2=' ';
}
break;
case '-':
r_flag_unset(&core->flags, input+1);
break;
case 'b':
r_flag_set_base(&core->flags, r_num_math(&core->num, input+1));
break;
case 's':
if (input[1]==' ')
r_flag_space_set(&core->flags, input+2);
else r_flag_space_list(&core->flags);
break;
case '*':
r_flag_list(&core->flags, 1);
break;
case '\0':
r_flag_list(&core->flags, 0);
break;
case '?':
fprintf(stderr, "Usage: f[ ] [flagname]\n"
" fb 0x8048000 ; set base address for flagging\n"
" f name 12 @ 33 ; set flag 'name' with size 12 at 33\n"
" f name 12 33 ; same as above\n"
" f+name 12 @ 33 ; like above but creates new one if doesnt exist\n"
" f-name ; remove flag 'name'\n"
" f ; list flags\n"
" f* ; list flags in r commands\n");
break;
}
return 0;
}
static int cmd_anal(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
int l, len = core->blocksize;
u32 tbs = core->blocksize;
if (input[0] && input[1]) {
l = (int) r_num_get(&core->num, input+2);
if (l>0) len = l;
if (l>tbs) {
r_core_block_size(core, l);
len = l;
}
}
switch(input[0]) {
case '\0':
r_anal_list(&core->anal);
break;
case 'o':
{
/* XXX hardcoded */
int ret, idx;
u8 *buf = core->block;
struct r_asm_aop_t aop;
struct r_anal_aop_t analop;
DISASM *disasm_obj;
r_asm_set(&core->assembler, "asm_x86_bea");
r_anal_set(&core->anal, "anal_x86_bea");
r_asm_set_pc(&core->assembler, core->seek);
r_anal_set_pc(&core->anal, core->seek);
for(idx=ret=0; idx < len; idx+=ret) {
r_asm_set_pc(&core->assembler, core->assembler.pc + ret);
ret = r_asm_disassemble(&core->assembler, &aop, buf+idx, len-idx);
disasm_obj = aop.disasm_obj;
r_anal_aop(&core->anal, &analop, disasm_obj);
}
}
break;
default:
fprintf(stderr, "Usage: a[o] [len]\n"
" ao [len] Analyze raw bytes\n");
break;
}
if (tbs != core->blocksize)
r_core_block_size(core, tbs);
return 0;
}
/* TODO: simplify using r_write */
static int cmd_write(void *data, const char *input)
{
int i, len = strlen(input);
char *tmp, *str = alloca(len)+1;
struct r_core_t *core = (struct r_core_t *)data;
memcpy(str, input+1, len);
switch(input[0]) {
case ' ':
// write strifng
len = r_str_escape(str);
r_io_lseek(&core->io, core->file->fd, core->seek, R_IO_SEEK_SET);
r_io_write(&core->io, core->file->fd, str, len);
r_core_block_read(core, 0);
break;
case 't':
case 'f':
fprintf(stderr, "TODO: wf, wt\n");
break;
case 'w':
str = str+1;
len = len-1;
len *= 2;
tmp = alloca(len);
for(i=0;i<len;i++) {
if (i%2) tmp[i] = 0;
else tmp[i] = str[i>>1];
}
str = tmp;
// write strifng
r_io_lseek(&core->io, core->file->fd, core->seek, R_IO_SEEK_SET);
r_io_write(&core->io, core->file->fd, str, len);
r_core_block_read(core, 0);
break;
case 'x':
{
int len = strlen(input);
char *buf = alloca(len);
len = r_hex_str2bin(input+1, buf);
r_io_lseek(&core->io, core->file->fd, core->seek, R_IO_SEEK_SET);
r_io_write(&core->io, core->file->fd, buf, len);
r_core_block_read(core, 0);
}
// write hexpairs
break;
case 'b':
{
int len = strlen(input);
char *buf = alloca(len);
len = r_hex_str2bin(input+1, buf);
r_mem_copyloop(core->block, buf, core->blocksize, len);
r_io_lseek(&core->io, core->file->fd, core->seek, R_IO_SEEK_SET);
r_io_write(&core->io, core->file->fd, core->block, core->blocksize);
r_core_block_read(core, 0);
}
break;
case 'm':
{
int len = strlen(input+1);
len = r_hex_str2bin(input+1, str);
switch(input[1]) {
case '\0':
fprintf(stderr, "Current write mask: TODO\n");
// TODO
break;
case '?':
break;
case '-':
r_io_set_write_mask(&core->io, -1, 0, 0);
fprintf(stderr, "Write mask disabled\n");
break;
case ' ':
if (len == 0) {
fprintf(stderr, "Invalid string\n");
} else {
r_io_set_write_mask(&core->io, core->file->fd, str, len);
fprintf(stderr, "Write mask set to '");
for (i=0;i<len;i++)
fprintf(stderr, "%02x", str[i]);
fprintf(stderr, "'\n");
}
break;
}
}
break;
case 'v':
{
u64 off = r_num_math(&core->num, input+1);
r_io_lseek(&core->io, core->file->fd, core->seek, R_IO_SEEK_SET);
if (off&U64_32U) {
/* 8 byte addr */
u64 addr8;
memcpy((u8*)&addr8, (u8*)&off, 8); // XXX needs endian here
// endian_memcpy((u8*)&addr8, (u8*)&off, 8);
r_io_write(&core->io, core->file->fd, (const u8 *)&addr8, 8);
} else {
/* 4 byte addr */
u32 addr4, addr4_ = (u32)off;
//drop_endian((u8*)&addr4_, (u8*)&addr4, 4); /* addr4_ = addr4 */
//endian_memcpy((u8*)&addr4, (u8*)&addr4_, 4); /* addr4 = addr4_ */
memcpy((u8*)&addr4, (u8*)&addr4_, 4); // XXX needs endian here too
r_io_write(&core->io, core->file->fd, (const u8 *)&addr4, 4);
}
r_core_block_read(core, 0);
}
break;
case 'o':
switch(input[1]) {
case 'a':
case 's':
case 'A':
case 'x':
case 'r':
case 'l':
case 'm':
case 'd':
case 'o':
if (input[2]!=' ') {
fprintf(stderr, "Usage: 'wo%c 00 11 22'\n", input[1]);
return 0;
}
case '2':
case '4':
r_core_write_op(core, input+3, input[1]);
break;
case '\0':
case '?':
default:
fprintf(stderr,
"Usage: wo[xrlasmd] [hexpairs]\n"
"Example: wox 90 ; xor cur block with 90\n"
"Example: woa 02 03 ; add 2, 3 to all bytes of cur block\n"
"Supported operations:\n"
" woa addition +=\n"
" wos substraction -=\n"
" wom multiply *=\n"
" wod divide /=\n"
" wox xor ^=\n"
" woo or |=\n"
" woA and &=\n"
" wor shift right >>=\n"
" wol shift left <<=\n"
" wo2 2 byte endian swap 2=\n"
" wo4 4 byte endian swap 4=\n"
);
break;
}
break;
default:
case '?':
if (core->oobi) {
fprintf(stderr, "Writing oobi buffer!\n");
r_io_write(&core->io, core->file->fd, core->oobi, core->oobi_len);
} else
r_cons_printf("Usage: w[x] [str] [<file] [<<EOF] [@addr]\n"
" w foobar ; write string 'foobar'\n"
" ww foobar ; write wide string 'f\\x00o\\x00o\\x00b\\x00a\\x00r\\x00'\n"
" wb 010203 ; fill current block with cyclic hexpairs\n"
" wx 9090 ; write two intel nops\n"
" wv eip+34 ; write 32-64 bit value\n"
" wo[] hex ; write in block with operation. 'wo?' fmi\n"
" wm f0ff ; cyclic write mask\n");
break;
}
return 0;
}
static int __cb_hit(struct r_search_kw_t *kw, void *user, u64 addr)
{
r_cons_printf("f hit0_%d %d 0x%08llx\n", kw->count, kw->keyword_length, addr);
return R_TRUE;
}
static int cmd_search(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
u64 at;
u32 n32;
int ret, dosearch = 0;
u8 *buf;
switch (input[0]) {
case '/':
r_search_begin(core->search);
dosearch = 1;
break;
case 'v':
r_search_free(core->search);
core->search = r_search_new(R_SEARCH_KEYWORD);
n32 = r_num_math(&core->num, input+1);
r_search_kw_add_bin(core->search, &n32, 4, "",0);
r_search_begin(core->search);
dosearch = 1;
break;
case ' ': /* search string */
r_search_free(core->search);
core->search = r_search_new(R_SEARCH_KEYWORD);
r_search_kw_add(core->search, input+1, "");
r_search_begin(core->search);
dosearch = 1;
break;
case 'm': /* match regexp */
{
char *inp = strdup(input+2);
char *res = r_str_lchr(inp+1, inp[0]);
char *opt = NULL;
if (res > inp) {
opt = strdup(res+1);
res[1]='\0';
}
r_search_free(core->search);
core->search = r_search_new(R_SEARCH_REGEXP);
r_search_kw_add(core->search, inp, opt);
r_search_begin(core->search);
dosearch = 1;
free(inp);
free(opt);
}
break;
case 'x': /* search hex */
r_search_free(core->search);
core->search = r_search_new(R_SEARCH_KEYWORD);
r_search_kw_add_hex(core->search, input+2, "");
r_search_begin(core->search);
dosearch = 1;
break;
default:
r_cons_printf("Usage: /[xm/] [arg]\n"
" / foo ; search for string 'foo'\n"
" /m /E.F/i ; match regular expression\n"
" /x ff0033 ; search for hex string\n"
" // ; repeat last search\n");
break;
}
if (core->search->n_kws==0) {
printf("No keywords defined\n");
} else
if (dosearch) {
/* set callback */
/* TODO: handle last block of data */
/* TODO: handle ^C */
/* TODO: launch search in background support */
buf = (u8 *)malloc(core->blocksize);
r_search_set_callback(core->search, &__cb_hit, &core);
r_cons_break(NULL, NULL);
for(at = core->seek; at < core->file->size; at += core->blocksize) {
if (r_cons_breaked)
break;
r_io_lseek(&core->io, core->file->fd, at, R_IO_SEEK_SET);
ret = r_io_read(&core->io, core->file->fd, buf, core->blocksize);
if (ret != core->blocksize)
break;
if (r_search_update(core->search, &at, buf, ret) == -1) {
printf("search:update error\n");
break;
}
}
r_cons_break_end();
}
return R_TRUE;
}
static int cmd_eval(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case '\0':
r_config_list(&core->config, NULL, 0);
break;
case '-':
r_config_init(&core->config, core);
break;
case '*':
r_config_list(&core->config, NULL, 1);
break;
case '?':
r_cons_printf(
"Usage: e[?] [var[=value]]\n"
" e ; list config vars\n"
" e- ; reset config vars\n"
" e* ; dump config vars in r commands\n"
" e a ; get value of var 'a'\n"
" e a=b ; set var 'a' the 'b' value\n");
//r_cmd_help(&core->cmd, "e");
break;
default:
r_config_eval(&core->config, input);
}
return 0;
}
static int cmd_hash(void *data, const char *input)
{
char algo[32];
struct r_core_t *core = (struct r_core_t *)data;
u32 len = core->blocksize;
const char *ptr;
if (input[0]=='!') {
#if 0
#!lua < file
#!lua <<EOF
#!lua
#!lua foo bar
#endif
if (input[1]=='\0') {
r_lang_list(&core->lang);
return R_TRUE;
}
// TODO: set argv here
r_lang_set(&core->lang, input+1);
if (core->oobi)
r_lang_run(&core->lang, core->oobi, core->oobi_len);
else r_lang_prompt(&core->lang);
return R_TRUE;
}
ptr = strchr(input, ' ');
sscanf(input, "%31s", algo);
if (ptr != NULL)
len = r_num_math(&core->num, ptr+1);
if (!r_str_ccmp(input, "crc32", ' ')) {
r_cons_printf("%04x\n", r_hash_crc32(core->block, len));
} else
if (!r_str_ccmp(input, "crc16", ' ')) {
r_cons_printf("%02x\n", r_hash_crc16(0, core->block, len));
} else {
r_cons_printf(
"Usage: #algo <size> @ addr\n"
" #crc32 ; calculate crc32 of current block\n"
" #crc32 < /etc/fstab ; calculate crc32 of this file\n"
" #md5 128K @ edi ; calculate md5 of 128K from 'edi'\n"
"Usage #!interpreter [<args>] [<file] [<<eof]\n"
" #! ; list all available interpreters\n"
" #!python ; run python commandline\n"
" #!python < foo.py ; run foo.py python script\n"
" #!python <<EOF ; get python code until 'EOF' mark\n"
" #!python arg0 a1 <<q ; set arg0 and arg1 and read until 'q'\n");
}
return 0;
}
static int cmd_visual(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
r_core_visual(core, input);
return 0;
}
static int cmd_system(void *data, const char *input)
{
//struct r_core_t *core = (struct r_core_t *)data;
// slurped from teh old radare_system
#if __FreeBSD__
/* freebsd system() is broken */
int fds[2];
int st,pid;
char *argv[] ={ "/bin/sh", "-c", input, NULL};
pipe(fds);
/* not working ?? */
//pid = rfork(RFPROC|RFCFDG);
pid = vfork();
if (pid == 0) {
dup2(1, fds[1]);
execv(argv[0], argv);
_exit(127); /* error */
} else {
dup2(1, fds[0]);
waitpid(pid, &st, 0);
}
return WEXITSTATUS(st);
#else
return system(input);
#endif
}
static int cmd_meta(void *data, const char *input)
{
//struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case '\0':
/* meta help */
break;
case 'C': /* add comment */
// r_meta_add(&core->meta);
break;
case 'F': /* add function */
break;
}
return R_TRUE;
}
static int cmd_io_system(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
return r_io_system(&core->io, core->file->fd, input);
}
static int cmd_macro(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case ')':
r_macro_break(&core->macro, input+1);
break;
case '-':
r_macro_rm(&core->macro, input+1);
break;
case '\0':
r_macro_list(&core->macro);
break;
case '?':
eprintf(
"Usage: (foo\\n..cmds..\\n)\n"
" Record macros grouping commands\n"
" (foo args\\n ..) ; define a macro\n"
" (-foo) ; remove a macro\n"
" .(foo) ; to call it\n"
" () ; break inside macro\n"
"Argument support:\n"
" (foo x y\\n$1 @ $2) ; define fun with args\n"
" .(foo 128 0x804800) ; call it with args\n"
"Iterations:\n"
" .(foo\\n() $@) ; define iterator returning iter index\n"
" x @@ .(foo) ; iterate over them\n"
);
break;
default:
r_macro_add(&core->macro, input);
break;
}
return 0;
}
static int r_core_cmd_subst(struct r_core_t *core, char *cmd, int *rs, int *rfd, int *times)
{
char *ptr, *ptr2, *str, *p;
int i, len = strlen(cmd);
len = atoi(cmd);
if (len>0) {
for(i=0;cmd[i]>='0'&&cmd[i]<='9'; i++);
if (i>0) strcpy(cmd, cmd+i);
*times = len;
}
if (cmd[0]=='\0')
return 0;
ptr = strchr(cmd, ';');
if (ptr)
ptr[0]='\0';
ptr = strchr(cmd+1, '|');
if (ptr) {
ptr[0] = '\0';
eprintf("System pipes not yet supported.\n");
}
/* Out Of Band Input */
free(core->oobi);
core->oobi = NULL;
ptr = strchr(cmd, '<');
if (ptr) {
ptr[0] = '\0';
if (ptr[1]=='<') {
/* this is a bit mess */
char *oprompt = r_line_prompt;
oprompt = ">";
for(str=ptr+2;str[0]== ' ';str=str+1);
eprintf("==> Reading from stdin until '%s'\n", str);
free(core->oobi);
core->oobi = malloc(1);
core->oobi[0] = '\0';
core->oobi_len = 0;
for (;;) {
char buf[1024];
int ret;
printf("> "); fflush(stdout);
fgets(buf, 1023, stdin);
if (feof(stdin))
break;
buf[strlen(buf)-1]='\0';
ret = strlen(buf);
core->oobi_len+=ret;
core->oobi = realloc(core->oobi, core->oobi_len+1);
if (!strcmp(buf, str))
break;
strcat(core->oobi, buf);
}
r_line_prompt = oprompt;
} else {
for(str=ptr+1;str[0]== ' ';str=str+1);
eprintf("SLURPING FILE '%s'\n", str);
core->oobi = r_file_slurp(str, &core->oobi_len);
if (core->oobi == NULL)
eprintf("Cannot open file\n");
else if (ptr == cmd)
return r_core_cmd_buffer(core, core->oobi);
}
}
/* Pipe console to file */
ptr = strchr(cmd, '>');
if (ptr) {
ptr[0] = '\0';
for(str=ptr+1; str[0]== ' '; str=str+1);
*rfd = r_cons_pipe_open(str, ptr[1]=='>');
}
ptr = strchr(cmd, '@');
if (ptr) {
char *pt = ptr;
ptr[0]='\0';
while(pt[0]==' '||pt[0]=='\t') {
pt[0]='\0';
pt = pt-1;
}
*rs = 1;
if (ptr[1]=='@') {
//fprintf(stderr, "TODO: foreach @ loop\n");
// TODO: remove
u64 tmpoff = core->seek;
r_core_cmd_foreach(core, cmd, ptr+2);
//config.seek = tmpoff;
r_core_seek(core, tmpoff);
return 0;
} else r_core_seek(core, r_num_math(&core->num, ptr+1));
}
while(((ptr = strchr(cmd, '`')))) {
ptr2 = strchr(ptr+1, '`');
if (ptr2==NULL) {
fprintf(stderr, "parse: Missing '`' in expression (%s).\n", ptr+1);
return 0;
}
ptr2[0]='\0';
fprintf(stderr, "Piping (%s)\n", ptr+1);
str = r_core_cmd_str(core, ptr+1);
for(i=0;str[i];i++) if (str[i]=='\n') str[i]=' ';
r_str_inject(ptr, ptr2+1, str, 1024); // XXX overflow here, fix maxlength
free(str);
}
ptr = strchr(cmd, '~');
if (ptr) {
ptr[0]='\0';
r_cons_grep(ptr+1);
} else r_cons_grep(NULL);
return 0;
}
int r_core_cmd_foreach(struct r_core_t *core, const char *cmd, char *each)
{
//void radare_cmd_foreach(const char *cmd, const char *each)
int i=0,j;
char ch;
char *word = NULL;
char *str;
struct list_head *pos;
u64 oseek, addr;
for(;*each==' ';each=each+1);
for(;*cmd==' ';cmd=cmd+1);
oseek = core->seek;
str = strdup(each);
//radare_controlc();
switch(each[0]) {
case '?':
eprintf("Foreach '@@' iterator command:\n");
eprintf(" This command is used to repeat a command over a list of offsets.\n");
eprintf(" x @@ sym. ; run 'x' over all flags matching 'sym.'\n");
eprintf(" x @@.file ; \"\" over the offsets specified in the file (one offset per line)\n");
eprintf(" x @@=off1 off2 .. ; manual list of offsets\n");
eprintf(" x @@=`pdf~call[0] ; run 'x' at every call offset of the current function\n");
break;
case '=':
/* foreach list of items */
each = each+1;
do {
while(each[0]==' ') each=each+1;
str = strchr(each, ' ');
if (str == NULL) {
addr = r_num_math(&core->num, each);
} else {
str[0]='\0';
addr = r_num_math(&core->num, each);
str[0]=' ';
}
each = str+1;
r_core_seek(core, addr);
eprintf("\n"); //eprintf("===(%s)at(0x%08llx)\n", cmd, addr);
r_core_cmd(core, cmd, 0);
} while(str != NULL);
break;
case '.':
if (each[1]=='(') {
char cmd2[1024];
// TODO: use controlc() here
for(core->macro.counter=0;i<999;core->macro.counter++) {
r_macro_call(&core->macro, each+2);
if (core->macro.brk_value == NULL) {
//eprintf("==>breaks(%s)\n", each);
break;
}
addr = core->macro._brk_value;
sprintf(cmd2, "%s @ 0x%08llx", cmd, addr);
eprintf("0x%08llx (%s)\n", addr, cmd2);
r_core_seek(core, addr);
r_core_cmd(core, cmd2, 0);
i++;
}
} else {
char buf[1024];
char cmd2[1024];
FILE *fd = fopen(each+1, "r");
if (fd == NULL) {
eprintf("Cannot open file '%s' for reading one offset per line.\n", each+1);
} else {
core->macro.counter=0;
while(!feof(fd)) {
buf[0]='\0';
fgets(buf, 1024, fd);
addr = r_num_math(&core->num, buf);
eprintf("0x%08llx\n", addr, cmd);
sprintf(cmd2, "%s @ 0x%08llx", cmd, addr);
r_core_seek(core, buf);
r_core_cmd(core, cmd2, 0);
core->macro.counter++;
}
fclose(fd);
}
}
break;
default:
core->macro.counter = 0;
//while(str[i]) && !core->interrupted) {
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 == NULL)
break;
str[i] = ch;
if (strchr(word, '*')) {
#if 0
/* for all flags in current flagspace */
list_for_each(pos, &flags) {
flag_t *flag = (flag_t *)list_entry(pos, flag_t, list);
//if (core->interrupted)
// break;
/* filter per flag spaces */
// if ((flag_space_idx != -1) && (flag->space != flag_space_idx))
// continue;
config.seek = flag->offset;
radare_read(0);
cons_printf("; @@ 0x%08llx (%s)\n", config.seek, flag->name);
radare_cmd(cmd,0);
}
#else
printf("No flags foreach implemented\n");
#endif
} else {
/* for all flags in current flagspace */
list_for_each(pos, &core->flags.flags) {
struct r_flag_item_t *flag =
(struct r_flag_item_t *)list_entry(pos, struct r_flag_item_t, list);
// if (config.interrupted)
// break;
#if 0
/* filter per flag spaces */
if ((flag_space_idx != -1) && (flag->space != flag_space_idx))
continue;
#endif
//eprintf("polla(%s)(%s)\n", flag->name, word);
if (word[0]=='\0' || strstr(flag->name, word) != NULL) {
r_core_seek(core, flag->offset);
r_cons_printf("; @@ 0x%08llx (%s)\n", core->seek, flag->name);
r_core_cmd(core, cmd, 0);
}
}
#if 0
/* ugly copypasta from tmpseek .. */
if (strstr(word, each)) {
if (word[i]=='+'||word[i]=='-')
config.seek = config.seek + get_math(word);
else config.seek = get_math(word);
radare_read(0);
cons_printf("; @@ 0x%08llx\n", config.seek);
radare_cmd(cmd,0);
}
#endif
//radare_controlc();
core->macro.counter++ ;
free(word);
word = NULL;
}
}
}
// radare_controlc_end();
core->seek = oseek;
free(word);
word = NULL;
free(str);
}
int r_core_cmd(struct r_core_t *core, const char *command, int log)
{
int i, len;
char *cmd;
int ret = -1;
int times = 1;
int newfd = 1;
int quoted = 0;
u64 tmpseek = core->seek;
int restoreseek = 0;
if (command == NULL )
return 0;
while(command[0]==' ') // TODO: handle tabs to with iswhitespace()
command = command+1;
len = strlen(command)+1;
cmd = alloca(len)+4096;
memcpy(cmd, command, len);
/* quoted / raw command */
len = strlen(cmd);
if (cmd[0]=='"') {
if (cmd[len-1]!='"') {
fprintf(stderr, "parse: Missing ending '\"': '%s' (%c) len=%d\n", cmd, cmd[2], len);
return 0;
}
cmd[len-1]='\0';
strcpy(cmd, cmd+1);
return r_cmd_call(&core->cmd, cmd);
}
ret = r_core_cmd_subst(core, cmd, &restoreseek, &newfd, &times);
if (ret == -1) {
fprintf(stderr, "Invalid conversion: '%s'\n", command);
ret = -1;
} else {
for(i=0;i<times;i++) {
if (quoted) {
ret = r_cmd_call(&core->cmd, cmd);
if (ret == -1) // stop on error?
break;
} else {
char *ptr;
ptr = strchr(cmd, '&');
while (ptr&&ptr[1]=='&') {
ptr[0]='\0';
ret = r_cmd_call(&core->cmd, cmd);
if (ret == -1){
fprintf(stderr, "command error(%s)\n", cmd);
break;
}
for(cmd=ptr+2;cmd&&cmd[0]==' ';cmd=cmd+1);
ptr = strchr(cmd, '&');
}
r_cmd_call(&core->cmd, cmd);
}
}
if (ret == -1){
if (cmd[0])
fprintf(stderr, "Invalid command: '%s'\n", command);
ret = 1;
} else
if (log) r_line_hist_add(command);
if (restoreseek)
r_core_seek(core, tmpseek);
}
if (newfd != 1) {
r_cons_flush();
r_cons_pipe_close(newfd);
}
free (core->oobi);
core->oobi = NULL;
core->oobi_len = 0;
return ret;
}
int r_core_cmd_file(struct r_core_t *core, const char *file)
{
char buf[1024];
FILE *fd = fopen(file, "r");
if (fd == NULL)
return -1;
while (!feof(fd)) {
if (fgets(buf, 1023, fd) != NULL) {
buf[strlen(buf)-1]='\0';
if (r_core_cmd(core, buf, 0) == -1) {
fprintf(stderr, "Error running command '%s'\n", buf);
break;
}
}
}
fclose(fd);
return 0;
}
static int cmd_debug(void *data, const char *input)
{
struct r_core_t *core = (struct r_core_t *)data;
switch(input[0]) {
case 's':
fprintf(stderr, "step\n");
r_debug_step(&core->dbg, 1);
break;
case 'b':
fprintf(stderr, "breakpoint\n");
break;
case 'c':
fprintf(stderr, "continue\n");
r_debug_continue(&core->dbg);
break;
case 'r':
r_core_cmd(core, "|reg", 0); // XXX
break;
case 'p':
if (input[1]==' ')
r_debug_select(&core->dbg, core->dbg.pid, atoi(input+2));
else fprintf(stderr, "TODO: List processes..\n");
break;
case 'h':
if (input[1]==' ')
r_debug_handle_set(&core->dbg, input+2);
else r_debug_handle_list(&core->dbg, "");
break;
default:
r_cons_printf("Usage: d[sbhcrbo] [arg]\n"
" dh [handler] ; list or set debugger handler\n"
" ds ; perform one step\n"
" ds 3 ; perform 3 steps\n"
" do 3 ; perform 3 steps overs\n"
" dc ; continue execution\n"
" dr ; show registers\n"
" dr* ; show registers in radare commands\n"
" dr eax ; show value of eax register\n"
" dr eax = 33 ; set register value. eax = 33\n"
" db ; list breakpoints\n"
" db sym.main ; set breakpoint\n"
" db -sym.main ; drop breakpoint\n"
" dm ; show memory maps\n"
" dm 4096 ; allocate 4KB in child process\n"
" dm rw- esp 9K; set 9KB of the stack as read+write (no exec)\n");
break;
}
return 0;
}
int r_core_cmd_buffer(void *user, const char *buf)
{
char *str = strdup(buf);
char *ptr = strchr(str, '\n');
char *optr = str;
while(ptr) {
ptr[0]='\0';
r_core_cmd(user, optr, 0);
optr = ptr+1;
ptr = strchr(str,'\n');
}
r_core_cmd(user, optr, 0);
free(str);
return R_TRUE;
}
int r_core_cmdf(void *user, const char *fmt, ...)
{
char string[1024];
int ret;
va_list ap;
va_start(ap, fmt);
vsnprintf(string, 1023, fmt, ap);
ret = r_core_cmd((struct r_core_t *)user, string, 0);
va_end(ap);
return ret;
}
int r_core_cmd0(void *user, const char *cmd)
{
return r_core_cmd((struct r_core_t *)user, cmd, 0);
}
char *r_core_cmd_str(struct r_core_t *core, const char *cmd)
{
char *retstr;
r_cons_reset();
if (r_core_cmd(core, cmd, 0) == -1) {
fprintf(stderr, "Invalid command: %s\n", cmd);
retstr = strdup("");
} else {
const char *static_str = r_cons_get_buffer();
if (retstr==NULL)
retstr = strdup("");
else retstr = strdup(static_str);
r_cons_reset();
}
return retstr;
}
int r_core_cmd_init(struct r_core_t *core)
{
r_cmd_init(&core->cmd);
r_cmd_set_data(&core->cmd, core);
r_cmd_add(&core->cmd, "x", "alias for px", &cmd_hexdump);
r_cmd_add(&core->cmd, "analysis", "analysis", &cmd_anal);
r_cmd_add(&core->cmd, "flag", "get/set flags", &cmd_flag);
r_cmd_add(&core->cmd, "debug", "debugger operations", &cmd_debug);
r_cmd_add(&core->cmd, "info", "get file info", &cmd_info);
r_cmd_add(&core->cmd, "seek", "seek to an offset", &cmd_seek);
r_cmd_add(&core->cmd, "bsize", "change block size", &cmd_bsize);
r_cmd_add(&core->cmd, "eval", "evaluate configuration variable", &cmd_eval);
r_cmd_add(&core->cmd, "print", "print current block", &cmd_print);
r_cmd_add(&core->cmd, "write", "write bytes", &cmd_write);
r_cmd_add(&core->cmd, "Code", "code metadata", &cmd_meta);
r_cmd_add(&core->cmd, "yank", "yank bytes", &cmd_yank);
r_cmd_add(&core->cmd, "Visual", "enter visual mode", &cmd_visual);
r_cmd_add(&core->cmd, "!", "run system command", &cmd_system);
r_cmd_add(&core->cmd, "|", "run io system command", &cmd_io_system);
r_cmd_add(&core->cmd, "#", "calculate hash", &cmd_hash);
r_cmd_add(&core->cmd, "?", "help message", &cmd_help);
r_cmd_add(&core->cmd, ".", "interpret", &cmd_interpret);
r_cmd_add(&core->cmd, "/", "search kw, pattern aes", &cmd_search);
r_cmd_add(&core->cmd, "(", "macro", &cmd_macro);
r_cmd_add(&core->cmd, "|", "io pipe", &cmd_iopipe);
r_cmd_add(&core->cmd, "quit", "exit program session", &cmd_quit);
return 0;
}