Add support for ICE/STUN/TURN in res_rtp_asterisk and chan_sip.
[asterisk/asterisk.git] / res / pjproject / pjlib / src / pj / addr_resolv_sock.c
1 /* $Id$ */
2 /* 
3  * Copyright (C) 2008-2011 Teluu Inc. (http://www.teluu.com)
4  * Copyright (C) 2003-2008 Benny Prijono <benny@prijono.org>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
19  */
20 #include <pj/addr_resolv.h>
21 #include <pj/assert.h>
22 #include <pj/string.h>
23 #include <pj/errno.h>
24 #include <pj/ip_helper.h>
25 #include <pj/compat/socket.h>
26
27 #if defined(PJ_GETADDRINFO_USE_CFHOST) && PJ_GETADDRINFO_USE_CFHOST!=0
28 #   include <CoreFoundation/CFString.h>
29 #   include <CFNetwork/CFHost.h>
30 #endif
31
32 PJ_DEF(pj_status_t) pj_gethostbyname(const pj_str_t *hostname, pj_hostent *phe)
33 {
34     struct hostent *he;
35     char copy[PJ_MAX_HOSTNAME];
36
37     pj_assert(hostname && hostname ->slen < PJ_MAX_HOSTNAME);
38     
39     if (hostname->slen >= PJ_MAX_HOSTNAME)
40         return PJ_ENAMETOOLONG;
41
42     pj_memcpy(copy, hostname->ptr, hostname->slen);
43     copy[ hostname->slen ] = '\0';
44
45     he = gethostbyname(copy);
46     if (!he) {
47         return PJ_ERESOLVE;
48         /* DO NOT use pj_get_netos_error() since host resolution error
49          * is reported in h_errno instead of errno!
50         return pj_get_netos_error();
51          */
52     }
53
54     phe->h_name = he->h_name;
55     phe->h_aliases = he->h_aliases;
56     phe->h_addrtype = he->h_addrtype;
57     phe->h_length = he->h_length;
58     phe->h_addr_list = he->h_addr_list;
59
60     return PJ_SUCCESS;
61 }
62
63 /* Resolve IPv4/IPv6 address */
64 PJ_DEF(pj_status_t) pj_getaddrinfo(int af, const pj_str_t *nodename,
65                                    unsigned *count, pj_addrinfo ai[])
66 {
67 #if defined(PJ_SOCK_HAS_GETADDRINFO) && PJ_SOCK_HAS_GETADDRINFO!=0
68     char nodecopy[PJ_MAX_HOSTNAME];
69     pj_bool_t has_addr = PJ_FALSE;
70     unsigned i;
71 #if defined(PJ_GETADDRINFO_USE_CFHOST) && PJ_GETADDRINFO_USE_CFHOST!=0
72     CFStringRef hostname;
73     CFHostRef hostRef;
74     pj_status_t status = PJ_SUCCESS;
75 #else
76     int rc;
77     struct addrinfo hint, *res, *orig_res;
78 #endif
79
80     PJ_ASSERT_RETURN(nodename && count && *count && ai, PJ_EINVAL);
81     PJ_ASSERT_RETURN(nodename->ptr && nodename->slen, PJ_EINVAL);
82     PJ_ASSERT_RETURN(af==PJ_AF_INET || af==PJ_AF_INET6 ||
83                      af==PJ_AF_UNSPEC, PJ_EINVAL);
84
85     /* Check if nodename is IP address */
86     pj_bzero(&ai[0], sizeof(ai[0]));
87     if ((af==PJ_AF_INET || af==PJ_AF_UNSPEC) &&
88         pj_inet_pton(PJ_AF_INET, nodename,
89                      &ai[0].ai_addr.ipv4.sin_addr) == PJ_SUCCESS)
90     {
91         af = PJ_AF_INET;
92         has_addr = PJ_TRUE;
93     } else if ((af==PJ_AF_INET6 || af==PJ_AF_UNSPEC) &&
94                pj_inet_pton(PJ_AF_INET6, nodename,
95                             &ai[0].ai_addr.ipv6.sin6_addr) == PJ_SUCCESS)
96     {
97         af = PJ_AF_INET6;
98         has_addr = PJ_TRUE;
99     }
100
101     if (has_addr) {
102         pj_str_t tmp;
103
104         tmp.ptr = ai[0].ai_canonname;
105         pj_strncpy_with_null(&tmp, nodename, PJ_MAX_HOSTNAME);
106         ai[0].ai_addr.addr.sa_family = (pj_uint16_t)af;
107         *count = 1;
108
109         return PJ_SUCCESS;
110     }
111
112     /* Copy node name to null terminated string. */
113     if (nodename->slen >= PJ_MAX_HOSTNAME)
114         return PJ_ENAMETOOLONG;
115     pj_memcpy(nodecopy, nodename->ptr, nodename->slen);
116     nodecopy[nodename->slen] = '\0';
117
118 #if defined(PJ_GETADDRINFO_USE_CFHOST) && PJ_GETADDRINFO_USE_CFHOST!=0
119     hostname =  CFStringCreateWithCStringNoCopy(kCFAllocatorDefault, nodecopy,
120                                                 kCFStringEncodingASCII,
121                                                 kCFAllocatorNull);
122     hostRef = CFHostCreateWithName(kCFAllocatorDefault, hostname);
123     if (CFHostStartInfoResolution(hostRef, kCFHostAddresses, nil)) {
124         CFArrayRef addrRef = CFHostGetAddressing(hostRef, nil);
125         i = 0;
126         if (addrRef != nil) {
127             CFIndex idx, naddr;
128             
129             naddr = CFArrayGetCount(addrRef);
130             for (idx = 0; idx < naddr && i < *count; idx++) {
131                 struct sockaddr *addr;
132                 
133                 addr = (struct sockaddr *)
134                        CFDataGetBytePtr(CFArrayGetValueAtIndex(addrRef, idx));
135                 /* This should not happen. */
136                 pj_assert(addr);
137                 
138                 /* Ignore unwanted address families */
139                 if (af!=PJ_AF_UNSPEC && addr->sa_family != af)
140                     continue;
141
142                 /* Store canonical name */
143                 pj_ansi_strcpy(ai[i].ai_canonname, nodecopy);
144                 
145                 /* Store address */
146                 PJ_ASSERT_ON_FAIL(sizeof(*addr) <= sizeof(pj_sockaddr),
147                                   continue);
148                 pj_memcpy(&ai[i].ai_addr, addr, sizeof(*addr));
149                 PJ_SOCKADDR_RESET_LEN(&ai[i].ai_addr);
150                 
151                 i++;
152             }
153         }
154         
155         *count = i;
156     } else {
157         status = PJ_ERESOLVE;
158     }
159     
160     CFRelease(hostRef);
161     CFRelease(hostname);
162     
163     return status;
164 #else
165     /* Call getaddrinfo() */
166     pj_bzero(&hint, sizeof(hint));
167     hint.ai_family = af;
168
169     rc = getaddrinfo(nodecopy, NULL, &hint, &res);
170     if (rc != 0)
171         return PJ_ERESOLVE;
172
173     orig_res = res;
174
175     /* Enumerate each item in the result */
176     for (i=0; i<*count && res; res=res->ai_next) {
177         /* Ignore unwanted address families */
178         if (af!=PJ_AF_UNSPEC && res->ai_family != af)
179             continue;
180
181         /* Store canonical name (possibly truncating the name) */
182         if (res->ai_canonname) {
183             pj_ansi_strncpy(ai[i].ai_canonname, res->ai_canonname,
184                             sizeof(ai[i].ai_canonname));
185             ai[i].ai_canonname[sizeof(ai[i].ai_canonname)-1] = '\0';
186         } else {
187             pj_ansi_strcpy(ai[i].ai_canonname, nodecopy);
188         }
189
190         /* Store address */
191         PJ_ASSERT_ON_FAIL(res->ai_addrlen <= sizeof(pj_sockaddr), continue);
192         pj_memcpy(&ai[i].ai_addr, res->ai_addr, res->ai_addrlen);
193         PJ_SOCKADDR_RESET_LEN(&ai[i].ai_addr);
194
195         /* Next slot */
196         ++i;
197     }
198
199     *count = i;
200
201     freeaddrinfo(orig_res);
202
203     /* Done */
204     return PJ_SUCCESS;
205 #endif
206
207 #else   /* PJ_SOCK_HAS_GETADDRINFO */
208     pj_bool_t has_addr = PJ_FALSE;
209
210     PJ_ASSERT_RETURN(count && *count, PJ_EINVAL);
211
212     /* Check if nodename is IP address */
213     pj_bzero(&ai[0], sizeof(ai[0]));
214     if ((af==PJ_AF_INET || af==PJ_AF_UNSPEC) &&
215         pj_inet_pton(PJ_AF_INET, nodename,
216                      &ai[0].ai_addr.ipv4.sin_addr) == PJ_SUCCESS)
217     {
218         af = PJ_AF_INET;
219         has_addr = PJ_TRUE;
220     }
221     else if ((af==PJ_AF_INET6 || af==PJ_AF_UNSPEC) &&
222              pj_inet_pton(PJ_AF_INET6, nodename,
223                           &ai[0].ai_addr.ipv6.sin6_addr) == PJ_SUCCESS)
224     {
225         af = PJ_AF_INET6;
226         has_addr = PJ_TRUE;
227     }
228
229     if (has_addr) {
230         pj_str_t tmp;
231
232         tmp.ptr = ai[0].ai_canonname;
233         pj_strncpy_with_null(&tmp, nodename, PJ_MAX_HOSTNAME);
234         ai[0].ai_addr.addr.sa_family = (pj_uint16_t)af;
235         *count = 1;
236
237         return PJ_SUCCESS;
238     }
239
240     if (af == PJ_AF_INET || af == PJ_AF_UNSPEC) {
241         pj_hostent he;
242         unsigned i, max_count;
243         pj_status_t status;
244         
245         /* VC6 complains that "he" is uninitialized */
246         #ifdef _MSC_VER
247         pj_bzero(&he, sizeof(he));
248         #endif
249
250         status = pj_gethostbyname(nodename, &he);
251         if (status != PJ_SUCCESS)
252             return status;
253
254         max_count = *count;
255         *count = 0;
256
257         pj_bzero(ai, max_count * sizeof(pj_addrinfo));
258
259         for (i=0; he.h_addr_list[i] && *count<max_count; ++i) {
260             pj_ansi_strncpy(ai[*count].ai_canonname, he.h_name,
261                             sizeof(ai[*count].ai_canonname));
262             ai[*count].ai_canonname[sizeof(ai[*count].ai_canonname)-1] = '\0';
263
264             ai[*count].ai_addr.ipv4.sin_family = PJ_AF_INET;
265             pj_memcpy(&ai[*count].ai_addr.ipv4.sin_addr,
266                       he.h_addr_list[i], he.h_length);
267             PJ_SOCKADDR_RESET_LEN(&ai[*count].ai_addr);
268
269             (*count)++;
270         }
271
272         return PJ_SUCCESS;
273
274     } else {
275         /* IPv6 is not supported */
276         *count = 0;
277
278         return PJ_EIPV6NOTSUP;
279     }
280 #endif  /* PJ_SOCK_HAS_GETADDRINFO */
281 }
282