WebKitWebSourceGStreamer.cpp [plain text]
#include "config.h"
#include "WebKitWebSourceGStreamer.h"
#if ENABLE(VIDEO) && USE(GSTREAMER)
#include "GRefPtrGStreamer.h"
#include "GStreamerUtilities.h"
#include "GUniquePtrGStreamer.h"
#include "HTTPHeaderNames.h"
#include "MainThreadNotifier.h"
#include "MediaPlayer.h"
#include "NotImplemented.h"
#include "PlatformMediaResourceLoader.h"
#include "ResourceError.h"
#include "ResourceHandle.h"
#include "ResourceHandleClient.h"
#include "ResourceRequest.h"
#include "ResourceResponse.h"
#include "SharedBuffer.h"
#include <gst/app/gstappsrc.h>
#include <gst/gst.h>
#include <gst/pbutils/missing-plugins.h>
#include <wtf/MainThread.h>
#include <wtf/Noncopyable.h>
#include <wtf/glib/GMutexLocker.h>
#include <wtf/glib/GRefPtr.h>
#include <wtf/glib/GUniquePtr.h>
#include <wtf/text/CString.h>
#if USE(SOUP)
#include "SoupNetworkSession.h"
#endif
using namespace WebCore;
class StreamingClient {
public:
StreamingClient(WebKitWebSrc*, ResourceRequest&&);
virtual ~StreamingClient();
protected:
char* createReadBuffer(size_t requestedSize, size_t& actualSize);
void handleResponseReceived(const ResourceResponse&);
void handleDataReceived(const char*, int);
void handleNotifyFinished();
GRefPtr<GstElement> m_src;
ResourceRequest m_request;
};
class CachedResourceStreamingClient final : public PlatformMediaResourceClient, public StreamingClient {
WTF_MAKE_NONCOPYABLE(CachedResourceStreamingClient);
public:
CachedResourceStreamingClient(WebKitWebSrc*, ResourceRequest&&);
virtual ~CachedResourceStreamingClient();
private:
#if USE(SOUP)
char* getOrCreateReadBuffer(PlatformMediaResource&, size_t requestedSize, size_t& actualSize) override;
#endif
void responseReceived(PlatformMediaResource&, const ResourceResponse&) override;
void dataReceived(PlatformMediaResource&, const char*, int) override;
void accessControlCheckFailed(PlatformMediaResource&, const ResourceError&) override;
void loadFailed(PlatformMediaResource&, const ResourceError&) override;
void loadFinished(PlatformMediaResource&) override;
};
class ResourceHandleStreamingClient : public ThreadSafeRefCounted<ResourceHandleStreamingClient>, public ResourceHandleClient, public StreamingClient {
public:
static Ref<ResourceHandleStreamingClient> create(WebKitWebSrc* src, ResourceRequest&& request)
{
return adoptRef(*new ResourceHandleStreamingClient(src, WTFMove(request)));
}
virtual ~ResourceHandleStreamingClient();
void invalidate();
bool loadFailed() const;
void setDefersLoading(bool);
private:
ResourceHandleStreamingClient(WebKitWebSrc*, ResourceRequest&&);
void cleanupAndStopRunLoop();
#if USE(SOUP)
char* getOrCreateReadBuffer(size_t requestedSize, size_t& actualSize) override;
#endif
ResourceRequest willSendRequest(ResourceHandle*, ResourceRequest&&, ResourceResponse&&) override;
void didReceiveResponse(ResourceHandle*, ResourceResponse&&) override;
void didReceiveData(ResourceHandle*, const char*, unsigned, int) override;
void didReceiveBuffer(ResourceHandle*, Ref<SharedBuffer>&&, int encodedLength) override;
void didFinishLoading(ResourceHandle*) override;
void didFail(ResourceHandle*, const ResourceError&) override;
void wasBlocked(ResourceHandle*) override;
void cannotShowURL(ResourceHandle*) override;
RefPtr<Thread> m_thread;
Lock m_initializeRunLoopConditionMutex;
Condition m_initializeRunLoopCondition;
RunLoop* m_runLoop { nullptr };
Lock m_terminateRunLoopConditionMutex;
Condition m_terminateRunLoopCondition;
RefPtr<ResourceHandle> m_resource;
#if USE(SOUP)
std::unique_ptr<SoupNetworkSession> m_session;
#endif
};
enum MainThreadSourceNotification {
Start = 1 << 0,
Stop = 1 << 1,
NeedData = 1 << 2,
EnoughData = 1 << 3,
Seek = 1 << 4
};
#define WEBKIT_WEB_SRC_GET_PRIVATE(obj) (G_TYPE_INSTANCE_GET_PRIVATE((obj), WEBKIT_TYPE_WEB_SRC, WebKitWebSrcPrivate))
struct _WebKitWebSrcPrivate {
GstAppSrc* appsrc;
GstPad* srcpad;
CString originalURI;
CString redirectedURI;
bool keepAlive;
GUniquePtr<GstStructure> extraHeaders;
bool compress;
GUniquePtr<gchar> httpMethod;
WebCore::MediaPlayer* player;
RefPtr<PlatformMediaResourceLoader> loader;
RefPtr<PlatformMediaResource> resource;
RefPtr<ResourceHandleStreamingClient> client;
bool didPassAccessControlCheck;
guint64 offset;
guint64 size;
gboolean seekable;
bool paused;
bool isSeeking;
guint64 requestedOffset;
bool createdInMainThread;
RefPtr<MainThreadNotifier<MainThreadSourceNotification>> notifier;
GRefPtr<GstBuffer> buffer;
};
enum {
PROP_0,
PROP_LOCATION,
PROP_RESOLVED_LOCATION,
PROP_KEEP_ALIVE,
PROP_EXTRA_HEADERS,
PROP_COMPRESS,
PROP_METHOD
};
static GstStaticPadTemplate srcTemplate = GST_STATIC_PAD_TEMPLATE("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS_ANY);
GST_DEBUG_CATEGORY_STATIC(webkit_web_src_debug);
#define GST_CAT_DEFAULT webkit_web_src_debug
static void webKitWebSrcUriHandlerInit(gpointer gIface, gpointer ifaceData);
static void webKitWebSrcDispose(GObject*);
static void webKitWebSrcFinalize(GObject*);
static void webKitWebSrcSetProperty(GObject*, guint propertyID, const GValue*, GParamSpec*);
static void webKitWebSrcGetProperty(GObject*, guint propertyID, GValue*, GParamSpec*);
static GstStateChangeReturn webKitWebSrcChangeState(GstElement*, GstStateChange);
static gboolean webKitWebSrcQueryWithParent(GstPad*, GstObject*, GstQuery*);
static void webKitWebSrcNeedData(WebKitWebSrc*);
static void webKitWebSrcEnoughData(WebKitWebSrc*);
static gboolean webKitWebSrcSeek(WebKitWebSrc*, guint64);
static GstAppSrcCallbacks appsrcCallbacks = {
[](GstAppSrc*, guint, gpointer userData) {
webKitWebSrcNeedData(WEBKIT_WEB_SRC(userData));
},
[](GstAppSrc*, gpointer userData) {
webKitWebSrcEnoughData(WEBKIT_WEB_SRC(userData));
},
[](GstAppSrc*, guint64 offset, gpointer userData) -> gboolean {
return webKitWebSrcSeek(WEBKIT_WEB_SRC(userData), offset);
},
{ nullptr }
};
#define webkit_web_src_parent_class parent_class
#define WEBKIT_WEB_SRC_CATEGORY_INIT GST_DEBUG_CATEGORY_INIT(webkit_web_src_debug, "webkitwebsrc", 0, "websrc element");
G_DEFINE_TYPE_WITH_CODE(WebKitWebSrc, webkit_web_src, GST_TYPE_BIN,
G_IMPLEMENT_INTERFACE(GST_TYPE_URI_HANDLER, webKitWebSrcUriHandlerInit);
WEBKIT_WEB_SRC_CATEGORY_INIT);
static void webkit_web_src_class_init(WebKitWebSrcClass* klass)
{
GObjectClass* oklass = G_OBJECT_CLASS(klass);
GstElementClass* eklass = GST_ELEMENT_CLASS(klass);
oklass->dispose = webKitWebSrcDispose;
oklass->finalize = webKitWebSrcFinalize;
oklass->set_property = webKitWebSrcSetProperty;
oklass->get_property = webKitWebSrcGetProperty;
gst_element_class_add_pad_template(eklass,
gst_static_pad_template_get(&srcTemplate));
gst_element_class_set_metadata(eklass, "WebKit Web source element", "Source", "Handles HTTP/HTTPS uris",
"Sebastian Dröge <sebastian.droege@collabora.co.uk>");
g_object_class_install_property(oklass, PROP_LOCATION,
g_param_spec_string("location", "location", "Location to read from",
nullptr, static_cast<GParamFlags>(G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
g_object_class_install_property(oklass, PROP_RESOLVED_LOCATION,
g_param_spec_string("resolved-location", "Resolved location", "The location resolved by the server",
nullptr, static_cast<GParamFlags>(G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)));
g_object_class_install_property(oklass, PROP_KEEP_ALIVE,
g_param_spec_boolean("keep-alive", "keep-alive", "Use HTTP persistent connections",
FALSE, static_cast<GParamFlags>(G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
g_object_class_install_property(oklass, PROP_EXTRA_HEADERS,
g_param_spec_boxed("extra-headers", "Extra Headers", "Extra headers to append to the HTTP request",
GST_TYPE_STRUCTURE, static_cast<GParamFlags>(G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
g_object_class_install_property(oklass, PROP_COMPRESS,
g_param_spec_boolean("compress", "Compress", "Allow compressed content encodings",
FALSE, static_cast<GParamFlags>(G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
g_object_class_install_property(oklass, PROP_METHOD,
g_param_spec_string("method", "method", "The HTTP method to use (default: GET)",
nullptr, static_cast<GParamFlags>(G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
eklass->change_state = webKitWebSrcChangeState;
g_type_class_add_private(klass, sizeof(WebKitWebSrcPrivate));
}
static void webkit_web_src_init(WebKitWebSrc* src)
{
WebKitWebSrcPrivate* priv = WEBKIT_WEB_SRC_GET_PRIVATE(src);
src->priv = priv;
new (priv) WebKitWebSrcPrivate();
priv->createdInMainThread = isMainThread();
priv->notifier = MainThreadNotifier<MainThreadSourceNotification>::create();
priv->appsrc = GST_APP_SRC(gst_element_factory_make("appsrc", nullptr));
if (!priv->appsrc) {
GST_ERROR_OBJECT(src, "Failed to create appsrc");
return;
}
gst_bin_add(GST_BIN(src), GST_ELEMENT(priv->appsrc));
GRefPtr<GstPad> targetPad = adoptGRef(gst_element_get_static_pad(GST_ELEMENT(priv->appsrc), "src"));
priv->srcpad = webkitGstGhostPadFromStaticTemplate(&srcTemplate, "src", targetPad.get());
gst_element_add_pad(GST_ELEMENT(src), priv->srcpad);
GST_OBJECT_FLAG_SET(priv->srcpad, GST_PAD_FLAG_NEED_PARENT);
gst_pad_set_query_function(priv->srcpad, webKitWebSrcQueryWithParent);
gst_app_src_set_callbacks(priv->appsrc, &appsrcCallbacks, src, nullptr);
gst_app_src_set_emit_signals(priv->appsrc, FALSE);
gst_app_src_set_stream_type(priv->appsrc, GST_APP_STREAM_TYPE_SEEKABLE);
gst_app_src_set_max_bytes(priv->appsrc, 512 * 1024);
g_object_set(priv->appsrc, "min-percent", 20, nullptr);
gst_base_src_set_automatic_eos(GST_BASE_SRC(priv->appsrc), FALSE);
gst_app_src_set_caps(priv->appsrc, nullptr);
gst_app_src_set_size(priv->appsrc, -1);
}
static void webKitWebSrcDispose(GObject* object)
{
WebKitWebSrcPrivate* priv = WEBKIT_WEB_SRC(object)->priv;
if (priv->notifier) {
priv->notifier->invalidate();
priv->notifier = nullptr;
}
priv->player = nullptr;
GST_CALL_PARENT(G_OBJECT_CLASS, dispose, (object));
}
static void webKitWebSrcFinalize(GObject* object)
{
WebKitWebSrcPrivate* priv = WEBKIT_WEB_SRC(object)->priv;
priv->~WebKitWebSrcPrivate();
GST_CALL_PARENT(G_OBJECT_CLASS, finalize, (object));
}
static void webKitWebSrcSetProperty(GObject* object, guint propID, const GValue* value, GParamSpec* pspec)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(object);
switch (propID) {
case PROP_LOCATION:
gst_uri_handler_set_uri(reinterpret_cast<GstURIHandler*>(src), g_value_get_string(value), nullptr);
break;
case PROP_KEEP_ALIVE:
src->priv->keepAlive = g_value_get_boolean(value);
break;
case PROP_EXTRA_HEADERS: {
const GstStructure* s = gst_value_get_structure(value);
src->priv->extraHeaders.reset(s ? gst_structure_copy(s) : nullptr);
break;
}
case PROP_COMPRESS:
src->priv->compress = g_value_get_boolean(value);
break;
case PROP_METHOD:
src->priv->httpMethod.reset(g_value_dup_string(value));
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID(object, propID, pspec);
break;
}
}
static void webKitWebSrcGetProperty(GObject* object, guint propID, GValue* value, GParamSpec* pspec)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(object);
WebKitWebSrcPrivate* priv = src->priv;
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
switch (propID) {
case PROP_LOCATION:
g_value_set_string(value, priv->originalURI.data());
break;
case PROP_RESOLVED_LOCATION:
g_value_set_string(value, priv->redirectedURI.isNull() ? priv->originalURI.data() : priv->redirectedURI.data());
break;
case PROP_KEEP_ALIVE:
g_value_set_boolean(value, priv->keepAlive);
break;
case PROP_EXTRA_HEADERS:
gst_value_set_structure(value, priv->extraHeaders.get());
break;
case PROP_COMPRESS:
g_value_set_boolean(value, priv->compress);
break;
case PROP_METHOD:
g_value_set_string(value, priv->httpMethod.get());
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID(object, propID, pspec);
break;
}
}
static void webKitWebSrcStop(WebKitWebSrc* src)
{
WebKitWebSrcPrivate* priv = src->priv;
if (priv->resource || (priv->loader && !priv->keepAlive)) {
GRefPtr<WebKitWebSrc> protector = WTF::ensureGRef(src);
priv->notifier->cancelPendingNotifications(MainThreadSourceNotification::NeedData | MainThreadSourceNotification::EnoughData | MainThreadSourceNotification::Seek);
priv->notifier->notify(MainThreadSourceNotification::Stop, [protector, keepAlive = priv->keepAlive] {
WebKitWebSrcPrivate* priv = protector->priv;
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(protector.get()));
if (priv->resource) {
priv->resource->stop();
priv->resource->setClient(nullptr);
priv->resource = nullptr;
}
if (!keepAlive)
priv->loader = nullptr;
});
}
if (priv->client) {
priv->client->invalidate();
priv->client = nullptr;
}
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
bool wasSeeking = std::exchange(priv->isSeeking, false);
if (priv->buffer) {
unmapGstBuffer(priv->buffer.get());
priv->buffer.clear();
}
priv->paused = false;
priv->offset = 0;
priv->seekable = FALSE;
if (!wasSeeking) {
priv->size = 0;
priv->requestedOffset = 0;
priv->player = nullptr;
}
locker.unlock();
if (priv->appsrc) {
gst_app_src_set_caps(priv->appsrc, nullptr);
if (!wasSeeking)
gst_app_src_set_size(priv->appsrc, -1);
}
GST_DEBUG_OBJECT(src, "Stopped request");
}
static bool webKitWebSrcSetExtraHeader(GQuark fieldId, const GValue* value, gpointer userData)
{
GUniquePtr<gchar> fieldContent;
if (G_VALUE_HOLDS_STRING(value))
fieldContent.reset(g_value_dup_string(value));
else {
GValue dest = G_VALUE_INIT;
g_value_init(&dest, G_TYPE_STRING);
if (g_value_transform(value, &dest))
fieldContent.reset(g_value_dup_string(&dest));
}
const gchar* fieldName = g_quark_to_string(fieldId);
if (!fieldContent.get()) {
GST_ERROR("extra-headers field '%s' contains no value or can't be converted to a string", fieldName);
return false;
}
GST_DEBUG("Appending extra header: \"%s: %s\"", fieldName, fieldContent.get());
ResourceRequest* request = static_cast<ResourceRequest*>(userData);
request->setHTTPHeaderField(fieldName, fieldContent.get());
return true;
}
static gboolean webKitWebSrcProcessExtraHeaders(GQuark fieldId, const GValue* value, gpointer userData)
{
if (G_VALUE_TYPE(value) == GST_TYPE_ARRAY) {
unsigned size = gst_value_array_get_size(value);
for (unsigned i = 0; i < size; i++) {
if (!webKitWebSrcSetExtraHeader(fieldId, gst_value_array_get_value(value, i), userData))
return FALSE;
}
return TRUE;
}
if (G_VALUE_TYPE(value) == GST_TYPE_LIST) {
unsigned size = gst_value_list_get_size(value);
for (unsigned i = 0; i < size; i++) {
if (!webKitWebSrcSetExtraHeader(fieldId, gst_value_list_get_value(value, i), userData))
return FALSE;
}
return TRUE;
}
return webKitWebSrcSetExtraHeader(fieldId, value, userData);
}
static void webKitWebSrcStart(WebKitWebSrc* src)
{
WebKitWebSrcPrivate* priv = src->priv;
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
priv->didPassAccessControlCheck = false;
if (priv->originalURI.isNull()) {
GST_ERROR_OBJECT(src, "No URI provided");
locker.unlock();
webKitWebSrcStop(src);
return;
}
ASSERT(!priv->client);
GST_DEBUG_OBJECT(src, "Fetching %s", priv->originalURI.data());
URL url = URL(URL(), priv->originalURI.data());
ResourceRequest request(url);
request.setAllowCookies(true);
request.setFirstPartyForCookies(url);
priv->size = 0;
if (priv->player)
request.setHTTPReferrer(priv->player->referrer());
if (priv->httpMethod.get())
request.setHTTPMethod(priv->httpMethod.get());
#if USE(SOUP)
if (!priv->compress)
request.setAcceptEncoding(false);
#endif
if (!g_ascii_strcasecmp("movies.apple.com", url.host().utf8().data())
|| !g_ascii_strcasecmp("trailers.apple.com", url.host().utf8().data()))
request.setHTTPUserAgent("Quicktime/7.6.6");
if (priv->requestedOffset) {
GUniquePtr<gchar> val(g_strdup_printf("bytes=%" G_GUINT64_FORMAT "-", priv->requestedOffset));
request.setHTTPHeaderField(HTTPHeaderName::Range, val.get());
}
priv->offset = priv->requestedOffset;
if (!priv->keepAlive) {
GST_DEBUG_OBJECT(src, "Persistent connection support disabled");
request.setHTTPHeaderField(HTTPHeaderName::Connection, "close");
}
if (priv->extraHeaders)
gst_structure_foreach(priv->extraHeaders.get(), webKitWebSrcProcessExtraHeaders, &request);
request.setHTTPHeaderField(HTTPHeaderName::IcyMetadata, "1");
if (!priv->player || !priv->createdInMainThread) {
priv->client = ResourceHandleStreamingClient::create(src, WTFMove(request));
if (priv->client->loadFailed()) {
GST_ERROR_OBJECT(src, "Failed to setup streaming client");
locker.unlock();
webKitWebSrcStop(src);
} else
GST_DEBUG_OBJECT(src, "Started request");
return;
}
locker.unlock();
GRefPtr<WebKitWebSrc> protector = WTF::ensureGRef(src);
priv->notifier->notify(MainThreadSourceNotification::Start, [protector, request = WTFMove(request)] {
WebKitWebSrcPrivate* priv = protector->priv;
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(protector.get()));
if (!priv->loader)
priv->loader = priv->player->createResourceLoader();
PlatformMediaResourceLoader::LoadOptions loadOptions = 0;
if (request.url().protocolIsBlob())
loadOptions |= PlatformMediaResourceLoader::LoadOption::BufferData;
priv->resource = priv->loader->requestResource(ResourceRequest(request), loadOptions);
if (priv->resource) {
priv->resource->setClient(std::make_unique<CachedResourceStreamingClient>(protector.get(), ResourceRequest(request)));
GST_DEBUG_OBJECT(protector.get(), "Started request");
} else {
GST_ERROR_OBJECT(protector.get(), "Failed to setup streaming client");
priv->loader = nullptr;
locker.unlock();
webKitWebSrcStop(protector.get());
}
});
}
static GstStateChangeReturn webKitWebSrcChangeState(GstElement* element, GstStateChange transition)
{
GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
WebKitWebSrc* src = WEBKIT_WEB_SRC(element);
WebKitWebSrcPrivate* priv = src->priv;
switch (transition) {
case GST_STATE_CHANGE_NULL_TO_READY:
if (!priv->appsrc) {
gst_element_post_message(element,
gst_missing_element_message_new(element, "appsrc"));
GST_ELEMENT_ERROR(src, CORE, MISSING_PLUGIN, (nullptr), ("no appsrc"));
return GST_STATE_CHANGE_FAILURE;
}
break;
default:
break;
}
ret = GST_ELEMENT_CLASS(parent_class)->change_state(element, transition);
if (G_UNLIKELY(ret == GST_STATE_CHANGE_FAILURE)) {
GST_DEBUG_OBJECT(src, "State change failed");
return ret;
}
switch (transition) {
case GST_STATE_CHANGE_READY_TO_PAUSED:
{
GST_DEBUG_OBJECT(src, "READY->PAUSED");
webKitWebSrcStart(src);
break;
}
case GST_STATE_CHANGE_PAUSED_TO_READY:
{
GST_DEBUG_OBJECT(src, "PAUSED->READY");
webKitWebSrcStop(src);
break;
}
default:
break;
}
return ret;
}
static gboolean webKitWebSrcQueryWithParent(GstPad* pad, GstObject* parent, GstQuery* query)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(GST_ELEMENT(parent));
gboolean result = FALSE;
switch (GST_QUERY_TYPE(query)) {
case GST_QUERY_DURATION: {
GstFormat format;
gst_query_parse_duration(query, &format, nullptr);
GST_DEBUG_OBJECT(src, "duration query in format %s", gst_format_get_name(format));
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
if (format == GST_FORMAT_BYTES && src->priv->size > 0) {
gst_query_set_duration(query, format, src->priv->size);
result = TRUE;
}
break;
}
case GST_QUERY_URI: {
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
gst_query_set_uri(query, src->priv->originalURI.data());
if (!src->priv->redirectedURI.isNull())
gst_query_set_uri_redirection(query, src->priv->redirectedURI.data());
result = TRUE;
break;
}
case GST_QUERY_SCHEDULING: {
GstSchedulingFlags flags;
int minSize, maxSize, align;
gst_query_parse_scheduling(query, &flags, &minSize, &maxSize, &align);
gst_query_set_scheduling(query, static_cast<GstSchedulingFlags>(flags | GST_SCHEDULING_FLAG_BANDWIDTH_LIMITED), minSize, maxSize, align);
result = TRUE;
break;
}
default: {
GRefPtr<GstPad> target = adoptGRef(gst_ghost_pad_get_target(GST_GHOST_PAD_CAST(pad)));
if (target)
result = gst_pad_query(target.get(), query);
break;
}
}
return result;
}
static bool urlHasSupportedProtocol(const URL& url)
{
return url.isValid() && (url.protocolIsInHTTPFamily() || url.protocolIsBlob());
}
static GstURIType webKitWebSrcUriGetType(GType)
{
return GST_URI_SRC;
}
const gchar* const* webKitWebSrcGetProtocols(GType)
{
static const char* protocols[] = {"webkit+http", "webkit+https", "blob", nullptr };
return protocols;
}
static URL convertPlaybinURI(const char* uriString)
{
URL url(URL(), uriString);
ASSERT(url.protocol().substring(0, 7) == "webkit+");
url.setProtocol(url.protocol().substring(7).toString());
return url;
}
static gchar* webKitWebSrcGetUri(GstURIHandler* handler)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(handler);
gchar* ret;
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
ret = g_strdup(src->priv->originalURI.data());
return ret;
}
static gboolean webKitWebSrcSetUri(GstURIHandler* handler, const gchar* uri, GError** error)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(handler);
WebKitWebSrcPrivate* priv = src->priv;
if (GST_STATE(src) >= GST_STATE_PAUSED) {
GST_ERROR_OBJECT(src, "URI can only be set in states < PAUSED");
return FALSE;
}
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
priv->redirectedURI = CString();
priv->originalURI = CString();
if (!uri)
return TRUE;
URL url = convertPlaybinURI(uri);
if (!urlHasSupportedProtocol(url)) {
g_set_error(error, GST_URI_ERROR, GST_URI_ERROR_BAD_URI, "Invalid URI '%s'", uri);
return FALSE;
}
priv->originalURI = url.string().utf8();
return TRUE;
}
static void webKitWebSrcUriHandlerInit(gpointer gIface, gpointer)
{
GstURIHandlerInterface* iface = (GstURIHandlerInterface *) gIface;
iface->get_type = webKitWebSrcUriGetType;
iface->get_protocols = webKitWebSrcGetProtocols;
iface->get_uri = webKitWebSrcGetUri;
iface->set_uri = webKitWebSrcSetUri;
}
static void webKitWebSrcNeedData(WebKitWebSrc* src)
{
WebKitWebSrcPrivate* priv = src->priv;
GST_DEBUG_OBJECT(src, "Need more data");
{
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
if (!priv->paused)
return;
priv->paused = false;
if (priv->client) {
priv->client->setDefersLoading(false);
return;
}
}
GRefPtr<WebKitWebSrc> protector = WTF::ensureGRef(src);
priv->notifier->notify(MainThreadSourceNotification::NeedData, [protector] {
WebKitWebSrcPrivate* priv = protector->priv;
if (priv->resource)
priv->resource->setDefersLoading(false);
});
}
static void webKitWebSrcEnoughData(WebKitWebSrc* src)
{
WebKitWebSrcPrivate* priv = src->priv;
GST_DEBUG_OBJECT(src, "Have enough data");
{
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
if (priv->paused)
return;
priv->paused = true;
if (priv->client) {
priv->client->setDefersLoading(true);
return;
}
}
GRefPtr<WebKitWebSrc> protector = WTF::ensureGRef(src);
priv->notifier->notify(MainThreadSourceNotification::EnoughData, [protector] {
WebKitWebSrcPrivate* priv = protector->priv;
if (priv->resource)
priv->resource->setDefersLoading(true);
});
}
static gboolean webKitWebSrcSeek(WebKitWebSrc* src, guint64 offset)
{
WebKitWebSrcPrivate* priv = src->priv;
{
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
if (offset == priv->offset && priv->requestedOffset == priv->offset)
return TRUE;
if (!priv->seekable)
return FALSE;
priv->isSeeking = true;
priv->requestedOffset = offset;
}
GST_DEBUG_OBJECT(src, "Seeking to offset: %" G_GUINT64_FORMAT, src->priv->requestedOffset);
if (priv->client) {
webKitWebSrcStop(src);
webKitWebSrcStart(src);
return TRUE;
}
GRefPtr<WebKitWebSrc> protector = WTF::ensureGRef(src);
priv->notifier->notify(MainThreadSourceNotification::Seek, [protector] {
webKitWebSrcStop(protector.get());
webKitWebSrcStart(protector.get());
});
return TRUE;
}
void webKitWebSrcSetMediaPlayer(WebKitWebSrc* src, WebCore::MediaPlayer* player)
{
ASSERT(player);
ASSERT(src->priv->createdInMainThread);
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
src->priv->player = player;
}
bool webKitSrcPassedCORSAccessCheck(WebKitWebSrc* src)
{
return src->priv->didPassAccessControlCheck;
}
StreamingClient::StreamingClient(WebKitWebSrc* src, ResourceRequest&& request)
: m_src(GST_ELEMENT(src))
, m_request(WTFMove(request))
{
}
StreamingClient::~StreamingClient()
{
}
char* StreamingClient::createReadBuffer(size_t requestedSize, size_t& actualSize)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
WebKitWebSrcPrivate* priv = src->priv;
ASSERT(!priv->buffer);
GstBuffer* buffer = gst_buffer_new_and_alloc(requestedSize);
mapGstBuffer(buffer, GST_MAP_WRITE);
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
priv->buffer = adoptGRef(buffer);
locker.unlock();
actualSize = gst_buffer_get_size(buffer);
return getGstBufferDataPointer(buffer);
}
void StreamingClient::handleResponseReceived(const ResourceResponse& response)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
WebKitWebSrcPrivate* priv = src->priv;
GST_DEBUG_OBJECT(src, "Received response: %d", response.httpStatusCode());
auto responseURI = response.url().string().utf8();
if (priv->originalURI != responseURI)
priv->redirectedURI = WTFMove(responseURI);
if (response.httpStatusCode() >= 400) {
GST_ELEMENT_ERROR(src, RESOURCE, READ, ("Received %d HTTP error code", response.httpStatusCode()), (nullptr));
gst_app_src_end_of_stream(priv->appsrc);
webKitWebSrcStop(src);
return;
}
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
if (priv->isSeeking) {
GST_DEBUG_OBJECT(src, "Seek in progress, ignoring response");
return;
}
if (priv->requestedOffset) {
if (response.httpStatusCode() == 200) {
priv->offset = 0;
} else if (response.httpStatusCode() != 206) {
locker.unlock();
GST_ELEMENT_ERROR(src, RESOURCE, READ, ("Received unexpected %d HTTP status code", response.httpStatusCode()), (nullptr));
gst_app_src_end_of_stream(priv->appsrc);
webKitWebSrcStop(src);
return;
}
}
long long length = response.expectedContentLength();
if (length > 0 && priv->requestedOffset && response.httpStatusCode() == 206)
length += priv->requestedOffset;
priv->size = length >= 0 ? length : 0;
priv->seekable = length > 0 && g_ascii_strcasecmp("none", response.httpHeaderField(HTTPHeaderName::AcceptRanges).utf8().data());
locker.unlock();
if (length > 0) {
gst_app_src_set_size(priv->appsrc, length);
} else
gst_app_src_set_size(priv->appsrc, -1);
GRefPtr<GstCaps> caps;
String metadataIntervalAsString = response.httpHeaderField(HTTPHeaderName::IcyMetaInt);
if (!metadataIntervalAsString.isEmpty()) {
bool isMetadataIntervalParsed;
int metadataInterval = metadataIntervalAsString.toInt(&isMetadataIntervalParsed);
if (isMetadataIntervalParsed && metadataInterval > 0)
caps = adoptGRef(gst_caps_new_simple("application/x-icy", "metadata-interval", G_TYPE_INT, metadataInterval, nullptr));
}
gst_app_src_set_caps(priv->appsrc, caps.get());
GstStructure* httpHeaders = gst_structure_new_empty("http-headers");
gst_structure_set(httpHeaders, "uri", G_TYPE_STRING, priv->originalURI.data(), nullptr);
if (!priv->redirectedURI.isNull())
gst_structure_set(httpHeaders, "redirection-uri", G_TYPE_STRING, priv->redirectedURI.data(), nullptr);
GUniquePtr<GstStructure> headers(gst_structure_new_empty("request-headers"));
for (const auto& header : m_request.httpHeaderFields())
gst_structure_set(headers.get(), header.key.utf8().data(), G_TYPE_STRING, header.value.utf8().data(), nullptr);
gst_structure_set(httpHeaders, "request-headers", GST_TYPE_STRUCTURE, headers.get(), nullptr);
headers.reset(gst_structure_new_empty("response-headers"));
for (const auto& header : response.httpHeaderFields())
gst_structure_set(headers.get(), header.key.utf8().data(), G_TYPE_STRING, header.value.utf8().data(), nullptr);
gst_structure_set(httpHeaders, "response-headers", GST_TYPE_STRUCTURE, headers.get(), nullptr);
gst_pad_push_event(GST_BASE_SRC_PAD(priv->appsrc), gst_event_new_custom(GST_EVENT_CUSTOM_DOWNSTREAM_STICKY, httpHeaders));
}
void StreamingClient::handleDataReceived(const char* data, int length)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
WebKitWebSrcPrivate* priv = src->priv;
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
GST_LOG_OBJECT(src, "Have %lld bytes of data", priv->buffer ? static_cast<long long>(gst_buffer_get_size(priv->buffer.get())) : length);
ASSERT(!priv->buffer || data == getGstBufferDataPointer(priv->buffer.get()));
if (priv->buffer)
unmapGstBuffer(priv->buffer.get());
if (priv->isSeeking) {
GST_DEBUG_OBJECT(src, "Seek in progress, ignoring data");
priv->buffer.clear();
return;
}
if (priv->offset < priv->requestedOffset) {
if (priv->offset + length <= priv->requestedOffset) {
priv->offset += length;
priv->buffer.clear();
return;
}
if (priv->offset + length > priv->requestedOffset) {
guint64 offset = priv->requestedOffset - priv->offset;
data += offset;
length -= offset;
if (priv->buffer)
gst_buffer_resize(priv->buffer.get(), offset, -1);
priv->offset = priv->requestedOffset;
}
priv->requestedOffset = 0;
}
if (!priv->buffer)
priv->buffer = adoptGRef(createGstBufferForData(data, length));
else
gst_buffer_set_size(priv->buffer.get(), static_cast<gssize>(length));
GST_BUFFER_OFFSET(priv->buffer.get()) = priv->offset;
if (priv->requestedOffset == priv->offset)
priv->requestedOffset += length;
priv->offset += length;
if (priv->size > 0 && priv->offset > priv->size) {
GST_DEBUG_OBJECT(src, "Updating internal size from %" G_GUINT64_FORMAT " to %" G_GUINT64_FORMAT, priv->size, priv->offset);
gst_app_src_set_size(priv->appsrc, priv->offset);
priv->size = priv->offset;
}
GST_BUFFER_OFFSET_END(priv->buffer.get()) = priv->offset;
locker.unlock();
GstFlowReturn ret = gst_app_src_push_buffer(priv->appsrc, priv->buffer.leakRef());
if (ret != GST_FLOW_OK && ret != GST_FLOW_EOS)
GST_ELEMENT_ERROR(src, CORE, FAILED, (nullptr), (nullptr));
}
void StreamingClient::handleNotifyFinished()
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
WebKitWebSrcPrivate* priv = src->priv;
GST_DEBUG_OBJECT(src, "Have EOS");
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
if (!priv->isSeeking) {
locker.unlock();
gst_app_src_end_of_stream(priv->appsrc);
}
}
CachedResourceStreamingClient::CachedResourceStreamingClient(WebKitWebSrc* src, ResourceRequest&& request)
: StreamingClient(src, WTFMove(request))
{
}
CachedResourceStreamingClient::~CachedResourceStreamingClient()
{
}
#if USE(SOUP)
char* CachedResourceStreamingClient::getOrCreateReadBuffer(PlatformMediaResource&, size_t requestedSize, size_t& actualSize)
{
return createReadBuffer(requestedSize, actualSize);
}
#endif
void CachedResourceStreamingClient::responseReceived(PlatformMediaResource&, const ResourceResponse& response)
{
WebKitWebSrcPrivate* priv = WEBKIT_WEB_SRC(m_src.get())->priv;
priv->didPassAccessControlCheck = priv->resource->didPassAccessControlCheck();
handleResponseReceived(response);
}
void CachedResourceStreamingClient::dataReceived(PlatformMediaResource&, const char* data, int length)
{
handleDataReceived(data, length);
}
void CachedResourceStreamingClient::accessControlCheckFailed(PlatformMediaResource&, const ResourceError& error)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
GST_ELEMENT_ERROR(src, RESOURCE, READ, ("%s", error.localizedDescription().utf8().data()), (nullptr));
gst_app_src_end_of_stream(src->priv->appsrc);
webKitWebSrcStop(src);
}
void CachedResourceStreamingClient::loadFailed(PlatformMediaResource&, const ResourceError& error)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
if (!error.isCancellation()) {
GST_ERROR_OBJECT(src, "Have failure: %s", error.localizedDescription().utf8().data());
GST_ELEMENT_ERROR(src, RESOURCE, FAILED, ("%s", error.localizedDescription().utf8().data()), (nullptr));
}
gst_app_src_end_of_stream(src->priv->appsrc);
}
void CachedResourceStreamingClient::loadFinished(PlatformMediaResource&)
{
handleNotifyFinished();
}
ResourceHandleStreamingClient::ResourceHandleStreamingClient(WebKitWebSrc* src, ResourceRequest&& request)
: StreamingClient(src, WTFMove(request))
{
LockHolder locker(m_initializeRunLoopConditionMutex);
m_thread = Thread::create("ResourceHandleStreamingClient", [this] {
{
LockHolder locker(m_initializeRunLoopConditionMutex);
m_runLoop = &RunLoop::current();
#if USE(SOUP)
m_session = std::make_unique<SoupNetworkSession>();
m_resource = ResourceHandle::create(*m_session, m_request, this, true, false);
#else
m_resource = ResourceHandle::create(nullptr, m_request, this, true, false);
#endif
m_initializeRunLoopCondition.notifyOne();
}
if (!m_resource)
return;
m_runLoop->dispatch([this] { m_resource->setDefersLoading(false); });
m_runLoop->run();
});
m_initializeRunLoopCondition.wait(m_initializeRunLoopConditionMutex);
}
ResourceHandleStreamingClient::~ResourceHandleStreamingClient()
{
if (m_thread) {
m_thread->detach();
m_thread = nullptr;
}
}
void ResourceHandleStreamingClient::cleanupAndStopRunLoop()
{
m_resource->clearClient();
m_resource->cancel();
m_resource = nullptr;
#if USE(SOUP)
m_session = nullptr;
#endif
m_runLoop->stop();
}
void ResourceHandleStreamingClient::invalidate()
{
if (m_runLoop == &RunLoop::current()) {
cleanupAndStopRunLoop();
return;
}
LockHolder locker(m_terminateRunLoopConditionMutex);
m_runLoop->dispatch([this, protectedThis = makeRef(*this)] {
cleanupAndStopRunLoop();
LockHolder locker(m_terminateRunLoopConditionMutex);
m_terminateRunLoopCondition.notifyOne();
});
m_terminateRunLoopCondition.wait(m_terminateRunLoopConditionMutex);
}
bool ResourceHandleStreamingClient::loadFailed() const
{
return !m_resource;
}
void ResourceHandleStreamingClient::setDefersLoading(bool defers)
{
m_runLoop->dispatch([this, protectedThis = makeRef(*this), defers] {
if (m_resource)
m_resource->setDefersLoading(defers);
});
}
#if USE(SOUP)
char* ResourceHandleStreamingClient::getOrCreateReadBuffer(size_t requestedSize, size_t& actualSize)
{
return createReadBuffer(requestedSize, actualSize);
}
#endif
ResourceRequest ResourceHandleStreamingClient::willSendRequest(ResourceHandle*, ResourceRequest&& request, ResourceResponse&&)
{
return WTFMove(request);
}
void ResourceHandleStreamingClient::didReceiveResponse(ResourceHandle*, ResourceResponse&& response)
{
if (m_resource)
handleResponseReceived(response);
}
void ResourceHandleStreamingClient::didReceiveData(ResourceHandle*, const char* , unsigned , int)
{
ASSERT_NOT_REACHED();
}
void ResourceHandleStreamingClient::didReceiveBuffer(ResourceHandle*, Ref<SharedBuffer>&& buffer, int )
{
if (!m_resource)
return;
for (const auto& element : buffer.get())
handleDataReceived(element.segment->data(), element.segment->size());
}
void ResourceHandleStreamingClient::didFinishLoading(ResourceHandle*)
{
if (m_resource)
handleNotifyFinished();
}
void ResourceHandleStreamingClient::didFail(ResourceHandle*, const ResourceError& error)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
GST_ERROR_OBJECT(src, "Have failure: %s", error.localizedDescription().utf8().data());
GST_ELEMENT_ERROR(src, RESOURCE, FAILED, ("%s", error.localizedDescription().utf8().data()), (nullptr));
gst_app_src_end_of_stream(src->priv->appsrc);
}
void ResourceHandleStreamingClient::wasBlocked(ResourceHandle*)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
GUniquePtr<gchar> uri;
GST_ERROR_OBJECT(src, "Request was blocked");
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
uri.reset(g_strdup(src->priv->originalURI.data()));
locker.unlock();
GST_ELEMENT_ERROR(src, RESOURCE, OPEN_READ, ("Access to \"%s\" was blocked", uri.get()), (nullptr));
}
void ResourceHandleStreamingClient::cannotShowURL(ResourceHandle*)
{
WebKitWebSrc* src = WEBKIT_WEB_SRC(m_src.get());
GUniquePtr<gchar> uri;
GST_ERROR_OBJECT(src, "Cannot show URL");
WTF::GMutexLocker<GMutex> locker(*GST_OBJECT_GET_LOCK(src));
uri.reset(g_strdup(src->priv->originalURI.data()));
locker.unlock();
GST_ELEMENT_ERROR(src, RESOURCE, OPEN_READ, ("Can't show \"%s\"", uri.get()), (nullptr));
}
#endif // USE(GSTREAMER)