dlls/dsquery/main.c

   1 /*
   2  * Copyright 2017 Zebediah Figura 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
  20 #include <stdarg.h>
  21
  22 #define COBJMACROS
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "objbase.h"
  27 #include "rpcproxy.h"
  28 #include "initguid.h"
  29 #include "cmnquery.h"
  30 #include "wine/debug.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(dsquery);
  33
  34 static HINSTANCE instance;
  35
  36 /******************************************************************
  37  *      IClassFactory implementation
  38  */
  39 struct query_class_factory {
  40     IClassFactory IClassFactory_iface;
  41     LONG ref;
  42     HRESULT (*pfnCreateInstance)(IUnknown *outer, REFIID riid, void **out);
  43 };
  44
  45 static inline struct query_class_factory *impl_from_IClassFactory(IClassFactory *iface)
  46 {
  47     return CONTAINING_RECORD(iface, struct query_class_factory, IClassFactory_iface);
  48 }
  49
  50 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **out)
  51 {
  52     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), out);
  53
  54     if (IsEqualGUID(riid, &IID_IClassFactory) || IsEqualGUID(riid, &IID_IUnknown))
  55     {
  56         IClassFactory_AddRef(iface);
  57         *out = iface;
  58         return S_OK;
  59     }
  60
  61     FIXME("interface %s not implemented\n", debugstr_guid(riid));
  62     *out = NULL;
  63     return E_NOINTERFACE;
  64 }
  65
  66 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
  67 {
  68     struct query_class_factory *This = impl_from_IClassFactory(iface);
  69     ULONG ref = InterlockedIncrement(&This->ref);
  70
  71     TRACE("(%p) increasing refcount to %u\n", iface, ref);
  72
  73     return ref;
  74 }
  75
  76 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
  77 {
  78     struct query_class_factory *This = impl_from_IClassFactory(iface);
  79     ULONG ref = InterlockedDecrement(&This->ref);
  80
  81     TRACE("(%p) decreasing refcount to %u\n", iface, ref);
  82
  83     if (ref == 0)
  84         HeapFree(GetProcessHeap(), 0, This);
  85
  86     return ref;
  87 }
  88
  89 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface,
  90         IUnknown *outer, REFIID riid, void **out)
  91 {
  92     struct query_class_factory *This = impl_from_IClassFactory(iface);
  93
  94     TRACE("(%p)->(%p, %s, %p)\n", iface, outer, debugstr_guid(riid), out);
  95
  96     return This->pfnCreateInstance(outer, riid, out);
  97 }
  98
  99 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
 100 {
 101     FIXME("(%p)->(%d)\n", iface, dolock);
 102
 103     return S_OK;
 104 }
 105
 106 static const IClassFactoryVtbl query_class_factory_vtbl =
 107 {
 108     ClassFactory_QueryInterface,
 109     ClassFactory_AddRef,
 110     ClassFactory_Release,
 111     ClassFactory_CreateInstance,
 112     ClassFactory_LockServer,
 113 };
 114
 115 /******************************************************************
 116  *      ICommonQuery implementation
 117  */
 118 struct common_query {
 119     ICommonQuery ICommonQuery_iface;
 120     LONG ref;
 121 };
 122
 123 static inline struct common_query *impl_from_ICommonQuery(ICommonQuery *iface)
 124 {
 125     return CONTAINING_RECORD(iface, struct common_query, ICommonQuery_iface);
 126 }
 127
 128 static HRESULT WINAPI CommonQuery_QueryInterface(ICommonQuery *iface, REFIID riid, void **out)
 129 {
 130     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), out);
 131
 132     if (IsEqualGUID(riid, &IID_ICommonQuery) || IsEqualGUID(riid, &IID_IUnknown))
 133     {
 134         ICommonQuery_AddRef(iface);
 135         *out = iface;
 136         return S_OK;
 137     }
 138
 139     FIXME("interface %s not implemented\n", debugstr_guid(riid));
 140     *out = NULL;
 141     return E_NOINTERFACE;
 142 }
 143
 144 static ULONG WINAPI CommonQuery_AddRef(ICommonQuery *iface)
 145 {
 146     struct common_query *This = impl_from_ICommonQuery(iface);
 147     ULONG ref = InterlockedIncrement(&This->ref);
 148
 149     TRACE("(%p) increasing refcount to %u\n", iface, ref);
 150
 151     return ref;
 152 }
 153
 154 static ULONG WINAPI CommonQuery_Release(ICommonQuery *iface)
 155 {
 156     struct common_query *This = impl_from_ICommonQuery(iface);
 157     ULONG ref = InterlockedDecrement(&This->ref);
 158
 159     TRACE("(%p) decreasing refcount to %u\n", iface, ref);
 160
 161     if (ref == 0)
 162         HeapFree(GetProcessHeap(), 0, This);
 163
 164     return ref;
 165 }
 166
 167 static HRESULT WINAPI CommonQuery_OpenQueryWindow(ICommonQuery *iface,
 168         HWND parent, LPOPENQUERYWINDOW query_window, IDataObject **data_object)
 169 {
 170     FIXME("(%p)->(%p, %p, %p) stub!\n", iface, parent, query_window, data_object);
 171
 172     return E_NOTIMPL;
 173 }
 174
 175 static const ICommonQueryVtbl CommonQuery_vtbl =
 176 {
 177     CommonQuery_QueryInterface,
 178     CommonQuery_AddRef,
 179     CommonQuery_Release,
 180     CommonQuery_OpenQueryWindow,
 181 };
 182
 183 static HRESULT CommonQuery_create(IUnknown *outer, REFIID riid, void **out)
 184 {
 185     struct common_query *query;
 186
 187     TRACE("outer %p, riid %s, out %p\n", outer, debugstr_guid(riid), out);
 188
 189     if (outer)
 190         return CLASS_E_NOAGGREGATION;
 191
 192     if (!(query = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*query))))
 193         return E_OUTOFMEMORY;
 194
 195     query->ICommonQuery_iface.lpVtbl = &CommonQuery_vtbl;
 196     return ICommonQuery_QueryInterface(&query->ICommonQuery_iface, riid, out);
 197 }
 198
 199 /***********************************************************************
 200  *              DllMain
 201  */
 202 BOOL WINAPI DllMain(HINSTANCE inst, DWORD reason, void *reserved)
 203 {
 204     TRACE("(%p, %u, %p)\n", inst, reason, reserved);
 205
 206     switch (reason)
 207     {
 208         case DLL_PROCESS_ATTACH:
 209             instance = inst;
 210             DisableThreadLibraryCalls(inst);
 211             break;
 212     }
 213
 214     return TRUE;
 215 }
 216
 217 /***********************************************************************
 218  *              DllCanUnloadNow (DSQUERY.@)
 219  */
 220 HRESULT WINAPI DllCanUnloadNow(void)
 221 {
 222     return S_FALSE;
 223 }
 224
 225 /***********************************************************************
 226  *              DllGetClassObject (DSQUERY.@)
 227  */
 228 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **out)
 229 {
 230     TRACE("rclsid %s, riid %s, out %p\n", debugstr_guid(rclsid), debugstr_guid(riid), out);
 231
 232     if (!IsEqualGUID( &IID_IClassFactory, riid)
 233         && !IsEqualGUID( &IID_IUnknown, riid))
 234     {
 235         FIXME("interface %s not implemented\n", debugstr_guid(riid));
 236         *out = NULL;
 237         return E_NOINTERFACE;
 238     }
 239
 240     if (IsEqualGUID(&CLSID_CommonQuery, rclsid))
 241     {
 242         struct query_class_factory *factory = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*factory));
 243         if (!factory) return E_OUTOFMEMORY;
 244
 245         factory->IClassFactory_iface.lpVtbl = &query_class_factory_vtbl;
 246         factory->ref = 1;
 247         factory->pfnCreateInstance = CommonQuery_create;
 248         *out = factory;
 249         return S_OK;
 250     }
 251
 252     FIXME("%s: no class found\n", debugstr_guid(rclsid));
 253     *out = NULL;
 254     return CLASS_E_CLASSNOTAVAILABLE;
 255 }
 256
 257 /***********************************************************************
 258  *              DllRegisterServer (DSQUERY.@)
 259  */
 260 HRESULT WINAPI DllRegisterServer(void)
 261 {
 262     return __wine_register_resources( instance );
 263 }
 264
 265 /***********************************************************************
 266  *              DllUnregisterServer (DSQUERY.@)
 267  */
 268 HRESULT WINAPI DllUnregisterServer(void)
 269 {
 270     return __wine_unregister_resources( instance );
 271 }