output: pe_output_image_type(): implementation and integration.

2 files changed, 109 insertions, 0 deletions

diff --git a/src/driver/pe_amain.c b/src/driver/pe_amain.c
index 42ec2d1..1ded100 100644
--- a/src/driver/pe_amain.c
+++ b/src/driver/pe_amain.c
@@ -66,6 +66,9 @@ static void pe_perform_unit_actions(
 	int      fpara = 0;
 	uint64_t flags = dctx->cctx->fmtflags;
 
+	if (flags & PERK_OUTPUT_IMAGE_TYPE)
+		pe_output_image_type(dctx,uctx,0);
+
 	if (flags & PERK_OUTPUT_EXPORT_SYMS) {
 		pe_output_export_symbols(dctx,uctx->meta,0);
 		fpara += uctx->meta->summary.nexpsyms;
diff --git a/src/output/pe_output_image_type.c b/src/output/pe_output_image_type.c
new file mode 100644
index 0000000..3d36ce5
--- /dev/null
+++ b/src/output/pe_output_image_type.c
@@ -0,0 +1,106 @@
+/***************************************************************/
+/*  perk: PE Resource Kit                                      */
+/*  Copyright (C) 2015--2016  Z. Gilboa                        */
+/*  Released under GPLv2 and GPLv3; see COPYING.PERK.          */
+/***************************************************************/
+
+#include <stdint.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <errno.h>
+
+#include <perk/perk.h>
+#include <perk/perk_output.h>
+#include "perk_output_impl.h"
+#include "perk_errinfo_impl.h"
+
+static const char const * pe_subsystem_name[0x10] = {
+	[PE_IMAGE_SUBSYSTEM_UNKNOWN]                  = "unknown",
+	[PE_IMAGE_SUBSYSTEM_NATIVE]                   = "native",
+	[PE_IMAGE_SUBSYSTEM_WINDOWS_GUI]              = "windows",
+	[PE_IMAGE_SUBSYSTEM_WINDOWS_CUI]              = "console",
+	[PE_IMAGE_SUBSYSTEM_POSIX_CUI]                = "posix",
+	[PE_IMAGE_SUBSYSTEM_WINDOWS_CE_GUI]           = "wince",
+	[PE_IMAGE_SUBSYSTEM_EFI_APPLICATION]          = "efi_app",
+	[PE_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER]  = "efi_driver",
+	[PE_IMAGE_SUBSYSTEM_EFI_ROM]                  = "efi_rom",
+	[PE_IMAGE_SUBSYSTEM_XBOX]                     = "xbox"
+};
+
+static const char * pretty_abi(const struct pe_unit_ctx * uctx)
+{
+	switch (uctx->meta->opt.std.magic) {
+		case PE_MAGIC_PE32:
+			return "PE32";
+
+		case PE_MAGIC_PE32_PLUS:
+			return "PE64";
+
+		default:
+			return "INTERNAL_ERROR";
+	}
+}
+
+static const char * pretty_type(const struct pe_unit_ctx * uctx)
+{
+	if (uctx->meta->coff.characteristics & PE_IMAGE_FILE_DLL)
+		return "dll";
+	else
+		return "exe";
+}
+
+static const char * pretty_subsystem(const struct pe_unit_ctx * uctx)
+{
+	if (uctx->meta->opt.img.subsystem >= 0x10)
+		return "UNEXPECTED_SUBSYSTEM";
+
+	else if (!pe_subsystem_name[uctx->meta->opt.img.subsystem])
+		return "UNEXPECTED_SUBSYSTEM";
+
+	else
+		return pe_subsystem_name[uctx->meta->opt.img.subsystem];
+}
+
+static const char * pretty_framework(const struct pe_unit_ctx * uctx)
+{
+	if (pe_get_named_section_index(uctx->meta,".midipix") >= 0)
+		return "midipix";
+	else
+		return "win32";
+}
+
+int pe_output_image_type(
+	const struct pe_driver_ctx *	dctx,
+	const struct pe_unit_ctx *	uctx,
+	FILE *				fout)
+{
+	FILE *				ftmp;
+	const struct pe_common_ctx *	cctx = dctx->cctx;
+
+	if (!(fout = pe_output_prolog(cctx,fout,&ftmp)))
+		return PERK_SYSTEM_ERROR(dctx);
+
+	if (dctx->cctx->fmtflags & PERK_PRETTY_YAML) {
+		if (fprintf(fout,"%s:\n- %s:\n- %s:\n- %s:\n- %s:\n",
+				*uctx->path,
+				pretty_abi(uctx),
+				pretty_type(uctx),
+				pretty_subsystem(uctx),
+				pretty_framework(uctx)) < 0)
+			return pe_output_epilog(
+				PERK_FILE_ERROR(dctx),
+				ftmp);
+	} else {
+		if (fprintf(fout,"%s-%s-%s-%s\n",
+				pretty_abi(uctx),
+				pretty_type(uctx),
+				pretty_subsystem(uctx),
+				pretty_framework(uctx)) < 0)
+			return pe_output_epilog(
+				PERK_FILE_ERROR(dctx),
+				ftmp);
+	}
+
+	return pe_output_epilog(0,ftmp);
+}