jory/upx
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

src: clang-format stub/tools/armpe/armpe_tester.c

Browse Source
This commit is contained in:
Markus F.X.J. Oberhumer
2023-10-30 13:07:20 +01:00
parent fadf7cfc7b
commit 19bf9f59e9
2 changed files with 110 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Makefile
+1
View File
@@ -96,6 +96,7 @@ ifeq ($(shell uname),Linux)
CLANG_FORMAT_EXCLUDE_FILES += miniacc.h stub/%.h CLANG_FORMAT_EXCLUDE_FILES += miniacc.h stub/%.h
CLANG_FORMAT_EXCLUDE_FILES += p_elf.h p_elf_enum.h p_lx_% p_mach% p_unix% p_vmlin% CLANG_FORMAT_EXCLUDE_FILES += p_elf.h p_elf_enum.h p_lx_% p_mach% p_unix% p_vmlin%
CLANG_FORMAT_FILES := $(sort $(wildcard *.[ch]* ../maint/src/*.[ch]* */*.[ch]*)) CLANG_FORMAT_FILES := $(sort $(wildcard *.[ch]* ../maint/src/*.[ch]* */*.[ch]*))
CLANG_FORMAT_FILES += stub/tools/armpe/armpe_tester.c
CLANG_FORMAT_FILES := $(filter-out $(CLANG_FORMAT_EXCLUDE_FILES),$(CLANG_FORMAT_FILES)) CLANG_FORMAT_FILES := $(filter-out $(CLANG_FORMAT_EXCLUDE_FILES),$(CLANG_FORMAT_FILES))
clang-format: PHONY $(CLANG_FORMAT_FILES) clang-format: PHONY $(CLANG_FORMAT_FILES)
@echo "running upx-clang-format" @echo "running upx-clang-format"
src/stub/tools/armpe/armpe_tester.c
+109 -145
View File
@@ -42,7 +42,6 @@
// arm-wince-pe-gcc -Wl,--image-base,0x400000 // arm-wince-pe-gcc -Wl,--image-base,0x400000
#include <stddef.h> #include <stddef.h>
#include <stdarg.h> #include <stdarg.h>
#include <string.h> #include <string.h>
@@ -50,89 +49,83 @@
#include <errno.h> #include <errno.h>
#ifdef __i386__ #ifdef __i386__
# define UPX_MMAP_ADDRESS 0x20000000 #define UPX_MMAP_ADDRESS 0x20000000
#else #else
# define UPX_MMAP_ADDRESS 0x410000 // 0x10000 #define UPX_MMAP_ADDRESS 0x410000 // 0x10000
#endif #endif
#ifdef __linux__ #ifdef __linux__
# include <sys/mman.h> #include <sys/mman.h>
#else #else
void *VirtualAlloc(void *address, unsigned size, unsigned type, unsigned protect); void *VirtualAlloc(void *address, unsigned size, unsigned type, unsigned protect);
# define MEM_COMMIT 0x1000 #define MEM_COMMIT 0x1000
# define PAGE_EXECUTE_READWRITE 0x0040 #define PAGE_EXECUTE_READWRITE 0x0040
#endif #endif
typedef size_t upx_uintptr_t; typedef unsigned short LE16;
typedef unsigned short LE16; typedef unsigned int LE32;
typedef unsigned int LE32; #define get_le32(p) (*(const unsigned *) (p))
#define get_le32(p) (* (const unsigned *) (p)) #define set_le32(p, v) (*(unsigned *) (p) = (v))
#define set_le32(p,v) (* (unsigned *) (p) = (v)) #define get_le16(p) (*(const unsigned short *) (p))
#define get_le16(p) (* (const unsigned short *) (p))
#if !defined(__packed_struct) #if !defined(packed_struct)
# define __packed_struct(s) struct s { #define packed_struct(s) struct s
# define __packed_struct_end() };
#endif #endif
packed_struct(ddirs_t) {
LE32 vaddr;
LE32 size;
};
__packed_struct(ddirs_t) packed_struct(pe_header_t) {
LE32 vaddr;
LE32 size;
__packed_struct_end()
__packed_struct(pe_header_t)
// 0x0 // 0x0
char _[4]; char _[4];
LE16 cpu; LE16 cpu;
LE16 objects; LE16 objects;
char __[12]; char __[12];
LE16 opthdrsize; LE16 opthdrsize;
LE16 flags; LE16 flags;
// optional header // optional header
char ___[4]; char ___[4];
LE32 codesize; LE32 codesize;
// 0x20 // 0x20
LE32 datasize; LE32 datasize;
LE32 bsssize; LE32 bsssize;
LE32 entry; LE32 entry;
LE32 codebase; LE32 codebase;
// 0x30 // 0x30
LE32 database; LE32 database;
// nt specific fields // nt specific fields
LE32 imagebase; LE32 imagebase;
LE32 objectalign; LE32 objectalign;
LE32 filealign; LE32 filealign;
// 0x40 // 0x40
char ____[16]; char ____[16];
// 0x50 // 0x50
LE32 imagesize; LE32 imagesize;
LE32 headersize; LE32 headersize;
LE32 chksum; LE32 chksum;
LE16 subsystem; LE16 subsystem;
LE16 dllflags; LE16 dllflags;
// 0x60 // 0x60
char _____[20]; char _____[20];
// 0x74 // 0x74
LE32 ddirsentries; LE32 ddirsentries;
// //
struct ddirs_t ddirs[16]; struct ddirs_t ddirs[16];
__packed_struct_end() };
packed_struct(pe_section_t) {
char name[8];
LE32 vsize;
LE32 vaddr;
LE32 size;
LE32 rawdataptr;
char _[12];
LE32 flags;
};
__packed_struct(pe_section_t) packed_struct(exe_header_t) {
char name[8];
LE32 vsize;
LE32 vaddr;
LE32 size;
LE32 rawdataptr;
char _[12];
LE32 flags;
__packed_struct_end()
__packed_struct(exe_header_t)
LE16 mz; LE16 mz;
LE16 m512; LE16 m512;
LE16 p512; LE16 p512;
@@ -140,27 +133,25 @@ __packed_struct(exe_header_t)
LE16 relocoffs; LE16 relocoffs;
char __[34]; char __[34];
LE32 nexepos; LE32 nexepos;
__packed_struct_end()
enum {
PEDIR_EXPORT = 0,
PEDIR_IMPORT = 1,
PEDIR_RESOURCE = 2,
PEDIR_EXCEPTION = 3,
PEDIR_SEC = 4,
PEDIR_RELOC = 5,
PEDIR_DEBUG = 6,
PEDIR_COPYRIGHT = 7,
PEDIR_GLOBALPTR = 8,
PEDIR_TLS = 9,
PEDIR_LOADCONF = 10,
PEDIR_BOUNDIM = 11,
PEDIR_IAT = 12,
PEDIR_DELAYIMP = 13,
PEDIR_COMRT = 14
}; };
enum {
PEDIR_EXPORT = 0,
PEDIR_IMPORT = 1,
PEDIR_RESOURCE = 2,
PEDIR_EXCEPTION = 3, // Exception table
PEDIR_SECURITY = 4, // Certificate table (file pointer)
PEDIR_BASERELOC = 5,
PEDIR_DEBUG = 6,
PEDIR_ARCHITECTURE = 7, // Architecture-specific data
PEDIR_GLOBALPTR = 8, // Global pointer
PEDIR_TLS = 9,
PEDIR_LOAD_CONFIG = 10, // Load Config Table
PEDIR_BOUND_IMPORT = 11,
PEDIR_IAT = 12,
PEDIR_DELAY_IMPORT = 13, // Delay Import Descriptor
PEDIR_COM_DESCRIPTOR = 14 // Com+ Runtime Header
};
static struct pe_header_t ih; static struct pe_header_t ih;
static struct pe_section_t isections[4]; static struct pe_section_t isections[4];
@@ -184,8 +175,7 @@ static int print(const char *format, ...)
#define print printf #define print printf
#endif #endif
static int load(const char *file) static int load(const char *file) {
{
struct exe_header_t h; struct exe_header_t h;
int ic; int ic;
unsigned pe_offset = 0; unsigned pe_offset = 0;
@@ -193,20 +183,17 @@ static int load(const char *file)
if ((f = fopen(file, "rb")) == NULL) if ((f = fopen(file, "rb")) == NULL)
return print("can not open file: %s\n", file); return print("can not open file: %s\n", file);
for (ic = 0; ic < 20; ic++) for (ic = 0; ic < 20; ic++) {
{ if (fseek(f, pe_offset, SEEK_SET) || fread(&h, sizeof(h), 1, f) != 1)
if (fseek(f, pe_offset, SEEK_SET)
|| fread(&h, sizeof(h), 1, f) != 1)
return print("read error at %u\n", pe_offset); return print("read error at %u\n", pe_offset);
if (h.mz == 'M' + 'Z'*256) // dos exe if (h.mz == 'M' + 'Z' * 256) // dos exe
{ {
if (h.relocoffs >= 0x40) // new format exe if (h.relocoffs >= 0x40) // new format exe
pe_offset += h.nexepos; pe_offset += h.nexepos;
else else
pe_offset += h.p512 * 512 + h.m512 - h.m512 ? 512 : 0; pe_offset += h.p512 * 512 + h.m512 - h.m512 ? 512 : 0;
} } else if (get_le32(&h) == 'P' + 'E' * 256)
else if (get_le32(&h) == 'P' + 'E'*256)
break; break;
else else
return print("bad header at %u\n", pe_offset); return print("bad header at %u\n", pe_offset);
@@ -214,47 +201,40 @@ static int load(const char *file)
if (ic == 20) if (ic == 20)
return print("pe header not found\n"); return print("pe header not found\n");
printf("pe header found at offset: %u\n", pe_offset); printf("pe header found at offset: %u\n", pe_offset);
if (fseek(f, pe_offset, SEEK_SET) if (fseek(f, pe_offset, SEEK_SET) || fread(&ih, sizeof(ih), 1, f) != 1)
|| fread(&ih, sizeof(ih), 1, f) != 1)
return print("can not load pe header\n"); return print("can not load pe header\n");
print("ih.imagesize=0x%x\n", ih.imagesize); print("ih.imagesize=0x%x\n", ih.imagesize);
if (ih.cpu != 0x1c0 && ih.cpu != 0x1c2) if (ih.cpu != 0x1c0 && ih.cpu != 0x1c2)
return print("unsupported processor type: %x\n", ih.cpu); return print("unsupported processor type: %x\n", ih.cpu);
if ((ih.objects != 3 && ih.objects != 4) if ((ih.objects != 3 && ih.objects != 4) || fread(isections, sizeof(isections), 1, f) != 1)
|| fread(isections, sizeof(isections), 1, f) != 1)
return print("error reading section descriptors\n"); return print("error reading section descriptors\n");
return 0; return 0;
} }
static int read(void) static int read(void) {
{
unsigned ic; unsigned ic;
#ifdef __linux__ #ifdef __linux__
vaddr = mmap((void *) UPX_MMAP_ADDRESS, ih.imagesize, vaddr = mmap((void *) UPX_MMAP_ADDRESS, ih.imagesize, PROT_WRITE | PROT_READ | PROT_EXEC,
PROT_WRITE | PROT_READ | PROT_EXEC,
MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED, -1, 0); MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED, -1, 0);
if (((int) vaddr) == -1) if (((int) vaddr) == -1)
return print("mmap() failed: %d\n", errno); return print("mmap() failed: %d\n", errno);
print("mmap for %p (size %x) successful\n", vaddr, ih.imagesize); print("mmap for %p (size %x) successful\n", vaddr, ih.imagesize);
#else #else
if ((vaddr = VirtualAlloc(0, ih.imagesize, if ((vaddr = VirtualAlloc(0, ih.imagesize, MEM_COMMIT, PAGE_EXECUTE_READWRITE)) == 0)
MEM_COMMIT, PAGE_EXECUTE_READWRITE)) == 0)
return print("VirtualAlloc() failed\n"); return print("VirtualAlloc() failed\n");
print("VirtualAlloc() ok %x\n", vaddr); print("VirtualAlloc() ok %x\n", vaddr);
#endif #endif
for (ic = 1; ic <= (unsigned) ih.objects - 1; ic++) for (ic = 1; ic <= (unsigned) ih.objects - 1; ic++)
if (fseek(f, isections[ic].rawdataptr, SEEK_SET) if (fseek(f, isections[ic].rawdataptr, SEEK_SET) ||
|| fread(vaddr + isections[ic].vaddr, fread(vaddr + isections[ic].vaddr, isections[ic].vsize, 1, f) != 1)
isections[ic].vsize, 1, f) != 1)
return print("error reading section %u\n", ic); return print("error reading section %u\n", ic);
return 0; return 0;
} }
static void dump(char n) static void dump(char n) {
{
char buf[100]; char buf[100];
#ifdef __linux__ #ifdef __linux__
snprintf(buf, sizeof(buf), "a.dump%c", n); snprintf(buf, sizeof(buf), "a.dump%c", n);
@@ -266,49 +246,37 @@ static void dump(char n)
fclose(f2); fclose(f2);
} }
static int loadlibraryw(const unsigned short *name) static int loadlibraryw(const unsigned short *name) {
{
return name[0] + name[1] * 0x100 + name[2] * 0x10000; return name[0] + name[1] * 0x100 + name[2] * 0x10000;
} }
static int getprocaddressa(unsigned h, const char *proc) static int getprocaddressa(unsigned h, const char *proc) {
{
unsigned p = (unsigned) proc; unsigned p = (unsigned) proc;
if (p < 0x10000) if (p < 0x10000) {
{ print("getprocaddressa called %c%c%c, ordinal %u\n", h, h >> 8, h >> 16, p);
print("getprocaddressa called %c%c%c, ordinal %u\n",
h, h >> 8, h >> 16, p);
return h + p * 0x10000; return h + p * 0x10000;
} }
print("getprocaddressa called %c%c%c, name %s\n", print("getprocaddressa called %c%c%c, name %s\n", h, h >> 8, h >> 16, proc);
h, h >> 8, h >> 16, proc);
return h + proc[0] * 0x10000 + proc[1] * 0x1000000; return h + proc[0] * 0x10000 + proc[1] * 0x1000000;
} }
static void cachesync(unsigned v) static void cachesync(unsigned v) { print("cachesync called %u\n", v); }
{
print("cachesync called %u\n", v);
}
static int import(void) static int import(void) {
{
if (ih.ddirs[PEDIR_IMPORT].vaddr == 0) if (ih.ddirs[PEDIR_IMPORT].vaddr == 0)
return print("no imports?\n"); return print("no imports?\n");
print("loadlibraryw=%p,getprocaddressa=%p,cachesync=%p\n", print("loadlibraryw=%p,getprocaddressa=%p,cachesync=%p\n", loadlibraryw, getprocaddressa,
loadlibraryw, getprocaddressa, cachesync); cachesync);
void *imports = vaddr + ih.ddirs[PEDIR_IMPORT].vaddr; void *imports = vaddr + ih.ddirs[PEDIR_IMPORT].vaddr;
while (get_le32(imports + 12)) while (get_le32(imports + 12)) {
{ if (strcasecmp(vaddr + get_le32(imports + 12), "coredll.dll") == 0) {
if (strcasecmp(vaddr + get_le32(imports + 12), "coredll.dll") == 0)
{
void *coredll_imports = vaddr + get_le32(imports + 16); void *coredll_imports = vaddr + get_le32(imports + 16);
print("coredll_imports=%p\n", coredll_imports); print("coredll_imports=%p\n", coredll_imports);
void *oft = vaddr + get_le32(imports); void *oft = vaddr + get_le32(imports);
unsigned pos = 0; unsigned pos = 0;
while (get_le32(oft + pos)) while (get_le32(oft + pos)) {
{
void *name = vaddr + get_le32(oft + pos) + 2; void *name = vaddr + get_le32(oft + pos) + 2;
print("name=%s\n", (char*) name); print("name=%s\n", (char *) name);
if (strcasecmp(name, "loadlibraryw") == 0) if (strcasecmp(name, "loadlibraryw") == 0)
set_le32(coredll_imports + pos, (unsigned) loadlibraryw); set_le32(coredll_imports + pos, (unsigned) loadlibraryw);
else if (strcasecmp(name, "getprocaddressa") == 0) else if (strcasecmp(name, "getprocaddressa") == 0)
@@ -326,18 +294,16 @@ static int import(void)
return 1; return 1;
} }
static int reloc(void) static int reloc(void) {
{ if (ih.ddirs[PEDIR_BASERELOC].vaddr == 0)
if (ih.ddirs[PEDIR_RELOC].vaddr == 0)
return 0; return 0;
void *relocs = vaddr + ih.ddirs[PEDIR_RELOC].vaddr; void *relocs = vaddr + ih.ddirs[PEDIR_BASERELOC].vaddr;
void *page = vaddr + get_le32(relocs); void *page = vaddr + get_le32(relocs);
unsigned size = get_le32(relocs + 4); unsigned size = get_le32(relocs + 4);
if (size != ih.ddirs[PEDIR_RELOC].size) if (size != ih.ddirs[PEDIR_BASERELOC].size)
return print("only 1 page can be relocated\n"); return print("only 1 page can be relocated\n");
unsigned num = (size - 8) / 2; unsigned num = (size - 8) / 2;
while (num--) while (num--) {
{
unsigned pos = get_le16(relocs + 8 + num * 2); unsigned pos = get_le16(relocs + 8 + num * 2);
if (pos == 0) if (pos == 0)
continue; continue;
@@ -350,14 +316,14 @@ static int reloc(void)
return 0; return 0;
} }
static void dump2(int c) #ifndef __i386__
{ static void dump2(int c) {
print("dump2 %c\n", c); print("dump2 %c\n", c);
dump(c); dump(c);
} }
#endif
static void call(void) static void call(void) {
{
#ifndef __i386__ #ifndef __i386__
void (*entry)(void (*)(int), unsigned) = vaddr + ih.entry; void (*entry)(void (*)(int), unsigned) = vaddr + ih.entry;
entry(dump2, 1); entry(dump2, 1);
@@ -365,8 +331,7 @@ static void call(void)
#endif #endif
} }
static int main2(int argc, char **argv) static int main2(int argc, char **argv) {
{
if (argc != 2) if (argc != 2)
return print("usage: %s arm_pe_file\n", argv[0]), 1; return print("usage: %s arm_pe_file\n", argv[0]), 1;
if (load(argv[1])) if (load(argv[1]))
@@ -386,8 +351,7 @@ static int main2(int argc, char **argv)
return 0; return 0;
} }
int main(int argc, char **argv) int main(int argc, char **argv) {
{
out = stdout; out = stdout;
#ifndef __linux__ #ifndef __linux__
out = fopen("/wtest.log", "wt"); out = fopen("/wtest.log", "wt");