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