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