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 <sys/time.h>
  33 #include <signal.h>
  34 #include <sys/ioctl.h>
  35
  36 #if defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__)
  37 #include <fcntl.h>
  38 #endif
  39
  40 #if defined (SOLARIS)
  41 #include <sys/sockio.h>
  42 #endif
  43
  44 #include "asterisk/netsock.h"
  45 #include "asterisk/channel.h"
  46 #include "asterisk/utils.h"
  47 #include "asterisk/lock.h"
  48 #include "asterisk/srv.h"
  49
  50 struct ast_netsock {
  51         ASTOBJ_COMPONENTS(struct ast_netsock);
  52         struct sockaddr_in bindaddr;
  53         int sockfd;
  54         int *ioref;
  55         struct io_context *ioc;
  56         void *data;
  57 };
  58
  59 struct ast_netsock_list {
  60         ASTOBJ_CONTAINER_COMPONENTS(struct ast_netsock);
  61         struct io_context *ioc;
  62 };
  63
  64 static void ast_netsock_destroy(struct ast_netsock *netsock)
  65 {
  66         ast_io_remove(netsock->ioc, netsock->ioref);
  67         close(netsock->sockfd);
  68         ast_free(netsock);
  69 }
  70
  71 struct ast_netsock_list *ast_netsock_list_alloc(void)
  72 {
  73         return ast_calloc(1, sizeof(struct ast_netsock_list));
  74 }
  75
  76 int ast_netsock_init(struct ast_netsock_list *list)
  77 {
  78         memset(list, 0, sizeof(*list));
  79         ASTOBJ_CONTAINER_INIT(list);
  80
  81         return 0;
  82 }
  83
  84 int ast_netsock_release(struct ast_netsock_list *list)
  85 {
  86         ASTOBJ_CONTAINER_DESTROYALL(list, ast_netsock_destroy);
  87         ASTOBJ_CONTAINER_DESTROY(list);
  88
  89         return 0;
  90 }
  91
  92 struct ast_netsock *ast_netsock_find(struct ast_netsock_list *list,
  93                                      struct sockaddr_in *sa)
  94 {
  95         struct ast_netsock *sock = NULL;
  96
  97         ASTOBJ_CONTAINER_TRAVERSE(list, !sock, {
  98                 ASTOBJ_RDLOCK(iterator);
  99                 if (!inaddrcmp(&iterator->bindaddr, sa))
 100                         sock = iterator;
 101                 ASTOBJ_UNLOCK(iterator);
 102         });
 103
 104         return sock;
 105 }
 106
 107 struct ast_netsock *ast_netsock_bindaddr(struct ast_netsock_list *list, struct io_context *ioc, struct sockaddr_in *bindaddr, int tos, int cos, ast_io_cb callback, void *data)
 108 {
 109         int netsocket = -1;
 110         int *ioref;
 111
 112         struct ast_netsock *ns;
 113         const int reuseFlag = 1;
 114
 115         /* Make a UDP socket */
 116         netsocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
 117
 118         if (netsocket < 0) {
 119                 ast_log(LOG_ERROR, "Unable to create network socket: %s\n", strerror(errno));
 120                 return NULL;
 121         }
 122         if (setsockopt(netsocket, SOL_SOCKET, SO_REUSEADDR, (char *)&reuseFlag, sizeof reuseFlag) < 0) {
 123                         ast_log(LOG_WARNING, "Error setting SO_REUSEADDR on sockfd '%d'\n", netsocket);
 124         }
 125         if (bind(netsocket,(struct sockaddr *)bindaddr, sizeof(struct sockaddr_in))) {
 126                 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));
 127                 close(netsocket);
 128                 return NULL;
 129         }
 130
 131         ast_netsock_set_qos(netsocket, tos, cos);
 132
 133         ast_enable_packet_fragmentation(netsocket);
 134
 135         if (!(ns = ast_calloc(1, sizeof(*ns)))) {
 136                 close(netsocket);
 137                 return NULL;
 138         }
 139
 140         /* Establish I/O callback for socket read */
 141         if (!(ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns))) {
 142                 close(netsocket);
 143                 ast_free(ns);
 144                 return NULL;
 145         }
 146         ASTOBJ_INIT(ns);
 147         ns->ioref = ioref;
 148         ns->ioc = ioc;
 149         ns->sockfd = netsocket;
 150         ns->data = data;
 151         memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
 152         ASTOBJ_CONTAINER_LINK(list, ns);
 153
 154         return ns;
 155 }
 156
 157 int ast_netsock_set_qos(int netsocket, int tos, int cos)
 158 {
 159         int res;
 160
 161         if ((res = setsockopt(netsocket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos))))
 162                 ast_log(LOG_WARNING, "Unable to set TOS to %d\n", tos);
 163         else {
 164             if (option_verbose > 1)
 165                 ast_verbose(VERBOSE_PREFIX_2 "Using TOS bits %d\n", tos);
 166         }
 167
 168 #if defined(linux)
 169         if (setsockopt(netsocket, SOL_SOCKET, SO_PRIORITY, &cos, sizeof(cos)))
 170             ast_log(LOG_WARNING, "Unable to set CoS to %d\n", cos);
 171         else {
 172             if (option_verbose > 1)
 173                 ast_verbose(VERBOSE_PREFIX_2 "Using CoS mark %d\n", cos);
 174         }
 175 #endif
 176
 177         return res;
 178 }
 179
 180
 181 struct ast_netsock *ast_netsock_bind(struct ast_netsock_list *list, struct io_context *ioc, const char *bindinfo, int defaultport, int tos, int cos, ast_io_cb callback, void *data)
 182 {
 183         struct sockaddr_in sin;
 184         char *tmp;
 185         char *host;
 186         char *port;
 187         int portno;
 188
 189         memset(&sin, 0, sizeof(sin));
 190         sin.sin_family = AF_INET;
 191         sin.sin_port = htons(defaultport);
 192         tmp = ast_strdupa(bindinfo);
 193
 194         host = strsep(&tmp, ":");
 195         port = tmp;
 196
 197         if (port && ((portno = atoi(port)) > 0))
 198                 sin.sin_port = htons(portno);
 199
 200         inet_aton(host, &sin.sin_addr);
 201
 202         return ast_netsock_bindaddr(list, ioc, &sin, tos, cos, callback, data);
 203 }
 204
 205 int ast_netsock_sockfd(const struct ast_netsock *ns)
 206 {
 207         return ns ? ns-> sockfd : -1;
 208 }
 209
 210 const struct sockaddr_in *ast_netsock_boundaddr(const struct ast_netsock *ns)
 211 {
 212         return &(ns->bindaddr);
 213 }
 214
 215 void *ast_netsock_data(const struct ast_netsock *ns)
 216 {
 217         return ns->data;
 218 }
 219
 220 void ast_netsock_unref(struct ast_netsock *ns)
 221 {
 222         ASTOBJ_UNREF(ns, ast_netsock_destroy);
 223 }