Let ExtensionState resolve dynamic hints.
[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 #if defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__) || defined(__Darwin__)
33 #include <net/if_dl.h>
34 #endif
35
36 #if defined (SOLARIS)
37 #include <sys/sockio.h>
38 #elif defined(HAVE_GETIFADDRS)
39 #include <ifaddrs.h>
40 #endif
41
42 #include "asterisk/netsock.h"
43 #include "asterisk/utils.h"
44 #include "asterisk/astobj.h"
45
46 struct ast_netsock {
47         ASTOBJ_COMPONENTS(struct ast_netsock);
48         struct sockaddr_in bindaddr;
49         int sockfd;
50         int *ioref;
51         struct io_context *ioc;
52         void *data;
53 };
54
55 struct ast_netsock_list {
56         ASTOBJ_CONTAINER_COMPONENTS(struct ast_netsock);
57         struct io_context *ioc;
58 };
59
60 static void ast_netsock_destroy(struct ast_netsock *netsock)
61 {
62         ast_io_remove(netsock->ioc, netsock->ioref);
63         close(netsock->sockfd);
64         ast_free(netsock);
65 }
66
67 struct ast_netsock_list *ast_netsock_list_alloc(void)
68 {
69         return ast_calloc(1, sizeof(struct ast_netsock_list));
70 }
71
72 int ast_netsock_init(struct ast_netsock_list *list)
73 {
74         memset(list, 0, sizeof(*list));
75         ASTOBJ_CONTAINER_INIT(list);
76
77         return 0;
78 }
79
80 int ast_netsock_release(struct ast_netsock_list *list)
81 {
82         ASTOBJ_CONTAINER_DESTROYALL(list, ast_netsock_destroy);
83         ASTOBJ_CONTAINER_DESTROY(list);
84         ast_free(list);
85
86         return 0;
87 }
88
89 struct ast_netsock *ast_netsock_find(struct ast_netsock_list *list,
90                                      struct sockaddr_in *sa)
91 {
92         struct ast_netsock *sock = NULL;
93
94         ASTOBJ_CONTAINER_TRAVERSE(list, !sock, {
95                 ASTOBJ_RDLOCK(iterator);
96                 if (!inaddrcmp(&iterator->bindaddr, sa))
97                         sock = iterator;
98                 ASTOBJ_UNLOCK(iterator);
99         });
100
101         return sock;
102 }
103
104 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)
105 {
106         int netsocket = -1;
107         int *ioref;
108         
109         struct ast_netsock *ns;
110         const int reuseFlag = 1;
111         
112         /* Make a UDP socket */
113         netsocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
114         
115         if (netsocket < 0) {
116                 ast_log(LOG_ERROR, "Unable to create network socket: %s\n", strerror(errno));
117                 return NULL;
118         }
119         if (setsockopt(netsocket, SOL_SOCKET, SO_REUSEADDR, (char *)&reuseFlag, sizeof reuseFlag) < 0) {
120                         ast_log(LOG_WARNING, "Error setting SO_REUSEADDR on sockfd '%d'\n", netsocket);
121         }
122         if (bind(netsocket,(struct sockaddr *)bindaddr, sizeof(struct sockaddr_in))) {
123                 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));
124                 close(netsocket);
125                 return NULL;
126         }
127
128         ast_netsock_set_qos(netsocket, tos, cos, "IAX2");
129                 
130         ast_enable_packet_fragmentation(netsocket);
131
132         if (!(ns = ast_calloc(1, sizeof(*ns)))) {
133                 close(netsocket);
134                 return NULL;
135         }
136         
137         /* Establish I/O callback for socket read */
138         if (!(ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns))) {
139                 close(netsocket);
140                 ast_free(ns);
141                 return NULL;
142         }       
143         ASTOBJ_INIT(ns);
144         ns->ioref = ioref;
145         ns->ioc = ioc;
146         ns->sockfd = netsocket;
147         ns->data = data;
148         memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
149         ASTOBJ_CONTAINER_LINK(list, ns);
150
151         return ns;
152 }
153
154 int ast_netsock_set_qos(int netsocket, int tos, int cos, const char *desc)
155 {
156         int res;
157         
158         if ((res = setsockopt(netsocket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos))))
159                 ast_log(LOG_WARNING, "Unable to set %s TOS to %d, may be you have no root privileges\n", desc, tos);
160         else if (tos)
161                 ast_verb(2, "Using %s TOS bits %d\n", desc, tos);
162
163 #if defined(linux)                                                              
164         if (setsockopt(netsocket, SOL_SOCKET, SO_PRIORITY, &cos, sizeof(cos)))
165                 ast_log(LOG_WARNING, "Unable to set %s CoS to %d\n", desc, cos);
166         else if (cos)
167                 ast_verb(2, "Using %s CoS mark %d\n", desc, cos);
168 #endif
169                                                         
170         return res;
171 }
172                                                                                                         
173
174 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)
175 {
176         struct sockaddr_in sin;
177         char *tmp;
178         char *host;
179         char *port;
180         int portno;
181
182         memset(&sin, 0, sizeof(sin));
183         sin.sin_family = AF_INET;
184         sin.sin_port = htons(defaultport);
185         tmp = ast_strdupa(bindinfo);
186
187         host = strsep(&tmp, ":");
188         port = tmp;
189
190         if (port && ((portno = atoi(port)) > 0))
191                 sin.sin_port = htons(portno);
192
193         inet_aton(host, &sin.sin_addr);
194
195         return ast_netsock_bindaddr(list, ioc, &sin, tos, cos, callback, data);
196 }
197
198 int ast_netsock_sockfd(const struct ast_netsock *ns)
199 {
200         return ns ? ns-> sockfd : -1;
201 }
202
203 const struct sockaddr_in *ast_netsock_boundaddr(const struct ast_netsock *ns)
204 {
205         return &(ns->bindaddr);
206 }
207
208 void *ast_netsock_data(const struct ast_netsock *ns)
209 {
210         return ns->data;
211 }
212
213 void ast_netsock_unref(struct ast_netsock *ns)
214 {
215         ASTOBJ_UNREF(ns, ast_netsock_destroy);
216 }
217
218 char *ast_eid_to_str(char *s, int maxlen, struct ast_eid *eid)
219 {
220         int x;
221         char *os = s;
222         if (maxlen < 18) {
223                 if (s && (maxlen > 0))
224                         *s = '\0';
225         } else {
226                 for (x = 0; x < 5; x++) {
227                         sprintf(s, "%02x:", eid->eid[x]);
228                         s += 3;
229                 }
230                 sprintf(s, "%02x", eid->eid[5]);
231         }
232         return os;
233 }
234
235 void ast_set_default_eid(struct ast_eid *eid)
236 {
237 #if defined(SIOCGIFHWADDR)
238         int s, x = 0;
239         char eid_str[20];
240         struct ifreq ifr;
241
242         s = socket(AF_INET, SOCK_STREAM, 0);
243         if (s < 0)
244                 return;
245         for (x = 0; x < 10; x++) {
246                 memset(&ifr, 0, sizeof(ifr));
247                 snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "eth%d", x);
248                 if (ioctl(s, SIOCGIFHWADDR, &ifr))
249                         continue;
250                 memcpy(eid, ((unsigned char *)&ifr.ifr_hwaddr) + 2, sizeof(*eid));
251                 ast_debug(1, "Seeding global EID '%s' from '%s' using 'siocgifhwaddr'\n", ast_eid_to_str(eid_str, sizeof(eid_str), eid), ifr.ifr_name);
252                 close(s);
253                 return;
254         }
255         close(s);
256 #else
257 #if defined(ifa_broadaddr) && !defined(SOLARIS)
258         char eid_str[20];
259         struct ifaddrs *ifap;
260         
261         if (getifaddrs(&ifap) == 0) {
262                 struct ifaddrs *p;
263                 for (p = ifap; p; p = p->ifa_next) {
264                         if ((p->ifa_addr->sa_family == AF_LINK) && !(p->ifa_flags & IFF_LOOPBACK) && (p->ifa_flags & IFF_RUNNING)) {
265                                 struct sockaddr_dl* sdp = (struct sockaddr_dl*) p->ifa_addr;
266                                 memcpy(&(eid->eid), sdp->sdl_data + sdp->sdl_nlen, 6);
267                                 ast_debug(1, "Seeding global EID '%s' from '%s' using 'getifaddrs'\n", ast_eid_to_str(eid_str, sizeof(eid_str), eid), p->ifa_name);
268                                 freeifaddrs(ifap);
269                                 return;
270                         }
271                 }
272                 freeifaddrs(ifap);
273         }
274 #endif
275 #endif
276         ast_debug(1, "No ethernet interface found for seeding global EID. You will have to set it manually.\n");
277 }
278
279 int ast_str_to_eid(struct ast_eid *eid, const char *s)
280 {
281         unsigned int eid_int[6];
282         int x;
283
284         if (sscanf(s, "%2x:%2x:%2x:%2x:%2x:%2x", &eid_int[0], &eid_int[1], &eid_int[2],
285                  &eid_int[3], &eid_int[4], &eid_int[5]) != 6)
286                         return -1;
287         
288         for (x = 0; x < 6; x++)
289                 eid->eid[x] = eid_int[x];
290
291         return 0;
292 }
293
294 int ast_eid_cmp(const struct ast_eid *eid1, const struct ast_eid *eid2)
295 {
296         return memcmp(eid1, eid2, sizeof(*eid1));
297 }