1 files changed, 83 insertions, 0 deletions
diff --git a/src/cmds/pe_cmd_perk.c b/src/cmds/pe_cmd_perk.c
new file mode 100644
index 0000000..d2b11a2
--- /dev/null
+++ b/src/cmds/pe_cmd_perk.c
@@ -0,0 +1,83 @@
+/***************************************************************/
+/*  perk: PE Resource Kit                                      */
+/*  Copyright (C) 2015--2025  SysDeer Technologies, LLC        */
+/*  Released under GPLv2 and GPLv3; see COPYING.PERK.          */
+/***************************************************************/
+
+#include <stdio.h>
+#include <unistd.h>
+#include <perk/perk.h>
+#include <perk/perk_output.h>
+#include "perk_driver_impl.h"
+#include "perk_dprintf_impl.h"
+
+static void pe_perform_hdrdump_actions(
+	const struct pe_driver_ctx *	dctx,
+	const struct pe_unit_ctx *	uctx)
+{
+	if (dctx->cctx->hdrdump & PERK_HDRDUMP_IMAGE_DOS_HEADER)
+		pe_hdrdump_image_dos_hdr(dctx,uctx->meta);
+
+	if (dctx->cctx->hdrdump & PERK_HDRDUMP_COFF_IMAGE_HEADER)
+		pe_hdrdump_coff_image_hdr(dctx,uctx->meta);
+
+	if (dctx->cctx->hdrdump & PERK_HDRDUMP_COFF_OBJECT_HEADER)
+		pe_hdrdump_coff_object_hdr(dctx,uctx->meta);
+
+	if (dctx->cctx->hdrdump & PERK_HDRDUMP_COFF_OPT_HEADER)
+		pe_hdrdump_coff_opt_hdr(dctx,uctx->meta);
+
+	if (dctx->cctx->hdrdump & PERK_HDRDUMP_SECTION_TABLE)
+		pe_hdrdump_sec_tbl(dctx,uctx->meta);
+
+	if (dctx->cctx->hdrdump & PERK_HDRDUMP_EXPORT_HEADER)
+		pe_hdrdump_export_hdr(dctx,uctx->meta);
+
+	if (dctx->cctx->hdrdump & PERK_HDRDUMP_IMPORT_TABLE)
+		pe_hdrdump_import_tbl(dctx,uctx->meta);
+}
+
+static void pe_perform_unit_actions(
+	const struct pe_driver_ctx *	dctx,
+	const struct pe_unit_ctx *	uctx)
+{
+	uint64_t flags = dctx->cctx->fmtflags;
+
+	if (flags & PERK_OUTPUT_IMAGE_CATEGORY)
+		pe_output_image_category(dctx,uctx->meta);
+
+	if (flags & PERK_OUTPUT_IMAGE_SECTIONS)
+		pe_output_image_sections(dctx,uctx->meta);
+
+	if (flags & PERK_OUTPUT_IMAGE_SYMBOLS)
+		pe_output_image_symbols(dctx,uctx->meta);
+
+	if (flags & PERK_OUTPUT_IMAGE_STRINGS)
+		pe_output_image_strings(dctx,uctx->meta);
+
+	if (flags & PERK_OUTPUT_EXPORT_SYMS)
+		pe_output_export_symbols(dctx,uctx->meta);
+
+	if ((flags & PERK_OUTPUT_IMPORT_LIBS) || (flags & PERK_OUTPUT_IMPORT_SYMS))
+		pe_output_import_libraries(dctx,uctx->meta);
+
+	if ((flags & PERK_OUTPUT_MDSO_LIBS) || (flags & PERK_OUTPUT_MDSO_SYMS))
+		pe_output_mdso_libraries(dctx,uctx->meta);
+
+	if (dctx->cctx->hdrdump)
+		pe_perform_hdrdump_actions(dctx,uctx);
+}
+
+int pe_cmd_perk(const struct pe_driver_ctx * dctx, const char * path)
+{
+	int                     ret;
+	struct pe_unit_ctx *	uctx;
+
+	if ((ret = pe_get_unit_ctx(dctx,path,&uctx)) < 0)
+		return ret;
+
+	pe_perform_unit_actions(dctx,uctx);
+	pe_free_unit_ctx(uctx);
+
+	return 0;
+}