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