Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194/* * Copyright 2017 Piotr Caban for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */
#include "vbscript.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
typedef struct { IEnumVARIANT IEnumVARIANT_iface;
LONG ref;
SAFEARRAY *sa; ULONG i, size;} safearray_iter;
static inline safearray_iter *impl_from_IEnumVARIANT(IEnumVARIANT *iface){ return CONTAINING_RECORD(iface, safearray_iter, IEnumVARIANT_iface);}
static HRESULT WINAPI safearray_iter_IEnumVARIANT_QueryInterface( IEnumVARIANT *iface, REFIID riid, void **ppv){ safearray_iter *This = impl_from_IEnumVARIANT(iface);
if(IsEqualGUID(riid, &IID_IUnknown)) { TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv); *ppv = &This->IEnumVARIANT_iface; }else if(IsEqualGUID(riid, &IID_IEnumVARIANT)) { TRACE("(%p)->(IID_IEnumVARIANT %p)\n", This, ppv); *ppv = &This->IEnumVARIANT_iface; }else { FIXME("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv); *ppv = NULL; return E_NOINTERFACE; }
IUnknown_AddRef((IUnknown*)*ppv); return S_OK;}
static ULONG WINAPI safearray_iter_IEnumVARIANT_AddRef(IEnumVARIANT *iface){ safearray_iter *This = impl_from_IEnumVARIANT(iface); LONG ref = InterlockedIncrement(&This->ref);
TRACE("(%p) ref=%d\n", This, ref);
return ref;}
static ULONG WINAPI safearray_iter_IEnumVARIANT_Release(IEnumVARIANT *iface){ safearray_iter *This = impl_from_IEnumVARIANT(iface); LONG ref = InterlockedDecrement(&This->ref);
TRACE("(%p) ref=%d\n", iface, ref);
if(!ref) { if(This->sa) SafeArrayUnlock(This->sa); heap_free(This); }
return ref;}
static HRESULT WINAPI safearray_iter_IEnumVARIANT_Next(IEnumVARIANT *iface, ULONG celt, VARIANT *rgVar, ULONG *pCeltFetched){ safearray_iter *This = impl_from_IEnumVARIANT(iface); HRESULT hres; VARIANT *v;
TRACE("(%p)->(%u %p %p)\n", This, celt, rgVar, pCeltFetched);
if(celt != 1) { FIXME("celt != 1\n"); return E_NOTIMPL; }
if(This->i >= This->size) { if(pCeltFetched) *pCeltFetched = 0; return S_FALSE; }
if(!This->sa->cLocks) ERR("SAFEARRAY not locked\n");
v = (VARIANT*)(((BYTE*)This->sa->pvData) + This->i * This->sa->cbElements); V_VT(rgVar) = VT_EMPTY; hres = VariantCopy(rgVar, v); if(FAILED(hres)) return hres;
This->i++; if(pCeltFetched) *pCeltFetched = 1; return S_OK;}
static HRESULT WINAPI safearray_iter_IEnumVARIANT_Skip(IEnumVARIANT *iface, ULONG celt){ FIXME("\n"); return E_NOTIMPL;}
static HRESULT WINAPI safearray_iter_IEnumVARIANT_Reset(IEnumVARIANT *iface){ FIXME("\n"); return E_NOTIMPL;}
static HRESULT WINAPI safearray_iter_IEnumVARIANT_Clone( IEnumVARIANT *iface, IEnumVARIANT **ppEnum){ FIXME("\n"); return E_NOTIMPL;}
static const IEnumVARIANTVtbl safearray_iter_EnumVARIANTVtbl = { safearray_iter_IEnumVARIANT_QueryInterface, safearray_iter_IEnumVARIANT_AddRef, safearray_iter_IEnumVARIANT_Release, safearray_iter_IEnumVARIANT_Next, safearray_iter_IEnumVARIANT_Skip, safearray_iter_IEnumVARIANT_Reset, safearray_iter_IEnumVARIANT_Clone};
static ULONG get_safearray_size(SAFEARRAY *sa){ ULONG ret = 1; USHORT i;
if(!sa) return 0;
for(i=0; i<sa->cDims && ret; i++) ret *= sa->rgsabound[i].cElements; return ret;}
HRESULT create_safearray_iter(SAFEARRAY *sa, IEnumVARIANT **ev){ safearray_iter *iter; HRESULT hres;
if(sa && !(sa->fFeatures & FADF_VARIANT)) { FIXME("enumeration not supported: %x\n", sa->fFeatures); return E_NOTIMPL; }
iter = heap_alloc(sizeof(*iter)); if(!iter) return E_OUTOFMEMORY;
if(sa) { hres = SafeArrayLock(sa); if(FAILED(hres)) { heap_free(iter); return hres; } }
iter->IEnumVARIANT_iface.lpVtbl = &safearray_iter_EnumVARIANTVtbl; iter->ref = 1; iter->sa = sa; iter->i = 0; iter->size = get_safearray_size(sa);
*ev = &iter->IEnumVARIANT_iface; return S_OK;}