|
 |
59b920 |
/***************************************************************/
|
|
|
59b920 |
/* perk: PE Resource Kit */
|
|
|
59b920 |
/* Copyright (C) 2015--2025 SysDeer Technologies, LLC */
|
|
|
59b920 |
/* Released under GPLv2 and GPLv3; see COPYING.PERK. */
|
|
|
59b920 |
/***************************************************************/
|
|
|
59b920 |
|
|
|
59b920 |
#include <stdio.h>
|
|
|
59b920 |
#include <unistd.h>
|
|
|
59b920 |
#include <perk/perk.h>
|
|
|
59b920 |
#include <perk/perk_output.h>
|
|
|
59b920 |
#include "perk_driver_impl.h"
|
|
|
59b920 |
#include "perk_dprintf_impl.h"
|
|
|
59b920 |
|
|
|
59b920 |
static void pe_perform_hdrdump_actions(
|
|
|
59b920 |
const struct pe_driver_ctx * dctx,
|
|
|
7b3da5 |
const struct pe_image_meta * meta)
|
|
|
59b920 |
{
|
|
|
59b920 |
if (dctx->cctx->hdrdump & PERK_HDRDUMP_IMAGE_DOS_HEADER)
|
|
|
7b3da5 |
pe_hexdump_image_dos_hdr(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (dctx->cctx->hdrdump & PERK_HDRDUMP_COFF_IMAGE_HEADER)
|
|
|
7b3da5 |
pe_hexdump_coff_image_hdr(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (dctx->cctx->hdrdump & PERK_HDRDUMP_COFF_OBJECT_HEADER)
|
|
|
7b3da5 |
pe_hexdump_coff_object_hdr(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (dctx->cctx->hdrdump & PERK_HDRDUMP_COFF_OPT_HEADER)
|
|
|
7b3da5 |
pe_hexdump_coff_opt_hdr(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (dctx->cctx->hdrdump & PERK_HDRDUMP_SECTION_TABLE)
|
|
|
7b3da5 |
pe_hexdump_sec_tbl(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (dctx->cctx->hdrdump & PERK_HDRDUMP_EXPORT_HEADER)
|
|
|
7b3da5 |
pe_hexdump_export_hdr(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (dctx->cctx->hdrdump & PERK_HDRDUMP_IMPORT_TABLE)
|
|
|
7b3da5 |
pe_hexdump_import_tbl(dctx,meta);
|
|
|
59b920 |
}
|
|
|
59b920 |
|
|
|
7b3da5 |
static void pe_perform_unit_actions_impl(
|
|
|
59b920 |
const struct pe_driver_ctx * dctx,
|
|
|
7b3da5 |
const struct pe_image_meta * meta)
|
|
|
59b920 |
{
|
|
|
59b920 |
uint64_t flags = dctx->cctx->fmtflags;
|
|
|
59b920 |
|
|
|
59b920 |
if (flags & PERK_OUTPUT_IMAGE_CATEGORY)
|
|
|
7b3da5 |
pe_output_image_category(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (flags & PERK_OUTPUT_IMAGE_SECTIONS)
|
|
|
7b3da5 |
pe_output_image_sections(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (flags & PERK_OUTPUT_IMAGE_SYMBOLS)
|
|
|
7b3da5 |
pe_output_image_symbols(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (flags & PERK_OUTPUT_IMAGE_STRINGS)
|
|
|
7b3da5 |
pe_output_image_strings(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (flags & PERK_OUTPUT_EXPORT_SYMS)
|
|
|
7b3da5 |
pe_output_export_symbols(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if ((flags & PERK_OUTPUT_IMPORT_LIBS) || (flags & PERK_OUTPUT_IMPORT_SYMS))
|
|
|
7b3da5 |
pe_output_import_libraries(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if ((flags & PERK_OUTPUT_MDSO_LIBS) || (flags & PERK_OUTPUT_MDSO_SYMS))
|
|
|
7b3da5 |
pe_output_mdso_libraries(dctx,meta);
|
|
|
59b920 |
|
|
|
59b920 |
if (dctx->cctx->hdrdump)
|
|
|
7b3da5 |
pe_perform_hdrdump_actions(dctx,meta);
|
|
|
7b3da5 |
}
|
|
|
7b3da5 |
|
|
|
7b3da5 |
static void pe_perform_unit_actions(
|
|
|
7b3da5 |
const struct pe_driver_ctx * dctx,
|
|
|
7b3da5 |
const struct pe_unit_ctx * uctx)
|
|
|
7b3da5 |
{
|
|
|
7b3da5 |
const struct pe_image_meta ** objmeta;
|
|
|
7b3da5 |
|
|
|
7b3da5 |
if (uctx->meta)
|
|
|
7b3da5 |
pe_perform_unit_actions_impl(dctx,uctx->meta);
|
|
|
7b3da5 |
|
|
|
7b3da5 |
if ((objmeta = uctx->objmeta))
|
|
|
7b3da5 |
for (; *objmeta; )
|
|
|
7b3da5 |
pe_perform_unit_actions_impl(dctx,*objmeta++);
|
|
|
59b920 |
}
|
|
|
59b920 |
|
|
|
59b920 |
int pe_cmd_perk(const struct pe_driver_ctx * dctx, const char * path)
|
|
|
59b920 |
{
|
|
|
59b920 |
int ret;
|
|
|
59b920 |
struct pe_unit_ctx * uctx;
|
|
|
59b920 |
|
|
|
118b74 |
if ((ret = pe_lib_get_unit_ctx(dctx,path,&uctx)) < 0)
|
|
|
59b920 |
return ret;
|
|
|
59b920 |
|
|
|
59b920 |
pe_perform_unit_actions(dctx,uctx);
|
|
|
118b74 |
pe_lib_free_unit_ctx(uctx);
|
|
|
59b920 |
|
|
|
59b920 |
return 0;
|
|
|
59b920 |
}
|