98e7a510fed9686f375f0d076ec87d3112e9de34
[reactos.git] / reactos / tools / widl / widl.c
1 /*
2 * IDL Compiler
3 *
4 * Copyright 2002 Ove Kaaven
5 * based on WRC code by Bertho Stultiens
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 */
21
22 #include "config.h"
23 #include "wine/port.h"
24
25 #include <errno.h>
26 #include <limits.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #ifdef HAVE_UNISTD_H
30 # include <unistd.h>
31 #endif
32 #include <string.h>
33 #include <assert.h>
34 #include <ctype.h>
35 #include <signal.h>
36 #ifdef HAVE_GETOPT_H
37 # include <getopt.h>
38 #endif
39
40 #include "widl.h"
41 #include "utils.h"
42 #include "parser.h"
43 #include "wine/wpp.h"
44 #include "header.h"
45
46 /* future options to reserve characters for: */
47 /* a = alignment of structures */
48 /* A = ACF input filename */
49 /* J = do not search standard include path */
50 /* O = generate interpreted stubs */
51 /* w = select win16/win32 output (?) */
52
53 static const char usage[] =
54 "Usage: widl [options...] infile.idl\n"
55 " or: widl [options...] --dlldata-only name1 [name2...]\n"
56 " -b arch Set the target architecture\n"
57 " -c Generate client stub\n"
58 " -C file Name of client stub file (default is infile_c.c)\n"
59 " -d n Set debug level to 'n'\n"
60 " -D id[=val] Define preprocessor identifier id=val\n"
61 " --dlldata=file Name of the dlldata file (default is dlldata.c)\n"
62 " -E Preprocess only\n"
63 " -h Generate headers\n"
64 " -H file Name of header file (default is infile.h)\n"
65 " -I path Set include search dir to path (multiple -I allowed)\n"
66 " --local-stubs=file Write empty stubs for call_as/local methods to file\n"
67 " -m32, -m64 Set the kind of typelib to build (Win32 or Win64)\n"
68 " -N Do not preprocess input\n"
69 " --oldnames Use old naming conventions\n"
70 " -p Generate proxy\n"
71 " -P file Name of proxy file (default is infile_p.c)\n"
72 " --prefix-all=p Prefix names of client stubs / server functions with 'p'\n"
73 " --prefix-client=p Prefix names of client stubs with 'p'\n"
74 " --prefix-server=p Prefix names of server functions with 'p'\n"
75 " -s Generate server stub\n"
76 " -S file Name of server stub file (default is infile_s.c)\n"
77 " -t Generate typelib\n"
78 " -T file Name of typelib file (default is infile.tlb)\n"
79 " -u Generate interface identifiers file\n"
80 " -U file Name of interface identifiers file (default is infile_i.c)\n"
81 " -V Print version and exit\n"
82 " -W Enable pedantic warnings\n"
83 " --win32 Only generate 32-bit code\n"
84 " --win64 Only generate 64-bit code\n"
85 "Debug level 'n' is a bitmask with following meaning:\n"
86 " * 0x01 Tell which resource is parsed (verbose mode)\n"
87 " * 0x02 Dump internal structures\n"
88 " * 0x04 Create a parser trace (yydebug=1)\n"
89 " * 0x08 Preprocessor messages\n"
90 " * 0x10 Preprocessor lex messages\n"
91 " * 0x20 Preprocessor yacc trace\n"
92 ;
93
94 static const char version_string[] = "Wine IDL Compiler version " PACKAGE_VERSION "\n"
95 "Copyright 2002 Ove Kaaven\n";
96
97 int debuglevel = DEBUGLEVEL_NONE;
98 int parser_debug, yy_flex_debug;
99
100 int pedantic = 0;
101 int do_everything = 1;
102 int preprocess_only = 0;
103 int do_header = 0;
104 int do_typelib = 0;
105 int do_proxies = 0;
106 int do_client = 0;
107 int do_server = 0;
108 int do_idfile = 0;
109 int do_dlldata = 0;
110 int no_preprocess = 0;
111 int old_names = 0;
112 int do_win32 = 1;
113 int do_win64 = 1;
114
115 char *input_name;
116 char *header_name;
117 char *local_stubs_name;
118 char *header_token;
119 char *typelib_name;
120 char *dlldata_name;
121 char *proxy_name;
122 char *proxy_token;
123 char *client_name;
124 char *client_token;
125 char *server_name;
126 char *server_token;
127 char *idfile_name;
128 char *idfile_token;
129 char *temp_name;
130 const char *prefix_client = "";
131 const char *prefix_server = "";
132
133 int line_number = 1;
134
135 FILE *header;
136 FILE *idfile;
137
138 size_t pointer_size = 0;
139 syskind_t typelib_kind = sizeof(void*) == 8 ? SYS_WIN64 : SYS_WIN32;
140
141 time_t now;
142
143 enum {
144 OLDNAMES_OPTION = CHAR_MAX + 1,
145 DLLDATA_OPTION,
146 DLLDATA_ONLY_OPTION,
147 LOCAL_STUBS_OPTION,
148 PREFIX_ALL_OPTION,
149 PREFIX_CLIENT_OPTION,
150 PREFIX_SERVER_OPTION,
151 WIN32_OPTION,
152 WIN64_OPTION
153 };
154
155 static const char short_options[] =
156 "b:cC:d:D:EhH:I:m:NpP:sS:tT:uU:VW";
157 static const struct option long_options[] = {
158 { "dlldata", 1, 0, DLLDATA_OPTION },
159 { "dlldata-only", 0, 0, DLLDATA_ONLY_OPTION },
160 { "local-stubs", 1, 0, LOCAL_STUBS_OPTION },
161 { "oldnames", 0, 0, OLDNAMES_OPTION },
162 { "prefix-all", 1, 0, PREFIX_ALL_OPTION },
163 { "prefix-client", 1, 0, PREFIX_CLIENT_OPTION },
164 { "prefix-server", 1, 0, PREFIX_SERVER_OPTION },
165 { "win32", 0, 0, WIN32_OPTION },
166 { "win64", 0, 0, WIN64_OPTION },
167 { 0, 0, 0, 0 }
168 };
169
170 static void rm_tempfile(void);
171
172 static char *make_token(const char *name)
173 {
174 char *token;
175 char *slash;
176 int i;
177
178 slash = strrchr(name, '/');
179 if(!slash)
180 slash = strrchr(name, '\\');
181
182 if (slash) name = slash + 1;
183
184 token = xstrdup(name);
185 for (i=0; token[i]; i++) {
186 if (!isalnum(token[i])) token[i] = '_';
187 else token[i] = toupper(token[i]);
188 }
189 return token;
190 }
191
192 /* duplicate a basename into a valid C token */
193 static char *dup_basename_token(const char *name, const char *ext)
194 {
195 char *p, *ret = dup_basename( name, ext );
196 /* map invalid characters to '_' */
197 for (p = ret; *p; p++) if (!isalnum(*p)) *p = '_';
198 return ret;
199 }
200
201 static void add_widl_version_define(void)
202 {
203 unsigned int version;
204 const char *p = PACKAGE_VERSION;
205
206 /* major */
207 version = atoi(p) * 0x10000;
208 p = strchr(p, '.');
209
210 /* minor */
211 if (p)
212 {
213 version += atoi(p + 1) * 0x100;
214 p = strchr(p + 1, '.');
215 }
216
217 /* build */
218 if (p)
219 version += atoi(p + 1);
220
221 if (version != 0)
222 {
223 char version_str[11];
224 snprintf(version_str, sizeof(version_str), "0x%x", version);
225 wpp_add_define("__WIDL__", version_str);
226 }
227 else
228 wpp_add_define("__WIDL__", NULL);
229 }
230
231 /* set the target platform */
232 static void set_target( const char *target )
233 {
234 static const struct
235 {
236 const char *name;
237 syskind_t kind;
238 } cpu_names[] =
239 {
240 { "i386", SYS_WIN32 },
241 { "i486", SYS_WIN32 },
242 { "i586", SYS_WIN32 },
243 { "i686", SYS_WIN32 },
244 { "i786", SYS_WIN32 },
245 { "x86_64", SYS_WIN64 },
246 { "sparc", SYS_WIN32 },
247 { "alpha", SYS_WIN32 },
248 { "powerpc", SYS_WIN32 }
249 };
250
251 unsigned int i;
252 char *p, *spec = xstrdup( target );
253
254 /* target specification is in the form CPU-MANUFACTURER-OS or CPU-MANUFACTURER-KERNEL-OS */
255
256 if (!(p = strchr( spec, '-' ))) error( "Invalid target specification '%s'\n", target );
257 *p++ = 0;
258 for (i = 0; i < sizeof(cpu_names)/sizeof(cpu_names[0]); i++)
259 {
260 if (!strcmp( cpu_names[i].name, spec ))
261 {
262 typelib_kind = cpu_names[i].kind;
263 free( spec );
264 return;
265 }
266 }
267 error( "Unrecognized CPU '%s'\n", spec );
268 }
269
270 /* clean things up when aborting on a signal */
271 static void exit_on_signal( int sig )
272 {
273 exit(1); /* this will call the atexit functions */
274 }
275
276 static void set_everything(int x)
277 {
278 do_header = x;
279 do_typelib = x;
280 do_proxies = x;
281 do_client = x;
282 do_server = x;
283 do_idfile = x;
284 do_dlldata = x;
285 }
286
287 void start_cplusplus_guard(FILE *fp)
288 {
289 fprintf(fp, "#ifdef __cplusplus\n");
290 fprintf(fp, "extern \"C\" {\n");
291 fprintf(fp, "#endif\n\n");
292 }
293
294 void end_cplusplus_guard(FILE *fp)
295 {
296 fprintf(fp, "#ifdef __cplusplus\n");
297 fprintf(fp, "}\n");
298 fprintf(fp, "#endif\n\n");
299 }
300
301 typedef struct
302 {
303 char *filename;
304 struct list link;
305 } filename_node_t;
306
307 static void add_filename_node(struct list *list, const char *name)
308 {
309 filename_node_t *node = xmalloc(sizeof *node);
310 node->filename = dup_basename( name, ".idl" );
311 list_add_tail(list, &node->link);
312 }
313
314 static void free_filename_nodes(struct list *list)
315 {
316 filename_node_t *node, *next;
317 LIST_FOR_EACH_ENTRY_SAFE(node, next, list, filename_node_t, link) {
318 list_remove(&node->link);
319 free(node->filename);
320 free(node);
321 }
322 }
323
324 static void write_dlldata_list(struct list *filenames)
325 {
326 FILE *dlldata;
327 filename_node_t *node;
328
329 dlldata = fopen(dlldata_name, "w");
330 if (!dlldata)
331 error("couldn't open %s: %s\n", dlldata_name, strerror(errno));
332
333 fprintf(dlldata, "/*** Autogenerated by WIDL %s ", PACKAGE_VERSION);
334 fprintf(dlldata, "- Do not edit ***/\n\n");
335 fprintf(dlldata, "#include <objbase.h>\n");
336 fprintf(dlldata, "#include <rpcproxy.h>\n\n");
337 start_cplusplus_guard(dlldata);
338
339 LIST_FOR_EACH_ENTRY(node, filenames, filename_node_t, link)
340 fprintf(dlldata, "EXTERN_PROXY_FILE(%s)\n", node->filename);
341
342 fprintf(dlldata, "\nPROXYFILE_LIST_START\n");
343 fprintf(dlldata, "/* Start of list */\n");
344 LIST_FOR_EACH_ENTRY(node, filenames, filename_node_t, link)
345 fprintf(dlldata, " REFERENCE_PROXY_FILE(%s),\n", node->filename);
346 fprintf(dlldata, "/* End of list */\n");
347 fprintf(dlldata, "PROXYFILE_LIST_END\n\n");
348
349 fprintf(dlldata, "DLLDATA_ROUTINES(aProxyFileList, GET_DLL_CLSID)\n\n");
350 end_cplusplus_guard(dlldata);
351 fclose(dlldata);
352 }
353
354 static char *eat_space(char *s)
355 {
356 while (isspace((unsigned char) *s))
357 ++s;
358 return s;
359 }
360
361 void write_dlldata(const statement_list_t *stmts)
362 {
363 struct list filenames = LIST_INIT(filenames);
364 filename_node_t *node;
365 FILE *dlldata;
366
367 if (!do_dlldata || !need_proxy_file(stmts))
368 return;
369
370 dlldata = fopen(dlldata_name, "r");
371 if (dlldata) {
372 static char marker[] = "REFERENCE_PROXY_FILE";
373 char *line = NULL;
374 size_t len = 0;
375
376 while (widl_getline(&line, &len, dlldata)) {
377 char *start, *end;
378 start = eat_space(line);
379 if (strncmp(start, marker, sizeof marker - 1) == 0) {
380 start = eat_space(start + sizeof marker - 1);
381 if (*start != '(')
382 continue;
383 end = start = eat_space(start + 1);
384 while (*end && *end != ')')
385 ++end;
386 if (*end != ')')
387 continue;
388 while (isspace((unsigned char) end[-1]))
389 --end;
390 *end = '\0';
391 if (start < end)
392 add_filename_node(&filenames, start);
393 }
394 }
395
396 if (ferror(dlldata))
397 error("couldn't read from %s: %s\n", dlldata_name, strerror(errno));
398
399 free(line);
400 fclose(dlldata);
401 }
402
403 LIST_FOR_EACH_ENTRY(node, &filenames, filename_node_t, link)
404 if (strcmp(proxy_token, node->filename) == 0) {
405 /* We're already in the list, no need to regenerate this file. */
406 free_filename_nodes(&filenames);
407 return;
408 }
409
410 add_filename_node(&filenames, proxy_token);
411 write_dlldata_list(&filenames);
412 free_filename_nodes(&filenames);
413 }
414
415 static void write_id_data_stmts(const statement_list_t *stmts)
416 {
417 const statement_t *stmt;
418 if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
419 {
420 if (stmt->type == STMT_TYPE)
421 {
422 const type_t *type = stmt->u.type;
423 if (type_get_type(type) == TYPE_INTERFACE)
424 {
425 const UUID *uuid;
426 if (!is_object(type->attrs) && !is_attr(type->attrs, ATTR_DISPINTERFACE))
427 continue;
428 uuid = get_attrp(type->attrs, ATTR_UUID);
429 write_guid(idfile, is_attr(type->attrs, ATTR_DISPINTERFACE) ? "DIID" : "IID",
430 type->name, uuid);
431 }
432 else if (type_get_type(type) == TYPE_COCLASS)
433 {
434 const UUID *uuid = get_attrp(type->attrs, ATTR_UUID);
435 write_guid(idfile, "CLSID", type->name, uuid);
436 }
437 }
438 else if (stmt->type == STMT_LIBRARY)
439 {
440 const UUID *uuid = get_attrp(stmt->u.lib->attrs, ATTR_UUID);
441 write_guid(idfile, "LIBID", stmt->u.lib->name, uuid);
442 write_id_data_stmts(stmt->u.lib->stmts);
443 }
444 }
445 }
446
447 void write_id_data(const statement_list_t *stmts)
448 {
449 if (!do_idfile) return;
450
451 idfile_token = make_token(idfile_name);
452
453 idfile = fopen(idfile_name, "w");
454 if (! idfile) {
455 error("Could not open %s for output\n", idfile_name);
456 return;
457 }
458
459 fprintf(idfile, "/*** Autogenerated by WIDL %s ", PACKAGE_VERSION);
460 fprintf(idfile, "from %s - Do not edit ***/\n\n", input_name);
461 fprintf(idfile, "#include <rpc.h>\n");
462 fprintf(idfile, "#include <rpcndr.h>\n\n");
463 fprintf(idfile, "#include <initguid.h>\n\n");
464 start_cplusplus_guard(idfile);
465
466 write_id_data_stmts(stmts);
467
468 fprintf(idfile, "\n");
469 end_cplusplus_guard(idfile);
470
471 fclose(idfile);
472 }
473
474 int main(int argc,char *argv[])
475 {
476 extern char* optarg;
477 extern int optind;
478 int optc;
479 int ret = 0;
480 int opti = 0;
481
482 signal( SIGTERM, exit_on_signal );
483 signal( SIGINT, exit_on_signal );
484 #ifdef SIGHUP
485 signal( SIGHUP, exit_on_signal );
486 #endif
487
488 now = time(NULL);
489
490 while((optc = getopt_long(argc, argv, short_options, long_options, &opti)) != EOF) {
491 switch(optc) {
492 case DLLDATA_OPTION:
493 dlldata_name = xstrdup(optarg);
494 break;
495 case DLLDATA_ONLY_OPTION:
496 do_everything = 0;
497 do_dlldata = 1;
498 break;
499 case LOCAL_STUBS_OPTION:
500 do_everything = 0;
501 local_stubs_name = xstrdup(optarg);
502 break;
503 case OLDNAMES_OPTION:
504 old_names = 1;
505 break;
506 case PREFIX_ALL_OPTION:
507 prefix_client = xstrdup(optarg);
508 prefix_server = xstrdup(optarg);
509 break;
510 case PREFIX_CLIENT_OPTION:
511 prefix_client = xstrdup(optarg);
512 break;
513 case PREFIX_SERVER_OPTION:
514 prefix_server = xstrdup(optarg);
515 break;
516 case WIN32_OPTION:
517 do_win32 = 1;
518 do_win64 = 0;
519 break;
520 case WIN64_OPTION:
521 do_win32 = 0;
522 do_win64 = 1;
523 break;
524 case 'b':
525 set_target( optarg );
526 break;
527 case 'c':
528 do_everything = 0;
529 do_client = 1;
530 break;
531 case 'C':
532 client_name = xstrdup(optarg);
533 break;
534 case 'd':
535 debuglevel = strtol(optarg, NULL, 0);
536 break;
537 case 'D':
538 wpp_add_cmdline_define(optarg);
539 break;
540 case 'E':
541 do_everything = 0;
542 preprocess_only = 1;
543 break;
544 case 'h':
545 do_everything = 0;
546 do_header = 1;
547 break;
548 case 'H':
549 header_name = xstrdup(optarg);
550 break;
551 case 'I':
552 wpp_add_include_path(optarg);
553 break;
554 case 'm':
555 if (!strcmp( optarg, "32" )) typelib_kind = SYS_WIN32;
556 else if (!strcmp( optarg, "64" )) typelib_kind = SYS_WIN64;
557 else error( "Invalid -m argument '%s'\n", optarg );
558 break;
559 case 'N':
560 no_preprocess = 1;
561 break;
562 case 'p':
563 do_everything = 0;
564 do_proxies = 1;
565 break;
566 case 'P':
567 proxy_name = xstrdup(optarg);
568 break;
569 case 's':
570 do_everything = 0;
571 do_server = 1;
572 break;
573 case 'S':
574 server_name = xstrdup(optarg);
575 break;
576 case 't':
577 do_everything = 0;
578 do_typelib = 1;
579 break;
580 case 'T':
581 typelib_name = xstrdup(optarg);
582 break;
583 case 'u':
584 do_everything = 0;
585 do_idfile = 1;
586 break;
587 case 'U':
588 idfile_name = xstrdup(optarg);
589 break;
590 case 'V':
591 printf("%s", version_string);
592 return 0;
593 case 'W':
594 pedantic = 1;
595 break;
596 default:
597 fprintf(stderr, "%s", usage);
598 return 1;
599 }
600 }
601
602 if(do_everything) {
603 set_everything(TRUE);
604 }
605
606 if (!dlldata_name && do_dlldata)
607 dlldata_name = xstrdup("dlldata.c");
608
609 if(optind < argc) {
610 if (do_dlldata && !do_everything) {
611 struct list filenames = LIST_INIT(filenames);
612 for ( ; optind < argc; ++optind)
613 add_filename_node(&filenames, argv[optind]);
614
615 write_dlldata_list(&filenames);
616 free_filename_nodes(&filenames);
617 return 0;
618 }
619 else if (optind != argc - 1) {
620 fprintf(stderr, "%s", usage);
621 return 1;
622 }
623 else
624 input_name = xstrdup(argv[optind]);
625 }
626 else {
627 fprintf(stderr, "%s", usage);
628 return 1;
629 }
630
631 if(debuglevel)
632 {
633 setbuf(stdout,0);
634 setbuf(stderr,0);
635 }
636
637 parser_debug = debuglevel & DEBUGLEVEL_TRACE ? 1 : 0;
638 yy_flex_debug = debuglevel & DEBUGLEVEL_TRACE ? 1 : 0;
639
640 wpp_set_debug( (debuglevel & DEBUGLEVEL_PPLEX) != 0,
641 (debuglevel & DEBUGLEVEL_PPTRACE) != 0,
642 (debuglevel & DEBUGLEVEL_PPMSG) != 0 );
643
644 if (!header_name) {
645 header_name = dup_basename(input_name, ".idl");
646 strcat(header_name, ".h");
647 }
648
649 if (!typelib_name && do_typelib) {
650 typelib_name = dup_basename(input_name, ".idl");
651 strcat(typelib_name, ".tlb");
652 }
653
654 if (!proxy_name && do_proxies) {
655 proxy_name = dup_basename(input_name, ".idl");
656 strcat(proxy_name, "_p.c");
657 }
658
659 if (!client_name && do_client) {
660 client_name = dup_basename(input_name, ".idl");
661 strcat(client_name, "_c.c");
662 }
663
664 if (!server_name && do_server) {
665 server_name = dup_basename(input_name, ".idl");
666 strcat(server_name, "_s.c");
667 }
668
669 if (!idfile_name && do_idfile) {
670 idfile_name = dup_basename(input_name, ".idl");
671 strcat(idfile_name, "_i.c");
672 }
673
674 if (do_proxies) proxy_token = dup_basename_token(proxy_name,"_p.c");
675 if (do_client) client_token = dup_basename_token(client_name,"_c.c");
676 if (do_server) server_token = dup_basename_token(server_name,"_s.c");
677
678 add_widl_version_define();
679
680 atexit(rm_tempfile);
681 if (!no_preprocess)
682 {
683 chat("Starting preprocess\n");
684
685 if (!preprocess_only)
686 {
687 ret = wpp_parse_temp( input_name, header_name, &temp_name );
688 }
689 else
690 {
691 ret = wpp_parse( input_name, stdout );
692 }
693
694 if(ret) exit(1);
695 if(preprocess_only) exit(0);
696 if(!(parser_in = fopen(temp_name, "r"))) {
697 fprintf(stderr, "Could not open %s for input\n", temp_name);
698 return 1;
699 }
700 }
701 else {
702 if(!(parser_in = fopen(input_name, "r"))) {
703 fprintf(stderr, "Could not open %s for input\n", input_name);
704 return 1;
705 }
706 }
707
708 header_token = make_token(header_name);
709
710 init_types();
711 ret = parser_parse();
712
713 fclose(parser_in);
714
715 if(ret) {
716 exit(1);
717 }
718
719 /* Everything has been done successfully, don't delete any files. */
720 set_everything(FALSE);
721 local_stubs_name = NULL;
722
723 return 0;
724 }
725
726 static void rm_tempfile(void)
727 {
728 abort_import();
729 if(temp_name)
730 unlink(temp_name);
731 if (do_header)
732 unlink(header_name);
733 if (local_stubs_name)
734 unlink(local_stubs_name);
735 if (do_client)
736 unlink(client_name);
737 if (do_server)
738 unlink(server_name);
739 if (do_idfile)
740 unlink(idfile_name);
741 if (do_proxies)
742 unlink(proxy_name);
743 if (do_typelib)
744 unlink(typelib_name);
745 }