win32/tools/tiny_impdef.c

   1 /* -------------------------------------------------------------- */
   2 /*
   3  * tiny_impdef creates an export definition file (.def) from a dll
   4  * on MS-Windows. Usage: tiny_impdef library.dll [-o outputfile]"
   5  *
   6  *  Copyright (c) 2005,2007 grischka
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21  */
  22
  23 #ifndef TINY_IMPDEF_GET_EXPORT_NAMES_ONLY
  24
  25 #define WIN32_LEAN_AND_MEAN
  26 #include <windows.h>
  27 #include <stdio.h>
  28 #include <malloc.h>
  29
  30 char *get_export_names(FILE *fp);
  31 #define tcc_free free
  32 #define tcc_realloc realloc
  33
  34 /* extract the basename of a file */
  35 static char *file_basename(const char *name)
  36 {
  37     const char *p = strchr(name, 0);
  38     while (p > name
  39         && p[-1] != '/'
  40         && p[-1] != '\\'
  41         )
  42         --p;
  43     return (char*)p;
  44 }
  45
  46 int main(int argc, char **argv)
  47 {
  48     int ret, v, i;
  49     char infile[MAX_PATH];
  50     char outfile[MAX_PATH];
  51
  52     static const char *ext[] = { ".dll", ".exe", NULL };
  53     const char *file, **pp;
  54     char path[MAX_PATH], *p, *q;
  55     FILE *fp, *op;
  56
  57     infile[0] = 0;
  58     outfile[0] = 0;
  59     fp = op = NULL;
  60     v = 0;
  61     ret = 1;
  62     p = NULL;
  63
  64     for (i = 1; i < argc; ++i) {
  65         const char *a = argv[i];
  66         if ('-' == a[0]) {
  67             if (0 == strcmp(a, "-v")) {
  68                 v = 1;
  69             } else if (0 == strcmp(a, "-o")) {
  70                 if (++i == argc)
  71                     goto usage;
  72                 strcpy(outfile, argv[i]);
  73             } else
  74                 goto usage;
  75         } else if (0 == infile[0])
  76             strcpy(infile, a);
  77         else
  78             goto usage;
  79     }
  80
  81     if (0 == infile[0]) {
  82 usage:
  83         fprintf(stderr,
  84             "tiny_impdef: create export definition file (.def) from a dll\n"
  85             "Usage: tiny_impdef library.dll [-o outputfile]\n"
  86             );
  87         goto the_end;
  88     }
  89
  90     if (0 == outfile[0])
  91     {
  92         strcpy(outfile, file_basename(infile));
  93         q = strrchr(outfile, '.');
  94         if (NULL == q)
  95             q = strchr(outfile, 0);
  96         strcpy(q, ".def");
  97     }
  98
  99     file = infile;
 100
 101 #ifdef _WIN32
 102     pp = ext;
 103     do if (SearchPath(NULL, file, *pp, sizeof path, path, NULL)) {
 104        file = path;
 105        break;
 106     } while (*pp++);
 107 #endif
 108
 109     fp = fopen(file, "rb");
 110     if (NULL == fp) {
 111         fprintf(stderr, "tiny_impdef: no such file: %s\n", infile);
 112         goto the_end;
 113     }
 114     if (v)
 115         printf("--> %s\n", file);
 116
 117     p = get_export_names(fp);
 118     if (NULL == p) {
 119         fprintf(stderr, "tiny_impdef: could not get exported function names.\n");
 120         goto the_end;
 121     }
 122
 123     op = fopen(outfile, "w");
 124     if (NULL == op) {
 125         fprintf(stderr, "tiny_impdef: could not create output file: %s\n", outfile);
 126         goto the_end;
 127     }
 128
 129     fprintf(op, "LIBRARY %s\n\nEXPORTS\n", file_basename(file));
 130     for (q = p, i = 0; *q; ++i) {
 131         fprintf(op, "%s\n", q);
 132         q += strlen(q) + 1;
 133     }
 134
 135     if (v) {
 136         printf("<-- %s\n", outfile);
 137         printf("%d symbol(s) found\n", i);
 138     }
 139
 140     ret = 0;
 141
 142 the_end:
 143     if (p)
 144         free(p);
 145     if (fp)
 146         fclose(fp);
 147     if (op)
 148         fclose(op);
 149     return ret;
 150 }
 151
 152 /* -------------------------------------------------------------- */
 153
 154 int read_mem(FILE *fp, unsigned offset, void *buffer, unsigned len)
 155 {
 156     fseek(fp, offset, 0);
 157     return len == fread(buffer, 1, len, fp);
 158 }
 159
 160 /* -------------------------------------------------------------- */
 161 #endif
 162
 163 char *get_export_names(FILE *fp)
 164 {
 165     int l, i, n, n0;
 166     char *p;
 167
 168     IMAGE_SECTION_HEADER ish;
 169     IMAGE_EXPORT_DIRECTORY ied;
 170     IMAGE_DOS_HEADER dh;
 171     IMAGE_FILE_HEADER ih;
 172     DWORD sig, ref, addr, ptr, namep;
 173 #ifdef TCC_TARGET_X86_64
 174     IMAGE_OPTIONAL_HEADER64 oh;
 175     const int MACHINE = 0x8664;
 176 #else
 177     IMAGE_OPTIONAL_HEADER32 oh;
 178     const int MACHINE = 0x014C;
 179 #endif
 180     int pef_hdroffset, opt_hdroffset, sec_hdroffset;
 181
 182     n = n0 = 0;
 183     p = NULL;
 184
 185     if (!read_mem(fp, 0, &dh, sizeof dh))
 186         goto the_end;
 187     if (!read_mem(fp, dh.e_lfanew, &sig, sizeof sig))
 188         goto the_end;
 189     if (sig != 0x00004550)
 190         goto the_end;
 191     pef_hdroffset = dh.e_lfanew + sizeof sig;
 192     if (!read_mem(fp, pef_hdroffset, &ih, sizeof ih))
 193         goto the_end;
 194     if (MACHINE != ih.Machine)
 195         goto the_end;
 196     opt_hdroffset = pef_hdroffset + sizeof ih;
 197     sec_hdroffset = opt_hdroffset + sizeof oh;
 198     if (!read_mem(fp, opt_hdroffset, &oh, sizeof oh))
 199         goto the_end;
 200
 201     if (IMAGE_DIRECTORY_ENTRY_EXPORT >= oh.NumberOfRvaAndSizes)
 202         goto the_end;
 203
 204     addr = oh.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress;
 205     //printf("addr: %08x\n", addr);
 206     for (i = 0; i < ih.NumberOfSections; ++i) {
 207         if (!read_mem(fp, sec_hdroffset + i * sizeof ish, &ish, sizeof ish))
 208             goto the_end;
 209         //printf("vaddr: %08x\n", ish.VirtualAddress);
 210         if (addr >= ish.VirtualAddress && addr < ish.VirtualAddress + ish.SizeOfRawData)
 211             goto found;
 212     }
 213     goto the_end;
 214
 215 found:
 216     ref = ish.VirtualAddress - ish.PointerToRawData;
 217     if (!read_mem(fp, addr - ref, &ied, sizeof ied))
 218         goto the_end;
 219
 220     namep = ied.AddressOfNames - ref;
 221     for (i = 0; i < ied.NumberOfNames; ++i) {
 222         if (!read_mem(fp, namep, &ptr, sizeof ptr))
 223             goto the_end;
 224         namep += sizeof ptr;
 225         for (l = 0;;) {
 226             if (n+1 >= n0)
 227                 p = tcc_realloc(p, n0 = n0 ? n0 * 2 : 256);
 228             if (!read_mem(fp, ptr - ref + l, p + n, 1) || ++l >= 80) {
 229                 tcc_free(p), p = NULL;
 230                 goto the_end;
 231             }
 232             if (p[n++] == 0)
 233                 break;
 234         }
 235     }
 236     if (p)
 237         p[n] = 0;
 238 the_end:
 239     return p;
 240 }
 241
 242 /* -------------------------------------------------------------- */