msi: Load the component states in CostFinalize instead of CostInitialize.
[wine] / dlls / msi / handle.c
1 /*
2  * Implementation of the Microsoft Installer (msi.dll)
3  *
4  * Copyright 2002-2004 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <stdarg.h>
22
23 #include "windef.h"
24 #include "winbase.h"
25 #include "winreg.h"
26 #include "shlwapi.h"
27 #include "wine/debug.h"
28 #include "msi.h"
29 #include "msiquery.h"
30 #include "msipriv.h"
31
32 WINE_DEFAULT_DEBUG_CHANNEL(msi);
33
34 static CRITICAL_SECTION MSI_handle_cs;
35 static CRITICAL_SECTION_DEBUG MSI_handle_cs_debug =
36 {
37     0, 0, &MSI_handle_cs,
38     { &MSI_handle_cs_debug.ProcessLocksList, 
39       &MSI_handle_cs_debug.ProcessLocksList },
40       0, 0, { (DWORD_PTR)(__FILE__ ": MSI_handle_cs") }
41 };
42 static CRITICAL_SECTION MSI_handle_cs = { &MSI_handle_cs_debug, -1, 0, 0, 0, 0 };
43
44 static CRITICAL_SECTION MSI_object_cs;
45 static CRITICAL_SECTION_DEBUG MSI_object_cs_debug =
46 {
47     0, 0, &MSI_object_cs,
48     { &MSI_object_cs_debug.ProcessLocksList, 
49       &MSI_object_cs_debug.ProcessLocksList },
50       0, 0, { (DWORD_PTR)(__FILE__ ": MSI_object_cs") }
51 };
52 static CRITICAL_SECTION MSI_object_cs = { &MSI_object_cs_debug, -1, 0, 0, 0, 0 };
53
54 typedef struct msi_handle_info_t
55 {
56     MSIOBJECTHDR *obj;
57     DWORD dwThreadId;
58 } msi_handle_info;
59
60 static msi_handle_info *msihandletable = NULL;
61 static int msihandletable_size = 0;
62
63 void msi_free_handle_table(void)
64 {
65     msi_free( msihandletable );
66     msihandletable = NULL;
67     msihandletable_size = 0;
68 }
69
70 MSIHANDLE alloc_msihandle( MSIOBJECTHDR *obj )
71 {
72     MSIHANDLE ret = 0;
73     UINT i;
74
75     EnterCriticalSection( &MSI_handle_cs );
76
77     /* find a slot */
78     for(i=0; i<msihandletable_size; i++)
79         if( !msihandletable[i].obj )
80             break;
81     if( i==msihandletable_size )
82     {
83         msi_handle_info *p;
84         int newsize;
85         if (msihandletable_size == 0)
86         {
87             newsize = 256;
88             p = msi_alloc_zero(newsize*sizeof(msi_handle_info));
89         }
90         else
91         {
92             newsize = msihandletable_size * 2;
93             p = msi_realloc_zero(msihandletable,
94                             newsize*sizeof(msi_handle_info));
95         }
96         if (!p)
97             goto out;
98         msihandletable = p;
99         msihandletable_size = newsize;
100     }
101
102     msiobj_addref( obj );
103     msihandletable[i].obj = obj;
104     msihandletable[i].dwThreadId = GetCurrentThreadId();
105     ret = (MSIHANDLE) (i+1);
106 out:
107     TRACE("%p -> %ld\n", obj, ret );
108
109     LeaveCriticalSection( &MSI_handle_cs );
110     return ret;
111 }
112
113 void *msihandle2msiinfo(MSIHANDLE handle, UINT type)
114 {
115     MSIOBJECTHDR *ret = NULL;
116
117     EnterCriticalSection( &MSI_handle_cs );
118     handle--;
119     if( handle<0 )
120         goto out;
121     if( handle>=msihandletable_size )
122         goto out;
123     if( !msihandletable[handle].obj )
124         goto out;
125     if( msihandletable[handle].obj->magic != MSIHANDLE_MAGIC )
126         goto out;
127     if( type && (msihandletable[handle].obj->type != type) )
128         goto out;
129     ret = msihandletable[handle].obj;
130     msiobj_addref( ret );
131     
132 out:
133     LeaveCriticalSection( &MSI_handle_cs );
134
135     return (void*) ret;
136 }
137
138 void *alloc_msiobject(UINT type, UINT size, msihandledestructor destroy )
139 {
140     MSIOBJECTHDR *info;
141
142     info = msi_alloc_zero( size );
143     if( info )
144     {
145         info->magic = MSIHANDLE_MAGIC;
146         info->type = type;
147         info->refcount = 1;
148         info->destructor = destroy;
149     }
150
151     return info;
152 }
153
154 void msiobj_addref( MSIOBJECTHDR *info )
155 {
156     TRACE("%p\n", info);
157
158     if( !info )
159         return;
160
161     if( info->magic != MSIHANDLE_MAGIC )
162     {
163         ERR("Invalid handle!\n");
164         return;
165     }
166
167     InterlockedIncrement(&info->refcount);
168 }
169
170 void msiobj_lock( MSIOBJECTHDR *info )
171 {
172     EnterCriticalSection( &MSI_object_cs );
173 }
174
175 void msiobj_unlock( MSIOBJECTHDR *info )
176 {
177     LeaveCriticalSection( &MSI_object_cs );
178 }
179
180 int msiobj_release( MSIOBJECTHDR *info )
181 {
182     int ret;
183
184     TRACE("%p\n",info);
185
186     if( !info )
187         return -1;
188
189     if( info->magic != MSIHANDLE_MAGIC )
190     {
191         ERR("Invalid handle!\n");
192         return -1;
193     }
194
195     ret = InterlockedDecrement( &info->refcount );
196     if( ret==0 )
197     {
198         if( info->destructor )
199             info->destructor( info );
200         msi_free( info );
201         TRACE("object %p destroyed\n", info);
202     }
203
204     return ret;
205 }
206
207 /***********************************************************
208  *   MsiCloseHandle   [MSI.@]
209  */
210 UINT WINAPI MsiCloseHandle(MSIHANDLE handle)
211 {
212     MSIOBJECTHDR *info;
213     UINT ret = ERROR_INVALID_HANDLE;
214
215     TRACE("%lx\n",handle);
216
217     if (!handle)
218         return ERROR_SUCCESS;
219
220     EnterCriticalSection( &MSI_handle_cs );
221
222     info = msihandle2msiinfo(handle, 0);
223     if( !info )
224         goto out;
225
226     if( info->magic != MSIHANDLE_MAGIC )
227     {
228         ERR("Invalid handle!\n");
229         goto out;
230     }
231
232     msiobj_release( info );
233     msihandletable[handle-1].obj = NULL;
234     ret = ERROR_SUCCESS;
235
236     TRACE("handle %lx Destroyed\n", handle);
237 out:
238     LeaveCriticalSection( &MSI_handle_cs );
239     if( info )
240         msiobj_release( info );
241
242     return ret;
243 }
244
245 /***********************************************************
246  *   MsiCloseAllHandles   [MSI.@]
247  *
248  *  Closes all handles owned by the current thread
249  *
250  *  RETURNS:
251  *   The number of handles closed
252  */
253 UINT WINAPI MsiCloseAllHandles(void)
254 {
255     UINT i, n=0;
256
257     TRACE("\n");
258
259     EnterCriticalSection( &MSI_handle_cs );
260     for(i=0; i<msihandletable_size; i++)
261     {
262         if(msihandletable[i].dwThreadId == GetCurrentThreadId())
263         {
264             LeaveCriticalSection( &MSI_handle_cs );
265             MsiCloseHandle( i+1 );
266             EnterCriticalSection( &MSI_handle_cs );
267             n++;
268         }
269     }
270     LeaveCriticalSection( &MSI_handle_cs );
271
272     return n;
273 }