git.oblomov.eu Git - wine/blob - tools/widl/write_msft.c

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