hlink: Added HlinkGetSpecialReference implementation.
[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 HINSTANCE msi_hInstance;
48
49 static WCHAR msi_path[MAX_PATH];
50 static ITypeLib *msi_typelib;
51
52 /*
53  * Dll lifetime tracking declaration
54  */
55 static void LockModule(void)
56 {
57     InterlockedIncrement(&dll_count);
58 }
59
60 static void UnlockModule(void)
61 {
62     InterlockedDecrement(&dll_count);
63 }
64
65 /******************************************************************
66  *      DllMain
67  */
68 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
69 {
70     switch (fdwReason)
71     {
72     case DLL_PROCESS_ATTACH:
73         msi_hInstance = hinstDLL;
74         DisableThreadLibraryCalls(hinstDLL);
75         msi_dialog_register_class();
76         break;
77     case DLL_PROCESS_DETACH:
78         if (msi_typelib) ITypeLib_Release( msi_typelib );
79         msi_dialog_unregister_class();
80         msi_free_handle_table();
81         break;
82     }
83     return TRUE;
84 }
85
86 static CRITICAL_SECTION MSI_typelib_cs;
87 static CRITICAL_SECTION_DEBUG MSI_typelib_cs_debug =
88 {
89     0, 0, &MSI_typelib_cs,
90     { &MSI_typelib_cs_debug.ProcessLocksList,
91       &MSI_typelib_cs_debug.ProcessLocksList },
92       0, 0, { (DWORD_PTR)(__FILE__ ": MSI_typelib_cs") }
93 };
94 static CRITICAL_SECTION MSI_typelib_cs = { &MSI_typelib_cs_debug, -1, 0, 0, 0, 0 };
95
96 ITypeLib *get_msi_typelib( LPWSTR *path )
97 {
98     EnterCriticalSection( &MSI_typelib_cs );
99
100     if (!msi_typelib)
101     {
102         TRACE("loading typelib\n");
103
104         if (GetModuleFileNameW( msi_hInstance, msi_path, MAX_PATH ))
105             LoadTypeLib( msi_path, &msi_typelib );
106     }
107
108     LeaveCriticalSection( &MSI_typelib_cs );
109
110     if (path)
111         *path = msi_path;
112
113     if (msi_typelib)
114         ITypeLib_AddRef( msi_typelib );
115
116     return msi_typelib;
117 }
118
119 typedef struct tagIClassFactoryImpl {
120     const IClassFactoryVtbl *lpVtbl;
121     HRESULT (*create_object)( IUnknown*, LPVOID* );
122 } IClassFactoryImpl;
123
124 static HRESULT WINAPI MsiCF_QueryInterface(LPCLASSFACTORY iface,
125                 REFIID riid,LPVOID *ppobj)
126 {
127     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
128
129     TRACE("%p %s %p\n",This,debugstr_guid(riid),ppobj);
130
131     if( IsEqualCLSID( riid, &IID_IUnknown ) ||
132         IsEqualCLSID( riid, &IID_IClassFactory ) )
133     {
134         IClassFactory_AddRef( iface );
135         *ppobj = iface;
136         return S_OK;
137     }
138     return E_NOINTERFACE;
139 }
140
141 static ULONG WINAPI MsiCF_AddRef(LPCLASSFACTORY iface)
142 {
143     LockModule();
144     return 2;
145 }
146
147 static ULONG WINAPI MsiCF_Release(LPCLASSFACTORY iface)
148 {
149     UnlockModule();
150     return 1;
151 }
152
153 static HRESULT WINAPI MsiCF_CreateInstance(LPCLASSFACTORY iface,
154     LPUNKNOWN pOuter, REFIID riid, LPVOID *ppobj)
155 {
156     IClassFactoryImpl *This = (IClassFactoryImpl *)iface;
157     IUnknown *unk = NULL;
158     HRESULT r;
159
160     TRACE("%p %p %s %p\n", This, pOuter, debugstr_guid(riid), ppobj);
161
162     r = This->create_object( pOuter, (LPVOID*) &unk );
163     if (SUCCEEDED(r))
164     {
165         r = IUnknown_QueryInterface( unk, riid, ppobj );
166         IUnknown_Release( unk );
167     }
168     return r;
169 }
170
171 static HRESULT WINAPI MsiCF_LockServer(LPCLASSFACTORY iface, BOOL dolock)
172 {
173     TRACE("%p %d\n", iface, dolock);
174
175     if (dolock)
176         LockModule();
177     else
178         UnlockModule();
179
180     return S_OK;
181 }
182
183 static const IClassFactoryVtbl MsiCF_Vtbl =
184 {
185     MsiCF_QueryInterface,
186     MsiCF_AddRef,
187     MsiCF_Release,
188     MsiCF_CreateInstance,
189     MsiCF_LockServer
190 };
191
192 static IClassFactoryImpl MsiServer_CF = { &MsiCF_Vtbl, create_msiserver };
193 static IClassFactoryImpl WineMsiCustomRemote_CF = { &MsiCF_Vtbl, create_msi_custom_remote };
194 static IClassFactoryImpl WineMsiRemotePackage_CF = { &MsiCF_Vtbl, create_msi_remote_package };
195
196 /******************************************************************
197  * DllGetClassObject          [MSI.@]
198  */
199 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
200 {
201     TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
202
203     if ( IsEqualCLSID (rclsid, &CLSID_IMsiServerX2) )
204     {
205         *ppv = (LPVOID) &MsiServer_CF;
206         return S_OK;
207     }
208
209     if ( IsEqualCLSID (rclsid, &CLSID_IWineMsiRemoteCustomAction) )
210     {
211         *ppv = (LPVOID) &WineMsiCustomRemote_CF;
212         return S_OK;
213     }
214
215     if ( IsEqualCLSID (rclsid, &CLSID_IWineMsiRemotePackage) )
216     {
217         *ppv = (LPVOID) &WineMsiRemotePackage_CF;
218         return S_OK;
219     }
220
221     if( IsEqualCLSID (rclsid, &CLSID_IMsiServerMessage) ||
222         IsEqualCLSID (rclsid, &CLSID_IMsiServer) ||
223         IsEqualCLSID (rclsid, &CLSID_IMsiServerX1) ||
224         IsEqualCLSID (rclsid, &CLSID_IMsiServerX3) )
225     {
226         FIXME("create %s object\n", debugstr_guid( rclsid ));
227     }
228
229     return CLASS_E_CLASSNOTAVAILABLE;
230 }
231
232 /******************************************************************
233  * DllGetVersion              [MSI.@]
234  */
235 HRESULT WINAPI DllGetVersion(DLLVERSIONINFO *pdvi)
236 {
237     TRACE("%p\n",pdvi);
238
239     if (pdvi->cbSize < sizeof(DLLVERSIONINFO))
240         return E_INVALIDARG;
241
242     pdvi->dwMajorVersion = MSI_MAJORVERSION;
243     pdvi->dwMinorVersion = MSI_MINORVERSION;
244     pdvi->dwBuildNumber = MSI_BUILDNUMBER;
245     pdvi->dwPlatformID = DLLVER_PLATFORM_WINDOWS;
246
247     return S_OK;
248 }
249
250 /******************************************************************
251  * DllCanUnloadNow            [MSI.@]
252  */
253 HRESULT WINAPI DllCanUnloadNow(void)
254 {
255     return dll_count == 0 ? S_OK : S_FALSE;
256 }