mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-08 13:18:52 +00:00
multisocketsink now understands the new GstNetControlMessageMeta to allow sending control messages (ancillary data) with data when writing to Unix domain sockets. Thanks to glib's `GSocketControlMessage` abstraction the code introduced in this commit is entirely portable and doesn't introduce and additional dependencies or conditionally compiled code, even if it is unlikely to be of much use on non-UNIX systems.
330 lines
9.4 KiB
C
330 lines
9.4 KiB
C
/* GStreamer
|
|
* Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
|
|
* Copyright (C) <2004> Thomas Vander Stichele <thomas at apestaart dot org>
|
|
* Copyright (C) <2011> Collabora Ltd.
|
|
* Author: Sebastian Dröge <sebastian.droege@collabora.co.uk>
|
|
* Copyright (C) <2014> William Manley <will@williammanley.net>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 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
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library 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.
|
|
*/
|
|
|
|
/**
|
|
* SECTION:element-socketsrc
|
|
*
|
|
* Receive data from a socket.
|
|
*
|
|
* As compared to other elements:
|
|
*
|
|
* socketsrc can be considered a source counterpart to the #multisocketsink
|
|
* sink.
|
|
*
|
|
* socketsrc can also be considered a generalization of #tcpclientsrc and
|
|
* #tcpserversrc: it contains all the logic required to communicate over the
|
|
* socket but none of the logic for creating the sockets/establishing the
|
|
* connection in the first place, allowing the user to accomplish this
|
|
* externally in whatever manner they wish making it applicable to other types
|
|
* of sockets besides TCP.
|
|
*
|
|
* As compared to #fdsrc socketsrc is socket specific and deals with #GSocket
|
|
* objects rather than sockets via integer file-descriptors.
|
|
*
|
|
* @see_also: #multisocketsink
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <gst/gst-i18n-plugin.h>
|
|
#include <gst/net/gstnetcontrolmessagemeta.h>
|
|
#include "gstsocketsrc.h"
|
|
#include "gsttcp.h"
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (socketsrc_debug);
|
|
#define GST_CAT_DEFAULT socketsrc_debug
|
|
|
|
#define MAX_READ_SIZE 4 * 1024
|
|
|
|
|
|
static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",
|
|
GST_PAD_SRC,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS_ANY);
|
|
|
|
|
|
enum
|
|
{
|
|
PROP_0,
|
|
PROP_SOCKET,
|
|
};
|
|
|
|
enum
|
|
{
|
|
CONNECTION_CLOSED_BY_PEER,
|
|
LAST_SIGNAL
|
|
};
|
|
|
|
static guint gst_socket_src_signals[LAST_SIGNAL] = { 0 };
|
|
|
|
#define gst_socket_src_parent_class parent_class
|
|
G_DEFINE_TYPE (GstSocketSrc, gst_socket_src, GST_TYPE_PUSH_SRC);
|
|
|
|
|
|
static void gst_socket_src_finalize (GObject * gobject);
|
|
|
|
static GstFlowReturn gst_socket_src_fill (GstPushSrc * psrc,
|
|
GstBuffer * outbuf);
|
|
static gboolean gst_socket_src_unlock (GstBaseSrc * bsrc);
|
|
static gboolean gst_socket_src_unlock_stop (GstBaseSrc * bsrc);
|
|
|
|
static void gst_socket_src_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec);
|
|
static void gst_socket_src_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec);
|
|
|
|
#define SWAP(a, b) do { GSocket* _swap_tmp = a; a = b; b = _swap_tmp; } while (0);
|
|
|
|
static void
|
|
gst_socket_src_class_init (GstSocketSrcClass * klass)
|
|
{
|
|
GObjectClass *gobject_class;
|
|
GstElementClass *gstelement_class;
|
|
GstBaseSrcClass *gstbasesrc_class;
|
|
GstPushSrcClass *gstpush_src_class;
|
|
|
|
gobject_class = (GObjectClass *) klass;
|
|
gstelement_class = (GstElementClass *) klass;
|
|
gstbasesrc_class = (GstBaseSrcClass *) klass;
|
|
gstpush_src_class = (GstPushSrcClass *) klass;
|
|
|
|
gobject_class->set_property = gst_socket_src_set_property;
|
|
gobject_class->get_property = gst_socket_src_get_property;
|
|
gobject_class->finalize = gst_socket_src_finalize;
|
|
|
|
g_object_class_install_property (gobject_class, PROP_SOCKET,
|
|
g_param_spec_object ("socket", "Socket",
|
|
"The socket to receive packets from", G_TYPE_SOCKET,
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
gst_socket_src_signals[CONNECTION_CLOSED_BY_PEER] =
|
|
g_signal_new ("connection-closed-by-peer", G_TYPE_FROM_CLASS (klass),
|
|
G_SIGNAL_RUN_FIRST, G_STRUCT_OFFSET (GstSocketSrcClass,
|
|
connection_closed_by_peer), NULL, NULL, NULL, G_TYPE_NONE, 0);
|
|
|
|
gst_element_class_add_pad_template (gstelement_class,
|
|
gst_static_pad_template_get (&srctemplate));
|
|
|
|
gst_element_class_set_static_metadata (gstelement_class,
|
|
"socket source", "Source/Network",
|
|
"Receive data from a socket",
|
|
"Thomas Vander Stichele <thomas at apestaart dot org>, "
|
|
"William Manley <will@williammanley.net>");
|
|
|
|
gstbasesrc_class->unlock = gst_socket_src_unlock;
|
|
gstbasesrc_class->unlock_stop = gst_socket_src_unlock_stop;
|
|
|
|
gstpush_src_class->fill = gst_socket_src_fill;
|
|
|
|
GST_DEBUG_CATEGORY_INIT (socketsrc_debug, "socketsrc", 0, "Socket Source");
|
|
}
|
|
|
|
static void
|
|
gst_socket_src_init (GstSocketSrc * this)
|
|
{
|
|
this->socket = NULL;
|
|
this->cancellable = g_cancellable_new ();
|
|
}
|
|
|
|
static void
|
|
gst_socket_src_finalize (GObject * gobject)
|
|
{
|
|
GstSocketSrc *this = GST_SOCKET_SRC (gobject);
|
|
|
|
g_clear_object (&this->cancellable);
|
|
g_clear_object (&this->socket);
|
|
|
|
G_OBJECT_CLASS (parent_class)->finalize (gobject);
|
|
}
|
|
|
|
static GstFlowReturn
|
|
gst_socket_src_fill (GstPushSrc * psrc, GstBuffer * outbuf)
|
|
{
|
|
GstSocketSrc *src;
|
|
GstFlowReturn ret = GST_FLOW_OK;
|
|
gssize rret;
|
|
GError *err = NULL;
|
|
GstMapInfo map;
|
|
GSocket *socket = NULL;
|
|
GSocketControlMessage **messages = NULL;
|
|
gint num_messages = 0;
|
|
gint i;
|
|
GInputVector ivec;
|
|
gint flags = 0;
|
|
|
|
src = GST_SOCKET_SRC (psrc);
|
|
|
|
GST_OBJECT_LOCK (src);
|
|
|
|
if (src->socket)
|
|
socket = g_object_ref (src->socket);
|
|
|
|
GST_OBJECT_UNLOCK (src);
|
|
|
|
if (socket == NULL)
|
|
goto no_socket;
|
|
|
|
GST_LOG_OBJECT (src, "asked for a buffer");
|
|
|
|
retry:
|
|
gst_buffer_map (outbuf, &map, GST_MAP_READWRITE);
|
|
ivec.buffer = map.data;
|
|
ivec.size = map.size;
|
|
rret =
|
|
g_socket_receive_message (socket, NULL, &ivec, 1, &messages,
|
|
&num_messages, &flags, src->cancellable, &err);
|
|
gst_buffer_unmap (outbuf, &map);
|
|
|
|
for (i = 0; i < num_messages; i++) {
|
|
gst_buffer_add_net_control_message_meta (outbuf, messages[i]);
|
|
g_object_unref (messages[i]);
|
|
messages[i] = NULL;
|
|
}
|
|
g_free (messages);
|
|
|
|
if (rret == 0) {
|
|
GSocket *tmp = NULL;
|
|
GST_DEBUG_OBJECT (src, "Received EOS on socket %p fd %i", socket,
|
|
g_socket_get_fd (socket));
|
|
|
|
/* We've hit EOS but we'll send this signal to allow someone to change
|
|
* our socket before we send EOS downstream. */
|
|
g_signal_emit (src, gst_socket_src_signals[CONNECTION_CLOSED_BY_PEER], 0);
|
|
|
|
GST_OBJECT_LOCK (src);
|
|
|
|
if (src->socket)
|
|
tmp = g_object_ref (src->socket);
|
|
|
|
GST_OBJECT_UNLOCK (src);
|
|
|
|
/* Do this dance with tmp to avoid unreffing with the lock held */
|
|
if (tmp != NULL && tmp != socket) {
|
|
SWAP (socket, tmp);
|
|
g_clear_object (&tmp);
|
|
|
|
GST_INFO_OBJECT (src, "New socket available after EOS %p fd %i: Retrying",
|
|
socket, g_socket_get_fd (socket));
|
|
|
|
/* retry with our new socket: */
|
|
goto retry;
|
|
} else {
|
|
g_clear_object (&tmp);
|
|
GST_INFO_OBJECT (src, "Forwarding EOS downstream");
|
|
ret = GST_FLOW_EOS;
|
|
}
|
|
} else if (rret < 0) {
|
|
if (g_error_matches (err, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
|
|
ret = GST_FLOW_FLUSHING;
|
|
GST_DEBUG_OBJECT (src, "Cancelled reading from socket");
|
|
} else {
|
|
ret = GST_FLOW_ERROR;
|
|
GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL),
|
|
("Failed to read from socket: %s", err->message));
|
|
}
|
|
} else {
|
|
ret = GST_FLOW_OK;
|
|
gst_buffer_resize (outbuf, 0, rret);
|
|
|
|
GST_LOG_OBJECT (src,
|
|
"Returning buffer from _get of size %" G_GSIZE_FORMAT ", ts %"
|
|
GST_TIME_FORMAT ", dur %" GST_TIME_FORMAT
|
|
", offset %" G_GINT64_FORMAT ", offset_end %" G_GINT64_FORMAT,
|
|
gst_buffer_get_size (outbuf),
|
|
GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (outbuf)),
|
|
GST_TIME_ARGS (GST_BUFFER_DURATION (outbuf)),
|
|
GST_BUFFER_OFFSET (outbuf), GST_BUFFER_OFFSET_END (outbuf));
|
|
}
|
|
g_clear_error (&err);
|
|
g_clear_object (&socket);
|
|
|
|
return ret;
|
|
|
|
no_socket:
|
|
{
|
|
GST_ELEMENT_ERROR (src, RESOURCE, NOT_FOUND, (NULL),
|
|
("Cannot receive: No socket set on socketsrc"));
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
}
|
|
|
|
static void
|
|
gst_socket_src_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstSocketSrc *socketsrc = GST_SOCKET_SRC (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_SOCKET:{
|
|
GSocket *socket = G_SOCKET (g_value_dup_object (value));
|
|
GST_OBJECT_LOCK (socketsrc);
|
|
SWAP (socket, socketsrc->socket);
|
|
GST_OBJECT_UNLOCK (socketsrc);
|
|
g_clear_object (&socket);
|
|
break;
|
|
}
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
gst_socket_src_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstSocketSrc *socketsrc = GST_SOCKET_SRC (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_SOCKET:
|
|
g_value_set_object (value, socketsrc->socket);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static gboolean
|
|
gst_socket_src_unlock (GstBaseSrc * bsrc)
|
|
{
|
|
GstSocketSrc *src = GST_SOCKET_SRC (bsrc);
|
|
|
|
GST_DEBUG_OBJECT (src, "set to flushing");
|
|
g_cancellable_cancel (src->cancellable);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
gst_socket_src_unlock_stop (GstBaseSrc * bsrc)
|
|
{
|
|
GstSocketSrc *src = GST_SOCKET_SRC (bsrc);
|
|
|
|
GST_DEBUG_OBJECT (src, "unset flushing");
|
|
g_cancellable_reset (src->cancellable);
|
|
|
|
return TRUE;
|
|
}
|