Commit 9f1c6bf6 authored by Robert Shearman's avatar Robert Shearman Committed by Alexandre Julliard

OLE: Fix SafeArrayCopy for NULL pvData.

It is allowed to copy a SAFEARRAY with a NULL pvData, as long as cbElements is non-zero. Add a test for this and fix the safe array code.
parent 6c3e1f9a
...@@ -355,7 +355,9 @@ static HRESULT SAFEARRAY_DestroyData(SAFEARRAY *psa, ULONG ulStartCell) ...@@ -355,7 +355,9 @@ static HRESULT SAFEARRAY_DestroyData(SAFEARRAY *psa, ULONG ulStartCell)
/* Copy data items from one array to another */ /* Copy data items from one array to another */
static HRESULT SAFEARRAY_CopyData(SAFEARRAY *psa, SAFEARRAY *dest) static HRESULT SAFEARRAY_CopyData(SAFEARRAY *psa, SAFEARRAY *dest)
{ {
if (!psa->pvData || !dest->pvData || psa->fFeatures & FADF_DATADELETED) if (!psa->pvData)
return S_OK;
else if (!dest->pvData || psa->fFeatures & FADF_DATADELETED)
return E_INVALIDARG; return E_INVALIDARG;
else else
{ {
...@@ -1378,6 +1380,12 @@ HRESULT WINAPI SafeArrayCopy(SAFEARRAY *psa, SAFEARRAY **ppsaOut) ...@@ -1378,6 +1380,12 @@ HRESULT WINAPI SafeArrayCopy(SAFEARRAY *psa, SAFEARRAY **ppsaOut)
if (!psa) if (!psa)
return S_OK; /* Handles copying of NULL arrays */ return S_OK; /* Handles copying of NULL arrays */
if (!psa->cbElements)
{
ERR("not copying an array of 0 elements\n");
return E_INVALIDARG;
}
if (psa->fFeatures & (FADF_RECORD|FADF_HAVEIID|FADF_HAVEVARTYPE)) if (psa->fFeatures & (FADF_RECORD|FADF_HAVEIID|FADF_HAVEVARTYPE))
{ {
VARTYPE vt; VARTYPE vt;
......
...@@ -1418,7 +1418,7 @@ static void test_SafeArrayClear(void) ...@@ -1418,7 +1418,7 @@ static void test_SafeArrayClear(void)
static void test_SafeArrayCopy(void) static void test_SafeArrayCopy(void)
{ {
SAFEARRAYBOUND sab; SAFEARRAYBOUND sab;
SAFEARRAY *sa; SAFEARRAY *sa, *sa2;
VARIANTARG vSrc, vDst; VARIANTARG vSrc, vDst;
HRESULT hres; HRESULT hres;
...@@ -1450,6 +1450,19 @@ static void test_SafeArrayCopy(void) ...@@ -1450,6 +1450,19 @@ static void test_SafeArrayCopy(void)
SafeArrayDestroy(V_ARRAY(&vSrc)); SafeArrayDestroy(V_ARRAY(&vSrc));
SafeArrayDestroy(V_ARRAY(&vDst)); SafeArrayDestroy(V_ARRAY(&vDst));
hres = SafeArrayAllocDescriptor(1, &sa);
ok(hres == S_OK, "SafeArrayAllocDescriptor failed with error 0x%08lx\n", hres);
hres = SafeArrayCopy(sa, &sa2);
ok(hres == E_INVALIDARG,
"SafeArrayCopy with empty array should have failed with error E_INVALIDARG instead of 0x%08lx\n",
hres);
sa->cbElements = 16;
hres = SafeArrayCopy(sa, &sa2);
ok(hres == S_OK, "SafeArrayCopy failed with error 0x%08lx\n", hres);
SafeArrayDestroy(sa);
} }
#define MKARRAY(low,num,typ) sab.lLbound = low; sab.cElements = num; \ #define MKARRAY(low,num,typ) sab.lLbound = low; sab.cElements = num; \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment