[PATCH 1/7] winegstreamer: Introduce a new wg_source interface.

[Rémi Bernon]

From: Rémi Bernon

---
 MAINTAINERS                       |  1 +
 dlls/winegstreamer/Makefile.in    |  1 +
 dlls/winegstreamer/gst_private.h  |  3 +
 dlls/winegstreamer/main.c         | 25 ++++++++
 dlls/winegstreamer/media_source.c | 15 ++++-
 dlls/winegstreamer/unix_private.h |  5 ++
 dlls/winegstreamer/unixlib.c      |  1 +
 dlls/winegstreamer/unixlib.h      |  9 +++
 dlls/winegstreamer/wg_parser.c    |  6 ++
 dlls/winegstreamer/wg_source.c    | 94 +++++++++++++++++++++++++++++++
 10 files changed, 159 insertions(+), 1 deletion(-)
 create mode 100644 dlls/winegstreamer/wg_source.c

diff --git a/MAINTAINERS b/MAINTAINERS
index dc5c85cc4ec..2fd7340efb0 100644
--- a/MAINTAINERS
+++ b/MAINTAINERS
@@ -223,6 +223,7 @@ F:	dlls/winegstreamer/mfplat.c
 F:	dlls/winegstreamer/resampler.c
 F:	dlls/winegstreamer/video_decoder.c
 F:	dlls/winegstreamer/video_processor.c
+F:	dlls/winegstreamer/wg_source.c
 F:	dlls/winegstreamer/wg_sample.c
 F:	dlls/winegstreamer/wg_transform.c
 F:	dlls/winegstreamer/wma_decoder.c
diff --git a/dlls/winegstreamer/Makefile.in b/dlls/winegstreamer/Makefile.in
index 78bdd6c0ef1..dbf0be035f1 100644
--- a/dlls/winegstreamer/Makefile.in
+++ b/dlls/winegstreamer/Makefile.in
@@ -24,6 +24,7 @@ C_SRCS = \
    wg_format.c \
    wg_parser.c \
    wg_sample.c \
+	wg_source.c \
    wg_transform.c \
    wm_reader.c \
    wma_decoder.c \
diff --git a/dlls/winegstreamer/gst_private.h b/dlls/winegstreamer/gst_private.h
index ed867f741d9..fd33ac42c19 100644
--- a/dlls/winegstreamer/gst_private.h
+++ b/dlls/winegstreamer/gst_private.h
@@ -110,6 +110,9 @@ bool wg_transform_get_status(wg_transform_t transform, bool *accepts_input);
 HRESULT wg_transform_drain(wg_transform_t transform);
 HRESULT wg_transform_flush(wg_transform_t transform);

+HRESULT wg_source_create(wg_source_t *out);
+void wg_source_destroy(wg_source_t source);
+
 unsigned int wg_format_get_max_size(const struct wg_format *format);

HRESULT avi_splitter_create(IUnknown *outer, IUnknown **out);
diff --git a/dlls/winegstreamer/main.c b/dlls/winegstreamer/main.c
index 9624c469314..8237936891c 100644
--- a/dlls/winegstreamer/main.c
+++ b/dlls/winegstreamer/main.c
@@ -456,6 +456,31 @@ HRESULT wg_transform_flush(wg_transform_t transform)
     return S_OK;
 }

+HRESULT wg_source_create(wg_source_t *out)
+{
+    struct wg_source_create_params params = {0};
+    NTSTATUS status;
+
+    TRACE("out %p\n", out);
+
+    if ((status = WINE_UNIX_CALL(unix_wg_source_create, &params)))
+        WARN("wg_source_create returned status %#lx\n", status);
+    else
+    {
+        TRACE("Returning source %#I64x.\n", params.source);
+        *out = params.source;
+    }
+
+    return HRESULT_FROM_NT(status);
+}
+
+void wg_source_destroy(wg_source_t source)
+{
+    TRACE("source %#I64x.\n", source);
+
+    WINE_UNIX_CALL(unix_wg_source_destroy, &source);
+}
+
 #define ALIGN(n, alignment) (((n) + (alignment) - 1) & ~((alignment) - 1))

unsigned int wg_format_get_stride(const struct wg_format *format)
diff --git a/dlls/winegstreamer/media_source.c b/dlls/winegstreamer/media_source.c
index 8b9d42ea3f0..c6403661c0f 100644
--- a/dlls/winegstreamer/media_source.c
+++ b/dlls/winegstreamer/media_source.c
@@ -36,6 +36,8 @@ struct object_context
     IMFByteStream *stream;
     UINT64 file_size;
     WCHAR *url;
+
+    wg_source_t wg_source;
 };

static struct object_context *impl_from_IUnknown(IUnknown *iface)
@@ -78,6 +80,8 @@ static ULONG WINAPI object_context_Release(IUnknown *iface)

if (!refcount)
     {
+        if (context->wg_source)
+            wg_source_destroy(context->wg_source);
         IMFAsyncResult_Release(context->result);
         IMFByteStream_Release(context->stream);
         free(context->url);
@@ -182,6 +186,7 @@ struct media_source

CRITICAL_SECTION cs;

+    wg_source_t wg_source;
     UINT64 file_size;
     wg_parser_t wg_parser;
     UINT64 duration;
@@ -1359,6 +1364,7 @@ static ULONG WINAPI media_source_Release(IMFMediaSource *iface)
         IMFMediaSource_Shutdown(iface);
         IMFMediaEventQueue_Release(source->event_queue);
         IMFByteStream_Release(source->byte_stream);
+        wg_source_destroy(source->wg_source);
         wg_parser_destroy(source->wg_parser);
         source->cs.DebugInfo->Spare[0] = 0;
         DeleteCriticalSection(&source->cs);
@@ -1637,6 +1643,9 @@ static HRESULT media_source_create(struct object_context *context, IMFMediaSourc
     InitializeCriticalSection(&object->cs);
     object->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": cs");

+    object->wg_source = context->wg_source;
+    context->wg_source = 0;
+
     if (FAILED(hr = MFCreateEventQueue(&object->event_queue)))
         goto fail;

@@ -1716,6 +1725,8 @@ fail:
         WaitForSingleObject(object->read_thread, INFINITE);
         CloseHandle(object->read_thread);
     }
+    if (object->wg_source)
+        wg_source_destroy(object->wg_source);
     if (object->wg_parser)
         wg_parser_destroy(object->wg_parser);
     if (object->async_commands_queue)
@@ -1998,7 +2009,9 @@ static HRESULT WINAPI stream_handler_callback_Invoke(IMFAsyncCallback *iface, IM
     if (!state || !(context = impl_from_IUnknown(state)))
         return E_INVALIDARG;

-    if (FAILED(hr = media_source_create(context, (IMFMediaSource **)&object)))
+    if (FAILED(hr = wg_source_create(&context->wg_source)))
+        WARN("Failed to create wg_source, hr %#lx\n", hr);
+    else if (FAILED(hr = media_source_create(context, (IMFMediaSource **)&object)))
         WARN("Failed to create media source, hr %#lx\n", hr);
     else
     {
diff --git a/dlls/winegstreamer/unix_private.h b/dlls/winegstreamer/unix_private.h
index db6bf3ebb2e..8574cac797f 100644
--- a/dlls/winegstreamer/unix_private.h
+++ b/dlls/winegstreamer/unix_private.h
@@ -57,6 +57,11 @@ extern NTSTATUS wg_transform_get_status(void *args) DECLSPEC_HIDDEN;
 extern NTSTATUS wg_transform_drain(void *args) DECLSPEC_HIDDEN;
 extern NTSTATUS wg_transform_flush(void *args) DECLSPEC_HIDDEN;

+/* wg_source.c */
+
+extern NTSTATUS wg_source_create(void *args) DECLSPEC_HIDDEN;
+extern NTSTATUS wg_source_destroy(void *args) DECLSPEC_HIDDEN;
+
 /* wg_allocator.c */

static inline BYTE *wg_sample_data(struct wg_sample *sample)
diff --git a/dlls/winegstreamer/unixlib.c b/dlls/winegstreamer/unixlib.c
index 6ed38260536..cb28b76b65e 100644
--- a/dlls/winegstreamer/unixlib.c
+++ b/dlls/winegstreamer/unixlib.c
@@ -31,6 +31,7 @@
 #include <gst/gst.h>
 #include <gst/video/video.h>
 #include <gst/audio/audio.h>
+#include <gst/base/base.h>
 #include <gst/tag/tag.h>

#include "ntstatus.h"
diff --git a/dlls/winegstreamer/unixlib.h b/dlls/winegstreamer/unixlib.h
index 702bd7aa69b..7fb177c4ed5 100644
--- a/dlls/winegstreamer/unixlib.h
+++ b/dlls/winegstreamer/unixlib.h
@@ -207,6 +207,7 @@ enum wg_parser_type
 typedef UINT64 wg_parser_t;
 typedef UINT64 wg_parser_stream_t;
 typedef UINT64 wg_transform_t;
+typedef UINT64 wg_source_t;

struct wg_parser_create_params
 {
@@ -363,6 +364,11 @@ struct wg_transform_get_status_params
     UINT32 accepts_input;
 };

+struct wg_source_create_params
+{
+    wg_source_t source;
+};
+
 enum unix_funcs
 {
     unix_wg_init_gstreamer,
@@ -402,6 +408,9 @@ enum unix_funcs
     unix_wg_transform_get_status,
     unix_wg_transform_drain,
     unix_wg_transform_flush,
+
+    unix_wg_source_create,
+    unix_wg_source_destroy,
 };

#endif /* __WINE_WINEGSTREAMER_UNIXLIB_H */
diff --git a/dlls/winegstreamer/wg_parser.c b/dlls/winegstreamer/wg_parser.c
index 298c05c6b88..387871d116a 100644
--- a/dlls/winegstreamer/wg_parser.c
+++ b/dlls/winegstreamer/wg_parser.c
@@ -1945,6 +1945,9 @@ const unixlib_entry_t __wine_unix_call_funcs[] =
     X(wg_transform_get_status),
     X(wg_transform_drain),
     X(wg_transform_flush),
+
+    X(wg_source_create),
+    X(wg_source_destroy),
 };

#ifdef _WIN64
@@ -2192,6 +2195,9 @@ const unixlib_entry_t __wine_unix_call_wow64_funcs[] =
     X(wg_transform_get_status),
     X(wg_transform_drain),
     X(wg_transform_flush),
+
+    X(wg_source_create),
+    X(wg_source_destroy),
 };

#endif  /* _WIN64 */
diff --git a/dlls/winegstreamer/wg_source.c b/dlls/winegstreamer/wg_source.c
new file mode 100644
index 00000000000..52a54196ea9
--- /dev/null
+++ b/dlls/winegstreamer/wg_source.c
@@ -0,0 +1,94 @@
+/*
+ * Copyright 2023 Rémi Bernon for CodeWeavers
+ *
+ * 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
+ */
+
+#if 0
+#pragma makedep unix
+#endif
+
+#include "config.h"
+
+#include <assert.h>
+#include <stdarg.h>
+#include <stdio.h>
+
+#include <gst/gst.h>
+#include <gst/video/video.h>
+#include <gst/audio/audio.h>
+#include <gst/base/base.h>
+#include <gst/tag/tag.h>
+
+#include "ntstatus.h"
+#define WIN32_NO_STATUS
+#include "winternl.h"
+#include "mferror.h"
+
+#include "unix_private.h"
+
+struct wg_source
+{
+    GstElement *container;
+};
+
+static struct wg_source *get_source(wg_source_t source)
+{
+    return (struct wg_source *)(ULONG_PTR)source;
+}
+
+NTSTATUS wg_source_create(void *args)
+{
+    struct wg_source_create_params *params = args;
+    struct wg_source *source;
+
+    if (!(source = calloc(1, sizeof(*source))))
+        return STATUS_UNSUCCESSFUL;
+
+    if (!(source->container = gst_bin_new("wg_source")))
+        goto error;
+
+    gst_element_set_state(source->container, GST_STATE_PAUSED);
+    if (!gst_element_get_state(source->container, NULL, NULL, -1))
+        goto error;
+
+    params->source = (wg_source_t)(ULONG_PTR)source;
+    GST_INFO("Created winegstreamer source %p.", source);
+    return STATUS_SUCCESS;
+
+error:
+    if (source->container)
+    {
+        gst_element_set_state(source->container, GST_STATE_NULL);
+        gst_object_unref(source->container);
+    }
+    free(source);
+
+    GST_ERROR("Failed to create winegstreamer source.");
+    return STATUS_UNSUCCESSFUL;
+}
+
+NTSTATUS wg_source_destroy(void *args)
+{
+    struct wg_source *source = get_source(*(wg_source_t *)args);
+
+    GST_TRACE("source %p", source);
+
+    gst_element_set_state(source->container, GST_STATE_NULL);
+    gst_object_unref(source->container);
+    free(source);
+
+    return STATUS_SUCCESS;
+}