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

   1 /*
   2  * Format String Generator for IDL Compiler
   3  *
   4  * Copyright 2005-2006 Eric Kohl
   5  * Copyright 2005-2006 Robert Shearman
   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 <stdio.h>
  26 #include <stdlib.h>
  27 #ifdef HAVE_UNISTD_H
  28 # include <unistd.h>
  29 #endif
  30 #include <string.h>
  31 #include <assert.h>
  32 #include <ctype.h>
  33 #include <limits.h>
  34
  35 #include "widl.h"
  36 #include "utils.h"
  37 #include "parser.h"
  38 #include "header.h"
  39 #include "typetree.h"
  40
  41 #include "typegen.h"
  42 #include "expr.h"
  43
  44 /* round size up to multiple of alignment */
  45 #define ROUND_SIZE(size, alignment) (((size) + ((alignment) - 1)) & ~((alignment) - 1))
  46 /* value to add on to round size up to a multiple of alignment */
  47 #define ROUNDING(size, alignment) (((alignment) - 1) - (((size) + ((alignment) - 1)) & ((alignment) - 1)))
  48
  49 static const type_t *current_structure;
  50 static const type_t *current_iface;
  51
  52 static struct list expr_eval_routines = LIST_INIT(expr_eval_routines);
  53 struct expr_eval_routine
  54 {
  55     struct list entry;
  56     const type_t *structure;
  57     unsigned int baseoff;
  58     const expr_t *expr;
  59 };
  60
  61 enum type_context
  62 {
  63     TYPE_CONTEXT_TOPLEVELPARAM,
  64     TYPE_CONTEXT_PARAM,
  65     TYPE_CONTEXT_CONTAINER,
  66     TYPE_CONTEXT_CONTAINER_NO_POINTERS,
  67 };
  68
  69 static unsigned int field_memsize(const type_t *type, unsigned int *offset);
  70 static unsigned int fields_memsize(const var_list_t *fields, unsigned int *align);
  71 static unsigned int type_memsize_and_alignment(const type_t *t, unsigned int *align);
  72 static unsigned int write_struct_tfs(FILE *file, type_t *type, const char *name, unsigned int *tfsoff);
  73 static int write_embedded_types(FILE *file, const attr_list_t *attrs, type_t *type,
  74                                 const char *name, int write_ptr, unsigned int *tfsoff);
  75 static const var_t *find_array_or_string_in_struct(const type_t *type);
  76 static unsigned int write_string_tfs(FILE *file, const attr_list_t *attrs,
  77                                type_t *type, int toplevel_param,
  78                                const char *name, unsigned int *typestring_offset);
  79
  80 static const char *string_of_type(unsigned char type)
  81 {
  82     switch (type)
  83     {
  84     case RPC_FC_BYTE: return "FC_BYTE";
  85     case RPC_FC_CHAR: return "FC_CHAR";
  86     case RPC_FC_SMALL: return "FC_SMALL";
  87     case RPC_FC_USMALL: return "FC_USMALL";
  88     case RPC_FC_WCHAR: return "FC_WCHAR";
  89     case RPC_FC_SHORT: return "FC_SHORT";
  90     case RPC_FC_USHORT: return "FC_USHORT";
  91     case RPC_FC_LONG: return "FC_LONG";
  92     case RPC_FC_ULONG: return "FC_ULONG";
  93     case RPC_FC_FLOAT: return "FC_FLOAT";
  94     case RPC_FC_HYPER: return "FC_HYPER";
  95     case RPC_FC_DOUBLE: return "FC_DOUBLE";
  96     case RPC_FC_ENUM16: return "FC_ENUM16";
  97     case RPC_FC_ENUM32: return "FC_ENUM32";
  98     case RPC_FC_IGNORE: return "FC_IGNORE";
  99     case RPC_FC_ERROR_STATUS_T: return "FC_ERROR_STATUS_T";
 100     case RPC_FC_RP: return "FC_RP";
 101     case RPC_FC_UP: return "FC_UP";
 102     case RPC_FC_OP: return "FC_OP";
 103     case RPC_FC_FP: return "FC_FP";
 104     case RPC_FC_ENCAPSULATED_UNION: return "FC_ENCAPSULATED_UNION";
 105     case RPC_FC_NON_ENCAPSULATED_UNION: return "FC_NON_ENCAPSULATED_UNION";
 106     case RPC_FC_STRUCT: return "FC_STRUCT";
 107     case RPC_FC_PSTRUCT: return "FC_PSTRUCT";
 108     case RPC_FC_CSTRUCT: return "FC_CSTRUCT";
 109     case RPC_FC_CPSTRUCT: return "FC_CPSTRUCT";
 110     case RPC_FC_CVSTRUCT: return "FC_CVSTRUCT";
 111     case RPC_FC_BOGUS_STRUCT: return "FC_BOGUS_STRUCT";
 112     case RPC_FC_SMFARRAY: return "FC_SMFARRAY";
 113     case RPC_FC_LGFARRAY: return "FC_LGFARRAY";
 114     case RPC_FC_SMVARRAY: return "FC_SMVARRAY";
 115     case RPC_FC_LGVARRAY: return "FC_LGVARRAY";
 116     case RPC_FC_CARRAY: return "FC_CARRAY";
 117     case RPC_FC_CVARRAY: return "FC_CVARRAY";
 118     case RPC_FC_BOGUS_ARRAY: return "FC_BOGUS_ARRAY";
 119     case RPC_FC_ALIGNM2: return "FC_ALIGNM2";
 120     case RPC_FC_ALIGNM4: return "FC_ALIGNM4";
 121     case RPC_FC_ALIGNM8: return "FC_ALIGNM8";
 122     case RPC_FC_POINTER: return "FC_POINTER";
 123     case RPC_FC_C_CSTRING: return "FC_C_CSTRING";
 124     case RPC_FC_C_WSTRING: return "FC_C_WSTRING";
 125     case RPC_FC_CSTRING: return "FC_CSTRING";
 126     case RPC_FC_WSTRING: return "FC_WSTRING";
 127     case RPC_FC_BYTE_COUNT_POINTER: return "FC_BYTE_COUNT_POINTER";
 128     case RPC_FC_TRANSMIT_AS: return "FC_TRANSMIT_AS";
 129     case RPC_FC_REPRESENT_AS: return "FC_REPRESENT_AS";
 130     case RPC_FC_IP: return "FC_IP";
 131     case RPC_FC_BIND_CONTEXT: return "FC_BIND_CONTEXT";
 132     case RPC_FC_BIND_GENERIC: return "FC_BIND_GENERIC";
 133     case RPC_FC_BIND_PRIMITIVE: return "FC_BIND_PRIMITIVE";
 134     case RPC_FC_AUTO_HANDLE: return "FC_AUTO_HANDLE";
 135     case RPC_FC_CALLBACK_HANDLE: return "FC_CALLBACK_HANDLE";
 136     case RPC_FC_STRUCTPAD1: return "FC_STRUCTPAD1";
 137     case RPC_FC_STRUCTPAD2: return "FC_STRUCTPAD2";
 138     case RPC_FC_STRUCTPAD3: return "FC_STRUCTPAD3";
 139     case RPC_FC_STRUCTPAD4: return "FC_STRUCTPAD4";
 140     case RPC_FC_STRUCTPAD5: return "FC_STRUCTPAD5";
 141     case RPC_FC_STRUCTPAD6: return "FC_STRUCTPAD6";
 142     case RPC_FC_STRUCTPAD7: return "FC_STRUCTPAD7";
 143     case RPC_FC_STRING_SIZED: return "FC_STRING_SIZED";
 144     case RPC_FC_NO_REPEAT: return "FC_NO_REPEAT";
 145     case RPC_FC_FIXED_REPEAT: return "FC_FIXED_REPEAT";
 146     case RPC_FC_VARIABLE_REPEAT: return "FC_VARIABLE_REPEAT";
 147     case RPC_FC_FIXED_OFFSET: return "FC_FIXED_OFFSET";
 148     case RPC_FC_VARIABLE_OFFSET: return "FC_VARIABLE_OFFSET";
 149     case RPC_FC_PP: return "FC_PP";
 150     case RPC_FC_EMBEDDED_COMPLEX: return "FC_EMBEDDED_COMPLEX";
 151     case RPC_FC_DEREFERENCE: return "FC_DEREFERENCE";
 152     case RPC_FC_DIV_2: return "FC_DIV_2";
 153     case RPC_FC_MULT_2: return "FC_MULT_2";
 154     case RPC_FC_ADD_1: return "FC_ADD_1";
 155     case RPC_FC_SUB_1: return "FC_SUB_1";
 156     case RPC_FC_CALLBACK: return "FC_CALLBACK";
 157     case RPC_FC_CONSTANT_IID: return "FC_CONSTANT_IID";
 158     case RPC_FC_END: return "FC_END";
 159     case RPC_FC_PAD: return "FC_PAD";
 160     case RPC_FC_USER_MARSHAL: return "FC_USER_MARSHAL";
 161     case RPC_FC_RANGE: return "FC_RANGE";
 162     case RPC_FC_INT3264: return "FC_INT3264";
 163     case RPC_FC_UINT3264: return "FC_UINT3264";
 164     default:
 165         error("string_of_type: unknown type 0x%02x\n", type);
 166         return NULL;
 167     }
 168 }
 169
 170 static void *get_aliaschain_attrp(const type_t *type, enum attr_type attr)
 171 {
 172     const type_t *t = type;
 173     for (;;)
 174     {
 175         if (is_attr(t->attrs, attr))
 176             return get_attrp(t->attrs, attr);
 177         else if (type_is_alias(t))
 178             t = type_alias_get_aliasee(t);
 179         else return NULL;
 180     }
 181 }
 182
 183 unsigned char get_basic_fc(const type_t *type)
 184 {
 185     int sign = type_basic_get_sign(type);
 186     switch (type_basic_get_type(type))
 187     {
 188     case TYPE_BASIC_INT8: return (sign <= 0 ? RPC_FC_SMALL : RPC_FC_USMALL);
 189     case TYPE_BASIC_INT16: return (sign <= 0 ? RPC_FC_SHORT : RPC_FC_USHORT);
 190     case TYPE_BASIC_INT32: return (sign <= 0 ? RPC_FC_LONG : RPC_FC_ULONG);
 191     case TYPE_BASIC_INT64: return RPC_FC_HYPER;
 192     case TYPE_BASIC_INT: return (sign <= 0 ? RPC_FC_LONG : RPC_FC_ULONG);
 193     case TYPE_BASIC_INT3264: return (sign <= 0 ? RPC_FC_INT3264 : RPC_FC_UINT3264);
 194     case TYPE_BASIC_BYTE: return RPC_FC_BYTE;
 195     case TYPE_BASIC_CHAR: return RPC_FC_CHAR;
 196     case TYPE_BASIC_WCHAR: return RPC_FC_WCHAR;
 197     case TYPE_BASIC_HYPER: return RPC_FC_HYPER;
 198     case TYPE_BASIC_FLOAT: return RPC_FC_FLOAT;
 199     case TYPE_BASIC_DOUBLE: return RPC_FC_DOUBLE;
 200     case TYPE_BASIC_ERROR_STATUS_T: return RPC_FC_ERROR_STATUS_T;
 201     case TYPE_BASIC_HANDLE: return RPC_FC_BIND_PRIMITIVE;
 202     }
 203     return 0;
 204 }
 205
 206 static inline unsigned int clamp_align(unsigned int align)
 207 {
 208     unsigned int packing = (pointer_size == 4) ? win32_packing : win64_packing;
 209     if(align > packing) align = packing;
 210     return align;
 211 }
 212
 213 unsigned char get_pointer_fc(const type_t *type, const attr_list_t *attrs, int toplevel_param)
 214 {
 215     const type_t *t;
 216     int pointer_type;
 217
 218     assert(is_ptr(type) || is_array(type));
 219
 220     pointer_type = get_attrv(attrs, ATTR_POINTERTYPE);
 221     if (pointer_type)
 222         return pointer_type;
 223
 224     for (t = type; type_is_alias(t); t = type_alias_get_aliasee(t))
 225     {
 226         pointer_type = get_attrv(t->attrs, ATTR_POINTERTYPE);
 227         if (pointer_type)
 228             return pointer_type;
 229     }
 230
 231     if (toplevel_param)
 232         return RPC_FC_RP;
 233     else if (is_ptr(type))
 234         return type_pointer_get_default_fc(type);
 235     else
 236         return type_array_get_ptr_default_fc(type);
 237 }
 238
 239 static unsigned char get_enum_fc(const type_t *type)
 240 {
 241     assert(type_get_type(type) == TYPE_ENUM);
 242     if (is_aliaschain_attr(type, ATTR_V1ENUM))
 243         return RPC_FC_ENUM32;
 244     else
 245         return RPC_FC_ENUM16;
 246 }
 247
 248 static type_t *get_user_type(const type_t *t, const char **pname)
 249 {
 250     for (;;)
 251     {
 252         type_t *ut = get_attrp(t->attrs, ATTR_WIREMARSHAL);
 253         if (ut)
 254         {
 255             if (pname)
 256                 *pname = t->name;
 257             return ut;
 258         }
 259
 260         if (type_is_alias(t))
 261             t = type_alias_get_aliasee(t);
 262         else
 263             return NULL;
 264     }
 265 }
 266
 267 static int is_user_type(const type_t *t)
 268 {
 269     return get_user_type(t, NULL) != NULL;
 270 }
 271
 272 enum typegen_type typegen_detect_type(const type_t *type, const attr_list_t *attrs, unsigned int flags)
 273 {
 274     if (is_user_type(type))
 275         return TGT_USER_TYPE;
 276
 277     if (is_aliaschain_attr(type, ATTR_CONTEXTHANDLE))
 278         return TGT_CTXT_HANDLE;
 279
 280     if (!(flags & TDT_IGNORE_STRINGS) && is_string_type(attrs, type))
 281         return TGT_STRING;
 282
 283     switch (type_get_type(type))
 284     {
 285     case TYPE_BASIC:
 286         if (!(flags & TDT_IGNORE_RANGES) &&
 287             (is_attr(attrs, ATTR_RANGE) || is_aliaschain_attr(type, ATTR_RANGE)))
 288             return TGT_RANGE;
 289         return TGT_BASIC;
 290     case TYPE_ENUM:
 291         if (!(flags & TDT_IGNORE_RANGES) &&
 292             (is_attr(attrs, ATTR_RANGE) || is_aliaschain_attr(type, ATTR_RANGE)))
 293             return TGT_RANGE;
 294         return TGT_ENUM;
 295     case TYPE_POINTER:
 296         if (type_get_type(type_pointer_get_ref(type)) == TYPE_INTERFACE ||
 297             (type_get_type(type_pointer_get_ref(type)) == TYPE_VOID && is_attr(attrs, ATTR_IIDIS)))
 298             return TGT_IFACE_POINTER;
 299         else if (is_aliaschain_attr(type_pointer_get_ref(type), ATTR_CONTEXTHANDLE))
 300             return TGT_CTXT_HANDLE_POINTER;
 301         else
 302             return TGT_POINTER;
 303     case TYPE_STRUCT:
 304         return TGT_STRUCT;
 305     case TYPE_ENCAPSULATED_UNION:
 306     case TYPE_UNION:
 307         return TGT_UNION;
 308     case TYPE_ARRAY:
 309         return TGT_ARRAY;
 310     case TYPE_FUNCTION:
 311     case TYPE_COCLASS:
 312     case TYPE_INTERFACE:
 313     case TYPE_MODULE:
 314     case TYPE_VOID:
 315     case TYPE_ALIAS:
 316     case TYPE_BITFIELD:
 317         break;
 318     }
 319     return TGT_INVALID;
 320 }
 321
 322 static int get_padding(const var_list_t *fields)
 323 {
 324     unsigned short offset = 0;
 325     unsigned int salign = 1;
 326     const var_t *f;
 327
 328     if (!fields)
 329         return 0;
 330
 331     LIST_FOR_EACH_ENTRY(f, fields, const var_t, entry)
 332     {
 333         type_t *ft = f->type;
 334         unsigned int align = 0;
 335         unsigned int size = type_memsize_and_alignment(ft, &align);
 336         align = clamp_align(align);
 337         if (align > salign) salign = align;
 338         offset = ROUND_SIZE(offset, align);
 339         offset += size;
 340     }
 341
 342     return ROUNDING(offset, salign);
 343 }
 344
 345 unsigned char get_struct_fc(const type_t *type)
 346 {
 347   int has_pointer = 0;
 348   int has_conformance = 0;
 349   int has_variance = 0;
 350   var_t *field;
 351   var_list_t *fields;
 352
 353   fields = type_struct_get_fields(type);
 354
 355   if (get_padding(fields))
 356     return RPC_FC_BOGUS_STRUCT;
 357
 358   if (fields) LIST_FOR_EACH_ENTRY( field, fields, var_t, entry )
 359   {
 360     type_t *t = field->type;
 361     enum typegen_type typegen_type;
 362
 363     typegen_type = typegen_detect_type(t, field->attrs, TDT_IGNORE_STRINGS);
 364
 365     if (typegen_type == TGT_ARRAY && !type_array_is_decl_as_ptr(t))
 366     {
 367         if (is_string_type(field->attrs, field->type))
 368         {
 369             if (is_conformant_array(t))
 370                 has_conformance = 1;
 371             has_variance = 1;
 372             continue;
 373         }
 374
 375         if (is_array(type_array_get_element(field->type)))
 376             return RPC_FC_BOGUS_STRUCT;
 377
 378         if (type_array_has_conformance(field->type))
 379         {
 380             has_conformance = 1;
 381             if (list_next(fields, &field->entry))
 382                 error_loc("field '%s' deriving from a conformant array must be the last field in the structure\n",
 383                         field->name);
 384         }
 385         if (type_array_has_variance(t))
 386             has_variance = 1;
 387
 388         t = type_array_get_element(t);
 389         typegen_type = typegen_detect_type(t, field->attrs, TDT_IGNORE_STRINGS);
 390     }
 391
 392     switch (typegen_type)
 393     {
 394     case TGT_USER_TYPE:
 395     case TGT_IFACE_POINTER:
 396         return RPC_FC_BOGUS_STRUCT;
 397     case TGT_BASIC:
 398         if (type_basic_get_type(t) == TYPE_BASIC_INT3264 && pointer_size != 4)
 399             return RPC_FC_BOGUS_STRUCT;
 400         break;
 401     case TGT_ENUM:
 402         if (get_enum_fc(t) == RPC_FC_ENUM16)
 403             return RPC_FC_BOGUS_STRUCT;
 404         break;
 405     case TGT_POINTER:
 406     case TGT_ARRAY:
 407         if (get_pointer_fc(t, field->attrs, FALSE) == RPC_FC_RP || pointer_size != 4)
 408             return RPC_FC_BOGUS_STRUCT;
 409         has_pointer = 1;
 410         break;
 411     case TGT_UNION:
 412         return RPC_FC_BOGUS_STRUCT;
 413     case TGT_STRUCT:
 414     {
 415         unsigned char fc = get_struct_fc(t);
 416         switch (fc)
 417         {
 418         case RPC_FC_STRUCT:
 419             break;
 420         case RPC_FC_CVSTRUCT:
 421             has_conformance = 1;
 422             has_variance = 1;
 423             has_pointer = 1;
 424             break;
 425
 426         case RPC_FC_CPSTRUCT:
 427             has_conformance = 1;
 428             if (list_next( fields, &field->entry ))
 429                 error_loc("field '%s' deriving from a conformant array must be the last field in the structure\n",
 430                         field->name);
 431             has_pointer = 1;
 432             break;
 433
 434         case RPC_FC_CSTRUCT:
 435             has_conformance = 1;
 436             if (list_next( fields, &field->entry ))
 437                 error_loc("field '%s' deriving from a conformant array must be the last field in the structure\n",
 438                       field->name);
 439             break;
 440
 441         case RPC_FC_PSTRUCT:
 442             has_pointer = 1;
 443             break;
 444
 445         default:
 446             error_loc("Unknown struct member %s with type (0x%02x)\n", field->name, fc);
 447             /* fallthru - treat it as complex */
 448
 449         /* as soon as we see one of these these members, it's bogus... */
 450         case RPC_FC_BOGUS_STRUCT:
 451             return RPC_FC_BOGUS_STRUCT;
 452         }
 453         break;
 454     }
 455     case TGT_RANGE:
 456         return RPC_FC_BOGUS_STRUCT;
 457     case TGT_STRING:
 458         /* shouldn't get here because of TDT_IGNORE_STRINGS above. fall through */
 459     case TGT_INVALID:
 460     case TGT_CTXT_HANDLE:
 461     case TGT_CTXT_HANDLE_POINTER:
 462         /* checking after parsing should mean that we don't get here. if we do,
 463          * it's a checker bug */
 464         assert(0);
 465     }
 466   }
 467
 468   if( has_variance )
 469   {
 470     if ( has_conformance )
 471       return RPC_FC_CVSTRUCT;
 472     else
 473       return RPC_FC_BOGUS_STRUCT;
 474   }
 475   if( has_conformance && has_pointer )
 476     return RPC_FC_CPSTRUCT;
 477   if( has_conformance )
 478     return RPC_FC_CSTRUCT;
 479   if( has_pointer )
 480     return RPC_FC_PSTRUCT;
 481   return RPC_FC_STRUCT;
 482 }
 483
 484 static unsigned char get_array_fc(const type_t *type)
 485 {
 486     unsigned char fc;
 487     const expr_t *size_is;
 488     const type_t *elem_type;
 489
 490     elem_type = type_array_get_element(type);
 491     size_is = type_array_get_conformance(type);
 492
 493     if (!size_is)
 494     {
 495         unsigned int size = type_memsize(elem_type);
 496         if (size * type_array_get_dim(type) > 0xffffuL)
 497             fc = RPC_FC_LGFARRAY;
 498         else
 499             fc = RPC_FC_SMFARRAY;
 500     }
 501     else
 502         fc = RPC_FC_CARRAY;
 503
 504     if (type_array_has_variance(type))
 505     {
 506         if (fc == RPC_FC_SMFARRAY)
 507             fc = RPC_FC_SMVARRAY;
 508         else if (fc == RPC_FC_LGFARRAY)
 509             fc = RPC_FC_LGVARRAY;
 510         else if (fc == RPC_FC_CARRAY)
 511             fc = RPC_FC_CVARRAY;
 512     }
 513
 514     switch (typegen_detect_type(elem_type, NULL, TDT_IGNORE_STRINGS))
 515     {
 516     case TGT_USER_TYPE:
 517         fc = RPC_FC_BOGUS_ARRAY;
 518         break;
 519     case TGT_BASIC:
 520         if (type_basic_get_type(elem_type) == TYPE_BASIC_INT3264 &&
 521             pointer_size != 4)
 522             fc = RPC_FC_BOGUS_ARRAY;
 523         break;
 524     case TGT_STRUCT:
 525         switch (get_struct_fc(elem_type))
 526         {
 527         case RPC_FC_BOGUS_STRUCT:
 528             fc = RPC_FC_BOGUS_ARRAY;
 529             break;
 530         }
 531         break;
 532     case TGT_ENUM:
 533         /* is 16-bit enum - if so, wire size differs from mem size and so
 534          * the array cannot be block copied, which means the array is complex */
 535         if (get_enum_fc(elem_type) == RPC_FC_ENUM16)
 536             fc = RPC_FC_BOGUS_ARRAY;
 537         break;
 538     case TGT_UNION:
 539     case TGT_IFACE_POINTER:
 540         fc = RPC_FC_BOGUS_ARRAY;
 541         break;
 542     case TGT_POINTER:
 543         /* ref pointers cannot just be block copied. unique pointers to
 544          * interfaces need special treatment. either case means the array is
 545          * complex */
 546         if (get_pointer_fc(elem_type, NULL, FALSE) == RPC_FC_RP || pointer_size != 4)
 547             fc = RPC_FC_BOGUS_ARRAY;
 548         break;
 549     case TGT_RANGE:
 550         fc = RPC_FC_BOGUS_ARRAY;
 551         break;
 552     case TGT_CTXT_HANDLE:
 553     case TGT_CTXT_HANDLE_POINTER:
 554     case TGT_STRING:
 555     case TGT_INVALID:
 556     case TGT_ARRAY:
 557         /* nothing to do for everything else */
 558         break;
 559     }
 560
 561     return fc;
 562 }
 563
 564 static int is_non_complex_struct(const type_t *type)
 565 {
 566     return (type_get_type(type) == TYPE_STRUCT &&
 567             get_struct_fc(type) != RPC_FC_BOGUS_STRUCT);
 568 }
 569
 570 static int type_has_pointers(const type_t *type)
 571 {
 572     switch (typegen_detect_type(type, NULL, TDT_IGNORE_STRINGS))
 573     {
 574     case TGT_USER_TYPE:
 575         return FALSE;
 576     case TGT_POINTER:
 577         return TRUE;
 578     case TGT_ARRAY:
 579         return type_array_is_decl_as_ptr(type) || type_has_pointers(type_array_get_element(type));
 580     case TGT_STRUCT:
 581     {
 582         var_list_t *fields = type_struct_get_fields(type);
 583         const var_t *field;
 584         if (fields) LIST_FOR_EACH_ENTRY( field, fields, const var_t, entry )
 585         {
 586             if (type_has_pointers(field->type))
 587                 return TRUE;
 588         }
 589         break;
 590     }
 591     case TGT_UNION:
 592     {
 593         var_list_t *fields;
 594         const var_t *field;
 595         fields = type_union_get_cases(type);
 596         if (fields) LIST_FOR_EACH_ENTRY( field, fields, const var_t, entry )
 597         {
 598             if (field->type && type_has_pointers(field->type))
 599                 return TRUE;
 600         }
 601         break;
 602     }
 603     case TGT_CTXT_HANDLE:
 604     case TGT_CTXT_HANDLE_POINTER:
 605     case TGT_STRING:
 606     case TGT_IFACE_POINTER:
 607     case TGT_BASIC:
 608     case TGT_ENUM:
 609     case TGT_RANGE:
 610     case TGT_INVALID:
 611         break;
 612     }
 613
 614     return FALSE;
 615 }
 616
 617 static int type_has_full_pointer(const type_t *type, const attr_list_t *attrs,
 618                                  int toplevel_param)
 619 {
 620     switch (typegen_detect_type(type, NULL, TDT_IGNORE_STRINGS))
 621     {
 622     case TGT_USER_TYPE:
 623         return FALSE;
 624     case TGT_POINTER:
 625         if (get_pointer_fc(type, attrs, toplevel_param) == RPC_FC_FP)
 626             return TRUE;
 627         else
 628             return FALSE;
 629     case TGT_ARRAY:
 630         if (get_pointer_fc(type, attrs, toplevel_param) == RPC_FC_FP)
 631             return TRUE;
 632         else
 633             return type_has_full_pointer(type_array_get_element(type), NULL, FALSE);
 634     case TGT_STRUCT:
 635     {
 636         var_list_t *fields = type_struct_get_fields(type);
 637         const var_t *field;
 638         if (fields) LIST_FOR_EACH_ENTRY( field, fields, const var_t, entry )
 639         {
 640             if (type_has_full_pointer(field->type, field->attrs, FALSE))
 641                 return TRUE;
 642         }
 643         break;
 644     }
 645     case TGT_UNION:
 646     {
 647         var_list_t *fields;
 648         const var_t *field;
 649         fields = type_union_get_cases(type);
 650         if (fields) LIST_FOR_EACH_ENTRY( field, fields, const var_t, entry )
 651         {
 652             if (field->type && type_has_full_pointer(field->type, field->attrs, FALSE))
 653                 return TRUE;
 654         }
 655         break;
 656     }
 657     case TGT_CTXT_HANDLE:
 658     case TGT_CTXT_HANDLE_POINTER:
 659     case TGT_STRING:
 660     case TGT_IFACE_POINTER:
 661     case TGT_BASIC:
 662     case TGT_ENUM:
 663     case TGT_RANGE:
 664     case TGT_INVALID:
 665         break;
 666     }
 667
 668     return FALSE;
 669 }
 670
 671 static unsigned short user_type_offset(const char *name)
 672 {
 673     user_type_t *ut;
 674     unsigned short off = 0;
 675     LIST_FOR_EACH_ENTRY(ut, &user_type_list, user_type_t, entry)
 676     {
 677         if (strcmp(name, ut->name) == 0)
 678             return off;
 679         ++off;
 680     }
 681     error("user_type_offset: couldn't find type (%s)\n", name);
 682     return 0;
 683 }
 684
 685 static void update_tfsoff(type_t *type, unsigned int offset, FILE *file)
 686 {
 687     type->typestring_offset = offset;
 688     if (file) type->tfswrite = FALSE;
 689 }
 690
 691 static void guard_rec(type_t *type)
 692 {
 693     /* types that contain references to themselves (like a linked list),
 694        need to be shielded from infinite recursion when writing embedded
 695        types  */
 696     if (type->typestring_offset)
 697         type->tfswrite = FALSE;
 698     else
 699         type->typestring_offset = 1;
 700 }
 701
 702 static int is_embedded_complex(const type_t *type)
 703 {
 704     switch (typegen_detect_type(type, NULL, TDT_ALL_TYPES))
 705     {
 706     case TGT_USER_TYPE:
 707     case TGT_STRUCT:
 708     case TGT_UNION:
 709     case TGT_ARRAY:
 710     case TGT_IFACE_POINTER:
 711         return TRUE;
 712     default:
 713         return FALSE;
 714     }
 715 }
 716
 717 static const char *get_context_handle_type_name(const type_t *type)
 718 {
 719     const type_t *t;
 720     for (t = type;
 721          is_ptr(t) || type_is_alias(t);
 722          t = type_is_alias(t) ? type_alias_get_aliasee(t) : type_pointer_get_ref(t))
 723         if (is_attr(t->attrs, ATTR_CONTEXTHANDLE))
 724             return t->name;
 725     assert(0);
 726     return NULL;
 727 }
 728
 729 #define WRITE_FCTYPE(file, fctype, typestring_offset) \
 730     do { \
 731         if (file) \
 732             fprintf(file, "/* %2u */\n", typestring_offset); \
 733         print_file((file), 2, "0x%02x,    /* " #fctype " */\n", RPC_##fctype); \
 734     } \
 735     while (0)
 736
 737 static void print_file(FILE *file, int indent, const char *format, ...) __attribute__((format (printf, 3, 4)));
 738 static void print_file(FILE *file, int indent, const char *format, ...)
 739 {
 740     va_list va;
 741     va_start(va, format);
 742     print(file, indent, format, va);
 743     va_end(va);
 744 }
 745
 746 void print(FILE *file, int indent, const char *format, va_list va)
 747 {
 748     if (file)
 749     {
 750         if (format[0] != '\n')
 751             while (0 < indent--)
 752                 fprintf(file, "    ");
 753         vfprintf(file, format, va);
 754     }
 755 }
 756
 757
 758 static void write_var_init(FILE *file, int indent, const type_t *t, const char *n, const char *local_var_prefix)
 759 {
 760     if (decl_indirect(t))
 761     {
 762         print_file(file, indent, "MIDL_memset(&%s%s, 0, sizeof(%s%s));\n",
 763                    local_var_prefix, n, local_var_prefix, n);
 764         print_file(file, indent, "%s_p_%s = &%s%s;\n", local_var_prefix, n, local_var_prefix, n);
 765     }
 766     else if (is_ptr(t) || is_array(t))
 767         print_file(file, indent, "%s%s = 0;\n", local_var_prefix, n);
 768 }
 769
 770 void write_parameters_init(FILE *file, int indent, const var_t *func, const char *local_var_prefix)
 771 {
 772     const var_t *var;
 773
 774     if (!is_void(type_function_get_rettype(func->type)))
 775         write_var_init(file, indent, type_function_get_rettype(func->type), "_RetVal", local_var_prefix);
 776
 777     if (!type_get_function_args(func->type))
 778         return;
 779
 780     LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
 781         write_var_init(file, indent, var->type, var->name, local_var_prefix);
 782
 783     fprintf(file, "\n");
 784 }
 785
 786 static void write_formatdesc(FILE *f, int indent, const char *str)
 787 {
 788     print_file(f, indent, "typedef struct _MIDL_%s_FORMAT_STRING\n", str);
 789     print_file(f, indent, "{\n");
 790     print_file(f, indent + 1, "short Pad;\n");
 791     print_file(f, indent + 1, "unsigned char Format[%s_FORMAT_STRING_SIZE];\n", str);
 792     print_file(f, indent, "} MIDL_%s_FORMAT_STRING;\n", str);
 793     print_file(f, indent, "\n");
 794 }
 795
 796 void write_formatstringsdecl(FILE *f, int indent, const statement_list_t *stmts, type_pred_t pred)
 797 {
 798     clear_all_offsets();
 799
 800     print_file(f, indent, "#define TYPE_FORMAT_STRING_SIZE %d\n",
 801                get_size_typeformatstring(stmts, pred));
 802
 803     print_file(f, indent, "#define PROC_FORMAT_STRING_SIZE %d\n",
 804                get_size_procformatstring(stmts, pred));
 805
 806     fprintf(f, "\n");
 807     write_formatdesc(f, indent, "TYPE");
 808     write_formatdesc(f, indent, "PROC");
 809     fprintf(f, "\n");
 810     print_file(f, indent, "static const MIDL_TYPE_FORMAT_STRING __MIDL_TypeFormatString;\n");
 811     print_file(f, indent, "static const MIDL_PROC_FORMAT_STRING __MIDL_ProcFormatString;\n");
 812     print_file(f, indent, "\n");
 813 }
 814
 815 int decl_indirect(const type_t *t)
 816 {
 817     if (is_user_type(t))
 818         return TRUE;
 819     return (type_get_type(t) != TYPE_BASIC &&
 820             type_get_type(t) != TYPE_ENUM &&
 821             type_get_type(t) != TYPE_POINTER &&
 822             type_get_type(t) != TYPE_ARRAY);
 823 }
 824
 825 static unsigned int write_procformatstring_type(FILE *file, int indent,
 826                                                 const type_t *type,
 827                                                 const attr_list_t *attrs,
 828                                                 int is_return)
 829 {
 830     unsigned int size;
 831
 832     int is_in = is_attr(attrs, ATTR_IN);
 833     int is_out = is_attr(attrs, ATTR_OUT);
 834
 835     if (!is_in && !is_out) is_in = TRUE;
 836
 837     if (type_get_type(type) == TYPE_BASIC ||
 838         type_get_type(type) == TYPE_ENUM)
 839     {
 840         unsigned char fc;
 841
 842         if (is_return)
 843             print_file(file, indent, "0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");
 844         else
 845             print_file(file, indent, "0x4e,    /* FC_IN_PARAM_BASETYPE */\n");
 846
 847         if (type_get_type(type) == TYPE_ENUM)
 848         {
 849             fc = get_enum_fc(type);
 850         }
 851         else
 852         {
 853             fc = get_basic_fc(type);
 854
 855             if (fc == RPC_FC_BIND_PRIMITIVE)
 856                 fc = RPC_FC_IGNORE;
 857         }
 858
 859         print_file(file, indent, "0x%02x,    /* %s */\n",
 860                    fc, string_of_type(fc));
 861         size = 2; /* includes param type prefix */
 862     }
 863     else
 864     {
 865         if (is_return)
 866             print_file(file, indent, "0x52,    /* FC_RETURN_PARAM */\n");
 867         else if (is_in && is_out)
 868             print_file(file, indent, "0x50,    /* FC_IN_OUT_PARAM */\n");
 869         else if (is_out)
 870             print_file(file, indent, "0x51,    /* FC_OUT_PARAM */\n");
 871         else
 872             print_file(file, indent, "0x4d,    /* FC_IN_PARAM */\n");
 873
 874         print_file(file, indent, "0x01,\n");
 875         print_file(file, indent, "NdrFcShort(0x%x),     /* type offset = %u */\n",
 876                    type->typestring_offset, type->typestring_offset);
 877         size = 4; /* includes param type prefix */
 878     }
 879     return size;
 880 }
 881
 882 static void write_procformatstring_func( FILE *file, int indent,
 883                                          const var_t *func, unsigned int *offset )
 884 {
 885     /* emit argument data */
 886     if (type_get_function_args(func->type))
 887     {
 888         const var_t *var;
 889         LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
 890         {
 891             print_file( file, 0, "/* %u (parameter %s) */\n", *offset, var->name );
 892             *offset += write_procformatstring_type(file, indent, var->type, var->attrs, FALSE);
 893         }
 894     }
 895
 896     /* emit return value data */
 897     if (is_void(type_function_get_rettype(func->type)))
 898     {
 899         print_file(file, 0, "/* %u (void) */\n", *offset);
 900         print_file(file, indent, "0x5b,    /* FC_END */\n");
 901         print_file(file, indent, "0x5c,    /* FC_PAD */\n");
 902         *offset += 2;
 903     }
 904     else
 905     {
 906         print_file( file, 0, "/* %u (return value) */\n", *offset );
 907         *offset += write_procformatstring_type(file, indent, type_function_get_rettype(func->type), NULL, TRUE);
 908     }
 909 }
 910
 911 static void write_procformatstring_stmts(FILE *file, int indent, const statement_list_t *stmts,
 912                                          type_pred_t pred, unsigned int *offset)
 913 {
 914     const statement_t *stmt;
 915     if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
 916     {
 917         if (stmt->type == STMT_TYPE && type_get_type(stmt->u.type) == TYPE_INTERFACE)
 918         {
 919             const statement_t *stmt_func;
 920             if (!pred(stmt->u.type))
 921                 continue;
 922             STATEMENTS_FOR_EACH_FUNC(stmt_func, type_iface_get_stmts(stmt->u.type))
 923             {
 924                 const var_t *func = stmt_func->u.var;
 925                 if (is_local(func->attrs)) continue;
 926                 write_procformatstring_func( file, indent, func, offset );
 927             }
 928         }
 929         else if (stmt->type == STMT_LIBRARY)
 930             write_procformatstring_stmts(file, indent, stmt->u.lib->stmts, pred, offset);
 931     }
 932 }
 933
 934 void write_procformatstring(FILE *file, const statement_list_t *stmts, type_pred_t pred)
 935 {
 936     int indent = 0;
 937     unsigned int offset = 0;
 938
 939     print_file(file, indent, "static const MIDL_PROC_FORMAT_STRING __MIDL_ProcFormatString =\n");
 940     print_file(file, indent, "{\n");
 941     indent++;
 942     print_file(file, indent, "0,\n");
 943     print_file(file, indent, "{\n");
 944     indent++;
 945
 946     write_procformatstring_stmts(file, indent, stmts, pred, &offset);
 947
 948     print_file(file, indent, "0x0\n");
 949     indent--;
 950     print_file(file, indent, "}\n");
 951     indent--;
 952     print_file(file, indent, "};\n");
 953     print_file(file, indent, "\n");
 954 }
 955
 956 void write_procformatstring_offsets( FILE *file, const type_t *iface )
 957 {
 958     const statement_t *stmt;
 959     int indent = 0;
 960
 961     print_file( file, indent,  "static const unsigned short %s_FormatStringOffsetTable[] =\n",
 962                 iface->name );
 963     print_file( file, indent,  "{\n" );
 964     indent++;
 965     STATEMENTS_FOR_EACH_FUNC( stmt, type_iface_get_stmts(iface) )
 966     {
 967         var_t *func = stmt->u.var;
 968         if (is_local( func->attrs )) continue;
 969         print_file( file, indent,  "%u,  /* %s */\n", func->procstring_offset, func->name );
 970     }
 971     indent--;
 972     print_file( file, indent,  "};\n\n" );
 973 }
 974
 975 static int write_base_type(FILE *file, const type_t *type, int convert_to_signed_type, unsigned int *typestring_offset)
 976 {
 977     unsigned char fc;
 978
 979     if (type_get_type(type) == TYPE_BASIC)
 980         fc = get_basic_fc(type);
 981     else if (type_get_type(type) == TYPE_ENUM)
 982         fc = get_enum_fc(type);
 983     else
 984         return 0;
 985
 986     if (convert_to_signed_type)
 987     {
 988         switch(fc)
 989         {
 990         case RPC_FC_USMALL:
 991             fc = RPC_FC_SMALL;
 992             break;
 993         case RPC_FC_USHORT:
 994             fc = RPC_FC_SHORT;
 995             break;
 996         case RPC_FC_ULONG:
 997             fc = RPC_FC_LONG;
 998             break;
 999         }
1000     }
1001
1002     print_file(file, 2, "0x%02x,\t/* %s */\n", fc, string_of_type(fc));
1003     *typestring_offset += 1;
1004     return 1;
1005 }
1006
1007 /* write conformance / variance descriptor */
1008 static unsigned int write_conf_or_var_desc(FILE *file, const type_t *structure,
1009                                            unsigned int baseoff, const type_t *type,
1010                                            const expr_t *expr)
1011 {
1012     unsigned char operator_type = 0;
1013     unsigned char conftype = RPC_FC_NORMAL_CONFORMANCE;
1014     const char *conftype_string = "";
1015     const expr_t *subexpr;
1016
1017     if (!expr)
1018     {
1019         print_file(file, 2, "NdrFcLong(0xffffffff),\t/* -1 */\n");
1020         return 4;
1021     }
1022
1023     if (expr->is_const)
1024     {
1025         if (expr->cval > UCHAR_MAX * (USHRT_MAX + 1) + USHRT_MAX)
1026             error("write_conf_or_var_desc: constant value %d is greater than "
1027                   "the maximum constant size of %d\n", expr->cval,
1028                   UCHAR_MAX * (USHRT_MAX + 1) + USHRT_MAX);
1029
1030         print_file(file, 2, "0x%x, /* Corr desc: constant, val = %d */\n",
1031                    RPC_FC_CONSTANT_CONFORMANCE, expr->cval);
1032         print_file(file, 2, "0x%x,\n", expr->cval >> 16);
1033         print_file(file, 2, "NdrFcShort(0x%hx),\n", (unsigned short)expr->cval);
1034
1035         return 4;
1036     }
1037
1038     if (!structure)
1039     {
1040         /* Top-level conformance calculations are done inline.  */
1041         print_file (file, 2, "0x%x,\t/* Corr desc: parameter */\n",
1042                     RPC_FC_TOP_LEVEL_CONFORMANCE);
1043         print_file (file, 2, "0x0,\n");
1044         print_file (file, 2, "NdrFcShort(0x0),\n");
1045         return 4;
1046     }
1047
1048     if (is_ptr(type) || (is_array(type) && type_array_is_decl_as_ptr(type)))
1049     {
1050         conftype = RPC_FC_POINTER_CONFORMANCE;
1051         conftype_string = "field pointer, ";
1052     }
1053
1054     subexpr = expr;
1055     switch (subexpr->type)
1056     {
1057     case EXPR_PPTR:
1058         subexpr = subexpr->ref;
1059         operator_type = RPC_FC_DEREFERENCE;
1060         break;
1061     case EXPR_DIV:
1062         if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 2))
1063         {
1064             subexpr = subexpr->ref;
1065             operator_type = RPC_FC_DIV_2;
1066         }
1067         break;
1068     case EXPR_MUL:
1069         if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 2))
1070         {
1071             subexpr = subexpr->ref;
1072             operator_type = RPC_FC_MULT_2;
1073         }
1074         break;
1075     case EXPR_SUB:
1076         if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 1))
1077         {
1078             subexpr = subexpr->ref;
1079             operator_type = RPC_FC_SUB_1;
1080         }
1081         break;
1082     case EXPR_ADD:
1083         if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 1))
1084         {
1085             subexpr = subexpr->ref;
1086             operator_type = RPC_FC_ADD_1;
1087         }
1088         break;
1089     default:
1090         break;
1091     }
1092
1093     if (subexpr->type == EXPR_IDENTIFIER)
1094     {
1095         const type_t *correlation_variable = NULL;
1096         unsigned char param_type = 0;
1097         unsigned int offset = 0;
1098         const var_t *var;
1099         var_list_t *fields = type_struct_get_fields(structure);
1100
1101         if (fields) LIST_FOR_EACH_ENTRY( var, fields, const var_t, entry )
1102         {
1103             unsigned int size = field_memsize( var->type, &offset );
1104             if (var->name && !strcmp(var->name, subexpr->u.sval))
1105             {
1106                 correlation_variable = var->type;
1107                 break;
1108             }
1109             offset += size;
1110         }
1111         if (!correlation_variable)
1112             error("write_conf_or_var_desc: couldn't find variable %s in structure\n",
1113                   subexpr->u.sval);
1114
1115         correlation_variable = expr_resolve_type(NULL, structure, expr);
1116
1117         offset -= baseoff;
1118
1119         if (type_get_type(correlation_variable) == TYPE_BASIC)
1120         {
1121             switch (get_basic_fc(correlation_variable))
1122             {
1123             case RPC_FC_CHAR:
1124             case RPC_FC_SMALL:
1125                 param_type = RPC_FC_SMALL;
1126                 break;
1127             case RPC_FC_BYTE:
1128             case RPC_FC_USMALL:
1129                 param_type = RPC_FC_USMALL;
1130                 break;
1131             case RPC_FC_WCHAR:
1132             case RPC_FC_SHORT:
1133                 param_type = RPC_FC_SHORT;
1134                 break;
1135             case RPC_FC_USHORT:
1136                 param_type = RPC_FC_USHORT;
1137                 break;
1138             case RPC_FC_LONG:
1139                 param_type = RPC_FC_LONG;
1140                 break;
1141             case RPC_FC_ULONG:
1142                 param_type = RPC_FC_ULONG;
1143                 break;
1144             default:
1145                 error("write_conf_or_var_desc: conformance variable type not supported 0x%x\n",
1146                       get_basic_fc(correlation_variable));
1147             }
1148         }
1149         else if (type_get_type(correlation_variable) == TYPE_ENUM)
1150         {
1151             if (get_enum_fc(correlation_variable) == RPC_FC_ENUM32)
1152                 param_type = RPC_FC_LONG;
1153             else
1154                 param_type = RPC_FC_SHORT;
1155         }
1156         else if (type_get_type(correlation_variable) == TYPE_POINTER)
1157         {
1158             if (pointer_size == 8)
1159                 param_type = RPC_FC_HYPER;
1160             else
1161                 param_type = RPC_FC_LONG;
1162         }
1163         else
1164         {
1165             error("write_conf_or_var_desc: non-arithmetic type used as correlation variable %s\n",
1166                   subexpr->u.sval);
1167             return 0;
1168         }
1169
1170         print_file(file, 2, "0x%x, /* Corr desc: %s%s */\n",
1171                    conftype | param_type, conftype_string, string_of_type(param_type));
1172         print_file(file, 2, "0x%x, /* %s */\n", operator_type,
1173                    operator_type ? string_of_type(operator_type) : "no operators");
1174         print_file(file, 2, "NdrFcShort(0x%hx),\t/* offset = %d */\n",
1175                    (unsigned short)offset, offset);
1176     }
1177     else
1178     {
1179         unsigned int callback_offset = 0;
1180         struct expr_eval_routine *eval;
1181         int found = 0;
1182
1183         LIST_FOR_EACH_ENTRY(eval, &expr_eval_routines, struct expr_eval_routine, entry)
1184         {
1185             if (eval->structure == structure ||
1186                 (eval->structure->name && structure->name &&
1187                  !strcmp(eval->structure->name, structure->name) &&
1188                  !compare_expr(eval->expr, expr)))
1189             {
1190                 found = 1;
1191                 break;
1192             }
1193             callback_offset++;
1194         }
1195
1196         if (!found)
1197         {
1198             eval = xmalloc (sizeof(*eval));
1199             eval->structure = structure;
1200             eval->baseoff = baseoff;
1201             eval->expr = expr;
1202             list_add_tail (&expr_eval_routines, &eval->entry);
1203         }
1204
1205         if (callback_offset > USHRT_MAX)
1206             error("Maximum number of callback routines reached\n");
1207
1208         print_file(file, 2, "0x%x, /* Corr desc: %s */\n", conftype, conftype_string);
1209         print_file(file, 2, "0x%x, /* %s */\n", RPC_FC_CALLBACK, "FC_CALLBACK");
1210         print_file(file, 2, "NdrFcShort(0x%hx),\t/* %u */\n", (unsigned short)callback_offset, callback_offset);
1211     }
1212     return 4;
1213 }
1214
1215 /* return size and start offset of a data field based on current offset */
1216 static unsigned int field_memsize(const type_t *type, unsigned int *offset)
1217 {
1218     unsigned int align = 0;
1219     unsigned int size = type_memsize_and_alignment( type, &align );
1220
1221     *offset = ROUND_SIZE( *offset, align );
1222     return size;
1223 }
1224
1225 static unsigned int fields_memsize(const var_list_t *fields, unsigned int *align)
1226 {
1227     unsigned int size = 0;
1228     unsigned int max_align;
1229     const var_t *v;
1230
1231     if (!fields) return 0;
1232     LIST_FOR_EACH_ENTRY( v, fields, const var_t, entry )
1233     {
1234         unsigned int falign = 0;
1235         unsigned int fsize = type_memsize_and_alignment(v->type, &falign);
1236         if (*align < falign) *align = falign;
1237         falign = clamp_align(falign);
1238         size = ROUND_SIZE(size, falign);
1239         size += fsize;
1240     }
1241
1242     max_align = clamp_align(*align);
1243     size = ROUND_SIZE(size, max_align);
1244
1245     return size;
1246 }
1247
1248 static unsigned int union_memsize(const var_list_t *fields, unsigned int *pmaxa)
1249 {
1250     unsigned int size, maxs = 0;
1251     unsigned int align = *pmaxa;
1252     const var_t *v;
1253
1254     if (fields) LIST_FOR_EACH_ENTRY( v, fields, const var_t, entry )
1255     {
1256         /* we could have an empty default field with NULL type */
1257         if (v->type)
1258         {
1259             size = type_memsize_and_alignment(v->type, &align);
1260             if (maxs < size) maxs = size;
1261             if (*pmaxa < align) *pmaxa = align;
1262         }
1263     }
1264
1265     return maxs;
1266 }
1267
1268 static unsigned int type_memsize_and_alignment(const type_t *t, unsigned int *align)
1269 {
1270     unsigned int size = 0;
1271
1272     switch (type_get_type(t))
1273     {
1274     case TYPE_BASIC:
1275         switch (get_basic_fc(t))
1276         {
1277         case RPC_FC_BYTE:
1278         case RPC_FC_CHAR:
1279         case RPC_FC_USMALL:
1280         case RPC_FC_SMALL:
1281             size = 1;
1282             if (size > *align) *align = size;
1283             break;
1284         case RPC_FC_WCHAR:
1285         case RPC_FC_USHORT:
1286         case RPC_FC_SHORT:
1287             size = 2;
1288             if (size > *align) *align = size;
1289             break;
1290         case RPC_FC_ULONG:
1291         case RPC_FC_LONG:
1292         case RPC_FC_ERROR_STATUS_T:
1293         case RPC_FC_FLOAT:
1294             size = 4;
1295             if (size > *align) *align = size;
1296             break;
1297         case RPC_FC_HYPER:
1298         case RPC_FC_DOUBLE:
1299             size = 8;
1300             if (size > *align) *align = size;
1301             break;
1302         case RPC_FC_INT3264:
1303         case RPC_FC_UINT3264:
1304             assert( pointer_size );
1305             size = pointer_size;
1306             if (size > *align) *align = size;
1307             break;
1308         default:
1309             error("type_memsize: Unknown type 0x%x\n", get_basic_fc(t));
1310             size = 0;
1311         }
1312         break;
1313     case TYPE_ENUM:
1314         switch (get_enum_fc(t))
1315         {
1316         case RPC_FC_ENUM16:
1317         case RPC_FC_ENUM32:
1318             size = 4;
1319             if (size > *align) *align = size;
1320             break;
1321         default:
1322             error("type_memsize: Unknown enum type\n");
1323             size = 0;
1324         }
1325         break;
1326     case TYPE_STRUCT:
1327         size = fields_memsize(type_struct_get_fields(t), align);
1328         break;
1329     case TYPE_ENCAPSULATED_UNION:
1330         size = fields_memsize(type_encapsulated_union_get_fields(t), align);
1331         break;
1332     case TYPE_UNION:
1333         size = union_memsize(type_union_get_cases(t), align);
1334         break;
1335     case TYPE_POINTER:
1336         assert( pointer_size );
1337         size = pointer_size;
1338         if (size > *align) *align = size;
1339         break;
1340     case TYPE_ARRAY:
1341         if (!type_array_is_decl_as_ptr(t))
1342         {
1343             if (is_conformant_array(t))
1344             {
1345                 type_memsize_and_alignment(type_array_get_element(t), align);
1346                 size = 0;
1347             }
1348             else
1349                 size = type_array_get_dim(t) *
1350                     type_memsize_and_alignment(type_array_get_element(t), align);
1351         }
1352         else /* declared as a pointer */
1353         {
1354             assert( pointer_size );
1355             size = pointer_size;
1356             if (size > *align) *align = size;
1357         }
1358         break;
1359     case TYPE_INTERFACE:
1360     case TYPE_ALIAS:
1361     case TYPE_VOID:
1362     case TYPE_COCLASS:
1363     case TYPE_MODULE:
1364     case TYPE_FUNCTION:
1365     case TYPE_BITFIELD:
1366         /* these types should not be encountered here due to language
1367          * restrictions (interface, void, coclass, module), logical
1368          * restrictions (alias - due to type_get_type call above) or
1369          * checking restrictions (function, bitfield). */
1370         assert(0);
1371     }
1372
1373     return size;
1374 }
1375
1376 unsigned int type_memsize(const type_t *t)
1377 {
1378     unsigned int align = 0;
1379     return type_memsize_and_alignment( t, &align );
1380 }
1381
1382 static unsigned int type_buffer_alignment(const type_t *t)
1383 {
1384     const var_list_t *fields;
1385     const var_t *var;
1386     unsigned int max = 0, align;
1387
1388     switch (type_get_type(t))
1389     {
1390     case TYPE_BASIC:
1391         switch (get_basic_fc(t))
1392         {
1393         case RPC_FC_BYTE:
1394         case RPC_FC_CHAR:
1395         case RPC_FC_USMALL:
1396         case RPC_FC_SMALL:
1397             return 1;
1398         case RPC_FC_WCHAR:
1399         case RPC_FC_USHORT:
1400         case RPC_FC_SHORT:
1401             return 2;
1402         case RPC_FC_ULONG:
1403         case RPC_FC_LONG:
1404         case RPC_FC_ERROR_STATUS_T:
1405         case RPC_FC_FLOAT:
1406         case RPC_FC_INT3264:
1407         case RPC_FC_UINT3264:
1408             return 4;
1409         case RPC_FC_HYPER:
1410         case RPC_FC_DOUBLE:
1411             return 8;
1412         default:
1413             error("type_buffer_alignment: Unknown type 0x%x\n", get_basic_fc(t));
1414         }
1415         break;
1416     case TYPE_ENUM:
1417         switch (get_enum_fc(t))
1418         {
1419         case RPC_FC_ENUM16:
1420             return 2;
1421         case RPC_FC_ENUM32:
1422             return 4;
1423         default:
1424             error("type_buffer_alignment: Unknown enum type\n");
1425         }
1426         break;
1427     case TYPE_STRUCT:
1428         if (!(fields = type_struct_get_fields(t))) break;
1429         LIST_FOR_EACH_ENTRY( var, fields, const var_t, entry )
1430         {
1431             if (!var->type) continue;
1432             align = type_buffer_alignment( var->type );
1433             if (max < align) max = align;
1434         }
1435         break;
1436     case TYPE_ENCAPSULATED_UNION:
1437         if (!(fields = type_encapsulated_union_get_fields(t))) break;
1438         LIST_FOR_EACH_ENTRY( var, fields, const var_t, entry )
1439         {
1440             if (!var->type) continue;
1441             align = type_buffer_alignment( var->type );
1442             if (max < align) max = align;
1443         }
1444         break;
1445     case TYPE_UNION:
1446         if (!(fields = type_union_get_cases(t))) break;
1447         LIST_FOR_EACH_ENTRY( var, fields, const var_t, entry )
1448         {
1449             if (!var->type) continue;
1450             align = type_buffer_alignment( var->type );
1451             if (max < align) max = align;
1452         }
1453         break;
1454     case TYPE_ARRAY:
1455         if (!type_array_is_decl_as_ptr(t))
1456             return type_buffer_alignment( type_array_get_element(t) );
1457         /* else fall through */
1458     case TYPE_POINTER:
1459         return 4;
1460     case TYPE_INTERFACE:
1461     case TYPE_ALIAS:
1462     case TYPE_VOID:
1463     case TYPE_COCLASS:
1464     case TYPE_MODULE:
1465     case TYPE_FUNCTION:
1466     case TYPE_BITFIELD:
1467         /* these types should not be encountered here due to language
1468          * restrictions (interface, void, coclass, module), logical
1469          * restrictions (alias - due to type_get_type call above) or
1470          * checking restrictions (function, bitfield). */
1471         assert(0);
1472     }
1473     return max;
1474 }
1475
1476 int is_full_pointer_function(const var_t *func)
1477 {
1478     const var_t *var;
1479     if (type_has_full_pointer(type_function_get_rettype(func->type), func->attrs, TRUE))
1480         return TRUE;
1481     if (!type_get_function_args(func->type))
1482         return FALSE;
1483     LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
1484         if (type_has_full_pointer( var->type, var->attrs, TRUE ))
1485             return TRUE;
1486     return FALSE;
1487 }
1488
1489 void write_full_pointer_init(FILE *file, int indent, const var_t *func, int is_server)
1490 {
1491     print_file(file, indent, "__frame->_StubMsg.FullPtrXlatTables = NdrFullPointerXlatInit(0,%s);\n",
1492                    is_server ? "XLAT_SERVER" : "XLAT_CLIENT");
1493     fprintf(file, "\n");
1494 }
1495
1496 void write_full_pointer_free(FILE *file, int indent, const var_t *func)
1497 {
1498     print_file(file, indent, "NdrFullPointerXlatFree(__frame->_StubMsg.FullPtrXlatTables);\n");
1499     fprintf(file, "\n");
1500 }
1501
1502 static unsigned int write_nonsimple_pointer(FILE *file, const attr_list_t *attrs,
1503                                             const type_t *type,
1504                                             int toplevel_param,
1505                                             unsigned int offset,
1506                                             unsigned int *typeformat_offset)
1507 {
1508     unsigned int start_offset = *typeformat_offset;
1509     short reloff = offset - (*typeformat_offset + 2);
1510     int in_attr, out_attr;
1511     int pointer_type;
1512     unsigned char flags = 0;
1513
1514     pointer_type = get_pointer_fc(type, attrs, toplevel_param);
1515
1516     in_attr = is_attr(attrs, ATTR_IN);
1517     out_attr = is_attr(attrs, ATTR_OUT);
1518     if (!in_attr && !out_attr) in_attr = 1;
1519
1520     if (out_attr && !in_attr && pointer_type == RPC_FC_RP)
1521         flags |= RPC_FC_P_ONSTACK;
1522
1523     if (is_ptr(type))
1524     {
1525         type_t *ref = type_pointer_get_ref(type);
1526         if(is_declptr(ref) && !is_user_type(ref))
1527             flags |= RPC_FC_P_DEREF;
1528     }
1529
1530     print_file(file, 2, "0x%x, 0x%x,\t\t/* %s",
1531                pointer_type,
1532                flags,
1533                string_of_type(pointer_type));
1534     if (file)
1535     {
1536         if (flags & RPC_FC_P_ONSTACK)
1537             fprintf(file, " [allocated_on_stack]");
1538         if (flags & RPC_FC_P_DEREF)
1539             fprintf(file, " [pointer_deref]");
1540         fprintf(file, " */\n");
1541     }
1542
1543     print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n", reloff, reloff, offset);
1544     *typeformat_offset += 4;
1545
1546     return start_offset;
1547 }
1548
1549 static unsigned int write_simple_pointer(FILE *file, const attr_list_t *attrs, const type_t *type, int toplevel_param)
1550 {
1551     unsigned char fc;
1552     unsigned char pointer_fc;
1553     const type_t *ref;
1554     int in_attr = is_attr(attrs, ATTR_IN);
1555     int out_attr = is_attr(attrs, ATTR_OUT);
1556     unsigned char flags = RPC_FC_P_SIMPLEPOINTER;
1557
1558     /* for historical reasons, write_simple_pointer also handled string types,
1559      * but no longer does. catch bad uses of the function with this check */
1560     if (is_string_type(attrs, type))
1561         error("write_simple_pointer: can't handle type %s which is a string type\n", type->name);
1562
1563     pointer_fc = get_pointer_fc(type, attrs, toplevel_param);
1564
1565     ref = type_pointer_get_ref(type);
1566     if (type_get_type(ref) == TYPE_ENUM)
1567         fc = get_enum_fc(ref);
1568     else
1569         fc = get_basic_fc(ref);
1570
1571     if (out_attr && !in_attr)
1572         flags |= RPC_FC_P_ONSTACK;
1573
1574     print_file(file, 2, "0x%02x, 0x%x,\t/* %s %s[simple_pointer] */\n",
1575                pointer_fc, flags, string_of_type(pointer_fc),
1576                flags & RPC_FC_P_ONSTACK ? "[allocated_on_stack] " : "");
1577     print_file(file, 2, "0x%02x,\t/* %s */\n", fc, string_of_type(fc));
1578     print_file(file, 2, "0x5c,\t/* FC_PAD */\n");
1579     return 4;
1580 }
1581
1582 static void print_start_tfs_comment(FILE *file, type_t *t, unsigned int tfsoff)
1583 {
1584     print_file(file, 0, "/* %u (", tfsoff);
1585     write_type_decl(file, t, NULL);
1586     print_file(file, 0, ") */\n");
1587 }
1588
1589 static unsigned int write_pointer_tfs(FILE *file, const attr_list_t *attrs,
1590                                       type_t *type, unsigned int ref_offset,
1591                                       int toplevel_param,
1592                                       unsigned int *typestring_offset)
1593 {
1594     unsigned int offset = *typestring_offset;
1595     type_t *ref = type_pointer_get_ref(type);
1596
1597     print_start_tfs_comment(file, type, offset);
1598     update_tfsoff(type, offset, file);
1599
1600     switch (typegen_detect_type(ref, attrs, TDT_ALL_TYPES))
1601     {
1602     case TGT_BASIC:
1603     case TGT_ENUM:
1604         *typestring_offset += write_simple_pointer(file, attrs, type,
1605                                                    toplevel_param);
1606         break;
1607     default:
1608         if (ref_offset)
1609             write_nonsimple_pointer(file, attrs, type,
1610                                     toplevel_param,
1611                                     ref_offset,
1612                                     typestring_offset);
1613         break;
1614     }
1615
1616     return offset;
1617 }
1618
1619 static int processed(const type_t *type)
1620 {
1621     return type->typestring_offset && !type->tfswrite;
1622 }
1623
1624 static int user_type_has_variable_size(const type_t *t)
1625 {
1626     if (is_ptr(t))
1627         return TRUE;
1628     else if (type_get_type(t) == TYPE_STRUCT)
1629     {
1630         switch (get_struct_fc(t))
1631         {
1632         case RPC_FC_PSTRUCT:
1633         case RPC_FC_CSTRUCT:
1634         case RPC_FC_CPSTRUCT:
1635         case RPC_FC_CVSTRUCT:
1636             return TRUE;
1637         }
1638     }
1639     /* Note: Since this only applies to user types, we can't have a conformant
1640        array here, and strings should get filed under pointer in this case.  */
1641     return FALSE;
1642 }
1643
1644 static void write_user_tfs(FILE *file, type_t *type, unsigned int *tfsoff)
1645 {
1646     unsigned int start, absoff, flags;
1647     const char *name = NULL;
1648     type_t *utype = get_user_type(type, &name);
1649     unsigned int usize = type_memsize(utype);
1650     unsigned int ualign = type_buffer_alignment(utype);
1651     unsigned int size = type_memsize(type);
1652     unsigned short funoff = user_type_offset(name);
1653     short reloff;
1654
1655     guard_rec(type);
1656
1657     if(user_type_has_variable_size(utype)) usize = 0;
1658
1659     if (type_get_type(utype) == TYPE_BASIC ||
1660         type_get_type(utype) == TYPE_ENUM)
1661     {
1662         unsigned char fc;
1663
1664         if (type_get_type(utype) == TYPE_ENUM)
1665             fc = get_enum_fc(utype);
1666         else
1667             fc = get_basic_fc(utype);
1668
1669         absoff = *tfsoff;
1670         print_start_tfs_comment(file, utype, absoff);
1671         print_file(file, 2, "0x%x,\t/* %s */\n", fc, string_of_type(fc));
1672         print_file(file, 2, "0x5c,\t/* FC_PAD */\n");
1673         *tfsoff += 2;
1674     }
1675     else
1676     {
1677         if (!processed(utype))
1678             write_embedded_types(file, NULL, utype, utype->name, TRUE, tfsoff);
1679         absoff = utype->typestring_offset;
1680     }
1681
1682     if (type_get_type(utype) == TYPE_POINTER && get_pointer_fc(utype, NULL, FALSE) == RPC_FC_RP)
1683         flags = 0x40;
1684     else if (type_get_type(utype) == TYPE_POINTER && get_pointer_fc(utype, NULL, FALSE) == RPC_FC_UP)
1685         flags = 0x80;
1686     else
1687         flags = 0;
1688
1689     start = *tfsoff;
1690     update_tfsoff(type, start, file);
1691     print_start_tfs_comment(file, type, start);
1692     print_file(file, 2, "0x%x,\t/* FC_USER_MARSHAL */\n", RPC_FC_USER_MARSHAL);
1693     print_file(file, 2, "0x%x,\t/* Alignment= %d, Flags= %02x */\n",
1694                flags | (ualign - 1), ualign - 1, flags);
1695     print_file(file, 2, "NdrFcShort(0x%hx),\t/* Function offset= %hu */\n", funoff, funoff);
1696     print_file(file, 2, "NdrFcShort(0x%hx),\t/* %u */\n", (unsigned short)size, size);
1697     print_file(file, 2, "NdrFcShort(0x%hx),\t/* %u */\n", (unsigned short)usize, usize);
1698     *tfsoff += 8;
1699     reloff = absoff - *tfsoff;
1700     print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n", reloff, reloff, absoff);
1701     *tfsoff += 2;
1702 }
1703
1704 static void write_member_type(FILE *file, const type_t *cont,
1705                               int cont_is_complex, const attr_list_t *attrs,
1706                               const type_t *type, unsigned int *corroff,
1707                               unsigned int *tfsoff)
1708 {
1709     if (is_embedded_complex(type) && !is_conformant_array(type))
1710     {
1711         unsigned int absoff;
1712         short reloff;
1713
1714         if (type_get_type(type) == TYPE_UNION && is_attr(attrs, ATTR_SWITCHIS))
1715         {
1716             absoff = *corroff;
1717             *corroff += 8;
1718         }
1719         else
1720         {
1721             absoff = type->typestring_offset;
1722         }
1723         reloff = absoff - (*tfsoff + 2);
1724
1725         print_file(file, 2, "0x4c,\t/* FC_EMBEDDED_COMPLEX */\n");
1726         /* padding is represented using FC_STRUCTPAD* types, so presumably
1727          * this is left over in the format for historical purposes in MIDL
1728          * or rpcrt4. */
1729         print_file(file, 2, "0x0,\n");
1730         print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n",
1731                    reloff, reloff, absoff);
1732         *tfsoff += 4;
1733     }
1734     else if (is_ptr(type) || is_conformant_array(type))
1735     {
1736         unsigned char fc = cont_is_complex ? RPC_FC_POINTER : RPC_FC_LONG;
1737         print_file(file, 2, "0x%x,\t/* %s */\n", fc, string_of_type(fc));
1738         *tfsoff += 1;
1739     }
1740     else if (!write_base_type(file, type, TRUE, tfsoff))
1741         error("Unsupported member type %d\n", type_get_type(type));
1742 }
1743
1744 static void write_array_element_type(FILE *file, const type_t *type,
1745                                      int cont_is_complex, unsigned int *tfsoff)
1746 {
1747     type_t *elem = type_array_get_element(type);
1748
1749     if (!is_embedded_complex(elem) && is_ptr(elem))
1750     {
1751         type_t *ref = type_pointer_get_ref(elem);
1752
1753         if (processed(ref))
1754         {
1755             write_nonsimple_pointer(file, NULL, elem, FALSE, ref->typestring_offset, tfsoff);
1756             return;
1757         }
1758         if (!is_string_type(NULL, elem) &&
1759             (type_get_type(ref) == TYPE_BASIC || type_get_type(ref) == TYPE_ENUM))
1760         {
1761             *tfsoff += write_simple_pointer(file, NULL, elem, FALSE);
1762             return;
1763         }
1764     }
1765     return write_member_type(file, type, cont_is_complex, NULL, elem, NULL, tfsoff);
1766 }
1767
1768 static void write_end(FILE *file, unsigned int *tfsoff)
1769 {
1770     if (*tfsoff % 2 == 0)
1771     {
1772         print_file(file, 2, "0x%x,\t\t/* FC_PAD */\n", RPC_FC_PAD);
1773         *tfsoff += 1;
1774     }
1775     print_file(file, 2, "0x%x,\t\t/* FC_END */\n", RPC_FC_END);
1776     *tfsoff += 1;
1777 }
1778
1779 static void write_descriptors(FILE *file, type_t *type, unsigned int *tfsoff)
1780 {
1781     unsigned int offset = 0;
1782     var_list_t *fs = type_struct_get_fields(type);
1783     var_t *f;
1784
1785     if (fs) LIST_FOR_EACH_ENTRY(f, fs, var_t, entry)
1786     {
1787         type_t *ft = f->type;
1788         unsigned int size = field_memsize( ft, &offset );
1789         if (type_get_type(ft) == TYPE_UNION && is_attr(f->attrs, ATTR_SWITCHIS))
1790         {
1791             short reloff;
1792             unsigned int absoff = ft->typestring_offset;
1793             if (is_attr(ft->attrs, ATTR_SWITCHTYPE))
1794                 absoff += 8; /* we already have a corr descr, skip it */
1795             reloff = absoff - (*tfsoff + 6);
1796             print_file(file, 0, "/* %d */\n", *tfsoff);
1797             print_file(file, 2, "0x%x,\t/* FC_NON_ENCAPSULATED_UNION */\n", RPC_FC_NON_ENCAPSULATED_UNION);
1798             print_file(file, 2, "0x%x,\t/* FIXME: always FC_LONG */\n", RPC_FC_LONG);
1799             write_conf_or_var_desc(file, current_structure, offset, ft,
1800                                    get_attrp(f->attrs, ATTR_SWITCHIS));
1801             print_file(file, 2, "NdrFcShort(%hd),\t/* Offset= %hd (%u) */\n",
1802                        reloff, reloff, absoff);
1803             *tfsoff += 8;
1804         }
1805         offset += size;
1806     }
1807 }
1808
1809 static int write_pointer_description_offsets(
1810     FILE *file, const attr_list_t *attrs, type_t *type,
1811     unsigned int *offset_in_memory, unsigned int *offset_in_buffer,
1812     unsigned int *typestring_offset)
1813 {
1814     int written = 0;
1815
1816     if ((is_ptr(type) && type_get_type(type_pointer_get_ref(type)) != TYPE_INTERFACE) ||
1817         (is_array(type) && type_array_is_decl_as_ptr(type)))
1818     {
1819         if (offset_in_memory && offset_in_buffer)
1820         {
1821             unsigned int memsize;
1822
1823             /* pointer instance
1824              *
1825              * note that MSDN states that for pointer layouts in structures,
1826              * this is a negative offset from the end of the structure, but
1827              * this statement is incorrect. all offsets are positive */
1828             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Memory offset = %d */\n", (unsigned short)*offset_in_memory, *offset_in_memory);
1829             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Buffer offset = %d */\n", (unsigned short)*offset_in_buffer, *offset_in_buffer);
1830
1831             memsize = type_memsize(type);
1832             *offset_in_memory += memsize;
1833             /* increment these separately as in the case of conformant (varying)
1834              * structures these start at different values */
1835             *offset_in_buffer += memsize;
1836         }
1837         *typestring_offset += 4;
1838
1839         if (is_ptr(type))
1840         {
1841             type_t *ref = type_pointer_get_ref(type);
1842
1843             if (is_string_type(attrs, type))
1844                 write_string_tfs(file, attrs, type, FALSE, NULL, typestring_offset);
1845             else if (processed(ref))
1846                 write_nonsimple_pointer(file, attrs, type, FALSE, ref->typestring_offset, typestring_offset);
1847             else if (type_get_type(ref) == TYPE_BASIC || type_get_type(ref) == TYPE_ENUM)
1848                 *typestring_offset += write_simple_pointer(file, attrs, type, FALSE);
1849             else
1850                 error("write_pointer_description_offsets: type format string unknown\n");
1851         }
1852         else
1853         {
1854             unsigned int offset = type->typestring_offset;
1855             /* skip over the pointer that is written for strings, since a
1856              * pointer has to be written in-place here */
1857             if (is_string_type(attrs, type))
1858                 offset += 4;
1859             write_nonsimple_pointer(file, attrs, type, FALSE, offset, typestring_offset);
1860         }
1861
1862         return 1;
1863     }
1864
1865     if (is_array(type))
1866     {
1867         return write_pointer_description_offsets(
1868             file, attrs, type_array_get_element(type), offset_in_memory,
1869             offset_in_buffer, typestring_offset);
1870     }
1871     else if (is_non_complex_struct(type))
1872     {
1873         /* otherwise search for interesting fields to parse */
1874         const var_t *v;
1875         LIST_FOR_EACH_ENTRY( v, type_struct_get_fields(type), const var_t, entry )
1876         {
1877             if (offset_in_memory && offset_in_buffer)
1878             {
1879                 unsigned int padding;
1880                 unsigned int align = 0;
1881                 type_memsize_and_alignment(v->type, &align);
1882                 padding = ROUNDING(*offset_in_memory, align);
1883                 *offset_in_memory += padding;
1884                 *offset_in_buffer += padding;
1885             }
1886             written += write_pointer_description_offsets(
1887                 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
1888                 typestring_offset);
1889         }
1890     }
1891     else
1892     {
1893         if (offset_in_memory && offset_in_buffer)
1894         {
1895             unsigned int memsize = type_memsize(type);
1896             *offset_in_memory += memsize;
1897             /* increment these separately as in the case of conformant (varying)
1898              * structures these start at different values */
1899             *offset_in_buffer += memsize;
1900         }
1901     }
1902
1903     return written;
1904 }
1905
1906 static int write_no_repeat_pointer_descriptions(
1907     FILE *file, const attr_list_t *attrs, type_t *type,
1908     unsigned int *offset_in_memory, unsigned int *offset_in_buffer,
1909     unsigned int *typestring_offset)
1910 {
1911     int written = 0;
1912
1913     if (is_ptr(type) ||
1914         (is_conformant_array(type) && type_array_is_decl_as_ptr(type)))
1915     {
1916         print_file(file, 2, "0x%02x, /* FC_NO_REPEAT */\n", RPC_FC_NO_REPEAT);
1917         print_file(file, 2, "0x%02x, /* FC_PAD */\n", RPC_FC_PAD);
1918         *typestring_offset += 2;
1919
1920         return write_pointer_description_offsets(file, attrs, type,
1921                        offset_in_memory, offset_in_buffer, typestring_offset);
1922     }
1923
1924     if (is_non_complex_struct(type))
1925     {
1926         const var_t *v;
1927         LIST_FOR_EACH_ENTRY( v, type_struct_get_fields(type), const var_t, entry )
1928         {
1929             if (offset_in_memory && offset_in_buffer)
1930             {
1931                 unsigned int padding;
1932                 unsigned int align = 0;
1933                 type_memsize_and_alignment(v->type, &align);
1934                 padding = ROUNDING(*offset_in_memory, align);
1935                 *offset_in_memory += padding;
1936                 *offset_in_buffer += padding;
1937             }
1938             written += write_no_repeat_pointer_descriptions(
1939                 file, v->attrs, v->type,
1940                 offset_in_memory, offset_in_buffer, typestring_offset);
1941         }
1942     }
1943     else
1944     {
1945         unsigned int memsize = type_memsize(type);
1946         *offset_in_memory += memsize;
1947         /* increment these separately as in the case of conformant (varying)
1948          * structures these start at different values */
1949         *offset_in_buffer += memsize;
1950     }
1951
1952     return written;
1953 }
1954
1955 /* Note: if file is NULL return value is number of pointers to write, else
1956  * it is the number of type format characters written */
1957 static int write_fixed_array_pointer_descriptions(
1958     FILE *file, const attr_list_t *attrs, type_t *type,
1959     unsigned int *offset_in_memory, unsigned int *offset_in_buffer,
1960     unsigned int *typestring_offset)
1961 {
1962     int pointer_count = 0;
1963
1964     if (type_get_type(type) == TYPE_ARRAY &&
1965         !type_array_has_conformance(type) && !type_array_has_variance(type))
1966     {
1967         unsigned int temp = 0;
1968         /* unfortunately, this needs to be done in two passes to avoid
1969          * writing out redundant FC_FIXED_REPEAT descriptions */
1970         pointer_count = write_pointer_description_offsets(
1971             NULL, attrs, type_array_get_element(type), NULL, NULL, &temp);
1972         if (pointer_count > 0)
1973         {
1974             unsigned int increment_size;
1975             unsigned int offset_of_array_pointer_mem = 0;
1976             unsigned int offset_of_array_pointer_buf = 0;
1977
1978             increment_size = type_memsize(type_array_get_element(type));
1979
1980             print_file(file, 2, "0x%02x, /* FC_FIXED_REPEAT */\n", RPC_FC_FIXED_REPEAT);
1981             print_file(file, 2, "0x%02x, /* FC_PAD */\n", RPC_FC_PAD);
1982             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Iterations = %d */\n", (unsigned short)type_array_get_dim(type), type_array_get_dim(type));
1983             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Increment = %d */\n", (unsigned short)increment_size, increment_size);
1984             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset to array = %d */\n", (unsigned short)*offset_in_memory, *offset_in_memory);
1985             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Number of pointers = %d */\n", (unsigned short)pointer_count, pointer_count);
1986             *typestring_offset += 10;
1987
1988             pointer_count = write_pointer_description_offsets(
1989                 file, attrs, type, &offset_of_array_pointer_mem,
1990                 &offset_of_array_pointer_buf, typestring_offset);
1991         }
1992     }
1993     else if (type_get_type(type) == TYPE_STRUCT)
1994     {
1995         const var_t *v;
1996         LIST_FOR_EACH_ENTRY( v, type_struct_get_fields(type), const var_t, entry )
1997         {
1998             if (offset_in_memory && offset_in_buffer)
1999             {
2000                 unsigned int padding;
2001                 unsigned int align = 0;
2002                 type_memsize_and_alignment(v->type, &align);
2003                 padding = ROUNDING(*offset_in_memory, align);
2004                 *offset_in_memory += padding;
2005                 *offset_in_buffer += padding;
2006             }
2007             pointer_count += write_fixed_array_pointer_descriptions(
2008                 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
2009                 typestring_offset);
2010         }
2011     }
2012     else
2013     {
2014         if (offset_in_memory && offset_in_buffer)
2015         {
2016             unsigned int memsize;
2017             memsize = type_memsize(type);
2018             *offset_in_memory += memsize;
2019             /* increment these separately as in the case of conformant (varying)
2020              * structures these start at different values */
2021             *offset_in_buffer += memsize;
2022         }
2023     }
2024
2025     return pointer_count;
2026 }
2027
2028 /* Note: if file is NULL return value is number of pointers to write, else
2029  * it is the number of type format characters written */
2030 static int write_conformant_array_pointer_descriptions(
2031     FILE *file, const attr_list_t *attrs, type_t *type,
2032     unsigned int offset_in_memory, unsigned int *typestring_offset)
2033 {
2034     int pointer_count = 0;
2035
2036     if (is_conformant_array(type) && !type_array_has_variance(type))
2037     {
2038         unsigned int temp = 0;
2039         /* unfortunately, this needs to be done in two passes to avoid
2040          * writing out redundant FC_VARIABLE_REPEAT descriptions */
2041         pointer_count = write_pointer_description_offsets(
2042             NULL, attrs, type_array_get_element(type), NULL, NULL, &temp);
2043         if (pointer_count > 0)
2044         {
2045             unsigned int increment_size;
2046             unsigned int offset_of_array_pointer_mem = offset_in_memory;
2047             unsigned int offset_of_array_pointer_buf = offset_in_memory;
2048
2049             increment_size = type_memsize(type_array_get_element(type));
2050
2051             if (increment_size > USHRT_MAX)
2052                 error("array size of %u bytes is too large\n", increment_size);
2053
2054             print_file(file, 2, "0x%02x, /* FC_VARIABLE_REPEAT */\n", RPC_FC_VARIABLE_REPEAT);
2055             print_file(file, 2, "0x%02x, /* FC_FIXED_OFFSET */\n", RPC_FC_FIXED_OFFSET);
2056             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Increment = %d */\n", (unsigned short)increment_size, increment_size);
2057             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset to array = %d */\n", (unsigned short)offset_in_memory, offset_in_memory);
2058             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Number of pointers = %d */\n", (unsigned short)pointer_count, pointer_count);
2059             *typestring_offset += 8;
2060
2061             pointer_count = write_pointer_description_offsets(
2062                 file, attrs, type_array_get_element(type),
2063                 &offset_of_array_pointer_mem, &offset_of_array_pointer_buf,
2064                 typestring_offset);
2065         }
2066     }
2067
2068     return pointer_count;
2069 }
2070
2071 /* Note: if file is NULL return value is number of pointers to write, else
2072  * it is the number of type format characters written */
2073 static int write_varying_array_pointer_descriptions(
2074     FILE *file, const attr_list_t *attrs, type_t *type,
2075     unsigned int *offset_in_memory, unsigned int *offset_in_buffer,
2076     unsigned int *typestring_offset)
2077 {
2078     int pointer_count = 0;
2079
2080     if (is_array(type) && type_array_has_variance(type))
2081     {
2082         unsigned int temp = 0;
2083         /* unfortunately, this needs to be done in two passes to avoid
2084          * writing out redundant FC_VARIABLE_REPEAT descriptions */
2085         pointer_count = write_pointer_description_offsets(
2086             NULL, attrs, type_array_get_element(type), NULL, NULL, &temp);
2087         if (pointer_count > 0)
2088         {
2089             unsigned int increment_size;
2090
2091             increment_size = type_memsize(type_array_get_element(type));
2092
2093             if (increment_size > USHRT_MAX)
2094                 error("array size of %u bytes is too large\n", increment_size);
2095
2096             print_file(file, 2, "0x%02x, /* FC_VARIABLE_REPEAT */\n", RPC_FC_VARIABLE_REPEAT);
2097             print_file(file, 2, "0x%02x, /* FC_VARIABLE_OFFSET */\n", RPC_FC_VARIABLE_OFFSET);
2098             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Increment = %d */\n", (unsigned short)increment_size, increment_size);
2099             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset to array = %d */\n", (unsigned short)*offset_in_memory, *offset_in_memory);
2100             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Number of pointers = %d */\n", (unsigned short)pointer_count, pointer_count);
2101             *typestring_offset += 8;
2102
2103             pointer_count = write_pointer_description_offsets(
2104                 file, attrs, type_array_get_element(type), offset_in_memory,
2105                 offset_in_buffer, typestring_offset);
2106         }
2107     }
2108     else if (type_get_type(type) == TYPE_STRUCT)
2109     {
2110         const var_t *v;
2111         LIST_FOR_EACH_ENTRY( v, type_struct_get_fields(type), const var_t, entry )
2112         {
2113             if (offset_in_memory && offset_in_buffer)
2114             {
2115                 unsigned int align = 0, padding;
2116
2117                 if (is_array(v->type) && type_array_has_variance(v->type))
2118                 {
2119                     *offset_in_buffer = ROUND_SIZE(*offset_in_buffer, 4);
2120                     /* skip over variance and offset in buffer */
2121                     *offset_in_buffer += 8;
2122                 }
2123
2124                 type_memsize_and_alignment(v->type, &align);
2125                 padding = ROUNDING(*offset_in_memory, align);
2126                 *offset_in_memory += padding;
2127                 *offset_in_buffer += padding;
2128             }
2129             pointer_count += write_varying_array_pointer_descriptions(
2130                 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
2131                 typestring_offset);
2132         }
2133     }
2134     else
2135     {
2136         if (offset_in_memory && offset_in_buffer)
2137         {
2138             unsigned int memsize = type_memsize(type);
2139             *offset_in_memory += memsize;
2140             /* increment these separately as in the case of conformant (varying)
2141              * structures these start at different values */
2142             *offset_in_buffer += memsize;
2143         }
2144     }
2145
2146     return pointer_count;
2147 }
2148
2149 static void write_pointer_description(FILE *file, type_t *type,
2150                                       unsigned int *typestring_offset)
2151 {
2152     unsigned int offset_in_buffer;
2153     unsigned int offset_in_memory;
2154
2155     /* pass 1: search for single instance of a pointer (i.e. don't descend
2156      * into arrays) */
2157     if (!is_array(type))
2158     {
2159         offset_in_memory = 0;
2160         offset_in_buffer = 0;
2161         write_no_repeat_pointer_descriptions(
2162             file, NULL, type,
2163             &offset_in_memory, &offset_in_buffer, typestring_offset);
2164     }
2165
2166     /* pass 2: search for pointers in fixed arrays */
2167     offset_in_memory = 0;
2168     offset_in_buffer = 0;
2169     write_fixed_array_pointer_descriptions(
2170         file, NULL, type,
2171         &offset_in_memory, &offset_in_buffer, typestring_offset);
2172
2173     /* pass 3: search for pointers in conformant only arrays (but don't descend
2174      * into conformant varying or varying arrays) */
2175     if (is_conformant_array(type) &&
2176         (type_array_is_decl_as_ptr(type) || !current_structure))
2177         write_conformant_array_pointer_descriptions(
2178             file, NULL, type, 0, typestring_offset);
2179     else if (type_get_type(type) == TYPE_STRUCT &&
2180              get_struct_fc(type) == RPC_FC_CPSTRUCT)
2181     {
2182         type_t *carray = find_array_or_string_in_struct(type)->type;
2183         write_conformant_array_pointer_descriptions( file, NULL, carray,
2184                                                      type_memsize(type), typestring_offset);
2185     }
2186
2187     /* pass 4: search for pointers in varying arrays */
2188     offset_in_memory = 0;
2189     offset_in_buffer = 0;
2190     write_varying_array_pointer_descriptions(
2191             file, NULL, type,
2192             &offset_in_memory, &offset_in_buffer, typestring_offset);
2193 }
2194
2195 int is_declptr(const type_t *t)
2196 {
2197   return is_ptr(t) || (type_get_type(t) == TYPE_ARRAY && type_array_is_decl_as_ptr(t));
2198 }
2199
2200 static unsigned int write_string_tfs(FILE *file, const attr_list_t *attrs,
2201                                      type_t *type, int toplevel_param,
2202                                      const char *name, unsigned int *typestring_offset)
2203 {
2204     unsigned int start_offset;
2205     unsigned char rtype;
2206     type_t *elem_type;
2207
2208     start_offset = *typestring_offset;
2209     update_tfsoff(type, start_offset, file);
2210
2211     if (is_declptr(type))
2212     {
2213         unsigned char flag = is_conformant_array(type) ? 0 : RPC_FC_P_SIMPLEPOINTER;
2214         int pointer_type = get_pointer_fc(type, attrs, toplevel_param);
2215         if (!pointer_type)
2216             pointer_type = RPC_FC_RP;
2217         print_start_tfs_comment(file, type, *typestring_offset);
2218         print_file(file, 2,"0x%x, 0x%x,\t/* %s%s */\n",
2219                    pointer_type, flag, string_of_type(pointer_type),
2220                    flag ? " [simple_pointer]" : "");
2221         *typestring_offset += 2;
2222         if (!flag)
2223         {
2224             print_file(file, 2, "NdrFcShort(0x2),\n");
2225             *typestring_offset += 2;
2226         }
2227     }
2228
2229     if (is_array(type))
2230         elem_type = type_array_get_element(type);
2231     else
2232         elem_type = type_pointer_get_ref(type);
2233
2234     if (type_get_type(elem_type) != TYPE_BASIC)
2235     {
2236         error("write_string_tfs: Unimplemented for non-basic type %s\n", name);
2237         return start_offset;
2238     }
2239
2240     rtype = get_basic_fc(elem_type);
2241     if ((rtype != RPC_FC_BYTE) && (rtype != RPC_FC_CHAR) && (rtype != RPC_FC_WCHAR))
2242     {
2243         error("write_string_tfs: Unimplemented for type 0x%x of name: %s\n", rtype, name);
2244         return start_offset;
2245     }
2246
2247     if (type_get_type(type) == TYPE_ARRAY && !type_array_has_conformance(type))
2248     {
2249         unsigned int dim = type_array_get_dim(type);
2250
2251         /* FIXME: multi-dimensional array */
2252         if (0xffffu < dim)
2253             error("array size for parameter %s exceeds %u bytes by %u bytes\n",
2254                   name, 0xffffu, dim - 0xffffu);
2255
2256         if (rtype == RPC_FC_WCHAR)
2257             WRITE_FCTYPE(file, FC_WSTRING, *typestring_offset);
2258         else
2259             WRITE_FCTYPE(file, FC_CSTRING, *typestring_offset);
2260         print_file(file, 2, "0x%x, /* FC_PAD */\n", RPC_FC_PAD);
2261         *typestring_offset += 2;
2262
2263         print_file(file, 2, "NdrFcShort(0x%hx),\t/* %d */\n", (unsigned short)dim, dim);
2264         *typestring_offset += 2;
2265
2266         return start_offset;
2267     }
2268     else if (is_conformant_array(type))
2269     {
2270         if (rtype == RPC_FC_WCHAR)
2271             WRITE_FCTYPE(file, FC_C_WSTRING, *typestring_offset);
2272         else
2273             WRITE_FCTYPE(file, FC_C_CSTRING, *typestring_offset);
2274         print_file(file, 2, "0x%x, /* FC_STRING_SIZED */\n", RPC_FC_STRING_SIZED);
2275         *typestring_offset += 2;
2276
2277         *typestring_offset += write_conf_or_var_desc(
2278             file, current_structure,
2279             (!type_array_is_decl_as_ptr(type) && current_structure
2280              ? type_memsize(current_structure)
2281              : 0),
2282             type, type_array_get_conformance(type));
2283
2284         return start_offset;
2285     }
2286     else
2287     {
2288         if (rtype == RPC_FC_WCHAR)
2289             WRITE_FCTYPE(file, FC_C_WSTRING, *typestring_offset);
2290         else
2291             WRITE_FCTYPE(file, FC_C_CSTRING, *typestring_offset);
2292         print_file(file, 2, "0x%x, /* FC_PAD */\n", RPC_FC_PAD);
2293         *typestring_offset += 2;
2294
2295         return start_offset;
2296     }
2297 }
2298
2299 static unsigned int write_array_tfs(FILE *file, const attr_list_t *attrs, type_t *type,
2300                                     const char *name, unsigned int *typestring_offset)
2301 {
2302     const expr_t *length_is = type_array_get_variance(type);
2303     const expr_t *size_is = type_array_get_conformance(type);
2304     unsigned int align;
2305     unsigned int size;
2306     unsigned int start_offset;
2307     unsigned char fc;
2308     int pointer_type = get_attrv(attrs, ATTR_POINTERTYPE);
2309     unsigned int baseoff
2310         = !type_array_is_decl_as_ptr(type) && current_structure
2311         ? type_memsize(current_structure)
2312         : 0;
2313
2314     if (!pointer_type)
2315         pointer_type = RPC_FC_RP;
2316
2317     write_embedded_types(file, attrs, type_array_get_element(type), name, FALSE, typestring_offset);
2318
2319     size = type_memsize(is_conformant_array(type) ? type_array_get_element(type) : type);
2320     align = type_buffer_alignment(is_conformant_array(type) ? type_array_get_element(type) : type);
2321     fc = get_array_fc(type);
2322
2323     start_offset = *typestring_offset;
2324     update_tfsoff(type, start_offset, file);
2325     print_start_tfs_comment(file, type, start_offset);
2326     print_file(file, 2, "0x%02x,\t/* %s */\n", fc, string_of_type(fc));
2327     print_file(file, 2, "0x%x,\t/* %d */\n", align - 1, align - 1);
2328     *typestring_offset += 2;
2329
2330     align = 0;
2331     if (fc != RPC_FC_BOGUS_ARRAY)
2332     {
2333         if (fc == RPC_FC_LGFARRAY || fc == RPC_FC_LGVARRAY)
2334         {
2335             print_file(file, 2, "NdrFcLong(0x%x),\t/* %u */\n", size, size);
2336             *typestring_offset += 4;
2337         }
2338         else
2339         {
2340             print_file(file, 2, "NdrFcShort(0x%hx),\t/* %u */\n", (unsigned short)size, size);
2341             *typestring_offset += 2;
2342         }
2343
2344         if (is_conformant_array(type))
2345             *typestring_offset
2346                 += write_conf_or_var_desc(file, current_structure, baseoff,
2347                                           type, size_is);
2348
2349         if (fc == RPC_FC_SMVARRAY || fc == RPC_FC_LGVARRAY)
2350         {
2351             unsigned int elsize = type_memsize(type_array_get_element(type));
2352             unsigned int dim = type_array_get_dim(type);
2353
2354             if (fc == RPC_FC_LGVARRAY)
2355             {
2356                 print_file(file, 2, "NdrFcLong(0x%x),\t/* %u */\n", dim, dim);
2357                 *typestring_offset += 4;
2358             }
2359             else
2360             {
2361                 print_file(file, 2, "NdrFcShort(0x%hx),\t/* %u */\n", (unsigned short)dim, dim);
2362                 *typestring_offset += 2;
2363             }
2364
2365             print_file(file, 2, "NdrFcShort(0x%hx),\t/* %u */\n", (unsigned short)elsize, elsize);
2366             *typestring_offset += 2;
2367         }
2368
2369         if (length_is)
2370             *typestring_offset
2371                 += write_conf_or_var_desc(file, current_structure, baseoff,
2372                                           type, length_is);
2373
2374         if (type_has_pointers(type_array_get_element(type)) &&
2375             (type_array_is_decl_as_ptr(type) || !current_structure))
2376         {
2377             print_file(file, 2, "0x%x, /* FC_PP */\n", RPC_FC_PP);
2378             print_file(file, 2, "0x%x, /* FC_PAD */\n", RPC_FC_PAD);
2379             *typestring_offset += 2;
2380             write_pointer_description(file, type, typestring_offset);
2381             print_file(file, 2, "0x%x, /* FC_END */\n", RPC_FC_END);
2382             *typestring_offset += 1;
2383         }
2384
2385         write_array_element_type(file, type, FALSE, typestring_offset);
2386         write_end(file, typestring_offset);
2387     }
2388     else
2389     {
2390         unsigned int dim = size_is ? 0 : type_array_get_dim(type);
2391         print_file(file, 2, "NdrFcShort(0x%hx),\t/* %u */\n", (unsigned short)dim, dim);
2392         *typestring_offset += 2;
2393         *typestring_offset
2394             += write_conf_or_var_desc(file, current_structure, baseoff,
2395                                       type, size_is);
2396         *typestring_offset
2397             += write_conf_or_var_desc(file, current_structure, baseoff,
2398                                       type, length_is);
2399
2400         write_array_element_type(file, type, TRUE, typestring_offset);
2401         write_end(file, typestring_offset);
2402     }
2403
2404     return start_offset;
2405 }
2406
2407 static const var_t *find_array_or_string_in_struct(const type_t *type)
2408 {
2409     const var_list_t *fields = type_struct_get_fields(type);
2410     const var_t *last_field;
2411     const type_t *ft;
2412
2413     if (!fields || list_empty(fields))
2414         return NULL;
2415
2416     last_field = LIST_ENTRY( list_tail(fields), const var_t, entry );
2417     ft = last_field->type;
2418
2419     if (is_conformant_array(ft) && !type_array_is_decl_as_ptr(ft))
2420         return last_field;
2421
2422     if (type_get_type(ft) == TYPE_STRUCT)
2423         return find_array_or_string_in_struct(ft);
2424     else
2425         return NULL;
2426 }
2427
2428 static void write_struct_members(FILE *file, const type_t *type,
2429                                  int is_complex, unsigned int *corroff,
2430                                  unsigned int *typestring_offset)
2431 {
2432     const var_t *field;
2433     unsigned short offset = 0;
2434     unsigned int salign = 1;
2435     int padding;
2436     var_list_t *fields = type_struct_get_fields(type);
2437
2438     if (fields) LIST_FOR_EACH_ENTRY( field, fields, const var_t, entry )
2439     {
2440         type_t *ft = field->type;
2441         unsigned int align = 0;
2442         unsigned int size = type_memsize_and_alignment(ft, &align);
2443         align = clamp_align(align);
2444         if (salign < align) salign = align;
2445
2446         if (!is_conformant_array(ft) || type_array_is_decl_as_ptr(ft))
2447         {
2448             if ((align - 1) & offset)
2449             {
2450                 unsigned char fc = 0;
2451                 switch (align)
2452                 {
2453                 case 2:
2454                     fc = RPC_FC_ALIGNM2;
2455                     break;
2456                 case 4:
2457                     fc = RPC_FC_ALIGNM4;
2458                     break;
2459                 case 8:
2460                     fc = RPC_FC_ALIGNM8;
2461                     break;
2462                 default:
2463                     error("write_struct_members: cannot align type %d\n", type_get_type(ft));
2464                 }
2465                 print_file(file, 2, "0x%x,\t/* %s */\n", fc, string_of_type(fc));
2466                 offset = ROUND_SIZE(offset, align);
2467                 *typestring_offset += 1;
2468             }
2469             write_member_type(file, type, is_complex, field->attrs, field->type, corroff,
2470                               typestring_offset);
2471             offset += size;
2472         }
2473     }
2474
2475     padding = ROUNDING(offset, salign);
2476     if (padding)
2477     {
2478         print_file(file, 2, "0x%x,\t/* FC_STRUCTPAD%d */\n",
2479                    RPC_FC_STRUCTPAD1 + padding - 1,
2480                    padding);
2481         *typestring_offset += 1;
2482     }
2483
2484     write_end(file, typestring_offset);
2485 }
2486
2487 static unsigned int write_struct_tfs(FILE *file, type_t *type,
2488                                      const char *name, unsigned int *tfsoff)
2489 {
2490     const type_t *save_current_structure = current_structure;
2491     unsigned int total_size;
2492     const var_t *array;
2493     unsigned int start_offset;
2494     unsigned int align;
2495     unsigned int corroff;
2496     var_t *f;
2497     unsigned char fc = get_struct_fc(type);
2498     var_list_t *fields = type_struct_get_fields(type);
2499
2500     guard_rec(type);
2501     current_structure = type;
2502
2503     total_size = type_memsize(type);
2504     align = type_buffer_alignment(type);
2505     if (total_size > USHRT_MAX)
2506         error("structure size for %s exceeds %d bytes by %d bytes\n",
2507               name, USHRT_MAX, total_size - USHRT_MAX);
2508
2509     if (fields) LIST_FOR_EACH_ENTRY(f, fields, var_t, entry)
2510         write_embedded_types(file, f->attrs, f->type, f->name, FALSE, tfsoff);
2511
2512     array = find_array_or_string_in_struct(type);
2513     if (array && !processed(array->type))
2514     {
2515         if(is_string_type(array->attrs, array->type))
2516             write_string_tfs(file, array->attrs, array->type, FALSE, array->name, tfsoff);
2517         else
2518             write_array_tfs(file, array->attrs, array->type, array->name, tfsoff);
2519     }
2520
2521     corroff = *tfsoff;
2522     write_descriptors(file, type, tfsoff);
2523
2524     start_offset = *tfsoff;
2525     update_tfsoff(type, start_offset, file);
2526     print_start_tfs_comment(file, type, start_offset);
2527     print_file(file, 2, "0x%x,\t/* %s */\n", fc, string_of_type(fc));
2528     print_file(file, 2, "0x%x,\t/* %d */\n", align - 1, align - 1);
2529     print_file(file, 2, "NdrFcShort(0x%hx),\t/* %d */\n", (unsigned short)total_size, total_size);
2530     *tfsoff += 4;
2531
2532     if (array)
2533     {
2534         unsigned int absoff = array->type->typestring_offset;
2535         short reloff = absoff - *tfsoff;
2536         print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n",
2537                    reloff, reloff, absoff);
2538         *tfsoff += 2;
2539     }
2540     else if (fc == RPC_FC_BOGUS_STRUCT)
2541     {
2542         print_file(file, 2, "NdrFcShort(0x0),\n");
2543         *tfsoff += 2;
2544     }
2545
2546     if (fc == RPC_FC_BOGUS_STRUCT)
2547     {
2548         /* On the sizing pass, type->ptrdesc may be zero, but it's ok as
2549            nothing is written to file yet.  On the actual writing pass,
2550            this will have been updated.  */
2551         unsigned int absoff = type->ptrdesc ? type->ptrdesc : *tfsoff;
2552         int reloff = absoff - *tfsoff;
2553         assert( reloff >= 0 );
2554         print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %d (%u) */\n",
2555                    (unsigned short)reloff, reloff, absoff);
2556         *tfsoff += 2;
2557     }
2558     else if ((fc == RPC_FC_PSTRUCT) ||
2559              (fc == RPC_FC_CPSTRUCT) ||
2560              (fc == RPC_FC_CVSTRUCT && type_has_pointers(type)))
2561     {
2562         print_file(file, 2, "0x%x, /* FC_PP */\n", RPC_FC_PP);
2563         print_file(file, 2, "0x%x, /* FC_PAD */\n", RPC_FC_PAD);
2564         *tfsoff += 2;
2565         write_pointer_description(file, type, tfsoff);
2566         print_file(file, 2, "0x%x, /* FC_END */\n", RPC_FC_END);
2567         *tfsoff += 1;
2568     }
2569
2570     write_struct_members(file, type, fc == RPC_FC_BOGUS_STRUCT, &corroff,
2571                          tfsoff);
2572
2573     if (fc == RPC_FC_BOGUS_STRUCT)
2574     {
2575         const var_t *f;
2576
2577         type->ptrdesc = *tfsoff;
2578         if (fields) LIST_FOR_EACH_ENTRY(f, fields, const var_t, entry)
2579         {
2580             type_t *ft = f->type;
2581             switch (typegen_detect_type(ft, f->attrs, TDT_IGNORE_STRINGS))
2582             {
2583             case TGT_POINTER:
2584                 if (is_string_type(f->attrs, ft))
2585                     write_string_tfs(file, f->attrs, ft, FALSE, f->name, tfsoff);
2586                 else
2587                     write_pointer_tfs(file, f->attrs, ft,
2588                                       type_pointer_get_ref(ft)->typestring_offset,
2589                                       FALSE, tfsoff);
2590                 break;
2591             case TGT_ARRAY:
2592                 if (type_array_is_decl_as_ptr(ft))
2593                 {
2594                     unsigned int offset;
2595
2596                     print_file(file, 0, "/* %d */\n", *tfsoff);
2597
2598                     offset = ft->typestring_offset;
2599                     /* skip over the pointer that is written for strings, since a
2600                      * pointer has to be written in-place here */
2601                     if (is_string_type(f->attrs, ft))
2602                         offset += 4;
2603                     write_nonsimple_pointer(file, f->attrs, ft, FALSE, offset, tfsoff);
2604                 }
2605                 break;
2606             default:
2607                 break;
2608             }
2609         }
2610         if (type->ptrdesc == *tfsoff)
2611             type->ptrdesc = 0;
2612     }
2613
2614     current_structure = save_current_structure;
2615     return start_offset;
2616 }
2617
2618 static void write_branch_type(FILE *file, const type_t *t, unsigned int *tfsoff)
2619 {
2620     if (t == NULL)
2621     {
2622         print_file(file, 2, "NdrFcShort(0x0),\t/* No type */\n");
2623     }
2624     else
2625     {
2626         if (type_get_type(t) == TYPE_BASIC || type_get_type(t) == TYPE_ENUM)
2627         {
2628             unsigned char fc;
2629             if (type_get_type(t) == TYPE_BASIC)
2630                 fc = get_basic_fc(t);
2631             else
2632                 fc = get_enum_fc(t);
2633             print_file(file, 2, "NdrFcShort(0x80%02x),\t/* Simple arm type: %s */\n",
2634                        fc, string_of_type(fc));
2635         }
2636         else if (t->typestring_offset)
2637         {
2638             short reloff = t->typestring_offset - *tfsoff;
2639             print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %d (%d) */\n",
2640                        reloff, reloff, t->typestring_offset);
2641         }
2642         else
2643             error("write_branch_type: type unimplemented %d\n", type_get_type(t));
2644     }
2645
2646     *tfsoff += 2;
2647 }
2648
2649 static unsigned int write_union_tfs(FILE *file, type_t *type, unsigned int *tfsoff)
2650 {
2651     unsigned int start_offset;
2652     unsigned int size;
2653     var_list_t *fields;
2654     unsigned int nbranch = 0;
2655     type_t *deftype = NULL;
2656     short nodeftype = 0xffff;
2657     var_t *f;
2658
2659     guard_rec(type);
2660
2661     size = type_memsize(type);
2662
2663     fields = type_union_get_cases(type);
2664
2665     if (fields) LIST_FOR_EACH_ENTRY(f, fields, var_t, entry)
2666     {
2667         expr_list_t *cases = get_attrp(f->attrs, ATTR_CASE);
2668         if (cases)
2669             nbranch += list_count(cases);
2670         if (f->type)
2671             write_embedded_types(file, f->attrs, f->type, f->name, TRUE, tfsoff);
2672     }
2673
2674     start_offset = *tfsoff;
2675     update_tfsoff(type, start_offset, file);
2676     print_start_tfs_comment(file, type, start_offset);
2677     if (type_get_type(type) == TYPE_ENCAPSULATED_UNION)
2678     {
2679         const var_t *sv = type_union_get_switch_value(type);
2680         const type_t *st = sv->type;
2681         unsigned char fc;
2682
2683         if (type_get_type(st) == TYPE_BASIC)
2684         {
2685             switch (get_basic_fc(st))
2686             {
2687             case RPC_FC_CHAR:
2688             case RPC_FC_SMALL:
2689             case RPC_FC_BYTE:
2690             case RPC_FC_USMALL:
2691             case RPC_FC_WCHAR:
2692             case RPC_FC_SHORT:
2693             case RPC_FC_USHORT:
2694             case RPC_FC_LONG:
2695             case RPC_FC_ULONG:
2696                 fc = get_basic_fc(st);
2697                 break;
2698             default:
2699                 fc = 0;
2700                 error("union switch type must be an integer, char, or enum\n");
2701             }
2702         }
2703         else if (type_get_type(st) == TYPE_ENUM)
2704             fc = get_enum_fc(st);
2705         else
2706             error("union switch type must be an integer, char, or enum\n");
2707
2708         print_file(file, 2, "0x%x,\t/* FC_ENCAPSULATED_UNION */\n", RPC_FC_ENCAPSULATED_UNION);
2709         print_file(file, 2, "0x%x,\t/* Switch type= %s */\n",
2710                    0x40 | fc, string_of_type(fc));
2711         *tfsoff += 2;
2712     }
2713     else if (is_attr(type->attrs, ATTR_SWITCHTYPE))
2714     {
2715         static const expr_t dummy_expr;  /* FIXME */
2716         const type_t *st = get_attrp(type->attrs, ATTR_SWITCHTYPE);
2717         unsigned char fc;
2718
2719         if (type_get_type(st) == TYPE_BASIC)
2720         {
2721             switch (get_basic_fc(st))
2722             {
2723             case RPC_FC_CHAR:
2724             case RPC_FC_SMALL:
2725             case RPC_FC_USMALL:
2726             case RPC_FC_SHORT:
2727             case RPC_FC_USHORT:
2728             case RPC_FC_LONG:
2729             case RPC_FC_ULONG:
2730             case RPC_FC_ENUM16:
2731             case RPC_FC_ENUM32:
2732                 fc = get_basic_fc(st);
2733                 break;
2734             default:
2735                 fc = 0;
2736                 error("union switch type must be an integer, char, or enum\n");
2737             }
2738         }
2739         else if (type_get_type(st) == TYPE_ENUM)
2740             fc = get_enum_fc(st);
2741         else
2742             error("union switch type must be an integer, char, or enum\n");
2743
2744         print_file(file, 2, "0x%x,\t/* FC_NON_ENCAPSULATED_UNION */\n", RPC_FC_NON_ENCAPSULATED_UNION);
2745         print_file(file, 2, "0x%x,\t/* Switch type= %s */\n",
2746                    fc, string_of_type(fc));
2747         *tfsoff += 2;
2748
2749         *tfsoff += write_conf_or_var_desc(file, NULL, *tfsoff, st, &dummy_expr );
2750         print_file(file, 2, "NdrFcShort(0x2),\t/* Offset= 2 (%u) */\n", *tfsoff + 2);
2751         *tfsoff += 2;
2752         print_file(file, 0, "/* %u */\n", *tfsoff);
2753     }
2754
2755     print_file(file, 2, "NdrFcShort(0x%hx),\t/* %d */\n", (unsigned short)size, size);
2756     print_file(file, 2, "NdrFcShort(0x%hx),\t/* %d */\n", (unsigned short)nbranch, nbranch);
2757     *tfsoff += 4;
2758
2759     if (fields) LIST_FOR_EACH_ENTRY(f, fields, var_t, entry)
2760     {
2761         type_t *ft = f->type;
2762         expr_list_t *cases = get_attrp(f->attrs, ATTR_CASE);
2763         int deflt = is_attr(f->attrs, ATTR_DEFAULT);
2764         expr_t *c;
2765
2766         if (cases == NULL && !deflt)
2767             error("union field %s with neither case nor default attribute\n", f->name);
2768
2769         if (cases) LIST_FOR_EACH_ENTRY(c, cases, expr_t, entry)
2770         {
2771             /* MIDL doesn't check for duplicate cases, even though that seems
2772                like a reasonable thing to do, it just dumps them to the TFS
2773                like we're going to do here.  */
2774             print_file(file, 2, "NdrFcLong(0x%x),\t/* %d */\n", c->cval, c->cval);
2775             *tfsoff += 4;
2776             write_branch_type(file, ft, tfsoff);
2777         }
2778
2779         /* MIDL allows multiple default branches, even though that seems
2780            illogical, it just chooses the last one, which is what we will
2781            do.  */
2782         if (deflt)
2783         {
2784             deftype = ft;
2785             nodeftype = 0;
2786         }
2787     }
2788
2789     if (deftype)
2790     {
2791         write_branch_type(file, deftype, tfsoff);
2792     }
2793     else
2794     {
2795         print_file(file, 2, "NdrFcShort(0x%hx),\n", nodeftype);
2796         *tfsoff += 2;
2797     }
2798
2799     return start_offset;
2800 }
2801
2802 static unsigned int write_ip_tfs(FILE *file, const attr_list_t *attrs, type_t *type,
2803                                  unsigned int *typeformat_offset)
2804 {
2805     unsigned int i;
2806     unsigned int start_offset = *typeformat_offset;
2807     expr_t *iid = get_attrp(attrs, ATTR_IIDIS);
2808
2809     print_start_tfs_comment(file, type, start_offset);
2810
2811     if (iid)
2812     {
2813         print_file(file, 2, "0x2f,  /* FC_IP */\n");
2814         print_file(file, 2, "0x5c,  /* FC_PAD */\n");
2815         *typeformat_offset
2816             += write_conf_or_var_desc(file, current_structure, 0, type, iid) + 2;
2817     }
2818     else
2819     {
2820         const type_t *base = is_ptr(type) ? type_pointer_get_ref(type) : type;
2821         const UUID *uuid = get_attrp(base->attrs, ATTR_UUID);
2822
2823         if (! uuid)
2824             error("%s: interface %s missing UUID\n", __FUNCTION__, base->name);
2825
2826         update_tfsoff(type, start_offset, file);
2827         print_file(file, 2, "0x2f,\t/* FC_IP */\n");
2828         print_file(file, 2, "0x5a,\t/* FC_CONSTANT_IID */\n");
2829         print_file(file, 2, "NdrFcLong(0x%08x),\n", uuid->Data1);
2830         print_file(file, 2, "NdrFcShort(0x%04x),\n", uuid->Data2);
2831         print_file(file, 2, "NdrFcShort(0x%04x),\n", uuid->Data3);
2832         for (i = 0; i < 8; ++i)
2833             print_file(file, 2, "0x%02x,\n", uuid->Data4[i]);
2834
2835         if (file)
2836             fprintf(file, "\n");
2837
2838         *typeformat_offset += 18;
2839     }
2840     return start_offset;
2841 }
2842
2843 static unsigned int write_contexthandle_tfs(FILE *file,
2844                                             const attr_list_t *attrs,
2845                                             const type_t *type,
2846                                             unsigned int *typeformat_offset)
2847 {
2848     unsigned int start_offset = *typeformat_offset;
2849     unsigned char flags = 0;
2850
2851     if (is_attr(current_iface->attrs, ATTR_STRICTCONTEXTHANDLE))
2852         flags |= NDR_STRICT_CONTEXT_HANDLE;
2853
2854     if (is_ptr(type))
2855         flags |= 0x80;
2856     if (is_attr(attrs, ATTR_IN))
2857     {
2858         flags |= 0x40;
2859         if (!is_attr(attrs, ATTR_OUT))
2860             flags |= NDR_CONTEXT_HANDLE_CANNOT_BE_NULL;
2861     }
2862     if (is_attr(attrs, ATTR_OUT))
2863         flags |= 0x20;
2864
2865     WRITE_FCTYPE(file, FC_BIND_CONTEXT, *typeformat_offset);
2866     print_file(file, 2, "0x%x,\t/* Context flags: ", flags);
2867     /* return and can't be null values overlap */
2868     if (((flags & 0x21) != 0x21) && (flags & NDR_CONTEXT_HANDLE_CANNOT_BE_NULL))
2869         print_file(file, 0, "can't be null, ");
2870     if (flags & NDR_CONTEXT_HANDLE_SERIALIZE)
2871         print_file(file, 0, "serialize, ");
2872     if (flags & NDR_CONTEXT_HANDLE_NO_SERIALIZE)
2873         print_file(file, 0, "no serialize, ");
2874     if (flags & NDR_STRICT_CONTEXT_HANDLE)
2875         print_file(file, 0, "strict, ");
2876     if ((flags & 0x21) == 0x20)
2877         print_file(file, 0, "out, ");
2878     if ((flags & 0x21) == 0x21)
2879         print_file(file, 0, "return, ");
2880     if (flags & 0x40)
2881         print_file(file, 0, "in, ");
2882     if (flags & 0x80)
2883         print_file(file, 0, "via ptr, ");
2884     print_file(file, 0, "*/\n");
2885     print_file(file, 2, "0, /* FIXME: rundown routine index*/\n");
2886     print_file(file, 2, "0, /* FIXME: param num */\n");
2887     *typeformat_offset += 4;
2888
2889     return start_offset;
2890 }
2891
2892 static unsigned int write_range_tfs(FILE *file, const attr_list_t *attrs,
2893                                     type_t *type, expr_list_t *range_list,
2894                                     unsigned int *typeformat_offset)
2895 {
2896     unsigned char fc;
2897     unsigned int start_offset = *typeformat_offset;
2898     const expr_t *range_min = LIST_ENTRY(list_head(range_list), const expr_t, entry);
2899     const expr_t *range_max = LIST_ENTRY(list_next(range_list, list_head(range_list)), const expr_t, entry);
2900
2901     if (type_get_type(type) == TYPE_BASIC)
2902         fc = get_basic_fc(type);
2903     else
2904         fc = get_enum_fc(type);
2905
2906     /* fc must fit in lower 4-bits of 8-bit field below */
2907     assert(fc <= 0xf);
2908
2909     print_file(file, 0, "/* %u */\n", *typeformat_offset);
2910     print_file(file, 2, "0x%x,\t/* FC_RANGE */\n", RPC_FC_RANGE);
2911     print_file(file, 2, "0x%x,\t/* %s */\n", fc, string_of_type(fc));
2912     print_file(file, 2, "NdrFcLong(0x%x),\t/* %u */\n", range_min->cval, range_min->cval);
2913     print_file(file, 2, "NdrFcLong(0x%x),\t/* %u */\n", range_max->cval, range_max->cval);
2914     *typeformat_offset += 10;
2915
2916     return start_offset;
2917 }
2918
2919 static unsigned int write_type_tfs(FILE *file, int indent,
2920                                    const attr_list_t *attrs, type_t *type,
2921                                    const char *name,
2922                                    enum type_context context,
2923                                    unsigned int *typeformat_offset)
2924 {
2925     unsigned int offset;
2926
2927     switch (typegen_detect_type(type, attrs, TDT_ALL_TYPES))
2928     {
2929     case TGT_CTXT_HANDLE:
2930     case TGT_CTXT_HANDLE_POINTER:
2931         return write_contexthandle_tfs(file, attrs, type, typeformat_offset);
2932     case TGT_USER_TYPE:
2933         write_user_tfs(file, type, typeformat_offset);
2934         return type->typestring_offset;
2935     case TGT_STRING:
2936         return write_string_tfs(file, attrs, type,
2937                                 context == TYPE_CONTEXT_TOPLEVELPARAM,
2938                                 name, typeformat_offset);
2939     case TGT_ARRAY:
2940     {
2941         unsigned int off;
2942         /* conformant and pointer arrays are handled specially */
2943         if ((context != TYPE_CONTEXT_CONTAINER &&
2944              context != TYPE_CONTEXT_CONTAINER_NO_POINTERS) ||
2945             !is_conformant_array(type) || type_array_is_decl_as_ptr(type))
2946             off = write_array_tfs(file, attrs, type, name, typeformat_offset);
2947         else
2948             off = 0;
2949         if (context != TYPE_CONTEXT_CONTAINER &&
2950             context != TYPE_CONTEXT_CONTAINER_NO_POINTERS)
2951         {
2952             int ptr_type;
2953             ptr_type = get_pointer_fc(type, attrs,
2954                                       context == TYPE_CONTEXT_TOPLEVELPARAM);
2955             if (ptr_type != RPC_FC_RP)
2956             {
2957                 unsigned int absoff = type->typestring_offset;
2958                 short reloff = absoff - (*typeformat_offset + 2);
2959                 off = *typeformat_offset;
2960                 print_file(file, 0, "/* %d */\n", off);
2961                 print_file(file, 2, "0x%x, 0x0,\t/* %s */\n", ptr_type,
2962                            string_of_type(ptr_type));
2963                 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n",
2964                            reloff, reloff, absoff);
2965                 *typeformat_offset += 4;
2966             }
2967         }
2968         return off;
2969     }
2970     case TGT_STRUCT:
2971         if (processed(type)) return type->typestring_offset;
2972         return write_struct_tfs(file, type, name, typeformat_offset);
2973     case TGT_UNION:
2974         if (processed(type)) return type->typestring_offset;
2975         return write_union_tfs(file, type, typeformat_offset);
2976     case TGT_ENUM:
2977     case TGT_BASIC:
2978         /* nothing to do */
2979         return 0;
2980     case TGT_RANGE:
2981     {
2982         expr_list_t *range_list = get_attrp(attrs, ATTR_RANGE);
2983         if (!range_list)
2984             range_list = get_aliaschain_attrp(type, ATTR_RANGE);
2985         return write_range_tfs(file, attrs, type, range_list, typeformat_offset);
2986     }
2987     case TGT_IFACE_POINTER:
2988         return write_ip_tfs(file, attrs, type, typeformat_offset);
2989     case TGT_POINTER:
2990         if (processed(type_pointer_get_ref(type)))
2991             offset = type_pointer_get_ref(type)->typestring_offset;
2992         else
2993         {
2994             enum type_context ref_context;
2995             if (context == TYPE_CONTEXT_TOPLEVELPARAM)
2996                 ref_context = TYPE_CONTEXT_PARAM;
2997             else if (context == TYPE_CONTEXT_CONTAINER_NO_POINTERS)
2998                 ref_context = TYPE_CONTEXT_CONTAINER;
2999             else
3000                 ref_context = context;
3001             offset = write_type_tfs(
3002                 file, indent, attrs, type_pointer_get_ref(type), name,
3003                 ref_context, typeformat_offset);
3004         }
3005         if (context == TYPE_CONTEXT_CONTAINER_NO_POINTERS)
3006             return 0;
3007         else
3008             return write_pointer_tfs(file, attrs, type, offset,
3009                                      context == TYPE_CONTEXT_TOPLEVELPARAM,
3010                                      typeformat_offset);
3011     case TGT_INVALID:
3012         break;
3013     }
3014     error("invalid type %s for var %s\n", type->name, name);
3015     return 0;
3016 }
3017
3018 static int write_embedded_types(FILE *file, const attr_list_t *attrs, type_t *type,
3019                                 const char *name, int write_ptr, unsigned int *tfsoff)
3020 {
3021     return write_type_tfs(file, 2, attrs, type, name, write_ptr ? TYPE_CONTEXT_CONTAINER : TYPE_CONTEXT_CONTAINER_NO_POINTERS, tfsoff);
3022 }
3023
3024 static unsigned int process_tfs_stmts(FILE *file, const statement_list_t *stmts,
3025                                       type_pred_t pred, unsigned int *typeformat_offset)
3026 {
3027     const var_t *var;
3028     const statement_t *stmt;
3029
3030     if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
3031     {
3032         const type_t *iface;
3033         const statement_t *stmt_func;
3034
3035         if (stmt->type == STMT_LIBRARY)
3036         {
3037             process_tfs_stmts(file, stmt->u.lib->stmts, pred, typeformat_offset);
3038             continue;
3039         }
3040         else if (stmt->type != STMT_TYPE || type_get_type(stmt->u.type) != TYPE_INTERFACE)
3041             continue;
3042
3043         iface = stmt->u.type;
3044         if (!pred(iface))
3045             continue;
3046
3047         current_iface = iface;
3048         STATEMENTS_FOR_EACH_FUNC( stmt_func, type_iface_get_stmts(iface) )
3049         {
3050             const var_t *func = stmt_func->u.var;
3051                 if (is_local(func->attrs)) continue;
3052
3053                 if (!is_void(type_function_get_rettype(func->type)))
3054                 {
3055                     update_tfsoff(type_function_get_rettype(func->type),
3056                                   write_type_tfs(
3057                                       file, 2, func->attrs,
3058                                       type_function_get_rettype(func->type),
3059                                       func->name, TYPE_CONTEXT_PARAM,
3060                                       typeformat_offset),
3061                                   file);
3062                 }
3063
3064                 if (type_get_function_args(func->type))
3065                     LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
3066                         update_tfsoff(
3067                             var->type,
3068                             write_type_tfs(
3069                                 file, 2, var->attrs, var->type, var->name,
3070                                 TYPE_CONTEXT_TOPLEVELPARAM,
3071                                 typeformat_offset),
3072                             file);
3073         }
3074     }
3075
3076     return *typeformat_offset + 1;
3077 }
3078
3079 static unsigned int process_tfs(FILE *file, const statement_list_t *stmts, type_pred_t pred)
3080 {
3081     unsigned int typeformat_offset = 2;
3082
3083     return process_tfs_stmts(file, stmts, pred, &typeformat_offset);
3084 }
3085
3086
3087 void write_typeformatstring(FILE *file, const statement_list_t *stmts, type_pred_t pred)
3088 {
3089     int indent = 0;
3090
3091     print_file(file, indent, "static const MIDL_TYPE_FORMAT_STRING __MIDL_TypeFormatString =\n");
3092     print_file(file, indent, "{\n");
3093     indent++;
3094     print_file(file, indent, "0,\n");
3095     print_file(file, indent, "{\n");
3096     indent++;
3097     print_file(file, indent, "NdrFcShort(0x0),\n");
3098
3099     set_all_tfswrite(TRUE);
3100     process_tfs(file, stmts, pred);
3101
3102     print_file(file, indent, "0x0\n");
3103     indent--;
3104     print_file(file, indent, "}\n");
3105     indent--;
3106     print_file(file, indent, "};\n");
3107     print_file(file, indent, "\n");
3108 }
3109
3110 static unsigned int get_required_buffer_size_type(
3111     const type_t *type, const char *name, const attr_list_t *attrs, int toplevel_param, unsigned int *alignment)
3112 {
3113     *alignment = 0;
3114     switch (typegen_detect_type(type, NULL, TDT_IGNORE_STRINGS|TDT_IGNORE_RANGES))
3115     {
3116     case TGT_USER_TYPE:
3117     {
3118         const char *uname;
3119         const type_t *utype = get_user_type(type, &uname);
3120         return get_required_buffer_size_type(utype, uname, NULL, FALSE, alignment);
3121     }
3122     case TGT_BASIC:
3123         switch (get_basic_fc(type))
3124         {
3125         case RPC_FC_BYTE:
3126         case RPC_FC_CHAR:
3127         case RPC_FC_USMALL:
3128         case RPC_FC_SMALL:
3129             *alignment = 4;
3130             return 1;
3131
3132         case RPC_FC_WCHAR:
3133         case RPC_FC_USHORT:
3134         case RPC_FC_SHORT:
3135             *alignment = 4;
3136             return 2;
3137
3138         case RPC_FC_ULONG:
3139         case RPC_FC_LONG:
3140         case RPC_FC_FLOAT:
3141         case RPC_FC_ERROR_STATUS_T:
3142             *alignment = 4;
3143             return 4;
3144
3145         case RPC_FC_HYPER:
3146         case RPC_FC_DOUBLE:
3147             *alignment = 8;
3148             return 8;
3149
3150         case RPC_FC_INT3264:
3151         case RPC_FC_UINT3264:
3152             assert( pointer_size );
3153             *alignment = pointer_size;
3154             return pointer_size;
3155
3156         case RPC_FC_IGNORE:
3157         case RPC_FC_BIND_PRIMITIVE:
3158             return 0;
3159
3160         default:
3161             error("get_required_buffer_size: unknown basic type 0x%02x\n",
3162                   get_basic_fc(type));
3163             return 0;
3164         }
3165         break;
3166
3167     case TGT_ENUM:
3168         switch (get_enum_fc(type))
3169         {
3170         case RPC_FC_ENUM32:
3171             *alignment = 4;
3172             return 4;
3173         case RPC_FC_ENUM16:
3174             *alignment = 4;
3175             return 2;
3176         }
3177         break;
3178
3179     case TGT_STRUCT:
3180         if (get_struct_fc(type) == RPC_FC_STRUCT)
3181         {
3182             if (!type_struct_get_fields(type)) return 0;
3183             return fields_memsize(type_struct_get_fields(type), alignment);
3184         }
3185         break;
3186
3187     case TGT_POINTER:
3188         if (get_pointer_fc(type, attrs, toplevel_param) == RPC_FC_RP)
3189         {
3190             const type_t *ref = type_pointer_get_ref(type);
3191             switch (typegen_detect_type(ref, NULL, TDT_ALL_TYPES))
3192             {
3193             case TGT_BASIC:
3194             case TGT_ENUM:
3195             case TGT_RANGE:
3196                 return get_required_buffer_size_type( ref, name, NULL, FALSE, alignment );
3197             case TGT_STRUCT:
3198                 if (get_struct_fc(ref) == RPC_FC_STRUCT)
3199                     return get_required_buffer_size_type( ref, name, NULL, FALSE, alignment );
3200                 break;
3201             case TGT_USER_TYPE:
3202             case TGT_CTXT_HANDLE:
3203             case TGT_CTXT_HANDLE_POINTER:
3204             case TGT_STRING:
3205             case TGT_POINTER:
3206             case TGT_ARRAY:
3207             case TGT_IFACE_POINTER:
3208             case TGT_UNION:
3209             case TGT_INVALID:
3210                 break;
3211             }
3212         }
3213         break;
3214
3215     case TGT_ARRAY:
3216         if (get_pointer_fc(type, attrs, toplevel_param) == RPC_FC_RP)
3217         {
3218             switch (get_array_fc(type))
3219             {
3220             case RPC_FC_SMFARRAY:
3221             case RPC_FC_LGFARRAY:
3222                 return type_array_get_dim(type) *
3223                     get_required_buffer_size_type(type_array_get_element(type), name,
3224                                                   NULL, FALSE, alignment);
3225             }
3226         }
3227         break;
3228
3229     default:
3230         break;
3231     }
3232     return 0;
3233 }
3234
3235 static unsigned int get_required_buffer_size(const var_t *var, unsigned int *alignment, enum pass pass)
3236 {
3237     int in_attr = is_attr(var->attrs, ATTR_IN);
3238     int out_attr = is_attr(var->attrs, ATTR_OUT);
3239
3240     if (!in_attr && !out_attr)
3241         in_attr = 1;
3242
3243     *alignment = 0;
3244
3245     if ((pass == PASS_IN && in_attr) || (pass == PASS_OUT && out_attr) ||
3246         pass == PASS_RETURN)
3247     {
3248         if (is_ptrchain_attr(var, ATTR_CONTEXTHANDLE))
3249         {
3250             *alignment = 4;
3251             return 20;
3252         }
3253
3254         if (!is_string_type(var->attrs, var->type))
3255             return get_required_buffer_size_type(var->type, var->name,
3256                                                  var->attrs, TRUE, alignment);
3257     }
3258     return 0;
3259 }
3260
3261 static unsigned int get_function_buffer_size( const var_t *func, enum pass pass )
3262 {
3263     const var_t *var;
3264     unsigned int total_size = 0, alignment;
3265
3266     if (type_get_function_args(func->type))
3267     {
3268         LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
3269         {
3270             total_size += get_required_buffer_size(var, &alignment, pass);
3271             total_size += alignment;
3272         }
3273     }
3274
3275     if (pass == PASS_OUT && !is_void(type_function_get_rettype(func->type)))
3276     {
3277         var_t v = *func;
3278         v.type = type_function_get_rettype(func->type);
3279         total_size += get_required_buffer_size(&v, &alignment, PASS_RETURN);
3280         total_size += alignment;
3281     }
3282     return total_size;
3283 }
3284
3285 static void print_phase_function(FILE *file, int indent, const char *type,
3286                                  const char *local_var_prefix, enum remoting_phase phase,
3287                                  const var_t *var, unsigned int type_offset)
3288 {
3289     const char *function;
3290     switch (phase)
3291     {
3292     case PHASE_BUFFERSIZE:
3293         function = "BufferSize";
3294         break;
3295     case PHASE_MARSHAL:
3296         function = "Marshall";
3297         break;
3298     case PHASE_UNMARSHAL:
3299         function = "Unmarshall";
3300         break;
3301     case PHASE_FREE:
3302         function = "Free";
3303         break;
3304     default:
3305         assert(0);
3306         return;
3307     }
3308
3309     print_file(file, indent, "Ndr%s%s(\n", type, function);
3310     indent++;
3311     print_file(file, indent, "&__frame->_StubMsg,\n");
3312     print_file(file, indent, "%s%s%s%s%s,\n",
3313                (phase == PHASE_UNMARSHAL) ? "(unsigned char **)" : "(unsigned char *)",
3314                (phase == PHASE_UNMARSHAL || decl_indirect(var->type)) ? "&" : "",
3315                local_var_prefix,
3316                (phase == PHASE_UNMARSHAL && decl_indirect(var->type)) ? "_p_" : "",
3317                var->name);
3318     print_file(file, indent, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]%s\n",
3319                type_offset, (phase == PHASE_UNMARSHAL) ? "," : ");");
3320     if (phase == PHASE_UNMARSHAL)
3321         print_file(file, indent, "0);\n");
3322     indent--;
3323 }
3324
3325 void print_phase_basetype(FILE *file, int indent, const char *local_var_prefix,
3326                           enum remoting_phase phase, enum pass pass, const var_t *var,
3327                           const char *varname)
3328 {
3329     type_t *type = var->type;
3330     unsigned int alignment = 0;
3331
3332     /* no work to do for other phases, buffer sizing is done elsewhere */
3333     if (phase != PHASE_MARSHAL && phase != PHASE_UNMARSHAL)
3334         return;
3335
3336     if (type_get_type(type) == TYPE_ENUM ||
3337         (type_get_type(type) == TYPE_BASIC &&
3338          type_basic_get_type(type) == TYPE_BASIC_INT3264 &&
3339          pointer_size != 4))
3340     {
3341         unsigned char fc;
3342
3343         if (type_get_type(type) == TYPE_ENUM)
3344             fc = get_enum_fc(type);
3345         else
3346             fc = get_basic_fc(type);
3347
3348         if (phase == PHASE_MARSHAL)
3349             print_file(file, indent, "NdrSimpleTypeMarshall(\n");
3350         else
3351             print_file(file, indent, "NdrSimpleTypeUnmarshall(\n");
3352         print_file(file, indent+1, "&__frame->_StubMsg,\n");
3353         print_file(file, indent+1, "(unsigned char *)&%s%s,\n",
3354                    local_var_prefix,
3355                    var->name);
3356         print_file(file, indent+1, "0x%02x /* %s */);\n", fc, string_of_type(fc));
3357     }
3358     else
3359     {
3360         const type_t *ref = is_ptr(type) ? type_pointer_get_ref(type) : type;
3361         switch (get_basic_fc(ref))
3362         {
3363         case RPC_FC_BYTE:
3364         case RPC_FC_CHAR:
3365         case RPC_FC_SMALL:
3366         case RPC_FC_USMALL:
3367             alignment = 1;
3368             break;
3369
3370         case RPC_FC_WCHAR:
3371         case RPC_FC_USHORT:
3372         case RPC_FC_SHORT:
3373             alignment = 2;
3374             break;
3375
3376         case RPC_FC_ULONG:
3377         case RPC_FC_LONG:
3378         case RPC_FC_FLOAT:
3379         case RPC_FC_ERROR_STATUS_T:
3380         /* pointer_size must be 4 if we got here in these two cases */
3381         case RPC_FC_INT3264:
3382         case RPC_FC_UINT3264:
3383             alignment = 4;
3384             break;
3385
3386         case RPC_FC_HYPER:
3387         case RPC_FC_DOUBLE:
3388             alignment = 8;
3389             break;
3390
3391         case RPC_FC_IGNORE:
3392         case RPC_FC_BIND_PRIMITIVE:
3393             /* no marshalling needed */
3394             return;
3395
3396         default:
3397             error("print_phase_basetype: Unsupported type: %s (0x%02x, ptr_level: 0)\n",
3398                   var->name, get_basic_fc(ref));
3399         }
3400
3401         if (phase == PHASE_MARSHAL && alignment > 1)
3402             print_file(file, indent, "MIDL_memset(__frame->_StubMsg.Buffer, 0, (0x%x - (ULONG_PTR)__frame->_StubMsg.Buffer) & 0x%x);\n", alignment, alignment - 1);
3403         print_file(file, indent, "__frame->_StubMsg.Buffer = (unsigned char *)(((ULONG_PTR)__frame->_StubMsg.Buffer + %u) & ~0x%x);\n",
3404                     alignment - 1, alignment - 1);
3405
3406         if (phase == PHASE_MARSHAL)
3407         {
3408             print_file(file, indent, "*(");
3409             write_type_decl(file, is_ptr(type) ? type_pointer_get_ref(type) : type, NULL);
3410             if (is_ptr(type))
3411                 fprintf(file, " *)__frame->_StubMsg.Buffer = *");
3412             else
3413                 fprintf(file, " *)__frame->_StubMsg.Buffer = ");
3414             fprintf(file, "%s%s", local_var_prefix, varname);
3415             fprintf(file, ";\n");
3416         }
3417         else if (phase == PHASE_UNMARSHAL)
3418         {
3419             print_file(file, indent, "if (__frame->_StubMsg.Buffer + sizeof(");
3420             write_type_decl(file, is_ptr(type) ? type_pointer_get_ref(type) : type, NULL);
3421             fprintf(file, ") > __frame->_StubMsg.BufferEnd)\n");
3422             print_file(file, indent, "{\n");
3423             print_file(file, indent + 1, "RpcRaiseException(RPC_X_BAD_STUB_DATA);\n");
3424             print_file(file, indent, "}\n");
3425             print_file(file, indent, "%s%s%s",
3426                        (pass == PASS_IN || pass == PASS_RETURN) ? "" : "*",
3427                        local_var_prefix, varname);
3428             if (pass == PASS_IN && is_ptr(type))
3429                 fprintf(file, " = (");
3430             else
3431                 fprintf(file, " = *(");
3432             write_type_decl(file, is_ptr(type) ? type_pointer_get_ref(type) : type, NULL);
3433             fprintf(file, " *)__frame->_StubMsg.Buffer;\n");
3434         }
3435
3436         print_file(file, indent, "__frame->_StubMsg.Buffer += sizeof(");
3437         write_type_decl(file, is_ptr(type) ? type_pointer_get_ref(type) : type, NULL);
3438         fprintf(file, ");\n");
3439     }
3440 }
3441
3442 /* returns whether the MaxCount, Offset or ActualCount members need to be
3443  * filled in for the specified phase */
3444 static inline int is_conformance_needed_for_phase(enum remoting_phase phase)
3445 {
3446     return (phase != PHASE_UNMARSHAL);
3447 }
3448
3449 expr_t *get_size_is_expr(const type_t *t, const char *name)
3450 {
3451     expr_t *x = NULL;
3452
3453     for ( ; is_array(t); t = type_array_get_element(t))
3454         if (type_array_has_conformance(t) &&
3455             type_array_get_conformance(t)->type != EXPR_VOID)
3456         {
3457             if (!x)
3458                 x = type_array_get_conformance(t);
3459             else
3460                 error("%s: multidimensional conformant"
3461                       " arrays not supported at the top level\n",
3462                       name);
3463         }
3464
3465     return x;
3466 }
3467
3468 static void write_parameter_conf_or_var_exprs(FILE *file, int indent, const char *local_var_prefix,
3469                                               enum remoting_phase phase, const var_t *var)
3470 {
3471     const type_t *type = var->type;
3472     /* get fundamental type for the argument */
3473     for (;;)
3474     {
3475         switch (typegen_detect_type(type, var->attrs, TDT_IGNORE_STRINGS|TDT_IGNORE_RANGES))
3476         {
3477         case TGT_ARRAY:
3478             if (is_conformance_needed_for_phase(phase))
3479             {
3480                 if (type_array_has_conformance(type) &&
3481                     type_array_get_conformance(type)->type != EXPR_VOID)
3482                 {
3483                     print_file(file, indent, "__frame->_StubMsg.MaxCount = (ULONG_PTR)");
3484                     write_expr(file, type_array_get_conformance(type), 1, 1, NULL, NULL, local_var_prefix);
3485                     fprintf(file, ";\n\n");
3486                 }
3487                 if (type_array_has_variance(type))
3488                 {
3489                     print_file(file, indent, "__frame->_StubMsg.Offset = 0;\n"); /* FIXME */
3490                     print_file(file, indent, "__frame->_StubMsg.ActualCount = (ULONG_PTR)");
3491                     write_expr(file, type_array_get_variance(type), 1, 1, NULL, NULL, local_var_prefix);
3492                     fprintf(file, ";\n\n");
3493                 }
3494             }
3495             break;
3496         case TGT_UNION:
3497             if (type_get_type(type) == TYPE_UNION &&
3498                 is_conformance_needed_for_phase(phase))
3499             {
3500                 print_file(file, indent, "__frame->_StubMsg.MaxCount = (ULONG_PTR)");
3501                 write_expr(file, get_attrp(var->attrs, ATTR_SWITCHIS), 1, 1, NULL, NULL, local_var_prefix);
3502                 fprintf(file, ";\n\n");
3503             }
3504             break;
3505         case TGT_IFACE_POINTER:
3506         {
3507             expr_t *iid;
3508
3509             if (is_conformance_needed_for_phase(phase) && (iid = get_attrp( var->attrs, ATTR_IIDIS )))
3510             {
3511                 print_file( file, indent, "__frame->_StubMsg.MaxCount = (ULONG_PTR) " );
3512                 write_expr( file, iid, 1, 1, NULL, NULL, local_var_prefix );
3513                 fprintf( file, ";\n\n" );
3514             }
3515             break;
3516         }
3517         case TGT_POINTER:
3518             type = type_pointer_get_ref(type);
3519             continue;
3520         case TGT_INVALID:
3521         case TGT_USER_TYPE:
3522         case TGT_CTXT_HANDLE:
3523         case TGT_CTXT_HANDLE_POINTER:
3524         case TGT_STRING:
3525         case TGT_BASIC:
3526         case TGT_ENUM:
3527         case TGT_STRUCT:
3528         case TGT_RANGE:
3529             break;
3530         }
3531         break;
3532     }
3533 }
3534
3535 static void write_remoting_arg(FILE *file, int indent, const var_t *func, const char *local_var_prefix,
3536                                enum pass pass, enum remoting_phase phase, const var_t *var)
3537 {
3538     int in_attr, out_attr, pointer_type;
3539     const type_t *type = var->type;
3540     unsigned int start_offset = type->typestring_offset;
3541
3542     if (is_ptr(type) || is_array(type))
3543         pointer_type = get_pointer_fc(type, var->attrs, pass != PASS_RETURN);
3544     else
3545         pointer_type = 0;
3546
3547     in_attr = is_attr(var->attrs, ATTR_IN);
3548     out_attr = is_attr(var->attrs, ATTR_OUT);
3549     if (!in_attr && !out_attr)
3550         in_attr = 1;
3551
3552     if (phase != PHASE_FREE)
3553         switch (pass)
3554         {
3555         case PASS_IN:
3556             if (!in_attr) return;
3557             break;
3558         case PASS_OUT:
3559             if (!out_attr) return;
3560             break;
3561         case PASS_RETURN:
3562             break;
3563         }
3564
3565     write_parameter_conf_or_var_exprs(file, indent, local_var_prefix, phase, var);
3566
3567     switch (typegen_detect_type(type, var->attrs, TDT_ALL_TYPES))
3568     {
3569     case TGT_CTXT_HANDLE:
3570     case TGT_CTXT_HANDLE_POINTER:
3571         if (phase == PHASE_MARSHAL)
3572         {
3573             if (pass == PASS_IN)
3574             {
3575                 /* if the context_handle attribute appears in the chain of types
3576                  * without pointers being followed, then the context handle must
3577                  * be direct, otherwise it is a pointer */
3578                 int is_ch_ptr = is_aliaschain_attr(type, ATTR_CONTEXTHANDLE) ? FALSE : TRUE;
3579                 print_file(file, indent, "NdrClientContextMarshall(\n");
3580                 print_file(file, indent + 1, "&__frame->_StubMsg,\n");
3581                 print_file(file, indent + 1, "(NDR_CCONTEXT)%s%s%s,\n", is_ch_ptr ? "*" : "", local_var_prefix, var->name);
3582                 print_file(file, indent + 1, "%s);\n", in_attr && out_attr ? "1" : "0");
3583             }
3584             else
3585             {
3586                 print_file(file, indent, "NdrServerContextNewMarshall(\n");
3587                 print_file(file, indent + 1, "&__frame->_StubMsg,\n");
3588                 print_file(file, indent + 1, "(NDR_SCONTEXT)%s%s,\n", local_var_prefix, var->name);
3589                 print_file(file, indent + 1, "(NDR_RUNDOWN)%s_rundown,\n", get_context_handle_type_name(var->type));
3590                 print_file(file, indent + 1, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]);\n", start_offset);
3591             }
3592         }
3593         else if (phase == PHASE_UNMARSHAL)
3594         {
3595             if (pass == PASS_OUT)
3596             {
3597                 if (!in_attr)
3598                     print_file(file, indent, "*%s%s = 0;\n", local_var_prefix, var->name);
3599                 print_file(file, indent, "NdrClientContextUnmarshall(\n");
3600                 print_file(file, indent + 1, "&__frame->_StubMsg,\n");
3601                 print_file(file, indent + 1, "(NDR_CCONTEXT *)%s%s,\n", local_var_prefix, var->name);
3602                 print_file(file, indent + 1, "__frame->_Handle);\n");
3603             }
3604             else
3605             {
3606                 print_file(file, indent, "%s%s = NdrServerContextNewUnmarshall(\n", local_var_prefix, var->name);
3607                 print_file(file, indent + 1, "&__frame->_StubMsg,\n");
3608                 print_file(file, indent + 1, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]);\n", start_offset);
3609             }
3610         }
3611         break;
3612     case TGT_USER_TYPE:
3613         print_phase_function(file, indent, "UserMarshal", local_var_prefix, phase, var, start_offset);
3614         break;
3615     case TGT_STRING:
3616         if (phase == PHASE_FREE || pass == PASS_RETURN ||
3617             pointer_type != RPC_FC_RP)
3618         {
3619             if (pointer_type == RPC_FC_RP && phase == PHASE_FREE &&
3620                 !in_attr && is_conformant_array(type))
3621             {
3622                 print_file(file, indent, "if (%s%s)\n", local_var_prefix, var->name);
3623                 indent++;
3624                 print_file(file, indent, "__frame->_StubMsg.pfnFree(%s%s);\n", local_var_prefix, var->name);
3625             }
3626             /* strings returned are assumed to be global and hence don't
3627              * need freeing */
3628             else if (is_declptr(type) &&
3629                      !(phase == PHASE_FREE && pass == PASS_RETURN))
3630                 print_phase_function(file, indent, "Pointer", local_var_prefix,
3631                                      phase, var, start_offset);
3632         }
3633         else
3634         {
3635             unsigned int real_start_offset = start_offset;
3636             /* skip over pointer description straight to string description */
3637             if (is_declptr(type))
3638             {
3639                 if (is_conformant_array(type))
3640                     real_start_offset += 4;
3641                 else
3642                     real_start_offset += 2;
3643             }
3644             if (is_array(type) && !is_conformant_array(type))
3645                 print_phase_function(file, indent, "NonConformantString",
3646                                      local_var_prefix, phase, var,
3647                                      real_start_offset);
3648             else
3649                 print_phase_function(file, indent, "ConformantString", local_var_prefix,
3650                                      phase, var, real_start_offset);
3651         }
3652         break;
3653     case TGT_ARRAY:
3654     {
3655         unsigned char tc = get_array_fc(type);
3656         const char *array_type = NULL;
3657
3658         /* We already have the size_is expression since it's at the
3659            top level, but do checks for multidimensional conformant
3660            arrays.  When we handle them, we'll need to extend this
3661            function to return a list, and then we'll actually use
3662            the return value.  */
3663         get_size_is_expr(type, var->name);
3664
3665         switch (tc)
3666         {
3667         case RPC_FC_SMFARRAY:
3668         case RPC_FC_LGFARRAY:
3669             array_type = "FixedArray";
3670             break;
3671         case RPC_FC_SMVARRAY:
3672         case RPC_FC_LGVARRAY:
3673             array_type = "VaryingArray";
3674             break;
3675         case RPC_FC_CARRAY:
3676             array_type = "ConformantArray";
3677             break;
3678         case RPC_FC_CVARRAY:
3679             array_type = "ConformantVaryingArray";
3680             break;
3681         case RPC_FC_BOGUS_ARRAY:
3682             array_type = "ComplexArray";
3683             break;
3684         }
3685
3686         if (pointer_type != RPC_FC_RP) array_type = "Pointer";
3687         print_phase_function(file, indent, array_type, local_var_prefix, phase, var, start_offset);
3688         if (phase == PHASE_FREE && pointer_type == RPC_FC_RP)
3689         {
3690             /* these are all unmarshalled by allocating memory */
3691             if (tc == RPC_FC_BOGUS_ARRAY ||
3692                 tc == RPC_FC_CVARRAY ||
3693                 ((tc == RPC_FC_SMVARRAY || tc == RPC_FC_LGVARRAY) && in_attr) ||
3694                 (tc == RPC_FC_CARRAY && !in_attr))
3695             {
3696                 print_file(file, indent, "if (%s%s)\n", local_var_prefix, var->name);
3697                 indent++;
3698                 print_file(file, indent, "__frame->_StubMsg.pfnFree(%s%s);\n", local_var_prefix, var->name);
3699             }
3700         }
3701         break;
3702     }
3703     case TGT_BASIC:
3704         print_phase_basetype(file, indent, local_var_prefix, phase, pass, var, var->name);
3705         break;
3706     case TGT_ENUM:
3707         print_phase_basetype(file, indent, local_var_prefix, phase, pass, var, var->name);
3708         break;
3709     case TGT_RANGE:
3710         print_phase_basetype(file, indent, local_var_prefix, phase, pass, var, var->name);
3711         /* Note: this goes beyond what MIDL does - it only supports arguments
3712          * with the [range] attribute in Oicf mode */
3713         if (phase == PHASE_UNMARSHAL)
3714         {
3715             const expr_t *range_min;
3716             const expr_t *range_max;
3717             expr_list_t *range_list = get_attrp(var->attrs, ATTR_RANGE);
3718             if (!range_list)
3719                 range_list = get_aliaschain_attrp(type, ATTR_RANGE);
3720             range_min = LIST_ENTRY(list_head(range_list), const expr_t, entry);
3721             range_max = LIST_ENTRY(list_next(range_list, list_head(range_list)), const expr_t, entry);
3722
3723             print_file(file, indent, "if ((%s%s < (", local_var_prefix, var->name);
3724             write_type_decl(file, var->type, NULL);
3725             fprintf(file, ")0x%x) || (%s%s > (", range_min->cval, local_var_prefix, var->name);
3726             write_type_decl(file, var->type, NULL);
3727             fprintf(file, ")0x%x))\n", range_max->cval);
3728             print_file(file, indent, "{\n");
3729             print_file(file, indent+1, "RpcRaiseException(RPC_S_INVALID_BOUND);\n");
3730             print_file(file, indent, "}\n");
3731         }
3732         break;
3733     case TGT_STRUCT:
3734         switch (get_struct_fc(type))
3735         {
3736         case RPC_FC_STRUCT:
3737             if (phase == PHASE_MARSHAL || phase == PHASE_UNMARSHAL)
3738                 print_phase_function(file, indent, "SimpleStruct", local_var_prefix, phase, var, start_offset);
3739             break;
3740         case RPC_FC_PSTRUCT:
3741             print_phase_function(file, indent, "SimpleStruct", local_var_prefix, phase, var, start_offset);
3742             break;
3743         case RPC_FC_CSTRUCT:
3744         case RPC_FC_CPSTRUCT:
3745             print_phase_function(file, indent, "ConformantStruct", local_var_prefix, phase, var, start_offset);
3746             break;
3747         case RPC_FC_CVSTRUCT:
3748             print_phase_function(file, indent, "ConformantVaryingStruct", local_var_prefix, phase, var, start_offset);
3749             break;
3750         case RPC_FC_BOGUS_STRUCT:
3751             print_phase_function(file, indent, "ComplexStruct", local_var_prefix, phase, var, start_offset);
3752             break;
3753         default:
3754             error("write_remoting_arguments: Unsupported type: %s (0x%02x)\n", var->name, get_struct_fc(type));
3755         }
3756         break;
3757     case TGT_UNION:
3758     {
3759         const char *union_type = NULL;
3760
3761         if (type_get_type(type) == TYPE_UNION)
3762             union_type = "NonEncapsulatedUnion";
3763         else if (type_get_type(type) == TYPE_ENCAPSULATED_UNION)
3764             union_type = "EncapsulatedUnion";
3765
3766         print_phase_function(file, indent, union_type, local_var_prefix,
3767                              phase, var, start_offset);
3768         break;
3769     }
3770     case TGT_POINTER:
3771     {
3772         const type_t *ref = type_pointer_get_ref(type);
3773         if (pointer_type == RPC_FC_RP) switch (typegen_detect_type(ref, NULL, TDT_ALL_TYPES))
3774         {
3775         case TGT_BASIC:
3776             print_phase_basetype(file, indent, local_var_prefix, phase, pass, var, var->name);
3777             break;
3778         case TGT_ENUM:
3779             /* base types have known sizes, so don't need a sizing pass
3780              * and don't have any memory to free and so don't need a
3781              * freeing pass */
3782             if (phase == PHASE_MARSHAL || phase == PHASE_UNMARSHAL)
3783                 print_phase_function(file, indent, "Pointer", local_var_prefix, phase, var, start_offset);
3784             break;
3785         case TGT_STRUCT:
3786         {
3787             const char *struct_type = NULL;
3788             switch (get_struct_fc(ref))
3789             {
3790             case RPC_FC_STRUCT:
3791                 /* simple structs have known sizes, so don't need a sizing
3792                  * pass and don't have any memory to free and so don't
3793                  * need a freeing pass */
3794                 if (phase == PHASE_MARSHAL || phase == PHASE_UNMARSHAL)
3795                     struct_type = "SimpleStruct";
3796                 else if (phase == PHASE_FREE && pass == PASS_RETURN)
3797                 {
3798                     print_file(file, indent, "if (%s%s)\n", local_var_prefix, var->name);
3799                     indent++;
3800                     print_file(file, indent, "__frame->_StubMsg.pfnFree(%s%s);\n", local_var_prefix, var->name);
3801                     indent--;
3802                 }
3803                 break;
3804             case RPC_FC_PSTRUCT:
3805                 struct_type = "SimpleStruct";
3806                 break;
3807             case RPC_FC_CSTRUCT:
3808             case RPC_FC_CPSTRUCT:
3809                 struct_type = "ConformantStruct";
3810                 break;
3811             case RPC_FC_CVSTRUCT:
3812                 struct_type = "ConformantVaryingStruct";
3813                 break;
3814             case RPC_FC_BOGUS_STRUCT:
3815                 struct_type = "ComplexStruct";
3816                 break;
3817             default:
3818                 error("write_remoting_arguments: Unsupported type: %s (0x%02x)\n", var->name, get_struct_fc(ref));
3819             }
3820
3821             if (struct_type)
3822             {
3823                 if (phase == PHASE_FREE)
3824                     struct_type = "Pointer";
3825                 else
3826                     start_offset = ref->typestring_offset;
3827                 print_phase_function(file, indent, struct_type, local_var_prefix, phase, var, start_offset);
3828             }
3829             break;
3830         }
3831         case TGT_UNION:
3832         {
3833             const char *union_type = NULL;
3834             if (phase == PHASE_FREE)
3835                 union_type = "Pointer";
3836             else
3837             {
3838                 if (type_get_type(ref) == TYPE_UNION)
3839                     union_type = "NonEncapsulatedUnion";
3840                 else if (type_get_type(ref) == TYPE_ENCAPSULATED_UNION)
3841                     union_type = "EncapsulatedUnion";
3842
3843                 start_offset = ref->typestring_offset;
3844             }
3845
3846             print_phase_function(file, indent, union_type, local_var_prefix,
3847                                  phase, var, start_offset);
3848             break;
3849         }
3850         case TGT_STRING:
3851         case TGT_POINTER:
3852         case TGT_ARRAY:
3853         case TGT_RANGE:
3854         case TGT_IFACE_POINTER:
3855         case TGT_USER_TYPE:
3856         case TGT_CTXT_HANDLE:
3857         case TGT_CTXT_HANDLE_POINTER:
3858             print_phase_function(file, indent, "Pointer", local_var_prefix, phase, var, start_offset);
3859             break;
3860         case TGT_INVALID:
3861             assert(0);
3862             break;
3863         }
3864         else
3865             print_phase_function(file, indent, "Pointer", local_var_prefix, phase, var, start_offset);
3866         break;
3867     }
3868     case TGT_IFACE_POINTER:
3869         print_phase_function(file, indent, "InterfacePointer", local_var_prefix, phase, var, start_offset);
3870         break;
3871     case TGT_INVALID:
3872         assert(0);
3873         break;
3874     }
3875     fprintf(file, "\n");
3876 }
3877
3878 void write_remoting_arguments(FILE *file, int indent, const var_t *func, const char *local_var_prefix,
3879                               enum pass pass, enum remoting_phase phase)
3880 {
3881     if (phase == PHASE_BUFFERSIZE && pass != PASS_RETURN)
3882     {
3883         unsigned int size = get_function_buffer_size( func, pass );
3884         print_file(file, indent, "__frame->_StubMsg.BufferLength = %u;\n", size);
3885     }
3886
3887     if (pass == PASS_RETURN)
3888     {
3889         var_t var;
3890         var = *func;
3891         var.type = type_function_get_rettype(func->type);
3892         var.name = xstrdup( "_RetVal" );
3893         write_remoting_arg( file, indent, func, local_var_prefix, pass, phase, &var );
3894         free( var.name );
3895     }
3896     else
3897     {
3898         const var_t *var;
3899         if (!type_get_function_args(func->type))
3900             return;
3901         LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
3902             write_remoting_arg( file, indent, func, local_var_prefix, pass, phase, var );
3903     }
3904 }
3905
3906
3907 unsigned int get_size_procformatstring_func(const var_t *func)
3908 {
3909     unsigned int offset = 0;
3910     write_procformatstring_func( NULL, 0, func, &offset );
3911     return offset;
3912 }
3913
3914 unsigned int get_size_procformatstring(const statement_list_t *stmts, type_pred_t pred)
3915 {
3916     const statement_t *stmt;
3917     unsigned int size = 1;
3918
3919     if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
3920     {
3921         const type_t *iface;
3922         const statement_t *stmt_func;
3923
3924         if (stmt->type == STMT_LIBRARY)
3925         {
3926             size += get_size_procformatstring(stmt->u.lib->stmts, pred) - 1;
3927             continue;
3928         }
3929         else if (stmt->type != STMT_TYPE || type_get_type(stmt->u.type) != TYPE_INTERFACE)
3930             continue;
3931
3932         iface = stmt->u.type;
3933         if (!pred(iface))
3934             continue;
3935
3936         STATEMENTS_FOR_EACH_FUNC( stmt_func, type_iface_get_stmts(iface) )
3937         {
3938             const var_t *func = stmt_func->u.var;
3939             if (!is_local(func->attrs))
3940                 size += get_size_procformatstring_func( func );
3941         }
3942     }
3943     return size;
3944 }
3945
3946 unsigned int get_size_typeformatstring(const statement_list_t *stmts, type_pred_t pred)
3947 {
3948     set_all_tfswrite(FALSE);
3949     return process_tfs(NULL, stmts, pred);
3950 }
3951
3952 void declare_stub_args( FILE *file, int indent, const var_t *func )
3953 {
3954     int in_attr, out_attr;
3955     int i = 0;
3956     const var_t *var;
3957
3958     /* declare return value '_RetVal' */
3959     if (!is_void(type_function_get_rettype(func->type)))
3960     {
3961         print_file(file, indent, "%s", "");
3962         write_type_decl_left(file, type_function_get_rettype(func->type));
3963         fprintf(file, " _RetVal;\n");
3964     }
3965
3966     if (!type_get_function_args(func->type))
3967         return;
3968
3969     LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
3970     {
3971         in_attr = is_attr(var->attrs, ATTR_IN);
3972         out_attr = is_attr(var->attrs, ATTR_OUT);
3973         if (!out_attr && !in_attr)
3974             in_attr = 1;
3975
3976         if (is_context_handle(var->type))
3977             print_file(file, indent, "NDR_SCONTEXT %s;\n", var->name);
3978         else
3979         {
3980             if (!in_attr && !is_conformant_array(var->type))
3981             {
3982                 type_t *type_to_print;
3983                 char name[16];
3984                 print_file(file, indent, "%s", "");
3985                 if (type_get_type(var->type) == TYPE_ARRAY &&
3986                     !type_array_is_decl_as_ptr(var->type))
3987                     type_to_print = var->type;
3988                 else
3989                     type_to_print = type_pointer_get_ref(var->type);
3990                 sprintf(name, "_W%u", i++);
3991                 write_type_decl(file, type_to_print, name);
3992                 fprintf(file, ";\n");
3993             }
3994
3995             print_file(file, indent, "%s", "");
3996             write_type_decl_left(file, var->type);
3997             fprintf(file, " ");
3998             if (type_get_type(var->type) == TYPE_ARRAY &&
3999                 !type_array_is_decl_as_ptr(var->type)) {
4000                 fprintf(file, "(*%s)", var->name);
4001             } else
4002                 fprintf(file, "%s", var->name);
4003             write_type_right(file, var->type, FALSE);
4004             fprintf(file, ";\n");
4005
4006             if (decl_indirect(var->type))
4007                 print_file(file, indent, "void *_p_%s;\n", var->name);
4008         }
4009     }
4010 }
4011
4012
4013 void assign_stub_out_args( FILE *file, int indent, const var_t *func, const char *local_var_prefix )
4014 {
4015     int in_attr, out_attr;
4016     int i = 0, sep = 0;
4017     const var_t *var;
4018
4019     if (!type_get_function_args(func->type))
4020         return;
4021
4022     LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
4023     {
4024         in_attr = is_attr(var->attrs, ATTR_IN);
4025         out_attr = is_attr(var->attrs, ATTR_OUT);
4026         if (!out_attr && !in_attr)
4027             in_attr = 1;
4028
4029         if (!in_attr)
4030         {
4031             print_file(file, indent, "%s%s", local_var_prefix, var->name);
4032
4033             switch (typegen_detect_type(var->type, var->attrs, TDT_IGNORE_STRINGS))
4034             {
4035             case TGT_CTXT_HANDLE_POINTER:
4036                 fprintf(file, " = NdrContextHandleInitialize(\n");
4037                 print_file(file, indent + 1, "&__frame->_StubMsg,\n");
4038                 print_file(file, indent + 1, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]);\n",
4039                            var->type->typestring_offset);
4040                 break;
4041             case TGT_ARRAY:
4042                 if (type_array_has_conformance(var->type))
4043                 {
4044                     unsigned int size;
4045                     type_t *type;
4046
4047                     fprintf(file, " = NdrAllocate(&__frame->_StubMsg, ");
4048                     for (type = var->type;
4049                          is_array(type) && type_array_has_conformance(type);
4050                          type = type_array_get_element(type))
4051                     {
4052                         write_expr(file, type_array_get_conformance(type), TRUE,
4053                                    TRUE, NULL, NULL, local_var_prefix);
4054                         fprintf(file, " * ");
4055                     }
4056                     size = type_memsize(type);
4057                     fprintf(file, "%u);\n", size);
4058
4059                     print_file(file, indent, "memset(%s%s, 0, ", local_var_prefix, var->name);
4060                     for (type = var->type;
4061                          is_array(type) && type_array_has_conformance(type);
4062                          type = type_array_get_element(type))
4063                     {
4064                         write_expr(file, type_array_get_conformance(type), TRUE,
4065                                    TRUE, NULL, NULL, local_var_prefix);
4066                         fprintf(file, " * ");
4067                     }
4068                     size = type_memsize(type);
4069                     fprintf(file, "%u);\n", size);
4070                 }
4071                 else
4072                     fprintf(file, " = &%s_W%u;\n", local_var_prefix, i++);
4073                 break;
4074             case TGT_POINTER:
4075                 fprintf(file, " = &%s_W%u;\n", local_var_prefix, i);
4076                 switch (typegen_detect_type(type_pointer_get_ref(var->type), var->attrs, TDT_IGNORE_STRINGS))
4077                 {
4078                 case TGT_BASIC:
4079                 case TGT_ENUM:
4080                 case TGT_POINTER:
4081                 case TGT_RANGE:
4082                 case TGT_IFACE_POINTER:
4083                     print_file(file, indent, "%s_W%u = 0;\n", local_var_prefix, i);
4084                     break;
4085                 case TGT_USER_TYPE:
4086                     print_file(file, indent, "memset(&%s_W%u, 0, sizeof(%s_W%u));\n",
4087                                local_var_prefix, i, local_var_prefix, i);
4088                     break;
4089                 case TGT_STRUCT:
4090                 case TGT_UNION:
4091                 case TGT_ARRAY:
4092                 case TGT_CTXT_HANDLE:
4093                 case TGT_CTXT_HANDLE_POINTER:
4094                 case TGT_INVALID:
4095                 case TGT_STRING:
4096                     /* not initialised */
4097                     break;
4098                 }
4099                 i++;
4100                 break;
4101             default:
4102                 break;
4103             }
4104
4105             sep = 1;
4106         }
4107     }
4108     if (sep)
4109         fprintf(file, "\n");
4110 }
4111
4112
4113 int write_expr_eval_routines(FILE *file, const char *iface)
4114 {
4115     static const char *var_name = "pS";
4116     static const char *var_name_expr = "pS->";
4117     int result = 0;
4118     struct expr_eval_routine *eval;
4119     unsigned short callback_offset = 0;
4120
4121     LIST_FOR_EACH_ENTRY(eval, &expr_eval_routines, struct expr_eval_routine, entry)
4122     {
4123         const char *name = eval->structure->name;
4124         result = 1;
4125
4126         print_file(file, 0, "static void __RPC_USER %s_%sExprEval_%04u(PMIDL_STUB_MESSAGE pStubMsg)\n",
4127                    iface, name, callback_offset);
4128         print_file(file, 0, "{\n");
4129         print_file(file, 1, "%s", "");
4130         write_type_left(file, (type_t *)eval->structure, TRUE);
4131         fprintf(file, " *%s = (", var_name);
4132         write_type_left(file, (type_t *)eval->structure, TRUE);
4133         fprintf(file, " *)(pStubMsg->StackTop - %u);\n", eval->baseoff);
4134         print_file(file, 1, "pStubMsg->Offset = 0;\n"); /* FIXME */
4135         print_file(file, 1, "pStubMsg->MaxCount = (ULONG_PTR)");
4136         write_expr(file, eval->expr, 1, 1, var_name_expr, eval->structure, "");
4137         fprintf(file, ";\n");
4138         print_file(file, 0, "}\n\n");
4139         callback_offset++;
4140     }
4141     return result;
4142 }
4143
4144 void write_expr_eval_routine_list(FILE *file, const char *iface)
4145 {
4146     struct expr_eval_routine *eval;
4147     struct expr_eval_routine *cursor;
4148     unsigned short callback_offset = 0;
4149
4150     fprintf(file, "static const EXPR_EVAL ExprEvalRoutines[] =\n");
4151     fprintf(file, "{\n");
4152
4153     LIST_FOR_EACH_ENTRY_SAFE(eval, cursor, &expr_eval_routines, struct expr_eval_routine, entry)
4154     {
4155         const char *name = eval->structure->name;
4156         print_file(file, 1, "%s_%sExprEval_%04u,\n", iface, name, callback_offset);
4157         callback_offset++;
4158         list_remove(&eval->entry);
4159         free(eval);
4160     }
4161
4162     fprintf(file, "};\n\n");
4163 }
4164
4165 void write_user_quad_list(FILE *file)
4166 {
4167     user_type_t *ut;
4168
4169     if (list_empty(&user_type_list))
4170         return;
4171
4172     fprintf(file, "static const USER_MARSHAL_ROUTINE_QUADRUPLE UserMarshalRoutines[] =\n");
4173     fprintf(file, "{\n");
4174     LIST_FOR_EACH_ENTRY(ut, &user_type_list, user_type_t, entry)
4175     {
4176         const char *sep = &ut->entry == list_tail(&user_type_list) ? "" : ",";
4177         print_file(file, 1, "{\n");
4178         print_file(file, 2, "(USER_MARSHAL_SIZING_ROUTINE)%s_UserSize,\n", ut->name);
4179         print_file(file, 2, "(USER_MARSHAL_MARSHALLING_ROUTINE)%s_UserMarshal,\n", ut->name);
4180         print_file(file, 2, "(USER_MARSHAL_UNMARSHALLING_ROUTINE)%s_UserUnmarshal,\n", ut->name);
4181         print_file(file, 2, "(USER_MARSHAL_FREEING_ROUTINE)%s_UserFree\n", ut->name);
4182         print_file(file, 1, "}%s\n", sep);
4183     }
4184     fprintf(file, "};\n\n");
4185 }
4186
4187 void write_endpoints( FILE *f, const char *prefix, const str_list_t *list )
4188 {
4189     const struct str_list_entry_t *endpoint;
4190     const char *p;
4191
4192     /* this should be an array of RPC_PROTSEQ_ENDPOINT but we want const strings */
4193     print_file( f, 0, "static const unsigned char * const %s__RpcProtseqEndpoint[][2] =\n{\n", prefix );
4194     LIST_FOR_EACH_ENTRY( endpoint, list, const struct str_list_entry_t, entry )
4195     {
4196         print_file( f, 1, "{ (const unsigned char *)\"" );
4197         for (p = endpoint->str; *p && *p != ':'; p++)
4198         {
4199             if (*p == '"' || *p == '\\') fputc( '\\', f );
4200             fputc( *p, f );
4201         }
4202         if (!*p) goto error;
4203         if (p[1] != '[') goto error;
4204
4205         fprintf( f, "\", (const unsigned char *)\"" );
4206         for (p += 2; *p && *p != ']'; p++)
4207         {
4208             if (*p == '"' || *p == '\\') fputc( '\\', f );
4209             fputc( *p, f );
4210         }
4211         if (*p != ']') goto error;
4212         fprintf( f, "\" },\n" );
4213     }
4214     print_file( f, 0, "};\n\n" );
4215     return;
4216
4217 error:
4218     error("Invalid endpoint syntax '%s'\n", endpoint->str);
4219 }
4220
4221 void write_exceptions( FILE *file )
4222 {
4223     fprintf( file, "#ifndef USE_COMPILER_EXCEPTIONS\n");
4224     fprintf( file, "\n");
4225     fprintf( file, "#include \"wine/exception.h\"\n");
4226     fprintf( file, "#undef RpcTryExcept\n");
4227     fprintf( file, "#undef RpcExcept\n");
4228     fprintf( file, "#undef RpcEndExcept\n");
4229     fprintf( file, "#undef RpcTryFinally\n");
4230     fprintf( file, "#undef RpcFinally\n");
4231     fprintf( file, "#undef RpcEndFinally\n");
4232     fprintf( file, "#undef RpcExceptionCode\n");
4233     fprintf( file, "#undef RpcAbnormalTermination\n");
4234     fprintf( file, "\n");
4235     fprintf( file, "struct __exception_frame;\n");
4236     fprintf( file, "typedef int (*__filter_func)(struct __exception_frame *);\n");
4237     fprintf( file, "typedef void (*__finally_func)(struct __exception_frame *);\n");
4238     fprintf( file, "\n");
4239     fprintf( file, "#define __DECL_EXCEPTION_FRAME \\\n");
4240     fprintf( file, "    EXCEPTION_REGISTRATION_RECORD frame; \\\n");
4241     fprintf( file, "    __filter_func                 filter; \\\n");
4242     fprintf( file, "    __finally_func                finally; \\\n");
4243     fprintf( file, "    sigjmp_buf                    jmp; \\\n");
4244     fprintf( file, "    DWORD                         code; \\\n");
4245     fprintf( file, "    unsigned char                 abnormal_termination; \\\n");
4246     fprintf( file, "    unsigned char                 filter_level; \\\n");
4247     fprintf( file, "    unsigned char                 finally_level;\n");
4248     fprintf( file, "\n");
4249     fprintf( file, "struct __exception_frame\n{\n");
4250     fprintf( file, "    __DECL_EXCEPTION_FRAME\n");
4251     fprintf( file, "};\n");
4252     fprintf( file, "\n");
4253     fprintf( file, "static inline void __widl_unwind_target(void)\n" );
4254     fprintf( file, "{\n");
4255     fprintf( file, "    struct __exception_frame *exc_frame = (struct __exception_frame *)__wine_get_frame();\n" );
4256     fprintf( file, "    if (exc_frame->finally_level > exc_frame->filter_level)\n" );
4257     fprintf( file, "    {\n");
4258     fprintf( file, "        exc_frame->abnormal_termination = 1;\n");
4259     fprintf( file, "        exc_frame->finally( exc_frame );\n");
4260     fprintf( file, "        __wine_pop_frame( &exc_frame->frame );\n");
4261     fprintf( file, "    }\n");
4262     fprintf( file, "    exc_frame->filter_level = 0;\n");
4263     fprintf( file, "    siglongjmp( exc_frame->jmp, 1 );\n");
4264     fprintf( file, "}\n");
4265     fprintf( file, "\n");
4266     fprintf( file, "static DWORD __widl_exception_handler( EXCEPTION_RECORD *record,\n");
4267     fprintf( file, "                                       EXCEPTION_REGISTRATION_RECORD *frame,\n");
4268     fprintf( file, "                                       CONTEXT *context,\n");
4269     fprintf( file, "                                       EXCEPTION_REGISTRATION_RECORD **pdispatcher )\n");
4270     fprintf( file, "{\n");
4271     fprintf( file, "    struct __exception_frame *exc_frame = (struct __exception_frame *)frame;\n");
4272     fprintf( file, "\n");
4273     fprintf( file, "    if (record->ExceptionFlags & (EH_UNWINDING | EH_EXIT_UNWIND | EH_NESTED_CALL))\n");
4274     fprintf( file, "    {\n" );
4275     fprintf( file, "        if (exc_frame->finally_level && (record->ExceptionFlags & (EH_UNWINDING | EH_EXIT_UNWIND)))\n");
4276     fprintf( file, "        {\n" );
4277     fprintf( file, "            exc_frame->abnormal_termination = 1;\n");
4278     fprintf( file, "            exc_frame->finally( exc_frame );\n");
4279     fprintf( file, "        }\n" );
4280     fprintf( file, "        return ExceptionContinueSearch;\n");
4281     fprintf( file, "    }\n" );
4282     fprintf( file, "    exc_frame->code = record->ExceptionCode;\n");
4283     fprintf( file, "    if (exc_frame->filter_level && exc_frame->filter( exc_frame ) == EXCEPTION_EXECUTE_HANDLER)\n" );
4284     fprintf( file, "        __wine_rtl_unwind( frame, record, __widl_unwind_target );\n");
4285     fprintf( file, "    return ExceptionContinueSearch;\n");
4286     fprintf( file, "}\n");
4287     fprintf( file, "\n");
4288     fprintf( file, "#define RpcTryExcept \\\n");
4289     fprintf( file, "    if (!sigsetjmp( __frame->jmp, 0 )) \\\n");
4290     fprintf( file, "    { \\\n");
4291     fprintf( file, "        if (!__frame->finally_level) \\\n" );
4292     fprintf( file, "            __wine_push_frame( &__frame->frame ); \\\n");
4293     fprintf( file, "        __frame->filter_level = __frame->finally_level + 1;\n" );
4294     fprintf( file, "\n");
4295     fprintf( file, "#define RpcExcept(expr) \\\n");
4296     fprintf( file, "        if (!__frame->finally_level) \\\n" );
4297     fprintf( file, "            __wine_pop_frame( &__frame->frame ); \\\n");
4298     fprintf( file, "        __frame->filter_level = 0; \\\n" );
4299     fprintf( file, "    } \\\n");
4300     fprintf( file, "    else \\\n");
4301     fprintf( file, "\n");
4302     fprintf( file, "#define RpcEndExcept\n");
4303     fprintf( file, "\n");
4304     fprintf( file, "#define RpcExceptionCode() (__frame->code)\n");
4305     fprintf( file, "\n");
4306     fprintf( file, "#define RpcTryFinally \\\n");
4307     fprintf( file, "    if (!__frame->filter_level) \\\n");
4308     fprintf( file, "        __wine_push_frame( &__frame->frame ); \\\n");
4309     fprintf( file, "    __frame->finally_level = __frame->filter_level + 1;\n");
4310     fprintf( file, "\n");
4311     fprintf( file, "#define RpcFinally \\\n");
4312     fprintf( file, "    if (!__frame->filter_level) \\\n");
4313     fprintf( file, "        __wine_pop_frame( &__frame->frame ); \\\n");
4314     fprintf( file, "    __frame->finally_level = 0;\n");
4315     fprintf( file, "\n");
4316     fprintf( file, "#define RpcEndFinally\n");
4317     fprintf( file, "\n");
4318     fprintf( file, "#define RpcAbnormalTermination() (__frame->abnormal_termination)\n");
4319     fprintf( file, "\n");
4320     fprintf( file, "#define RpcExceptionInit(filter_func,finally_func) \\\n");
4321     fprintf( file, "    do { \\\n");
4322     fprintf( file, "        __frame->frame.Handler = __widl_exception_handler; \\\n");
4323     fprintf( file, "        __frame->filter = (__filter_func)(filter_func); \\\n" );
4324     fprintf( file, "        __frame->finally = (__finally_func)(finally_func); \\\n");
4325     fprintf( file, "        __frame->abnormal_termination = 0; \\\n");
4326     fprintf( file, "        __frame->filter_level = 0; \\\n");
4327     fprintf( file, "        __frame->finally_level = 0; \\\n");
4328     fprintf( file, "    } while (0)\n");
4329     fprintf( file, "\n");
4330     fprintf( file, "#else /* USE_COMPILER_EXCEPTIONS */\n");
4331     fprintf( file, "\n");
4332     fprintf( file, "#define RpcExceptionInit(filter_func,finally_func) \\\n");
4333     fprintf( file, "    do { (void)(filter_func); } while(0)\n");
4334     fprintf( file, "\n");
4335     fprintf( file, "#define __DECL_EXCEPTION_FRAME \\\n");
4336     fprintf( file, "    DWORD code;\n");
4337     fprintf( file, "\n");
4338     fprintf( file, "#endif /* USE_COMPILER_EXCEPTIONS */\n");
4339 }