Fix DEBUG_THREADS when lock is acquired in __constructor__
[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 /*** MODULEINFO
29         <support_level>core</support_level>
30  ***/
31
32 #include "asterisk.h"
33
34 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
35
36 #ifndef __linux__
37 #if defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__) || defined(__Darwin__) || defined(__GLIBC__)
38 #include <net/if_dl.h>
39 #endif
40 #endif
41
42 #if defined (SOLARIS)
43 #include <sys/sockio.h>
44 #elif defined(HAVE_GETIFADDRS)
45 #include <ifaddrs.h>
46 #endif
47
48 #include "asterisk/netsock.h"
49 #include "asterisk/netsock2.h"
50 #include "asterisk/utils.h"
51 #include "asterisk/astobj.h"
52
53 struct ast_netsock {
54         ASTOBJ_COMPONENTS(struct ast_netsock);
55         struct ast_sockaddr bindaddr;
56         int sockfd;
57         int *ioref;
58         struct io_context *ioc;
59         void *data;
60 };
61
62 struct ast_netsock_list {
63         ASTOBJ_CONTAINER_COMPONENTS(struct ast_netsock);
64         struct io_context *ioc;
65 };
66
67 static void ast_netsock_destroy(struct ast_netsock *netsock)
68 {
69         ast_io_remove(netsock->ioc, netsock->ioref);
70         close(netsock->sockfd);
71         ast_free(netsock);
72 }
73
74 struct ast_netsock_list *ast_netsock_list_alloc(void)
75 {
76         return ast_calloc(1, sizeof(struct ast_netsock_list));
77 }
78
79 int ast_netsock_init(struct ast_netsock_list *list)
80 {
81         memset(list, 0, sizeof(*list));
82         ASTOBJ_CONTAINER_INIT(list);
83
84         return 0;
85 }
86
87 int ast_netsock_release(struct ast_netsock_list *list)
88 {
89         ASTOBJ_CONTAINER_DESTROYALL(list, ast_netsock_destroy);
90         ASTOBJ_CONTAINER_DESTROY(list);
91         ast_free(list);
92
93         return 0;
94 }
95
96 struct ast_netsock *ast_netsock_find(struct ast_netsock_list *list, struct ast_sockaddr *addr)
97 {
98         struct ast_netsock *sock = NULL;
99
100         ASTOBJ_CONTAINER_TRAVERSE(list, !sock, {
101                 ASTOBJ_RDLOCK(iterator);
102                 if (!ast_sockaddr_cmp(&iterator->bindaddr, addr)) {
103                         sock = iterator;
104                 }
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 ast_sockaddr *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 (ast_bind(netsocket, bindaddr)) {
130                 ast_log(LOG_ERROR,
131                         "Unable to bind to %s: %s\n",
132                         ast_sockaddr_stringify(bindaddr),
133                         strerror(errno));
134                 close(netsocket);
135                 return NULL;
136         }
137
138         ast_set_qos(netsocket, tos, cos, "IAX2");
139
140         ast_enable_packet_fragmentation(netsocket);
141
142         if (!(ns = ast_calloc(1, sizeof(*ns)))) {
143                 close(netsocket);
144                 return NULL;
145         }
146
147         /* Establish I/O callback for socket read */
148         if (!(ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns))) {
149                 close(netsocket);
150                 ast_free(ns);
151                 return NULL;
152         }
153         ASTOBJ_INIT(ns);
154         ns->ioref = ioref;
155         ns->ioc = ioc;
156         ns->sockfd = netsocket;
157         ns->data = data;
158         memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
159         ASTOBJ_CONTAINER_LINK(list, ns);
160
161         return ns;
162 }
163
164 int ast_netsock_set_qos(int sockfd, int tos, int cos, const char *desc)
165 {
166         return ast_set_qos(sockfd, tos, cos, desc);
167 }
168
169 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)
170 {
171         struct ast_sockaddr addr;
172
173         if (ast_sockaddr_parse(&addr, bindinfo, 0)) {
174                 if (!ast_sockaddr_is_ipv4(&addr)) {
175                         ast_log(LOG_WARNING, "Only IPv4 addresses are supported at this time.\n");
176                         return NULL;
177                 }
178
179                 if (!ast_sockaddr_port(&addr)) {
180                         ast_sockaddr_set_port(&addr, defaultport);
181                 }
182
183                 return ast_netsock_bindaddr(list, ioc, &addr, tos, cos, callback, data);
184         }
185
186         return NULL;
187 }
188
189 int ast_netsock_sockfd(const struct ast_netsock *ns)
190 {
191         return ns ? ns-> sockfd : -1;
192 }
193
194 const struct ast_sockaddr *ast_netsock_boundaddr(const struct ast_netsock *ns)
195 {
196         return &ns->bindaddr;
197 }
198
199 void *ast_netsock_data(const struct ast_netsock *ns)
200 {
201         return ns->data;
202 }
203
204 void ast_netsock_unref(struct ast_netsock *ns)
205 {
206         ASTOBJ_UNREF(ns, ast_netsock_destroy);
207 }
208
209 char *ast_eid_to_str(char *s, int maxlen, struct ast_eid *eid)
210 {
211         int x;
212         char *os = s;
213         if (maxlen < 18) {
214                 if (s && (maxlen > 0))
215                         *s = '\0';
216         } else {
217                 for (x = 0; x < 5; x++) {
218                         sprintf(s, "%02x:", eid->eid[x]);
219                         s += 3;
220                 }
221                 sprintf(s, "%02x", eid->eid[5]);
222         }
223         return os;
224 }
225
226 void ast_set_default_eid(struct ast_eid *eid)
227 {
228 #if defined(SIOCGIFHWADDR) && defined(HAVE_STRUCT_IFREQ_IFR_IFRU_IFRU_HWADDR)
229         int s, x = 0;
230         char eid_str[20];
231         struct ifreq ifr;
232         static const unsigned int MAXIF = 10;
233
234         s = socket(AF_INET, SOCK_STREAM, 0);
235         if (s < 0)
236                 return;
237         for (x = 0; x < MAXIF; x++) {
238                 static const char *prefixes[] = { "eth", "em" };
239                 unsigned int i;
240
241                 for (i = 0; i < ARRAY_LEN(prefixes); i++) {
242                         memset(&ifr, 0, sizeof(ifr));
243                         snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "%s%d", prefixes[i], x);
244                         if (!ioctl(s, SIOCGIFHWADDR, &ifr)) {
245                                 break;
246                         }
247                 }
248
249                 if (i == ARRAY_LEN(prefixes)) {
250                         /* Try pciX#[1..N] */
251                         for (i = 0; i < MAXIF; i++) {
252                                 memset(&ifr, 0, sizeof(ifr));
253                                 snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "pci%u#%u", x, i);
254                                 if (!ioctl(s, SIOCGIFHWADDR, &ifr)) {
255                                         break;
256                                 }
257                         }
258                         if (i == MAXIF) {
259                                 continue;
260                         }
261                 }
262
263                 memcpy(eid, ((unsigned char *)&ifr.ifr_hwaddr) + 2, sizeof(*eid));
264                 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);
265                 close(s);
266                 return;
267         }
268         close(s);
269 #else
270 #if defined(ifa_broadaddr) && !defined(SOLARIS)
271         char eid_str[20];
272         struct ifaddrs *ifap;
273
274         if (getifaddrs(&ifap) == 0) {
275                 struct ifaddrs *p;
276                 for (p = ifap; p; p = p->ifa_next) {
277                         if ((p->ifa_addr->sa_family == AF_LINK) && !(p->ifa_flags & IFF_LOOPBACK) && (p->ifa_flags & IFF_RUNNING)) {
278                                 struct sockaddr_dl* sdp = (struct sockaddr_dl*) p->ifa_addr;
279                                 memcpy(&(eid->eid), sdp->sdl_data + sdp->sdl_nlen, 6);
280                                 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);
281                                 freeifaddrs(ifap);
282                                 return;
283                         }
284                 }
285                 freeifaddrs(ifap);
286         }
287 #endif
288 #endif
289         ast_debug(1, "No ethernet interface found for seeding global EID. You will have to set it manually.\n");
290 }
291
292 int ast_str_to_eid(struct ast_eid *eid, const char *s)
293 {
294         unsigned int eid_int[6];
295         int x;
296
297         if (sscanf(s, "%2x:%2x:%2x:%2x:%2x:%2x", &eid_int[0], &eid_int[1], &eid_int[2],
298                  &eid_int[3], &eid_int[4], &eid_int[5]) != 6)
299                         return -1;
300
301         for (x = 0; x < 6; x++)
302                 eid->eid[x] = eid_int[x];
303
304         return 0;
305 }
306
307 int ast_eid_cmp(const struct ast_eid *eid1, const struct ast_eid *eid2)
308 {
309         return memcmp(eid1, eid2, sizeof(*eid1));
310 }