Fix handle leaks in the test cases.
[wine] / dlls / msi / msipriv.h
1 /*
2  * Implementation of the Microsoft Installer (msi.dll)
3  *
4  * Copyright 2002-2005 Mike McCormack for CodeWeavers
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #ifndef __WINE_MSI_PRIVATE__
22 #define __WINE_MSI_PRIVATE__
23
24 #include <stdarg.h>
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "msi.h"
29 #include "msiquery.h"
30 #include "objbase.h"
31 #include "objidl.h"
32 #include "winnls.h"
33 #include "wine/list.h"
34
35 #define MSI_DATASIZEMASK 0x00ff
36 #define MSITYPE_VALID    0x0100
37 #define MSITYPE_STRING   0x0800
38 #define MSITYPE_NULLABLE 0x1000
39 #define MSITYPE_KEY      0x2000
40
41 #define MSITYPE_BINARY 0x8900
42
43 struct tagMSITABLE;
44 typedef struct tagMSITABLE MSITABLE;
45
46 struct string_table;
47 typedef struct string_table string_table;
48
49 struct tagMSIOBJECTHDR;
50 typedef struct tagMSIOBJECTHDR MSIOBJECTHDR;
51
52 typedef VOID (*msihandledestructor)( MSIOBJECTHDR * );
53
54 struct tagMSIOBJECTHDR
55 {
56     UINT magic;
57     UINT type;
58     LONG refcount;
59     msihandledestructor destructor;
60 };
61
62 typedef struct tagMSIDATABASE
63 {
64     MSIOBJECTHDR hdr;
65     IStorage *storage;
66     string_table *strings;
67     LPCWSTR mode;
68     struct list tables;
69 } MSIDATABASE;
70
71 typedef struct tagMSIVIEW MSIVIEW;
72
73 typedef struct tagMSIQUERY
74 {
75     MSIOBJECTHDR hdr;
76     MSIVIEW *view;
77     UINT row;
78     MSIDATABASE *db;
79     struct list mem;
80 } MSIQUERY;
81
82 /* maybe we can use a Variant instead of doing it ourselves? */
83 typedef struct tagMSIFIELD
84 {
85     UINT type;
86     union
87     {
88         INT iVal;
89         LPWSTR szwVal;
90         IStream *stream;
91     } u;
92 } MSIFIELD;
93
94 typedef struct tagMSIRECORD
95 {
96     MSIOBJECTHDR hdr;
97     UINT count;       /* as passed to MsiCreateRecord */
98     MSIFIELD fields[1]; /* nb. array size is count+1 */
99 } MSIRECORD;
100
101 typedef struct tagMSIVIEWOPS
102 {
103     /*
104      * fetch_int - reads one integer from {row,col} in the table
105      *
106      *  This function should be called after the execute method.
107      *  Data returned by the function should not change until 
108      *   close or delete is called.
109      *  To get a string value, query the database's string table with
110      *   the integer value returned from this function.
111      */
112     UINT (*fetch_int)( struct tagMSIVIEW *, UINT row, UINT col, UINT *val );
113
114     /*
115      * fetch_int - reads one integer from {row,col} in the table
116      *
117      *  This function is similar to fetch_int, except fetches a
118      *    stream instead of an integer.
119      */
120     UINT (*fetch_stream)( struct tagMSIVIEW *, UINT row, UINT col, IStream **stm );
121
122     /*
123      * get_int - sets one integer at {row,col} in the table
124      *
125      *  Similar semantics to fetch_int
126      */
127     UINT (*set_int)( struct tagMSIVIEW *, UINT row, UINT col, UINT val );
128
129     /*
130      * Inserts a new row into the database from the records contents
131      */
132     UINT (*insert_row)( struct tagMSIVIEW *, MSIRECORD * );
133
134     /*
135      * execute - loads the underlying data into memory so it can be read
136      */
137     UINT (*execute)( struct tagMSIVIEW *, MSIRECORD * );
138
139     /*
140      * close - clears the data read by execute from memory
141      */
142     UINT (*close)( struct tagMSIVIEW * );
143
144     /*
145      * get_dimensions - returns the number of rows or columns in a table.
146      *
147      *  The number of rows can only be queried after the execute method
148      *   is called. The number of columns can be queried at any time.
149      */
150     UINT (*get_dimensions)( struct tagMSIVIEW *, UINT *rows, UINT *cols );
151
152     /*
153      * get_column_info - returns the name and type of a specific column
154      *
155      *  The name is HeapAlloc'ed by this function and should be freed by
156      *   the caller.
157      *  The column information can be queried at any time.
158      */
159     UINT (*get_column_info)( struct tagMSIVIEW *, UINT n, LPWSTR *name, UINT *type );
160
161     /*
162      * modify - not yet implemented properly
163      */
164     UINT (*modify)( struct tagMSIVIEW *, MSIMODIFY, MSIRECORD * );
165
166     /*
167      * delete - destroys the structure completely
168      */
169     UINT (*delete)( struct tagMSIVIEW * );
170
171 } MSIVIEWOPS;
172
173 struct tagMSIVIEW
174 {
175     MSIOBJECTHDR hdr;
176     MSIVIEWOPS   *ops;
177 };
178
179 struct msi_dialog_tag;
180 typedef struct msi_dialog_tag msi_dialog;
181
182 typedef struct tagMSIPACKAGE
183 {
184     MSIOBJECTHDR hdr;
185     MSIDATABASE *db;
186     struct list components;
187     struct list features;
188     struct list files;
189     struct list tempfiles;
190     struct list folders;
191     LPWSTR ActionFormat;
192     LPWSTR LastAction;
193
194     struct list classes;
195     struct list extensions;
196     struct list progids;
197     struct list mimes;
198     struct list appids;
199     
200     struct tagMSISCRIPT *script;
201
202     struct list RunningActions;
203
204     LPWSTR PackagePath;
205     LPWSTR msiFilePath;
206     LPWSTR ProductCode;
207
208     UINT CurrentInstallState;
209     msi_dialog *dialog;
210     LPWSTR next_dialog;
211     
212     struct list subscriptions;
213 } MSIPACKAGE;
214
215 typedef struct tagMSIPREVIEW
216 {
217     MSIOBJECTHDR hdr;
218     MSIPACKAGE *package;
219     msi_dialog *dialog;
220 } MSIPREVIEW;
221
222 #define MSIHANDLETYPE_ANY 0
223 #define MSIHANDLETYPE_DATABASE 1
224 #define MSIHANDLETYPE_SUMMARYINFO 2
225 #define MSIHANDLETYPE_VIEW 3
226 #define MSIHANDLETYPE_RECORD 4
227 #define MSIHANDLETYPE_PACKAGE 5
228 #define MSIHANDLETYPE_PREVIEW 6
229
230 #define MSI_MAJORVERSION 2
231 #define MSI_MINORVERSION 0
232 #define MSI_BUILDNUMBER 2600
233
234 #define GUID_SIZE 39
235
236 #define MSIHANDLE_MAGIC 0x4d434923
237 #define MSIMAXHANDLES 0xf0
238
239 DEFINE_GUID(CLSID_IMsiServer,   0x000C101C,0x0000,0x0000,0xC0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
240 DEFINE_GUID(CLSID_IMsiServerX1, 0x000C103E,0x0000,0x0000,0xC0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
241 DEFINE_GUID(CLSID_IMsiServerX2, 0x000C1090,0x0000,0x0000,0xC0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
242 DEFINE_GUID(CLSID_IMsiServerX3, 0x000C1094,0x0000,0x0000,0xC0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
243
244 DEFINE_GUID(CLSID_IMsiServerMessage, 0x000C101D,0x0000,0x0000,0xC0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
245
246 typedef struct {
247     BOOL unicode;
248     union {
249        LPSTR a;
250        LPWSTR w;
251     } str;
252 } awstring;
253
254 typedef struct {
255     BOOL unicode;
256     union {
257        LPCSTR a;
258        LPCWSTR w;
259     } str;
260 } awcstring;
261
262 /* handle functions */
263 extern void *msihandle2msiinfo(MSIHANDLE handle, UINT type);
264 extern MSIHANDLE alloc_msihandle( MSIOBJECTHDR * );
265 extern void *alloc_msiobject(UINT type, UINT size, msihandledestructor destroy );
266 extern void msiobj_addref(MSIOBJECTHDR *);
267 extern int msiobj_release(MSIOBJECTHDR *);
268 extern void msiobj_lock(MSIOBJECTHDR *);
269 extern void msiobj_unlock(MSIOBJECTHDR *);
270 extern MSIHANDLE msiobj_findhandle( MSIOBJECTHDR *hdr );
271
272 extern void free_cached_tables( MSIDATABASE *db );
273 extern string_table *load_string_table( IStorage *stg );
274 extern UINT MSI_CommitTables( MSIDATABASE *db );
275 extern HRESULT init_string_table( IStorage *stg );
276
277
278 /* string table functions */
279 extern BOOL msi_addstring( string_table *st, UINT string_no, const CHAR *data, int len, UINT refcount );
280 extern BOOL msi_addstringW( string_table *st, UINT string_no, const WCHAR *data, int len, UINT refcount );
281 extern UINT msi_id2stringW( string_table *st, UINT string_no, LPWSTR buffer, UINT *sz );
282 extern UINT msi_id2stringA( string_table *st, UINT string_no, LPSTR buffer, UINT *sz );
283
284 extern LPWSTR MSI_makestring( MSIDATABASE *db, UINT stringid);
285 extern UINT msi_string2idW( string_table *st, LPCWSTR buffer, UINT *id );
286 extern UINT msi_string2idA( string_table *st, LPCSTR str, UINT *id );
287 extern string_table *msi_init_stringtable( int entries, UINT codepage );
288 extern VOID msi_destroy_stringtable( string_table *st );
289 extern UINT msi_string_count( string_table *st );
290 extern UINT msi_id_refcount( string_table *st, UINT i );
291 extern UINT msi_string_totalsize( string_table *st, UINT *last );
292 extern UINT msi_strcmp( string_table *st, UINT lval, UINT rval, UINT *res );
293 extern const WCHAR *msi_string_lookup_id( string_table *st, UINT id );
294 extern UINT msi_string_get_codepage( string_table *st );
295
296
297 extern UINT VIEW_find_column( MSIVIEW *view, LPCWSTR name, UINT *n );
298
299 extern BOOL TABLE_Exists( MSIDATABASE *db, LPWSTR name );
300
301 extern UINT read_raw_stream_data( MSIDATABASE*, LPCWSTR stname,
302                               USHORT **pdata, UINT *psz );
303
304 /* action internals */
305 extern UINT ACTION_DoTopLevelINSTALL( MSIPACKAGE *, LPCWSTR, LPCWSTR, LPCWSTR );
306 extern void ACTION_free_package_structures( MSIPACKAGE* );
307 extern UINT ACTION_DialogBox( MSIPACKAGE*, LPCWSTR);
308 extern UINT MSI_Sequence( MSIPACKAGE *package, LPCWSTR szTable, INT iSequenceMode );
309
310 /* record internals */
311 extern UINT MSI_RecordSetIStream( MSIRECORD *, unsigned int, IStream *);
312 extern UINT MSI_RecordGetIStream( MSIRECORD *, unsigned int, IStream **);
313 extern const WCHAR *MSI_RecordGetString( MSIRECORD *, unsigned int );
314 extern MSIRECORD *MSI_CreateRecord( unsigned int );
315 extern UINT MSI_RecordSetInteger( MSIRECORD *, unsigned int, int );
316 extern UINT MSI_RecordSetStringW( MSIRECORD *, unsigned int, LPCWSTR );
317 extern UINT MSI_RecordSetStringA( MSIRECORD *, unsigned int, LPCSTR );
318 extern BOOL MSI_RecordIsNull( MSIRECORD *, unsigned int );
319 extern UINT MSI_RecordGetStringW( MSIRECORD * , unsigned int, LPWSTR, DWORD *);
320 extern UINT MSI_RecordGetStringA( MSIRECORD *, unsigned int, LPSTR, DWORD *);
321 extern int MSI_RecordGetInteger( MSIRECORD *, unsigned int );
322 extern UINT MSI_RecordReadStream( MSIRECORD *, unsigned int, char *, DWORD *);
323 extern unsigned int MSI_RecordGetFieldCount( MSIRECORD *rec );
324 extern UINT MSI_RecordSetStreamW( MSIRECORD *, unsigned int, LPCWSTR );
325 extern UINT MSI_RecordSetStreamA( MSIRECORD *, unsigned int, LPCSTR );
326 extern UINT MSI_RecordDataSize( MSIRECORD *, unsigned int );
327 extern UINT MSI_RecordStreamToFile( MSIRECORD *, unsigned int, LPCWSTR );
328
329 /* stream internals */
330 extern UINT get_raw_stream( MSIHANDLE hdb, LPCWSTR stname, IStream **stm );
331 extern UINT db_get_raw_stream( MSIDATABASE *db, LPCWSTR stname, IStream **stm );
332 extern void enum_stream_names( IStorage *stg );
333
334 /* database internals */
335 extern UINT MSI_OpenDatabaseW( LPCWSTR, LPCWSTR, MSIDATABASE ** );
336 extern UINT MSI_DatabaseOpenViewW(MSIDATABASE *, LPCWSTR, MSIQUERY ** );
337 extern UINT MSI_OpenQuery( MSIDATABASE *, MSIQUERY **, LPCWSTR, ... );
338 typedef UINT (*record_func)( MSIRECORD *, LPVOID );
339 extern UINT MSI_IterateRecords( MSIQUERY *, DWORD *, record_func, LPVOID );
340 extern MSIRECORD *MSI_QueryGetRecord( MSIDATABASE *db, LPCWSTR query, ... );
341 extern UINT MSI_DatabaseImport( MSIDATABASE *, LPCWSTR, LPCWSTR );
342 extern UINT MSI_DatabaseExport( MSIDATABASE *, LPCWSTR, LPCWSTR, LPCWSTR );
343 extern UINT MSI_DatabaseGetPrimaryKeys( MSIDATABASE *, LPCWSTR, MSIRECORD ** );
344
345 /* view internals */
346 extern UINT MSI_ViewExecute( MSIQUERY*, MSIRECORD * );
347 extern UINT MSI_ViewFetch( MSIQUERY*, MSIRECORD ** );
348 extern UINT MSI_ViewClose( MSIQUERY* );
349
350 /* package internals */
351 extern MSIPACKAGE *MSI_CreatePackage( MSIDATABASE * );
352 extern UINT MSI_OpenPackageW( LPCWSTR szPackage, MSIPACKAGE ** );
353 extern UINT MSI_SetTargetPathW( MSIPACKAGE *, LPCWSTR, LPCWSTR );
354 extern UINT MSI_SetPropertyW( MSIPACKAGE *, LPCWSTR, LPCWSTR );
355 extern INT MSI_ProcessMessage( MSIPACKAGE *, INSTALLMESSAGE, MSIRECORD * );
356 extern UINT MSI_GetPropertyW( MSIPACKAGE *, LPCWSTR, LPWSTR, DWORD * );
357 extern UINT MSI_GetPropertyA(MSIPACKAGE *, LPCSTR, LPSTR, DWORD* );
358 extern MSICONDITION MSI_EvaluateConditionW( MSIPACKAGE *, LPCWSTR );
359 extern UINT MSI_GetComponentStateW( MSIPACKAGE *, LPWSTR, INSTALLSTATE *, INSTALLSTATE * );
360 extern UINT MSI_GetFeatureStateW( MSIPACKAGE *, LPWSTR, INSTALLSTATE *, INSTALLSTATE * );
361 extern UINT WINAPI MSI_SetFeatureStateW(MSIPACKAGE*, LPCWSTR, INSTALLSTATE );
362
363 /* for deformating */
364 extern UINT MSI_FormatRecordW( MSIPACKAGE *, MSIRECORD *, LPWSTR, DWORD * );
365 extern UINT MSI_FormatRecordA( MSIPACKAGE *, MSIRECORD *, LPSTR, DWORD * );
366     
367 /* registry data encoding/decoding functions */
368 extern BOOL unsquash_guid(LPCWSTR in, LPWSTR out);
369 extern BOOL squash_guid(LPCWSTR in, LPWSTR out);
370 extern BOOL encode_base85_guid(GUID *,LPWSTR);
371 extern BOOL decode_base85_guid(LPCWSTR,GUID*);
372 extern UINT MSIREG_OpenUninstallKey(LPCWSTR szProduct, HKEY* key, BOOL create);
373 extern UINT MSIREG_OpenUserProductsKey(LPCWSTR szProduct, HKEY* key, BOOL create);
374 extern UINT MSIREG_OpenFeatures(HKEY* key);
375 extern UINT MSIREG_OpenFeaturesKey(LPCWSTR szProduct, HKEY* key, BOOL create);
376 extern UINT MSIREG_OpenComponents(HKEY* key);
377 extern UINT MSIREG_OpenUserComponentsKey(LPCWSTR szComponent, HKEY* key, BOOL create);
378 extern UINT MSIREG_OpenComponentsKey(LPCWSTR szComponent, HKEY* key, BOOL create);
379 extern UINT MSIREG_OpenProductsKey(LPCWSTR szProduct, HKEY* key, BOOL create);
380 extern UINT MSIREG_OpenUserFeaturesKey(LPCWSTR szProduct, HKEY* key, BOOL create);
381 extern UINT MSIREG_OpenUserComponentsKey(LPCWSTR szComponent, HKEY* key, BOOL create);
382 extern UINT MSIREG_OpenUpgradeCodesKey(LPCWSTR szProduct, HKEY* key, BOOL create);
383 extern UINT MSIREG_OpenUserUpgradeCodesKey(LPCWSTR szProduct, HKEY* key, BOOL create);
384
385 extern LONG msi_reg_set_val_str( HKEY hkey, LPCWSTR name, LPCWSTR value );
386 extern LONG msi_reg_set_val_multi_str( HKEY hkey, LPCWSTR name, LPCWSTR value );
387 extern LONG msi_reg_set_val_dword( HKEY hkey, LPCWSTR name, DWORD val );
388
389 /* msi dialog interface */
390 typedef UINT (*msi_dialog_event_handler)( MSIPACKAGE*, LPCWSTR, LPCWSTR, msi_dialog* );
391 extern msi_dialog *msi_dialog_create( MSIPACKAGE*, LPCWSTR, msi_dialog_event_handler );
392 extern UINT msi_dialog_run_message_loop( msi_dialog* );
393 extern void msi_dialog_end_dialog( msi_dialog* );
394 extern void msi_dialog_check_messages( HANDLE );
395 extern void msi_dialog_do_preview( msi_dialog* );
396 extern void msi_dialog_destroy( msi_dialog* );
397 extern BOOL msi_dialog_register_class( void );
398 extern void msi_dialog_unregister_class( void );
399 extern void msi_dialog_handle_event( msi_dialog*, LPCWSTR, LPCWSTR, MSIRECORD * );
400
401 /* preview */
402 extern MSIPREVIEW *MSI_EnableUIPreview( MSIDATABASE * );
403 extern UINT MSI_PreviewDialogW( MSIPREVIEW *, LPCWSTR );
404
405 /* undocumented functions */
406 UINT WINAPI MsiCreateAndVerifyInstallerDirectory( DWORD );
407 UINT WINAPI MsiDecomposeDescriptorW( LPCWSTR, LPWSTR, LPWSTR, LPWSTR, DWORD * );
408 UINT WINAPI MsiDecomposeDescriptorA( LPCSTR, LPSTR, LPSTR, LPSTR, DWORD * );
409 LANGID WINAPI MsiLoadStringW( MSIHANDLE, UINT, LPWSTR, int, LANGID );
410 LANGID WINAPI MsiLoadStringA( MSIHANDLE, UINT, LPSTR, int, LANGID );
411
412 /* UI globals */
413 extern INSTALLUILEVEL gUILevel;
414 extern HWND gUIhwnd;
415 extern INSTALLUI_HANDLERA gUIHandlerA;
416 extern INSTALLUI_HANDLERW gUIHandlerW;
417 extern DWORD gUIFilter;
418 extern LPVOID gUIContext;
419 extern WCHAR gszLogFile[MAX_PATH];
420
421 /* memory allocation macro functions */
422 static inline void *msi_alloc( size_t len )
423 {
424     return HeapAlloc( GetProcessHeap(), 0, len );
425 }
426
427 static inline void *msi_alloc_zero( size_t len )
428 {
429     return HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, len );
430 }
431
432 static inline void *msi_realloc( void *mem, size_t len )
433 {
434     return HeapReAlloc( GetProcessHeap(), 0, mem, len );
435 }
436
437 static inline void *msi_realloc_zero( void *mem, size_t len )
438 {
439     return HeapReAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, mem, len );
440 }
441
442 static inline BOOL msi_free( void *mem )
443 {
444     return HeapFree( GetProcessHeap(), 0, mem );
445 }
446
447 inline static char *strdupWtoA( LPCWSTR str )
448 {
449     LPSTR ret = NULL;
450     DWORD len;
451
452     if (!str) return ret;
453     len = WideCharToMultiByte( CP_ACP, 0, str, -1, NULL, 0, NULL, NULL);
454     ret = msi_alloc( len );
455     if (ret)
456         WideCharToMultiByte( CP_ACP, 0, str, -1, ret, len, NULL, NULL );
457     return ret;
458 }
459
460 inline static LPWSTR strdupAtoW( LPCSTR str )
461 {
462     LPWSTR ret = NULL;
463     DWORD len;
464
465     if (!str) return ret;
466     len = MultiByteToWideChar( CP_ACP, 0, str, -1, NULL, 0 );
467     ret = msi_alloc( len * sizeof(WCHAR) );
468     if (ret)
469         MultiByteToWideChar( CP_ACP, 0, str, -1, ret, len );
470     return ret;
471 }
472
473 inline static LPWSTR strdupW( LPCWSTR src )
474 {
475     LPWSTR dest;
476     if (!src) return NULL;
477     dest = msi_alloc( (lstrlenW(src)+1)*sizeof(WCHAR) );
478     if (dest)
479         lstrcpyW(dest, src);
480     return dest;
481 }
482
483 #endif /* __WINE_MSI_PRIVATE__ */