/***************************************************************/
/* perk: PE Resource Kit */
/* Copyright (C) 2015--2021 Z. Gilboa */
/* Released under GPLv2 and GPLv3; see COPYING.PERK. */
/***************************************************************/
#include <string.h>
#include <perk/perk.h>
#include <perk/perk_consts.h>
#include <perk/perk_structs.h>
#include "perk_endian_impl.h"
#include "perk_reader_impl.h"
int pe_read_coff_header(const struct pe_raw_coff_image_hdr * p, struct pe_meta_coff_file_hdr * m)
{
if ((p->cfh_signature[0] != 'P') || (p->cfh_signature[1] != 'E')
|| p->cfh_signature[2] || p->cfh_signature[3])
return PERK_ERR_BAD_COFF_HEADER;
m->cfh_signature[0] = p->cfh_signature[0];
m->cfh_signature[1] = p->cfh_signature[1];
m->cfh_signature[2] = p->cfh_signature[2];
m->cfh_signature[3] = p->cfh_signature[3];
m->cfh_machine = pe_read_short(p->cfh_machine);
m->cfh_num_of_sections = pe_read_short(p->cfh_num_of_sections);
m->cfh_time_date_stamp = pe_read_long(p->cfh_time_date_stamp);
m->cfh_ptr_to_sym_tbl = pe_read_long(p->cfh_ptr_to_sym_tbl);
m->cfh_num_of_syms = pe_read_long(p->cfh_num_of_syms);
m->cfh_size_of_opt_hdr = pe_read_short(p->cfh_size_of_opt_hdr);
m->cfh_characteristics = pe_read_short(p->cfh_characteristics);
return 0;
}
#define PE_IMAGE_FILE_CHARACTERISTICS (\
PE_IMAGE_FILE_RELOCS_STRIPPED \
| PE_IMAGE_FILE_EXECUTABLE_IMAGE \
| PE_IMAGE_FILE_LINE_NUMS_STRIPPED \
| PE_IMAGE_FILE_LOCAL_SYMS_STRIPPED \
| PE_IMAGE_FILE_AGGRESSIVE_WS_TRIM \
| PE_IMAGE_FILE_LARGE_ADDRESS_AWARE \
| PE_IMAGE_FILE_RESERVED_CHARACTERISTIC \
| PE_IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP \
| PE_IMAGE_FILE_BYTES_REVERSED_LO \
| PE_IMAGE_FILE_32BIT_MACHINE \
| PE_IMAGE_FILE_DEBUG_STRIPPED \
| PE_IMAGE_FILE_NET_RUN_FROM_SWAP \
| PE_IMAGE_FILE_SYSTEM \
| PE_IMAGE_FILE_DLL \
| PE_IMAGE_FILE_UP_SYSTEM_ONLY \
| PE_IMAGE_FILE_BYTES_REVERSED_HI)
int pe_read_object_header(const struct pe_raw_coff_object_hdr * p, struct pe_meta_coff_file_hdr * m)
{
if (pe_read_short(p->cfh_size_of_opt_hdr))
return PERK_ERR_BAD_COFF_HEADER;
if (pe_read_short(p->cfh_characteristics) & ~PE_IMAGE_FILE_CHARACTERISTICS)
return PERK_ERR_BAD_COFF_HEADER;
switch (pe_read_short(p->cfh_machine)) {
case PE_IMAGE_FILE_MACHINE_I386:
case PE_IMAGE_FILE_MACHINE_R4000:
case PE_IMAGE_FILE_MACHINE_WCEMIPSV2:
case PE_IMAGE_FILE_MACHINE_SH3:
case PE_IMAGE_FILE_MACHINE_SH3DSP:
case PE_IMAGE_FILE_MACHINE_SH4:
case PE_IMAGE_FILE_MACHINE_SH5:
case PE_IMAGE_FILE_MACHINE_ARM:
case PE_IMAGE_FILE_MACHINE_THUMB:
case PE_IMAGE_FILE_MACHINE_ARMV7:
case PE_IMAGE_FILE_MACHINE_AM33:
case PE_IMAGE_FILE_MACHINE_POWERPC:
case PE_IMAGE_FILE_MACHINE_POWERPCFP:
case PE_IMAGE_FILE_MACHINE_IA64:
case PE_IMAGE_FILE_MACHINE_MIPS16:
case PE_IMAGE_FILE_MACHINE_MIPSFPU:
case PE_IMAGE_FILE_MACHINE_MIPSFPU16:
case PE_IMAGE_FILE_MACHINE_EBC:
case PE_IMAGE_FILE_MACHINE_AMD64:
case PE_IMAGE_FILE_MACHINE_M32R:
break;
default:
return PERK_ERR_BAD_COFF_HEADER;
}
m->cfh_signature[0] = 0;
m->cfh_signature[1] = 0;
m->cfh_signature[2] = 0;
m->cfh_signature[3] = 0;
m->cfh_machine = pe_read_short(p->cfh_machine);
m->cfh_num_of_sections = pe_read_short(p->cfh_num_of_sections);
m->cfh_time_date_stamp = pe_read_long(p->cfh_time_date_stamp);
m->cfh_ptr_to_sym_tbl = pe_read_long(p->cfh_ptr_to_sym_tbl);
m->cfh_num_of_syms = pe_read_long(p->cfh_num_of_syms);
m->cfh_size_of_opt_hdr = pe_read_short(p->cfh_size_of_opt_hdr);
m->cfh_characteristics = pe_read_short(p->cfh_characteristics);
return 0;
}