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