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