widl: Fix size parameter for xrealloc (Coccinelle).
[wine] / tools / widl / write_msft.c
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 #define NONAMELESSSTRUCT
42
43 #include "winerror.h"
44 #include "windef.h"
45 #include "winbase.h"
46 #include "winnls.h"
47
48 #include "widl.h"
49 #include "typelib.h"
50 #include "typelib_struct.h"
51 #include "utils.h"
52 #include "header.h"
53 #include "hash.h"
54 #include "typetree.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     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 < 15; 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         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[((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     hash_key = ctl2_hash_guid(&guid->guid);
499
500     offset = ctl2_find_guid(typelib, hash_key, &guid->guid);
501     if (offset != -1) return offset;
502
503     offset = ctl2_alloc_segment(typelib, MSFT_SEG_GUID, sizeof(MSFT_GuidEntry), 0);
504
505     guid_space = (void *)(typelib->typelib_segment_data[MSFT_SEG_GUID] + offset);
506     *guid_space = *guid;
507
508     guid_space->next_hash = typelib->typelib_guidhash_segment[hash_key];
509     typelib->typelib_guidhash_segment[hash_key] = offset;
510
511     return offset;
512 }
513
514 /****************************************************************************
515  *      ctl2_alloc_name
516  *
517  *  Allocates and initializes a name within a type library. Also updates the
518  *  name hash table as needed.
519  *
520  * RETURNS
521  *
522  *  Success: The offset within the segment of the new name.
523  *  Failure: -1 (this is invariably an out of memory condition).
524  */
525 static int ctl2_alloc_name(
526         msft_typelib_t *typelib,  /* [I] The type library to allocate in. */
527         const char *name)         /* [I] The name to store. */
528 {
529     int length;
530     int offset;
531     MSFT_NameIntro *name_space;
532     char *encoded_name;
533
534     length = ctl2_encode_name(typelib, name, &encoded_name);
535
536     offset = ctl2_find_name(typelib, encoded_name);
537     if (offset != -1) return offset;
538
539     offset = ctl2_alloc_segment(typelib, MSFT_SEG_NAME, length + 8, 0);
540
541     name_space = (void *)(typelib->typelib_segment_data[MSFT_SEG_NAME] + offset);
542     name_space->hreftype = -1;
543     name_space->next_hash = -1;
544     memcpy(&name_space->namelen, encoded_name, length);
545
546     if (typelib->typelib_namehash_segment[encoded_name[2] & 0x7f] != -1)
547         name_space->next_hash = typelib->typelib_namehash_segment[encoded_name[2] & 0x7f];
548
549     typelib->typelib_namehash_segment[encoded_name[2] & 0x7f] = offset;
550
551     typelib->typelib_header.nametablecount += 1;
552     typelib->typelib_header.nametablechars += *encoded_name;
553
554     return offset;
555 }
556
557 /****************************************************************************
558  *      ctl2_alloc_string
559  *
560  *  Allocates and initializes a string in a type library.
561  *
562  * RETURNS
563  *
564  *  Success: The offset within the segment of the new string.
565  *  Failure: -1 (this is invariably an out of memory condition).
566  */
567 static int ctl2_alloc_string(
568         msft_typelib_t *typelib,  /* [I] The type library to allocate in. */
569         const char *string)       /* [I] The string to store. */
570 {
571     int length;
572     int offset;
573     char *string_space;
574     char *encoded_string;
575
576     length = ctl2_encode_string(string, &encoded_string);
577
578     for (offset = 0; offset < typelib->typelib_segdir[MSFT_SEG_STRING].length;
579          offset += ((((typelib->typelib_segment_data[MSFT_SEG_STRING][offset + 1] << 8) & 0xff)
580              | (typelib->typelib_segment_data[MSFT_SEG_STRING][offset + 0] & 0xff)) + 5) & ~3) {
581         if (!memcmp(encoded_string, typelib->typelib_segment_data[MSFT_SEG_STRING] + offset, length)) return offset;
582     }
583
584     offset = ctl2_alloc_segment(typelib, MSFT_SEG_STRING, length, 0);
585
586     string_space = typelib->typelib_segment_data[MSFT_SEG_STRING] + offset;
587     memcpy(string_space, encoded_string, length);
588
589     return offset;
590 }
591
592 /****************************************************************************
593  *      alloc_msft_importinfo
594  *
595  *  Allocates and initializes an import information structure in a type library.
596  *
597  * RETURNS
598  *
599  *  Success: The offset of the new importinfo.
600  *  Failure: -1 (this is invariably an out of memory condition).
601  */
602 static int alloc_msft_importinfo(
603         msft_typelib_t *typelib,   /* [I] The type library to allocate in. */
604         MSFT_ImpInfo *impinfo)     /* [I] The import information to store. */
605 {
606     int offset;
607     MSFT_ImpInfo *impinfo_space;
608
609     for (offset = 0;
610          offset < typelib->typelib_segdir[MSFT_SEG_IMPORTINFO].length;
611          offset += sizeof(MSFT_ImpInfo)) {
612         if (!memcmp(&(typelib->typelib_segment_data[MSFT_SEG_IMPORTINFO][offset]),
613                     impinfo, sizeof(MSFT_ImpInfo))) {
614             return offset;
615         }
616     }
617
618     impinfo->flags |= typelib->typelib_header.nimpinfos++;
619
620     offset = ctl2_alloc_segment(typelib, MSFT_SEG_IMPORTINFO, sizeof(MSFT_ImpInfo), 0);
621
622     impinfo_space = (void *)(typelib->typelib_segment_data[MSFT_SEG_IMPORTINFO] + offset);
623     *impinfo_space = *impinfo;
624
625     return offset;
626 }
627
628 /****************************************************************************
629  *      alloc_importfile
630  *
631  *  Allocates and initializes an import file definition in a type library.
632  *
633  * RETURNS
634  *
635  *  Success: The offset of the new importinfo.
636  *  Failure: -1 (this is invariably an out of memory condition).
637  */
638 static int alloc_importfile(
639         msft_typelib_t *typelib,   /* [I] The type library to allocate in. */
640         int guidoffset,            /* [I] The offset to the GUID for the imported library. */
641         int major_version,         /* [I] The major version number of the imported library. */
642         int minor_version,         /* [I] The minor version number of the imported library. */
643         const char *filename)      /* [I] The filename of the imported library. */
644 {
645     int length;
646     int offset;
647     MSFT_ImpFile *importfile;
648     char *encoded_string;
649
650     length = ctl2_encode_string(filename, &encoded_string);
651
652     encoded_string[0] <<= 2;
653     encoded_string[0] |= 1;
654
655     for (offset = 0; offset < typelib->typelib_segdir[MSFT_SEG_IMPORTFILES].length;
656          offset += ((((typelib->typelib_segment_data[MSFT_SEG_IMPORTFILES][offset + 0xd] << 8) & 0xff)
657              | (typelib->typelib_segment_data[MSFT_SEG_IMPORTFILES][offset + 0xc] & 0xff)) >> 2) + 0xc) {
658         if (!memcmp(encoded_string, typelib->typelib_segment_data[MSFT_SEG_IMPORTFILES] + offset + 0xc, length)) return offset;
659     }
660
661     offset = ctl2_alloc_segment(typelib, MSFT_SEG_IMPORTFILES, length + 0xc, 0);
662
663     importfile = (MSFT_ImpFile *)&typelib->typelib_segment_data[MSFT_SEG_IMPORTFILES][offset];
664     importfile->guid = guidoffset;
665     importfile->lcid = typelib->typelib_header.lcid2;
666     importfile->version = major_version | (minor_version << 16);
667     memcpy(&importfile->filename, encoded_string, length);
668
669     return offset;
670 }
671
672 static void alloc_importinfo(msft_typelib_t *typelib, importinfo_t *importinfo)
673 {
674     importlib_t *importlib = importinfo->importlib;
675
676     chat("alloc_importinfo: %s\n", importinfo->name);
677
678     if(!importlib->allocated) {
679         MSFT_GuidEntry guid;
680         int guid_idx;
681
682         chat("allocating importlib %s\n", importlib->name);
683
684         importlib->allocated = -1;
685
686         memcpy(&guid.guid, &importlib->guid, sizeof(GUID));
687         guid.hreftype = 2;
688
689         guid_idx = ctl2_alloc_guid(typelib, &guid);
690
691         alloc_importfile(typelib, guid_idx, importlib->version&0xffff,
692                          importlib->version>>16, importlib->name);
693     }
694
695     if(importinfo->offset == -1 || !(importinfo->flags & MSFT_IMPINFO_OFFSET_IS_GUID)) {
696         MSFT_ImpInfo impinfo;
697
698         impinfo.flags = importinfo->flags;
699         impinfo.oImpFile = 0;
700
701         if(importinfo->flags & MSFT_IMPINFO_OFFSET_IS_GUID) {
702             MSFT_GuidEntry guid;
703
704             guid.hreftype = 0;
705             memcpy(&guid.guid, &importinfo->guid, sizeof(GUID));
706
707             impinfo.oGuid = ctl2_alloc_guid(typelib, &guid);
708
709             importinfo->offset = alloc_msft_importinfo(typelib, &impinfo);
710
711             typelib->typelib_segment_data[MSFT_SEG_GUID][impinfo.oGuid+sizeof(GUID)]
712                 = importinfo->offset+1;
713
714             if(!strcmp(importinfo->name, "IDispatch"))
715                 typelib->typelib_header.dispatchpos = importinfo->offset+1;
716         }else {
717             impinfo.oGuid = importinfo->id;
718             importinfo->offset = alloc_msft_importinfo(typelib, &impinfo);
719         }
720     }
721 }
722
723 static importinfo_t *find_importinfo(msft_typelib_t *typelib, const char *name)
724 {
725     importlib_t *importlib;
726     int i;
727
728     chat("search importlib %s\n", name);
729
730     if(!name)
731         return NULL;
732
733     LIST_FOR_EACH_ENTRY( importlib, &typelib->typelib->importlibs, importlib_t, entry )
734     {
735         for(i=0; i < importlib->ntypeinfos; i++) {
736             if(!strcmp(name, importlib->importinfos[i].name)) {
737                 chat("Found %s in importlib.\n", name);
738                 return importlib->importinfos+i;
739             }
740         }
741     }
742
743     return NULL;
744 }
745
746 static void add_structure_typeinfo(msft_typelib_t *typelib, type_t *structure);
747 static void add_interface_typeinfo(msft_typelib_t *typelib, type_t *interface);
748 static void add_enum_typeinfo(msft_typelib_t *typelib, type_t *enumeration);
749 static void add_coclass_typeinfo(msft_typelib_t *typelib, type_t *cls);
750 static void add_dispinterface_typeinfo(msft_typelib_t *typelib, type_t *dispinterface);
751
752
753 /****************************************************************************
754  *      encode_type
755  *
756  *  Encodes a type, storing information in the TYPEDESC and ARRAYDESC
757  *  segments as needed.
758  *
759  * RETURNS
760  *
761  *  Success: 0.
762  *  Failure: -1.
763  */
764 static int encode_type(
765         msft_typelib_t *typelib,   /* [I] The type library in which to encode the TYPEDESC. */
766         int vt,                    /* [I] vt to encode */
767         type_t *type,              /* [I] type */
768         int *encoded_type,         /* [O] The encoded type description. */
769         int *width,                /* [O] The width of the type, or NULL. */
770         int *alignment,            /* [O] The alignment of the type, or NULL. */
771         int *decoded_size)         /* [O] The total size of the unencoded TYPEDESCs, including nested descs. */
772 {
773     int default_type;
774     int scratch;
775     int typeoffset;
776     int *typedata;
777     int target_type;
778     int child_size = 0;
779
780     chat("encode_type vt %d type %p\n", vt, type);
781
782     default_type = 0x80000000 | (vt << 16) | vt;
783     if (!width) width = &scratch;
784     if (!alignment) alignment = &scratch;
785     if (!decoded_size) decoded_size = &scratch;
786
787
788     switch (vt) {
789     case VT_I1:
790     case VT_UI1:
791         *encoded_type = default_type;
792         *width = 1;
793         *alignment = 1;
794         break;
795
796     case VT_INT:
797         *encoded_type = 0x80000000 | (VT_I4 << 16) | VT_INT;
798         if ((typelib->typelib_header.varflags & 0x0f) == SYS_WIN16) {
799             *width = 2;
800             *alignment = 2;
801         } else {
802             *width = 4;
803             *alignment = 4;
804         }
805         break;
806
807     case VT_UINT:
808         *encoded_type = 0x80000000 | (VT_UI4 << 16) | VT_UINT;
809         if ((typelib->typelib_header.varflags & 0x0f) == SYS_WIN16) {
810             *width = 2;
811             *alignment = 2;
812         } else {
813             *width = 4;
814             *alignment = 4;
815         }
816         break;
817
818     case VT_UI2:
819     case VT_I2:
820     case VT_BOOL:
821         *encoded_type = default_type;
822         *width = 2;
823         *alignment = 2;
824         break;
825
826     case VT_I4:
827     case VT_UI4:
828     case VT_R4:
829     case VT_ERROR:
830     case VT_BSTR:
831     case VT_HRESULT:
832         *encoded_type = default_type;
833         *width = 4;
834         *alignment = 4;
835         break;
836
837     case VT_R8:
838     case VT_I8:
839     case VT_UI8:
840         *encoded_type = default_type;
841         *width = 8;
842         *alignment = 8;
843         break;
844
845     case VT_CY:
846     case VT_DATE:
847         *encoded_type = default_type;
848         *width = 8;
849         *alignment = 8;
850         break;
851
852     case VT_VOID:
853         *encoded_type = 0x80000000 | (VT_EMPTY << 16) | vt;
854         *width = 0;
855         *alignment = 1;
856         break;
857
858     case VT_UNKNOWN:
859     case VT_DISPATCH:
860         *encoded_type = default_type;
861         *width = 4;
862         *alignment = 4;
863         break;
864
865     case VT_VARIANT:
866         *encoded_type = default_type;
867         break;
868
869     case VT_LPSTR:
870     case VT_LPWSTR:
871         *encoded_type = 0xfffe0000 | vt;
872         *width = 4;
873         *alignment = 4;
874         break;
875
876     case VT_PTR:
877       {
878         int next_vt;
879         for(next_vt = 0; is_ptr(type); type = type_pointer_get_ref(type)) {
880             next_vt = get_type_vt(type_pointer_get_ref(type));
881             if (next_vt != 0)
882                 break;
883         }
884         /* if no type found then it must be void */
885         if (next_vt == 0)
886             next_vt = VT_VOID;
887
888         encode_type(typelib, next_vt, type_pointer_get_ref(type),
889                     &target_type, NULL, NULL, &child_size);
890         /* these types already have an implicit pointer, so we don't need to
891          * add another */
892         if(next_vt == VT_DISPATCH || next_vt == VT_UNKNOWN) {
893             chat("encode_type: skipping ptr\n");
894             *encoded_type = target_type;
895             *width = 4;
896             *alignment = 4;
897             *decoded_size = child_size;
898             break;
899         }
900
901         for (typeoffset = 0; typeoffset < typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length; typeoffset += 8) {
902             typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
903             if (((typedata[0] & 0xffff) == VT_PTR) && (typedata[1] == target_type)) break;
904         }
905
906         if (typeoffset == typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length) {
907             int mix_field;
908             
909             if (target_type & 0x80000000) {
910                 mix_field = ((target_type >> 16) & 0x3fff) | VT_BYREF;
911             } else {
912                 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][target_type];
913                 mix_field = ((typedata[0] >> 16) == 0x7fff)? 0x7fff: 0x7ffe;
914             }
915
916             typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
917             typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
918
919             typedata[0] = (mix_field << 16) | VT_PTR;
920             typedata[1] = target_type;
921         }
922
923         *encoded_type = typeoffset;
924
925         *width = 4;
926         *alignment = 4;
927         *decoded_size = 8 /*sizeof(TYPEDESC)*/ + child_size;
928         break;
929     }
930
931
932     case VT_SAFEARRAY:
933         {
934         type_t *element_type = type_alias_get_aliasee(type_array_get_element(type));
935         int next_vt = get_type_vt(element_type);
936
937         encode_type(typelib, next_vt, type_alias_get_aliasee(type_array_get_element(type)), &target_type, NULL, NULL, &child_size);
938
939         for (typeoffset = 0; typeoffset < typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length; typeoffset += 8) {
940             typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
941             if (((typedata[0] & 0xffff) == VT_SAFEARRAY) && (typedata[1] == target_type)) break;
942         }
943
944         if (typeoffset == typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length) {
945             int mix_field;
946             
947             if (target_type & 0x80000000) {
948                 mix_field = ((target_type >> 16) & VT_TYPEMASK) | VT_ARRAY;
949             } else {
950                 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][target_type];
951                 mix_field = ((typedata[0] >> 16) == 0x7fff)? 0x7fff: 0x7ffe;
952             }
953
954             typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
955             typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
956
957             typedata[0] = (mix_field << 16) | VT_SAFEARRAY;
958             typedata[1] = target_type;
959         }
960
961         *encoded_type = typeoffset;
962
963         *width = 4;
964         *alignment = 4;
965         *decoded_size = 8 /*sizeof(TYPEDESC)*/ + child_size;
966         break;
967         }
968
969
970     case VT_USERDEFINED:
971       {
972         int typeinfo_offset;
973
974         /* typedef'd types without public attribute aren't included in the typelib */
975         while (type->typelib_idx < 0 && type_is_alias(type) && !is_attr(type->attrs, ATTR_PUBLIC))
976           type = type_alias_get_aliasee(type);
977
978         chat("encode_type: VT_USERDEFINED - type %p name = %s real type %d idx %d\n", type,
979              type->name, type_get_type(type), type->typelib_idx);
980
981         if(type->typelib_idx == -1) {
982             chat("encode_type: trying to ref not added type\n");
983             switch (type_get_type(type)) {
984             case TYPE_STRUCT:
985                 add_structure_typeinfo(typelib, type);
986                 break;
987             case TYPE_INTERFACE:
988                 add_interface_typeinfo(typelib, type);
989                 break;
990             case TYPE_ENUM:
991                 add_enum_typeinfo(typelib, type);
992                 break;
993             case TYPE_COCLASS:
994                 add_coclass_typeinfo(typelib, type);
995                 break;
996             default:
997                 error("encode_type: VT_USERDEFINED - unhandled type %d\n",
998                       type_get_type(type));
999             }
1000         }
1001
1002         typeinfo_offset = typelib->typelib_typeinfo_offsets[type->typelib_idx];
1003         for (typeoffset = 0; typeoffset < typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length; typeoffset += 8) {
1004             typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1005             if ((typedata[0] == ((0x7fff << 16) | VT_USERDEFINED)) && (typedata[1] == typeinfo_offset)) break;
1006         }
1007
1008         if (typeoffset == typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length) {
1009             typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
1010             typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1011
1012             typedata[0] = (0x7fff << 16) | VT_USERDEFINED;
1013             typedata[1] = typeinfo_offset;
1014         }
1015
1016         *encoded_type = typeoffset;
1017         *width = 0;
1018         *alignment = 1;
1019         break;
1020       }
1021
1022     default:
1023         error("encode_type: unrecognized type %d.\n", vt);
1024         *encoded_type = default_type;
1025         *width = 0;
1026         *alignment = 1;
1027         break;
1028     }
1029
1030     return 0;
1031 }
1032
1033 static void dump_type(type_t *t)
1034 {
1035     chat("dump_type: %p name %s type %d attrs %p\n", t, t->name, type_get_type(t), t->attrs);
1036 }
1037
1038 static int encode_var(
1039         msft_typelib_t *typelib,   /* [I] The type library in which to encode the TYPEDESC. */
1040         type_t *type,              /* [I] The type description to encode. */
1041         var_t *var,                /* [I] The var to encode. */
1042         int *encoded_type,         /* [O] The encoded type description. */
1043         int *width,                /* [O] The width of the type, or NULL. */
1044         int *alignment,            /* [O] The alignment of the type, or NULL. */
1045         int *decoded_size)         /* [O] The total size of the unencoded TYPEDESCs, including nested descs. */
1046 {
1047     int typeoffset;
1048     int *typedata;
1049     int target_type;
1050     int child_size;
1051     int vt;
1052     int scratch;
1053
1054     if (!width) width = &scratch;
1055     if (!alignment) alignment = &scratch;
1056     if (!decoded_size) decoded_size = &scratch;
1057     *decoded_size = 0;
1058
1059     chat("encode_var: var %p type %p type->name %s\n",
1060          var, type, type->name ? type->name : "NULL");
1061
1062     if (is_array(type) && !type_array_is_decl_as_ptr(type)) {
1063         int num_dims, elements = 1, arrayoffset;
1064         type_t *atype;
1065         int *arraydata;
1066
1067         num_dims = 0;
1068         for (atype = type;
1069              is_array(atype) && !type_array_is_decl_as_ptr(atype);
1070              atype = type_array_get_element(atype))
1071             ++num_dims;
1072
1073         chat("array with %d dimensions\n", num_dims);
1074         encode_var(typelib, atype, var, &target_type, width, alignment, NULL);
1075         arrayoffset = ctl2_alloc_segment(typelib, MSFT_SEG_ARRAYDESC, (2 + 2 * num_dims) * sizeof(int), 0);
1076         arraydata = (void *)&typelib->typelib_segment_data[MSFT_SEG_ARRAYDESC][arrayoffset];
1077
1078         arraydata[0] = target_type;
1079         arraydata[1] = num_dims;
1080         arraydata[1] |= ((num_dims * 2 * sizeof(int)) << 16);
1081
1082         arraydata += 2;
1083         for (atype = type;
1084              is_array(atype) && !type_array_is_decl_as_ptr(atype);
1085              atype = type_array_get_element(atype))
1086         {
1087             arraydata[0] = type_array_get_dim(atype);
1088             arraydata[1] = 0;
1089             arraydata += 2;
1090             elements *= type_array_get_dim(atype);
1091         }
1092
1093         typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
1094         typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1095
1096         typedata[0] = (0x7ffe << 16) | VT_CARRAY;
1097         typedata[1] = arrayoffset;
1098
1099         *encoded_type = typeoffset;
1100         *width = *width * elements;
1101         *decoded_size = 20 /*sizeof(ARRAYDESC)*/ + (num_dims - 1) * 8 /*sizeof(SAFEARRAYBOUND)*/;
1102         return 0;
1103     }
1104
1105     vt = get_type_vt(type);
1106     if (vt == VT_PTR) {
1107         type_t *ref = is_ptr(type) ?
1108             type_pointer_get_ref(type) : type_array_get_element(type);
1109         int skip_ptr = encode_var(typelib, ref, var,
1110                                   &target_type, NULL, NULL, &child_size);
1111
1112         if(skip_ptr == 2) {
1113             chat("encode_var: skipping ptr\n");
1114             *encoded_type = target_type;
1115             *decoded_size = child_size;
1116             *width = 4;
1117             *alignment = 4;
1118             return 0;
1119         }
1120
1121         for (typeoffset = 0; typeoffset < typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length; typeoffset += 8) {
1122             typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1123             if (((typedata[0] & 0xffff) == VT_PTR) && (typedata[1] == target_type)) break;
1124         }
1125
1126         if (typeoffset == typelib->typelib_segdir[MSFT_SEG_TYPEDESC].length) {
1127             int mix_field;
1128             
1129             if (target_type & 0x80000000) {
1130                 mix_field = ((target_type >> 16) & 0x3fff) | VT_BYREF;
1131             } else {
1132                 typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][target_type];
1133                 mix_field = ((typedata[0] >> 16) == 0x7fff)? 0x7fff: 0x7ffe;
1134             }
1135
1136             typeoffset = ctl2_alloc_segment(typelib, MSFT_SEG_TYPEDESC, 8, 0);
1137             typedata = (void *)&typelib->typelib_segment_data[MSFT_SEG_TYPEDESC][typeoffset];
1138
1139             typedata[0] = (mix_field << 16) | VT_PTR;
1140             typedata[1] = target_type;
1141         }
1142
1143         *encoded_type = typeoffset;
1144
1145         *width = 4;
1146         *alignment = 4;
1147         *decoded_size = 8 /*sizeof(TYPEDESC)*/ + child_size;
1148         return 0;
1149     }
1150
1151     dump_type(type);
1152
1153     encode_type(typelib, vt, type, encoded_type, width, alignment, decoded_size);
1154     /* these types already have an implicit pointer, so we don't need to
1155      * add another */
1156     if(vt == VT_DISPATCH || vt == VT_UNKNOWN) return 2;
1157     return 0;
1158 }
1159
1160 static unsigned long get_ulong_val(unsigned long val, int vt)
1161 {
1162     switch(vt) {
1163     case VT_I2:
1164     case VT_BOOL:
1165     case VT_UI2:
1166         return val & 0xffff;
1167     case VT_I1:
1168     case VT_UI1:
1169         return val & 0xff;
1170     }
1171
1172     return val;
1173 }
1174
1175 static void write_value(msft_typelib_t* typelib, int *out, int vt, const void *value)
1176 {
1177     switch(vt) {
1178     case VT_I2:
1179     case VT_I4:
1180     case VT_R4:
1181     case VT_BOOL:
1182     case VT_I1:
1183     case VT_UI1:
1184     case VT_UI2:
1185     case VT_UI4:
1186     case VT_INT:
1187     case VT_UINT:
1188     case VT_HRESULT:
1189     case VT_PTR:
1190       {
1191         const unsigned long lv = get_ulong_val(*(const unsigned long*)value, vt);
1192         if((lv & 0x3ffffff) == lv) {
1193             *out = 0x80000000;
1194             *out |= vt << 26;
1195             *out |= lv;
1196         } else {
1197             int offset = ctl2_alloc_segment(typelib, MSFT_SEG_CUSTDATA, 8, 0);
1198             *((unsigned short *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset]) = vt;
1199             memcpy(&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+2], value, 4);
1200             *((unsigned short *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+6]) = 0x5757;
1201             *out = offset;
1202         }
1203         return;
1204       }
1205     case VT_BSTR:
1206       {
1207         const char *s = (const char *) value;
1208         int len = strlen(s), seg_len = (len + 6 + 3) & ~0x3;
1209         int offset = ctl2_alloc_segment(typelib, MSFT_SEG_CUSTDATA, seg_len, 0);
1210         *((unsigned short *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset]) = vt;
1211         *((unsigned int *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+2]) = len;        
1212         memcpy(&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+6], value, len);
1213         len += 6;
1214         while(len < seg_len) {
1215             *((char *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATA][offset+len]) = 0x57;
1216             len++;
1217         }
1218         *out = offset;
1219         return;
1220       }
1221
1222     default:
1223         warning("can't write value of type %d yet\n", vt);
1224     }
1225     return;
1226 }
1227
1228 static HRESULT set_custdata(msft_typelib_t *typelib, REFGUID guid,
1229                             int vt, void *value, int *offset)
1230 {
1231     MSFT_GuidEntry guidentry;
1232     int guidoffset;
1233     int custoffset;
1234     int *custdata;
1235     int data_out;
1236
1237     guidentry.guid = *guid;
1238
1239     guidentry.hreftype = -1;
1240     guidentry.next_hash = -1;
1241
1242     guidoffset = ctl2_alloc_guid(typelib, &guidentry);
1243     write_value(typelib, &data_out, vt, value);
1244
1245     custoffset = ctl2_alloc_segment(typelib, MSFT_SEG_CUSTDATAGUID, 12, 0);
1246
1247     custdata = (int *)&typelib->typelib_segment_data[MSFT_SEG_CUSTDATAGUID][custoffset];
1248     custdata[0] = guidoffset;
1249     custdata[1] = data_out;
1250     custdata[2] = *offset;
1251     *offset = custoffset;
1252
1253     return S_OK;
1254 }
1255
1256 static HRESULT add_func_desc(msft_typeinfo_t* typeinfo, var_t *func, int index)
1257 {
1258     int offset, name_offset;
1259     int *typedata, typedata_size;
1260     int i, id, next_idx;
1261     int decoded_size, extra_attr = 0;
1262     int num_params = 0, num_optional = 0, num_defaults = 0;
1263     var_t *arg;
1264     char *namedata;
1265     const attr_t *attr;
1266     unsigned int funcflags = 0, callconv = 4 /* CC_STDCALL */;
1267     unsigned int funckind, invokekind = 1 /* INVOKE_FUNC */;
1268     int help_context = 0, help_string_context = 0, help_string_offset = -1;
1269     int entry = -1, entry_is_ord = 0;
1270     int lcid_retval_count = 0;
1271
1272     chat("add_func_desc(%p,%d)\n", typeinfo, index);
1273
1274     id = ((0x6000 | (typeinfo->typeinfo->datatype2 & 0xffff)) << 16) | index;
1275
1276     switch(typeinfo->typekind) {
1277     case TKIND_DISPATCH:
1278         funckind = 0x4; /* FUNC_DISPATCH */
1279         break;
1280     case TKIND_MODULE:
1281         funckind = 0x3; /* FUNC_STATIC */
1282         break;
1283     default:
1284         funckind = 0x1; /* FUNC_PUREVIRTUAL */
1285         break;
1286     }
1287
1288     if (is_local( func->attrs )) {
1289         chat("add_func_desc: skipping local function\n");
1290         return S_FALSE;
1291     }
1292
1293     if (type_get_function_args(func->type))
1294       LIST_FOR_EACH_ENTRY( arg, type_get_function_args(func->type), var_t, entry )
1295       {
1296         num_params++;
1297         if (arg->attrs) LIST_FOR_EACH_ENTRY( attr, arg->attrs, const attr_t, entry ) {
1298             if(attr->type == ATTR_DEFAULTVALUE)
1299                 num_defaults++;
1300             else if(attr->type == ATTR_OPTIONAL)
1301                 num_optional++;
1302         }
1303       }
1304
1305     chat("add_func_desc: num of params %d\n", num_params);
1306
1307     name_offset = ctl2_alloc_name(typeinfo->typelib, func->name);
1308
1309     if (func->attrs) LIST_FOR_EACH_ENTRY( attr, func->attrs, const attr_t, entry ) {
1310         expr_t *expr = attr->u.pval;
1311         switch(attr->type) {
1312         case ATTR_BINDABLE:
1313             funcflags |= 0x4; /* FUNCFLAG_FBINDABLE */
1314             break;
1315         /* FIXME: FUNCFLAG_FDEFAULTBIND */
1316         case ATTR_DEFAULTCOLLELEM:
1317             funcflags |= 0x100; /* FUNCFLAG_FDEFAULTCOLLELEM */
1318             break;
1319         case ATTR_DISPLAYBIND:
1320             funcflags |= 0x10; /* FUNCFLAG_FDISPLAYBIND */
1321             break;
1322         case ATTR_ENTRY:
1323             extra_attr = max(extra_attr, 3);
1324             if (expr->type == EXPR_STRLIT || expr->type == EXPR_WSTRLIT)
1325               entry = ctl2_alloc_string(typeinfo->typelib, attr->u.pval);
1326             else {
1327               entry = expr->cval;
1328               entry_is_ord = 1;
1329             }
1330             break;
1331         case ATTR_HELPCONTEXT:
1332             extra_attr = max(extra_attr, 1);
1333             help_context = expr->u.lval;
1334             break;
1335         case ATTR_HELPSTRING:
1336             extra_attr = max(extra_attr, 2);
1337             help_string_offset = ctl2_alloc_string(typeinfo->typelib, attr->u.pval);
1338             break;
1339         case ATTR_HELPSTRINGCONTEXT:
1340             extra_attr = max(extra_attr, 6);
1341             help_string_context = expr->u.lval;
1342             break;
1343         case ATTR_HIDDEN:
1344             funcflags |= 0x40; /* FUNCFLAG_FHIDDEN */
1345             break;
1346         case ATTR_ID:
1347             id = expr->cval;
1348             break;
1349         case ATTR_IMMEDIATEBIND:
1350             funcflags |= 0x1000; /* FUNCFLAG_FIMMEDIATEBIND */
1351             break;
1352         case ATTR_NONBROWSABLE:
1353             funcflags |= 0x400; /* FUNCFLAG_FNONBROWSABLE */
1354             break;
1355         case ATTR_OUT:
1356             break;
1357         case ATTR_PROPGET:
1358             invokekind = 0x2; /* INVOKE_PROPERTYGET */
1359             break;
1360         case ATTR_PROPPUT:
1361             invokekind = 0x4; /* INVOKE_PROPERTYPUT */
1362             break;
1363         case ATTR_PROPPUTREF:
1364             invokekind = 0x8; /* INVOKE_PROPERTYPUTREF */
1365             break;
1366         /* FIXME: FUNCFLAG_FREPLACEABLE */
1367         case ATTR_REQUESTEDIT:
1368             funcflags |= 0x8; /* FUNCFLAG_FREQUESTEDIT */
1369             break;
1370         case ATTR_RESTRICTED:
1371             funcflags |= 0x1; /* FUNCFLAG_FRESTRICTED */
1372             break;
1373         case ATTR_SOURCE:
1374             funcflags |= 0x2; /* FUNCFLAG_FSOURCE */
1375             break;
1376         /* FIXME: FUNCFLAG_FUIDEFAULT */
1377         /* FIXME: FUNCFLAG_FUSESGETLASTERROR */
1378         case ATTR_VARARG:
1379             if (num_optional || num_defaults)
1380                 warning("add_func_desc: ignoring vararg in function with optional or defaultvalue params\n");
1381             else
1382                 num_optional = -1;
1383             break;
1384         default:
1385             break;
1386         }
1387     }
1388
1389     /* allocate type data space for us */
1390     typedata_size = 0x18 + extra_attr * sizeof(int) + (num_params * (num_defaults ? 16 : 12));
1391
1392     if (!typeinfo->func_data) {
1393         typeinfo->func_data = xmalloc(0x100);
1394         typeinfo->func_data_allocated = 0x100;
1395         typeinfo->func_data[0] = 0;
1396     }
1397
1398     if(typeinfo->func_data[0] + typedata_size + sizeof(int) > typeinfo->func_data_allocated) {
1399         typeinfo->func_data_allocated = max(typeinfo->func_data_allocated * 2,
1400                                             typeinfo->func_data[0] + typedata_size + sizeof(int));
1401         typeinfo->func_data = xrealloc(typeinfo->func_data, typeinfo->func_data_allocated);
1402     }
1403
1404     offset = typeinfo->func_data[0];
1405     typeinfo->func_data[0] += typedata_size;
1406     typedata = typeinfo->func_data + (offset >> 2) + 1;
1407
1408
1409     /* find func with the same name - if it exists use its id */
1410     for(i = 0; i < (typeinfo->typeinfo->cElement & 0xffff); i++) {
1411         if(name_offset == typeinfo->func_names[i]) {
1412             id = typeinfo->func_indices[i];
1413             break;
1414         }
1415     }
1416
1417     /* find the first func with the same id and link via the hiword of typedata[4] */
1418     next_idx = index;
1419     for(i = 0; i < (typeinfo->typeinfo->cElement & 0xffff); i++) {
1420         if(id == typeinfo->func_indices[i]) {
1421             next_idx = typeinfo->func_data[(typeinfo->func_offsets[i] >> 2) + 1 + 4] >> 16;
1422             typeinfo->func_data[(typeinfo->func_offsets[i] >> 2) + 1 + 4] &= 0xffff;
1423             typeinfo->func_data[(typeinfo->func_offsets[i] >> 2) + 1 + 4] |= (index << 16);
1424             break;
1425         }
1426     }
1427
1428     /* fill out the basic type information */
1429     typedata[0] = typedata_size | (index << 16);
1430     encode_var(typeinfo->typelib, type_function_get_rettype(func->type), func, &typedata[1], NULL, NULL, &decoded_size);
1431     typedata[2] = funcflags;
1432     typedata[3] = ((52 /*sizeof(FUNCDESC)*/ + decoded_size) << 16) | typeinfo->typeinfo->cbSizeVft;
1433     typedata[4] = (next_idx << 16) | (callconv << 8) | (invokekind << 3) | funckind;
1434     if(num_defaults) typedata[4] |= 0x1000;
1435     if(entry_is_ord) typedata[4] |= 0x2000;
1436     typedata[5] = (num_optional << 16) | num_params;
1437
1438     /* NOTE: High word of typedata[3] is total size of FUNCDESC + size of all ELEMDESCs for params + TYPEDESCs for pointer params and return types. */
1439     /* That is, total memory allocation required to reconstitute the FUNCDESC in its entirety. */
1440     typedata[3] += (16 /*sizeof(ELEMDESC)*/ * num_params) << 16;
1441     typedata[3] += (24 /*sizeof(PARAMDESCEX)*/ * num_defaults) << 16;
1442
1443     switch(extra_attr) {
1444     case 6: typedata[11] = help_string_context;
1445     case 5: typedata[10] = -1;
1446     case 4: typedata[9] = -1;
1447     case 3: typedata[8] = entry;
1448     case 2: typedata[7] = help_string_offset;
1449     case 1: typedata[6] = help_context;
1450     case 0:
1451         break;
1452     default:
1453         warning("unknown number of optional attrs\n");
1454     }
1455
1456     if (type_get_function_args(func->type))
1457     {
1458       i = 0;
1459       LIST_FOR_EACH_ENTRY( arg, type_get_function_args(func->type), var_t, entry )
1460       {
1461         int paramflags = 0;
1462         int *paramdata = typedata + 6 + extra_attr + (num_defaults ? num_params : 0) + i * 3;
1463         int *defaultdata = num_defaults ? typedata + 6 + extra_attr + i : NULL;
1464
1465         if(defaultdata) *defaultdata = -1;
1466
1467         encode_var(typeinfo->typelib, arg->type, arg, paramdata, NULL, NULL, &decoded_size);
1468         if (arg->attrs) LIST_FOR_EACH_ENTRY( attr, arg->attrs, const attr_t, entry ) {
1469             switch(attr->type) {
1470             case ATTR_DEFAULTVALUE:
1471               {
1472                 int vt;
1473                 expr_t *expr = (expr_t *)attr->u.pval;
1474                 if (type_get_type(arg->type) == TYPE_ENUM)
1475                     vt = VT_INT;
1476                 else
1477                     vt = get_type_vt(arg->type);
1478                 paramflags |= 0x30; /* PARAMFLAG_FHASDEFAULT | PARAMFLAG_FOPT */
1479                 if (expr->type == EXPR_STRLIT || expr->type == EXPR_WSTRLIT)
1480                 {
1481                   if (vt != VT_BSTR) error("string default value applied to non-string type\n");
1482                   chat("default value '%s'\n", expr->u.sval);
1483                   write_value(typeinfo->typelib, defaultdata, vt, expr->u.sval);
1484                 }
1485                 else
1486                 {
1487                   chat("default value %ld\n", expr->cval);
1488                   write_value(typeinfo->typelib, defaultdata, vt, &expr->cval);
1489                 }
1490                 break;
1491               }
1492             case ATTR_IN:
1493                 paramflags |= 0x01; /* PARAMFLAG_FIN */
1494                 break;
1495             case ATTR_OPTIONAL:
1496                 paramflags |= 0x10; /* PARAMFLAG_FOPT */
1497                 break;
1498             case ATTR_OUT:
1499                 paramflags |= 0x02; /* PARAMFLAG_FOUT */
1500                 break;
1501             case ATTR_PARAMLCID:
1502                 paramflags |= 0x04; /* PARAMFLAG_LCID */
1503                 lcid_retval_count++;
1504                 break;
1505             case ATTR_RETVAL:
1506                 paramflags |= 0x08; /* PARAMFLAG_FRETVAL */
1507                 lcid_retval_count++;
1508                 break;
1509             default:
1510                 chat("unhandled param attr %d\n", attr->type);
1511                 break;
1512             }
1513         }
1514         paramdata[1] = -1;
1515         paramdata[2] = paramflags;
1516         typedata[3] += decoded_size << 16;
1517
1518         i++;
1519       }
1520     }
1521
1522     if(lcid_retval_count == 1)
1523         typedata[4] |= 0x4000;
1524     else if(lcid_retval_count == 2)
1525         typedata[4] |= 0x8000;
1526
1527     if(typeinfo->funcs_allocated == 0) {
1528         typeinfo->funcs_allocated = 10;
1529         typeinfo->func_indices = xmalloc(typeinfo->funcs_allocated * sizeof(int));
1530         typeinfo->func_names   = xmalloc(typeinfo->funcs_allocated * sizeof(int));
1531         typeinfo->func_offsets = xmalloc(typeinfo->funcs_allocated * sizeof(int));
1532     }
1533     if(typeinfo->funcs_allocated == (typeinfo->typeinfo->cElement & 0xffff)) {
1534         typeinfo->funcs_allocated *= 2;
1535         typeinfo->func_indices = xrealloc(typeinfo->func_indices, typeinfo->funcs_allocated * sizeof(int));
1536         typeinfo->func_names   = xrealloc(typeinfo->func_names,   typeinfo->funcs_allocated * sizeof(int));
1537         typeinfo->func_offsets = xrealloc(typeinfo->func_offsets, typeinfo->funcs_allocated * sizeof(int));
1538     }
1539
1540     /* update the index data */
1541     typeinfo->func_indices[typeinfo->typeinfo->cElement & 0xffff] = id; 
1542     typeinfo->func_offsets[typeinfo->typeinfo->cElement & 0xffff] = offset;
1543     typeinfo->func_names[typeinfo->typeinfo->cElement & 0xffff] = name_offset;
1544
1545     /* ??? */
1546     if (!typeinfo->typeinfo->res2) typeinfo->typeinfo->res2 = 0x20;
1547     typeinfo->typeinfo->res2 <<= 1;
1548     /* ??? */
1549     if (index < 2) typeinfo->typeinfo->res2 += num_params << 4;
1550
1551     if (typeinfo->typeinfo->res3 == -1) typeinfo->typeinfo->res3 = 0;
1552     typeinfo->typeinfo->res3 += 0x38 + num_params * 0x10;
1553     if(num_defaults) typeinfo->typeinfo->res3 += num_params * 0x4;
1554
1555     /* adjust size of VTBL */
1556     if(funckind != 0x3 /* FUNC_STATIC */)
1557         typeinfo->typeinfo->cbSizeVft += pointer_size;
1558
1559     /* Increment the number of function elements */
1560     typeinfo->typeinfo->cElement += 1;
1561
1562     namedata = typeinfo->typelib->typelib_segment_data[MSFT_SEG_NAME] + name_offset;
1563     if (*((INT *)namedata) == -1) {
1564         *((INT *)namedata) = typeinfo->typelib->typelib_typeinfo_offsets[typeinfo->typeinfo->typekind >> 16];
1565         if(typeinfo->typekind == TKIND_MODULE)
1566             namedata[9] |= 0x10;
1567     } else
1568         namedata[9] &= ~0x10;
1569
1570     if(typeinfo->typekind == TKIND_MODULE)
1571         namedata[9] |= 0x20;
1572
1573     if (type_get_function_args(func->type))
1574     {
1575         i = 0;
1576         LIST_FOR_EACH_ENTRY( arg, type_get_function_args(func->type), var_t, entry )
1577         {
1578             /* don't give the last arg of a [propput*] func a name */
1579             if(i != num_params - 1 || (invokekind != 0x4 /* INVOKE_PROPERTYPUT */ && invokekind != 0x8 /* INVOKE_PROPERTYPUTREF */))
1580             {
1581                 int *paramdata = typedata + 6 + extra_attr + (num_defaults ? num_params : 0) + i * 3;
1582                 offset = ctl2_alloc_name(typeinfo->typelib, arg->name);
1583                 paramdata[1] = offset;
1584             }
1585             i++;
1586         }
1587     }
1588     return S_OK;
1589 }
1590
1591 static HRESULT add_var_desc(msft_typeinfo_t *typeinfo, UINT index, var_t* var)
1592 {
1593     int offset, id;
1594     unsigned int typedata_size;
1595     INT *typedata;
1596     int var_datawidth;
1597     int var_alignment;
1598     int var_type_size, var_kind = 0 /* VAR_PERINSTANCE */; 
1599     int alignment;
1600     int varflags = 0;
1601     const attr_t *attr;
1602     char *namedata;
1603     int var_num = (typeinfo->typeinfo->cElement >> 16) & 0xffff;
1604
1605     chat("add_var_desc(%d, %s)\n", index, var->name);
1606
1607     id = 0x40000000 + index;
1608
1609     if (var->attrs) LIST_FOR_EACH_ENTRY( attr, var->attrs, const attr_t, entry ) {
1610         expr_t *expr = attr->u.pval;
1611         switch(attr->type) {
1612         case ATTR_BINDABLE:
1613             varflags |= 0x04; /* VARFLAG_FBINDABLE */
1614             break;
1615         /* FIXME: VARFLAG_FDEFAULTBIND */
1616         case ATTR_DEFAULTCOLLELEM:
1617             varflags |= 0x100; /* VARFLAG_FDEFAULTCOLLELEM */
1618             break;
1619         case ATTR_DISPLAYBIND:
1620             varflags |= 0x10; /* VARFLAG_FDISPLAYBIND */
1621             break;
1622         case ATTR_HIDDEN:
1623             varflags |= 0x40; /* VARFLAG_FHIDDEN */
1624             break;
1625         case ATTR_ID:
1626             id = expr->cval;
1627             break;
1628         case ATTR_IMMEDIATEBIND:
1629             varflags |= 0x1000; /* VARFLAG_FIMMEDIATEBIND */
1630             break;
1631         case ATTR_NONBROWSABLE:
1632             varflags |= 0x400; /* VARFLAG_FNONBROWSABLE */
1633             break;
1634         case ATTR_READONLY:
1635             varflags |= 0x01; /* VARFLAG_FREADONLY */
1636             break;
1637         /* FIXME: VARFLAG_FREPLACEABLE */
1638         case ATTR_REQUESTEDIT:
1639             varflags |= 0x08; /* VARFLAG_FREQUESTEDIT */
1640             break;
1641         case ATTR_RESTRICTED:
1642             varflags |= 0x80; /* VARFLAG_FRESTRICTED */
1643             break;
1644         case ATTR_SOURCE:
1645             varflags |= 0x02; /* VARFLAG_FSOURCE */
1646             break;
1647         /* FIXME: VARFLAG_FUIDEFAULT */
1648         default:
1649             break;
1650         }
1651     }
1652
1653     /* allocate type data space for us */
1654     typedata_size = 0x14;
1655
1656     if (!typeinfo->var_data) {
1657         typeinfo->var_data = xmalloc(0x100);
1658         typeinfo->var_data_allocated = 0x100;
1659         typeinfo->var_data[0] = 0;
1660     }
1661
1662     if(typeinfo->var_data[0] + typedata_size + sizeof(int) > typeinfo->var_data_allocated) {
1663         typeinfo->var_data_allocated = max(typeinfo->var_data_allocated * 2,
1664                                             typeinfo->var_data[0] + typedata_size + sizeof(int));
1665         typeinfo->var_data = xrealloc(typeinfo->var_data, typeinfo->var_data_allocated);
1666     }
1667
1668     offset = typeinfo->var_data[0];
1669     typeinfo->var_data[0] += typedata_size;
1670     typedata = typeinfo->var_data + (offset >> 2) + 1;
1671
1672     /* fill out the basic type information */
1673     typedata[0] = typedata_size | (index << 16);
1674     typedata[2] = varflags;
1675     typedata[3] = (36 /*sizeof(VARDESC)*/ << 16) | 0;
1676
1677     if(typeinfo->vars_allocated == 0) {
1678         typeinfo->vars_allocated = 10;
1679         typeinfo->var_indices = xmalloc(typeinfo->vars_allocated * sizeof(int));
1680         typeinfo->var_names   = xmalloc(typeinfo->vars_allocated * sizeof(int));
1681         typeinfo->var_offsets = xmalloc(typeinfo->vars_allocated * sizeof(int));
1682     }
1683     if(typeinfo->vars_allocated == var_num) {
1684         typeinfo->vars_allocated *= 2;
1685         typeinfo->var_indices = xrealloc(typeinfo->var_indices, typeinfo->vars_allocated * sizeof(int));
1686         typeinfo->var_names   = xrealloc(typeinfo->var_names,   typeinfo->vars_allocated * sizeof(int));
1687         typeinfo->var_offsets = xrealloc(typeinfo->var_offsets, typeinfo->vars_allocated * sizeof(int));
1688     }
1689     /* update the index data */
1690     typeinfo->var_indices[var_num] = id;
1691     typeinfo->var_names[var_num] = -1;
1692     typeinfo->var_offsets[var_num] = offset;
1693
1694     /* figure out type widths and whatnot */
1695     encode_var(typeinfo->typelib, var->type, var, &typedata[1], &var_datawidth,
1696                &var_alignment, &var_type_size);
1697
1698     /* pad out starting position to data width */
1699     typeinfo->datawidth += var_alignment - 1;
1700     typeinfo->datawidth &= ~(var_alignment - 1);
1701
1702     switch(typeinfo->typekind) {
1703     case TKIND_ENUM:
1704         write_value(typeinfo->typelib, &typedata[4], VT_I4, &var->eval->cval);
1705         var_kind = 2; /* VAR_CONST */
1706         var_type_size += 16; /* sizeof(VARIANT) */
1707         typeinfo->datawidth = var_datawidth;
1708         break;
1709     case TKIND_RECORD:
1710         typedata[4] = typeinfo->datawidth;
1711         typeinfo->datawidth += var_datawidth;
1712         break;
1713     case TKIND_DISPATCH:
1714         var_kind = 3; /* VAR_DISPATCH */
1715         typeinfo->datawidth = 4;
1716         var_alignment = 4;
1717         break;
1718     default:
1719         error("add_var_desc: unhandled type kind %d\n", typeinfo->typekind);
1720         break;
1721     }
1722
1723     /* add type description size to total required allocation */
1724     typedata[3] += var_type_size << 16 | var_kind;
1725
1726     /* fix type alignment */
1727     alignment = (typeinfo->typeinfo->typekind >> 11) & 0x1f;
1728     if (alignment < var_alignment) {
1729         alignment = var_alignment;
1730         typeinfo->typeinfo->typekind &= ~0xffc0;
1731         typeinfo->typeinfo->typekind |= alignment << 11 | alignment << 6;
1732     }
1733
1734     /* ??? */
1735     if (!typeinfo->typeinfo->res2) typeinfo->typeinfo->res2 = 0x1a;
1736     if ((index == 0) || (index == 1) || (index == 2) || (index == 4) || (index == 9)) {
1737         typeinfo->typeinfo->res2 <<= 1;
1738     }
1739
1740     /* ??? */
1741     if (typeinfo->typeinfo->res3 == -1) typeinfo->typeinfo->res3 = 0;
1742     typeinfo->typeinfo->res3 += 0x2c;
1743
1744     /* increment the number of variable elements */
1745     typeinfo->typeinfo->cElement += 0x10000;
1746
1747     /* pad data width to alignment */
1748     typeinfo->typeinfo->size = (typeinfo->datawidth + (alignment - 1)) & ~(alignment - 1);
1749
1750     offset = ctl2_alloc_name(typeinfo->typelib, var->name);
1751     if (offset == -1) return E_OUTOFMEMORY;
1752
1753     namedata = typeinfo->typelib->typelib_segment_data[MSFT_SEG_NAME] + offset;
1754     if (*((INT *)namedata) == -1) {
1755         *((INT *)namedata) = typeinfo->typelib->typelib_typeinfo_offsets[typeinfo->typeinfo->typekind >> 16];
1756         if(typeinfo->typekind != TKIND_DISPATCH)
1757             namedata[9] |= 0x10;
1758     } else
1759         namedata[9] &= ~0x10;
1760
1761     if (typeinfo->typekind == TKIND_ENUM) {
1762         namedata[9] |= 0x20;
1763     }
1764     typeinfo->var_names[var_num] = offset;
1765
1766     return S_OK;
1767 }
1768
1769 static HRESULT add_impl_type(msft_typeinfo_t *typeinfo, type_t *ref, importinfo_t *importinfo)
1770 {
1771     if(importinfo) {
1772         alloc_importinfo(typeinfo->typelib, importinfo);
1773         typeinfo->typeinfo->datatype1 = importinfo->offset+1;
1774     }else {
1775         if(ref->typelib_idx == -1)
1776             add_interface_typeinfo(typeinfo->typelib, ref);
1777         if(ref->typelib_idx == -1)
1778             error("add_impl_type: unable to add inherited interface\n");
1779
1780         typeinfo->typeinfo->datatype1 = typeinfo->typelib->typelib_typeinfo_offsets[ref->typelib_idx];
1781     }
1782
1783     typeinfo->typeinfo->cImplTypes++;
1784     return S_OK;
1785 }
1786
1787 static msft_typeinfo_t *create_msft_typeinfo(msft_typelib_t *typelib, enum type_kind kind,
1788                                              const char *name, const attr_list_t *attrs)
1789 {
1790     const attr_t *attr;
1791     msft_typeinfo_t *msft_typeinfo;
1792     int nameoffset;
1793     int typeinfo_offset;
1794     MSFT_TypeInfoBase *typeinfo;
1795     MSFT_GuidEntry guidentry;
1796
1797     chat("create_msft_typeinfo: name %s kind %d\n", name, kind);
1798
1799     msft_typeinfo = xmalloc(sizeof(*msft_typeinfo));
1800     memset( msft_typeinfo, 0, sizeof(*msft_typeinfo) );
1801
1802     msft_typeinfo->typelib = typelib;
1803
1804     nameoffset = ctl2_alloc_name(typelib, name);
1805     typeinfo_offset = ctl2_alloc_typeinfo(typelib, nameoffset);
1806     typeinfo = (MSFT_TypeInfoBase *)&typelib->typelib_segment_data[MSFT_SEG_TYPEINFO][typeinfo_offset];
1807
1808     typelib->typelib_segment_data[MSFT_SEG_NAME][nameoffset + 9] = 0x38;
1809     *((int *)&typelib->typelib_segment_data[MSFT_SEG_NAME][nameoffset]) = typeinfo_offset;
1810
1811     msft_typeinfo->typekind = kind;
1812     msft_typeinfo->typeinfo = typeinfo;
1813
1814     typeinfo->typekind |= kind | 0x20;
1815
1816     if(kind == TKIND_COCLASS)
1817         typeinfo->flags |= 0x2; /* TYPEFLAG_FCANCREATE */
1818
1819     if (attrs) LIST_FOR_EACH_ENTRY( attr, attrs, const attr_t, entry ) {
1820         switch(attr->type) {
1821         case ATTR_AGGREGATABLE:
1822             if (kind == TKIND_COCLASS)
1823                 typeinfo->flags |= 0x400; /* TYPEFLAG_FAGGREGATABLE */
1824             break;
1825
1826         case ATTR_APPOBJECT:
1827             if (kind == TKIND_COCLASS)
1828                 typeinfo->flags |= 0x1; /* TYPEFLAG_FAPPOBJECT */
1829             break;
1830
1831         case ATTR_CONTROL:
1832             if (kind == TKIND_COCLASS)
1833                 typeinfo->flags |= 0x20; /* TYPEFLAG_FCONTROL */
1834             break;
1835
1836         case ATTR_DLLNAME:
1837           {
1838             int offset = ctl2_alloc_string(typelib, attr->u.pval);
1839             typeinfo->datatype1 = offset;
1840             break;
1841           }
1842
1843         case ATTR_DUAL:
1844             /* FIXME: check interface is compatible */
1845             typeinfo->typekind = (typeinfo->typekind & ~0xff) | 0x34;
1846             typeinfo->flags |= 0x140; /* TYPEFLAG_FDUAL | TYPEFLAG_FOLEAUTOMATION */
1847             break;
1848
1849         case ATTR_HELPCONTEXT:
1850           {
1851             expr_t *expr = (expr_t*)attr->u.pval;
1852             typeinfo->helpcontext = expr->cval;
1853             break;
1854           }
1855         case ATTR_HELPSTRING:
1856           {
1857             int offset = ctl2_alloc_string(typelib, attr->u.pval);
1858             if (offset == -1) break;
1859             typeinfo->docstringoffs = offset;
1860             break;
1861           }
1862         case ATTR_HELPSTRINGCONTEXT:
1863           {
1864             expr_t *expr = (expr_t*)attr->u.pval;
1865             typeinfo->helpstringcontext = expr->cval;
1866             break;
1867           }
1868         case ATTR_HIDDEN:
1869             typeinfo->flags |= 0x10; /* TYPEFLAG_FHIDDEN */
1870             break;
1871
1872         /* FIXME: TYPEFLAG_FLICENSED */
1873
1874         case ATTR_NONCREATABLE:
1875             typeinfo->flags &= ~0x2; /* TYPEFLAG_FCANCREATE */
1876             break;
1877
1878         case ATTR_NONEXTENSIBLE:
1879             typeinfo->flags |= 0x80; /* TYPEFLAG_FNONEXTENSIBLE */
1880             break;
1881
1882         case ATTR_OLEAUTOMATION:
1883             typeinfo->flags |= 0x100; /* TYPEFLAG_FOLEAUTOMATION */
1884             break;
1885
1886         /* FIXME: TYPEFLAG_FPREDCLID */
1887
1888         /* FIXME: TYPEFLAG_FPROXY */
1889
1890         /* FIXME: TYPEFLAG_FREPLACEABLE */
1891
1892         case ATTR_RESTRICTED:
1893             typeinfo->flags |= 0x200; /* TYPEFLAG_FRESTRICTED */
1894             break;
1895
1896         case ATTR_UUID:
1897             guidentry.guid = *(GUID*)attr->u.pval;
1898             guidentry.hreftype = typelib->typelib_typeinfo_offsets[typeinfo->typekind >> 16];
1899             guidentry.next_hash = -1;
1900             typeinfo->posguid = ctl2_alloc_guid(typelib, &guidentry);
1901 #if 0
1902             if (IsEqualIID(guid, &IID_IDispatch)) {
1903                 typelib->typelib_header.dispatchpos = typelib->typelib_typeinfo_offsets[typeinfo->typekind >> 16];
1904             }
1905 #endif
1906             break;
1907
1908         case ATTR_VERSION:
1909             typeinfo->version = attr->u.ival;
1910             break;
1911
1912         default:
1913             break;
1914         }
1915     }
1916
1917     if (typelib->last_typeinfo) typelib->last_typeinfo->next_typeinfo = msft_typeinfo;
1918     typelib->last_typeinfo = msft_typeinfo;
1919     if (!typelib->typeinfos) typelib->typeinfos = msft_typeinfo;
1920
1921
1922     return msft_typeinfo;
1923 }
1924
1925 static void add_dispatch(msft_typelib_t *typelib)
1926 {
1927     int guid_offset, impfile_offset;
1928     MSFT_GuidEntry guidentry;
1929     MSFT_ImpInfo impinfo;
1930     GUID stdole =        {0x00020430,0x0000,0x0000,{0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46}};
1931     GUID iid_idispatch = {0x00020400,0x0000,0x0000,{0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46}};
1932
1933     if(typelib->typelib_header.dispatchpos != -1) return;
1934
1935     guidentry.guid = stdole;
1936     guidentry.hreftype = 2;
1937     guidentry.next_hash = -1;
1938     guid_offset = ctl2_alloc_guid(typelib, &guidentry);
1939     impfile_offset = alloc_importfile(typelib, guid_offset, 2, 0, "stdole2.tlb");
1940
1941     guidentry.guid = iid_idispatch;
1942     guidentry.hreftype = 1;
1943     guidentry.next_hash = -1;
1944     impinfo.flags = TKIND_INTERFACE << 24 | MSFT_IMPINFO_OFFSET_IS_GUID;
1945     impinfo.oImpFile = impfile_offset;
1946     impinfo.oGuid = ctl2_alloc_guid(typelib, &guidentry);
1947     typelib->typelib_header.dispatchpos = alloc_msft_importinfo(typelib, &impinfo) | 0x01;
1948 }
1949
1950 static void add_dispinterface_typeinfo(msft_typelib_t *typelib, type_t *dispinterface)
1951 {
1952     int idx = 0;
1953     const func_t *func;
1954     var_t *var;
1955     msft_typeinfo_t *msft_typeinfo;
1956
1957     if (-1 < dispinterface->typelib_idx)
1958         return;
1959
1960     dispinterface->typelib_idx = typelib->typelib_header.nrtypeinfos;
1961     msft_typeinfo = create_msft_typeinfo(typelib, TKIND_DISPATCH, dispinterface->name,
1962                                          dispinterface->attrs);
1963
1964     msft_typeinfo->typeinfo->size = 4;
1965     msft_typeinfo->typeinfo->typekind |= 0x2100;
1966
1967     msft_typeinfo->typeinfo->flags |= 0x1000; /* TYPEFLAG_FDISPATCHABLE */
1968     add_dispatch(typelib);
1969     msft_typeinfo->typeinfo->cImplTypes = 1;
1970
1971     /* count the no of methods, as the variable indices come after the funcs */
1972     if (dispinterface->details.iface->disp_methods)
1973         LIST_FOR_EACH_ENTRY( func, dispinterface->details.iface->disp_methods, const func_t, entry )
1974             idx++;
1975
1976     if (type_dispiface_get_props(dispinterface))
1977         LIST_FOR_EACH_ENTRY( var, type_dispiface_get_props(dispinterface), var_t, entry )
1978             add_var_desc(msft_typeinfo, idx++, var);
1979
1980     if (type_dispiface_get_methods(dispinterface))
1981     {
1982         idx = 0;
1983         LIST_FOR_EACH_ENTRY( func, type_dispiface_get_methods(dispinterface), const func_t, entry )
1984             if(add_func_desc(msft_typeinfo, func->def, idx) == S_OK)
1985                 idx++;
1986     }
1987 }
1988
1989 static void add_interface_typeinfo(msft_typelib_t *typelib, type_t *interface)
1990 {
1991     int idx = 0;
1992     const statement_t *stmt_func;
1993     type_t *ref;
1994     msft_typeinfo_t *msft_typeinfo;
1995     importinfo_t *ref_importinfo = NULL;
1996     int num_parents = 0, num_funcs = 0;
1997     type_t *inherit;
1998     const type_t *derived;
1999
2000     if (-1 < interface->typelib_idx)
2001         return;
2002
2003     if (is_attr(interface->attrs, ATTR_DISPINTERFACE))
2004         return add_dispinterface_typeinfo(typelib, interface);
2005
2006     /* midl adds the parent interface first, unless the parent itself
2007        has no parent (i.e. it stops before IUnknown). */
2008
2009     inherit = type_iface_get_inherit(interface);
2010
2011     if(inherit) {
2012         ref_importinfo = find_importinfo(typelib, inherit->name);
2013
2014         if(!ref_importinfo && type_iface_get_inherit(inherit) &&
2015            inherit->typelib_idx == -1)
2016             add_interface_typeinfo(typelib, inherit);
2017     }
2018
2019     interface->typelib_idx = typelib->typelib_header.nrtypeinfos;
2020     msft_typeinfo = create_msft_typeinfo(typelib, TKIND_INTERFACE, interface->name, interface->attrs);
2021     msft_typeinfo->typeinfo->size = 4;
2022     msft_typeinfo->typeinfo->typekind |= 0x2200;
2023
2024     for (derived = inherit; derived; derived = type_iface_get_inherit(derived))
2025         if (derived->name && !strcmp(derived->name, "IDispatch"))
2026             msft_typeinfo->typeinfo->flags |= 0x1000; /* TYPEFLAG_FDISPATCHABLE */
2027
2028     /* can't be dual if it doesn't derive from IDispatch */
2029     if (!(msft_typeinfo->typeinfo->flags & 0x1000)) /* TYPEFLAG_FDISPATCHABLE */
2030         msft_typeinfo->typeinfo->flags &= ~0x40; /* TYPEFLAG_FDUAL */
2031
2032     if(type_iface_get_inherit(interface))
2033         add_impl_type(msft_typeinfo, type_iface_get_inherit(interface),
2034                       ref_importinfo);
2035
2036     /* count the number of inherited interfaces and non-local functions */
2037     for(ref = inherit; ref; ref = type_iface_get_inherit(ref)) {
2038         num_parents++;
2039         STATEMENTS_FOR_EACH_FUNC( stmt_func, type_iface_get_stmts(ref) ) {
2040             var_t *func = stmt_func->u.var;
2041             if (!is_local(func->attrs)) num_funcs++;
2042         }
2043     }
2044     msft_typeinfo->typeinfo->datatype2 = num_funcs << 16 | num_parents;
2045     msft_typeinfo->typeinfo->cbSizeVft = num_funcs * pointer_size;
2046
2047     STATEMENTS_FOR_EACH_FUNC( stmt_func, type_iface_get_stmts(interface) ) {
2048         var_t *func = stmt_func->u.var;
2049         if(add_func_desc(msft_typeinfo, func, idx) == S_OK)
2050             idx++;
2051     }
2052 }
2053
2054 static void add_structure_typeinfo(msft_typelib_t *typelib, type_t *structure)
2055 {
2056     int idx = 0;
2057     var_t *cur;
2058     msft_typeinfo_t *msft_typeinfo;
2059
2060     if (-1 < structure->typelib_idx)
2061         return;
2062
2063     structure->typelib_idx = typelib->typelib_header.nrtypeinfos;
2064     msft_typeinfo = create_msft_typeinfo(typelib, TKIND_RECORD, structure->name, structure->attrs);
2065     msft_typeinfo->typeinfo->size = 0;
2066
2067     if (type_struct_get_fields(structure))
2068         LIST_FOR_EACH_ENTRY( cur, type_struct_get_fields(structure), var_t, entry )
2069             add_var_desc(msft_typeinfo, idx++, cur);
2070 }
2071
2072 static void add_enum_typeinfo(msft_typelib_t *typelib, type_t *enumeration)
2073 {
2074     int idx = 0;
2075     var_t *cur;
2076     msft_typeinfo_t *msft_typeinfo;
2077
2078     enumeration->typelib_idx = typelib->typelib_header.nrtypeinfos;
2079     msft_typeinfo = create_msft_typeinfo(typelib, TKIND_ENUM, enumeration->name, enumeration->attrs);
2080     msft_typeinfo->typeinfo->size = 0;
2081
2082     if (type_enum_get_values(enumeration))
2083         LIST_FOR_EACH_ENTRY( cur, type_enum_get_values(enumeration), var_t, entry )
2084             add_var_desc(msft_typeinfo, idx++, cur);
2085 }
2086
2087 static void add_typedef_typeinfo(msft_typelib_t *typelib, type_t *tdef)
2088 {
2089     msft_typeinfo_t *msft_typeinfo;
2090     int alignment;
2091
2092     if (-1 < tdef->typelib_idx)
2093         return;
2094
2095     tdef->typelib_idx = typelib->typelib_header.nrtypeinfos;
2096     msft_typeinfo = create_msft_typeinfo(typelib, TKIND_ALIAS, tdef->name, tdef->attrs);
2097     encode_type(typelib, get_type_vt(type_alias_get_aliasee(tdef)),
2098                 type_alias_get_aliasee(tdef),
2099                 &msft_typeinfo->typeinfo->datatype1,
2100                 &msft_typeinfo->typeinfo->size,
2101                 &alignment, &msft_typeinfo->typeinfo->datatype2);
2102     msft_typeinfo->typeinfo->typekind |= (alignment << 11 | alignment << 6);
2103 }
2104
2105 static void add_coclass_typeinfo(msft_typelib_t *typelib, type_t *cls)
2106 {
2107     msft_typeinfo_t *msft_typeinfo;
2108     ifref_t *iref;
2109     int num_ifaces = 0, offset, i;
2110     MSFT_RefRecord *ref, *first = NULL, *first_source = NULL;
2111     int have_default = 0, have_default_source = 0;
2112     const attr_t *attr;
2113     ifref_list_t *ifaces;
2114
2115     if (-1 < cls->typelib_idx)
2116         return;
2117
2118     cls->typelib_idx = typelib->typelib_header.nrtypeinfos;
2119     msft_typeinfo = create_msft_typeinfo(typelib, TKIND_COCLASS, cls->name, cls->attrs);
2120
2121     ifaces = type_coclass_get_ifaces(cls);
2122     if (ifaces) LIST_FOR_EACH_ENTRY( iref, ifaces, ifref_t, entry ) num_ifaces++;
2123
2124     offset = msft_typeinfo->typeinfo->datatype1 = ctl2_alloc_segment(typelib, MSFT_SEG_REFERENCES,
2125                                                                      num_ifaces * sizeof(*ref), 0);
2126
2127     i = 0;
2128     if (ifaces) LIST_FOR_EACH_ENTRY( iref, ifaces, ifref_t, entry ) {
2129         if(iref->iface->typelib_idx == -1)
2130             add_interface_typeinfo(typelib, iref->iface);
2131         ref = (MSFT_RefRecord*) (typelib->typelib_segment_data[MSFT_SEG_REFERENCES] + offset + i * sizeof(*ref));
2132         ref->reftype = typelib->typelib_typeinfo_offsets[iref->iface->typelib_idx];
2133         ref->flags = 0;
2134         ref->oCustData = -1;
2135         ref->onext = -1;
2136         if(i < num_ifaces - 1)
2137             ref->onext = offset + (i + 1) * sizeof(*ref);
2138
2139         if (iref->attrs) LIST_FOR_EACH_ENTRY( attr, iref->attrs, const attr_t, entry ) {
2140             switch(attr->type) {
2141             case ATTR_DEFAULT:
2142                 ref->flags |= 0x1; /* IMPLTYPEFLAG_FDEFAULT */
2143                 break;
2144             case ATTR_DEFAULTVTABLE:
2145                 ref->flags |= 0x8; /* IMPLTYPEFLAG_FDEFAULTVTABLE */
2146                 break;
2147             case ATTR_RESTRICTED:
2148                 ref->flags |= 0x4; /* IMPLTYPEFLAG_FRESTRICTED */
2149                 break;
2150             case ATTR_SOURCE:
2151                 ref->flags |= 0x2; /* IMPLTYPEFLAG_FSOURCE */
2152                 break;
2153             default:
2154                 warning("add_coclass_typeinfo: unhandled attr %d\n", attr->type);
2155             }
2156         }
2157         if(ref->flags & 0x1) { /* IMPLTYPEFLAG_FDEFAULT */
2158             if(ref->flags & 0x2) /* IMPLTYPEFLAG_SOURCE */
2159                 have_default_source = 1;
2160             else
2161                 have_default = 1;
2162         }
2163
2164         /* If the interface is non-restricted and we haven't already had one then
2165            remember it so that we can use it as a default later */
2166         if((ref->flags & 0x4) == 0) { /* IMPLTYPEFLAG_FRESTRICTED */
2167             if(ref->flags & 0x2) { /* IMPLTYPEFLAG_FSOURCE */
2168                 if(!first_source)
2169                     first_source = ref;
2170             }
2171             else if(!first)
2172                 first = ref;
2173         }
2174         i++;
2175     }
2176
2177     /* If we haven't had a default interface, then set the default flags on the
2178        first ones */
2179     if(!have_default && first)
2180         first->flags |= 0x1;
2181     if(!have_default_source && first_source)
2182         first_source->flags |= 0x1;
2183
2184     msft_typeinfo->typeinfo->cImplTypes = num_ifaces;
2185     msft_typeinfo->typeinfo->size = 4;
2186     msft_typeinfo->typeinfo->typekind |= 0x2200;
2187 }
2188
2189 static void add_module_typeinfo(msft_typelib_t *typelib, type_t *module)
2190 {
2191     int idx = 0;
2192     const statement_t *stmt;
2193     msft_typeinfo_t *msft_typeinfo;
2194
2195     if (-1 < module->typelib_idx)
2196         return;
2197
2198     module->typelib_idx = typelib->typelib_header.nrtypeinfos;
2199     msft_typeinfo = create_msft_typeinfo(typelib, TKIND_MODULE, module->name, module->attrs);
2200     msft_typeinfo->typeinfo->typekind |= 0x0a00;
2201
2202     STATEMENTS_FOR_EACH_FUNC( stmt, module->details.module->stmts ) {
2203         var_t *func = stmt->u.var;
2204         if(add_func_desc(msft_typeinfo, func, idx) == S_OK)
2205             idx++;
2206     }
2207
2208     msft_typeinfo->typeinfo->size = idx;
2209 }
2210
2211 static void add_type_typeinfo(msft_typelib_t *typelib, type_t *type)
2212 {
2213     switch (type_get_type(type)) {
2214     case TYPE_INTERFACE:
2215         add_interface_typeinfo(typelib, type);
2216         break;
2217     case TYPE_STRUCT:
2218         add_structure_typeinfo(typelib, type);
2219         break;
2220     case TYPE_ENUM:
2221         add_enum_typeinfo(typelib, type);
2222         break;
2223     case TYPE_COCLASS:
2224         add_coclass_typeinfo(typelib, type);
2225         break;
2226     case TYPE_BASIC:
2227     case TYPE_POINTER:
2228         break;
2229     default:
2230         error("add_entry: unhandled type 0x%x for %s\n",
2231               type_get_type(type), type->name);
2232         break;
2233     }
2234 }
2235
2236 static void add_entry(msft_typelib_t *typelib, const statement_t *stmt)
2237 {
2238     switch(stmt->type) {
2239     case STMT_LIBRARY:
2240     case STMT_IMPORT:
2241     case STMT_CPPQUOTE:
2242     case STMT_DECLARATION:
2243         /* not included in typelib */
2244         break;
2245     case STMT_IMPORTLIB:
2246         /* not processed here */
2247         break;
2248     case STMT_TYPEDEF:
2249     {
2250         const type_list_t *type_entry = stmt->u.type_list;
2251         for (; type_entry; type_entry = type_entry->next) {
2252             /* if the type is public then add the typedef, otherwise attempt
2253              * to add the aliased type */
2254             if (is_attr(type_entry->type->attrs, ATTR_PUBLIC))
2255                 add_typedef_typeinfo(typelib, type_entry->type);
2256             else
2257                 add_type_typeinfo(typelib, type_alias_get_aliasee(type_entry->type));
2258         }
2259         break;
2260     }
2261     case STMT_MODULE:
2262         add_module_typeinfo(typelib, stmt->u.type);
2263         break;
2264     case STMT_TYPE:
2265     case STMT_TYPEREF:
2266     {
2267         type_t *type = stmt->u.type;
2268         add_type_typeinfo(typelib, type);
2269         break;
2270     }
2271     }
2272 }
2273
2274 static void set_name(msft_typelib_t *typelib)
2275 {
2276     int offset;
2277
2278     offset = ctl2_alloc_name(typelib, typelib->typelib->name);
2279     if (offset == -1) return;
2280     typelib->typelib_header.NameOffset = offset;
2281     return;
2282 }
2283
2284 static void set_version(msft_typelib_t *typelib)
2285 {
2286     typelib->typelib_header.version = get_attrv( typelib->typelib->attrs, ATTR_VERSION );
2287 }
2288
2289 static void set_guid(msft_typelib_t *typelib)
2290 {
2291     MSFT_GuidEntry guidentry;
2292     int offset;
2293     void *ptr;
2294     GUID guid = {0,0,0,{0,0,0,0,0,0}};
2295
2296     guidentry.guid = guid;
2297     guidentry.hreftype = -2;
2298     guidentry.next_hash = -1;
2299
2300     ptr = get_attrp( typelib->typelib->attrs, ATTR_UUID );
2301     if (ptr) guidentry.guid = *(GUID *)ptr;
2302
2303     offset = ctl2_alloc_guid(typelib, &guidentry);
2304     typelib->typelib_header.posguid = offset;
2305
2306     return;
2307 }
2308
2309 static void set_doc_string(msft_typelib_t *typelib)
2310 {
2311     char *str = get_attrp( typelib->typelib->attrs, ATTR_HELPSTRING );
2312
2313     if (str)
2314     {
2315         int offset = ctl2_alloc_string(typelib, str);
2316         if (offset != -1) typelib->typelib_header.helpstring = offset;
2317     }
2318 }
2319
2320 static void set_help_file_name(msft_typelib_t *typelib)
2321 {
2322     char *str = get_attrp( typelib->typelib->attrs, ATTR_HELPFILE );
2323
2324     if (str)
2325     {
2326         int offset = ctl2_alloc_string(typelib, str);
2327         if (offset != -1)
2328         {
2329             typelib->typelib_header.helpfile = offset;
2330             typelib->typelib_header.varflags |= 0x10;
2331         }
2332     }
2333 }
2334
2335 static void set_help_context(msft_typelib_t *typelib)
2336 {
2337     const expr_t *expr = get_attrp( typelib->typelib->attrs, ATTR_HELPCONTEXT );
2338     if (expr) typelib->typelib_header.helpcontext = expr->cval;
2339 }
2340
2341 static void set_help_string_dll(msft_typelib_t *typelib)
2342 {
2343     char *str = get_attrp( typelib->typelib->attrs, ATTR_HELPSTRINGDLL );
2344
2345     if (str)
2346     {
2347         int offset = ctl2_alloc_string(typelib, str);
2348         if (offset != -1)
2349         {
2350             typelib->help_string_dll_offset = offset;
2351             typelib->typelib_header.varflags |= 0x100;
2352         }
2353     }
2354 }
2355
2356 static void set_help_string_context(msft_typelib_t *typelib)
2357 {
2358     const expr_t *expr = get_attrp( typelib->typelib->attrs, ATTR_HELPSTRINGCONTEXT );
2359     if (expr) typelib->typelib_header.helpstringcontext = expr->cval;
2360 }
2361
2362 static void set_lcid(msft_typelib_t *typelib)
2363 {
2364     const expr_t *lcid_expr = get_attrp( typelib->typelib->attrs, ATTR_LIBLCID );
2365     if(lcid_expr)
2366     {
2367         typelib->typelib_header.lcid  = lcid_expr->cval;
2368         typelib->typelib_header.lcid2 = lcid_expr->cval;
2369     }
2370 }
2371
2372 static void set_lib_flags(msft_typelib_t *typelib)
2373 {
2374     const attr_t *attr;
2375
2376     typelib->typelib_header.flags = 0;
2377     if (!typelib->typelib->attrs) return;
2378     LIST_FOR_EACH_ENTRY( attr, typelib->typelib->attrs, const attr_t, entry )
2379     {
2380         switch(attr->type) {
2381         case ATTR_CONTROL:
2382             typelib->typelib_header.flags |= 0x02; /* LIBFLAG_FCONTROL */
2383             break;
2384         case ATTR_HIDDEN:
2385             typelib->typelib_header.flags |= 0x04; /* LIBFLAG_FHIDDEN */
2386             break;
2387         case ATTR_RESTRICTED:
2388             typelib->typelib_header.flags |= 0x01; /* LIBFLAG_FRESTRICTED */
2389             break;
2390         default:
2391             break;
2392         }
2393     }
2394     return;
2395 }
2396
2397 static int ctl2_write_chunk(int fd, void *segment, int length)
2398 {
2399     if (write(fd, segment, length) != length) {
2400         close(fd);
2401         return 0;
2402     }
2403     return -1;
2404 }
2405
2406 static int ctl2_write_segment(msft_typelib_t *typelib, int fd, int segment)
2407 {
2408     if (write(fd, typelib->typelib_segment_data[segment], typelib->typelib_segdir[segment].length)
2409         != typelib->typelib_segdir[segment].length) {
2410         close(fd);
2411         return 0;
2412     }
2413
2414     return -1;
2415 }
2416
2417 static void ctl2_finalize_typeinfos(msft_typelib_t *typelib, int filesize)
2418 {
2419     msft_typeinfo_t *typeinfo;
2420
2421     for (typeinfo = typelib->typeinfos; typeinfo; typeinfo = typeinfo->next_typeinfo) {
2422         typeinfo->typeinfo->memoffset = filesize;
2423         if (typeinfo->func_data)
2424             filesize += typeinfo->func_data[0] + ((typeinfo->typeinfo->cElement & 0xffff) * 12);
2425         if (typeinfo->var_data)
2426             filesize += typeinfo->var_data[0] + (((typeinfo->typeinfo->cElement >> 16) & 0xffff) * 12);
2427         if (typeinfo->func_data || typeinfo->var_data)
2428             filesize += 4;
2429     }
2430 }
2431
2432 static int ctl2_finalize_segment(msft_typelib_t *typelib, int filepos, int segment)
2433 {
2434     if (typelib->typelib_segdir[segment].length) {
2435         typelib->typelib_segdir[segment].offset = filepos;
2436     } else {
2437         typelib->typelib_segdir[segment].offset = -1;
2438     }
2439
2440     return typelib->typelib_segdir[segment].length;
2441 }
2442
2443
2444 static void ctl2_write_typeinfos(msft_typelib_t *typelib, int fd)
2445 {
2446     msft_typeinfo_t *typeinfo;
2447     int typedata_size;
2448
2449     for (typeinfo = typelib->typeinfos; typeinfo; typeinfo = typeinfo->next_typeinfo) {
2450         if (!typeinfo->func_data && !typeinfo->var_data) continue;
2451         typedata_size = 0;
2452         if (typeinfo->func_data)
2453             typedata_size = typeinfo->func_data[0];
2454         if (typeinfo->var_data)
2455             typedata_size += typeinfo->var_data[0];
2456         ctl2_write_chunk(fd, &typedata_size, sizeof(int));
2457         if (typeinfo->func_data)
2458             ctl2_write_chunk(fd, typeinfo->func_data + 1, typeinfo->func_data[0]);
2459         if (typeinfo->var_data)
2460             ctl2_write_chunk(fd, typeinfo->var_data + 1, typeinfo->var_data[0]);
2461         if (typeinfo->func_indices)
2462             ctl2_write_chunk(fd, typeinfo->func_indices, (typeinfo->typeinfo->cElement & 0xffff) * 4);
2463         if (typeinfo->var_indices)
2464             ctl2_write_chunk(fd, typeinfo->var_indices, (typeinfo->typeinfo->cElement >> 16) * 4);
2465         if (typeinfo->func_names)
2466             ctl2_write_chunk(fd, typeinfo->func_names,   (typeinfo->typeinfo->cElement & 0xffff) * 4);
2467         if (typeinfo->var_names)
2468             ctl2_write_chunk(fd, typeinfo->var_names,   (typeinfo->typeinfo->cElement >> 16) * 4); 
2469         if (typeinfo->func_offsets)
2470             ctl2_write_chunk(fd, typeinfo->func_offsets, (typeinfo->typeinfo->cElement & 0xffff) * 4);
2471         if (typeinfo->var_offsets) {
2472             int add = 0, i, offset;
2473             if(typeinfo->func_data)
2474                 add = typeinfo->func_data[0];
2475             for(i = 0; i < (typeinfo->typeinfo->cElement >> 16); i++) {
2476                 offset = typeinfo->var_offsets[i];
2477                 offset += add;
2478                 ctl2_write_chunk(fd, &offset, 4);
2479             }
2480         }
2481     }
2482 }
2483
2484 static int save_all_changes(msft_typelib_t *typelib)
2485 {
2486     int retval;
2487     int filepos;
2488     int fd;
2489
2490     chat("save_all_changes(%p)\n", typelib);
2491
2492     retval = TYPE_E_IOERROR;
2493
2494     fd = open(typelib->typelib->filename, O_CREAT | O_WRONLY | O_TRUNC | O_BINARY, 0666);
2495     if (fd == -1) return retval;
2496
2497     filepos = sizeof(MSFT_Header) + sizeof(MSFT_SegDir);
2498     if(typelib->typelib_header.varflags & 0x100) filepos += 4; /* helpstringdll */
2499     filepos += typelib->typelib_header.nrtypeinfos * 4;
2500
2501     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_TYPEINFO);
2502     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_GUIDHASH);
2503     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_GUID);
2504     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_REFERENCES);
2505     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_IMPORTINFO);
2506     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_IMPORTFILES);
2507     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_NAMEHASH);
2508     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_NAME);
2509     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_STRING);
2510     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_TYPEDESC);
2511     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_ARRAYDESC);
2512     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_CUSTDATA);
2513     filepos += ctl2_finalize_segment(typelib, filepos, MSFT_SEG_CUSTDATAGUID);
2514
2515     ctl2_finalize_typeinfos(typelib, filepos);
2516
2517     if (!ctl2_write_chunk(fd, &typelib->typelib_header, sizeof(typelib->typelib_header))) return retval;
2518     if(typelib->typelib_header.varflags & 0x100)
2519         if (!ctl2_write_chunk(fd, &typelib->help_string_dll_offset, sizeof(typelib->help_string_dll_offset)))
2520             return retval;
2521
2522     if (!ctl2_write_chunk(fd, typelib->typelib_typeinfo_offsets, typelib->typelib_header.nrtypeinfos * 4)) return retval;
2523     if (!ctl2_write_chunk(fd, &typelib->typelib_segdir, sizeof(typelib->typelib_segdir))) return retval;
2524     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_TYPEINFO    )) return retval;
2525     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_GUIDHASH    )) return retval;
2526     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_GUID        )) return retval;
2527     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_REFERENCES  )) return retval;
2528     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_IMPORTINFO  )) return retval;
2529     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_IMPORTFILES )) return retval;
2530     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_NAMEHASH    )) return retval;
2531     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_NAME        )) return retval;
2532     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_STRING      )) return retval;
2533     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_TYPEDESC    )) return retval;
2534     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_ARRAYDESC   )) return retval;
2535     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_CUSTDATA    )) return retval;
2536     if (!ctl2_write_segment(typelib, fd, MSFT_SEG_CUSTDATAGUID)) return retval;
2537
2538     ctl2_write_typeinfos(typelib, fd);
2539
2540     if (close(fd) == -1) return retval;
2541
2542     retval = S_OK;
2543     return retval;
2544 }
2545
2546 int create_msft_typelib(typelib_t *typelib)
2547 {
2548     msft_typelib_t *msft;
2549     int failed = 0;
2550     const statement_t *stmt;
2551     time_t cur_time;
2552     char *time_override;
2553     unsigned int version = 5 << 24 | 1 << 16 | 164; /* 5.01.0164 */
2554     GUID midl_time_guid    = {0xde77ba63,0x517c,0x11d1,{0xa2,0xda,0x00,0x00,0xf8,0x77,0x3c,0xe9}}; 
2555     GUID midl_version_guid = {0xde77ba64,0x517c,0x11d1,{0xa2,0xda,0x00,0x00,0xf8,0x77,0x3c,0xe9}}; 
2556
2557     pointer_size = (typelib_kind == SYS_WIN64) ? 8 : 4;
2558
2559     msft = xmalloc(sizeof(*msft));
2560     memset(msft, 0, sizeof(*msft));
2561     msft->typelib = typelib;
2562
2563     ctl2_init_header(msft);
2564     ctl2_init_segdir(msft);
2565
2566     msft->typelib_header.varflags |= typelib_kind;
2567
2568     /*
2569      * The following two calls return an offset or -1 if out of memory. We
2570      * specifically need an offset of 0, however, so...
2571      */
2572     if (ctl2_alloc_segment(msft, MSFT_SEG_GUIDHASH, 0x80, 0x80)) { failed = 1; }
2573     if (ctl2_alloc_segment(msft, MSFT_SEG_NAMEHASH, 0x200, 0x200)) { failed = 1; }
2574
2575     if(failed)
2576     {
2577         free(msft);
2578         return 0;
2579     }
2580
2581     msft->typelib_guidhash_segment = (int *)msft->typelib_segment_data[MSFT_SEG_GUIDHASH];
2582     msft->typelib_namehash_segment = (int *)msft->typelib_segment_data[MSFT_SEG_NAMEHASH];
2583
2584     memset(msft->typelib_guidhash_segment, 0xff, 0x80);
2585     memset(msft->typelib_namehash_segment, 0xff, 0x200);
2586
2587     set_lib_flags(msft);
2588     set_lcid(msft);
2589     set_help_file_name(msft);
2590     set_doc_string(msft);
2591     set_guid(msft);
2592     set_version(msft);
2593     set_name(msft);
2594     set_help_context(msft);
2595     set_help_string_dll(msft);
2596     set_help_string_context(msft);
2597     
2598     /* midl adds two sets of custom data to the library: the current unix time
2599        and midl's version number */
2600     time_override = getenv( "WIDL_TIME_OVERRIDE");
2601     cur_time = time_override ? atol( time_override) : time(NULL);
2602     set_custdata(msft, &midl_time_guid, VT_UI4, &cur_time, &msft->typelib_header.CustomDataOffset);
2603     set_custdata(msft, &midl_version_guid, VT_UI4, &version, &msft->typelib_header.CustomDataOffset);
2604
2605     if (typelib->stmts)
2606         LIST_FOR_EACH_ENTRY( stmt, typelib->stmts, const statement_t, entry )
2607             add_entry(msft, stmt);
2608
2609     save_all_changes(msft);
2610     free(msft);
2611     return 1;
2612 }