[PATCH v9 1/1] activeds: Implement ADsBuildVarArrayInt.

[Daniel Martin]

From: Daniel Martin

---
 dlls/activeds/activeds_main.c  | 32 +++++++++++++++++++++--
 dlls/activeds/tests/activeds.c | 48 ++++++++++++++++++++++++++++++++++
 include/adshlp.h               |  1 +
 3 files changed, 79 insertions(+), 2 deletions(-)

diff --git a/dlls/activeds/activeds_main.c b/dlls/activeds/activeds_main.c
index 0f13a5629b6..ee0ea80088b 100644
--- a/dlls/activeds/activeds_main.c
+++ b/dlls/activeds/activeds_main.c
@@ -128,8 +128,36 @@ fail:
  */
 HRESULT WINAPI ADsBuildVarArrayInt(LPDWORD lpdwObjectTypes, DWORD dwObjectTypes, VARIANT* pvar)
 {
-    FIXME("(%p, %ld, %p)!stub\n",lpdwObjectTypes, dwObjectTypes, pvar);
-    return E_NOTIMPL;
+    HRESULT hr;
+    SAFEARRAY *sa;
+    LONG idx, end = dwObjectTypes;
+
+    TRACE("(%p, %lu, %p)\n", lpdwObjectTypes, dwObjectTypes, pvar);
+
+    if (!pvar) return E_ADS_BAD_PARAMETER;
+
+    sa = SafeArrayCreateVector(VT_VARIANT, 0, dwObjectTypes);
+    if (!sa) return E_OUTOFMEMORY;
+
+    VariantInit(pvar);
+    for (idx = 0; idx < end; idx++)
+    {
+        VARIANT item;
+
+        V_VT(&item) = VT_UI4;
+        V_UI4(&item) = lpdwObjectTypes[idx];
+
+        hr = SafeArrayPutElement(sa, &idx, &item);
+        if (hr != S_OK) 
+        {
+            SafeArrayDestroy(sa);
+            return hr;
+        }
+    }
+
+    V_VT(var) = VT_ARRAY | VT_VARIANT;
+    V_ARRAY(pvar) = sa;
+    return S_OK;
 }

/*****************************************************
diff --git a/dlls/activeds/tests/activeds.c b/dlls/activeds/tests/activeds.c
index bc6a6bfc568..e8b8d0152ee 100644
--- a/dlls/activeds/tests/activeds.c
+++ b/dlls/activeds/tests/activeds.c
@@ -84,6 +84,53 @@ static void test_ADsBuildVarArrayStr(void)
     VariantClear(&var);
 }

+static void test_ADsBuildVarArrayInt(void)
+{
+    DWORD props[] = { 1, 2, 3, 4 };
+    HRESULT hr;
+    VARIANT var, item;
+    LONG start, end, idx;
+
+    hr = ADsBuildVarArrayInt(NULL, 0, NULL);
+    ok(hr == E_ADS_BAD_PARAMETER || hr == E_FAIL /* XP */, "got %#lx\n", hr);
+
+    hr = ADsBuildVarArrayInt(NULL, 0, &var);
+    ok(hr == S_OK, "got %#lx\n", hr);
+    ok(V_VT(&var) == (VT_ARRAY | VT_VARIANT), "got %d\n", V_VT(&var));
+    start = 0xdeadbeef;
+    hr = SafeArrayGetLBound(V_ARRAY(&var), 1, &start);
+    ok(hr == S_OK, "got %#lx\n", hr);
+    ok(start == 0, "got %ld\n", start);
+    end = 0xdeadbeef;
+    hr = SafeArrayGetUBound(V_ARRAY(&var), 1, &end);
+    ok(hr == S_OK, "got %#lx\n", hr);
+    ok(end == -1, "got %ld\n", end);
+    VariantClear(&var);
+
+    hr = ADsBuildVarArrayInt(props, ARRAY_SIZE(props), &var);
+    ok(hr == S_OK, "got %#lx\n", hr);
+    ok(V_VT(&var) == (VT_ARRAY | VT_VARIANT), "got %d\n", V_VT(&var));
+    start = 0xdeadbeef;
+    hr = SafeArrayGetLBound(V_ARRAY(&var), 1, &start);
+    ok(hr == S_OK, "got %#lx\n", hr);
+    ok(start == 0, "got %ld\n", start);
+    end = 0xdeadbeef;
+    hr = SafeArrayGetUBound(V_ARRAY(&var), 1, &end);
+    ok(hr == S_OK, "got %#lx\n", hr);
+    ok(end == 3, "got %ld\n", end);
+
+    for (idx = 0; idx < ARRAY_SIZE(props); idx++)
+    {
+        hr = SafeArrayGetElement(V_ARRAY(&var), &idx, &item);
+        ok(hr == S_OK, "got %#lx\n", hr);
+        ok(V_VT(&item) == VT_UI4, "got %d\n", V_VT(&item));
+        ok(V_UI4(&item) == props[idx], "got %lu\n", V_UI4(&item));
+        VariantClear(&item);
+    }
+
+    VariantClear(&var);
+}
+
 static void test_Pathname(void)
 {
     static const WCHAR * const elem[3] = { L"a=b",L"c=d",L"e=f" };
@@ -167,6 +214,7 @@ START_TEST(activeds)

test_Pathname();
     test_ADsBuildVarArrayStr();
+    test_ADsBuildVarArrayInt();

CoUninitialize();
 }
diff --git a/include/adshlp.h b/include/adshlp.h
index 7366c077788..4e85c0efb5c 100644
--- a/include/adshlp.h
+++ b/include/adshlp.h
@@ -31,6 +31,7 @@ WCHAR*  WINAPI AllocADsStr(WCHAR*) __WINE_DEALLOC(FreeADsStr) __WINE_MALLOC;

HRESULT WINAPI ADsBuildEnumerator(IADsContainer*,IEnumVARIANT**);
 HRESULT WINAPI ADsBuildVarArrayStr(LPWSTR*,DWORD,VARIANT*);
+HRESULT WINAPI ADsBuildVarArrayInt(LPDWORD,DWORD,VARIANT*);
 HRESULT WINAPI ADsEnumerateNext(IEnumVARIANT*,ULONG,VARIANT*,ULONG*);
 HRESULT WINAPI ADsGetObject(LPCWSTR,REFIID,VOID**);
 HRESULT WINAPI ADsOpenObject(LPCWSTR,LPCWSTR,LPCWSTR,DWORD,REFIID,VOID**);