push 9d6e510d5bb9560a6aa2e6ec402cb2d2444b5081
[wine/hacks.git] / tools / widl / typegen.c
blobdfcb31e13e73214e738f5442620adf94e114b4b7
1 /*
2 * Format String Generator for IDL Compiler
4 * Copyright 2005-2006 Eric Kohl
5 * Copyright 2005-2006 Robert Shearman
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.
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.
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
22 #include "config.h"
23 #include "wine/port.h"
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>
35 #include "widl.h"
36 #include "utils.h"
37 #include "parser.h"
38 #include "header.h"
39 #include "typetree.h"
41 #include "typegen.h"
42 #include "expr.h"
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)))
49 static const var_t *current_func;
50 static const type_t *current_structure;
51 static const type_t *current_iface;
53 static struct list expr_eval_routines = LIST_INIT(expr_eval_routines);
54 struct expr_eval_routine
56 struct list entry;
57 const type_t *structure;
58 unsigned int baseoff;
59 const expr_t *expr;
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);
71 const char *string_of_type(unsigned char type)
73 switch (type)
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;
123 static unsigned char get_pointer_fc(const type_t *type)
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;
132 static int get_struct_type(const type_t *type)
134 int has_pointer = 0;
135 int has_conformance = 0;
136 int has_variance = 0;
137 var_t *field;
138 var_list_t *fields;
140 if (type->type != RPC_FC_STRUCT) return type->type;
142 fields = type_struct_get_fields(type);
144 if (get_padding(fields))
145 return RPC_FC_BOGUS_STRUCT;
147 if (fields) LIST_FOR_EACH_ENTRY( field, fields, var_t, entry )
149 type_t *t = field->type;
151 if (is_user_type(t))
152 return RPC_FC_BOGUS_STRUCT;
154 if (field->type->declarray)
156 if (is_string_type(field->attrs, field->type))
158 if (is_conformant_array(field->type))
159 has_conformance = 1;
160 has_variance = 1;
161 continue;
164 if (is_array(type_array_get_element(field->type)))
165 return RPC_FC_BOGUS_STRUCT;
167 if (type_array_has_conformance(field->type))
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);
174 if (type_array_has_variance(field->type))
175 has_variance = 1;
177 t = type_array_get_element(field->type);
180 switch (get_struct_type(t))
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.
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;
207 case RPC_FC_RP:
208 return RPC_FC_BOGUS_STRUCT;
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;
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:
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;
240 * Propagate member attributes
241 * a struct should be at least as complex as its member
243 case RPC_FC_CVSTRUCT:
244 has_conformance = 1;
245 has_variance = 1;
246 has_pointer = 1;
247 break;
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;
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;
264 case RPC_FC_PSTRUCT:
265 has_pointer = 1;
266 break;
268 default:
269 error_loc("Unknown struct member %s with type (0x%02x)\n", field->name, t->type);
270 /* fallthru - treat it as complex */
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;
281 if( has_variance )
283 if ( has_conformance )
284 return RPC_FC_CVSTRUCT;
285 else
286 return RPC_FC_BOGUS_STRUCT;
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;
297 static unsigned char get_array_type(const type_t *type)
299 unsigned char fc;
300 const expr_t *size_is;
301 const type_t *elem_type;
303 if (!is_array(type))
304 return type->type;
306 elem_type = type_array_get_element(type);
307 size_is = type_array_get_conformance(type);
309 if (!size_is)
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;
318 else
319 fc = RPC_FC_CARRAY;
321 if (type_array_has_variance(type))
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;
331 if (is_user_type(elem_type))
332 fc = RPC_FC_BOGUS_ARRAY;
333 else if (is_struct(elem_type->type))
335 switch (get_struct_type(elem_type))
337 case RPC_FC_BOGUS_STRUCT:
338 fc = RPC_FC_BOGUS_ARRAY;
339 break;
342 else if (elem_type->type == RPC_FC_ENUM16)
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;
348 else if (is_union(elem_type->type))
349 fc = RPC_FC_BOGUS_ARRAY;
350 else if (is_ptr(elem_type))
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;
360 return fc;
363 int is_struct(unsigned char type)
365 switch (type)
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;
379 static int is_non_complex_struct(const type_t *type)
381 switch (get_struct_type(type))
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;
394 int is_union(unsigned char type)
396 switch (type)
398 case RPC_FC_ENCAPSULATED_UNION:
399 case RPC_FC_NON_ENCAPSULATED_UNION:
400 return 1;
401 default:
402 return 0;
406 static int type_has_pointers(const type_t *type)
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))
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 )
420 if (type_has_pointers(field->type))
421 return TRUE;
424 else if (is_union(type->type))
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 )
431 if (field->type && type_has_pointers(field->type))
432 return TRUE;
436 return FALSE;
439 static int type_has_full_pointer(const type_t *type)
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))
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 )
455 if (type_has_full_pointer(field->type))
456 return TRUE;
459 else if (is_union(type->type))
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 )
466 if (field->type && type_has_full_pointer(field->type))
467 return TRUE;
471 return FALSE;
474 static unsigned short user_type_offset(const char *name)
476 user_type_t *ut;
477 unsigned short off = 0;
478 LIST_FOR_EACH_ENTRY(ut, &user_type_list, user_type_t, entry)
480 if (strcmp(name, ut->name) == 0)
481 return off;
482 ++off;
484 error("user_type_offset: couldn't find type (%s)\n", name);
485 return 0;
488 static void update_tfsoff(type_t *type, unsigned int offset, FILE *file)
490 type->typestring_offset = offset;
491 if (file) type->tfswrite = FALSE;
494 static void guard_rec(type_t *type)
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;
505 static type_t *get_user_type(const type_t *t, const char **pname)
507 for (;;)
509 type_t *ut = get_attrp(t->attrs, ATTR_WIREMARSHAL);
510 if (ut)
512 if (pname)
513 *pname = t->name;
514 return ut;
517 if (type_is_alias(t))
518 t = t->orig;
519 else
520 return 0;
524 int is_user_type(const type_t *t)
526 return get_user_type(t, NULL) != NULL;
529 static int is_embedded_complex(const type_t *type)
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);
536 static const char *get_context_handle_type_name(const type_t *type)
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;
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); \
552 while (0)
554 static void print_file(FILE *file, int indent, const char *format, ...)
556 va_list va;
557 va_start(va, format);
558 print(file, indent, format, va);
559 va_end(va);
562 void print(FILE *file, int indent, const char *format, va_list va)
564 if (file)
566 if (format[0] != '\n')
567 while (0 < indent--)
568 fprintf(file, " ");
569 vfprintf(file, format, va);
574 static void write_var_init(FILE *file, int indent, const type_t *t, const char *n, const char *local_var_prefix)
576 if (decl_indirect(t))
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);
582 else if (is_ptr(t) || is_array(t))
583 print_file(file, indent, "%s%s = 0;\n", local_var_prefix, n);
586 void write_parameters_init(FILE *file, int indent, const var_t *func, const char *local_var_prefix)
588 const var_t *var;
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);
593 if (!type_get_function_args(func->type))
594 return;
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);
599 fprintf(file, "\n");
602 static void write_formatdesc(FILE *f, int indent, const char *str)
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");
612 void write_formatstringsdecl(FILE *f, int indent, const statement_list_t *stmts, type_pred_t pred)
614 clear_all_offsets();
616 print_file(f, indent, "#define TYPE_FORMAT_STRING_SIZE %d\n",
617 get_size_typeformatstring(stmts, pred));
619 print_file(f, indent, "#define PROC_FORMAT_STRING_SIZE %d\n",
620 get_size_procformatstring(stmts, pred));
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");
631 static inline int is_base_type(unsigned char type)
633 switch (type)
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;
654 default:
655 return FALSE;
659 int decl_indirect(const type_t *t)
661 return is_user_type(t)
662 || (!is_base_type(t->type)
663 && !is_ptr(t)
664 && !is_array(t));
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)
673 size_t size;
675 int is_in = is_attr(attrs, ATTR_IN);
676 int is_out = is_attr(attrs, ATTR_OUT);
678 if (!is_in && !is_out) is_in = TRUE;
680 if (!type->declarray && is_base_type(type->type))
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");
687 if (type->type == RPC_FC_BIND_PRIMITIVE)
689 print_file(file, indent, "0x%02x, /* FC_IGNORE */\n", RPC_FC_IGNORE);
690 size = 2; /* includes param type prefix */
692 else if (is_base_type(type->type))
694 print_file(file, indent, "0x%02x, /* %s */\n", type->type, string_of_type(type->type));
695 size = 2; /* includes param type prefix */
697 else
699 error("Unknown/unsupported type: %s (0x%02x)\n", name, type->type);
700 size = 0;
703 else
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");
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 */
718 return size;
721 static void write_procformatstring_stmts(FILE *file, int indent, const statement_list_t *stmts, type_pred_t pred)
723 const statement_t *stmt;
724 if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
726 if (stmt->type == STMT_TYPE && stmt->u.type->type == RPC_FC_IP)
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))
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))
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);
743 /* emit return value data */
744 if (is_void(type_function_get_rettype(func->type)))
746 print_file(file, indent, "0x5b, /* FC_END */\n");
747 print_file(file, indent, "0x5c, /* FC_PAD */\n");
749 else
750 write_procformatstring_type(file, indent, "return value", type_function_get_rettype(func->type), NULL, TRUE);
753 else if (stmt->type == STMT_LIBRARY)
754 write_procformatstring_stmts(file, indent, stmt->u.lib->stmts, pred);
758 void write_procformatstring(FILE *file, const statement_list_t *stmts, type_pred_t pred)
760 int indent = 0;
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++;
769 write_procformatstring_stmts(file, indent, stmts, pred);
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");
779 static int write_base_type(FILE *file, const type_t *type, unsigned int *typestring_offset)
781 if (is_base_type(type->type))
783 print_file(file, 2, "0x%02x,\t/* %s */\n", type->type, string_of_type(type->type));
784 *typestring_offset += 1;
785 return 1;
788 return 0;
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)
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;
802 if (!expr)
804 print_file(file, 2, "NdrFcLong(0xffffffff),\t/* -1 */\n");
805 return 4;
808 if (!structure)
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;
818 if (expr->is_const)
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);
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);
830 return 4;
833 if (is_ptr(type) || (is_array(type) && !type->declarray))
835 conftype = RPC_FC_POINTER_CONFORMANCE;
836 conftype_string = "field pointer, ";
839 subexpr = expr;
840 switch (subexpr->type)
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))
850 subexpr = subexpr->ref;
851 operator_type = RPC_FC_DIV_2;
852 operator_string = "FC_DIV_2";
854 break;
855 case EXPR_MUL:
856 if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 2))
858 subexpr = subexpr->ref;
859 operator_type = RPC_FC_MULT_2;
860 operator_string = "FC_MULT_2";
862 break;
863 case EXPR_SUB:
864 if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 1))
866 subexpr = subexpr->ref;
867 operator_type = RPC_FC_SUB_1;
868 operator_string = "FC_SUB_1";
870 break;
871 case EXPR_ADD:
872 if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 1))
874 subexpr = subexpr->ref;
875 operator_type = RPC_FC_ADD_1;
876 operator_string = "FC_ADD_1";
878 break;
879 default:
880 break;
883 if (subexpr->type == EXPR_IDENTIFIER)
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);
892 if (fields) LIST_FOR_EACH_ENTRY( var, fields, const var_t, entry )
894 unsigned int align = 0;
895 /* FIXME: take alignment into account */
896 if (var->name && !strcmp(var->name, subexpr->u.sval))
898 correlation_variable = var->type;
899 break;
901 offset += type_memsize(var->type, &align);
903 if (!correlation_variable)
904 error("write_conf_or_var_desc: couldn't find variable %s in structure\n",
905 subexpr->u.sval);
907 correlation_variable = expr_resolve_type(NULL, structure, expr);
909 offset -= baseoff;
910 correlation_variable_type = correlation_variable->type;
912 switch (correlation_variable_type)
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);
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);
948 else
950 unsigned int callback_offset = 0;
951 struct expr_eval_routine *eval;
952 int found = 0;
954 LIST_FOR_EACH_ENTRY(eval, &expr_eval_routines, struct expr_eval_routine, entry)
956 if (!strcmp (eval->structure->name, structure->name)
957 && !compare_expr (eval->expr, expr))
959 found = 1;
960 break;
962 callback_offset++;
965 if (!found)
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);
974 if (callback_offset > USHRT_MAX)
975 error("Maximum number of callback routines reached\n");
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);
981 return 4;
984 static size_t fields_memsize(const var_list_t *fields, unsigned int *align)
986 int have_align = FALSE;
987 size_t size = 0;
988 const var_t *v;
990 if (!fields) return 0;
991 LIST_FOR_EACH_ENTRY( v, fields, const var_t, entry )
993 unsigned int falign = 0;
994 size_t fsize = type_memsize(v->type, &falign);
995 if (!have_align)
997 *align = falign;
998 have_align = TRUE;
1000 size = ROUND_SIZE(size, falign);
1001 size += fsize;
1004 size = ROUND_SIZE(size, *align);
1005 return size;
1008 static size_t union_memsize(const var_list_t *fields, unsigned int *pmaxa)
1010 size_t size, maxs = 0;
1011 unsigned int align = *pmaxa;
1012 const var_t *v;
1014 if (fields) LIST_FOR_EACH_ENTRY( v, fields, const var_t, entry )
1016 /* we could have an empty default field with NULL type */
1017 if (v->type)
1019 size = type_memsize(v->type, &align);
1020 if (maxs < size) maxs = size;
1021 if (*pmaxa < align) *pmaxa = align;
1025 return maxs;
1028 int get_padding(const var_list_t *fields)
1030 unsigned short offset = 0;
1031 int salign = -1;
1032 const var_t *f;
1034 if (!fields)
1035 return 0;
1037 LIST_FOR_EACH_ENTRY(f, fields, const var_t, entry)
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;
1048 return ROUNDING(offset, salign);
1051 size_t type_memsize(const type_t *t, unsigned int *align)
1053 size_t size = 0;
1055 if (type_is_alias(t))
1056 size = type_memsize(t->orig, align);
1057 else if (t->declarray && is_conformant_array(t))
1059 type_memsize(type_array_get_element(t), align);
1060 size = 0;
1062 else if (is_ptr(t) || is_conformant_array(t))
1064 assert( pointer_size );
1065 size = pointer_size;
1066 if (size > *align) *align = size;
1068 else switch (t->type)
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;
1123 return size;
1126 int is_full_pointer_function(const var_t *func)
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;
1139 void write_full_pointer_init(FILE *file, int indent, const var_t *func, int is_server)
1141 print_file(file, indent, "__frame->_StubMsg.FullPtrXlatTables = NdrFullPointerXlatInit(0,%s);\n",
1142 is_server ? "XLAT_SERVER" : "XLAT_CLIENT");
1143 fprintf(file, "\n");
1146 void write_full_pointer_free(FILE *file, int indent, const var_t *func)
1148 print_file(file, indent, "NdrFullPointerXlatFree(__frame->_StubMsg.FullPtrXlatTables);\n");
1149 fprintf(file, "\n");
1152 static unsigned int write_nonsimple_pointer(FILE *file, const type_t *type, size_t offset)
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;
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;
1165 static unsigned int write_simple_pointer(FILE *file, const type_t *type)
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;
1179 static void print_start_tfs_comment(FILE *file, type_t *t, unsigned int tfsoff)
1181 print_file(file, 0, "/* %u (", tfsoff);
1182 write_type_decl(file, t, NULL);
1183 print_file(file, 0, ") */\n");
1186 static size_t write_pointer_tfs(FILE *file, type_t *type, unsigned int *typestring_offset)
1188 unsigned int offset = *typestring_offset;
1190 print_start_tfs_comment(file, type, offset);
1191 update_tfsoff(type, offset, file);
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);
1198 return offset;
1201 static int processed(const type_t *type)
1203 return type->typestring_offset && !type->tfswrite;
1206 static int user_type_has_variable_size(const type_t *t)
1208 if (is_ptr(t))
1209 return TRUE;
1210 else
1211 switch (get_struct_type(t))
1213 case RPC_FC_PSTRUCT:
1214 case RPC_FC_CSTRUCT:
1215 case RPC_FC_CPSTRUCT:
1216 case RPC_FC_CVSTRUCT:
1217 return TRUE;
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;
1224 static void write_user_tfs(FILE *file, type_t *type, unsigned int *tfsoff)
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;
1235 guard_rec(type);
1237 if (is_base_type(utype->type))
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;
1245 else
1247 if (!processed(utype))
1248 write_embedded_types(file, NULL, utype, utype->name, TRUE, tfsoff);
1249 absoff = utype->typestring_offset;
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;
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;
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)
1278 if (is_embedded_complex(type) && !is_conformant_array(type))
1280 size_t absoff;
1281 short reloff;
1283 if (is_union(type->type) && is_attr(attrs, ATTR_SWITCHIS))
1285 absoff = *corroff;
1286 *corroff += 8;
1288 else
1290 absoff = type->typestring_offset;
1292 reloff = absoff - (*tfsoff + 2);
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;
1301 else if (is_ptr(type) || is_conformant_array(type))
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;
1309 else if (!write_base_type(file, type, tfsoff))
1310 error("Unsupported member type 0x%x\n", type->type);
1313 static void write_end(FILE *file, unsigned int *tfsoff)
1315 if (*tfsoff % 2 == 0)
1317 print_file(file, 2, "0x%x,\t\t/* FC_PAD */\n", RPC_FC_PAD);
1318 *tfsoff += 1;
1320 print_file(file, 2, "0x%x,\t\t/* FC_END */\n", RPC_FC_END);
1321 *tfsoff += 1;
1324 static void write_descriptors(FILE *file, type_t *type, unsigned int *tfsoff)
1326 unsigned int offset = 0;
1327 var_list_t *fs = type_struct_get_fields(type);
1328 var_t *f;
1330 if (fs) LIST_FOR_EACH_ENTRY(f, fs, var_t, entry)
1332 unsigned int align = 0;
1333 type_t *ft = f->type;
1334 if (is_union(ft->type) && is_attr(f->attrs, ATTR_SWITCHIS))
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;
1348 /* FIXME: take alignment into account */
1349 offset += type_memsize(ft, &align);
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)
1358 int written = 0;
1359 unsigned int align;
1361 if (is_ptr(type) || (!type->declarray && is_conformant_array(type)))
1363 size_t memsize;
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);
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;
1373 if (is_ptr(type))
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);
1380 else
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;
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;
1398 return 1;
1401 if (is_non_complex_struct(type))
1403 const var_t *v;
1404 LIST_FOR_EACH_ENTRY( v, type_struct_get_fields(type), const var_t, entry )
1406 if (offset_in_memory && offset_in_buffer)
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;
1415 written += write_no_repeat_pointer_descriptions(
1416 file, v->type,
1417 offset_in_memory, offset_in_buffer, typestring_offset);
1420 else
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;
1431 return written;
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)
1439 int written = 0;
1440 unsigned int align;
1442 if (is_ptr(type) && type_pointer_get_ref(type)->type != RPC_FC_IP)
1444 if (offset_in_memory && offset_in_buffer)
1446 size_t memsize;
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);
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;
1460 *typestring_offset += 4;
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");
1470 return 1;
1473 if (is_array(type))
1475 return write_pointer_description_offsets(
1476 file, attrs, type_array_get_element(type), offset_in_memory,
1477 offset_in_buffer, typestring_offset);
1479 else if (is_non_complex_struct(type))
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 )
1485 if (offset_in_memory && offset_in_buffer)
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;
1494 written += write_pointer_description_offsets(
1495 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
1496 typestring_offset);
1499 else
1501 if (offset_in_memory && offset_in_buffer)
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;
1513 return written;
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)
1523 unsigned int align;
1524 int pointer_count = 0;
1525 int real_type = get_array_type( type );
1527 if (real_type == RPC_FC_SMFARRAY || real_type == RPC_FC_LGFARRAY)
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)
1536 unsigned int increment_size;
1537 size_t offset_of_array_pointer_mem = 0;
1538 size_t offset_of_array_pointer_buf = 0;
1540 align = 0;
1541 increment_size = type_memsize(type_array_get_element(type), &align);
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;
1551 pointer_count = write_pointer_description_offsets(
1552 file, attrs, type, &offset_of_array_pointer_mem,
1553 &offset_of_array_pointer_buf, typestring_offset);
1556 else if (is_struct(type->type))
1558 const var_t *v;
1559 LIST_FOR_EACH_ENTRY( v, type_struct_get_fields(type), const var_t, entry )
1561 if (offset_in_memory && offset_in_buffer)
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;
1570 pointer_count += write_fixed_array_pointer_descriptions(
1571 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
1572 typestring_offset);
1575 else
1577 if (offset_in_memory && offset_in_buffer)
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;
1589 return pointer_count;
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)
1598 unsigned int align;
1599 int pointer_count = 0;
1601 if (is_conformant_array(type) && !type_array_has_variance(type))
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)
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;
1614 align = 0;
1615 increment_size = type_memsize(type_array_get_element(type), &align);
1617 if (increment_size > USHRT_MAX)
1618 error("array size of %u bytes is too large\n", increment_size);
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;
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);
1634 return pointer_count;
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)
1644 unsigned int align;
1645 int pointer_count = 0;
1647 if (is_array(type) && type_array_has_variance(type))
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)
1656 unsigned int increment_size;
1658 align = 0;
1659 increment_size = type_memsize(type_array_get_element(type), &align);
1661 if (increment_size > USHRT_MAX)
1662 error("array size of %u bytes is too large\n", increment_size);
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;
1671 pointer_count = write_pointer_description_offsets(
1672 file, attrs, type, offset_in_memory,
1673 offset_in_buffer, typestring_offset);
1676 else if (is_struct(type->type))
1678 const var_t *v;
1679 LIST_FOR_EACH_ENTRY( v, type_struct_get_fields(type), const var_t, entry )
1681 if (offset_in_memory && offset_in_buffer)
1683 size_t padding;
1685 if (is_array(v->type) && type_array_has_variance(v->type))
1687 *offset_in_buffer = ROUND_SIZE(*offset_in_buffer, 4);
1688 /* skip over variance and offset in buffer */
1689 *offset_in_buffer += 8;
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;
1698 pointer_count += write_varying_array_pointer_descriptions(
1699 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
1700 typestring_offset);
1703 else
1705 if (offset_in_memory && offset_in_buffer)
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;
1717 return pointer_count;
1720 static void write_pointer_description(FILE *file, type_t *type,
1721 unsigned int *typestring_offset)
1723 size_t offset_in_buffer;
1724 size_t offset_in_memory;
1726 /* pass 1: search for single instance of a pointer (i.e. don't descend
1727 * into arrays) */
1728 if (!is_array(type))
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);
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);
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)
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);
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);
1767 int is_declptr(const type_t *t)
1769 return is_ptr(t) || (is_conformant_array(t) && !t->declarray);
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)
1776 size_t start_offset;
1777 unsigned char rtype;
1779 if (is_declptr(type))
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)
1792 print_file(file, 2, "NdrFcShort(0x2),\n");
1793 *typestring_offset += 2;
1797 start_offset = *typestring_offset;
1798 update_tfsoff(type, start_offset, file);
1800 if (is_array(type))
1801 rtype = type_array_get_element(type)->type;
1802 else
1803 rtype = type_pointer_get_ref(type)->type;
1805 if ((rtype != RPC_FC_BYTE) && (rtype != RPC_FC_CHAR) && (rtype != RPC_FC_WCHAR))
1807 error("write_string_tfs: Unimplemented for type 0x%x of name: %s\n", rtype, name);
1808 return start_offset;
1811 if (type->declarray && !is_conformant_array(type))
1813 unsigned long dim = type_array_get_dim(type);
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);
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;
1827 print_file(file, 2, "NdrFcShort(0x%x), /* %d */\n", dim, dim);
1828 *typestring_offset += 2;
1830 return start_offset;
1832 else if (is_conformant_array(type))
1834 unsigned int align = 0;
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;
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));
1850 return start_offset;
1852 else
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;
1861 return start_offset;
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)
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;
1881 if (!pointer_type)
1882 pointer_type = RPC_FC_RP;
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));
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 );
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;
1900 align = 0;
1901 if (real_type != RPC_FC_BOGUS_ARRAY)
1903 if (real_type == RPC_FC_LGFARRAY || real_type == RPC_FC_LGVARRAY)
1905 print_file(file, 2, "NdrFcLong(0x%x),\t/* %lu */\n", size, size);
1906 *typestring_offset += 4;
1908 else
1910 print_file(file, 2, "NdrFcShort(0x%x),\t/* %lu */\n", size, size);
1911 *typestring_offset += 2;
1914 if (is_conformant_array(type))
1915 *typestring_offset
1916 += write_conf_or_var_desc(file, current_structure, baseoff,
1917 type, size_is);
1919 if (real_type == RPC_FC_SMVARRAY || real_type == RPC_FC_LGVARRAY)
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);
1925 if (real_type == RPC_FC_LGVARRAY)
1927 print_file(file, 2, "NdrFcLong(0x%x),\t/* %lu */\n", dim, dim);
1928 *typestring_offset += 4;
1930 else
1932 print_file(file, 2, "NdrFcShort(0x%x),\t/* %lu */\n", dim, dim);
1933 *typestring_offset += 2;
1936 print_file(file, 2, "NdrFcShort(0x%x),\t/* %lu */\n", elsize, elsize);
1937 *typestring_offset += 2;
1940 if (length_is)
1941 *typestring_offset
1942 += write_conf_or_var_desc(file, current_structure, baseoff,
1943 type, length_is);
1945 if (has_pointer && (!type->declarray || !current_structure))
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;
1955 write_member_type(file, type, NULL, type_array_get_element(type), NULL, typestring_offset);
1956 write_end(file, typestring_offset);
1958 else
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);
1973 return start_offset;
1976 static const var_t *find_array_or_string_in_struct(const type_t *type)
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;
1983 if (!fields || list_empty(fields))
1984 return NULL;
1986 last_field = LIST_ENTRY( list_tail(fields), const var_t, entry );
1987 ft = last_field->type;
1989 if (ft->declarray && is_conformant_array(ft))
1990 return last_field;
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;
1999 static void write_struct_members(FILE *file, const type_t *type,
2000 unsigned int *corroff, unsigned int *typestring_offset)
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);
2008 if (fields) LIST_FOR_EACH_ENTRY( field, fields, const var_t, entry )
2010 type_t *ft = field->type;
2011 if (!ft->declarray || !is_conformant_array(ft))
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)
2019 unsigned char fc = 0;
2020 switch (align)
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);
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;
2035 write_member_type(file, type, field->attrs, field->type, corroff,
2036 typestring_offset);
2037 offset += size;
2041 padding = ROUNDING(offset, salign);
2042 if (padding)
2044 print_file(file, 2, "0x%x,\t/* FC_STRUCTPAD%d */\n",
2045 RPC_FC_STRUCTPAD1 + padding - 1,
2046 padding);
2047 *typestring_offset += 1;
2050 write_end(file, typestring_offset);
2053 static size_t write_struct_tfs(FILE *file, type_t *type,
2054 const char *name, unsigned int *tfsoff)
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);
2068 guard_rec(type);
2069 current_structure = type;
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);
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);
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);
2088 corroff = *tfsoff;
2089 write_descriptors(file, type, tfsoff);
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;
2099 if (array)
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;
2107 else if (real_type == RPC_FC_BOGUS_STRUCT)
2109 print_file(file, 2, "NdrFcShort(0x0),\n");
2110 *tfsoff += 2;
2113 if (real_type == RPC_FC_BOGUS_STRUCT)
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;
2125 else if ((real_type == RPC_FC_PSTRUCT) ||
2126 (real_type == RPC_FC_CPSTRUCT) ||
2127 (real_type == RPC_FC_CVSTRUCT && has_pointers))
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;
2137 write_struct_members(file, type, &corroff, tfsoff);
2139 if (real_type == RPC_FC_BOGUS_STRUCT)
2141 const var_t *f;
2143 type->ptrdesc = *tfsoff;
2144 if (fields) LIST_FOR_EACH_ENTRY(f, fields, const var_t, entry)
2146 type_t *ft = f->type;
2147 if (is_ptr(ft))
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);
2154 else if (!ft->declarray && is_conformant_array(ft))
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;
2171 if (type->ptrdesc == *tfsoff)
2172 type->ptrdesc = 0;
2175 current_structure = save_current_structure;
2176 return start_offset;
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)
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;
2190 if (out_attr && !in_attr && pointer_type == RPC_FC_RP)
2191 flags |= 0x04;
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)
2199 if (flags & 0x04)
2200 fprintf(file, " [allocated_on_stack]");
2201 if (flags & 0x10)
2202 fprintf(file, " [pointer_deref]");
2203 fprintf(file, " */\n");
2206 print_file(file, 2, "NdrFcShort(0x%x),\t/* %d */\n", reloff, offset);
2207 *typeformat_offset += 4;
2209 return start_offset;
2212 static void write_branch_type(FILE *file, const type_t *t, unsigned int *tfsoff)
2214 if (t == NULL)
2216 print_file(file, 2, "NdrFcShort(0x0),\t/* No type */\n");
2218 else if (is_base_type(t->type))
2220 print_file(file, 2, "NdrFcShort(0x80%02x),\t/* Simple arm type: %s */\n",
2221 t->type, string_of_type(t->type));
2223 else if (t->typestring_offset)
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);
2229 else
2230 error("write_branch_type: type unimplemented (0x%x)\n", t->type);
2232 *tfsoff += 2;
2235 static size_t write_union_tfs(FILE *file, type_t *type, unsigned int *tfsoff)
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;
2246 guard_rec(type);
2248 fields = type_union_get_cases(type);
2250 if (fields) LIST_FOR_EACH_ENTRY(f, fields, var_t, entry)
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);
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)
2264 const var_t *sv = type_union_get_switch_value(type);
2265 const type_t *st = sv->type;
2267 switch (st->type)
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");
2287 else if (is_attr(type->attrs, ATTR_SWITCHTYPE))
2289 static const expr_t dummy_expr; /* FIXME */
2290 const type_t *st = get_attrp(type->attrs, ATTR_SWITCHTYPE);
2292 switch (st->type)
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");
2312 *tfsoff += write_conf_or_var_desc(file, NULL, *tfsoff, st, &dummy_expr );
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;
2319 if (fields) LIST_FOR_EACH_ENTRY(f, fields, var_t, entry)
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;
2326 if (cases == NULL && !deflt)
2327 error("union field %s with neither case nor default attribute\n", f->name);
2329 if (cases) LIST_FOR_EACH_ENTRY(c, cases, expr_t, entry)
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);
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)
2344 deftype = ft;
2345 nodeftype = 0;
2349 if (deftype)
2351 write_branch_type(file, deftype, tfsoff);
2353 else
2355 print_file(file, 2, "NdrFcShort(0x%x),\n", nodeftype);
2356 *tfsoff += 2;
2359 return start_offset;
2362 static size_t write_ip_tfs(FILE *file, const attr_list_t *attrs, type_t *type,
2363 unsigned int *typeformat_offset)
2365 size_t i;
2366 size_t start_offset = *typeformat_offset;
2367 expr_t *iid = get_attrp(attrs, ATTR_IIDIS);
2369 if (iid)
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;
2376 else
2378 const type_t *base = is_ptr(type) ? type_pointer_get_ref(type) : type;
2379 const UUID *uuid = get_attrp(base->attrs, ATTR_UUID);
2381 if (! uuid)
2382 error("%s: interface %s missing UUID\n", __FUNCTION__, base->name);
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]);
2394 if (file)
2395 fprintf(file, "\n");
2397 *typeformat_offset += 18;
2399 return start_offset;
2402 static size_t write_contexthandle_tfs(FILE *file, const type_t *type,
2403 const var_t *var,
2404 unsigned int *typeformat_offset)
2406 size_t start_offset = *typeformat_offset;
2407 unsigned char flags = 0;
2409 if (is_attr(current_iface->attrs, ATTR_STRICTCONTEXTHANDLE))
2410 flags |= NDR_STRICT_CONTEXT_HANDLE;
2412 if (is_ptr(type))
2413 flags |= 0x80;
2414 if (is_attr(var->attrs, ATTR_IN))
2416 flags |= 0x40;
2417 if (!is_attr(var->attrs, ATTR_OUT))
2418 flags |= NDR_CONTEXT_HANDLE_CANNOT_BE_NULL;
2420 if (is_attr(var->attrs, ATTR_OUT))
2421 flags |= 0x20;
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;
2447 return start_offset;
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)
2454 size_t offset;
2456 if (is_context_handle(type))
2457 return write_contexthandle_tfs(file, type, var, typeformat_offset);
2459 if (is_user_type(type))
2461 write_user_tfs(file, type, typeformat_offset);
2462 return type->typestring_offset;
2465 if (is_string_type(var->attrs, type))
2466 return write_string_tfs(file, var->attrs, type, var->name, typeformat_offset);
2468 if (is_array(type))
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)
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;
2494 return off;
2497 if (!is_ptr(type))
2499 /* basic types don't need a type format string */
2500 if (is_base_type(type->type))
2501 return 0;
2503 if (processed(type)) return type->typestring_offset;
2505 switch (type->type)
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);
2525 else if (last_ptr(type))
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);
2532 if (base->type == RPC_FC_IP
2533 || (base->type == 0
2534 && is_attr(var->attrs, ATTR_IIDIS)))
2536 return write_ip_tfs(file, var->attrs, type, typeformat_offset);
2539 /* special case for pointers to base types */
2540 if (is_base_type(base->type))
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;
2553 assert(is_ptr(type));
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);
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)
2568 int retmask = 0;
2570 if (is_user_type(type))
2572 write_user_tfs(file, type, tfsoff);
2574 else if (is_string_type(attrs, type))
2576 write_string_tfs(file, attrs, type, name, tfsoff);
2578 else if (is_ptr(type))
2580 type_t *ref = type_pointer_get_ref(type);
2582 if (ref->type == RPC_FC_IP
2583 || (ref->type == 0
2584 && is_attr(attrs, ATTR_IIDIS)))
2586 write_ip_tfs(file, attrs, type, tfsoff);
2588 else
2590 if (!processed(ref) && !is_base_type(ref->type))
2591 retmask |= write_embedded_types(file, NULL, ref, name, TRUE, tfsoff);
2593 if (write_ptr)
2594 write_pointer_tfs(file, type, tfsoff);
2596 retmask |= 1;
2599 else if (type->declarray && is_conformant_array(type))
2600 ; /* conformant arrays and strings are handled specially */
2601 else if (is_array(type))
2603 write_array_tfs(file, attrs, type, name, tfsoff);
2604 if (is_conformant_array(type))
2605 retmask |= 1;
2607 else if (is_struct(type->type))
2609 if (!processed(type))
2610 write_struct_tfs(file, type, name, tfsoff);
2612 else if (is_union(type->type))
2614 if (!processed(type))
2615 write_union_tfs(file, type, tfsoff);
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);
2621 return retmask;
2624 static size_t process_tfs_stmts(FILE *file, const statement_list_t *stmts,
2625 type_pred_t pred, unsigned int *typeformat_offset)
2627 const var_t *var;
2628 const statement_t *stmt;
2630 if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
2632 const type_t *iface;
2633 const statement_t *stmt_func;
2635 if (stmt->type == STMT_LIBRARY)
2637 process_tfs_stmts(file, stmt->u.lib->stmts, pred, typeformat_offset);
2638 continue;
2640 else if (stmt->type != STMT_TYPE || stmt->u.type->type != RPC_FC_IP)
2641 continue;
2643 iface = stmt->u.type;
2644 if (!pred(iface))
2645 continue;
2647 current_iface = iface;
2648 STATEMENTS_FOR_EACH_FUNC( stmt_func, type_iface_get_stmts(iface) )
2650 const var_t *func = stmt_func->u.var;
2651 if (is_local(func->attrs)) continue;
2653 if (!is_void(type_function_get_rettype(func->type)))
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);
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);
2677 return *typeformat_offset + 1;
2680 static size_t process_tfs(FILE *file, const statement_list_t *stmts, type_pred_t pred)
2682 unsigned int typeformat_offset = 2;
2684 return process_tfs_stmts(file, stmts, pred, &typeformat_offset);
2688 void write_typeformatstring(FILE *file, const statement_list_t *stmts, type_pred_t pred)
2690 int indent = 0;
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");
2700 set_all_tfswrite(TRUE);
2701 process_tfs(file, stmts, pred);
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");
2711 static unsigned int get_required_buffer_size_type(
2712 const type_t *type, const char *name, unsigned int *alignment)
2714 const char *uname;
2715 const type_t *utype;
2717 *alignment = 0;
2718 if ((utype = get_user_type(type, &uname)))
2720 return get_required_buffer_size_type(utype, uname, alignment);
2722 else
2724 switch (get_struct_type(type))
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;
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;
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;
2748 case RPC_FC_HYPER:
2749 case RPC_FC_DOUBLE:
2750 *alignment = 8;
2751 return 8;
2753 case RPC_FC_IGNORE:
2754 case RPC_FC_BIND_PRIMITIVE:
2755 return 0;
2757 case RPC_FC_STRUCT:
2758 if (!type_struct_get_fields(type)) return 0;
2759 return fields_memsize(type_struct_get_fields(type), alignment);
2761 case RPC_FC_RP:
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;
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);
2772 default:
2773 return 0;
2778 static unsigned int get_required_buffer_size(const var_t *var, unsigned int *alignment, enum pass pass)
2780 int in_attr = is_attr(var->attrs, ATTR_IN);
2781 int out_attr = is_attr(var->attrs, ATTR_OUT);
2783 if (!in_attr && !out_attr)
2784 in_attr = 1;
2786 *alignment = 0;
2788 if ((pass == PASS_IN && in_attr) || (pass == PASS_OUT && out_attr) ||
2789 pass == PASS_RETURN)
2791 if (is_ptrchain_attr(var, ATTR_CONTEXTHANDLE))
2793 *alignment = 4;
2794 return 20;
2797 if (!is_string_type(var->attrs, var->type))
2798 return get_required_buffer_size_type(var->type, var->name,
2799 alignment);
2801 return 0;
2804 static unsigned int get_function_buffer_size( const var_t *func, enum pass pass )
2806 const var_t *var;
2807 unsigned int total_size = 0, alignment;
2809 if (type_get_function_args(func->type))
2811 LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
2813 total_size += get_required_buffer_size(var, &alignment, pass);
2814 total_size += alignment;
2818 if (pass == PASS_OUT && !is_void(type_function_get_rettype(func->type)))
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;
2825 return total_size;
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)
2832 const char *function;
2833 switch (phase)
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;
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--;
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)
2872 type_t *type = var->type;
2873 unsigned int size;
2874 unsigned int alignment = 0;
2875 unsigned char rtype;
2877 /* no work to do for other phases, buffer sizing is done elsewhere */
2878 if (phase != PHASE_MARSHAL && phase != PHASE_UNMARSHAL)
2879 return;
2881 rtype = is_ptr(type) ? type_pointer_get_ref(type)->type : type->type;
2883 switch (rtype)
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;
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;
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;
2910 case RPC_FC_HYPER:
2911 case RPC_FC_DOUBLE:
2912 size = 8;
2913 alignment = 8;
2914 break;
2916 case RPC_FC_IGNORE:
2917 case RPC_FC_BIND_PRIMITIVE:
2918 /* no marshalling needed */
2919 return;
2921 default:
2922 error("print_phase_basetype: Unsupported type: %s (0x%02x, ptr_level: 0)\n", var->name, rtype);
2923 size = 0;
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);
2931 if (phase == PHASE_MARSHAL)
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");
2942 else if (phase == PHASE_UNMARSHAL)
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");
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");
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)
2972 return (phase != PHASE_UNMARSHAL);
2975 expr_t *get_size_is_expr(const type_t *t, const char *name)
2977 expr_t *x = NULL;
2979 for ( ; is_array(t); t = type_array_get_element(t))
2980 if (type_array_has_conformance(t))
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);
2990 return x;
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)
2996 const type_t *type = var->type;
2997 /* get fundamental type for the argument */
2998 for (;;)
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))
3006 if (is_conformance_needed_for_phase(phase) && is_array(type))
3008 if (type_array_has_conformance(type))
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");
3014 if (type_array_has_variance(type))
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");
3022 break;
3024 else if (type->type == RPC_FC_NON_ENCAPSULATED_UNION)
3026 if (is_conformance_needed_for_phase(phase))
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");
3032 break;
3034 else if (type->type == RPC_FC_IP)
3036 expr_t *iid;
3038 if (is_conformance_needed_for_phase(phase) && (iid = get_attrp( var->attrs, ATTR_IIDIS )))
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" );
3044 break;
3046 else if (is_ptr(type))
3047 type = type_pointer_get_ref(type);
3048 else
3049 break;
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)
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;
3061 pointer_type = get_attrv(var->attrs, ATTR_POINTERTYPE);
3062 if (!pointer_type)
3063 pointer_type = RPC_FC_RP;
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;
3070 if (phase != PHASE_FREE)
3071 switch (pass)
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;
3083 write_parameter_conf_or_var_exprs(file, indent, local_var_prefix, phase, var);
3084 rtype = get_struct_type(type);
3086 if (is_context_handle(type))
3088 if (phase == PHASE_MARSHAL)
3090 if (pass == PASS_IN)
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");
3101 else
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);
3110 else if (phase == PHASE_UNMARSHAL)
3112 if (pass == PASS_OUT)
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");
3121 else
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);
3129 else if (is_user_type(var->type))
3131 print_phase_function(file, indent, "UserMarshal", local_var_prefix, phase, var, start_offset);
3133 else if (is_string_type(var->attrs, var->type))
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
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);
3148 else if (is_array(type))
3150 unsigned char tc = get_array_type( type );
3151 const char *array_type = "FixedArray";
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);
3160 if (tc == RPC_FC_SMVARRAY || tc == RPC_FC_LGVARRAY)
3162 array_type = "VaryingArray";
3164 else if (tc == RPC_FC_CARRAY)
3166 array_type = "ConformantArray";
3168 else if (tc == RPC_FC_CVARRAY || tc == RPC_FC_BOGUS_ARRAY)
3170 array_type = (tc == RPC_FC_BOGUS_ARRAY
3171 ? "ComplexArray"
3172 : "ConformantVaryingArray");
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)
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))
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);
3191 else if (!is_ptr(var->type) && is_base_type(rtype))
3193 if (phase != PHASE_FREE)
3194 print_phase_basetype(file, indent, local_var_prefix, phase, pass, var, var->name);
3196 else if (!is_ptr(var->type))
3198 switch (rtype)
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);
3218 else
3220 const type_t *ref = type_pointer_get_ref(type);
3221 if (type->type == RPC_FC_RP && is_base_type(ref->type))
3223 if (phase != PHASE_FREE)
3224 print_phase_basetype(file, indent, local_var_prefix, phase, pass, var, var->name);
3226 else if (type->type == RPC_FC_RP && get_struct_type(ref) == RPC_FC_STRUCT &&
3227 !is_user_type(ref))
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);
3234 else
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);
3242 fprintf(file, "\n");
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)
3248 if (phase == PHASE_BUFFERSIZE && pass != PASS_RETURN)
3250 unsigned int size = get_function_buffer_size( func, pass );
3251 print_file(file, indent, "__frame->_StubMsg.BufferLength = %u;\n", size);
3254 if (pass == PASS_RETURN)
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 );
3263 else
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 );
3274 size_t get_size_procformatstring_type(const char *name, const type_t *type, const attr_list_t *attrs)
3276 return write_procformatstring_type(NULL, 0, name, type, attrs, FALSE);
3280 size_t get_size_procformatstring_func(const var_t *func)
3282 const var_t *var;
3283 size_t size = 0;
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);
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);
3296 return size;
3299 size_t get_size_procformatstring(const statement_list_t *stmts, type_pred_t pred)
3301 const statement_t *stmt;
3302 size_t size = 1;
3304 if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
3306 const type_t *iface;
3307 const statement_t *stmt_func;
3309 if (stmt->type == STMT_LIBRARY)
3311 size += get_size_procformatstring(stmt->u.lib->stmts, pred) - 1;
3312 continue;
3314 else if (stmt->type != STMT_TYPE || stmt->u.type->type != RPC_FC_IP)
3315 continue;
3317 iface = stmt->u.type;
3318 if (!pred(iface))
3319 continue;
3321 STATEMENTS_FOR_EACH_FUNC( stmt_func, type_iface_get_stmts(iface) )
3323 const var_t *func = stmt_func->u.var;
3324 if (!is_local(func->attrs))
3325 size += get_size_procformatstring_func( func );
3328 return size;
3331 size_t get_size_typeformatstring(const statement_list_t *stmts, type_pred_t pred)
3333 set_all_tfswrite(FALSE);
3334 return process_tfs(NULL, stmts, pred);
3337 void declare_stub_args( FILE *file, int indent, const var_t *func )
3339 int in_attr, out_attr;
3340 int i = 0;
3341 const var_t *var;
3343 /* declare return value '_RetVal' */
3344 if (!is_void(type_function_get_rettype(func->type)))
3346 print_file(file, indent, "");
3347 write_type_decl_left(file, type_function_get_rettype(func->type));
3348 fprintf(file, " _RetVal;\n");
3351 if (!type_get_function_args(func->type))
3352 return;
3354 LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
3356 int is_string = is_string_type(var->attrs, var->type);
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;
3363 if (is_context_handle(var->type))
3364 print_file(file, indent, "NDR_SCONTEXT %s;\n", var->name);
3365 else
3367 if (!in_attr && !is_conformant_array(var->type) && !is_string)
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");
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");
3389 if (decl_indirect(var->type))
3390 print_file(file, indent, "void *_p_%s;\n", var->name);
3396 void assign_stub_out_args( FILE *file, int indent, const var_t *func, const char *local_var_prefix )
3398 int in_attr, out_attr;
3399 int i = 0, sep = 0;
3400 const var_t *var;
3402 if (!type_get_function_args(func->type))
3403 return;
3405 LIST_FOR_EACH_ENTRY( var, type_get_function_args(func->type), const var_t, entry )
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;
3413 if (!in_attr)
3415 print_file(file, indent, "%s%s", local_var_prefix, var->name);
3417 if (is_context_handle(var->type))
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);
3424 else if (is_array(var->type) &&
3425 type_array_has_conformance(var->type))
3427 unsigned int size, align = 0;
3428 type_t *type = var->type;
3430 fprintf(file, " = NdrAllocate(&__frame->_StubMsg, ");
3431 for ( ;
3432 is_array(type) && type_array_has_conformance(type);
3433 type = type_array_get_element(type))
3435 write_expr(file, type_array_get_conformance(type), TRUE,
3436 TRUE, NULL, NULL, local_var_prefix);
3437 fprintf(file, " * ");
3439 size = type_memsize(type, &align);
3440 fprintf(file, "%u);\n", size);
3442 else if (!is_string)
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++;
3450 sep = 1;
3453 if (sep)
3454 fprintf(file, "\n");
3458 int write_expr_eval_routines(FILE *file, const char *iface)
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;
3466 LIST_FOR_EACH_ENTRY(eval, &expr_eval_routines, struct expr_eval_routine, entry)
3468 const char *name = eval->structure->name;
3469 result = 1;
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++;
3483 return result;
3486 void write_expr_eval_routine_list(FILE *file, const char *iface)
3488 struct expr_eval_routine *eval;
3489 struct expr_eval_routine *cursor;
3490 unsigned short callback_offset = 0;
3492 fprintf(file, "static const EXPR_EVAL ExprEvalRoutines[] =\n");
3493 fprintf(file, "{\n");
3495 LIST_FOR_EACH_ENTRY_SAFE(eval, cursor, &expr_eval_routines, struct expr_eval_routine, entry)
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);
3504 fprintf(file, "};\n\n");
3507 void write_user_quad_list(FILE *file)
3509 user_type_t *ut;
3511 if (list_empty(&user_type_list))
3512 return;
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)
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);
3526 fprintf(file, "};\n\n");
3529 void write_endpoints( FILE *f, const char *prefix, const str_list_t *list )
3531 const struct str_list_entry_t *endpoint;
3532 const char *p;
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 )
3538 print_file( f, 1, "{ (const unsigned char *)\"" );
3539 for (p = endpoint->str; *p && *p != ':'; p++)
3541 if (*p == '"' || *p == '\\') fputc( '\\', f );
3542 fputc( *p, f );
3544 if (!*p) goto error;
3545 if (p[1] != '[') goto error;
3547 fprintf( f, "\", (const unsigned char *)\"" );
3548 for (p += 2; *p && *p != ']'; p++)
3550 if (*p == '"' || *p == '\\') fputc( '\\', f );
3551 fputc( *p, f );
3553 if (*p != ']') goto error;
3554 fprintf( f, "\" },\n" );
3556 print_file( f, 0, "};\n\n" );
3557 return;
3559 error:
3560 error("Invalid endpoint syntax '%s'\n", endpoint->str);
3563 void write_exceptions( FILE *file )
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");