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