qcap: Use separate functions when searching from a pin and filter.

Previously, if we tried to search from a matching but connected source, we
would loop forever.

Signed-off-by: Zebediah Figura <zfigura@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Zebediah Figura 2020-04-08 15:58:53 -05:00 committed by Alexandre Julliard
parent 18462a2f16
commit 74014b5539
2 changed files with 101 additions and 90 deletions

View File

@ -499,83 +499,88 @@ end:
return hr;
}
static HRESULT find_unconnected_pin(CaptureGraphImpl *This,
const GUID *pCategory, const GUID *pType, IUnknown *pSource, IPin **out_pin)
static HRESULT find_unconnected_source_from_filter(CaptureGraphImpl *capture_graph,
const GUID *category, const GUID *majortype, IBaseFilter *filter, IPin **ret);
static HRESULT find_unconnected_source_from_pin(CaptureGraphImpl *capture_graph,
const GUID *category, const GUID *majortype, IPin *pin, IPin **ret)
{
int index = 0;
IPin *source_out;
HRESULT hr;
BOOL usedSmartTeePreviewPin = FALSE;
/* depth-first search the graph for the first unconnected pin that matches
* the given category and type */
for(;;){
IPin *nextpin;
if (pCategory && (IsEqualIID(pCategory, &PIN_CATEGORY_CAPTURE) || IsEqualIID(pCategory, &PIN_CATEGORY_PREVIEW))){
IBaseFilter *sourceFilter = NULL;
hr = IUnknown_QueryInterface(pSource, &IID_IBaseFilter, (void**)&sourceFilter);
if (SUCCEEDED(hr)) {
hr = match_smart_tee_pin(This, pCategory, pType, pSource, &source_out);
if (hr == VFW_S_NOPREVIEWPIN)
usedSmartTeePreviewPin = TRUE;
IBaseFilter_Release(sourceFilter);
} else {
hr = ICaptureGraphBuilder2_FindPin(&This->ICaptureGraphBuilder2_iface, pSource, PINDIR_OUTPUT, pCategory, pType, FALSE, index, &source_out);
}
if (FAILED(hr))
return E_INVALIDARG;
} else {
hr = ICaptureGraphBuilder2_FindPin(&This->ICaptureGraphBuilder2_iface, pSource, PINDIR_OUTPUT, pCategory, pType, FALSE, index, &source_out);
if (FAILED(hr))
return E_INVALIDARG;
}
hr = IPin_ConnectedTo(source_out, &nextpin);
if(SUCCEEDED(hr)){
PIN_INFO info;
HRESULT hr;
IPin *peer;
IPin_Release(source_out);
if (category && (IsEqualGUID(category, &PIN_CATEGORY_CAPTURE)
|| IsEqualGUID(category, &PIN_CATEGORY_PREVIEW)))
hr = match_smart_tee_pin(capture_graph, category, majortype, (IUnknown *)pin, &pin);
else
hr = ICaptureGraphBuilder2_FindPin(&capture_graph->ICaptureGraphBuilder2_iface,
(IUnknown *)pin, PINDIR_OUTPUT, category, majortype, FALSE, -1, &pin);
if (FAILED(hr))
return hr;
hr = IPin_QueryPinInfo(nextpin, &info);
if(FAILED(hr) || !info.pFilter){
WARN("QueryPinInfo failed: %08x\n", hr);
if (FAILED(IPin_ConnectedTo(pin, &peer)))
{
*ret = pin;
return hr;
}
hr = find_unconnected_pin(This, pCategory, pType, (IUnknown*)info.pFilter, out_pin);
IPin_QueryPinInfo(peer, &info);
hr = find_unconnected_source_from_filter(capture_graph, category, majortype, info.pFilter, ret);
IBaseFilter_Release(info.pFilter);
IPin_Release(peer);
IPin_Release(pin);
return hr;
}
static HRESULT find_unconnected_source_from_filter(CaptureGraphImpl *capture_graph,
const GUID *category, const GUID *majortype, IBaseFilter *filter, IPin **ret)
{
IPin *pin, *peer;
HRESULT hr;
int index;
if (category && (IsEqualGUID(category, &PIN_CATEGORY_CAPTURE)
|| IsEqualGUID(category, &PIN_CATEGORY_PREVIEW)))
{
if (FAILED(hr = match_smart_tee_pin(capture_graph, category, majortype, (IUnknown *)filter, &pin)))
return hr;
if (FAILED(IPin_ConnectedTo(pin, &peer)))
{
*ret = pin;
return hr;
}
IPin_Release(peer);
IPin_Release(pin);
return E_INVALIDARG;
}
for (index = 0; SUCCEEDED(ICaptureGraphBuilder2_FindPin(&capture_graph->ICaptureGraphBuilder2_iface,
(IUnknown *)filter, PINDIR_OUTPUT, category, majortype, FALSE, index, &pin)); ++index)
{
hr = find_unconnected_source_from_pin(capture_graph, category, majortype, pin, ret);
IPin_Release(pin);
if (SUCCEEDED(hr))
return hr;
}else{
*out_pin = source_out;
if(usedSmartTeePreviewPin)
return VFW_S_NOPREVIEWPIN;
return S_OK;
}
index++;
}
return E_INVALIDARG;
}
static HRESULT WINAPI
fnCaptureGraphBuilder2_RenderStream(ICaptureGraphBuilder2 * iface,
const GUID *pCategory,
const GUID *pType,
IUnknown *pSource,
IBaseFilter *pfCompressor,
IBaseFilter *pfRenderer)
static HRESULT WINAPI fnCaptureGraphBuilder2_RenderStream(ICaptureGraphBuilder2 *iface,
const GUID *category, const GUID *majortype, IUnknown *source,
IBaseFilter *pfCompressor, IBaseFilter *pfRenderer)
{
CaptureGraphImpl *This = impl_from_ICaptureGraphBuilder2(iface);
IPin *source_out = NULL, *renderer_in;
BOOL rendererNeedsRelease = FALSE;
HRESULT hr, return_hr = S_OK;
IBaseFilter *filter;
IPin *pin;
FIXME("(%p/%p)->(%s, %s, %p, %p, %p) semi-stub!\n", This, iface,
debugstr_guid(pCategory), debugstr_guid(pType),
pSource, pfCompressor, pfRenderer);
TRACE("graph %p, category %s, majortype %s, source %p, intermediate %p, sink %p.\n",
This, debugstr_guid(category), debugstr_guid(majortype), source, pfCompressor, pfRenderer);
if (!This->mygraph)
{
@ -583,12 +588,27 @@ fnCaptureGraphBuilder2_RenderStream(ICaptureGraphBuilder2 * iface,
return E_UNEXPECTED;
}
if (pCategory && IsEqualIID(pCategory, &PIN_CATEGORY_VBI)) {
if (category && IsEqualGUID(category, &PIN_CATEGORY_VBI))
{
FIXME("Tee/Sink-to-Sink filter not supported\n");
return E_NOTIMPL;
}
hr = find_unconnected_pin(This, pCategory, pType, pSource, &source_out);
if (IUnknown_QueryInterface(source, &IID_IPin, (void **)&pin) == S_OK)
{
hr = find_unconnected_source_from_pin(This, category, majortype, pin, &source_out);
IPin_Release(pin);
}
else if (IUnknown_QueryInterface(source, &IID_IBaseFilter, (void **)&filter) == S_OK)
{
hr = find_unconnected_source_from_filter(This, category, majortype, filter, &source_out);
IBaseFilter_Release(filter);
}
else
{
WARN("Source object does not expose IBaseFilter or IPin.\n");
return E_INVALIDARG;
}
if (FAILED(hr))
return hr;
return_hr = hr;

View File

@ -929,12 +929,9 @@ static void test_render_stream(void)
todo_wine ok(!transform.source1.pin.pin.peer, "Pin should not be connected.\n");
todo_wine ok(!sink.sink1.pin.pin.peer, "Pin should not be connected.\n");
if (0)
{
hr = ICaptureGraphBuilder2_RenderStream(capture_graph, &PIN_CATEGORY_CC, NULL,
(IUnknown *)&source.source1.pin.pin.IPin_iface, NULL, &sink.filter.IBaseFilter_iface);
ok(hr == E_INVALIDARG, "Got hr %#x.\n", hr);
}
disconnect_pins(graph, &source.source1);
@ -983,12 +980,9 @@ static void test_render_stream(void)
check_smart_tee_pin(transform.sink1.pin.pin.peer, L"Capture");
ok(transform.source1.pin.pin.peer == &sink.sink1.pin.pin.IPin_iface, "Got wrong connection.\n");
if (0)
{
hr = ICaptureGraphBuilder2_RenderStream(capture_graph, &PIN_CATEGORY_CAPTURE, NULL,
(IUnknown *)&source.filter.IBaseFilter_iface, NULL, &sink.filter.IBaseFilter_iface);
ok(hr == E_INVALIDARG, "Got hr %#x.\n", hr);
}
hr = ICaptureGraphBuilder2_RenderStream(capture_graph, &PIN_CATEGORY_PREVIEW, NULL,
(IUnknown *)&source.filter.IBaseFilter_iface, NULL, &sink.filter.IBaseFilter_iface);
@ -1014,12 +1008,9 @@ static void test_render_stream(void)
check_smart_tee_pin(transform.sink1.pin.pin.peer, L"Preview");
ok(transform.source1.pin.pin.peer == &sink.sink1.pin.pin.IPin_iface, "Got wrong connection.\n");
if (0)
{
hr = ICaptureGraphBuilder2_RenderStream(capture_graph, &PIN_CATEGORY_PREVIEW, NULL,
(IUnknown *)&source.filter.IBaseFilter_iface, NULL, &sink.filter.IBaseFilter_iface);
ok(hr == E_INVALIDARG, "Got hr %#x.\n", hr);
}
disconnect_pins(graph, &source.source1);
IGraphBuilder_RemoveFilter(graph, &transform.filter.IBaseFilter_iface);
@ -1031,8 +1022,8 @@ static void test_render_stream(void)
hr = ICaptureGraphBuilder2_RenderStream(capture_graph, &PIN_CATEGORY_CAPTURE, NULL,
(IUnknown *)&source.source1.pin.pin.IPin_iface, NULL, &sink.filter.IBaseFilter_iface);
ok(hr == S_OK, "Got hr %#x.\n", hr);
todo_wine check_smart_tee_pin(source.source1.pin.pin.peer, L"Input");
todo_wine check_smart_tee_pin(transform.sink1.pin.pin.peer, L"Capture");
check_smart_tee_pin(source.source1.pin.pin.peer, L"Input");
check_smart_tee_pin(transform.sink1.pin.pin.peer, L"Capture");
ok(transform.source1.pin.pin.peer == &sink.sink1.pin.pin.IPin_iface, "Got wrong connection.\n");
disconnect_pins(graph, &source.source1);
IGraphBuilder_RemoveFilter(graph, &transform.filter.IBaseFilter_iface);
@ -1062,8 +1053,8 @@ static void test_render_stream(void)
hr = ICaptureGraphBuilder2_RenderStream(capture_graph, &PIN_CATEGORY_CAPTURE, NULL,
(IUnknown *)&source.source1.pin.pin.IPin_iface, NULL, &sink.filter.IBaseFilter_iface);
ok(hr == S_OK, "Got hr %#x.\n", hr);
todo_wine check_smart_tee_pin(source.source1.pin.pin.peer, L"Input");
todo_wine check_smart_tee_pin(transform.sink1.pin.pin.peer, L"Capture");
check_smart_tee_pin(source.source1.pin.pin.peer, L"Input");
check_smart_tee_pin(transform.sink1.pin.pin.peer, L"Capture");
ok(transform.source1.pin.pin.peer == &sink.sink1.pin.pin.IPin_iface, "Got wrong connection.\n");
disconnect_pins(graph, &source.source1);
disconnect_pins(graph, &transform.source1);
@ -1075,9 +1066,9 @@ static void test_render_stream(void)
ref = IBaseFilter_Release(&source.filter.IBaseFilter_iface);
ok(!ref, "Got outstanding refcount %d.\n", ref);
ref = IBaseFilter_Release(&transform.filter.IBaseFilter_iface);
todo_wine ok(!ref, "Got outstanding refcount %d.\n", ref);
ok(!ref, "Got outstanding refcount %d.\n", ref);
ref = IBaseFilter_Release(&sink.filter.IBaseFilter_iface);
todo_wine ok(!ref, "Got outstanding refcount %d.\n", ref);
ok(!ref, "Got outstanding refcount %d.\n", ref);
}
START_TEST(capturegraph)