diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/driver/pe_amain.c | 3 | ||||
| -rw-r--r-- | src/output/pe_output_image_type.c | 106 |
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); +} |