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 "   -c          Generate client stub\n"
  57 "   -C file     Name of client stub file (default is infile_c.c)\n"
  58 "   -d n        Set debug level to 'n'\n"
  59 "   -D id[=val] Define preprocessor identifier id=val\n"
  60 "   --dlldata=file  Name of the dlldata file (default is dlldata.c)\n"
  61 "   -E          Preprocess only\n"
  62 "   -h          Generate headers\n"
  63 "   -H file     Name of header file (default is infile.h)\n"
  64 "   -I path     Set include search dir to path (multiple -I allowed)\n"
  65 "   --local-stubs=file  Write empty stubs for call_as/local methods to file\n"
  66 "   -N          Do not preprocess input\n"
  67 "   --oldnames  Use old naming conventions\n"
  68 "   -p          Generate proxy\n"
  69 "   -P file     Name of proxy file (default is infile_p.c)\n"
  70 "   --prefix-all=p  Prefix names of client stubs / server functions with 'p'\n"
  71 "   --prefix-client=p  Prefix names of client stubs with 'p'\n"
  72 "   --prefix-server=p  Prefix names of server functions with 'p'\n"
  73 "   -s          Generate server stub\n"
  74 "   -S file     Name of server stub file (default is infile_s.c)\n"
  75 "   -t          Generate typelib\n"
  76 "   -T file     Name of typelib file (default is infile.tlb)\n"
  77 "   -u          Generate interface identifiers file\n"
  78 "   -U file     Name of interface identifiers file (default is infile_i.c)\n"
  79 "   -V          Print version and exit\n"
  80 "   -W          Enable pedantic warnings\n"
  81 "Debug level 'n' is a bitmask with following meaning:\n"
  82 "    * 0x01 Tell which resource is parsed (verbose mode)\n"
  83 "    * 0x02 Dump internal structures\n"
  84 "    * 0x04 Create a parser trace (yydebug=1)\n"
  85 "    * 0x08 Preprocessor messages\n"
  86 "    * 0x10 Preprocessor lex messages\n"
  87 "    * 0x20 Preprocessor yacc trace\n"
  88 ;
  89
  90 static const char version_string[] = "Wine IDL Compiler version " PACKAGE_VERSION "\n"
  91                         "Copyright 2002 Ove Kaaven\n";
  92
  93 int win32 = 1;
  94 int debuglevel = DEBUGLEVEL_NONE;
  95 int parser_debug, yy_flex_debug;
  96
  97 int pedantic = 0;
  98 int do_everything = 1;
  99 int preprocess_only = 0;
 100 int do_header = 0;
 101 int do_typelib = 0;
 102 int do_proxies = 0;
 103 int do_client = 0;
 104 int do_server = 0;
 105 int do_idfile = 0;
 106 int do_dlldata = 0;
 107 int no_preprocess = 0;
 108 int old_names = 0;
 109
 110 char *input_name;
 111 char *header_name;
 112 char *local_stubs_name;
 113 char *header_token;
 114 char *typelib_name;
 115 char *dlldata_name;
 116 char *proxy_name;
 117 char *proxy_token;
 118 char *client_name;
 119 char *client_token;
 120 char *server_name;
 121 char *server_token;
 122 char *idfile_name;
 123 char *idfile_token;
 124 char *temp_name;
 125 const char *prefix_client = "";
 126 const char *prefix_server = "";
 127
 128 int line_number = 1;
 129
 130 FILE *header;
 131 FILE *local_stubs;
 132 FILE *proxy;
 133 FILE *idfile;
 134
 135 time_t now;
 136
 137 enum {
 138     OLDNAMES_OPTION = CHAR_MAX + 1,
 139     DLLDATA_OPTION,
 140     DLLDATA_ONLY_OPTION,
 141     LOCAL_STUBS_OPTION,
 142     PREFIX_ALL_OPTION,
 143     PREFIX_CLIENT_OPTION,
 144     PREFIX_SERVER_OPTION
 145 };
 146
 147 static const char short_options[] =
 148     "cC:d:D:EhH:I:NpP:sS:tT:uU:VW";
 149 static const struct option long_options[] = {
 150     { "dlldata", 1, 0, DLLDATA_OPTION },
 151     { "dlldata-only", 0, 0, DLLDATA_ONLY_OPTION },
 152     { "local-stubs", 1, 0, LOCAL_STUBS_OPTION },
 153     { "oldnames", 0, 0, OLDNAMES_OPTION },
 154     { "prefix-all", 1, 0, PREFIX_ALL_OPTION },
 155     { "prefix-client", 1, 0, PREFIX_CLIENT_OPTION },
 156     { "prefix-server", 1, 0, PREFIX_SERVER_OPTION },
 157     { 0, 0, 0, 0 }
 158 };
 159
 160 static void rm_tempfile(void);
 161
 162 static char *make_token(const char *name)
 163 {
 164   char *token;
 165   char *slash;
 166   int i;
 167
 168   slash = strrchr(name, '/');
 169   if(!slash)
 170     slash = strrchr(name, '\\');
 171
 172   if (slash) name = slash + 1;
 173
 174   token = xstrdup(name);
 175   for (i=0; token[i]; i++) {
 176     if (!isalnum(token[i])) token[i] = '_';
 177     else token[i] = tolower(token[i]);
 178   }
 179   return token;
 180 }
 181
 182 /* duplicate a basename into a valid C token */
 183 static char *dup_basename_token(const char *name, const char *ext)
 184 {
 185     char *p, *ret = dup_basename( name, ext );
 186     /* map invalid characters to '_' */
 187     for (p = ret; *p; p++) if (!isalnum(*p)) *p = '_';
 188     return ret;
 189 }
 190
 191 /* clean things up when aborting on a signal */
 192 static void exit_on_signal( int sig )
 193 {
 194     exit(1);  /* this will call the atexit functions */
 195 }
 196
 197 static void set_everything(int x)
 198 {
 199   do_header = x;
 200   do_typelib = x;
 201   do_proxies = x;
 202   do_client = x;
 203   do_server = x;
 204   do_idfile = x;
 205   do_dlldata = x;
 206 }
 207
 208 static void start_cplusplus_guard(FILE *fp)
 209 {
 210   fprintf(fp, "#ifdef __cplusplus\n");
 211   fprintf(fp, "extern \"C\" {\n");
 212   fprintf(fp, "#endif\n\n");
 213 }
 214
 215 static void end_cplusplus_guard(FILE *fp)
 216 {
 217   fprintf(fp, "#ifdef __cplusplus\n");
 218   fprintf(fp, "}\n");
 219   fprintf(fp, "#endif\n\n");
 220 }
 221
 222 typedef struct
 223 {
 224   char *filename;
 225   struct list link;
 226 } filename_node_t;
 227
 228 static void add_filename_node(struct list *list, const char *name)
 229 {
 230   filename_node_t *node = xmalloc(sizeof *node);
 231   node->filename = dup_basename( name, ".idl" );
 232   list_add_tail(list, &node->link);
 233 }
 234
 235 static void free_filename_nodes(struct list *list)
 236 {
 237   filename_node_t *node, *next;
 238   LIST_FOR_EACH_ENTRY_SAFE(node, next, list, filename_node_t, link) {
 239     list_remove(&node->link);
 240     free(node->filename);
 241     free(node);
 242   }
 243 }
 244
 245 static void write_dlldata_list(struct list *filenames)
 246 {
 247   FILE *dlldata;
 248   filename_node_t *node;
 249
 250   dlldata = fopen(dlldata_name, "w");
 251   if (!dlldata)
 252     error("couldn't open %s: %s\n", dlldata_name, strerror(errno));
 253
 254   fprintf(dlldata, "/*** Autogenerated by WIDL %s ", PACKAGE_VERSION);
 255   fprintf(dlldata, "- Do not edit ***/\n\n");
 256   fprintf(dlldata, "#include <objbase.h>\n");
 257   fprintf(dlldata, "#include <rpcproxy.h>\n\n");
 258   start_cplusplus_guard(dlldata);
 259
 260   LIST_FOR_EACH_ENTRY(node, filenames, filename_node_t, link)
 261     fprintf(dlldata, "EXTERN_PROXY_FILE(%s)\n", node->filename);
 262
 263   fprintf(dlldata, "\nPROXYFILE_LIST_START\n");
 264   fprintf(dlldata, "/* Start of list */\n");
 265   LIST_FOR_EACH_ENTRY(node, filenames, filename_node_t, link)
 266     fprintf(dlldata, "  REFERENCE_PROXY_FILE(%s),\n", node->filename);
 267   fprintf(dlldata, "/* End of list */\n");
 268   fprintf(dlldata, "PROXYFILE_LIST_END\n\n");
 269
 270   fprintf(dlldata, "DLLDATA_ROUTINES(aProxyFileList, GET_DLL_CLSID)\n\n");
 271   end_cplusplus_guard(dlldata);
 272   fclose(dlldata);
 273 }
 274
 275 static char *eat_space(char *s)
 276 {
 277   while (isspace((unsigned char) *s))
 278     ++s;
 279   return s;
 280 }
 281
 282 void write_dlldata(const statement_list_t *stmts)
 283 {
 284   struct list filenames = LIST_INIT(filenames);
 285   filename_node_t *node;
 286   FILE *dlldata;
 287
 288   if (!do_dlldata || !need_proxy_file(stmts))
 289     return;
 290
 291   dlldata = fopen(dlldata_name, "r");
 292   if (dlldata) {
 293     static char marker[] = "REFERENCE_PROXY_FILE";
 294     char *line = NULL;
 295     size_t len = 0;
 296
 297     while (widl_getline(&line, &len, dlldata)) {
 298       char *start, *end;
 299       start = eat_space(line);
 300       if (strncmp(start, marker, sizeof marker - 1) == 0) {
 301         start = eat_space(start + sizeof marker - 1);
 302         if (*start != '(')
 303           continue;
 304         end = start = eat_space(start + 1);
 305         while (*end && *end != ')')
 306           ++end;
 307         if (*end != ')')
 308           continue;
 309         while (isspace((unsigned char) end[-1]))
 310           --end;
 311         *end = '\0';
 312         if (start < end)
 313           add_filename_node(&filenames, start);
 314       }
 315     }
 316
 317     if (ferror(dlldata))
 318       error("couldn't read from %s: %s\n", dlldata_name, strerror(errno));
 319
 320     free(line);
 321     fclose(dlldata);
 322   }
 323
 324   LIST_FOR_EACH_ENTRY(node, &filenames, filename_node_t, link)
 325     if (strcmp(proxy_token, node->filename) == 0) {
 326       /* We're already in the list, no need to regenerate this file.  */
 327       free_filename_nodes(&filenames);
 328       return;
 329     }
 330
 331   add_filename_node(&filenames, proxy_token);
 332   write_dlldata_list(&filenames);
 333   free_filename_nodes(&filenames);
 334 }
 335
 336 int main(int argc,char *argv[])
 337 {
 338   extern char* optarg;
 339   extern int   optind;
 340   int optc;
 341   int ret = 0;
 342   int opti = 0;
 343
 344   signal( SIGTERM, exit_on_signal );
 345   signal( SIGINT, exit_on_signal );
 346 #ifdef SIGHUP
 347   signal( SIGHUP, exit_on_signal );
 348 #endif
 349
 350   now = time(NULL);
 351
 352   while((optc = getopt_long(argc, argv, short_options, long_options, &opti)) != EOF) {
 353     switch(optc) {
 354     case DLLDATA_OPTION:
 355       dlldata_name = xstrdup(optarg);
 356       break;
 357     case DLLDATA_ONLY_OPTION:
 358       do_everything = 0;
 359       do_dlldata = 1;
 360       break;
 361     case LOCAL_STUBS_OPTION:
 362       do_everything = 0;
 363       local_stubs_name = xstrdup(optarg);
 364       break;
 365     case OLDNAMES_OPTION:
 366       old_names = 1;
 367       break;
 368     case PREFIX_ALL_OPTION:
 369       prefix_client = xstrdup(optarg);
 370       prefix_server = xstrdup(optarg);
 371       break;
 372     case PREFIX_CLIENT_OPTION:
 373       prefix_client = xstrdup(optarg);
 374       break;
 375     case PREFIX_SERVER_OPTION:
 376       prefix_server = xstrdup(optarg);
 377       break;
 378     case 'c':
 379       do_everything = 0;
 380       do_client = 1;
 381       break;
 382     case 'C':
 383       client_name = xstrdup(optarg);
 384       break;
 385     case 'd':
 386       debuglevel = strtol(optarg, NULL, 0);
 387       break;
 388     case 'D':
 389       wpp_add_cmdline_define(optarg);
 390       break;
 391     case 'E':
 392       do_everything = 0;
 393       preprocess_only = 1;
 394       break;
 395     case 'h':
 396       do_everything = 0;
 397       do_header = 1;
 398       break;
 399     case 'H':
 400       header_name = xstrdup(optarg);
 401       break;
 402     case 'I':
 403       wpp_add_include_path(optarg);
 404       break;
 405     case 'N':
 406       no_preprocess = 1;
 407       break;
 408     case 'p':
 409       do_everything = 0;
 410       do_proxies = 1;
 411       break;
 412     case 'P':
 413       proxy_name = xstrdup(optarg);
 414       break;
 415     case 's':
 416       do_everything = 0;
 417       do_server = 1;
 418       break;
 419     case 'S':
 420       server_name = xstrdup(optarg);
 421       break;
 422     case 't':
 423       do_everything = 0;
 424       do_typelib = 1;
 425       break;
 426     case 'T':
 427       typelib_name = xstrdup(optarg);
 428       break;
 429     case 'u':
 430       do_everything = 0;
 431       do_idfile = 1;
 432       break;
 433     case 'U':
 434       idfile_name = xstrdup(optarg);
 435       break;
 436     case 'V':
 437       printf("%s", version_string);
 438       return 0;
 439     case 'W':
 440       pedantic = 1;
 441       break;
 442     default:
 443       fprintf(stderr, "%s", usage);
 444       return 1;
 445     }
 446   }
 447
 448   if(do_everything) {
 449     set_everything(TRUE);
 450   }
 451
 452   if (!dlldata_name && do_dlldata)
 453     dlldata_name = xstrdup("dlldata.c");
 454
 455   if(optind < argc) {
 456     if (do_dlldata && !do_everything) {
 457       struct list filenames = LIST_INIT(filenames);
 458       for ( ; optind < argc; ++optind)
 459         add_filename_node(&filenames, argv[optind]);
 460
 461       write_dlldata_list(&filenames);
 462       free_filename_nodes(&filenames);
 463       return 0;
 464     }
 465     else if (optind != argc - 1) {
 466       fprintf(stderr, "%s", usage);
 467       return 1;
 468     }
 469     else
 470       input_name = xstrdup(argv[optind]);
 471   }
 472   else {
 473     fprintf(stderr, "%s", usage);
 474     return 1;
 475   }
 476
 477   if(debuglevel)
 478   {
 479     setbuf(stdout,0);
 480     setbuf(stderr,0);
 481   }
 482
 483   parser_debug = debuglevel & DEBUGLEVEL_TRACE ? 1 : 0;
 484   yy_flex_debug = debuglevel & DEBUGLEVEL_TRACE ? 1 : 0;
 485
 486   wpp_set_debug( (debuglevel & DEBUGLEVEL_PPLEX) != 0,
 487                  (debuglevel & DEBUGLEVEL_PPTRACE) != 0,
 488                  (debuglevel & DEBUGLEVEL_PPMSG) != 0 );
 489
 490   if (!header_name) {
 491     header_name = dup_basename(input_name, ".idl");
 492     strcat(header_name, ".h");
 493   }
 494
 495   if (!typelib_name && do_typelib) {
 496     typelib_name = dup_basename(input_name, ".idl");
 497     strcat(typelib_name, ".tlb");
 498   }
 499
 500   if (!proxy_name && do_proxies) {
 501     proxy_name = dup_basename(input_name, ".idl");
 502     strcat(proxy_name, "_p.c");
 503   }
 504
 505   if (!client_name && do_client) {
 506     client_name = dup_basename(input_name, ".idl");
 507     strcat(client_name, "_c.c");
 508   }
 509
 510   if (!server_name && do_server) {
 511     server_name = dup_basename(input_name, ".idl");
 512     strcat(server_name, "_s.c");
 513   }
 514
 515   if (!idfile_name && do_idfile) {
 516     idfile_name = dup_basename(input_name, ".idl");
 517     strcat(idfile_name, "_i.c");
 518   }
 519
 520   if (do_proxies) proxy_token = dup_basename_token(proxy_name,"_p.c");
 521   if (do_client) client_token = dup_basename_token(client_name,"_c.c");
 522   if (do_server) server_token = dup_basename_token(server_name,"_s.c");
 523
 524   wpp_add_cmdline_define("__WIDL__");
 525
 526   atexit(rm_tempfile);
 527   if (!no_preprocess)
 528   {
 529     chat("Starting preprocess\n");
 530
 531     if (!preprocess_only)
 532     {
 533         ret = wpp_parse_temp( input_name, header_name, &temp_name );
 534     }
 535     else
 536     {
 537         ret = wpp_parse( input_name, stdout );
 538     }
 539
 540     if(ret) exit(1);
 541     if(preprocess_only) exit(0);
 542     if(!(parser_in = fopen(temp_name, "r"))) {
 543       fprintf(stderr, "Could not open %s for input\n", temp_name);
 544       return 1;
 545     }
 546   }
 547   else {
 548     if(!(parser_in = fopen(input_name, "r"))) {
 549       fprintf(stderr, "Could not open %s for input\n", input_name);
 550       return 1;
 551     }
 552   }
 553
 554   if(do_header) {
 555     header_token = make_token(header_name);
 556
 557     if(!(header = fopen(header_name, "w"))) {
 558       fprintf(stderr, "Could not open %s for output\n", header_name);
 559       return 1;
 560     }
 561     fprintf(header, "/*** Autogenerated by WIDL %s from %s - Do not edit ***/\n", PACKAGE_VERSION, input_name);
 562     fprintf(header, "#include <rpc.h>\n" );
 563     fprintf(header, "#include <rpcndr.h>\n\n" );
 564     fprintf(header, "#ifndef __%s__\n", header_token);
 565     fprintf(header, "#define __%s__\n", header_token);
 566     start_cplusplus_guard(header);
 567   }
 568
 569   if (local_stubs_name) {
 570     local_stubs = fopen(local_stubs_name, "w");
 571     if (!local_stubs) {
 572       fprintf(stderr, "Could not open %s for output\n", local_stubs_name);
 573       return 1;
 574     }
 575     fprintf(local_stubs, "/* call_as/local stubs for %s */\n\n", input_name);
 576     fprintf(local_stubs, "#include <objbase.h>\n");
 577     fprintf(local_stubs, "#include \"%s\"\n\n", header_name);
 578   }
 579
 580   if (do_idfile) {
 581     idfile_token = make_token(idfile_name);
 582
 583     idfile = fopen(idfile_name, "w");
 584     if (! idfile) {
 585       fprintf(stderr, "Could not open %s for output\n", idfile_name);
 586       return 1;
 587     }
 588
 589     fprintf(idfile, "/*** Autogenerated by WIDL %s ", PACKAGE_VERSION);
 590     fprintf(idfile, "from %s - Do not edit ***/\n\n", input_name);
 591     fprintf(idfile, "#include <rpc.h>\n");
 592     fprintf(idfile, "#include <rpcndr.h>\n\n");
 593     fprintf(idfile, "#include <initguid.h>\n\n");
 594     start_cplusplus_guard(idfile);
 595   }
 596
 597   init_types();
 598   ret = parser_parse();
 599
 600   if(do_header) {
 601     fprintf(header, "/* Begin additional prototypes for all interfaces */\n");
 602     fprintf(header, "\n");
 603     write_user_types();
 604     write_generic_handle_routines();
 605     write_context_handle_rundowns();
 606     fprintf(header, "\n");
 607     fprintf(header, "/* End additional prototypes */\n");
 608     fprintf(header, "\n");
 609     end_cplusplus_guard(header);
 610     fprintf(header, "#endif /* __%s__ */\n", header_token);
 611     fclose(header);
 612   }
 613
 614   if (local_stubs) {
 615     fclose(local_stubs);
 616   }
 617
 618   if (do_idfile) {
 619     fprintf(idfile, "\n");
 620     end_cplusplus_guard(idfile);
 621
 622     fclose(idfile);
 623   }
 624
 625   fclose(parser_in);
 626
 627   if(ret) {
 628     exit(1);
 629   }
 630
 631   /* Everything has been done successfully, don't delete any files.  */
 632   set_everything(FALSE);
 633   local_stubs_name = NULL;
 634
 635   return 0;
 636 }
 637
 638 static void rm_tempfile(void)
 639 {
 640   abort_import();
 641   if(temp_name)
 642     unlink(temp_name);
 643   if (do_header)
 644     unlink(header_name);
 645   if (local_stubs_name)
 646     unlink(local_stubs_name);
 647   if (do_client)
 648     unlink(client_name);
 649   if (do_server)
 650     unlink(server_name);
 651   if (do_idfile)
 652     unlink(idfile_name);
 653   if (do_proxies)
 654     unlink(proxy_name);
 655   if (do_typelib)
 656     unlink(typelib_name);
 657 }