dlls/diasymreader/main.c

   1 /* Copyright (C) 2022 Esme Povirk
   2  *
   3  * This program is free software; you can redistribute it and/or
   4  * modify it under the terms of the GNU Lesser General Public
   5  * License as published by the Free Software Foundation; either
   6  * version 2.1 of the License, or (at your option) any later version.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  * Lesser General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU Lesser General Public
  14  * License along with this program; if not, write to the Free Software
  15  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  16  */
  17
  18 #include <stdarg.h>
  19
  20 #define COBJMACROS
  21
  22 #include "windef.h"
  23 #include "winbase.h"
  24 #include "winnls.h"
  25 #include "objbase.h"
  26 #include "ocidl.h"
  27 #include "rpcproxy.h"
  28
  29 #include "wine/debug.h"
  30
  31 #include "initguid.h"
  32
  33 #include "diasymreader_private.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(diasymreader);
  36
  37 typedef struct {
  38     REFCLSID classid;
  39     HRESULT (*constructor)(REFIID,void**);
  40 } classinfo;
  41
  42 static const classinfo classes[] = {
  43     {&CLSID_CorSymWriter_SxS, SymWriter_CreateInstance}
  44 };
  45
  46 typedef struct {
  47     IClassFactory           IClassFactory_iface;
  48     LONG                    ref;
  49     const classinfo         *info;
  50 } ClassFactoryImpl;
  51
  52 static inline ClassFactoryImpl *impl_from_IClassFactory(IClassFactory *iface)
  53 {
  54     return CONTAINING_RECORD(iface, ClassFactoryImpl, IClassFactory_iface);
  55 }
  56
  57 static HRESULT WINAPI ClassFactoryImpl_QueryInterface(IClassFactory *iface,
  58     REFIID iid, void **ppv)
  59 {
  60     ClassFactoryImpl *This = impl_from_IClassFactory(iface);
  61     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
  62
  63     if (!ppv) return E_INVALIDARG;
  64
  65     if (IsEqualIID(&IID_IUnknown, iid) ||
  66         IsEqualIID(&IID_IClassFactory, iid))
  67     {
  68         *ppv = &This->IClassFactory_iface;
  69     }
  70     else
  71     {
  72         *ppv = NULL;
  73         return E_NOINTERFACE;
  74     }
  75
  76     IUnknown_AddRef((IUnknown*)*ppv);
  77     return S_OK;
  78 }
  79
  80 static ULONG WINAPI ClassFactoryImpl_AddRef(IClassFactory *iface)
  81 {
  82     ClassFactoryImpl *This = impl_from_IClassFactory(iface);
  83     ULONG ref = InterlockedIncrement(&This->ref);
  84
  85     TRACE("(%p) refcount=%lu\n", iface, ref);
  86
  87     return ref;
  88 }
  89
  90 static ULONG WINAPI ClassFactoryImpl_Release(IClassFactory *iface)
  91 {
  92     ClassFactoryImpl *This = impl_from_IClassFactory(iface);
  93     ULONG ref = InterlockedDecrement(&This->ref);
  94
  95     TRACE("(%p) refcount=%lu\n", iface, ref);
  96
  97     if (ref == 0)
  98         free(This);
  99
 100     return ref;
 101 }
 102
 103 static HRESULT WINAPI ClassFactoryImpl_CreateInstance(IClassFactory *iface,
 104     IUnknown *pUnkOuter, REFIID riid, void **ppv)
 105 {
 106     ClassFactoryImpl *This = impl_from_IClassFactory(iface);
 107
 108     *ppv = NULL;
 109
 110     if (pUnkOuter) return CLASS_E_NOAGGREGATION;
 111
 112     return This->info->constructor(riid, ppv);
 113 }
 114
 115 static HRESULT WINAPI ClassFactoryImpl_LockServer(IClassFactory *iface, BOOL lock)
 116 {
 117     TRACE("(%p, %i): stub\n", iface, lock);
 118     return E_NOTIMPL;
 119 }
 120
 121 static const IClassFactoryVtbl ClassFactoryImpl_Vtbl = {
 122     ClassFactoryImpl_QueryInterface,
 123     ClassFactoryImpl_AddRef,
 124     ClassFactoryImpl_Release,
 125     ClassFactoryImpl_CreateInstance,
 126     ClassFactoryImpl_LockServer
 127 };
 128
 129 static HRESULT ClassFactoryImpl_Constructor(const classinfo *info, REFIID riid, LPVOID *ppv)
 130 {
 131     ClassFactoryImpl *This;
 132     HRESULT ret;
 133
 134     *ppv = NULL;
 135
 136     This = malloc(sizeof(ClassFactoryImpl));
 137     if (!This) return E_OUTOFMEMORY;
 138
 139     This->IClassFactory_iface.lpVtbl = &ClassFactoryImpl_Vtbl;
 140     This->ref = 1;
 141     This->info = info;
 142
 143     ret = IClassFactory_QueryInterface(&This->IClassFactory_iface, riid, ppv);
 144     IClassFactory_Release(&This->IClassFactory_iface);
 145
 146     return ret;
 147 }
 148
 149 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID iid, LPVOID *ppv)
 150 {
 151     HRESULT ret;
 152     const classinfo *info=NULL;
 153     int i;
 154
 155     TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(iid), ppv);
 156
 157     if (!rclsid || !iid || !ppv)
 158         return E_INVALIDARG;
 159
 160     *ppv = NULL;
 161
 162     for (i=0; classes[i].classid; i++)
 163     {
 164         if (IsEqualCLSID(classes[i].classid, rclsid))
 165         {
 166             info = &classes[i];
 167             break;
 168         }
 169     }
 170
 171     if (info)
 172         ret = ClassFactoryImpl_Constructor(info, iid, ppv);
 173     else
 174         ret = CLASS_E_CLASSNOTAVAILABLE;
 175
 176     TRACE("<-- %08lx\n", ret);
 177     return ret;
 178 }