cpdb.c File Reference

#include <rz_core.h>
#include <rz_pdb.h>
#include "../bin/pdb/pdb.h"

Go to the source code of this file.

Functions
static void	pdb_types_print_standard (const RzTypeDB *db, const RzPdb *pdb, const RzList *types)
static void	pdb_types_print_json (const RzTypeDB *db, const RzPdb *pdb, const RzList *types, PJ *pj)
static void	rz_core_bin_pdb_types_print (const RzTypeDB *db, const RzPdb *pdb, const RzCmdStateOutput *state)
RZ_API char *	rz_core_bin_pdb_gvars_as_string (RZ_NONNULL const RzPdb *pdb, const ut64 img_base, PJ *pj, const RzOutputMode mode)
	Return the PDB global vars string. More...
static void	rz_core_bin_pdb_gvars_print (const RzPdb *pdb, const ut64 img_base, const RzCmdStateOutput *state)
static void	pdb_set_symbols (const RzCore *core, const RzPdb *pdb, const ut64 img_base, const char *pdbfile)
RZ_API RzPdb *	rz_core_pdb_load_info (RZ_NONNULL RzCore *core, RZ_NONNULL const char *file)
	Parse PDB file info and integrate with typedb. More...
RZ_API void	rz_core_pdb_info_print (RZ_NONNULL RzCore *core, RZ_NONNULL RzTypeDB *db, RZ_NONNULL RzPdb *pdb, RZ_NONNULL RzCmdStateOutput *state)
	Print parsed PDB file info. More...

Function Documentation

pdb_set_symbols()

static void pdb_set_symbols ( const RzCore *  core, const RzPdb *  pdb, const ut64  img_base, const char *  pdbfile  )

static

Definition at line 199 of file cpdb.c.

                                                                                                            {
    rz_return_if_fail(core && pdb);
    PeImageSectionHeader *sctn_header = 0;
    RzPdbGDataStream *gsym_data_stream = 0;
    RzPdbPeStream *pe_stream = 0;
    RzPdbOmapStream *omap_stream;
    GDataGlobal *gdata = 0;
    RzListIter *it = 0;
    char *name;
    char *filtered_name;
    gsym_data_stream = pdb->s_gdata;
    pe_stream = pdb->s_pe;
    omap_stream = pdb->s_omap;
    if (!pe_stream) {
        return;
    }
    char *file = rz_str_replace(strdup(pdbfile), ".pdb", "", 0);
    rz_flag_space_push(core->flags, RZ_FLAGS_FS_SYMBOLS);
    rz_list_foreach (gsym_data_stream->global_list, it, gdata) {
        sctn_header = rz_list_get_n(pe_stream->sections_hdrs, (gdata->segment - 1));
        if (sctn_header) {
            name = rz_demangler_msvc(gdata->name);
            name = (name) ? name : strdup(gdata->name);
            filtered_name = rz_name_filter2(name, true);
            char *fname = rz_str_newf("pdb.%s.%s", file, filtered_name);
            ut64 addr = (ut64)(img_base + rz_bin_pdb_omap_remap(omap_stream, gdata->offset + sctn_header->virtual_address));
            RzFlagItem *item = rz_flag_set(core->flags, fname, addr, 0);
            if (item) {
                rz_flag_item_set_realname(item, name);
            }
            free(filtered_name);
            free(name);
        }
    }
    rz_flag_space_pop(core->flags);
    free(file);
    return;
}

References addr, rz_core_t::flags, create_tags_rz::fname, free(), RzPdbGDataStream::global_list, name, GDataGlobal::name, GDataGlobal::offset, rz_bin_pdb_omap_remap(), rz_demangler_msvc(), rz_flag_item_set_realname(), rz_flag_set(), RZ_FLAGS_FS_SYMBOLS, rz_list_get_n(), rz_name_filter2(), rz_return_if_fail, rz_str_newf(), rz_str_replace(), rz_pdb_t::s_gdata, rz_pdb_t::s_omap, rz_pdb_t::s_pe, RzPdbPeStream::sections_hdrs, GDataGlobal::segment, strdup(), ut64(), and PeImageSectionHeader::virtual_address.

Referenced by rz_core_pdb_load_info().

pdb_types_print_json()

static void pdb_types_print_json ( const RzTypeDB *  db, const RzPdb *  pdb, const RzList *  types, PJ *  pj  )

static

Definition at line 24 of file cpdb.c.

                                                                                                    {
    rz_return_if_fail(db && pdb && types && pj);
    RzListIter *it;
    RzBaseType *type;
    pj_o(pj);
    pj_ka(pj, "types");
    rz_list_foreach (types, it, type) {
        switch (type->kind) {
        case RZ_BASE_TYPE_KIND_STRUCT: {
            pj_o(pj);
            pj_ks(pj, "type", "structure");
            pj_ks(pj, "name", type->name);
            pj_kn(pj, "size", type->size);
            pj_ka(pj, "members");
            RzTypeStructMember *memb;
            rz_vector_foreach(&type->struct_data.members, memb) {
                pj_o(pj);
                char *typ = rz_type_as_string(db, memb->type);
                pj_ks(pj, "member_type", typ);
                RZ_FREE(typ);
                pj_ks(pj, "member_name", memb->name);
                pj_kN(pj, "offset", memb->offset);
                pj_end(pj);
            }
            pj_end(pj);
            pj_end(pj);
            break;
        }
        case RZ_BASE_TYPE_KIND_UNION: {
            pj_o(pj);
            pj_ks(pj, "type", "union");
            pj_ks(pj, "name", type->name);
            pj_kn(pj, "size", type->size);
            pj_ka(pj, "members");
            RzTypeUnionMember *memb;
            rz_vector_foreach(&type->union_data.members, memb) {
                pj_o(pj);
                char *typ = rz_type_as_string(db, memb->type);
                pj_ks(pj, "member_type", typ);
                RZ_FREE(typ);
                pj_ks(pj, "member_name", memb->name);
                pj_kN(pj, "offset", memb->offset);
                pj_end(pj);
            }
            pj_end(pj);
            pj_end(pj);
            break;
        }
        case RZ_BASE_TYPE_KIND_ENUM: {
            pj_o(pj);
            pj_ks(pj, "type", "enum");
            pj_ks(pj, "name", type->name);
            char *typ = rz_type_as_string(db, type->type);
            pj_ks(pj, "base_type", typ);
            RZ_FREE(typ);
            pj_ka(pj, "cases");
            RzTypeEnumCase *cas;
            rz_vector_foreach(&type->enum_data.cases, cas) {
                pj_o(pj);
                pj_ks(pj, "enum_name", cas->name);
                pj_kn(pj, "enum_val", cas->val);
                pj_end(pj);
            }
            pj_end(pj);
            pj_end(pj);
            break;
        }
        default:
            break;
        }
    }
    pj_end(pj);
    pj_end(pj);
    return;
}

References rz_type_enum_case_t::name, rz_type_struct_member_t::name, rz_type_union_member_t::name, rz_type_struct_member_t::offset, rz_type_union_member_t::offset, pj_end(), pj_ka(), pj_kN(), pj_kn(), pj_ks(), pj_o(), RZ_BASE_TYPE_KIND_ENUM, RZ_BASE_TYPE_KIND_STRUCT, RZ_BASE_TYPE_KIND_UNION, RZ_FREE, rz_return_if_fail, rz_type_as_string(), rz_vector_foreach, type, rz_type_struct_member_t::type, rz_type_union_member_t::type, types, and rz_type_enum_case_t::val.

Referenced by rz_core_bin_pdb_types_print().

pdb_types_print_standard()

static void pdb_types_print_standard ( const RzTypeDB *  db, const RzPdb *  pdb, const RzList *  types  )

static

Definition at line 8 of file cpdb.c.

                                                                                                {
    rz_return_if_fail(pdb && db && types);
    if (!types) {
        eprintf("there is nothing to print!\n");
    }
    RzListIter *it;
    RzBaseType *type;
    RzStrBuf *buf = rz_strbuf_new(NULL);
    rz_list_foreach (types, it, type) {
        rz_strbuf_append(buf, rz_type_db_base_type_as_pretty_string(db, type, RZ_TYPE_PRINT_MULTILINE | RZ_TYPE_PRINT_END_NEWLINE, 1));
    }
    rz_cons_print(rz_strbuf_get(buf));
    rz_strbuf_free(buf);
    return;
}

References eprintf, NULL, rz_return_if_fail, rz_strbuf_append(), rz_strbuf_free(), rz_strbuf_get(), rz_strbuf_new(), rz_type_db_base_type_as_pretty_string(), RZ_TYPE_PRINT_END_NEWLINE, RZ_TYPE_PRINT_MULTILINE, type, and types.

Referenced by rz_core_bin_pdb_types_print().

rz_core_bin_pdb_gvars_as_string()

RZ_API char* rz_core_bin_pdb_gvars_as_string	(	RZ_NONNULL const RzPdb *	pdb,
		const ut64	img_base,
		PJ *	pj,
		const RzOutputMode	mode
	)

Return the PDB global vars string.

Parameters

pdb	PDB instance
img_base	image base addr
pj	JSON instance
mode	RzOutputMode

Returns

char *

Definition at line 128 of file cpdb.c.

                                                                                                                                {
    rz_return_val_if_fail(pdb, NULL);
    PeImageSectionHeader *sctn_header = 0;
    RzPdbGDataStream *gsym_data_stream = 0;
    RzPdbPeStream *pe_stream = 0;
    RzPdbOmapStream *omap_stream;
    GDataGlobal *gdata = 0;
    RzListIter *it = 0;
    char *name;
    RzStrBuf *buf = rz_strbuf_new(NULL);
    if (!buf) {
        return NULL;
    }
    if (mode == RZ_OUTPUT_MODE_JSON) {
        pj_o(pj);
        pj_ka(pj, "gvars");
    }
    gsym_data_stream = pdb->s_gdata;
    pe_stream = pdb->s_pe;
    omap_stream = pdb->s_omap;
    if (!pe_stream) {
        rz_strbuf_free(buf);
        return NULL;
    }
    rz_list_foreach (gsym_data_stream->global_list, it, gdata) {
        sctn_header = rz_list_get_n(pe_stream->sections_hdrs, (gdata->segment - 1));
        if (sctn_header) {
            name = rz_demangler_msvc(gdata->name);
            name = (name) ? name : strdup(gdata->name);
            switch (mode) {
            case RZ_OUTPUT_MODE_JSON: // JSON
                pj_o(pj);
                pj_kN(pj, "address", (img_base + rz_bin_pdb_omap_remap(omap_stream, gdata->offset + sctn_header->virtual_address)));
                pj_kN(pj, "symtype", gdata->symtype);
                pj_ks(pj, "section_name", sctn_header->name);
                pj_ks(pj, "gdata_name", name);
                pj_end(pj);
                break;
            case RZ_OUTPUT_MODE_STANDARD:
                rz_strbuf_appendf(buf, "0x%08" PFMT64x "  %d  %.*s  %s\n",
                    (ut64)(img_base + rz_bin_pdb_omap_remap(omap_stream, gdata->offset + sctn_header->virtual_address)),
                    gdata->symtype, PDB_SIZEOF_SECTION_NAME, sctn_header->name, name);
                break;
            default:
                break;
            }
            free(name);
        }
    }
    if (mode == RZ_OUTPUT_MODE_JSON) {
        pj_end(pj);
        pj_end(pj);
        // We will need this for Windows Heap.
        rz_strbuf_append(buf, pj_string(pj));
    }
    char *str = strdup(rz_strbuf_get(buf));
    rz_strbuf_free(buf);
    return str;
}

References free(), RzPdbGDataStream::global_list, name, GDataGlobal::name, PeImageSectionHeader::name, NULL, GDataGlobal::offset, PDB_SIZEOF_SECTION_NAME, PFMT64x, pj_end(), pj_ka(), pj_kN(), pj_ks(), pj_o(), pj_string(), rz_bin_pdb_omap_remap(), rz_demangler_msvc(), rz_list_get_n(), RZ_OUTPUT_MODE_JSON, RZ_OUTPUT_MODE_STANDARD, rz_return_val_if_fail, rz_strbuf_append(), rz_strbuf_appendf(), rz_strbuf_free(), rz_strbuf_get(), rz_strbuf_new(), RzPdbPeStream::sections_hdrs, GDataGlobal::segment, cmd_descs_generate::str, strdup(), GDataGlobal::symtype, ut64(), and PeImageSectionHeader::virtual_address.

Referenced by GetHeapGlobalsOffset(), and rz_core_bin_pdb_gvars_print().

rz_core_bin_pdb_gvars_print()

static void rz_core_bin_pdb_gvars_print ( const RzPdb *  pdb, const ut64  img_base, const RzCmdStateOutput *  state  )

static

Definition at line 188 of file cpdb.c.

                                                                                                              {
    rz_return_if_fail(pdb && state);
    char *str = rz_core_bin_pdb_gvars_as_string(pdb, img_base, state->d.pj, state->mode);
    // We don't need to print the output of JSON because the RzCmdStateOutput will handle it.
    if (state->mode == RZ_OUTPUT_MODE_STANDARD) {
        rz_cons_print(str);
    }
    free(str);
    return;
}

References free(), rz_core_bin_pdb_gvars_as_string(), RZ_OUTPUT_MODE_STANDARD, rz_return_if_fail, and cmd_descs_generate::str.

Referenced by rz_core_pdb_info_print().

rz_core_bin_pdb_types_print()

static void rz_core_bin_pdb_types_print ( const RzTypeDB *  db, const RzPdb *  pdb, const RzCmdStateOutput *  state  )

static

Definition at line 100 of file cpdb.c.

                                                                                                             {
    rz_return_if_fail(db && pdb && state);
    RzPdbTpiStream *stream = pdb->s_tpi;
    if (!stream) {
        eprintf("There is no tpi stream in current pdb\n");
        return;
    }
    switch (state->mode) {
    case RZ_OUTPUT_MODE_STANDARD:
        pdb_types_print_standard(db, pdb, stream->print_type);
        break;
    case RZ_OUTPUT_MODE_JSON:
        pdb_types_print_json(db, pdb, stream->print_type, state->d.pj);
        break;
    default:
        return;
    }
}

References eprintf, pdb_types_print_json(), pdb_types_print_standard(), RZ_OUTPUT_MODE_JSON, RZ_OUTPUT_MODE_STANDARD, rz_return_if_fail, and rz_pdb_t::s_tpi.

Referenced by rz_core_pdb_info_print().

rz_core_pdb_info_print()

RZ_API void rz_core_pdb_info_print	(	RZ_NONNULL RzCore *	core,
		RZ_NONNULL RzTypeDB *	db,
		RZ_NONNULL RzPdb *	pdb,
		RZ_NONNULL RzCmdStateOutput *	state
	)

Print parsed PDB file info.

Parameters

db	RzTypeDB
pdb	instance of PDB
state	Output State

Returns

void

Definition at line 273 of file cpdb.c.

                                                                                                                                                {
    rz_return_if_fail(db && pdb && state);
 
    ut64 baddr = rz_config_get_i(core->config, "bin.baddr");
    if (core->bin->cur && core->bin->cur->o && core->bin->cur->o->opts.baseaddr) {
        baddr = core->bin->cur->o->opts.baseaddr;
    } else {
        eprintf("Warning: Cannot find base address, flags will probably be misplaced\n");
    }
 
    rz_cmd_state_output_array_start(state);
    rz_core_bin_pdb_types_print(core->analysis->typedb, pdb, state);
    rz_core_bin_pdb_gvars_print(pdb, baddr, state);
    rz_cmd_state_output_array_end(state);
    return;
}

References baddr(), eprintf, rz_cmd_state_output_array_end(), rz_cmd_state_output_array_start(), rz_config_get_i(), rz_core_bin_pdb_gvars_print(), rz_core_bin_pdb_types_print(), rz_return_if_fail, and ut64().

Referenced by rz_cmd_info_pdb_show_handler(), and rz_core_bin_print().

rz_core_pdb_load_info()

RZ_API RzPdb* rz_core_pdb_load_info	(	RZ_NONNULL RzCore *	core,
		RZ_NONNULL const char *	file
	)

Parse PDB file info and integrate with typedb.

Parameters

core	RzCore instance
file	Path of PDB file

Returns

bool

Definition at line 245 of file cpdb.c.

                                                                                          {
    rz_return_val_if_fail(core && file, NULL);
 
    ut64 baddr = rz_bin_get_baddr(core->bin);
    if (!baddr || baddr == UT64_MAX) {
        baddr = rz_config_get_i(core->config, "bin.baddr");
        eprintf("Warning: Cannot find base address, flags will probably be misplaced\n");
    }
 
    RzPdb *pdb = rz_bin_pdb_parse_from_file(file);
    if (!pdb) {
        return NULL;
    }
 
    // Save compound types into types database
    rz_parse_pdb_types(core->analysis->typedb, pdb);
    pdb_set_symbols(core, pdb, baddr, rz_file_basename(file));
    return pdb;
}

References baddr(), eprintf, NULL, pdb_set_symbols(), rz_bin_get_baddr(), rz_bin_pdb_parse_from_file(), rz_config_get_i(), rz_file_basename(), rz_parse_pdb_types(), rz_return_val_if_fail, ut64(), and UT64_MAX.

Referenced by rz_cmd_info_pdb_show_handler(), rz_core_bin_pdb_load(), and rz_core_bin_print().