dlls/wbemdisp/main.c

   1 /*
   2  * Copyright 2013 Hans Leidekker for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <stdarg.h>
  20
  21 #define COBJMACROS
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "objbase.h"
  26 #include "wmiutils.h"
  27 #include "wbemdisp.h"
  28 #include "rpcproxy.h"
  29
  30 #include "wine/debug.h"
  31 #include "wine/heap.h"
  32 #include "wbemdisp_private.h"
  33 #include "wbemdisp_classes.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(wbemdisp);
  36
  37 static HINSTANCE instance;
  38
  39 static HRESULT WINAPI WinMGMTS_QueryInterface(IParseDisplayName *iface, REFIID riid, void **ppv)
  40 {
  41     if(IsEqualGUID(riid, &IID_IUnknown)) {
  42         TRACE("(IID_IUnknown %p)\n", ppv);
  43         *ppv = iface;
  44     }else if(IsEqualGUID(riid, &IID_IParseDisplayName)) {
  45         TRACE("(IID_IParseDisplayName %p)\n", ppv);
  46         *ppv = iface;
  47     }else {
  48         WARN("Unsupported riid %s\n", debugstr_guid(riid));
  49         *ppv = NULL;
  50         return E_NOINTERFACE;
  51     }
  52
  53     IUnknown_AddRef((IUnknown*)*ppv);
  54     return S_OK;
  55 }
  56
  57 static ULONG WINAPI WinMGMTS_AddRef(IParseDisplayName *iface)
  58 {
  59     return 2;
  60 }
  61
  62 static ULONG WINAPI WinMGMTS_Release(IParseDisplayName *iface)
  63 {
  64     return 1;
  65 }
  66
  67 static HRESULT parse_path( const WCHAR *str, BSTR *server, BSTR *namespace, BSTR *relative )
  68 {
  69     IWbemPath *path;
  70     ULONG len;
  71     HRESULT hr;
  72
  73     *server = *namespace = *relative = NULL;
  74
  75     hr = CoCreateInstance( &CLSID_WbemDefPath, NULL, CLSCTX_INPROC_SERVER, &IID_IWbemPath, (void **)&path );
  76     if (hr != S_OK) return hr;
  77
  78     hr = IWbemPath_SetText( path, WBEMPATH_CREATE_ACCEPT_ALL, str );
  79     if (hr != S_OK) goto done;
  80
  81     len = 0;
  82     hr = IWbemPath_GetServer( path, &len, NULL );
  83     if (hr == S_OK)
  84     {
  85         if (!(*server = SysAllocStringLen( NULL, len )))
  86         {
  87             hr = E_OUTOFMEMORY;
  88             goto done;
  89         }
  90         hr = IWbemPath_GetServer( path, &len, *server );
  91         if (hr != S_OK) goto done;
  92     }
  93
  94     len = 0;
  95     hr = IWbemPath_GetText( path, WBEMPATH_GET_NAMESPACE_ONLY, &len, NULL );
  96     if (hr == S_OK)
  97     {
  98         if (!(*namespace = SysAllocStringLen( NULL, len )))
  99         {
 100             hr = E_OUTOFMEMORY;
 101             goto done;
 102         }
 103         hr = IWbemPath_GetText( path, WBEMPATH_GET_NAMESPACE_ONLY, &len, *namespace );
 104         if (hr != S_OK) goto done;
 105     }
 106     len = 0;
 107     hr = IWbemPath_GetText( path, WBEMPATH_GET_RELATIVE_ONLY, &len, NULL );
 108     if (hr == S_OK)
 109     {
 110         if (!(*relative = SysAllocStringLen( NULL, len )))
 111         {
 112             hr = E_OUTOFMEMORY;
 113             goto done;
 114         }
 115         hr = IWbemPath_GetText( path, WBEMPATH_GET_RELATIVE_ONLY, &len, *relative );
 116     }
 117
 118 done:
 119     IWbemPath_Release( path );
 120     if (hr != S_OK)
 121     {
 122         SysFreeString( *server );
 123         SysFreeString( *namespace );
 124         SysFreeString( *relative );
 125     }
 126     return hr;
 127 }
 128
 129 static HRESULT WINAPI WinMGMTS_ParseDisplayName(IParseDisplayName *iface, IBindCtx *pbc, LPOLESTR pszDisplayName,
 130         ULONG *pchEaten, IMoniker **ppmkOut)
 131 {
 132     const DWORD prefix_len = ARRAY_SIZE(L"winmgmts:") - 1;
 133     ISWbemLocator *locator = NULL;
 134     ISWbemServices *services = NULL;
 135     ISWbemObject *obj = NULL;
 136     BSTR server, namespace, relative;
 137     WCHAR *p;
 138     HRESULT hr;
 139
 140     TRACE( "%p, %p, %s, %p, %p\n", iface, pbc, debugstr_w(pszDisplayName), pchEaten, ppmkOut );
 141
 142     if (wcsnicmp( pszDisplayName, L"winmgmts:", prefix_len )) return MK_E_SYNTAX;
 143
 144     p = pszDisplayName + prefix_len;
 145     if (*p == '{')
 146     {
 147         FIXME( "ignoring security settings\n" );
 148         while (*p && *p != '}') p++;
 149         if (*p == '}') p++;
 150         if (*p == '!') p++;
 151     }
 152     hr = parse_path( p, &server, &namespace, &relative );
 153     if (hr != S_OK) return hr;
 154
 155     hr = SWbemLocator_create( (void **)&locator );
 156     if (hr != S_OK) goto done;
 157
 158     hr = ISWbemLocator_ConnectServer( locator, server, namespace, NULL, NULL, NULL, NULL, 0, NULL, &services );
 159     if (hr != S_OK) goto done;
 160
 161     if (!relative || !*relative) CreatePointerMoniker( (IUnknown *)services, ppmkOut );
 162     else
 163     {
 164         hr = ISWbemServices_Get( services, relative, 0, NULL, &obj );
 165         if (hr != S_OK) goto done;
 166         hr = CreatePointerMoniker( (IUnknown *)obj, ppmkOut );
 167     }
 168
 169 done:
 170     if (obj) ISWbemObject_Release( obj );
 171     if (services) ISWbemServices_Release( services );
 172     if (locator) ISWbemLocator_Release( locator );
 173     SysFreeString( server );
 174     SysFreeString( namespace );
 175     SysFreeString( relative );
 176     if (hr == S_OK) *pchEaten = lstrlenW( pszDisplayName );
 177     return hr;
 178 }
 179
 180 static const IParseDisplayNameVtbl WinMGMTSVtbl = {
 181     WinMGMTS_QueryInterface,
 182     WinMGMTS_AddRef,
 183     WinMGMTS_Release,
 184     WinMGMTS_ParseDisplayName
 185 };
 186
 187 static IParseDisplayName winmgmts = { &WinMGMTSVtbl };
 188
 189 static HRESULT WinMGMTS_create(void **ppv)
 190 {
 191     *ppv = &winmgmts;
 192     return S_OK;
 193 }
 194
 195 struct factory
 196 {
 197     IClassFactory IClassFactory_iface;
 198     HRESULT (*fnCreateInstance)( LPVOID * );
 199 };
 200
 201 static inline struct factory *impl_from_IClassFactory( IClassFactory *iface )
 202 {
 203     return CONTAINING_RECORD( iface, struct factory, IClassFactory_iface );
 204 }
 205
 206 static HRESULT WINAPI factory_QueryInterface( IClassFactory *iface, REFIID riid, LPVOID *obj )
 207 {
 208     if (IsEqualGUID( riid, &IID_IUnknown ) || IsEqualGUID( riid, &IID_IClassFactory ))
 209     {
 210         IClassFactory_AddRef( iface );
 211         *obj = iface;
 212         return S_OK;
 213     }
 214     WARN( "interface %s not implemented\n", debugstr_guid(riid) );
 215     return E_NOINTERFACE;
 216 }
 217
 218 static ULONG WINAPI factory_AddRef( IClassFactory *iface )
 219 {
 220     return 2;
 221 }
 222
 223 static ULONG WINAPI factory_Release( IClassFactory *iface )
 224 {
 225     return 1;
 226 }
 227
 228 static HRESULT WINAPI factory_CreateInstance( IClassFactory *iface, LPUNKNOWN outer, REFIID riid,
 229                                               LPVOID *obj )
 230 {
 231     struct factory *factory = impl_from_IClassFactory( iface );
 232     IUnknown *unk;
 233     HRESULT hr;
 234
 235     TRACE( "%p, %s, %p\n", outer, debugstr_guid(riid), obj );
 236
 237     *obj = NULL;
 238     if (outer) return CLASS_E_NOAGGREGATION;
 239
 240     hr = factory->fnCreateInstance( (LPVOID *)&unk );
 241     if (FAILED( hr ))
 242         return hr;
 243
 244     hr = IUnknown_QueryInterface( unk, riid, obj );
 245     IUnknown_Release( unk );
 246     return hr;
 247 }
 248
 249 static HRESULT WINAPI factory_LockServer( IClassFactory *iface, BOOL lock )
 250 {
 251     FIXME( "%p, %d\n", iface, lock );
 252     return S_OK;
 253 }
 254
 255 static const struct IClassFactoryVtbl factory_vtbl =
 256 {
 257     factory_QueryInterface,
 258     factory_AddRef,
 259     factory_Release,
 260     factory_CreateInstance,
 261     factory_LockServer
 262 };
 263
 264 static struct factory swbem_locator_cf = { { &factory_vtbl }, SWbemLocator_create };
 265 static struct factory swbem_namedvalueset_cf = { { &factory_vtbl }, SWbemNamedValueSet_create };
 266 static struct factory winmgmts_cf = { { &factory_vtbl }, WinMGMTS_create };
 267
 268 BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID reserved )
 269 {
 270
 271     switch (reason)
 272     {
 273         case DLL_PROCESS_ATTACH:
 274             instance = hinst;
 275             DisableThreadLibraryCalls( hinst );
 276             break;
 277     }
 278     return TRUE;
 279 }
 280
 281 HRESULT WINAPI DllGetClassObject( REFCLSID rclsid, REFIID iid, LPVOID *obj )
 282 {
 283     IClassFactory *cf = NULL;
 284
 285     TRACE( "%s, %s, %p\n", debugstr_guid(rclsid), debugstr_guid(iid), obj );
 286
 287     if (IsEqualGUID( rclsid, &CLSID_SWbemLocator ))
 288         cf = &swbem_locator_cf.IClassFactory_iface;
 289     else if (IsEqualGUID( rclsid, &CLSID_WinMGMTS ))
 290         cf = &winmgmts_cf.IClassFactory_iface;
 291     else if (IsEqualGUID( rclsid, &CLSID_SWbemNamedValueSet ))
 292         cf = &swbem_namedvalueset_cf.IClassFactory_iface;
 293     else
 294         return CLASS_E_CLASSNOTAVAILABLE;
 295
 296     return IClassFactory_QueryInterface( cf, iid, obj );
 297 }
 298
 299 /***********************************************************************
 300  *      DllCanUnloadNow (WBEMDISP.@)
 301  */
 302 HRESULT WINAPI DllCanUnloadNow(void)
 303 {
 304     return S_FALSE;
 305 }
 306
 307 /***********************************************************************
 308  *      DllRegisterServer (WBEMDISP.@)
 309  */
 310 HRESULT WINAPI DllRegisterServer(void)
 311 {
 312     return __wine_register_resources( instance );
 313 }
 314
 315 /***********************************************************************
 316  *      DllUnregisterServer (WBEMDISP.@)
 317  */
 318 HRESULT WINAPI DllUnregisterServer(void)
 319 {
 320     return __wine_unregister_resources( instance );
 321 }