[PATCH v2 09/10] winegstreamer/media_source: Handle GST_QUERY_DURATION on wg_source src pad.

[Rémi Bernon]

From: Rémi Bernon

---
 dlls/winegstreamer/gst_private.h  |  3 ++-
 dlls/winegstreamer/main.c         |  7 +++++--
 dlls/winegstreamer/media_source.c |  3 ++-
 dlls/winegstreamer/unixlib.h      |  1 +
 dlls/winegstreamer/wg_source.c    | 17 +++++++++++++++++
 5 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/dlls/winegstreamer/gst_private.h b/dlls/winegstreamer/gst_private.h
index 527fdf47b07..cf46768122f 100644
--- a/dlls/winegstreamer/gst_private.h
+++ b/dlls/winegstreamer/gst_private.h
@@ -104,7 +104,8 @@ char *wg_parser_stream_get_tag(wg_parser_stream_t stream, enum wg_parser_tag tag
 void wg_parser_stream_seek(wg_parser_stream_t stream, double rate,
         uint64_t start_pos, uint64_t stop_pos, DWORD start_flags, DWORD stop_flags);

-HRESULT wg_source_create(const WCHAR *url, const void *data, uint32_t size, wg_source_t *out);
+HRESULT wg_source_create(const WCHAR *url, uint64_t file_size,
+        const void *data, uint32_t size, wg_source_t *out);
 void wg_source_destroy(wg_source_t source);
 HRESULT wg_source_push_data(wg_source_t source, UINT64 offset, const void *data, uint32_t size);

diff --git a/dlls/winegstreamer/main.c b/dlls/winegstreamer/main.c
index 7caf1fb0782..038ddf3e045 100644
--- a/dlls/winegstreamer/main.c
+++ b/dlls/winegstreamer/main.c
@@ -435,17 +435,20 @@ void wg_parser_stream_seek(wg_parser_stream_t stream, double rate,
     WINE_UNIX_CALL(unix_wg_parser_stream_seek, &params);
 }

-HRESULT wg_source_create(const WCHAR *url, const void *data, uint32_t size, wg_source_t *out)
+HRESULT wg_source_create(const WCHAR *url, uint64_t file_size,
+        const void *data, uint32_t size, wg_source_t *out)
 {
     struct wg_source_create_params params =
     {
+        .file_size = file_size,
         .data = data, .size = size,
     };
     char *tmp = NULL;
     NTSTATUS status;
     UINT len;

-    TRACE("url %s, data %p, size %#x\n", debugstr_w(url), data, size);
+    TRACE("url %s, file_size %#I64x, data %p, size %#x\n", debugstr_w(url),
+            file_size, data, size);

if (url && (len = WideCharToMultiByte(CP_ACP, 0, url, -1, NULL, 0, NULL, NULL)))
         tmp = malloc(len);
diff --git a/dlls/winegstreamer/media_source.c b/dlls/winegstreamer/media_source.c
index 0cb83437b61..3393ff33dd5 100644
--- a/dlls/winegstreamer/media_source.c
+++ b/dlls/winegstreamer/media_source.c
@@ -2049,7 +2049,8 @@ static HRESULT WINAPI stream_handler_callback_Invoke(IMFAsyncCallback *iface, IM

if (FAILED(hr = IMFByteStream_EndRead(context->stream, result, &size)))
         WARN("Failed to complete stream read, hr %#lx\n", hr);
-    else if (FAILED(hr = wg_source_create(context->url, context->buffer, size, &context->wg_source)))
+    else if (FAILED(hr = wg_source_create(context->url, context->file_size,
+            context->buffer, size, &context->wg_source)))
         WARN("Failed to create wg_source, hr %#lx\n", hr);
     else if (FAILED(hr = wg_source_push_data(context->wg_source, context->read_offset, context->buffer, size)))
         WARN("Failed to push wg_source data, hr %#lx\n", hr);
diff --git a/dlls/winegstreamer/unixlib.h b/dlls/winegstreamer/unixlib.h
index a70ecf9cd25..f7ed2fafd54 100644
--- a/dlls/winegstreamer/unixlib.h
+++ b/dlls/winegstreamer/unixlib.h
@@ -333,6 +333,7 @@ struct wg_parser_stream_seek_params
 struct wg_source_create_params
 {
     const char *url;
+    UINT64 file_size;
     const void *data;
     UINT32 size;
     wg_source_t source;
diff --git a/dlls/winegstreamer/wg_source.c b/dlls/winegstreamer/wg_source.c
index 77c40179314..e1a907db3e6 100644
--- a/dlls/winegstreamer/wg_source.c
+++ b/dlls/winegstreamer/wg_source.c
@@ -108,6 +108,20 @@ static GstPad *create_pad_with_caps(GstPadDirection direction, GstCaps *caps)
     return pad;
 }

+static gboolean src_query_duration(struct wg_source *source, GstQuery *query)
+{
+    GstFormat format;
+
+    GST_LOG("source %p, query %" GST_PTR_FORMAT, source, query);
+
+    gst_query_parse_duration(query, &format, NULL);
+    if (format != GST_FORMAT_BYTES)
+        return false;
+
+    gst_query_set_duration(query, format, source->segment.stop);
+    return true;
+}
+
 static gboolean src_query_uri(struct wg_source *source, GstQuery *query)
 {
     gchar *uri;
@@ -126,6 +140,8 @@ static gboolean src_query_cb(GstPad *pad, GstObject *parent, GstQuery *query)

switch (GST_QUERY_TYPE(query))
     {
+    case GST_QUERY_DURATION:
+        return src_query_duration(source, query);
     case GST_QUERY_URI:
         if (!source->url)
             return false;
@@ -152,6 +168,7 @@ NTSTATUS wg_source_create(void *args)
     }
     source->url = params->url ? strdup(params->url) : NULL;
     gst_segment_init(&source->segment, GST_FORMAT_BYTES);
+    source->segment.stop = params->file_size;

if (!(source->container = gst_bin_new("wg_source")))
         goto error;