urlmon: Check pointer before use.

This commit is contained in:
Alistair Leslie-Hughes 2008-06-18 22:14:30 +10:00 committed by Alexandre Julliard
parent 3bacdaf664
commit 031e1ae441
1 changed files with 8 additions and 6 deletions

View File

@ -294,7 +294,7 @@ static void create_file(void)
static void test_URLOpenBlockingStreamW(void) static void test_URLOpenBlockingStreamW(void)
{ {
HRESULT hr; HRESULT hr;
IStream *pStream; IStream *pStream = NULL;
char buffer[256]; char buffer[256];
hr = URLOpenBlockingStreamW(NULL, NULL, &pStream, 0, &BindStatusCallback); hr = URLOpenBlockingStreamW(NULL, NULL, &pStream, 0, &BindStatusCallback);
@ -324,12 +324,14 @@ static void test_URLOpenBlockingStreamW(void)
CHECK_CALLED(OnStopBinding); CHECK_CALLED(OnStopBinding);
ok(pStream != NULL, "pStream is NULL\n"); ok(pStream != NULL, "pStream is NULL\n");
if(pStream)
{
hr = IStream_Read(pStream, buffer, sizeof(buffer), NULL);
ok(hr == S_OK, "IStream_Read failed with error 0x%08x\n", hr);
ok(!memcmp(buffer, szHtmlDoc, sizeof(szHtmlDoc)-1), "read data differs from file\n");
hr = IStream_Read(pStream, buffer, sizeof(buffer), NULL); IStream_Release(pStream);
ok(hr == S_OK, "IStream_Read failed with error 0x%08x\n", hr); }
ok(!memcmp(buffer, szHtmlDoc, sizeof(szHtmlDoc)-1), "read data differs from file\n");
IStream_Release(pStream);
} }
static void test_URLOpenStreamW(void) static void test_URLOpenStreamW(void)