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