Initial revision

Original commit message from CVS:
Initial revision

1 files changed, 300 insertions, 0 deletions

diff --git a/gst/udp/gstudpsrc.c b/gst/udp/gstudpsrc.c
new file mode 100644
index 00000000..4400a064
--- /dev/null
+++ b/gst/udp/gstudpsrc.c
@@ -0,0 +1,300 @@
+/* Gnome-Streamer
+ * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
+ *
+ * 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., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+
+#include "gstudpsrc.h"
+
+#define UDP_DEFAULT_PORT	4951
+
+/* elementfactory information */
+GstElementDetails gst_udpsrc_details = {
+  "UDP packet receiver",
+  "Transport/",
+  "",
+  VERSION,
+  "Wim Taymans <wim.taymans@chello.be>",
+  "(C) 2001",
+};
+
+/* UDPSrc signals and args */
+enum {
+  /* FILL ME */
+  LAST_SIGNAL
+};
+
+enum {
+  ARG_0,
+  ARG_PORT,
+  /* FILL ME */
+};
+
+static void		gst_udpsrc_class_init		(GstUDPSrc *klass);
+static void		gst_udpsrc_init			(GstUDPSrc *udpsrc);
+
+static GstBuffer*	gst_udpsrc_get			(GstPad *pad);
+static GstElementStateReturn
+			gst_udpsrc_change_state 	(GstElement *element);
+
+static void 		gst_udpsrc_set_property 	(GObject *object, guint prop_id, 
+							 const GValue *value, GParamSpec *pspec);
+static void 		gst_udpsrc_get_property 	(GObject *object, guint prop_id, 
+							 GValue *value, GParamSpec *pspec);
+
+static GstElementClass *parent_class = NULL;
+//static guint gst_udpsrc_signals[LAST_SIGNAL] = { 0 };
+
+GType
+gst_udpsrc_get_type (void)
+{
+  static GType udpsrc_type = 0;
+
+
+  if (!udpsrc_type) {
+    static const GTypeInfo udpsrc_info = {
+      sizeof(GstUDPSrcClass),
+      NULL,
+      NULL,
+      (GClassInitFunc)gst_udpsrc_class_init,
+      NULL,
+      NULL,
+      sizeof(GstUDPSrc),
+      0,
+      (GInstanceInitFunc)gst_udpsrc_init,
+      NULL
+    };
+    udpsrc_type = g_type_register_static (GST_TYPE_ELEMENT, "GstUDPSrc", &udpsrc_info, 0);
+  }
+  return udpsrc_type;
+}
+
+static void
+gst_udpsrc_class_init (GstUDPSrc *klass)
+{
+  GObjectClass *gobject_class;
+  GstElementClass *gstelement_class;
+
+  gobject_class = (GObjectClass*) klass;
+  gstelement_class = (GstElementClass*) klass;
+
+  parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
+
+  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_PORT,
+    g_param_spec_int ("port", "port", "The port to receive the packets from",
+                       0, 32768, UDP_DEFAULT_PORT, G_PARAM_READWRITE)); 
+
+  gobject_class->set_property = gst_udpsrc_set_property;
+  gobject_class->get_property = gst_udpsrc_get_property;
+
+  gstelement_class->change_state = gst_udpsrc_change_state;
+}
+
+
+static void
+gst_udpsrc_init (GstUDPSrc *udpsrc)
+{
+  /* create the src and src pads */
+  udpsrc->srcpad = gst_pad_new ("src", GST_PAD_SRC);
+  gst_element_add_pad (GST_ELEMENT (udpsrc), udpsrc->srcpad);
+  gst_pad_set_get_function (udpsrc->srcpad, gst_udpsrc_get);
+
+  udpsrc->port = UDP_DEFAULT_PORT;
+}
+
+static GstBuffer*
+gst_udpsrc_get (GstPad *pad)
+{
+  GstUDPSrc *udpsrc;
+  GstBuffer *outbuf;
+  struct sockaddr_in tmpaddr;
+  int len, numbytes;
+  fd_set read_fds;
+
+  g_return_val_if_fail (pad != NULL, NULL);
+  g_return_val_if_fail (GST_IS_PAD (pad), NULL);
+
+  udpsrc = GST_UDPSRC (GST_OBJECT_PARENT (pad));
+
+  FD_ZERO (&read_fds);
+  FD_SET (udpsrc->control_sock, &read_fds);
+  FD_SET (udpsrc->sock, &read_fds);
+
+  if (select (udpsrc->control_sock+1, &read_fds, NULL, NULL, NULL) > 0) {
+    if (FD_ISSET (udpsrc->control_sock, &read_fds)) {
+#ifndef GST_DISABLE_LOADSAVE
+      guchar *buf;
+      int ret;
+      int fdread;
+      struct sockaddr addr;
+      socklen_t len;
+      xmlDocPtr doc;
+      GstCaps *caps;
+
+      buf = g_malloc (1024*10);
+
+      len = sizeof (struct sockaddr);
+      fdread = accept (udpsrc->control_sock, &addr, &len);
+      if (fdread < 0) {
+	perror ("accept");
+      }
+      
+      ret = read (fdread, buf, 1024*10);
+      if (ret < 0) {
+	perror ("read");
+      }
+      buf[ret] = '\0';
+      doc = xmlParseMemory(buf, ret);
+      caps = gst_caps_load_thyself(doc->xmlRootNode);
+      
+      gst_pad_set_caps (udpsrc->srcpad, caps);
+
+#endif
+      outbuf = NULL;
+    }
+    else {
+      outbuf = gst_buffer_new ();
+      GST_BUFFER_DATA (outbuf) = g_malloc (24000);
+      GST_BUFFER_SIZE (outbuf) = 24000;
+
+      numbytes = recvfrom (udpsrc->sock, GST_BUFFER_DATA (outbuf),
+		  GST_BUFFER_SIZE (outbuf), 0, (struct sockaddr *)&tmpaddr, &len);
+
+      if (numbytes != -1) {
+        GST_BUFFER_SIZE (outbuf) = numbytes;
+      }
+      else {
+	perror ("recvfrom");
+        gst_buffer_unref (outbuf);
+        outbuf = NULL;
+      }
+  
+    }
+  }
+  else {
+    perror ("select");
+    outbuf = NULL;
+  }
+  return outbuf;
+}
+
+
+static void
+gst_udpsrc_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
+{
+  GstUDPSrc *udpsrc;
+
+  /* it's not null if we got it, but it might not be ours */
+  g_return_if_fail(GST_IS_UDPSRC(object));
+  udpsrc = GST_UDPSRC(object);
+
+  switch (prop_id) {
+    case ARG_PORT:
+        udpsrc->port = g_value_get_int (value);
+      break;
+    default:
+      break;
+  }
+}
+
+static void
+gst_udpsrc_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
+{
+  GstUDPSrc *udpsrc;
+
+  /* it's not null if we got it, but it might not be ours */
+  g_return_if_fail(GST_IS_UDPSRC(object));
+  udpsrc = GST_UDPSRC(object);
+
+  switch (prop_id) {
+    case ARG_PORT:
+      g_value_set_int (value, udpsrc->port);
+      break;
+    default:
+      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+      break;
+  }
+}
+
+// create a socket for sending to remote machine
+static gboolean
+gst_udpsrc_init_receive (GstUDPSrc *src)
+{
+  bzero (&src->myaddr, sizeof (src->myaddr));
+  src->myaddr.sin_family = AF_INET;         // host byte order
+  src->myaddr.sin_port = htons (src->port);     // short, network byte order
+  src->myaddr.sin_addr.s_addr = INADDR_ANY;
+
+  if ((src->sock = socket (AF_INET, SOCK_DGRAM, 0)) == -1) {
+    perror("socket");
+    return FALSE;
+  }
+
+  if (bind (src->sock, (struct sockaddr *) &src->myaddr, sizeof (src->myaddr)) == -1) {
+    perror("bind");
+    return FALSE;
+  }
+
+  if ((src->control_sock = socket (AF_INET, SOCK_STREAM, 0)) == -1) {
+    perror("control_socket");
+    return FALSE;
+  }
+
+  if (bind (src->control_sock, (struct sockaddr *) &src->myaddr, sizeof (src->myaddr)) == -1) {
+    perror("control_bind");
+    return FALSE;
+  }
+  if (listen (src->control_sock, 5) == -1) {
+    perror("listen");
+    return FALSE;
+  }
+  fcntl (src->control_sock, F_SETFL, O_NONBLOCK);
+
+  GST_FLAG_SET (src, GST_UDPSRC_OPEN);
+
+  return TRUE;
+}
+
+static void
+gst_udpsrc_close (GstUDPSrc *src)
+{
+  close (src->sock);
+
+  GST_FLAG_UNSET (src, GST_UDPSRC_OPEN);
+}
+
+static GstElementStateReturn
+gst_udpsrc_change_state (GstElement *element)
+{
+  g_return_val_if_fail (GST_IS_UDPSRC (element), GST_STATE_FAILURE);
+
+  if (GST_STATE_PENDING (element) == GST_STATE_NULL) {
+    if (GST_FLAG_IS_SET (element, GST_UDPSRC_OPEN))
+      gst_udpsrc_close (GST_UDPSRC (element));
+  } else {
+    if (!GST_FLAG_IS_SET (element, GST_UDPSRC_OPEN)) {
+      if (!gst_udpsrc_init_receive (GST_UDPSRC (element)))
+        return GST_STATE_FAILURE;
+    }
+  }
+
+  if (GST_ELEMENT_CLASS (parent_class)->change_state)
+    return GST_ELEMENT_CLASS (parent_class)->change_state (element);
+
+  return GST_STATE_SUCCESS;
+}
+