winmm: Use criticalsection for determining wether timer is synchronously killable.
[wine] / dlls / msi / msi_main.c
1 /*
2  * Implementation of the Microsoft Installer (msi.dll)
3  *
4  * Copyright 2006 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 #define COBJMACROS
24 #define NONAMELESSUNION
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winreg.h"
29 #include "shlwapi.h"
30 #include "oleauto.h"
31 #include "msipriv.h"
32
33 #include "wine/debug.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(msi);
36
37 static LONG dll_count;
38
39 /* the UI level */
40 INSTALLUILEVEL gUILevel = INSTALLUILEVEL_BASIC;
41 HWND           gUIhwnd = 0;
42 INSTALLUI_HANDLERA gUIHandlerA = NULL;
43 INSTALLUI_HANDLERW gUIHandlerW = NULL;
44 DWORD gUIFilter = 0;
45 LPVOID gUIContext = NULL;
46 WCHAR gszLogFile[MAX_PATH];
47 WCHAR msi_path[MAX_PATH];
48 ITypeLib *msi_typelib = NULL;
49 HINSTANCE msi_hInstance;
50
51 /*
52  * Dll lifetime tracking declaration
53  */
54 static void LockModule(void)
55 {
56     InterlockedIncrement(&dll_count);
57 }
58
59 static void UnlockModule(void)
60 {
61     InterlockedDecrement(&dll_count);
62 }
63
64 /******************************************************************
65  *      DllMain
66  */
67 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
68 {
69     switch (fdwReason)
70     {
71     case DLL_PROCESS_ATTACH:
72         msi_hInstance = hinstDLL;
73         DisableThreadLibraryCalls(hinstDLL);
74         msi_dialog_register_class();
75         break;
76     case DLL_PROCESS_DETACH:
77         if (msi_typelib) ITypeLib_Release( msi_typelib );
78         msi_dialog_unregister_class();
79         msi_free_handle_table();
80         break;
81     }
82     return TRUE;
83 }
84
85 static CRITICAL_SECTION MSI_typelib_cs;
86 static CRITICAL_SECTION_DEBUG MSI_typelib_cs_debug =
87 {
88     0, 0, &MSI_typelib_cs,
89     { &MSI_typelib_cs_debug.ProcessLocksList,
90       &MSI_typelib_cs_debug.ProcessLocksList },
91       0, 0, { (DWORD_PTR)(__FILE__ ": MSI_typelib_cs") }
92 };
93 static CRITICAL_SECTION MSI_typelib_cs = { &MSI_typelib_cs_debug, -1, 0, 0, 0, 0 };
94
95 ITypeLib *get_msi_typelib( LPWSTR *path )
96 {
97     EnterCriticalSection( &MSI_typelib_cs );
98
99     if (!msi_typelib)
100     {
101         TRACE("loading typelib\n");
102
103         if (GetModuleFileNameW( msi_hInstance, msi_path, MAX_PATH ))
104             LoadTypeLib( msi_path, &msi_typelib );
105     }
106
107     LeaveCriticalSection( &MSI_typelib_cs );
108
109     if (path)
110         *path = msi_path;
111
112     if (msi_typelib)
113         ITypeLib_AddRef( msi_typelib );
114
115     return msi_typelib;
116 }
117
118 typedef struct tagIClassFactoryImpl {
119     const IClassFactoryVtbl *lpVtbl;
120     HRESULT (*create_object)( IUnknown*, LPVOID* );
121 } IClassFactoryImpl;
122
123 static HRESULT WINAPI MsiCF_QueryInterface(LPCLASSFACTORY iface,
124                 REFIID riid,LPVOID *ppobj)
125 {
126     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
127
128     TRACE("%p %s %p\n",This,debugstr_guid(riid),ppobj);
129
130     if( IsEqualCLSID( riid, &IID_IUnknown ) ||
131         IsEqualCLSID( riid, &IID_IClassFactory ) )
132     {
133         IClassFactory_AddRef( iface );
134         *ppobj = iface;
135         return S_OK;
136     }
137     return E_NOINTERFACE;
138 }
139
140 static ULONG WINAPI MsiCF_AddRef(LPCLASSFACTORY iface)
141 {
142     LockModule();
143     return 2;
144 }
145
146 static ULONG WINAPI MsiCF_Release(LPCLASSFACTORY iface)
147 {
148     UnlockModule();
149     return 1;
150 }
151
152 static HRESULT WINAPI MsiCF_CreateInstance(LPCLASSFACTORY iface,
153     LPUNKNOWN pOuter, REFIID riid, LPVOID *ppobj)
154 {
155     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
156     IUnknown *unk = NULL;
157     HRESULT r;
158
159     TRACE("%p %p %s %p\n", This, pOuter, debugstr_guid(riid), ppobj);
160
161     r = This->create_object( pOuter, (LPVOID*) &unk );
162     if (SUCCEEDED(r))
163     {
164         r = IUnknown_QueryInterface( unk, riid, ppobj );
165         IUnknown_Release( unk );
166     }
167     return r;
168 }
169
170 static HRESULT WINAPI MsiCF_LockServer(LPCLASSFACTORY iface, BOOL dolock)
171 {
172     TRACE("%p %d\n", iface, dolock);
173
174     if (dolock)
175         LockModule();
176     else
177         UnlockModule();
178
179     return S_OK;
180 }
181
182 static const IClassFactoryVtbl MsiCF_Vtbl =
183 {
184     MsiCF_QueryInterface,
185     MsiCF_AddRef,
186     MsiCF_Release,
187     MsiCF_CreateInstance,
188     MsiCF_LockServer
189 };
190
191 static IClassFactoryImpl MsiServer_CF = { &MsiCF_Vtbl, create_msiserver };
192
193 /******************************************************************
194  * DllGetClassObject          [MSI.@]
195  */
196 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
197 {
198     TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
199
200     if ( IsEqualCLSID (rclsid, &CLSID_IMsiServerX2) )
201     {
202         *ppv = (LPVOID) &MsiServer_CF;
203         return S_OK;
204     }
205
206     if( IsEqualCLSID (rclsid, &CLSID_IMsiServerMessage) ||
207         IsEqualCLSID (rclsid, &CLSID_IMsiServer) ||
208         IsEqualCLSID (rclsid, &CLSID_IMsiServerX1) ||
209         IsEqualCLSID (rclsid, &CLSID_IMsiServerX3) )
210     {
211         FIXME("create %s object\n", debugstr_guid( rclsid ));
212     }
213
214     return CLASS_E_CLASSNOTAVAILABLE;
215 }
216
217 /******************************************************************
218  * DllGetVersion              [MSI.@]
219  */
220 HRESULT WINAPI DllGetVersion(DLLVERSIONINFO *pdvi)
221 {
222     TRACE("%p\n",pdvi);
223
224     if (pdvi->cbSize < sizeof(DLLVERSIONINFO))
225         return E_INVALIDARG;
226
227     pdvi->dwMajorVersion = MSI_MAJORVERSION;
228     pdvi->dwMinorVersion = MSI_MINORVERSION;
229     pdvi->dwBuildNumber = MSI_BUILDNUMBER;
230     pdvi->dwPlatformID = DLLVER_PLATFORM_WINDOWS;
231
232     return S_OK;
233 }
234
235 /******************************************************************
236  * DllCanUnloadNow            [MSI.@]
237  */
238 HRESULT WINAPI DllCanUnloadNow(void)
239 {
240     return dll_count == 0 ? S_OK : S_FALSE;
241 }