343 lines
10 KiB
C
343 lines
10 KiB
C
/*
|
|
* DLL imports support
|
|
*
|
|
* Copyright 2000 Alexandre Julliard
|
|
*/
|
|
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
|
|
#include "winnt.h"
|
|
#include "build.h"
|
|
|
|
struct import
|
|
{
|
|
char *dll; /* dll name */
|
|
char **exports; /* functions exported from this dll */
|
|
int nb_exports; /* number of exported functions */
|
|
char **imports; /* functions we want to import from this dll */
|
|
int nb_imports; /* number of imported functions */
|
|
};
|
|
|
|
static char **undef_symbols; /* list of undefined symbols */
|
|
static int nb_undef_symbols = -1;
|
|
static int undef_size;
|
|
|
|
static struct import **dll_imports = NULL;
|
|
static int nb_imports = 0; /* number of imported dlls */
|
|
static int total_imports = 0; /* total number of imported functions */
|
|
|
|
|
|
/* compare function names; helper for resolve_imports */
|
|
static int name_cmp( const void *name, const void *entry )
|
|
{
|
|
return strcmp( *(char **)name, *(char **)entry );
|
|
}
|
|
|
|
/* locate a symbol in a (sorted) list */
|
|
inline static const char *find_symbol( const char *name, char **table, int size )
|
|
{
|
|
char **res = bsearch( &name, table, size, sizeof(*table), name_cmp );
|
|
return res ? *res : NULL;
|
|
}
|
|
|
|
/* sort a symbol table */
|
|
inline static void sort_symbols( char **table, int size )
|
|
{
|
|
qsort( table, size, sizeof(*table), name_cmp );
|
|
}
|
|
|
|
/* open the .so library for a given dll in a specified path */
|
|
static char *try_library_path( const char *path, const char *name )
|
|
{
|
|
char *buffer, *p;
|
|
int fd;
|
|
|
|
buffer = xmalloc( strlen(path) + strlen(name) + 8 );
|
|
sprintf( buffer, "%s/lib%s", path, name );
|
|
p = buffer + strlen(buffer) - 4;
|
|
if (!strcmp( p, ".dll" )) *p = 0;
|
|
strcat( buffer, ".so" );
|
|
/* check if the file exists */
|
|
if ((fd = open( buffer, O_RDONLY )) == -1) return NULL;
|
|
close( fd );
|
|
return buffer;
|
|
}
|
|
|
|
/* open the .so library for a given dll */
|
|
static char *open_library( const char *name )
|
|
{
|
|
char *fullname;
|
|
int i;
|
|
|
|
for (i = 0; i < nb_lib_paths; i++)
|
|
{
|
|
if ((fullname = try_library_path( lib_path[i], name ))) return fullname;
|
|
}
|
|
if (!(fullname = try_library_path( ".", name )))
|
|
fatal_error( "could not open .so file for %s\n", name );
|
|
return fullname;
|
|
}
|
|
|
|
/* read in the list of exported symbols of a .so */
|
|
static void read_exported_symbols( const char *name, struct import *imp )
|
|
{
|
|
FILE *f;
|
|
char buffer[1024];
|
|
char *fullname, *cmdline;
|
|
const char *ext;
|
|
int size, err;
|
|
|
|
imp->exports = NULL;
|
|
imp->nb_exports = size = 0;
|
|
|
|
if (!(ext = strrchr( name, '.' ))) ext = name + strlen(name);
|
|
|
|
if (!(fullname = open_library( name ))) return;
|
|
cmdline = xmalloc( strlen(fullname) + 4 );
|
|
sprintf( cmdline, "nm %s", fullname );
|
|
free( fullname );
|
|
|
|
if (!(f = popen( cmdline, "r" )))
|
|
fatal_error( "Cannot execute '%s'\n", cmdline );
|
|
|
|
while (fgets( buffer, sizeof(buffer), f ))
|
|
{
|
|
char *p = buffer + strlen(buffer) - 1;
|
|
if (p < buffer) continue;
|
|
if (*p == '\n') *p-- = 0;
|
|
if (!(p = strstr( buffer, "__wine_dllexport_" ))) continue;
|
|
p += 17;
|
|
if (strncmp( p, name, ext - name )) continue;
|
|
p += ext - name;
|
|
if (*p++ != '_') continue;
|
|
|
|
if (imp->nb_exports == size)
|
|
{
|
|
size += 128;
|
|
imp->exports = xrealloc( imp->exports, size * sizeof(*imp->exports) );
|
|
}
|
|
imp->exports[imp->nb_exports++] = xstrdup( p );
|
|
}
|
|
if ((err = pclose( f ))) fatal_error( "%s error %d\n", cmdline, err );
|
|
free( cmdline );
|
|
sort_symbols( imp->exports, imp->nb_exports );
|
|
}
|
|
|
|
/* add a dll to the list of imports */
|
|
void add_import_dll( const char *name )
|
|
{
|
|
struct import *imp = xmalloc( sizeof(*imp) );
|
|
imp->dll = xstrdup( name );
|
|
imp->imports = NULL;
|
|
imp->nb_imports = 0;
|
|
|
|
read_exported_symbols( name, imp );
|
|
|
|
dll_imports = xrealloc( dll_imports, (nb_imports+1) * sizeof(*dll_imports) );
|
|
dll_imports[nb_imports++] = imp;
|
|
}
|
|
|
|
/* add a function to the list of imports from a given dll */
|
|
static void add_import_func( struct import *imp, const char *name )
|
|
{
|
|
imp->imports = xrealloc( imp->imports, (imp->nb_imports+1) * sizeof(*imp->imports) );
|
|
imp->imports[imp->nb_imports++] = xstrdup( name );
|
|
total_imports++;
|
|
}
|
|
|
|
/* add a symbol to the undef list */
|
|
inline static void add_undef_symbol( const char *name )
|
|
{
|
|
if (nb_undef_symbols == undef_size)
|
|
{
|
|
undef_size += 128;
|
|
undef_symbols = xrealloc( undef_symbols, undef_size * sizeof(*undef_symbols) );
|
|
}
|
|
undef_symbols[nb_undef_symbols++] = xstrdup( name );
|
|
}
|
|
|
|
/* add the extra undefined symbols that will be contained in the generated spec file itself */
|
|
static void add_extra_undef_symbols(void)
|
|
{
|
|
const char *extras[8];
|
|
int i, count = 0;
|
|
|
|
#define ADD_SYM(name) \
|
|
do { if (!find_symbol( extras[count] = (name), undef_symbols, \
|
|
nb_undef_symbols )) count++; } while(0)
|
|
|
|
sort_symbols( undef_symbols, nb_undef_symbols );
|
|
|
|
/* add symbols that will be contained in the spec file itself */
|
|
switch (SpecMode)
|
|
{
|
|
case SPEC_MODE_GUIEXE:
|
|
ADD_SYM( "GetCommandLineA" );
|
|
ADD_SYM( "GetStartupInfoA" );
|
|
ADD_SYM( "GetModuleHandleA" );
|
|
/* fall through */
|
|
case SPEC_MODE_CUIEXE:
|
|
ADD_SYM( "__wine_get_main_args" );
|
|
ADD_SYM( "ExitProcess" );
|
|
/* fall through */
|
|
case SPEC_MODE_DLL:
|
|
case SPEC_MODE_GUIEXE_NO_MAIN:
|
|
case SPEC_MODE_CUIEXE_NO_MAIN:
|
|
ADD_SYM( "RtlRaiseException" );
|
|
break;
|
|
}
|
|
if (count)
|
|
{
|
|
for (i = 0; i < count; i++) add_undef_symbol( extras[i] );
|
|
sort_symbols( undef_symbols, nb_undef_symbols );
|
|
}
|
|
}
|
|
|
|
/* warn if a given dll is not used, but check forwards first */
|
|
static void warn_unused( const char *dllname )
|
|
{
|
|
int i;
|
|
size_t len = strlen(dllname);
|
|
const char *p = strchr( dllname, '.' );
|
|
if (p && !strcasecmp( p, ".dll" )) len = p - dllname;
|
|
|
|
for (i = Base; i <= Limit; i++)
|
|
{
|
|
ORDDEF *odp = Ordinals[i];
|
|
if (!odp || odp->type != TYPE_FORWARD) continue;
|
|
if (!strncasecmp( odp->u.fwd.link_name, dllname, len ) &&
|
|
odp->u.fwd.link_name[len] == '.')
|
|
return; /* found an import, do not warn */
|
|
}
|
|
warning( "%s imported but no symbols used\n", dllname );
|
|
}
|
|
|
|
/* read in the list of undefined symbols */
|
|
void read_undef_symbols( const char *name )
|
|
{
|
|
FILE *f;
|
|
char buffer[1024];
|
|
int err;
|
|
|
|
undef_size = nb_undef_symbols = 0;
|
|
|
|
sprintf( buffer, "nm -u %s", name );
|
|
if (!(f = popen( buffer, "r" )))
|
|
fatal_error( "Cannot execute '%s'\n", buffer );
|
|
|
|
while (fgets( buffer, sizeof(buffer), f ))
|
|
{
|
|
char *p = buffer + strlen(buffer) - 1;
|
|
if (p < buffer) continue;
|
|
if (*p == '\n') *p-- = 0;
|
|
add_undef_symbol( buffer );
|
|
}
|
|
if ((err = pclose( f ))) fatal_error( "nm -u %s error %d\n", name, err );
|
|
}
|
|
|
|
/* resolve the imports for a Win32 module */
|
|
int resolve_imports( FILE *outfile )
|
|
{
|
|
int i, j, off;
|
|
char **p;
|
|
|
|
if (nb_undef_symbols == -1) return 0; /* no symbol file specified */
|
|
|
|
add_extra_undef_symbols();
|
|
|
|
for (i = 0; i < nb_imports; i++)
|
|
{
|
|
struct import *imp = dll_imports[i];
|
|
|
|
for (j = 0; j < nb_undef_symbols; j++)
|
|
{
|
|
const char *res = find_symbol( undef_symbols[j], imp->exports, imp->nb_exports );
|
|
if (res)
|
|
{
|
|
add_import_func( imp, res );
|
|
undef_symbols[j] = NULL;
|
|
}
|
|
}
|
|
/* remove all the holes in the undef symbols list */
|
|
p = undef_symbols;
|
|
for (j = off = 0; j < nb_undef_symbols; j++)
|
|
{
|
|
if (!undef_symbols[j]) off++;
|
|
else undef_symbols[j - off] = undef_symbols[j];
|
|
}
|
|
nb_undef_symbols -= off;
|
|
if (!off) warn_unused( imp->dll );
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/* output the import table of a Win32 module */
|
|
int output_imports( FILE *outfile )
|
|
{
|
|
int i, j, pos;
|
|
|
|
if (!nb_imports) goto done;
|
|
|
|
/* main import header */
|
|
|
|
fprintf( outfile, "\nstatic struct {\n" );
|
|
fprintf( outfile, " struct {\n" );
|
|
fprintf( outfile, " void *OriginalFirstThunk;\n" );
|
|
fprintf( outfile, " unsigned int TimeDateStamp;\n" );
|
|
fprintf( outfile, " unsigned int ForwarderChain;\n" );
|
|
fprintf( outfile, " const char *Name;\n" );
|
|
fprintf( outfile, " void *FirstThunk;\n" );
|
|
fprintf( outfile, " } imp[%d];\n", nb_imports+1 );
|
|
fprintf( outfile, " const char *data[%d];\n", total_imports + nb_imports );
|
|
fprintf( outfile, "} imports = {\n {\n" );
|
|
|
|
/* list of dlls */
|
|
|
|
for (i = j = 0; i < nb_imports; i++)
|
|
{
|
|
fprintf( outfile, " { 0, 0, 0, \"%s\", &imports.data[%d] },\n",
|
|
dll_imports[i]->dll, j );
|
|
j += dll_imports[i]->nb_imports + 1;
|
|
}
|
|
fprintf( outfile, " { 0, 0, 0, 0, 0 },\n" );
|
|
fprintf( outfile, " },\n {\n" );
|
|
|
|
/* list of imported functions */
|
|
|
|
for (i = 0; i < nb_imports; i++)
|
|
{
|
|
fprintf( outfile, " /* %s */\n", dll_imports[i]->dll );
|
|
for (j = 0; j < dll_imports[i]->nb_imports; j++)
|
|
fprintf( outfile, " \"\\0\\0%s\",\n", dll_imports[i]->imports[j] );
|
|
fprintf( outfile, " 0,\n" );
|
|
}
|
|
fprintf( outfile, " }\n};\n\n" );
|
|
|
|
/* thunks for imported functions */
|
|
|
|
fprintf( outfile, "#ifndef __GNUC__\nstatic void __asm__dummy_import(void) {\n#endif\n\n" );
|
|
pos = 20 * (nb_imports + 1); /* offset of imports.data from start of imports */
|
|
fprintf( outfile, "asm(\".align 8\\n\"\n" );
|
|
for (i = 0; i < nb_imports; i++, pos += 4)
|
|
{
|
|
for (j = 0; j < dll_imports[i]->nb_imports; j++, pos += 4)
|
|
{
|
|
fprintf( outfile, " \"\\t.type " PREFIX "%s,@function\\n\"\n",
|
|
dll_imports[i]->imports[j] );
|
|
fprintf( outfile, " \"\\t.globl " PREFIX "%s\\n\"\n",
|
|
dll_imports[i]->imports[j] );
|
|
fprintf( outfile, " \"" PREFIX "%s:\\t", dll_imports[i]->imports[j] );
|
|
if (strstr( dll_imports[i]->imports[j], "__wine_call_from_16" ))
|
|
fprintf( outfile, ".byte 0x2e\\n\\tjmp *(imports+%d)\\n\\tnop\\n\"\n", pos );
|
|
else
|
|
fprintf( outfile, "jmp *(imports+%d)\\n\\tmovl %%esi,%%esi\\n\"\n", pos );
|
|
}
|
|
}
|
|
fprintf( outfile, ");\n#ifndef __GNUC__\n}\n#endif\n\n" );
|
|
|
|
done:
|
|
return nb_imports;
|
|
}
|