qbit/go
1
0
Fork 0
mirror of https://github.com/golang/go synced 2024-11-22 02:34:40 -07:00
Code Releases Activity

8l: pe executable building code changed to include import table for kernel32.dll functions

Browse Source 
Fixes #586.

R=rsc
CC=golang-dev
https://golang.org/cl/203060
This commit is contained in:
Alex Brainman 2010-02-10 00:47:52 -08:00 committed by Russ Cox
parent 1ca39fb1fd
commit 3b1a71862e
2 changed files with 99 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
src/cmd/ld/pe.c
View File

@ -95,6 +95,78 @@ dope(void)
IMAGE_SCN_MEM_READ|IMAGE_SCN_MEM_WRITE; IMAGE_SCN_MEM_READ|IMAGE_SCN_MEM_WRITE;
} }
static void
strput(char *s)
{
while(*s)
cput(*s++);
cput('\0');
}
static void
add_import_table(void)
{
IMAGE_IMPORT_DESCRIPTOR ds[2], *d;
char *dllname = "kernel32.dll";
struct {
char *name;
uint32 thunk;
} *f, fs[] = {
{ "GetProcAddress", 0 },
{ "LoadLibraryExA", 0 },
{ 0, 0 }
};
uint32 size = 0;
memset(ds, 0, sizeof(ds));
size += sizeof(ds);
ds[0].Name = size;
size += strlen(dllname) + 1;
for(f=fs; f->name; f++) {
f->thunk = size;
size += sizeof(uint16) + strlen(f->name) + 1;
}
ds[0].FirstThunk = size;
for(f=fs; f->name; f++)
size += sizeof(fs[0].thunk);
IMAGE_SECTION_HEADER *isect;
isect = new_section(".idata", size, 0);
isect->Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA|
IMAGE_SCN_MEM_READ|IMAGE_SCN_MEM_WRITE;
uint32 va = isect->VirtualAddress;
oh.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress = va;
oh.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size = isect->VirtualSize;
ds[0].Name += va;
ds[0].FirstThunk += va;
for(f=fs; f->name; f++)
f->thunk += va;
vlong off = seek(cout, 0, 1);
seek(cout, 0, 2);
for(d=ds; ; d++) {
lputl(d->OriginalFirstThunk);
lputl(d->TimeDateStamp);
lputl(d->ForwarderChain);
lputl(d->Name);
lputl(d->FirstThunk);
if(!d->Name)
break;
}
strput(dllname);
for(f=fs; f->name; f++) {
wputl(0);
strput(f->name);
}
for(f=fs; f->name; f++)
lputl(f->thunk);
strnput("", isect->SizeOfRawData - size);
cflush();
seek(cout, off, 0);
}
void void
asmbpe(void) asmbpe(void)
{ {
@ -117,6 +189,8 @@ asmbpe(void)
IMAGE_SCN_CNT_INITIALIZED_DATA; IMAGE_SCN_CNT_INITIALIZED_DATA;
} }
add_import_table();
fh.NumberOfSections = nsect; fh.NumberOfSections = nsect;
fh.TimeDateStamp = time(0); fh.TimeDateStamp = time(0);
fh.SizeOfOptionalHeader = sizeof(oh); fh.SizeOfOptionalHeader = sizeof(oh);

25
src/cmd/ld/pe.h
View File

@ -64,6 +64,14 @@ typedef struct {
uint32 Characteristics; uint32 Characteristics;
} IMAGE_SECTION_HEADER; } IMAGE_SECTION_HEADER;
typedef struct {
uint32 OriginalFirstThunk;
uint32 TimeDateStamp;
uint32 ForwarderChain;
uint32 Name;
uint32 FirstThunk;
} IMAGE_IMPORT_DESCRIPTOR;
#define PERESERVE 0x400 #define PERESERVE 0x400
#define PEALIGN 0x200 #define PEALIGN 0x200
#define PEBASE 0x00400000 #define PEBASE 0x00400000
@ -84,6 +92,23 @@ enum {
IMAGE_SCN_MEM_EXECUTE = 0x20000000, IMAGE_SCN_MEM_EXECUTE = 0x20000000,
IMAGE_SCN_MEM_READ = 0x40000000, IMAGE_SCN_MEM_READ = 0x40000000,
IMAGE_SCN_MEM_WRITE = 0x80000000, IMAGE_SCN_MEM_WRITE = 0x80000000,
IMAGE_DIRECTORY_ENTRY_EXPORT = 0,
IMAGE_DIRECTORY_ENTRY_IMPORT = 1,
IMAGE_DIRECTORY_ENTRY_RESOURCE = 2,
IMAGE_DIRECTORY_ENTRY_EXCEPTION = 3,
IMAGE_DIRECTORY_ENTRY_SECURITY = 4,
IMAGE_DIRECTORY_ENTRY_BASERELOC = 5,
IMAGE_DIRECTORY_ENTRY_DEBUG = 6,
IMAGE_DIRECTORY_ENTRY_COPYRIGHT = 7,
IMAGE_DIRECTORY_ENTRY_ARCHITECTURE = 7,
IMAGE_DIRECTORY_ENTRY_GLOBALPTR = 8,
IMAGE_DIRECTORY_ENTRY_TLS = 9,
IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG = 10,
IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT = 11,
IMAGE_DIRECTORY_ENTRY_IAT = 12,
IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT = 13,
IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR = 14,
}; };
void peinit(void); void peinit(void);