Nikolay Sivov : mf/topoloader: Switch to public interface for initial topology validation.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 8e343024b577892bd4908304ded34b758579698d
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=8e343024b577892bd4908304ded34b758579698d

Author: Nikolay Sivov <nsivov at codeweavers.com>
Date:   Wed Jul 15 15:55:45 2020 +0300

mf/topoloader: Switch to public interface for initial topology validation.

Signed-off-by: Nikolay Sivov <nsivov at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/mf/topology.c | 36 ++++++++++++++++++++++++------------
 1 file changed, 24 insertions(+), 12 deletions(-)

diff --git a/dlls/mf/topology.c b/dlls/mf/topology.c
index abe66c45fd..8522f56969 100644
--- a/dlls/mf/topology.c
+++ b/dlls/mf/topology.c
@@ -1936,38 +1936,50 @@ static ULONG WINAPI topology_loader_Release(IMFTopoLoader *iface)
 static HRESULT WINAPI topology_loader_Load(IMFTopoLoader *iface, IMFTopology *input_topology,
         IMFTopology **output_topology, IMFTopology *current_topology)
 {
-    struct topology *topology = unsafe_impl_from_IMFTopology(input_topology);
+    MF_TOPOLOGY_TYPE node_type;
+    IMFTopologyNode *node;
+    unsigned short i = 0;
     IMFStreamSink *sink;
-    HRESULT hr;
-    size_t i;
+    IUnknown *object;
+    HRESULT hr = E_FAIL;
 
     FIXME("%p, %p, %p, %p.\n", iface, input_topology, output_topology, current_topology);
 
     if (current_topology)
         FIXME("Current topology instance is ignored.\n");
 
-    for (i = 0; i < topology->nodes.count; ++i)
+    /* Basic sanity checks for input topology:
+
+       - source nodes must have stream descriptor set;
+       - sink nodes must be resolved to stream sink objects;
+    */
+    while (SUCCEEDED(IMFTopology_GetNode(input_topology, i++, &node)))
     {
-        struct topology_node *node = topology->nodes.nodes[i];
+        IMFTopologyNode_GetNodeType(node, &node_type);
 
-        switch (node->node_type)
+        switch (node_type)
         {
             case MF_TOPOLOGY_OUTPUT_NODE:
-                if (node->object)
+                if (SUCCEEDED(hr = IMFTopologyNode_GetObject(node, &object)))
                 {
                     /* Sinks must be bound beforehand. */
-                    if (FAILED(IUnknown_QueryInterface(node->object, &IID_IMFStreamSink, (void **)&sink)))
-                        return MF_E_TOPO_SINK_ACTIVATES_UNSUPPORTED;
-                    IMFStreamSink_Release(sink);
+                    if (FAILED(IUnknown_QueryInterface(object, &IID_IMFStreamSink, (void **)&sink)))
+                        hr = MF_E_TOPO_SINK_ACTIVATES_UNSUPPORTED;
+                    else if (sink)
+                        IMFStreamSink_Release(sink);
+                    IUnknown_Release(object);
                 }
                 break;
             case MF_TOPOLOGY_SOURCESTREAM_NODE:
-                if (FAILED(hr = IMFAttributes_GetItem(node->attributes, &MF_TOPONODE_STREAM_DESCRIPTOR, NULL)))
-                    return hr;
+                hr = IMFTopologyNode_GetItem(node, &MF_TOPONODE_STREAM_DESCRIPTOR, NULL);
                 break;
             default:
                 ;
         }
+
+        IMFTopologyNode_Release(node);
+        if (FAILED(hr))
+            return hr;
     }
 
     if (FAILED(hr = MFCreateTopology(output_topology)))