gst/udp/gstudpsrc.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19
  20
  21 #include "gstudpsrc.h"
  22
  23 #define UDP_DEFAULT_PORT        4951
  24
  25 /* elementfactory information */
  26 GstElementDetails gst_udpsrc_details = {
  27   "UDP packet receiver",
  28   "Source/Network",
  29   "LGPL",
  30   "",
  31   VERSION,
  32   "Wim Taymans <wim.taymans@chello.be>",
  33   "(C) 2001",
  34 };
  35
  36 /* UDPSrc signals and args */
  37 enum {
  38   /* FILL ME */
  39   LAST_SIGNAL
  40 };
  41
  42 enum {
  43   ARG_0,
  44   ARG_PORT,
  45   /* FILL ME */
  46 };
  47
  48 static void             gst_udpsrc_class_init           (GstUDPSrc *klass);
  49 static void             gst_udpsrc_init                 (GstUDPSrc *udpsrc);
  50
  51 static GstBuffer*       gst_udpsrc_get                  (GstPad *pad);
  52 static GstElementStateReturn
  53                         gst_udpsrc_change_state         (GstElement *element);
  54
  55 static void             gst_udpsrc_set_property         (GObject *object, guint prop_id,
  56                                                          const GValue *value, GParamSpec *pspec);
  57 static void             gst_udpsrc_get_property         (GObject *object, guint prop_id,
  58                                                          GValue *value, GParamSpec *pspec);
  59
  60 static GstElementClass *parent_class = NULL;
  61 /*static guint gst_udpsrc_signals[LAST_SIGNAL] = { 0 }; */
  62
  63 GType
  64 gst_udpsrc_get_type (void)
  65 {
  66   static GType udpsrc_type = 0;
  67
  68
  69   if (!udpsrc_type) {
  70     static const GTypeInfo udpsrc_info = {
  71       sizeof(GstUDPSrcClass),
  72       NULL,
  73       NULL,
  74       (GClassInitFunc)gst_udpsrc_class_init,
  75       NULL,
  76       NULL,
  77       sizeof(GstUDPSrc),
  78       0,
  79       (GInstanceInitFunc)gst_udpsrc_init,
  80       NULL
  81     };
  82     udpsrc_type = g_type_register_static (GST_TYPE_ELEMENT, "GstUDPSrc", &udpsrc_info, 0);
  83   }
  84   return udpsrc_type;
  85 }
  86
  87 static void
  88 gst_udpsrc_class_init (GstUDPSrc *klass)
  89 {
  90   GObjectClass *gobject_class;
  91   GstElementClass *gstelement_class;
  92
  93   gobject_class = (GObjectClass*) klass;
  94   gstelement_class = (GstElementClass*) klass;
  95
  96   parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
  97
  98   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_PORT,
  99     g_param_spec_int ("port", "port", "The port to receive the packets from",
 100                        0, 32768, UDP_DEFAULT_PORT, G_PARAM_READWRITE));
 101
 102   gobject_class->set_property = gst_udpsrc_set_property;
 103   gobject_class->get_property = gst_udpsrc_get_property;
 104
 105   gstelement_class->change_state = gst_udpsrc_change_state;
 106 }
 107
 108
 109 static void
 110 gst_udpsrc_init (GstUDPSrc *udpsrc)
 111 {
 112   /* create the src and src pads */
 113   udpsrc->srcpad = gst_pad_new ("src", GST_PAD_SRC);
 114   gst_element_add_pad (GST_ELEMENT (udpsrc), udpsrc->srcpad);
 115   gst_pad_set_get_function (udpsrc->srcpad, gst_udpsrc_get);
 116
 117   udpsrc->port = UDP_DEFAULT_PORT;
 118 }
 119
 120 static GstBuffer*
 121 gst_udpsrc_get (GstPad *pad)
 122 {
 123   GstUDPSrc *udpsrc;
 124   GstBuffer *outbuf;
 125   struct sockaddr_in tmpaddr;
 126   int len, numbytes;
 127   fd_set read_fds;
 128
 129   g_return_val_if_fail (pad != NULL, NULL);
 130   g_return_val_if_fail (GST_IS_PAD (pad), NULL);
 131
 132   udpsrc = GST_UDPSRC (GST_OBJECT_PARENT (pad));
 133
 134   FD_ZERO (&read_fds);
 135   FD_SET (udpsrc->control_sock, &read_fds);
 136   FD_SET (udpsrc->sock, &read_fds);
 137
 138   if (select (udpsrc->control_sock+1, &read_fds, NULL, NULL, NULL) > 0) {
 139     if (FD_ISSET (udpsrc->control_sock, &read_fds)) {
 140 #ifndef GST_DISABLE_LOADSAVE
 141       guchar *buf;
 142       int ret;
 143       int fdread;
 144       struct sockaddr addr;
 145       socklen_t len;
 146       xmlDocPtr doc;
 147       GstCaps *caps;
 148
 149       buf = g_malloc (1024*10);
 150
 151       len = sizeof (struct sockaddr);
 152       fdread = accept (udpsrc->control_sock, &addr, &len);
 153       if (fdread < 0) {
 154         perror ("accept");
 155       }
 156
 157       ret = read (fdread, buf, 1024*10);
 158       if (ret < 0) {
 159         perror ("read");
 160       }
 161       buf[ret] = '\0';
 162       doc = xmlParseMemory(buf, ret);
 163       caps = gst_caps_load_thyself(doc->xmlRootNode);
 164
 165       gst_pad_try_set_caps (udpsrc->srcpad, caps);
 166
 167 #endif
 168       outbuf = NULL;
 169     }
 170     else {
 171       outbuf = gst_buffer_new ();
 172       GST_BUFFER_DATA (outbuf) = g_malloc (24000);
 173       GST_BUFFER_SIZE (outbuf) = 24000;
 174
 175       numbytes = recvfrom (udpsrc->sock, GST_BUFFER_DATA (outbuf),
 176                   GST_BUFFER_SIZE (outbuf), 0, (struct sockaddr *)&tmpaddr, &len);
 177
 178       if (numbytes != -1) {
 179         GST_BUFFER_SIZE (outbuf) = numbytes;
 180       }
 181       else {
 182         perror ("recvfrom");
 183         gst_buffer_unref (outbuf);
 184         outbuf = NULL;
 185       }
 186
 187     }
 188   }
 189   else {
 190     perror ("select");
 191     outbuf = NULL;
 192   }
 193   return outbuf;
 194 }
 195
 196
 197 static void
 198 gst_udpsrc_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
 199 {
 200   GstUDPSrc *udpsrc;
 201
 202   /* it's not null if we got it, but it might not be ours */
 203   g_return_if_fail(GST_IS_UDPSRC(object));
 204   udpsrc = GST_UDPSRC(object);
 205
 206   switch (prop_id) {
 207     case ARG_PORT:
 208         udpsrc->port = g_value_get_int (value);
 209       break;
 210     default:
 211       break;
 212   }
 213 }
 214
 215 static void
 216 gst_udpsrc_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
 217 {
 218   GstUDPSrc *udpsrc;
 219
 220   /* it's not null if we got it, but it might not be ours */
 221   g_return_if_fail(GST_IS_UDPSRC(object));
 222   udpsrc = GST_UDPSRC(object);
 223
 224   switch (prop_id) {
 225     case ARG_PORT:
 226       g_value_set_int (value, udpsrc->port);
 227       break;
 228     default:
 229       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 230       break;
 231   }
 232 }
 233
 234 /* create a socket for sending to remote machine */
 235 static gboolean
 236 gst_udpsrc_init_receive (GstUDPSrc *src)
 237 {
 238   bzero (&src->myaddr, sizeof (src->myaddr));
 239   src->myaddr.sin_family = AF_INET;         /* host byte order */
 240   src->myaddr.sin_port = htons (src->port);     /* short, network byte order */
 241   src->myaddr.sin_addr.s_addr = INADDR_ANY;
 242
 243   if ((src->sock = socket (AF_INET, SOCK_DGRAM, 0)) == -1) {
 244     perror("socket");
 245     return FALSE;
 246   }
 247
 248   if (bind (src->sock, (struct sockaddr *) &src->myaddr, sizeof (src->myaddr)) == -1) {
 249     perror("bind");
 250     return FALSE;
 251   }
 252
 253   if ((src->control_sock = socket (AF_INET, SOCK_STREAM, 0)) == -1) {
 254     perror("control_socket");
 255     return FALSE;
 256   }
 257
 258   if (bind (src->control_sock, (struct sockaddr *) &src->myaddr, sizeof (src->myaddr)) == -1) {
 259     perror("control_bind");
 260     return FALSE;
 261   }
 262   if (listen (src->control_sock, 5) == -1) {
 263     perror("listen");
 264     return FALSE;
 265   }
 266   fcntl (src->control_sock, F_SETFL, O_NONBLOCK);
 267
 268   GST_FLAG_SET (src, GST_UDPSRC_OPEN);
 269
 270   return TRUE;
 271 }
 272
 273 static void
 274 gst_udpsrc_close (GstUDPSrc *src)
 275 {
 276   close (src->sock);
 277
 278   GST_FLAG_UNSET (src, GST_UDPSRC_OPEN);
 279 }
 280
 281 static GstElementStateReturn
 282 gst_udpsrc_change_state (GstElement *element)
 283 {
 284   g_return_val_if_fail (GST_IS_UDPSRC (element), GST_STATE_FAILURE);
 285
 286   if (GST_STATE_PENDING (element) == GST_STATE_NULL) {
 287     if (GST_FLAG_IS_SET (element, GST_UDPSRC_OPEN))
 288       gst_udpsrc_close (GST_UDPSRC (element));
 289   } else {
 290     if (!GST_FLAG_IS_SET (element, GST_UDPSRC_OPEN)) {
 291       if (!gst_udpsrc_init_receive (GST_UDPSRC (element)))
 292         return GST_STATE_FAILURE;
 293     }
 294   }
 295
 296   if (GST_ELEMENT_CLASS (parent_class)->change_state)
 297     return GST_ELEMENT_CLASS (parent_class)->change_state (element);
 298
 299   return GST_STATE_SUCCESS;
 300 }
 301