Blame src/output/pe_output_image_type.c

007c7f
/***************************************************************/
007c7f
/*  perk: PE Resource Kit                                      */
007c7f
/*  Copyright (C) 2015--2016  Z. Gilboa                        */
007c7f
/*  Released under GPLv2 and GPLv3; see COPYING.PERK.          */
007c7f
/***************************************************************/
007c7f
007c7f
#include <stdint.h>
007c7f
#include <stdlib.h>
007c7f
#include <stdio.h>
007c7f
#include <string.h>
007c7f
#include <errno.h>
007c7f
007c7f
#include <perk/perk.h>
007c7f
#include <perk/perk_output.h>
007c7f
#include "perk_output_impl.h"
007c7f
#include "perk_errinfo_impl.h"
007c7f
007c7f
static const char const * pe_subsystem_name[0x10] = {
007c7f
	[PE_IMAGE_SUBSYSTEM_UNKNOWN]                  = "unknown",
007c7f
	[PE_IMAGE_SUBSYSTEM_NATIVE]                   = "native",
007c7f
	[PE_IMAGE_SUBSYSTEM_WINDOWS_GUI]              = "windows",
007c7f
	[PE_IMAGE_SUBSYSTEM_WINDOWS_CUI]              = "console",
007c7f
	[PE_IMAGE_SUBSYSTEM_POSIX_CUI]                = "posix",
007c7f
	[PE_IMAGE_SUBSYSTEM_WINDOWS_CE_GUI]           = "wince",
007c7f
	[PE_IMAGE_SUBSYSTEM_EFI_APPLICATION]          = "efi_app",
007c7f
	[PE_IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER]  = "efi_driver",
007c7f
	[PE_IMAGE_SUBSYSTEM_EFI_ROM]                  = "efi_rom",
007c7f
	[PE_IMAGE_SUBSYSTEM_XBOX]                     = "xbox"
007c7f
};
007c7f
007c7f
static const char * pretty_abi(const struct pe_unit_ctx * uctx)
007c7f
{
007c7f
	switch (uctx->meta->opt.std.magic) {
007c7f
		case PE_MAGIC_PE32:
007c7f
			return "PE32";
007c7f
007c7f
		case PE_MAGIC_PE32_PLUS:
007c7f
			return "PE64";
007c7f
007c7f
		default:
007c7f
			return "INTERNAL_ERROR";
007c7f
	}
007c7f
}
007c7f
007c7f
static const char * pretty_type(const struct pe_unit_ctx * uctx)
007c7f
{
007c7f
	if (uctx->meta->coff.characteristics & PE_IMAGE_FILE_DLL)
007c7f
		return "dll";
007c7f
	else
007c7f
		return "exe";
007c7f
}
007c7f
007c7f
static const char * pretty_subsystem(const struct pe_unit_ctx * uctx)
007c7f
{
007c7f
	if (uctx->meta->opt.img.subsystem >= 0x10)
007c7f
		return "UNEXPECTED_SUBSYSTEM";
007c7f
007c7f
	else if (!pe_subsystem_name[uctx->meta->opt.img.subsystem])
007c7f
		return "UNEXPECTED_SUBSYSTEM";
007c7f
007c7f
	else
007c7f
		return pe_subsystem_name[uctx->meta->opt.img.subsystem];
007c7f
}
007c7f
007c7f
static const char * pretty_framework(const struct pe_unit_ctx * uctx)
007c7f
{
007c7f
	if (pe_get_named_section_index(uctx->meta,".midipix") >= 0)
007c7f
		return "midipix";
a5cc53
a5cc53
	else if (pe_get_named_section_index(uctx->meta,".freestd") >= 0)
a5cc53
		return "freestd";
a5cc53
007c7f
	else
007c7f
		return "win32";
007c7f
}
007c7f
007c7f
int pe_output_image_type(
007c7f
	const struct pe_driver_ctx *	dctx,
007c7f
	const struct pe_unit_ctx *	uctx,
007c7f
	FILE *				fout)
007c7f
{
007c7f
	FILE *				ftmp;
007c7f
	const struct pe_common_ctx *	cctx = dctx->cctx;
007c7f
007c7f
	if (!(fout = pe_output_prolog(cctx,fout,&ftmp)))
007c7f
		return PERK_SYSTEM_ERROR(dctx);
007c7f
007c7f
	if (dctx->cctx->fmtflags & PERK_PRETTY_YAML) {
007c7f
		if (fprintf(fout,"%s:\n- %s:\n- %s:\n- %s:\n- %s:\n",
007c7f
				*uctx->path,
007c7f
				pretty_abi(uctx),
007c7f
				pretty_type(uctx),
007c7f
				pretty_subsystem(uctx),
007c7f
				pretty_framework(uctx)) < 0)
007c7f
			return pe_output_epilog(
007c7f
				PERK_FILE_ERROR(dctx),
007c7f
				ftmp);
007c7f
	} else {
007c7f
		if (fprintf(fout,"%s-%s-%s-%s\n",
007c7f
				pretty_abi(uctx),
007c7f
				pretty_type(uctx),
007c7f
				pretty_subsystem(uctx),
007c7f
				pretty_framework(uctx)) < 0)
007c7f
			return pe_output_epilog(
007c7f
				PERK_FILE_ERROR(dctx),
007c7f
				ftmp);
007c7f
	}
007c7f
007c7f
	return pe_output_epilog(0,ftmp);
007c7f
}