4581929c66e9c3e0d0d00df8c4b4326b3d30f0c0
[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 else if (is_attr(type->attrs, ATTR_SWITCHTYPE))
2036 {
2037 static const expr_t dummy_expr; /* FIXME */
2038 const type_t *st = get_attrp(type->attrs, ATTR_SWITCHTYPE);
2039
2040 switch (st->type)
2041 {
2042 case RPC_FC_CHAR:
2043 case RPC_FC_SMALL:
2044 case RPC_FC_USMALL:
2045 case RPC_FC_SHORT:
2046 case RPC_FC_USHORT:
2047 case RPC_FC_LONG:
2048 case RPC_FC_ULONG:
2049 case RPC_FC_ENUM16:
2050 case RPC_FC_ENUM32:
2051 print_file(file, 2, "0x%x,\t/* %s */\n", type->type, string_of_type(type->type));
2052 print_file(file, 2, "0x%x,\t/* Switch type= %s */\n",
2053 st->type, string_of_type(st->type));
2054 *tfsoff += 2;
2055 break;
2056 default:
2057 error("union switch type must be an integer, char, or enum\n");
2058 }
2059
2060 *tfsoff += write_conf_or_var_desc(file, NULL, *tfsoff, st, &dummy_expr );
2061 }
2062
2063 print_file(file, 2, "NdrFcShort(0x%x),\t/* %d */\n", size, size);
2064 print_file(file, 2, "NdrFcShort(0x%x),\t/* %d */\n", nbranch, nbranch);
2065 *tfsoff += 4;
2066
2067 if (fields) LIST_FOR_EACH_ENTRY(f, fields, var_t, entry)
2068 {
2069 type_t *ft = f->type;
2070 expr_list_t *cases = get_attrp(f->attrs, ATTR_CASE);
2071 int deflt = is_attr(f->attrs, ATTR_DEFAULT);
2072 expr_t *c;
2073
2074 if (cases == NULL && !deflt)
2075 error("union field %s with neither case nor default attribute\n", f->name);
2076
2077 if (cases) LIST_FOR_EACH_ENTRY(c, cases, expr_t, entry)
2078 {
2079 /* MIDL doesn't check for duplicate cases, even though that seems
2080 like a reasonable thing to do, it just dumps them to the TFS
2081 like we're going to do here. */
2082 print_file(file, 2, "NdrFcLong(0x%x),\t/* %d */\n", c->cval, c->cval);
2083 *tfsoff += 4;
2084 write_branch_type(file, ft, tfsoff);
2085 }
2086
2087 /* MIDL allows multiple default branches, even though that seems
2088 illogical, it just chooses the last one, which is what we will
2089 do. */
2090 if (deflt)
2091 {
2092 deftype = ft;
2093 nodeftype = 0;
2094 }
2095 }
2096
2097 if (deftype)
2098 {
2099 write_branch_type(file, deftype, tfsoff);
2100 }
2101 else
2102 {
2103 print_file(file, 2, "NdrFcShort(0x%x),\n", nodeftype);
2104 *tfsoff += 2;
2105 }
2106
2107 return start_offset;
2108 }
2109
2110 static size_t write_ip_tfs(FILE *file, const attr_list_t *attrs, type_t *type,
2111 unsigned int *typeformat_offset)
2112 {
2113 size_t i;
2114 size_t start_offset = *typeformat_offset;
2115 expr_t *iid = get_attrp(attrs, ATTR_IIDIS);
2116
2117 if (iid)
2118 {
2119 print_file(file, 2, "0x2f, /* FC_IP */\n");
2120 print_file(file, 2, "0x5c, /* FC_PAD */\n");
2121 *typeformat_offset
2122 += write_conf_or_var_desc(file, NULL, 0, type, iid) + 2;
2123 }
2124 else
2125 {
2126 const type_t *base = is_ptr(type) ? type->ref : type;
2127 const UUID *uuid = get_attrp(base->attrs, ATTR_UUID);
2128
2129 if (! uuid)
2130 error("%s: interface %s missing UUID\n", __FUNCTION__, base->name);
2131
2132 update_tfsoff(type, start_offset, file);
2133 print_start_tfs_comment(file, type, start_offset);
2134 print_file(file, 2, "0x2f,\t/* FC_IP */\n");
2135 print_file(file, 2, "0x5a,\t/* FC_CONSTANT_IID */\n");
2136 print_file(file, 2, "NdrFcLong(0x%08lx),\n", uuid->Data1);
2137 print_file(file, 2, "NdrFcShort(0x%04x),\n", uuid->Data2);
2138 print_file(file, 2, "NdrFcShort(0x%04x),\n", uuid->Data3);
2139 for (i = 0; i < 8; ++i)
2140 print_file(file, 2, "0x%02x,\n", uuid->Data4[i]);
2141
2142 if (file)
2143 fprintf(file, "\n");
2144
2145 *typeformat_offset += 18;
2146 }
2147 return start_offset;
2148 }
2149
2150 static size_t write_contexthandle_tfs(FILE *file, const type_t *type,
2151 const var_t *var,
2152 unsigned int *typeformat_offset)
2153 {
2154 size_t start_offset = *typeformat_offset;
2155 unsigned char flags = 0;
2156
2157 if (is_attr(current_iface->attrs, ATTR_STRICTCONTEXTHANDLE))
2158 flags |= NDR_STRICT_CONTEXT_HANDLE;
2159
2160 if (is_ptr(type))
2161 flags |= 0x80;
2162 if (is_attr(var->attrs, ATTR_IN))
2163 {
2164 flags |= 0x40;
2165 if (!is_attr(var->attrs, ATTR_OUT))
2166 flags |= NDR_CONTEXT_HANDLE_CANNOT_BE_NULL;
2167 }
2168 if (is_attr(var->attrs, ATTR_OUT))
2169 flags |= 0x20;
2170
2171 WRITE_FCTYPE(file, FC_BIND_CONTEXT, *typeformat_offset);
2172 print_file(file, 2, "0x%x,\t/* Context flags: ", flags);
2173 /* return and can't be null values overlap */
2174 if (((flags & 0x21) != 0x21) && (flags & NDR_CONTEXT_HANDLE_CANNOT_BE_NULL))
2175 print_file(file, 0, "can't be null, ");
2176 if (flags & NDR_CONTEXT_HANDLE_SERIALIZE)
2177 print_file(file, 0, "serialize, ");
2178 if (flags & NDR_CONTEXT_HANDLE_NO_SERIALIZE)
2179 print_file(file, 0, "no serialize, ");
2180 if (flags & NDR_STRICT_CONTEXT_HANDLE)
2181 print_file(file, 0, "strict, ");
2182 if ((flags & 0x21) == 0x20)
2183 print_file(file, 0, "out, ");
2184 if ((flags & 0x21) == 0x21)
2185 print_file(file, 0, "return, ");
2186 if (flags & 0x40)
2187 print_file(file, 0, "in, ");
2188 if (flags & 0x80)
2189 print_file(file, 0, "via ptr, ");
2190 print_file(file, 0, "*/\n");
2191 print_file(file, 2, "0, /* FIXME: rundown routine index*/\n");
2192 print_file(file, 2, "0, /* FIXME: param num */\n");
2193 *typeformat_offset += 4;
2194
2195 return start_offset;
2196 }
2197
2198 static size_t write_typeformatstring_var(FILE *file, int indent, const func_t *func,
2199 type_t *type, const var_t *var,
2200 unsigned int *typeformat_offset)
2201 {
2202 size_t offset;
2203
2204 if (is_context_handle(type))
2205 return write_contexthandle_tfs(file, type, var, typeformat_offset);
2206
2207 if (is_user_type(type))
2208 {
2209 write_user_tfs(file, type, typeformat_offset);
2210 return type->typestring_offset;
2211 }
2212
2213 if (is_string_type(var->attrs, type))
2214 return write_string_tfs(file, var->attrs, type, var->name, typeformat_offset);
2215
2216 if (is_array(type))
2217 {
2218 int ptr_type;
2219 size_t off;
2220 off = write_array_tfs(file, var->attrs, type, var->name, typeformat_offset);
2221 ptr_type = get_attrv(var->attrs, ATTR_POINTERTYPE);
2222 /* Top level pointers to conformant arrays may be handled specially
2223 since we can bypass the pointer, but if the array is buried
2224 beneath another pointer (e.g., "[size_is(,n)] int **p" then we
2225 always need to write the pointer. */
2226 if (!ptr_type && var->type != type)
2227 /* FIXME: This should use pointer_default, but the information
2228 isn't kept around for arrays. */
2229 ptr_type = RPC_FC_UP;
2230 if (ptr_type && ptr_type != RPC_FC_RP)
2231 {
2232 unsigned int absoff = type->typestring_offset;
2233 short reloff = absoff - (*typeformat_offset + 2);
2234 off = *typeformat_offset;
2235 print_file(file, 0, "/* %d */\n", off);
2236 print_file(file, 2, "0x%x, 0x0,\t/* %s */\n", ptr_type,
2237 string_of_type(ptr_type));
2238 print_file(file, 2, "NdrFcShort(0x%hx),\t/* Offset= %hd (%u) */\n",
2239 reloff, reloff, absoff);
2240 *typeformat_offset += 4;
2241 }
2242 return off;
2243 }
2244
2245 if (!is_ptr(type))
2246 {
2247 /* basic types don't need a type format string */
2248 if (is_base_type(type->type))
2249 return 0;
2250
2251 switch (type->type)
2252 {
2253 case RPC_FC_STRUCT:
2254 case RPC_FC_PSTRUCT:
2255 case RPC_FC_CSTRUCT:
2256 case RPC_FC_CPSTRUCT:
2257 case RPC_FC_CVSTRUCT:
2258 case RPC_FC_BOGUS_STRUCT:
2259 return write_struct_tfs(file, type, var->name, typeformat_offset);
2260 case RPC_FC_ENCAPSULATED_UNION:
2261 case RPC_FC_NON_ENCAPSULATED_UNION:
2262 return write_union_tfs(file, type, typeformat_offset);
2263 case RPC_FC_IGNORE:
2264 case RPC_FC_BIND_PRIMITIVE:
2265 /* nothing to do */
2266 return 0;
2267 default:
2268 error("write_typeformatstring_var: Unsupported type 0x%x for variable %s\n", type->type, var->name);
2269 }
2270 }
2271 else if (last_ptr(type))
2272 {
2273 size_t start_offset = *typeformat_offset;
2274 int in_attr = is_attr(var->attrs, ATTR_IN);
2275 int out_attr = is_attr(var->attrs, ATTR_OUT);
2276 const type_t *base = type->ref;
2277
2278 if (base->type == RPC_FC_IP
2279 || (base->type == 0
2280 && is_attr(var->attrs, ATTR_IIDIS)))
2281 {
2282 return write_ip_tfs(file, var->attrs, type, typeformat_offset);
2283 }
2284
2285 /* special case for pointers to base types */
2286 if (is_base_type(base->type))
2287 {
2288 print_file(file, indent, "0x%x, 0x%x, /* %s %s[simple_pointer] */\n",
2289 type->type, (!in_attr && out_attr) ? 0x0C : 0x08,
2290 string_of_type(type->type),
2291 (!in_attr && out_attr) ? "[allocated_on_stack] " : "");
2292 print_file(file, indent, "0x%02x, /* %s */\n", base->type, string_of_type(base->type));
2293 print_file(file, indent, "0x5c, /* FC_PAD */\n");
2294 *typeformat_offset += 4;
2295 return start_offset;
2296 }
2297 }
2298
2299 assert(is_ptr(type));
2300
2301 offset = write_typeformatstring_var(file, indent, func, type->ref, var, typeformat_offset);
2302 if (file)
2303 fprintf(file, "/* %2u */\n", *typeformat_offset);
2304 return write_pointer_only_tfs(file, var->attrs, type->type,
2305 !last_ptr(type) ? 0x10 : 0,
2306 offset, typeformat_offset);
2307 }
2308
2309 static int write_embedded_types(FILE *file, const attr_list_t *attrs, type_t *type,
2310 const char *name, int write_ptr, unsigned int *tfsoff)
2311 {
2312 int retmask = 0;
2313
2314 if (is_user_type(type))
2315 {
2316 write_user_tfs(file, type, tfsoff);
2317 }
2318 else if (is_string_type(attrs, type))
2319 {
2320 write_string_tfs(file, attrs, type, name, tfsoff);
2321 }
2322 else if (is_ptr(type))
2323 {
2324 type_t *ref = type->ref;
2325
2326 if (ref->type == RPC_FC_IP
2327 || (ref->type == 0
2328 && is_attr(attrs, ATTR_IIDIS)))
2329 {
2330 write_ip_tfs(file, attrs, type, tfsoff);
2331 }
2332 else
2333 {
2334 if (!processed(ref) && !is_base_type(ref->type))
2335 retmask |= write_embedded_types(file, NULL, ref, name, TRUE, tfsoff);
2336
2337 if (write_ptr)
2338 write_pointer_tfs(file, type, tfsoff);
2339
2340 retmask |= 1;
2341 }
2342 }
2343 else if (type->declarray && is_conformant_array(type))
2344 ; /* conformant arrays and strings are handled specially */
2345 else if (is_array(type))
2346 {
2347 write_array_tfs(file, attrs, type, name, tfsoff);
2348 if (is_conformant_array(type))
2349 retmask |= 1;
2350 }
2351 else if (is_struct(type->type))
2352 {
2353 if (!processed(type))
2354 write_struct_tfs(file, type, name, tfsoff);
2355 }
2356 else if (is_union(type->type))
2357 {
2358 if (!processed(type))
2359 write_union_tfs(file, type, tfsoff);
2360 }
2361 else if (!is_base_type(type->type))
2362 error("write_embedded_types: unknown embedded type for %s (0x%x)\n",
2363 name, type->type);
2364
2365 return retmask;
2366 }
2367
2368 static size_t process_tfs_stmts(FILE *file, const statement_list_t *stmts,
2369 type_pred_t pred, unsigned int *typeformat_offset)
2370 {
2371 const var_t *var;
2372 const statement_t *stmt;
2373
2374 if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
2375 {
2376 const type_t *iface;
2377 if (stmt->type == STMT_LIBRARY)
2378 {
2379 process_tfs_stmts(file, stmt->u.lib->stmts, pred, typeformat_offset);
2380 continue;
2381 }
2382 else if (stmt->type != STMT_TYPE || stmt->u.type->type != RPC_FC_IP)
2383 continue;
2384
2385 iface = stmt->u.type;
2386 if (!pred(iface))
2387 continue;
2388
2389 if (iface->funcs)
2390 {
2391 const func_t *func;
2392 current_iface = iface;
2393 LIST_FOR_EACH_ENTRY( func, iface->funcs, const func_t, entry )
2394 {
2395 if (is_local(func->def->attrs)) continue;
2396
2397 if (!is_void(get_func_return_type(func)))
2398 {
2399 var_t v = *func->def;
2400 v.type = get_func_return_type(func);
2401 update_tfsoff(get_func_return_type(func),
2402 write_typeformatstring_var(
2403 file, 2, NULL, get_func_return_type(func),
2404 &v, typeformat_offset),
2405 file);
2406 }
2407
2408 current_func = func;
2409 if (func->args)
2410 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
2411 update_tfsoff(
2412 var->type,
2413 write_typeformatstring_var(
2414 file, 2, func, var->type, var,
2415 typeformat_offset),
2416 file);
2417 }
2418 }
2419 }
2420
2421 return *typeformat_offset + 1;
2422 }
2423
2424 static size_t process_tfs(FILE *file, const statement_list_t *stmts, type_pred_t pred)
2425 {
2426 unsigned int typeformat_offset = 2;
2427
2428 return process_tfs_stmts(file, stmts, pred, &typeformat_offset);
2429 }
2430
2431
2432 void write_typeformatstring(FILE *file, const statement_list_t *stmts, type_pred_t pred)
2433 {
2434 int indent = 0;
2435
2436 print_file(file, indent, "static const MIDL_TYPE_FORMAT_STRING __MIDL_TypeFormatString =\n");
2437 print_file(file, indent, "{\n");
2438 indent++;
2439 print_file(file, indent, "0,\n");
2440 print_file(file, indent, "{\n");
2441 indent++;
2442 print_file(file, indent, "NdrFcShort(0x0),\n");
2443
2444 set_all_tfswrite(TRUE);
2445 process_tfs(file, stmts, pred);
2446
2447 print_file(file, indent, "0x0\n");
2448 indent--;
2449 print_file(file, indent, "}\n");
2450 indent--;
2451 print_file(file, indent, "};\n");
2452 print_file(file, indent, "\n");
2453 }
2454
2455 static unsigned int get_required_buffer_size_type(
2456 const type_t *type, const char *name, unsigned int *alignment)
2457 {
2458 *alignment = 0;
2459 if (is_user_type(type))
2460 {
2461 const char *uname;
2462 const type_t *utype = get_user_type(type, &uname);
2463 return get_required_buffer_size_type(utype, uname, alignment);
2464 }
2465 else
2466 {
2467 switch (type->type)
2468 {
2469 case RPC_FC_BYTE:
2470 case RPC_FC_CHAR:
2471 case RPC_FC_USMALL:
2472 case RPC_FC_SMALL:
2473 *alignment = 4;
2474 return 1;
2475
2476 case RPC_FC_WCHAR:
2477 case RPC_FC_USHORT:
2478 case RPC_FC_SHORT:
2479 case RPC_FC_ENUM16:
2480 *alignment = 4;
2481 return 2;
2482
2483 case RPC_FC_ULONG:
2484 case RPC_FC_LONG:
2485 case RPC_FC_ENUM32:
2486 case RPC_FC_FLOAT:
2487 case RPC_FC_ERROR_STATUS_T:
2488 *alignment = 4;
2489 return 4;
2490
2491 case RPC_FC_HYPER:
2492 case RPC_FC_DOUBLE:
2493 *alignment = 8;
2494 return 8;
2495
2496 case RPC_FC_IGNORE:
2497 case RPC_FC_BIND_PRIMITIVE:
2498 return 0;
2499
2500 case RPC_FC_STRUCT:
2501 case RPC_FC_PSTRUCT:
2502 {
2503 size_t size = 0;
2504 const var_t *field;
2505 if (!type->fields_or_args) return 0;
2506 LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
2507 {
2508 unsigned int alignment;
2509 size += get_required_buffer_size_type(field->type, field->name,
2510 &alignment);
2511 }
2512 return size;
2513 }
2514
2515 case RPC_FC_RP:
2516 return
2517 is_base_type( type->ref->type ) || type->ref->type == RPC_FC_STRUCT
2518 ? get_required_buffer_size_type( type->ref, name, alignment )
2519 : 0;
2520
2521 case RPC_FC_SMFARRAY:
2522 case RPC_FC_LGFARRAY:
2523 return type->dim * get_required_buffer_size_type(type->ref, name, alignment);
2524
2525 default:
2526 return 0;
2527 }
2528 }
2529 }
2530
2531 static unsigned int get_required_buffer_size(const var_t *var, unsigned int *alignment, enum pass pass)
2532 {
2533 int in_attr = is_attr(var->attrs, ATTR_IN);
2534 int out_attr = is_attr(var->attrs, ATTR_OUT);
2535 const type_t *t;
2536
2537 if (!in_attr && !out_attr)
2538 in_attr = 1;
2539
2540 *alignment = 0;
2541
2542 for (t = var->type; is_ptr(t); t = t->ref)
2543 if (is_attr(t->attrs, ATTR_CONTEXTHANDLE))
2544 {
2545 *alignment = 4;
2546 return 20;
2547 }
2548
2549 if (pass == PASS_OUT)
2550 {
2551 if (out_attr && is_ptr(var->type))
2552 {
2553 type_t *type = var->type;
2554
2555 if (type->type == RPC_FC_STRUCT)
2556 {
2557 const var_t *field;
2558 unsigned int size = 36;
2559
2560 if (!type->fields_or_args) return size;
2561 LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
2562 {
2563 unsigned int align;
2564 size += get_required_buffer_size_type(
2565 field->type, field->name, &align);
2566 }
2567 return size;
2568 }
2569 }
2570 return 0;
2571 }
2572 else
2573 {
2574 if ((!out_attr || in_attr) && !var->type->size_is
2575 && !is_attr(var->attrs, ATTR_STRING) && !var->type->declarray)
2576 {
2577 if (is_ptr(var->type))
2578 {
2579 type_t *type = var->type;
2580
2581 if (is_base_type(type->type))
2582 {
2583 return 25;
2584 }
2585 else if (type->type == RPC_FC_STRUCT)
2586 {
2587 unsigned int size = 36;
2588 const var_t *field;
2589
2590 if (!type->fields_or_args) return size;
2591 LIST_FOR_EACH_ENTRY( field, type->fields_or_args, const var_t, entry )
2592 {
2593 unsigned int align;
2594 size += get_required_buffer_size_type(
2595 field->type, field->name, &align);
2596 }
2597 return size;
2598 }
2599 }
2600 }
2601
2602 return get_required_buffer_size_type(var->type, var->name, alignment);
2603 }
2604 }
2605
2606 static unsigned int get_function_buffer_size( const func_t *func, enum pass pass )
2607 {
2608 const var_t *var;
2609 unsigned int total_size = 0, alignment;
2610
2611 if (func->args)
2612 {
2613 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
2614 {
2615 total_size += get_required_buffer_size(var, &alignment, pass);
2616 total_size += alignment;
2617 }
2618 }
2619
2620 if (pass == PASS_OUT && !is_void(get_func_return_type(func)))
2621 {
2622 var_t v = *func->def;
2623 v.type = get_func_return_type(func);
2624 total_size += get_required_buffer_size(&v, &alignment, PASS_RETURN);
2625 total_size += alignment;
2626 }
2627 return total_size;
2628 }
2629
2630 static void print_phase_function(FILE *file, int indent, const char *type,
2631 enum remoting_phase phase,
2632 const var_t *var, unsigned int type_offset)
2633 {
2634 const char *function;
2635 switch (phase)
2636 {
2637 case PHASE_BUFFERSIZE:
2638 function = "BufferSize";
2639 break;
2640 case PHASE_MARSHAL:
2641 function = "Marshall";
2642 break;
2643 case PHASE_UNMARSHAL:
2644 function = "Unmarshall";
2645 break;
2646 case PHASE_FREE:
2647 function = "Free";
2648 break;
2649 default:
2650 assert(0);
2651 return;
2652 }
2653
2654 print_file(file, indent, "Ndr%s%s(\n", type, function);
2655 indent++;
2656 print_file(file, indent, "&_StubMsg,\n");
2657 print_file(file, indent, "%s%s%s%s,\n",
2658 (phase == PHASE_UNMARSHAL) ? "(unsigned char **)" : "(unsigned char *)",
2659 (phase == PHASE_UNMARSHAL || decl_indirect(var->type)) ? "&" : "",
2660 (phase == PHASE_UNMARSHAL && decl_indirect(var->type)) ? "_p_" : "",
2661 var->name);
2662 print_file(file, indent, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]%s\n",
2663 type_offset, (phase == PHASE_UNMARSHAL) ? "," : ");");
2664 if (phase == PHASE_UNMARSHAL)
2665 print_file(file, indent, "0);\n");
2666 indent--;
2667 }
2668
2669 void print_phase_basetype(FILE *file, int indent, enum remoting_phase phase,
2670 enum pass pass, const var_t *var,
2671 const char *varname)
2672 {
2673 type_t *type = var->type;
2674 unsigned int size;
2675 unsigned int alignment = 0;
2676 unsigned char rtype;
2677
2678 /* no work to do for other phases, buffer sizing is done elsewhere */
2679 if (phase != PHASE_MARSHAL && phase != PHASE_UNMARSHAL)
2680 return;
2681
2682 rtype = is_ptr(type) ? type->ref->type : type->type;
2683
2684 switch (rtype)
2685 {
2686 case RPC_FC_BYTE:
2687 case RPC_FC_CHAR:
2688 case RPC_FC_SMALL:
2689 case RPC_FC_USMALL:
2690 size = 1;
2691 alignment = 1;
2692 break;
2693
2694 case RPC_FC_WCHAR:
2695 case RPC_FC_USHORT:
2696 case RPC_FC_SHORT:
2697 case RPC_FC_ENUM16:
2698 size = 2;
2699 alignment = 2;
2700 break;
2701
2702 case RPC_FC_ULONG:
2703 case RPC_FC_LONG:
2704 case RPC_FC_ENUM32:
2705 case RPC_FC_FLOAT:
2706 case RPC_FC_ERROR_STATUS_T:
2707 size = 4;
2708 alignment = 4;
2709 break;
2710
2711 case RPC_FC_HYPER:
2712 case RPC_FC_DOUBLE:
2713 size = 8;
2714 alignment = 8;
2715 break;
2716
2717 case RPC_FC_IGNORE:
2718 case RPC_FC_BIND_PRIMITIVE:
2719 /* no marshalling needed */
2720 return;
2721
2722 default:
2723 error("print_phase_basetype: Unsupported type: %s (0x%02x, ptr_level: 0)\n", var->name, rtype);
2724 size = 0;
2725 }
2726
2727 if (phase == PHASE_MARSHAL)
2728 print_file(file, indent, "MIDL_memset(_StubMsg.Buffer, 0, (0x%x - (long)_StubMsg.Buffer) & 0x%x);\n", alignment, alignment - 1);
2729 print_file(file, indent, "_StubMsg.Buffer = (unsigned char *)(((long)_StubMsg.Buffer + %u) & ~0x%x);\n",
2730 alignment - 1, alignment - 1);
2731
2732 if (phase == PHASE_MARSHAL)
2733 {
2734 print_file(file, indent, "*(");
2735 write_type_decl(file, is_ptr(type) ? type->ref : type, NULL);
2736 if (is_ptr(type))
2737 fprintf(file, " *)_StubMsg.Buffer = *");
2738 else
2739 fprintf(file, " *)_StubMsg.Buffer = ");
2740 fprintf(file, "%s", varname);
2741 fprintf(file, ";\n");
2742 }
2743 else if (phase == PHASE_UNMARSHAL)
2744 {
2745 print_file(file, indent, "if (_StubMsg.Buffer + sizeof(");
2746 write_type_decl(file, is_ptr(type) ? type->ref : type, NULL);
2747 fprintf(file, ") > _StubMsg.BufferEnd)\n");
2748 print_file(file, indent, "{\n");
2749 print_file(file, indent + 1, "RpcRaiseException(RPC_X_BAD_STUB_DATA);\n");
2750 print_file(file, indent, "}\n");
2751 if (pass == PASS_IN || pass == PASS_RETURN)
2752 print_file(file, indent, "");
2753 else
2754 print_file(file, indent, "*");
2755 fprintf(file, "%s", varname);
2756 if (pass == PASS_IN && is_ptr(type))
2757 fprintf(file, " = (");
2758 else
2759 fprintf(file, " = *(");
2760 write_type_decl(file, is_ptr(type) ? type->ref : type, NULL);
2761 fprintf(file, " *)_StubMsg.Buffer;\n");
2762 }
2763
2764 print_file(file, indent, "_StubMsg.Buffer += sizeof(");
2765 write_type_decl(file, var->type, NULL);
2766 fprintf(file, ");\n");
2767 }
2768
2769 /* returns whether the MaxCount, Offset or ActualCount members need to be
2770 * filled in for the specified phase */
2771 static inline int is_size_needed_for_phase(enum remoting_phase phase)
2772 {
2773 return (phase != PHASE_UNMARSHAL);
2774 }
2775
2776 expr_t *get_size_is_expr(const type_t *t, const char *name)
2777 {
2778 expr_t *x = NULL;
2779
2780 for ( ; is_ptr(t) || is_array(t); t = t->ref)
2781 if (t->size_is)
2782 {
2783 if (!x)
2784 x = t->size_is;
2785 else
2786 error("%s: multidimensional conformant"
2787 " arrays not supported at the top level\n",
2788 name);
2789 }
2790
2791 return x;
2792 }
2793
2794 static void write_remoting_arg(FILE *file, int indent, const func_t *func,
2795 enum pass pass, enum remoting_phase phase,
2796 const var_t *var)
2797 {
2798 int in_attr, out_attr, pointer_type;
2799 const type_t *type = var->type;
2800 unsigned char rtype;
2801 size_t start_offset = type->typestring_offset;
2802
2803 pointer_type = get_attrv(var->attrs, ATTR_POINTERTYPE);
2804 if (!pointer_type)
2805 pointer_type = RPC_FC_RP;
2806
2807 in_attr = is_attr(var->attrs, ATTR_IN);
2808 out_attr = is_attr(var->attrs, ATTR_OUT);
2809 if (!in_attr && !out_attr)
2810 in_attr = 1;
2811
2812 if (phase != PHASE_FREE)
2813 switch (pass)
2814 {
2815 case PASS_IN:
2816 if (!in_attr) return;
2817 break;
2818 case PASS_OUT:
2819 if (!out_attr) return;
2820 break;
2821 case PASS_RETURN:
2822 break;
2823 }
2824
2825 rtype = type->type;
2826
2827 if (is_context_handle(type))
2828 {
2829 if (phase == PHASE_MARSHAL)
2830 {
2831 if (pass == PASS_IN)
2832 {
2833 /* if the context_handle attribute appears in the chain of types
2834 * without pointers being followed, then the context handle must
2835 * be direct, otherwise it is a pointer */
2836 int is_ch_ptr = is_aliaschain_attr(type, ATTR_CONTEXTHANDLE) ? FALSE : TRUE;
2837 print_file(file, indent, "NdrClientContextMarshall(\n");
2838 print_file(file, indent + 1, "&_StubMsg,\n");
2839 print_file(file, indent + 1, "(NDR_CCONTEXT)%s%s,\n", is_ch_ptr ? "*" : "", var->name);
2840 print_file(file, indent + 1, "%s);\n", in_attr && out_attr ? "1" : "0");
2841 }
2842 else
2843 {
2844 print_file(file, indent, "NdrServerContextNewMarshall(\n");
2845 print_file(file, indent + 1, "&_StubMsg,\n");
2846 print_file(file, indent + 1, "(NDR_SCONTEXT)%s,\n", var->name);
2847 print_file(file, indent + 1, "(NDR_RUNDOWN)%s_rundown,\n", get_context_handle_type_name(var->type));
2848 print_file(file, indent + 1, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]);\n", start_offset);
2849 }
2850 }
2851 else if (phase == PHASE_UNMARSHAL)
2852 {
2853 if (pass == PASS_OUT)
2854 {
2855 if (!in_attr)
2856 print_file(file, indent, "*%s = 0;\n", var->name);
2857 print_file(file, indent, "NdrClientContextUnmarshall(\n");
2858 print_file(file, indent + 1, "&_StubMsg,\n");
2859 print_file(file, indent + 1, "(NDR_CCONTEXT *)%s,\n", var->name);
2860 print_file(file, indent + 1, "_Handle);\n");
2861 }
2862 else
2863 {
2864 print_file(file, indent, "%s = NdrServerContextNewUnmarshall(\n", var->name);
2865 print_file(file, indent + 1, "&_StubMsg,\n");
2866 print_file(file, indent + 1, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]);\n", start_offset);
2867 }
2868 }
2869 }
2870 else if (is_user_type(var->type))
2871 {
2872 print_phase_function(file, indent, "UserMarshal", phase, var, start_offset);
2873 }
2874 else if (is_string_type(var->attrs, var->type))
2875 {
2876 if (is_array(type) && !is_conformant_array(type))
2877 print_phase_function(file, indent, "NonConformantString", phase, var, start_offset);
2878 else
2879 {
2880 if (type->size_is && is_size_needed_for_phase(phase))
2881 {
2882 print_file(file, indent, "_StubMsg.MaxCount = (unsigned long)");
2883 write_expr(file, type->size_is, 1, 1, NULL, NULL);
2884 fprintf(file, ";\n");
2885 }
2886
2887 if (phase == PHASE_FREE || pass == PASS_RETURN || pointer_type == RPC_FC_UP)
2888 print_phase_function(file, indent, "Pointer", phase, var,
2889 start_offset - (type->size_is ? 4 : 2));
2890 else
2891 print_phase_function(file, indent, "ConformantString", phase, var,
2892 start_offset);
2893 }
2894 }
2895 else if (is_array(type))
2896 {
2897 unsigned char tc = type->type;
2898 const char *array_type = "FixedArray";
2899
2900 /* We already have the size_is expression since it's at the
2901 top level, but do checks for multidimensional conformant
2902 arrays. When we handle them, we'll need to extend this
2903 function to return a list, and then we'll actually use
2904 the return value. */
2905 get_size_is_expr(type, var->name);
2906
2907 if (tc == RPC_FC_SMVARRAY || tc == RPC_FC_LGVARRAY)
2908 {
2909 if (is_size_needed_for_phase(phase))
2910 {
2911 print_file(file, indent, "_StubMsg.Offset = (unsigned long)0;\n"); /* FIXME */
2912 print_file(file, indent, "_StubMsg.ActualCount = (unsigned long)");
2913 write_expr(file, type->length_is, 1, 1, NULL, NULL);
2914 fprintf(file, ";\n\n");
2915 }
2916 array_type = "VaryingArray";
2917 }
2918 else if (tc == RPC_FC_CARRAY)
2919 {
2920 if (is_size_needed_for_phase(phase))
2921 {
2922 print_file(file, indent, "_StubMsg.MaxCount = (unsigned long)");
2923 write_expr(file, type->size_is, 1, 1, NULL, NULL);
2924 fprintf(file, ";\n\n");
2925 }
2926 array_type = "ConformantArray";
2927 }
2928 else if (tc == RPC_FC_CVARRAY || tc == RPC_FC_BOGUS_ARRAY)
2929 {
2930 if (is_size_needed_for_phase(phase))
2931 {
2932 if (type->size_is)
2933 {
2934 print_file(file, indent, "_StubMsg.MaxCount = (unsigned long)");
2935 write_expr(file, type->size_is, 1, 1, NULL, NULL);
2936 fprintf(file, ";\n");
2937 }
2938 if (type->length_is)
2939 {
2940 print_file(file, indent, "_StubMsg.Offset = (unsigned long)0;\n"); /* FIXME */
2941 print_file(file, indent, "_StubMsg.ActualCount = (unsigned long)");
2942 write_expr(file, type->length_is, 1, 1, NULL, NULL);
2943 fprintf(file, ";\n\n");
2944 }
2945 }
2946 array_type = (tc == RPC_FC_BOGUS_ARRAY
2947 ? "ComplexArray"
2948 : "ConformantVaryingArray");
2949 }
2950
2951 if (pointer_type != RPC_FC_RP) array_type = "Pointer";
2952 print_phase_function(file, indent, array_type, phase, var, start_offset);
2953 if (phase == PHASE_FREE && pointer_type == RPC_FC_RP)
2954 {
2955 /* these are all unmarshalled by allocating memory */
2956 if (type->type == RPC_FC_BOGUS_ARRAY ||
2957 type->type == RPC_FC_CVARRAY ||
2958 ((type->type == RPC_FC_SMVARRAY || type->type == RPC_FC_LGVARRAY) && in_attr) ||
2959 (type->type == RPC_FC_CARRAY && !in_attr))
2960 {
2961 print_file(file, indent, "if (%s)\n", var->name);
2962 indent++;
2963 print_file(file, indent, "_StubMsg.pfnFree(%s);\n", var->name);
2964 }
2965 }
2966 }
2967 else if (!is_ptr(var->type) && is_base_type(rtype))
2968 {
2969 if (phase != PHASE_FREE)
2970 print_phase_basetype(file, indent, phase, pass, var, var->name);
2971 }
2972 else if (!is_ptr(var->type))
2973 {
2974 switch (rtype)
2975 {
2976 case RPC_FC_STRUCT:
2977 case RPC_FC_PSTRUCT:
2978 print_phase_function(file, indent, "SimpleStruct", phase, var, start_offset);
2979 break;
2980 case RPC_FC_CSTRUCT:
2981 case RPC_FC_CPSTRUCT:
2982 print_phase_function(file, indent, "ConformantStruct", phase, var, start_offset);
2983 break;
2984 case RPC_FC_CVSTRUCT:
2985 print_phase_function(file, indent, "ConformantVaryingStruct", phase, var, start_offset);
2986 break;
2987 case RPC_FC_BOGUS_STRUCT:
2988 print_phase_function(file, indent, "ComplexStruct", phase, var, start_offset);
2989 break;
2990 case RPC_FC_RP:
2991 if (is_base_type( var->type->ref->type ))
2992 {
2993 print_phase_basetype(file, indent, phase, pass, var, var->name);
2994 }
2995 else if (var->type->ref->type == RPC_FC_STRUCT)
2996 {
2997 if (phase != PHASE_BUFFERSIZE && phase != PHASE_FREE)
2998 print_phase_function(file, indent, "SimpleStruct", phase, var, start_offset + 4);
2999 }
3000 else
3001 {
3002 expr_t *iid;
3003 if ((iid = get_attrp( var->attrs, ATTR_IIDIS )))
3004 {
3005 print_file( file, indent, "_StubMsg.MaxCount = (unsigned long) " );
3006 write_expr( file, iid, 1, 1, NULL, NULL );
3007 fprintf( file, ";\n\n" );
3008 }
3009 print_phase_function(file, indent, "Pointer", phase, var, start_offset);
3010 }
3011 break;
3012 default:
3013 error("write_remoting_arguments: Unsupported type: %s (0x%02x)\n", var->name, rtype);
3014 }
3015 }
3016 else
3017 {
3018 if (last_ptr(var->type) && (pointer_type == RPC_FC_RP) && is_base_type(rtype))
3019 {
3020 if (phase != PHASE_FREE)
3021 print_phase_basetype(file, indent, phase, pass, var, var->name);
3022 }
3023 else if (last_ptr(var->type) && (pointer_type == RPC_FC_RP) && (rtype == RPC_FC_STRUCT))
3024 {
3025 if (phase != PHASE_BUFFERSIZE && phase != PHASE_FREE)
3026 print_phase_function(file, indent, "SimpleStruct", phase, var, start_offset + 4);
3027 }
3028 else
3029 {
3030 expr_t *iid;
3031 expr_t *sx = get_size_is_expr(type, var->name);
3032
3033 if ((iid = get_attrp( var->attrs, ATTR_IIDIS )))
3034 {
3035 print_file( file, indent, "_StubMsg.MaxCount = (unsigned long) " );
3036 write_expr( file, iid, 1, 1, NULL, NULL );
3037 fprintf( file, ";\n\n" );
3038 }
3039 else if (sx)
3040 {
3041 print_file(file, indent, "_StubMsg.MaxCount = (unsigned long) ");
3042 write_expr(file, sx, 1, 1, NULL, NULL);
3043 fprintf(file, ";\n\n");
3044 }
3045 if (var->type->ref->type == RPC_FC_IP)
3046 print_phase_function(file, indent, "InterfacePointer", phase, var, start_offset);
3047 else
3048 print_phase_function(file, indent, "Pointer", phase, var, start_offset);
3049 }
3050 }
3051 fprintf(file, "\n");
3052 }
3053
3054 void write_remoting_arguments(FILE *file, int indent, const func_t *func,
3055 enum pass pass, enum remoting_phase phase)
3056 {
3057 if (phase == PHASE_BUFFERSIZE && pass != PASS_RETURN)
3058 {
3059 unsigned int size = get_function_buffer_size( func, pass );
3060 print_file(file, indent, "_StubMsg.BufferLength = %u;\n", size);
3061 }
3062
3063 if (pass == PASS_RETURN)
3064 {
3065 var_t var;
3066 var = *func->def;
3067 var.type = get_func_return_type(func);
3068 var.name = xstrdup( "_RetVal" );
3069 write_remoting_arg( file, indent, func, pass, phase, &var );
3070 free( var.name );
3071 }
3072 else
3073 {
3074 const var_t *var;
3075 if (!func->args)
3076 return;
3077 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
3078 write_remoting_arg( file, indent, func, pass, phase, var );
3079 }
3080 }
3081
3082
3083 size_t get_size_procformatstring_type(const char *name, const type_t *type, const attr_list_t *attrs)
3084 {
3085 return write_procformatstring_type(NULL, 0, name, type, attrs, FALSE);
3086 }
3087
3088
3089 size_t get_size_procformatstring_func(const func_t *func)
3090 {
3091 const var_t *var;
3092 size_t size = 0;
3093
3094 /* argument list size */
3095 if (func->args)
3096 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
3097 size += get_size_procformatstring_type(var->name, var->type, var->attrs);
3098
3099 /* return value size */
3100 if (is_void(get_func_return_type(func)))
3101 size += 2; /* FC_END and FC_PAD */
3102 else
3103 size += get_size_procformatstring_type("return value", get_func_return_type(func), NULL);
3104
3105 return size;
3106 }
3107
3108 size_t get_size_procformatstring(const statement_list_t *stmts, type_pred_t pred)
3109 {
3110 const statement_t *stmt;
3111 size_t size = 1;
3112 const func_t *func;
3113
3114 if (stmts) LIST_FOR_EACH_ENTRY( stmt, stmts, const statement_t, entry )
3115 {
3116 const type_t *iface;
3117 if (stmt->type == STMT_LIBRARY)
3118 {
3119 size += get_size_procformatstring(stmt->u.lib->stmts, pred) - 1;
3120 continue;
3121 }
3122 else if (stmt->type != STMT_TYPE || stmt->u.type->type != RPC_FC_IP)
3123 continue;
3124
3125 iface = stmt->u.type;
3126 if (!pred(iface))
3127 continue;
3128
3129 if (iface->funcs)
3130 LIST_FOR_EACH_ENTRY( func, iface->funcs, const func_t, entry )
3131 if (!is_local(func->def->attrs))
3132 size += get_size_procformatstring_func( func );
3133 }
3134 return size;
3135 }
3136
3137 size_t get_size_typeformatstring(const statement_list_t *stmts, type_pred_t pred)
3138 {
3139 set_all_tfswrite(FALSE);
3140 return process_tfs(NULL, stmts, pred);
3141 }
3142
3143 void declare_stub_args( FILE *file, int indent, const func_t *func )
3144 {
3145 int in_attr, out_attr;
3146 int i = 0;
3147 const var_t *var;
3148
3149 /* declare return value '_RetVal' */
3150 if (!is_void(get_func_return_type(func)))
3151 {
3152 print_file(file, indent, "");
3153 write_type_decl_left(file, get_func_return_type(func));
3154 fprintf(file, " _RetVal;\n");
3155 }
3156
3157 if (!func->args)
3158 return;
3159
3160 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
3161 {
3162 int is_string = is_attr(var->attrs, ATTR_STRING);
3163
3164 in_attr = is_attr(var->attrs, ATTR_IN);
3165 out_attr = is_attr(var->attrs, ATTR_OUT);
3166 if (!out_attr && !in_attr)
3167 in_attr = 1;
3168
3169 if (is_context_handle(var->type))
3170 print_file(file, indent, "NDR_SCONTEXT %s;\n", var->name);
3171 else
3172 {
3173 if (!in_attr && !var->type->size_is && !is_string)
3174 {
3175 print_file(file, indent, "");
3176 write_type_decl(file, var->type->declarray ? var->type : var->type->ref,
3177 "_W%u", i++);
3178 fprintf(file, ";\n");
3179 }
3180
3181 print_file(file, indent, "");
3182 write_type_decl_left(file, var->type);
3183 fprintf(file, " ");
3184 if (var->type->declarray) {
3185 fprintf(file, "( *");
3186 write_name(file, var);
3187 fprintf(file, " )");
3188 } else
3189 write_name(file, var);
3190 write_type_right(file, var->type, FALSE);
3191 fprintf(file, ";\n");
3192
3193 if (decl_indirect(var->type))
3194 print_file(file, indent, "void *_p_%s = &%s;\n",
3195 var->name, var->name);
3196 }
3197 }
3198 }
3199
3200
3201 void assign_stub_out_args( FILE *file, int indent, const func_t *func )
3202 {
3203 int in_attr, out_attr;
3204 int i = 0, sep = 0;
3205 const var_t *var;
3206
3207 if (!func->args)
3208 return;
3209
3210 LIST_FOR_EACH_ENTRY( var, func->args, const var_t, entry )
3211 {
3212 int is_string = is_attr(var->attrs, ATTR_STRING);
3213 in_attr = is_attr(var->attrs, ATTR_IN);
3214 out_attr = is_attr(var->attrs, ATTR_OUT);
3215 if (!out_attr && !in_attr)
3216 in_attr = 1;
3217
3218 if (!in_attr)
3219 {
3220 print_file(file, indent, "");
3221 write_name(file, var);
3222
3223 if (is_context_handle(var->type))
3224 {
3225 fprintf(file, " = NdrContextHandleInitialize(\n");
3226 print_file(file, indent + 1, "&_StubMsg,\n");
3227 print_file(file, indent + 1, "(PFORMAT_STRING)&__MIDL_TypeFormatString.Format[%d]);\n",
3228 var->type->typestring_offset);
3229 }
3230 else if (var->type->size_is)
3231 {
3232 unsigned int size, align = 0;
3233 type_t *type = var->type;
3234
3235 fprintf(file, " = NdrAllocate(&_StubMsg, ");
3236 for ( ; type->size_is ; type = type->ref)
3237 {
3238 write_expr(file, type->size_is, TRUE, TRUE, NULL, NULL);
3239 fprintf(file, " * ");
3240 }
3241 size = type_memsize(type, &align);
3242 fprintf(file, "%u);\n", size);
3243 }
3244 else if (!is_string)
3245 {
3246 fprintf(file, " = &_W%u;\n", i);
3247 if (is_ptr(var->type) && !last_ptr(var->type))
3248 print_file(file, indent, "_W%u = 0;\n", i);
3249 i++;
3250 }
3251
3252 sep = 1;
3253 }
3254 }
3255 if (sep)
3256 fprintf(file, "\n");
3257 }
3258
3259
3260 int write_expr_eval_routines(FILE *file, const char *iface)
3261 {
3262 static const char *var_name = "pS";
3263 static const char *var_name_expr = "pS->";
3264 int result = 0;
3265 struct expr_eval_routine *eval;
3266 unsigned short callback_offset = 0;
3267
3268 LIST_FOR_EACH_ENTRY(eval, &expr_eval_routines, struct expr_eval_routine, entry)
3269 {
3270 const char *name = eval->structure->name;
3271 result = 1;
3272
3273 print_file(file, 0, "static void __RPC_USER %s_%sExprEval_%04u(PMIDL_STUB_MESSAGE pStubMsg)\n",
3274 iface, name, callback_offset);
3275 print_file(file, 0, "{\n");
3276 print_file (file, 1, "%s *%s = (%s *)(pStubMsg->StackTop - %u);\n",
3277 name, var_name, name, eval->baseoff);
3278 print_file(file, 1, "pStubMsg->Offset = 0;\n"); /* FIXME */
3279 print_file(file, 1, "pStubMsg->MaxCount = (unsigned long)");
3280 write_expr(file, eval->expr, 1, 1, var_name_expr, eval->structure);
3281 fprintf(file, ";\n");
3282 print_file(file, 0, "}\n\n");
3283 callback_offset++;
3284 }
3285 return result;
3286 }
3287
3288 void write_expr_eval_routine_list(FILE *file, const char *iface)
3289 {
3290 struct expr_eval_routine *eval;
3291 struct expr_eval_routine *cursor;
3292 unsigned short callback_offset = 0;
3293
3294 fprintf(file, "static const EXPR_EVAL ExprEvalRoutines[] =\n");
3295 fprintf(file, "{\n");
3296
3297 LIST_FOR_EACH_ENTRY_SAFE(eval, cursor, &expr_eval_routines, struct expr_eval_routine, entry)
3298 {
3299 const char *name = eval->structure->name;
3300 print_file(file, 1, "%s_%sExprEval_%04u,\n", iface, name, callback_offset);
3301 callback_offset++;
3302 list_remove(&eval->entry);
3303 free(eval);
3304 }
3305
3306 fprintf(file, "};\n\n");
3307 }
3308
3309 void write_user_quad_list(FILE *file)
3310 {
3311 user_type_t *ut;
3312
3313 if (list_empty(&user_type_list))
3314 return;
3315
3316 fprintf(file, "static const USER_MARSHAL_ROUTINE_QUADRUPLE UserMarshalRoutines[] =\n");
3317 fprintf(file, "{\n");
3318 LIST_FOR_EACH_ENTRY(ut, &user_type_list, user_type_t, entry)
3319 {
3320 const char *sep = &ut->entry == list_tail(&user_type_list) ? "" : ",";
3321 print_file(file, 1, "{\n");
3322 print_file(file, 2, "(USER_MARSHAL_SIZING_ROUTINE)%s_UserSize,\n", ut->name);
3323 print_file(file, 2, "(USER_MARSHAL_MARSHALLING_ROUTINE)%s_UserMarshal,\n", ut->name);
3324 print_file(file, 2, "(USER_MARSHAL_UNMARSHALLING_ROUTINE)%s_UserUnmarshal,\n", ut->name);
3325 print_file(file, 2, "(USER_MARSHAL_FREEING_ROUTINE)%s_UserFree\n", ut->name);
3326 print_file(file, 1, "}%s\n", sep);
3327 }
3328 fprintf(file, "};\n\n");
3329 }
3330
3331 void write_endpoints( FILE *f, const char *prefix, const str_list_t *list )
3332 {
3333 const struct str_list_entry_t *endpoint;
3334 const char *p;
3335
3336 /* this should be an array of RPC_PROTSEQ_ENDPOINT but we want const strings */
3337 print_file( f, 0, "static const unsigned char * %s__RpcProtseqEndpoint[][2] =\n{\n", prefix );
3338 LIST_FOR_EACH_ENTRY( endpoint, list, const struct str_list_entry_t, entry )
3339 {
3340 print_file( f, 1, "{ (const unsigned char *)\"" );
3341 for (p = endpoint->str; *p && *p != ':'; p++)
3342 {
3343 if (*p == '"' || *p == '\\') fputc( '\\', f );
3344 fputc( *p, f );
3345 }
3346 if (!*p) goto error;
3347 if (p[1] != '[') goto error;
3348
3349 fprintf( f, "\", (const unsigned char *)\"" );
3350 for (p += 2; *p && *p != ']'; p++)
3351 {
3352 if (*p == '"' || *p == '\\') fputc( '\\', f );
3353 fputc( *p, f );
3354 }
3355 if (*p != ']') goto error;
3356 fprintf( f, "\" },\n" );
3357 }
3358 print_file( f, 0, "};\n\n" );
3359 return;
3360
3361 error:
3362 error("Invalid endpoint syntax '%s'\n", endpoint->str);
3363 }