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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 #include <stdarg.h>
  20
  21 #define COBJMACROS
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winuser.h"
  26 #include "ole2.h"
  27 #include "urlmon.h"
  28 #include "urlmon_main.h"
  29
  30 #include "wine/debug.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
  33
  34 typedef struct {
  35     const IEnumFORMATETCVtbl *lpEnumFORMATETCVtbl;
  36
  37     FORMATETC *fetc;
  38     UINT fetc_cnt;
  39     UINT it;
  40
  41     LONG ref;
  42 } EnumFORMATETC;
  43
  44 static IEnumFORMATETC *EnumFORMATETC_Create(UINT cfmtetc, FORMATETC *rgfmtetc, UINT it);
  45
  46 #define ENUMF_THIS(iface) ICOM_THIS_MULTI(EnumFORMATETC, lpEnumFORMATETCVtbl, iface)
  47
  48 static HRESULT WINAPI EnumFORMATETC_QueryInterface(IEnumFORMATETC *iface, REFIID riid, void **ppv)
  49 {
  50     TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
  51
  52     *ppv = NULL;
  53
  54     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IEnumFORMATETC, riid)) {
  55         *ppv = iface;
  56         return S_OK;
  57     }
  58
  59     WARN("not supported interface %s\n", debugstr_guid(riid));
  60     return E_NOINTERFACE;
  61 }
  62
  63 static ULONG WINAPI EnumFORMATETC_AddRef(IEnumFORMATETC *iface)
  64 {
  65     ENUMF_THIS(iface);
  66     LONG ref = InterlockedIncrement(&This->ref);
  67     TRACE("(%p) ref=%ld\n", This, ref);
  68     return ref;
  69 }
  70
  71 static ULONG WINAPI EnumFORMATETC_Release(IEnumFORMATETC *iface)
  72 {
  73     ENUMF_THIS(iface);
  74     LONG ref = InterlockedDecrement(&This->ref);
  75
  76     TRACE("(%p) ref=%ld\n", This, ref);
  77
  78     if(!ref) {
  79         HeapFree(GetProcessHeap(), 0, This->fetc);
  80         HeapFree(GetProcessHeap(), 0, This);
  81
  82         URLMON_UnlockModule();
  83     }
  84
  85     return ref;
  86 }
  87
  88 static HRESULT WINAPI EnumFORMATETC_Next(IEnumFORMATETC *iface, ULONG celt,
  89         FORMATETC *rgelt, ULONG *pceltFetched)
  90 {
  91     ENUMF_THIS(iface);
  92     ULONG cnt;
  93
  94     TRACE("(%p)->(%ld %p %p)\n", This, celt, rgelt, pceltFetched);
  95
  96     if(!rgelt)
  97         return E_INVALIDARG;
  98
  99     if(This->it >= This->fetc_cnt || !celt) {
 100         if(pceltFetched)
 101             *pceltFetched = 0;
 102         return celt ? S_FALSE : S_OK;
 103     }
 104
 105     cnt = This->fetc_cnt-This->it > celt ? celt : This->fetc_cnt-This->it;
 106
 107     memcpy(rgelt, This->fetc+This->it, cnt*sizeof(FORMATETC));
 108     This->it += cnt;
 109
 110     if(pceltFetched)
 111         *pceltFetched = cnt;
 112
 113     return cnt == celt ? S_OK : S_FALSE;
 114 }
 115
 116 static HRESULT WINAPI EnumFORMATETC_Skip(IEnumFORMATETC *iface, ULONG celt)
 117 {
 118     ENUMF_THIS(iface);
 119
 120     TRACE("(%p)->(%ld)\n", This, celt);
 121
 122     This->it += celt;
 123     return This->it > This->fetc_cnt ? S_FALSE : S_OK;
 124 }
 125
 126 static HRESULT WINAPI EnumFORMATETC_Reset(IEnumFORMATETC *iface)
 127 {
 128     ENUMF_THIS(iface);
 129
 130     TRACE("(%p)\n", This);
 131
 132     This->it = 0;
 133     return S_OK;
 134 }
 135
 136 static HRESULT WINAPI EnumFORMATETC_Clone(IEnumFORMATETC *iface, IEnumFORMATETC **ppenum)
 137 {
 138     ENUMF_THIS(iface);
 139
 140     TRACE("(%p)->(%p)\n", This, ppenum);
 141
 142     if(!ppenum)
 143         return E_INVALIDARG;
 144
 145     *ppenum = EnumFORMATETC_Create(This->fetc_cnt, This->fetc, This->it);
 146     return S_OK;
 147 }
 148
 149 static const IEnumFORMATETCVtbl EnumFORMATETCVtbl = {
 150     EnumFORMATETC_QueryInterface,
 151     EnumFORMATETC_AddRef,
 152     EnumFORMATETC_Release,
 153     EnumFORMATETC_Next,
 154     EnumFORMATETC_Skip,
 155     EnumFORMATETC_Reset,
 156     EnumFORMATETC_Clone
 157 };
 158
 159 static IEnumFORMATETC *EnumFORMATETC_Create(UINT cfmtetc, FORMATETC *rgfmtetc, UINT it)
 160 {
 161     EnumFORMATETC *ret = HeapAlloc(GetProcessHeap(), 0, sizeof(EnumFORMATETC));
 162
 163     URLMON_LockModule();
 164
 165     ret->lpEnumFORMATETCVtbl = &EnumFORMATETCVtbl;
 166     ret->ref = 1;
 167     ret->it = it;
 168     ret->fetc_cnt = cfmtetc;
 169
 170     ret->fetc = HeapAlloc(GetProcessHeap(), 0, cfmtetc*sizeof(FORMATETC));
 171     memcpy(ret->fetc, rgfmtetc, cfmtetc*sizeof(FORMATETC));
 172
 173     return (IEnumFORMATETC*)ret;
 174 }
 175
 176 /**********************************************************
 177  *      CreateFormatEnumerator (urlmon.@)
 178  */
 179 HRESULT WINAPI CreateFormatEnumerator(UINT cfmtetc, FORMATETC *rgfmtetc,
 180         IEnumFORMATETC** ppenumfmtetc)
 181 {
 182     TRACE("(%d %p %p)\n", cfmtetc, rgfmtetc, ppenumfmtetc);
 183
 184     if(!ppenumfmtetc)
 185         return E_INVALIDARG;
 186     if(!cfmtetc)
 187         return E_FAIL;
 188
 189     *ppenumfmtetc = EnumFORMATETC_Create(cfmtetc, rgfmtetc, 0);
 190     return S_OK;
 191 }