git.oblomov.eu Git - wine/blob - dlls/urlmon/format.c

   1 /*
   2  * Copyright 2005 Jacek Caban
   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 "urlmon_main.h"
  20 #include "wine/debug.h"
  21
  22 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
  23
  24 static WCHAR wszEnumFORMATETC[] = {'_','E','n','u','m','F','O','R','M','A','T','E','T','C','_',0};
  25
  26 typedef struct {
  27     const IEnumFORMATETCVtbl *lpEnumFORMATETCVtbl;
  28
  29     FORMATETC *fetc;
  30     UINT fetc_cnt;
  31     UINT it;
  32
  33     LONG ref;
  34 } EnumFORMATETC;
  35
  36 static IEnumFORMATETC *EnumFORMATETC_Create(UINT cfmtetc, const FORMATETC *rgfmtetc, UINT it);
  37
  38 #define ENUMF_THIS(iface) ICOM_THIS_MULTI(EnumFORMATETC, lpEnumFORMATETCVtbl, iface)
  39
  40 static HRESULT WINAPI EnumFORMATETC_QueryInterface(IEnumFORMATETC *iface, REFIID riid, void **ppv)
  41 {
  42     TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
  43
  44     *ppv = NULL;
  45
  46     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IEnumFORMATETC, riid)) {
  47         *ppv = iface;
  48         return S_OK;
  49     }
  50
  51     WARN("not supported interface %s\n", debugstr_guid(riid));
  52     return E_NOINTERFACE;
  53 }
  54
  55 static ULONG WINAPI EnumFORMATETC_AddRef(IEnumFORMATETC *iface)
  56 {
  57     ENUMF_THIS(iface);
  58     LONG ref = InterlockedIncrement(&This->ref);
  59     TRACE("(%p) ref=%d\n", This, ref);
  60     return ref;
  61 }
  62
  63 static ULONG WINAPI EnumFORMATETC_Release(IEnumFORMATETC *iface)
  64 {
  65     ENUMF_THIS(iface);
  66     LONG ref = InterlockedDecrement(&This->ref);
  67
  68     TRACE("(%p) ref=%d\n", This, ref);
  69
  70     if(!ref) {
  71         heap_free(This->fetc);
  72         heap_free(This);
  73
  74         URLMON_UnlockModule();
  75     }
  76
  77     return ref;
  78 }
  79
  80 static HRESULT WINAPI EnumFORMATETC_Next(IEnumFORMATETC *iface, ULONG celt,
  81         FORMATETC *rgelt, ULONG *pceltFetched)
  82 {
  83     ENUMF_THIS(iface);
  84     ULONG cnt;
  85
  86     TRACE("(%p)->(%d %p %p)\n", This, celt, rgelt, pceltFetched);
  87
  88     if(!rgelt)
  89         return E_INVALIDARG;
  90
  91     if(This->it >= This->fetc_cnt || !celt) {
  92         if(pceltFetched)
  93             *pceltFetched = 0;
  94         return celt ? S_FALSE : S_OK;
  95     }
  96
  97     cnt = This->fetc_cnt-This->it > celt ? celt : This->fetc_cnt-This->it;
  98
  99     memcpy(rgelt, This->fetc+This->it, cnt*sizeof(FORMATETC));
 100     This->it += cnt;
 101
 102     if(pceltFetched)
 103         *pceltFetched = cnt;
 104
 105     return cnt == celt ? S_OK : S_FALSE;
 106 }
 107
 108 static HRESULT WINAPI EnumFORMATETC_Skip(IEnumFORMATETC *iface, ULONG celt)
 109 {
 110     ENUMF_THIS(iface);
 111
 112     TRACE("(%p)->(%d)\n", This, celt);
 113
 114     This->it += celt;
 115     return This->it > This->fetc_cnt ? S_FALSE : S_OK;
 116 }
 117
 118 static HRESULT WINAPI EnumFORMATETC_Reset(IEnumFORMATETC *iface)
 119 {
 120     ENUMF_THIS(iface);
 121
 122     TRACE("(%p)\n", This);
 123
 124     This->it = 0;
 125     return S_OK;
 126 }
 127
 128 static HRESULT WINAPI EnumFORMATETC_Clone(IEnumFORMATETC *iface, IEnumFORMATETC **ppenum)
 129 {
 130     ENUMF_THIS(iface);
 131
 132     TRACE("(%p)->(%p)\n", This, ppenum);
 133
 134     if(!ppenum)
 135         return E_INVALIDARG;
 136
 137     *ppenum = EnumFORMATETC_Create(This->fetc_cnt, This->fetc, This->it);
 138     return S_OK;
 139 }
 140
 141 static const IEnumFORMATETCVtbl EnumFORMATETCVtbl = {
 142     EnumFORMATETC_QueryInterface,
 143     EnumFORMATETC_AddRef,
 144     EnumFORMATETC_Release,
 145     EnumFORMATETC_Next,
 146     EnumFORMATETC_Skip,
 147     EnumFORMATETC_Reset,
 148     EnumFORMATETC_Clone
 149 };
 150
 151 static IEnumFORMATETC *EnumFORMATETC_Create(UINT cfmtetc, const FORMATETC *rgfmtetc, UINT it)
 152 {
 153     EnumFORMATETC *ret = heap_alloc(sizeof(EnumFORMATETC));
 154
 155     URLMON_LockModule();
 156
 157     ret->lpEnumFORMATETCVtbl = &EnumFORMATETCVtbl;
 158     ret->ref = 1;
 159     ret->it = it;
 160     ret->fetc_cnt = cfmtetc;
 161
 162     ret->fetc = heap_alloc(cfmtetc*sizeof(FORMATETC));
 163     memcpy(ret->fetc, rgfmtetc, cfmtetc*sizeof(FORMATETC));
 164
 165     return (IEnumFORMATETC*)ret;
 166 }
 167
 168 /**********************************************************
 169  *      CreateFormatEnumerator (urlmon.@)
 170  */
 171 HRESULT WINAPI CreateFormatEnumerator(UINT cfmtetc, FORMATETC *rgfmtetc,
 172         IEnumFORMATETC** ppenumfmtetc)
 173 {
 174     TRACE("(%d %p %p)\n", cfmtetc, rgfmtetc, ppenumfmtetc);
 175
 176     if(!ppenumfmtetc)
 177         return E_INVALIDARG;
 178     if(!cfmtetc)
 179         return E_FAIL;
 180
 181     *ppenumfmtetc = EnumFORMATETC_Create(cfmtetc, rgfmtetc, 0);
 182     return S_OK;
 183 }
 184
 185 /**********************************************************
 186  *      RegisterFormatEnumerator (urlmon.@)
 187  */
 188 HRESULT WINAPI RegisterFormatEnumerator(LPBC pBC, IEnumFORMATETC *pEFetc, DWORD reserved)
 189 {
 190     TRACE("(%p %p %d)\n", pBC, pEFetc, reserved);
 191
 192     if(reserved)
 193         WARN("reserved != 0\n");
 194
 195     if(!pBC || !pEFetc)
 196         return E_INVALIDARG;
 197
 198     return IBindCtx_RegisterObjectParam(pBC, wszEnumFORMATETC, (IUnknown*)pEFetc);
 199 }
 200
 201 /**********************************************************
 202  *      RevokeFormatEnumerator (urlmon.@)
 203  */
 204 HRESULT WINAPI RevokeFormatEnumerator(LPBC pbc, IEnumFORMATETC *pEFetc)
 205 {
 206     TRACE("(%p %p)\n", pbc, pEFetc);
 207
 208     if(!pbc)
 209         return E_INVALIDARG;
 210
 211     return IBindCtx_RevokeObjectParam(pbc, wszEnumFORMATETC);
 212 }