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