wnaspi32: Make winaspi.dll into a stand-alone 16-bit module.
[wine] / tools / winebuild / main.c
1 /*
2  * Main function
3  *
4  * Copyright 1993 Robert J. Amstadt
5  * Copyright 1995 Martin von Loewis
6  * Copyright 1995, 1996, 1997 Alexandre Julliard
7  * Copyright 1997 Eric Youngdale
8  * Copyright 1999 Ulrich Weigand
9  *
10  * This library is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Lesser General Public
12  * License as published by the Free Software Foundation; either
13  * version 2.1 of the License, or (at your option) any later version.
14  *
15  * This library is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  * Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public
21  * License along with this library; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23  */
24
25 #include "config.h"
26 #include "wine/port.h"
27
28 #include <assert.h>
29 #include <stdio.h>
30 #include <signal.h>
31 #include <errno.h>
32 #include <string.h>
33 #include <stdarg.h>
34 #include <ctype.h>
35 #ifdef HAVE_GETOPT_H
36 # include <getopt.h>
37 #endif
38
39 #include "windef.h"
40 #include "winbase.h"
41 #include "build.h"
42
43 int UsePIC = 0;
44 int nb_lib_paths = 0;
45 int nb_errors = 0;
46 int display_warnings = 0;
47 int kill_at = 0;
48 int verbose = 0;
49 int save_temps = 0;
50 int link_ext_symbols = 0;
51 int force_pointer_size = 0;
52
53 #ifdef __i386__
54 enum target_cpu target_cpu = CPU_x86;
55 #elif defined(__x86_64__)
56 enum target_cpu target_cpu = CPU_x86_64;
57 #elif defined(__sparc__)
58 enum target_cpu target_cpu = CPU_SPARC;
59 #elif defined(__ALPHA__)
60 enum target_cpu target_cpu = CPU_ALPHA;
61 #elif defined(__powerpc__)
62 enum target_cpu target_cpu = CPU_POWERPC;
63 #else
64 #error Unsupported CPU
65 #endif
66
67 #ifdef __APPLE__
68 enum target_platform target_platform = PLATFORM_APPLE;
69 #elif defined(__sun)
70 enum target_platform target_platform = PLATFORM_SOLARIS;
71 #elif defined(_WINDOWS)
72 enum target_platform target_platform = PLATFORM_WINDOWS;
73 #else
74 enum target_platform target_platform = PLATFORM_UNSPECIFIED;
75 #endif
76
77 char *target_alias = NULL;
78 char **lib_path = NULL;
79
80 char *input_file_name = NULL;
81 char *spec_file_name = NULL;
82 FILE *output_file = NULL;
83 const char *output_file_name = NULL;
84 static const char *output_file_source_name;
85
86 char *as_command = NULL;
87 char *ld_command = NULL;
88 char *nm_command = NULL;
89
90 static int nb_res_files;
91 static char **res_files;
92
93 /* execution mode */
94 enum exec_mode_values
95 {
96     MODE_NONE,
97     MODE_DLL,
98     MODE_EXE,
99     MODE_DEF,
100     MODE_RELAY16,
101     MODE_RELAY32
102 };
103
104 static enum exec_mode_values exec_mode = MODE_NONE;
105
106 static const struct
107 {
108     const char *name;
109     enum target_platform platform;
110 } platform_names[] =
111 {
112     { "macos",   PLATFORM_APPLE },
113     { "darwin",  PLATFORM_APPLE },
114     { "solaris", PLATFORM_SOLARIS },
115     { "windows", PLATFORM_WINDOWS },
116     { "winnt",   PLATFORM_WINDOWS }
117 };
118
119 /* set the dll file name from the input file name */
120 static void set_dll_file_name( const char *name, DLLSPEC *spec )
121 {
122     char *p;
123
124     if (spec->file_name) return;
125
126     if ((p = strrchr( name, '\\' ))) name = p + 1;
127     if ((p = strrchr( name, '/' ))) name = p + 1;
128     spec->file_name = xmalloc( strlen(name) + 5 );
129     strcpy( spec->file_name, name );
130     if ((p = strrchr( spec->file_name, '.' )))
131     {
132         if (!strcmp( p, ".spec" ) || !strcmp( p, ".def" )) *p = 0;
133     }
134 }
135
136 /* set the dll subsystem */
137 static void set_subsystem( const char *subsystem, DLLSPEC *spec )
138 {
139     char *major, *minor, *str = xstrdup( subsystem );
140
141     if ((major = strchr( str, ':' ))) *major++ = 0;
142     if (!strcmp( str, "native" )) spec->subsystem = IMAGE_SUBSYSTEM_NATIVE;
143     else if (!strcmp( str, "windows" )) spec->subsystem = IMAGE_SUBSYSTEM_WINDOWS_GUI;
144     else if (!strcmp( str, "console" )) spec->subsystem = IMAGE_SUBSYSTEM_WINDOWS_CUI;
145     else if (!strcmp( str, "win16" )) spec->type = SPEC_WIN16;
146     else fatal_error( "Invalid subsystem name '%s'\n", subsystem );
147     if (major)
148     {
149         if ((minor = strchr( major, '.' )))
150         {
151             *minor++ = 0;
152             spec->subsystem_minor = atoi( minor );
153         }
154         spec->subsystem_major = atoi( major );
155     }
156     free( str );
157 }
158
159 /* set the target CPU and platform */
160 static void set_target( const char *target )
161 {
162     unsigned int i;
163     char *p, *platform, *spec = xstrdup( target );
164
165     /* target specification is in the form CPU-MANUFACTURER-OS or CPU-MANUFACTURER-KERNEL-OS */
166
167     target_alias = xstrdup( target );
168
169     /* get the CPU part */
170
171     if (!(p = strchr( spec, '-' ))) fatal_error( "Invalid target specification '%s'\n", target );
172     *p++ = 0;
173     if ((target_cpu = get_cpu_from_name( spec )) == -1)
174         fatal_error( "Unrecognized CPU '%s'\n", spec );
175     platform = p;
176     if ((p = strrchr( p, '-' ))) platform = p + 1;
177
178     /* get the OS part */
179
180     target_platform = PLATFORM_UNSPECIFIED;  /* default value */
181     for (i = 0; i < sizeof(platform_names)/sizeof(platform_names[0]); i++)
182     {
183         if (!strncmp( platform_names[i].name, platform, strlen(platform_names[i].name) ))
184         {
185             target_platform = platform_names[i].platform;
186             break;
187         }
188     }
189
190     free( spec );
191 }
192
193 /* cleanup on program exit */
194 static void cleanup(void)
195 {
196     if (output_file_name) unlink( output_file_name );
197 }
198
199 /* clean things up when aborting on a signal */
200 static void exit_on_signal( int sig )
201 {
202     exit(1);  /* this will call atexit functions */
203 }
204
205 /*******************************************************************
206  *         command-line option handling
207  */
208 static const char usage_str[] =
209 "Usage: winebuild [OPTIONS] [FILES]\n\n"
210 "Options:\n"
211 "       --as-cmd=AS          Command to use for assembling (default: as)\n"
212 "   -b, --target=TARGET      Specify target CPU and platform for cross-compiling\n"
213 "   -d, --delay-lib=LIB      Import the specified library in delayed mode\n"
214 "   -D SYM                   Ignored for C flags compatibility\n"
215 "   -e, --entry=FUNC         Set the DLL entry point function (default: DllMain)\n"
216 "   -E, --export=FILE        Export the symbols defined in the .spec or .def file\n"
217 "       --external-symbols   Allow linking to external symbols\n"
218 "   -f FLAGS                 Compiler flags (only -fPIC is supported)\n"
219 "   -F, --filename=DLLFILE   Set the DLL filename (default: from input file name)\n"
220 "   -h, --help               Display this help message\n"
221 "   -H, --heap=SIZE          Set the heap size for a Win16 dll\n"
222 "   -i, --ignore=SYM[,SYM]   Ignore specified symbols when resolving imports\n"
223 "   -I DIR                   Ignored for C flags compatibility\n"
224 "   -k, --kill-at            Kill stdcall decorations in generated .def files\n"
225 "   -K, FLAGS                Compiler flags (only -KPIC is supported)\n"
226 "       --ld-cmd=LD          Command to use for linking (default: ld)\n"
227 "   -l, --library=LIB        Import the specified library\n"
228 "   -L, --library-path=DIR   Look for imports libraries in DIR\n"
229 "   -m32, -m64               Force building 32-bit resp. 64-bit code\n"
230 "   -M, --main-module=MODULE Set the name of the main module for a Win16 dll\n"
231 "       --nm-cmd=NM          Command to use to get undefined symbols (default: nm)\n"
232 "       --nxcompat=y|n       Set the NX compatibility flag (default: yes)\n"
233 "   -N, --dll-name=DLLNAME   Set the DLL name (default: from input file name)\n"
234 "   -o, --output=NAME        Set the output file name (default: stdout)\n"
235 "   -r, --res=RSRC.RES       Load resources from RSRC.RES\n"
236 "       --save-temps         Do not delete the generated intermediate files\n"
237 "       --subsystem=SUBSYS   Set the subsystem (one of native, windows, console)\n"
238 "   -u, --undefined=SYMBOL   Add an undefined reference to SYMBOL when linking\n"
239 "   -v, --verbose            Display the programs invoked\n"
240 "       --version            Print the version and exit\n"
241 "   -w, --warnings           Turn on warnings\n"
242 "\nMode options:\n"
243 "       --dll                Build a .c file from a .spec or .def file\n"
244 "       --def                Build a .def file from a .spec file\n"
245 "       --exe                Build a .c file for an executable\n"
246 "       --relay16            Build the 16-bit relay assembly routines\n"
247 "       --relay32            Build the 32-bit relay assembly routines\n\n"
248 "The mode options are mutually exclusive; you must specify one and only one.\n\n";
249
250 enum long_options_values
251 {
252     LONG_OPT_DLL = 1,
253     LONG_OPT_DEF,
254     LONG_OPT_EXE,
255     LONG_OPT_ASCMD,
256     LONG_OPT_EXTERNAL_SYMS,
257     LONG_OPT_LDCMD,
258     LONG_OPT_NMCMD,
259     LONG_OPT_NXCOMPAT,
260     LONG_OPT_RELAY16,
261     LONG_OPT_RELAY32,
262     LONG_OPT_SAVE_TEMPS,
263     LONG_OPT_SUBSYSTEM,
264     LONG_OPT_VERSION
265 };
266
267 static const char short_options[] = "C:D:E:F:H:I:K:L:M:N:b:d:e:f:hi:kl:m:o:r:u:vw";
268
269 static const struct option long_options[] =
270 {
271     { "dll",           0, 0, LONG_OPT_DLL },
272     { "def",           0, 0, LONG_OPT_DEF },
273     { "exe",           0, 0, LONG_OPT_EXE },
274     { "as-cmd",        1, 0, LONG_OPT_ASCMD },
275     { "external-symbols", 0, 0, LONG_OPT_EXTERNAL_SYMS },
276     { "ld-cmd",        1, 0, LONG_OPT_LDCMD },
277     { "nm-cmd",        1, 0, LONG_OPT_NMCMD },
278     { "nxcompat",      1, 0, LONG_OPT_NXCOMPAT },
279     { "relay16",       0, 0, LONG_OPT_RELAY16 },
280     { "relay32",       0, 0, LONG_OPT_RELAY32 },
281     { "save-temps",    0, 0, LONG_OPT_SAVE_TEMPS },
282     { "subsystem",     1, 0, LONG_OPT_SUBSYSTEM },
283     { "version",       0, 0, LONG_OPT_VERSION },
284     /* aliases for short options */
285     { "target",        1, 0, 'b' },
286     { "delay-lib",     1, 0, 'd' },
287     { "export",        1, 0, 'E' },
288     { "entry",         1, 0, 'e' },
289     { "filename",      1, 0, 'F' },
290     { "help",          0, 0, 'h' },
291     { "heap",          1, 0, 'H' },
292     { "ignore",        1, 0, 'i' },
293     { "kill-at",       0, 0, 'k' },
294     { "library",       1, 0, 'l' },
295     { "library-path",  1, 0, 'L' },
296     { "main-module",   1, 0, 'M' },
297     { "dll-name",      1, 0, 'N' },
298     { "output",        1, 0, 'o' },
299     { "res",           1, 0, 'r' },
300     { "undefined",     1, 0, 'u' },
301     { "verbose",       0, 0, 'v' },
302     { "warnings",      0, 0, 'w' },
303     { NULL,            0, 0, 0 }
304 };
305
306 static void usage( int exit_code )
307 {
308     fprintf( stderr, "%s", usage_str );
309     exit( exit_code );
310 }
311
312 static void set_exec_mode( enum exec_mode_values mode )
313 {
314     if (exec_mode != MODE_NONE) usage(1);
315     exec_mode = mode;
316 }
317
318 /* parse options from the argv array and remove all the recognized ones */
319 static char **parse_options( int argc, char **argv, DLLSPEC *spec )
320 {
321     char *p;
322     int optc;
323
324     while ((optc = getopt_long( argc, argv, short_options, long_options, NULL )) != -1)
325     {
326         switch(optc)
327         {
328         case 'D':
329             /* ignored */
330             break;
331         case 'E':
332             spec_file_name = xstrdup( optarg );
333             set_dll_file_name( optarg, spec );
334             break;
335         case 'F':
336             spec->file_name = xstrdup( optarg );
337             break;
338         case 'H':
339             if (!isdigit(optarg[0]))
340                 fatal_error( "Expected number argument with -H option instead of '%s'\n", optarg );
341             spec->heap_size = atoi(optarg);
342             if (spec->heap_size > 65535)
343                 fatal_error( "Invalid heap size %d, maximum is 65535\n", spec->heap_size );
344             break;
345         case 'I':
346             /* ignored */
347             break;
348         case 'K':
349             /* ignored, because cc generates correct code. */
350             break;
351         case 'L':
352             lib_path = xrealloc( lib_path, (nb_lib_paths+1) * sizeof(*lib_path) );
353             lib_path[nb_lib_paths++] = xstrdup( optarg );
354             break;
355         case 'm':
356             if (strcmp( optarg, "32" ) && strcmp( optarg, "64" ))
357                 fatal_error( "Invalid -m option '%s', expected -m32 or -m64\n", optarg );
358             if (!strcmp( optarg, "32" )) force_pointer_size = 4;
359             else force_pointer_size = 8;
360             break;
361         case 'M':
362             spec->main_module = xstrdup( optarg );
363             break;
364         case 'N':
365             spec->dll_name = xstrdup( optarg );
366             break;
367         case 'b':
368             set_target( optarg );
369             break;
370         case 'd':
371             add_delayed_import( optarg );
372             break;
373         case 'e':
374             spec->init_func = xstrdup( optarg );
375             if ((p = strchr( spec->init_func, '@' ))) *p = 0;  /* kill stdcall decoration */
376             break;
377         case 'f':
378             if (!strcmp( optarg, "PIC") || !strcmp( optarg, "pic")) UsePIC = 1;
379             /* ignore all other flags */
380             break;
381         case 'h':
382             usage(0);
383             break;
384         case 'i':
385             {
386                 char *str = xstrdup( optarg );
387                 char *token = strtok( str, "," );
388                 while (token)
389                 {
390                     add_ignore_symbol( token );
391                     token = strtok( NULL, "," );
392                 }
393                 free( str );
394             }
395             break;
396         case 'k':
397             kill_at = 1;
398             break;
399         case 'l':
400             add_import_dll( optarg, NULL );
401             break;
402         case 'o':
403             {
404                 char *ext = strrchr( optarg, '.' );
405
406                 if (unlink( optarg ) == -1 && errno != ENOENT)
407                     fatal_error( "Unable to create output file '%s'\n", optarg );
408                 if (ext && !strcmp( ext, ".o" ))
409                 {
410                     output_file_source_name = get_temp_file_name( optarg, ".s" );
411                     if (!(output_file = fopen( output_file_source_name, "w" )))
412                         fatal_error( "Unable to create output file '%s'\n", optarg );
413                 }
414                 else
415                 {
416                     if (!(output_file = fopen( optarg, "w" )))
417                         fatal_error( "Unable to create output file '%s'\n", optarg );
418                 }
419                 output_file_name = xstrdup(optarg);
420                 atexit( cleanup );  /* make sure we remove the output file on exit */
421             }
422             break;
423         case 'r':
424             res_files = xrealloc( res_files, (nb_res_files+1) * sizeof(*res_files) );
425             res_files[nb_res_files++] = xstrdup( optarg );
426             break;
427         case 'u':
428             add_extra_ld_symbol( optarg );
429             break;
430         case 'v':
431             verbose++;
432             break;
433         case 'w':
434             display_warnings = 1;
435             break;
436         case LONG_OPT_DLL:
437             set_exec_mode( MODE_DLL );
438             break;
439         case LONG_OPT_DEF:
440             set_exec_mode( MODE_DEF );
441             break;
442         case LONG_OPT_EXE:
443             set_exec_mode( MODE_EXE );
444             if (!spec->subsystem) spec->subsystem = IMAGE_SUBSYSTEM_WINDOWS_GUI;
445             break;
446         case LONG_OPT_ASCMD:
447             as_command = xstrdup( optarg );
448             break;
449         case LONG_OPT_EXTERNAL_SYMS:
450             link_ext_symbols = 1;
451             break;
452         case LONG_OPT_LDCMD:
453             ld_command = xstrdup( optarg );
454             break;
455         case LONG_OPT_NMCMD:
456             nm_command = xstrdup( optarg );
457             break;
458         case LONG_OPT_NXCOMPAT:
459             if (optarg[0] == 'n' || optarg[0] == 'N')
460                 spec->dll_characteristics &= ~IMAGE_DLLCHARACTERISTICS_NX_COMPAT;
461             break;
462         case LONG_OPT_RELAY16:
463             set_exec_mode( MODE_RELAY16 );
464             break;
465         case LONG_OPT_RELAY32:
466             set_exec_mode( MODE_RELAY32 );
467             break;
468         case LONG_OPT_SAVE_TEMPS:
469             save_temps = 1;
470             break;
471         case LONG_OPT_SUBSYSTEM:
472             set_subsystem( optarg, spec );
473             break;
474         case LONG_OPT_VERSION:
475             printf( "winebuild version " PACKAGE_VERSION "\n" );
476             exit(0);
477         case '?':
478             usage(1);
479             break;
480         }
481     }
482
483     if (spec->file_name && !strchr( spec->file_name, '.' ))
484         strcat( spec->file_name, exec_mode == MODE_EXE ? ".exe" : ".dll" );
485
486     switch (target_cpu)
487     {
488     case CPU_x86:
489         if (force_pointer_size == 8) target_cpu = CPU_x86_64;
490         break;
491     case CPU_x86_64:
492         if (force_pointer_size == 4) target_cpu = CPU_x86;
493         break;
494     default:
495         if (force_pointer_size == 8)
496             fatal_error( "Cannot build 64-bit code for this CPU\n" );
497         break;
498     }
499
500     return &argv[optind];
501 }
502
503
504 /* load all specified resource files */
505 static void load_resources( char *argv[], DLLSPEC *spec )
506 {
507     int i;
508     char **ptr, **last;
509
510     switch (spec->type)
511     {
512     case SPEC_WIN16:
513         for (i = 0; i < nb_res_files; i++) load_res16_file( res_files[i], spec );
514         break;
515
516     case SPEC_WIN32:
517         for (i = 0; i < nb_res_files; i++)
518         {
519             if (!load_res32_file( res_files[i], spec ))
520                 fatal_error( "%s is not a valid Win32 resource file\n", res_files[i] );
521         }
522
523         /* load any resource file found in the remaining arguments */
524         for (ptr = last = argv; *ptr; ptr++)
525         {
526             if (!load_res32_file( *ptr, spec ))
527                 *last++ = *ptr; /* not a resource file, keep it in the list */
528         }
529         *last = NULL;
530         break;
531     }
532 }
533
534 /* add input files that look like import libs to the import list */
535 static void load_import_libs( char *argv[] )
536 {
537     char **ptr, **last;
538
539     for (ptr = last = argv; *ptr; ptr++)
540     {
541         if (strendswith( *ptr, ".def" ))
542             add_import_dll( NULL, *ptr );
543         else
544             *last++ = *ptr; /* not an import dll, keep it in the list */
545     }
546     *last = NULL;
547 }
548
549 static int parse_input_file( DLLSPEC *spec )
550 {
551     FILE *input_file = open_input_file( NULL, spec_file_name );
552     char *extension = strrchr( spec_file_name, '.' );
553     int result;
554
555     spec->src_name = xstrdup( input_file_name );
556     if (extension && !strcmp( extension, ".def" ))
557         result = parse_def_file( input_file, spec );
558     else
559         result = parse_spec_file( input_file, spec );
560     close_input_file( input_file );
561     return result;
562 }
563
564
565 /*******************************************************************
566  *         main
567  */
568 int main(int argc, char **argv)
569 {
570     DLLSPEC *spec = alloc_dll_spec();
571
572 #ifdef SIGHUP
573     signal( SIGHUP, exit_on_signal );
574 #endif
575     signal( SIGTERM, exit_on_signal );
576     signal( SIGINT, exit_on_signal );
577
578     output_file = stdout;
579     argv = parse_options( argc, argv, spec );
580
581     switch(exec_mode)
582     {
583     case MODE_DLL:
584         if (spec->subsystem != IMAGE_SUBSYSTEM_NATIVE)
585             spec->characteristics |= IMAGE_FILE_DLL;
586         if (!spec_file_name) fatal_error( "missing .spec file\n" );
587         if (spec->type == SPEC_WIN32 && spec->main_module)  /* embedded 16-bit module */
588         {
589             spec->type = SPEC_WIN16;
590             load_resources( argv, spec );
591             if (parse_input_file( spec )) BuildSpec16File( spec );
592             break;
593         }
594         /* fall through */
595     case MODE_EXE:
596         load_resources( argv, spec );
597         load_import_libs( argv );
598         if (spec_file_name && !parse_input_file( spec )) break;
599         read_undef_symbols( spec, argv );
600         switch (spec->type)
601         {
602             case SPEC_WIN16:
603                 output_spec16_file( spec );
604                 break;
605             case SPEC_WIN32:
606                 BuildSpec32File( spec );
607                 break;
608             default: assert(0);
609         }
610         break;
611     case MODE_DEF:
612         if (argv[0]) fatal_error( "file argument '%s' not allowed in this mode\n", argv[0] );
613         if (spec->type == SPEC_WIN16) fatal_error( "Cannot yet build .def file for 16-bit dlls\n" );
614         if (!spec_file_name) fatal_error( "missing .spec file\n" );
615         if (!parse_input_file( spec )) break;
616         BuildDef32File( spec );
617         break;
618     case MODE_RELAY16:
619         if (argv[0]) fatal_error( "file argument '%s' not allowed in this mode\n", argv[0] );
620         BuildRelays16();
621         break;
622     case MODE_RELAY32:
623         if (argv[0]) fatal_error( "file argument '%s' not allowed in this mode\n", argv[0] );
624         BuildRelays32();
625         break;
626     default:
627         usage(1);
628         break;
629     }
630     if (nb_errors) exit(1);
631     if (output_file_name)
632     {
633         if (fclose( output_file ) < 0) fatal_perror( "fclose" );
634         if (output_file_source_name) assemble_file( output_file_source_name, output_file_name );
635         output_file_name = NULL;
636     }
637     return 0;
638 }