2 * Asterisk -- An open source telephony toolkit.
4 * Copyright (C) 2010, Digium, Inc.
6 * Viagénie <asteriskv6@viagenie.ca>
8 * See http://www.asterisk.org for more information about
9 * the Asterisk project. Please do not directly contact
10 * any of the maintainers of this project for assistance;
11 * the project provides a web site, mailing lists and IRC
12 * channels for your use.
14 * This program is free software, distributed under the terms of
15 * the GNU General Public License Version 2. See the LICENSE file
16 * at the top of the source tree.
21 * \brief Network socket handling
23 * \author Viagénie <asteriskv6@viagenie.ca>
27 <support_level>core</support_level>
32 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
34 #include "asterisk/config.h"
35 #include "asterisk/netsock2.h"
36 #include "asterisk/utils.h"
37 #include "asterisk/threadstorage.h"
39 int ast_sockaddr_ipv4_mapped(const struct ast_sockaddr *addr, struct ast_sockaddr *ast_mapped)
41 const struct sockaddr_in6 *sin6;
42 struct sockaddr_in sin4;
44 if (!ast_sockaddr_is_ipv6(addr)) {
48 if (!ast_sockaddr_is_ipv4_mapped(addr)) {
52 sin6 = (const struct sockaddr_in6*)&addr->ss;
54 memset(&sin4, 0, sizeof(sin4));
55 sin4.sin_family = AF_INET;
56 sin4.sin_port = sin6->sin6_port;
57 sin4.sin_addr.s_addr = ((uint32_t *)&sin6->sin6_addr)[3];
59 ast_sockaddr_from_sin(ast_mapped, &sin4);
65 AST_THREADSTORAGE(ast_sockaddr_stringify_buf);
67 char *ast_sockaddr_stringify_fmt(const struct ast_sockaddr *sa, int format)
69 struct ast_sockaddr sa_ipv4;
70 const struct ast_sockaddr *sa_tmp;
71 char host[NI_MAXHOST];
72 char port[NI_MAXSERV];
75 static const size_t size = sizeof(host) - 1 + sizeof(port) - 1 + 4;
78 if (ast_sockaddr_isnull(sa)) {
82 if (!(str = ast_str_thread_get(&ast_sockaddr_stringify_buf, size))) {
86 if (ast_sockaddr_ipv4_mapped(sa, &sa_ipv4)) {
92 if ((e = getnameinfo((struct sockaddr *)&sa_tmp->ss, sa_tmp->len,
93 format & AST_SOCKADDR_STR_ADDR ? host : NULL,
94 format & AST_SOCKADDR_STR_ADDR ? sizeof(host) : 0,
95 format & AST_SOCKADDR_STR_PORT ? port : 0,
96 format & AST_SOCKADDR_STR_PORT ? sizeof(port): 0,
97 NI_NUMERICHOST | NI_NUMERICSERV))) {
98 ast_log(LOG_ERROR, "getnameinfo(): %s\n", gai_strerror(e));
102 if ((format & AST_SOCKADDR_STR_REMOTE) == AST_SOCKADDR_STR_REMOTE) {
104 if (ast_sockaddr_is_ipv6_link_local(sa) && (p = strchr(host, '%'))) {
109 switch ((format & AST_SOCKADDR_STR_FORMAT_MASK)) {
110 case AST_SOCKADDR_STR_DEFAULT:
111 ast_str_set(&str, 0, sa_tmp->ss.ss_family == AF_INET6 ?
112 "[%s]:%s" : "%s:%s", host, port);
114 case AST_SOCKADDR_STR_ADDR:
115 ast_str_set(&str, 0, "%s", host);
117 case AST_SOCKADDR_STR_HOST:
119 sa_tmp->ss.ss_family == AF_INET6 ? "[%s]" : "%s", host);
121 case AST_SOCKADDR_STR_PORT:
122 ast_str_set(&str, 0, "%s", port);
125 ast_log(LOG_ERROR, "Invalid format\n");
129 return ast_str_buffer(str);
132 int ast_sockaddr_split_hostport(char *str, char **host, char **port, int flags)
135 char *orig_str = str;/* Original string in case the port presence is incorrect. */
136 char *host_end = NULL;/* Delay terminating the host in case the port presence is incorrect. */
138 ast_debug(5, "Splitting '%s' into...\n", str);
143 for (; *s && *s != ']'; ++s) {
170 switch (flags & PARSE_PORT_MASK) {
171 case PARSE_PORT_IGNORE:
174 case PARSE_PORT_REQUIRE:
176 ast_log(LOG_WARNING, "Port missing in %s\n", orig_str);
180 case PARSE_PORT_FORBID:
182 ast_log(LOG_WARNING, "Port disallowed in %s\n", orig_str);
188 /* Can terminate the host string now if needed. */
192 ast_debug(5, "...host '%s' and port '%s'.\n", *host, *port ? *port : "");
198 int ast_sockaddr_parse(struct ast_sockaddr *addr, const char *str, int flags)
200 struct addrinfo hints;
201 struct addrinfo *res;
207 s = ast_strdupa(str);
208 if (!ast_sockaddr_split_hostport(s, &host, &port, flags)) {
212 memset(&hints, 0, sizeof(hints));
213 /* Hint to get only one entry from getaddrinfo */
214 hints.ai_socktype = SOCK_DGRAM;
216 #ifdef AI_NUMERICSERV
217 hints.ai_flags = AI_NUMERICHOST | AI_NUMERICSERV;
219 hints.ai_flags = AI_NUMERICHOST;
221 if ((e = getaddrinfo(host, port, &hints, &res))) {
222 if (e != EAI_NONAME) { /* if this was just a host name rather than a ip address, don't print error */
223 ast_log(LOG_ERROR, "getaddrinfo(\"%s\", \"%s\", ...): %s\n",
224 host, S_OR(port, "(null)"), gai_strerror(e));
230 * I don't see how this could be possible since we're not resolving host
231 * names. But let's be careful...
233 if (res->ai_next != NULL) {
234 ast_log(LOG_WARNING, "getaddrinfo() returned multiple "
235 "addresses. Ignoring all but the first.\n");
239 addr->len = res->ai_addrlen;
240 memcpy(&addr->ss, res->ai_addr, addr->len);
248 int ast_sockaddr_resolve(struct ast_sockaddr **addrs, const char *str,
249 int flags, int family)
251 struct addrinfo hints, *res, *ai;
252 char *s, *host, *port;
259 s = ast_strdupa(str);
260 if (!ast_sockaddr_split_hostport(s, &host, &port, flags)) {
264 memset(&hints, 0, sizeof(hints));
265 hints.ai_family = family;
266 hints.ai_socktype = SOCK_DGRAM;
268 if ((e = getaddrinfo(host, port, &hints, &res))) {
269 ast_log(LOG_ERROR, "getaddrinfo(\"%s\", \"%s\", ...): %s\n",
270 host, S_OR(port, "(null)"), gai_strerror(e));
275 for (ai = res; ai; ai = ai->ai_next) {
283 if ((*addrs = ast_malloc(res_cnt * sizeof(struct ast_sockaddr))) == NULL) {
289 for (ai = res; ai; ai = ai->ai_next) {
290 (*addrs)[i].len = ai->ai_addrlen;
291 memcpy(&(*addrs)[i].ss, ai->ai_addr, ai->ai_addrlen);
300 int ast_sockaddr_apply_netmask(const struct ast_sockaddr *addr, const struct ast_sockaddr *netmask,
301 struct ast_sockaddr *result)
305 if (ast_sockaddr_is_ipv4(addr)) {
306 struct sockaddr_in result4 = { 0, };
307 struct sockaddr_in *addr4 = (struct sockaddr_in *) &addr->ss;
308 struct sockaddr_in *mask4 = (struct sockaddr_in *) &netmask->ss;
309 result4.sin_family = AF_INET;
310 result4.sin_addr.s_addr = addr4->sin_addr.s_addr & mask4->sin_addr.s_addr;
311 ast_sockaddr_from_sin(result, &result4);
312 } else if (ast_sockaddr_is_ipv6(addr)) {
313 struct sockaddr_in6 result6 = { 0, };
314 struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *) &addr->ss;
315 struct sockaddr_in6 *mask6 = (struct sockaddr_in6 *) &netmask->ss;
317 result6.sin6_family = AF_INET6;
318 for (i = 0; i < 4; ++i) {
319 V6_WORD(&result6, i) = V6_WORD(addr6, i) & V6_WORD(mask6, i);
321 memcpy(&result->ss, &result6, sizeof(result6));
322 result->len = sizeof(result6);
324 /* Unsupported address scheme */
331 int ast_sockaddr_cmp(const struct ast_sockaddr *a, const struct ast_sockaddr *b)
333 const struct ast_sockaddr *a_tmp, *b_tmp;
334 struct ast_sockaddr ipv4_mapped;
339 if (a_tmp->len != b_tmp->len) {
340 if (ast_sockaddr_ipv4_mapped(a, &ipv4_mapped)) {
341 a_tmp = &ipv4_mapped;
342 } else if (ast_sockaddr_ipv4_mapped(b, &ipv4_mapped)) {
343 b_tmp = &ipv4_mapped;
347 if (a_tmp->len < b_tmp->len) {
349 } else if (a_tmp->len > b_tmp->len) {
353 return memcmp(&a_tmp->ss, &b_tmp->ss, a_tmp->len);
356 int ast_sockaddr_cmp_addr(const struct ast_sockaddr *a, const struct ast_sockaddr *b)
358 const struct ast_sockaddr *a_tmp, *b_tmp;
359 struct ast_sockaddr ipv4_mapped;
360 const struct in_addr *ip4a, *ip4b;
361 const struct in6_addr *ip6a, *ip6b;
367 if (a_tmp->len != b_tmp->len) {
368 if (ast_sockaddr_ipv4_mapped(a, &ipv4_mapped)) {
369 a_tmp = &ipv4_mapped;
370 } else if (ast_sockaddr_ipv4_mapped(b, &ipv4_mapped)) {
371 b_tmp = &ipv4_mapped;
375 if (a->len < b->len) {
377 } else if (a->len > b->len) {
381 switch (a_tmp->ss.ss_family) {
383 ip4a = &((const struct sockaddr_in*)&a_tmp->ss)->sin_addr;
384 ip4b = &((const struct sockaddr_in*)&b_tmp->ss)->sin_addr;
385 ret = memcmp(ip4a, ip4b, sizeof(*ip4a));
388 ip6a = &((const struct sockaddr_in6*)&a_tmp->ss)->sin6_addr;
389 ip6b = &((const struct sockaddr_in6*)&b_tmp->ss)->sin6_addr;
390 ret = memcmp(ip6a, ip6b, sizeof(*ip6a));
396 uint16_t _ast_sockaddr_port(const struct ast_sockaddr *addr, const char *file, int line, const char *func)
398 if (addr->ss.ss_family == AF_INET &&
399 addr->len == sizeof(struct sockaddr_in)) {
400 return ntohs(((struct sockaddr_in *)&addr->ss)->sin_port);
401 } else if (addr->ss.ss_family == AF_INET6 &&
402 addr->len == sizeof(struct sockaddr_in6)) {
403 return ntohs(((struct sockaddr_in6 *)&addr->ss)->sin6_port);
405 if (option_debug >= 1) {
406 ast_log(__LOG_DEBUG, file, line, func, "Not an IPv4 nor IPv6 address, cannot get port.\n");
411 void _ast_sockaddr_set_port(struct ast_sockaddr *addr, uint16_t port, const char *file, int line, const char *func)
413 if (addr->ss.ss_family == AF_INET &&
414 addr->len == sizeof(struct sockaddr_in)) {
415 ((struct sockaddr_in *)&addr->ss)->sin_port = htons(port);
416 } else if (addr->ss.ss_family == AF_INET6 &&
417 addr->len == sizeof(struct sockaddr_in6)) {
418 ((struct sockaddr_in6 *)&addr->ss)->sin6_port = htons(port);
419 } else if (option_debug >= 1) {
420 ast_log(__LOG_DEBUG, file, line, func,
421 "Not an IPv4 nor IPv6 address, cannot set port.\n");
425 uint32_t ast_sockaddr_ipv4(const struct ast_sockaddr *addr)
427 const struct sockaddr_in *sin = (struct sockaddr_in *)&addr->ss;
428 return ntohl(sin->sin_addr.s_addr);
431 int ast_sockaddr_is_ipv4(const struct ast_sockaddr *addr)
433 return addr->ss.ss_family == AF_INET &&
434 addr->len == sizeof(struct sockaddr_in);
437 int ast_sockaddr_is_ipv4_mapped(const struct ast_sockaddr *addr)
439 const struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)&addr->ss;
440 return addr->len && IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr);
443 int ast_sockaddr_is_ipv4_multicast(const struct ast_sockaddr *addr)
445 return ((ast_sockaddr_ipv4(addr) & 0xf0000000) == 0xe0000000);
448 int ast_sockaddr_is_ipv6_link_local(const struct ast_sockaddr *addr)
450 const struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)&addr->ss;
451 return ast_sockaddr_is_ipv6(addr) && IN6_IS_ADDR_LINKLOCAL(&sin6->sin6_addr);
454 int ast_sockaddr_is_ipv6(const struct ast_sockaddr *addr)
456 return addr->ss.ss_family == AF_INET6 &&
457 addr->len == sizeof(struct sockaddr_in6);
460 int ast_sockaddr_is_any(const struct ast_sockaddr *addr)
463 struct sockaddr_storage ss;
464 struct sockaddr_in sin;
465 struct sockaddr_in6 sin6;
470 return (ast_sockaddr_is_ipv4(addr) && (tmp_addr.sin.sin_addr.s_addr == INADDR_ANY)) ||
471 (ast_sockaddr_is_ipv6(addr) && IN6_IS_ADDR_UNSPECIFIED(&tmp_addr.sin6.sin6_addr));
474 int ast_sockaddr_hash(const struct ast_sockaddr *addr)
477 * For IPv4, return the IP address as-is. For IPv6, return the last 32
480 switch (addr->ss.ss_family) {
482 return ((const struct sockaddr_in *)&addr->ss)->sin_addr.s_addr;
484 return ((uint32_t *)&((const struct sockaddr_in6 *)&addr->ss)->sin6_addr)[3];
486 ast_log(LOG_ERROR, "Unknown address family '%d'.\n",
492 const char *ast_transport2str(enum ast_transport transport)
495 case AST_TRANSPORT_TLS:
497 case AST_TRANSPORT_UDP:
499 case AST_TRANSPORT_TCP:
501 case AST_TRANSPORT_WS:
503 case AST_TRANSPORT_WSS:
510 int ast_accept(int sockfd, struct ast_sockaddr *addr)
512 addr->len = sizeof(addr->ss);
513 return accept(sockfd, (struct sockaddr *)&addr->ss, &addr->len);
516 int ast_bind(int sockfd, const struct ast_sockaddr *addr)
518 return bind(sockfd, (const struct sockaddr *)&addr->ss, addr->len);
521 int ast_connect(int sockfd, const struct ast_sockaddr *addr)
523 return connect(sockfd, (const struct sockaddr *)&addr->ss, addr->len);
526 int ast_getsockname(int sockfd, struct ast_sockaddr *addr)
528 addr->len = sizeof(addr->ss);
529 return getsockname(sockfd, (struct sockaddr *)&addr->ss, &addr->len);
532 ssize_t ast_recvfrom(int sockfd, void *buf, size_t len, int flags,
533 struct ast_sockaddr *src_addr)
535 src_addr->len = sizeof(src_addr->ss);
536 return recvfrom(sockfd, buf, len, flags,
537 (struct sockaddr *)&src_addr->ss, &src_addr->len);
540 ssize_t ast_sendto(int sockfd, const void *buf, size_t len, int flags,
541 const struct ast_sockaddr *dest_addr)
543 return sendto(sockfd, buf, len, flags,
544 (const struct sockaddr *)&dest_addr->ss, dest_addr->len);
547 int ast_set_qos(int sockfd, int tos, int cos, const char *desc)
552 struct ast_sockaddr addr;
554 /* If the sock address is IPv6, the TCLASS field must be set. */
555 set_tclass = !ast_getsockname(sockfd, &addr) && ast_sockaddr_is_ipv6(&addr) ? 1 : 0;
557 /* If the the sock address is IPv4 or (IPv6 set to any address [::]) set TOS bits */
558 set_tos = (!set_tclass || (set_tclass && ast_sockaddr_is_any(&addr))) ? 1 : 0;
561 if ((res = setsockopt(sockfd, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)))) {
562 ast_log(LOG_WARNING, "Unable to set %s DSCP TOS value to %d (may be you have no "
563 "root privileges): %s\n", desc, tos, strerror(errno));
565 ast_verb(2, "Using %s TOS bits %d\n", desc, tos);
569 #if defined(IPV6_TCLASS) && defined(IPPROTO_IPV6)
571 if (!ast_getsockname(sockfd, &addr) && ast_sockaddr_is_ipv6(&addr)) {
572 if ((res = setsockopt(sockfd, IPPROTO_IPV6, IPV6_TCLASS, &tos, sizeof(tos)))) {
573 ast_log(LOG_WARNING, "Unable to set %s DSCP TCLASS field to %d (may be you have no "
574 "root privileges): %s\n", desc, tos, strerror(errno));
576 ast_verb(2, "Using %s TOS bits %d in TCLASS field.\n", desc, tos);
583 if (setsockopt(sockfd, SOL_SOCKET, SO_PRIORITY, &cos, sizeof(cos))) {
584 ast_log(LOG_WARNING, "Unable to set %s CoS to %d: %s\n", desc, cos,
587 ast_verb(2, "Using %s CoS mark %d\n", desc, cos);
594 int _ast_sockaddr_to_sin(const struct ast_sockaddr *addr,
595 struct sockaddr_in *sin, const char *file, int line, const char *func)
597 if (ast_sockaddr_isnull(addr)) {
598 memset(sin, 0, sizeof(*sin));
602 if (addr->len != sizeof(*sin)) {
603 ast_log(__LOG_ERROR, file, line, func, "Bad address cast to IPv4\n");
607 if (addr->ss.ss_family != AF_INET && option_debug >= 1) {
608 ast_log(__LOG_DEBUG, file, line, func, "Address family is not AF_INET\n");
611 *sin = *(struct sockaddr_in *)&addr->ss;
615 void _ast_sockaddr_from_sin(struct ast_sockaddr *addr, const struct sockaddr_in *sin,
616 const char *file, int line, const char *func)
618 memcpy(&addr->ss, sin, sizeof(*sin));
620 if (addr->ss.ss_family != AF_INET && option_debug >= 1) {
621 ast_log(__LOG_DEBUG, file, line, func, "Address family is not AF_INET\n");
624 addr->len = sizeof(*sin);