mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-05 16:06:43 +00:00
36006c61e9
Without this patch the following pipeline would send packets containing garbage in the data section. $ gst-launch-1.0 avtpsrc ! avtpsink Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/1077>
310 lines
8.8 KiB
C
310 lines
8.8 KiB
C
/*
|
|
* GStreamer AVTP Plugin
|
|
* Copyright (C) 2019 Intel Corporation
|
|
*
|
|
* 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 Street, Fifth Floor,
|
|
* Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
/**
|
|
* SECTION:element-avtpsrc
|
|
* @see_also: avtpsink
|
|
*
|
|
* avtpsrc is a network source that receives AVTPDUs from the network. It
|
|
* should be combined with AVTP depayloaders to implement an AVTP listener.
|
|
* For more information see https://standards.ieee.org/standard/1722-2016.html.
|
|
*
|
|
* <note>
|
|
* Applications must have CAP_NET_RAW capability in order to successfully use
|
|
* this element. See avtpsink documentation for further information.
|
|
* </note>
|
|
*
|
|
* <refsect2>
|
|
* <title>Example pipeline</title>
|
|
* |[
|
|
* gst-launch-1.0 avtpsrc ! avtpaafdepay ! autoaudiosink
|
|
* ]| This example pipeline implements an AVTP listener that plays an AAF
|
|
* stream back.
|
|
* </refsect2>
|
|
*/
|
|
|
|
#include <arpa/inet.h>
|
|
#include <linux/if_packet.h>
|
|
#include <net/ethernet.h>
|
|
#include <net/if.h>
|
|
#include <stdio.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
|
|
#include "gstavtpsrc.h"
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (avtpsrc_debug);
|
|
#define GST_CAT_DEFAULT (avtpsrc_debug)
|
|
|
|
#define DEFAULT_IFNAME "eth0"
|
|
#define DEFAULT_ADDRESS "01:AA:AA:AA:AA:AA"
|
|
|
|
#define MAX_AVTPDU_SIZE 1500
|
|
|
|
enum
|
|
{
|
|
PROP_0,
|
|
PROP_IFNAME,
|
|
PROP_ADDRESS,
|
|
};
|
|
|
|
static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
|
|
GST_PAD_SRC,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS ("application/x-avtp")
|
|
);
|
|
|
|
#define gst_avtp_src_parent_class parent_class
|
|
G_DEFINE_TYPE (GstAvtpSrc, gst_avtp_src, GST_TYPE_PUSH_SRC);
|
|
GST_ELEMENT_REGISTER_DEFINE (avtpsrc, "avtpsrc", GST_RANK_NONE,
|
|
GST_TYPE_AVTP_SRC);
|
|
|
|
static void gst_avtp_src_finalize (GObject * gobject);
|
|
static void gst_avtp_src_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec);
|
|
static void gst_avtp_src_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec);
|
|
|
|
static gboolean gst_avtp_src_start (GstBaseSrc * basesrc);
|
|
static gboolean gst_avtp_src_stop (GstBaseSrc * basesrc);
|
|
static GstFlowReturn gst_avtp_src_fill (GstPushSrc * pushsrc, GstBuffer *
|
|
buffer);
|
|
|
|
static void
|
|
gst_avtp_src_class_init (GstAvtpSrcClass * klass)
|
|
{
|
|
GObjectClass *object_class = G_OBJECT_CLASS (klass);
|
|
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
|
|
GstBaseSrcClass *basesrc_class = GST_BASE_SRC_CLASS (klass);
|
|
GstPushSrcClass *pushsrc_class = GST_PUSH_SRC_CLASS (klass);
|
|
|
|
object_class->finalize = gst_avtp_src_finalize;
|
|
object_class->get_property = gst_avtp_src_get_property;
|
|
object_class->set_property = gst_avtp_src_set_property;
|
|
|
|
g_object_class_install_property (object_class, PROP_IFNAME,
|
|
g_param_spec_string ("ifname", "Interface Name",
|
|
"Network interface utilized to receive AVTPDUs",
|
|
DEFAULT_IFNAME, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS |
|
|
GST_PARAM_MUTABLE_READY));
|
|
g_object_class_install_property (object_class, PROP_ADDRESS,
|
|
g_param_spec_string ("address", "Destination MAC address",
|
|
"Destination MAC address to listen to",
|
|
DEFAULT_ADDRESS, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS |
|
|
GST_PARAM_MUTABLE_READY));
|
|
|
|
gst_element_class_add_static_pad_template (element_class, &src_template);
|
|
|
|
gst_element_class_set_static_metadata (element_class,
|
|
"Audio/Video Transport Protocol (AVTP) Source",
|
|
"Src/Network", "Receive AVTPDUs from the network",
|
|
"Andre Guedes <andre.guedes@intel.com>");
|
|
|
|
basesrc_class->start = GST_DEBUG_FUNCPTR (gst_avtp_src_start);
|
|
basesrc_class->stop = GST_DEBUG_FUNCPTR (gst_avtp_src_stop);
|
|
pushsrc_class->fill = GST_DEBUG_FUNCPTR (gst_avtp_src_fill);
|
|
|
|
GST_DEBUG_CATEGORY_INIT (avtpsrc_debug, "avtpsrc", 0, "AVTP Source");
|
|
}
|
|
|
|
static void
|
|
gst_avtp_src_init (GstAvtpSrc * avtpsrc)
|
|
{
|
|
gst_base_src_set_live (GST_BASE_SRC (avtpsrc), TRUE);
|
|
gst_base_src_set_format (GST_BASE_SRC (avtpsrc), GST_FORMAT_TIME);
|
|
gst_base_src_set_blocksize (GST_BASE_SRC (avtpsrc), MAX_AVTPDU_SIZE);
|
|
|
|
avtpsrc->ifname = g_strdup (DEFAULT_IFNAME);
|
|
avtpsrc->address = g_strdup (DEFAULT_ADDRESS);
|
|
avtpsrc->sk_fd = -1;
|
|
}
|
|
|
|
static void
|
|
gst_avtp_src_finalize (GObject * object)
|
|
{
|
|
GstAvtpSrc *avtpsrc = GST_AVTP_SRC (object);
|
|
|
|
g_free (avtpsrc->ifname);
|
|
g_free (avtpsrc->address);
|
|
|
|
G_OBJECT_CLASS (parent_class)->finalize (object);
|
|
}
|
|
|
|
static void
|
|
gst_avtp_src_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstAvtpSrc *avtpsrc = GST_AVTP_SRC (object);
|
|
|
|
GST_DEBUG_OBJECT (avtpsrc, "prop_id %u", prop_id);
|
|
|
|
switch (prop_id) {
|
|
case PROP_IFNAME:
|
|
g_free (avtpsrc->ifname);
|
|
avtpsrc->ifname = g_value_dup_string (value);
|
|
break;
|
|
case PROP_ADDRESS:
|
|
g_free (avtpsrc->address);
|
|
avtpsrc->address = g_value_dup_string (value);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
gst_avtp_src_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstAvtpSrc *avtpsrc = GST_AVTP_SRC (object);
|
|
|
|
GST_DEBUG_OBJECT (avtpsrc, "prop_id %u", prop_id);
|
|
|
|
switch (prop_id) {
|
|
case PROP_IFNAME:
|
|
g_value_set_string (value, avtpsrc->ifname);
|
|
break;
|
|
case PROP_ADDRESS:
|
|
g_value_set_string (value, avtpsrc->address);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static gboolean
|
|
gst_avtp_src_start (GstBaseSrc * basesrc)
|
|
{
|
|
int fd, res;
|
|
unsigned int index;
|
|
guint8 addr[ETH_ALEN];
|
|
struct sockaddr_ll sk_addr = { 0 };
|
|
struct packet_mreq mreq = { 0 };
|
|
GstAvtpSrc *avtpsrc = GST_AVTP_SRC (basesrc);
|
|
|
|
index = if_nametoindex (avtpsrc->ifname);
|
|
if (!index) {
|
|
GST_ERROR_OBJECT (avtpsrc, "Failed to get if_index: %s",
|
|
g_strerror (errno));
|
|
return FALSE;
|
|
}
|
|
|
|
fd = socket (AF_PACKET, SOCK_DGRAM, htons (ETH_P_TSN));
|
|
if (fd < 0) {
|
|
GST_ERROR_OBJECT (avtpsrc, "Failed to open socket: %s", g_strerror (errno));
|
|
return FALSE;
|
|
}
|
|
|
|
sk_addr.sll_family = AF_PACKET;
|
|
sk_addr.sll_protocol = htons (ETH_P_TSN);
|
|
sk_addr.sll_ifindex = index;
|
|
|
|
res = bind (fd, (struct sockaddr *) &sk_addr, sizeof (sk_addr));
|
|
if (res < 0) {
|
|
GST_ERROR_OBJECT (avtpsrc, "Failed to bind socket: %s", g_strerror (errno));
|
|
goto err;
|
|
}
|
|
|
|
res = sscanf (avtpsrc->address, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
|
|
&addr[0], &addr[1], &addr[2], &addr[3], &addr[4], &addr[5]);
|
|
if (res != 6) {
|
|
GST_ERROR_OBJECT (avtpsrc, "Destination MAC address format not valid");
|
|
goto err;
|
|
}
|
|
|
|
mreq.mr_ifindex = index;
|
|
mreq.mr_type = PACKET_MR_MULTICAST;
|
|
mreq.mr_alen = ETH_ALEN;
|
|
memcpy (&mreq.mr_address, addr, ETH_ALEN);
|
|
res = setsockopt (fd, SOL_PACKET, PACKET_ADD_MEMBERSHIP, &mreq,
|
|
sizeof (struct packet_mreq));
|
|
if (res < 0) {
|
|
GST_ERROR_OBJECT (avtpsrc, "Failed to set multicast address: %s",
|
|
g_strerror (errno));
|
|
goto err;
|
|
}
|
|
|
|
avtpsrc->sk_fd = fd;
|
|
|
|
GST_DEBUG_OBJECT (avtpsrc, "AVTP source started");
|
|
return TRUE;
|
|
|
|
err:
|
|
close (fd);
|
|
return FALSE;
|
|
}
|
|
|
|
static gboolean
|
|
gst_avtp_src_stop (GstBaseSrc * basesrc)
|
|
{
|
|
GstAvtpSrc *avtpsrc = GST_AVTP_SRC (basesrc);
|
|
|
|
close (avtpsrc->sk_fd);
|
|
|
|
GST_DEBUG_OBJECT (avtpsrc, "AVTP source stopped");
|
|
return TRUE;
|
|
}
|
|
|
|
static GstFlowReturn
|
|
gst_avtp_src_fill (GstPushSrc * pushsrc, GstBuffer * buffer)
|
|
{
|
|
GstMapInfo map;
|
|
gsize buffer_size;
|
|
ssize_t n = MAX_AVTPDU_SIZE;
|
|
ssize_t received = -1;
|
|
GstAvtpSrc *avtpsrc = GST_AVTP_SRC (pushsrc);
|
|
|
|
buffer_size = gst_buffer_get_size (buffer);
|
|
if (G_UNLIKELY (buffer_size < MAX_AVTPDU_SIZE)) {
|
|
GST_WARNING_OBJECT (avtpsrc,
|
|
"Buffer size (%" G_GSIZE_FORMAT
|
|
") may not be enough to hold AVTPDU (max AVTPDU size %d)", buffer_size,
|
|
MAX_AVTPDU_SIZE);
|
|
n = buffer_size;
|
|
}
|
|
|
|
if (!gst_buffer_map (buffer, &map, GST_MAP_WRITE)) {
|
|
GST_WARNING_OBJECT (avtpsrc, "Failed to map buffer");
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
retry:
|
|
errno = 0;
|
|
received = recv (avtpsrc->sk_fd, map.data, n, 0);
|
|
if (received < 0) {
|
|
if (errno == EINTR) {
|
|
goto retry;
|
|
}
|
|
GST_ELEMENT_ERROR (avtpsrc, RESOURCE, READ, (NULL),
|
|
("Failed to receive AVTPDU: %s", g_strerror (errno)));
|
|
gst_buffer_unmap (buffer, &map);
|
|
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
|
|
gst_buffer_set_size (buffer, received);
|
|
gst_buffer_unmap (buffer, &map);
|
|
|
|
return GST_FLOW_OK;
|
|
}
|