?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
bcrypt: Implement BCryptGenerateSymmetricKey and BCryptDestroyKey.
[wine.git] / tools / widl / write_msft.c
blob4b09317acfa8a66b6ca57ae4cf5217ffddb3b128
1 /*
2 * Typelib v2 (MSFT) generation
3 *
4 * Copyright 2004 Alastair Bridgewater
5 * 2004, 2005 Huw Davies
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 * Known problems:
23 *
24 * Badly incomplete.
25 *
26 * Only works on little-endian systems.
27 *
28 */
29
30 #include "config.h"
31 #include "wine/port.h"
32
33 #include <stdlib.h>
34 #include <string.h>
35 #include <stdarg.h>
36 #include <stdio.h>
37 #include <ctype.h>
38 #include <time.h>
39
40 #define NONAMELESSUNION
41
42 #include "winerror.h"
43 #include "windef.h"
44 #include "winbase.h"
45 #include "winnls.h"
46
47 #include "widl.h"
48 #include "typelib.h"
49 #include "typelib_struct.h"
50 #include "utils.h"
51 #include "header.h"
52 #include "hash.h"
53 #include "typetree.h"
54 #include "parser.h"
55
56 enum MSFT_segment_index {
57 MSFT_SEG_TYPEINFO = 0, /* type information */
58 MSFT_SEG_IMPORTINFO, /* import information */
59 MSFT_SEG_IMPORTFILES, /* import filenames */
60 MSFT_SEG_REFERENCES, /* references (?) */
61 MSFT_SEG_GUIDHASH, /* hash table for guids? */
62 MSFT_SEG_GUID, /* guid storage */
63 MSFT_SEG_NAMEHASH, /* hash table for names */
64 MSFT_SEG_NAME, /* name storage */
65 MSFT_SEG_STRING, /* string storage */
66 MSFT_SEG_TYPEDESC, /* type descriptions */
67 MSFT_SEG_ARRAYDESC, /* array descriptions */
68 MSFT_SEG_CUSTDATA, /* custom data */
69 MSFT_SEG_CUSTDATAGUID, /* custom data guids */
70 MSFT_SEG_UNKNOWN, /* ??? */
71 MSFT_SEG_UNKNOWN2, /* ??? */
72 MSFT_SEG_MAX /* total number of segments */
73 };
74
75 typedef struct tagMSFT_ImpFile {
76 int guid;
77 LCID lcid;
78 int version;
79 char filename[0]; /* preceded by two bytes of encoded (length << 2) + flags in the low two bits. */
80 } MSFT_ImpFile;
81
82 typedef struct _msft_typelib_t
83 {
84 typelib_t *typelib;
85 MSFT_Header typelib_header;
86 MSFT_pSeg typelib_segdir[MSFT_SEG_MAX];
87 unsigned char *typelib_segment_data[MSFT_SEG_MAX];
88 int typelib_segment_block_length[MSFT_SEG_MAX];
89
90 INT typelib_typeinfo_offsets[0x200]; /* Hope that's enough. */
91
92 INT *typelib_namehash_segment;
93 INT *typelib_guidhash_segment;
94
95 INT help_string_dll_offset;
96
97 struct _msft_typeinfo_t *typeinfos;
98 struct _msft_typeinfo_t *last_typeinfo;
99 } msft_typelib_t;
100
101 typedef struct _msft_typeinfo_t
102 {
103 msft_typelib_t *typelib;
104 MSFT_TypeInfoBase *typeinfo;
105
106 int typekind;
107
108 unsigned int var_data_allocated;
109 int *var_data;
110
111 unsigned int func_data_allocated;
112 int *func_data;
113
114 int vars_allocated;
115 int *var_indices;
116 int *var_names;
117 int *var_offsets;
118
119 int funcs_allocated;
120 int *func_indices;
121 int *func_names;
122 int *func_offsets;
123
124 int datawidth;
125
126 struct _msft_typeinfo_t *next_typeinfo;
127 } msft_typeinfo_t;
128
129
130
131 /*================== Internal functions ===================================*/
132
133 /****************************************************************************
134 * ctl2_init_header
135 *
136 * Initializes the type library header of a new typelib.
137 */
138 static void ctl2_init_header(
139 msft_typelib_t *typelib) /* [I] The typelib to initialize. */
140 {
141 typelib->typelib_header.magic1 = 0x5446534d;
142 typelib->typelib_header.magic2 = 0x00010002;
143 typelib->typelib_header.posguid = -1;
144 typelib->typelib_header.lcid = 0x0409;
145 typelib->typelib_header.lcid2 = 0x0;
146 typelib->typelib_header.varflags = 0x40;
147 typelib->typelib_header.version = 0;
148 typelib->typelib_header.flags = 0;
149 typelib->typelib_header.nrtypeinfos = 0;
150 typelib->typelib_header.helpstring = -1;
151 typelib->typelib_header.helpstringcontext = 0;
152 typelib->typelib_header.helpcontext = 0;
153 typelib->typelib_header.nametablecount = 0;
154 typelib->typelib_header.nametablechars = 0;
155 typelib->typelib_header.NameOffset = -1;
156 typelib->typelib_header.helpfile = -1;
157 typelib->typelib_header.CustomDataOffset = -1;
158 typelib->typelib_header.res44 = 0x20;
159 typelib->typelib_header.res48 = 0x80;
160 typelib->typelib_header.dispatchpos = -1;
161 typelib->typelib_header.nimpinfos = 0;
162 }
163
164 /****************************************************************************
165 * ctl2_init_segdir
166 *
167 * Initializes the segment directory of a new typelib.
168 */
169 static void ctl2_init_segdir(
170 msft_typelib_t *typelib) /* [I] The typelib to initialize. */
171 {
172 int i;
173 MSFT_pSeg *segdir;
174
175 segdir = &typelib->typelib_segdir[MSFT_SEG_TYPEINFO];
176
177 for (i = 0; i < MSFT_SEG_MAX; i++) {
178 segdir[i].offset = -1;
179 segdir[i].length = 0;
180 segdir[i].res08 = -1;
181 segdir[i].res0c = 0x0f;
182 }
183 }
184
185 /****************************************************************************
186 * ctl2_hash_guid
187 *
188 * Generates a hash key from a GUID.
189 *
190 * RETURNS
191 *
192 * The hash key for the GUID.
193 */
194 static int ctl2_hash_guid(
195 REFGUID guid) /* [I] The guid to hash. */
196 {
197 int hash;
198 int i;
199
200 hash = 0;
201 for (i = 0; i < 8; i ++) {
202 hash ^= ((const short *)guid)[i];
203 }
204
205 return hash & 0x1f;
206 }
207
208 /****************************************************************************
209 * ctl2_find_guid
210 *
211 * Locates a guid in a type library.
212 *
213 * RETURNS
214 *
215 * The offset into the GUID segment of the guid, or -1 if not found.
216 */
217 static int ctl2_find_guid(
218 msft_typelib_t *typelib, /* [I] The typelib to operate against. */
219 int hash_key, /* [I] The hash key for the guid. */
220 REFGUID guid) /* [I] The guid to find. */
221 {
222 int offset;
223 MSFT_GuidEntry *guidentry;
224
225 offset = typelib->typelib_guidhash_segment[hash_key];
226 while (offset != -1) {
227 guidentry = (MSFT_GuidEntry *)&typelib->typelib_segment_data[MSFT_SEG_GUID][offset];
228
229 if (!memcmp(guidentry, guid, sizeof(GUID))) return offset;
230
231 offset = guidentry->next_hash;
232 }
233
234 return offset;
235 }
236
237 /****************************************************************************
238 * ctl2_find_name
239 *
240 * Locates a name in a type library.
241 *
242 * RETURNS
243 *
244 * The offset into the NAME segment of the name, or -1 if not found.
245 *
246 * NOTES
247 *
248 * The name must be encoded as with ctl2_encode_name().
249 */
250 static int ctl2_find_name(
251 msft_typelib_t *typelib, /* [I] The typelib to operate against. */
252 char *name) /* [I] The encoded name to find. */
253 {
254 int offset;
255 int *namestruct;
256
257 offset = typelib->typelib_namehash_segment[name[2] & 0x7f];
258 while (offset != -1) {
259 namestruct = (int *)&typelib->typelib_segment_data[MSFT_SEG_NAME][offset];
260
261 if (!((namestruct[2] ^ *((int *)name)) & 0xffff00ff)) {
262 /* hash codes and lengths match, final test */
263 if (!strncasecmp(name+4, (void *)(namestruct+3), name[0])) break;
264 }
265
266 /* move to next item in hash bucket */
267 offset = namestruct[1];
268 }
269
270 return offset;
271 }
272
273 /****************************************************************************
274 * ctl2_encode_name
275 *
276 * Encodes a name string to a form suitable for storing into a type library
277 * or comparing to a name stored in a type library.
278 *
279 * RETURNS
280 *
281 * The length of the encoded name, including padding and length+hash fields.
282 *
283 * NOTES
284 *
285 * Will throw an exception if name or result are NULL. Is not multithread
286 * safe in the slightest.
287 */
288 static int ctl2_encode_name(
289 msft_typelib_t *typelib, /* [I] The typelib to operate against (used for LCID only). */
290 const char *name, /* [I] The name string to encode. */
291 char **result) /* [O] A pointer to a pointer to receive the encoded name. */
292 {
293 int length;
294 static char converted_name[0x104];
295 int offset;
296 int value;
297
298 length = strlen(name);
299 memcpy(converted_name + 4, name, length);
300
301 converted_name[length + 4] = 0;
302
303
304 value = lhash_val_of_name_sys(typelib->typelib_header.varflags & 0x0f, typelib->typelib_header.lcid, converted_name + 4);
305
306 #ifdef WORDS_BIGENDIAN
307 converted_name[3] = length & 0xff;
308 converted_name[2] = 0x00;
309 converted_name[1] = value;
310 converted_name[0] = value >> 8;
311 #else
312 converted_name[0] = length & 0xff;
313 converted_name[1] = 0x00;
314 converted_name[2] = value;
315 converted_name[3] = value >> 8;
316 #endif
317
318 for (offset = (4 - length) & 3; offset; offset--) converted_name[length + offset + 3] = 0x57;
319
320 *result = converted_name;
321
322 return (length + 7) & ~3;
323 }
324
325 /****************************************************************************
326 * ctl2_encode_string
327 *
328 * Encodes a string to a form suitable for storing into a type library or
329 * comparing to a string stored in a type library.
330 *
331 * RETURNS
332 *
333 * The length of the encoded string, including padding and length fields.
334 *
335 * NOTES
336 *
337 * Will throw an exception if string or result are NULL. Is not multithread
338 * safe in the slightest.
339 */
340 static int ctl2_encode_string(
341 const char *string, /* [I] The string to encode. */
342 char **result) /* [O] A pointer to a pointer to receive the encoded string. */
343 {
344 int length;
345 static char converted_string[0x104];
346 int offset;
347
348 length = strlen(string);
349 memcpy(converted_string + 2, string, length);
350
351 #ifdef WORDS_BIGENDIAN
352 converted_string[1] = length & 0xff;
353 converted_string[0] = (length >> 8) & 0xff;
354 #else
355 converted_string[0] = length & 0xff;
356 converted_string[1] = (length >> 8) & 0xff;
357 #endif
358
359 if(length < 3) { /* strings of this length are padded with up to 8 bytes incl the 2 byte length */
360 for(offset = 0; offset < 4; offset++)
361 converted_string[length + offset + 2] = 0x57;
362 length += 4;
363 }
364 for (offset = (4 - (length + 2)) & 3; offset; offset--) converted_string[length + offset + 1] = 0x57;
365
366 *result = converted_string;
367
368 return (length + 5) & ~3;
369 }
370
371 /****************************************************************************
372 * ctl2_alloc_segment
373 *
374 * Allocates memory from a segment in a type library.
375 *
376 * RETURNS
377 *
378 * Success: The offset within the segment of the new data area.
379 *
380 * BUGS
381 *
382 * Does not (yet) handle the case where the allocated segment memory needs to grow.
383 */
384 static int ctl2_alloc_segment(
385 msft_typelib_t *typelib, /* [I] The type library in which to allocate. */
386 enum MSFT_segment_index segment, /* [I] The segment in which to allocate. */
387 int size, /* [I] The amount to allocate. */
388 int block_size) /* [I] Initial allocation block size, or 0 for default. */
389 {
390 int offset;
391
392 if(!typelib->typelib_segment_data[segment]) {
393 if (!block_size) block_size = 0x2000;
394
395 typelib->typelib_segment_block_length[segment] = block_size;
396 typelib->typelib_segment_data[segment] = xmalloc(block_size);
397 if (!typelib->typelib_segment_data[segment]) return -1;
398 memset(typelib->typelib_segment_data[segment], 0x57, block_size);
399 }
400
401 while ((typelib->typelib_segdir[segment].length + size) > typelib->typelib_segment_block_length[segment]) {
402 unsigned char *block;
403
404 block_size = typelib->typelib_segment_block_length[segment];
405 block = xrealloc(typelib->typelib_segment_data[segment], block_size << 1);
406
407 if (segment == MSFT_SEG_TYPEINFO) {
408 /* TypeInfos have a direct pointer to their memory space, so we have to fix them up. */
409 msft_typeinfo_t *typeinfo;
410
411 for (typeinfo = typelib->typeinfos; typeinfo; typeinfo = typeinfo->next_typeinfo) {
412 typeinfo->typeinfo = (void *)&block[((unsigned char *)typeinfo->typeinfo) - typelib->typelib_segment_data[segment]];
413 }
414 }
415
416 memset(block + block_size, 0x57, block_size);
417 typelib->typelib_segment_block_length[segment] = block_size << 1;
418 typelib->typelib_segment_data[segment] = block;
419 }
420
421 offset = typelib->typelib_segdir[segment].length;
422 typelib->typelib_segdir[segment].length += size;
423
424 return offset;
425 }
426
427 /****************************************************************************
428 * ctl2_alloc_typeinfo
429 *
430 * Allocates and initializes a typeinfo structure in a type library.
431 *
432 * RETURNS
433 *
434 * Success: The offset of the new typeinfo.
435 * Failure: -1 (this is invariably an out of memory condition).
436 */
437 static int ctl2_alloc_typeinfo(
438 msft_typelib_t *typelib, /* [I] The type library to allocate in. */
439 int nameoffset) /* [I] The offset of the name for this typeinfo. */
440 {
441 int offset;
442 MSFT_TypeInfoBase *typeinfo;
443
444 offset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEINFO, sizeof(MSFT_TypeInfoBase), 0);
445
446 typelib->typelib_typeinfo_offsets[typelib->typelib_header.nrtypeinfos++] = offset;
447
448 typeinfo = (void *)(typelib->typelib_segment_data[MSFT_SEG_TYPEINFO] + offset);
449
450 typeinfo->typekind = (typelib->typelib_header.nrtypeinfos - 1) << 16;
451 typeinfo->memoffset = -1; /* should be EOF if no elements */
452 typeinfo->res2 = 0;
453 typeinfo->res3 = -1;
454 typeinfo->res4 = 3;
455 typeinfo->res5 = 0;
456 typeinfo->cElement = 0;
457 typeinfo->res7 = 0;
458 typeinfo->res8 = 0;
459 typeinfo->res9 = 0;
460 typeinfo->resA = 0;
461 typeinfo->posguid = -1;
462 typeinfo->flags = 0;
463 typeinfo->NameOffset = nameoffset;
464 typeinfo->version = 0;
465 typeinfo->docstringoffs = -1;
466 typeinfo->helpstringcontext = 0;
467 typeinfo->helpcontext = 0;
468 typeinfo->oCustData = -1;
469 typeinfo->cbSizeVft = 0;
470 typeinfo->cImplTypes = 0;
471 typeinfo->size = 0;
472 typeinfo->datatype1 = -1;
473 typeinfo->datatype2 = 0;
474 typeinfo->res18 = 0;
475 typeinfo->res19 = -1;
476
477 return offset;
478 }
479
480 /****************************************************************************
481 * ctl2_alloc_guid
482 *
483 * Allocates and initializes a GUID structure in a type library. Also updates
484 * the GUID hash table as needed.
485 *
486 * RETURNS
487 *
488 * Success: The offset of the new GUID.
489 */
490 static int ctl2_alloc_guid(
491 msft_typelib_t *typelib, /* [I] The type library to allocate in. */
492 MSFT_GuidEntry *guid) /* [I] The GUID to store. */
493 {
494 int offset;
495 MSFT_GuidEntry *guid_space;
496 int hash_key;
497
498 chat("adding uuid {%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}\n",
499 guid->guid.Data1, guid->guid.Data2, guid->guid.Data3,
500 guid->guid.Data4[0], guid->guid.Data4[1], guid->guid.Data4[2], guid->guid.Data4[3],
501 guid->guid.Data4[4], guid->guid.Data4[5], guid->guid.Data4[6], guid->guid.Data4[7]);
502
503 hash_key = ctl2_hash_guid(&guid->guid);
504
505 offset = ctl2_find_guid(typelib, hash_key, &guid->guid);
506 if (offset != -1)
507 {
508 if (is_warning_enabled(2368))
509 warning("duplicate uuid {%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}\n",
510 guid->guid.Data1, guid->guid.Data2, guid->guid.Data3,
511 guid->guid.Data4[0], guid->guid.Data4[1], guid->guid.Data4[2], guid->guid.Data4[3],
512 guid->guid.Data4[4], guid->guid.Data4[5], guid->guid.Data4[6], guid->guid.Data4[7]);
513 return -1;
514 }
515
516 offset = ctl2_alloc_segment(typelib, MSFT_SEG_GUID, sizeof(MSFT_GuidEntry), 0);
517
518 guid_space = (void *)(typelib->typelib_segment_data[MSFT_SEG_GUID] + offset);
519 *guid_space = *guid;
520
521 guid_space->next_hash = typelib->typelib_guidhash_segment[hash_key];
522 typelib->typelib_guidhash_segment[hash_key] = offset;
523
524 return offset;
525 }
526
527 /****************************************************************************
528 * ctl2_alloc_name
529 *
530 * Allocates and initializes a name within a type library. Also updates the
531 * name hash table as needed.
532 *
533 * RETURNS
534 *
535 * Success: The offset within the segment of the new name.
536 * Failure: -1 (this is invariably an out of memory condition).
537 */
538 static int ctl2_alloc_name(
539 msft_typelib_t *typelib, /* [I] The type library to allocate in. */
540 const char *name) /* [I] The name to store. */
541 {
542 int length;
543 int offset;
544 MSFT_NameIntro *name_space;
545 char *encoded_name;
546
547 length = ctl2_encode_name(typelib, name, &encoded_name);
548
549 offset = ctl2_find_name(typelib, encoded_name);
550 if (offset != -1) return offset;
551
552 offset = ctl2_alloc_segment(typelib, MSFT_SEG_NAME, length + 8, 0);
553
554 name_space = (void *)(typelib->typelib_segment_data[MSFT_SEG_NAME] + offset);
555 name_space->hreftype = -1;
556 name_space->next_hash = -1;
557 memcpy(&name_space->namelen, encoded_name, length);
558
559 if (typelib->typelib_namehash_segment[encoded_name[2] & 0x7f] != -1)
560 name_space->next_hash = typelib->typelib_namehash_segment[encoded_name[2] & 0x7f];
561
562 typelib->typelib_namehash_segment[encoded_name[2] & 0x7f] = offset;
563
564 typelib->typelib_header.nametablecount += 1;
565 typelib->typelib_header.nametablechars += *encoded_name;
566
567 return offset;
568 }
569
570 /****************************************************************************
571 * ctl2_alloc_string
572 *
573 * Allocates and initializes a string in a type library.
574 *
575 * RETURNS
576 *
577 * Success: The offset within the segment of the new string.
578 * Failure: -1 (this is invariably an out of memory condition).
579 */
580 static int ctl2_alloc_string(
581 msft_typelib_t *typelib, /* [I] The type library to allocate in. */
582 const char *string) /* [I] The string to store. */
583 {
584 int length;
585 int offset;
586 unsigned char *string_space;
587 char *encoded_string;
588
589 length = ctl2_encode_string(string, &encoded_string);
590
591 for (offset = 0; offset < typelib->typelib_segdir[MSFT_SEG_STRING].length;
592 offset += (((typelib->typelib_segment_data[MSFT_SEG_STRING][offset + 1] << 8) |
593 typelib->typelib_segment_data[MSFT_SEG_STRING][offset + 0]) + 5) & ~3) {
594 if (!memcmp(encoded_string, typelib->typelib_segment_data[MSFT_SEG_STRING] + offset, length)) return offset;
595 }
596
597 offset = ctl2_alloc_segment(typelib, MSFT_SEG_STRING, length, 0);
598
599 string_space = typelib->typelib_segment_data[MSFT_SEG_STRING] + offset;
600 memcpy(string_space, encoded_string, length);
601
602 return offset;
603 }
604
605 /****************************************************************************
606 * alloc_msft_importinfo
607 *
608 * Allocates and initializes an import information structure in a type library.
609 *
610 * RETURNS
611 *
612 * Success: The offset of the new importinfo.
613 * Failure: -1 (this is invariably an out of memory condition).
614 */
615 static int alloc_msft_importinfo(
616 msft_typelib_t *typelib, /* [I] The type library to allocate in. */
617 MSFT_ImpInfo *impinfo) /* [I] The import information to store. */
618 {
619 int offset;
620 MSFT_ImpInfo *impinfo_space;
621
622 for (offset = 0;
623 offset < typelib->typelib_segdir[MSFT_SEG_IMPORTINFO].length;
624 offset += sizeof(MSFT_ImpInfo)) {
625 if (!memcmp(&(typelib->typelib_segment_data[MSFT_SEG_IMPORTINFO][offset]),
626 impinfo, sizeof(MSFT_ImpInfo))) {
627 return offset;
628 }
629 }
630
631 impinfo->flags |= typelib->typelib_header.nimpinfos++;
632
633 offset = ctl2_alloc_segment(typelib, MSFT_SEG_IMPORTINFO, sizeof(MSFT_ImpInfo), 0);
634
635 impinfo_space = (void *)(typelib->typelib_segment_data[MSFT_SEG_IMPORTINFO] + offset);
636 *impinfo_space = *impinfo;
637
638 return offset;
639 }
640
641 /****************************************************************************
642 * alloc_importfile
643 *
644 * Allocates and initializes an import file definition in a type library.
645 *
646 * RETURNS
647 *
648 * Success: The offset of the new importinfo.
649 * Failure: -1 (this is invariably an out of memory condition).
650 */
651 static int alloc_importfile(
652 msft_typelib_t *typelib, /* [I] The type library to allocate in. */
653 int guidoffset, /* [I] The offset to the GUID for the imported library. */
654 int major_version, /* [I] The major version number of the imported library. */
655 int minor_version, /* [I] The minor version number of the imported library. */
656 const char *filename) /* [I] The filename of the imported library. */
657 {
658 int length;
659 int offset;
660 MSFT_ImpFile *importfile;
661 char *encoded_string;
662
663 length = ctl2_encode_string(filename, &encoded_string);
664
665 encoded_string[0] <<= 2;
666 encoded_string[0] |= 1;
667
668 for (offset = 0; offset < typelib->typelib_segdir[MSFT_SEG_IMPORTFILES].length;
669 offset += (((typelib->typelib_segment_data[MSFT_SEG_IMPORTFILES][offset + 0xd] << 8) |
670 typelib->typelib_segment_data[MSFT_SEG_IMPORTFILES][offset + 0xc]) >> 2) + 0xc) {
671 if (!memcmp(encoded_string, typelib->typelib_segment_data[MSFT_SEG_IMPORTFILES] + offset + 0xc, length)) return offset;
672 }
673
674 offset = ctl2_alloc_segment(typelib, MSFT_SEG_IMPORTFILES, length + 0xc, 0);
675
676 importfile = (MSFT_ImpFile *)&typelib->typelib_segment_data[MSFT_SEG_IMPORTFILES][offset];
677 importfile->guid = guidoffset;
678 importfile->lcid = typelib->typelib_header.lcid2;
679 importfile->version = major_version | (minor_version << 16);
680 memcpy(&importfile->filename, encoded_string, length);
681
682 return offset;
683 }
684
685 static void alloc_importinfo(msft_typelib_t *typelib, importinfo_t *importinfo)
686 {
687 importlib_t *importlib = importinfo->importlib;
688
689 chat("alloc_importinfo: %s\n", importinfo->name);
690
691 if(!importlib->allocated) {
692 MSFT_GuidEntry guid;
693 int guid_idx;
694
695 chat("allocating importlib %s\n", importlib->name);
696
697 importlib->allocated = -1;
698
699 memcpy(&guid.guid, &importlib->guid, sizeof(GUID));
700 guid.hreftype = 2;
701
702 guid_idx = ctl2_alloc_guid(typelib, &guid);
703
704 alloc_importfile(typelib, guid_idx, importlib->version&0xffff,
705 importlib->version>>16, importlib->name);
706 }
707
708 if(importinfo->offset == -1 || !(importinfo->flags & MSFT_IMPINFO_OFFSET_IS_GUID)) {
709 MSFT_ImpInfo impinfo;
710
711 impinfo.flags = importinfo->flags;
712 impinfo.oImpFile = 0;
713
714 if(importinfo->flags & MSFT_IMPINFO_OFFSET_IS_GUID) {
715 MSFT_GuidEntry guid;
716
717 guid.hreftype = 0;
718 memcpy(&guid.guid, &importinfo->guid, sizeof(GUID));
719
720 impinfo.oGuid = ctl2_alloc_guid(typelib, &guid);
721
722 importinfo->offset = alloc_msft_importinfo(typelib, &impinfo);
723
724 typelib->typelib_segment_data[MSFT_SEG_GUID][impinfo.oGuid+sizeof(GUID)]
725 = importinfo->offset+1;
726
727 if(!strcmp(importinfo->name, "IDispatch"))
728 typelib->typelib_header.dispatchpos = importinfo->offset+1;
729 }else {
730 impinfo.oGuid = importinfo->id;
731 importinfo->offset = alloc_msft_importinfo(typelib, &impinfo);
732 }
733 }
734 }
735
736 static importinfo_t *find_importinfo(msft_typelib_t *typelib, const char *name)
737 {
738 importlib_t *importlib;
739 int i;
740
741 chat("search importlib %s\n", name);
742
743 if(!name)
744 return NULL;
745
746 LIST_FOR_EACH_ENTRY( importlib, &typelib->typelib->importlibs, importlib_t, entry )
747 {
748 for(i=0; i < importlib->ntypeinfos; i++) {
749 if(!strcmp(name, importlib->importinfos[i].name)) {
750 chat("Found %s in importlib.\n", name);
751 return importlib->importinfos+i;
752 }
753 }
754 }
755
756 return NULL;
757 }
758
759 static void add_structure_typeinfo(msft_typelib_t *typelib, type_t *structure);
760 static void add_interface_typeinfo(msft_typelib_t *typelib, type_t *interface);
761 static void add_enum_typeinfo(msft_typelib_t *typelib, type_t *enumeration);
762 static void add_union_typeinfo(msft_typelib_t *typelib, type_t *tunion);
763 static void add_coclass_typeinfo(msft_typelib_t *typelib, type_t *cls);
764 static void add_dispinterface_typeinfo(msft_typelib_t *typelib, type_t *dispinterface);
765
766
767 /****************************************************************************
768 * encode_type
769 *
770 * Encodes a type, storing information in the TYPEDESC and ARRAYDESC
771 * segments as needed.
772 *
773 * RETURNS
774 *
775 * Success: 0.
776 * Failure: -1.
777 */
778 static int encode_type(
779 msft_typelib_t *typelib, /* [I] The type library in which to encode the TYPEDESC. */
780 int vt, /* [I] vt to encode */
781 type_t *type, /* [I] type */
782 int *encoded_type, /* [O] The encoded type description. */
783 int *width, /* [O] The width of the type, or NULL. */
784 int *alignment, /* [O] The alignment of the type, or NULL. */
785 int *decoded_size) /* [O] The total size of the unencoded TYPEDESCs, including nested descs. */
786 {
787 int default_type;
788 int scratch;
789 int typeoffset;
790 int *typedata;
791 int target_type;
792 int child_size = 0;
793
794 chat("encode_type vt %d type %p\n", vt, type);
795
796 default_type = 0x80000000 | (vt << 16) | vt;
797 if (!width) width = &scratch;
798 if (!alignment) alignment = &scratch;
799 if (!decoded_size) decoded_size = &scratch;
800
801 *decoded_size = 0;
802
803 switch (vt) {
804 case VT_I1:
805 case VT_UI1:
806 *encoded_type = default_type;
807 *width = 1;
808 *alignment = 1;
809 break;
810
811 case VT_INT:
812 *encoded_type = 0x80000000 | (VT_I4 << 16) | VT_INT;
813 if ((typelib->typelib_header.varflags & 0x0f) == SYS_WIN16) {
814 *width = 2;
815 *alignment = 2;
816 } else {
817 *width = 4;
818 *alignment = 4;
819 }
820 break;
821
822 case VT_UINT:
823 *encoded_type = 0x80000000 | (VT_UI4 << 16) | VT_UINT;
824 if ((typelib->typelib_header.varflags & 0x0f) == SYS_WIN16) {
825 *width = 2;
826 *alignment = 2;
827 } else {
828 *width = 4;
829 *alignment = 4;
830 }
831 break;
832
833 case VT_UI2:
834 case VT_I2:
835 case VT_BOOL:
836 *encoded_type = default_type;
837 *width = 2;
838 *alignment = 2;
839 break;
840
841 case VT_I4:
842 case VT_UI4:
843 case VT_R4:
844 case VT_ERROR:
845 case VT_HRESULT:
846 *encoded_type = default_type;
847 *width = 4;
848 *alignment = 4;
849 break;
850
851 case VT_R8:
852 case VT_I8:
853 case VT_UI8:
854 *encoded_type = default_type;
855 *width = 8;
856 *alignment = 8;
857 break;
858
859 case VT_CY:
860 case VT_DATE:
861 *encoded_type = default_type;
862 *width = 8;
863 *alignment = 8;
864 break;
865
866 case VT_DECIMAL:
867 *encoded_type = default_type;
868 *width = 16;
869 *alignment = 8;
870 break;
871
872 case VT_VOID:
873 *encoded_type = 0x80000000 | (VT_EMPTY << 16) | vt;
874 *width = 0;
875 *alignment = 1;
876 break;
877
878 case VT_UNKNOWN:
879 case VT_DISPATCH:
880 case VT_BSTR:
881 *encoded_type = default_type;
882 *width = pointer_size;
883 *alignment = 4;
884 break;
885
886 case VT_VARIANT:
887 *encoded_type = default_type;
888 *width = 8 + 2 * pointer_size;
889 *alignment = 8;
890 break;
891
892 case VT_LPSTR:
893 case VT_LPWSTR:
894 *encoded_type = 0xfffe0000 | vt;
895 *width = pointer_size;
896 *alignment = 4;
897 break;
898
899 case VT_PTR:
900 {
901 int next_vt;
902 for(next_vt = 0; is_ptr(type); type = type_pointer_get_ref(type)) {
903 next_vt = get_type_vt(type_pointer_get_ref(type));
904 if (next_vt != 0)
905 break;
906 }
907 /* if no type found then it must be void */
908 if (next_vt == 0)
909 next_vt = VT_VOID;
910
911 encode_type(typelib, next_vt, type_pointer_get_ref(type),
912 &target_type, NULL, NULL, &child_size);
913 /* these types already have an implicit pointer, so we don't need to
914 * add another */
915 if(next_vt == VT_DISPATCH || next_vt == VT_UNKNOWN) {
916 chat("encode_type: skipping ptr\n");
917 *encoded_type = target_type;
918 *width = pointer_size;
919 *alignment = 4;
920 *decoded_size = child_size;
921 break;
922 }
923
924 for (typeoffset = 0; typeoffset < typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length; typeoffset += 8) {
925 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
926 if (((typedata[0] & 0xffff) == VT_PTR) && (typedata[1] == target_type)) break;
927 }
928
929 if (typeoffset == typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length) {
930 int mix_field;
931
932 if (target_type & 0x80000000) {
933 mix_field = ((target_type >> 16) & 0x3fff) | VT_BYREF;
934 } else {
935 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][target_type];
936 mix_field = ((typedata[0] >> 16) == 0x7fff)? 0x7fff: 0x7ffe;
937 }
938
939 typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
940 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
941
942 typedata[0] = (mix_field << 16) | VT_PTR;
943 typedata[1] = target_type;
944 }
945
946 *encoded_type = typeoffset;
947
948 *width = pointer_size;
949 *alignment = 4;
950 *decoded_size = 8 /*sizeof(TYPEDESC)*/ + child_size;
951 break;
952 }
953
954 case VT_SAFEARRAY:
955 {
956 type_t *element_type = type_alias_get_aliasee(type_array_get_element(type));
957 int next_vt = get_type_vt(element_type);
958
959 encode_type(typelib, next_vt, type_alias_get_aliasee(type_array_get_element(type)), &target_type, NULL, NULL, &child_size);
960
961 for (typeoffset = 0; typeoffset < typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length; typeoffset += 8) {
962 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
963 if (((typedata[0] & 0xffff) == VT_SAFEARRAY) && (typedata[1] == target_type)) break;
964 }
965
966 if (typeoffset == typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length) {
967 int mix_field;
968
969 if (target_type & 0x80000000) {
970 mix_field = ((target_type >> 16) & VT_TYPEMASK) | VT_ARRAY;
971 } else {
972 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][target_type];
973 mix_field = ((typedata[0] >> 16) == 0x7fff)? 0x7fff: 0x7ffe;
974 }
975
976 typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
977 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
978
979 typedata[0] = (mix_field << 16) | VT_SAFEARRAY;
980 typedata[1] = target_type;
981 }
982
983 *encoded_type = typeoffset;
984
985 *width = pointer_size;
986 *alignment = 4;
987 *decoded_size = 8 /*sizeof(TYPEDESC)*/ + child_size;
988 break;
989 }
990
991 case VT_USERDEFINED:
992 {
993 int typeinfo_offset;
994
995 /* typedef'd types without public attribute aren't included in the typelib */
996 while (type->typelib_idx < 0 && type_is_alias(type) && !is_attr(type->attrs, ATTR_PUBLIC))
997 type = type_alias_get_aliasee(type);
998
999 chat("encode_type: VT_USERDEFINED - type %p name = %s real type %d idx %d\n", type,
1000 type->name, type_get_type(type), type->typelib_idx);
1001
1002 if(type->typelib_idx == -1) {
1003 chat("encode_type: trying to ref not added type\n");
1004 switch (type_get_type(type)) {
1005 case TYPE_STRUCT:
1006 add_structure_typeinfo(typelib, type);
1007 break;
1008 case TYPE_INTERFACE:
1009 add_interface_typeinfo(typelib, type);
1010 break;
1011 case TYPE_ENUM:
1012 add_enum_typeinfo(typelib, type);
1013 break;
1014 case TYPE_UNION:
1015 add_union_typeinfo(typelib, type);
1016 break;
1017 case TYPE_COCLASS:
1018 add_coclass_typeinfo(typelib, type);
1019 break;
1020 default:
1021 error("encode_type: VT_USERDEFINED - unhandled type %d\n",
1022 type_get_type(type));
1023 }
1024 }
1025
1026 typeinfo_offset = typelib->typelib_typeinfo_offsets[type->typelib_idx];
1027 for (typeoffset = 0; typeoffset < typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length; typeoffset += 8) {
1028 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1029 if ((typedata[0] == ((0x7fff << 16) | VT_USERDEFINED)) && (typedata[1] == typeinfo_offset)) break;
1030 }
1031
1032 if (typeoffset == typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length) {
1033 typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
1034 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1035
1036 typedata[0] = (0x7fff << 16) | VT_USERDEFINED;
1037 typedata[1] = typeinfo_offset;
1038 }
1039
1040 *encoded_type = typeoffset;
1041 *width = 0;
1042 *alignment = 1;
1043 break;
1044 }
1045
1046 default:
1047 error("encode_type: unrecognized type %d.\n", vt);
1048 *encoded_type = default_type;
1049 *width = 0;
1050 *alignment = 1;
1051 break;
1052 }
1053
1054 return 0;
1055 }
1056
1057 static void dump_type(type_t *t)
1058 {
1059 chat("dump_type: %p name %s type %d attrs %p\n", t, t->name, type_get_type(t), t->attrs);
1060 }
1061
1062 static int encode_var(
1063 msft_typelib_t *typelib, /* [I] The type library in which to encode the TYPEDESC. */
1064 type_t *type, /* [I] The type description to encode. */
1065 var_t *var, /* [I] The var to encode. */
1066 int *encoded_type, /* [O] The encoded type description. */
1067 int *width, /* [O] The width of the type, or NULL. */
1068 int *alignment, /* [O] The alignment of the type, or NULL. */
1069 int *decoded_size) /* [O] The total size of the unencoded TYPEDESCs, including nested descs. */
1070 {
1071 int typeoffset;
1072 int *typedata;
1073 int target_type;
1074 int child_size;
1075 int vt;
1076 int scratch;
1077
1078 if (!width) width = &scratch;
1079 if (!alignment) alignment = &scratch;
1080 if (!decoded_size) decoded_size = &scratch;
1081 *decoded_size = 0;
1082
1083 chat("encode_var: var %p type %p type->name %s\n",
1084 var, type, type->name ? type->name : "NULL");
1085
1086 if (is_array(type) && !type_array_is_decl_as_ptr(type)) {
1087 int num_dims, elements = 1, arrayoffset;
1088 type_t *atype;
1089 int *arraydata;
1090
1091 num_dims = 0;
1092 for (atype = type;
1093 is_array(atype) && !type_array_is_decl_as_ptr(atype);
1094 atype = type_array_get_element(atype))
1095 ++num_dims;
1096
1097 chat("array with %d dimensions\n", num_dims);
1098 encode_var(typelib, atype, var, &target_type, width, alignment, NULL);
1099 arrayoffset = ctl2_alloc_segment(typelib, MSFT_SEG_ARRAYDESC, (2 + 2 * num_dims) * sizeof(int), 0);
1100 arraydata = (void *)&typelib->typelib_segment_data[MSFT_SEG_ARRAYDESC][arrayoffset];
1101
1102 arraydata[0] = target_type;
1103 arraydata[1] = num_dims;
1104 arraydata[1] |= ((num_dims * 2 * sizeof(int)) << 16);
1105
1106 arraydata += 2;
1107 for (atype = type;
1108 is_array(atype) && !type_array_is_decl_as_ptr(atype);
1109 atype = type_array_get_element(atype))
1110 {
1111 arraydata[0] = type_array_get_dim(atype);
1112 arraydata[1] = 0;
1113 arraydata += 2;
1114 elements *= type_array_get_dim(atype);
1115 }
1116
1117 typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
1118 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1119
1120 typedata[0] = (0x7ffe << 16) | VT_CARRAY;
1121 typedata[1] = arrayoffset;
1122
1123 *encoded_type = typeoffset;
1124 *width = *width * elements;
1125 *decoded_size = 20 /*sizeof(ARRAYDESC)*/ + (num_dims - 1) * 8 /*sizeof(SAFEARRAYBOUND)*/;
1126 return 0;
1127 }
1128
1129 vt = get_type_vt(type);
1130 if (vt == VT_PTR) {
1131 type_t *ref = is_ptr(type) ?
1132 type_pointer_get_ref(type) : type_array_get_element(type);
1133 int skip_ptr = encode_var(typelib, ref, var,
1134 &target_type, NULL, NULL, &child_size);
1135
1136 if(skip_ptr == 2) {
1137 chat("encode_var: skipping ptr\n");
1138 *encoded_type = target_type;
1139 *decoded_size = child_size;
1140 *width = pointer_size;
1141 *alignment = 4;
1142 return 0;
1143 }
1144
1145 for (typeoffset = 0; typeoffset < typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length; typeoffset += 8) {
1146 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1147 if (((typedata[0] & 0xffff) == VT_PTR) && (typedata[1] == target_type)) break;
1148 }
1149
1150 if (typeoffset == typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length) {
1151 int mix_field;
1152
1153 if (target_type & 0x80000000) {
1154 mix_field = ((target_type >> 16) & 0x3fff) | VT_BYREF;
1155 } else if (is_array(ref)) {
1156 type_t *element_type = type_alias_get_aliasee(type_array_get_element(ref));
1157 mix_field = get_type_vt(element_type) | VT_ARRAY | VT_BYREF;
1158 } else {
1159 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][target_type];
1160 mix_field = ((typedata[0] >> 16) == 0x7fff)? 0x7fff: 0x7ffe;
1161 }
1162
1163 typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
1164 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1165
1166 typedata[0] = (mix_field << 16) | VT_PTR;
1167 typedata[1] = target_type;
1168 }
1169
1170 *encoded_type = typeoffset;
1171
1172 *width = pointer_size;
1173 *alignment = 4;
1174 *decoded_size = 8 /*sizeof(TYPEDESC)*/ + child_size;
1175 return 0;
1176 }
1177
1178 dump_type(type);
1179
1180 encode_type(typelib, vt, type, encoded_type, width, alignment, decoded_size);
1181 /* these types already have an implicit pointer, so we don't need to
1182 * add another */
1183 if(vt == VT_DISPATCH || vt == VT_UNKNOWN) return 2;
1184 return 0;
1185 }
1186
1187 static unsigned int get_ulong_val(unsigned int val, int vt)
1188 {
1189 switch(vt) {
1190 case VT_I2:
1191 case VT_BOOL:
1192 case VT_UI2:
1193 return val & 0xffff;
1194 case VT_I1:
1195 case VT_UI1:
1196 return val & 0xff;
1197 }
1198
1199 return val;
1200 }
1201
1202 static void write_value(msft_typelib_t* typelib, int *out, int vt, const void *value)
1203 {
1204 switch(vt) {
1205 case VT_I2:
1206 case VT_I4:
1207 case VT_R4:
1208 case VT_BOOL:
1209 case VT_I1:
1210 case VT_UI1:
1211 case VT_UI2:
1212 case VT_UI4:
1213 case VT_INT:
1214 case VT_UINT:
1215 case VT_HRESULT:
1216 case VT_PTR:
1217 case VT_UNKNOWN:
1218 case VT_DISPATCH:
1219 {
1220 const unsigned int lv = get_ulong_val(*(const unsigned int *)value, vt);
1221 if((lv & 0x3ffffff) == lv) {
1222 *out = 0x80000000;
1223 *out |= vt << 26;
1224 *out |= lv;
1225 } else {
1226 int offset = ctl2_alloc_segment(typelib, MSFT_SEG_CUSTDATA, 8, 0);
1227 *((unsigned short *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset]) = vt;
1228 memcpy(&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+2], value, 4);
1229 *((unsigned short *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+6]) = 0x5757;
1230 *out = offset;
1231 }
1232 return;
1233 }
1234 case VT_BSTR:
1235 {
1236 const char *s = (const char *) value;
1237 int len = strlen(s), seg_len = (len + 6 + 3) & ~0x3;
1238 int offset = ctl2_alloc_segment(typelib, MSFT_SEG_CUSTDATA, seg_len, 0);
1239 *((unsigned short *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset]) = vt;
1240 memcpy(&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+2], &len, sizeof(len));
1241 memcpy(&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+6], value, len);
1242 len += 6;
1243 while(len < seg_len) {
1244 *((char *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+len]) = 0x57;
1245 len++;
1246 }
1247 *out = offset;
1248 return;
1249 }
1250
1251 default:
1252 warning("can't write value of type %d yet\n", vt);
1253 }
1254 return;
1255 }
1256
1257 static HRESULT set_custdata(msft_typelib_t *typelib, REFGUID guid,
1258 int vt, const void *value, int *offset)
1259 {
1260 MSFT_GuidEntry guidentry;
1261 int guidoffset;
1262 int custoffset;
1263 int *custdata;
1264 int data_out;
1265
1266 guidentry.guid = *guid;
1267
1268 guidentry.hreftype = -1;
1269 guidentry.next_hash = -1;
1270
1271 guidoffset = ctl2_alloc_guid(typelib, &guidentry);
1272 write_value(typelib, &data_out, vt, value);
1273
1274 custoffset = ctl2_alloc_segment(typelib, MSFT_SEG_CUSTDATAGUID, 12, 0);
1275
1276 custdata = (int *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATAGUID][custoffset];
1277 custdata[0] = guidoffset;
1278 custdata[1] = data_out;
1279 custdata[2] = *offset;
1280 *offset = custoffset;
1281
1282 return S_OK;
1283 }
1284
1285 /* It's possible to have a default value for pointer arguments too.
1286 In this case default value has a referenced type, e.g.
1287 'LONG*' argument gets VT_I4, 'DOUBLE*' - VT_R8. IUnknown* and IDispatch*
1288 are recognised too and stored as VT_UNKNOWN and VT_DISPATCH.
1289 But IUnknown/IDispatch arguments can only have default value of 0
1290 (or expression that resolves to zero) while other pointers can have
1291 any default value. */
1292 static int get_defaultvalue_vt(type_t *type)
1293 {
1294 int vt;
1295 if (type_get_type(type) == TYPE_ENUM)
1296 vt = VT_I4;
1297 else
1298 {
1299 vt = get_type_vt(type);
1300 if (vt == VT_PTR && is_ptr(type)) {
1301 vt = get_type_vt(type_pointer_get_ref(type));
1302 /* The only acceptable value for pointers to non-basic types
1303 is NULL, it's stored as VT_I4 for both 32 and 64 bit typelibs. */
1304 if (vt == VT_USERDEFINED)
1305 vt = VT_I4;
1306 }
1307 }
1308
1309 return vt;
1310 }
1311
1312 static HRESULT add_func_desc(msft_typeinfo_t* typeinfo, var_t *func, int index)
1313 {
1314 int offset, name_offset;
1315 int *typedata, typedata_size;
1316 int i, id, next_idx;
1317 int decoded_size, extra_attr = 0;
1318 int num_params = 0, num_optional = 0, num_defaults = 0;
1319 var_t *arg;
1320 unsigned char *namedata;
1321 const attr_t *attr;
1322 unsigned int funcflags = 0, callconv = 4 /* CC_STDCALL */;
1323 unsigned int funckind, invokekind = 1 /* INVOKE_FUNC */;
1324 int help_context = 0, help_string_context = 0, help_string_offset = -1;
1325 int entry = -1, entry_is_ord = 0;
1326 int lcid_retval_count = 0;
1327
1328 chat("add_func_desc(%p,%d)\n", typeinfo, index);
1329
1330 id = ((0x6000 | (typeinfo->typeinfo->datatype2 & 0xffff)) << 16) | index;
1331
1332 switch(typeinfo->typekind) {
1333 case TKIND_DISPATCH:
1334 funckind = 0x4; /* FUNC_DISPATCH */
1335 break;
1336 case TKIND_MODULE:
1337 funckind = 0x3; /* FUNC_STATIC */
1338 break;
1339 default:
1340 funckind = 0x1; /* FUNC_PUREVIRTUAL */
1341 break;
1342 }
1343
1344 if (is_local( func->attrs )) {
1345 chat("add_func_desc: skipping local function\n");
1346 return S_FALSE;
1347 }
1348
1349 if (type_get_function_args(func->type))
1350 LIST_FOR_EACH_ENTRY( arg, type_get_function_args(func->type), var_t, entry )
1351 {
1352 num_params++;
1353 if (arg->attrs) LIST_FOR_EACH_ENTRY( attr, arg->attrs, const attr_t, entry ) {
1354 if(attr->type == ATTR_DEFAULTVALUE)
1355 num_defaults++;
1356 else if(attr->type == ATTR_OPTIONAL)
1357 num_optional++;
1358 }
1359 }
1360
1361 chat("add_func_desc: num of params %d\n", num_params);
1362
1363 name_offset = ctl2_alloc_name(typeinfo->typelib, func->name);
1364
1365 if (func->attrs) LIST_FOR_EACH_ENTRY( attr, func->attrs, const attr_t, entry ) {
1366 expr_t *expr = attr->u.pval;
1367 switch(attr->type) {
1368 case ATTR_BINDABLE:
1369 funcflags |= 0x4; /* FUNCFLAG_FBINDABLE */
1370 break;
1371 case ATTR_DEFAULTBIND:
1372 funcflags |= 0x20; /* FUNCFLAG_FDEFAULTBIND */
1373 break;
1374 case ATTR_DEFAULTCOLLELEM:
1375 funcflags |= 0x100; /* FUNCFLAG_FDEFAULTCOLLELEM */
1376 break;
1377 case ATTR_DISPLAYBIND:
1378 funcflags |= 0x10; /* FUNCFLAG_FDISPLAYBIND */
1379 break;
1380 case ATTR_ENTRY:
1381 extra_attr = max(extra_attr, 3);
1382 if (expr->type == EXPR_STRLIT || expr->type == EXPR_WSTRLIT)
1383 entry = ctl2_alloc_string(typeinfo->typelib, attr->u.pval);
1384 else {
1385 entry = expr->cval;
1386 entry_is_ord = 1;
1387 }
1388 break;
1389 case ATTR_HELPCONTEXT:
1390 extra_attr = max(extra_attr, 1);
1391 help_context = expr->u.lval;
1392 break;
1393 case ATTR_HELPSTRING:
1394 extra_attr = max(extra_attr, 2);
1395 help_string_offset = ctl2_alloc_string(typeinfo->typelib, attr->u.pval);
1396 break;
1397 case ATTR_HELPSTRINGCONTEXT:
1398 extra_attr = max(extra_attr, 6);
1399 help_string_context = expr->u.lval;
1400 break;
1401 case ATTR_HIDDEN:
1402 funcflags |= 0x40; /* FUNCFLAG_FHIDDEN */
1403 break;
1404 case ATTR_ID:
1405 id = expr->cval;
1406 break;
1407 case ATTR_IMMEDIATEBIND:
1408 funcflags |= 0x1000; /* FUNCFLAG_FIMMEDIATEBIND */
1409 break;
1410 case ATTR_NONBROWSABLE:
1411 funcflags |= 0x400; /* FUNCFLAG_FNONBROWSABLE */
1412 break;
1413 case ATTR_OUT:
1414 break;
1415 case ATTR_PROPGET:
1416 invokekind = 0x2; /* INVOKE_PROPERTYGET */
1417 break;
1418 case ATTR_PROPPUT:
1419 invokekind = 0x4; /* INVOKE_PROPERTYPUT */
1420 break;
1421 case ATTR_PROPPUTREF:
1422 invokekind = 0x8; /* INVOKE_PROPERTYPUTREF */
1423 break;
1424 /* FIXME: FUNCFLAG_FREPLACEABLE */
1425 case ATTR_REQUESTEDIT:
1426 funcflags |= 0x8; /* FUNCFLAG_FREQUESTEDIT */
1427 break;
1428 case ATTR_RESTRICTED:
1429 funcflags |= 0x1; /* FUNCFLAG_FRESTRICTED */
1430 break;
1431 case ATTR_SOURCE:
1432 funcflags |= 0x2; /* FUNCFLAG_FSOURCE */
1433 break;
1434 case ATTR_UIDEFAULT:
1435 funcflags |= 0x200; /* FUNCFLAG_FUIDEFAULT */
1436 break;
1437 case ATTR_USESGETLASTERROR:
1438 funcflags |= 0x80; /* FUNCFLAG_FUSESGETLASTERROR */
1439 break;
1440 case ATTR_VARARG:
1441 if (num_optional || num_defaults)
1442 warning("add_func_desc: ignoring vararg in function with optional or defaultvalue params\n");
1443 else
1444 num_optional = -1;
1445 break;
1446 default:
1447 break;
1448 }
1449 }
1450
1451 /* allocate type data space for us */
1452 typedata_size = 0x18 + extra_attr * sizeof(int) + (num_params * (num_defaults ? 16 : 12));
1453
1454 if (!typeinfo->func_data) {
1455 typeinfo->func_data = xmalloc(0x100);
1456 typeinfo->func_data_allocated = 0x100;
1457 typeinfo->func_data[0] = 0;
1458 }
1459
1460 if(typeinfo->func_data[0] + typedata_size + sizeof(int) > typeinfo->func_data_allocated) {
1461 typeinfo->func_data_allocated = max(typeinfo->func_data_allocated * 2,
1462 typeinfo->func_data[0] + typedata_size + sizeof(int));
1463 typeinfo->func_data = xrealloc(typeinfo->func_data, typeinfo->func_data_allocated);
1464 }
1465
1466 offset = typeinfo->func_data[0];
1467 typeinfo->func_data[0] += typedata_size;
1468 typedata = typeinfo->func_data + (offset >> 2) + 1;
1469
1470
1471 /* find func with the same name - if it exists use its id */
1472 for(i = 0; i < (typeinfo->typeinfo->cElement & 0xffff); i++) {
1473 if(name_offset == typeinfo->func_names[i]) {
1474 id = typeinfo->func_indices[i];
1475 break;
1476 }
1477 }
1478
1479 /* find the first func with the same id and link via the hiword of typedata[4] */
1480 next_idx = index;
1481 for(i = 0; i < (typeinfo->typeinfo->cElement & 0xffff); i++) {
1482 if(id == typeinfo->func_indices[i]) {
1483 next_idx = typeinfo->func_data[(typeinfo->func_offsets[i] >> 2) + 1 + 4] >> 16;
1484 typeinfo->func_data[(typeinfo->func_offsets[i] >> 2) + 1 + 4] &= 0xffff;
1485 typeinfo->func_data[(typeinfo->func_offsets[i] >> 2) + 1 + 4] |= (index << 16);
1486 break;
1487 }
1488 }
1489
1490 /* fill out the basic type information */
1491 typedata[0] = typedata_size | (index << 16);
1492 encode_var(typeinfo->typelib, type_function_get_rettype(func->type), func, &typedata[1], NULL, NULL, &decoded_size);
1493 typedata[2] = funcflags;
1494 typedata[3] = ((52 /*sizeof(FUNCDESC)*/ + decoded_size) << 16) | typeinfo->typeinfo->cbSizeVft;
1495 typedata[4] = (next_idx << 16) | (callconv << 8) | (invokekind << 3) | funckind;
1496 if(num_defaults) typedata[4] |= 0x1000;
1497 if(entry_is_ord) typedata[4] |= 0x2000;
1498 typedata[5] = (num_optional << 16) | num_params;
1499
1500 /* NOTE: High word of typedata[3] is total size of FUNCDESC + size of all ELEMDESCs for params + TYPEDESCs for pointer params and return types. */
1501 /* That is, total memory allocation required to reconstitute the FUNCDESC in its entirety. */
1502 typedata[3] += (16 /*sizeof(ELEMDESC)*/ * num_params) << 16;
1503 typedata[3] += (24 /*sizeof(PARAMDESCEX)*/ * num_defaults) << 16;
1504
1505 switch(extra_attr) {
1506 case 6: typedata[11] = help_string_context;
1507 case 5: typedata[10] = -1;
1508 case 4: typedata[9] = -1;
1509 case 3: typedata[8] = entry;
1510 case 2: typedata[7] = help_string_offset;
1511 case 1: typedata[6] = help_context;
1512 case 0:
1513 break;
1514 default:
1515 warning("unknown number of optional attrs\n");
1516 }
1517
1518 if (type_get_function_args(func->type))
1519 {
1520 i = 0;
1521 LIST_FOR_EACH_ENTRY( arg, type_get_function_args(func->type), var_t, entry )
1522 {
1523 int paramflags = 0;
1524 int *paramdata = typedata + 6 + extra_attr + (num_defaults ? num_params : 0) + i * 3;
1525 int *defaultdata = num_defaults ? typedata + 6 + extra_attr + i : NULL;
1526
1527 if(defaultdata) *defaultdata = -1;
1528
1529 encode_var(typeinfo->typelib, arg->type, arg, paramdata, NULL, NULL, &decoded_size);
1530 if (arg->attrs) LIST_FOR_EACH_ENTRY( attr, arg->attrs, const attr_t, entry ) {
1531 switch(attr->type) {
1532 case ATTR_DEFAULTVALUE:
1533 {
1534 int vt;
1535 expr_t *expr = (expr_t *)attr->u.pval;
1536 vt = get_defaultvalue_vt(arg->type);
1537 paramflags |= 0x30; /* PARAMFLAG_FHASDEFAULT | PARAMFLAG_FOPT */
1538 if (expr->type == EXPR_STRLIT || expr->type == EXPR_WSTRLIT)
1539 {
1540 if (vt != VT_BSTR) error("string default value applied to non-string type\n");
1541 chat("default value '%s'\n", expr->u.sval);
1542 write_value(typeinfo->typelib, defaultdata, vt, expr->u.sval);
1543 }
1544 else
1545 {
1546 chat("default value %d\n", expr->cval);
1547 write_value(typeinfo->typelib, defaultdata, vt, &expr->cval);
1548 }
1549 break;
1550 }
1551 case ATTR_IN:
1552 paramflags |= 0x01; /* PARAMFLAG_FIN */
1553 break;
1554 case ATTR_OPTIONAL:
1555 paramflags |= 0x10; /* PARAMFLAG_FOPT */
1556 break;
1557 case ATTR_OUT:
1558 paramflags |= 0x02; /* PARAMFLAG_FOUT */
1559 break;
1560 case ATTR_PARAMLCID:
1561 paramflags |= 0x04; /* PARAMFLAG_LCID */
1562 lcid_retval_count++;
1563 break;
1564 case ATTR_RETVAL:
1565 paramflags |= 0x08; /* PARAMFLAG_FRETVAL */
1566 lcid_retval_count++;
1567 break;
1568 default:
1569 chat("unhandled param attr %d\n", attr->type);
1570 break;
1571 }
1572 }
1573 paramdata[1] = -1;
1574 paramdata[2] = paramflags;
1575 typedata[3] += decoded_size << 16;
1576
1577 i++;
1578 }
1579 }
1580
1581 if(lcid_retval_count == 1)
1582 typedata[4] |= 0x4000;
1583 else if(lcid_retval_count == 2)
1584 typedata[4] |= 0x8000;
1585
1586 if(typeinfo->funcs_allocated == 0) {
1587 typeinfo->funcs_allocated = 10;
1588 typeinfo->func_indices = xmalloc(typeinfo->funcs_allocated * sizeof(int));
1589 typeinfo->func_names = xmalloc(typeinfo->funcs_allocated * sizeof(int));
1590 typeinfo->func_offsets = xmalloc(typeinfo->funcs_allocated * sizeof(int));
1591 }
1592 if(typeinfo->funcs_allocated == (typeinfo->typeinfo->cElement & 0xffff)) {
1593 typeinfo->funcs_allocated *= 2;
1594 typeinfo->func_indices = xrealloc(typeinfo->func_indices, typeinfo->funcs_allocated * sizeof(int));
1595 typeinfo->func_names = xrealloc(typeinfo->func_names, typeinfo->funcs_allocated * sizeof(int));
1596 typeinfo->func_offsets = xrealloc(typeinfo->func_offsets, typeinfo->funcs_allocated * sizeof(int));
1597 }
1598
1599 /* update the index data */
1600 typeinfo->func_indices[typeinfo->typeinfo->cElement & 0xffff] = id;
1601 typeinfo->func_offsets[typeinfo->typeinfo->cElement & 0xffff] = offset;
1602 typeinfo->func_names[typeinfo->typeinfo->cElement & 0xffff] = name_offset;
1603
1604 /* ??? */
1605 if (!typeinfo->typeinfo->res2) typeinfo->typeinfo->res2 = 0x20;
1606 typeinfo->typeinfo->res2 <<= 1;
1607 /* ??? */
1608 if (index < 2) typeinfo->typeinfo->res2 += num_params << 4;
1609
1610 if (typeinfo->typeinfo->res3 == -1) typeinfo->typeinfo->res3 = 0;
1611 typeinfo->typeinfo->res3 += 0x38 + num_params * 0x10;
1612 if(num_defaults) typeinfo->typeinfo->res3 += num_params * 0x4;
1613
1614 /* adjust size of VTBL */
1615 if(funckind != 0x3 /* FUNC_STATIC */)
1616 typeinfo->typeinfo->cbSizeVft += pointer_size;
1617
1618 /* Increment the number of function elements */
1619 typeinfo->typeinfo->cElement += 1;
1620
1621 namedata = typeinfo->typelib->typelib_segment_data[MSFT_SEG_NAME] + name_offset;
1622 if (*((INT *)namedata) == -1) {
1623 *((INT *)namedata) = typeinfo->typelib->typelib_typeinfo_offsets[typeinfo->typeinfo->typekind >> 16];
1624 if(typeinfo->typekind == TKIND_MODULE)
1625 namedata[9] |= 0x10;
1626 } else
1627 namedata[9] &= ~0x10;
1628
1629 if(typeinfo->typekind == TKIND_MODULE)
1630 namedata[9] |= 0x20;
1631
1632 if (type_get_function_args(func->type))
1633 {
1634 i = 0;
1635 LIST_FOR_EACH_ENTRY( arg, type_get_function_args(func->type), var_t, entry )
1636 {
1637 /* don't give the last arg of a [propput*] func a name */
1638 if(i != num_params - 1 || (invokekind != 0x4 /* INVOKE_PROPERTYPUT */ && invokekind != 0x8 /* INVOKE_PROPERTYPUTREF */))
1639 {
1640 int *paramdata = typedata + 6 + extra_attr + (num_defaults ? num_params : 0) + i * 3;
1641 offset = ctl2_alloc_name(typeinfo->typelib, arg->name);
1642 paramdata[1] = offset;
1643 }
1644 i++;
1645 }
1646 }
1647 return S_OK;
1648 }
1649
1650 static HRESULT add_var_desc(msft_typeinfo_t *typeinfo, UINT index, var_t* var)
1651 {
1652 int offset, id;
1653 unsigned int typedata_size;
1654 INT *typedata;
1655 int var_datawidth;
1656 int var_alignment;
1657 int var_type_size, var_kind = 0 /* VAR_PERINSTANCE */;
1658 int alignment;
1659 int varflags = 0;
1660 const attr_t *attr;
1661 unsigned char *namedata;
1662 int var_num = (typeinfo->typeinfo->cElement >> 16) & 0xffff;
1663
1664 chat("add_var_desc(%d, %s)\n", index, var->name);
1665
1666 id = 0x40000000 + index;
1667
1668 if (var->attrs) LIST_FOR_EACH_ENTRY( attr, var->attrs, const attr_t, entry ) {
1669 expr_t *expr = attr->u.pval;
1670 switch(attr->type) {
1671 case ATTR_BINDABLE:
1672 varflags |= 0x04; /* VARFLAG_FBINDABLE */
1673 break;
1674 case ATTR_DEFAULTBIND:
1675 varflags |= 0x20; /* VARFLAG_FDEFAULTBIND */
1676 break;
1677 case ATTR_DEFAULTCOLLELEM:
1678 varflags |= 0x100; /* VARFLAG_FDEFAULTCOLLELEM */
1679 break;
1680 case ATTR_DISPLAYBIND:
1681 varflags |= 0x10; /* VARFLAG_FDISPLAYBIND */
1682 break;
1683 case ATTR_HIDDEN:
1684 varflags |= 0x40; /* VARFLAG_FHIDDEN */
1685 break;
1686 case ATTR_ID:
1687 id = expr->cval;
1688 break;
1689 case ATTR_IMMEDIATEBIND:
1690 varflags |= 0x1000; /* VARFLAG_FIMMEDIATEBIND */
1691 break;
1692 case ATTR_NONBROWSABLE:
1693 varflags |= 0x400; /* VARFLAG_FNONBROWSABLE */
1694 break;
1695 case ATTR_READONLY:
1696 varflags |= 0x01; /* VARFLAG_FREADONLY */
1697 break;
1698 /* FIXME: VARFLAG_FREPLACEABLE */
1699 case ATTR_REQUESTEDIT:
1700 varflags |= 0x08; /* VARFLAG_FREQUESTEDIT */
1701 break;
1702 case ATTR_RESTRICTED:
1703 varflags |= 0x80; /* VARFLAG_FRESTRICTED */
1704 break;
1705 case ATTR_SOURCE:
1706 varflags |= 0x02; /* VARFLAG_FSOURCE */
1707 break;
1708 case ATTR_UIDEFAULT:
1709 varflags |= 0x0200; /* VARFLAG_FUIDEFAULT */
1710 break;
1711 default:
1712 break;
1713 }
1714 }
1715
1716 /* allocate type data space for us */
1717 typedata_size = 0x14;
1718
1719 if (!typeinfo->var_data) {
1720 typeinfo->var_data = xmalloc(0x100);
1721 typeinfo->var_data_allocated = 0x100;
1722 typeinfo->var_data[0] = 0;
1723 }
1724
1725 if(typeinfo->var_data[0] + typedata_size + sizeof(int) > typeinfo->var_data_allocated) {
1726 typeinfo->var_data_allocated = max(typeinfo->var_data_allocated * 2,
1727 typeinfo->var_data[0] + typedata_size + sizeof(int));
1728 typeinfo->var_data = xrealloc(typeinfo->var_data, typeinfo->var_data_allocated);
1729 }
1730
1731 offset = typeinfo->var_data[0];
1732 typeinfo->var_data[0] += typedata_size;
1733 typedata = typeinfo->var_data + (offset >> 2) + 1;
1734
1735 /* fill out the basic type information */
1736 typedata[0] = typedata_size | (index << 16);
1737 typedata[2] = varflags;
1738 typedata[3] = (36 /*sizeof(VARDESC)*/ << 16) | 0;
1739
1740 if(typeinfo->vars_allocated == 0) {
1741 typeinfo->vars_allocated = 10;
1742 typeinfo->var_indices = xmalloc(typeinfo->vars_allocated * sizeof(int));
1743 typeinfo->var_names = xmalloc(typeinfo->vars_allocated * sizeof(int));
1744 typeinfo->var_offsets = xmalloc(typeinfo->vars_allocated * sizeof(int));
1745 }
1746 if(typeinfo->vars_allocated == var_num) {
1747 typeinfo->vars_allocated *= 2;
1748 typeinfo->var_indices = xrealloc(typeinfo->var_indices, typeinfo->vars_allocated * sizeof(int));
1749 typeinfo->var_names = xrealloc(typeinfo->var_names, typeinfo->vars_allocated * sizeof(int));
1750 typeinfo->var_offsets = xrealloc(typeinfo->var_offsets, typeinfo->vars_allocated * sizeof(int));
1751 }
1752 /* update the index data */
1753 typeinfo->var_indices[var_num] = id;
1754 typeinfo->var_names[var_num] = -1;
1755 typeinfo->var_offsets[var_num] = offset;
1756
1757 /* figure out type widths and whatnot */
1758 encode_var(typeinfo->typelib, var->type, var, &typedata[1], &var_datawidth,
1759 &var_alignment, &var_type_size);
1760
1761 /* pad out starting position to data width */
1762 typeinfo->datawidth += var_alignment - 1;
1763 typeinfo->datawidth &= ~(var_alignment - 1);
1764
1765 switch(typeinfo->typekind) {
1766 case TKIND_ENUM:
1767 write_value(typeinfo->typelib, &typedata[4], VT_I4, &var->eval->cval);
1768 var_kind = 2; /* VAR_CONST */
1769 var_type_size += 16; /* sizeof(VARIANT) */
1770 typeinfo->datawidth = var_datawidth;
1771 break;
1772 case TKIND_RECORD:
1773 typedata[4] = typeinfo->datawidth;
1774 typeinfo->datawidth += var_datawidth;
1775 break;
1776 case TKIND_UNION:
1777 typedata[4] = typeinfo->datawidth;
1778 typeinfo->datawidth = max(typeinfo->datawidth, var_datawidth);
1779 break;
1780 case TKIND_DISPATCH:
1781 var_kind = 3; /* VAR_DISPATCH */
1782 typeinfo->datawidth = pointer_size;
1783 var_alignment = 4;
1784 break;
1785 default:
1786 error("add_var_desc: unhandled type kind %d\n", typeinfo->typekind);
1787 break;
1788 }
1789
1790 /* add type description size to total required allocation */
1791 typedata[3] += var_type_size << 16 | var_kind;
1792
1793 /* fix type alignment */
1794 alignment = (typeinfo->typeinfo->typekind >> 11) & 0x1f;
1795 if (alignment < var_alignment) {
1796 alignment = var_alignment;
1797 typeinfo->typeinfo->typekind &= ~0xffc0;
1798 typeinfo->typeinfo->typekind |= alignment << 11 | alignment << 6;
1799 }
1800
1801 /* ??? */
1802 if (!typeinfo->typeinfo->res2) typeinfo->typeinfo->res2 = 0x1a;
1803 if ((index == 0) || (index == 1) || (index == 2) || (index == 4) || (index == 9)) {
1804 typeinfo->typeinfo->res2 <<= 1;
1805 }
1806
1807 /* ??? */
1808 if (typeinfo->typeinfo->res3 == -1) typeinfo->typeinfo->res3 = 0;
1809 typeinfo->typeinfo->res3 += 0x2c;
1810
1811 /* increment the number of variable elements */
1812 typeinfo->typeinfo->cElement += 0x10000;
1813
1814 /* pad data width to alignment */
1815 typeinfo->typeinfo->size = (typeinfo->datawidth + (alignment - 1)) & ~(alignment - 1);
1816
1817 offset = ctl2_alloc_name(typeinfo->typelib, var->name);
1818 if (offset == -1) return E_OUTOFMEMORY;
1819
1820 namedata = typeinfo->typelib->typelib_segment_data[MSFT_SEG_NAME] + offset;
1821 if (*((INT *)namedata) == -1) {
1822 *((INT *)namedata) = typeinfo->typelib->typelib_typeinfo_offsets[typeinfo->typeinfo->typekind >> 16];
1823 if(typeinfo->typekind != TKIND_DISPATCH)
1824 namedata[9] |= 0x10;
1825 } else
1826 namedata[9] &= ~0x10;
1827
1828 if (typeinfo->typekind == TKIND_ENUM) {
1829 namedata[9] |= 0x20;
1830 }
1831 typeinfo->var_names[var_num] = offset;
1832
1833 return S_OK;
1834 }
1835
1836 static HRESULT add_impl_type(msft_typeinfo_t *typeinfo, type_t *ref, importinfo_t *importinfo)
1837 {
1838 if(importinfo) {
1839 alloc_importinfo(typeinfo->typelib, importinfo);
1840 typeinfo->typeinfo->datatype1 = importinfo->offset+1;
1841 }else {
1842 if(ref->typelib_idx == -1)
1843 add_interface_typeinfo(typeinfo->typelib, ref);
1844 if(ref->typelib_idx == -1)
1845 error("add_impl_type: unable to add inherited interface\n");
1846
1847 typeinfo->typeinfo->datatype1 = typeinfo->typelib->typelib_typeinfo_offsets[ref->typelib_idx];
1848 }
1849
1850 typeinfo->typeinfo->cImplTypes++;
1851 return S_OK;
1852 }
1853
1854 static msft_typeinfo_t *create_msft_typeinfo(msft_typelib_t *typelib, enum type_kind kind,
1855 const char *name, const attr_list_t *attrs)
1856 {
1857 const attr_t *attr;
1858 msft_typeinfo_t *msft_typeinfo;
1859 int nameoffset;
1860 int typeinfo_offset;
1861 MSFT_TypeInfoBase *typeinfo;
1862 MSFT_GuidEntry guidentry;
1863
1864 chat("create_msft_typeinfo: name %s kind %d index %d\n", name, kind, typelib->typelib_header.nrtypeinfos);
1865
1866 msft_typeinfo = xmalloc(sizeof(*msft_typeinfo));
1867 memset( msft_typeinfo, 0, sizeof(*msft_typeinfo) );
1868
1869 msft_typeinfo->typelib = typelib;
1870
1871 nameoffset = ctl2_alloc_name(typelib, name);
1872 typeinfo_offset = ctl2_alloc_typeinfo(typelib, nameoffset);
1873 typeinfo = (MSFT_TypeInfoBase *)&typelib->typelib_segment_data[MSFT_SEG_TYPEINFO][typeinfo_offset];
1874
1875 typelib->typelib_segment_data[MSFT_SEG_NAME][nameoffset + 9] = 0x38;
1876 *((int *)&typelib->typelib_segment_data[MSFT_SEG_NAME][nameoffset]) = typeinfo_offset;
1877
1878 msft_typeinfo->typekind = kind;
1879 msft_typeinfo->typeinfo = typeinfo;
1880
1881 typeinfo->typekind |= kind | 0x20;
1882
1883 if(kind == TKIND_COCLASS)
1884 typeinfo->flags |= 0x2; /* TYPEFLAG_FCANCREATE */
1885
1886 if (attrs) LIST_FOR_EACH_ENTRY( attr, attrs, const attr_t, entry ) {
1887 switch(attr->type) {
1888 case ATTR_AGGREGATABLE:
1889 if (kind == TKIND_COCLASS)
1890 typeinfo->flags |= 0x400; /* TYPEFLAG_FAGGREGATABLE */
1891 break;
1892
1893 case ATTR_APPOBJECT:
1894 if (kind == TKIND_COCLASS)
1895 typeinfo->flags |= 0x1; /* TYPEFLAG_FAPPOBJECT */
1896 break;
1897
1898 case ATTR_CONTROL:
1899 if (kind == TKIND_COCLASS)
1900 typeinfo->flags |= 0x20; /* TYPEFLAG_FCONTROL */
1901 break;
1902
1903 case ATTR_DLLNAME:
1904 {
1905 int offset = ctl2_alloc_string(typelib, attr->u.pval);
1906 typeinfo->datatype1 = offset;
1907 break;
1908 }
1909
1910 case ATTR_DUAL:
1911 /* FIXME: check interface is compatible */
1912 typeinfo->typekind = (typeinfo->typekind & ~0xff) | 0x34;
1913 typeinfo->flags |= 0x140; /* TYPEFLAG_FDUAL | TYPEFLAG_FOLEAUTOMATION */
1914 break;
1915
1916 case ATTR_HELPCONTEXT:
1917 {
1918 expr_t *expr = (expr_t*)attr->u.pval;
1919 typeinfo->helpcontext = expr->cval;
1920 break;
1921 }
1922 case ATTR_HELPSTRING:
1923 {
1924 int offset = ctl2_alloc_string(typelib, attr->u.pval);
1925 if (offset == -1) break;
1926 typeinfo->docstringoffs = offset;
1927 break;
1928 }
1929 case ATTR_HELPSTRINGCONTEXT:
1930 {
1931 expr_t *expr = (expr_t*)attr->u.pval;
1932 typeinfo->helpstringcontext = expr->cval;
1933 break;
1934 }
1935 case ATTR_HIDDEN:
1936 typeinfo->flags |= 0x10; /* TYPEFLAG_FHIDDEN */
1937 break;
1938
1939 case ATTR_LICENSED:
1940 typeinfo->flags |= 0x04; /* TYPEFLAG_FLICENSED */
1941 break;
1942
1943 case ATTR_NONCREATABLE:
1944 typeinfo->flags &= ~0x2; /* TYPEFLAG_FCANCREATE */
1945 break;
1946
1947 case ATTR_NONEXTENSIBLE:
1948 typeinfo->flags |= 0x80; /* TYPEFLAG_FNONEXTENSIBLE */
1949 break;
1950
1951 case ATTR_OLEAUTOMATION:
1952 typeinfo->flags |= 0x100; /* TYPEFLAG_FOLEAUTOMATION */
1953 break;
1954
1955 /* FIXME: TYPEFLAG_FPREDCLID */
1956
1957 case ATTR_PROXY:
1958 typeinfo->flags |= 0x4000; /* TYPEFLAG_FPROXY */
1959 break;
1960
1961 /* FIXME: TYPEFLAG_FREPLACEABLE */
1962
1963 case ATTR_RESTRICTED:
1964 typeinfo->flags |= 0x200; /* TYPEFLAG_FRESTRICTED */
1965 break;
1966
1967 case ATTR_UUID:
1968 guidentry.guid = *(GUID*)attr->u.pval;
1969 guidentry.hreftype = typelib->typelib_typeinfo_offsets[typeinfo->typekind >> 16];
1970 guidentry.next_hash = -1;
1971 typeinfo->posguid = ctl2_alloc_guid(typelib, &guidentry);
1972 #if 0
1973 if (IsEqualIID(guid, &IID_IDispatch)) {
1974 typelib->typelib_header.dispatchpos = typelib->typelib_typeinfo_offsets[typeinfo->typekind >> 16];
1975 }
1976 #endif
1977 break;
1978
1979 case ATTR_VERSION:
1980 typeinfo->version = attr->u.ival;
1981 break;
1982
1983 default:
1984 break;
1985 }
1986 }
1987
1988 if (typelib->last_typeinfo) typelib->last_typeinfo->next_typeinfo = msft_typeinfo;
1989 typelib->last_typeinfo = msft_typeinfo;
1990 if (!typelib->typeinfos) typelib->typeinfos = msft_typeinfo;
1991
1992
1993 return msft_typeinfo;
1994 }
1995
1996 static void add_dispatch(msft_typelib_t *typelib)
1997 {
1998 int guid_offset, impfile_offset, hash_key;
1999 MSFT_GuidEntry guidentry;
2000 MSFT_ImpInfo impinfo;
2001 GUID stdole = {0x00020430,0x0000,0x0000,{0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46}};
2002 GUID iid_idispatch = {0x00020400,0x0000,0x0000,{0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46}};
2003
2004 if(typelib->typelib_header.dispatchpos != -1) return;
2005
2006 guidentry.guid = stdole;
2007 guidentry.hreftype = 2;
2008 guidentry.next_hash = -1;
2009 hash_key = ctl2_hash_guid(&guidentry.guid);
2010 guid_offset = ctl2_find_guid(typelib, hash_key, &guidentry.guid);
2011 if (guid_offset == -1)
2012 guid_offset = ctl2_alloc_guid(typelib, &guidentry);
2013 impfile_offset = alloc_importfile(typelib, guid_offset, 2, 0, "stdole2.tlb");
2014
2015 guidentry.guid = iid_idispatch;
2016 guidentry.hreftype = 1;
2017 guidentry.next_hash = -1;
2018 impinfo.flags = TKIND_INTERFACE << 24 | MSFT_IMPINFO_OFFSET_IS_GUID;
2019 impinfo.oImpFile = impfile_offset;
2020 hash_key = ctl2_hash_guid(&guidentry.guid);
2021 guid_offset = ctl2_find_guid(typelib, hash_key, &guidentry.guid);
2022 if (guid_offset == -1)
2023 guid_offset = ctl2_alloc_guid(typelib, &guidentry);
2024 impinfo.oGuid = guid_offset;
2025 typelib->typelib_header.dispatchpos = alloc_msft_importinfo(typelib, &impinfo) | 0x01;
2026 }
2027
2028 static void add_dispinterface_typeinfo(msft_typelib_t *typelib, type_t *dispinterface)
2029 {
2030 int idx = 0;
2031 var_t *func;
2032 var_t *var;
2033 msft_typeinfo_t *msft_typeinfo;
2034
2035 if (-1 < dispinterface->typelib_idx)
2036 return;
2037
2038 dispinterface->typelib_idx = typelib->typelib_header.nrtypeinfos;
2039 msft_typeinfo = create_msft_typeinfo(typelib, TKIND_DISPATCH, dispinterface->name,
2040 dispinterface->attrs);
2041
2042 msft_typeinfo->typeinfo->size = pointer_size;
2043 msft_typeinfo->typeinfo->typekind |= 0x2100;
2044
2045 msft_typeinfo->typeinfo->flags |= 0x1000; /* TYPEFLAG_FDISPATCHABLE */
2046 add_dispatch(typelib);
2047 msft_typeinfo->typeinfo->cImplTypes = 1;
2048
2049 /* count the no of methods, as the variable indices come after the funcs */
2050 if (dispinterface->details.iface->disp_methods)
2051 LIST_FOR_EACH_ENTRY( func, dispinterface->details.iface->disp_methods, var_t, entry )
2052 idx++;
2053
2054 if (type_dispiface_get_props(dispinterface))
2055 LIST_FOR_EACH_ENTRY( var, type_dispiface_get_props(dispinterface), var_t, entry )
2056 add_var_desc(msft_typeinfo, idx++, var);
2057
2058 if (type_dispiface_get_methods(dispinterface))
2059 {
2060 idx = 0;
2061 LIST_FOR_EACH_ENTRY( func, type_dispiface_get_methods(dispinterface), var_t, entry )
2062 if(add_func_desc(msft_typeinfo, func, idx) == S_OK)
2063 idx++;
2064 }
2065 }
2066
2067 static void add_interface_typeinfo(msft_typelib_t *typelib, type_t *interface)
2068 {
2069 int idx = 0;
2070 const statement_t *stmt_func;
2071 type_t *ref;
2072 msft_typeinfo_t *msft_typeinfo;
2073 importinfo_t *ref_importinfo = NULL;
2074 int num_parents = 0, num_funcs = 0;
2075 type_t *inherit;
2076 const type_t *derived;
2077
2078 if (-1 < interface->typelib_idx)
2079 return;
2080
2081 if (!interface->details.iface)
2082 {
2083 error( "interface %s is referenced but not defined\n", interface->name );
2084 return;
2085 }
2086
2087 if (is_attr(interface->attrs, ATTR_DISPINTERFACE)) {
2088 add_dispinterface_typeinfo(typelib, interface);
2089 return;
2090 }
2091
2092 /* midl adds the parent interface first, unless the parent itself
2093 has no parent (i.e. it stops before IUnknown). */
2094
2095 inherit = type_iface_get_inherit(interface);
2096
2097 if(inherit) {
2098 ref_importinfo = find_importinfo(typelib, inherit->name);
2099
2100 if(!ref_importinfo && type_iface_get_inherit(inherit) &&
2101 inherit->typelib_idx == -1)
2102 add_interface_typeinfo(typelib, inherit);
2103 }
2104
2105 /* check typelib_idx again, it could have been added while resolving the parent interface */
2106 if (-1 < interface->typelib_idx)
2107 return;
2108
2109 interface->typelib_idx = typelib->typelib_header.nrtypeinfos;
2110 msft_typeinfo = create_msft_typeinfo(typelib, TKIND_INTERFACE, interface->name, interface->attrs);
2111 msft_typeinfo->typeinfo->size = pointer_size;
2112 msft_typeinfo->typeinfo->typekind |= 0x2200;
2113
2114 for (derived = inherit; derived; derived = type_iface_get_inherit(derived))
2115 if (derived->name && !strcmp(derived->name, "IDispatch"))
2116 msft_typeinfo->typeinfo->flags |= 0x1000; /* TYPEFLAG_FDISPATCHABLE */
2117
2118 /* can't be dual if it doesn't derive from IDispatch */
2119 if (!(msft_typeinfo->typeinfo->flags & 0x1000)) /* TYPEFLAG_FDISPATCHABLE */
2120 msft_typeinfo->typeinfo->flags &= ~0x40; /* TYPEFLAG_FDUAL */
2121
2122 if(type_iface_get_inherit(interface))
2123 add_impl_type(msft_typeinfo, type_iface_get_inherit(interface),
2124 ref_importinfo);
2125
2126 /* count the number of inherited interfaces and non-local functions */
2127 for(ref = inherit; ref; ref = type_iface_get_inherit(ref)) {
2128 num_parents++;
2129 STATEMENTS_FOR_EACH_FUNC( stmt_func, type_iface_get_stmts(ref) ) {
2130 var_t *func = stmt_func->u.var;
2131 if (!is_local(func->attrs)) num_funcs++;
2132 }
2133 }
2134 msft_typeinfo->typeinfo->datatype2 = num_funcs << 16 | num_parents;
2135 msft_typeinfo->typeinfo->cbSizeVft = num_funcs * pointer_size;
2136
2137 STATEMENTS_FOR_EACH_FUNC( stmt_func, type_iface_get_stmts(interface) ) {
2138 var_t *func = stmt_func->u.var;
2139 if(add_func_desc(msft_typeinfo, func, idx) == S_OK)
2140 idx++;
2141 }
2142 }
2143
2144 static void add_structure_typeinfo(msft_typelib_t *typelib, type_t *structure)
2145 {
2146 int idx = 0;
2147 var_t *cur;
2148 msft_typeinfo_t *msft_typeinfo;
2149
2150 if (-1 < structure->typelib_idx)
2151 return;
2152
2153 structure->typelib_idx = typelib->typelib_header.nrtypeinfos;
2154 msft_typeinfo = create_msft_typeinfo(typelib, TKIND_RECORD, structure->name, structure->attrs);
2155 msft_typeinfo->typeinfo->size = 0;
2156
2157 if (type_struct_get_fields(structure))
2158 LIST_FOR_EACH_ENTRY( cur, type_struct_get_fields(structure), var_t, entry )
2159 add_var_desc(msft_typeinfo, idx++, cur);
2160 }
2161
2162 static void add_enum_typeinfo(msft_typelib_t *typelib, type_t *enumeration)
2163 {
2164 int idx = 0;
2165 var_t *cur;
2166 msft_typeinfo_t *msft_typeinfo;
2167
2168 if (-1 < enumeration->typelib_idx)
2169 return;
2170
2171 enumeration->typelib_idx = typelib->typelib_header.nrtypeinfos;
2172 msft_typeinfo = create_msft_typeinfo(typelib, TKIND_ENUM, enumeration->name, enumeration->attrs);
2173 msft_typeinfo->typeinfo->size = 0;
2174
2175 if (type_enum_get_values(enumeration))
2176 LIST_FOR_EACH_ENTRY( cur, type_enum_get_values(enumeration), var_t, entry )
2177 add_var_desc(msft_typeinfo, idx++, cur);
2178 }
2179
2180 static void add_union_typeinfo(msft_typelib_t *typelib, type_t *tunion)
2181 {
2182 int idx = 0;
2183 var_t *cur;
2184 msft_typeinfo_t *msft_typeinfo;
2185
2186 if (-1 < tunion->typelib_idx)
2187 return;
2188
2189 tunion->typelib_idx = typelib->typelib_header.nrtypeinfos;
2190 msft_typeinfo = create_msft_typeinfo(typelib, TKIND_UNION, tunion->name, tunion->attrs);
2191 msft_typeinfo->typeinfo->size = 0;
2192
2193 if (type_union_get_cases(tunion))
2194 LIST_FOR_EACH_ENTRY(cur, type_union_get_cases(tunion), var_t, entry)
2195 add_var_desc(msft_typeinfo, idx++, cur);
2196 }
2197
2198 static void add_typedef_typeinfo(msft_typelib_t *typelib, type_t *tdef)
2199 {
2200 msft_typeinfo_t *msft_typeinfo = NULL;
2201 int alignment, datatype1, datatype2, size, duplicate = 0;
2202 type_t *type;
2203
2204 if (-1 < tdef->typelib_idx)
2205 return;
2206
2207 type = type_alias_get_aliasee(tdef);
2208
2209 if (!type->name || strcmp(tdef->name, type->name) != 0)
2210 {
2211 tdef->typelib_idx = typelib->typelib_header.nrtypeinfos;
2212 msft_typeinfo = create_msft_typeinfo(typelib, TKIND_ALIAS, tdef->name, tdef->attrs);
2213 }
2214 else
2215 duplicate = 1;
2216
2217 encode_type(typelib, get_type_vt(type), type,
2218 &datatype1, &size, &alignment, &datatype2);
2219
2220 if (msft_typeinfo)
2221 {
2222 msft_typeinfo->typeinfo->datatype1 = datatype1;
2223 msft_typeinfo->typeinfo->size = size;
2224 msft_typeinfo->typeinfo->datatype2 = datatype2;
2225 msft_typeinfo->typeinfo->typekind |= (alignment << 11 | alignment << 6);
2226 }
2227
2228 /* avoid adding duplicate type definitions */
2229 if (duplicate)
2230 tdef->typelib_idx = type->typelib_idx;
2231 }
2232
2233 static void add_coclass_typeinfo(msft_typelib_t *typelib, type_t *cls)
2234 {
2235 msft_typeinfo_t *msft_typeinfo;
2236 ifref_t *iref;
2237 int num_ifaces = 0, offset, i;
2238 MSFT_RefRecord *ref, *first = NULL, *first_source = NULL;
2239 int have_default = 0, have_default_source = 0;
2240 const attr_t *attr;
2241 ifref_list_t *ifaces;
2242
2243 if (-1 < cls->typelib_idx)
2244 return;
2245
2246 cls->typelib_idx = typelib->typelib_header.nrtypeinfos;
2247 msft_typeinfo = create_msft_typeinfo(typelib, TKIND_COCLASS, cls->name, cls->attrs);
2248
2249 ifaces = type_coclass_get_ifaces(cls);
2250 if (ifaces) LIST_FOR_EACH_ENTRY( iref, ifaces, ifref_t, entry ) num_ifaces++;
2251
2252 offset = msft_typeinfo->typeinfo->datatype1 = ctl2_alloc_segment(typelib, MSFT_SEG_REFERENCES,
2253 num_ifaces * sizeof(*ref), 0);
2254
2255 i = 0;
2256 if (ifaces) LIST_FOR_EACH_ENTRY( iref, ifaces, ifref_t, entry ) {
2257 if(iref->iface->typelib_idx == -1)
2258 add_interface_typeinfo(typelib, iref->iface);
2259 ref = (MSFT_RefRecord*) (typelib->typelib_segment_data[MSFT_SEG_REFERENCES] + offset + i * sizeof(*ref));
2260 ref->reftype = typelib->typelib_typeinfo_offsets[iref->iface->typelib_idx];
2261 ref->flags = 0;
2262 ref->oCustData = -1;
2263 ref->onext = -1;
2264 if(i < num_ifaces - 1)
2265 ref->onext = offset + (i + 1) * sizeof(*ref);
2266
2267 if (iref->attrs) LIST_FOR_EACH_ENTRY( attr, iref->attrs, const attr_t, entry ) {
2268 switch(attr->type) {
2269 case ATTR_DEFAULT:
2270 ref->flags |= 0x1; /* IMPLTYPEFLAG_FDEFAULT */
2271 break;
2272 case ATTR_DEFAULTVTABLE:
2273 ref->flags |= 0x8; /* IMPLTYPEFLAG_FDEFAULTVTABLE */
2274 break;
2275 case ATTR_RESTRICTED:
2276 ref->flags |= 0x4; /* IMPLTYPEFLAG_FRESTRICTED */
2277 break;
2278 case ATTR_SOURCE:
2279 ref->flags |= 0x2; /* IMPLTYPEFLAG_FSOURCE */
2280 break;
2281 default:
2282 warning("add_coclass_typeinfo: unhandled attr %d\n", attr->type);
2283 }
2284 }
2285 if(ref->flags & 0x1) { /* IMPLTYPEFLAG_FDEFAULT */
2286 if(ref->flags & 0x2) /* IMPLTYPEFLAG_SOURCE */
2287 have_default_source = 1;
2288 else
2289 have_default = 1;
2290 }
2291
2292 /* If the interface is non-restricted and we haven't already had one then
2293 remember it so that we can use it as a default later */
2294 if((ref->flags & 0x4) == 0) { /* IMPLTYPEFLAG_FRESTRICTED */
2295 if(ref->flags & 0x2) { /* IMPLTYPEFLAG_FSOURCE */
2296 if(!first_source)
2297 first_source = ref;
2298 }
2299 else if(!first)
2300 first = ref;
2301 }
2302 i++;
2303 }
2304
2305 /* If we haven't had a default interface, then set the default flags on the
2306 first ones */
2307 if(!have_default && first)
2308 first->flags |= 0x1;
2309 if(!have_default_source && first_source)
2310 first_source->flags |= 0x1;
2311
2312 msft_typeinfo->typeinfo->cImplTypes = num_ifaces;
2313 msft_typeinfo->typeinfo->size = pointer_size;
2314 msft_typeinfo->typeinfo->typekind |= 0x2200;
2315 }
2316
2317 static void add_module_typeinfo(msft_typelib_t *typelib, type_t *module)
2318 {
2319 int idx = 0;
2320 const statement_t *stmt;
2321 msft_typeinfo_t *msft_typeinfo;
2322
2323 if (-1 < module->typelib_idx)
2324 return;
2325
2326 module->typelib_idx = typelib->typelib_header.nrtypeinfos;
2327 msft_typeinfo = create_msft_typeinfo(typelib, TKIND_MODULE, module->name, module->attrs);
2328 msft_typeinfo->typeinfo->typekind |= 0x0a00;
2329
2330 STATEMENTS_FOR_EACH_FUNC( stmt, module->details.module->stmts ) {
2331 var_t *func = stmt->u.var;
2332 if(add_func_desc(msft_typeinfo, func, idx) == S_OK)
2333 idx++;
2334 }
2335
2336 msft_typeinfo->typeinfo->size = idx;
2337 }
2338
2339 static void add_type_typeinfo(msft_typelib_t *typelib, type_t *type)
2340 {
2341 switch (type_get_type(type)) {
2342 case TYPE_INTERFACE:
2343 add_interface_typeinfo(typelib, type);
2344 break;
2345 case TYPE_STRUCT:
2346 add_structure_typeinfo(typelib, type);
2347 break;
2348 case TYPE_ENUM:
2349 add_enum_typeinfo(typelib, type);
2350 break;
2351 case TYPE_UNION:
2352 add_union_typeinfo(typelib, type);
2353 break;
2354 case TYPE_COCLASS:
2355 add_coclass_typeinfo(typelib, type);
2356 break;
2357 case TYPE_BASIC:
2358 case TYPE_POINTER:
2359 break;
2360 default:
2361 error("add_entry: unhandled type 0x%x for %s\n",
2362 type_get_type(type), type->name);
2363 break;
2364 }
2365 }
2366
2367 static void add_entry(msft_typelib_t *typelib, const statement_t *stmt)
2368 {
2369 switch(stmt->type) {
2370 case STMT_LIBRARY:
2371 case STMT_IMPORT:
2372 case STMT_PRAGMA:
2373 case STMT_CPPQUOTE:
2374 case STMT_DECLARATION:
2375 /* not included in typelib */
2376 break;
2377 case STMT_IMPORTLIB:
2378 /* not processed here */
2379 break;
2380 case STMT_TYPEDEF:
2381 {
2382 const type_list_t *type_entry = stmt->u.type_list;
2383 for (; type_entry; type_entry = type_entry->next) {
2384 /* if the type is public then add the typedef, otherwise attempt
2385 * to add the aliased type */
2386 if (is_attr(type_entry->type->attrs, ATTR_PUBLIC))
2387 add_typedef_typeinfo(typelib, type_entry->type);
2388 else
2389 add_type_typeinfo(typelib, type_alias_get_aliasee(type_entry->type));
2390 }
2391 break;
2392 }
2393 case STMT_MODULE:
2394 add_module_typeinfo(typelib, stmt->u.type);
2395 break;
2396 case STMT_TYPE:
2397 case STMT_TYPEREF:
2398 {
2399 type_t *type = stmt->u.type;
2400 add_type_typeinfo(typelib, type);
2401 break;
2402 }
2403 }
2404 }
2405
2406 static void set_name(msft_typelib_t *typelib)
2407 {
2408 int offset;
2409
2410 offset = ctl2_alloc_name(typelib, typelib->typelib->name);
2411 if (offset == -1) return;
2412 typelib->typelib_header.NameOffset = offset;
2413 return;
2414 }
2415
2416 static void set_version(msft_typelib_t *typelib)
2417 {
2418 typelib->typelib_header.version = get_attrv( typelib->typelib->attrs, ATTR_VERSION );
2419 }
2420
2421 static void set_guid(msft_typelib_t *typelib)
2422 {
2423 MSFT_GuidEntry guidentry;
2424 int offset;
2425 void *ptr;
2426 GUID guid = {0,0,0,{0,0,0,0,0,0}};
2427
2428 guidentry.guid = guid;
2429 guidentry.hreftype = -2;
2430 guidentry.next_hash = -1;
2431
2432 ptr = get_attrp( typelib->typelib->attrs, ATTR_UUID );
2433 if (ptr) guidentry.guid = *(GUID *)ptr;
2434
2435 offset = ctl2_alloc_guid(typelib, &guidentry);
2436 typelib->typelib_header.posguid = offset;
2437
2438 return;
2439 }
2440
2441 static void set_doc_string(msft_typelib_t *typelib)
2442 {
2443 char *str = get_attrp( typelib->typelib->attrs, ATTR_HELPSTRING );
2444
2445 if (str)
2446 {
2447 int offset = ctl2_alloc_string(typelib, str);
2448 if (offset != -1) typelib->typelib_header.helpstring = offset;
2449 }
2450 }
2451
2452 static void set_help_file_name(msft_typelib_t *typelib)
2453 {
2454 char *str = get_attrp( typelib->typelib->attrs, ATTR_HELPFILE );
2455
2456 if (str)
2457 {
2458 int offset = ctl2_alloc_string(typelib, str);
2459 if (offset != -1)
2460 {
2461 typelib->typelib_header.helpfile = offset;
2462 typelib->typelib_header.varflags |= 0x10;
2463 }
2464 }
2465 }
2466
2467 static void set_help_context(msft_typelib_t *typelib)
2468 {
2469 const expr_t *expr = get_attrp( typelib->typelib->attrs, ATTR_HELPCONTEXT );
2470 if (expr) typelib->typelib_header.helpcontext = expr->cval;
2471 }
2472
2473 static void set_help_string_dll(msft_typelib_t *typelib)
2474 {
2475 char *str = get_attrp( typelib->typelib->attrs, ATTR_HELPSTRINGDLL );
2476
2477 if (str)
2478 {
2479 int offset = ctl2_alloc_string(typelib, str);
2480 if (offset != -1)
2481 {
2482 typelib->help_string_dll_offset = offset;
2483 typelib->typelib_header.varflags |= 0x100;
2484 }
2485 }
2486 }
2487
2488 static void set_help_string_context(msft_typelib_t *typelib)
2489 {
2490 const expr_t *expr = get_attrp( typelib->typelib->attrs, ATTR_HELPSTRINGCONTEXT );
2491 if (expr) typelib->typelib_header.helpstringcontext = expr->cval;
2492 }
2493
2494 static void set_lcid(msft_typelib_t *typelib)
2495 {
2496 const expr_t *lcid_expr = get_attrp( typelib->typelib->attrs, ATTR_LIBLCID );
2497 if(lcid_expr)
2498 {
2499 typelib->typelib_header.lcid = lcid_expr->cval;
2500 typelib->typelib_header.lcid2 = lcid_expr->cval;
2501 }
2502 }
2503
2504 static void set_lib_flags(msft_typelib_t *typelib)
2505 {
2506 const attr_t *attr;
2507
2508 typelib->typelib_header.flags = 0;
2509 if (!typelib->typelib->attrs) return;
2510 LIST_FOR_EACH_ENTRY( attr, typelib->typelib->attrs, const attr_t, entry )
2511 {
2512 switch(attr->type) {
2513 case ATTR_CONTROL:
2514 typelib->typelib_header.flags |= 0x02; /* LIBFLAG_FCONTROL */
2515 break;
2516 case ATTR_HIDDEN:
2517 typelib->typelib_header.flags |= 0x04; /* LIBFLAG_FHIDDEN */
2518 break;
2519 case ATTR_RESTRICTED:
2520 typelib->typelib_header.flags |= 0x01; /* LIBFLAG_FRESTRICTED */
2521 break;
2522 default:
2523 break;
2524 }
2525 }
2526 return;
2527 }
2528
2529 static void ctl2_write_segment(msft_typelib_t *typelib, int segment)
2530 {
2531 put_data(typelib->typelib_segment_data[segment], typelib->typelib_segdir[segment].length);
2532 }
2533
2534 static void ctl2_finalize_typeinfos(msft_typelib_t *typelib, int filesize)
2535 {
2536 msft_typeinfo_t *typeinfo;
2537
2538 for (typeinfo = typelib->typeinfos; typeinfo; typeinfo = typeinfo->next_typeinfo) {
2539 typeinfo->typeinfo->memoffset = filesize;
2540 if (typeinfo->func_data)
2541 filesize += typeinfo->func_data[0] + ((typeinfo->typeinfo->cElement & 0xffff) * 12);
2542 if (typeinfo->var_data)
2543 filesize += typeinfo->var_data[0] + (((typeinfo->typeinfo->cElement >> 16) & 0xffff) * 12);
2544 if (typeinfo->func_data || typeinfo->var_data)
2545 filesize += 4;
2546 }
2547 }
2548
2549 static int ctl2_finalize_segment(msft_typelib_t *typelib, int filepos, int segment)
2550 {
2551 if (typelib->typelib_segdir[segment].length) {
2552 typelib->typelib_segdir[segment].offset = filepos;
2553 } else {
2554 typelib->typelib_segdir[segment].offset = -1;
2555 }
2556
2557 return typelib->typelib_segdir[segment].length;
2558 }
2559
2560
2561 static void ctl2_write_typeinfos(msft_typelib_t *typelib)
2562 {
2563 msft_typeinfo_t *typeinfo;
2564 int typedata_size;
2565
2566 for (typeinfo = typelib->typeinfos; typeinfo; typeinfo = typeinfo->next_typeinfo) {
2567 if (!typeinfo->func_data && !typeinfo->var_data) continue;
2568 typedata_size = 0;
2569 if (typeinfo->func_data)
2570 typedata_size = typeinfo->func_data[0];
2571 if (typeinfo->var_data)
2572 typedata_size += typeinfo->var_data[0];
2573 put_data(&typedata_size, sizeof(int));
2574 if (typeinfo->func_data)
2575 put_data(typeinfo->func_data + 1, typeinfo->func_data[0]);
2576 if (typeinfo->var_data)
2577 put_data(typeinfo->var_data + 1, typeinfo->var_data[0]);
2578 if (typeinfo->func_indices)
2579 put_data(typeinfo->func_indices, (typeinfo->typeinfo->cElement & 0xffff) * 4);
2580 if (typeinfo->var_indices)
2581 put_data(typeinfo->var_indices, (typeinfo->typeinfo->cElement >> 16) * 4);
2582 if (typeinfo->func_names)
2583 put_data(typeinfo->func_names, (typeinfo->typeinfo->cElement & 0xffff) * 4);
2584 if (typeinfo->var_names)
2585 put_data(typeinfo->var_names, (typeinfo->typeinfo->cElement >> 16) * 4);
2586 if (typeinfo->func_offsets)
2587 put_data(typeinfo->func_offsets, (typeinfo->typeinfo->cElement & 0xffff) * 4);
2588 if (typeinfo->var_offsets) {
2589 int add = 0, i, offset;
2590 if(typeinfo->func_data)
2591 add = typeinfo->func_data[0];
2592 for(i = 0; i < (typeinfo->typeinfo->cElement >> 16); i++) {
2593 offset = typeinfo->var_offsets[i];
2594 offset += add;
2595 put_data(&offset, 4);
2596 }
2597 }
2598 }
2599 }
2600
2601 static void save_all_changes(msft_typelib_t *typelib)
2602 {
2603 int filepos;
2604
2605 chat("save_all_changes(%p)\n", typelib);
2606
2607 filepos = sizeof(MSFT_Header) + sizeof(MSFT_SegDir);
2608 if(typelib->typelib_header.varflags & 0x100) filepos += 4; /* helpstringdll */
2609 filepos += typelib->typelib_header.nrtypeinfos * 4;
2610
2611 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_TYPEINFO);
2612 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_GUIDHASH);
2613 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_GUID);
2614 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_REFERENCES);
2615 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_IMPORTINFO);
2616 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_IMPORTFILES);
2617 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_NAMEHASH);
2618 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_NAME);
2619 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_STRING);
2620 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_TYPEDESC);
2621 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_ARRAYDESC);
2622 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_CUSTDATA);
2623 filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_CUSTDATAGUID);
2624
2625 ctl2_finalize_typeinfos(typelib, filepos);
2626
2627 byte_swapped = 0;
2628 init_output_buffer();
2629
2630 put_data(&typelib->typelib_header, sizeof(typelib->typelib_header));
2631 if(typelib->typelib_header.varflags & 0x100)
2632 put_data(&typelib->help_string_dll_offset, sizeof(typelib->help_string_dll_offset));
2633
2634 put_data(typelib->typelib_typeinfo_offsets, typelib->typelib_header.nrtypeinfos * 4);
2635 put_data(&typelib->typelib_segdir, sizeof(typelib->typelib_segdir));
2636 ctl2_write_segment( typelib, MSFT_SEG_TYPEINFO );
2637 ctl2_write_segment( typelib, MSFT_SEG_GUIDHASH );
2638 ctl2_write_segment( typelib, MSFT_SEG_GUID );
2639 ctl2_write_segment( typelib, MSFT_SEG_REFERENCES );
2640 ctl2_write_segment( typelib, MSFT_SEG_IMPORTINFO );
2641 ctl2_write_segment( typelib, MSFT_SEG_IMPORTFILES );
2642 ctl2_write_segment( typelib, MSFT_SEG_NAMEHASH );
2643 ctl2_write_segment( typelib, MSFT_SEG_NAME );
2644 ctl2_write_segment( typelib, MSFT_SEG_STRING );
2645 ctl2_write_segment( typelib, MSFT_SEG_TYPEDESC );
2646 ctl2_write_segment( typelib, MSFT_SEG_ARRAYDESC );
2647 ctl2_write_segment( typelib, MSFT_SEG_CUSTDATA );
2648 ctl2_write_segment( typelib, MSFT_SEG_CUSTDATAGUID );
2649
2650 ctl2_write_typeinfos(typelib);
2651
2652 if (strendswith( typelib_name, ".res" )) /* create a binary resource file */
2653 {
2654 char typelib_id[13] = "#1";
2655
2656 expr_t *expr = get_attrp( typelib->typelib->attrs, ATTR_ID );
2657 if (expr)
2658 sprintf( typelib_id, "#%d", expr->cval );
2659 add_output_to_resources( "TYPELIB", typelib_id );
2660 output_typelib_regscript( typelib->typelib );
2661 flush_output_resources( typelib_name );
2662 }
2663 else flush_output_buffer( typelib_name );
2664 }
2665
2666 int create_msft_typelib(typelib_t *typelib)
2667 {
2668 msft_typelib_t *msft;
2669 int failed = 0;
2670 const statement_t *stmt;
2671 time_t cur_time;
2672 char *time_override;
2673 unsigned int version = 7 << 24 | 555; /* 7.00.0555 */
2674 GUID midl_time_guid = {0xde77ba63,0x517c,0x11d1,{0xa2,0xda,0x00,0x00,0xf8,0x77,0x3c,0xe9}};
2675 GUID midl_version_guid = {0xde77ba64,0x517c,0x11d1,{0xa2,0xda,0x00,0x00,0xf8,0x77,0x3c,0xe9}};
2676 GUID midl_info_guid = {0xde77ba65,0x517c,0x11d1,{0xa2,0xda,0x00,0x00,0xf8,0x77,0x3c,0xe9}};
2677 char info_string[128];
2678
2679 pointer_size = (typelib_kind == SYS_WIN64) ? 8 : 4;
2680
2681 msft = xmalloc(sizeof(*msft));
2682 memset(msft, 0, sizeof(*msft));
2683 msft->typelib = typelib;
2684
2685 ctl2_init_header(msft);
2686 ctl2_init_segdir(msft);
2687
2688 msft->typelib_header.varflags |= typelib_kind;
2689
2690 /*
2691 * The following two calls return an offset or -1 if out of memory. We
2692 * specifically need an offset of 0, however, so...
2693 */
2694 if (ctl2_alloc_segment(msft, MSFT_SEG_GUIDHASH, 0x80, 0x80)) { failed = 1; }
2695 if (ctl2_alloc_segment(msft, MSFT_SEG_NAMEHASH, 0x200, 0x200)) { failed = 1; }
2696
2697 if(failed)
2698 {
2699 free(msft);
2700 return 0;
2701 }
2702
2703 msft->typelib_guidhash_segment = (int *)msft->typelib_segment_data[MSFT_SEG_GUIDHASH];
2704 msft->typelib_namehash_segment = (int *)msft->typelib_segment_data[MSFT_SEG_NAMEHASH];
2705
2706 memset(msft->typelib_guidhash_segment, 0xff, 0x80);
2707 memset(msft->typelib_namehash_segment, 0xff, 0x200);
2708
2709 set_lib_flags(msft);
2710 set_lcid(msft);
2711 set_help_file_name(msft);
2712 set_doc_string(msft);
2713 set_guid(msft);
2714 set_version(msft);
2715 set_name(msft);
2716 set_help_context(msft);
2717 set_help_string_dll(msft);
2718 set_help_string_context(msft);
2719
2720 /* midl adds two sets of custom data to the library: the current unix time
2721 and midl's version number */
2722 time_override = getenv( "WIDL_TIME_OVERRIDE");
2723 cur_time = time_override ? atol( time_override) : time(NULL);
2724 sprintf(info_string, "Created by WIDL version %s at %s\n", PACKAGE_VERSION, ctime(&cur_time));
2725 set_custdata(msft, &midl_info_guid, VT_BSTR, info_string, &msft->typelib_header.CustomDataOffset);
2726 set_custdata(msft, &midl_time_guid, VT_UI4, &cur_time, &msft->typelib_header.CustomDataOffset);
2727 set_custdata(msft, &midl_version_guid, VT_UI4, &version, &msft->typelib_header.CustomDataOffset);
2728
2729 if (typelib->stmts)
2730 LIST_FOR_EACH_ENTRY( stmt, typelib->stmts, const statement_t, entry )
2731 add_entry(msft, stmt);
2732
2733 save_all_changes(msft);
2734 free(msft);
2735 return 1;
2736 }
Windows API implementation