183 lines
6.4 KiB
C
183 lines
6.4 KiB
C
/*
|
|
* Memory allocator unit tests
|
|
*
|
|
* Copyright (C) 2005 Christian Costa
|
|
*
|
|
* 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
|
|
*/
|
|
|
|
#define COBJMACROS
|
|
#include "dshow.h"
|
|
#include "wine/test.h"
|
|
|
|
static IMemAllocator *create_allocator(void)
|
|
{
|
|
IMemAllocator *allocator = NULL;
|
|
HRESULT hr = CoCreateInstance(&CLSID_MemoryAllocator, NULL, CLSCTX_INPROC_SERVER,
|
|
&IID_IMemAllocator, (void **)&allocator);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
return allocator;
|
|
}
|
|
|
|
static void test_properties(void)
|
|
{
|
|
ALLOCATOR_PROPERTIES req_props = {0}, ret_props;
|
|
IMemAllocator *allocator = create_allocator();
|
|
IMediaSample *sample;
|
|
LONG size, ret_size;
|
|
unsigned int i;
|
|
HRESULT hr;
|
|
|
|
static const ALLOCATOR_PROPERTIES tests[] =
|
|
{
|
|
{0, 0, 1, 0},
|
|
{1, 0, 1, 0},
|
|
{1, 1, 1, 0},
|
|
{1, 1, 4, 0},
|
|
{2, 1, 1, 0},
|
|
{1, 2, 4, 0},
|
|
{1, 1, 16, 0},
|
|
{1, 16, 16, 0},
|
|
{1, 17, 16, 0},
|
|
{1, 32, 16, 0},
|
|
{1, 1, 16, 1},
|
|
{1, 15, 16, 1},
|
|
{1, 16, 16, 1},
|
|
{1, 17, 16, 1},
|
|
{1, 0, 16, 16},
|
|
{1, 1, 16, 16},
|
|
{1, 16, 16, 16},
|
|
};
|
|
|
|
memset(&ret_props, 0xcc, sizeof(ret_props));
|
|
hr = IMemAllocator_GetProperties(allocator, &ret_props);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
ok(!ret_props.cBuffers, "Got %d buffers.\n", ret_props.cBuffers);
|
|
ok(!ret_props.cbBuffer, "Got size %d.\n", ret_props.cbBuffer);
|
|
ok(!ret_props.cbAlign, "Got align %d.\n", ret_props.cbAlign);
|
|
ok(!ret_props.cbPrefix, "Got prefix %d.\n", ret_props.cbPrefix);
|
|
|
|
hr = IMemAllocator_SetProperties(allocator, &req_props, &ret_props);
|
|
ok(hr == VFW_E_BADALIGN, "Got hr %#x.\n", hr);
|
|
|
|
for (i = 0; i < ARRAY_SIZE(tests); i++)
|
|
{
|
|
req_props = tests[i];
|
|
hr = IMemAllocator_SetProperties(allocator, &req_props, &ret_props);
|
|
ok(hr == S_OK, "Test %u: Got hr %#x.\n", i, hr);
|
|
ok(!memcmp(&req_props, &tests[i], sizeof(req_props)), "Test %u: Requested props should not be changed.\n", i);
|
|
ok(ret_props.cBuffers == req_props.cBuffers, "Test %u: Got %d buffers.\n", i, ret_props.cBuffers);
|
|
ok(ret_props.cbBuffer >= req_props.cbBuffer, "Test %u: Got size %d.\n", i, ret_props.cbBuffer);
|
|
ok(ret_props.cbAlign == req_props.cbAlign, "Test %u: Got alignment %d.\n", i, ret_props.cbAlign);
|
|
ok(ret_props.cbPrefix == req_props.cbPrefix, "Test %u: Got prefix %d.\n", i, ret_props.cbPrefix);
|
|
ret_size = ret_props.cbBuffer;
|
|
|
|
hr = IMemAllocator_GetProperties(allocator, &ret_props);
|
|
ok(hr == S_OK, "Test %u: Got hr %#x.\n", i, hr);
|
|
ok(ret_props.cBuffers == req_props.cBuffers, "Test %u: Got %d buffers.\n", i, ret_props.cBuffers);
|
|
ok(ret_props.cbBuffer == ret_size, "Test %u: Got size %d.\n", i, ret_props.cbBuffer);
|
|
ok(ret_props.cbAlign == req_props.cbAlign, "Test %u: Got alignment %d.\n", i, ret_props.cbAlign);
|
|
ok(ret_props.cbPrefix == req_props.cbPrefix, "Test %u: Got prefix %d.\n", i, ret_props.cbPrefix);
|
|
|
|
hr = IMemAllocator_Commit(allocator);
|
|
if (!req_props.cbBuffer)
|
|
{
|
|
ok(hr == VFW_E_SIZENOTSET, "Test %u: Got hr %#x.\n", i, hr);
|
|
continue;
|
|
}
|
|
ok(hr == S_OK, "Test %u: Got hr %#x.\n", i, hr);
|
|
|
|
hr = IMemAllocator_SetProperties(allocator, &req_props, &ret_props);
|
|
ok(hr == VFW_E_ALREADY_COMMITTED, "Test %u: Got hr %#x.\n", i, hr);
|
|
|
|
hr = IMemAllocator_GetBuffer(allocator, &sample, NULL, NULL, 0);
|
|
ok(hr == S_OK, "Test %u: Got hr %#x.\n", i, hr);
|
|
|
|
size = IMediaSample_GetSize(sample);
|
|
ok(size == ret_size, "Test %u: Got size %d.\n", i, size);
|
|
|
|
hr = IMemAllocator_Decommit(allocator);
|
|
ok(hr == S_OK, "Test %u: Got hr %#x.\n", i, hr);
|
|
|
|
hr = IMemAllocator_SetProperties(allocator, &req_props, &ret_props);
|
|
ok(hr == VFW_E_BUFFERS_OUTSTANDING, "Test %u: Got hr %#x.\n", i, hr);
|
|
|
|
hr = IMediaSample_Release(sample);
|
|
ok(hr == S_OK, "Test %u: Got hr %#x.\n", i, hr);
|
|
}
|
|
|
|
IMemAllocator_Release(allocator);
|
|
}
|
|
|
|
static void test_commit(void)
|
|
{
|
|
ALLOCATOR_PROPERTIES req_props = {2, 65536, 1, 0}, ret_props;
|
|
IMemAllocator *allocator = create_allocator();
|
|
IMediaSample *sample, *sample2;
|
|
HRESULT hr;
|
|
BYTE *data;
|
|
|
|
hr = IMemAllocator_SetProperties(allocator, &req_props, &ret_props);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
|
|
hr = IMemAllocator_GetBuffer(allocator, &sample, NULL, NULL, 0);
|
|
ok(hr == VFW_E_NOT_COMMITTED, "Got hr %#x.\n", hr);
|
|
|
|
hr = IMemAllocator_Commit(allocator);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
hr = IMemAllocator_Commit(allocator);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
|
|
hr = IMemAllocator_GetBuffer(allocator, &sample, NULL, NULL, 0);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
IMediaSample_Release(sample);
|
|
|
|
hr = IMemAllocator_Decommit(allocator);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
hr = IMemAllocator_Decommit(allocator);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
|
|
/* Extant samples remain valid even after Decommit() is called. */
|
|
hr = IMemAllocator_Commit(allocator);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
|
|
hr = IMemAllocator_GetBuffer(allocator, &sample, NULL, NULL, 0);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
|
|
hr = IMediaSample_GetPointer(sample, &data);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
|
|
hr = IMemAllocator_Decommit(allocator);
|
|
ok(hr == S_OK, "Got hr %#x.\n", hr);
|
|
|
|
memset(data, 0xcc, 65536);
|
|
|
|
hr = IMemAllocator_GetBuffer(allocator, &sample2, NULL, NULL, 0);
|
|
ok(hr == VFW_E_NOT_COMMITTED, "Got hr %#x.\n", hr);
|
|
|
|
IMediaSample_Release(sample);
|
|
IMemAllocator_Release(allocator);
|
|
}
|
|
|
|
START_TEST(memallocator)
|
|
{
|
|
CoInitialize(NULL);
|
|
|
|
test_properties();
|
|
test_commit();
|
|
|
|
CoUninitialize();
|
|
}
|