git.oblomov.eu Git - wine/blob - tools/widl/header.c

   1 /*
   2  * IDL Compiler
   3  *
   4  * Copyright 2002 Ove Kaaven
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22
  23 #include <stdarg.h>
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #ifdef HAVE_UNISTD_H
  27 # include <unistd.h>
  28 #endif
  29 #include <string.h>
  30 #include <assert.h>
  31 #include <ctype.h>
  32 #include <signal.h>
  33
  34 #include "windef.h"
  35 #include "widl.h"
  36 #include "utils.h"
  37 #include "parser.h"
  38 #include "header.h"
  39
  40 static int indentation = 0;
  41 user_type_list_t user_type_list = LIST_INIT(user_type_list);
  42 static context_handle_list_t context_handle_list = LIST_INIT(context_handle_list);
  43
  44 static void indent(FILE *h, int delta)
  45 {
  46   int c;
  47   if (delta < 0) indentation += delta;
  48   for (c=0; c<indentation; c++) fprintf(h, "    ");
  49   if (delta > 0) indentation += delta;
  50 }
  51
  52 int is_ptrchain_attr(const var_t *var, enum attr_type t)
  53 {
  54     if (is_attr(var->attrs, t))
  55         return 1;
  56     else
  57     {
  58         type_t *type = var->type;
  59         for (;;)
  60         {
  61             if (is_attr(type->attrs, t))
  62                 return 1;
  63             else if (type->kind == TKIND_ALIAS)
  64                 type = type->orig;
  65             else if (is_ptr(type))
  66                 type = type->ref;
  67             else return 0;
  68         }
  69     }
  70 }
  71
  72 int is_attr(const attr_list_t *list, enum attr_type t)
  73 {
  74     const attr_t *attr;
  75     if (list) LIST_FOR_EACH_ENTRY( attr, list, const attr_t, entry )
  76         if (attr->type == t) return 1;
  77     return 0;
  78 }
  79
  80 void *get_attrp(const attr_list_t *list, enum attr_type t)
  81 {
  82     const attr_t *attr;
  83     if (list) LIST_FOR_EACH_ENTRY( attr, list, const attr_t, entry )
  84         if (attr->type == t) return attr->u.pval;
  85     return NULL;
  86 }
  87
  88 unsigned long get_attrv(const attr_list_t *list, enum attr_type t)
  89 {
  90     const attr_t *attr;
  91     if (list) LIST_FOR_EACH_ENTRY( attr, list, const attr_t, entry )
  92         if (attr->type == t) return attr->u.ival;
  93     return 0;
  94 }
  95
  96 int is_void(const type_t *t)
  97 {
  98   if (!t->type && !t->ref) return 1;
  99   return 0;
 100 }
 101
 102 int is_conformant_array(const type_t *t)
 103 {
 104     return t->type == RPC_FC_CARRAY
 105         || t->type == RPC_FC_CVARRAY
 106         || (t->type == RPC_FC_BOGUS_ARRAY && t->size_is);
 107 }
 108
 109 void write_guid(FILE *f, const char *guid_prefix, const char *name, const UUID *uuid)
 110 {
 111   if (!uuid) return;
 112   fprintf(f, "DEFINE_GUID(%s_%s, 0x%08x, 0x%04x, 0x%04x, 0x%02x,0x%02x, 0x%02x,"
 113         "0x%02x,0x%02x,0x%02x,0x%02x,0x%02x);\n",
 114         guid_prefix, name, uuid->Data1, uuid->Data2, uuid->Data3, uuid->Data4[0],
 115         uuid->Data4[1], uuid->Data4[2], uuid->Data4[3], uuid->Data4[4], uuid->Data4[5],
 116         uuid->Data4[6], uuid->Data4[7]);
 117 }
 118
 119 void write_name(FILE *h, const var_t *v)
 120 {
 121   if (is_attr( v->attrs, ATTR_PROPGET ))
 122     fprintf(h, "get_" );
 123   else if (is_attr( v->attrs, ATTR_PROPPUT ))
 124     fprintf(h, "put_" );
 125   else if (is_attr( v->attrs, ATTR_PROPPUTREF ))
 126     fprintf(h, "putref_" );
 127   fprintf(h, "%s", v->name);
 128 }
 129
 130 void write_prefix_name(FILE *h, const char *prefix, const var_t *v)
 131 {
 132   fprintf(h, "%s", prefix);
 133   write_name(h, v);
 134 }
 135
 136 static void write_field(FILE *h, var_t *v)
 137 {
 138   if (!v) return;
 139   if (v->type) {
 140     const char *name = v->name;
 141     if (name == NULL) {
 142       switch (v->type->type) {
 143       case RPC_FC_STRUCT:
 144       case RPC_FC_CVSTRUCT:
 145       case RPC_FC_CPSTRUCT:
 146       case RPC_FC_CSTRUCT:
 147       case RPC_FC_PSTRUCT:
 148       case RPC_FC_BOGUS_STRUCT:
 149       case RPC_FC_ENCAPSULATED_UNION:
 150         name = "DUMMYSTRUCTNAME";
 151         break;
 152       case RPC_FC_NON_ENCAPSULATED_UNION:
 153         name = "DUMMYUNIONNAME";
 154         break;
 155       default:
 156         /* ? */
 157         break;
 158       }
 159     }
 160     indent(h, 0);
 161     write_type_def_or_decl(h, v->type, TRUE, "%s", name);
 162     fprintf(h, ";\n");
 163   }
 164 }
 165
 166 static void write_fields(FILE *h, var_list_t *fields)
 167 {
 168     var_t *v;
 169     if (!fields) return;
 170     LIST_FOR_EACH_ENTRY( v, fields, var_t, entry ) write_field(h, v);
 171 }
 172
 173 static void write_enums(FILE *h, var_list_t *enums)
 174 {
 175   var_t *v;
 176   if (!enums) return;
 177   LIST_FOR_EACH_ENTRY( v, enums, var_t, entry )
 178   {
 179     if (v->name) {
 180       indent(h, 0);
 181       write_name(h, v);
 182       if (v->eval) {
 183         fprintf(h, " = ");
 184         write_expr(h, v->eval, 0);
 185       }
 186     }
 187     if (list_next( enums, &v->entry )) fprintf(h, ",\n");
 188   }
 189   fprintf(h, "\n");
 190 }
 191
 192 int needs_space_after(type_t *t)
 193 {
 194   return (t->kind == TKIND_ALIAS
 195           || (!is_ptr(t) && (!is_conformant_array(t) || t->declarray)));
 196 }
 197
 198 void write_type_left(FILE *h, type_t *t, int declonly)
 199 {
 200   if (!h) return;
 201
 202   if (t->is_const) fprintf(h, "const ");
 203
 204   if (t->kind == TKIND_ALIAS) fprintf(h, "%s", t->name);
 205   else if (t->declarray) write_type_left(h, t->ref, declonly);
 206   else {
 207     if (t->sign > 0) fprintf(h, "signed ");
 208     else if (t->sign < 0) fprintf(h, "unsigned ");
 209     switch (t->type) {
 210       case RPC_FC_ENUM16:
 211       case RPC_FC_ENUM32:
 212         if (!declonly && t->defined && !t->written && !t->ignore) {
 213           if (t->name) fprintf(h, "enum %s {\n", t->name);
 214           else fprintf(h, "enum {\n");
 215           t->written = TRUE;
 216           indentation++;
 217           write_enums(h, t->fields);
 218           indent(h, -1);
 219           fprintf(h, "}");
 220         }
 221         else fprintf(h, "enum %s", t->name ? t->name : "");
 222         break;
 223       case RPC_FC_STRUCT:
 224       case RPC_FC_CVSTRUCT:
 225       case RPC_FC_CPSTRUCT:
 226       case RPC_FC_CSTRUCT:
 227       case RPC_FC_PSTRUCT:
 228       case RPC_FC_BOGUS_STRUCT:
 229       case RPC_FC_ENCAPSULATED_UNION:
 230         if (!declonly && t->defined && !t->written && !t->ignore) {
 231           if (t->name) fprintf(h, "struct %s {\n", t->name);
 232           else fprintf(h, "struct {\n");
 233           t->written = TRUE;
 234           indentation++;
 235           write_fields(h, t->fields);
 236           indent(h, -1);
 237           fprintf(h, "}");
 238         }
 239         else fprintf(h, "struct %s", t->name ? t->name : "");
 240         break;
 241       case RPC_FC_NON_ENCAPSULATED_UNION:
 242         if (!declonly && t->defined && !t->written && !t->ignore) {
 243           if (t->name) fprintf(h, "union %s {\n", t->name);
 244           else fprintf(h, "union {\n");
 245           t->written = TRUE;
 246           indentation++;
 247           write_fields(h, t->fields);
 248           indent(h, -1);
 249           fprintf(h, "}");
 250         }
 251         else fprintf(h, "union %s", t->name ? t->name : "");
 252         break;
 253       case RPC_FC_RP:
 254       case RPC_FC_UP:
 255       case RPC_FC_FP:
 256       case RPC_FC_OP:
 257       case RPC_FC_CARRAY:
 258       case RPC_FC_CVARRAY:
 259       case RPC_FC_BOGUS_ARRAY:
 260         write_type_left(h, t->ref, declonly);
 261         fprintf(h, "%s*", needs_space_after(t->ref) ? " " : "");
 262         break;
 263       default:
 264         fprintf(h, "%s", t->name);
 265     }
 266   }
 267 }
 268
 269 void write_type_right(FILE *h, type_t *t, int is_field)
 270 {
 271   if (!h) return;
 272
 273   if (t->declarray) {
 274     if (is_conformant_array(t)) {
 275       fprintf(h, "[%s]", is_field ? "1" : "");
 276       t = t->ref;
 277     }
 278     for ( ; t->declarray; t = t->ref)
 279       fprintf(h, "[%lu]", t->dim);
 280   }
 281 }
 282
 283 void write_type_v(FILE *h, type_t *t, int is_field, int declonly,
 284                   const char *fmt, va_list args)
 285 {
 286   if (!h) return;
 287
 288   write_type_left(h, t, declonly);
 289   if (fmt) {
 290     if (needs_space_after(t))
 291       fprintf(h, " ");
 292     vfprintf(h, fmt, args);
 293   }
 294   write_type_right(h, t, is_field);
 295 }
 296
 297 void write_type_def_or_decl(FILE *f, type_t *t, int field, const char *fmt, ...)
 298 {
 299   va_list args;
 300   va_start(args, fmt);
 301   write_type_v(f, t, field, FALSE, fmt, args);
 302   va_end(args);
 303 }
 304
 305 void write_type_decl(FILE *f, type_t *t, const char *fmt, ...)
 306 {
 307   va_list args;
 308   va_start(args, fmt);
 309   write_type_v(f, t, FALSE, TRUE, fmt, args);
 310   va_end(args);
 311 }
 312
 313 void write_type_decl_left(FILE *f, type_t *t)
 314 {
 315   write_type_left(f, t, TRUE);
 316 }
 317
 318 static int user_type_registered(const char *name)
 319 {
 320   user_type_t *ut;
 321   LIST_FOR_EACH_ENTRY(ut, &user_type_list, user_type_t, entry)
 322     if (!strcmp(name, ut->name))
 323       return 1;
 324   return 0;
 325 }
 326
 327 static int context_handle_registered(const char *name)
 328 {
 329   context_handle_t *ch;
 330   LIST_FOR_EACH_ENTRY(ch, &context_handle_list, context_handle_t, entry)
 331     if (!strcmp(name, ch->name))
 332       return 1;
 333   return 0;
 334 }
 335
 336 void check_for_user_types_and_context_handles(const var_list_t *list)
 337 {
 338   const var_t *v;
 339
 340   if (!list) return;
 341   LIST_FOR_EACH_ENTRY( v, list, const var_t, entry )
 342   {
 343     type_t *type;
 344     for (type = v->type; type; type = type->kind == TKIND_ALIAS ? type->orig : type->ref) {
 345       const char *name = type->name;
 346       if (type->user_types_registered) continue;
 347       type->user_types_registered = 1;
 348       if (is_attr(type->attrs, ATTR_CONTEXTHANDLE)) {
 349         if (!context_handle_registered(name))
 350         {
 351           context_handle_t *ch = xmalloc(sizeof(*ch));
 352           ch->name = xstrdup(name);
 353           list_add_tail(&context_handle_list, &ch->entry);
 354         }
 355         /* don't carry on parsing fields within this type */
 356         break;
 357       }
 358       if (is_attr(type->attrs, ATTR_WIREMARSHAL)) {
 359         if (!user_type_registered(name))
 360         {
 361           user_type_t *ut = xmalloc(sizeof *ut);
 362           ut->name = xstrdup(name);
 363           list_add_tail(&user_type_list, &ut->entry);
 364         }
 365         /* don't carry on parsing fields within this type as we are already
 366          * using a wire marshaled type */
 367         break;
 368       }
 369       else
 370       {
 371         check_for_user_types_and_context_handles(type->fields);
 372       }
 373     }
 374   }
 375 }
 376
 377 void write_user_types(void)
 378 {
 379   user_type_t *ut;
 380   LIST_FOR_EACH_ENTRY(ut, &user_type_list, user_type_t, entry)
 381   {
 382     const char *name = ut->name;
 383     fprintf(header, "ULONG           __RPC_USER %s_UserSize     (ULONG *, ULONG, %s *);\n", name, name);
 384     fprintf(header, "unsigned char * __RPC_USER %s_UserMarshal  (ULONG *, unsigned char *, %s *);\n", name, name);
 385     fprintf(header, "unsigned char * __RPC_USER %s_UserUnmarshal(ULONG *, unsigned char *, %s *);\n", name, name);
 386     fprintf(header, "void            __RPC_USER %s_UserFree     (ULONG *, %s *);\n", name, name);
 387   }
 388 }
 389
 390 void write_context_handle_rundowns(void)
 391 {
 392   context_handle_t *ch;
 393   LIST_FOR_EACH_ENTRY(ch, &context_handle_list, context_handle_t, entry)
 394   {
 395     const char *name = ch->name;
 396     fprintf(header, "void __RPC_USER %s_rundown(%s);\n", name, name);
 397   }
 398 }
 399
 400 void write_typedef(type_t *type)
 401 {
 402   fprintf(header, "typedef ");
 403   write_type_def_or_decl(header, type->orig, FALSE, "%s", type->name);
 404   fprintf(header, ";\n");
 405 }
 406
 407 void write_expr(FILE *h, const expr_t *e, int brackets)
 408 {
 409   switch (e->type) {
 410   case EXPR_VOID:
 411     break;
 412   case EXPR_NUM:
 413     fprintf(h, "%lu", e->u.lval);
 414     break;
 415   case EXPR_HEXNUM:
 416     fprintf(h, "0x%lx", e->u.lval);
 417     break;
 418   case EXPR_DOUBLE:
 419     fprintf(h, "%#.15g", e->u.dval);
 420     break;
 421   case EXPR_TRUEFALSE:
 422     if (e->u.lval == 0)
 423       fprintf(h, "FALSE");
 424     else
 425       fprintf(h, "TRUE");
 426     break;
 427   case EXPR_IDENTIFIER:
 428     fprintf(h, "%s", e->u.sval);
 429     break;
 430   case EXPR_NEG:
 431     fprintf(h, "-");
 432     write_expr(h, e->ref, 1);
 433     break;
 434   case EXPR_NOT:
 435     fprintf(h, "~");
 436     write_expr(h, e->ref, 1);
 437     break;
 438   case EXPR_PPTR:
 439     fprintf(h, "*");
 440     write_expr(h, e->ref, 1);
 441     break;
 442   case EXPR_CAST:
 443     fprintf(h, "(");
 444     write_type_decl(h, e->u.tref, NULL);
 445     fprintf(h, ")");
 446     write_expr(h, e->ref, 1);
 447     break;
 448   case EXPR_SIZEOF:
 449     fprintf(h, "sizeof(");
 450     write_type_decl(h, e->u.tref, NULL);
 451     fprintf(h, ")");
 452     break;
 453   case EXPR_SHL:
 454   case EXPR_SHR:
 455   case EXPR_MUL:
 456   case EXPR_DIV:
 457   case EXPR_ADD:
 458   case EXPR_SUB:
 459   case EXPR_AND:
 460   case EXPR_OR:
 461     if (brackets) fprintf(h, "(");
 462     write_expr(h, e->ref, 1);
 463     switch (e->type) {
 464     case EXPR_SHL: fprintf(h, " << "); break;
 465     case EXPR_SHR: fprintf(h, " >> "); break;
 466     case EXPR_MUL: fprintf(h, " * "); break;
 467     case EXPR_DIV: fprintf(h, " / "); break;
 468     case EXPR_ADD: fprintf(h, " + "); break;
 469     case EXPR_SUB: fprintf(h, " - "); break;
 470     case EXPR_AND: fprintf(h, " & "); break;
 471     case EXPR_OR:  fprintf(h, " | "); break;
 472     default: break;
 473     }
 474     write_expr(h, e->u.ext, 1);
 475     if (brackets) fprintf(h, ")");
 476     break;
 477   case EXPR_COND:
 478     if (brackets) fprintf(h, "(");
 479     write_expr(h, e->ref, 1);
 480     fprintf(h, " ? ");
 481     write_expr(h, e->u.ext, 1);
 482     fprintf(h, " : ");
 483     write_expr(h, e->ext2, 1);
 484     if (brackets) fprintf(h, ")");
 485     break;
 486   }
 487 }
 488
 489 void write_constdef(const var_t *v)
 490 {
 491   fprintf(header, "#define %s (", v->name);
 492   write_expr(header, v->eval, 0);
 493   fprintf(header, ")\n\n");
 494 }
 495
 496 void write_externdef(const var_t *v)
 497 {
 498   fprintf(header, "extern const ");
 499   write_type_def_or_decl(header, v->type, FALSE, "%s", v->name);
 500   fprintf(header, ";\n\n");
 501 }
 502
 503 void write_library(const char *name, const attr_list_t *attr)
 504 {
 505   const UUID *uuid = get_attrp(attr, ATTR_UUID);
 506   fprintf(header, "\n");
 507   write_guid(header, "LIBID", name, uuid);
 508   fprintf(header, "\n");
 509 }
 510
 511
 512 const var_t* get_explicit_handle_var(const func_t* func)
 513 {
 514     const var_t* var;
 515
 516     if (!func->args)
 517         return NULL;
 518
 519     LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
 520         if (var->type->type == RPC_FC_BIND_PRIMITIVE)
 521             return var;
 522
 523     return NULL;
 524 }
 525
 526 int has_out_arg_or_return(const func_t *func)
 527 {
 528     const var_t *var;
 529
 530     if (!is_void(func->def->type))
 531         return 1;
 532
 533     if (!func->args)
 534         return 0;
 535
 536     LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
 537         if (is_attr(var->attrs, ATTR_OUT))
 538             return 1;
 539
 540     return 0;
 541 }
 542
 543
 544 /********** INTERFACES **********/
 545
 546 int is_object(const attr_list_t *list)
 547 {
 548     const attr_t *attr;
 549     if (list) LIST_FOR_EACH_ENTRY( attr, list, const attr_t, entry )
 550         if (attr->type == ATTR_OBJECT || attr->type == ATTR_ODL) return 1;
 551     return 0;
 552 }
 553
 554 int is_local(const attr_list_t *a)
 555 {
 556   return is_attr(a, ATTR_LOCAL);
 557 }
 558
 559 const var_t *is_callas(const attr_list_t *a)
 560 {
 561   return get_attrp(a, ATTR_CALLAS);
 562 }
 563
 564 static void write_method_macro(const type_t *iface, const char *name)
 565 {
 566   const func_t *cur;
 567
 568   if (iface->ref) write_method_macro(iface->ref, name);
 569
 570   if (!iface->funcs) return;
 571
 572   fprintf(header, "/*** %s methods ***/\n", iface->name);
 573   LIST_FOR_EACH_ENTRY( cur, iface->funcs, const func_t, entry )
 574   {
 575     var_t *def = cur->def;
 576     if (!is_callas(def->attrs)) {
 577       const var_t *arg;
 578       int argc = 0;
 579       int c;
 580
 581       if (cur->args) LIST_FOR_EACH_ENTRY( arg, cur->args, const var_t, entry ) argc++;
 582
 583       fprintf(header, "#define %s_", name);
 584       write_name(header,def);
 585       fprintf(header, "(p");
 586       for (c=0; c<argc; c++)
 587         fprintf(header, ",%c", c+'a');
 588       fprintf(header, ") ");
 589
 590       fprintf(header, "(p)->lpVtbl->");
 591       write_name(header, def);
 592       fprintf(header, "(p");
 593       for (c=0; c<argc; c++)
 594         fprintf(header, ",%c", c+'a');
 595       fprintf(header, ")\n");
 596     }
 597   }
 598 }
 599
 600 void write_args(FILE *h, const var_list_t *args, const char *name, int method, int do_indent)
 601 {
 602   const var_t *arg;
 603   int count = 0;
 604
 605   if (do_indent)
 606   {
 607       indentation++;
 608       indent(h, 0);
 609   }
 610   if (method == 1) {
 611     fprintf(h, "%s* This", name);
 612     count++;
 613   }
 614   if (args) LIST_FOR_EACH_ENTRY( arg, args, const var_t, entry ) {
 615     if (count) {
 616         if (do_indent)
 617         {
 618             fprintf(h, ",\n");
 619             indent(h, 0);
 620         }
 621         else fprintf(h, ",");
 622     }
 623     if (arg->args)
 624     {
 625       write_type_decl_left(h, arg->type);
 626       fprintf(h, " (STDMETHODCALLTYPE *");
 627       write_name(h,arg);
 628       fprintf(h, ")(");
 629       write_args(h, arg->args, NULL, 0, FALSE);
 630       fprintf(h, ")");
 631     }
 632     else
 633       write_type_decl(h, arg->type, "%s", arg->name);
 634     count++;
 635   }
 636   if (do_indent) indentation--;
 637 }
 638
 639 static void write_cpp_method_def(const type_t *iface)
 640 {
 641   const func_t *cur;
 642
 643   if (!iface->funcs) return;
 644
 645   LIST_FOR_EACH_ENTRY( cur, iface->funcs, const func_t, entry )
 646   {
 647     var_t *def = cur->def;
 648     if (!is_callas(def->attrs)) {
 649       indent(header, 0);
 650       fprintf(header, "virtual ");
 651       write_type_decl_left(header, def->type);
 652       fprintf(header, " STDMETHODCALLTYPE ");
 653       write_name(header, def);
 654       fprintf(header, "(\n");
 655       write_args(header, cur->args, iface->name, 2, TRUE);
 656       fprintf(header, ") = 0;\n");
 657       fprintf(header, "\n");
 658     }
 659   }
 660 }
 661
 662 static void do_write_c_method_def(const type_t *iface, const char *name)
 663 {
 664   const func_t *cur;
 665
 666   if (iface->ref) do_write_c_method_def(iface->ref, name);
 667
 668   if (!iface->funcs) return;
 669   indent(header, 0);
 670   fprintf(header, "/*** %s methods ***/\n", iface->name);
 671   LIST_FOR_EACH_ENTRY( cur, iface->funcs, const func_t, entry )
 672   {
 673     const var_t *def = cur->def;
 674     if (!is_callas(def->attrs)) {
 675       indent(header, 0);
 676       write_type_decl_left(header, def->type);
 677       fprintf(header, " (STDMETHODCALLTYPE *");
 678       write_name(header, def);
 679       fprintf(header, ")(\n");
 680       write_args(header, cur->args, name, 1, TRUE);
 681       fprintf(header, ");\n");
 682       fprintf(header, "\n");
 683     }
 684   }
 685 }
 686
 687 static void write_c_method_def(const type_t *iface)
 688 {
 689   do_write_c_method_def(iface, iface->name);
 690 }
 691
 692 static void write_c_disp_method_def(const type_t *iface)
 693 {
 694   do_write_c_method_def(iface->ref, iface->name);
 695 }
 696
 697 static void write_method_proto(const type_t *iface)
 698 {
 699   const func_t *cur;
 700
 701   if (!iface->funcs) return;
 702   LIST_FOR_EACH_ENTRY( cur, iface->funcs, const func_t, entry )
 703   {
 704     const var_t *def = cur->def;
 705
 706     if (!is_local(def->attrs)) {
 707       /* proxy prototype */
 708       write_type_decl_left(header, def->type);
 709       fprintf(header, " CALLBACK %s_", iface->name);
 710       write_name(header, def);
 711       fprintf(header, "_Proxy(\n");
 712       write_args(header, cur->args, iface->name, 1, TRUE);
 713       fprintf(header, ");\n");
 714       /* stub prototype */
 715       fprintf(header, "void __RPC_STUB %s_", iface->name);
 716       write_name(header,def);
 717       fprintf(header, "_Stub(\n");
 718       fprintf(header, "    IRpcStubBuffer* This,\n");
 719       fprintf(header, "    IRpcChannelBuffer* pRpcChannelBuffer,\n");
 720       fprintf(header, "    PRPC_MESSAGE pRpcMessage,\n");
 721       fprintf(header, "    DWORD* pdwStubPhase);\n");
 722     }
 723   }
 724 }
 725
 726 void write_locals(FILE *fp, const type_t *iface, int body)
 727 {
 728   static const char comment[]
 729     = "/* WIDL-generated stub.  You must provide an implementation for this.  */";
 730   const func_list_t *funcs = iface->funcs;
 731   const func_t *cur;
 732
 733   if (!is_object(iface->attrs) || !funcs)
 734     return;
 735
 736   LIST_FOR_EACH_ENTRY(cur, funcs, const func_t, entry) {
 737     const var_t *def = cur->def;
 738     const var_t *cas = is_callas(def->attrs);
 739
 740     if (cas) {
 741       const func_t *m;
 742       LIST_FOR_EACH_ENTRY(m, iface->funcs, const func_t, entry)
 743         if (!strcmp(m->def->name, cas->name))
 744           break;
 745       if (&m->entry != iface->funcs) {
 746         const var_t *mdef = m->def;
 747         /* proxy prototype - use local prototype */
 748         write_type_decl_left(fp, mdef->type);
 749         fprintf(fp, " CALLBACK %s_", iface->name);
 750         write_name(fp, mdef);
 751         fprintf(fp, "_Proxy(\n");
 752         write_args(fp, m->args, iface->name, 1, TRUE);
 753         fprintf(fp, ")");
 754         if (body) {
 755           type_t *rt = mdef->type;
 756           fprintf(fp, "\n{\n");
 757           fprintf(fp, "    %s\n", comment);
 758           if (rt->name && strcmp(rt->name, "HRESULT") == 0)
 759             fprintf(fp, "    return E_NOTIMPL;\n");
 760           else if (rt->type) {
 761             fprintf(fp, "    ");
 762             write_type_decl(fp, rt, "rv");
 763             fprintf(fp, ";\n");
 764             fprintf(fp, "    memset(&rv, 0, sizeof rv);\n");
 765             fprintf(fp, "    return rv;\n");
 766           }
 767           fprintf(fp, "}\n\n");
 768         }
 769         else
 770           fprintf(fp, ";\n");
 771         /* stub prototype - use remotable prototype */
 772         write_type_decl_left(fp, def->type);
 773         fprintf(fp, " __RPC_STUB %s_", iface->name);
 774         write_name(fp, mdef);
 775         fprintf(fp, "_Stub(\n");
 776         write_args(fp, cur->args, iface->name, 1, TRUE);
 777         fprintf(fp, ")");
 778         if (body)
 779           /* Remotable methods must all return HRESULTs.  */
 780           fprintf(fp, "\n{\n    %s\n    return E_NOTIMPL;\n}\n\n", comment);
 781         else
 782           fprintf(fp, ";\n");
 783       }
 784       else
 785         error_loc("invalid call_as attribute (%s -> %s)\n", def->name, cas->name);
 786     }
 787   }
 788 }
 789
 790 static void write_function_proto(const type_t *iface, const func_t *fun, const char *prefix)
 791 {
 792   var_t *def = fun->def;
 793
 794   /* FIXME: do we need to handle call_as? */
 795   write_type_decl_left(header, def->type);
 796   fprintf(header, " ");
 797   write_prefix_name(header, prefix, def);
 798   fprintf(header, "(\n");
 799   if (fun->args)
 800     write_args(header, fun->args, iface->name, 0, TRUE);
 801   else
 802     fprintf(header, "    void");
 803   fprintf(header, ");\n");
 804 }
 805
 806 static void write_function_protos(const type_t *iface)
 807 {
 808   const char *implicit_handle = get_attrp(iface->attrs, ATTR_IMPLICIT_HANDLE);
 809   int explicit_handle = is_attr(iface->attrs, ATTR_EXPLICIT_HANDLE);
 810   const var_t* explicit_handle_var;
 811   const func_t *cur;
 812   int prefixes_differ = strcmp(prefix_client, prefix_server);
 813
 814   if (!iface->funcs) return;
 815   LIST_FOR_EACH_ENTRY( cur, iface->funcs, const func_t, entry )
 816   {
 817     var_t *def = cur->def;
 818
 819     /* check for a defined binding handle */
 820     explicit_handle_var = get_explicit_handle_var(cur);
 821     if (explicit_handle) {
 822       if (!explicit_handle_var) {
 823         error("%s() does not define an explicit binding handle!\n", def->name);
 824         return;
 825       }
 826     } else if (implicit_handle) {
 827       if (explicit_handle_var) {
 828         error("%s() must not define a binding handle!\n", def->name);
 829         return;
 830       }
 831     }
 832
 833     if (prefixes_differ) {
 834       fprintf(header, "/* client prototype */\n");
 835       write_function_proto(iface, cur, prefix_client);
 836       fprintf(header, "/* server prototype */\n");
 837     }
 838     write_function_proto(iface, cur, prefix_server);
 839   }
 840 }
 841
 842 void write_forward(type_t *iface)
 843 {
 844   /* C/C++ forwards should only be written for object interfaces, so if we
 845    * have a full definition we only write one if we find [object] among the
 846    * attributes - however, if we don't have a full definition at this point
 847    * (i.e. this is an IDL forward), then we also assume that it is an object
 848    * interface, since non-object interfaces shouldn't need forwards */
 849   if ((!iface->defined || is_object(iface->attrs) || is_attr(iface->attrs, ATTR_DISPINTERFACE))
 850         && !iface->written) {
 851     fprintf(header, "#ifndef __%s_FWD_DEFINED__\n", iface->name);
 852     fprintf(header, "#define __%s_FWD_DEFINED__\n", iface->name);
 853     fprintf(header, "typedef interface %s %s;\n", iface->name, iface->name);
 854     fprintf(header, "#endif\n\n" );
 855     iface->written = TRUE;
 856   }
 857 }
 858
 859 static void write_iface_guid(const type_t *iface)
 860 {
 861   const UUID *uuid = get_attrp(iface->attrs, ATTR_UUID);
 862   write_guid(header, "IID", iface->name, uuid);
 863 }
 864
 865 static void write_dispiface_guid(const type_t *iface)
 866 {
 867   const UUID *uuid = get_attrp(iface->attrs, ATTR_UUID);
 868   write_guid(header, "DIID", iface->name, uuid);
 869 }
 870
 871 static void write_coclass_guid(type_t *cocl)
 872 {
 873   const UUID *uuid = get_attrp(cocl->attrs, ATTR_UUID);
 874   write_guid(header, "CLSID", cocl->name, uuid);
 875 }
 876
 877 static void write_com_interface(type_t *iface)
 878 {
 879   if (!iface->funcs && !iface->ref) {
 880     parser_warning("%s has no methods\n", iface->name);
 881     return;
 882   }
 883
 884   fprintf(header, "/*****************************************************************************\n");
 885   fprintf(header, " * %s interface\n", iface->name);
 886   fprintf(header, " */\n");
 887   fprintf(header,"#ifndef __%s_INTERFACE_DEFINED__\n", iface->name);
 888   fprintf(header,"#define __%s_INTERFACE_DEFINED__\n\n", iface->name);
 889   write_iface_guid(iface);
 890   write_forward(iface);
 891   /* C++ interface */
 892   fprintf(header, "#if defined(__cplusplus) && !defined(CINTERFACE)\n");
 893   if (iface->ref)
 894   {
 895       fprintf(header, "interface %s : public %s\n", iface->name, iface->ref->name);
 896       fprintf(header, "{\n");
 897       indentation++;
 898       write_cpp_method_def(iface);
 899       indentation--;
 900       fprintf(header, "};\n");
 901   }
 902   else
 903   {
 904       fprintf(header, "interface %s\n", iface->name);
 905       fprintf(header, "{\n");
 906       fprintf(header, "    BEGIN_INTERFACE\n");
 907       fprintf(header, "\n");
 908       indentation++;
 909       write_cpp_method_def(iface);
 910       indentation--;
 911       fprintf(header, "    END_INTERFACE\n");
 912       fprintf(header, "};\n");
 913   }
 914   fprintf(header, "#else\n");
 915   /* C interface */
 916   fprintf(header, "typedef struct %sVtbl {\n", iface->name);
 917   indentation++;
 918   fprintf(header, "    BEGIN_INTERFACE\n");
 919   fprintf(header, "\n");
 920   write_c_method_def(iface);
 921   indentation--;
 922   fprintf(header, "    END_INTERFACE\n");
 923   fprintf(header, "} %sVtbl;\n", iface->name);
 924   fprintf(header, "interface %s {\n", iface->name);
 925   fprintf(header, "    CONST_VTBL %sVtbl* lpVtbl;\n", iface->name);
 926   fprintf(header, "};\n");
 927   fprintf(header, "\n");
 928   fprintf(header, "#ifdef COBJMACROS\n");
 929   write_method_macro(iface, iface->name);
 930   fprintf(header, "#endif\n");
 931   fprintf(header, "\n");
 932   fprintf(header, "#endif\n");
 933   fprintf(header, "\n");
 934   write_method_proto(iface);
 935   write_locals(header, iface, FALSE);
 936   fprintf(header,"\n#endif  /* __%s_INTERFACE_DEFINED__ */\n\n", iface->name);
 937 }
 938
 939 static void write_rpc_interface(const type_t *iface)
 940 {
 941   unsigned long ver = get_attrv(iface->attrs, ATTR_VERSION);
 942   const char *var = get_attrp(iface->attrs, ATTR_IMPLICIT_HANDLE);
 943   static int allocate_written = 0;
 944
 945   if (!allocate_written)
 946   {
 947     allocate_written = 1;
 948     fprintf(header, "void * __RPC_USER MIDL_user_allocate(size_t);\n");
 949     fprintf(header, "void __RPC_USER MIDL_user_free(void *);\n\n");
 950   }
 951
 952   fprintf(header, "/*****************************************************************************\n");
 953   fprintf(header, " * %s interface (v%d.%d)\n", iface->name, LOWORD(ver), HIWORD(ver));
 954   fprintf(header, " */\n");
 955   fprintf(header,"#ifndef __%s_INTERFACE_DEFINED__\n", iface->name);
 956   fprintf(header,"#define __%s_INTERFACE_DEFINED__\n\n", iface->name);
 957   if (iface->funcs)
 958   {
 959     write_iface_guid(iface);
 960     if (var) fprintf(header, "extern handle_t %s;\n", var);
 961     if (old_names)
 962     {
 963         fprintf(header, "extern RPC_IF_HANDLE %s%s_ClientIfHandle;\n", prefix_client, iface->name);
 964         fprintf(header, "extern RPC_IF_HANDLE %s%s_ServerIfHandle;\n", prefix_server, iface->name);
 965     }
 966     else
 967     {
 968         fprintf(header, "extern RPC_IF_HANDLE %s%s_v%d_%d_c_ifspec;\n",
 969                 prefix_client, iface->name, LOWORD(ver), HIWORD(ver));
 970         fprintf(header, "extern RPC_IF_HANDLE %s%s_v%d_%d_s_ifspec;\n",
 971                 prefix_server, iface->name, LOWORD(ver), HIWORD(ver));
 972     }
 973     write_function_protos(iface);
 974   }
 975   fprintf(header,"\n#endif  /* __%s_INTERFACE_DEFINED__ */\n\n", iface->name);
 976
 977   /* FIXME: server/client code */
 978 }
 979
 980 void write_interface(type_t *iface)
 981 {
 982   if (is_object(iface->attrs))
 983     write_com_interface(iface);
 984   else
 985     write_rpc_interface(iface);
 986 }
 987
 988 void write_dispinterface(type_t *iface)
 989 {
 990   fprintf(header, "/*****************************************************************************\n");
 991   fprintf(header, " * %s dispinterface\n", iface->name);
 992   fprintf(header, " */\n");
 993   fprintf(header,"#ifndef __%s_DISPINTERFACE_DEFINED__\n", iface->name);
 994   fprintf(header,"#define __%s_DISPINTERFACE_DEFINED__\n\n", iface->name);
 995   write_dispiface_guid(iface);
 996   write_forward(iface);
 997   /* C++ interface */
 998   fprintf(header, "#if defined(__cplusplus) && !defined(CINTERFACE)\n");
 999   fprintf(header, "interface %s : public %s\n", iface->name, iface->ref->name);
1000   fprintf(header, "{\n");
1001   fprintf(header, "};\n");
1002   fprintf(header, "#else\n");
1003   /* C interface */
1004   fprintf(header, "typedef struct %sVtbl {\n", iface->name);
1005   indentation++;
1006   fprintf(header, "    BEGIN_INTERFACE\n");
1007   fprintf(header, "\n");
1008   write_c_disp_method_def(iface);
1009   indentation--;
1010   fprintf(header, "    END_INTERFACE\n");
1011   fprintf(header, "} %sVtbl;\n", iface->name);
1012   fprintf(header, "interface %s {\n", iface->name);
1013   fprintf(header, "    CONST_VTBL %sVtbl* lpVtbl;\n", iface->name);
1014   fprintf(header, "};\n");
1015   fprintf(header, "\n");
1016   fprintf(header, "#ifdef COBJMACROS\n");
1017   write_method_macro(iface->ref, iface->name);
1018   fprintf(header, "#endif\n");
1019   fprintf(header, "\n");
1020   fprintf(header, "#endif\n");
1021   fprintf(header, "\n");
1022   fprintf(header,"#endif  /* __%s_DISPINTERFACE_DEFINED__ */\n\n", iface->name);
1023 }
1024
1025 void write_coclass(type_t *cocl)
1026 {
1027   fprintf(header, "/*****************************************************************************\n");
1028   fprintf(header, " * %s coclass\n", cocl->name);
1029   fprintf(header, " */\n\n");
1030   write_coclass_guid(cocl);
1031   fprintf(header, "\n");
1032 }
1033
1034 void write_coclass_forward(type_t *cocl)
1035 {
1036   fprintf(header, "#ifndef __%s_FWD_DEFINED__\n", cocl->name);
1037   fprintf(header, "#define __%s_FWD_DEFINED__\n", cocl->name);
1038   fprintf(header, "typedef struct %s %s;\n", cocl->name, cocl->name);
1039   fprintf(header, "#endif /* defined __%s_FWD_DEFINED__ */\n\n", cocl->name );
1040 }