dlls/winegstreamer/unixlib.c

   1 /*
   2  * winegstreamer Unix library interface
   3  *
   4  * Copyright 2020-2021 Zebediah Figura for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #if 0
  22 #pragma makedep unix
  23 #endif
  24
  25 #include "config.h"
  26
  27 #include <assert.h>
  28 #include <stdarg.h>
  29 #include <stdio.h>
  30
  31 #include <gst/gst.h>
  32 #include <gst/video/video.h>
  33 #include <gst/audio/audio.h>
  34 #include <gst/tag/tag.h>
  35
  36 #include "ntstatus.h"
  37 #define WIN32_NO_STATUS
  38 #include "winternl.h"
  39 #include "dshow.h"
  40
  41 #include "unix_private.h"
  42
  43 /* GStreamer callbacks may be called on threads not created by Wine, and
  44  * therefore cannot access the Wine TEB. This means that we must use GStreamer
  45  * debug logging instead of Wine debug logging. In order to be safe we forbid
  46  * any use of Wine debug logging in this entire file. */
  47
  48 GST_DEBUG_CATEGORY(wine);
  49
  50 GstStreamType stream_type_from_caps(GstCaps *caps)
  51 {
  52     const gchar *media_type;
  53
  54     if (!caps || !gst_caps_get_size(caps))
  55         return GST_STREAM_TYPE_UNKNOWN;
  56
  57     media_type = gst_structure_get_name(gst_caps_get_structure(caps, 0));
  58     if (g_str_has_prefix(media_type, "video/")
  59             || g_str_has_prefix(media_type, "image/"))
  60         return GST_STREAM_TYPE_VIDEO;
  61     if (g_str_has_prefix(media_type, "audio/"))
  62         return GST_STREAM_TYPE_AUDIO;
  63     if (g_str_has_prefix(media_type, "text/")
  64             || g_str_has_prefix(media_type, "subpicture/")
  65             || g_str_has_prefix(media_type, "closedcaption/"))
  66         return GST_STREAM_TYPE_TEXT;
  67
  68     return GST_STREAM_TYPE_UNKNOWN;
  69 }
  70
  71 GstElement *create_element(const char *name, const char *plugin_set)
  72 {
  73     GstElement *element;
  74
  75     if (!(element = gst_element_factory_make(name, NULL)))
  76         fprintf(stderr, "winegstreamer: failed to create %s, are %u-bit GStreamer \"%s\" plugins installed?\n",
  77                 name, 8 * (unsigned int)sizeof(void *), plugin_set);
  78     return element;
  79 }
  80
  81 GstElement *find_element(GstElementFactoryListType type, GstCaps *src_caps, GstCaps *sink_caps)
  82 {
  83     GstElement *element = NULL;
  84     GList *tmp, *transforms;
  85     const gchar *name;
  86
  87     if (!(transforms = gst_element_factory_list_get_elements(type, GST_RANK_MARGINAL)))
  88         goto done;
  89
  90     tmp = gst_element_factory_list_filter(transforms, src_caps, GST_PAD_SINK, FALSE);
  91     gst_plugin_feature_list_free(transforms);
  92     if (!(transforms = tmp))
  93         goto done;
  94
  95     tmp = gst_element_factory_list_filter(transforms, sink_caps, GST_PAD_SRC, FALSE);
  96     gst_plugin_feature_list_free(transforms);
  97     if (!(transforms = tmp))
  98         goto done;
  99
 100     transforms = g_list_sort(transforms, gst_plugin_feature_rank_compare_func);
 101     for (tmp = transforms; tmp != NULL && element == NULL; tmp = tmp->next)
 102     {
 103         name = gst_plugin_feature_get_name(GST_PLUGIN_FEATURE(tmp->data));
 104
 105         if (!strcmp(name, "vaapidecodebin"))
 106         {
 107             /* vaapidecodebin adds asynchronicity which breaks wg_transform synchronous drain / flush
 108              * requirements. Ignore it and use VA-API decoders directly instead.
 109              */
 110             GST_WARNING("Ignoring vaapidecodebin decoder.");
 111             continue;
 112         }
 113
 114         if (!(element = gst_element_factory_create(GST_ELEMENT_FACTORY(tmp->data), NULL)))
 115             GST_WARNING("Failed to create %s element.", name);
 116     }
 117     gst_plugin_feature_list_free(transforms);
 118
 119 done:
 120     if (element)
 121     {
 122         GST_DEBUG("Created %s element %p.", name, element);
 123     }
 124     else
 125     {
 126         gchar *src_str = gst_caps_to_string(src_caps), *sink_str = gst_caps_to_string(sink_caps);
 127         GST_WARNING("Failed to create element matching caps %s / %s.", src_str, sink_str);
 128         g_free(sink_str);
 129         g_free(src_str);
 130     }
 131
 132     return element;
 133 }
 134
 135 bool append_element(GstElement *container, GstElement *element, GstElement **first, GstElement **last)
 136 {
 137     gchar *name = gst_element_get_name(element);
 138     bool success = false;
 139
 140     if (!gst_bin_add(GST_BIN(container), element) ||
 141             !gst_element_sync_state_with_parent(element) ||
 142             (*last && !gst_element_link(*last, element)))
 143     {
 144         GST_ERROR("Failed to link %s element.", name);
 145     }
 146     else
 147     {
 148         GST_DEBUG("Linked %s element %p.", name, element);
 149         if (!*first)
 150             *first = element;
 151         *last = element;
 152         success = true;
 153     }
 154
 155     g_free(name);
 156     return success;
 157 }
 158
 159 bool link_src_to_element(GstPad *src_pad, GstElement *element)
 160 {
 161     GstPadLinkReturn ret;
 162     GstPad *sink_pad;
 163
 164     if (!(sink_pad = gst_element_get_static_pad(element, "sink")))
 165     {
 166         gchar *name = gst_element_get_name(element);
 167         GST_ERROR("Failed to find sink pad on %s", name);
 168         g_free(name);
 169         return false;
 170     }
 171     if ((ret = gst_pad_link(src_pad, sink_pad)))
 172     {
 173         gchar *src_name = gst_pad_get_name(src_pad), *sink_name = gst_pad_get_name(sink_pad);
 174         GST_ERROR("Failed to link element pad %s with pad %s", src_name, sink_name);
 175         g_free(sink_name);
 176         g_free(src_name);
 177     }
 178     gst_object_unref(sink_pad);
 179     return !ret;
 180 }
 181
 182 bool link_element_to_sink(GstElement *element, GstPad *sink_pad)
 183 {
 184     GstPadLinkReturn ret;
 185     GstPad *src_pad;
 186
 187     if (!(src_pad = gst_element_get_static_pad(element, "src")))
 188     {
 189         gchar *name = gst_element_get_name(element);
 190         GST_ERROR("Failed to find src pad on %s", name);
 191         g_free(name);
 192         return false;
 193     }
 194     if ((ret = gst_pad_link(src_pad, sink_pad)))
 195     {
 196         gchar *src_name = gst_pad_get_name(src_pad), *sink_name = gst_pad_get_name(sink_pad);
 197         GST_ERROR("Failed to link pad %s with element pad %s", src_name, sink_name);
 198         g_free(sink_name);
 199         g_free(src_name);
 200     }
 201     gst_object_unref(src_pad);
 202     return !ret;
 203 }
 204
 205 NTSTATUS wg_init_gstreamer(void *arg)
 206 {
 207     char arg0[] = "wine";
 208     char arg1[] = "--gst-disable-registry-fork";
 209     char *args[] = {arg0, arg1, NULL};
 210     int argc = ARRAY_SIZE(args) - 1;
 211     char **argv = args;
 212     GError *err;
 213
 214     if (!gst_init_check(&argc, &argv, &err))
 215     {
 216         fprintf(stderr, "winegstreamer: failed to initialize GStreamer: %s\n", err->message);
 217         g_error_free(err);
 218         return STATUS_UNSUCCESSFUL;
 219     }
 220
 221     GST_DEBUG_CATEGORY_INIT(wine, "WINE", GST_DEBUG_FG_RED, "Wine GStreamer support");
 222
 223     GST_INFO("GStreamer library version %s; wine built with %d.%d.%d.",
 224             gst_version_string(), GST_VERSION_MAJOR, GST_VERSION_MINOR, GST_VERSION_MICRO);
 225     return STATUS_SUCCESS;
 226 }