emit_arm.c File Reference

#include <rz_egg.h>

Go to the source code of this file.

Macros
#define	attsyntax   0
#define	EMIT_NAME   emit_arm
#define	RZ_ARCH   "arm"
#define	RZ_SZ   8
#define	RZ_SP   "sp"
#define	RZ_BP   "fp"
#define	RZ_AX   "r7"
#define	RZ_GP    { "r0", "r1", "r2", "r3", "r4" }
#define	RZ_TMP   "r9"
#define	RZ_NGP   5

Functions
static void	emit_init (RzEgg *egg)
static char *	emit_syscall (RzEgg *egg, int num)
static void	emit_frame (RzEgg *egg, int sz)
static void	emit_frame_end (RzEgg *egg, int sz, int ctx)
static void	emit_comment (RzEgg *egg, const char *fmt,...)
static void	emit_equ (RzEgg *egg, const char *key, const char *value)
static void	emit_syscall_args (RzEgg *egg, int nargs)
static void	emit_set_string (RzEgg *egg, const char *dstvar, const char *str, int j)
static void	emit_jmp (RzEgg *egg, const char *str, int atr)
static void	emit_call (RzEgg *egg, const char *str, int atr)
static void	emit_arg (RzEgg *egg, int xs, int num, const char *str)
static void	emit_get_result (RzEgg *egg, const char *ocn)
static void	emit_restore_stack (RzEgg *egg, int size)
static void	emit_get_while_end (RzEgg *egg, char *str, const char *ctxpush, const char *label)
static void	emit_while_end (RzEgg *egg, const char *labelback)
static void	emit_get_var (RzEgg *egg, int type, char *out, int idx)
static void	emit_trap (RzEgg *egg)
static void	emit_load_ptr (RzEgg *egg, const char *dst)
static void	emit_branch (RzEgg *egg, char *b, char *g, char *e, char *n, int sz, const char *dst)
static void	emit_load (RzEgg *egg, const char *dst, int sz)
static void	emit_mathop (RzEgg *egg, int ch, int vs, int type, const char *eq, const char *p)
static const char *	emit_regs (RzEgg *egg, int idx)

Variables
static char *	regs [] = RZ_GP
static int	lastarg = 0
static char	lastargs [16][32]
RzEggEmit	EMIT_NAME

Macro Definition Documentation

attsyntax

#define attsyntax   0

Definition at line 5 of file emit_arm.c.

EMIT_NAME

#define EMIT_NAME   emit_arm

Definition at line 7 of file emit_arm.c.

RZ_ARCH

#define RZ_ARCH   "arm"

Definition at line 8 of file emit_arm.c.

RZ_AX

#define RZ_AX   "r7"

Definition at line 12 of file emit_arm.c.

RZ_BP

#define RZ_BP   "fp"

Definition at line 11 of file emit_arm.c.

RZ_GP

#define RZ_GP    { "r0", "r1", "r2", "r3", "r4" }

Definition at line 13 of file emit_arm.c.

RZ_NGP

#define RZ_NGP   5

Definition at line 16 of file emit_arm.c.

RZ_SP

#define RZ_SP   "sp"

Definition at line 10 of file emit_arm.c.

RZ_SZ

#define RZ_SZ   8

Definition at line 9 of file emit_arm.c.

RZ_TMP

#define RZ_TMP   "r9"

Definition at line 15 of file emit_arm.c.

Function Documentation

emit_arg()

static void emit_arg ( RzEgg *  egg, int  xs, int  num, const char *  str  )

static

Definition at line 139 of file emit_arm.c.

                                                                   {
    int d = atoi(str);
    if (!attsyntax && (*str == '$')) {
        str++;
    }
    lastarg = num;
    switch (xs) {
    case 0:
        if (strchr(str, ',')) {
            // rz_egg_printf (egg, ".  str r0, [%s]\n", str);
            strncpy(lastargs[num - 1], str, sizeof(lastargs[0]) - 1);
        } else {
            if (!atoi(str)) {
                eprintf("WARNING: probably a bug?\n");
            }
            rz_egg_printf(egg, "  mov r0, %s\n", str);
            snprintf(lastargs[num - 1], sizeof(lastargs[0]), "sp, %d", 8 + (num * 4));
            rz_egg_printf(egg, "  str r0, [%s]\n", lastargs[num - 1]);
        }
        break;
    case '*':
        rz_egg_printf(egg, "  push {%s}\n", str);
        break;
    case '&':
        if (d) {
            rz_egg_printf(egg, "  add " RZ_BP ", %d\n", d);
        }
        rz_egg_printf(egg, "  push {" RZ_BP "}\n");
        if (d) {
            rz_egg_printf(egg, "  sub " RZ_BP ", %d\n", d);
        }
        break;
    }
}

References attsyntax, d, eprintf, lastarg, lastargs, num, RZ_BP, rz_egg_printf(), snprintf, and cmd_descs_generate::str.

emit_branch()

static void emit_branch ( RzEgg *  egg, char *  b, char *  g, char *  e, char *  n, int  sz, const char *  dst  )

static

Definition at line 213 of file emit_arm.c.

                                                                                                 {
    char *p, str[64];
    char *arg = NULL;
    char *op = "beq";
    /* NOTE that jb/ja are inverted to fit cmp opcode */
    if (b) {
        *b = '\0';
        op = e ? "bge" : "bgt";
        arg = b + 1;
    } else if (g) {
        *g = '\0';
        op = e ? "ble" : "blt";
        arg = g + 1;
    }
    if (!arg) {
        if (e) {
            arg = e + 1;
            op = "bne";
        } else {
            arg = "0";
            op = n ? "bne" : "beq";
        }
    }
 
    if (*arg == '=') {
        arg++; /* for <=, >=, ... */
    }
    p = rz_egg_mkvar(egg, str, arg, 0);
    rz_egg_printf(egg, "  pop " RZ_AX "\n"); /* TODO: add support for more than one arg get arg0 */
    rz_egg_printf(egg, "  cmp %s, " RZ_AX "\n", p);
    // if (context>0)
    rz_egg_printf(egg, "  %s %s\n", op, dst);
    free(p);
}

References b, dst, e, free(), g, n, NULL, p, RZ_AX, rz_egg_mkvar(), rz_egg_printf(), and cmd_descs_generate::str.

emit_call()

static void emit_call ( RzEgg *  egg, const char *  str, int  atr  )

static

Definition at line 123 of file emit_arm.c.

                                                            {
    int i;
    // rz_egg_printf (egg, " ARGS=%d CALL(%s,%d)\n", lastarg, str, atr);
    for (i = 0; i < lastarg; i++) {
        rz_egg_printf(egg, "  ldr r%d, [%s]\n", lastarg - 1 - i, lastargs[i]);
        lastargs[i][0] = 0;
    }
 
    if (atr) {
        rz_egg_printf(egg, "  ldr r0, %s", str);
        rz_egg_printf(egg, "  blx r0\n");
    } else {
        rz_egg_printf(egg, "  bl %s\n", str);
    }
}

References i, lastarg, lastargs, rz_egg_printf(), and cmd_descs_generate::str.

emit_comment()

static void emit_comment ( RzEgg *  egg, const char *  fmt,   ...  )

static

Definition at line 66 of file emit_arm.c.

                                                           {
    va_list ap;
    char buf[1024];
    va_start(ap, fmt);
    vsnprintf(buf, sizeof(buf), fmt, ap);
    rz_egg_printf(egg, "# %s\n", buf);
    va_end(ap);
}

References rz_egg_printf(), and vsnprintf.

emit_equ()

static void emit_equ ( RzEgg *  egg, const char *  key, const char *  value  )

static

Definition at line 75 of file emit_arm.c.

                                                                     {
    rz_egg_printf(egg, ".equ %s, %s\n", key, value);
}

References key, rz_egg_printf(), and value.

emit_frame()

static void emit_frame ( RzEgg *  egg, int  sz  )

static

Definition at line 46 of file emit_arm.c.

                                           {
    rz_egg_printf(egg, "  push {fp,lr}\n");
    if (sz > 0) {
        rz_egg_printf(egg,
            // "  mov "RZ_BP", "RZ_SP"\n"
            "  add fp, sp, $4\n" // size of arguments
            "  sub sp, %d\n",
            sz); // size of stackframe 8, 16, ..
    }
}

References rz_egg_printf().

emit_frame_end()

static void emit_frame_end ( RzEgg *  egg, int  sz, int  ctx  )

static

Definition at line 57 of file emit_arm.c.

                                                        {
    if (sz > 0) {
        rz_egg_printf(egg, "  add sp, fp, %d\n", sz);
    }
    if (ctx > 0) {
        rz_egg_printf(egg, "  pop {fp,pc}\n");
    }
}

References rz_egg_printf().

emit_get_result()

static void emit_get_result ( RzEgg *  egg, const char *  ocn  )

static

Definition at line 174 of file emit_arm.c.

                                                         {
    rz_egg_printf(egg, "  mov %s, r0\n", ocn);
}

References rz_egg_printf().

emit_get_var()

static void emit_get_var ( RzEgg *  egg, int  type, char *  out, int  idx  )

static

Definition at line 195 of file emit_arm.c.

                                                                   {
    switch (type) {
    case 0: sprintf(out, "sp, %d", idx - 1); break; /* variable */
    case 1:
        sprintf(out, "r%d", idx);
        break; /* registers */
        // sp,$%d", idx); break; /* argument */ // XXX: MUST BE r0, r1, r2, ..
    }
}

References setup::idx, out, sprintf, and type.

emit_get_while_end()

static void emit_get_while_end ( RzEgg *  egg, char *  str, const char *  ctxpush, const char *  label  )

static

Definition at line 183 of file emit_arm.c.

                                                                                              {
    sprintf(str, "  push {%s}\n  b %s\n", ctxpush, label);
}

References sprintf, and cmd_descs_generate::str.

emit_init()

static void emit_init ( RzEgg *  egg )

static

Definition at line 23 of file emit_arm.c.

                                  {
    /* TODO */
}

emit_jmp()

static void emit_jmp ( RzEgg *  egg, const char *  str, int  atr  )

static

Definition at line 114 of file emit_arm.c.

                                                           {
    if (atr) {
        rz_egg_printf(egg, "  ldr r0, %s", str);
        rz_egg_printf(egg, "  bx r0\n");
    } else {
        rz_egg_printf(egg, "  b %s\n", str);
    }
}

References rz_egg_printf(), and cmd_descs_generate::str.

emit_load()

static void emit_load ( RzEgg *  egg, const char *  dst, int  sz  )

static

Definition at line 248 of file emit_arm.c.

                                                           {
    switch (sz) {
    case 'l':
        rz_egg_printf(egg, "  mov " RZ_AX ", %s\n", dst);
        rz_egg_printf(egg, "  mov " RZ_AX ", [" RZ_AX "]\n");
        break;
    case 'b':
        rz_egg_printf(egg, "  mov " RZ_AX ", %s\n", dst);
        rz_egg_printf(egg, "  movz " RZ_AX ", [" RZ_AX "]\n");
        break;
    default:
        // TODO: unhandled?!?
        rz_egg_printf(egg, "  mov " RZ_AX ", %s\n", dst);
        rz_egg_printf(egg, "  mov " RZ_AX ", [" RZ_AX "]\n");
    }
}

References dst, RZ_AX, and rz_egg_printf().

emit_load_ptr()

static void emit_load_ptr ( RzEgg *  egg, const char *  dst  )

static

Definition at line 209 of file emit_arm.c.

                                                       {
    rz_egg_printf(egg, "  ldr r0, [fp, %d]\n", atoi(dst));
}

References dst, and rz_egg_printf().

emit_mathop()

static void emit_mathop ( RzEgg *  egg, int  ch, int  vs, int  type, const char *  eq, const char *  p  )

static

Definition at line 265 of file emit_arm.c.

                                                                                             {
    char *op;
    switch (ch) {
    case '^': op = "eor"; break;
    case '&': op = "and"; break;
    case '|': op = "orr"; break;
    case '-': op = "sub"; break;
    case '+': op = "add"; break;
    case '*': op = "mul"; break;
    case '/': op = "div"; break;
    default: op = "mov"; break;
    }
    if (!eq) {
        eq = RZ_AX;
    }
    if (!p) {
        p = RZ_AX;
    }
#if 0
    // TODO:
    eprintf ("TYPE = %c\n", type);
    eprintf ("  %s%c %c%s, %s\n", op, vs, type, eq, p);
    eprintf ("  %s %s, [%s]\n", op, p, eq);
#endif
    if (type == '*') {
        rz_egg_printf(egg, "  %s %s, [%s]\n", op, p, eq);
    } else {
        rz_egg_printf(egg, "  %s %s, %s\n", op, p, eq);
    }
}

References eprintf, eq, op, p, RZ_AX, rz_egg_printf(), and type.

emit_regs()

static const char* emit_regs ( RzEgg *  egg, int  idx  )

static

Definition at line 296 of file emit_arm.c.

                                                  {
    return regs[idx % RZ_NGP];
}

References setup::idx, regs, and RZ_NGP.

emit_restore_stack()

static void emit_restore_stack ( RzEgg *  egg, int  size  )

static

Definition at line 178 of file emit_arm.c.

                                                     {
    // XXX: must die.. or add emit_store_stack. not needed by ARM
    // rz_egg_printf (egg, "  add sp, %d\n", size);
}

emit_set_string()

static void emit_set_string ( RzEgg *  egg, const char *  dstvar, const char *  str, int  j  )

static

Definition at line 88 of file emit_arm.c.

                                                                                    {
    int rest, off = 0;
    off = strlen(str) + 1;
    rest = (off % 4);
    if (rest) {
        rest = 4 - rest;
    }
    off += rest - 4;
    rz_egg_printf(egg, "  add pc, %d\n", (off));
    // XXX: does not handle \n and so on.. must use rz_util
    // use rz_str_escape to handle \n
    // do not forget mem leak
    rz_egg_printf(egg, ".string \"%s\"\n", str = rz_str_escape(str));
    free((char *)str);
    if (rest) {
        rz_egg_printf(egg, ".fill %d, 1, 0\n", (rest));
    }
    rz_egg_printf(egg, "  sub r0, pc, %d\n", off + 12);
    {
        char str[32], *p = rz_egg_mkvar(egg, str, dstvar, 0);
        // rz_egg_printf (egg, "DSTVAR=%s --> %s\n", dstvar, p);
        rz_egg_printf(egg, "  str r0, [%s]\n", p);
        free(p);
    }
}

References free(), off, p, rz_egg_mkvar(), rz_egg_printf(), rz_str_escape(), and cmd_descs_generate::str.

emit_syscall()

static char* emit_syscall ( RzEgg *  egg, int  num  )

static

Definition at line 27 of file emit_arm.c.

                                               {
    int svc = 0;
    switch (egg->os) {
    case RZ_EGG_OS_DARWIN:
    case RZ_EGG_OS_OSX:
    case RZ_EGG_OS_IOS:
    case RZ_EGG_OS_MACOS:
        svc = 0x80;
        break;
    case RZ_EGG_OS_WATCHOS:
        svc = 0x8000;
        break;
    case RZ_EGG_OS_LINUX:
        svc = 0;
        break;
    }
    return rz_str_newf(": mov " RZ_AX ", `.arg`\n: svc 0x%x\n", svc);
}

References rz_egg_t::os, RZ_AX, RZ_EGG_OS_DARWIN, RZ_EGG_OS_IOS, RZ_EGG_OS_LINUX, RZ_EGG_OS_MACOS, RZ_EGG_OS_OSX, RZ_EGG_OS_WATCHOS, rz_str_newf(), and svc().

emit_syscall_args()

static void emit_syscall_args ( RzEgg *  egg, int  nargs  )

static

Definition at line 79 of file emit_arm.c.

                                                     {
    int j, k;
    for (j = 0; j < nargs; j++) {
        k = j * RZ_SZ;
        rz_egg_printf(egg, "  ldr %s, [sp, %d]\n",
            regs[j + 1], k ? k + 4 : k + 8);
    }
}

References k, cmd_descs_generate::nargs, regs, rz_egg_printf(), and RZ_SZ.

emit_trap()

static void emit_trap ( RzEgg *  egg )

static

Definition at line 205 of file emit_arm.c.

                                  {
    rz_egg_printf(egg, "  udf 16\n");
}

References rz_egg_printf().

emit_while_end()

static void emit_while_end ( RzEgg *  egg, const char *  labelback  )

static

Definition at line 187 of file emit_arm.c.

                                                              {
    rz_egg_printf(egg,
        "  pop " RZ_AX "\n"
        "  cmp " RZ_AX ", " RZ_AX "\n" // XXX MUST SUPPORT != 0 COMPARE HERE
        "  beq %s\n",
        labelback);
}

References RZ_AX, and rz_egg_printf().

Variable Documentation

EMIT_NAME

RzEggEmit EMIT_NAME

Initial value:

= {
    .arch = RZ_ARCH,
    .size = RZ_SZ,
    .jmp = emit_jmp,
    .call = emit_call,
    .init = emit_init,
    .equ = emit_equ,
    .regs = emit_regs,
    
    .trap = emit_trap,
    .frame = emit_frame,
    .frame_end = emit_frame_end,
    .comment = emit_comment,
    .push_arg = emit_arg,
    .restore_stack = emit_restore_stack,
    .get_result = emit_get_result,
    .syscall_args = emit_syscall_args,
    .set_string = emit_set_string,
    .get_var = emit_get_var,
    .while_end = emit_while_end,
    .get_while_end = emit_get_while_end,
    .branch = emit_branch,
    .load = emit_load,
    .load_ptr = emit_load_ptr,
    .mathop = emit_mathop,
    .syscall = emit_syscall,
}

Definition at line 300 of file emit_arm.c.

lastarg

int lastarg = 0

static

Definition at line 20 of file emit_arm.c.

Referenced by emit_arg(), and emit_call().

lastargs

char lastargs[16][32]

static

Definition at line 21 of file emit_arm.c.

Referenced by emit_arg(), and emit_call().

regs

char* regs[] = RZ_GP

static

Definition at line 19 of file emit_arm.c.

Referenced by emit_regs(), and emit_syscall_args().