radare2/libr/parse/p/parse_x86_pseudo.c
2015-07-12 19:05:33 +02:00

318 lines
7.4 KiB
C

/* radare - LGPL - Copyright 2009-2015 - nibble, pancake */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <r_lib.h>
#include <r_util.h>
#include <r_flags.h>
#include <r_anal.h>
#include <r_parse.h>
// 16 bit examples
// 0x0001f3a4 9a67620eca call word 0xca0e:0x6267
// 0x0001f41c eabe76de12 jmp word 0x12de:0x76be [2]
// 0x0001f56a ea7ed73cd3 jmp word 0xd33c:0xd77e [6]
static int replace(int argc, const char *argv[], char *newstr) {
int i,j,k;
struct {
char *op;
char *str;
} ops[] = {
{ "adc", "1 += 2"},
{ "add", "1 += 2"},
{ "and", "1 &= 2"},
{ "call", "1 ()"},
{ "cmove", "1 = 2"},
{ "cmovl","ifnot zf,1 = 2"},
{ "cmp", "if (1 == 2"},
{ "cmpsq", "if (1 == 2"},
{ "cmsb", "if (1 == 2"},
{ "cmsw", "if (1 == 2"},
{ "dec", "1--"},
{ "div", "1 /= 2"},
{ "fabs", "abs(1)"},
{ "fadd", "1 = 1 + 2"},
{ "fcomp", "if (1 == 2"},
{ "fcos", "1 = cos(1)"},
{ "fdiv", "1 = 1 / 2"},
{ "fiadd", "1 = 1 / 2"},
{ "ficom", "if (1 == 2"},
{ "fidiv", "1 = 1 / 2"},
{ "fidiv", "1 = 1 * 2"},
{ "fisub", "1 = 1 - 2"},
{ "fnul", "1 = 1 * 2"},
{ "fnop", " "},
{ "frndint", "1 = (int) 1"},
{ "fsin", "1 = sin(1)"},
{ "fsqrt", "1 = sqrt(1)"},
{ "fsub", "1 = 1 - 2"},
{ "fxch", "1,2 = 2,1"},
{ "idiv", "1 /= 2"},
{ "imul", "1 *= 2"},
{ "in", "1 = io[2]"},
{ "inc", "1++"},
{ "ja", "isAbove 1)"},
{ "jbe", "isBelowOrEqual 1)"},
{ "je", "isZero 1)"},
{ "jg", "isGreater 1)"},
{ "jge", "isGreaterOrEqual 1)"},
{ "jle", "isLessOrEqual 1)"},
{ "jmp", "goto 1"},
{ "jne", "notZero 1)"},
{ "lea", "1 = 2"},
{ "mov", "1 = 2"},
{ "movsd", "1 = 2"},
{ "movsx","1 = 2"},
{ "movsxd","1 = 2"},
{ "movzx", "1 = 2"},
{ "movntdq", "1 = 2"},
{ "movnti", "1 = 2"},
{ "movntpd", "1 = 2"},
{ "mul", "1 *= 2"},
{ "neg", "1 ~= 1"},
{ "nop", ""},
{ "not", "1 = !1"},
{ "or", "1 |= 2"},
{ "out", "io[1] = 2"},
{ "pop", "pop 1"},
{ "push", "push 1"},
{ "sal", "1 <<= 2"},
{ "sar", "1 >>= 2"},
{ "sete", "1 = e"},
{ "setne", "1 = ne"},
{ "shl", "1 <<<= 2"},
{ "shld", "1 <<<= 2"},
{ "sbb", "1 = 1 - 2"},
{ "shr", "1 >>>= 2"},
{ "shlr", "1 >>>= 2"},
//{ "strd", "1 = 2 - 3"},
{ "sub", "1 -= 2"},
{ "swap", "swap 1, 2"},
{ "test", "if (1 == 2"},
{ "xchg", "1,2 = 2,1"},
{ "xadd", "1,2 = 2,1+2"},
{ "xor", "1 ^= 2"},
{ NULL }
};
if (argc>2 && !strcmp (argv[0], "xor")) {
if (!strcmp (argv[1], argv[2])) {
argv[0] = "mov";
argv[2] = "0";
}
}
for (i=0; ops[i].op != NULL; i++) {
if (!strcmp (ops[i].op, argv[0])) {
if (newstr != NULL) {
for (j=k=0; ops[i].str[j]!='\0'; j++, k++) {
if (ops[i].str[j]>='0' && ops[i].str[j]<='9') {
const char *w = argv[ ops[i].str[j]-'0' ];
if (w != NULL) {
strcpy (newstr+k, w);
k += strlen(w)-1;
}
} else newstr[k] = ops[i].str[j];
}
newstr[k]='\0';
}
return R_TRUE;
}
}
/* TODO: this is slow */
if (newstr != NULL) {
newstr[0] = '\0';
for (i=0; i<argc; i++) {
strcat (newstr, argv[i]);
strcat (newstr, (i == 0 || i== argc - 1)?" ":",");
}
}
return R_FALSE;
}
static int parse(RParse *p, const char *data, char *str) {
char w0[256], w1[256], w2[256], w3[256];
int i, len = strlen (data);
char *buf, *ptr, *optr;
if (len>=sizeof (w0))
return R_FALSE;
// malloc can be slow here :?
if ((buf = malloc (len+1)) == NULL)
return R_FALSE;
memcpy (buf, data, len+1);
if (*buf) {
*w0 = *w1 = *w2 = *w3 = '\0';
ptr = strchr (buf, ' ');
if (ptr == NULL)
ptr = strchr (buf, '\t');
if (ptr) {
*ptr = '\0';
for (++ptr; *ptr==' '; ptr++);
strncpy (w0, buf, sizeof (w0) - 1);
strncpy (w1, ptr, sizeof (w1) - 1);
optr = ptr;
ptr = strchr (ptr, ',');
if (ptr) {
*ptr = '\0';
for (++ptr; *ptr==' '; ptr++);
strncpy (w1, optr, sizeof (w1) - 1);
strncpy (w2, ptr, sizeof (w2) - 1);
optr = ptr;
ptr = strchr (ptr, ',');
if (ptr) {
*ptr = '\0';
for (++ptr; *ptr==' '; ptr++);
strncpy (w2, optr, sizeof (w2) - 1);
strncpy (w3, ptr, sizeof (w3) - 1);
}
}
}
{
const char *wa[] = { w0, w1, w2, w3 };
int nw = 0;
for (i=0; i<4; i++) {
if (wa[i][0] != '\0')
nw++;
}
replace (nw, wa, str);
}
}
free (buf);
return R_TRUE;
}
#if 0
static inline int ishexch (char c) {
if (c>=0 && c<=9) return 1;
if (c>='a' && c<='f') return 1;
if (c>='A' && c<='F') return 1;
return 0;
}
static inline int issegoff (const char *w) {
if (!ishexch (w[0])) return 0;
if (!ishexch (w[1])) return 0;
if (!ishexch (w[2])) return 0;
if (!ishexch (w[3])) return 0;
// :
if (!ishexch (w[5])) return 0;
if (!ishexch (w[6])) return 0;
if (!ishexch (w[7])) return 0;
if (!ishexch (w[8])) return 0;
return 1;
}
#endif
static int varsub(RParse *p, RAnalFunction *f, char *data, char *str, int len) {
#if USE_VARSUBS
int i;
char *ptr, *ptr2;
for (i = 0; i < R_ANAL_VARSUBS; i++)
if (f->varsubs[i].pat[0] != '\0' && \
f->varsubs[i].sub[0] != '\0' && \
(ptr = strstr (data, f->varsubs[i].pat))) {
*ptr = '\0';
ptr2 = ptr + strlen (f->varsubs[i].pat);
snprintf (str, len, "%s%s%s", data,
f->varsubs[i].sub, ptr2);
}
return R_TRUE;
#else
RAnalVar *var, *arg;
RListIter *variter, *argiter;
char oldstr[64], newstr[64];
char *tstr = strdup (data);
RList *vars, *args;
if (!p->varlist) {
free(tstr);
return R_FALSE;
}
vars = p->varlist (p->anal, f, 'v');
args = p->varlist (p->anal, f, 'a');
r_list_foreach (args, argiter, arg) {
if (arg->delta < 10) snprintf (oldstr, sizeof (oldstr)-1,
"[%s + %d]",
p->anal->reg->name[R_REG_NAME_BP],
arg->delta);
else snprintf (oldstr, sizeof (oldstr)-1,
"[%s + 0x%x]",
p->anal->reg->name[R_REG_NAME_BP],
arg->delta);
snprintf (newstr, sizeof (newstr)-1, "[%s+%s]",
p->anal->reg->name[R_REG_NAME_BP],
arg->name);
if (strstr (tstr, oldstr) != NULL) {
tstr = r_str_replace (tstr, oldstr, newstr, 1);
break;
}
// Try with no spaces
snprintf (oldstr, sizeof (oldstr)-1, "[%s+0x%x]",
p->anal->reg->name[R_REG_NAME_BP],
arg->delta);
if (strstr (tstr, oldstr) != NULL) {
tstr = r_str_replace (tstr, oldstr, newstr, 1);
break;
}
}
r_list_foreach (vars, variter, var) {
if (var->delta < 10) snprintf (oldstr, sizeof (oldstr)-1,
"[%s - %d]",
p->anal->reg->name[R_REG_NAME_BP],
var->delta);
else snprintf (oldstr, sizeof (oldstr)-1,
"[%s - 0x%x]",
p->anal->reg->name[R_REG_NAME_BP],
var->delta);
snprintf (newstr, sizeof (newstr)-1, "[%s-%s]",
p->anal->reg->name[R_REG_NAME_BP],
var->name);
if (strstr (tstr, oldstr) != NULL) {
tstr = r_str_replace (tstr, oldstr, newstr, 1);
break;
}
// Try with no spaces
snprintf (oldstr, sizeof (oldstr)-1, "[%s-0x%x]",
p->anal->reg->name[R_REG_NAME_BP],
var->delta);
if (strstr (tstr, oldstr) != NULL) {
tstr = r_str_replace (tstr, oldstr, newstr, 1);
break;
}
}
if (len > strlen (tstr)) {
strncpy (str, tstr, strlen(tstr));
str[strlen (tstr)] = 0;
} else {
// TOO BIG STRING CANNOT REPLACE HERE
free (tstr);
return R_FALSE;
}
free (tstr);
return R_TRUE;
#endif
}
struct r_parse_plugin_t r_parse_plugin_x86_pseudo = {
.name = "x86.pseudo",
.desc = "X86 pseudo syntax",
.init = NULL,
.fini = NULL,
.parse = &parse,
.filter = NULL,
.varsub = &varsub,
};
#ifndef CORELIB
struct r_lib_struct_t radare_plugin = {
.type = R_LIB_TYPE_PARSE,
.data = &r_parse_plugin_x86_pseudo,
.version = R2_VERSION
};
#endif