EVMDisassembler.c

Go to the documentation of this file.

/* Capstone Disassembly Engine */
/* By Nguyen Anh Quynh, 2018 */
 
#include <string.h>
#include <stddef.h> // offsetof macro
                    // alternatively #include "../../utils.h" like everyone else
 
#include "EVMDisassembler.h"
#include "EVMMapping.h"
 
static short opcodes[256] = {
    EVM_INS_STOP,
    EVM_INS_ADD,
    EVM_INS_MUL,
    EVM_INS_SUB,
    EVM_INS_DIV,
    EVM_INS_SDIV,
    EVM_INS_MOD,
    EVM_INS_SMOD,
    EVM_INS_ADDMOD,
    EVM_INS_MULMOD,
    EVM_INS_EXP,
    EVM_INS_SIGNEXTEND,
    -1,
    -1,
    -1,
    -1,
    EVM_INS_LT,
    EVM_INS_GT,
    EVM_INS_SLT,
    EVM_INS_SGT,
    EVM_INS_EQ,
    EVM_INS_ISZERO,
    EVM_INS_AND,
    EVM_INS_OR,
    EVM_INS_XOR,
    EVM_INS_NOT,
    EVM_INS_BYTE,
    -1,
    -1,
    -1,
    -1,
    -1,
    EVM_INS_SHA3,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    EVM_INS_ADDRESS,
    EVM_INS_BALANCE,
    EVM_INS_ORIGIN,
    EVM_INS_CALLER,
    EVM_INS_CALLVALUE,
    EVM_INS_CALLDATALOAD,
    EVM_INS_CALLDATASIZE,
    EVM_INS_CALLDATACOPY,
    EVM_INS_CODESIZE,
    EVM_INS_CODECOPY,
    EVM_INS_GASPRICE,
    EVM_INS_EXTCODESIZE,
    EVM_INS_EXTCODECOPY,
    EVM_INS_RETURNDATASIZE,
    EVM_INS_RETURNDATACOPY,
    -1,
    EVM_INS_BLOCKHASH,
    EVM_INS_COINBASE,
    EVM_INS_TIMESTAMP,
    EVM_INS_NUMBER,
    EVM_INS_DIFFICULTY,
    EVM_INS_GASLIMIT,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    EVM_INS_POP,
    EVM_INS_MLOAD,
    EVM_INS_MSTORE,
    EVM_INS_MSTORE8,
    EVM_INS_SLOAD,
    EVM_INS_SSTORE,
    EVM_INS_JUMP,
    EVM_INS_JUMPI,
    EVM_INS_PC,
    EVM_INS_MSIZE,
    EVM_INS_GAS,
    EVM_INS_JUMPDEST,
    -1,
    -1,
    -1,
    -1,
    EVM_INS_PUSH1,
    EVM_INS_PUSH2,
    EVM_INS_PUSH3,
    EVM_INS_PUSH4,
    EVM_INS_PUSH5,
    EVM_INS_PUSH6,
    EVM_INS_PUSH7,
    EVM_INS_PUSH8,
    EVM_INS_PUSH9,
    EVM_INS_PUSH10,
    EVM_INS_PUSH11,
    EVM_INS_PUSH12,
    EVM_INS_PUSH13,
    EVM_INS_PUSH14,
    EVM_INS_PUSH15,
    EVM_INS_PUSH16,
    EVM_INS_PUSH17,
    EVM_INS_PUSH18,
    EVM_INS_PUSH19,
    EVM_INS_PUSH20,
    EVM_INS_PUSH21,
    EVM_INS_PUSH22,
    EVM_INS_PUSH23,
    EVM_INS_PUSH24,
    EVM_INS_PUSH25,
    EVM_INS_PUSH26,
    EVM_INS_PUSH27,
    EVM_INS_PUSH28,
    EVM_INS_PUSH29,
    EVM_INS_PUSH30,
    EVM_INS_PUSH31,
    EVM_INS_PUSH32,
    EVM_INS_DUP1,
    EVM_INS_DUP2,
    EVM_INS_DUP3,
    EVM_INS_DUP4,
    EVM_INS_DUP5,
    EVM_INS_DUP6,
    EVM_INS_DUP7,
    EVM_INS_DUP8,
    EVM_INS_DUP9,
    EVM_INS_DUP10,
    EVM_INS_DUP11,
    EVM_INS_DUP12,
    EVM_INS_DUP13,
    EVM_INS_DUP14,
    EVM_INS_DUP15,
    EVM_INS_DUP16,
    EVM_INS_SWAP1,
    EVM_INS_SWAP2,
    EVM_INS_SWAP3,
    EVM_INS_SWAP4,
    EVM_INS_SWAP5,
    EVM_INS_SWAP6,
    EVM_INS_SWAP7,
    EVM_INS_SWAP8,
    EVM_INS_SWAP9,
    EVM_INS_SWAP10,
    EVM_INS_SWAP11,
    EVM_INS_SWAP12,
    EVM_INS_SWAP13,
    EVM_INS_SWAP14,
    EVM_INS_SWAP15,
    EVM_INS_SWAP16,
    EVM_INS_LOG0,
    EVM_INS_LOG1,
    EVM_INS_LOG2,
    EVM_INS_LOG3,
    EVM_INS_LOG4,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    -1,
    EVM_INS_CREATE,
    EVM_INS_CALL,
    EVM_INS_CALLCODE,
    EVM_INS_RETURN,
    EVM_INS_DELEGATECALL,
    EVM_INS_CALLBLACKBOX,
    -1,
    -1,
    -1,
    -1,
    EVM_INS_STATICCALL,
    -1,
    -1,
    EVM_INS_REVERT,
    -1,
    EVM_INS_SUICIDE,
};
 
bool EVM_getInstruction(csh ud, const uint8_t *code, size_t code_len,
    MCInst *MI, uint16_t *size, uint64_t address, void *inst_info)
{
    unsigned char opcode;
 
    if (code_len == 0)
        return false;
 
    opcode = code[0];
    if (opcodes[opcode] == -1) {
        // invalid opcode
        return false;
    }
 
    // valid opcode
    MI->address = address;
    MI->OpcodePub = MI->Opcode = opcode;
 
    if (opcode >= EVM_INS_PUSH1 && opcode <= EVM_INS_PUSH32) {
        unsigned char len = (opcode - EVM_INS_PUSH1 + 1);
        if (code_len < 1 + len) {
            // not enough data
            return false;
        }
 
        *size = 1 + len;
        memcpy(MI->evm_data, code + 1, len);
    } else
        *size = 1;
 
    if (MI->flat_insn->detail) {
        memset(MI->flat_insn->detail, 0, offsetof(cs_detail, evm)+sizeof(cs_evm));
        EVM_get_insn_id((cs_struct *)ud, MI->flat_insn, opcode);
 
        if (MI->flat_insn->detail->evm.pop) {
            MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_STACK_READ;
            MI->flat_insn->detail->groups_count++;
        }
 
        if (MI->flat_insn->detail->evm.push) {
            MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_STACK_WRITE;
            MI->flat_insn->detail->groups_count++;
        }
 
        // setup groups
        switch(opcode) {
            default:
                break;
            case EVM_INS_ADD:
            case EVM_INS_MUL:
            case EVM_INS_SUB:
            case EVM_INS_DIV:
            case EVM_INS_SDIV:
            case EVM_INS_MOD:
            case EVM_INS_SMOD:
            case EVM_INS_ADDMOD:
            case EVM_INS_MULMOD:
            case EVM_INS_EXP:
            case EVM_INS_SIGNEXTEND:
                MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_MATH;
                MI->flat_insn->detail->groups_count++;
                break;
 
            case EVM_INS_MSTORE:
            case EVM_INS_MSTORE8:
            case EVM_INS_CALLDATACOPY:
            case EVM_INS_CODECOPY:
            case EVM_INS_EXTCODECOPY:
                MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_MEM_WRITE;
                MI->flat_insn->detail->groups_count++;
                break;
 
            case EVM_INS_MLOAD:
            case EVM_INS_CREATE:
            case EVM_INS_CALL:
            case EVM_INS_CALLCODE:
            case EVM_INS_RETURN:
            case EVM_INS_DELEGATECALL:
            case EVM_INS_REVERT:
                MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_MEM_READ;
                MI->flat_insn->detail->groups_count++;
                break;
 
            case EVM_INS_SSTORE:
                MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_STORE_WRITE;
                MI->flat_insn->detail->groups_count++;
                break;
 
            case EVM_INS_SLOAD:
                MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_STORE_READ;
                MI->flat_insn->detail->groups_count++;
                break;
 
            case EVM_INS_JUMP:
            case EVM_INS_JUMPI:
                MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_JUMP;
                MI->flat_insn->detail->groups_count++;
                break;
 
            case EVM_INS_STOP:
            case EVM_INS_SUICIDE:
                MI->flat_insn->detail->groups[MI->flat_insn->detail->groups_count] = EVM_GRP_HALT;
                MI->flat_insn->detail->groups_count++;
                break;
 
        }
    }
 
    return true;
}