8148ba716772217b5f1dd36da71011cc085552c0
[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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 */
24
25 #include "config.h"
26
27 #include <assert.h>
28 #include <stdio.h>
29 #include <signal.h>
30 #include <errno.h>
31 #include <string.h>
32 #include <stdarg.h>
33 #include <ctype.h>
34 #ifdef HAVE_GETOPT_H
35 # include <getopt.h>
36 #endif
37
38 #include "winglue.h"
39 #include "build.h"
40
41 int UsePIC = 0;
42 int nb_debug_channels = 0;
43 int nb_lib_paths = 0;
44 int nb_errors = 0;
45 int display_warnings = 0;
46 int kill_at = 0;
47
48 /* we only support relay debugging on i386 */
49 #ifdef __i386__
50 int debugging = 1;
51 #else
52 int debugging = 0;
53 #endif
54
55 char **debug_channels = NULL;
56 char **lib_path = NULL;
57
58 char *input_file_name = NULL;
59 char *spec_file_name = NULL;
60 const char *output_file_name = NULL;
61
62 char *ld_command = "ld";
63 char *nm_command = "nm";
64
65 static FILE *output_file;
66 static const char *current_src_dir;
67 static int nb_res_files;
68 static char **res_files;
69
70 /* execution mode */
71 enum exec_mode_values
72 {
73 MODE_NONE,
74 MODE_DLL,
75 MODE_EXE,
76 MODE_DEF,
77 MODE_DEBUG,
78 MODE_RELAY16,
79 MODE_RELAY32,
80 MODE_PEDLL
81 };
82
83 static enum exec_mode_values exec_mode = MODE_NONE;
84
85 /* set the dll file name from the input file name */
86 static void set_dll_file_name( const char *name, DLLSPEC *spec )
87 {
88 char *p;
89
90 if (spec->file_name) return;
91
92 if ((p = strrchr( name, '\\' ))) name = p + 1;
93 if ((p = strrchr( name, '/' ))) name = p + 1;
94 spec->file_name = xmalloc( strlen(name) + 5 );
95 strcpy( spec->file_name, name );
96 if ((p = strrchr( spec->file_name, '.' )))
97 {
98 if (!strcmp( p, ".spec" ) || !strcmp( p, ".def" )) *p = 0;
99 }
100 }
101
102 /* set the dll subsystem */
103 static void set_subsystem( const char *subsystem, DLLSPEC *spec )
104 {
105 char *major, *minor, *str = xstrdup( subsystem );
106
107 if ((major = strchr( str, ':' ))) *major++ = 0;
108 if (!strcmp( str, "native" )) spec->subsystem = IMAGE_SUBSYSTEM_NATIVE;
109 else if (!strcmp( str, "windows" )) spec->subsystem = IMAGE_SUBSYSTEM_WINDOWS_GUI;
110 else if (!strcmp( str, "console" )) spec->subsystem = IMAGE_SUBSYSTEM_WINDOWS_CUI;
111 else fatal_error( "Invalid subsystem name '%s'\n", subsystem );
112 if (major)
113 {
114 if ((minor = strchr( major, '.' )))
115 {
116 *minor++ = 0;
117 spec->subsystem_minor = atoi( minor );
118 }
119 spec->subsystem_major = atoi( major );
120 }
121 free( str );
122 }
123
124 /* cleanup on program exit */
125 static void cleanup(void)
126 {
127 if (output_file_name) unlink( output_file_name );
128 }
129
130 /* clean things up when aborting on a signal */
131 static void exit_on_signal( int sig )
132 {
133 exit(1); /* this will call atexit functions */
134 }
135
136 /*******************************************************************
137 * command-line option handling
138 */
139 static const char usage_str[] =
140 "Usage: winebuild [OPTIONS] [FILES]\n\n"
141 "Options:\n"
142 " -C --source-dir=DIR Look for source files in DIR\n"
143 " -d --delay-lib=LIB Import the specified library in delayed mode\n"
144 " -D SYM Ignored for C flags compatibility\n"
145 " -E --export=FILE Export the symbols defined in the .spec or .def file\n"
146 " -e --entry=FUNC Set the DLL entry point function (default: DllMain)\n"
147 " -f FLAGS Compiler flags (only -fPIC is supported)\n"
148 " -F --filename=DLLFILE Set the DLL filename (default: from input file name)\n"
149 " -h --help Display this help message\n"
150 " -H --heap=SIZE Set the heap size for a Win16 dll\n"
151 " -i --ignore=SYM[,SYM] Ignore specified symbols when resolving imports\n"
152 " -I DIR Ignored for C flags compatibility\n"
153 " -k --kill-at Kill stdcall decorations in generated .def files\n"
154 " -K FLAGS Compiler flags (only -KPIC is supported)\n"
155 " --ld-cmd=LD Command to use for linking (default: ld)\n"
156 " -l --library=LIB Import the specified library\n"
157 " -L --library-path=DIR Look for imports libraries in DIR\n"
158 " -M --main-module=MODULE Set the name of the main module for a Win16 dll\n"
159 " --nm-cmd=NM Command to use to get undefined symbols (default: nm)\n"
160 " -N --dll-name=DLLNAME Set the DLL name (default: from input file name)\n"
161 " -o --output=NAME Set the output file name (default: stdout)\n"
162 " -r --res=RSRC.RES Load resources from RSRC.RES\n"
163 " --subsystem=SUBSYS Set the subsystem (one of native, windows, console)\n"
164 " --version Print the version and exit\n"
165 " -w --warnings Turn on warnings\n"
166 "\nMode options:\n"
167 " --dll Build a .c file from a .spec or .def file\n"
168 " --def Build a .def file from a .spec file\n"
169 " --exe Build a .c file for an executable\n"
170 " --debug [FILES] Build a .c file with the debug channels declarations\n"
171 " --relay16 Build the 16-bit relay assembly routines\n"
172 " --relay32 Build the 32-bit relay assembly routines\n"
173 " --pedll Build a .c file for PE dll\n\n"
174 "The mode options are mutually exclusive; you must specify one and only one.\n\n";
175
176 enum long_options_values
177 {
178 LONG_OPT_DLL = 1,
179 LONG_OPT_DEF,
180 LONG_OPT_EXE,
181 LONG_OPT_DEBUG,
182 LONG_OPT_LDCMD,
183 LONG_OPT_NMCMD,
184 LONG_OPT_RELAY16,
185 LONG_OPT_RELAY32,
186 LONG_OPT_SUBSYSTEM,
187 LONG_OPT_VERSION,
188 LONG_OPT_PEDLL
189 };
190
191 static const char short_options[] = "C:D:E:F:H:I:K:L:M:N:d:e:f:hi:kl:m:o:r:w";
192
193 static const struct option long_options[] =
194 {
195 { "dll", 0, 0, LONG_OPT_DLL },
196 { "def", 0, 0, LONG_OPT_DEF },
197 { "exe", 0, 0, LONG_OPT_EXE },
198 { "debug", 0, 0, LONG_OPT_DEBUG },
199 { "ld-cmd", 1, 0, LONG_OPT_LDCMD },
200 { "nm-cmd", 1, 0, LONG_OPT_NMCMD },
201 { "relay16", 0, 0, LONG_OPT_RELAY16 },
202 { "relay32", 0, 0, LONG_OPT_RELAY32 },
203 { "subsystem",1, 0, LONG_OPT_SUBSYSTEM },
204 { "version", 0, 0, LONG_OPT_VERSION },
205 { "pedll", 1, 0, LONG_OPT_PEDLL },
206 /* aliases for short options */
207 { "source-dir", 1, 0, 'C' },
208 { "delay-lib", 1, 0, 'd' },
209 { "export", 1, 0, 'E' },
210 { "entry", 1, 0, 'e' },
211 { "filename", 1, 0, 'F' },
212 { "help", 0, 0, 'h' },
213 { "heap", 1, 0, 'H' },
214 { "ignore", 1, 0, 'i' },
215 { "kill-at", 0, 0, 'k' },
216 { "library", 1, 0, 'l' },
217 { "library-path", 1, 0, 'L' },
218 { "main-module", 1, 0, 'M' },
219 { "dll-name", 1, 0, 'N' },
220 { "output", 1, 0, 'o' },
221 { "res", 1, 0, 'r' },
222 { "warnings", 0, 0, 'w' },
223 { NULL, 0, 0, 0 }
224 };
225
226 static void usage( int exit_code )
227 {
228 fprintf( stderr, "%s", usage_str );
229 exit( exit_code );
230 }
231
232 static void set_exec_mode( enum exec_mode_values mode )
233 {
234 if (exec_mode != MODE_NONE) usage(1);
235 exec_mode = mode;
236 }
237
238 /* parse options from the argv array and remove all the recognized ones */
239 static char **parse_options( int argc, char **argv, DLLSPEC *spec )
240 {
241 char *p;
242 int optc;
243
244 while ((optc = getopt_long( argc, argv, short_options, long_options, NULL )) != -1)
245 {
246 switch(optc)
247 {
248 case 'C':
249 current_src_dir = optarg;
250 break;
251 case 'D':
252 /* ignored */
253 break;
254 case 'E':
255 spec_file_name = xstrdup( optarg );
256 set_dll_file_name( optarg, spec );
257 break;
258 case 'F':
259 spec->file_name = xstrdup( optarg );
260 break;
261 case 'H':
262 if (!isdigit(optarg[0]))
263 fatal_error( "Expected number argument with -H option instead of '%s'\n", optarg );
264 spec->heap_size = atoi(optarg);
265 if (spec->heap_size > 65535)
266 fatal_error( "Invalid heap size %d, maximum is 65535\n", spec->heap_size );
267 break;
268 case 'I':
269 /* ignored */
270 break;
271 case 'K':
272 /* ignored, because cc generates correct code. */
273 break;
274 case 'L':
275 lib_path = xrealloc( lib_path, (nb_lib_paths+1) * sizeof(*lib_path) );
276 lib_path[nb_lib_paths++] = xstrdup( optarg );
277 break;
278 case 'M':
279 spec->owner_name = xstrdup( optarg );
280 spec->type = SPEC_WIN16;
281 break;
282 case 'N':
283 spec->dll_name = xstrdup( optarg );
284 break;
285 case 'd':
286 add_import_dll( optarg, 1 );
287 break;
288 case 'e':
289 spec->init_func = xstrdup( optarg );
290 if ((p = strchr( spec->init_func, '@' ))) *p = 0; /* kill stdcall decoration */
291 break;
292 case 'f':
293 if (!strcmp( optarg, "PIC") || !strcmp( optarg, "pic")) UsePIC = 1;
294 /* ignore all other flags */
295 break;
296 case 'h':
297 usage(0);
298 break;
299 case 'i':
300 {
301 char *str = xstrdup( optarg );
302 char *token = strtok( str, "," );
303 while (token)
304 {
305 add_ignore_symbol( token );
306 token = strtok( NULL, "," );
307 }
308 free( str );
309 }
310 break;
311 case 'k':
312 kill_at = 1;
313 break;
314 case 'l':
315 add_import_dll( optarg, 0 );
316 break;
317 case 'o':
318 if (unlink( optarg ) == -1 && errno != ENOENT)
319 fatal_error( "Unable to create output file '%s'\n", optarg );
320 if (!(output_file = fopen( optarg, "w" )))
321 fatal_error( "Unable to create output file '%s'\n", optarg );
322 output_file_name = xstrdup(optarg);
323 atexit( cleanup ); /* make sure we remove the output file on exit */
324 break;
325 case 'r':
326 res_files = xrealloc( res_files, (nb_res_files+1) * sizeof(*res_files) );
327 res_files[nb_res_files++] = xstrdup( optarg );
328 break;
329 case 'w':
330 display_warnings = 1;
331 break;
332 case LONG_OPT_DLL:
333 set_exec_mode( MODE_DLL );
334 break;
335 case LONG_OPT_DEF:
336 set_exec_mode( MODE_DEF );
337 break;
338 case LONG_OPT_EXE:
339 set_exec_mode( MODE_EXE );
340 if (!spec->subsystem) spec->subsystem = IMAGE_SUBSYSTEM_WINDOWS_GUI;
341 break;
342 case LONG_OPT_DEBUG:
343 set_exec_mode( MODE_DEBUG );
344 break;
345 case LONG_OPT_LDCMD:
346 ld_command = xstrdup( optarg );
347 break;
348 case LONG_OPT_NMCMD:
349 nm_command = xstrdup( optarg );
350 break;
351 case LONG_OPT_RELAY16:
352 set_exec_mode( MODE_RELAY16 );
353 break;
354 case LONG_OPT_RELAY32:
355 set_exec_mode( MODE_RELAY32 );
356 break;
357 case LONG_OPT_SUBSYSTEM:
358 set_subsystem( optarg, spec );
359 break;
360 case LONG_OPT_VERSION:
361 printf( "winebuild version " PACKAGE_VERSION "\n" );
362 exit(0);
363 case LONG_OPT_PEDLL:
364 set_exec_mode( MODE_PEDLL );
365 spec_file_name = xstrdup( optarg );
366 set_dll_file_name( optarg, spec );
367 break;
368 case '?':
369 usage(1);
370 break;
371 }
372 }
373
374 if (spec->file_name && !strchr( spec->file_name, '.' ))
375 strcat( spec->file_name, exec_mode == MODE_EXE ? ".exe" : ".dll" );
376
377 return &argv[optind];
378 }
379
380
381 /* load all specified resource files */
382 static void load_resources( char *argv[], DLLSPEC *spec )
383 {
384 int i;
385 char **ptr, **last;
386
387 switch (spec->type)
388 {
389 case SPEC_WIN16:
390 for (i = 0; i < nb_res_files; i++) load_res16_file( res_files[i], spec );
391 break;
392
393 case SPEC_WIN32:
394 for (i = 0; i < nb_res_files; i++)
395 {
396 if (!load_res32_file( res_files[i], spec ))
397 fatal_error( "%s is not a valid Win32 resource file\n", res_files[i] );
398 }
399
400 /* load any resource file found in the remaining arguments */
401 for (ptr = last = argv; *ptr; ptr++)
402 {
403 if (!load_res32_file( *ptr, spec ))
404 *last++ = *ptr; /* not a resource file, keep it in the list */
405 }
406 *last = NULL;
407 break;
408 }
409 }
410
411 static int parse_input_file( DLLSPEC *spec )
412 {
413 FILE *input_file = open_input_file( NULL, spec_file_name );
414 char *extension = strrchr( spec_file_name, '.' );
415 int result;
416
417 if (extension && !strcmp( extension, ".def" ))
418 result = parse_def_file( input_file, spec );
419 else
420 result = parse_spec_file( input_file, spec );
421 close_input_file( input_file );
422 return result;
423 }
424
425
426 /*******************************************************************
427 * main
428 */
429 int main(int argc, char **argv)
430 {
431 DLLSPEC *spec = alloc_dll_spec();
432
433 #ifdef SIGHUP
434 signal( SIGHUP, exit_on_signal );
435 #endif
436 signal( SIGTERM, exit_on_signal );
437 signal( SIGINT, exit_on_signal );
438
439 output_file = stdout;
440 argv = parse_options( argc, argv, spec );
441
442 switch(exec_mode)
443 {
444 case MODE_DLL:
445 spec->characteristics |= IMAGE_FILE_DLL;
446 load_resources( argv, spec );
447 if (!spec_file_name) fatal_error( "missing .spec file\n" );
448 if (!parse_input_file( spec )) break;
449 switch (spec->type)
450 {
451 case SPEC_WIN16:
452 fatal_error( "Win16 specs are not supported in ReactOS version of winebuild\n" );
453 break;
454 case SPEC_WIN32:
455 read_undef_symbols( argv );
456 BuildSpec32File( output_file, spec );
457 break;
458 default: assert(0);
459 }
460 break;
461 case MODE_EXE:
462 if (spec->type == SPEC_WIN16) fatal_error( "Cannot build 16-bit exe files\n" );
463 if (!spec->file_name) fatal_error( "executable must be named via the -F option\n" );
464 load_resources( argv, spec );
465 if (spec_file_name && !parse_input_file( spec )) break;
466 read_undef_symbols( argv );
467 BuildSpec32File( output_file, spec );
468 break;
469 case MODE_DEF:
470 if (argv[0]) fatal_error( "file argument '%s' not allowed in this mode\n", argv[0] );
471 if (spec->type == SPEC_WIN16) fatal_error( "Cannot yet build .def file for 16-bit dlls\n" );
472 if (!spec_file_name) fatal_error( "missing .spec file\n" );
473 if (!parse_input_file( spec )) break;
474 BuildDef32File( output_file, spec );
475 break;
476 case MODE_DEBUG:
477 BuildDebugFile( output_file, current_src_dir, argv );
478 break;
479 case MODE_RELAY16:
480 fatal_error( "Win16 relays are not supported in ReactOS version of winebuild\n" );
481 break;
482 case MODE_RELAY32:
483 fatal_error( "Win32 relays are not supported in ReactOS version of winebuild\n" );
484 break;
485 case MODE_PEDLL:
486 if (argv[0]) fatal_error( "file argument '%s' not allowed in this mode\n", argv[0] );
487 if (!parse_input_file( spec )) break;
488 BuildPedllFile( output_file, spec );
489 break;
490 default:
491 usage(1);
492 break;
493 }
494 if (nb_errors) exit(1);
495 if (output_file_name)
496 {
497 fclose( output_file );
498 output_file_name = NULL;
499 }
500 return 0;
501 }