test_evm.c

Go to the documentation of this file.

/* Capstone Disassembler Engine */
/* By Nguyen Anh Quynh <aquynh@gmail.com>, 2018 */
 
#include <stdio.h>
#include <stdlib.h>
 
#include <capstone/platform.h>
#include <capstone/capstone.h>
 
static csh handle;
 
struct platform {
        cs_arch arch;
        cs_mode mode;
        unsigned char *code;
        size_t size;
        const char *comment;
};
 
static void print_string_hex(const char *comment, unsigned char *str, size_t len)
{
        unsigned char *c;
 
        printf("%s", comment);
        for (c = str; c < str + len; c++) {
                printf("0x%02x ", *c & 0xff);
        }
 
        printf("\n");
}
 
static void print_insn_detail(csh cs_handle, cs_insn *ins)
{
        cs_evm *evm;
 
        // detail can be NULL on "data" instruction if SKIPDATA option is turned ON
        if (ins->detail == NULL)
                return;
 
        evm = &(ins->detail->evm);
 
        if (evm->pop)
                printf("\tPop:     %u\n", evm->pop);
 
        if (evm->push)
                printf("\tPush:    %u\n", evm->push);
 
        if (evm->fee)
                printf("\tGas fee: %u\n", evm->fee);
 
        if (ins->detail->groups_count) {
                int j;
 
                printf("\tGroups: ");
                for(j = 0; j < ins->detail->groups_count; j++) {
                        printf("%s ", cs_group_name(handle, ins->detail->groups[j]));
                }
                printf("\n");
        }
}
 
static void test()
{
#define EVM_CODE "\x60\x61\x50"
 
        struct platform platforms[] = {
                {
                        CS_ARCH_EVM,
                        0,
                        (unsigned char *)EVM_CODE,
                        sizeof(EVM_CODE) - 1,
                        "EVM"
                },
        };
 
        uint64_t address = 0x80001000;
        cs_insn *insn;
        int i;
        size_t count;
 
        for (i = 0; i < sizeof(platforms)/sizeof(platforms[0]); i++) {
                cs_err err = cs_open(platforms[i].arch, platforms[i].mode, &handle);
                if (err) {
                        printf("Failed on cs_open() with error returned: %u\n", err);
                        abort();
                }
 
                cs_option(handle, CS_OPT_DETAIL, CS_OPT_ON);
 
                count = cs_disasm(handle, platforms[i].code, platforms[i].size, address, 0, &insn);
                if (count) {
                        size_t j;
                        printf("****************\n");
                        printf("Platform: %s\n", platforms[i].comment);
                        print_string_hex("Code:", platforms[i].code, platforms[i].size);
                        printf("Disasm:\n");
 
                        for (j = 0; j < count; j++) {
                                printf("0x%" PRIx64 ":\t%s\t%s\n", insn[j].address, insn[j].mnemonic, insn[j].op_str);
                                print_insn_detail(handle, &insn[j]);
                        }
                        printf("0x%" PRIx64 ":\n", insn[j-1].address + insn[j-1].size);
 
                        // free memory allocated by cs_disasm()
                        cs_free(insn, count);
                } else {
                        printf("****************\n");
                        printf("Platform: %s\n", platforms[i].comment);
                        print_string_hex("Code:", platforms[i].code, platforms[i].size);
                        printf("ERROR: Failed to disasm given code!\n");
                        abort();
                }
 
                printf("\n");
 
                cs_close(&handle);
        }
}
 
int main()
{
        test();
 
        return 0;
}