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 /******************************************************************
  35  *      IClassFactory implementation
  36  */
  37 struct query_class_factory {
  38     IClassFactory IClassFactory_iface;
  39     LONG ref;
  40     HRESULT (*pfnCreateInstance)(IUnknown *outer, REFIID riid, void **out);
  41 };
  42
  43 static inline struct query_class_factory *impl_from_IClassFactory(IClassFactory *iface)
  44 {
  45     return CONTAINING_RECORD(iface, struct query_class_factory, IClassFactory_iface);
  46 }
  47
  48 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **out)
  49 {
  50     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), out);
  51
  52     if (IsEqualGUID(riid, &IID_IClassFactory) || IsEqualGUID(riid, &IID_IUnknown))
  53     {
  54         IClassFactory_AddRef(iface);
  55         *out = iface;
  56         return S_OK;
  57     }
  58
  59     FIXME("interface %s not implemented\n", debugstr_guid(riid));
  60     *out = NULL;
  61     return E_NOINTERFACE;
  62 }
  63
  64 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
  65 {
  66     struct query_class_factory *This = impl_from_IClassFactory(iface);
  67     ULONG ref = InterlockedIncrement(&This->ref);
  68
  69     TRACE("(%p) increasing refcount to %lu\n", iface, ref);
  70
  71     return ref;
  72 }
  73
  74 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
  75 {
  76     struct query_class_factory *This = impl_from_IClassFactory(iface);
  77     ULONG ref = InterlockedDecrement(&This->ref);
  78
  79     TRACE("(%p) decreasing refcount to %lu\n", iface, ref);
  80
  81     if (ref == 0)
  82         HeapFree(GetProcessHeap(), 0, This);
  83
  84     return ref;
  85 }
  86
  87 static HRESULT WINAPI ClassFactory_CreateInstance(IClassFactory *iface,
  88         IUnknown *outer, REFIID riid, void **out)
  89 {
  90     struct query_class_factory *This = impl_from_IClassFactory(iface);
  91
  92     TRACE("(%p)->(%p, %s, %p)\n", iface, outer, debugstr_guid(riid), out);
  93
  94     return This->pfnCreateInstance(outer, riid, out);
  95 }
  96
  97 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL dolock)
  98 {
  99     FIXME("(%p)->(%d)\n", iface, dolock);
 100
 101     return S_OK;
 102 }
 103
 104 static const IClassFactoryVtbl query_class_factory_vtbl =
 105 {
 106     ClassFactory_QueryInterface,
 107     ClassFactory_AddRef,
 108     ClassFactory_Release,
 109     ClassFactory_CreateInstance,
 110     ClassFactory_LockServer,
 111 };
 112
 113 /******************************************************************
 114  *      ICommonQuery implementation
 115  */
 116 struct common_query {
 117     ICommonQuery ICommonQuery_iface;
 118     LONG ref;
 119 };
 120
 121 static inline struct common_query *impl_from_ICommonQuery(ICommonQuery *iface)
 122 {
 123     return CONTAINING_RECORD(iface, struct common_query, ICommonQuery_iface);
 124 }
 125
 126 static HRESULT WINAPI CommonQuery_QueryInterface(ICommonQuery *iface, REFIID riid, void **out)
 127 {
 128     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), out);
 129
 130     if (IsEqualGUID(riid, &IID_ICommonQuery) || IsEqualGUID(riid, &IID_IUnknown))
 131     {
 132         ICommonQuery_AddRef(iface);
 133         *out = iface;
 134         return S_OK;
 135     }
 136
 137     FIXME("interface %s not implemented\n", debugstr_guid(riid));
 138     *out = NULL;
 139     return E_NOINTERFACE;
 140 }
 141
 142 static ULONG WINAPI CommonQuery_AddRef(ICommonQuery *iface)
 143 {
 144     struct common_query *This = impl_from_ICommonQuery(iface);
 145     ULONG ref = InterlockedIncrement(&This->ref);
 146
 147     TRACE("(%p) increasing refcount to %lu\n", iface, ref);
 148
 149     return ref;
 150 }
 151
 152 static ULONG WINAPI CommonQuery_Release(ICommonQuery *iface)
 153 {
 154     struct common_query *This = impl_from_ICommonQuery(iface);
 155     ULONG ref = InterlockedDecrement(&This->ref);
 156
 157     TRACE("(%p) decreasing refcount to %lu\n", iface, ref);
 158
 159     if (ref == 0)
 160         HeapFree(GetProcessHeap(), 0, This);
 161
 162     return ref;
 163 }
 164
 165 static HRESULT WINAPI CommonQuery_OpenQueryWindow(ICommonQuery *iface,
 166         HWND parent, LPOPENQUERYWINDOW query_window, IDataObject **data_object)
 167 {
 168     FIXME("(%p)->(%p, %p, %p) stub!\n", iface, parent, query_window, data_object);
 169
 170     return E_NOTIMPL;
 171 }
 172
 173 static const ICommonQueryVtbl CommonQuery_vtbl =
 174 {
 175     CommonQuery_QueryInterface,
 176     CommonQuery_AddRef,
 177     CommonQuery_Release,
 178     CommonQuery_OpenQueryWindow,
 179 };
 180
 181 static HRESULT CommonQuery_create(IUnknown *outer, REFIID riid, void **out)
 182 {
 183     struct common_query *query;
 184
 185     TRACE("outer %p, riid %s, out %p\n", outer, debugstr_guid(riid), out);
 186
 187     if (outer)
 188         return CLASS_E_NOAGGREGATION;
 189
 190     if (!(query = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*query))))
 191         return E_OUTOFMEMORY;
 192
 193     query->ICommonQuery_iface.lpVtbl = &CommonQuery_vtbl;
 194     return ICommonQuery_QueryInterface(&query->ICommonQuery_iface, riid, out);
 195 }
 196
 197 /***********************************************************************
 198  *              DllGetClassObject (DSQUERY.@)
 199  */
 200 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **out)
 201 {
 202     TRACE("rclsid %s, riid %s, out %p\n", debugstr_guid(rclsid), debugstr_guid(riid), out);
 203
 204     if (!IsEqualGUID( &IID_IClassFactory, riid)
 205         && !IsEqualGUID( &IID_IUnknown, riid))
 206     {
 207         FIXME("interface %s not implemented\n", debugstr_guid(riid));
 208         *out = NULL;
 209         return E_NOINTERFACE;
 210     }
 211
 212     if (IsEqualGUID(&CLSID_CommonQuery, rclsid))
 213     {
 214         struct query_class_factory *factory = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*factory));
 215         if (!factory) return E_OUTOFMEMORY;
 216
 217         factory->IClassFactory_iface.lpVtbl = &query_class_factory_vtbl;
 218         factory->ref = 1;
 219         factory->pfnCreateInstance = CommonQuery_create;
 220         *out = factory;
 221         return S_OK;
 222     }
 223
 224     FIXME("%s: no class found\n", debugstr_guid(rclsid));
 225     *out = NULL;
 226     return CLASS_E_CLASSNOTAVAILABLE;
 227 }