[PATCH v4 1/2] qedit/tests: Test adding a reference to the sample from ISampleGrabberCB::SampleCB().
Brendan McGrath
brendan at redmandi.com
Sat Jul 16 21:38:08 CDT 2022
Unravel Two adds a reference to the IMediaSample during its callback.
This patch adds a test that checks if an application does do this, that it can
free it later and the reference count will finish at zero.
Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=51616
Signed-off-by: Brendan McGrath <brendan at redmandi.com>
Signed-off-by: Zebediah Figura <zfigura at codeweavers.com>
---
dlls/qedit/tests/mediadet.c | 28 +++++++++++++++++++++++-----
1 file changed, 23 insertions(+), 5 deletions(-)
diff --git a/dlls/qedit/tests/mediadet.c b/dlls/qedit/tests/mediadet.c
index 3202baca9244..6e7017f865ce 100644
--- a/dlls/qedit/tests/mediadet.c
+++ b/dlls/qedit/tests/mediadet.c
@@ -866,6 +866,17 @@ static void test_put_filter(void)
ok(!ref, "Got outstanding refcount %ld.\n", ref);
}
+struct test_sample
+{
+ IMediaSample sample;
+ LONG refcount;
+};
+
+static struct test_sample *impl_from_IMediaSample(IMediaSample *iface)
+{
+ return CONTAINING_RECORD(iface, struct test_sample, sample);
+}
+
static HRESULT WINAPI ms_QueryInterface(IMediaSample *iface, REFIID riid,
void **ppvObject)
{
@@ -874,12 +885,14 @@ static HRESULT WINAPI ms_QueryInterface(IMediaSample *iface, REFIID riid,
static ULONG WINAPI ms_AddRef(IMediaSample *iface)
{
- return 2;
+ struct test_sample *sample = impl_from_IMediaSample(iface);
+ return InterlockedIncrement(&sample->refcount);
}
static ULONG WINAPI ms_Release(IMediaSample *iface)
{
- return 1;
+ struct test_sample *sample = impl_from_IMediaSample(iface);
+ return InterlockedDecrement(&sample->refcount);
}
static HRESULT WINAPI ms_GetPointer(IMediaSample *iface, BYTE **ppBuffer)
@@ -989,7 +1002,7 @@ static const IMediaSampleVtbl my_sample_vt = {
ms_SetMediaTime
};
-static IMediaSample my_sample = { &my_sample_vt };
+static struct test_sample my_sample = { {&my_sample_vt}, 0 };
static BOOL samplecb_called = FALSE;
@@ -1012,8 +1025,9 @@ static ULONG WINAPI sgcb_Release(ISampleGrabberCB *iface)
static HRESULT WINAPI sgcb_SampleCB(ISampleGrabberCB *iface, double SampleTime,
IMediaSample *pSample)
{
- ok(pSample == &my_sample, "Got wrong IMediaSample: %p, expected %p\n", pSample, &my_sample);
+ ok(pSample == &my_sample.sample, "Got wrong IMediaSample: %p, expected %p\n", pSample, &my_sample);
samplecb_called = TRUE;
+ IMediaSample_AddRef(pSample);
return E_NOTIMPL;
}
@@ -1043,6 +1057,7 @@ static void test_samplegrabber(void)
IEnumPins *pins;
HRESULT hr;
FILTER_STATE fstate;
+ ULONG refcount;
/* Invalid RIID */
hr = CoCreateInstance(&CLSID_SampleGrabber, NULL, CLSCTX_INPROC_SERVER, &IID_IClassFactory,
@@ -1074,10 +1089,13 @@ static void test_samplegrabber(void)
hr = IPin_QueryInterface(pin, &IID_IMemInputPin, (void**)&inpin);
ok(hr == S_OK, "Got hr %#lx.\n", hr);
- hr = IMemInputPin_Receive(inpin, &my_sample);
+ hr = IMemInputPin_Receive(inpin, &my_sample.sample);
ok(hr == S_OK, "Got hr %#lx.\n", hr);
ok(samplecb_called == TRUE, "SampleCB should have been called\n");
+ refcount = IUnknown_Release(&my_sample.sample);
+ todo_wine ok(!refcount, "Got unexpected refcount %ld.\n", refcount);
+
IMemInputPin_Release(inpin);
IPin_Release(pin);
--
2.34.1
More information about the wine-devel
mailing list