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