mirror of
https://github.com/frida/tinycc
synced 2024-12-28 23:59:41 +03:00
win64: update tiny_impdef, tiny_libmaker (Elf64)
This commit is contained in:
grischka
parent
09ac9faf59
commit
06aed3d171
@ -23,201 +23,92 @@
|
||||
#define WIN32_LEAN_AND_MEAN
|
||||
#include <windows.h>
|
||||
#include <stdio.h>
|
||||
#include <malloc.h>
|
||||
#include "../../config.h"
|
||||
|
||||
/* Offset to PE file signature */
|
||||
#define NTSIGNATURE(a) ((LPVOID)((BYTE *)a + \
|
||||
((PIMAGE_DOS_HEADER)a)->e_lfanew))
|
||||
|
||||
/* MS-OS header identifies the NT PEFile signature dword;
|
||||
the PEFILE header exists just after that dword. */
|
||||
#define PEFHDROFFSET(a) ((LPVOID)((BYTE *)a + \
|
||||
((PIMAGE_DOS_HEADER)a)->e_lfanew + \
|
||||
SIZE_OF_NT_SIGNATURE))
|
||||
|
||||
/* PE optional header is immediately after PEFile header. */
|
||||
#define OPTHDROFFSET(a) ((LPVOID)((BYTE *)a + \
|
||||
((PIMAGE_DOS_HEADER)a)->e_lfanew + \
|
||||
SIZE_OF_NT_SIGNATURE + \
|
||||
sizeof (IMAGE_FILE_HEADER)))
|
||||
|
||||
/* Section headers are immediately after PE optional header. */
|
||||
#define SECHDROFFSET(a) ((LPVOID)((BYTE *)a + \
|
||||
((PIMAGE_DOS_HEADER)a)->e_lfanew + \
|
||||
SIZE_OF_NT_SIGNATURE + \
|
||||
sizeof (IMAGE_FILE_HEADER) + \
|
||||
sizeof (IMAGE_OPTIONAL_HEADER)))
|
||||
|
||||
|
||||
#define SIZE_OF_NT_SIGNATURE 4
|
||||
|
||||
/* -------------------------------------------------------------- */
|
||||
|
||||
int WINAPI NumOfSections (
|
||||
LPVOID lpFile)
|
||||
int read_mem(FILE *fp, unsigned offset, void *buffer, unsigned len)
|
||||
{
|
||||
/* Number of sections is indicated in file header. */
|
||||
return (int)
|
||||
((PIMAGE_FILE_HEADER)
|
||||
PEFHDROFFSET(lpFile))->NumberOfSections;
|
||||
fseek(fp, offset, 0);
|
||||
return len == fread(buffer, 1, len, fp);
|
||||
}
|
||||
|
||||
|
||||
/* -------------------------------------------------------------- */
|
||||
|
||||
LPVOID WINAPI ImageDirectoryOffset (
|
||||
LPVOID lpFile,
|
||||
DWORD dwIMAGE_DIRECTORY)
|
||||
char *get_export_names(FILE *fp)
|
||||
{
|
||||
PIMAGE_OPTIONAL_HEADER poh;
|
||||
PIMAGE_SECTION_HEADER psh;
|
||||
int nSections = NumOfSections (lpFile);
|
||||
int i = 0;
|
||||
LPVOID VAImageDir;
|
||||
int l, i, n, n0;
|
||||
char *p;
|
||||
|
||||
/* Retrieve offsets to optional and section headers. */
|
||||
poh = (PIMAGE_OPTIONAL_HEADER)OPTHDROFFSET (lpFile);
|
||||
psh = (PIMAGE_SECTION_HEADER)SECHDROFFSET (lpFile);
|
||||
IMAGE_SECTION_HEADER ish;
|
||||
IMAGE_EXPORT_DIRECTORY ied;
|
||||
IMAGE_DOS_HEADER dh;
|
||||
IMAGE_FILE_HEADER ih;
|
||||
DWORD sig, ref, addr, ptr, namep;
|
||||
#ifdef TCC_TARGET_X86_64
|
||||
IMAGE_OPTIONAL_HEADER64 oh;
|
||||
const int MACHINE = 0x8664;
|
||||
#else
|
||||
IMAGE_OPTIONAL_HEADER32 oh;
|
||||
const int MACHINE = 0x014C;
|
||||
#endif
|
||||
int pef_hdroffset, opt_hdroffset, sec_hdroffset;
|
||||
|
||||
/* Must be 0 thru (NumberOfRvaAndSizes-1). */
|
||||
if (dwIMAGE_DIRECTORY >= poh->NumberOfRvaAndSizes)
|
||||
return NULL;
|
||||
n = n0 = 0;
|
||||
p = NULL;
|
||||
|
||||
/* Locate image directory's relative virtual address. */
|
||||
VAImageDir = (LPVOID)poh->DataDirectory[dwIMAGE_DIRECTORY].VirtualAddress;
|
||||
if (!read_mem(fp, 0, &dh, sizeof dh))
|
||||
goto the_end;
|
||||
if (!read_mem(fp, dh.e_lfanew, &sig, sizeof sig))
|
||||
goto the_end;
|
||||
if (sig != 0x00004550)
|
||||
goto the_end;
|
||||
pef_hdroffset = dh.e_lfanew + sizeof sig;
|
||||
if (!read_mem(fp, pef_hdroffset, &ih, sizeof ih))
|
||||
goto the_end;
|
||||
if (MACHINE != ih.Machine)
|
||||
goto the_end;
|
||||
opt_hdroffset = pef_hdroffset + sizeof ih;
|
||||
sec_hdroffset = opt_hdroffset + sizeof oh;
|
||||
if (!read_mem(fp, opt_hdroffset, &oh, sizeof oh))
|
||||
goto the_end;
|
||||
|
||||
/* Locate section containing image directory. */
|
||||
while (i++<nSections)
|
||||
{
|
||||
if (psh->VirtualAddress <= (DWORD)VAImageDir
|
||||
&& psh->VirtualAddress + psh->SizeOfRawData > (DWORD)VAImageDir)
|
||||
break;
|
||||
psh++;
|
||||
if (IMAGE_DIRECTORY_ENTRY_EXPORT >= oh.NumberOfRvaAndSizes)
|
||||
goto the_end;
|
||||
|
||||
addr = oh.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress;
|
||||
//printf("addr: %08x\n", addr);
|
||||
for (i = 0; i < ih.NumberOfSections; ++i) {
|
||||
if (!read_mem(fp, sec_hdroffset + i * sizeof ish, &ish, sizeof ish))
|
||||
goto the_end;
|
||||
//printf("vaddr: %08x\n", ish.VirtualAddress);
|
||||
if (addr >= ish.VirtualAddress && addr < ish.VirtualAddress + ish.SizeOfRawData)
|
||||
goto found;
|
||||
}
|
||||
goto the_end;
|
||||
|
||||
if (i > nSections)
|
||||
return NULL;
|
||||
found:
|
||||
ref = ish.VirtualAddress - ish.PointerToRawData;
|
||||
if (!read_mem(fp, addr - ref, &ied, sizeof ied))
|
||||
goto the_end;
|
||||
|
||||
/* Return image import directory offset. */
|
||||
return (LPVOID)(((int)lpFile +
|
||||
(int)VAImageDir - psh->VirtualAddress) +
|
||||
(int)psh->PointerToRawData);
|
||||
}
|
||||
|
||||
/* -------------------------------------------------------------- */
|
||||
|
||||
BOOL WINAPI GetSectionHdrByName (
|
||||
LPVOID lpFile,
|
||||
IMAGE_SECTION_HEADER *sh,
|
||||
char *szSection)
|
||||
{
|
||||
PIMAGE_SECTION_HEADER psh;
|
||||
int nSections = NumOfSections (lpFile);
|
||||
int i;
|
||||
|
||||
if ((psh = (PIMAGE_SECTION_HEADER)SECHDROFFSET (lpFile)) != NULL)
|
||||
{
|
||||
/* find the section by name */
|
||||
for (i=0; i<nSections; i++)
|
||||
{
|
||||
if (!strcmp (psh->Name, szSection))
|
||||
{
|
||||
/* copy data to header */
|
||||
memcpy ((LPVOID)sh, (LPVOID)psh, sizeof (IMAGE_SECTION_HEADER));
|
||||
return TRUE;
|
||||
namep = ied.AddressOfNames - ref;
|
||||
for (i = 0; i < ied.NumberOfNames; ++i) {
|
||||
if (!read_mem(fp, namep, &ptr, sizeof ptr))
|
||||
goto the_end;
|
||||
namep += sizeof ptr;
|
||||
for (l = 0;;) {
|
||||
if (n+1 >= n0)
|
||||
p = realloc(p, n0 = n0 ? n0 * 2 : 256);
|
||||
if (!read_mem(fp, ptr - ref + l, p + n, 1) || ++l >= 80) {
|
||||
free(p), p = NULL;
|
||||
goto the_end;
|
||||
}
|
||||
else
|
||||
psh++;
|
||||
if (p[n++] == 0)
|
||||
break;
|
||||
}
|
||||
}
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/* -------------------------------------------------------------- */
|
||||
|
||||
BOOL WINAPI GetSectionHdrByAddress (
|
||||
LPVOID lpFile,
|
||||
IMAGE_SECTION_HEADER *sh,
|
||||
DWORD addr)
|
||||
{
|
||||
PIMAGE_SECTION_HEADER psh;
|
||||
int nSections = NumOfSections (lpFile);
|
||||
int i;
|
||||
|
||||
if ((psh = (PIMAGE_SECTION_HEADER)SECHDROFFSET (lpFile)) != NULL)
|
||||
{
|
||||
/* find the section by name */
|
||||
for (i=0; i<nSections; i++)
|
||||
{
|
||||
if (addr >= psh->VirtualAddress
|
||||
&& addr < psh->VirtualAddress + psh->SizeOfRawData)
|
||||
{
|
||||
/* copy data to header */
|
||||
memcpy ((LPVOID)sh, (LPVOID)psh, sizeof (IMAGE_SECTION_HEADER));
|
||||
return TRUE;
|
||||
}
|
||||
else
|
||||
psh++;
|
||||
}
|
||||
}
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/* -------------------------------------------------------------- */
|
||||
|
||||
int WINAPI GetExportFunctionNames (
|
||||
LPVOID lpFile,
|
||||
HANDLE hHeap,
|
||||
char **pszFunctions)
|
||||
{
|
||||
IMAGE_SECTION_HEADER sh;
|
||||
PIMAGE_EXPORT_DIRECTORY ped;
|
||||
int *pNames, *pCnt;
|
||||
char *pSrc, *pDest;
|
||||
int i, nCnt;
|
||||
DWORD VAImageDir;
|
||||
PIMAGE_OPTIONAL_HEADER poh;
|
||||
char *pOffset;
|
||||
|
||||
/* Get section header and pointer to data directory
|
||||
for .edata section. */
|
||||
if (NULL == (ped = (PIMAGE_EXPORT_DIRECTORY)
|
||||
ImageDirectoryOffset (lpFile, IMAGE_DIRECTORY_ENTRY_EXPORT)))
|
||||
return 0;
|
||||
|
||||
poh = (PIMAGE_OPTIONAL_HEADER)OPTHDROFFSET (lpFile);
|
||||
VAImageDir = poh->DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress;
|
||||
|
||||
if (FALSE == GetSectionHdrByAddress (lpFile, &sh, VAImageDir))
|
||||
return 0;
|
||||
|
||||
pOffset = (char *)lpFile + (sh.PointerToRawData - sh.VirtualAddress);
|
||||
|
||||
pNames = (int *)(pOffset + (DWORD)ped->AddressOfNames);
|
||||
|
||||
/* Figure out how much memory to allocate for all strings. */
|
||||
nCnt = 1;
|
||||
for (i=0, pCnt = pNames; i<(int)ped->NumberOfNames; i++)
|
||||
{
|
||||
pSrc = (pOffset + *pCnt++);
|
||||
if (pSrc)
|
||||
nCnt += strlen(pSrc)+1;
|
||||
}
|
||||
|
||||
/* Allocate memory off heap for function names. */
|
||||
pDest = *pszFunctions = HeapAlloc (hHeap, HEAP_ZERO_MEMORY, nCnt);
|
||||
|
||||
/* Copy all strings to buffer. */
|
||||
for (i=0, pCnt = pNames; i<(int)ped->NumberOfNames; i++)
|
||||
{
|
||||
pSrc = (pOffset + *pCnt++);
|
||||
if (pSrc) {
|
||||
strcpy(pDest, pSrc);
|
||||
pDest += strlen(pSrc)+1;
|
||||
}
|
||||
}
|
||||
*pDest = 0;
|
||||
|
||||
return ped->NumberOfNames;
|
||||
if (p)
|
||||
p[n] = 0;
|
||||
the_end:
|
||||
return p;
|
||||
}
|
||||
|
||||
/* -------------------------------------------------------------- */
|
||||
@ -238,61 +129,49 @@ static char *file_basename(const char *name)
|
||||
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
HANDLE hHeap;
|
||||
HANDLE hFile;
|
||||
HANDLE hMapObject;
|
||||
VOID *pMem;
|
||||
|
||||
int nCnt, ret, n;
|
||||
char *pNames;
|
||||
int ret, v, i;
|
||||
char infile[MAX_PATH];
|
||||
char buffer[MAX_PATH];
|
||||
char outfile[MAX_PATH];
|
||||
FILE *op;
|
||||
char *p;
|
||||
|
||||
hHeap = NULL;
|
||||
hFile = NULL;
|
||||
hMapObject = NULL;
|
||||
pMem = NULL;
|
||||
static const char *ext[] = { ".dll", ".exe", NULL };
|
||||
const char *file, **pp;
|
||||
char path[MAX_PATH], *p, *q;
|
||||
FILE *fp, *op;
|
||||
|
||||
infile[0] = 0;
|
||||
outfile[0] = 0;
|
||||
fp = op = NULL;
|
||||
v = 0;
|
||||
ret = 1;
|
||||
|
||||
for (n = 1; n < argc; ++n)
|
||||
{
|
||||
const char *a = argv[n];
|
||||
for (i = 1; i < argc; ++i) {
|
||||
const char *a = argv[i];
|
||||
if ('-' == a[0]) {
|
||||
if (0 == strcmp(a, "-o")) {
|
||||
if (++n == argc)
|
||||
if (0 == strcmp(a, "-v")) {
|
||||
v = 1;
|
||||
} else if (0 == strcmp(a, "-o")) {
|
||||
if (++i == argc)
|
||||
goto usage;
|
||||
strcpy(outfile, argv[n]);
|
||||
}
|
||||
else
|
||||
strcpy(outfile, argv[i]);
|
||||
} else
|
||||
goto usage;
|
||||
|
||||
} else if (0 == infile[0])
|
||||
strcpy(infile, a);
|
||||
else
|
||||
goto usage;
|
||||
}
|
||||
|
||||
if (0 == infile[0])
|
||||
{
|
||||
if (0 == infile[0]) {
|
||||
usage:
|
||||
fprintf(stderr,
|
||||
"tiny_impdef creates an export definition file (.def) from a dll\n"
|
||||
"tiny_impdef: create export definition file (.def) from a dll\n"
|
||||
"Usage: tiny_impdef library.dll [-o outputfile]\n"
|
||||
);
|
||||
goto the_end;
|
||||
}
|
||||
|
||||
if (SearchPath(NULL, infile, ".dll", sizeof buffer, buffer, NULL))
|
||||
strcpy(infile, buffer);
|
||||
|
||||
if (0 == outfile[0])
|
||||
{
|
||||
char *p;
|
||||
strcpy(outfile, file_basename(infile));
|
||||
p = strrchr(outfile, '.');
|
||||
if (NULL == p)
|
||||
@ -300,93 +179,55 @@ usage:
|
||||
strcpy(p, ".def");
|
||||
}
|
||||
|
||||
hFile = CreateFile(
|
||||
infile,
|
||||
GENERIC_READ,
|
||||
FILE_SHARE_READ,
|
||||
NULL,
|
||||
OPEN_EXISTING,
|
||||
0,
|
||||
NULL
|
||||
);
|
||||
file = infile;
|
||||
#ifdef _WIN32
|
||||
for (pp = ext; *pp; ++pp)
|
||||
if (SearchPath(NULL, file, *pp, sizeof path, path, NULL)) {
|
||||
file = path;
|
||||
break;
|
||||
}
|
||||
#endif
|
||||
|
||||
if (hFile == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
fprintf(stderr, "No such file: %s\n", infile);
|
||||
fp = fopen(file, "rb");
|
||||
if (NULL == fp) {
|
||||
fprintf(stderr, "tiny_impdef: no such file: %s\n", infile);
|
||||
goto the_end;
|
||||
}
|
||||
|
||||
|
||||
hMapObject = CreateFileMapping(
|
||||
hFile,
|
||||
NULL,
|
||||
PAGE_READONLY,
|
||||
0, 0,
|
||||
NULL
|
||||
);
|
||||
|
||||
if (NULL == hMapObject)
|
||||
{
|
||||
fprintf(stderr, "Could not create file mapping: %s\n", infile);
|
||||
goto the_end;
|
||||
}
|
||||
|
||||
pMem = MapViewOfFile(
|
||||
hMapObject, // object to map view of
|
||||
FILE_MAP_READ, // read access
|
||||
0, // high offset: map from
|
||||
0, // low offset: beginning
|
||||
0); // default: map entire file
|
||||
|
||||
if (NULL == pMem)
|
||||
{
|
||||
fprintf(stderr, "Could not map view of file: %s\n", infile);
|
||||
goto the_end;
|
||||
}
|
||||
|
||||
if (0 != strncmp(NTSIGNATURE(pMem), "PE", 2))
|
||||
{
|
||||
fprintf(stderr, "Not a PE file: %s\n", infile);
|
||||
goto the_end;
|
||||
}
|
||||
|
||||
|
||||
hHeap = GetProcessHeap();
|
||||
nCnt = GetExportFunctionNames(pMem, hHeap, &pNames);
|
||||
if (0 == nCnt) {
|
||||
fprintf(stderr, "Could not get exported function names: %s\n", infile);
|
||||
goto the_end;
|
||||
}
|
||||
|
||||
printf("--> %s\n", infile);
|
||||
|
||||
op = fopen(outfile, "w");
|
||||
if (NULL == op)
|
||||
{
|
||||
fprintf(stderr, "Could not create file: %s\n", outfile);
|
||||
if (NULL == op) {
|
||||
fprintf(stderr, "tiny_impdef: could not create output file: %s\n", outfile);
|
||||
goto the_end;
|
||||
}
|
||||
|
||||
printf("<-- %s\n", outfile);
|
||||
if (v)
|
||||
printf("--> %s\n", infile);
|
||||
|
||||
fprintf(op, "LIBRARY %s\n\nEXPORTS\n", file_basename(infile));
|
||||
for (n = 0, p = pNames; n < nCnt; ++n)
|
||||
{
|
||||
fprintf(op, "%s\n", p);
|
||||
while (*p++);
|
||||
p = get_export_names(fp);
|
||||
if (NULL == p) {
|
||||
fprintf(stderr, "tiny_impdef: could not get exported function names.\n");
|
||||
goto the_end;
|
||||
}
|
||||
|
||||
for (q = p, i = 0; *q; ++i) {
|
||||
fprintf(op, "%s\n", q);
|
||||
q += strlen(q) + 1;
|
||||
}
|
||||
free(p);
|
||||
|
||||
if (v) {
|
||||
printf("<-- %s\n", outfile);
|
||||
printf("%d symbol(s) found\n", i);
|
||||
}
|
||||
|
||||
ret = 0;
|
||||
|
||||
the_end:
|
||||
if (pMem)
|
||||
UnmapViewOfFile(pMem);
|
||||
|
||||
if (hMapObject)
|
||||
CloseHandle(hMapObject);
|
||||
|
||||
if (hFile)
|
||||
CloseHandle(hFile);
|
||||
|
||||
if (fp)
|
||||
fclose(fp);
|
||||
if (op)
|
||||
fclose(op);
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
@ -13,98 +13,8 @@
|
||||
#include <io.h> /* for mktemp */
|
||||
#endif
|
||||
|
||||
/* #include "ar-elf.h" */
|
||||
/* "ar-elf.h" */
|
||||
/* ELF_v1.2.pdf */
|
||||
typedef unsigned short int Elf32_Half;
|
||||
typedef int Elf32_Sword;
|
||||
typedef unsigned int Elf32_Word;
|
||||
typedef unsigned int Elf32_Addr;
|
||||
typedef unsigned int Elf32_Off;
|
||||
typedef unsigned short int Elf32_Section;
|
||||
|
||||
#define EI_NIDENT 16
|
||||
typedef struct {
|
||||
unsigned char e_ident[EI_NIDENT];
|
||||
Elf32_Half e_type;
|
||||
Elf32_Half e_machine;
|
||||
Elf32_Word e_version;
|
||||
Elf32_Addr e_entry;
|
||||
Elf32_Off e_phoff;
|
||||
Elf32_Off e_shoff;
|
||||
Elf32_Word e_flags;
|
||||
Elf32_Half e_ehsize;
|
||||
Elf32_Half e_phentsize;
|
||||
Elf32_Half e_phnum;
|
||||
Elf32_Half e_shentsize;
|
||||
Elf32_Half e_shnum;
|
||||
Elf32_Half e_shstrndx;
|
||||
} Elf32_Ehdr;
|
||||
|
||||
typedef struct {
|
||||
Elf32_Word sh_name;
|
||||
Elf32_Word sh_type;
|
||||
Elf32_Word sh_flags;
|
||||
Elf32_Addr sh_addr;
|
||||
Elf32_Off sh_offset;
|
||||
Elf32_Word sh_size;
|
||||
Elf32_Word sh_link;
|
||||
Elf32_Word sh_info;
|
||||
Elf32_Word sh_addralign;
|
||||
Elf32_Word sh_entsize;
|
||||
} Elf32_Shdr;
|
||||
|
||||
#define SHT_NULL 0
|
||||
#define SHT_PROGBITS 1
|
||||
#define SHT_SYMTAB 2
|
||||
#define SHT_STRTAB 3
|
||||
#define SHT_RELA 4
|
||||
#define SHT_HASH 5
|
||||
#define SHT_DYNAMIC 6
|
||||
#define SHT_NOTE 7
|
||||
#define SHT_NOBITS 8
|
||||
#define SHT_REL 9
|
||||
#define SHT_SHLIB 10
|
||||
#define SHT_DYNSYM 11
|
||||
|
||||
typedef struct {
|
||||
Elf32_Word st_name;
|
||||
Elf32_Addr st_value;
|
||||
Elf32_Word st_size;
|
||||
unsigned char st_info;
|
||||
unsigned char st_other;
|
||||
Elf32_Half st_shndx;
|
||||
} Elf32_Sym;
|
||||
|
||||
#define ELF32_ST_BIND(i) ((i)>>4)
|
||||
#define ELF32_ST_TYPE(i) ((i)&0xf)
|
||||
#define ELF32_ST_INFO(b,t) (((b)<<4)+((t)&0xf))
|
||||
|
||||
#define STT_NOTYPE 0
|
||||
#define STT_OBJECT 1
|
||||
#define STT_FUNC 2
|
||||
#define STT_SECTION 3
|
||||
#define STT_FILE 4
|
||||
#define STT_LOPROC 13
|
||||
#define STT_HIPROC 15
|
||||
|
||||
#define STB_LOCAL 0
|
||||
#define STB_GLOBAL 1
|
||||
#define STB_WEAK 2
|
||||
#define STB_LOPROC 13
|
||||
#define STB_HIPROC 15
|
||||
|
||||
typedef struct {
|
||||
Elf32_Word p_type;
|
||||
Elf32_Off p_offset;
|
||||
Elf32_Addr p_vaddr;
|
||||
Elf32_Addr p_paddr;
|
||||
Elf32_Word p_filesz;
|
||||
Elf32_Word p_memsz;
|
||||
Elf32_Word p_flags;
|
||||
Elf32_Word p_align;
|
||||
} Elf32_Phdr;
|
||||
/* "ar-elf.h" ends */
|
||||
#include "../../config.h"
|
||||
#include "../../elf.h"
|
||||
|
||||
#define ARMAG "!<arch>\n"
|
||||
#define ARFMAG "`\n"
|
||||
@ -119,7 +29,6 @@ typedef struct ArHdr {
|
||||
char ar_fmag[2];
|
||||
} ArHdr;
|
||||
|
||||
|
||||
unsigned long le2belong(unsigned long ul) {
|
||||
return ((ul & 0xFF0000)>>8)+((ul & 0xFF000000)>>24) +
|
||||
((ul & 0xFF)<<24)+((ul & 0xFF00)<<8);
|
||||
@ -148,9 +57,9 @@ ArHdr arhdro = {
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
FILE *fi, *fh, *fo;
|
||||
Elf32_Ehdr *ehdr;
|
||||
Elf32_Shdr *shdr;
|
||||
Elf32_Sym *sym;
|
||||
ElfW(Ehdr) *ehdr;
|
||||
ElfW(Shdr) *shdr;
|
||||
ElfW(Sym) *sym;
|
||||
int i, fsize, iarg;
|
||||
char *buf, *shstr, *symtab = NULL, *strtab = NULL;
|
||||
int symtabsize = 0, strtabsize = 0;
|
||||
@ -207,7 +116,7 @@ int main(int argc, char **argv)
|
||||
}
|
||||
if ((fi = fopen(argv[iarg], "rb")) == NULL)
|
||||
{
|
||||
fprintf(stderr, "Can't open file %s \n", argv[iarg]);
|
||||
fprintf(stderr, "Can't open file %s \n", argv[iarg]);
|
||||
remove(tfile);
|
||||
return 2;
|
||||
}
|
||||
@ -218,14 +127,21 @@ int main(int argc, char **argv)
|
||||
fread(buf, fsize, 1, fi);
|
||||
fclose(fi);
|
||||
|
||||
printf("%s:\n", argv[iarg]);
|
||||
//printf("%s:\n", argv[iarg]);
|
||||
// elf header
|
||||
ehdr = (Elf32_Ehdr *)buf;
|
||||
shdr = (Elf32_Shdr *) (buf + ehdr->e_shoff + ehdr->e_shstrndx * ehdr->e_shentsize);
|
||||
ehdr = (ElfW(Ehdr) *)buf;
|
||||
if (ehdr->e_ident[4] != TCC_ELFCLASS)
|
||||
{
|
||||
fprintf(stderr, "Unsupported Elf Class: %s\n", argv[iarg]);
|
||||
remove(tfile);
|
||||
return 2;
|
||||
}
|
||||
|
||||
shdr = (ElfW(Shdr) *) (buf + ehdr->e_shoff + ehdr->e_shstrndx * ehdr->e_shentsize);
|
||||
shstr = (char *)(buf + shdr->sh_offset);
|
||||
for (i = 0; i < ehdr->e_shnum; i++)
|
||||
{
|
||||
shdr = (Elf32_Shdr *) (buf + ehdr->e_shoff + i * ehdr->e_shentsize);
|
||||
shdr = (ElfW(Shdr) *) (buf + ehdr->e_shoff + i * ehdr->e_shentsize);
|
||||
if (!shdr->sh_offset) continue;
|
||||
if (shdr->sh_type == SHT_SYMTAB)
|
||||
{
|
||||
@ -244,12 +160,16 @@ int main(int argc, char **argv)
|
||||
|
||||
if (symtab && symtabsize)
|
||||
{
|
||||
int nsym = symtabsize / sizeof(Elf32_Sym);
|
||||
int nsym = symtabsize / sizeof(ElfW(Sym));
|
||||
//printf("symtab: info size shndx name\n");
|
||||
for (i = 1; i < nsym; i++)
|
||||
{
|
||||
sym = (Elf32_Sym *) (symtab + i * sizeof(Elf32_Sym));
|
||||
if (sym->st_shndx && (sym->st_info == 0x11 || sym->st_info == 0x12)) {
|
||||
sym = (ElfW(Sym) *) (symtab + i * sizeof(ElfW(Sym)));
|
||||
if (sym->st_shndx &&
|
||||
(sym->st_info == 0x10
|
||||
|| sym->st_info == 0x11
|
||||
|| sym->st_info == 0x12
|
||||
)) {
|
||||
//printf("symtab: %2Xh %4Xh %2Xh %s\n", sym->st_info, sym->st_size, sym->st_shndx, strtab + sym->st_name);
|
||||
istrlen = strlen(strtab + sym->st_name)+1;
|
||||
anames = realloc(anames, strpos+istrlen);
|
||||
@ -281,7 +201,7 @@ int main(int argc, char **argv)
|
||||
} else fpos = 0;
|
||||
// write header
|
||||
fwrite("!<arch>\n", 8, 1, fh);
|
||||
sprintf(stmp, "%-10d", strpos + (funccnt+1) * sizeof(int));
|
||||
sprintf(stmp, "%-10d", (int)(strpos + (funccnt+1) * sizeof(int)));
|
||||
memcpy(&arhdr.ar_size, stmp, 10);
|
||||
fwrite(&arhdr, sizeof(arhdr), 1, fh);
|
||||
afpos[0] = le2belong(funccnt);
|
||||
|
||||
Loading…
Reference in New Issue
Block a user