b0a57efaf7e99fb2bb8638324667d62595a7a89c
[reactos.git] / reactos / 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 "winglue.h"
40 #include "build.h"
41
42 int UsePIC = 0;
43 int nb_lib_paths = 0;
44 int nb_errors = 0;
45 int display_warnings = 0;
46 int kill_at = 0;
47 int verbose = 0;
48 int save_temps = 0;
49 int link_ext_symbols = 0;
50
51 #if defined(__i386__)
52 enum target_cpu target_cpu = CPU_x86;
53 #elif defined(__x86_64__)
54 enum target_cpu target_cpu = CPU_x86_64;
55 #elif defined(__sparc__)
56 enum target_cpu target_cpu = CPU_SPARC;
57 #elif defined(__ALPHA__)
58 enum target_cpu target_cpu = CPU_ALPHA;
59 #elif defined(__powerpc__)
60 enum target_cpu target_cpu = CPU_POWERPC;
61 #else
62 #error Unsupported CPU
63 #endif
64
65 #ifdef __APPLE__
66 enum target_platform target_platform = PLATFORM_APPLE;
67 #elif defined(_WINDOWS)
68 enum target_platform target_platform = PLATFORM_WINDOWS;
69 #else
70 enum target_platform target_platform = PLATFORM_UNSPECIFIED;
71 #endif
72
73 char **lib_path = NULL;
74
75 char *input_file_name = NULL;
76 char *spec_file_name = NULL;
77 FILE *output_file = NULL;
78 const char *output_file_name = NULL;
79 static const char *output_file_source_name;
80
81 char *as_command = NULL;
82 char *ld_command = NULL;
83 char *nm_command = NULL;
84
85 static int nb_res_files;
86 static char **res_files;
87
88 /* execution mode */
89 enum exec_mode_values
90 {
91 MODE_NONE,
92 MODE_DLL,
93 MODE_EXE,
94 MODE_DEF,
95 MODE_RELAY16,
96 MODE_RELAY32,
97 MODE_PEDLL
98 };
99
100 static enum exec_mode_values exec_mode = MODE_NONE;
101
102 static const struct
103 {
104 const char *name;
105 enum target_cpu cpu;
106 } cpu_names[] =
107 {
108 { "i386", CPU_x86 },
109 { "i486", CPU_x86 },
110 { "i586", CPU_x86 },
111 { "i686", CPU_x86 },
112 { "i786", CPU_x86 },
113 { "x86_64", CPU_x86_64 },
114 { "sparc", CPU_SPARC },
115 { "alpha", CPU_ALPHA },
116 { "powerpc", CPU_POWERPC }
117 };
118
119 static const struct
120 {
121 const char *name;
122 enum target_platform platform;
123 } platform_names[] =
124 {
125 { "macos", PLATFORM_APPLE },
126 { "darwin", PLATFORM_APPLE },
127 { "windows", PLATFORM_WINDOWS },
128 { "winnt", PLATFORM_WINDOWS }
129 };
130
131 /* set the dll file name from the input file name */
132 static void set_dll_file_name( const char *name, DLLSPEC *spec )
133 {
134 char *p;
135
136 if (spec->file_name) return;
137
138 if ((p = strrchr( name, '\\' ))) name = p + 1;
139 if ((p = strrchr( name, '/' ))) name = p + 1;
140 spec->file_name = xmalloc( strlen(name) + 5 );
141 strcpy( spec->file_name, name );
142 if ((p = strrchr( spec->file_name, '.' )))
143 {
144 if (!strcmp( p, ".spec" ) || !strcmp( p, ".def" )) *p = 0;
145 }
146 }
147
148 /* set the dll subsystem */
149 static void set_subsystem( const char *subsystem, DLLSPEC *spec )
150 {
151 char *major, *minor, *str = xstrdup( subsystem );
152
153 if ((major = strchr( str, ':' ))) *major++ = 0;
154 if (!strcmp( str, "native" )) spec->subsystem = IMAGE_SUBSYSTEM_NATIVE;
155 else if (!strcmp( str, "windows" )) spec->subsystem = IMAGE_SUBSYSTEM_WINDOWS_GUI;
156 else if (!strcmp( str, "console" )) spec->subsystem = IMAGE_SUBSYSTEM_WINDOWS_CUI;
157 else fatal_error( "Invalid subsystem name '%s'\n", subsystem );
158 if (major)
159 {
160 if ((minor = strchr( major, '.' )))
161 {
162 *minor++ = 0;
163 spec->subsystem_minor = atoi( minor );
164 }
165 spec->subsystem_major = atoi( major );
166 }
167 free( str );
168 }
169
170 /* set the target CPU and platform */
171 static void set_target( const char *target )
172 {
173 unsigned int i;
174 char *p, *platform, *spec = xstrdup( target );
175
176 /* target specification is in the form CPU-MANUFACTURER-OS or CPU-MANUFACTURER-KERNEL-OS */
177
178 /* get the CPU part */
179
180 if (!(p = strchr( spec, '-' ))) fatal_error( "Invalid target specification '%s'\n", target );
181 *p++ = 0;
182 for (i = 0; i < sizeof(cpu_names)/sizeof(cpu_names[0]); i++)
183 {
184 if (!strcmp( cpu_names[i].name, spec )) break;
185 }
186 if (i < sizeof(cpu_names)/sizeof(cpu_names[0])) target_cpu = cpu_names[i].cpu;
187 else fatal_error( "Unrecognized CPU '%s'\n", spec );
188
189 platform = p;
190 if ((p = strrchr( p, '-' ))) platform = p + 1;
191
192 /* get the OS part */
193
194 target_platform = PLATFORM_UNSPECIFIED; /* default value */
195 for (i = 0; i < sizeof(platform_names)/sizeof(platform_names[0]); i++)
196 {
197 if (!strncmp( platform_names[i].name, platform, strlen(platform_names[i].name) ))
198 {
199 target_platform = platform_names[i].platform;
200 break;
201 }
202 }
203
204 free( spec );
205
206 if (!as_command)
207 {
208 as_command = xmalloc( strlen(target) + sizeof("-as") );
209 strcpy( as_command, target );
210 strcat( as_command, "-as" );
211 }
212 if (!ld_command)
213 {
214 ld_command = xmalloc( strlen(target) + sizeof("-ld") );
215 strcpy( ld_command, target );
216 strcat( ld_command, "-ld" );
217 }
218 if (!nm_command)
219 {
220 nm_command = xmalloc( strlen(target) + sizeof("-nm") );
221 strcpy( nm_command, target );
222 strcat( nm_command, "-nm" );
223 }
224 }
225
226 /* cleanup on program exit */
227 static void cleanup(void)
228 {
229 if (output_file_name) unlink( output_file_name );
230 }
231
232 /* clean things up when aborting on a signal */
233 static void exit_on_signal( int sig )
234 {
235 exit(1); /* this will call atexit functions */
236 }
237
238 /*******************************************************************
239 * command-line option handling
240 */
241 static const char usage_str[] =
242 "Usage: winebuild [OPTIONS] [FILES]\n\n"
243 "Options:\n"
244 " --as-cmd=AS Command to use for assembling (default: as)\n"
245 " -d, --delay-lib=LIB Import the specified library in delayed mode\n"
246 " -D SYM Ignored for C flags compatibility\n"
247 " -e, --entry=FUNC Set the DLL entry point function (default: DllMain)\n"
248 " -E, --export=FILE Export the symbols defined in the .spec or .def file\n"
249 " --external-symbols Allow linking to external symbols\n"
250 " -f FLAGS Compiler flags (only -fPIC is supported)\n"
251 " -F, --filename=DLLFILE Set the DLL filename (default: from input file name)\n"
252 " -h, --help Display this help message\n"
253 " -H, --heap=SIZE Set the heap size for a Win16 dll\n"
254 " -i, --ignore=SYM[,SYM] Ignore specified symbols when resolving imports\n"
255 " -I DIR Ignored for C flags compatibility\n"
256 " -k, --kill-at Kill stdcall decorations in generated .def files\n"
257 " -K, FLAGS Compiler flags (only -KPIC is supported)\n"
258 " --ld-cmd=LD Command to use for linking (default: ld)\n"
259 " -l, --library=LIB Import the specified library\n"
260 " -L, --library-path=DIR Look for imports libraries in DIR\n"
261 " -M, --main-module=MODULE Set the name of the main module for a Win16 dll\n"
262 " --nm-cmd=NM Command to use to get undefined symbols (default: nm)\n"
263 " --nxcompat=y|n Set the NX compatibility flag (default: yes)\n"
264 " -N, --dll-name=DLLNAME Set the DLL name (default: from input file name)\n"
265 " -o, --output=NAME Set the output file name (default: stdout)\n"
266 " -r, --res=RSRC.RES Load resources from RSRC.RES\n"
267 " --save-temps Do not delete the generated intermediate files\n"
268 " --subsystem=SUBSYS Set the subsystem (one of native, windows, console)\n"
269 " --target=TARGET Specify target CPU and platform for cross-compiling\n"
270 " -u, --undefined=SYMBOL Add an undefined reference to SYMBOL when linking\n"
271 " -v, --verbose Display the programs invoked\n"
272 " --version Print the version and exit\n"
273 " -w, --warnings Turn on warnings\n"
274 "\nMode options:\n"
275 " --dll Build a .c file from a .spec or .def file\n"
276 " --def Build a .def file from a .spec file\n"
277 " --exe Build a .c file for an executable\n"
278 " --relay16 Build the 16-bit relay assembly routines\n"
279 " --relay32 Build the 32-bit relay assembly routines\n\n"
280 " --pedll Build a .c file for PE dll\n\n"
281 "The mode options are mutually exclusive; you must specify one and only one.\n\n";
282
283 enum long_options_values
284 {
285 LONG_OPT_DLL = 1,
286 LONG_OPT_DEF,
287 LONG_OPT_EXE,
288 LONG_OPT_ASCMD,
289 LONG_OPT_EXTERNAL_SYMS,
290 LONG_OPT_LDCMD,
291 LONG_OPT_NMCMD,
292 LONG_OPT_NXCOMPAT,
293 LONG_OPT_RELAY16,
294 LONG_OPT_RELAY32,
295 LONG_OPT_SAVE_TEMPS,
296 LONG_OPT_SUBSYSTEM,
297 LONG_OPT_TARGET,
298 LONG_OPT_VERSION,
299 LONG_OPT_PEDLL
300 };
301
302 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";
303
304 static const struct option long_options[] =
305 {
306 { "dll", 0, 0, LONG_OPT_DLL },
307 { "def", 0, 0, LONG_OPT_DEF },
308 { "exe", 0, 0, LONG_OPT_EXE },
309 { "as-cmd", 1, 0, LONG_OPT_ASCMD },
310 { "external-symbols", 0, 0, LONG_OPT_EXTERNAL_SYMS },
311 { "ld-cmd", 1, 0, LONG_OPT_LDCMD },
312 { "nm-cmd", 1, 0, LONG_OPT_NMCMD },
313 { "nxcompat", 1, 0, LONG_OPT_NXCOMPAT },
314 { "relay16", 0, 0, LONG_OPT_RELAY16 },
315 { "relay32", 0, 0, LONG_OPT_RELAY32 },
316 { "save-temps", 0, 0, LONG_OPT_SAVE_TEMPS },
317 { "subsystem", 1, 0, LONG_OPT_SUBSYSTEM },
318 { "target", 1, 0, LONG_OPT_TARGET },
319 { "version", 0, 0, LONG_OPT_VERSION },
320 { "pedll", 1, 0, LONG_OPT_PEDLL },
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 LONG_OPT_PEDLL:
508 set_exec_mode( MODE_PEDLL );
509 spec_file_name = xstrdup( optarg );
510 set_dll_file_name( optarg, spec );
511 break;
512 case '?':
513 usage(1);
514 break;
515 }
516 }
517
518 if (spec->file_name && !strchr( spec->file_name, '.' ))
519 strcat( spec->file_name, exec_mode == MODE_EXE ? ".exe" : ".dll" );
520
521 return &argv[optind];
522 }
523
524
525 /* load all specified resource files */
526 static void load_resources( char *argv[], DLLSPEC *spec )
527 {
528 int i;
529 char **ptr, **last;
530
531 switch (spec->type)
532 {
533 case SPEC_WIN16:
534 for (i = 0; i < nb_res_files; i++) load_res16_file( res_files[i], spec );
535 break;
536
537 case SPEC_WIN32:
538 for (i = 0; i < nb_res_files; i++)
539 {
540 if (!load_res32_file( res_files[i], spec ))
541 fatal_error( "%s is not a valid Win32 resource file\n", res_files[i] );
542 }
543
544 /* load any resource file found in the remaining arguments */
545 for (ptr = last = argv; *ptr; ptr++)
546 {
547 if (!load_res32_file( *ptr, spec ))
548 *last++ = *ptr; /* not a resource file, keep it in the list */
549 }
550 *last = NULL;
551 break;
552 }
553 }
554
555 /* add input files that look like import libs to the import list */
556 static void load_import_libs( char *argv[] )
557 {
558 char **ptr, **last;
559
560 for (ptr = last = argv; *ptr; ptr++)
561 {
562 if (strendswith( *ptr, ".def" ))
563 add_import_dll( NULL, *ptr );
564 else
565 *last++ = *ptr; /* not an import dll, keep it in the list */
566 }
567 *last = NULL;
568 }
569
570 static int parse_input_file( DLLSPEC *spec )
571 {
572 FILE *input_file = open_input_file( NULL, spec_file_name );
573 char *extension = strrchr( spec_file_name, '.' );
574 int result;
575
576 spec->src_name = xstrdup( input_file_name );
577 if (extension && !strcmp( extension, ".def" ))
578 result = parse_def_file( input_file, spec );
579 else
580 result = parse_spec_file( input_file, spec );
581 close_input_file( input_file );
582 return result;
583 }
584
585
586 /*******************************************************************
587 * main
588 */
589 int main(int argc, char **argv)
590 {
591 DLLSPEC *spec = alloc_dll_spec();
592
593 #ifdef SIGHUP
594 signal( SIGHUP, exit_on_signal );
595 #endif
596 signal( SIGTERM, exit_on_signal );
597 signal( SIGINT, exit_on_signal );
598
599 output_file = stdout;
600 argv = parse_options( argc, argv, spec );
601
602 switch(exec_mode)
603 {
604 case MODE_DLL:
605 if (spec->subsystem != IMAGE_SUBSYSTEM_NATIVE)
606 spec->characteristics |= IMAGE_FILE_DLL;
607 load_resources( argv, spec );
608 load_import_libs( argv );
609 if (!spec_file_name) fatal_error( "missing .spec file\n" );
610 if (!parse_input_file( spec )) break;
611 switch (spec->type)
612 {
613 case SPEC_WIN16:
614 fatal_error( "Win16 specs are not supported in ReactOS version of winebuild\n" );
615 break;
616 case SPEC_WIN32:
617 read_undef_symbols( spec, argv );
618 BuildSpec32File( spec );
619 break;
620 default: assert(0);
621 }
622 break;
623 case MODE_EXE:
624 if (spec->type == SPEC_WIN16) fatal_error( "Cannot build 16-bit exe files\n" );
625 if (!spec->file_name) fatal_error( "executable must be named via the -F option\n" );
626 load_resources( argv, spec );
627 load_import_libs( argv );
628 if (spec_file_name && !parse_input_file( spec )) break;
629 read_undef_symbols( spec, argv );
630 BuildSpec32File( spec );
631 break;
632 case MODE_DEF:
633 if (argv[0]) fatal_error( "file argument '%s' not allowed in this mode\n", argv[0] );
634 if (spec->type == SPEC_WIN16) fatal_error( "Cannot yet build .def file for 16-bit dlls\n" );
635 if (!spec_file_name) fatal_error( "missing .spec file\n" );
636 if (!parse_input_file( spec )) break;
637 BuildDef32File( spec );
638 break;
639 case MODE_RELAY16:
640 fatal_error( "Win16 relays are not supported in ReactOS version of winebuild\n" );
641 break;
642 case MODE_RELAY32:
643 fatal_error( "Win32 relays are not supported in ReactOS version of winebuild\n" );
644 break;
645 case MODE_PEDLL:
646 if (argv[0]) fatal_error( "file argument '%s' not allowed in this mode\n", argv[0] );
647 if (!parse_input_file( spec )) break;
648 BuildPedllFile( spec );
649 break;
650 default:
651 usage(1);
652 break;
653 }
654 if (nb_errors) exit(1);
655 if (output_file_name)
656 {
657 if (fclose( output_file ) < 0) fatal_perror( "fclose" );
658 if (output_file_source_name) assemble_file( output_file_source_name, output_file_name );
659 output_file_name = NULL;
660 }
661 return 0;
662 }