Fix a couple of places where "tos" was used instead of "cos".
[asterisk/asterisk.git] / 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, int cos, 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
146         ast_netsock_set_qos(netsocket, tos, cos);
147                 
148         ast_enable_packet_fragmentation(netsocket);
149
150         if (!(ns = ast_calloc(1, sizeof(struct ast_netsock)))) {
151                 close(netsocket);
152                 return NULL;
153         }
154         
155         /* Establish I/O callback for socket read */
156         if (!(ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns))) {
157                 close(netsocket);
158                 free(ns);
159                 return NULL;
160         }       
161         ASTOBJ_INIT(ns);
162         ns->ioref = ioref;
163         ns->ioc = ioc;
164         ns->sockfd = netsocket;
165         ns->data = data;
166         memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
167         ASTOBJ_CONTAINER_LINK(list, ns);
168
169         return ns;
170 }
171
172 int ast_netsock_set_qos(int netsocket, int tos, int cos)
173 {
174         int res;
175         
176         if ((res = setsockopt(netsocket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos))))
177                 ast_log(LOG_WARNING, "Unable to set TOS to %d\n", tos);
178         else {
179             if (option_verbose > 1)
180                 ast_verbose(VERBOSE_PREFIX_2 "Using TOS bits %d\n", tos);
181         }
182
183 #if defined(linux)                                                              
184         if (setsockopt(netsocket, SOL_SOCKET, SO_PRIORITY, &cos, sizeof(cos)))
185             ast_log(LOG_WARNING, "Unable to set CoS to %d\n", cos);
186         else {
187             if (option_verbose > 1)
188                 ast_verbose(VERBOSE_PREFIX_2 "Using CoS mark %d\n", cos);
189         }
190 #endif
191                                                         
192         return res;
193 }
194                                                                                                         
195
196 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)
197 {
198         struct sockaddr_in sin;
199         char *tmp;
200         char *host;
201         char *port;
202         int portno;
203
204         memset(&sin, 0, sizeof(sin));
205         sin.sin_family = AF_INET;
206         sin.sin_port = htons(defaultport);
207         tmp = ast_strdupa(bindinfo);
208
209         host = strsep(&tmp, ":");
210         port = tmp;
211
212         if (port && ((portno = atoi(port)) > 0))
213                 sin.sin_port = htons(portno);
214
215         inet_aton(host, &sin.sin_addr);
216
217         return ast_netsock_bindaddr(list, ioc, &sin, tos, cos, callback, data);
218 }
219
220 int ast_netsock_sockfd(const struct ast_netsock *ns)
221 {
222         return ns ? ns-> sockfd : -1;
223 }
224
225 const struct sockaddr_in *ast_netsock_boundaddr(const struct ast_netsock *ns)
226 {
227         return &(ns->bindaddr);
228 }
229
230 void *ast_netsock_data(const struct ast_netsock *ns)
231 {
232         return ns->data;
233 }
234
235 void ast_netsock_unref(struct ast_netsock *ns)
236 {
237         ASTOBJ_UNREF(ns, ast_netsock_destroy);
238 }