main/netsock.c

   1 /*
   2  * Asterisk -- An open source telephony toolkit.
   3  *
   4  * Copyright (C) 1999 - 2005, Digium, Inc.
   5  *
   6  * Kevin P. Fleming <kpfleming@digium.com>
   7  * Mark Spencer <markster@digium.com>
   8  *
   9  * See http://www.asterisk.org for more information about
  10  * the Asterisk project. Please do not directly contact
  11  * any of the maintainers of this project for assistance;
  12  * the project provides a web site, mailing lists and IRC
  13  * channels for your use.
  14  *
  15  * This program is free software, distributed under the terms of
  16  * the GNU General Public License Version 2. See the LICENSE file
  17  * at the top of the source tree.
  18  */
  19
  20 /*! \file
  21  *
  22  * \brief Network socket handling
  23  *
  24  * \author Kevin P. Fleming <kpfleming@digium.com>
  25  * \author Mark Spencer <markster@digium.com>
  26  */
  27
  28 #include "asterisk.h"
  29
  30 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
  31
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <string.h>
  35 #include <sys/time.h>
  36 #include <signal.h>
  37 #include <errno.h>
  38 #include <unistd.h>
  39 #include <netinet/in.h>
  40 #include <arpa/inet.h>
  41 #include <sys/socket.h>
  42 #include <netdb.h>
  43 #include <net/if.h>
  44 #include <netinet/in_systm.h>
  45 #include <netinet/ip.h>
  46 #include <sys/ioctl.h>
  47
  48 #if defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__)
  49 #include <fcntl.h>
  50 #include <net/route.h>
  51 #endif
  52
  53 #if defined (SOLARIS)
  54 #include <sys/sockio.h>
  55 #endif
  56
  57 #include "asterisk/netsock.h"
  58 #include "asterisk/logger.h"
  59 #include "asterisk/channel.h"
  60 #include "asterisk/options.h"
  61 #include "asterisk/utils.h"
  62 #include "asterisk/lock.h"
  63 #include "asterisk/srv.h"
  64
  65 struct ast_netsock {
  66         ASTOBJ_COMPONENTS(struct ast_netsock);
  67         struct sockaddr_in bindaddr;
  68         int sockfd;
  69         int *ioref;
  70         struct io_context *ioc;
  71         void *data;
  72 };
  73
  74 struct ast_netsock_list {
  75         ASTOBJ_CONTAINER_COMPONENTS(struct ast_netsock);
  76         struct io_context *ioc;
  77 };
  78
  79 static void ast_netsock_destroy(struct ast_netsock *netsock)
  80 {
  81         ast_io_remove(netsock->ioc, netsock->ioref);
  82         close(netsock->sockfd);
  83         free(netsock);
  84 }
  85
  86 struct ast_netsock_list *ast_netsock_list_alloc(void)
  87 {
  88         return ast_calloc(1, sizeof(struct ast_netsock_list));
  89 }
  90
  91 int ast_netsock_init(struct ast_netsock_list *list)
  92 {
  93         memset(list, 0, sizeof(*list));
  94         ASTOBJ_CONTAINER_INIT(list);
  95
  96         return 0;
  97 }
  98
  99 int ast_netsock_release(struct ast_netsock_list *list)
 100 {
 101         ASTOBJ_CONTAINER_DESTROYALL(list, ast_netsock_destroy);
 102         ASTOBJ_CONTAINER_DESTROY(list);
 103
 104         return 0;
 105 }
 106
 107 struct ast_netsock *ast_netsock_find(struct ast_netsock_list *list,
 108                                      struct sockaddr_in *sa)
 109 {
 110         struct ast_netsock *sock = NULL;
 111
 112         ASTOBJ_CONTAINER_TRAVERSE(list, !sock, {
 113                 ASTOBJ_RDLOCK(iterator);
 114                 if (!inaddrcmp(&iterator->bindaddr, sa))
 115                         sock = iterator;
 116                 ASTOBJ_UNLOCK(iterator);
 117         });
 118
 119         return sock;
 120 }
 121
 122 struct ast_netsock *ast_netsock_bindaddr(struct ast_netsock_list *list, struct io_context *ioc, struct sockaddr_in *bindaddr, int tos, ast_io_cb callback, void *data)
 123 {
 124         int netsocket = -1;
 125         int *ioref;
 126
 127         struct ast_netsock *ns;
 128         const int reuseFlag = 1;
 129
 130         /* Make a UDP socket */
 131         netsocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
 132
 133         if (netsocket < 0) {
 134                 ast_log(LOG_ERROR, "Unable to create network socket: %s\n", strerror(errno));
 135                 return NULL;
 136         }
 137         if (setsockopt(netsocket, SOL_SOCKET, SO_REUSEADDR, (char *)&reuseFlag, sizeof reuseFlag) < 0) {
 138                         ast_log(LOG_WARNING, "Error setting SO_REUSEADDR on sockfd '%d'\n", netsocket);
 139         }
 140         if (bind(netsocket,(struct sockaddr *)bindaddr, sizeof(struct sockaddr_in))) {
 141                 ast_log(LOG_ERROR, "Unable to bind to %s port %d: %s\n", ast_inet_ntoa(bindaddr->sin_addr), ntohs(bindaddr->sin_port), strerror(errno));
 142                 close(netsocket);
 143                 return NULL;
 144         }
 145         if (option_verbose > 1)
 146                 ast_verbose(VERBOSE_PREFIX_2 "Using TOS bits %d\n", tos);
 147
 148         if (setsockopt(netsocket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)))
 149                 ast_log(LOG_WARNING, "Unable to set TOS to %d\n", tos);
 150
 151         ast_enable_packet_fragmentation(netsocket);
 152
 153         if (!(ns = ast_calloc(1, sizeof(struct ast_netsock)))) {
 154                 close(netsocket);
 155                 return NULL;
 156         }
 157
 158         /* Establish I/O callback for socket read */
 159         if (!(ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns))) {
 160                 close(netsocket);
 161                 free(ns);
 162                 return NULL;
 163         }
 164         ASTOBJ_INIT(ns);
 165         ns->ioref = ioref;
 166         ns->ioc = ioc;
 167         ns->sockfd = netsocket;
 168         ns->data = data;
 169         memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
 170         ASTOBJ_CONTAINER_LINK(list, ns);
 171
 172         return ns;
 173 }
 174
 175 struct ast_netsock *ast_netsock_bind(struct ast_netsock_list *list, struct io_context *ioc, const char *bindinfo, int defaultport, int tos, ast_io_cb callback, void *data)
 176 {
 177         struct sockaddr_in sin;
 178         char *tmp;
 179         char *host;
 180         char *port;
 181         int portno;
 182
 183         memset(&sin, 0, sizeof(sin));
 184         sin.sin_family = AF_INET;
 185         sin.sin_port = htons(defaultport);
 186         tmp = ast_strdupa(bindinfo);
 187
 188         host = strsep(&tmp, ":");
 189         port = tmp;
 190
 191         if (port && ((portno = atoi(port)) > 0))
 192                 sin.sin_port = htons(portno);
 193
 194         inet_aton(host, &sin.sin_addr);
 195
 196         return ast_netsock_bindaddr(list, ioc, &sin, tos, callback, data);
 197 }
 198
 199 int ast_netsock_sockfd(const struct ast_netsock *ns)
 200 {
 201         return ns ? ns-> sockfd : -1;
 202 }
 203
 204 const struct sockaddr_in *ast_netsock_boundaddr(const struct ast_netsock *ns)
 205 {
 206         return &(ns->bindaddr);
 207 }
 208
 209 void *ast_netsock_data(const struct ast_netsock *ns)
 210 {
 211         return ns->data;
 212 }
 213
 214 void ast_netsock_unref(struct ast_netsock *ns)
 215 {
 216         ASTOBJ_UNREF(ns, ast_netsock_destroy);
 217 }