2 * Copyright (C) <2005> Philippe Khalaf <burger@speedy.org>
3 * Copyright (C) <2005> Nokia Corporation <kai.vehmanen@nokia.com>
4 * Copyright (C) <2006> Joni Valtanen <joni.valtanen@movial.fi>
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Library General Public
8 * License as published by the Free Software Foundation; either
9 * version 2 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Library General Public License for more details.
16 * You should have received a copy of the GNU Library General Public
17 * License along with this library; if not, write to the
18 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
19 * Boston, MA 02111-1307, USA.
25 #include "gstudp-marshal.h"
26 #include "gstdynudpsink.h"
35 #ifdef HAVE_SYS_TIME_H
38 #include <sys/types.h>
39 #include <gst/netbuffer/gstnetbuffer.h>
41 GST_DEBUG_CATEGORY_STATIC (dynudpsink_debug);
42 #define GST_CAT_DEFAULT (dynudpsink_debug)
44 static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
49 /* elementfactory information */
50 static const GstElementDetails gst_dynudpsink_details =
51 GST_ELEMENT_DETAILS ("UDP packet sender",
53 "Send data over the network via UDP",
54 "Philippe Khalaf <burger@speedy.org>");
56 /* DynUDPSink signals and args */
68 #define UDP_DEFAULT_SOCKFD -1
69 #define UDP_DEFAULT_CLOSEFD TRUE
78 #define CLOSE_IF_REQUESTED(udpctx) \
80 if ((!udpctx->externalfd) || (udpctx->externalfd && udpctx->closefd)) { \
81 CLOSE_SOCKET(udpctx->sock); \
82 if (udpctx->sock == udpctx->sockfd) \
83 udpctx->sockfd = UDP_DEFAULT_SOCKFD; \
88 static void gst_dynudpsink_base_init (gpointer g_class);
89 static void gst_dynudpsink_class_init (GstDynUDPSink * klass);
90 static void gst_dynudpsink_init (GstDynUDPSink * udpsink);
91 static void gst_dynudpsink_finalize (GObject * object);
93 static GstFlowReturn gst_dynudpsink_render (GstBaseSink * sink,
95 static void gst_dynudpsink_close (GstDynUDPSink * sink);
96 static GstStateChangeReturn gst_dynudpsink_change_state (GstElement * element,
97 GstStateChange transition);
99 static void gst_dynudpsink_set_property (GObject * object, guint prop_id,
100 const GValue * value, GParamSpec * pspec);
101 static void gst_dynudpsink_get_property (GObject * object, guint prop_id,
102 GValue * value, GParamSpec * pspec);
104 static GstElementClass *parent_class = NULL;
106 static guint gst_dynudpsink_signals[LAST_SIGNAL] = { 0 };
109 gst_dynudpsink_get_type (void)
111 static GType dynudpsink_type = 0;
113 if (!dynudpsink_type) {
114 static const GTypeInfo dynudpsink_info = {
115 sizeof (GstDynUDPSinkClass),
116 gst_dynudpsink_base_init,
118 (GClassInitFunc) gst_dynudpsink_class_init,
121 sizeof (GstDynUDPSink),
123 (GInstanceInitFunc) gst_dynudpsink_init,
128 g_type_register_static (GST_TYPE_BASE_SINK, "GstDynUDPSink",
129 &dynudpsink_info, 0);
131 return dynudpsink_type;
135 gst_dynudpsink_base_init (gpointer g_class)
137 GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
139 gst_element_class_add_pad_template (element_class,
140 gst_static_pad_template_get (&sink_template));
142 gst_element_class_set_details (element_class, &gst_dynudpsink_details);
146 gst_dynudpsink_class_init (GstDynUDPSink * klass)
148 GObjectClass *gobject_class;
149 GstElementClass *gstelement_class;
150 GstBaseSinkClass *gstbasesink_class;
152 gobject_class = (GObjectClass *) klass;
153 gstelement_class = (GstElementClass *) klass;
154 gstbasesink_class = (GstBaseSinkClass *) klass;
156 parent_class = g_type_class_peek_parent (klass);
158 gobject_class->set_property = gst_dynudpsink_set_property;
159 gobject_class->get_property = gst_dynudpsink_get_property;
160 gobject_class->finalize = gst_dynudpsink_finalize;
162 gst_dynudpsink_signals[SIGNAL_GET_STATS] =
163 g_signal_new ("get-stats", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
164 G_STRUCT_OFFSET (GstDynUDPSinkClass, get_stats),
165 NULL, NULL, gst_udp_marshal_BOXED__STRING_INT, G_TYPE_VALUE_ARRAY, 2,
166 G_TYPE_STRING, G_TYPE_INT);
168 g_object_class_install_property (gobject_class, PROP_SOCKFD,
169 g_param_spec_int ("sockfd", "socket handle",
170 "Socket to use for UDP sending. (-1 == allocate)",
171 -1, G_MAXINT16, UDP_DEFAULT_SOCKFD, G_PARAM_READWRITE));
172 g_object_class_install_property (gobject_class, PROP_CLOSEFD,
173 g_param_spec_boolean ("closefd", "Close sockfd",
174 "Close sockfd if passed as property on state change",
175 UDP_DEFAULT_CLOSEFD, G_PARAM_READWRITE));
177 gstelement_class->change_state = gst_dynudpsink_change_state;
179 gstbasesink_class->render = gst_dynudpsink_render;
181 GST_DEBUG_CATEGORY_INIT (dynudpsink_debug, "dynudpsink", 0, "UDP sink");
185 gst_dynudpsink_init (GstDynUDPSink * sink)
189 sink->sockfd = UDP_DEFAULT_SOCKFD;
190 sink->closefd = UDP_DEFAULT_CLOSEFD;
191 sink->externalfd = FALSE;
197 gst_dynudpsink_finalize (GObject * object)
199 GstDynUDPSink *udpsink;
201 udpsink = GST_DYNUDPSINK (object);
203 if (udpsink->sockfd >= 0 && udpsink->closefd)
204 CLOSE_SOCKET (udpsink->sockfd);
206 G_OBJECT_CLASS (parent_class)->finalize (object);
208 WSA_CLEANUP (object);
212 gst_dynudpsink_render (GstBaseSink * bsink, GstBuffer * buffer)
217 GstNetBuffer *netbuf;
218 struct sockaddr_in theiraddr;
222 memset (&theiraddr, 0, sizeof (theiraddr));
224 if (GST_IS_NETBUFFER (buffer)) {
225 netbuf = GST_NETBUFFER (buffer);
227 GST_DEBUG ("Received buffer is not a GstNetBuffer, skipping");
231 sink = GST_DYNUDPSINK (bsink);
233 size = GST_BUFFER_SIZE (netbuf);
234 data = GST_BUFFER_DATA (netbuf);
236 GST_DEBUG ("about to send %d bytes", size);
238 // let's get the address from the netbuffer
239 gst_netaddress_get_ip4_address (&netbuf->to, &destaddr, &destport);
241 GST_DEBUG ("sending %d bytes to client %d port %d", size, destaddr, destport);
243 theiraddr.sin_family = AF_INET;
244 theiraddr.sin_addr.s_addr = destaddr;
245 theiraddr.sin_port = destport;
247 ret = sendto (sink->sock, (char *) data, size, 0,
249 ret = sendto (sink->sock, data, size, 0,
251 (struct sockaddr *) &theiraddr, sizeof (theiraddr));
254 if (errno != EINTR && errno != EAGAIN) {
259 GST_DEBUG ("sent %d bytes", size);
265 GST_DEBUG ("got send error %s (%d)", g_strerror (errno), errno);
266 return GST_FLOW_ERROR;
271 gst_dynudpsink_set_property (GObject * object, guint prop_id,
272 const GValue * value, GParamSpec * pspec)
274 GstDynUDPSink *udpsink;
276 udpsink = GST_DYNUDPSINK (object);
280 if (udpsink->sockfd >= 0 && udpsink->sockfd != udpsink->sock &&
282 CLOSE_SOCKET (udpsink->sockfd);
283 udpsink->sockfd = g_value_get_int (value);
284 GST_DEBUG ("setting SOCKFD to %d", udpsink->sockfd);
287 udpsink->closefd = g_value_get_boolean (value);
291 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
297 gst_dynudpsink_get_property (GObject * object, guint prop_id, GValue * value,
300 GstDynUDPSink *udpsink;
302 udpsink = GST_DYNUDPSINK (object);
306 g_value_set_int (value, udpsink->sockfd);
309 g_value_set_boolean (value, udpsink->closefd);
312 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
318 /* create a socket for sending to remote machine */
320 gst_dynudpsink_init_send (GstDynUDPSink * sink)
324 if (sink->sockfd == -1) {
325 /* create sender socket if none available */
326 if ((sink->sock = socket (AF_INET, SOCK_DGRAM, 0)) < 0)
330 if (setsockopt (sink->sock, SOL_SOCKET, SO_BROADCAST, &bc_val,
331 sizeof (bc_val)) < 0)
334 sink->externalfd = TRUE;
336 sink->sock = sink->sockfd;
337 sink->externalfd = TRUE;
349 perror ("setsockopt");
350 CLOSE_IF_REQUESTED (sink);
356 gst_dynudpsink_get_stats (GstDynUDPSink * sink, const gchar * host, gint port)
362 gst_dynudpsink_close (GstDynUDPSink * sink)
364 CLOSE_IF_REQUESTED (sink);
367 static GstStateChangeReturn
368 gst_dynudpsink_change_state (GstElement * element, GstStateChange transition)
370 GstStateChangeReturn ret;
373 sink = GST_DYNUDPSINK (element);
375 switch (transition) {
376 case GST_STATE_CHANGE_READY_TO_PAUSED:
377 if (!gst_dynudpsink_init_send (sink))
384 ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
386 switch (transition) {
387 case GST_STATE_CHANGE_PAUSED_TO_READY:
388 gst_dynudpsink_close (sink);
398 return GST_STATE_CHANGE_FAILURE;