bin_pyc.c File Reference

#include <rz_bin.h>
#include "../format/pyc/pyc.h"

Go to the source code of this file.

Functions
static bool	check_buffer (RzBuffer *b)
static bool	load_buffer (RzBinFile *bf, RzBinObject *obj, RzBuffer *buf, Sdb *sdb)
static ut64	get_entrypoint (RzBinFile *arch)
static RzBinInfo *	info (RzBinFile *arch)
static RzList *	sections (RzBinFile *arch)
static RzList *	entries (RzBinFile *arch)
static ut64	baddr (RzBinFile *bf)
static RzList *	symbols (RzBinFile *arch)
static void	destroy (RzBinFile *bf)

Variables
RzBinPlugin	rz_bin_plugin_pyc
RZ_API RzLibStruct	rizin_plugin

Function Documentation

baddr()

static ut64 baddr ( RzBinFile *  bf )

static

Definition at line 92 of file bin_pyc.c.

                                 {
    return 0;
}

check_buffer()

static bool check_buffer ( RzBuffer *  b )

static

Definition at line 8 of file bin_pyc.c.

                                      {
    if (rz_buf_size(b) > 4) {
        ut32 buf;
        rz_buf_read_at(b, 0, (ut8 *)&buf, sizeof(buf));
        struct pyc_version version = get_pyc_version(buf);
        return version.magic != -1;
    }
    return false;
}

References b, get_pyc_version(), rz_buf_read_at(), and rz_buf_size().

destroy()

static void destroy ( RzBinFile *  bf )

static

Definition at line 134 of file bin_pyc.c.

                                   {
    RzBinPycObj *pyc = bf->o->bin_obj;
    rz_list_free(pyc->shared);
    RZ_FREE(bf->o->bin_obj);
}

References rz_bin_object_t::bin_obj, rz_bin_file_t::o, RZ_FREE, rz_list_free(), and pyc_context::shared.

entries()

static RzList* entries ( RzBinFile *  arch )

static

Definition at line 74 of file bin_pyc.c.

                                        {
    RzList *entries = rz_list_newf((RzListFree)free);
    if (!entries) {
        return NULL;
    }
    RzBinAddr *addr = RZ_NEW0(RzBinAddr);
    if (!addr) {
        rz_list_free(entries);
        return NULL;
    }
    ut64 entrypoint = get_entrypoint(arch);
    addr->paddr = entrypoint;
    addr->vaddr = entrypoint;
    rz_buf_seek(arch->buf, entrypoint, RZ_IO_SEEK_SET);
    rz_list_append(entries, addr);
    return entries;
}

References addr, arch, free(), get_entrypoint(), NULL, rz_buf_seek(), RZ_IO_SEEK_SET, rz_list_append(), rz_list_free(), rz_list_newf(), RZ_NEW0, and ut64().

get_entrypoint()

static ut64 get_entrypoint ( RzBinFile *  arch )

static

Definition at line 30 of file bin_pyc.c.

                                            {
    RzBinPycObj *pyc = arch->o->bin_obj;
    RzBuffer *buf = arch->buf;
    ut8 b;
    ut64 result;
    int addr;
    for (addr = 0x8; addr <= 0x10; addr += 0x4) {
        rz_buf_read_at(buf, addr, &b, sizeof(b));
        if (pyc_is_code(b, pyc->version.magic)) {
            pyc->code_start_offset = addr;
            rz_buf_seek(buf, addr + 1, RZ_BUF_SET);
            if ((result = get_code_object_addr(pyc, buf, pyc->version.magic)) == 0) {
                return addr;
            }
            return result;
        }
    }
    return 0;
}

References addr, arch, b, pyc_context::code_start_offset, get_code_object_addr(), pyc_version::magic, pyc_is_code(), rz_buf_read_at(), rz_buf_seek(), RZ_BUF_SET, ut64(), and pyc_context::version.

Referenced by entries().

info()

static RzBinInfo* info ( RzBinFile *  arch )

static

Definition at line 50 of file bin_pyc.c.

                                        {
    RzBinPycObj *ctx = arch->o->bin_obj;
    RzBinInfo *ret = RZ_NEW0(RzBinInfo);
    if (!ret) {
        return NULL;
    }
    ret->file = strdup(arch->file);
    ret->type = rz_str_newf("Python %s byte-compiled file", ctx->version.version);
    ret->bclass = strdup("Python byte-compiled file");
    ret->rclass = strdup("pyc");
    ret->arch = strdup("pyc");
    ret->machine = rz_str_newf("Python %s VM (rev %s)", ctx->version.version,
        ctx->version.revision);
    ret->os = strdup("any");
    ret->bits = version2double(ctx->version.version) < 3.6 ? 16 : 8;
    ret->cpu = strdup(ctx->version.version); // pass version info in cpu, Asm plugin will get it
    return ret;
}

References rz_bin_info_t::arch, arch, rz_bin_info_t::bclass, rz_bin_info_t::bits, rz_bin_info_t::cpu, rz_bin_info_t::file, rz_bin_info_t::machine, NULL, rz_bin_info_t::os, rz_bin_info_t::rclass, RZ_NEW0, rz_str_newf(), strdup(), rz_bin_info_t::type, and version2double().

load_buffer()

static bool load_buffer ( RzBinFile *  bf, RzBinObject *  obj, RzBuffer *  buf, Sdb *  sdb  )

static

Definition at line 18 of file bin_pyc.c.

                                                                                  {
    RzBinPycObj *ctx = RZ_NEW0(RzBinPycObj);
    if (!ctx) {
        return false;
    }
    ut32 ver;
    rz_buf_read_at(buf, 0, (ut8 *)&ver, sizeof(ver));
    ctx->version = get_pyc_version(ver);
    obj->bin_obj = ctx;
    return true;
}

References rz_bin_object_t::bin_obj, get_pyc_version(), rz_buf_read_at(), and RZ_NEW0.

sections()

static RzList* sections ( RzBinFile *  arch )

static

Definition at line 69 of file bin_pyc.c.

                                         {
    RzBinPycObj *ctx = arch->o->bin_obj;
    return ctx->sections_cache;
}

References arch.

Referenced by symbols().

symbols()

static RzList* symbols ( RzBinFile *  arch )

static

Definition at line 96 of file bin_pyc.c.

                                        {
    RzBinPycObj *pyc = arch->o->bin_obj;
    RzList *shared = rz_list_newf((RzListFree)rz_list_free);
    if (!shared) {
        return NULL;
    }
    RzList *cobjs = rz_list_newf((RzListFree)free);
    if (!cobjs) {
        rz_list_free(shared);
        return NULL;
    }
    pyc->interned_table = rz_list_newf((RzListFree)free);
    if (!pyc->interned_table) {
        rz_list_free(shared);
        rz_list_free(cobjs);
        return NULL;
    }
    rz_list_append(shared, cobjs);
    rz_list_append(shared, pyc->interned_table);
    pyc->shared = shared;
    RzList *sections = rz_list_newf((RzListFree)free);
    if (!sections) {
        rz_list_free(shared);
        return NULL;
    }
    RzList *symbols = rz_list_newf((RzListFree)free);
    if (!symbols) {
        rz_list_free(shared);
        rz_list_free(sections);
        return NULL;
    }
    RzBuffer *buffer = arch->buf;
    rz_buf_seek(buffer, pyc->code_start_offset, RZ_BUF_SET);
    pyc_get_sections_symbols(pyc, sections, symbols, cobjs, buffer, pyc->version.magic);
    pyc->sections_cache = sections;
    return symbols;
}

References arch, pyc_context::code_start_offset, free(), pyc_context::interned_table, pyc_version::magic, NULL, pyc_get_sections_symbols(), rz_buf_seek(), RZ_BUF_SET, rz_list_append(), rz_list_free(), rz_list_newf(), sections(), pyc_context::sections_cache, pyc_context::shared, and pyc_context::version.

Variable Documentation

rizin_plugin

RZ_API RzLibStruct rizin_plugin

Initial value:

= {
    .type = RZ_LIB_TYPE_BIN,
    .data = &rz_bin_plugin_pyc,
    .version = RZ_VERSION,
}

Definition at line 155 of file bin_pyc.c.

rz_bin_plugin_pyc

RzBinPlugin rz_bin_plugin_pyc

Initial value:

= {
    .name = "pyc",
    .desc = "Python byte-compiled file plugin",
    .license = "LGPL3",
    .info = &info,
    .load_buffer = &load_buffer,
    .check_buffer = &check_buffer,
    .entries = &entries,
    .sections = &sections,
    .baddr = &baddr,
    .symbols = &symbols,
    .destroy = &destroy
}

Definition at line 140 of file bin_pyc.c.