Files
2024-04-03 21:57:41 +03:00

206 lines
7.4 KiB
C++

#include "pe/FileHdrWrapper.h"
#include "pe/PEFile.h"
#include <time.h>
#include <QDateTime>
namespace util {
QString getDateString(const quint64 timestamp)
{
const time_t rawtime = (const time_t)timestamp;
QString format = "dddd, dd.MM.yyyy hh:mm:ss";
#if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
QDateTime date1(QDateTime::fromSecsSinceEpoch(rawtime));
#else
QDateTime date1(QDateTime::fromTime_t(rawtime));
#endif
return date1.toUTC().toString(format) + " UTC";
}
};
std::map<DWORD, QString> FileHdrWrapper::s_fHdrCharact;
std::map<DWORD, QString> FileHdrWrapper::s_machine;
void FileHdrWrapper::initCharact()
{
if (s_fHdrCharact.size() != 0) {
return; //already initialized
}
s_fHdrCharact[F_RELOCS_STRIPPED] = "Relocation info stripped from file.";
s_fHdrCharact[F_EXECUTABLE_IMAGE] = "File is executable (i.e. no unresolved external references).";
s_fHdrCharact[F_LINE_NUMS_STRIPPED] = "Line numbers stripped from file.";
s_fHdrCharact[F_LOCAL_SYMS_STRIPPED] = "Local symbols stripped from file.";
s_fHdrCharact[F_AGGRESIVE_WS_TRIM] = "Aggressively trim working set";
s_fHdrCharact[F_LARGE_ADDRESS_AWARE] = "App can handle >2gb addresses";
s_fHdrCharact[F_BYTES_REVERSED_LO] = "Bytes of machine word are reversed.";
s_fHdrCharact[F_MACHINE_32BIT] = "32 bit word machine.";
s_fHdrCharact[F_DEBUG_STRIPPED] = "Debugging info stripped from file in .DBG file";
s_fHdrCharact[F_REMOVABLE_RUN_FROM_SWAP] = "If Image is on removable media, copy and run from the swap file.";
s_fHdrCharact[F_NET_RUN_FROM_SWAP] = "If Image is on Net, copy and run from the swap file.";
s_fHdrCharact[F_SYSTEM] = "System File.";
s_fHdrCharact[F_DLL] = "File is a DLL.";
s_fHdrCharact[F_UP_SYSTEM_ONLY] = "File should only be run on a UP machine";
s_fHdrCharact[F_BYTES_REVERSED_HI] = "Bytes of machine word are reversed.";
}
std::vector<DWORD> FileHdrWrapper::splitCharact(DWORD characteristics)
{
if (s_fHdrCharact.size() == 0) initCharact();
std::vector<DWORD> chSet;
for (std::map<DWORD, QString>::iterator iter = s_fHdrCharact.begin(); iter != s_fHdrCharact.end(); ++iter) {
if (characteristics & iter->first) {
chSet.push_back(iter->first);
}
}
return chSet;
}
QString FileHdrWrapper::translateCharacteristics(DWORD charact)
{
if (s_fHdrCharact.size() == 0)
initCharact();
if (s_fHdrCharact.find(charact) == s_fHdrCharact.end()) return "";
return s_fHdrCharact[charact];
}
void FileHdrWrapper::initMachine()
{
s_machine[M_UNKNOWN] = "s_machine unknown";
s_machine[M_I386] = "Intel 386";
s_machine[M_R3000] = "MIPS little-endian, 0x160 big-endian";
s_machine[M_R4000] = "MIPS little-endian";
s_machine[M_R10000] = "MIPS little-endian";
s_machine[M_WCEMIPSV2] = " MIPS little-endian WCE v2";
s_machine[M_ALPHA] = "Alpha_AXP";
s_machine[M_SH3] = "SH3 little-endian";
s_machine[M_SH3DSP] = "SH3DSP";
s_machine[M_SH3E] = "SH3E little-endian";
s_machine[M_SH4] = "SH4 little-endian";
s_machine[M_SH5] = "SH5";
s_machine[M_ARM] = "ARM Little-Endian";
s_machine[M_THUMB] = "Thumb";
s_machine[M_THUMB2] = "Thumb2";
s_machine[M_AM33] = "AM33";
s_machine[M_POWERPC] = "IBM PowerPC Little-Endian";
s_machine[M_POWERPCFP] = "PowerRPCFP";
s_machine[M_IA64] = "Intel 64";
s_machine[M_MIPS16] = "MIPS";
s_machine[M_ALPHA64] = "ALPHA64";
s_machine[M_MIPSFPU] = "MIPS";
s_machine[M_MIPSFPU16] = "MIPS";
s_machine[M_AXP64] = "M_ALPHA64";
s_machine[M_TRICORE] = " Infineon";
s_machine[M_CEF] = "CEF";
s_machine[M_EBC] = "EFI Byte Code";
s_machine[M_AMD64] = "AMD64 (K8)";
s_machine[M_M32R] = "M32R little-endian";
s_machine[M_CEE] = "CEE";
s_machine[M_RISCV32] = "RISC-V 32-bit Address Space";
s_machine[M_RISCV64] = "RISC-V 64-bit Address Space";
s_machine[M_RISCV128] = "RISC-V 128-bit Address Space";
s_machine[M_ARM64LE] = "ARM64 Little Endian";
s_machine[M_LOONGARCH32] = "LoongArch 32-bit Processor Family";
s_machine[M_LOONGARCH64] = "LoongArch 64-bit Processor Family";
s_machine[M_LINUXDOTNET64] = "AMD64 .Net For Linux";
s_machine[M_OSXDOTNET64] = "AMD64 .Net For Mac OS";
s_machine[M_FREEBSDDOTNET64] = "AMD64 .Net For Free BSD";
s_machine[M_NETBSDDOTNET64] = "AMD64 .Net For Net BSD";
s_machine[M_SUNDOTNET64] = "AMD64 .Net For Sun (Oracle Solaris)";
s_machine[M_LINUXDOTNET32] = "Intel 386 .Net For Linux";
s_machine[M_OSXDOTNET32] = "Intel 386 .Net For Mac OS";
s_machine[M_FREEBSDDOTNET32] = "Intel 386 .Net For Free BSD";
s_machine[M_NETBSDDOTNET32] = "Intel 386 .Net For Net BSD";
s_machine[M_SUNDOTNET32] = "Intel 386 .Net For Sun (Oracle Solaris)";
}
QString FileHdrWrapper::translateMachine(DWORD val)
{
if (s_machine.size() == 0)
initMachine();
if (s_machine.find(val) == s_machine.end()) return "";
return s_machine[val];
}
void* FileHdrWrapper::getPtr()
{
if (this->hdr) {
// use cached if exists
return (void*) hdr;
}
if (m_PE == NULL) return NULL;
offset_t myOff = m_PE->peFileHdrOffset();
IMAGE_FILE_HEADER* hdr = (IMAGE_FILE_HEADER*) m_Exe->getContentAt(myOff, sizeof(IMAGE_FILE_HEADER));
if (!hdr) return NULL; //error
return (void*) hdr;
}
void* FileHdrWrapper::getFieldPtr(size_t fieldId, size_t subField)
{
IMAGE_FILE_HEADER * hdr = reinterpret_cast<IMAGE_FILE_HEADER*>(getPtr());
if (!hdr) return NULL;
IMAGE_FILE_HEADER &fileHeader = (*hdr);
switch (fieldId) {
case MACHINE: return (void*) &fileHeader.Machine;
case SEC_NUM: return (void*) &fileHeader.NumberOfSections;
case TIMESTAMP: return (void*) &fileHeader.TimeDateStamp;
case SYMBOL_PTR: return (void*) &fileHeader.PointerToSymbolTable;
case SYMBOL_NUM: return (void*) &fileHeader.NumberOfSymbols;
case OPTHDR_SIZE: return (void*) &fileHeader.SizeOfOptionalHeader;
case CHARACT: return (void*) &fileHeader.Characteristics;
}
return (void*) &fileHeader;
}
QString FileHdrWrapper::getFieldName(size_t fieldId)
{
switch (fieldId) {
case MACHINE: return("Machine");
case SEC_NUM: return ("Sections Count");
case TIMESTAMP: {
PEFile* myPe = dynamic_cast<PEFile*>(this->m_Exe);
if (myPe && myPe->isReproBuild()) {
return "ReproChecksum";
}
return("Time Date Stamp");
}
case SYMBOL_PTR: return("Ptr to Symbol Table");
case SYMBOL_NUM: return("Num. of Symbols");
case OPTHDR_SIZE: return("Size of OptionalHeader");
case CHARACT: return("Characteristics");
}
return "";
}
Executable::addr_type FileHdrWrapper::containsAddrType(size_t fieldId, size_t subField)
{
return Executable::NOT_ADDR;
}
QString FileHdrWrapper::translateFieldContent(size_t fieldId)
{
IMAGE_FILE_HEADER * hdr = reinterpret_cast<IMAGE_FILE_HEADER*>(getPtr());
if (!hdr) return "";
IMAGE_FILE_HEADER &fileHeader = (*hdr);
switch (fieldId) {
case MACHINE: return FileHdrWrapper::translateMachine(fileHeader.Machine);
case TIMESTAMP: {
PEFile* myPe = dynamic_cast<PEFile*>(this->m_Exe);
if (myPe && myPe->isReproBuild()) {
return ""; // This is not a real timestamp
}
return util::getDateString(fileHeader.TimeDateStamp);
}
}
return "";
}