110 lines
2.7 KiB
C
110 lines
2.7 KiB
C
// SPDX-FileCopyrightText: 2013-2018 pancake <pancake@nopcode.org>
|
|
// SPDX-License-Identifier: LGPL-3.0-only
|
|
|
|
#include <rz_asm.h>
|
|
#include <rz_lib.h>
|
|
#include "cs_helper.h"
|
|
|
|
CAPSTONE_DEFINE_PLUGIN_FUNCTIONS(mips);
|
|
|
|
RZ_IPI int mips_assemble(const char *str, ut64 pc, ut8 *out);
|
|
|
|
static int mips_disassemble(RzAsm *a, RzAsmOp *op, const ut8 *buf, int len) {
|
|
CapstoneContext *ctx = (CapstoneContext *)a->plugin_data;
|
|
|
|
cs_insn *insn;
|
|
int mode, n, ret = -1;
|
|
mode = (a->big_endian) ? CS_MODE_BIG_ENDIAN : CS_MODE_LITTLE_ENDIAN;
|
|
if (!op) {
|
|
return 0;
|
|
}
|
|
if (a->cpu && *a->cpu) {
|
|
if (!strcmp(a->cpu, "micro")) {
|
|
mode |= CS_MODE_MICRO;
|
|
} else if (!strcmp(a->cpu, "r6")) {
|
|
mode |= CS_MODE_MIPS32R6;
|
|
} else if (!strcmp(a->cpu, "v3")) {
|
|
mode |= CS_MODE_MIPS3;
|
|
} else if (!strcmp(a->cpu, "v2")) {
|
|
mode |= CS_MODE_MIPS2;
|
|
}
|
|
}
|
|
mode |= (a->bits == 64) ? CS_MODE_MIPS64 : CS_MODE_MIPS32;
|
|
memset(op, 0, sizeof(RzAsmOp));
|
|
op->size = 4;
|
|
if (ctx->omode != mode) {
|
|
cs_close(&ctx->handle);
|
|
ctx->handle = 0;
|
|
ctx->omode = -1;
|
|
}
|
|
if (!ctx->handle) {
|
|
ret = cs_open(CS_ARCH_MIPS, mode, &ctx->handle);
|
|
if (ret) {
|
|
goto fin;
|
|
}
|
|
ctx->omode = mode;
|
|
cs_option(ctx->handle, CS_OPT_DETAIL, CS_OPT_OFF);
|
|
}
|
|
if (a->syntax == RZ_ASM_SYNTAX_REGNUM) {
|
|
cs_option(ctx->handle, CS_OPT_SYNTAX, CS_OPT_SYNTAX_NOREGNAME);
|
|
} else {
|
|
cs_option(ctx->handle, CS_OPT_SYNTAX, CS_OPT_SYNTAX_DEFAULT);
|
|
}
|
|
n = cs_disasm(ctx->handle, (ut8 *)buf, len, a->pc, 1, &insn);
|
|
if (n < 1) {
|
|
rz_asm_op_set_asm(op, "invalid");
|
|
op->size = 4;
|
|
goto fin;
|
|
}
|
|
if (insn->size < 1) {
|
|
goto fin;
|
|
}
|
|
op->size = insn->size;
|
|
rz_asm_op_setf_asm(op, "%s%s%s", insn->mnemonic, insn->op_str[0] ? " " : "", insn->op_str);
|
|
char *str = rz_asm_op_get_asm(op);
|
|
if (str) {
|
|
// remove the '$'<registername> in the string
|
|
rz_str_replace_char(str, '$', 0);
|
|
}
|
|
cs_free(insn, n);
|
|
fin:
|
|
return op->size;
|
|
}
|
|
|
|
static int assemble(RzAsm *a, RzAsmOp *op, const char *str) {
|
|
ut8 *opbuf = (ut8 *)rz_strbuf_get(&op->buf);
|
|
int ret = mips_assemble(str, a->pc, opbuf);
|
|
if (a->big_endian) {
|
|
ut8 *buf = opbuf;
|
|
ut8 tmp = buf[0];
|
|
buf[0] = buf[3];
|
|
buf[3] = tmp;
|
|
tmp = buf[1];
|
|
buf[1] = buf[2];
|
|
buf[2] = tmp;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
RzAsmPlugin rz_asm_plugin_mips_cs = {
|
|
.name = "mips",
|
|
.desc = "Capstone MIPS disassembler",
|
|
.license = "BSD",
|
|
.arch = "mips",
|
|
.cpus = "mips32/64,micro,r6,v3,v2",
|
|
.bits = 16 | 32 | 64,
|
|
.endian = RZ_SYS_ENDIAN_LITTLE | RZ_SYS_ENDIAN_BIG,
|
|
.init = mips_init,
|
|
.fini = mips_fini,
|
|
.disassemble = &mips_disassemble,
|
|
.mnemonics = mips_mnemonics,
|
|
.assemble = &assemble
|
|
};
|
|
|
|
#ifndef RZ_PLUGIN_INCORE
|
|
RZ_API RzLibStruct rizin_plugin = {
|
|
.type = RZ_LIB_TYPE_ASM,
|
|
.data = &rz_asm_plugin_mips_cs,
|
|
.version = RZ_VERSION
|
|
};
|
|
#endif
|