/***************************************************************/
/*  perk: PE Resource Kit                                      */
/*  Copyright (C) 2015--2016  Z. Gilboa                        */
/*  Released under GPLv2 and GPLv3; see COPYING.PERK.          */
/***************************************************************/

#include <stdint.h>
#include <stdbool.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>

#include <perk/perk.h>
#include <perk/perk_output.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 bool pe_image_is_psxscl(const struct pe_unit_ctx * uctx)
{
	return (!uctx->meta->summary.nimplibs
		&& !pe_get_expsym_by_name(uctx->meta,"__psx_init",0));
}

static bool pe_image_is_cygwin(const struct pe_unit_ctx * uctx)
{
	int				i;
	const struct pe_image_meta *	meta = uctx->meta;

	for (i=0; i<uctx->meta->summary.nimplibs; i++)
		if (!(strcmp(meta->idata[i].name,"cygwin1.dll")))
			return true;

	return false;
}

static bool pe_image_is_msys(const struct pe_unit_ctx * uctx)
{
	int				i;
	const struct pe_image_meta *	meta = uctx->meta;

	for (i=0; i<uctx->meta->summary.nimplibs; i++)
		if (!(strcmp(meta->idata[i].name,"msys-2.0.dll")))
			return true;

	return false;
}

static bool pe_image_is_mingw(const struct pe_unit_ctx * uctx)
{
	return ((pe_get_named_section_index(uctx->meta,".CRT") >= 0)
		&& (pe_get_named_section_index(uctx->meta,".bss") >= 0)
		&& (pe_get_named_section_index(uctx->meta,".tls") >= 0));
}

static const char * pretty_framework(const struct pe_unit_ctx * uctx)
{
	if (pe_get_named_section_index(uctx->meta,".midipix") >= 0)
		return "midipix";

	else if (pe_get_named_section_index(uctx->meta,".freestd") >= 0)
		return "freestd";

	else if (pe_get_named_section_index(uctx->meta,".cygheap") >= 0)
		return "cygone";

	else if (pe_image_is_psxscl(uctx))
		return "psxscl";

	else if (pe_image_is_cygwin(uctx))
		return "cygwin";

	else if (pe_image_is_msys(uctx))
		return "msys";

	else if (pe_image_is_mingw(uctx))
		return "mingw";

	else if (uctx->meta->opt.img.subsystem == PE_IMAGE_SUBSYSTEM_POSIX_CUI)
		return "suacon";

	else if (uctx->meta->opt.img.subsystem == PE_IMAGE_SUBSYSTEM_WINDOWS_CUI)
		return "wincon";

	else if (uctx->meta->opt.img.subsystem == PE_IMAGE_SUBSYSTEM_WINDOWS_GUI)
		return "win32";

	else
		return "unknown";
}

int pe_output_image_type(
	const struct pe_driver_ctx *	dctx,
	const struct pe_unit_ctx *	uctx,
	FILE *				fout)
{
	if (!fout)
		fout = stdout;

	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 PERK_FILE_ERROR(dctx);
	} else {
		if (fprintf(fout,"%s-%s-%s-%s\n",
				pretty_abi(uctx),
				pretty_type(uctx),
				pretty_subsystem(uctx),
				pretty_framework(uctx)) < 0)
			return PERK_FILE_ERROR(dctx);
	}

	return 0;
}