rsaenh: Store a key's value to the registry when a key is imported.
[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         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 static IClassFactoryImpl WineMsiCustomRemote_CF = { &MsiCF_Vtbl, create_msi_custom_remote };
193 static IClassFactoryImpl WineMsiRemotePackage_CF = { &MsiCF_Vtbl, create_msi_remote_package };
194
195 /******************************************************************
196  * DllGetClassObject          [MSI.@]
197  */
198 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
199 {
200     TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
201
202     if ( IsEqualCLSID (rclsid, &CLSID_IMsiServerX2) )
203     {
204         *ppv = &MsiServer_CF;
205         return S_OK;
206     }
207
208     if ( IsEqualCLSID (rclsid, &CLSID_IWineMsiRemoteCustomAction) )
209     {
210         *ppv = &WineMsiCustomRemote_CF;
211         return S_OK;
212     }
213
214     if ( IsEqualCLSID (rclsid, &CLSID_IWineMsiRemotePackage) )
215     {
216         *ppv = &WineMsiRemotePackage_CF;
217         return S_OK;
218     }
219
220     if( IsEqualCLSID (rclsid, &CLSID_IMsiServerMessage) ||
221         IsEqualCLSID (rclsid, &CLSID_IMsiServer) ||
222         IsEqualCLSID (rclsid, &CLSID_IMsiServerX1) ||
223         IsEqualCLSID (rclsid, &CLSID_IMsiServerX3) )
224     {
225         FIXME("create %s object\n", debugstr_guid( rclsid ));
226     }
227
228     return CLASS_E_CLASSNOTAVAILABLE;
229 }
230
231 /******************************************************************
232  * DllGetVersion              [MSI.@]
233  */
234 HRESULT WINAPI DllGetVersion(DLLVERSIONINFO *pdvi)
235 {
236     TRACE("%p\n",pdvi);
237
238     if (pdvi->cbSize < sizeof(DLLVERSIONINFO))
239         return E_INVALIDARG;
240
241     pdvi->dwMajorVersion = MSI_MAJORVERSION;
242     pdvi->dwMinorVersion = MSI_MINORVERSION;
243     pdvi->dwBuildNumber = MSI_BUILDNUMBER;
244     pdvi->dwPlatformID = DLLVER_PLATFORM_WINDOWS;
245
246     return S_OK;
247 }
248
249 /******************************************************************
250  * DllCanUnloadNow            [MSI.@]
251  */
252 HRESULT WINAPI DllCanUnloadNow(void)
253 {
254     return dll_count == 0 ? S_OK : S_FALSE;
255 }