diff -r 000000000000 -r 0e761a78d257 gst_plugins_base/gst/tcp/gsttcpserversink.c --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/gst_plugins_base/gst/tcp/gsttcpserversink.c Thu Dec 17 08:53:32 2009 +0200 @@ -0,0 +1,348 @@ +/* GStreamer + * Copyright (C) <1999> Erik Walthinsen + * Copyright (C) <2004> Thomas Vander Stichele + * + * 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. + */ + +/** + * SECTION:tcpserversink + * @short_description: a sink that acts as a TCP server and sends data to + * multiple clients + * @see_also: #multifdsink + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif +#ifdef __SYMBIAN32__ +#include "gst/gst-i18n-plugin.h" +#else +#include +#endif +#include /* memset */ + +#include + +#ifdef HAVE_FIONREAD_IN_SYS_FILIO +#include +#endif + +#include "gsttcp.h" +#include "gsttcpserversink.h" +#include "gsttcp-marshal.h" + +#define TCP_BACKLOG 5 + +/* elementfactory information */ +static const GstElementDetails gst_tcp_server_sink_details = +GST_ELEMENT_DETAILS ("TCP server sink", + "Sink/Network", + "Send data as a server over the network via TCP", + "Thomas Vander Stichele "); + +GST_DEBUG_CATEGORY_STATIC (tcpserversink_debug); +#define GST_CAT_DEFAULT (tcpserversink_debug) + +enum +{ + ARG_0, + ARG_HOST, + ARG_PORT, +}; + +static void gst_tcp_server_sink_finalize (GObject * gobject); + +static gboolean gst_tcp_server_sink_handle_wait (GstMultiFdSink * sink, + GstPoll * set); +static gboolean gst_tcp_server_sink_init_send (GstMultiFdSink * this); +static gboolean gst_tcp_server_sink_close (GstMultiFdSink * this); +static void gst_tcp_server_sink_removed (GstMultiFdSink * sink, int fd); + +static void gst_tcp_server_sink_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec); +static void gst_tcp_server_sink_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec); + + +GST_BOILERPLATE (GstTCPServerSink, gst_tcp_server_sink, GstMultiFdSink, + GST_TYPE_MULTI_FD_SINK); + + +static void +gst_tcp_server_sink_base_init (gpointer g_class) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); + + gst_element_class_set_details (element_class, &gst_tcp_server_sink_details); +} + +static void +gst_tcp_server_sink_class_init (GstTCPServerSinkClass * klass) +{ + GObjectClass *gobject_class; + GstMultiFdSinkClass *gstmultifdsink_class; + + gobject_class = (GObjectClass *) klass; + gstmultifdsink_class = (GstMultiFdSinkClass *) klass; + + gobject_class->set_property = gst_tcp_server_sink_set_property; + gobject_class->get_property = gst_tcp_server_sink_get_property; + gobject_class->finalize = gst_tcp_server_sink_finalize; + + g_object_class_install_property (gobject_class, ARG_HOST, + g_param_spec_string ("host", "host", "The host/IP to send the packets to", + TCP_DEFAULT_HOST, G_PARAM_READWRITE)); + g_object_class_install_property (gobject_class, ARG_PORT, + g_param_spec_int ("port", "port", "The port to send the packets to", + 0, TCP_HIGHEST_PORT, TCP_DEFAULT_PORT, G_PARAM_READWRITE)); + + gstmultifdsink_class->init = gst_tcp_server_sink_init_send; + gstmultifdsink_class->wait = gst_tcp_server_sink_handle_wait; + gstmultifdsink_class->close = gst_tcp_server_sink_close; + gstmultifdsink_class->removed = gst_tcp_server_sink_removed; + + GST_DEBUG_CATEGORY_INIT (tcpserversink_debug, "tcpserversink", 0, "TCP sink"); +} + +static void +gst_tcp_server_sink_init (GstTCPServerSink * this, + GstTCPServerSinkClass * klass) +{ + this->server_port = TCP_DEFAULT_PORT; + /* should support as minimum 576 for IPV4 and 1500 for IPV6 */ + /* this->mtu = 1500; */ + this->host = g_strdup (TCP_DEFAULT_HOST); + + this->server_sock.fd = -1; +} + +static void +gst_tcp_server_sink_finalize (GObject * gobject) +{ + GstTCPServerSink *this = GST_TCP_SERVER_SINK (gobject); + + g_free (this->host); + + G_OBJECT_CLASS (parent_class)->finalize (gobject); +} + +/* handle a read request on the server, + * which indicates a new client connection */ +static gboolean +gst_tcp_server_sink_handle_server_read (GstTCPServerSink * sink) +{ + /* new client */ + int client_sock_fd; + struct sockaddr_in client_address; + unsigned int client_address_len; + + /* For some stupid reason, client_address and client_address_len has to be + * zeroed */ + memset (&client_address, 0, sizeof (client_address)); + client_address_len = 0; + + client_sock_fd = + accept (sink->server_sock.fd, (struct sockaddr *) &client_address, + &client_address_len); + if (client_sock_fd == -1) { + GST_ELEMENT_ERROR (sink, RESOURCE, OPEN_WRITE, (NULL), + ("Could not accept client on server socket %d: %s (%d)", + sink->server_sock.fd, g_strerror (errno), errno)); + return FALSE; + } + + gst_multi_fd_sink_add (GST_MULTI_FD_SINK (sink), client_sock_fd); + + GST_DEBUG_OBJECT (sink, "added new client ip %s with fd %d", + inet_ntoa (client_address.sin_addr), client_sock_fd); + + return TRUE; +} + +static void +gst_tcp_server_sink_removed (GstMultiFdSink * sink, int fd) +{ +#ifndef GST_DISABLE_GST_DEBUG + GstTCPServerSink *this = GST_TCP_SERVER_SINK (sink); +#endif + + GST_LOG_OBJECT (this, "closing fd %d", fd); + if (close (fd) < 0) { + GST_WARNING_OBJECT (this, "error closing fd %d: %s", fd, + g_strerror (errno)); + } +} + +static gboolean +gst_tcp_server_sink_handle_wait (GstMultiFdSink * sink, GstPoll * set) +{ + GstTCPServerSink *this = GST_TCP_SERVER_SINK (sink); + + if (gst_poll_fd_can_read (set, &this->server_sock)) { + /* handle new client connection on server socket */ + if (!gst_tcp_server_sink_handle_server_read (this)) { + GST_ELEMENT_ERROR (this, RESOURCE, READ, (NULL), + ("client connection failed: %s", g_strerror (errno))); + return FALSE; + } + } + return TRUE; +} + +static void +gst_tcp_server_sink_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec) +{ + GstTCPServerSink *sink; + + g_return_if_fail (GST_IS_TCP_SERVER_SINK (object)); + sink = GST_TCP_SERVER_SINK (object); + + switch (prop_id) { + case ARG_HOST: + if (!g_value_get_string (value)) { + g_warning ("host property cannot be NULL"); + break; + } + g_free (sink->host); + sink->host = g_strdup (g_value_get_string (value)); + break; + case ARG_PORT: + sink->server_port = g_value_get_int (value); + break; + + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static void +gst_tcp_server_sink_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec) +{ + GstTCPServerSink *sink; + + g_return_if_fail (GST_IS_TCP_SERVER_SINK (object)); + sink = GST_TCP_SERVER_SINK (object); + + switch (prop_id) { + case ARG_HOST: + g_value_set_string (value, sink->host); + break; + case ARG_PORT: + g_value_set_int (value, sink->server_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_tcp_server_sink_init_send (GstMultiFdSink * parent) +{ + int ret; + GstTCPServerSink *this = GST_TCP_SERVER_SINK (parent); + + /* create sending server socket */ + if ((this->server_sock.fd = socket (AF_INET, SOCK_STREAM, 0)) == -1) { + GST_ELEMENT_ERROR (this, RESOURCE, OPEN_WRITE, (NULL), GST_ERROR_SYSTEM); + return FALSE; + } + GST_DEBUG_OBJECT (this, "opened sending server socket with fd %d", + this->server_sock.fd); + + /* make address reusable */ + ret = 1; + if (setsockopt (this->server_sock.fd, SOL_SOCKET, SO_REUSEADDR, + (void *) &ret, sizeof (ret)) < 0) { + gst_tcp_socket_close (&this->server_sock); + GST_ELEMENT_ERROR (this, RESOURCE, SETTINGS, (NULL), + ("Could not setsockopt: %s", g_strerror (errno))); + return FALSE; + } + /* keep connection alive; avoids SIGPIPE during write */ + ret = 1; + if (setsockopt (this->server_sock.fd, SOL_SOCKET, SO_KEEPALIVE, + (void *) &ret, sizeof (ret)) < 0) { + gst_tcp_socket_close (&this->server_sock); + GST_ELEMENT_ERROR (this, RESOURCE, SETTINGS, (NULL), + ("Could not setsockopt: %s", g_strerror (errno))); + return FALSE; + } + + /* name the socket */ + memset (&this->server_sin, 0, sizeof (this->server_sin)); + this->server_sin.sin_family = AF_INET; /* network socket */ + this->server_sin.sin_port = htons (this->server_port); /* on port */ + this->server_sin.sin_addr.s_addr = htonl (INADDR_ANY); /* for hosts */ + + /* bind it */ + GST_DEBUG_OBJECT (this, "binding server socket to address"); + ret = bind (this->server_sock.fd, (struct sockaddr *) &this->server_sin, + sizeof (this->server_sin)); + + if (ret) { + gst_tcp_socket_close (&this->server_sock); + switch (errno) { + default: + GST_ELEMENT_ERROR (this, RESOURCE, OPEN_READ, (NULL), + ("bind on port %d failed: %s", this->server_port, + g_strerror (errno))); + return FALSE; + break; + } + } + + /* set the server socket to nonblocking */ + fcntl (this->server_sock.fd, F_SETFL, O_NONBLOCK); + + GST_DEBUG_OBJECT (this, "listening on server socket %d with queue of %d", + this->server_sock.fd, TCP_BACKLOG); + if (listen (this->server_sock.fd, TCP_BACKLOG) == -1) { + gst_tcp_socket_close (&this->server_sock); + GST_ELEMENT_ERROR (this, RESOURCE, OPEN_READ, (NULL), + ("Could not listen on server socket: %s", g_strerror (errno))); + return FALSE; + } + GST_DEBUG_OBJECT (this, + "listened on server socket %d, returning from connection setup", + this->server_sock.fd); + + gst_poll_add_fd (parent->fdset, &this->server_sock); + gst_poll_fd_ctl_read (parent->fdset, &this->server_sock, TRUE); + + return TRUE; +} + +static gboolean +gst_tcp_server_sink_close (GstMultiFdSink * parent) +{ + GstTCPServerSink *this = GST_TCP_SERVER_SINK (parent); + + if (this->server_sock.fd != -1) { + gst_poll_remove_fd (parent->fdset, &this->server_sock); + + close (this->server_sock.fd); + this->server_sock.fd = -1; + } + return TRUE; +}