dlls/dsuiext/dsuiext.c

   1 /*
   2  * Copyright 2020 Dmitry Timoshkov
   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 "rpcproxy.h"
  27 #include "initguid.h"
  28 #include "iads.h"
  29 #include "dsclient.h"
  30
  31 #include "wine/heap.h"
  32 #include "wine/debug.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(dsuiext);
  35
  36 typedef struct
  37 {
  38     IDsDisplaySpecifier IDsDisplaySpecifier_iface;
  39     LONG ref;
  40 } DisplaySpec;
  41
  42 static inline DisplaySpec *impl_from_IDsDisplaySpecifier(IDsDisplaySpecifier *iface)
  43 {
  44     return CONTAINING_RECORD(iface, DisplaySpec, IDsDisplaySpecifier_iface);
  45 }
  46
  47 static HRESULT WINAPI dispspec_QueryInterface(IDsDisplaySpecifier *iface, REFIID iid, void **obj)
  48 {
  49     TRACE("%p,%s,%p\n", iface, debugstr_guid(iid), obj);
  50
  51     if (!iid || !obj) return E_INVALIDARG;
  52
  53     if (IsEqualGUID(iid, &IID_IUnknown) ||
  54         IsEqualGUID(iid, &IID_IDsDisplaySpecifier))
  55     {
  56         iface->lpVtbl->AddRef(iface);
  57         *obj = iface;
  58         return S_OK;
  59     }
  60
  61     FIXME("interface %s is not implemented\n", debugstr_guid(iid));
  62     return E_NOINTERFACE;
  63 }
  64
  65 static ULONG WINAPI dispspec_AddRef(IDsDisplaySpecifier *iface)
  66 {
  67     DisplaySpec *dispspec = impl_from_IDsDisplaySpecifier(iface);
  68     return InterlockedIncrement(&dispspec->ref);
  69 }
  70
  71 static ULONG WINAPI dispspec_Release(IDsDisplaySpecifier *iface)
  72 {
  73     DisplaySpec *dispspec = impl_from_IDsDisplaySpecifier(iface);
  74     LONG ref = InterlockedDecrement(&dispspec->ref);
  75
  76     if (!ref)
  77     {
  78         TRACE("destroying %p\n", iface);
  79         heap_free(dispspec);
  80     }
  81
  82     return ref;
  83 }
  84
  85 static HRESULT WINAPI dispspec_SetServer(IDsDisplaySpecifier *iface, LPCWSTR server, LPCWSTR user,
  86                                          LPCWSTR password, DWORD flags)
  87 {
  88     FIXME("%p,%s,%s,%p,%08x: stub\n", iface, debugstr_w(server), debugstr_w(user), password, flags);
  89     return E_NOTIMPL;
  90 }
  91
  92 static HRESULT WINAPI dispspec_SetLanguageID(IDsDisplaySpecifier *iface, LANGID lang)
  93 {
  94     FIXME("%p,%08x: stub\n", iface, lang);
  95     return E_NOTIMPL;
  96 }
  97
  98 static HRESULT WINAPI dispspec_GetDisplaySpecifier(IDsDisplaySpecifier *iface, LPCWSTR object,
  99                                                    REFIID iid, void **obj)
 100 {
 101     FIXME("%p,%s,%s,%p: stub\n", iface, debugstr_w(object), debugstr_guid(iid), obj);
 102     return E_NOTIMPL;
 103 }
 104
 105 static HRESULT WINAPI dispspec_GetIconLocation(IDsDisplaySpecifier *iface, LPCWSTR object,
 106                                                DWORD flags, LPWSTR buffer, INT size, INT *id)
 107 {
 108     FIXME("%p,%s,%08x,%p,%d,%p: stub\n", iface, debugstr_w(object), flags, buffer, size, id);
 109     return E_NOTIMPL;
 110 }
 111
 112 static HICON WINAPI dispspec_GetIcon(IDsDisplaySpecifier *iface, LPCWSTR object,
 113                                      DWORD flags, INT cx, INT cy)
 114 {
 115     FIXME("%p,%s,%08x,%dx%d: stub\n", iface, debugstr_w(object), flags, cx, cy);
 116     return 0;
 117 }
 118
 119 static HRESULT WINAPI dispspec_GetFriendlyClassName(IDsDisplaySpecifier *iface, LPCWSTR object,
 120                                                     LPWSTR buffer, INT size)
 121 {
 122     FIXME("%p,%s,%p,%d: stub\n", iface, debugstr_w(object), buffer, size);
 123     return E_NOTIMPL;
 124 }
 125
 126 static HRESULT WINAPI dispspec_GetFriendlyAttributeName(IDsDisplaySpecifier *iface, LPCWSTR object,
 127                                                         LPCWSTR name, LPWSTR buffer, UINT size)
 128 {
 129     FIXME("%p,%s,%s,%p,%d: stub\n", iface, debugstr_w(object), debugstr_w(name), buffer, size);
 130     return E_NOTIMPL;
 131 }
 132
 133 static BOOL WINAPI dispspec_IsClassContainer(IDsDisplaySpecifier *iface, LPCWSTR object,
 134                                              LPCWSTR path, DWORD flags)
 135 {
 136     FIXME("%p,%s,%s,%08x: stub\n", iface, debugstr_w(object), debugstr_w(path), flags);
 137     return FALSE;
 138 }
 139
 140 static HRESULT WINAPI dispspec_GetClassCreationInfo(IDsDisplaySpecifier *iface, LPCWSTR object,
 141                                                     LPDSCLASSCREATIONINFO *info)
 142 {
 143     FIXME("%p,%s,%p: stub\n", iface, debugstr_w(object), info);
 144     return E_NOTIMPL;
 145 }
 146
 147 static HRESULT WINAPI dispspec_EnumClassAttributes(IDsDisplaySpecifier *iface, LPCWSTR object,
 148                                                    LPDSENUMATTRIBUTES cb, LPARAM param)
 149 {
 150     FIXME("%p,%s,%p,%08lx: stub\n", iface, debugstr_w(object), cb, param);
 151     return E_NOTIMPL;
 152 }
 153
 154 static ADSTYPE WINAPI dispspec_GetAttributeADsType(IDsDisplaySpecifier *iface, LPCWSTR name)
 155 {
 156     FIXME("%p,%s: stub\n", iface, debugstr_w(name));
 157     return ADSTYPE_INVALID;
 158 }
 159
 160 static const IDsDisplaySpecifierVtbl IDsDisplaySpecifier_vtbl =
 161 {
 162     dispspec_QueryInterface,
 163     dispspec_AddRef,
 164     dispspec_Release,
 165     dispspec_SetServer,
 166     dispspec_SetLanguageID,
 167     dispspec_GetDisplaySpecifier,
 168     dispspec_GetIconLocation,
 169     dispspec_GetIcon,
 170     dispspec_GetFriendlyClassName,
 171     dispspec_GetFriendlyAttributeName,
 172     dispspec_IsClassContainer,
 173     dispspec_GetClassCreationInfo,
 174     dispspec_EnumClassAttributes,
 175     dispspec_GetAttributeADsType
 176 };
 177
 178 static HRESULT DsDisplaySpecifier_create(REFIID iid, void **obj)
 179 {
 180     DisplaySpec *dispspec;
 181     HRESULT hr;
 182
 183     dispspec = heap_alloc(sizeof(*dispspec));
 184     if (!dispspec) return E_OUTOFMEMORY;
 185
 186     dispspec->IDsDisplaySpecifier_iface.lpVtbl = &IDsDisplaySpecifier_vtbl;
 187     dispspec->ref = 1;
 188
 189     hr = dispspec->IDsDisplaySpecifier_iface.lpVtbl->QueryInterface(&dispspec->IDsDisplaySpecifier_iface, iid, obj);
 190     dispspec->IDsDisplaySpecifier_iface.lpVtbl->Release(&dispspec->IDsDisplaySpecifier_iface);
 191
 192     return hr;
 193 }
 194
 195 static const struct class_info
 196 {
 197     const CLSID *clsid;
 198     HRESULT (*constructor)(REFIID, void **);
 199 } class_info[] =
 200 {
 201     { &CLSID_DsDisplaySpecifier, DsDisplaySpecifier_create }
 202 };
 203
 204 typedef struct
 205 {
 206     IClassFactory IClassFactory_iface;
 207     LONG ref;
 208     const struct class_info *info;
 209 } class_factory;
 210
 211 static inline class_factory *impl_from_IClassFactory(IClassFactory *iface)
 212 {
 213     return CONTAINING_RECORD(iface, class_factory, IClassFactory_iface);
 214 }
 215
 216 static HRESULT WINAPI factory_QueryInterface(IClassFactory *iface, REFIID iid, LPVOID *obj)
 217 {
 218     TRACE("%p,%s,%p\n", iface, debugstr_guid(iid), obj);
 219
 220     if (!iid || !obj) return E_INVALIDARG;
 221
 222     if (IsEqualIID(iid, &IID_IUnknown) ||
 223         IsEqualIID(iid, &IID_IClassFactory))
 224     {
 225         IClassFactory_AddRef(iface);
 226         *obj = iface;
 227         return S_OK;
 228     }
 229
 230     *obj = NULL;
 231     FIXME("interface %s is not implemented\n", debugstr_guid(iid));
 232     return E_NOINTERFACE;
 233 }
 234
 235 static ULONG WINAPI factory_AddRef(IClassFactory *iface)
 236 {
 237     class_factory *factory = impl_from_IClassFactory(iface);
 238     ULONG ref = InterlockedIncrement(&factory->ref);
 239
 240     TRACE("(%p) ref %u\n", iface, ref);
 241
 242     return ref;
 243 }
 244
 245 static ULONG WINAPI factory_Release(IClassFactory *iface)
 246 {
 247     class_factory *factory = impl_from_IClassFactory(iface);
 248     ULONG ref = InterlockedDecrement(&factory->ref);
 249
 250     TRACE("(%p) ref %u\n", iface, ref);
 251
 252     if (!ref)
 253         heap_free(factory);
 254
 255     return ref;
 256 }
 257
 258 static HRESULT WINAPI factory_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID iid, void **obj)
 259 {
 260     class_factory *factory = impl_from_IClassFactory(iface);
 261
 262     TRACE("%p,%s,%p\n", outer, debugstr_guid(iid), obj);
 263
 264     if (!iid || !obj) return E_INVALIDARG;
 265
 266     *obj = NULL;
 267     if (outer) return CLASS_E_NOAGGREGATION;
 268
 269     return factory->info->constructor(iid, obj);
 270 }
 271
 272 static HRESULT WINAPI factory_LockServer(IClassFactory *iface, BOOL lock)
 273 {
 274     FIXME("%p,%d: stub\n", iface, lock);
 275     return S_OK;
 276 }
 277
 278 static const struct IClassFactoryVtbl factory_vtbl =
 279 {
 280     factory_QueryInterface,
 281     factory_AddRef,
 282     factory_Release,
 283     factory_CreateInstance,
 284     factory_LockServer
 285 };
 286
 287 static HRESULT factory_constructor(const struct class_info *info, REFIID riid, void **obj)
 288 {
 289     class_factory *factory;
 290     HRESULT hr;
 291
 292     factory = heap_alloc(sizeof(*factory));
 293     if (!factory) return E_OUTOFMEMORY;
 294
 295     factory->IClassFactory_iface.lpVtbl = &factory_vtbl;
 296     factory->ref = 1;
 297     factory->info = info;
 298
 299     hr = IClassFactory_QueryInterface(&factory->IClassFactory_iface, riid, obj);
 300     IClassFactory_Release(&factory->IClassFactory_iface);
 301
 302     return hr;
 303 }
 304
 305 HRESULT WINAPI DllGetClassObject(REFCLSID clsid, REFIID iid, LPVOID *obj)
 306 {
 307     int i;
 308
 309     TRACE("%s,%s,%p\n", debugstr_guid(clsid), debugstr_guid(iid), obj);
 310
 311     if (!clsid || !iid || !obj) return E_INVALIDARG;
 312
 313     *obj = NULL;
 314
 315     for (i = 0; i < ARRAY_SIZE(class_info); i++)
 316     {
 317         if (IsEqualCLSID(class_info[i].clsid, clsid))
 318             return factory_constructor(&class_info[i], iid, obj);
 319     }
 320
 321     FIXME("class %s/%s is not implemented\n", debugstr_guid(clsid), debugstr_guid(iid));
 322     return CLASS_E_CLASSNOTAVAILABLE;
 323 }