luac_bin.c File Reference

#include "luac_common.h"

Go to the source code of this file.

Functions
void	luac_add_section (RzList *section_list, char *name, ut64 offset, ut32 size, bool is_func)
void	luac_add_symbol (RzList *symbol_list, char *name, ut64 offset, ut64 size, const char *type)
void	luac_add_entry (RzList *entry_list, ut64 offset, int entry_type)
void	luac_add_string (RzList *string_list, char *string, ut64 offset, ut64 size)
static void	try_free_empty_list (RzList *list)
static void	free_rz_section (RzBinSection *section)
static void	free_rz_string (RzBinString *string)
static void	free_rz_addr (RzBinAddr *addr)
LuacBinInfo *	luac_build_info (LuaProto *proto)
static const char *	get_tag_string (ut8 tag)
static char *	get_constant_symbol_name (char *proto_name, LuaConstEntry *entry)
static char *	simple_build_upvalue_symbol (char *proto_name, LuaUpvalueEntry *entry)
static char *	get_upvalue_symbol_name (char *proto_name, LuaUpvalueEntry *entry, char *debug_name)
void	_luac_build_info (LuaProto *proto, LuacBinInfo *info)

Function Documentation

_luac_build_info()

void _luac_build_info	(	LuaProto *	proto,
		LuacBinInfo *	info
	)

Definition at line 225 of file luac_bin.c.

                                                          {
    /* process proto header info */
    char *section_name;
    char *symbol_name;
    char *proto_name;
    RzListIter *iter;
 
    ut64 current_offset;
    ut64 current_size;
 
    int i = 0; // iter
 
    // 0. check if stripped (proto name is lost)
    if (proto->name_size == 0 || proto->proto_name == NULL) {
        // replace name with current offset
        proto_name = rz_str_newf("fcn.%08llx", proto->offset);
    } else {
        proto_name = rz_str_new((char *)proto->proto_name);
    }
 
    // 1.1 set section name as function_name.header
    current_offset = proto->offset;
    current_size = proto->size;
    section_name = rz_str_newf("%s.header", proto_name);
    luac_add_section(info->section_list, section_name, current_offset, current_size, false);
    RZ_FREE(section_name);
 
    // 1.2 set section name as function_name.code
    current_offset = proto->code_offset;
    current_size = proto->code_size;
    section_name = rz_str_newf("%s.code", proto_name);
    luac_add_section(info->section_list, section_name, current_offset, current_size, true);
    RZ_FREE(section_name);
 
    // 1.3 set const section
    current_offset = proto->const_offset;
    current_size = proto->const_size;
    section_name = rz_str_newf("%s.const", proto_name);
    luac_add_section(info->section_list, section_name, current_offset, current_size, false);
    RZ_FREE(section_name);
 
    // 1.4 upvalue section
    current_offset = proto->upvalue_offset;
    current_size = proto->upvalue_size;
    section_name = rz_str_newf("%s.upvalues", proto_name);
    luac_add_section(info->section_list, section_name, current_offset, current_size, false);
    RZ_FREE(section_name);
 
    // 1.5 inner protos section
    current_offset = proto->inner_proto_offset;
    current_size = proto->inner_proto_size;
    section_name = rz_str_newf("%s.protos", proto_name);
    luac_add_section(info->section_list, section_name, current_offset, current_size, false);
    RZ_FREE(section_name);
 
    // 1.6 debug section
    current_offset = proto->debug_offset;
    current_size = proto->debug_size;
    section_name = rz_str_newf("%s.debug", proto_name);
    luac_add_section(info->section_list, section_name, current_offset, current_size, false);
    RZ_FREE(section_name);
 
    // 2.1 parse local var info
    LuaLocalVarEntry *local_var_entry;
    rz_list_foreach (proto->local_var_info_entries, iter, local_var_entry) {
        luac_add_string(
            info->string_list,
            (char *)local_var_entry->varname,
            local_var_entry->offset,
            local_var_entry->varname_len);
    }
 
    // 2.2 parse debug_upvalues
    char **upvalue_names;
    int real_upvalue_cnt;
    LuaDbgUpvalueEntry *debug_upv_entry;
    real_upvalue_cnt = rz_list_length(proto->upvalue_entries);
    upvalue_names = RZ_NEWS0(char *, real_upvalue_cnt);
    if (!upvalue_names) {
        return;
    }
    rz_list_foreach (proto->dbg_upvalue_entries, iter, debug_upv_entry) {
        upvalue_names[i] = (char *)debug_upv_entry->upvalue_name;
        luac_add_string(
            info->string_list,
            upvalue_names[i],
            debug_upv_entry->offset,
            debug_upv_entry->name_len);
    }
 
    // 3.1 construct constant symbols
    LuaConstEntry *const_entry;
    rz_list_foreach (proto->const_entries, iter, const_entry) {
        symbol_name = get_constant_symbol_name(proto_name, const_entry);
        luac_add_symbol(
            info->symbol_list,
            symbol_name,
            const_entry->offset,
            const_entry->data_len,
            get_tag_string(const_entry->tag));
        if (const_entry->tag == LUA_VLNGSTR || const_entry->tag == LUA_VSHRSTR) {
            luac_add_string(
                info->string_list,
                (char *)const_entry->data,
                const_entry->offset,
                const_entry->data_len);
        }
        RZ_FREE(symbol_name);
    }
 
    // 3.2 construct upvalue symbols
    LuaUpvalueEntry *upvalue_entry;
    i = 0;
    rz_list_foreach (proto->upvalue_entries, iter, upvalue_entry) {
        symbol_name = get_upvalue_symbol_name(proto_name, upvalue_entry, upvalue_names[i++]);
        luac_add_symbol(
            info->symbol_list,
            symbol_name,
            upvalue_entry->offset,
            3,
            "UPVALUE");
        RZ_FREE(symbol_name);
    }
 
    // 4. parse sub proto
    LuaProto *sub_proto;
    rz_list_foreach (proto->proto_entries, iter, sub_proto) {
        _luac_build_info(sub_proto, info);
    }
 
    RZ_FREE(proto_name);
}

References lua_proto_ex::code_offset, lua_proto_ex::code_size, lua_proto_ex::const_entries, lua_proto_ex::const_offset, lua_proto_ex::const_size, lua_constant_entry::data, lua_constant_entry::data_len, lua_proto_ex::dbg_upvalue_entries, lua_proto_ex::debug_offset, lua_proto_ex::debug_size, get_constant_symbol_name(), get_tag_string(), get_upvalue_symbol_name(), i, info(), lua_proto_ex::inner_proto_offset, lua_proto_ex::inner_proto_size, lua_proto_ex::local_var_info_entries, LUA_VLNGSTR, LUA_VSHRSTR, luac_add_section(), luac_add_string(), luac_add_symbol(), lua_dbg_upvalue_entry::name_len, lua_proto_ex::name_size, NULL, lua_proto_ex::offset, lua_constant_entry::offset, lua_upvalue_entry::offset, lua_local_var_entry::offset, lua_dbg_upvalue_entry::offset, lua_proto_ex::proto_entries, lua_proto_ex::proto_name, RZ_FREE, rz_list_length(), RZ_NEWS0, rz_str_new(), rz_str_newf(), lua_proto_ex::size, lua_constant_entry::tag, lua_proto_ex::upvalue_entries, lua_dbg_upvalue_entry::upvalue_name, lua_proto_ex::upvalue_offset, lua_proto_ex::upvalue_size, ut64(), lua_local_var_entry::varname, and lua_local_var_entry::varname_len.

Referenced by luac_build_info().

free_rz_addr()

static void free_rz_addr ( RzBinAddr *  addr )

static

Definition at line 107 of file luac_bin.c.

                                          {
    if (!addr) {
        return;
    }
    RZ_FREE(addr);
}

References addr, and RZ_FREE.

Referenced by luac_build_info().

free_rz_section()

static void free_rz_section ( RzBinSection *  section )

static

Definition at line 79 of file luac_bin.c.

                                                   {
    if (!section) {
        return;
    }
 
    if (section->name) {
        RZ_FREE(section->name);
    }
 
    if (section->format) {
        RZ_FREE(section->format);
    }
 
    RZ_FREE(section);
}

References RZ_FREE.

Referenced by luac_build_info().

free_rz_string()

static void free_rz_string ( RzBinString *  string )

static

Definition at line 95 of file luac_bin.c.

                                                {
    if (!string) {
        return;
    }
 
    if (string->string) {
        RZ_FREE(string->string);
    }
 
    RZ_FREE(string);
}

References RZ_FREE, and rz_bin_string_t::string.

Referenced by luac_build_info().

get_constant_symbol_name()

static char* get_constant_symbol_name ( char *  proto_name, LuaConstEntry *  entry  )

static

Definition at line 166 of file luac_bin.c.

                                                                              {
    rz_return_val_if_fail(entry || proto_name, NULL);
    ut8 tag = entry->tag;
    char *ret;
    int integer_value;
    double float_value;
 
    switch (tag) {
    case LUA_VNIL:
        ret = rz_str_newf("%s_const_nil", proto_name);
        break;
    case LUA_VTRUE:
        ret = rz_str_newf("%s_const_true", proto_name);
        break;
    case LUA_VFALSE:
        ret = rz_str_newf("%s_const_false", proto_name);
        break;
    case LUA_VSHRSTR:
    case LUA_VLNGSTR:
        rz_return_val_if_fail(entry->data, NULL);
        ret = rz_str_newf("%s_const_%s", proto_name, (char *)entry->data);
        break;
    case LUA_VNUMFLT:
        rz_return_val_if_fail(entry->data, NULL);
        if (entry->data_len < sizeof(double)) {
            return NULL;
        }
        float_value = *(double *)entry->data;
        ret = rz_str_newf("%s_const_%f", proto_name, float_value);
        break;
    case LUA_VNUMINT:
        rz_return_val_if_fail(entry->data, NULL);
        if (entry->data_len < sizeof(int)) {
            return NULL;
        }
        integer_value = *(int *)entry->data;
        ret = rz_str_newf("%s_const_%d", proto_name, integer_value);
        break;
    default:
        ret = rz_str_newf("%s_const_0x%llx", proto_name, entry->offset);
        break;
    }
    return ret;
}

References LUA_VFALSE, LUA_VLNGSTR, LUA_VNIL, LUA_VNUMFLT, LUA_VNUMINT, LUA_VSHRSTR, LUA_VTRUE, NULL, rz_return_val_if_fail, rz_str_newf(), and test-lz4-versions::tag.

Referenced by _luac_build_info().

get_tag_string()

static const char* get_tag_string ( ut8  tag )

static

Definition at line 147 of file luac_bin.c.

                                           {
    switch (tag) {
    case LUA_VNIL:
        return "CONST_NIL";
    case LUA_VTRUE:
    case LUA_VFALSE:
        return "CONST_BOOL";
    case LUA_VSHRSTR:
    case LUA_VLNGSTR:
        return "CONST_STRING";
    case LUA_VNUMFLT:
    case LUA_VNUMINT:
        return "CONST_NUM";
    default:
        return "CONST_UNKNOWN";
    }
}

References LUA_VFALSE, LUA_VLNGSTR, LUA_VNIL, LUA_VNUMFLT, LUA_VNUMINT, LUA_VSHRSTR, LUA_VTRUE, and test-lz4-versions::tag.

Referenced by _luac_build_info().

get_upvalue_symbol_name()

static char* get_upvalue_symbol_name ( char *  proto_name, LuaUpvalueEntry *  entry, char *  debug_name  )

static

Definition at line 216 of file luac_bin.c.

                                                                                                 {
    rz_return_val_if_fail(proto_name || entry, NULL);
    if (debug_name == NULL) {
        return simple_build_upvalue_symbol(proto_name, entry);
    }
 
    return rz_str_newf("%s_upvalue_%s", proto_name, debug_name);
}

References NULL, rz_return_val_if_fail, rz_str_newf(), and simple_build_upvalue_symbol().

Referenced by _luac_build_info().

luac_add_entry()

void luac_add_entry	(	RzList *	entry_list,
		ut64	offset,
		int	entry_type
	)

Definition at line 44 of file luac_bin.c.

                                                                     {
    RzBinAddr *entry = RZ_NEW0(RzBinAddr);
    if (!entry) {
        return;
    }
 
    entry->vaddr = offset;
    entry->paddr = offset;
    entry->type = entry_type;
 
    rz_list_append(entry_list, entry);
}

References rz_list_append(), and RZ_NEW0.

Referenced by luac_build_info().

luac_add_section()

void luac_add_section	(	RzList *	section_list,
		char *	name,
		ut64	offset,
		ut32	size,
		bool	is_func
	)

Definition at line 6 of file luac_bin.c.

                                                                                              {
    RzBinSection *bin_sec = RZ_NEW0(RzBinSection);
    if (!bin_sec) {
        return;
    }
 
    bin_sec->name = rz_str_new(name);
    bin_sec->vaddr = bin_sec->paddr = offset;
    bin_sec->size = bin_sec->vsize = size;
    bin_sec->is_data = false;
    bin_sec->bits = is_func ? sizeof(LUA_INSTRUCTION) * 8 : 8;
    // bin_sec->has_strings = !is_func;
    bin_sec->has_strings = false;
    bin_sec->arch = rz_str_new("luac");
 
    if (is_func) {
        bin_sec->perm = RZ_PERM_R | RZ_PERM_X;
    } else {
        bin_sec->perm = RZ_PERM_R;
    }
 
    rz_list_append(section_list, bin_sec);
}

References rz_bin_section_t::arch, rz_bin_section_t::bits, rz_bin_section_t::has_strings, rz_bin_section_t::is_data, rz_bin_section_t::name, rz_bin_section_t::paddr, rz_bin_section_t::perm, rz_list_append(), RZ_NEW0, RZ_PERM_R, RZ_PERM_X, rz_str_new(), rz_bin_section_t::size, rz_bin_section_t::vaddr, and rz_bin_section_t::vsize.

Referenced by _luac_build_info().

luac_add_string()

void luac_add_string	(	RzList *	string_list,
		char *	string,
		ut64	offset,
		ut64	size
	)

Definition at line 57 of file luac_bin.c.

                                                                                {
    RzBinString *bin_string = RZ_NEW0(RzBinString);
    if (!bin_string) {
        return;
    }
 
    bin_string->paddr = offset;
    bin_string->vaddr = offset;
    bin_string->size = size;
    bin_string->length = size;
    bin_string->string = rz_str_new(string);
    bin_string->type = RZ_STRING_ENC_UTF8;
 
    rz_list_append(string_list, bin_string);
}

References rz_bin_string_t::length, rz_bin_string_t::paddr, rz_list_append(), RZ_NEW0, rz_str_new(), RZ_STRING_ENC_UTF8, rz_bin_string_t::size, rz_bin_string_t::string, rz_bin_string_t::type, and rz_bin_string_t::vaddr.

Referenced by _luac_build_info().

luac_add_symbol()

void luac_add_symbol	(	RzList *	symbol_list,
		char *	name,
		ut64	offset,
		ut64	size,
		const char *	type
	)

Definition at line 30 of file luac_bin.c.

                                                                                                {
    RzBinSymbol *bin_sym = RZ_NEW0(RzBinSymbol);
    if (!bin_sym) {
        return;
    }
 
    bin_sym->name = rz_str_new(name);
    bin_sym->vaddr = bin_sym->paddr = offset;
    bin_sym->size = size;
    bin_sym->type = type;
 
    rz_list_append(symbol_list, bin_sym);
}

References rz_bin_symbol_t::name, rz_bin_symbol_t::paddr, rz_list_append(), RZ_NEW0, rz_str_new(), rz_bin_symbol_t::size, type, rz_bin_symbol_t::type, and rz_bin_symbol_t::vaddr.

Referenced by _luac_build_info().

luac_build_info()

LuacBinInfo* luac_build_info

(

LuaProto *

proto

)

Definition at line 114 of file luac_bin.c.

                                              {
    if (!proto) {
        RZ_LOG_ERROR("Invalid luac file\n");
        return NULL;
    }
 
    LuacBinInfo *ret = RZ_NEW0(LuacBinInfo);
    if (!ret) {
        return NULL;
    }
 
    ret->entry_list = rz_list_newf((RzListFree)free_rz_addr);
    ret->symbol_list = rz_list_newf((RzListFree)rz_bin_symbol_free);
    ret->section_list = rz_list_newf((RzListFree)free_rz_section);
    ret->string_list = rz_list_newf((RzListFree)free_rz_string);
 
    if (!(ret->entry_list && ret->symbol_list && ret->section_list && ret->string_list)) {
        try_free_empty_list(ret->entry_list);
        try_free_empty_list(ret->symbol_list);
        try_free_empty_list(ret->section_list);
        try_free_empty_list(ret->string_list);
    }
 
    _luac_build_info(proto, ret);
 
    // add entry of main
    ut64 main_entry_offset;
    main_entry_offset = proto->code_offset + proto->code_skipped;
    luac_add_entry(ret->entry_list, main_entry_offset, RZ_BIN_ENTRY_TYPE_PROGRAM);
 
    return ret;
}

References _luac_build_info(), lua_proto_ex::code_offset, lua_proto_ex::code_skipped, luac_bin_info::entry_list, free_rz_addr(), free_rz_section(), free_rz_string(), luac_add_entry(), NULL, RZ_BIN_ENTRY_TYPE_PROGRAM, rz_bin_symbol_free(), rz_list_newf(), RZ_LOG_ERROR, RZ_NEW0, luac_bin_info::section_list, luac_bin_info::string_list, luac_bin_info::symbol_list, try_free_empty_list(), and ut64().

Referenced by load_buffer().

simple_build_upvalue_symbol()

static char* simple_build_upvalue_symbol ( char *  proto_name, LuaUpvalueEntry *  entry  )

static

Definition at line 212 of file luac_bin.c.

                                                                                   {
    return rz_str_newf("%s_upvalue_0x%llx", proto_name, entry->offset);
}

References rz_str_newf().

Referenced by get_upvalue_symbol_name().

try_free_empty_list()

static void try_free_empty_list ( RzList *  list )

static

Definition at line 73 of file luac_bin.c.

                                              {
    if (list != NULL) {
        rz_list_free(list);
    }
}

References list(), NULL, and rz_list_free().

Referenced by luac_build_info().