From c7fd1b11e03b327d1e8c5cfecdcc2c907b393739 Mon Sep 17 00:00:00 2001 From: Christian Pointner Date: Wed, 24 Sep 2014 23:40:12 +0200 Subject: added udp socket handling to own file --- src/sydra.c | 121 +----------------------------------------------------------- 1 file changed, 2 insertions(+), 119 deletions(-) (limited to 'src/sydra.c') diff --git a/src/sydra.c b/src/sydra.c index 680836d..44abc38 100644 --- a/src/sydra.c +++ b/src/sydra.c @@ -53,6 +53,8 @@ #include "daemon.h" #include "utils.h" #include "pipelines.h" +#include "udp.h" + static gboolean sig_handler_terminate(gpointer user_data) { @@ -163,125 +165,6 @@ static gboolean bus_call(GstBus *bus, GstMessage *msg, gpointer data) return TRUE; } - - - - -static void udp_add_client(struct sockaddr_storage *addr, socklen_t addrlen, struct udp_sink* sink) -{ - struct addr_port* client = g_new(struct addr_port, 1); - if(!client) return; - - switch(addr->ss_family) - { - case AF_INET: client->type_ = IPv4; client->port_ = ntohs(((struct sockaddr_in*)addr)->sin_port); break; - case AF_INET6: client->type_ = IPv6; client->port_ = ntohs(((struct sockaddr_in6*)addr)->sin6_port); break; - default: return; - } - - int errcode = getnameinfo((struct sockaddr *)addr, addrlen, client->addr_, sizeof(client->addr_), NULL, 0, NI_NUMERICHOST | NI_NUMERICSERV); - if (errcode != 0) { - log_printf(WARNING, "getnameinfo() error: %s", gai_strerror(errcode)); - return; - } - gchar* name = gst_element_get_name(sink->udp_); - if(!g_list_find_custom(sink->clients_, client, cmp_addr_port)) { - log_printf(INFO, "adding host %s%c%d to list of receivers for element %s", client->addr_, client->type_ == IPv4 ? ':' : '.', client->port_, name); - g_signal_emit_by_name(G_OBJECT(sink->udp_), "add", client->addr_, client->port_, NULL); - sink->clients_ = g_list_append(sink->clients_, client); - } else { - log_printf(DEBUG, "not adding host %s%c%d to list of receivers for element %s - already added", client->addr_, client->type_ == IPv4 ? ':' : '.', client->port_, name); - g_free(client); - } - g_free(name); -} - -static void udp_remove_client(struct sockaddr_storage *addr, socklen_t addrlen, struct udp_sink* sink) -{ - // TODO: implement this! - gchar* name = gst_element_get_name(sink->udp_); - log_printf(ERROR, "removing client from %s failed! - removing not yet implemented!", name); - g_free(name); -} - -#define UDP_PROTO_MAGIC "SYDRA:" -#define UDP_PROTO_MAGIC_LEN (sizeof(UDP_PROTO_MAGIC)-1) -#define UDP_PROTO_CMD_ADD_CLIENT "add_client\n" -#define UDP_PROTO_CMD_ADD_CLIENT_LEN (sizeof(UDP_PROTO_CMD_ADD_CLIENT)-1) -#define UDP_PROTO_CMD_RM_CLIENT "remove_client\n" -#define UDP_PROTO_CMD_RM_CLIENT_LEN (sizeof(UDP_PROTO_CMD_RM_CLIENT)-1) - -static gboolean on_udp_desc_ready(gint fd, GIOCondition cond, gpointer user_data) -{ - struct udp_sink* sink = (struct udp_sink*)user_data; - if(!sink || !(sink->udp_)) { - log_printf(WARNING, "File descriptor %d is ready for reading - but supplied element is NULL removing callback", fd); - return FALSE; - } - - struct sockaddr_storage addr; - socklen_t addrlen = sizeof(addr); - u_int8_t buf[2048]; - ssize_t bytes = recvfrom(fd, buf, sizeof(buf), 0, (struct sockaddr *)&addr, &addrlen); - if(bytes < 1) { - if(errno == EINTR) - return TRUE; - - log_printf(WARNING, "Error while receiving UDP data on fd %d, will remove callback", fd); - return FALSE; - } - - if(bytes < UDP_PROTO_MAGIC_LEN || memcmp(UDP_PROTO_MAGIC, buf, UDP_PROTO_MAGIC_LEN)) { - log_printf(DEBUG, "client discovery: ignoring invalid incoming packet"); - return TRUE; - } - - u_int8_t* cmd = &(buf[UDP_PROTO_MAGIC_LEN]); - if(bytes == (UDP_PROTO_MAGIC_LEN+UDP_PROTO_CMD_ADD_CLIENT_LEN) && - !memcmp(UDP_PROTO_CMD_ADD_CLIENT, cmd, UDP_PROTO_CMD_ADD_CLIENT_LEN)) { - udp_add_client(&addr, addrlen, sink); - } else if(bytes == (UDP_PROTO_MAGIC_LEN+UDP_PROTO_CMD_RM_CLIENT_LEN) && - !memcmp(UDP_PROTO_CMD_RM_CLIENT, cmd, UDP_PROTO_CMD_RM_CLIENT_LEN)) { - udp_remove_client(&addr, addrlen, sink); - } else { - log_printf(DEBUG, "client discovery: ignoring invalid command"); - } - return TRUE; -} - -static gboolean attach_udp_descriptor(struct udp_sink* sink, const char* name, const char* prop) -{ - GSocket *sock; - g_object_get(G_OBJECT(sink->udp_), prop, &sock, NULL); - int fd = g_socket_get_fd(sock); - - guint id = g_unix_fd_add(fd, G_IO_IN, on_udp_desc_ready, sink); - if(id <= 0) { - log_printf(ERROR, "Error while adding %s file descriptor (%d) to main loop", name, fd); - return FALSE; - } - log_printf(DEBUG, "successfully installed callback for %s (fd: %d) for reading (id: %d)", name, fd, id); - return TRUE; -} - - -static gboolean attach_udp_descriptors(struct udp_sinks *sinks) -{ - if(!attach_udp_descriptor(&(sinks->rtp_video_), "RTP(video) IPv4", "used-socket") || - !attach_udp_descriptor(&(sinks->rtp_video_), "RTP(video) IPv6", "used-socket-v6") || - !attach_udp_descriptor(&(sinks->rtcp_video_), "RTCP(video) IPv4", "used-socket") || - !attach_udp_descriptor(&(sinks->rtcp_video_), "RTCP(video) IPv6", "used-socket-v6") || - !attach_udp_descriptor(&(sinks->rtp_audio_), "RTP(audio) IPv4", "used-socket") || - !attach_udp_descriptor(&(sinks->rtp_audio_), "RTP(audio) IPv6", "used-socket-v6") || - !attach_udp_descriptor(&(sinks->rtcp_audio_), "RTCP(audio) IPv4", "used-socket") || - !attach_udp_descriptor(&(sinks->rtcp_audio_), "RTCP(audio) IPv6", "used-socket-v6")) { - return FALSE; - } - return TRUE; -} - - - int main_loop(options_t* opt) { log_printf(INFO, "entering main loop"); -- cgit v1.2.3