1 /* GStreamer
2 * Copyright (C) <2007> Wim Taymans <wim.taymans@gmail.com>
3 * Copyright (C) <2009> Jarkko Palviainen <jarkko.palviainen@sesca.com>
4 * Copyright (C) <2012> Collabora Ltd.
5 * Author: Sebastian Dröge <sebastian.droege@collabora.co.uk>
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Library General Public
9 * License as published by the Free Software Foundation; either
10 * version 2 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Library General Public License for more details.
16 *
17 * You should have received a copy of the GNU Library General Public
18 * License along with this library; if not, write to the
19 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
20 * Boston, MA 02110-1301, USA.
21 */
22
23 /**
24 * SECTION:element-multiudpsink
25 * @see_also: udpsink, multifdsink
26 *
27 * multiudpsink is a network sink that sends UDP packets to multiple
28 * clients.
29 * It can be combined with rtp payload encoders to implement RTP streaming.
30 */
31
32 #ifdef HAVE_CONFIG_H
33 #include "config.h"
34 #endif
35 #include "gstmultiudpsink.h"
36
37 #include <string.h>
38 #ifdef HAVE_SYS_SOCKET_H
39 #include <sys/socket.h>
40 #endif
41
42 #ifndef G_OS_WIN32
43 #include <netinet/in.h>
44 #endif
45
46 #include "gst/glib-compat-private.h"
47
48 GST_DEBUG_CATEGORY_STATIC (multiudpsink_debug);
49 #define GST_CAT_DEFAULT (multiudpsink_debug)
50
51 #define UDP_MAX_SIZE 65507
52
53 static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
54 GST_PAD_SINK,
55 GST_PAD_ALWAYS,
56 GST_STATIC_CAPS_ANY);
57
58 /* MultiUDPSink signals and args */
59 enum
60 {
61 /* methods */
62 SIGNAL_ADD,
63 SIGNAL_REMOVE,
64 SIGNAL_CLEAR,
65 SIGNAL_GET_STATS,
66
67 /* signals */
68 SIGNAL_CLIENT_ADDED,
69 SIGNAL_CLIENT_REMOVED,
70
71 /* FILL ME */
72 LAST_SIGNAL
73 };
74
75 #define DEFAULT_SOCKET NULL
76 #define DEFAULT_CLOSE_SOCKET TRUE
77 #define DEFAULT_USED_SOCKET NULL
78 #define DEFAULT_CLIENTS NULL
79 /* FIXME, this should be disabled by default, we don't need to join a multicast
80 * group for sending, if this socket is also used for receiving, it should
81 * be configured in the element that does the receive. */
82 #define DEFAULT_AUTO_MULTICAST TRUE
83 #define DEFAULT_MULTICAST_IFACE NULL
84 #define DEFAULT_TTL 64
85 #define DEFAULT_TTL_MC 1
86 #define DEFAULT_LOOP TRUE
87 #define DEFAULT_FORCE_IPV4 FALSE
88 #define DEFAULT_QOS_DSCP -1
89 #define DEFAULT_SEND_DUPLICATES TRUE
90 #define DEFAULT_BUFFER_SIZE 0
91 #define DEFAULT_BIND_ADDRESS NULL
92 #define DEFAULT_BIND_PORT 0
93
94 enum
95 {
96 PROP_0,
97 PROP_BYTES_TO_SERVE,
98 PROP_BYTES_SERVED,
99 PROP_SOCKET,
100 PROP_SOCKET_V6,
101 PROP_CLOSE_SOCKET,
102 PROP_USED_SOCKET,
103 PROP_USED_SOCKET_V6,
104 PROP_CLIENTS,
105 PROP_AUTO_MULTICAST,
106 PROP_MULTICAST_IFACE,
107 PROP_TTL,
108 PROP_TTL_MC,
109 PROP_LOOP,
110 PROP_FORCE_IPV4,
111 PROP_QOS_DSCP,
112 PROP_SEND_DUPLICATES,
113 PROP_BUFFER_SIZE,
114 PROP_BIND_ADDRESS,
115 PROP_BIND_PORT
116 };
117
118 static void gst_multiudpsink_finalize (GObject * object);
119
120 static GstFlowReturn gst_multiudpsink_render (GstBaseSink * sink,
121 GstBuffer * buffer);
122 static GstFlowReturn gst_multiudpsink_render_list (GstBaseSink * bsink,
123 GstBufferList * buffer_list);
124
125 static gboolean gst_multiudpsink_start (GstBaseSink * bsink);
126 static gboolean gst_multiudpsink_stop (GstBaseSink * bsink);
127 static gboolean gst_multiudpsink_unlock (GstBaseSink * bsink);
128 static gboolean gst_multiudpsink_unlock_stop (GstBaseSink * bsink);
129
130 static void gst_multiudpsink_set_property (GObject * object, guint prop_id,
131 const GValue * value, GParamSpec * pspec);
132 static void gst_multiudpsink_get_property (GObject * object, guint prop_id,
133 GValue * value, GParamSpec * pspec);
134
135 static void gst_multiudpsink_add_internal (GstMultiUDPSink * sink,
136 const gchar * host, gint port, gboolean lock);
137 static void gst_multiudpsink_clear_internal (GstMultiUDPSink * sink,
138 gboolean lock);
139
140 static guint gst_multiudpsink_signals[LAST_SIGNAL] = { 0 };
141
142 #define gst_multiudpsink_parent_class parent_class
143 G_DEFINE_TYPE (GstMultiUDPSink, gst_multiudpsink, GST_TYPE_BASE_SINK);
144
145 static void
gst_multiudpsink_class_init(GstMultiUDPSinkClass * klass)146 gst_multiudpsink_class_init (GstMultiUDPSinkClass * klass)
147 {
148 GObjectClass *gobject_class;
149 GstElementClass *gstelement_class;
150 GstBaseSinkClass *gstbasesink_class;
151
152 gobject_class = (GObjectClass *) klass;
153 gstelement_class = (GstElementClass *) klass;
154 gstbasesink_class = (GstBaseSinkClass *) klass;
155
156 gobject_class->set_property = gst_multiudpsink_set_property;
157 gobject_class->get_property = gst_multiudpsink_get_property;
158 gobject_class->finalize = gst_multiudpsink_finalize;
159
160 /**
161 * GstMultiUDPSink::add:
162 * @gstmultiudpsink: the sink on which the signal is emitted
163 * @host: the hostname/IP address of the client to add
164 * @port: the port of the client to add
165 *
166 * Add a client with destination @host and @port to the list of
167 * clients. When the same host/port pair is added multiple times, the
168 * send-duplicates property defines if the packets are sent multiple times to
169 * the same host/port pair or not.
170 *
171 * When a host/port pair is added multiple times, an equal amount of remove
172 * calls must be performed to actually remove the host/port pair from the list
173 * of destinations.
174 */
175 gst_multiudpsink_signals[SIGNAL_ADD] =
176 g_signal_new ("add", G_TYPE_FROM_CLASS (klass),
177 G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
178 G_STRUCT_OFFSET (GstMultiUDPSinkClass, add),
179 NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 2,
180 G_TYPE_STRING, G_TYPE_INT);
181 /**
182 * GstMultiUDPSink::remove:
183 * @gstmultiudpsink: the sink on which the signal is emitted
184 * @host: the hostname/IP address of the client to remove
185 * @port: the port of the client to remove
186 *
187 * Remove the client with destination @host and @port from the list of
188 * clients.
189 */
190 gst_multiudpsink_signals[SIGNAL_REMOVE] =
191 g_signal_new ("remove", G_TYPE_FROM_CLASS (klass),
192 G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
193 G_STRUCT_OFFSET (GstMultiUDPSinkClass, remove),
194 NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 2,
195 G_TYPE_STRING, G_TYPE_INT);
196 /**
197 * GstMultiUDPSink::clear:
198 * @gstmultiudpsink: the sink on which the signal is emitted
199 *
200 * Clear the list of clients.
201 */
202 gst_multiudpsink_signals[SIGNAL_CLEAR] =
203 g_signal_new ("clear", G_TYPE_FROM_CLASS (klass),
204 G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
205 G_STRUCT_OFFSET (GstMultiUDPSinkClass, clear),
206 NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 0);
207 /**
208 * GstMultiUDPSink::get-stats:
209 * @gstmultiudpsink: the sink on which the signal is emitted
210 * @host: the hostname/IP address of the client to get stats on
211 * @port: the port of the client to get stats on
212 *
213 * Get the statistics of the client with destination @host and @port.
214 *
215 * Returns: a GstStructure: bytes_sent, packets_sent,
216 * connect_time (in epoch seconds), disconnect_time (in epoch seconds)
217 */
218 gst_multiudpsink_signals[SIGNAL_GET_STATS] =
219 g_signal_new ("get-stats", G_TYPE_FROM_CLASS (klass),
220 G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
221 G_STRUCT_OFFSET (GstMultiUDPSinkClass, get_stats),
222 NULL, NULL, g_cclosure_marshal_generic, GST_TYPE_STRUCTURE, 2,
223 G_TYPE_STRING, G_TYPE_INT);
224 /**
225 * GstMultiUDPSink::client-added:
226 * @gstmultiudpsink: the sink emitting the signal
227 * @host: the hostname/IP address of the added client
228 * @port: the port of the added client
229 *
230 * Signal emitted when a new client is added to the list of
231 * clients.
232 */
233 gst_multiudpsink_signals[SIGNAL_CLIENT_ADDED] =
234 g_signal_new ("client-added", G_TYPE_FROM_CLASS (klass),
235 G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstMultiUDPSinkClass, client_added),
236 NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 2,
237 G_TYPE_STRING, G_TYPE_INT);
238 /**
239 * GstMultiUDPSink::client-removed:
240 * @gstmultiudpsink: the sink emitting the signal
241 * @host: the hostname/IP address of the removed client
242 * @port: the port of the removed client
243 *
244 * Signal emitted when a client is removed from the list of
245 * clients.
246 */
247 gst_multiudpsink_signals[SIGNAL_CLIENT_REMOVED] =
248 g_signal_new ("client-removed", G_TYPE_FROM_CLASS (klass),
249 G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstMultiUDPSinkClass,
250 client_removed), NULL, NULL, g_cclosure_marshal_generic,
251 G_TYPE_NONE, 2, G_TYPE_STRING, G_TYPE_INT);
252
253 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_BYTES_TO_SERVE,
254 g_param_spec_uint64 ("bytes-to-serve", "Bytes to serve",
255 "Number of bytes received to serve to clients", 0, G_MAXUINT64, 0,
256 G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
257 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_BYTES_SERVED,
258 g_param_spec_uint64 ("bytes-served", "Bytes served",
259 "Total number of bytes sent to all clients", 0, G_MAXUINT64, 0,
260 G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
261 g_object_class_install_property (gobject_class, PROP_SOCKET,
262 g_param_spec_object ("socket", "Socket Handle",
263 "Socket to use for UDP sending. (NULL == allocate)",
264 G_TYPE_SOCKET, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
265 g_object_class_install_property (gobject_class, PROP_SOCKET_V6,
266 g_param_spec_object ("socket-v6", "Socket Handle IPv6",
267 "Socket to use for UDPv6 sending. (NULL == allocate)",
268 G_TYPE_SOCKET, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
269 g_object_class_install_property (gobject_class, PROP_CLOSE_SOCKET,
270 g_param_spec_boolean ("close-socket", "Close socket",
271 "Close socket if passed as property on state change",
272 DEFAULT_CLOSE_SOCKET, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
273 g_object_class_install_property (gobject_class, PROP_USED_SOCKET,
274 g_param_spec_object ("used-socket", "Used Socket Handle",
275 "Socket currently in use for UDP sending. (NULL == no socket)",
276 G_TYPE_SOCKET, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
277 g_object_class_install_property (gobject_class, PROP_USED_SOCKET_V6,
278 g_param_spec_object ("used-socket-v6", "Used Socket Handle IPv6",
279 "Socket currently in use for UDPv6 sending. (NULL == no socket)",
280 G_TYPE_SOCKET, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
281 g_object_class_install_property (gobject_class, PROP_CLIENTS,
282 g_param_spec_string ("clients", "Clients",
283 "A comma separated list of host:port pairs with destinations",
284 DEFAULT_CLIENTS, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
285 g_object_class_install_property (gobject_class, PROP_AUTO_MULTICAST,
286 g_param_spec_boolean ("auto-multicast",
287 "Automatically join/leave multicast groups",
288 "Automatically join/leave the multicast groups, FALSE means user"
289 " has to do it himself", DEFAULT_AUTO_MULTICAST,
290 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
291 g_object_class_install_property (gobject_class, PROP_MULTICAST_IFACE,
292 g_param_spec_string ("multicast-iface", "Multicast Interface",
293 "The network interface on which to join the multicast group",
294 DEFAULT_MULTICAST_IFACE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
295 g_object_class_install_property (gobject_class, PROP_TTL,
296 g_param_spec_int ("ttl", "Unicast TTL",
297 "Used for setting the unicast TTL parameter",
298 0, 255, DEFAULT_TTL, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
299 g_object_class_install_property (gobject_class, PROP_TTL_MC,
300 g_param_spec_int ("ttl-mc", "Multicast TTL",
301 "Used for setting the multicast TTL parameter",
302 0, 255, DEFAULT_TTL_MC, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
303 g_object_class_install_property (gobject_class, PROP_LOOP,
304 g_param_spec_boolean ("loop", "Multicast Loopback",
305 "Used for setting the multicast loop parameter. TRUE = enable,"
306 " FALSE = disable", DEFAULT_LOOP,
307 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
308 /**
309 * GstMultiUDPSink::force-ipv4:
310 *
311 * Force the use of an IPv4 socket.
312 *
313 * Since: 1.0.2
314 */
315 #ifndef GST_REMOVE_DEPRECATED
316 g_object_class_install_property (gobject_class, PROP_FORCE_IPV4,
317 g_param_spec_boolean ("force-ipv4", "Force IPv4",
318 "Forcing the use of an IPv4 socket (DEPRECATED, has no effect anymore)",
319 DEFAULT_FORCE_IPV4,
320 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | G_PARAM_DEPRECATED));
321 #endif
322 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_QOS_DSCP,
323 g_param_spec_int ("qos-dscp", "QoS diff srv code point",
324 "Quality of Service, differentiated services code point (-1 default)",
325 -1, 63, DEFAULT_QOS_DSCP,
326 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
327 /**
328 * GstMultiUDPSink::send-duplicates:
329 *
330 * When a host/port pair is added mutliple times, send the packet to the host
331 * multiple times as well.
332 */
333 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_SEND_DUPLICATES,
334 g_param_spec_boolean ("send-duplicates", "Send Duplicates",
335 "When a distination/port pair is added multiple times, send packets "
336 "multiple times as well", DEFAULT_SEND_DUPLICATES,
337 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
338
339 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_BUFFER_SIZE,
340 g_param_spec_int ("buffer-size", "Buffer Size",
341 "Size of the kernel send buffer in bytes, 0=default", 0, G_MAXINT,
342 DEFAULT_BUFFER_SIZE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
343
344 g_object_class_install_property (gobject_class, PROP_BIND_ADDRESS,
345 g_param_spec_string ("bind-address", "Bind Address",
346 "Address to bind the socket to", DEFAULT_BIND_ADDRESS,
347 G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
348 g_object_class_install_property (gobject_class, PROP_BIND_PORT,
349 g_param_spec_int ("bind-port", "Bind Port",
350 "Port to bind the socket to", 0, G_MAXUINT16,
351 DEFAULT_BIND_PORT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
352
353 gst_element_class_add_static_pad_template (gstelement_class, &sink_template);
354
355 gst_element_class_set_static_metadata (gstelement_class, "UDP packet sender",
356 "Sink/Network",
357 "Send data over the network via UDP to one or multiple recipients "
358 "which can be added or removed at runtime using action signals",
359 "Wim Taymans <wim.taymans@gmail.com>");
360
361 gstbasesink_class->render = gst_multiudpsink_render;
362 gstbasesink_class->render_list = gst_multiudpsink_render_list;
363 gstbasesink_class->start = gst_multiudpsink_start;
364 gstbasesink_class->stop = gst_multiudpsink_stop;
365 gstbasesink_class->unlock = gst_multiudpsink_unlock;
366 gstbasesink_class->unlock_stop = gst_multiudpsink_unlock_stop;
367 klass->add = gst_multiudpsink_add;
368 klass->remove = gst_multiudpsink_remove;
369 klass->clear = gst_multiudpsink_clear;
370 klass->get_stats = gst_multiudpsink_get_stats;
371
372 GST_DEBUG_CATEGORY_INIT (multiudpsink_debug, "multiudpsink", 0, "UDP sink");
373 }
374
375 static void
gst_multiudpsink_create_cancellable(GstMultiUDPSink * sink)376 gst_multiudpsink_create_cancellable (GstMultiUDPSink * sink)
377 {
378 GPollFD pollfd;
379
380 sink->cancellable = g_cancellable_new ();
381 sink->made_cancel_fd = g_cancellable_make_pollfd (sink->cancellable, &pollfd);
382 }
383
384 static void
gst_multiudpsink_free_cancellable(GstMultiUDPSink * sink)385 gst_multiudpsink_free_cancellable (GstMultiUDPSink * sink)
386 {
387 if (sink->made_cancel_fd) {
388 g_cancellable_release_fd (sink->cancellable);
389 sink->made_cancel_fd = FALSE;
390 }
391 g_object_unref (sink->cancellable);
392 sink->cancellable = NULL;
393 }
394
395 static void
gst_multiudpsink_init(GstMultiUDPSink * sink)396 gst_multiudpsink_init (GstMultiUDPSink * sink)
397 {
398 guint max_mem;
399
400 g_mutex_init (&sink->client_lock);
401 sink->clients = NULL;
402 sink->num_v4_unique = 0;
403 sink->num_v4_all = 0;
404 sink->num_v6_unique = 0;
405 sink->num_v6_all = 0;
406
407 sink->socket = DEFAULT_SOCKET;
408 sink->socket_v6 = DEFAULT_SOCKET;
409 sink->used_socket = DEFAULT_USED_SOCKET;
410 sink->used_socket_v6 = DEFAULT_USED_SOCKET;
411 sink->close_socket = DEFAULT_CLOSE_SOCKET;
412 sink->external_socket = (sink->socket != NULL);
413 sink->auto_multicast = DEFAULT_AUTO_MULTICAST;
414 sink->ttl = DEFAULT_TTL;
415 sink->ttl_mc = DEFAULT_TTL_MC;
416 sink->loop = DEFAULT_LOOP;
417 sink->force_ipv4 = DEFAULT_FORCE_IPV4;
418 sink->qos_dscp = DEFAULT_QOS_DSCP;
419 sink->send_duplicates = DEFAULT_SEND_DUPLICATES;
420 sink->multi_iface = g_strdup (DEFAULT_MULTICAST_IFACE);
421
422 gst_multiudpsink_create_cancellable (sink);
423
424 /* pre-allocate OutputVector, MapInfo and OutputMessage arrays
425 * for use in the render and render_list functions */
426 max_mem = gst_buffer_get_max_memory ();
427
428 sink->n_vecs = max_mem;
429 sink->vecs = g_new (GOutputVector, sink->n_vecs);
430
431 sink->n_maps = max_mem;
432 sink->maps = g_new (GstMapInfo, sink->n_maps);
433
434 sink->n_messages = 1;
435 sink->messages = g_new (GstOutputMessage, sink->n_messages);
436
437 /* we assume that the number of memories per buffer can fit into a guint8 */
438 g_warn_if_fail (max_mem <= G_MAXUINT8);
439 }
440
441 static GstUDPClient *
gst_udp_client_new(GstMultiUDPSink * sink,const gchar * host,gint port)442 gst_udp_client_new (GstMultiUDPSink * sink, const gchar * host, gint port)
443 {
444 GstUDPClient *client;
445 GInetAddress *addr;
446 GResolver *resolver;
447 GError *err = NULL;
448
449 addr = g_inet_address_new_from_string (host);
450 if (!addr) {
451 GList *results;
452
453 resolver = g_resolver_get_default ();
454 results =
455 g_resolver_lookup_by_name (resolver, host, sink->cancellable, &err);
456 if (!results)
457 goto name_resolve;
458 addr = G_INET_ADDRESS (g_object_ref (results->data));
459
460 g_resolver_free_addresses (results);
461 g_object_unref (resolver);
462 }
463 #ifndef GST_DISABLE_GST_DEBUG
464 {
465 gchar *ip = g_inet_address_to_string (addr);
466
467 GST_DEBUG_OBJECT (sink, "IP address for host %s is %s", host, ip);
468 g_free (ip);
469 }
470 #endif
471
472 client = g_slice_new0 (GstUDPClient);
473 client->ref_count = 1;
474 client->add_count = 0;
475 client->host = g_strdup (host);
476 client->port = port;
477 client->addr = g_inet_socket_address_new (addr, port);
478 g_object_unref (addr);
479
480 return client;
481
482 name_resolve:
483 {
484 g_clear_error (&err);
485 g_object_unref (resolver);
486
487 return NULL;
488 }
489 }
490
491 /* call with client lock held */
492 static void
gst_udp_client_unref(GstUDPClient * client)493 gst_udp_client_unref (GstUDPClient * client)
494 {
495 if (--client->ref_count == 0) {
496 g_object_unref (client->addr);
497 g_free (client->host);
498 g_slice_free (GstUDPClient, client);
499 }
500 }
501
502 /* call with client lock held */
503 static inline GstUDPClient *
gst_udp_client_ref(GstUDPClient * client)504 gst_udp_client_ref (GstUDPClient * client)
505 {
506 ++client->ref_count;
507 return client;
508 }
509
510 static gint
client_compare(GstUDPClient * a,GstUDPClient * b)511 client_compare (GstUDPClient * a, GstUDPClient * b)
512 {
513 if ((a->port == b->port) && (strcmp (a->host, b->host) == 0))
514 return 0;
515
516 return 1;
517 }
518
519 static void
gst_multiudpsink_finalize(GObject * object)520 gst_multiudpsink_finalize (GObject * object)
521 {
522 GstMultiUDPSink *sink;
523
524 sink = GST_MULTIUDPSINK (object);
525
526 g_list_foreach (sink->clients, (GFunc) gst_udp_client_unref, NULL);
527 g_list_free (sink->clients);
528
529 if (sink->socket)
530 g_object_unref (sink->socket);
531 sink->socket = NULL;
532
533 if (sink->socket_v6)
534 g_object_unref (sink->socket_v6);
535 sink->socket_v6 = NULL;
536
537 if (sink->used_socket)
538 g_object_unref (sink->used_socket);
539 sink->used_socket = NULL;
540
541 if (sink->used_socket_v6)
542 g_object_unref (sink->used_socket_v6);
543 sink->used_socket_v6 = NULL;
544
545 gst_multiudpsink_free_cancellable (sink);
546
547 g_free (sink->multi_iface);
548 sink->multi_iface = NULL;
549
550 g_free (sink->vecs);
551 sink->vecs = NULL;
552 g_free (sink->maps);
553 sink->maps = NULL;
554 g_free (sink->messages);
555 sink->messages = NULL;
556
557 g_free (sink->bind_address);
558 sink->bind_address = NULL;
559
560 g_mutex_clear (&sink->client_lock);
561
562 G_OBJECT_CLASS (parent_class)->finalize (object);
563 }
564
565 /* replacement until we can depend unconditionally on the real one in GLib */
566 #ifndef HAVE_G_SOCKET_SEND_MESSAGES
567 #define g_socket_send_messages gst_socket_send_messages
568
569 static gint
gst_socket_send_messages(GSocket * socket,GstOutputMessage * messages,guint num_messages,gint flags,GCancellable * cancellable,GError ** error)570 gst_socket_send_messages (GSocket * socket, GstOutputMessage * messages,
571 guint num_messages, gint flags, GCancellable * cancellable, GError ** error)
572 {
573 gssize result;
574 gint i;
575
576 for (i = 0; i < num_messages; ++i) {
577 GstOutputMessage *msg = &messages[i];
578 GError *msg_error = NULL;
579
580 result = g_socket_send_message (socket, msg->address,
581 msg->vectors, msg->num_vectors,
582 msg->control_messages, msg->num_control_messages,
583 flags, cancellable, &msg_error);
584
585 if (result < 0) {
586 /* if we couldn't send all messages, just return how many we did
587 * manage to send, provided we managed to send at least one */
588 if (msg_error->code == G_IO_ERROR_WOULD_BLOCK && i > 0) {
589 g_error_free (msg_error);
590 return i;
591 } else {
592 g_propagate_error (error, msg_error);
593 return -1;
594 }
595 }
596
597 msg->bytes_sent = result;
598 }
599
600 return i;
601 }
602 #endif /* HAVE_G_SOCKET_SEND_MESSAGES */
603
604 static gsize
fill_vectors(GOutputVector * vecs,GstMapInfo * maps,guint n,GstBuffer * buf)605 fill_vectors (GOutputVector * vecs, GstMapInfo * maps, guint n, GstBuffer * buf)
606 {
607 GstMemory *mem;
608 gsize size = 0;
609 guint i;
610
611 g_assert (gst_buffer_n_memory (buf) == n);
612
613 for (i = 0; i < n; ++i) {
614 mem = gst_buffer_peek_memory (buf, i);
615 if (gst_memory_map (mem, &maps[i], GST_MAP_READ)) {
616 vecs[i].buffer = maps[i].data;
617 vecs[i].size = maps[i].size;
618 } else {
619 GST_WARNING ("Failed to map memory %p for reading", mem);
620 vecs[i].buffer = "";
621 vecs[i].size = 0;
622 }
623 size += vecs[i].size;
624 }
625
626 return size;
627 }
628
629 static gsize
gst_udp_calc_message_size(GstOutputMessage * msg)630 gst_udp_calc_message_size (GstOutputMessage * msg)
631 {
632 gsize size = 0;
633 guint i;
634
635 for (i = 0; i < msg->num_vectors; ++i)
636 size += msg->vectors[i].size;
637
638 return size;
639 }
640
641 static gint
gst_udp_messsages_find_first_not_sent(GstOutputMessage * messages,guint num_messages)642 gst_udp_messsages_find_first_not_sent (GstOutputMessage * messages,
643 guint num_messages)
644 {
645 guint i;
646
647 for (i = 0; i < num_messages; ++i) {
648 GstOutputMessage *msg = &messages[i];
649
650 if (msg->bytes_sent == 0 && gst_udp_calc_message_size (msg) > 0)
651 return i;
652 }
653
654 return -1;
655 }
656
657 static inline gchar *
gst_udp_address_get_string(GSocketAddress * addr,gchar * s,gsize size)658 gst_udp_address_get_string (GSocketAddress * addr, gchar * s, gsize size)
659 {
660 GInetSocketAddress *isa = G_INET_SOCKET_ADDRESS (addr);
661 GInetAddress *ia;
662 gchar *addr_str;
663
664 ia = g_inet_socket_address_get_address (isa);
665 addr_str = g_inet_address_to_string (ia);
666 g_snprintf (s, size, "%s:%u", addr_str, g_inet_socket_address_get_port (isa));
667 g_free (addr_str);
668
669 return s;
670 }
671
672 /* Wrapper around g_socket_send_messages() plus error handling (ignoring).
673 * Returns FALSE if we got cancelled, otherwise TRUE. */
674 static GstFlowReturn
gst_multiudpsink_send_messages(GstMultiUDPSink * sink,GSocket * socket,GstOutputMessage * messages,guint num_messages)675 gst_multiudpsink_send_messages (GstMultiUDPSink * sink, GSocket * socket,
676 GstOutputMessage * messages, guint num_messages)
677 {
678 gboolean sent_max_size_warning = FALSE;
679
680 while (num_messages > 0) {
681 gchar astr[64] G_GNUC_UNUSED;
682 GError *err = NULL;
683 guint msg_size, skip, i;
684 gint ret, err_idx;
685
686 ret = g_socket_send_messages (socket, messages, num_messages, 0,
687 sink->cancellable, &err);
688
689 if (G_UNLIKELY (ret < 0)) {
690 GstOutputMessage *msg;
691
692 if (g_error_matches (err, G_IO_ERROR, G_IO_ERROR_CANCELLED)) {
693 GstFlowReturn flow_ret;
694
695 g_clear_error (&err);
696
697 flow_ret = gst_base_sink_wait_preroll (GST_BASE_SINK (sink));
698
699 if (flow_ret == GST_FLOW_OK)
700 continue;
701
702 return flow_ret;
703 }
704
705 err_idx = gst_udp_messsages_find_first_not_sent (messages, num_messages);
706 if (err_idx < 0)
707 break;
708
709 msg = &messages[err_idx];
710 msg_size = gst_udp_calc_message_size (msg);
711
712 GST_LOG_OBJECT (sink, "error sending %u bytes to client %s: %s", msg_size,
713 gst_udp_address_get_string (msg->address, astr, sizeof (astr)),
714 err->message);
715
716 skip = 1;
717 if (msg_size > UDP_MAX_SIZE) {
718 if (!sent_max_size_warning) {
719 GST_ELEMENT_WARNING (sink, RESOURCE, WRITE,
720 ("Attempting to send a UDP packets larger than maximum size "
721 "(%u > %d)", msg_size, UDP_MAX_SIZE),
722 ("Reason: %s", err ? err->message : "unknown reason"));
723 sent_max_size_warning = FALSE;
724 }
725 } else {
726 GST_ELEMENT_WARNING (sink, RESOURCE, WRITE,
727 ("Error sending UDP packets"), ("client %s, reason: %s",
728 gst_udp_address_get_string (msg->address, astr, sizeof (astr)),
729 (err != NULL) ? err->message : "unknown reason"));
730
731 for (i = err_idx + 1; i < num_messages; ++i, ++skip) {
732 if (messages[i].address != msg->address)
733 break;
734 }
735 GST_DEBUG_OBJECT (sink, "skipping %d message(s) to same client", skip);
736 }
737
738 /* ignore any errors and try sending the rest */
739 g_clear_error (&err);
740 ret = skip;
741 }
742
743 g_assert (ret <= num_messages);
744
745 messages += ret;
746 num_messages -= ret;
747 }
748
749 return GST_FLOW_OK;
750 }
751
752 static GstFlowReturn
gst_multiudpsink_render_buffers(GstMultiUDPSink * sink,GstBuffer ** buffers,guint num_buffers,guint8 * mem_nums,guint total_mem_num)753 gst_multiudpsink_render_buffers (GstMultiUDPSink * sink, GstBuffer ** buffers,
754 guint num_buffers, guint8 * mem_nums, guint total_mem_num)
755 {
756 GstOutputMessage *msgs;
757 gboolean send_duplicates;
758 GstUDPClient **clients;
759 GOutputVector *vecs;
760 GstMapInfo *map_infos;
761 GstFlowReturn flow_ret;
762 guint num_addr_v4, num_addr_v6;
763 guint num_addr, num_msgs;
764 guint i, j, mem;
765 gsize size = 0;
766 GList *l;
767
768 send_duplicates = sink->send_duplicates;
769
770 g_mutex_lock (&sink->client_lock);
771
772 if (send_duplicates) {
773 num_addr_v4 = sink->num_v4_all;
774 num_addr_v6 = sink->num_v6_all;
775 } else {
776 num_addr_v4 = sink->num_v4_unique;
777 num_addr_v6 = sink->num_v6_unique;
778 }
779 num_addr = num_addr_v4 + num_addr_v6;
780
781 if (num_addr == 0)
782 goto no_clients;
783
784 clients = g_newa (GstUDPClient *, num_addr);
785 for (l = sink->clients, i = 0; l != NULL; l = l->next) {
786 GstUDPClient *client = l->data;
787
788 clients[i++] = gst_udp_client_ref (client);
789 for (j = 1; send_duplicates && j < client->add_count; ++j)
790 clients[i++] = gst_udp_client_ref (client);
791 }
792 g_assert_cmpuint (i, ==, num_addr);
793
794 g_mutex_unlock (&sink->client_lock);
795
796 GST_LOG_OBJECT (sink, "%u buffers, %u memories -> to be sent to %u clients",
797 num_buffers, total_mem_num, num_addr);
798
799 /* ensure our pre-allocated scratch space arrays are large enough */
800 if (sink->n_vecs < total_mem_num) {
801 sink->n_vecs = GST_ROUND_UP_16 (total_mem_num);
802 g_free (sink->vecs);
803 sink->vecs = g_new (GOutputVector, sink->n_vecs);
804 }
805 vecs = sink->vecs;
806
807 if (sink->n_maps < total_mem_num) {
808 sink->n_maps = GST_ROUND_UP_16 (total_mem_num);
809 g_free (sink->maps);
810 sink->maps = g_new (GstMapInfo, sink->n_maps);
811 }
812 map_infos = sink->maps;
813
814 num_msgs = num_addr * num_buffers;
815 if (sink->n_messages < num_msgs) {
816 sink->n_messages = GST_ROUND_UP_16 (num_msgs);
817 g_free (sink->messages);
818 sink->messages = g_new (GstOutputMessage, sink->n_messages);
819 }
820 msgs = sink->messages;
821
822 /* populate first num_buffers messages with output vectors for the buffers */
823 for (i = 0, mem = 0; i < num_buffers; ++i) {
824 size += fill_vectors (&vecs[mem], &map_infos[mem], mem_nums[i], buffers[i]);
825 msgs[i].vectors = &vecs[mem];
826 msgs[i].num_vectors = mem_nums[i];
827 msgs[i].num_control_messages = 0;
828 msgs[i].bytes_sent = 0;
829 msgs[i].control_messages = NULL;
830 msgs[i].address = clients[0]->addr;
831 mem += mem_nums[i];
832 }
833
834 /* FIXME: how about some locking? (there wasn't any before either, but..) */
835 sink->bytes_to_serve += size;
836
837 /* now copy the pre-filled num_buffer messages over to the next num_buffer
838 * messages for the next client, where we also change the target adddress */
839 for (i = 1; i < num_addr; ++i) {
840 for (j = 0; j < num_buffers; ++j) {
841 msgs[i * num_buffers + j] = msgs[j];
842 msgs[i * num_buffers + j].address = clients[i]->addr;
843 }
844 }
845
846 /* now send it! */
847
848 /* no IPv4 socket? Send it all from the IPv6 socket then.. */
849 if (sink->used_socket == NULL) {
850 flow_ret = gst_multiudpsink_send_messages (sink, sink->used_socket_v6,
851 msgs, num_msgs);
852 } else {
853 guint num_msgs_v4 = num_buffers * num_addr_v4;
854 guint num_msgs_v6 = num_buffers * num_addr_v6;
855
856 /* our client list is sorted with IPv4 clients first and IPv6 ones last */
857 flow_ret = gst_multiudpsink_send_messages (sink, sink->used_socket,
858 msgs, num_msgs_v4);
859
860 if (flow_ret != GST_FLOW_OK)
861 goto cancelled;
862
863 flow_ret = gst_multiudpsink_send_messages (sink, sink->used_socket_v6,
864 msgs + num_msgs_v4, num_msgs_v6);
865 }
866
867 if (flow_ret != GST_FLOW_OK)
868 goto cancelled;
869
870 /* now update stats */
871 g_mutex_lock (&sink->client_lock);
872
873 for (i = 0; i < num_addr; ++i) {
874 GstUDPClient *client = clients[i];
875
876 for (j = 0; j < num_buffers; ++j) {
877 gsize bytes_sent;
878
879 bytes_sent = msgs[i * num_buffers + j].bytes_sent;
880
881 client->bytes_sent += bytes_sent;
882 client->packets_sent++;
883 sink->bytes_served += bytes_sent;
884 }
885 gst_udp_client_unref (client);
886 }
887
888 g_mutex_unlock (&sink->client_lock);
889
890 out:
891
892 for (i = 0; i < mem; ++i)
893 gst_memory_unmap (map_infos[i].memory, &map_infos[i]);
894
895 return flow_ret;
896
897 no_clients:
898 {
899 g_mutex_unlock (&sink->client_lock);
900 GST_LOG_OBJECT (sink, "no clients");
901 return GST_FLOW_OK;
902 }
903 cancelled:
904 {
905 GST_INFO_OBJECT (sink, "cancelled");
906
907 g_mutex_lock (&sink->client_lock);
908 for (i = 0; i < num_addr; ++i)
909 gst_udp_client_unref (clients[i]);
910 g_mutex_unlock (&sink->client_lock);
911 goto out;
912 }
913 }
914
915 static GstFlowReturn
gst_multiudpsink_render_list(GstBaseSink * bsink,GstBufferList * buffer_list)916 gst_multiudpsink_render_list (GstBaseSink * bsink, GstBufferList * buffer_list)
917 {
918 GstMultiUDPSink *sink;
919 GstBuffer **buffers;
920 GstFlowReturn flow;
921 guint8 *mem_nums;
922 guint total_mems;
923 guint i, num_buffers;
924
925 sink = GST_MULTIUDPSINK_CAST (bsink);
926
927 num_buffers = gst_buffer_list_length (buffer_list);
928 if (num_buffers == 0)
929 goto no_data;
930
931 buffers = g_newa (GstBuffer *, num_buffers);
932 mem_nums = g_newa (guint8, num_buffers);
933 for (i = 0, total_mems = 0; i < num_buffers; ++i) {
934 buffers[i] = gst_buffer_list_get (buffer_list, i);
935 mem_nums[i] = gst_buffer_n_memory (buffers[i]);
936 total_mems += mem_nums[i];
937 }
938
939 flow = gst_multiudpsink_render_buffers (sink, buffers, num_buffers,
940 mem_nums, total_mems);
941
942 return flow;
943
944 no_data:
945 {
946 GST_LOG_OBJECT (sink, "empty buffer");
947 return GST_FLOW_OK;
948 }
949 }
950
951 static GstFlowReturn
gst_multiudpsink_render(GstBaseSink * bsink,GstBuffer * buffer)952 gst_multiudpsink_render (GstBaseSink * bsink, GstBuffer * buffer)
953 {
954 GstMultiUDPSink *sink;
955 GstFlowReturn flow;
956 guint8 n_mem;
957
958 sink = GST_MULTIUDPSINK_CAST (bsink);
959
960 n_mem = gst_buffer_n_memory (buffer);
961
962 if (n_mem > 0)
963 flow = gst_multiudpsink_render_buffers (sink, &buffer, 1, &n_mem, n_mem);
964 else
965 flow = GST_FLOW_OK;
966
967 return flow;
968 }
969
970 static void
gst_multiudpsink_set_clients_string(GstMultiUDPSink * sink,const gchar * string)971 gst_multiudpsink_set_clients_string (GstMultiUDPSink * sink,
972 const gchar * string)
973 {
974 gchar **clients;
975 gint i;
976
977 clients = g_strsplit (string, ",", 0);
978
979 g_mutex_lock (&sink->client_lock);
980 /* clear all existing clients */
981 gst_multiudpsink_clear_internal (sink, FALSE);
982 for (i = 0; clients[i]; i++) {
983 gchar *host, *p;
984 gint64 port = 0;
985
986 host = clients[i];
987 p = strstr (clients[i], ":");
988 if (p != NULL) {
989 *p = '\0';
990 port = g_ascii_strtoll (p + 1, NULL, 10);
991 }
992 if (port != 0)
993 gst_multiudpsink_add_internal (sink, host, port, FALSE);
994 }
995 g_mutex_unlock (&sink->client_lock);
996
997 g_strfreev (clients);
998 }
999
1000 static gchar *
gst_multiudpsink_get_clients_string(GstMultiUDPSink * sink)1001 gst_multiudpsink_get_clients_string (GstMultiUDPSink * sink)
1002 {
1003 GString *str;
1004 GList *clients;
1005
1006 str = g_string_new ("");
1007
1008 g_mutex_lock (&sink->client_lock);
1009 clients = sink->clients;
1010 while (clients) {
1011 GstUDPClient *client;
1012 gint count;
1013
1014 client = (GstUDPClient *) clients->data;
1015
1016 clients = g_list_next (clients);
1017
1018 count = client->add_count;
1019 while (count--) {
1020 g_string_append_printf (str, "%s:%d%s", client->host, client->port,
1021 (clients || count > 1 ? "," : ""));
1022 }
1023 }
1024 g_mutex_unlock (&sink->client_lock);
1025
1026 return g_string_free (str, FALSE);
1027 }
1028
1029 static void
gst_multiudpsink_setup_qos_dscp(GstMultiUDPSink * sink,GSocket * socket)1030 gst_multiudpsink_setup_qos_dscp (GstMultiUDPSink * sink, GSocket * socket)
1031 {
1032 /* don't touch on -1 */
1033 if (sink->qos_dscp < 0)
1034 return;
1035
1036 if (socket == NULL)
1037 return;
1038
1039 #ifdef IP_TOS
1040 {
1041 gint tos;
1042 gint fd;
1043
1044 fd = g_socket_get_fd (socket);
1045
1046 GST_DEBUG_OBJECT (sink, "setting TOS to %d", sink->qos_dscp);
1047
1048 /* Extract and shift 6 bits of DSFIELD */
1049 tos = (sink->qos_dscp & 0x3f) << 2;
1050
1051 if (setsockopt (fd, IPPROTO_IP, IP_TOS, &tos, sizeof (tos)) < 0) {
1052 GST_ERROR_OBJECT (sink, "could not set TOS: %s", g_strerror (errno));
1053 }
1054 #ifdef IPV6_TCLASS
1055 if (g_socket_get_family (socket) == G_SOCKET_FAMILY_IPV6) {
1056 if (setsockopt (fd, IPPROTO_IPV6, IPV6_TCLASS, &tos, sizeof (tos)) < 0) {
1057 GST_ERROR_OBJECT (sink, "could not set TCLASS: %s", g_strerror (errno));
1058 }
1059 }
1060 #endif
1061 }
1062 #endif
1063 }
1064
1065 static void
gst_multiudpsink_set_property(GObject * object,guint prop_id,const GValue * value,GParamSpec * pspec)1066 gst_multiudpsink_set_property (GObject * object, guint prop_id,
1067 const GValue * value, GParamSpec * pspec)
1068 {
1069 GstMultiUDPSink *udpsink;
1070
1071 udpsink = GST_MULTIUDPSINK (object);
1072
1073 switch (prop_id) {
1074 case PROP_SOCKET:
1075 if (udpsink->socket != NULL && udpsink->socket != udpsink->used_socket &&
1076 udpsink->close_socket) {
1077 GError *err = NULL;
1078
1079 if (!g_socket_close (udpsink->socket, &err)) {
1080 GST_ERROR ("failed to close socket %p: %s", udpsink->socket,
1081 err->message);
1082 g_clear_error (&err);
1083 }
1084 }
1085 if (udpsink->socket)
1086 g_object_unref (udpsink->socket);
1087 udpsink->socket = g_value_dup_object (value);
1088 GST_DEBUG_OBJECT (udpsink, "setting socket to %p", udpsink->socket);
1089 break;
1090 case PROP_SOCKET_V6:
1091 if (udpsink->socket_v6 != NULL
1092 && udpsink->socket_v6 != udpsink->used_socket_v6
1093 && udpsink->close_socket) {
1094 GError *err = NULL;
1095
1096 if (!g_socket_close (udpsink->socket_v6, &err)) {
1097 GST_ERROR ("failed to close socket %p: %s", udpsink->socket_v6,
1098 err->message);
1099 g_clear_error (&err);
1100 }
1101 }
1102 if (udpsink->socket_v6)
1103 g_object_unref (udpsink->socket_v6);
1104 udpsink->socket_v6 = g_value_dup_object (value);
1105 GST_DEBUG_OBJECT (udpsink, "setting socket to %p", udpsink->socket_v6);
1106 break;
1107 case PROP_CLOSE_SOCKET:
1108 udpsink->close_socket = g_value_get_boolean (value);
1109 break;
1110 case PROP_CLIENTS:
1111 gst_multiudpsink_set_clients_string (udpsink, g_value_get_string (value));
1112 break;
1113 case PROP_AUTO_MULTICAST:
1114 udpsink->auto_multicast = g_value_get_boolean (value);
1115 break;
1116 case PROP_MULTICAST_IFACE:
1117 g_free (udpsink->multi_iface);
1118
1119 if (g_value_get_string (value) == NULL)
1120 udpsink->multi_iface = g_strdup (DEFAULT_MULTICAST_IFACE);
1121 else
1122 udpsink->multi_iface = g_value_dup_string (value);
1123 break;
1124 case PROP_TTL:
1125 udpsink->ttl = g_value_get_int (value);
1126 break;
1127 case PROP_TTL_MC:
1128 udpsink->ttl_mc = g_value_get_int (value);
1129 break;
1130 case PROP_LOOP:
1131 udpsink->loop = g_value_get_boolean (value);
1132 break;
1133 case PROP_FORCE_IPV4:
1134 udpsink->force_ipv4 = g_value_get_boolean (value);
1135 break;
1136 case PROP_QOS_DSCP:
1137 udpsink->qos_dscp = g_value_get_int (value);
1138 gst_multiudpsink_setup_qos_dscp (udpsink, udpsink->used_socket);
1139 gst_multiudpsink_setup_qos_dscp (udpsink, udpsink->used_socket_v6);
1140 break;
1141 case PROP_SEND_DUPLICATES:
1142 udpsink->send_duplicates = g_value_get_boolean (value);
1143 break;
1144 case PROP_BUFFER_SIZE:
1145 udpsink->buffer_size = g_value_get_int (value);
1146 break;
1147 case PROP_BIND_ADDRESS:
1148 g_free (udpsink->bind_address);
1149 udpsink->bind_address = g_value_dup_string (value);
1150 break;
1151 case PROP_BIND_PORT:
1152 udpsink->bind_port = g_value_get_int (value);
1153 break;
1154 default:
1155 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1156 break;
1157 }
1158 }
1159
1160 static void
gst_multiudpsink_get_property(GObject * object,guint prop_id,GValue * value,GParamSpec * pspec)1161 gst_multiudpsink_get_property (GObject * object, guint prop_id, GValue * value,
1162 GParamSpec * pspec)
1163 {
1164 GstMultiUDPSink *udpsink;
1165
1166 udpsink = GST_MULTIUDPSINK (object);
1167
1168 switch (prop_id) {
1169 case PROP_BYTES_TO_SERVE:
1170 g_value_set_uint64 (value, udpsink->bytes_to_serve);
1171 break;
1172 case PROP_BYTES_SERVED:
1173 g_value_set_uint64 (value, udpsink->bytes_served);
1174 break;
1175 case PROP_SOCKET:
1176 g_value_set_object (value, udpsink->socket);
1177 break;
1178 case PROP_SOCKET_V6:
1179 g_value_set_object (value, udpsink->socket_v6);
1180 break;
1181 case PROP_CLOSE_SOCKET:
1182 g_value_set_boolean (value, udpsink->close_socket);
1183 break;
1184 case PROP_USED_SOCKET:
1185 g_value_set_object (value, udpsink->used_socket);
1186 break;
1187 case PROP_USED_SOCKET_V6:
1188 g_value_set_object (value, udpsink->used_socket_v6);
1189 break;
1190 case PROP_CLIENTS:
1191 g_value_take_string (value,
1192 gst_multiudpsink_get_clients_string (udpsink));
1193 break;
1194 case PROP_AUTO_MULTICAST:
1195 g_value_set_boolean (value, udpsink->auto_multicast);
1196 break;
1197 case PROP_MULTICAST_IFACE:
1198 g_value_set_string (value, udpsink->multi_iface);
1199 break;
1200 case PROP_TTL:
1201 g_value_set_int (value, udpsink->ttl);
1202 break;
1203 case PROP_TTL_MC:
1204 g_value_set_int (value, udpsink->ttl_mc);
1205 break;
1206 case PROP_LOOP:
1207 g_value_set_boolean (value, udpsink->loop);
1208 break;
1209 case PROP_FORCE_IPV4:
1210 g_value_set_boolean (value, udpsink->force_ipv4);
1211 break;
1212 case PROP_QOS_DSCP:
1213 g_value_set_int (value, udpsink->qos_dscp);
1214 break;
1215 case PROP_SEND_DUPLICATES:
1216 g_value_set_boolean (value, udpsink->send_duplicates);
1217 break;
1218 case PROP_BUFFER_SIZE:
1219 g_value_set_int (value, udpsink->buffer_size);
1220 break;
1221 case PROP_BIND_ADDRESS:
1222 g_value_set_string (value, udpsink->bind_address);
1223 break;
1224 case PROP_BIND_PORT:
1225 g_value_set_int (value, udpsink->bind_port);
1226 break;
1227 default:
1228 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1229 break;
1230 }
1231 }
1232
1233 static gboolean
gst_multiudpsink_configure_client(GstMultiUDPSink * sink,GstUDPClient * client)1234 gst_multiudpsink_configure_client (GstMultiUDPSink * sink,
1235 GstUDPClient * client)
1236 {
1237 GInetSocketAddress *saddr = G_INET_SOCKET_ADDRESS (client->addr);
1238 GInetAddress *addr = g_inet_socket_address_get_address (saddr);
1239 GSocketFamily family = g_socket_address_get_family (G_SOCKET_ADDRESS (saddr));
1240 GSocket *socket;
1241 GError *err = NULL;
1242
1243 GST_DEBUG_OBJECT (sink, "configuring client %p", client);
1244
1245 if (family == G_SOCKET_FAMILY_IPV6 && !sink->used_socket_v6)
1246 goto invalid_family;
1247
1248 /* Select socket to send from for this address */
1249 if (family == G_SOCKET_FAMILY_IPV6 || !sink->used_socket)
1250 socket = sink->used_socket_v6;
1251 else
1252 socket = sink->used_socket;
1253
1254 if (g_inet_address_get_is_multicast (addr)) {
1255 GST_DEBUG_OBJECT (sink, "we have a multicast client %p", client);
1256 if (sink->auto_multicast) {
1257 GST_DEBUG_OBJECT (sink, "autojoining group");
1258 if (!g_socket_join_multicast_group (socket, addr, FALSE,
1259 sink->multi_iface, &err))
1260 goto join_group_failed;
1261 }
1262 GST_DEBUG_OBJECT (sink, "setting loop to %d", sink->loop);
1263 g_socket_set_multicast_loopback (socket, sink->loop);
1264 GST_DEBUG_OBJECT (sink, "setting ttl to %d", sink->ttl_mc);
1265 g_socket_set_multicast_ttl (socket, sink->ttl_mc);
1266 } else {
1267 GST_DEBUG_OBJECT (sink, "setting unicast ttl to %d", sink->ttl);
1268 g_socket_set_ttl (socket, sink->ttl);
1269 }
1270 return TRUE;
1271
1272 /* ERRORS */
1273 join_group_failed:
1274 {
1275 gst_multiudpsink_stop (GST_BASE_SINK (sink));
1276 GST_ELEMENT_ERROR (sink, RESOURCE, SETTINGS, (NULL),
1277 ("Could not join multicast group: %s",
1278 err ? err->message : "unknown reason"));
1279 g_clear_error (&err);
1280 return FALSE;
1281 }
1282 invalid_family:
1283 {
1284 gst_multiudpsink_stop (GST_BASE_SINK (sink));
1285 GST_ELEMENT_ERROR (sink, RESOURCE, SETTINGS, (NULL),
1286 ("Invalid address family (got %d)", family));
1287 return FALSE;
1288 }
1289 }
1290
1291 /* create a socket for sending to remote machine */
1292 static gboolean
gst_multiudpsink_start(GstBaseSink * bsink)1293 gst_multiudpsink_start (GstBaseSink * bsink)
1294 {
1295 GstMultiUDPSink *sink;
1296 GList *clients;
1297 GstUDPClient *client;
1298 GError *err = NULL;
1299
1300 sink = GST_MULTIUDPSINK (bsink);
1301
1302 sink->external_socket = FALSE;
1303
1304 if (sink->socket) {
1305 GST_DEBUG_OBJECT (sink, "using configured socket");
1306 if (g_socket_get_family (sink->socket) == G_SOCKET_FAMILY_IPV6) {
1307 sink->used_socket_v6 = G_SOCKET (g_object_ref (sink->socket));
1308 sink->external_socket = TRUE;
1309 } else {
1310 sink->used_socket = G_SOCKET (g_object_ref (sink->socket));
1311 sink->external_socket = TRUE;
1312 }
1313 }
1314
1315 if (sink->socket_v6) {
1316 GST_DEBUG_OBJECT (sink, "using configured IPv6 socket");
1317 g_return_val_if_fail (!sink->socket || g_socket_get_family (sink->socket) !=
1318 G_SOCKET_FAMILY_IPV6, FALSE);
1319
1320 if (sink->used_socket_v6 && sink->used_socket_v6 != sink->socket_v6) {
1321 GST_ERROR_OBJECT (sink,
1322 "Provided different IPv6 sockets in socket and socket-v6 properties");
1323 return FALSE;
1324 }
1325
1326 sink->used_socket_v6 = G_SOCKET (g_object_ref (sink->socket_v6));
1327 sink->external_socket = TRUE;
1328 }
1329
1330 if (!sink->used_socket && !sink->used_socket_v6) {
1331 GSocketAddress *bind_addr;
1332 GInetAddress *bind_iaddr;
1333
1334 if (sink->bind_address) {
1335 GSocketFamily family;
1336
1337 bind_iaddr = g_inet_address_new_from_string (sink->bind_address);
1338 if (!bind_iaddr) {
1339 GList *results;
1340 GResolver *resolver;
1341
1342 resolver = g_resolver_get_default ();
1343 results =
1344 g_resolver_lookup_by_name (resolver, sink->bind_address,
1345 sink->cancellable, &err);
1346 if (!results) {
1347 g_object_unref (resolver);
1348 goto name_resolve;
1349 }
1350 bind_iaddr = G_INET_ADDRESS (g_object_ref (results->data));
1351 g_resolver_free_addresses (results);
1352 g_object_unref (resolver);
1353 }
1354
1355 bind_addr = g_inet_socket_address_new (bind_iaddr, sink->bind_port);
1356 g_object_unref (bind_iaddr);
1357 family = g_socket_address_get_family (G_SOCKET_ADDRESS (bind_addr));
1358
1359 if ((sink->used_socket =
1360 g_socket_new (family, G_SOCKET_TYPE_DATAGRAM,
1361 G_SOCKET_PROTOCOL_UDP, &err)) == NULL) {
1362 g_object_unref (bind_addr);
1363 goto no_socket;
1364 }
1365
1366 g_socket_bind (sink->used_socket, bind_addr, TRUE, &err);
1367 g_object_unref (bind_addr);
1368 if (err != NULL)
1369 goto bind_error;
1370 } else {
1371 /* create sender sockets if none available */
1372 if ((sink->used_socket = g_socket_new (G_SOCKET_FAMILY_IPV4,
1373 G_SOCKET_TYPE_DATAGRAM, G_SOCKET_PROTOCOL_UDP, &err)) == NULL)
1374 goto no_socket;
1375
1376 bind_iaddr = g_inet_address_new_any (G_SOCKET_FAMILY_IPV4);
1377 bind_addr = g_inet_socket_address_new (bind_iaddr, sink->bind_port);
1378 g_socket_bind (sink->used_socket, bind_addr, TRUE, &err);
1379 g_object_unref (bind_addr);
1380 g_object_unref (bind_iaddr);
1381 if (err != NULL)
1382 goto bind_error;
1383
1384 if ((sink->used_socket_v6 = g_socket_new (G_SOCKET_FAMILY_IPV6,
1385 G_SOCKET_TYPE_DATAGRAM, G_SOCKET_PROTOCOL_UDP,
1386 &err)) == NULL) {
1387 GST_INFO_OBJECT (sink, "Failed to create IPv6 socket: %s",
1388 err->message);
1389 g_clear_error (&err);
1390 } else {
1391 bind_iaddr = g_inet_address_new_any (G_SOCKET_FAMILY_IPV6);
1392 bind_addr = g_inet_socket_address_new (bind_iaddr, sink->bind_port);
1393 g_socket_bind (sink->used_socket_v6, bind_addr, TRUE, &err);
1394 g_object_unref (bind_addr);
1395 g_object_unref (bind_iaddr);
1396 if (err != NULL)
1397 goto bind_error;
1398 }
1399 }
1400 }
1401 #ifdef SO_SNDBUF
1402 {
1403 socklen_t len;
1404 gint sndsize, ret;
1405
1406 len = sizeof (sndsize);
1407 if (sink->buffer_size != 0) {
1408 sndsize = sink->buffer_size;
1409
1410 GST_DEBUG_OBJECT (sink, "setting udp buffer of %d bytes", sndsize);
1411 /* set buffer size, Note that on Linux this is typically limited to a
1412 * maximum of around 100K. Also a minimum of 128 bytes is required on
1413 * Linux. */
1414
1415 if (sink->used_socket) {
1416 ret =
1417 setsockopt (g_socket_get_fd (sink->used_socket), SOL_SOCKET,
1418 SO_SNDBUF, (void *) &sndsize, len);
1419 if (ret != 0) {
1420 GST_ELEMENT_WARNING (sink, RESOURCE, SETTINGS, (NULL),
1421 ("Could not create a buffer of requested %d bytes, %d: %s",
1422 sndsize, ret, g_strerror (errno)));
1423 }
1424 }
1425
1426 if (sink->used_socket_v6) {
1427 ret =
1428 setsockopt (g_socket_get_fd (sink->used_socket_v6), SOL_SOCKET,
1429 SO_SNDBUF, (void *) &sndsize, len);
1430 if (ret != 0) {
1431 GST_ELEMENT_WARNING (sink, RESOURCE, SETTINGS, (NULL),
1432 ("Could not create a buffer of requested %d bytes, %d: %s",
1433 sndsize, ret, g_strerror (errno)));
1434 }
1435 }
1436 }
1437
1438 /* read the value of the receive buffer. Note that on linux this returns 2x the
1439 * value we set because the kernel allocates extra memory for metadata.
1440 * The default on Linux is about 100K (which is about 50K without metadata) */
1441 if (sink->used_socket) {
1442 ret =
1443 getsockopt (g_socket_get_fd (sink->used_socket), SOL_SOCKET,
1444 SO_SNDBUF, (void *) &sndsize, &len);
1445 if (ret == 0)
1446 GST_DEBUG_OBJECT (sink, "have UDP buffer of %d bytes", sndsize);
1447 else
1448 GST_DEBUG_OBJECT (sink, "could not get UDP buffer size");
1449 }
1450
1451 if (sink->used_socket_v6) {
1452 ret =
1453 getsockopt (g_socket_get_fd (sink->used_socket_v6), SOL_SOCKET,
1454 SO_SNDBUF, (void *) &sndsize, &len);
1455 if (ret == 0)
1456 GST_DEBUG_OBJECT (sink, "have UDPv6 buffer of %d bytes", sndsize);
1457 else
1458 GST_DEBUG_OBJECT (sink, "could not get UDPv6 buffer size");
1459 }
1460 }
1461 #endif
1462
1463 #ifdef SO_BINDTODEVICE
1464 if (sink->multi_iface) {
1465 if (sink->used_socket) {
1466 if (setsockopt (g_socket_get_fd (sink->used_socket), SOL_SOCKET,
1467 SO_BINDTODEVICE, sink->multi_iface,
1468 strlen (sink->multi_iface)) < 0)
1469 GST_WARNING_OBJECT (sink, "setsockopt SO_BINDTODEVICE failed: %s",
1470 strerror (errno));
1471 }
1472 if (sink->used_socket_v6) {
1473 if (setsockopt (g_socket_get_fd (sink->used_socket_v6), SOL_SOCKET,
1474 SO_BINDTODEVICE, sink->multi_iface,
1475 strlen (sink->multi_iface)) < 0)
1476 GST_WARNING_OBJECT (sink, "setsockopt SO_BINDTODEVICE failed (v6): %s",
1477 strerror (errno));
1478 }
1479 }
1480 #endif
1481
1482 if (sink->used_socket)
1483 g_socket_set_broadcast (sink->used_socket, TRUE);
1484 if (sink->used_socket_v6)
1485 g_socket_set_broadcast (sink->used_socket_v6, TRUE);
1486
1487 sink->bytes_to_serve = 0;
1488 sink->bytes_served = 0;
1489
1490 gst_multiudpsink_setup_qos_dscp (sink, sink->used_socket);
1491 gst_multiudpsink_setup_qos_dscp (sink, sink->used_socket_v6);
1492
1493 /* look for multicast clients and join multicast groups appropriately
1494 set also ttl and multicast loopback delivery appropriately */
1495 for (clients = sink->clients; clients; clients = g_list_next (clients)) {
1496 client = (GstUDPClient *) clients->data;
1497
1498 if (!gst_multiudpsink_configure_client (sink, client))
1499 return FALSE;
1500 }
1501 return TRUE;
1502
1503 /* ERRORS */
1504 no_socket:
1505 {
1506 GST_ELEMENT_ERROR (sink, RESOURCE, FAILED, (NULL),
1507 ("Could not create socket: %s", err->message));
1508 g_clear_error (&err);
1509 return FALSE;
1510 }
1511 bind_error:
1512 {
1513 GST_ELEMENT_ERROR (sink, RESOURCE, FAILED, (NULL),
1514 ("Failed to bind socket: %s", err->message));
1515 g_clear_error (&err);
1516 return FALSE;
1517 }
1518 name_resolve:
1519 {
1520 GST_ELEMENT_ERROR (sink, RESOURCE, FAILED, (NULL),
1521 ("Failed to resolve bind address %s: %s", sink->bind_address,
1522 err->message));
1523 g_clear_error (&err);
1524 return FALSE;
1525 }
1526 }
1527
1528 static gboolean
gst_multiudpsink_stop(GstBaseSink * bsink)1529 gst_multiudpsink_stop (GstBaseSink * bsink)
1530 {
1531 GstMultiUDPSink *udpsink;
1532
1533 udpsink = GST_MULTIUDPSINK (bsink);
1534
1535 if (udpsink->used_socket) {
1536 if (udpsink->close_socket || !udpsink->external_socket) {
1537 GError *err = NULL;
1538
1539 if (!g_socket_close (udpsink->used_socket, &err)) {
1540 GST_ERROR_OBJECT (udpsink, "Failed to close socket: %s", err->message);
1541 g_clear_error (&err);
1542 }
1543 }
1544
1545 g_object_unref (udpsink->used_socket);
1546 udpsink->used_socket = NULL;
1547 }
1548
1549 if (udpsink->used_socket_v6) {
1550 if (udpsink->close_socket || !udpsink->external_socket) {
1551 GError *err = NULL;
1552
1553 if (!g_socket_close (udpsink->used_socket_v6, &err)) {
1554 GST_ERROR_OBJECT (udpsink, "Failed to close socket: %s", err->message);
1555 g_clear_error (&err);
1556 }
1557 }
1558
1559 g_object_unref (udpsink->used_socket_v6);
1560 udpsink->used_socket_v6 = NULL;
1561 }
1562
1563 return TRUE;
1564 }
1565
1566 static gint
gst_udp_client_compare_socket_family(GstUDPClient * a,GstUDPClient * b)1567 gst_udp_client_compare_socket_family (GstUDPClient * a, GstUDPClient * b)
1568 {
1569 GSocketFamily fa = g_socket_address_get_family (a->addr);
1570 GSocketFamily fb = g_socket_address_get_family (b->addr);
1571
1572 if (fa == fb)
1573 return 0;
1574
1575 /* a should go before b */
1576 if (fa == G_SOCKET_FAMILY_IPV4 && fb == G_SOCKET_FAMILY_IPV6)
1577 return -1;
1578
1579 /* b should go before a */
1580 return 1;
1581 }
1582
1583 static void
gst_multiudpsink_add_internal(GstMultiUDPSink * sink,const gchar * host,gint port,gboolean lock)1584 gst_multiudpsink_add_internal (GstMultiUDPSink * sink, const gchar * host,
1585 gint port, gboolean lock)
1586 {
1587 GSocketFamily family;
1588 GstUDPClient *client;
1589 GstUDPClient udpclient;
1590 GTimeVal now;
1591 GList *find;
1592
1593 udpclient.host = (gchar *) host;
1594 udpclient.port = port;
1595
1596 GST_DEBUG_OBJECT (sink, "adding client on host %s, port %d", host, port);
1597
1598 if (lock)
1599 g_mutex_lock (&sink->client_lock);
1600
1601 find = g_list_find_custom (sink->clients, &udpclient,
1602 (GCompareFunc) client_compare);
1603
1604 if (!find) {
1605 find = g_list_find_custom (sink->clients_to_be_removed, &udpclient,
1606 (GCompareFunc) client_compare);
1607 if (find)
1608 gst_udp_client_ref (find->data);
1609 }
1610
1611 if (find) {
1612 client = (GstUDPClient *) find->data;
1613
1614 family = g_socket_address_get_family (client->addr);
1615
1616 GST_DEBUG_OBJECT (sink, "found %d existing clients with host %s, port %d",
1617 client->add_count, host, port);
1618 } else {
1619 client = gst_udp_client_new (sink, host, port);
1620 if (!client)
1621 goto error;
1622
1623 family = g_socket_address_get_family (client->addr);
1624
1625 g_get_current_time (&now);
1626 client->connect_time = GST_TIMEVAL_TO_TIME (now);
1627
1628 if (sink->used_socket)
1629 gst_multiudpsink_configure_client (sink, client);
1630
1631 GST_DEBUG_OBJECT (sink, "add client with host %s, port %d", host, port);
1632
1633 /* keep IPv4 clients at the beginning, and IPv6 at the end, we can make
1634 * use of this in gst_multiudpsink_render_buffers() */
1635 sink->clients = g_list_insert_sorted (sink->clients, client,
1636 (GCompareFunc) gst_udp_client_compare_socket_family);
1637
1638 if (family == G_SOCKET_FAMILY_IPV4)
1639 ++sink->num_v4_unique;
1640 else
1641 ++sink->num_v6_unique;
1642 }
1643
1644 ++client->add_count;
1645
1646 if (family == G_SOCKET_FAMILY_IPV4)
1647 ++sink->num_v4_all;
1648 else
1649 ++sink->num_v6_all;
1650
1651 if (lock)
1652 g_mutex_unlock (&sink->client_lock);
1653
1654 g_signal_emit (G_OBJECT (sink),
1655 gst_multiudpsink_signals[SIGNAL_CLIENT_ADDED], 0, host, port);
1656
1657 GST_DEBUG_OBJECT (sink, "added client on host %s, port %d", host, port);
1658 return;
1659
1660 /* ERRORS */
1661 error:
1662 {
1663 GST_DEBUG_OBJECT (sink, "did not add client on host %s, port %d", host,
1664 port);
1665 if (lock)
1666 g_mutex_unlock (&sink->client_lock);
1667 return;
1668 }
1669 }
1670
1671 void
gst_multiudpsink_add(GstMultiUDPSink * sink,const gchar * host,gint port)1672 gst_multiudpsink_add (GstMultiUDPSink * sink, const gchar * host, gint port)
1673 {
1674 gst_multiudpsink_add_internal (sink, host, port, TRUE);
1675 }
1676
1677 void
gst_multiudpsink_remove(GstMultiUDPSink * sink,const gchar * host,gint port)1678 gst_multiudpsink_remove (GstMultiUDPSink * sink, const gchar * host, gint port)
1679 {
1680 GSocketFamily family;
1681 GList *find;
1682 GstUDPClient udpclient;
1683 GstUDPClient *client;
1684 GTimeVal now;
1685
1686 udpclient.host = (gchar *) host;
1687 udpclient.port = port;
1688
1689 g_mutex_lock (&sink->client_lock);
1690 find = g_list_find_custom (sink->clients, &udpclient,
1691 (GCompareFunc) client_compare);
1692 if (!find)
1693 goto not_found;
1694
1695 client = (GstUDPClient *) find->data;
1696
1697 GST_DEBUG_OBJECT (sink, "found %d clients with host %s, port %d",
1698 client->add_count, host, port);
1699
1700 --client->add_count;
1701
1702 family = g_socket_address_get_family (client->addr);
1703 if (family == G_SOCKET_FAMILY_IPV4)
1704 --sink->num_v4_all;
1705 else
1706 --sink->num_v6_all;
1707
1708 if (client->add_count == 0) {
1709 GInetSocketAddress *saddr = G_INET_SOCKET_ADDRESS (client->addr);
1710 GInetAddress *addr = g_inet_socket_address_get_address (saddr);
1711 GSocket *socket;
1712
1713 /* Select socket to send from for this address */
1714 if (family == G_SOCKET_FAMILY_IPV6 || !sink->used_socket)
1715 socket = sink->used_socket_v6;
1716 else
1717 socket = sink->used_socket;
1718
1719 GST_DEBUG_OBJECT (sink, "remove client with host %s, port %d", host, port);
1720
1721 g_get_current_time (&now);
1722 client->disconnect_time = GST_TIMEVAL_TO_TIME (now);
1723
1724 if (socket && sink->auto_multicast
1725 && g_inet_address_get_is_multicast (addr)) {
1726 GError *err = NULL;
1727
1728 if (!g_socket_leave_multicast_group (socket, addr, FALSE,
1729 sink->multi_iface, &err)) {
1730 GST_DEBUG_OBJECT (sink, "Failed to leave multicast group: %s",
1731 err->message);
1732 g_clear_error (&err);
1733 }
1734 }
1735
1736 if (family == G_SOCKET_FAMILY_IPV4)
1737 --sink->num_v4_unique;
1738 else
1739 --sink->num_v6_unique;
1740
1741 /* Keep state consistent for streaming thread, so remove from client list,
1742 * but keep it around until after the signal has been emitted, in case a
1743 * callback wants to get stats for that client or so */
1744 sink->clients = g_list_delete_link (sink->clients, find);
1745
1746 sink->clients_to_be_removed =
1747 g_list_prepend (sink->clients_to_be_removed, client);
1748
1749 /* Unlock to emit signal before we delete the actual client */
1750 g_mutex_unlock (&sink->client_lock);
1751 g_signal_emit (G_OBJECT (sink),
1752 gst_multiudpsink_signals[SIGNAL_CLIENT_REMOVED], 0, host, port);
1753 g_mutex_lock (&sink->client_lock);
1754
1755 sink->clients_to_be_removed =
1756 g_list_remove (sink->clients_to_be_removed, client);
1757
1758 gst_udp_client_unref (client);
1759 }
1760 g_mutex_unlock (&sink->client_lock);
1761
1762 return;
1763
1764 /* ERRORS */
1765 not_found:
1766 {
1767 g_mutex_unlock (&sink->client_lock);
1768 GST_WARNING_OBJECT (sink, "client at host %s, port %d not found",
1769 host, port);
1770 return;
1771 }
1772 }
1773
1774 static void
gst_multiudpsink_clear_internal(GstMultiUDPSink * sink,gboolean lock)1775 gst_multiudpsink_clear_internal (GstMultiUDPSink * sink, gboolean lock)
1776 {
1777 GST_DEBUG_OBJECT (sink, "clearing");
1778 /* we only need to remove the client structure, there is no additional
1779 * socket or anything to free for UDP */
1780 if (lock)
1781 g_mutex_lock (&sink->client_lock);
1782 g_list_foreach (sink->clients, (GFunc) gst_udp_client_unref, sink);
1783 g_list_free (sink->clients);
1784 sink->clients = NULL;
1785 sink->num_v4_unique = 0;
1786 sink->num_v4_all = 0;
1787 sink->num_v6_unique = 0;
1788 sink->num_v6_all = 0;
1789 if (lock)
1790 g_mutex_unlock (&sink->client_lock);
1791 }
1792
1793 void
gst_multiudpsink_clear(GstMultiUDPSink * sink)1794 gst_multiudpsink_clear (GstMultiUDPSink * sink)
1795 {
1796 gst_multiudpsink_clear_internal (sink, TRUE);
1797 }
1798
1799 GstStructure *
gst_multiudpsink_get_stats(GstMultiUDPSink * sink,const gchar * host,gint port)1800 gst_multiudpsink_get_stats (GstMultiUDPSink * sink, const gchar * host,
1801 gint port)
1802 {
1803 GstUDPClient *client;
1804 GstStructure *result = NULL;
1805 GstUDPClient udpclient;
1806 GList *find;
1807
1808 udpclient.host = (gchar *) host;
1809 udpclient.port = port;
1810
1811 g_mutex_lock (&sink->client_lock);
1812
1813 find = g_list_find_custom (sink->clients, &udpclient,
1814 (GCompareFunc) client_compare);
1815
1816 if (!find)
1817 find = g_list_find_custom (sink->clients_to_be_removed, &udpclient,
1818 (GCompareFunc) client_compare);
1819
1820 if (!find)
1821 goto not_found;
1822
1823 GST_DEBUG_OBJECT (sink, "stats for client with host %s, port %d", host, port);
1824
1825 client = (GstUDPClient *) find->data;
1826
1827 result = gst_structure_new_empty ("multiudpsink-stats");
1828
1829 gst_structure_set (result,
1830 "bytes-sent", G_TYPE_UINT64, client->bytes_sent,
1831 "packets-sent", G_TYPE_UINT64, client->packets_sent,
1832 "connect-time", G_TYPE_UINT64, client->connect_time,
1833 "disconnect-time", G_TYPE_UINT64, client->disconnect_time, NULL);
1834
1835 g_mutex_unlock (&sink->client_lock);
1836
1837 return result;
1838
1839 /* ERRORS */
1840 not_found:
1841 {
1842 g_mutex_unlock (&sink->client_lock);
1843 GST_WARNING_OBJECT (sink, "client with host %s, port %d not found",
1844 host, port);
1845 /* Apparently (see comment in gstmultifdsink.c) returning NULL from here may
1846 * confuse/break python bindings */
1847 return gst_structure_new_empty ("multiudpsink-stats");
1848 }
1849 }
1850
1851 static gboolean
gst_multiudpsink_unlock(GstBaseSink * bsink)1852 gst_multiudpsink_unlock (GstBaseSink * bsink)
1853 {
1854 GstMultiUDPSink *sink;
1855
1856 sink = GST_MULTIUDPSINK (bsink);
1857
1858 g_cancellable_cancel (sink->cancellable);
1859
1860 return TRUE;
1861 }
1862
1863 static gboolean
gst_multiudpsink_unlock_stop(GstBaseSink * bsink)1864 gst_multiudpsink_unlock_stop (GstBaseSink * bsink)
1865 {
1866 GstMultiUDPSink *sink;
1867
1868 sink = GST_MULTIUDPSINK (bsink);
1869
1870 gst_multiudpsink_free_cancellable (sink);
1871 gst_multiudpsink_create_cancellable (sink);
1872
1873 return TRUE;
1874 }
1875