ntdll: Check and fix PE header protections.
[wine] / dlls / wuapi / main.c
1 /*
2  * WUAPI implementation
3  *
4  * Copyright 2008 Hans Leidekker
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 #define COBJMACROS
22
23 #include "config.h"
24 #include <stdarg.h>
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winuser.h"
29 #include "ole2.h"
30 #include "wuapi.h"
31
32 #include "wine/debug.h"
33 #include "wuapi_private.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(wuapi);
36
37 typedef HRESULT (*fnCreateInstance)( IUnknown *pUnkOuter, LPVOID *ppObj );
38
39 typedef struct _wucf
40 {
41     const struct IClassFactoryVtbl *vtbl;
42     fnCreateInstance pfnCreateInstance;
43 } wucf;
44
45 static inline wucf *impl_from_IClassFactory( IClassFactory *iface )
46 {
47     return (wucf *)((char *)iface - FIELD_OFFSET( wucf, vtbl ));
48 }
49
50 static HRESULT WINAPI wucf_QueryInterface( IClassFactory *iface, REFIID riid, LPVOID *ppobj )
51 {
52     if (IsEqualGUID(riid, &IID_IUnknown) ||
53         IsEqualGUID(riid, &IID_IClassFactory))
54     {
55         IClassFactory_AddRef( iface );
56         *ppobj = iface;
57         return S_OK;
58     }
59     FIXME("interface %s not implemented\n", debugstr_guid(riid));
60     return E_NOINTERFACE;
61 }
62
63 static ULONG WINAPI wucf_AddRef( IClassFactory *iface )
64 {
65     return 2;
66 }
67
68 static ULONG WINAPI wucf_Release( IClassFactory *iface )
69 {
70     return 1;
71 }
72
73 static HRESULT WINAPI wucf_CreateInstance( IClassFactory *iface, LPUNKNOWN pOuter,
74                                            REFIID riid, LPVOID *ppobj )
75 {
76     wucf *This = impl_from_IClassFactory( iface );
77     HRESULT r;
78     IUnknown *punk;
79
80     TRACE("%p %s %p\n", pOuter, debugstr_guid(riid), ppobj);
81
82     *ppobj = NULL;
83
84     if (pOuter)
85         return CLASS_E_NOAGGREGATION;
86
87     r = This->pfnCreateInstance( pOuter, (LPVOID *)&punk );
88     if (FAILED(r))
89         return r;
90
91     r = IUnknown_QueryInterface( punk, riid, ppobj );
92     if (FAILED(r))
93         return r;
94
95     IUnknown_Release( punk );
96     return r;
97 }
98
99 static HRESULT WINAPI wucf_LockServer( IClassFactory *iface, BOOL dolock )
100 {
101     FIXME("(%p)->(%d)\n", iface, dolock);
102     return S_OK;
103 }
104
105 static const struct IClassFactoryVtbl wucf_vtbl =
106 {
107     wucf_QueryInterface,
108     wucf_AddRef,
109     wucf_Release,
110     wucf_CreateInstance,
111     wucf_LockServer
112 };
113
114 static wucf sessioncf = { &wucf_vtbl, UpdateSession_create };
115 static wucf updatescf = { &wucf_vtbl, AutomaticUpdates_create };
116
117 BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID lpv )
118 {
119     switch(reason)
120     {
121     case DLL_WINE_PREATTACH:
122         return FALSE;  /* prefer native version */
123     case DLL_PROCESS_ATTACH:
124         DisableThreadLibraryCalls( hinst );
125         break;
126     case DLL_PROCESS_DETACH:
127         break;
128     }
129     return TRUE;
130 }
131
132 HRESULT WINAPI DllGetClassObject( REFCLSID rclsid, REFIID iid, LPVOID *ppv )
133 {
134     IClassFactory *cf = NULL;
135
136     TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(iid), ppv);
137
138     if (IsEqualGUID( rclsid, &CLSID_UpdateSession ))
139     {
140        cf = (IClassFactory *)&sessioncf.vtbl;
141     }
142     else if (IsEqualGUID( rclsid, &CLSID_AutomaticUpdates ))
143     {
144        cf = (IClassFactory *)&updatescf.vtbl;
145     }
146     if (!cf) return CLASS_E_CLASSNOTAVAILABLE;
147     return IClassFactory_QueryInterface( cf, iid, ppv );
148 }
149
150 HRESULT WINAPI DllCanUnloadNow( void )
151 {
152     return S_FALSE;
153 }