1 #include <gst/gst.h>
2 #include <gst/sdp/sdp.h>
3 #include <gst/webrtc/webrtc.h>
4 
5 #include <string.h>
6 
7 static GMainLoop *loop;
8 static GstElement *pipe1, *webrtc1, *webrtc2;
9 static GstBus *bus1;
10 
11 static gboolean
_bus_watch(GstBus * bus,GstMessage * msg,GstElement * pipe)12 _bus_watch (GstBus * bus, GstMessage * msg, GstElement * pipe)
13 {
14   switch (GST_MESSAGE_TYPE (msg)) {
15     case GST_MESSAGE_STATE_CHANGED:
16       if (GST_ELEMENT (msg->src) == pipe) {
17         GstState old, new, pending;
18 
19         gst_message_parse_state_changed (msg, &old, &new, &pending);
20 
21         {
22           gchar *dump_name = g_strconcat ("state_changed-",
23               gst_element_state_get_name (old), "_",
24               gst_element_state_get_name (new), NULL);
25           GST_DEBUG_BIN_TO_DOT_FILE_WITH_TS (GST_BIN (msg->src),
26               GST_DEBUG_GRAPH_SHOW_ALL, dump_name);
27           g_free (dump_name);
28         }
29       }
30       break;
31     case GST_MESSAGE_ERROR:{
32       GError *err = NULL;
33       gchar *dbg_info = NULL;
34 
35       GST_DEBUG_BIN_TO_DOT_FILE_WITH_TS (GST_BIN (pipe),
36           GST_DEBUG_GRAPH_SHOW_ALL, "error");
37 
38       gst_message_parse_error (msg, &err, &dbg_info);
39       g_printerr ("ERROR from element %s: %s\n",
40           GST_OBJECT_NAME (msg->src), err->message);
41       g_printerr ("Debugging info: %s\n", (dbg_info) ? dbg_info : "none");
42       g_error_free (err);
43       g_free (dbg_info);
44       g_main_loop_quit (loop);
45       break;
46     }
47     case GST_MESSAGE_EOS:{
48       GST_DEBUG_BIN_TO_DOT_FILE_WITH_TS (GST_BIN (pipe),
49           GST_DEBUG_GRAPH_SHOW_ALL, "eos");
50       g_print ("EOS received\n");
51       g_main_loop_quit (loop);
52       break;
53     }
54     default:
55       break;
56   }
57 
58   return TRUE;
59 }
60 
61 static void
_webrtc_pad_added(GstElement * webrtc,GstPad * new_pad,GstElement * pipe)62 _webrtc_pad_added (GstElement * webrtc, GstPad * new_pad, GstElement * pipe)
63 {
64   GstElement *out;
65   GstPad *sink;
66 
67   if (GST_PAD_DIRECTION (new_pad) != GST_PAD_SRC)
68     return;
69 
70   out = gst_parse_bin_from_description ("rtpvp8depay ! vp8dec ! "
71       "videoconvert ! queue ! xvimagesink", TRUE, NULL);
72   gst_bin_add (GST_BIN (pipe), out);
73   gst_element_sync_state_with_parent (out);
74 
75   sink = out->sinkpads->data;
76 
77   gst_pad_link (new_pad, sink);
78 }
79 
80 static void
_on_answer_received(GstPromise * promise,gpointer user_data)81 _on_answer_received (GstPromise * promise, gpointer user_data)
82 {
83   GstWebRTCSessionDescription *answer = NULL;
84   const GstStructure *reply;
85   gchar *desc;
86 
87   g_assert (gst_promise_wait (promise) == GST_PROMISE_RESULT_REPLIED);
88   reply = gst_promise_get_reply (promise);
89   gst_structure_get (reply, "answer",
90       GST_TYPE_WEBRTC_SESSION_DESCRIPTION, &answer, NULL);
91   gst_promise_unref (promise);
92   desc = gst_sdp_message_as_text (answer->sdp);
93   g_print ("Created answer:\n%s\n", desc);
94   g_free (desc);
95 
96   /* this is one way to tell webrtcbin that we don't want to be notified when
97    * this task is complete: set a NULL promise */
98   g_signal_emit_by_name (webrtc1, "set-remote-description", answer, NULL);
99   /* this is another way to tell webrtcbin that we don't want to be notified
100    * when this task is complete: interrupt the promise */
101   promise = gst_promise_new ();
102   g_signal_emit_by_name (webrtc2, "set-local-description", answer, NULL);
103   gst_promise_interrupt (promise);
104   gst_promise_unref (promise);
105 
106   gst_webrtc_session_description_free (answer);
107 }
108 
109 static void
_on_offer_received(GstPromise * promise,gpointer user_data)110 _on_offer_received (GstPromise * promise, gpointer user_data)
111 {
112   GstWebRTCSessionDescription *offer = NULL;
113   const GstStructure *reply;
114   gchar *desc;
115 
116   g_assert (gst_promise_wait (promise) == GST_PROMISE_RESULT_REPLIED);
117   reply = gst_promise_get_reply (promise);
118   gst_structure_get (reply, "offer",
119       GST_TYPE_WEBRTC_SESSION_DESCRIPTION, &offer, NULL);
120   gst_promise_unref (promise);
121   desc = gst_sdp_message_as_text (offer->sdp);
122   g_print ("Created offer:\n%s\n", desc);
123   g_free (desc);
124 
125   g_signal_emit_by_name (webrtc1, "set-local-description", offer, NULL);
126   g_signal_emit_by_name (webrtc2, "set-remote-description", offer, NULL);
127 
128   promise = gst_promise_new_with_change_func (_on_answer_received, user_data,
129       NULL);
130   g_signal_emit_by_name (webrtc2, "create-answer", NULL, promise);
131 
132   gst_webrtc_session_description_free (offer);
133 }
134 
135 static void
_on_negotiation_needed(GstElement * element,gpointer user_data)136 _on_negotiation_needed (GstElement * element, gpointer user_data)
137 {
138   GstPromise *promise;
139 
140   promise = gst_promise_new_with_change_func (_on_offer_received, user_data,
141       NULL);
142   g_signal_emit_by_name (webrtc1, "create-offer", NULL, promise);
143 }
144 
145 static void
_on_ice_candidate(GstElement * webrtc,guint mlineindex,gchar * candidate,GstElement * other)146 _on_ice_candidate (GstElement * webrtc, guint mlineindex, gchar * candidate,
147     GstElement * other)
148 {
149   g_signal_emit_by_name (other, "add-ice-candidate", mlineindex, candidate);
150 }
151 
152 int
main(int argc,char * argv[])153 main (int argc, char *argv[])
154 {
155   gst_init (&argc, &argv);
156 
157   loop = g_main_loop_new (NULL, FALSE);
158   pipe1 =
159       gst_parse_launch ("videotestsrc ! queue ! vp8enc ! rtpvp8pay ! queue ! "
160       "application/x-rtp,media=video,payload=96,encoding-name=VP8 ! "
161       "webrtcbin name=smpte videotestsrc pattern=ball ! queue ! vp8enc ! rtpvp8pay ! queue ! "
162       "application/x-rtp,media=video,payload=96,encoding-name=VP8 ! webrtcbin name=ball",
163       NULL);
164   bus1 = gst_pipeline_get_bus (GST_PIPELINE (pipe1));
165   gst_bus_add_watch (bus1, (GstBusFunc) _bus_watch, pipe1);
166 
167   webrtc1 = gst_bin_get_by_name (GST_BIN (pipe1), "smpte");
168   g_signal_connect (webrtc1, "on-negotiation-needed",
169       G_CALLBACK (_on_negotiation_needed), NULL);
170   g_signal_connect (webrtc1, "pad-added", G_CALLBACK (_webrtc_pad_added),
171       pipe1);
172   webrtc2 = gst_bin_get_by_name (GST_BIN (pipe1), "ball");
173   g_signal_connect (webrtc2, "pad-added", G_CALLBACK (_webrtc_pad_added),
174       pipe1);
175   g_signal_connect (webrtc1, "on-ice-candidate",
176       G_CALLBACK (_on_ice_candidate), webrtc2);
177   g_signal_connect (webrtc2, "on-ice-candidate",
178       G_CALLBACK (_on_ice_candidate), webrtc1);
179 
180   g_print ("Starting pipeline\n");
181   gst_element_set_state (GST_ELEMENT (pipe1), GST_STATE_PLAYING);
182 
183   g_main_loop_run (loop);
184 
185   gst_element_set_state (GST_ELEMENT (pipe1), GST_STATE_NULL);
186   g_print ("Pipeline stopped\n");
187 
188   gst_object_unref (webrtc1);
189   gst_object_unref (webrtc2);
190   gst_bus_remove_watch (bus1);
191   gst_object_unref (bus1);
192   gst_object_unref (pipe1);
193 
194   gst_deinit ();
195 
196   return 0;
197 }
198