f89f13fb9b85dc2cd6a6eec77b8dd3e9ae8c528c
[reactos.git] / reactos / tools / widl / typegen.c
1 /*
2 * Format String Generator for IDL Compiler
3 *
4 * Copyright 2005-2006 Eric Kohl
5 * Copyright 2005-2006 Robert Shearman
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 */
21
22 #include "config.h"
23 #include "wine/port.h"
24
25 #include <stdio.h>
26 #include <stdlib.h>
27 #ifdef HAVE_UNISTD_H
28 # include <unistd.h>
29 #endif
30 #include <string.h>
31 #include <assert.h>
32 #include <ctype.h>
33 #include <limits.h>
34
35 #include "widl.h"
36 #include "utils.h"
37 #include "parser.h"
38 #include "header.h"
39 #include "wine/list.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 func_t *current_func;
50 static const type_t *current_structure;
51 static const type_t *current_iface;
52
53 static struct list expr_eval_routines = LIST_INIT(expr_eval_routines);
54 struct expr_eval_routine
55 {
56 struct list entry;
57 const type_t *structure;
58 unsigned int baseoff;
59 const expr_t *expr;
60 };
61
62 static size_t fields_memsize(const var_list_t *fields, unsigned int *align);
63 static size_t write_struct_tfs(FILE *file, type_t *type, const char *name, unsigned int *tfsoff);
64 static int write_embedded_types(FILE *file, const attr_list_t *attrs, type_t *type,
65 const char *name, int write_ptr, unsigned int *tfsoff);
66 static const var_t *find_array_or_string_in_struct(const type_t *type);
67 static size_t write_string_tfs(FILE *file, const attr_list_t *attrs,
68 type_t *type,
69 const char *name, unsigned int *typestring_offset);
70
71 const char *string_of_type(unsigned char type)
72 {
73 switch (type)
74 {
75 case RPC_FC_BYTE: return "FC_BYTE";
76 case RPC_FC_CHAR: return "FC_CHAR";
77 case RPC_FC_SMALL: return "FC_SMALL";
78 case RPC_FC_USMALL: return "FC_USMALL";
79 case RPC_FC_WCHAR: return "FC_WCHAR";
80 case RPC_FC_SHORT: return "FC_SHORT";
81 case RPC_FC_USHORT: return "FC_USHORT";
82 case RPC_FC_LONG: return "FC_LONG";
83 case RPC_FC_ULONG: return "FC_ULONG";
84 case RPC_FC_FLOAT: return "FC_FLOAT";
85 case RPC_FC_HYPER: return "FC_HYPER";
86 case RPC_FC_DOUBLE: return "FC_DOUBLE";
87 case RPC_FC_ENUM16: return "FC_ENUM16";
88 case RPC_FC_ENUM32: return "FC_ENUM32";
89 case RPC_FC_IGNORE: return "FC_IGNORE";
90 case RPC_FC_ERROR_STATUS_T: return "FC_ERROR_STATUS_T";
91 case RPC_FC_RP: return "FC_RP";
92 case RPC_FC_UP: return "FC_UP";
93 case RPC_FC_OP: return "FC_OP";
94 case RPC_FC_FP: return "FC_FP";
95 case RPC_FC_ENCAPSULATED_UNION: return "FC_ENCAPSULATED_UNION";
96 case RPC_FC_NON_ENCAPSULATED_UNION: return "FC_NON_ENCAPSULATED_UNION";
97 case RPC_FC_STRUCT: return "FC_STRUCT";
98 case RPC_FC_PSTRUCT: return "FC_PSTRUCT";
99 case RPC_FC_CSTRUCT: return "FC_CSTRUCT";
100 case RPC_FC_CPSTRUCT: return "FC_CPSTRUCT";
101 case RPC_FC_CVSTRUCT: return "FC_CVSTRUCT";
102 case RPC_FC_BOGUS_STRUCT: return "FC_BOGUS_STRUCT";
103 case RPC_FC_SMFARRAY: return "FC_SMFARRAY";
104 case RPC_FC_LGFARRAY: return "FC_LGFARRAY";
105 case RPC_FC_SMVARRAY: return "FC_SMVARRAY";
106 case RPC_FC_LGVARRAY: return "FC_LGVARRAY";
107 case RPC_FC_CARRAY: return "FC_CARRAY";
108 case RPC_FC_CVARRAY: return "FC_CVARRAY";
109 case RPC_FC_BOGUS_ARRAY: return "FC_BOGUS_ARRAY";
110 case RPC_FC_ALIGNM4: return "FC_ALIGNM4";
111 case RPC_FC_ALIGNM8: return "FC_ALIGNM8";
112 case RPC_FC_POINTER: return "FC_POINTER";
113 case RPC_FC_C_CSTRING: return "FC_C_CSTRING";
114 case RPC_FC_C_WSTRING: return "FC_C_WSTRING";
115 case RPC_FC_CSTRING: return "FC_CSTRING";
116 case RPC_FC_WSTRING: return "FC_WSTRING";
117 default:
118 error("string_of_type: unknown type 0x%02x\n", type);
119 return NULL;
120 }
121 }
122
123 int is_struct(unsigned char type)
124 {
125 switch (type)
126 {
127 case RPC_FC_STRUCT:
128 case RPC_FC_PSTRUCT:
129 case RPC_FC_CSTRUCT:
130 case RPC_FC_CPSTRUCT:
131 case RPC_FC_CVSTRUCT:
132 case RPC_FC_BOGUS_STRUCT:
133 return 1;
134 default:
135 return 0;
136 }
137 }
138
139 static int is_non_complex_struct(const type_t *type)
140 {
141 switch (type->type)
142 {
143 case RPC_FC_STRUCT:
144 case RPC_FC_PSTRUCT:
145 case RPC_FC_CSTRUCT:
146 case RPC_FC_CPSTRUCT:
147 case RPC_FC_CVSTRUCT:
148 return 1;
149 default:
150 return 0;
151 }
152 }
153
154 int is_union(unsigned char type)
155 {
156 switch (type)
157 {
158 case RPC_FC_ENCAPSULATED_UNION:
159 case RPC_FC_NON_ENCAPSULATED_UNION:
160 return 1;
161 default:
162 return 0;
163 }
164 }
165
166 static int type_has_pointers(const type_t *type)
167 {
168 if (is_user_type(type))
169 return FALSE;
170 else if (is_ptr(type))
171 return TRUE;
172 else if (is_array(type))
173 return type_has_pointers(type->ref);
174 else if (is_struct(type->type))
175 {
176 const var_t *field;
177 if (type->fields_or_args) LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
178 {
179 if (type_has_pointers(field->type))
180 return TRUE;
181 }
182 }
183 else if (is_union(type->type))
184 {
185 var_list_t *fields;
186 const var_t *field;
187 if (type->type == RPC_FC_ENCAPSULATED_UNION)
188 {
189 const var_t *uv = LIST_ENTRY(list_tail(type->fields_or_args), const var_t, entry);
190 fields = uv->type->fields_or_args;
191 }
192 else
193 fields = type->fields_or_args;
194 if (fields) LIST_FOR_EACH_ENTRY( field, fields, const var_t, entry )
195 {
196 if (field->type && type_has_pointers(field->type))
197 return TRUE;
198 }
199 }
200
201 return FALSE;
202 }
203
204 static int type_has_full_pointer(const type_t *type)
205 {
206 if (is_user_type(type))
207 return FALSE;
208 else if (type->type == RPC_FC_FP)
209 return TRUE;
210 else if (is_ptr(type))
211 return FALSE;
212 else if (is_array(type))
213 return type_has_full_pointer(type->ref);
214 else if (is_struct(type->type))
215 {
216 const var_t *field;
217 if (type->fields_or_args) LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
218 {
219 if (type_has_full_pointer(field->type))
220 return TRUE;
221 }
222 }
223 else if (is_union(type->type))
224 {
225 var_list_t *fields;
226 const var_t *field;
227 if (type->type == RPC_FC_ENCAPSULATED_UNION)
228 {
229 const var_t *uv = LIST_ENTRY(list_tail(type->fields_or_args), const var_t, entry);
230 fields = uv->type->fields_or_args;
231 }
232 else
233 fields = type->fields_or_args;
234 if (fields) LIST_FOR_EACH_ENTRY( field, fields, const var_t, entry )
235 {
236 if (field->type && type_has_full_pointer(field->type))
237 return TRUE;
238 }
239 }
240
241 return FALSE;
242 }
243
244 static unsigned short user_type_offset(const char *name)
245 {
246 user_type_t *ut;
247 unsigned short off = 0;
248 LIST_FOR_EACH_ENTRY(ut, &user_type_list, user_type_t, entry)
249 {
250 if (strcmp(name, ut->name) == 0)
251 return off;
252 ++off;
253 }
254 error("user_type_offset: couldn't find type (%s)\n", name);
255 return 0;
256 }
257
258 static void update_tfsoff(type_t *type, unsigned int offset, FILE *file)
259 {
260 type->typestring_offset = offset;
261 if (file) type->tfswrite = FALSE;
262 }
263
264 static void guard_rec(type_t *type)
265 {
266 /* types that contain references to themselves (like a linked list),
267 need to be shielded from infinite recursion when writing embedded
268 types */
269 if (type->typestring_offset)
270 type->tfswrite = FALSE;
271 else
272 type->typestring_offset = 1;
273 }
274
275 static type_t *get_user_type(const type_t *t, const char **pname)
276 {
277 for (;;)
278 {
279 type_t *ut = get_attrp(t->attrs, ATTR_WIREMARSHAL);
280 if (ut)
281 {
282 if (pname)
283 *pname = t->name;
284 return ut;
285 }
286
287 if (t->kind == TKIND_ALIAS)
288 t = t->orig;
289 else
290 return 0;
291 }
292 }
293
294 int is_user_type(const type_t *t)
295 {
296 return get_user_type(t, NULL) != NULL;
297 }
298
299 static int is_embedded_complex(const type_t *type)
300 {
301 unsigned char tc = type->type;
302 return is_struct(tc) || is_union(tc) || is_array(type) || is_user_type(type)
303 || (is_ptr(type) && type->ref->type == RPC_FC_IP);
304 }
305
306 static const char *get_context_handle_type_name(const type_t *type)
307 {
308 const type_t *t;
309 for (t = type; is_ptr(t); t = t->ref)
310 if (is_attr(t->attrs, ATTR_CONTEXTHANDLE))
311 return t->name;
312 assert(0);
313 return NULL;
314 }
315
316 #define WRITE_FCTYPE(file, fctype, typestring_offset) \
317 do { \
318 if (file) \
319 fprintf(file, "/* %2u */\n", typestring_offset); \
320 print_file((file), 2, "0x%02x, /* " #fctype " */\n", RPC_##fctype); \
321 } \
322 while (0)
323
324 static void print_file(FILE *file, int indent, const char *format, ...)
325 {
326 va_list va;
327 va_start(va, format);
328 print(file, indent, format, va);
329 va_end(va);
330 }
331
332 void print(FILE *file, int indent, const char *format, va_list va)
333 {
334 if (file)
335 {
336 if (format[0] != '\n')
337 while (0 < indent--)
338 fprintf(file, " ");
339 vfprintf(file, format, va);
340 }
341 }
342
343
344 static void write_var_init(FILE *file, int indent, const type_t *t, const char *n)
345 {
346 if (decl_indirect(t))
347 print_file(file, indent, "MIDL_memset(&%s, 0, sizeof(%s));\n", n, n);
348 else if (is_ptr(t) || is_array(t))
349 print_file(file, indent, "%s = 0;\n", n);
350 }
351
352 void write_parameters_init(FILE *file, int indent, const func_t *func)
353 {
354 const var_t *var;
355
356 if (!is_void(get_func_return_type(func)))
357 write_var_init(file, indent, get_func_return_type(func), "_RetVal");
358
359 if (!func->args)
360 return;
361
362 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
363 write_var_init(file, indent, var->type, var->name);
364
365 fprintf(file, "\n");
366 }
367
368 static void write_formatdesc(FILE *f, int indent, const char *str)
369 {
370 print_file(f, indent, "typedef struct _MIDL_%s_FORMAT_STRING\n", str);
371 print_file(f, indent, "{\n");
372 print_file(f, indent + 1, "short Pad;\n");
373 print_file(f, indent + 1, "unsigned char Format[%s_FORMAT_STRING_SIZE];\n", str);
374 print_file(f, indent, "} MIDL_%s_FORMAT_STRING;\n", str);
375 print_file(f, indent, "\n");
376 }
377
378 void write_formatstringsdecl(FILE *f, int indent, const statement_list_t *stmts, type_pred_t pred)
379 {
380 print_file(f, indent, "#define TYPE_FORMAT_STRING_SIZE %d\n",
381 get_size_typeformatstring(stmts, pred));
382
383 print_file(f, indent, "#define PROC_FORMAT_STRING_SIZE %d\n",
384 get_size_procformatstring(stmts, pred));
385
386 fprintf(f, "\n");
387 write_formatdesc(f, indent, "TYPE");
388 write_formatdesc(f, indent, "PROC");
389 fprintf(f, "\n");
390 print_file(f, indent, "static const MIDL_TYPE_FORMAT_STRING __MIDL_TypeFormatString;\n");
391 print_file(f, indent, "static const MIDL_PROC_FORMAT_STRING __MIDL_ProcFormatString;\n");
392 print_file(f, indent, "\n");
393 }
394
395 static inline int is_base_type(unsigned char type)
396 {
397 switch (type)
398 {
399 case RPC_FC_BYTE:
400 case RPC_FC_CHAR:
401 case RPC_FC_USMALL:
402 case RPC_FC_SMALL:
403 case RPC_FC_WCHAR:
404 case RPC_FC_USHORT:
405 case RPC_FC_SHORT:
406 case RPC_FC_ULONG:
407 case RPC_FC_LONG:
408 case RPC_FC_HYPER:
409 case RPC_FC_IGNORE:
410 case RPC_FC_FLOAT:
411 case RPC_FC_DOUBLE:
412 case RPC_FC_ENUM16:
413 case RPC_FC_ENUM32:
414 case RPC_FC_ERROR_STATUS_T:
415 case RPC_FC_BIND_PRIMITIVE:
416 return TRUE;
417
418 default:
419 return FALSE;
420 }
421 }
422
423 int decl_indirect(const type_t *t)
424 {
425 return is_user_type(t)
426 || (!is_base_type(t->type)
427 && !is_ptr(t)
428 && !is_array(t));
429 }
430
431 static size_t write_procformatstring_type(FILE *file, int indent,
432 const char *name,
433 const type_t *type,
434 const attr_list_t *attrs,
435 int is_return)
436 {
437 size_t size;
438
439 int is_in = is_attr(attrs, ATTR_IN);
440 int is_out = is_attr(attrs, ATTR_OUT);
441
442 if (!is_in && !is_out) is_in = TRUE;
443
444 if (!type->declarray && is_base_type(type->type))
445 {
446 if (is_return)
447 print_file(file, indent, "0x53, /* FC_RETURN_PARAM_BASETYPE */\n");
448 else
449 print_file(file, indent, "0x4e, /* FC_IN_PARAM_BASETYPE */\n");
450
451 if (type->type == RPC_FC_BIND_PRIMITIVE)
452 {
453 print_file(file, indent, "0x%02x, /* FC_IGNORE */\n", RPC_FC_IGNORE);
454 size = 2; /* includes param type prefix */
455 }
456 else if (is_base_type(type->type))
457 {
458 print_file(file, indent, "0x%02x, /* %s */\n", type->type, string_of_type(type->type));
459 size = 2; /* includes param type prefix */
460 }
461 else
462 {
463 error("Unknown/unsupported type: %s (0x%02x)\n", name, type->type);
464 size = 0;
465 }
466 }
467 else
468 {
469 if (is_return)
470 print_file(file, indent, "0x52, /* FC_RETURN_PARAM */\n");
471 else if (is_in && is_out)
472 print_file(file, indent, "0x50, /* FC_IN_OUT_PARAM */\n");
473 else if (is_out)
474 print_file(file, indent, "0x51, /* FC_OUT_PARAM */\n");
475 else
476 print_file(file, indent, "0x4d, /* FC_IN_PARAM */\n");
477
478 print_file(file, indent, "0x01,\n");
479 print_file(file, indent, "NdrFcShort(0x%x),\n", type->typestring_offset);
480 size = 4; /* includes param type prefix */
481 }
482 return size;
483 }
484
485 static void write_procformatstring_stmts(FILE *file, int indent, const statement_list_t *stmts, type_pred_t pred)
486 {
487 const statement_t *stmt;
488 if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
489 {
490 if (stmt->type == STMT_TYPE && stmt->u.type->type == RPC_FC_IP)
491 {
492 const func_t *func;
493 if (!pred(stmt->u.type))
494 continue;
495 if (stmt->u.type->funcs) LIST_FOR_EACH_ENTRY( func, stmt->u.type->funcs, const func_t, entry )
496 {
497 if (is_local(func->def->attrs)) continue;
498 /* emit argument data */
499 if (func->args)
500 {
501 const var_t *var;
502 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
503 write_procformatstring_type(file, indent, var->name, var->type, var->attrs, FALSE);
504 }
505
506 /* emit return value data */
507 if (is_void(get_func_return_type(func)))
508 {
509 print_file(file, indent, "0x5b, /* FC_END */\n");
510 print_file(file, indent, "0x5c, /* FC_PAD */\n");
511 }
512 else
513 write_procformatstring_type(file, indent, "return value", get_func_return_type(func), NULL, TRUE);
514 }
515 }
516 else if (stmt->type == STMT_LIBRARY)
517 write_procformatstring_stmts(file, indent, stmt->u.lib->stmts, pred);
518 }
519 }
520
521 void write_procformatstring(FILE *file, const statement_list_t *stmts, type_pred_t pred)
522 {
523 int indent = 0;
524
525 print_file(file, indent, "static const MIDL_PROC_FORMAT_STRING __MIDL_ProcFormatString =\n");
526 print_file(file, indent, "{\n");
527 indent++;
528 print_file(file, indent, "0,\n");
529 print_file(file, indent, "{\n");
530 indent++;
531
532 write_procformatstring_stmts(file, indent, stmts, pred);
533
534 print_file(file, indent, "0x0\n");
535 indent--;
536 print_file(file, indent, "}\n");
537 indent--;
538 print_file(file, indent, "};\n");
539 print_file(file, indent, "\n");
540 }
541
542 static int write_base_type(FILE *file, const type_t *type, unsigned int *typestring_offset)
543 {
544 if (is_base_type(type->type))
545 {
546 print_file(file, 2, "0x%02x,\t/* %s */\n", type->type, string_of_type(type->type));
547 *typestring_offset += 1;
548 return 1;
549 }
550
551 return 0;
552 }
553
554 /* write conformance / variance descriptor */
555 static size_t write_conf_or_var_desc(FILE *file, const type_t *structure,
556 unsigned int baseoff, const type_t *type,
557 const expr_t *expr)
558 {
559 unsigned char operator_type = 0;
560 unsigned char conftype = RPC_FC_NORMAL_CONFORMANCE;
561 const char *conftype_string = "";
562 const char *operator_string = "no operators";
563 const expr_t *subexpr;
564
565 if (!expr)
566 {
567 print_file(file, 2, "NdrFcLong(0xffffffff),\t/* -1 */\n");
568 return 4;
569 }
570
571 if (!structure)
572 {
573 /* Top-level conformance calculations are done inline. */
574 print_file (file, 2, "0x%x,\t/* Corr desc: parameter */\n",
575 RPC_FC_TOP_LEVEL_CONFORMANCE);
576 print_file (file, 2, "0x0,\n");
577 print_file (file, 2, "NdrFcShort(0x0),\n");
578 return 4;
579 }
580
581 if (expr->is_const)
582 {
583 if (expr->cval > UCHAR_MAX * (USHRT_MAX + 1) + USHRT_MAX)
584 error("write_conf_or_var_desc: constant value %ld is greater than "
585 "the maximum constant size of %d\n", expr->cval,
586 UCHAR_MAX * (USHRT_MAX + 1) + USHRT_MAX);
587
588 print_file(file, 2, "0x%x, /* Corr desc: constant, val = %ld */\n",
589 RPC_FC_CONSTANT_CONFORMANCE, expr->cval);
590 print_file(file, 2, "0x%x,\n", expr->cval & ~USHRT_MAX);
591 print_file(file, 2, "NdrFcShort(0x%x),\n", expr->cval & USHRT_MAX);
592
593 return 4;
594 }
595
596 if (is_ptr(type) || (is_array(type) && !type->declarray))
597 {
598 conftype = RPC_FC_POINTER_CONFORMANCE;
599 conftype_string = "field pointer, ";
600 }
601
602 subexpr = expr;
603 switch (subexpr->type)
604 {
605 case EXPR_PPTR:
606 subexpr = subexpr->ref;
607 operator_type = RPC_FC_DEREFERENCE;
608 operator_string = "FC_DEREFERENCE";
609 break;
610 case EXPR_DIV:
611 if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 2))
612 {
613 subexpr = subexpr->ref;
614 operator_type = RPC_FC_DIV_2;
615 operator_string = "FC_DIV_2";
616 }
617 break;
618 case EXPR_MUL:
619 if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 2))
620 {
621 subexpr = subexpr->ref;
622 operator_type = RPC_FC_MULT_2;
623 operator_string = "FC_MULT_2";
624 }
625 break;
626 case EXPR_SUB:
627 if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 1))
628 {
629 subexpr = subexpr->ref;
630 operator_type = RPC_FC_SUB_1;
631 operator_string = "FC_SUB_1";
632 }
633 break;
634 case EXPR_ADD:
635 if (subexpr->u.ext->is_const && (subexpr->u.ext->cval == 1))
636 {
637 subexpr = subexpr->ref;
638 operator_type = RPC_FC_ADD_1;
639 operator_string = "FC_ADD_1";
640 }
641 break;
642 default:
643 break;
644 }
645
646 if (subexpr->type == EXPR_IDENTIFIER)
647 {
648 const type_t *correlation_variable = NULL;
649 unsigned char correlation_variable_type;
650 unsigned char param_type = 0;
651 size_t offset = 0;
652 const var_t *var;
653
654 if (structure->fields_or_args) LIST_FOR_EACH_ENTRY( var, structure->fields_or_args, const var_t, entry )
655 {
656 unsigned int align = 0;
657 /* FIXME: take alignment into account */
658 if (var->name && !strcmp(var->name, subexpr->u.sval))
659 {
660 correlation_variable = var->type;
661 break;
662 }
663 offset += type_memsize(var->type, &align);
664 }
665 if (!correlation_variable)
666 error("write_conf_or_var_desc: couldn't find variable %s in structure\n",
667 subexpr->u.sval);
668
669 correlation_variable = expr_resolve_type(NULL, structure, expr);
670
671 offset -= baseoff;
672 correlation_variable_type = correlation_variable->type;
673
674 switch (correlation_variable_type)
675 {
676 case RPC_FC_CHAR:
677 case RPC_FC_SMALL:
678 param_type = RPC_FC_SMALL;
679 break;
680 case RPC_FC_BYTE:
681 case RPC_FC_USMALL:
682 param_type = RPC_FC_USMALL;
683 break;
684 case RPC_FC_WCHAR:
685 case RPC_FC_SHORT:
686 case RPC_FC_ENUM16:
687 param_type = RPC_FC_SHORT;
688 break;
689 case RPC_FC_USHORT:
690 param_type = RPC_FC_USHORT;
691 break;
692 case RPC_FC_LONG:
693 case RPC_FC_ENUM32:
694 param_type = RPC_FC_LONG;
695 break;
696 case RPC_FC_ULONG:
697 param_type = RPC_FC_ULONG;
698 break;
699 default:
700 error("write_conf_or_var_desc: conformance variable type not supported 0x%x\n",
701 correlation_variable_type);
702 }
703
704 print_file(file, 2, "0x%x, /* Corr desc: %s%s */\n",
705 conftype | param_type, conftype_string, string_of_type(param_type));
706 print_file(file, 2, "0x%x, /* %s */\n", operator_type, operator_string);
707 print_file(file, 2, "NdrFcShort(0x%x), /* offset = %d */\n",
708 offset, offset);
709 }
710 else
711 {
712 unsigned int callback_offset = 0;
713 struct expr_eval_routine *eval;
714 int found = 0;
715
716 LIST_FOR_EACH_ENTRY(eval, &expr_eval_routines, struct expr_eval_routine, entry)
717 {
718 if (!strcmp (eval->structure->name, structure->name)
719 && !compare_expr (eval->expr, expr))
720 {
721 found = 1;
722 break;
723 }
724 callback_offset++;
725 }
726
727 if (!found)
728 {
729 eval = xmalloc (sizeof(*eval));
730 eval->structure = structure;
731 eval->baseoff = baseoff;
732 eval->expr = expr;
733 list_add_tail (&expr_eval_routines, &eval->entry);
734 }
735
736 if (callback_offset > USHRT_MAX)
737 error("Maximum number of callback routines reached\n");
738
739 print_file(file, 2, "0x%x, /* Corr desc: %s */\n", conftype, conftype_string);
740 print_file(file, 2, "0x%x, /* %s */\n", RPC_FC_CALLBACK, "FC_CALLBACK");
741 print_file(file, 2, "NdrFcShort(0x%x), /* %u */\n", callback_offset, callback_offset);
742 }
743 return 4;
744 }
745
746 static size_t fields_memsize(const var_list_t *fields, unsigned int *align)
747 {
748 int have_align = FALSE;
749 size_t size = 0;
750 const var_t *v;
751
752 if (!fields) return 0;
753 LIST_FOR_EACH_ENTRY( v, fields, const var_t, entry )
754 {
755 unsigned int falign = 0;
756 size_t fsize = type_memsize(v->type, &falign);
757 if (!have_align)
758 {
759 *align = falign;
760 have_align = TRUE;
761 }
762 size = ROUND_SIZE(size, falign);
763 size += fsize;
764 }
765
766 size = ROUND_SIZE(size, *align);
767 return size;
768 }
769
770 static size_t union_memsize(const var_list_t *fields, unsigned int *pmaxa)
771 {
772 size_t size, maxs = 0;
773 unsigned int align = *pmaxa;
774 const var_t *v;
775
776 if (fields) LIST_FOR_EACH_ENTRY( v, fields, const var_t, entry )
777 {
778 /* we could have an empty default field with NULL type */
779 if (v->type)
780 {
781 size = type_memsize(v->type, &align);
782 if (maxs < size) maxs = size;
783 if (*pmaxa < align) *pmaxa = align;
784 }
785 }
786
787 return maxs;
788 }
789
790 int get_padding(const var_list_t *fields)
791 {
792 unsigned short offset = 0;
793 int salign = -1;
794 const var_t *f;
795
796 if (!fields)
797 return 0;
798
799 LIST_FOR_EACH_ENTRY(f, fields, const var_t, entry)
800 {
801 type_t *ft = f->type;
802 unsigned int align = 0;
803 size_t size = type_memsize(ft, &align);
804 if (salign == -1)
805 salign = align;
806 offset = ROUND_SIZE(offset, align);
807 offset += size;
808 }
809
810 return ROUNDING(offset, salign);
811 }
812
813 size_t type_memsize(const type_t *t, unsigned int *align)
814 {
815 size_t size = 0;
816
817 if (t->kind == TKIND_ALIAS)
818 size = type_memsize(t->orig, align);
819 else if (t->declarray && is_conformant_array(t))
820 {
821 type_memsize(t->ref, align);
822 size = 0;
823 }
824 else if (is_ptr(t) || is_conformant_array(t))
825 {
826 size = sizeof(void *);
827 if (size > *align) *align = size;
828 }
829 else switch (t->type)
830 {
831 case RPC_FC_BYTE:
832 case RPC_FC_CHAR:
833 case RPC_FC_USMALL:
834 case RPC_FC_SMALL:
835 size = 1;
836 if (size > *align) *align = size;
837 break;
838 case RPC_FC_WCHAR:
839 case RPC_FC_USHORT:
840 case RPC_FC_SHORT:
841 case RPC_FC_ENUM16:
842 size = 2;
843 if (size > *align) *align = size;
844 break;
845 case RPC_FC_ULONG:
846 case RPC_FC_LONG:
847 case RPC_FC_ERROR_STATUS_T:
848 case RPC_FC_ENUM32:
849 case RPC_FC_FLOAT:
850 size = 4;
851 if (size > *align) *align = size;
852 break;
853 case RPC_FC_HYPER:
854 case RPC_FC_DOUBLE:
855 size = 8;
856 if (size > *align) *align = size;
857 break;
858 case RPC_FC_STRUCT:
859 case RPC_FC_CVSTRUCT:
860 case RPC_FC_CPSTRUCT:
861 case RPC_FC_CSTRUCT:
862 case RPC_FC_PSTRUCT:
863 case RPC_FC_BOGUS_STRUCT:
864 size = fields_memsize(t->fields_or_args, align);
865 break;
866 case RPC_FC_ENCAPSULATED_UNION:
867 case RPC_FC_NON_ENCAPSULATED_UNION:
868 size = union_memsize(t->fields_or_args, align);
869 break;
870 case RPC_FC_SMFARRAY:
871 case RPC_FC_LGFARRAY:
872 case RPC_FC_SMVARRAY:
873 case RPC_FC_LGVARRAY:
874 case RPC_FC_BOGUS_ARRAY:
875 size = t->dim * type_memsize(t->ref, align);
876 break;
877 default:
878 error("type_memsize: Unknown type %d\n", t->type);
879 size = 0;
880 }
881
882 return size;
883 }
884
885 int is_full_pointer_function(const func_t *func)
886 {
887 const var_t *var;
888 if (type_has_full_pointer(get_func_return_type(func)))
889 return TRUE;
890 if (!func->args)
891 return FALSE;
892 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
893 if (type_has_full_pointer( var->type ))
894 return TRUE;
895 return FALSE;
896 }
897
898 void write_full_pointer_init(FILE *file, int indent, const func_t *func, int is_server)
899 {
900 print_file(file, indent, "_StubMsg.FullPtrXlatTables = NdrFullPointerXlatInit(0,%s);\n",
901 is_server ? "XLAT_SERVER" : "XLAT_CLIENT");
902 fprintf(file, "\n");
903 }
904
905 void write_full_pointer_free(FILE *file, int indent, const func_t *func)
906 {
907 print_file(file, indent, "NdrFullPointerXlatFree(_StubMsg.FullPtrXlatTables);\n");
908 fprintf(file, "\n");
909 }
910
911 static unsigned int write_nonsimple_pointer(FILE *file, const type_t *type, size_t offset)
912 {
913 short absoff = type->ref->typestring_offset;
914 short reloff = absoff - (offset + 2);
915 int ptr_attr = is_ptr(type->ref) ? 0x10 : 0x0;
916
917 print_file(file, 2, "0x%02x, 0x%x,\t/* %s */\n",
918 type->type, ptr_attr, string_of_type(type->type));
919 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%hd) */\n",
920 reloff, reloff, absoff);
921 return 4;
922 }
923
924 static unsigned int write_simple_pointer(FILE *file, const type_t *type)
925 {
926 unsigned char fc = type->ref->type;
927 /* for historical reasons, write_simple_pointer also handled string types,
928 * but no longer does. catch bad uses of the function with this check */
929 if (is_string_type(type->attrs, type))
930 error("write_simple_pointer: can't handle type %s which is a string type\n", type->name);
931 print_file(file, 2, "0x%02x, 0x8,\t/* %s [simple_pointer] */\n",
932 type->type, string_of_type(type->type));
933 print_file(file, 2, "0x%02x,\t/* %s */\n", fc, string_of_type(fc));
934 print_file(file, 2, "0x5c,\t/* FC_PAD */\n");
935 return 4;
936 }
937
938 static void print_start_tfs_comment(FILE *file, type_t *t, unsigned int tfsoff)
939 {
940 print_file(file, 0, "/* %u (", tfsoff);
941 write_type_decl(file, t, NULL);
942 print_file(file, 0, ") */\n");
943 }
944
945 static size_t write_pointer_tfs(FILE *file, type_t *type, unsigned int *typestring_offset)
946 {
947 unsigned int offset = *typestring_offset;
948
949 print_start_tfs_comment(file, type, offset);
950 update_tfsoff(type, offset, file);
951
952 if (type->ref->typestring_offset)
953 *typestring_offset += write_nonsimple_pointer(file, type, offset);
954 else if (is_base_type(type->ref->type))
955 *typestring_offset += write_simple_pointer(file, type);
956
957 return offset;
958 }
959
960 static int processed(const type_t *type)
961 {
962 return type->typestring_offset && !type->tfswrite;
963 }
964
965 static int user_type_has_variable_size(const type_t *t)
966 {
967 if (is_ptr(t))
968 return TRUE;
969 else
970 switch (t->type)
971 {
972 case RPC_FC_PSTRUCT:
973 case RPC_FC_CSTRUCT:
974 case RPC_FC_CPSTRUCT:
975 case RPC_FC_CVSTRUCT:
976 return TRUE;
977 }
978 /* Note: Since this only applies to user types, we can't have a conformant
979 array here, and strings should get filed under pointer in this case. */
980 return FALSE;
981 }
982
983 static void write_user_tfs(FILE *file, type_t *type, unsigned int *tfsoff)
984 {
985 unsigned int start, absoff, flags;
986 unsigned int align = 0, ualign = 0;
987 const char *name;
988 type_t *utype = get_user_type(type, &name);
989 size_t usize = user_type_has_variable_size(utype) ? 0 : type_memsize(utype, &ualign);
990 size_t size = type_memsize(type, &align);
991 unsigned short funoff = user_type_offset(name);
992 short reloff;
993
994 guard_rec(type);
995
996 if (is_base_type(utype->type))
997 {
998 absoff = *tfsoff;
999 print_start_tfs_comment(file, utype, absoff);
1000 print_file(file, 2, "0x%x,\t/* %s */\n", utype->type, string_of_type(utype->type));
1001 print_file(file, 2, "0x5c,\t/* FC_PAD */\n");
1002 *tfsoff += 2;
1003 }
1004 else
1005 {
1006 if (!processed(utype))
1007 write_embedded_types(file, NULL, utype, utype->name, TRUE, tfsoff);
1008 absoff = utype->typestring_offset;
1009 }
1010
1011 if (utype->type == RPC_FC_RP)
1012 flags = 0x40;
1013 else if (utype->type == RPC_FC_UP)
1014 flags = 0x80;
1015 else
1016 flags = 0;
1017
1018 start = *tfsoff;
1019 update_tfsoff(type, start, file);
1020 print_start_tfs_comment(file, type, start);
1021 print_file(file, 2, "0x%x,\t/* FC_USER_MARSHAL */\n", RPC_FC_USER_MARSHAL);
1022 print_file(file, 2, "0x%x,\t/* Alignment= %d, Flags= %02x */\n",
1023 flags | (align - 1), align - 1, flags);
1024 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Function offset= %hu */\n", funoff, funoff);
1025 print_file(file, 2, "NdrFcShort(0x%lx),\t/* %lu */\n", size, size);
1026 print_file(file, 2, "NdrFcShort(0x%lx),\t/* %lu */\n", usize, usize);
1027 *tfsoff += 8;
1028 reloff = absoff - *tfsoff;
1029 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%lu) */\n", reloff, reloff, absoff);
1030 *tfsoff += 2;
1031 }
1032
1033 static void write_member_type(FILE *file, const type_t *cont,
1034 const attr_list_t *attrs, const type_t *type,
1035 unsigned int *corroff, unsigned int *tfsoff)
1036 {
1037 if (is_embedded_complex(type) && !is_conformant_array(type))
1038 {
1039 size_t absoff;
1040 short reloff;
1041
1042 if (is_union(type->type) && is_attr(attrs, ATTR_SWITCHIS))
1043 {
1044 absoff = *corroff;
1045 *corroff += 8;
1046 }
1047 else
1048 {
1049 absoff = type->typestring_offset;
1050 }
1051 reloff = absoff - (*tfsoff + 2);
1052
1053 print_file(file, 2, "0x4c,\t/* FC_EMBEDDED_COMPLEX */\n");
1054 /* FIXME: actually compute necessary padding */
1055 print_file(file, 2, "0x0,\t/* FIXME: padding */\n");
1056 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%lu) */\n",
1057 reloff, reloff, absoff);
1058 *tfsoff += 4;
1059 }
1060 else if (is_ptr(type) || is_conformant_array(type))
1061 {
1062 unsigned char fc = (cont->type == RPC_FC_BOGUS_STRUCT
1063 ? RPC_FC_POINTER
1064 : RPC_FC_LONG);
1065 print_file(file, 2, "0x%x,\t/* %s */\n", fc, string_of_type(fc));
1066 *tfsoff += 1;
1067 }
1068 else if (!write_base_type(file, type, tfsoff))
1069 error("Unsupported member type 0x%x\n", type->type);
1070 }
1071
1072 static void write_end(FILE *file, unsigned int *tfsoff)
1073 {
1074 if (*tfsoff % 2 == 0)
1075 {
1076 print_file(file, 2, "0x%x,\t\t/* FC_PAD */\n", RPC_FC_PAD);
1077 *tfsoff += 1;
1078 }
1079 print_file(file, 2, "0x%x,\t\t/* FC_END */\n", RPC_FC_END);
1080 *tfsoff += 1;
1081 }
1082
1083 static void write_descriptors(FILE *file, type_t *type, unsigned int *tfsoff)
1084 {
1085 unsigned int offset = 0;
1086 var_list_t *fs = type->fields_or_args;
1087 var_t *f;
1088
1089 if (fs) LIST_FOR_EACH_ENTRY(f, fs, var_t, entry)
1090 {
1091 unsigned int align = 0;
1092 type_t *ft = f->type;
1093 if (is_union(ft->type) && is_attr(f->attrs, ATTR_SWITCHIS))
1094 {
1095 unsigned int absoff = ft->typestring_offset;
1096 short reloff = absoff - (*tfsoff + 6);
1097 print_file(file, 0, "/* %d */\n", *tfsoff);
1098 print_file(file, 2, "0x%x,\t/* %s */\n", ft->type, string_of_type(ft->type));
1099 print_file(file, 2, "0x%x,\t/* FIXME: always FC_LONG */\n", RPC_FC_LONG);
1100 write_conf_or_var_desc(file, current_structure, offset, ft,
1101 get_attrp(f->attrs, ATTR_SWITCHIS));
1102 print_file(file, 2, "NdrFcShort(%hd),\t/* Offset= %hd (%u) */\n",
1103 reloff, reloff, absoff);
1104 *tfsoff += 8;
1105 }
1106
1107 /* FIXME: take alignment into account */
1108 offset += type_memsize(ft, &align);
1109 }
1110 }
1111
1112 static int write_no_repeat_pointer_descriptions(
1113 FILE *file, type_t *type,
1114 size_t *offset_in_memory, size_t *offset_in_buffer,
1115 unsigned int *typestring_offset)
1116 {
1117 int written = 0;
1118 unsigned int align;
1119
1120 if (is_ptr(type) || (!type->declarray && is_conformant_array(type)))
1121 {
1122 size_t memsize;
1123
1124 print_file(file, 2, "0x%02x, /* FC_NO_REPEAT */\n", RPC_FC_NO_REPEAT);
1125 print_file(file, 2, "0x%02x, /* FC_PAD */\n", RPC_FC_PAD);
1126
1127 /* pointer instance */
1128 print_file(file, 2, "NdrFcShort(0x%x), /* Memory offset = %d */\n", *offset_in_memory, *offset_in_memory);
1129 print_file(file, 2, "NdrFcShort(0x%x), /* Buffer offset = %d */\n", *offset_in_buffer, *offset_in_buffer);
1130 *typestring_offset += 6;
1131
1132 if (is_ptr(type))
1133 {
1134 if (is_string_type(type->attrs, type))
1135 write_string_tfs(file, NULL, type, NULL, typestring_offset);
1136 else
1137 write_pointer_tfs(file, type, typestring_offset);
1138 }
1139 else
1140 {
1141 unsigned absoff = type->typestring_offset;
1142 short reloff = absoff - (*typestring_offset + 2);
1143 /* FIXME: get pointer attributes from field */
1144 print_file(file, 2, "0x%02x, 0x0,\t/* %s */\n", RPC_FC_UP, "FC_UP");
1145 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n",
1146 reloff, reloff, absoff);
1147 *typestring_offset += 4;
1148 }
1149
1150 align = 0;
1151 memsize = type_memsize(type, &align);
1152 *offset_in_memory += memsize;
1153 /* increment these separately as in the case of conformant (varying)
1154 * structures these start at different values */
1155 *offset_in_buffer += memsize;
1156
1157 return 1;
1158 }
1159
1160 if (is_non_complex_struct(type))
1161 {
1162 const var_t *v;
1163 LIST_FOR_EACH_ENTRY( v, type->fields_or_args, const var_t, entry )
1164 {
1165 if (offset_in_memory && offset_in_buffer)
1166 {
1167 size_t padding;
1168 align = 0;
1169 type_memsize(v->type, &align);
1170 padding = ROUNDING(*offset_in_memory, align);
1171 *offset_in_memory += padding;
1172 *offset_in_buffer += padding;
1173 }
1174 written += write_no_repeat_pointer_descriptions(
1175 file, v->type,
1176 offset_in_memory, offset_in_buffer, typestring_offset);
1177 }
1178 }
1179 else
1180 {
1181 size_t memsize;
1182 align = 0;
1183 memsize = type_memsize(type, &align);
1184 *offset_in_memory += memsize;
1185 /* increment these separately as in the case of conformant (varying)
1186 * structures these start at different values */
1187 *offset_in_buffer += memsize;
1188 }
1189
1190 return written;
1191 }
1192
1193 static int write_pointer_description_offsets(
1194 FILE *file, const attr_list_t *attrs, type_t *type,
1195 size_t *offset_in_memory, size_t *offset_in_buffer,
1196 unsigned int *typestring_offset)
1197 {
1198 int written = 0;
1199 unsigned int align;
1200
1201 if (is_ptr(type) && type->ref->type != RPC_FC_IP)
1202 {
1203 if (offset_in_memory && offset_in_buffer)
1204 {
1205 size_t memsize;
1206
1207 /* pointer instance */
1208 /* FIXME: sometimes from end of structure, sometimes from beginning */
1209 print_file(file, 2, "NdrFcShort(0x%x), /* Memory offset = %d */\n", *offset_in_memory, *offset_in_memory);
1210 print_file(file, 2, "NdrFcShort(0x%x), /* Buffer offset = %d */\n", *offset_in_buffer, *offset_in_buffer);
1211
1212 align = 0;
1213 memsize = type_memsize(type, &align);
1214 *offset_in_memory += memsize;
1215 /* increment these separately as in the case of conformant (varying)
1216 * structures these start at different values */
1217 *offset_in_buffer += memsize;
1218 }
1219 *typestring_offset += 4;
1220
1221 if (is_string_type(attrs, type))
1222 write_string_tfs(file, NULL, type, NULL, typestring_offset);
1223 else if (processed(type->ref) || is_base_type(type->ref->type))
1224 write_pointer_tfs(file, type, typestring_offset);
1225 else
1226 error("write_pointer_description_offsets: type format string unknown\n");
1227
1228 return 1;
1229 }
1230
1231 if (is_array(type))
1232 {
1233 return write_pointer_description_offsets(
1234 file, attrs, type->ref, offset_in_memory, offset_in_buffer,
1235 typestring_offset);
1236 }
1237 else if (is_non_complex_struct(type))
1238 {
1239 /* otherwise search for interesting fields to parse */
1240 const var_t *v;
1241 LIST_FOR_EACH_ENTRY( v, type->fields_or_args, const var_t, entry )
1242 {
1243 if (offset_in_memory && offset_in_buffer)
1244 {
1245 size_t padding;
1246 align = 0;
1247 type_memsize(v->type, &align);
1248 padding = ROUNDING(*offset_in_memory, align);
1249 *offset_in_memory += padding;
1250 *offset_in_buffer += padding;
1251 }
1252 written += write_pointer_description_offsets(
1253 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
1254 typestring_offset);
1255 }
1256 }
1257 else
1258 {
1259 if (offset_in_memory && offset_in_buffer)
1260 {
1261 size_t memsize;
1262 align = 0;
1263 memsize = type_memsize(type, &align);
1264 *offset_in_memory += memsize;
1265 /* increment these separately as in the case of conformant (varying)
1266 * structures these start at different values */
1267 *offset_in_buffer += memsize;
1268 }
1269 }
1270
1271 return written;
1272 }
1273
1274 /* Note: if file is NULL return value is number of pointers to write, else
1275 * it is the number of type format characters written */
1276 static int write_fixed_array_pointer_descriptions(
1277 FILE *file, const attr_list_t *attrs, type_t *type,
1278 size_t *offset_in_memory, size_t *offset_in_buffer,
1279 unsigned int *typestring_offset)
1280 {
1281 unsigned int align;
1282 int pointer_count = 0;
1283
1284 if (type->type == RPC_FC_SMFARRAY || type->type == RPC_FC_LGFARRAY)
1285 {
1286 unsigned int temp = 0;
1287 /* unfortunately, this needs to be done in two passes to avoid
1288 * writing out redundant FC_FIXED_REPEAT descriptions */
1289 pointer_count = write_pointer_description_offsets(
1290 NULL, attrs, type->ref, NULL, NULL, &temp);
1291 if (pointer_count > 0)
1292 {
1293 unsigned int increment_size;
1294 size_t offset_of_array_pointer_mem = 0;
1295 size_t offset_of_array_pointer_buf = 0;
1296
1297 align = 0;
1298 increment_size = type_memsize(type->ref, &align);
1299
1300 print_file(file, 2, "0x%02x, /* FC_FIXED_REPEAT */\n", RPC_FC_FIXED_REPEAT);
1301 print_file(file, 2, "0x%02x, /* FC_PAD */\n", RPC_FC_PAD);
1302 print_file(file, 2, "NdrFcShort(0x%x), /* Iterations = %d */\n", type->dim, type->dim);
1303 print_file(file, 2, "NdrFcShort(0x%x), /* Increment = %d */\n", increment_size, increment_size);
1304 print_file(file, 2, "NdrFcShort(0x%x), /* Offset to array = %d */\n", *offset_in_memory, *offset_in_memory);
1305 print_file(file, 2, "NdrFcShort(0x%x), /* Number of pointers = %d */\n", pointer_count, pointer_count);
1306 *typestring_offset += 10;
1307
1308 pointer_count = write_pointer_description_offsets(
1309 file, attrs, type, &offset_of_array_pointer_mem,
1310 &offset_of_array_pointer_buf, typestring_offset);
1311 }
1312 }
1313 else if (is_struct(type->type))
1314 {
1315 const var_t *v;
1316 LIST_FOR_EACH_ENTRY( v, type->fields_or_args, const var_t, entry )
1317 {
1318 if (offset_in_memory && offset_in_buffer)
1319 {
1320 size_t padding;
1321 align = 0;
1322 type_memsize(v->type, &align);
1323 padding = ROUNDING(*offset_in_memory, align);
1324 *offset_in_memory += padding;
1325 *offset_in_buffer += padding;
1326 }
1327 pointer_count += write_fixed_array_pointer_descriptions(
1328 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
1329 typestring_offset);
1330 }
1331 }
1332 else
1333 {
1334 if (offset_in_memory && offset_in_buffer)
1335 {
1336 size_t memsize;
1337 align = 0;
1338 memsize = type_memsize(type, &align);
1339 *offset_in_memory += memsize;
1340 /* increment these separately as in the case of conformant (varying)
1341 * structures these start at different values */
1342 *offset_in_buffer += memsize;
1343 }
1344 }
1345
1346 return pointer_count;
1347 }
1348
1349 /* Note: if file is NULL return value is number of pointers to write, else
1350 * it is the number of type format characters written */
1351 static int write_conformant_array_pointer_descriptions(
1352 FILE *file, const attr_list_t *attrs, type_t *type,
1353 size_t offset_in_memory, unsigned int *typestring_offset)
1354 {
1355 unsigned int align;
1356 int pointer_count = 0;
1357
1358 if (is_conformant_array(type) && !type->length_is)
1359 {
1360 unsigned int temp = 0;
1361 /* unfortunately, this needs to be done in two passes to avoid
1362 * writing out redundant FC_VARIABLE_REPEAT descriptions */
1363 pointer_count = write_pointer_description_offsets(
1364 NULL, attrs, type->ref, NULL, NULL, &temp);
1365 if (pointer_count > 0)
1366 {
1367 unsigned int increment_size;
1368 size_t offset_of_array_pointer_mem = offset_in_memory;
1369 size_t offset_of_array_pointer_buf = offset_in_memory;
1370
1371 align = 0;
1372 increment_size = type_memsize(type->ref, &align);
1373
1374 if (increment_size > USHRT_MAX)
1375 error("array size of %u bytes is too large\n", increment_size);
1376
1377 print_file(file, 2, "0x%02x, /* FC_VARIABLE_REPEAT */\n", RPC_FC_VARIABLE_REPEAT);
1378 print_file(file, 2, "0x%02x, /* FC_FIXED_OFFSET */\n", RPC_FC_FIXED_OFFSET);
1379 print_file(file, 2, "NdrFcShort(0x%x), /* Increment = %d */\n", increment_size, increment_size);
1380 print_file(file, 2, "NdrFcShort(0x%x), /* Offset to array = %d */\n", offset_in_memory, offset_in_memory);
1381 print_file(file, 2, "NdrFcShort(0x%x), /* Number of pointers = %d */\n", pointer_count, pointer_count);
1382 *typestring_offset += 8;
1383
1384 pointer_count = write_pointer_description_offsets(
1385 file, attrs, type->ref, &offset_of_array_pointer_mem,
1386 &offset_of_array_pointer_buf, typestring_offset);
1387 }
1388 }
1389
1390 return pointer_count;
1391 }
1392
1393 /* Note: if file is NULL return value is number of pointers to write, else
1394 * it is the number of type format characters written */
1395 static int write_varying_array_pointer_descriptions(
1396 FILE *file, const attr_list_t *attrs, type_t *type,
1397 size_t *offset_in_memory, size_t *offset_in_buffer,
1398 unsigned int *typestring_offset)
1399 {
1400 unsigned int align;
1401 int pointer_count = 0;
1402
1403 if (is_array(type) && type->length_is)
1404 {
1405 unsigned int temp = 0;
1406 /* unfortunately, this needs to be done in two passes to avoid
1407 * writing out redundant FC_VARIABLE_REPEAT descriptions */
1408 pointer_count = write_pointer_description_offsets(
1409 NULL, attrs, type->ref, NULL, NULL, &temp);
1410 if (pointer_count > 0)
1411 {
1412 unsigned int increment_size;
1413
1414 align = 0;
1415 increment_size = type_memsize(type->ref, &align);
1416
1417 if (increment_size > USHRT_MAX)
1418 error("array size of %u bytes is too large\n", increment_size);
1419
1420 print_file(file, 2, "0x%02x, /* FC_VARIABLE_REPEAT */\n", RPC_FC_VARIABLE_REPEAT);
1421 print_file(file, 2, "0x%02x, /* FC_VARIABLE_OFFSET */\n", RPC_FC_VARIABLE_OFFSET);
1422 print_file(file, 2, "NdrFcShort(0x%x), /* Increment = %d */\n", increment_size, increment_size);
1423 print_file(file, 2, "NdrFcShort(0x%x), /* Offset to array = %d */\n", *offset_in_memory, *offset_in_memory);
1424 print_file(file, 2, "NdrFcShort(0x%x), /* Number of pointers = %d */\n", pointer_count, pointer_count);
1425 *typestring_offset += 8;
1426
1427 pointer_count = write_pointer_description_offsets(
1428 file, attrs, type, offset_in_memory,
1429 offset_in_buffer, typestring_offset);
1430 }
1431 }
1432 else if (is_struct(type->type))
1433 {
1434 const var_t *v;
1435 LIST_FOR_EACH_ENTRY( v, type->fields_or_args, const var_t, entry )
1436 {
1437 if (offset_in_memory && offset_in_buffer)
1438 {
1439 size_t padding;
1440 align = 0;
1441 type_memsize(v->type, &align);
1442 padding = ROUNDING(*offset_in_memory, align);
1443 *offset_in_memory += padding;
1444 *offset_in_buffer += padding;
1445 }
1446 pointer_count += write_varying_array_pointer_descriptions(
1447 file, v->attrs, v->type, offset_in_memory, offset_in_buffer,
1448 typestring_offset);
1449 }
1450 }
1451 else
1452 {
1453 if (offset_in_memory && offset_in_buffer)
1454 {
1455 size_t memsize;
1456 align = 0;
1457 memsize = type_memsize(type, &align);
1458 *offset_in_memory += memsize;
1459 /* increment these separately as in the case of conformant (varying)
1460 * structures these start at different values */
1461 *offset_in_buffer += memsize;
1462 }
1463 }
1464
1465 return pointer_count;
1466 }
1467
1468 static void write_pointer_description(FILE *file, type_t *type,
1469 unsigned int *typestring_offset)
1470 {
1471 size_t offset_in_buffer;
1472 size_t offset_in_memory;
1473 size_t conformance = 0;
1474
1475 if (type->type == RPC_FC_CVSTRUCT)
1476 conformance = 8;
1477 else if (type->type == RPC_FC_CSTRUCT || type->type == RPC_FC_CPSTRUCT)
1478 conformance = 4;
1479
1480 /* pass 1: search for single instance of a pointer (i.e. don't descend
1481 * into arrays) */
1482 if (!is_array(type))
1483 {
1484 offset_in_memory = 0;
1485 offset_in_buffer = conformance;
1486 write_no_repeat_pointer_descriptions(
1487 file, type,
1488 &offset_in_memory, &offset_in_buffer, typestring_offset);
1489 }
1490
1491 /* pass 2: search for pointers in fixed arrays */
1492 offset_in_memory = 0;
1493 offset_in_buffer = conformance;
1494 write_fixed_array_pointer_descriptions(
1495 file, NULL, type,
1496 &offset_in_memory, &offset_in_buffer, typestring_offset);
1497
1498 /* pass 3: search for pointers in conformant only arrays (but don't descend
1499 * into conformant varying or varying arrays) */
1500 if ((!type->declarray || !current_structure) && is_conformant_array(type))
1501 write_conformant_array_pointer_descriptions(
1502 file, NULL, type, 0, typestring_offset);
1503 else if (type->type == RPC_FC_CPSTRUCT)
1504 {
1505 unsigned int align = 0;
1506 type_t *carray = find_array_or_string_in_struct(type)->type;
1507 write_conformant_array_pointer_descriptions(
1508 file, NULL, carray,
1509 type_memsize(type, &align),
1510 typestring_offset);
1511 }
1512
1513 /* pass 4: search for pointers in varying arrays */
1514 offset_in_memory = 0;
1515 offset_in_buffer = conformance;
1516 write_varying_array_pointer_descriptions(
1517 file, NULL, type,
1518 &offset_in_memory, &offset_in_buffer, typestring_offset);
1519 }
1520
1521 int is_declptr(const type_t *t)
1522 {
1523 return is_ptr(t) || (is_conformant_array(t) && !t->declarray);
1524 }
1525
1526 static size_t write_string_tfs(FILE *file, const attr_list_t *attrs,
1527 type_t *type,
1528 const char *name, unsigned int *typestring_offset)
1529 {
1530 size_t start_offset;
1531 unsigned char rtype;
1532
1533 if (is_declptr(type))
1534 {
1535 unsigned char flag = is_conformant_array(type) ? 0 : RPC_FC_P_SIMPLEPOINTER;
1536 int pointer_type = is_ptr(type) ? type->type : get_attrv(attrs, ATTR_POINTERTYPE);
1537 if (!pointer_type)
1538 pointer_type = RPC_FC_RP;
1539 print_start_tfs_comment(file, type, *typestring_offset);
1540 print_file(file, 2,"0x%x, 0x%x,\t/* %s%s */\n",
1541 pointer_type, flag, string_of_type(pointer_type),
1542 flag ? " [simple_pointer]" : "");
1543 *typestring_offset += 2;
1544 if (!flag)
1545 {
1546 print_file(file, 2, "NdrFcShort(0x2),\n");
1547 *typestring_offset += 2;
1548 }
1549 }
1550
1551 start_offset = *typestring_offset;
1552 update_tfsoff(type, start_offset, file);
1553
1554 rtype = type->ref->type;
1555
1556 if ((rtype != RPC_FC_BYTE) && (rtype != RPC_FC_CHAR) && (rtype != RPC_FC_WCHAR))
1557 {
1558 error("write_string_tfs: Unimplemented for type 0x%x of name: %s\n", rtype, name);
1559 return start_offset;
1560 }
1561
1562 if (type->declarray && !is_conformant_array(type))
1563 {
1564 /* FIXME: multi-dimensional array */
1565 if (0xffffuL < type->dim)
1566 error("array size for parameter %s exceeds %u bytes by %lu bytes\n",
1567 name, 0xffffu, type->dim - 0xffffu);
1568
1569 if (rtype == RPC_FC_CHAR)
1570 WRITE_FCTYPE(file, FC_CSTRING, *typestring_offset);
1571 else
1572 WRITE_FCTYPE(file, FC_WSTRING, *typestring_offset);
1573 print_file(file, 2, "0x%x, /* FC_PAD */\n", RPC_FC_PAD);
1574 *typestring_offset += 2;
1575
1576 print_file(file, 2, "NdrFcShort(0x%x), /* %d */\n", type->dim, type->dim);
1577 *typestring_offset += 2;
1578
1579 return start_offset;
1580 }
1581 else if (type->size_is)
1582 {
1583 unsigned int align = 0;
1584
1585 if (rtype == RPC_FC_CHAR)
1586 WRITE_FCTYPE(file, FC_C_CSTRING, *typestring_offset);
1587 else
1588 WRITE_FCTYPE(file, FC_C_WSTRING, *typestring_offset);
1589 print_file(file, 2, "0x%x, /* FC_STRING_SIZED */\n", RPC_FC_STRING_SIZED);
1590 *typestring_offset += 2;
1591
1592 *typestring_offset += write_conf_or_var_desc(
1593 file, current_structure,
1594 (type->declarray && current_structure
1595 ? type_memsize(current_structure, &align)
1596 : 0),
1597 type, type->size_is);
1598
1599 return start_offset;
1600 }
1601 else
1602 {
1603 if (rtype == RPC_FC_WCHAR)
1604 WRITE_FCTYPE(file, FC_C_WSTRING, *typestring_offset);
1605 else
1606 WRITE_FCTYPE(file, FC_C_CSTRING, *typestring_offset);
1607 print_file(file, 2, "0x%x, /* FC_PAD */\n", RPC_FC_PAD);
1608 *typestring_offset += 2;
1609
1610 return start_offset;
1611 }
1612 }
1613
1614 static size_t write_array_tfs(FILE *file, const attr_list_t *attrs, type_t *type,
1615 const char *name, unsigned int *typestring_offset)
1616 {
1617 const expr_t *length_is = type->length_is;
1618 const expr_t *size_is = type->size_is;
1619 unsigned int align = 0;
1620 size_t size;
1621 size_t start_offset;
1622 int has_pointer;
1623 int pointer_type = get_attrv(attrs, ATTR_POINTERTYPE);
1624 unsigned int baseoff
1625 = type->declarray && current_structure
1626 ? type_memsize(current_structure, &align)
1627 : 0;
1628
1629 if (!pointer_type)
1630 pointer_type = RPC_FC_RP;
1631
1632 if (write_embedded_types(file, attrs, type->ref, name, FALSE, typestring_offset))
1633 has_pointer = TRUE;
1634 else
1635 has_pointer = type_has_pointers(type->ref);
1636
1637 align = 0;
1638 size = type_memsize((is_conformant_array(type) ? type->ref : type), &align);
1639
1640 start_offset = *typestring_offset;
1641 update_tfsoff(type, start_offset, file);
1642 print_start_tfs_comment(file, type, start_offset);
1643 print_file(file, 2, "0x%02x,\t/* %s */\n", type->type, string_of_type(type->type));
1644 print_file(file, 2, "0x%x,\t/* %d */\n", align - 1, align - 1);
1645 *typestring_offset += 2;
1646
1647 align = 0;
1648 if (type->type != RPC_FC_BOGUS_ARRAY)
1649 {
1650 unsigned char tc = type->type;
1651
1652 if (tc == RPC_FC_LGFARRAY || tc == RPC_FC_LGVARRAY)
1653 {
1654 print_file(file, 2, "NdrFcLong(0x%x),\t/* %lu */\n", size, size);
1655 *typestring_offset += 4;
1656 }
1657 else
1658 {
1659 print_file(file, 2, "NdrFcShort(0x%x),\t/* %lu */\n", size, size);
1660 *typestring_offset += 2;
1661 }
1662
1663 if (is_conformant_array(type))
1664 *typestring_offset
1665 += write_conf_or_var_desc(file, current_structure, baseoff,
1666 type, size_is);
1667
1668 if (type->type == RPC_FC_SMVARRAY || type->type == RPC_FC_LGVARRAY)
1669 {
1670 unsigned int elalign = 0;
1671 size_t elsize = type_memsize(type->ref, &elalign);
1672
1673 if (type->type == RPC_FC_LGVARRAY)
1674 {
1675 print_file(file, 2, "NdrFcLong(0x%x),\t/* %lu */\n", type->dim, type->dim);
1676 *typestring_offset += 4;
1677 }
1678 else
1679 {
1680 print_file(file, 2, "NdrFcShort(0x%x),\t/* %lu */\n", type->dim, type->dim);
1681 *typestring_offset += 2;
1682 }
1683
1684 print_file(file, 2, "NdrFcShort(0x%x),\t/* %lu */\n", elsize, elsize);
1685 *typestring_offset += 2;
1686 }
1687
1688 if (length_is)
1689 *typestring_offset
1690 += write_conf_or_var_desc(file, current_structure, baseoff,
1691 type, length_is);
1692
1693 if (has_pointer && (!type->declarray || !current_structure))
1694 {
1695 print_file(file, 2, "0x%x, /* FC_PP */\n", RPC_FC_PP);
1696 print_file(file, 2, "0x%x, /* FC_PAD */\n", RPC_FC_PAD);
1697 *typestring_offset += 2;
1698 write_pointer_description(file, type, typestring_offset);
1699 print_file(file, 2, "0x%x, /* FC_END */\n", RPC_FC_END);
1700 *typestring_offset += 1;
1701 }
1702
1703 write_member_type(file, type, NULL, type->ref, NULL, typestring_offset);
1704 write_end(file, typestring_offset);
1705 }
1706 else
1707 {
1708 unsigned int dim = size_is ? 0 : type->dim;
1709 print_file(file, 2, "NdrFcShort(0x%x),\t/* %u */\n", dim, dim);
1710 *typestring_offset += 2;
1711 *typestring_offset
1712 += write_conf_or_var_desc(file, current_structure, baseoff,
1713 type, size_is);
1714 *typestring_offset
1715 += write_conf_or_var_desc(file, current_structure, baseoff,
1716 type, length_is);
1717 write_member_type(file, type, NULL, type->ref, NULL, typestring_offset);
1718 write_end(file, typestring_offset);
1719 }
1720
1721 return start_offset;
1722 }
1723
1724 static const var_t *find_array_or_string_in_struct(const type_t *type)
1725 {
1726 const var_t *last_field;
1727 const type_t *ft;
1728
1729 if (!type->fields_or_args || list_empty(type->fields_or_args))
1730 return NULL;
1731
1732 last_field = LIST_ENTRY( list_tail(type->fields_or_args), const var_t, entry );
1733 ft = last_field->type;
1734
1735 if (ft->declarray && is_conformant_array(ft))
1736 return last_field;
1737
1738 if (ft->type == RPC_FC_CSTRUCT || ft->type == RPC_FC_CPSTRUCT || ft->type == RPC_FC_CVSTRUCT)
1739 return find_array_or_string_in_struct(ft);
1740 else
1741 return NULL;
1742 }
1743
1744 static void write_struct_members(FILE *file, const type_t *type,
1745 unsigned int *corroff, unsigned int *typestring_offset)
1746 {
1747 const var_t *field;
1748 unsigned short offset = 0;
1749 int salign = -1;
1750 int padding;
1751
1752 if (type->fields_or_args) LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
1753 {
1754 type_t *ft = field->type;
1755 if (!ft->declarray || !is_conformant_array(ft))
1756 {
1757 unsigned int align = 0;
1758 size_t size = type_memsize(ft, &align);
1759 if (salign == -1)
1760 salign = align;
1761 if ((align - 1) & offset)
1762 {
1763 unsigned char fc = 0;
1764 switch (align)
1765 {
1766 case 4:
1767 fc = RPC_FC_ALIGNM4;
1768 break;
1769 case 8:
1770 fc = RPC_FC_ALIGNM8;
1771 break;
1772 default:
1773 error("write_struct_members: cannot align type %d\n", ft->type);
1774 }
1775 print_file(file, 2, "0x%x,\t/* %s */\n", fc, string_of_type(fc));
1776 offset = ROUND_SIZE(offset, align);
1777 *typestring_offset += 1;
1778 }
1779 write_member_type(file, type, field->attrs, field->type, corroff,
1780 typestring_offset);
1781 offset += size;
1782 }
1783 }
1784
1785 padding = ROUNDING(offset, salign);
1786 if (padding)
1787 {
1788 print_file(file, 2, "0x%x,\t/* FC_STRUCTPAD%d */\n",
1789 RPC_FC_STRUCTPAD1 + padding - 1,
1790 padding);
1791 *typestring_offset += 1;
1792 }
1793
1794 write_end(file, typestring_offset);
1795 }
1796
1797 static size_t write_struct_tfs(FILE *file, type_t *type,
1798 const char *name, unsigned int *tfsoff)
1799 {
1800 const type_t *save_current_structure = current_structure;
1801 unsigned int total_size;
1802 const var_t *array;
1803 size_t start_offset;
1804 size_t array_offset;
1805 int has_pointers = 0;
1806 unsigned int align = 0;
1807 unsigned int corroff;
1808 var_t *f;
1809
1810 guard_rec(type);
1811 current_structure = type;
1812
1813 total_size = type_memsize(type, &align);
1814 if (total_size > USHRT_MAX)
1815 error("structure size for %s exceeds %d bytes by %d bytes\n",
1816 name, USHRT_MAX, total_size - USHRT_MAX);
1817
1818 if (type->fields_or_args) LIST_FOR_EACH_ENTRY(f, type->fields_or_args, var_t, entry)
1819 has_pointers |= write_embedded_types(file, f->attrs, f->type, f->name,
1820 FALSE, tfsoff);
1821 if (!has_pointers) has_pointers = type_has_pointers(type);
1822
1823 array = find_array_or_string_in_struct(type);
1824 if (array && !processed(array->type))
1825 array_offset
1826 = is_attr(array->attrs, ATTR_STRING)
1827 ? write_string_tfs(file, array->attrs, array->type, array->name, tfsoff)
1828 : write_array_tfs(file, array->attrs, array->type, array->name, tfsoff);
1829
1830 corroff = *tfsoff;
1831 write_descriptors(file, type, tfsoff);
1832
1833 start_offset = *tfsoff;
1834 update_tfsoff(type, start_offset, file);
1835 print_start_tfs_comment(file, type, start_offset);
1836 print_file(file, 2, "0x%x,\t/* %s */\n", type->type, string_of_type(type->type));
1837 print_file(file, 2, "0x%x,\t/* %d */\n", align - 1, align - 1);
1838 print_file(file, 2, "NdrFcShort(0x%x),\t/* %d */\n", total_size, total_size);
1839 *tfsoff += 4;
1840
1841 if (array)
1842 {
1843 unsigned int absoff = array->type->typestring_offset;
1844 short reloff = absoff - *tfsoff;
1845 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%lu) */\n",
1846 reloff, reloff, absoff);
1847 *tfsoff += 2;
1848 }
1849 else if (type->type == RPC_FC_BOGUS_STRUCT)
1850 {
1851 print_file(file, 2, "NdrFcShort(0x0),\n");
1852 *tfsoff += 2;
1853 }
1854
1855 if (type->type == RPC_FC_BOGUS_STRUCT)
1856 {
1857 /* On the sizing pass, type->ptrdesc may be zero, but it's ok as
1858 nothing is written to file yet. On the actual writing pass,
1859 this will have been updated. */
1860 unsigned int absoff = type->ptrdesc ? type->ptrdesc : *tfsoff;
1861 short reloff = absoff - *tfsoff;
1862 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n",
1863 reloff, reloff, absoff);
1864 *tfsoff += 2;
1865 }
1866 else if ((type->type == RPC_FC_PSTRUCT) ||
1867 (type->type == RPC_FC_CPSTRUCT) ||
1868 (type->type == RPC_FC_CVSTRUCT && has_pointers))
1869 {
1870 print_file(file, 2, "0x%x, /* FC_PP */\n", RPC_FC_PP);
1871 print_file(file, 2, "0x%x, /* FC_PAD */\n", RPC_FC_PAD);
1872 *tfsoff += 2;
1873 write_pointer_description(file, type, tfsoff);
1874 print_file(file, 2, "0x%x, /* FC_END */\n", RPC_FC_END);
1875 *tfsoff += 1;
1876 }
1877
1878 write_struct_members(file, type, &corroff, tfsoff);
1879
1880 if (type->type == RPC_FC_BOGUS_STRUCT)
1881 {
1882 const var_list_t *fs = type->fields_or_args;
1883 const var_t *f;
1884
1885 type->ptrdesc = *tfsoff;
1886 if (fs) LIST_FOR_EACH_ENTRY(f, fs, const var_t, entry)
1887 {
1888 type_t *ft = f->type;
1889 if (is_ptr(ft))
1890 {
1891 if (is_string_type(f->attrs, ft))
1892 write_string_tfs(file, f->attrs, ft, f->name, tfsoff);
1893 else
1894 write_pointer_tfs(file, ft, tfsoff);
1895 }
1896 else if (!ft->declarray && is_conformant_array(ft))
1897 {
1898 unsigned int absoff = ft->typestring_offset;
1899 short reloff = absoff - (*tfsoff + 2);
1900 int ptr_type = get_attrv(f->attrs, ATTR_POINTERTYPE);
1901 /* FIXME: We need to store pointer attributes for arrays
1902 so we don't lose pointer_default info. */
1903 if (ptr_type == 0)
1904 ptr_type = RPC_FC_UP;
1905 print_file(file, 0, "/* %d */\n", *tfsoff);
1906 print_file(file, 2, "0x%x, 0x0,\t/* %s */\n", ptr_type,
1907 string_of_type(ptr_type));
1908 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n",
1909 reloff, reloff, absoff);
1910 *tfsoff += 4;
1911 }
1912 }
1913 if (type->ptrdesc == *tfsoff)
1914 type->ptrdesc = 0;
1915 }
1916
1917 current_structure = save_current_structure;
1918 return start_offset;
1919 }
1920
1921 static size_t write_pointer_only_tfs(FILE *file, const attr_list_t *attrs, int pointer_type,
1922 unsigned char flags, size_t offset,
1923 unsigned int *typeformat_offset)
1924 {
1925 size_t start_offset = *typeformat_offset;
1926 short reloff = offset - (*typeformat_offset + 2);
1927 int in_attr, out_attr;
1928 in_attr = is_attr(attrs, ATTR_IN);
1929 out_attr = is_attr(attrs, ATTR_OUT);
1930 if (!in_attr && !out_attr) in_attr = 1;
1931
1932 if (out_attr && !in_attr && pointer_type == RPC_FC_RP)
1933 flags |= 0x04;
1934
1935 print_file(file, 2, "0x%x, 0x%x,\t\t/* %s",
1936 pointer_type,
1937 flags,
1938 string_of_type(pointer_type));
1939 if (file)
1940 {
1941 if (flags & 0x04)
1942 fprintf(file, " [allocated_on_stack]");
1943 if (flags & 0x10)
1944 fprintf(file, " [pointer_deref]");
1945 fprintf(file, " */\n");
1946 }
1947
1948 print_file(file, 2, "NdrFcShort(0x%x),\t/* %d */\n", reloff, offset);
1949 *typeformat_offset += 4;
1950
1951 return start_offset;
1952 }
1953
1954 static void write_branch_type(FILE *file, const type_t *t, unsigned int *tfsoff)
1955 {
1956 if (t == NULL)
1957 {
1958 print_file(file, 2, "NdrFcShort(0x0),\t/* No type */\n");
1959 }
1960 else if (is_base_type(t->type))
1961 {
1962 print_file(file, 2, "NdrFcShort(0x80%02x),\t/* Simple arm type: %s */\n",
1963 t->type, string_of_type(t->type));
1964 }
1965 else if (t->typestring_offset)
1966 {
1967 short reloff = t->typestring_offset - *tfsoff;
1968 print_file(file, 2, "NdrFcShort(0x%x),\t/* Offset= %d (%d) */\n",
1969 reloff, reloff, t->typestring_offset);
1970 }
1971 else
1972 error("write_branch_type: type unimplemented (0x%x)\n", t->type);
1973
1974 *tfsoff += 2;
1975 }
1976
1977 static size_t write_union_tfs(FILE *file, type_t *type, unsigned int *tfsoff)
1978 {
1979 unsigned int align = 0;
1980 unsigned int start_offset;
1981 size_t size = type_memsize(type, &align);
1982 var_list_t *fields;
1983 size_t nbranch = 0;
1984 type_t *deftype = NULL;
1985 short nodeftype = 0xffff;
1986 var_t *f;
1987
1988 guard_rec(type);
1989
1990 if (type->type == RPC_FC_ENCAPSULATED_UNION)
1991 {
1992 const var_t *uv = LIST_ENTRY(list_tail(type->fields_or_args), const var_t, entry);
1993 fields = uv->type->fields_or_args;
1994 }
1995 else
1996 fields = type->fields_or_args;
1997
1998 if (fields) LIST_FOR_EACH_ENTRY(f, fields, var_t, entry)
1999 {
2000 expr_list_t *cases = get_attrp(f->attrs, ATTR_CASE);
2001 if (cases)
2002 nbranch += list_count(cases);
2003 if (f->type)
2004 write_embedded_types(file, f->attrs, f->type, f->name, TRUE, tfsoff);
2005 }
2006
2007 start_offset = *tfsoff;
2008 update_tfsoff(type, start_offset, file);
2009 print_start_tfs_comment(file, type, start_offset);
2010 if (type->type == RPC_FC_ENCAPSULATED_UNION)
2011 {
2012 const var_t *sv = LIST_ENTRY(list_head(type->fields_or_args), const var_t, entry);
2013 const type_t *st = sv->type;
2014
2015 switch (st->type)
2016 {
2017 case RPC_FC_CHAR:
2018 case RPC_FC_SMALL:
2019 case RPC_FC_USMALL:
2020 case RPC_FC_SHORT:
2021 case RPC_FC_USHORT:
2022 case RPC_FC_LONG:
2023 case RPC_FC_ULONG:
2024 case RPC_FC_ENUM16:
2025 case RPC_FC_ENUM32:
2026 print_file(file, 2, "0x%x,\t/* %s */\n", type->type, string_of_type(type->type));
2027 print_file(file, 2, "0x%x,\t/* Switch type= %s */\n",
2028 0x40 | st->type, string_of_type(st->type));
2029 *tfsoff += 2;
2030 break;
2031 default:
2032 error("union switch type must be an integer, char, or enum\n");
2033 }
2034 }
2035 print_file(file, 2, "NdrFcShort(0x%x),\t/* %d */\n", size, size);
2036 print_file(file, 2, "NdrFcShort(0x%x),\t/* %d */\n", nbranch, nbranch);
2037 *tfsoff += 4;
2038
2039 if (fields) LIST_FOR_EACH_ENTRY(f, fields, var_t, entry)
2040 {
2041 type_t *ft = f->type;
2042 expr_list_t *cases = get_attrp(f->attrs, ATTR_CASE);
2043 int deflt = is_attr(f->attrs, ATTR_DEFAULT);
2044 expr_t *c;
2045
2046 if (cases == NULL && !deflt)
2047 error("union field %s with neither case nor default attribute\n", f->name);
2048
2049 if (cases) LIST_FOR_EACH_ENTRY(c, cases, expr_t, entry)
2050 {
2051 /* MIDL doesn't check for duplicate cases, even though that seems
2052 like a reasonable thing to do, it just dumps them to the TFS
2053 like we're going to do here. */
2054 print_file(file, 2, "NdrFcLong(0x%x),\t/* %d */\n", c->cval, c->cval);
2055 *tfsoff += 4;
2056 write_branch_type(file, ft, tfsoff);
2057 }
2058
2059 /* MIDL allows multiple default branches, even though that seems
2060 illogical, it just chooses the last one, which is what we will
2061 do. */
2062 if (deflt)
2063 {
2064 deftype = ft;
2065 nodeftype = 0;
2066 }
2067 }
2068
2069 if (deftype)
2070 {
2071 write_branch_type(file, deftype, tfsoff);
2072 }
2073 else
2074 {
2075 print_file(file, 2, "NdrFcShort(0x%x),\n", nodeftype);
2076 *tfsoff += 2;
2077 }
2078
2079 return start_offset;
2080 }
2081
2082 static size_t write_ip_tfs(FILE *file, const attr_list_t *attrs, type_t *type,
2083 unsigned int *typeformat_offset)
2084 {
2085 size_t i;
2086 size_t start_offset = *typeformat_offset;
2087 expr_t *iid = get_attrp(attrs, ATTR_IIDIS);
2088
2089 if (iid)
2090 {
2091 print_file(file, 2, "0x2f, /* FC_IP */\n");
2092 print_file(file, 2, "0x5c, /* FC_PAD */\n");
2093 *typeformat_offset
2094 += write_conf_or_var_desc(file, NULL, 0, type, iid) + 2;
2095 }
2096 else
2097 {
2098 const type_t *base = is_ptr(type) ? type->ref : type;
2099 const UUID *uuid = get_attrp(base->attrs, ATTR_UUID);
2100
2101 if (! uuid)
2102 error("%s: interface %s missing UUID\n", __FUNCTION__, base->name);
2103
2104 update_tfsoff(type, start_offset, file);
2105 print_start_tfs_comment(file, type, start_offset);
2106 print_file(file, 2, "0x2f,\t/* FC_IP */\n");
2107 print_file(file, 2, "0x5a,\t/* FC_CONSTANT_IID */\n");
2108 print_file(file, 2, "NdrFcLong(0x%08lx),\n", uuid->Data1);
2109 print_file(file, 2, "NdrFcShort(0x%04x),\n", uuid->Data2);
2110 print_file(file, 2, "NdrFcShort(0x%04x),\n", uuid->Data3);
2111 for (i = 0; i < 8; ++i)
2112 print_file(file, 2, "0x%02x,\n", uuid->Data4[i]);
2113
2114 if (file)
2115 fprintf(file, "\n");
2116
2117 *typeformat_offset += 18;
2118 }
2119 return start_offset;
2120 }
2121
2122 static size_t write_contexthandle_tfs(FILE *file, const type_t *type,
2123 const var_t *var,
2124 unsigned int *typeformat_offset)
2125 {
2126 size_t start_offset = *typeformat_offset;
2127 unsigned char flags = 0;
2128
2129 if (is_attr(current_iface->attrs, ATTR_STRICTCONTEXTHANDLE))
2130 flags |= NDR_STRICT_CONTEXT_HANDLE;
2131
2132 if (is_ptr(type))
2133 flags |= 0x80;
2134 if (is_attr(var->attrs, ATTR_IN))
2135 {
2136 flags |= 0x40;
2137 if (!is_attr(var->attrs, ATTR_OUT))
2138 flags |= NDR_CONTEXT_HANDLE_CANNOT_BE_NULL;
2139 }
2140 if (is_attr(var->attrs, ATTR_OUT))
2141 flags |= 0x20;
2142
2143 WRITE_FCTYPE(file, FC_BIND_CONTEXT, *typeformat_offset);
2144 print_file(file, 2, "0x%x,\t/* Context flags: ", flags);
2145 /* return and can't be null values overlap */
2146 if (((flags & 0x21) != 0x21) && (flags & NDR_CONTEXT_HANDLE_CANNOT_BE_NULL))
2147 print_file(file, 0, "can't be null, ");
2148 if (flags & NDR_CONTEXT_HANDLE_SERIALIZE)
2149 print_file(file, 0, "serialize, ");
2150 if (flags & NDR_CONTEXT_HANDLE_NO_SERIALIZE)
2151 print_file(file, 0, "no serialize, ");
2152 if (flags & NDR_STRICT_CONTEXT_HANDLE)
2153 print_file(file, 0, "strict, ");
2154 if ((flags & 0x21) == 0x20)
2155 print_file(file, 0, "out, ");
2156 if ((flags & 0x21) == 0x21)
2157 print_file(file, 0, "return, ");
2158 if (flags & 0x40)
2159 print_file(file, 0, "in, ");
2160 if (flags & 0x80)
2161 print_file(file, 0, "via ptr, ");
2162 print_file(file, 0, "*/\n");
2163 print_file(file, 2, "0, /* FIXME: rundown routine index*/\n");
2164 print_file(file, 2, "0, /* FIXME: param num */\n");
2165 *typeformat_offset += 4;
2166
2167 return start_offset;
2168 }
2169
2170 static size_t write_typeformatstring_var(FILE *file, int indent, const func_t *func,
2171 type_t *type, const var_t *var,
2172 unsigned int *typeformat_offset)
2173 {
2174 size_t offset;
2175
2176 if (is_context_handle(type))
2177 return write_contexthandle_tfs(file, type, var, typeformat_offset);
2178
2179 if (is_user_type(type))
2180 {
2181 write_user_tfs(file, type, typeformat_offset);
2182 return type->typestring_offset;
2183 }
2184
2185 if (is_string_type(var->attrs, type))
2186 return write_string_tfs(file, var->attrs, type, var->name, typeformat_offset);
2187
2188 if (is_array(type))
2189 {
2190 int ptr_type;
2191 size_t off;
2192 off = write_array_tfs(file, var->attrs, type, var->name, typeformat_offset);
2193 ptr_type = get_attrv(var->attrs, ATTR_POINTERTYPE);
2194 /* Top level pointers to conformant arrays may be handled specially
2195 since we can bypass the pointer, but if the array is buried
2196 beneath another pointer (e.g., "[size_is(,n)] int **p" then we
2197 always need to write the pointer. */
2198 if (!ptr_type && var->type != type)
2199 /* FIXME: This should use pointer_default, but the information
2200 isn't kept around for arrays. */
2201 ptr_type = RPC_FC_UP;
2202 if (ptr_type && ptr_type != RPC_FC_RP)
2203 {
2204 unsigned int absoff = type->typestring_offset;
2205 short reloff = absoff - (*typeformat_offset + 2);
2206 off = *typeformat_offset;
2207 print_file(file, 0, "/* %d */\n", off);
2208 print_file(file, 2, "0x%x, 0x0,\t/* %s */\n", ptr_type,
2209 string_of_type(ptr_type));
2210 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n",
2211 reloff, reloff, absoff);
2212 *typeformat_offset += 4;
2213 }
2214 return off;
2215 }
2216
2217 if (!is_ptr(type))
2218 {
2219 /* basic types don't need a type format string */
2220 if (is_base_type(type->type))
2221 return 0;
2222
2223 switch (type->type)
2224 {
2225 case RPC_FC_STRUCT:
2226 case RPC_FC_PSTRUCT:
2227 case RPC_FC_CSTRUCT:
2228 case RPC_FC_CPSTRUCT:
2229 case RPC_FC_CVSTRUCT:
2230 case RPC_FC_BOGUS_STRUCT:
2231 return write_struct_tfs(file, type, var->name, typeformat_offset);
2232 case RPC_FC_ENCAPSULATED_UNION:
2233 case RPC_FC_NON_ENCAPSULATED_UNION:
2234 return write_union_tfs(file, type, typeformat_offset);
2235 case RPC_FC_IGNORE:
2236 case RPC_FC_BIND_PRIMITIVE:
2237 /* nothing to do */
2238 return 0;
2239 default:
2240 error("write_typeformatstring_var: Unsupported type 0x%x for variable %s\n", type->type, var->name);
2241 }
2242 }
2243 else if (last_ptr(type))
2244 {
2245 size_t start_offset = *typeformat_offset;
2246 int in_attr = is_attr(var->attrs, ATTR_IN);
2247 int out_attr = is_attr(var->attrs, ATTR_OUT);
2248 const type_t *base = type->ref;
2249
2250 if (base->type == RPC_FC_IP
2251 || (base->type == 0
2252 && is_attr(var->attrs, ATTR_IIDIS)))
2253 {
2254 return write_ip_tfs(file, var->attrs, type, typeformat_offset);
2255 }
2256
2257 /* special case for pointers to base types */
2258 if (is_base_type(base->type))
2259 {
2260 print_file(file, indent, "0x%x, 0x%x, /* %s %s[simple_pointer] */\n",
2261 type->type, (!in_attr && out_attr) ? 0x0C : 0x08,
2262 string_of_type(type->type),
2263 (!in_attr && out_attr) ? "[allocated_on_stack] " : "");
2264 print_file(file, indent, "0x%02x, /* %s */\n", base->type, string_of_type(base->type));
2265 print_file(file, indent, "0x5c, /* FC_PAD */\n");
2266 *typeformat_offset += 4;
2267 return start_offset;
2268 }
2269 }
2270
2271 assert(is_ptr(type));
2272
2273 offset = write_typeformatstring_var(file, indent, func, type->ref, var, typeformat_offset);
2274 if (file)
2275 fprintf(file, "/* %2u */\n", *typeformat_offset);
2276 return write_pointer_only_tfs(file, var->attrs, type->type,
2277 !last_ptr(type) ? 0x10 : 0,
2278 offset, typeformat_offset);
2279 }
2280
2281 static int write_embedded_types(FILE *file, const attr_list_t *attrs, type_t *type,
2282 const char *name, int write_ptr, unsigned int *tfsoff)
2283 {
2284 int retmask = 0;
2285
2286 if (is_user_type(type))
2287 {
2288 write_user_tfs(file, type, tfsoff);
2289 }
2290 else if (is_string_type(attrs, type))
2291 {
2292 write_string_tfs(file, attrs, type, name, tfsoff);
2293 }
2294 else if (is_ptr(type))
2295 {
2296 type_t *ref = type->ref;
2297
2298 if (ref->type == RPC_FC_IP
2299 || (ref->type == 0
2300 && is_attr(attrs, ATTR_IIDIS)))
2301 {
2302 write_ip_tfs(file, attrs, type, tfsoff);
2303 }
2304 else
2305 {
2306 if (!processed(ref) && !is_base_type(ref->type))
2307 retmask |= write_embedded_types(file, NULL, ref, name, TRUE, tfsoff);
2308
2309 if (write_ptr)
2310 write_pointer_tfs(file, type, tfsoff);
2311
2312 retmask |= 1;
2313 }
2314 }
2315 else if (type->declarray && is_conformant_array(type))
2316 ; /* conformant arrays and strings are handled specially */
2317 else if (is_array(type))
2318 {
2319 write_array_tfs(file, attrs, type, name, tfsoff);
2320 if (is_conformant_array(type))
2321 retmask |= 1;
2322 }
2323 else if (is_struct(type->type))
2324 {
2325 if (!processed(type))
2326 write_struct_tfs(file, type, name, tfsoff);
2327 }
2328 else if (is_union(type->type))
2329 {
2330 if (!processed(type))
2331 write_union_tfs(file, type, tfsoff);
2332 }
2333 else if (!is_base_type(type->type))
2334 error("write_embedded_types: unknown embedded type for %s (0x%x)\n",
2335 name, type->type);
2336
2337 return retmask;
2338 }
2339
2340 static size_t process_tfs_stmts(FILE *file, const statement_list_t *stmts,
2341 type_pred_t pred, unsigned int *typeformat_offset)
2342 {
2343 const var_t *var;
2344 const statement_t *stmt;
2345
2346 if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
2347 {
2348 const type_t *iface;
2349 if (stmt->type == STMT_LIBRARY)
2350 {
2351 process_tfs_stmts(file, stmt->u.lib->stmts, pred, typeformat_offset);
2352 continue;
2353 }
2354 else if (stmt->type != STMT_TYPE || stmt->u.type->type != RPC_FC_IP)
2355 continue;
2356
2357 iface = stmt->u.type;
2358 if (!pred(iface))
2359 continue;
2360
2361 if (iface->funcs)
2362 {
2363 const func_t *func;
2364 current_iface = iface;
2365 LIST_FOR_EACH_ENTRY( func, iface->funcs, const func_t, entry )
2366 {
2367 if (is_local(func->def->attrs)) continue;
2368
2369 if (!is_void(get_func_return_type(func)))
2370 {
2371 var_t v = *func->def;
2372 v.type = get_func_return_type(func);
2373 update_tfsoff(get_func_return_type(func),
2374 write_typeformatstring_var(
2375 file, 2, NULL, get_func_return_type(func),
2376 &v, typeformat_offset),
2377 file);
2378 }
2379
2380 current_func = func;
2381 if (func->args)
2382 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
2383 update_tfsoff(
2384 var->type,
2385 write_typeformatstring_var(
2386 file, 2, func, var->type, var,
2387 typeformat_offset),
2388 file);
2389 }
2390 }
2391 }
2392
2393 return *typeformat_offset + 1;
2394 }
2395
2396 static size_t process_tfs(FILE *file, const statement_list_t *stmts, type_pred_t pred)
2397 {
2398 unsigned int typeformat_offset = 2;
2399
2400 return process_tfs_stmts(file, stmts, pred, &typeformat_offset);
2401 }
2402
2403
2404 void write_typeformatstring(FILE *file, const statement_list_t *stmts, type_pred_t pred)
2405 {
2406 int indent = 0;
2407
2408 print_file(file, indent, "static const MIDL_TYPE_FORMAT_STRING __MIDL_TypeFormatString =\n");
2409 print_file(file, indent, "{\n");
2410 indent++;
2411 print_file(file, indent, "0,\n");
2412 print_file(file, indent, "{\n");
2413 indent++;
2414 print_file(file, indent, "NdrFcShort(0x0),\n");
2415
2416 set_all_tfswrite(TRUE);
2417 process_tfs(file, stmts, pred);
2418
2419 print_file(file, indent, "0x0\n");
2420 indent--;
2421 print_file(file, indent, "}\n");
2422 indent--;
2423 print_file(file, indent, "};\n");
2424 print_file(file, indent, "\n");
2425 }
2426
2427 static unsigned int get_required_buffer_size_type(
2428 const type_t *type, const char *name, unsigned int *alignment)
2429 {
2430 *alignment = 0;
2431 if (is_user_type(type))
2432 {
2433 const char *uname;
2434 const type_t *utype = get_user_type(type, &uname);
2435 return get_required_buffer_size_type(utype, uname, alignment);
2436 }
2437 else
2438 {
2439 switch (type->type)
2440 {
2441 case RPC_FC_BYTE:
2442 case RPC_FC_CHAR:
2443 case RPC_FC_USMALL:
2444 case RPC_FC_SMALL:
2445 *alignment = 4;
2446 return 1;
2447
2448 case RPC_FC_WCHAR:
2449 case RPC_FC_USHORT:
2450 case RPC_FC_SHORT:
2451 case RPC_FC_ENUM16:
2452 *alignment = 4;
2453 return 2;
2454
2455 case RPC_FC_ULONG:
2456 case RPC_FC_LONG:
2457 case RPC_FC_ENUM32:
2458 case RPC_FC_FLOAT:
2459 case RPC_FC_ERROR_STATUS_T:
2460 *alignment = 4;
2461 return 4;
2462
2463 case RPC_FC_HYPER:
2464 case RPC_FC_DOUBLE:
2465 *alignment = 8;
2466 return 8;
2467
2468 case RPC_FC_IGNORE:
2469 case RPC_FC_BIND_PRIMITIVE:
2470 return 0;
2471
2472 case RPC_FC_STRUCT:
2473 case RPC_FC_PSTRUCT:
2474 {
2475 size_t size = 0;
2476 const var_t *field;
2477 if (!type->fields_or_args) return 0;
2478 LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
2479 {
2480 unsigned int alignment;
2481 size += get_required_buffer_size_type(field->type, field->name,
2482 &alignment);
2483 }
2484 return size;
2485 }
2486
2487 case RPC_FC_RP:
2488 return
2489 is_base_type( type->ref->type ) || type->ref->type == RPC_FC_STRUCT
2490 ? get_required_buffer_size_type( type->ref, name, alignment )
2491 : 0;
2492
2493 case RPC_FC_SMFARRAY:
2494 case RPC_FC_LGFARRAY:
2495 return type->dim * get_required_buffer_size_type(type->ref, name, alignment);
2496
2497 default:
2498 return 0;
2499 }
2500 }
2501 }
2502
2503 static unsigned int get_required_buffer_size(const var_t *var, unsigned int *alignment, enum pass pass)
2504 {
2505 int in_attr = is_attr(var->attrs, ATTR_IN);
2506 int out_attr = is_attr(var->attrs, ATTR_OUT);
2507 const type_t *t;
2508
2509 if (!in_attr && !out_attr)
2510 in_attr = 1;
2511
2512 *alignment = 0;
2513
2514 for (t = var->type; is_ptr(t); t = t->ref)
2515 if (is_attr(t->attrs, ATTR_CONTEXTHANDLE))
2516 {
2517 *alignment = 4;
2518 return 20;
2519 }
2520
2521 if (pass == PASS_OUT)
2522 {
2523 if (out_attr && is_ptr(var->type))
2524 {
2525 type_t *type = var->type;
2526
2527 if (type->type == RPC_FC_STRUCT)
2528 {
2529 const var_t *field;
2530 unsigned int size = 36;
2531
2532 if (!type->fields_or_args) return size;
2533 LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
2534 {
2535 unsigned int align;
2536 size += get_required_buffer_size_type(
2537 field->type, field->name, &align);
2538 }
2539 return size;
2540 }
2541 }
2542 return 0;
2543 }
2544 else
2545 {
2546 if ((!out_attr || in_attr) && !var->type->size_is
2547 && !is_attr(var->attrs, ATTR_STRING) && !var->type->declarray)
2548 {
2549 if (is_ptr(var->type))
2550 {
2551 type_t *type = var->type;
2552
2553 if (is_base_type(type->type))
2554 {
2555 return 25;
2556 }
2557 else if (type->type == RPC_FC_STRUCT)
2558 {
2559 unsigned int size = 36;
2560 const var_t *field;
2561
2562 if (!type->fields_or_args) return size;
2563 LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
2564 {
2565 unsigned int align;
2566 size += get_required_buffer_size_type(
2567 field->type, field->name, &align);
2568 }
2569 return size;
2570 }
2571 }
2572 }
2573
2574 return get_required_buffer_size_type(var->type, var->name, alignment);
2575 }
2576 }
2577
2578 static unsigned int get_function_buffer_size( const func_t *func, enum pass pass )
2579 {
2580 const var_t *var;
2581 unsigned int total_size = 0, alignment;
2582
2583 if (func->args)
2584 {
2585 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
2586 {
2587 total_size += get_required_buffer_size(var, &alignment, pass);
2588 total_size += alignment;
2589 }
2590 }
2591
2592 if (pass == PASS_OUT && !is_void(get_func_return_type(func)))
2593 {
2594 var_t v = *func->def;
2595 v.type = get_func_return_type(func);
2596 total_size += get_required_buffer_size(&v, &alignment, PASS_RETURN);
2597 total_size += alignment;
2598 }
2599 return total_size;
2600 }
2601
2602 static void print_phase_function(FILE *file, int indent, const char *type,
2603 enum remoting_phase phase,
2604 const var_t *var, unsigned int type_offset)
2605 {
2606 const char *function;
2607 switch (phase)
2608 {
2609 case PHASE_BUFFERSIZE:
2610 function = "BufferSize";
2611 break;
2612 case PHASE_MARSHAL:
2613 function = "Marshall";
2614 break;
2615 case PHASE_UNMARSHAL:
2616 function = "Unmarshall";
2617 break;
2618 case PHASE_FREE:
2619 function = "Free";
2620 break;
2621 default:
2622 assert(0);
2623 return;
2624 }
2625
2626 print_file(file, indent, "Ndr%s%s(\n", type, function);
2627 indent++;
2628 print_file(file, indent, "&_StubMsg,\n");
2629 print_file(file, indent, "%s%s%s%s,\n",
2630 (phase == PHASE_UNMARSHAL) ? "(unsigned char **)" : "(unsigned char *)",
2631 (phase == PHASE_UNMARSHAL || decl_indirect(var->type)) ? "&" : "",
2632 (phase == PHASE_UNMARSHAL && decl_indirect(var->type)) ? "_p_" : "",
2633 var->name);
2634 print_file(file, indent, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]%s\n",
2635 type_offset, (phase == PHASE_UNMARSHAL) ? "," : ");");
2636 if (phase == PHASE_UNMARSHAL)
2637 print_file(file, indent, "0);\n");
2638 indent--;
2639 }
2640
2641 void print_phase_basetype(FILE *file, int indent, enum remoting_phase phase,
2642 enum pass pass, const var_t *var,
2643 const char *varname)
2644 {
2645 type_t *type = var->type;
2646 unsigned int size;
2647 unsigned int alignment = 0;
2648 unsigned char rtype;
2649
2650 /* no work to do for other phases, buffer sizing is done elsewhere */
2651 if (phase != PHASE_MARSHAL && phase != PHASE_UNMARSHAL)
2652 return;
2653
2654 rtype = is_ptr(type) ? type->ref->type : type->type;
2655
2656 switch (rtype)
2657 {
2658 case RPC_FC_BYTE:
2659 case RPC_FC_CHAR:
2660 case RPC_FC_SMALL:
2661 case RPC_FC_USMALL:
2662 size = 1;
2663 alignment = 1;
2664 break;
2665
2666 case RPC_FC_WCHAR:
2667 case RPC_FC_USHORT:
2668 case RPC_FC_SHORT:
2669 case RPC_FC_ENUM16:
2670 size = 2;
2671 alignment = 2;
2672 break;
2673
2674 case RPC_FC_ULONG:
2675 case RPC_FC_LONG:
2676 case RPC_FC_ENUM32:
2677 case RPC_FC_FLOAT:
2678 case RPC_FC_ERROR_STATUS_T:
2679 size = 4;
2680 alignment = 4;
2681 break;
2682
2683 case RPC_FC_HYPER:
2684 case RPC_FC_DOUBLE:
2685 size = 8;
2686 alignment = 8;
2687 break;
2688
2689 case RPC_FC_IGNORE:
2690 case RPC_FC_BIND_PRIMITIVE:
2691 /* no marshalling needed */
2692 return;
2693
2694 default:
2695 error("print_phase_basetype: Unsupported type: %s (0x%02x, ptr_level: 0)\n", var->name, rtype);
2696 size = 0;
2697 }
2698
2699 if (phase == PHASE_MARSHAL)
2700 print_file(file, indent, "MIDL_memset(_StubMsg.Buffer, 0, (0x%x - (long)_StubMsg.Buffer) & 0x%x);\n", alignment, alignment - 1);
2701 print_file(file, indent, "_StubMsg.Buffer = (unsigned char *)(((long)_StubMsg.Buffer + %u) & ~0x%x);\n",
2702 alignment - 1, alignment - 1);
2703
2704 if (phase == PHASE_MARSHAL)
2705 {
2706 print_file(file, indent, "*(");
2707 write_type_decl(file, is_ptr(type) ? type->ref : type, NULL);
2708 if (is_ptr(type))
2709 fprintf(file, " *)_StubMsg.Buffer = *");
2710 else
2711 fprintf(file, " *)_StubMsg.Buffer = ");
2712 fprintf(file, "%s", varname);
2713 fprintf(file, ";\n");
2714 }
2715 else if (phase == PHASE_UNMARSHAL)
2716 {
2717 print_file(file, indent, "if (_StubMsg.Buffer + sizeof(");
2718 write_type_decl(file, is_ptr(type) ? type->ref : type, NULL);
2719 fprintf(file, ") > _StubMsg.BufferEnd)\n");
2720 print_file(file, indent, "{\n");
2721 print_file(file, indent + 1, "RpcRaiseException(RPC_X_BAD_STUB_DATA);\n");
2722 print_file(file, indent, "}\n");
2723 if (pass == PASS_IN || pass == PASS_RETURN)
2724 print_file(file, indent, "");
2725 else
2726 print_file(file, indent, "*");
2727 fprintf(file, "%s", varname);
2728 if (pass == PASS_IN && is_ptr(type))
2729 fprintf(file, " = (");
2730 else
2731 fprintf(file, " = *(");
2732 write_type_decl(file, is_ptr(type) ? type->ref : type, NULL);
2733 fprintf(file, " *)_StubMsg.Buffer;\n");
2734 }
2735
2736 print_file(file, indent, "_StubMsg.Buffer += sizeof(");
2737 write_type_decl(file, var->type, NULL);
2738 fprintf(file, ");\n");
2739 }
2740
2741 /* returns whether the MaxCount, Offset or ActualCount members need to be
2742 * filled in for the specified phase */
2743 static inline int is_size_needed_for_phase(enum remoting_phase phase)
2744 {
2745 return (phase != PHASE_UNMARSHAL);
2746 }
2747
2748 expr_t *get_size_is_expr(const type_t *t, const char *name)
2749 {
2750 expr_t *x = NULL;
2751
2752 for ( ; is_ptr(t) || is_array(t); t = t->ref)
2753 if (t->size_is)
2754 {
2755 if (!x)
2756 x = t->size_is;
2757 else
2758 error("%s: multidimensional conformant"
2759 " arrays not supported at the top level\n",
2760 name);
2761 }
2762
2763 return x;
2764 }
2765
2766 static void write_remoting_arg(FILE *file, int indent, const func_t *func,
2767 enum pass pass, enum remoting_phase phase,
2768 const var_t *var)
2769 {
2770 int in_attr, out_attr, pointer_type;
2771 const type_t *type = var->type;
2772 unsigned char rtype;
2773 size_t start_offset = type->typestring_offset;
2774
2775 pointer_type = get_attrv(var->attrs, ATTR_POINTERTYPE);
2776 if (!pointer_type)
2777 pointer_type = RPC_FC_RP;
2778
2779 in_attr = is_attr(var->attrs, ATTR_IN);
2780 out_attr = is_attr(var->attrs, ATTR_OUT);
2781 if (!in_attr && !out_attr)
2782 in_attr = 1;
2783
2784 if (phase != PHASE_FREE)
2785 switch (pass)
2786 {
2787 case PASS_IN:
2788 if (!in_attr) return;
2789 break;
2790 case PASS_OUT:
2791 if (!out_attr) return;
2792 break;
2793 case PASS_RETURN:
2794 break;
2795 }
2796
2797 rtype = type->type;
2798
2799 if (is_context_handle(type))
2800 {
2801 if (phase == PHASE_MARSHAL)
2802 {
2803 if (pass == PASS_IN)
2804 {
2805 /* if the context_handle attribute appears in the chain of types
2806 * without pointers being followed, then the context handle must
2807 * be direct, otherwise it is a pointer */
2808 int is_ch_ptr = is_aliaschain_attr(type, ATTR_CONTEXTHANDLE) ? FALSE : TRUE;
2809 print_file(file, indent, "NdrClientContextMarshall(\n");
2810 print_file(file, indent + 1, "&_StubMsg,\n");
2811 print_file(file, indent + 1, "(NDR_CCONTEXT)%s%s,\n", is_ch_ptr ? "*" : "", var->name);
2812 print_file(file, indent + 1, "%s);\n", in_attr && out_attr ? "1" : "0");
2813 }
2814 else
2815 {
2816 print_file(file, indent, "NdrServerContextNewMarshall(\n");
2817 print_file(file, indent + 1, "&_StubMsg,\n");
2818 print_file(file, indent + 1, "(NDR_SCONTEXT)%s,\n", var->name);
2819 print_file(file, indent + 1, "(NDR_RUNDOWN)%s_rundown,\n", get_context_handle_type_name(var->type));
2820 print_file(file, indent + 1, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]);\n", start_offset);
2821 }
2822 }
2823 else if (phase == PHASE_UNMARSHAL)
2824 {
2825 if (pass == PASS_OUT)
2826 {
2827 if (!in_attr)
2828 print_file(file, indent, "*%s = 0;\n", var->name);
2829 print_file(file, indent, "NdrClientContextUnmarshall(\n");
2830 print_file(file, indent + 1, "&_StubMsg,\n");
2831 print_file(file, indent + 1, "(NDR_CCONTEXT *)%s,\n", var->name);
2832 print_file(file, indent + 1, "_Handle);\n");
2833 }
2834 else
2835 {
2836 print_file(file, indent, "%s = NdrServerContextNewUnmarshall(\n", var->name);
2837 print_file(file, indent + 1, "&_StubMsg,\n");
2838 print_file(file, indent + 1, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]);\n", start_offset);
2839 }
2840 }
2841 }
2842 else if (is_user_type(var->type))
2843 {
2844 print_phase_function(file, indent, "UserMarshal", phase, var, start_offset);
2845 }
2846 else if (is_string_type(var->attrs, var->type))
2847 {
2848 if (is_array(type) && !is_conformant_array(type))
2849 print_phase_function(file, indent, "NonConformantString", phase, var, start_offset);
2850 else
2851 {
2852 if (type->size_is && is_size_needed_for_phase(phase))
2853 {
2854 print_file(file, indent, "_StubMsg.MaxCount = (unsigned long)");
2855 write_expr(file, type->size_is, 1, 1, NULL, NULL);
2856 fprintf(file, ";\n");
2857 }
2858
2859 if (phase == PHASE_FREE || pass == PASS_RETURN || pointer_type == RPC_FC_UP)
2860 print_phase_function(file, indent, "Pointer", phase, var,
2861 start_offset - (type->size_is ? 4 : 2));
2862 else
2863 print_phase_function(file, indent, "ConformantString", phase, var,
2864 start_offset);
2865 }
2866 }
2867 else if (is_array(type))
2868 {
2869 unsigned char tc = type->type;
2870 const char *array_type = "FixedArray";
2871
2872 /* We already have the size_is expression since it's at the
2873 top level, but do checks for multidimensional conformant
2874 arrays. When we handle them, we'll need to extend this
2875 function to return a list, and then we'll actually use
2876 the return value. */
2877 get_size_is_expr(type, var->name);
2878
2879 if (tc == RPC_FC_SMVARRAY || tc == RPC_FC_LGVARRAY)
2880 {
2881 if (is_size_needed_for_phase(phase))
2882 {
2883 print_file(file, indent, "_StubMsg.Offset = (unsigned long)0;\n"); /* FIXME */
2884 print_file(file, indent, "_StubMsg.ActualCount = (unsigned long)");
2885 write_expr(file, type->length_is, 1, 1, NULL, NULL);
2886 fprintf(file, ";\n\n");
2887 }
2888 array_type = "VaryingArray";
2889 }
2890 else if (tc == RPC_FC_CARRAY)
2891 {
2892 if (is_size_needed_for_phase(phase))
2893 {
2894 print_file(file, indent, "_StubMsg.MaxCount = (unsigned long)");
2895 write_expr(file, type->size_is, 1, 1, NULL, NULL);
2896 fprintf(file, ";\n\n");
2897 }
2898 array_type = "ConformantArray";
2899 }
2900 else if (tc == RPC_FC_CVARRAY || tc == RPC_FC_BOGUS_ARRAY)
2901 {
2902 if (is_size_needed_for_phase(phase))
2903 {
2904 if (type->size_is)
2905 {
2906 print_file(file, indent, "_StubMsg.MaxCount = (unsigned long)");
2907 write_expr(file, type->size_is, 1, 1, NULL, NULL);
2908 fprintf(file, ";\n");
2909 }
2910 if (type->length_is)
2911 {
2912 print_file(file, indent, "_StubMsg.Offset = (unsigned long)0;\n"); /* FIXME */
2913 print_file(file, indent, "_StubMsg.ActualCount = (unsigned long)");
2914 write_expr(file, type->length_is, 1, 1, NULL, NULL);
2915 fprintf(file, ";\n\n");
2916 }
2917 }
2918 array_type = (tc == RPC_FC_BOGUS_ARRAY
2919 ? "ComplexArray"
2920 : "ConformantVaryingArray");
2921 }
2922
2923 if (pointer_type != RPC_FC_RP) array_type = "Pointer";
2924 print_phase_function(file, indent, array_type, phase, var, start_offset);
2925 if (phase == PHASE_FREE && pointer_type == RPC_FC_RP)
2926 {
2927 /* these are all unmarshalled by allocating memory */
2928 if (type->type == RPC_FC_BOGUS_ARRAY ||
2929 type->type == RPC_FC_CVARRAY ||
2930 ((type->type == RPC_FC_SMVARRAY || type->type == RPC_FC_LGVARRAY) && in_attr) ||
2931 (type->type == RPC_FC_CARRAY && !in_attr))
2932 {
2933 print_file(file, indent, "if (%s)\n", var->name);
2934 indent++;
2935 print_file(file, indent, "_StubMsg.pfnFree(%s);\n", var->name);
2936 }
2937 }
2938 }
2939 else if (!is_ptr(var->type) && is_base_type(rtype))
2940 {
2941 if (phase != PHASE_FREE)
2942 print_phase_basetype(file, indent, phase, pass, var, var->name);
2943 }
2944 else if (!is_ptr(var->type))
2945 {
2946 switch (rtype)
2947 {
2948 case RPC_FC_STRUCT:
2949 case RPC_FC_PSTRUCT:
2950 print_phase_function(file, indent, "SimpleStruct", phase, var, start_offset);
2951 break;
2952 case RPC_FC_CSTRUCT:
2953 case RPC_FC_CPSTRUCT:
2954 print_phase_function(file, indent, "ConformantStruct", phase, var, start_offset);
2955 break;
2956 case RPC_FC_CVSTRUCT:
2957 print_phase_function(file, indent, "ConformantVaryingStruct", phase, var, start_offset);
2958 break;
2959 case RPC_FC_BOGUS_STRUCT:
2960 print_phase_function(file, indent, "ComplexStruct", phase, var, start_offset);
2961 break;
2962 case RPC_FC_RP:
2963 if (is_base_type( var->type->ref->type ))
2964 {
2965 print_phase_basetype(file, indent, phase, pass, var, var->name);
2966 }
2967 else if (var->type->ref->type == RPC_FC_STRUCT)
2968 {
2969 if (phase != PHASE_BUFFERSIZE && phase != PHASE_FREE)
2970 print_phase_function(file, indent, "SimpleStruct", phase, var, start_offset + 4);
2971 }
2972 else
2973 {
2974 expr_t *iid;
2975 if ((iid = get_attrp( var->attrs, ATTR_IIDIS )))
2976 {
2977 print_file( file, indent, "_StubMsg.MaxCount = (unsigned long) " );
2978 write_expr( file, iid, 1, 1, NULL, NULL );
2979 fprintf( file, ";\n\n" );
2980 }
2981 print_phase_function(file, indent, "Pointer", phase, var, start_offset);
2982 }
2983 break;
2984 default:
2985 error("write_remoting_arguments: Unsupported type: %s (0x%02x)\n", var->name, rtype);
2986 }
2987 }
2988 else
2989 {
2990 if (last_ptr(var->type) && (pointer_type == RPC_FC_RP) && is_base_type(rtype))
2991 {
2992 if (phase != PHASE_FREE)
2993 print_phase_basetype(file,<