opcode.c File Reference

#include "opcode.h"

Go to the source code of this file.

Functions
bool	pyc_opcodes_equal (pyc_opcodes *op, const char *version)
pyc_opcodes *	get_opcode_by_version (char *version)
pyc_opcodes *	new_pyc_opcodes ()
void	free_opcode (pyc_opcodes *opcodes)
void	add_arg_fmt (pyc_opcodes *ret, char *op_name, const char *(*formatter)(ut32 oparg))
void()	def_opN (struct op_parameter par)
void()	name_opN (struct op_parameter par)
void()	local_opN (struct op_parameter par)
void()	free_opN (struct op_parameter par)
void()	store_opN (struct op_parameter par)
void()	varargs_op (struct op_parameter par)
void()	const_opN (struct op_parameter par)
void()	compare_op (struct op_parameter par)
void()	jabs_opN (struct op_parameter par)
void()	jrel_opN (struct op_parameter par)
void()	nargs_op (struct op_parameter par)
void()	rm_op (struct op_parameter par)

Variables
static version_opcode	version_op []

Function Documentation

add_arg_fmt()

void add_arg_fmt	(	pyc_opcodes *	ret,
		char *	op_name,
		const char *(*)(ut32 oparg)	formatter
	)

Definition at line 234 of file opcode.c.

                                                                                        {
    pyc_arg_fmt *fmt = RZ_NEW0(pyc_arg_fmt);
    if (!fmt) {
        return;
    }
    fmt->op_name = op_name;
    fmt->formatter = formatter;
    rz_list_append(ret->opcode_arg_fmt, fmt);
}

References pyc_arg_fmt::formatter, pyc_arg_fmt::op_name, pyc_opcodes::opcode_arg_fmt, rz_list_append(), and RZ_NEW0.

Referenced by opcode_10(), opcode_11(), opcode_12(), opcode_13(), opcode_14(), opcode_15(), opcode_16(), opcode_20(), opcode_21(), opcode_22(), opcode_23(), opcode_24(), opcode_25(), opcode_26(), opcode_27(), opcode_30(), opcode_31(), opcode_32(), opcode_33(), opcode_34(), opcode_35(), opcode_36(), opcode_37(), opcode_38(), opcode_39(), and opcode_3x().

compare_op()

void() compare_op

(

struct op_parameter

par

)

Definition at line 301 of file opcode.c.

                                          {
    def_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
    par.op_obj[par.op_code].type |= HASCOMPARE;
}

References def_op, HASCOMPARE, pyc_opcode_object::op_name, and op_parameter::op_obj.

const_opN()

void() const_opN

(

struct op_parameter

par

)

Definition at line 296 of file opcode.c.

                                         {
    def_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
    par.op_obj[par.op_code].type |= HASCONST;
}

References def_op, HASCONST, pyc_opcode_object::op_name, and op_parameter::op_obj.

def_opN()

void() def_opN

(

struct op_parameter

par

)

Definition at line 244 of file opcode.c.

                                       {
    free(par.op_obj[par.op_code].op_name);
    par.op_obj[par.op_code].op_name = strdup(par.op_name);
    par.op_obj[par.op_code].op_code = par.op_code;
    par.op_obj[par.op_code].op_pop = par.pop;
    par.op_obj[par.op_code].op_push = par.push;
    if (!par.fallthrough) {
        par.op_obj[par.op_code].type |= NOFOLLOW;
    }
}

References free(), NOFOLLOW, and strdup().

Referenced by opcode_15().

free_opcode()

void free_opcode

(

pyc_opcodes *

opcodes

)

Definition at line 217 of file opcode.c.

                                       {
    size_t i;
    if (opcodes == NULL || opcodes->opcodes == NULL) {
        return;
    }
    for (i = 0; i < 256; i++) {
        if (opcodes->opcodes[i].op_name) {
            free(opcodes->opcodes[i].op_name);
        }
    }
    free(opcodes->opcodes);
    if (opcodes->opcode_arg_fmt) {
        rz_list_free(opcodes->opcode_arg_fmt);
    }
    free(opcodes);
}

References free(), i, NULL, opcodes, and rz_list_free().

Referenced by finish(), and pyc_op().

free_opN()

void() free_opN

(

struct op_parameter

par

)

Definition at line 265 of file opcode.c.

                                        {
    def_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
    par.op_obj[par.op_code].type |= HASFREE;
}

References def_op, HASFREE, pyc_opcode_object::op_name, and op_parameter::op_obj.

get_opcode_by_version()

pyc_opcodes* get_opcode_by_version

(

char *

version

)

Definition at line 169 of file opcode.c.

                                                  {
    if (version == NULL) {
        return NULL;
    }
    version_opcode *vop = version_op;
 
    while (vop->version) {
        if (!strcmp(vop->version, version)) {
            return vop->opcode_func();
        }
        vop++;
    }
 
    return NULL; // No match version
}

References NULL, version_opcode::opcode_func, version_opcode::version, and version_op.

Referenced by disassemble(), and pyc_op().

jabs_opN()

void() jabs_opN

(

struct op_parameter

par

)

Definition at line 306 of file opcode.c.

                                        {
    def_op00(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push, .fallthrough = par.fallthrough);
    par.op_obj[par.op_code].type |= HASJABS;
    if (par.conditional) {
        par.op_obj[par.op_code].type |= HASCONDITION;
    }
}

References def_op00, HASCONDITION, HASJABS, pyc_opcode_object::op_name, and op_parameter::op_obj.

jrel_opN()

void() jrel_opN

(

struct op_parameter

par

)

Definition at line 314 of file opcode.c.

                                        {
    def_op00(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push, .fallthrough = par.fallthrough);
    par.op_obj[par.op_code].type |= HASJREL;
    if (par.conditional) {
        par.op_obj[par.op_code].type |= HASCONDITION;
    }
}

References def_op00, HASCONDITION, HASJREL, pyc_opcode_object::op_name, and op_parameter::op_obj.

local_opN()

void() local_opN

(

struct op_parameter

par

)

Definition at line 260 of file opcode.c.

                                         {
    def_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
    par.op_obj[par.op_code].type |= HASLOCAL;
}

References def_op, HASLOCAL, pyc_opcode_object::op_name, and op_parameter::op_obj.

name_opN()

void() name_opN

(

struct op_parameter

par

)

Definition at line 255 of file opcode.c.

                                        {
    def_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
    par.op_obj[par.op_code].type |= HASNAME;
}

References def_op, HASNAME, pyc_opcode_object::op_name, and op_parameter::op_obj.

nargs_op()

void() nargs_op

(

struct op_parameter

par

)

Definition at line 322 of file opcode.c.

                                        {
    def_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
    par.op_obj[par.op_code].type |= HASNARGS;
}

References def_op, HASNARGS, pyc_opcode_object::op_name, and op_parameter::op_obj.

new_pyc_opcodes()

pyc_opcodes* new_pyc_opcodes

(

)

Definition at line 185 of file opcode.c.

                               {
    size_t i, j;
    pyc_opcodes *ret = RZ_NEW0(pyc_opcodes);
    if (!ret) {
        return NULL;
    }
    ret->have_argument = 90;
    ret->opcodes = malloc(sizeof(pyc_opcode_object) * 256);
    if (!ret->opcodes) {
        free(ret);
        return NULL;
    }
    for (i = 0; i < 256; i++) {
        ret->opcodes[i].op_name = rz_str_newf("<%zu>", i);
        if (!ret->opcodes[i].op_name) {
            for (j = 0; j < i; j++) {
                free(ret->opcodes[j].op_name);
            }
            free(ret->opcodes);
            RZ_FREE(ret);
            return NULL;
        }
        ret->opcodes[i].type = 0;
        ret->opcodes[i].op_code = i;
        ret->opcodes[i].op_push = 0;
        ret->opcodes[i].op_pop = 0;
    }
 
    ret->opcode_arg_fmt = rz_list_newf((RzListFree)free);
    return ret;
}

References free(), pyc_opcodes::have_argument, i, malloc(), NULL, pyc_opcode_object::op_code, pyc_opcode_object::op_name, pyc_opcode_object::op_pop, pyc_opcode_object::op_push, pyc_opcodes::opcode_arg_fmt, pyc_opcodes::opcodes, RZ_FREE, rz_list_newf(), RZ_NEW0, rz_str_newf(), and pyc_opcode_object::type.

Referenced by opcode_15(), opcode_2x(), and opcode_3x().

pyc_opcodes_equal()

bool pyc_opcodes_equal	(	pyc_opcodes *	op,
		const char *	version
	)

Definition at line 151 of file opcode.c.

                                                             {
    if (version == NULL || op == NULL) {
        return false;
    }
    version_opcode *vop = version_op;
 
    while (vop->version) {
        if (!strcmp(vop->version, version)) {
            if (vop->opcode_func == (pyc_opcodes * (*)())(op->version_sig)) {
                return true;
            }
        }
        vop++;
    }
 
    return false;
}

References NULL, version_opcode::opcode_func, version_opcode::version, and version_op.

Referenced by disassemble(), and pyc_op().

rm_op()

void() rm_op

(

struct op_parameter

par

)

Definition at line 327 of file opcode.c.

                                     {
    pyc_opcode_object *op_obj = &par.op_obj[par.op_code];
    if (op_obj->op_code == par.op_code && !strcmp(op_obj->op_name, par.op_name)) {
        free(op_obj->op_name);
        op_obj->op_name = rz_str_newf("<%u>", par.op_code);
        op_obj->type = op_obj->op_pop = op_obj->op_push = 0;
    } else {
        RZ_LOG_ERROR("Error in rm_op() while constructing opcodes for .pyc file: \n .op_code = %u, .op_name = %s", par.op_code, par.op_name);
    }
}

References free(), pyc_opcode_object::op_code, pyc_opcode_object::op_name, op_parameter::op_obj, pyc_opcode_object::op_pop, pyc_opcode_object::op_push, RZ_LOG_ERROR, rz_str_newf(), and pyc_opcode_object::type.

store_opN()

void() store_opN

(

struct op_parameter

par

)

Definition at line 270 of file opcode.c.

                                         {
    switch (par.func) {
    case NAME_OP:
        name_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
        break;
    case LOCAL_OP:
        local_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
        break;
    case FREE_OP:
        free_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
        break;
    case DEF_OP:
        def_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
        break;
    default:
        RZ_LOG_ERROR("Error in store_op in pyc/opcode.c, call function %u.\n", par.func);
        return;
    }
    par.op_obj[par.op_code].type |= HASSTORE;
}

References DEF_OP, def_op, FREE_OP, free_op, HASSTORE, LOCAL_OP, local_op, NAME_OP, name_op, pyc_opcode_object::op_name, op_parameter::op_obj, and RZ_LOG_ERROR.

varargs_op()

void() varargs_op

(

struct op_parameter

par

)

Definition at line 291 of file opcode.c.

                                          {
    def_op(.op_obj = par.op_obj, .op_name = par.op_name, .op_code = par.op_code, .pop = par.pop, .push = par.push);
    par.op_obj[par.op_code].type |= HASVARGS;
}

References def_op, HASVARGS, pyc_opcode_object::op_name, and op_parameter::op_obj.

Variable Documentation

version_op

version_opcode version_op[]

static

Definition at line 6 of file opcode.c.

Referenced by get_opcode_by_version(), and pyc_opcodes_equal().