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