[VBSCRIPT] Sync with Wine 3.0. CORE-14225

2025-08-05 18:33:10 +00:00 · 2018-01-20 13:39:40 +01:00 · 2018-01-20 13:39:40 +01:00 · e0b80a0929
commit e0b80a0929
parent eb91a73902
11 changed files with 978 additions and 699 deletions
--- a/dll/win32/vbscript/utils.c
+++ b/dll/win32/vbscript/utils.c
@ -0,0 +1,189 @@
+/*
+ * 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"
+
+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;
+}