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