mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-15 04:46:32 +00:00
297 lines
10 KiB
C
297 lines
10 KiB
C
/* GStreamer
|
|
* Copyright (C) 2013 Collabora Ltd.
|
|
* @author Torrie Fischer <torrie.fischer@collabora.co.uk>
|
|
*
|
|
* 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.
|
|
*/
|
|
#include <gst/gst.h>
|
|
#include <gst/rtp/rtp.h>
|
|
|
|
/*
|
|
* An RTP server
|
|
* creates two sessions and streams audio on one, video on the other, with RTCP
|
|
* on both sessions. The destination is 127.0.0.1.
|
|
*
|
|
* In both sessions, we set "rtprtxsend" as the session's "aux" element
|
|
* in rtpbin, which enables RFC4588 retransmission for that session.
|
|
*
|
|
* .-------. .-------. .-------. .------------. .-------.
|
|
* |audiots| |alawenc| |pcmapay| | rtpbin | |udpsink|
|
|
* | src->sink src->sink src->send_rtp_0 send_rtp_0->sink |
|
|
* '-------' '-------' '-------' | | '-------'
|
|
* | |
|
|
* .-------. .---------. .---------. | | .-------.
|
|
* |audiots| |theoraenc| |theorapay| | | |udpsink|
|
|
* | src->sink src->sink src->send_rtp_1 send_rtp_1->sink |
|
|
* '-------' '---------' '---------' | | '-------'
|
|
* | |
|
|
* .------. | |
|
|
* |udpsrc| | | .-------.
|
|
* | src->recv_rtcp_0 | |udpsink|
|
|
* '------' | send_rtcp_0->sink |
|
|
* | | '-------'
|
|
* .------. | |
|
|
* |udpsrc| | | .-------.
|
|
* | src->recv_rtcp_1 | |udpsink|
|
|
* '------' | send_rtcp_1->sink |
|
|
* '------------' '-------'
|
|
*
|
|
* To keep the set of ports consistent across both this server and the
|
|
* corresponding client, a SessionData struct maps a rtpbin session number to
|
|
* a GstBin and is used to create the corresponding udp sinks with correct
|
|
* ports.
|
|
*/
|
|
|
|
typedef struct _SessionData
|
|
{
|
|
int ref;
|
|
guint sessionNum;
|
|
GstElement *input;
|
|
} SessionData;
|
|
|
|
static SessionData *
|
|
session_ref (SessionData * data)
|
|
{
|
|
g_atomic_int_inc (&data->ref);
|
|
return data;
|
|
}
|
|
|
|
static void
|
|
session_unref (gpointer data)
|
|
{
|
|
SessionData *session = (SessionData *) data;
|
|
if (g_atomic_int_dec_and_test (&session->ref)) {
|
|
g_free (session);
|
|
}
|
|
}
|
|
|
|
static SessionData *
|
|
session_new (guint sessionNum)
|
|
{
|
|
SessionData *ret = g_new0 (SessionData, 1);
|
|
ret->sessionNum = sessionNum;
|
|
return session_ref (ret);
|
|
}
|
|
|
|
/*
|
|
* Used to generate informative messages during pipeline startup
|
|
*/
|
|
static void
|
|
cb_state (GstBus * bus, GstMessage * message, gpointer data)
|
|
{
|
|
GstObject *pipe = GST_OBJECT (data);
|
|
GstState old, new, pending;
|
|
gst_message_parse_state_changed (message, &old, &new, &pending);
|
|
if (message->src == pipe) {
|
|
g_print ("Pipeline %s changed state from %s to %s\n",
|
|
GST_OBJECT_NAME (message->src),
|
|
gst_element_state_get_name (old), gst_element_state_get_name (new));
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Creates a GstGhostPad named "src" on the given bin, pointed at the "src" pad
|
|
* of the given element
|
|
*/
|
|
static void
|
|
setup_ghost (GstElement * src, GstBin * bin)
|
|
{
|
|
GstPad *srcPad = gst_element_get_static_pad (src, "src");
|
|
GstPad *binPad = gst_ghost_pad_new ("src", srcPad);
|
|
gst_element_add_pad (GST_ELEMENT (bin), binPad);
|
|
}
|
|
|
|
static SessionData *
|
|
make_audio_session (guint sessionNum)
|
|
{
|
|
SessionData *session;
|
|
GstBin *audioBin = GST_BIN (gst_bin_new (NULL));
|
|
GstElement *audioSrc = gst_element_factory_make ("audiotestsrc", NULL);
|
|
GstElement *encoder = gst_element_factory_make ("alawenc", NULL);
|
|
GstElement *payloader = gst_element_factory_make ("rtppcmapay", NULL);
|
|
g_object_set (audioSrc, "is-live", TRUE, NULL);
|
|
|
|
gst_bin_add_many (audioBin, audioSrc, encoder, payloader, NULL);
|
|
gst_element_link_many (audioSrc, encoder, payloader, NULL);
|
|
|
|
setup_ghost (payloader, audioBin);
|
|
|
|
session = session_new (sessionNum);
|
|
session->input = GST_ELEMENT (audioBin);
|
|
|
|
return session;
|
|
}
|
|
|
|
static SessionData *
|
|
make_video_session (guint sessionNum)
|
|
{
|
|
GstBin *videoBin = GST_BIN (gst_bin_new (NULL));
|
|
GstElement *videoSrc = gst_element_factory_make ("videotestsrc", NULL);
|
|
GstElement *encoder = gst_element_factory_make ("theoraenc", NULL);
|
|
GstElement *payloader = gst_element_factory_make ("rtptheorapay", NULL);
|
|
GstCaps *videoCaps;
|
|
SessionData *session;
|
|
g_object_set (videoSrc, "is-live", TRUE, "horizontal-speed", 1, NULL);
|
|
g_object_set (payloader, "config-interval", 2, NULL);
|
|
|
|
gst_bin_add_many (videoBin, videoSrc, encoder, payloader, NULL);
|
|
videoCaps = gst_caps_new_simple ("video/x-raw",
|
|
"width", G_TYPE_INT, 352,
|
|
"height", G_TYPE_INT, 288, "framerate", GST_TYPE_FRACTION, 15, 1, NULL);
|
|
gst_element_link_filtered (videoSrc, encoder, videoCaps);
|
|
gst_element_link (encoder, payloader);
|
|
|
|
setup_ghost (payloader, videoBin);
|
|
|
|
session = session_new (sessionNum);
|
|
session->input = GST_ELEMENT (videoBin);
|
|
|
|
return session;
|
|
}
|
|
|
|
static GstElement *
|
|
request_aux_sender (GstElement * rtpbin, guint sessid, SessionData * session)
|
|
{
|
|
GstElement *rtx, *bin;
|
|
GstPad *pad;
|
|
gchar *name;
|
|
GstStructure *pt_map;
|
|
|
|
GST_INFO ("creating AUX sender");
|
|
bin = gst_bin_new (NULL);
|
|
rtx = gst_element_factory_make ("rtprtxsend", NULL);
|
|
pt_map = gst_structure_new ("application/x-rtp-pt-map",
|
|
"8", G_TYPE_UINT, 98, "96", G_TYPE_UINT, 99, NULL);
|
|
g_object_set (rtx, "payload-type-map", pt_map, NULL);
|
|
gst_structure_free (pt_map);
|
|
gst_bin_add (GST_BIN (bin), rtx);
|
|
|
|
pad = gst_element_get_static_pad (rtx, "src");
|
|
name = g_strdup_printf ("src_%u", sessid);
|
|
gst_element_add_pad (bin, gst_ghost_pad_new (name, pad));
|
|
g_free (name);
|
|
gst_object_unref (pad);
|
|
|
|
pad = gst_element_get_static_pad (rtx, "sink");
|
|
name = g_strdup_printf ("sink_%u", sessid);
|
|
gst_element_add_pad (bin, gst_ghost_pad_new (name, pad));
|
|
g_free (name);
|
|
gst_object_unref (pad);
|
|
|
|
return bin;
|
|
}
|
|
|
|
/*
|
|
* This function sets up the UDP sinks and sources for RTP/RTCP, adds the
|
|
* given session's bin into the pipeline, and links it to the properly numbered
|
|
* pads on the rtpbin
|
|
*/
|
|
static void
|
|
add_stream (GstPipeline * pipe, GstElement * rtpBin, SessionData * session)
|
|
{
|
|
GstElement *rtpSink = gst_element_factory_make ("udpsink", NULL);
|
|
GstElement *rtcpSink = gst_element_factory_make ("udpsink", NULL);
|
|
GstElement *rtcpSrc = gst_element_factory_make ("udpsrc", NULL);
|
|
GstElement *identity = gst_element_factory_make ("identity", NULL);
|
|
int basePort;
|
|
gchar *padName;
|
|
|
|
basePort = 5000 + (session->sessionNum * 6);
|
|
|
|
gst_bin_add_many (GST_BIN (pipe), rtpSink, rtcpSink, rtcpSrc, identity,
|
|
session->input, NULL);
|
|
|
|
/* enable retransmission by setting rtprtxsend as the "aux" element of rtpbin */
|
|
g_signal_connect (rtpBin, "request-aux-sender",
|
|
(GCallback) request_aux_sender, session);
|
|
|
|
g_object_set (rtpSink, "port", basePort, "host", "127.0.0.1", NULL);
|
|
g_object_set (rtcpSink, "port", basePort + 1, "host", "127.0.0.1", "sync",
|
|
FALSE, "async", FALSE, NULL);
|
|
g_object_set (rtcpSrc, "port", basePort + 5, NULL);
|
|
|
|
/* this is just to drop some rtp packets at random, to demonstrate
|
|
* that rtprtxsend actually works */
|
|
g_object_set (identity, "drop-probability", 0.01, NULL);
|
|
|
|
padName = g_strdup_printf ("send_rtp_sink_%u", session->sessionNum);
|
|
gst_element_link_pads (session->input, "src", rtpBin, padName);
|
|
g_free (padName);
|
|
|
|
/* link rtpbin to udpsink directly here if you don't want
|
|
* artificial packet loss */
|
|
padName = g_strdup_printf ("send_rtp_src_%u", session->sessionNum);
|
|
gst_element_link_pads (rtpBin, padName, identity, "sink");
|
|
gst_element_link (identity, rtpSink);
|
|
g_free (padName);
|
|
|
|
padName = g_strdup_printf ("send_rtcp_src_%u", session->sessionNum);
|
|
gst_element_link_pads (rtpBin, padName, rtcpSink, "sink");
|
|
g_free (padName);
|
|
|
|
padName = g_strdup_printf ("recv_rtcp_sink_%u", session->sessionNum);
|
|
gst_element_link_pads (rtcpSrc, "src", rtpBin, padName);
|
|
g_free (padName);
|
|
|
|
g_print ("New RTP stream on %i/%i/%i\n", basePort, basePort + 1,
|
|
basePort + 5);
|
|
|
|
session_unref (session);
|
|
}
|
|
|
|
int
|
|
main (int argc, char **argv)
|
|
{
|
|
GstPipeline *pipe;
|
|
GstBus *bus;
|
|
SessionData *videoSession;
|
|
SessionData *audioSession;
|
|
GstElement *rtpBin;
|
|
GMainLoop *loop;
|
|
|
|
gst_init (&argc, &argv);
|
|
|
|
loop = g_main_loop_new (NULL, FALSE);
|
|
|
|
pipe = GST_PIPELINE (gst_pipeline_new (NULL));
|
|
bus = gst_element_get_bus (GST_ELEMENT (pipe));
|
|
g_signal_connect (bus, "message::state-changed", G_CALLBACK (cb_state), pipe);
|
|
gst_bus_add_signal_watch (bus);
|
|
gst_object_unref (bus);
|
|
|
|
rtpBin = gst_element_factory_make ("rtpbin", NULL);
|
|
g_object_set (rtpBin, "rtp-profile", GST_RTP_PROFILE_AVPF, NULL);
|
|
|
|
gst_bin_add (GST_BIN (pipe), rtpBin);
|
|
|
|
videoSession = make_video_session (0);
|
|
audioSession = make_audio_session (1);
|
|
add_stream (pipe, rtpBin, videoSession);
|
|
add_stream (pipe, rtpBin, audioSession);
|
|
|
|
g_print ("starting server pipeline\n");
|
|
gst_element_set_state (GST_ELEMENT (pipe), GST_STATE_PLAYING);
|
|
|
|
g_main_loop_run (loop);
|
|
|
|
g_print ("stopping server pipeline\n");
|
|
gst_element_set_state (GST_ELEMENT (pipe), GST_STATE_NULL);
|
|
|
|
gst_object_unref (pipe);
|
|
g_main_loop_unref (loop);
|
|
|
|
return 0;
|
|
}
|