widl: Make the attrs parameter passed to start_typelib const.
[wine] / tools / widl / typelib.c
1 /*
2  * IDL Compiler
3  *
4  * Copyright 2004 Ove Kaaven
5  * Copyright 2006 Jacek Caban for CodeWeavers
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 #include "config.h"
23 #include "wine/port.h"
24 #include "wine/wpp.h"
25
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <stdarg.h>
29 #ifdef HAVE_UNISTD_H
30 # include <unistd.h>
31 #endif
32 #include <string.h>
33 #include <ctype.h>
34
35 #define NONAMELESSUNION
36 #define NONAMELESSSTRUCT
37
38 #include "windef.h"
39 #include "winbase.h"
40
41 #include "widl.h"
42 #include "utils.h"
43 #include "parser.h"
44 #include "header.h"
45 #include "typelib.h"
46 #include "widltypes.h"
47 #include "typelib_struct.h"
48
49 int in_typelib = 0;
50
51 static typelib_t *typelib;
52
53 type_t *duptype(type_t *t, int dupname)
54 {
55   type_t *d = alloc_type();
56
57   *d = *t;
58   if (dupname && t->name)
59     d->name = xstrdup(t->name);
60
61   d->orig = t;
62   return d;
63 }
64
65 type_t *alias(type_t *t, const char *name)
66 {
67   type_t *a = duptype(t, 0);
68
69   a->name = xstrdup(name);
70   a->kind = TKIND_ALIAS;
71   a->attrs = NULL;
72   a->declarray = FALSE;
73
74   return a;
75 }
76
77 int is_ptr(const type_t *t)
78 {
79   unsigned char c = t->type;
80   return c == RPC_FC_RP
81       || c == RPC_FC_UP
82       || c == RPC_FC_FP
83       || c == RPC_FC_OP;
84 }
85
86 int is_array(const type_t *t)
87 {
88     switch (t->type)
89     {
90     case RPC_FC_SMFARRAY:
91     case RPC_FC_LGFARRAY:
92     case RPC_FC_SMVARRAY:
93     case RPC_FC_LGVARRAY:
94     case RPC_FC_CARRAY:
95     case RPC_FC_CVARRAY:
96     case RPC_FC_BOGUS_ARRAY:
97         return TRUE;
98     default:
99         return FALSE;
100     }
101 }
102
103 /* List of oleauto types that should be recognized by name.
104  * (most of) these seem to be intrinsic types in mktyplib. */
105
106 static const struct oatype {
107   const char *kw;
108   unsigned short vt;
109 } oatypes[] = {
110   {"BSTR",          VT_BSTR},
111   {"CURRENCY",      VT_CY},
112   {"DATE",          VT_DATE},
113   {"DECIMAL",       VT_DECIMAL},
114   {"HRESULT",       VT_HRESULT},
115   {"LPSTR",         VT_LPSTR},
116   {"LPWSTR",        VT_LPWSTR},
117   {"SCODE",         VT_ERROR},
118   {"VARIANT",       VT_VARIANT},
119   {"VARIANT_BOOL",  VT_BOOL}
120 };
121 #define NTYPES (sizeof(oatypes)/sizeof(oatypes[0]))
122 #define KWP(p) ((const struct oatype *)(p))
123
124 static int kw_cmp_func(const void *s1, const void *s2)
125 {
126         return strcmp(KWP(s1)->kw, KWP(s2)->kw);
127 }
128
129 static unsigned short builtin_vt(const type_t *t)
130 {
131   const char *kw = t->name;
132   struct oatype key;
133   const struct oatype *kwp;
134   key.kw = kw;
135 #ifdef KW_BSEARCH
136   kwp = bsearch(&key, oatypes, NTYPES, sizeof(oatypes[0]), kw_cmp_func);
137 #else
138   {
139     unsigned int i;
140     for (kwp=NULL, i=0; i < NTYPES; i++)
141       if (!kw_cmp_func(&key, &oatypes[i])) {
142         kwp = &oatypes[i];
143         break;
144       }
145   }
146 #endif
147   if (kwp) {
148     return kwp->vt;
149   }
150   if (is_string_type (t->attrs, t))
151     switch (t->ref->type)
152       {
153       case RPC_FC_CHAR: return VT_LPSTR;
154       case RPC_FC_WCHAR: return VT_LPWSTR;
155       default: break;
156       }
157   return 0;
158 }
159
160 static int match(const char*n, const char*m)
161 {
162   if (!n) return 0;
163   return !strcmp(n, m);
164 }
165
166 unsigned short get_type_vt(type_t *t)
167 {
168   unsigned short vt;
169
170   chat("get_type_vt: %p type->name %s\n", t, t->name);
171   if (t->name) {
172     vt = builtin_vt(t);
173     if (vt) return vt;
174   }
175
176   if (t->kind == TKIND_ALIAS && t->attrs)
177     return VT_USERDEFINED;
178
179   switch (t->type) {
180   case RPC_FC_BYTE:
181   case RPC_FC_USMALL:
182     return VT_UI1;
183   case RPC_FC_CHAR:
184   case RPC_FC_SMALL:
185     return VT_I1;
186   case RPC_FC_WCHAR:
187     return VT_I2; /* mktyplib seems to parse wchar_t as short */
188   case RPC_FC_SHORT:
189     return VT_I2;
190   case RPC_FC_USHORT:
191     return VT_UI2;
192   case RPC_FC_LONG:
193     if (match(t->name, "int")) return VT_INT;
194     return VT_I4;
195   case RPC_FC_ULONG:
196     if (match(t->name, "int")) return VT_UINT;
197     return VT_UI4;
198   case RPC_FC_HYPER:
199     if (t->sign < 0) return VT_UI8;
200     if (match(t->name, "MIDL_uhyper")) return VT_UI8;
201     return VT_I8;
202   case RPC_FC_FLOAT:
203     return VT_R4;
204   case RPC_FC_DOUBLE:
205     return VT_R8;
206   case RPC_FC_RP:
207   case RPC_FC_UP:
208   case RPC_FC_OP:
209   case RPC_FC_FP:
210   case RPC_FC_CARRAY:
211   case RPC_FC_CVARRAY:
212     if(t->ref)
213     {
214       if (match(t->ref->name, "SAFEARRAY"))
215         return VT_SAFEARRAY;
216       return VT_PTR;
217     }
218
219     error("get_type_vt: unknown-deref-type: %d\n", t->ref->type);
220     break;
221   case RPC_FC_IP:
222     if(match(t->name, "IUnknown"))
223       return VT_UNKNOWN;
224     if(match(t->name, "IDispatch"))
225       return VT_DISPATCH;
226     return VT_USERDEFINED;
227
228   case RPC_FC_ENUM16:
229   case RPC_FC_STRUCT:
230   case RPC_FC_PSTRUCT:
231   case RPC_FC_CSTRUCT:
232   case RPC_FC_CPSTRUCT:
233   case RPC_FC_CVSTRUCT:
234   case RPC_FC_BOGUS_STRUCT:
235     return VT_USERDEFINED;
236   case 0:
237     return t->kind == TKIND_PRIMITIVE ? VT_VOID : VT_USERDEFINED;
238   default:
239     error("get_type_vt: unknown type: 0x%02x\n", t->type);
240   }
241   return 0;
242 }
243
244 void start_typelib(char *name, const attr_list_t *attrs)
245 {
246     in_typelib++;
247     if (!do_typelib) return;
248
249     typelib = xmalloc(sizeof(*typelib));
250     typelib->name = xstrdup(name);
251     typelib->filename = xstrdup(typelib_name);
252     typelib->attrs = attrs;
253     list_init( &typelib->entries );
254     list_init( &typelib->importlibs );
255
256     if (is_attr(attrs, ATTR_POINTERDEFAULT))
257         pointer_default = get_attrv(attrs, ATTR_POINTERDEFAULT);
258 }
259
260 void end_typelib(void)
261 {
262     in_typelib--;
263     if (!typelib) return;
264
265     create_msft_typelib(typelib);
266     pointer_default = RPC_FC_UP;
267     return;
268 }
269
270 void add_typelib_entry(type_t *t)
271 {
272     typelib_entry_t *entry;
273     if (!typelib) return;
274
275     chat("add kind %i: %s\n", t->kind, t->name);
276     entry = xmalloc(sizeof(*entry));
277     entry->type = t;
278     list_add_tail( &typelib->entries, &entry->entry );
279 }
280
281 static void tlb_read(int fd, void *buf, int count)
282 {
283     if(read(fd, buf, count) < count)
284         error("error while reading importlib.\n");
285 }
286
287 static void tlb_lseek(int fd, off_t offset)
288 {
289     if(lseek(fd, offset, SEEK_SET) == -1)
290         error("lseek failed\n");
291 }
292
293 static void msft_read_guid(int fd, MSFT_SegDir *segdir, int offset, GUID *guid)
294 {
295     tlb_lseek(fd, segdir->pGuidTab.offset+offset);
296     tlb_read(fd, guid, sizeof(GUID));
297 }
298
299 static void read_msft_importlib(importlib_t *importlib, int fd)
300 {
301     MSFT_Header header;
302     MSFT_SegDir segdir;
303     int *typeinfo_offs;
304     int i;
305
306     importlib->allocated = 0;
307
308     tlb_lseek(fd, 0);
309     tlb_read(fd, &header, sizeof(header));
310
311     importlib->version = header.version;
312
313     typeinfo_offs = xmalloc(header.nrtypeinfos*sizeof(INT));
314     tlb_read(fd, typeinfo_offs, header.nrtypeinfos*sizeof(INT));
315     tlb_read(fd, &segdir, sizeof(segdir));
316
317     msft_read_guid(fd, &segdir, header.posguid, &importlib->guid);
318
319     importlib->ntypeinfos = header.nrtypeinfos;
320     importlib->importinfos = xmalloc(importlib->ntypeinfos*sizeof(importinfo_t));
321
322     for(i=0; i < importlib->ntypeinfos; i++) {
323         MSFT_TypeInfoBase base;
324         MSFT_NameIntro nameintro;
325         int len;
326
327         tlb_lseek(fd, sizeof(MSFT_Header) + header.nrtypeinfos*sizeof(INT) + sizeof(MSFT_SegDir)
328                  + typeinfo_offs[i]);
329         tlb_read(fd, &base, sizeof(base));
330
331         importlib->importinfos[i].importlib = importlib;
332         importlib->importinfos[i].flags = (base.typekind&0xf)<<24;
333         importlib->importinfos[i].offset = -1;
334         importlib->importinfos[i].id = i;
335
336         if(base.posguid != -1) {
337             importlib->importinfos[i].flags |= MSFT_IMPINFO_OFFSET_IS_GUID;
338             msft_read_guid(fd, &segdir, base.posguid, &importlib->importinfos[i].guid);
339         }
340         else memset( &importlib->importinfos[i].guid, 0, sizeof(importlib->importinfos[i].guid));
341
342         tlb_lseek(fd, segdir.pNametab.offset + base.NameOffset);
343         tlb_read(fd, &nameintro, sizeof(nameintro));
344
345         len = nameintro.namelen & 0xff;
346
347         importlib->importinfos[i].name = xmalloc(len+1);
348         tlb_read(fd, importlib->importinfos[i].name, len);
349         importlib->importinfos[i].name[len] = 0;
350     }
351
352     free(typeinfo_offs);
353 }
354
355 static void read_importlib(importlib_t *importlib)
356 {
357     int fd;
358     INT magic;
359     char *file_name;
360
361     file_name = wpp_find_include(importlib->name, NULL);
362     if(file_name) {
363         fd = open(file_name, O_RDONLY);
364         free(file_name);
365     }else {
366         fd = open(importlib->name, O_RDONLY);
367     }
368
369     if(fd < 0)
370         error("Could not open importlib %s.\n", importlib->name);
371
372     tlb_read(fd, &magic, sizeof(magic));
373
374     switch(magic) {
375     case MSFT_MAGIC:
376         read_msft_importlib(importlib, fd);
377         break;
378     default:
379         error("Wrong or unsupported typelib magic %x\n", magic);
380     };
381
382     close(fd);
383 }
384
385 void add_importlib(const char *name)
386 {
387     importlib_t *importlib;
388
389     if(!typelib) return;
390
391     LIST_FOR_EACH_ENTRY( importlib, &typelib->importlibs, importlib_t, entry )
392         if(!strcmp(name, importlib->name))
393             return;
394
395     chat("add_importlib: %s\n", name);
396
397     importlib = xmalloc(sizeof(*importlib));
398     memset( importlib, 0, sizeof(*importlib) );
399     importlib->name = xstrdup(name);
400
401     read_importlib(importlib);
402     list_add_head( &typelib->importlibs, &importlib->entry );
403 }