belle_sip_resolver.c 19.2 KB
Newer Older
Simon Morlat's avatar
Simon Morlat committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
/*
	belle-sip - SIP (RFC3261) library.
    Copyright (C) 2010  Belledonne Communications SARL

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "belle_sip_resolver.h"

21
#include <stdlib.h>
22 23 24
#ifdef __APPLE__
#include "TargetConditionals.h"
#endif
25 26 27 28 29

#define DNS_EAGAIN  EAGAIN


static struct dns_resolv_conf *resconf(belle_sip_resolver_context_t *ctx) {
jehan's avatar
jehan committed
30
#if !_WIN32 && !HAVE_RESINIT
31
/*#if !_WIN32 && (!HAVE_RESINIT || !TARGET_OS_IPHONE)*/
32
	const char *path;
33
#endif
34 35 36 37 38 39 40 41 42 43
	int error;

	if (ctx->resconf)
		return ctx->resconf;

	if (!(ctx->resconf = dns_resconf_open(&error))) {
		belle_sip_error("%s dns_resconf_open error: %s", __FUNCTION__, dns_strerror(error));
		return NULL;
	}

44 45
#ifdef _WIN32
	error = dns_resconf_loadwin(ctx->resconf);
Ghislain MARY's avatar
Ghislain MARY committed
46 47 48
	if (error) {
		belle_sip_error("%s dns_resconf_loadwin error", __FUNCTION__);
	}
jehan's avatar
jehan committed
49
#elif ANDROID
Ghislain MARY's avatar
Ghislain MARY committed
50 51 52 53
	error = dns_resconf_loadandroid(ctx->resconf);
	if (error) {
		belle_sip_error("%s dns_resconf_loadandroid error", __FUNCTION__);
	}
jehan's avatar
jehan committed
54
#elif HAVE_RESINIT
55
/*#elif HAVE_RESINIT && TARGET_OS_IPHONE*/
jehan's avatar
jehan committed
56 57 58 59
	error = dns_resconf_loadfromresolv(ctx->resconf);
	if (error) {
		belle_sip_error("%s dns_resconf_loadfromresolv error", __FUNCTION__);
	}
60
#else
61 62 63 64 65 66 67 68 69 70
	path = "/etc/resolv.conf";
	error = dns_resconf_loadpath(ctx->resconf, path);
	if (error) {
		belle_sip_error("%s dns_resconf_loadpath error [%s]: %s", __FUNCTION__, path, dns_strerror(error));
		return NULL;
	}

	path = "/etc/nsswitch.conf";
	error = dns_nssconf_loadpath(ctx->resconf, path);
	if (error) {
71
		belle_sip_message("%s dns_nssconf_loadpath error [%s]: %s", __FUNCTION__, path, dns_strerror(error));
72
	}
Ghislain MARY's avatar
Ghislain MARY committed
73
#endif
jehan's avatar
jehan committed
74

75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95
	return ctx->resconf;
}

static struct dns_hosts *hosts(belle_sip_resolver_context_t *ctx) {
	int error;

	if (ctx->hosts)
		return ctx->hosts;

	if (!(ctx->hosts = dns_hosts_local(&error))) {
		belle_sip_error("%s dns_hosts_local error: %s", __FUNCTION__, dns_strerror(error));
		return NULL;
	}

	return ctx->hosts;
}

struct dns_cache *cache(belle_sip_resolver_context_t *ctx) {
	return NULL;
}

96 97 98 99 100 101 102 103 104 105 106
static struct addrinfo * ai_list_append(struct addrinfo *ai_list, struct addrinfo *ai_to_append) {
	struct addrinfo *ai_current = ai_list;
	if (ai_to_append == NULL) return ai_list;
	if (ai_list == NULL) return ai_to_append;
	while (ai_current->ai_next != NULL) {
		ai_current = ai_current->ai_next;
	}
	ai_current->ai_next = ai_to_append;
	return ai_list;
}

107 108 109 110 111 112 113 114
static int srv_compare_prio(const void *psrv1, const void *psrv2){
	struct dns_srv *srv1=(struct dns_srv*)psrv1;
	struct dns_srv *srv2=(struct dns_srv*)psrv2;
	if (srv1->priority < srv2->priority) return -1;
	if (srv1->priority == srv2->priority) return 0;
	return 1;
}

Ghislain MARY's avatar
Ghislain MARY committed
115
static int resolver_process_data(belle_sip_resolver_context_t *ctx, unsigned int revents) {
116 117
	char host[NI_MAXHOST + 1];
	char service[NI_MAXSERV + 1];
118 119
	struct dns_packet *ans;
	struct dns_rr_i *I;
120
	struct dns_rr_i dns_rr_it;
Ghislain MARY's avatar
Ghislain MARY committed
121
	int error;
122 123 124

	if (revents & BELLE_SIP_EVENT_TIMEOUT) {
		belle_sip_error("%s timed-out", __FUNCTION__);
125 126 127 128 129
		if ((ctx->type == DNS_T_A) || (ctx->type == DNS_T_AAAA)) {
			ctx->cb(ctx->cb_data, ctx->name, NULL);
		} else if (ctx->type == DNS_T_SRV) {
			ctx->srv_cb(ctx->cb_data, ctx->name, NULL);
		}
130
		ctx->done=TRUE;
131 132 133
		return BELLE_SIP_STOP;
	}
	if (ctx->cancelled) {
134
		ctx->done=TRUE;
135 136 137 138 139 140 141 142 143 144
		return BELLE_SIP_STOP;
	}

	error = dns_res_check(ctx->R);
	if (!error) {
		struct dns_rr rr;
		union dns_any any;
		enum dns_section section = DNS_S_AN;

		ans = dns_res_fetch(ctx->R, &error);
145 146
		memset(&dns_rr_it, 0, sizeof dns_rr_it);
		I = dns_rr_i_init(&dns_rr_it, ans);
147

148 149 150 151
		while (dns_rr_grep(&rr, 1, I, ans, &error)) {
			if (rr.section == section) {
				if ((error = dns_any_parse(dns_any_init(&any, sizeof(any)), &rr, ans))) {
					belle_sip_error("%s dns_any_parse error: %s", __FUNCTION__, dns_strerror(error));
152
					break;
153
				}
Ghislain MARY's avatar
Ghislain MARY committed
154
				if ((ctx->type == DNS_T_AAAA) && (rr.class == DNS_C_IN) && (rr.type == DNS_T_AAAA)) {
155
					struct dns_aaaa *aaaa = &any.aaaa;
156 157 158 159 160 161 162
					struct sockaddr_in6 sin6;
					memset(&sin6, 0, sizeof(sin6));
					memcpy(&sin6.sin6_addr, &aaaa->addr, sizeof(sin6.sin6_addr));
					sin6.sin6_family = AF_INET6;
					sin6.sin6_port = ctx->port;
					if (getnameinfo((struct sockaddr *)&sin6, sizeof(sin6), host, sizeof(host), service, sizeof(service), NI_NUMERICHOST) != 0)
						continue;
163
					ctx->ai_list = ai_list_append(ctx->ai_list, belle_sip_ip_address_to_addrinfo(ctx->family, host, ctx->port));
164
					belle_sip_message("%s resolved to %s", ctx->name, host);
Ghislain MARY's avatar
Ghislain MARY committed
165 166 167 168 169 170 171 172 173
				} else if ((ctx->type == DNS_T_A) && (rr.class == DNS_C_IN) && (rr.type == DNS_T_A)) {
					struct dns_a *a = &any.a;
					struct sockaddr_in sin;
					memset(&sin, 0, sizeof(sin));
					memcpy(&sin.sin_addr, &a->addr, sizeof(sin.sin_addr));
					sin.sin_family = AF_INET;
					sin.sin_port = ctx->port;
					if (getnameinfo((struct sockaddr *)&sin, sizeof(sin), host, sizeof(host), service, sizeof(service), NI_NUMERICHOST) != 0)
						continue;
174
					ctx->ai_list = ai_list_append(ctx->ai_list, belle_sip_ip_address_to_addrinfo(ctx->family, host, ctx->port));
Ghislain MARY's avatar
Ghislain MARY committed
175 176 177 178
					belle_sip_message("%s resolved to %s", ctx->name, host);
				} else if ((ctx->type == DNS_T_SRV) && (rr.class == DNS_C_IN) && (rr.type == DNS_T_SRV)) {
					struct dns_srv *srv = &any.srv;
					struct dns_srv *res = belle_sip_malloc(sizeof(struct dns_srv));
Ghislain MARY's avatar
Ghislain MARY committed
179
					memcpy(res, srv, sizeof(struct dns_srv));
Ghislain MARY's avatar
Ghislain MARY committed
180
					snprintf(host, sizeof(host), "[target:%s port:%d prio:%d weight:%d]", srv->target, srv->port, srv->priority, srv->weight);
181
					ctx->srv_list = belle_sip_list_insert_sorted(ctx->srv_list, res, srv_compare_prio);
Ghislain MARY's avatar
Ghislain MARY committed
182
					belle_sip_message("SRV %s resolved to %s", ctx->name, host);
183 184 185 186
				}
			}
		}
		free(ans);
Ghislain MARY's avatar
Ghislain MARY committed
187
		ctx->done=TRUE;
Ghislain MARY's avatar
Ghislain MARY committed
188 189 190
		if ((ctx->type == DNS_T_A) || (ctx->type == DNS_T_AAAA)) {
			ctx->cb(ctx->cb_data, ctx->name, ctx->ai_list);
		} else if (ctx->type == DNS_T_SRV) {
191
			ctx->srv_cb(ctx->cb_data, ctx->name, ctx->srv_list);
Ghislain MARY's avatar
Ghislain MARY committed
192
		}
Ghislain MARY's avatar
Ghislain MARY committed
193 194 195 196 197
		if (ctx->done == TRUE) {
			return BELLE_SIP_STOP;
		} else {
			return BELLE_SIP_CONTINUE;
		}
198 199 200
	}
	if (error != DNS_EAGAIN) {
		belle_sip_error("%s dns_res_check error: %s (%d)", __FUNCTION__, dns_strerror(error), error);
201
		ctx->done=TRUE;
202 203 204 205 206
		if ((ctx->type == DNS_T_A) || (ctx->type == DNS_T_AAAA)) {
			ctx->cb(ctx->cb_data, ctx->name, NULL);
		} else if (ctx->type == DNS_T_SRV) {
			ctx->srv_cb(ctx->cb_data, ctx->name, NULL);
		}
207 208 209 210 211 212 213
		return BELLE_SIP_STOP;
	}

	dns_res_poll(ctx->R, 0);
	return BELLE_SIP_CONTINUE;
}

Ghislain MARY's avatar
Ghislain MARY committed
214
static int _resolver_send_query(belle_sip_resolver_context_t *ctx, belle_sip_source_func_t datafunc, int timeout) {
215 216
	int error;

217
	if (!ctx->stack->resolver_send_error) {
Ghislain MARY's avatar
Ghislain MARY committed
218
		error = dns_res_submit(ctx->R, ctx->name, ctx->type, DNS_C_IN);
219 220 221 222
		if (error)
			belle_sip_error("%s dns_res_submit error [%s]: %s", __FUNCTION__, ctx->name, dns_strerror(error));
	} else {
		/* Error simulation */
223
		error = ctx->stack->resolver_send_error;
224 225 226 227 228 229
		belle_sip_error("%s dns_res_submit error [%s]: simulated error %d", __FUNCTION__, ctx->name, error);
	}
	if (error < 0) {
		return -1;
	}

230 231 232 233
	if ((*datafunc)(ctx, 0) == BELLE_SIP_CONTINUE) {
		/*only init source if res inprogress*/
		belle_sip_socket_source_init((belle_sip_source_t*)ctx, datafunc, ctx, dns_res_pollfd(ctx->R), BELLE_SIP_EVENT_READ | BELLE_SIP_EVENT_TIMEOUT, timeout);
	}
234 235 236 237 238 239 240 241 242 243 244
	return 0;
}

typedef struct delayed_send {
	belle_sip_resolver_context_t *ctx;
	belle_sip_source_func_t datafunc;
	int timeout;
} delayed_send_t;

static int on_delayed_send_do(delayed_send_t *ds) {
	belle_sip_message("%s sending now", __FUNCTION__);
Ghislain MARY's avatar
Ghislain MARY committed
245
	_resolver_send_query(ds->ctx, ds->datafunc, ds->timeout);
246 247 248 249 250
	belle_sip_object_unref(ds->ctx);
	belle_sip_free(ds);
	return FALSE;
}

Ghislain MARY's avatar
Ghislain MARY committed
251
static int resolver_send_query(belle_sip_resolver_context_t *ctx, belle_sip_source_func_t datafunc, int timeout) {
252
	struct dns_hints *(*hints)() = &dns_hints_local;
253 254 255 256
	struct dns_options *opts;
#ifndef HAVE_C99
	struct dns_options opts_st;
#endif
257 258 259 260 261 262 263 264 265 266 267
	int error;

	if (!ctx->name) return -1;

	if (resconf(ctx))
		resconf(ctx)->options.recurse = 0;
	else
		return -1;
	if (!hosts(ctx))
		return -1;

268 269
	memset(&opts_st, 0, sizeof opts_st);
	opts = &opts_st;
270

271
	if (!(ctx->R = dns_res_open(ctx->resconf, ctx->hosts, dns_hints_mortal(hints(ctx->resconf, &error)), cache(ctx), opts, &error))) {
272 273 274 275
		belle_sip_error("%s dns_res_open error [%s]: %s", __FUNCTION__, ctx->name, dns_strerror(error));
		return -1;
	}

276
	if (ctx->stack->resolver_tx_delay > 0) {
277 278 279 280
		delayed_send_t *ds = belle_sip_new(delayed_send_t);
		ds->ctx = (belle_sip_resolver_context_t *)belle_sip_object_ref(ctx);
		ds->datafunc = datafunc;
		ds->timeout = timeout;
281
		belle_sip_main_loop_add_timeout(ctx->stack->ml, (belle_sip_source_func_t)on_delayed_send_do, ds, ctx->stack->resolver_tx_delay);
282
		belle_sip_socket_source_init((belle_sip_source_t*)ctx, datafunc, ctx, dns_res_pollfd(ctx->R), BELLE_SIP_EVENT_READ | BELLE_SIP_EVENT_TIMEOUT, ctx->stack->resolver_tx_delay + 1000);
283
		belle_sip_message("%s DNS resolution delayed by %d ms", __FUNCTION__, ctx->stack->resolver_tx_delay);
284 285
		return 0;
	} else {
Ghislain MARY's avatar
Ghislain MARY committed
286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304
		return _resolver_send_query(ctx, datafunc, timeout);
	}
}

static int resolver_start_query(belle_sip_resolver_context_t *ctx) {
	if (resolver_send_query(ctx,
			(belle_sip_source_func_t)resolver_process_data,
			belle_sip_stack_get_dns_timeout(ctx->stack)) < 0) {
		belle_sip_object_unref(ctx);
		return 0;
	}
	if ((ctx->done == FALSE) && (ctx->started == FALSE)) {
		/* The resolver context must never be removed manually from the main loop */
		belle_sip_main_loop_add_source(ctx->ml, (belle_sip_source_t *)ctx);
		belle_sip_object_unref(ctx);	/* The main loop has a ref on it */
		ctx->started = TRUE;
		return ctx->source.id;
	} else {
		return 0; /*resolution done synchronously*/
305 306 307 308
	}
}


Simon Morlat's avatar
Simon Morlat committed
309 310 311 312 313 314 315 316 317 318 319 320

int belle_sip_addrinfo_to_ip(const struct addrinfo *ai, char *ip, size_t ip_size, int *port){
	char serv[16];
	int err=getnameinfo(ai->ai_addr,ai->ai_addrlen,ip,ip_size,serv,sizeof(serv),NI_NUMERICHOST|NI_NUMERICSERV);
	if (err!=0){
		belle_sip_error("getnameinfo() error: %s",gai_strerror(err));
		strncpy(ip,"<bug!!>",ip_size);
	}
	if (port) *port=atoi(serv);
	return 0;
}

321
struct addrinfo * belle_sip_ip_address_to_addrinfo(int family, const char *ipaddress, int port){
Simon Morlat's avatar
Simon Morlat committed
322 323 324 325 326 327
	struct addrinfo *res=NULL;
	struct addrinfo hints={0};
	char serv[10];
	int err;

	snprintf(serv,sizeof(serv),"%i",port);
328
	hints.ai_family=family;
Simon Morlat's avatar
Simon Morlat committed
329
	hints.ai_flags=AI_NUMERICSERV|AI_NUMERICHOST;
330
	hints.ai_socktype=SOCK_STREAM; //not used but it's needed to specify it because otherwise getaddrinfo returns one struct addrinfo per socktype.
331 332 333
	
	if (family==AF_INET6) hints.ai_flags|=AI_V4MAPPED;
	
Simon Morlat's avatar
Simon Morlat committed
334 335 336 337 338 339 340 341
	err=getaddrinfo(ipaddress,serv,&hints,&res);
	if (err!=0){
		return NULL;
	}
	return res;
}


342
static void belle_sip_resolver_context_destroy(belle_sip_resolver_context_t *ctx){
343 344
	/* Do not free elements of ctx->ai_list with freeaddrinfo(). Let the caller do it, otherwise
	   it will not be able to use them after the resolver has been destroyed. */
Ghislain MARY's avatar
Ghislain MARY committed
345
	if (ctx->name != NULL) {
Simon Morlat's avatar
Simon Morlat committed
346
		belle_sip_free(ctx->name);
Ghislain MARY's avatar
Ghislain MARY committed
347 348 349
		ctx->name = NULL;
	}
	if (ctx->R != NULL) {
350
		dns_res_close(ctx->R);
Ghislain MARY's avatar
Ghislain MARY committed
351 352 353
		ctx->R = NULL;
	}
	if (ctx->hosts != NULL) {
354
		dns_hosts_close(ctx->hosts);
Ghislain MARY's avatar
Ghislain MARY committed
355
		ctx->hosts = NULL;
356
	}
Ghislain MARY's avatar
Ghislain MARY committed
357
	if (ctx->resconf != NULL) {
358
		free(ctx->resconf);
Ghislain MARY's avatar
Ghislain MARY committed
359 360
		ctx->resconf = NULL;
	}
Simon Morlat's avatar
Simon Morlat committed
361 362
}

363
BELLE_SIP_DECLARE_NO_IMPLEMENTED_INTERFACES(belle_sip_resolver_context_t);
Simon Morlat's avatar
Simon Morlat committed
364
BELLE_SIP_INSTANCIATE_VPTR(belle_sip_resolver_context_t, belle_sip_source_t,belle_sip_resolver_context_destroy, NULL, NULL,FALSE);
Simon Morlat's avatar
Simon Morlat committed
365

Ghislain MARY's avatar
Ghislain MARY committed
366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450
struct belle_sip_recursive_resolve_data {
	belle_sip_resolver_context_t *ctx;
	belle_sip_resolver_callback_t cb;
	void *data;
	char *name;
	int port;
	int family;
	belle_sip_list_t *srv_list;
	int next_srv;
	struct addrinfo *ai_list;
};

static char * srv_prefix_from_transport(const char *transport) {
	char *prefix = "";

	if (strcasecmp(transport, "udp") == 0) {
		prefix = "_sip._udp.";
	} else if (strcasecmp(transport, "tcp") == 0) {
		prefix = "_sip._tcp.";
	} else if (strcasecmp(transport, "tls") == 0) {
		prefix = "_sips._tcp.";
	} else {
		prefix = "_sip._udp.";
	}

	return prefix;
}

static void process_a_results(void *data, const char *name, struct addrinfo *ai_list);

static int start_a_query_from_srv_results(struct belle_sip_recursive_resolve_data *rec_data) {
	if (rec_data->srv_list && belle_sip_list_size(rec_data->srv_list) > rec_data->next_srv) {
		struct dns_srv *srv;
		rec_data->next_srv++;
		srv = belle_sip_list_nth_data(rec_data->srv_list, 0);
		rec_data->ctx->cb = process_a_results;
		rec_data->ctx->name = srv->target;
		rec_data->ctx->port = srv->port;
		rec_data->ctx->family = rec_data->family;
		rec_data->ctx->type = (rec_data->family == AF_INET6) ? DNS_T_AAAA : DNS_T_A;
		resolver_start_query(rec_data->ctx);
		return 1;
	}

	return 0;
}

static void process_a_results(void *data, const char *name, struct addrinfo *ai_list) {
	struct belle_sip_recursive_resolve_data *rec_data = (struct belle_sip_recursive_resolve_data *)data;
	rec_data->ai_list = ai_list_append(rec_data->ai_list, ai_list);
	if (!start_a_query_from_srv_results(rec_data)) {
		/* All the SRV results have been queried, return the results */
		(*rec_data->cb)(rec_data->data, rec_data->name, rec_data->ai_list);
		rec_data->ctx->name = NULL;
		if (rec_data->srv_list != NULL) {
			belle_sip_list_for_each(rec_data->srv_list, belle_sip_free);
			belle_sip_list_free(rec_data->srv_list);
			rec_data->srv_list = NULL;
		}
		if (rec_data->name != NULL) {
			belle_sip_free(rec_data->name);
		}
		belle_sip_free(rec_data);
	} else {
		rec_data->ctx->done = FALSE;
	}
}

static void process_srv_results(void *data, const char *name, belle_sip_list_t *srv_list) {
	struct belle_sip_recursive_resolve_data *rec_data = (struct belle_sip_recursive_resolve_data *)data;
	rec_data->srv_list = srv_list;
	rec_data->ctx->done = FALSE;
	belle_sip_resolver_context_destroy(rec_data->ctx);
	if (!start_a_query_from_srv_results(rec_data)) {
		/* There was no SRV results, try to perform the A or AAAA directly. */
		rec_data->ctx->cb = process_a_results;
		rec_data->ctx->name = rec_data->name;
		rec_data->ctx->port = rec_data->port;
		rec_data->ctx->family = rec_data->family;
		rec_data->ctx->type = (rec_data->family == AF_INET6) ? DNS_T_AAAA : DNS_T_A;
		resolver_start_query(rec_data->ctx);
	}
}

unsigned long belle_sip_resolve(belle_sip_stack_t *stack, const char *transport, const char *name, int port, int family, belle_sip_resolver_callback_t cb, void *data, belle_sip_main_loop_t *ml) {
Ghislain MARY's avatar
Ghislain MARY committed
451
	struct addrinfo *res = belle_sip_ip_address_to_addrinfo(family, name, port);
452
	if (res == NULL) {
Ghislain MARY's avatar
Ghislain MARY committed
453 454
		/* Then perform asynchronous DNS SRV query */
		struct belle_sip_recursive_resolve_data *rec_data = belle_sip_malloc0(sizeof(struct belle_sip_recursive_resolve_data));
455
		belle_sip_resolver_context_t *ctx = belle_sip_object_new(belle_sip_resolver_context_t);
456
		ctx->stack = stack;
Ghislain MARY's avatar
Ghislain MARY committed
457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483
		ctx->ml = ml;
		ctx->cb_data = rec_data;
		ctx->srv_cb = process_srv_results;
		ctx->name = belle_sip_concat(srv_prefix_from_transport(transport), name, NULL);
		ctx->type = DNS_T_SRV;
		rec_data->ctx = ctx;
		rec_data->cb = cb;
		rec_data->data = data;
		rec_data->name = belle_sip_strdup(name);
		rec_data->port = port;
		if (family == 0) family = AF_UNSPEC;
		rec_data->family = family;
		return resolver_start_query(ctx);
	} else {
		/* There is no resolve to be done */
		cb(data, name, res);
		return 0;
	}
}

unsigned long belle_sip_resolve_a(belle_sip_stack_t *stack, const char *name, int port, int family, belle_sip_resolver_callback_t cb , void *data, belle_sip_main_loop_t *ml) {
	struct addrinfo *res = belle_sip_ip_address_to_addrinfo(family, name, port);
	if (res == NULL) {
		/* Then perform asynchronous DNS A or AAAA query */
		belle_sip_resolver_context_t *ctx = belle_sip_object_new(belle_sip_resolver_context_t);
		ctx->stack = stack;
		ctx->ml = ml;
484 485 486 487 488 489
		ctx->cb_data = data;
		ctx->cb = cb;
		ctx->name = belle_sip_strdup(name);
		ctx->port = port;
		if (family == 0) family = AF_UNSPEC;
		ctx->family = family;
Ghislain MARY's avatar
Ghislain MARY committed
490
		ctx->type = (ctx->family == AF_INET6) ? DNS_T_AAAA : DNS_T_A;
Ghislain MARY's avatar
Ghislain MARY committed
491
		return resolver_start_query(ctx);
492
	} else {
Ghislain MARY's avatar
Ghislain MARY committed
493
		/* There is no resolve to be done */
494
		cb(data, name, res);
495 496
		return 0;
	}
Simon Morlat's avatar
Simon Morlat committed
497 498
}

499
unsigned long belle_sip_resolve_srv(belle_sip_stack_t *stack, const char *name, const char *transport, belle_sip_resolver_srv_callback_t cb, void *data, belle_sip_main_loop_t *ml) {
Ghislain MARY's avatar
Ghislain MARY committed
500 501
	belle_sip_resolver_context_t *ctx = belle_sip_object_new(belle_sip_resolver_context_t);
	ctx->stack = stack;
Ghislain MARY's avatar
Ghislain MARY committed
502
	ctx->ml = ml;
Ghislain MARY's avatar
Ghislain MARY committed
503
	ctx->cb_data = data;
504
	ctx->srv_cb = cb;
Ghislain MARY's avatar
Ghislain MARY committed
505
	ctx->name = belle_sip_concat(srv_prefix_from_transport(transport), name, NULL);
Ghislain MARY's avatar
Ghislain MARY committed
506
	ctx->type = DNS_T_SRV;
Ghislain MARY's avatar
Ghislain MARY committed
507
	return resolver_start_query(ctx);
Ghislain MARY's avatar
Ghislain MARY committed
508 509
}

510 511 512 513 514 515 516 517 518 519
void belle_sip_resolve_cancel(belle_sip_main_loop_t *ml, unsigned long id){
	if (id!=0){
		belle_sip_source_t *s=belle_sip_main_loop_find_source(ml,id);
		if (s){
			belle_sip_resolver_context_t *res=BELLE_SIP_RESOLVER_CONTEXT(s);
			res->cancelled=1;
		}
	}
}

520 521

void belle_sip_get_src_addr_for(const struct sockaddr *dest, socklen_t destlen, struct sockaddr *src, socklen_t *srclen, int local_port){
522
	int af_type=dest->sa_family;
523 524
	int sock=socket(af_type,SOCK_DGRAM,IPPROTO_UDP);
	
525
	if (sock==(belle_sip_socket_t)-1){
526 527
		belle_sip_fatal("Could not create socket: %s",belle_sip_get_socket_error_string());
		goto fail;
528 529
	}
	if (connect(sock,dest,destlen)==-1){
530 531
		belle_sip_error("belle_sip_get_src_addr_for: connect() failed: %s",belle_sip_get_socket_error_string());
		goto fail;
532 533
	}
	if (getsockname(sock,src,srclen)==-1){
534 535 536
		belle_sip_error("belle_sip_get_src_addr_for: getsockname() failed: %s",belle_sip_get_socket_error_string());
		goto fail;
	}
537 538 539 540 541 542 543 544 545
	
	if (af_type==AF_INET6){
		struct sockaddr_in6 *sin6=(struct sockaddr_in6*)src;
		sin6->sin6_port=htons(local_port);
	}else{
		struct sockaddr_in *sin=(struct sockaddr_in*)src;
		sin->sin_port=htons(local_port);
	}
	
546 547 548 549 550 551 552 553 554
	close_socket(sock);
	return;
fail:
	{
		struct addrinfo hints={0},*res=NULL;
		int err;
		hints.ai_family=af_type;
		err=getaddrinfo(af_type==AF_INET ? "0.0.0.0" : "::0","0",&hints,&res);
		if (err!=0) belle_sip_fatal("belle_sip_get_src_addr_for(): getaddrinfo failed: %s",belle_sip_get_socket_error_string_from_code(err));
555
		memcpy(src,res->ai_addr,MIN((size_t)*srclen,res->ai_addrlen));
556 557
		*srclen=res->ai_addrlen;
		freeaddrinfo(res);
558
	}
559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585
	if (sock!=(belle_sip_socket_t)-1) close_socket(sock);
}

#ifndef IN6_GET_ADDR_V4MAPPED
#define IN6_GET_ADDR_V4MAPPED(sin6_addr)	*(unsigned int*)((unsigned char*)(sin6_addr)+12)
#endif


void belle_sip_address_remove_v4_mapping(const struct sockaddr *v6, struct sockaddr *result, socklen_t *result_len){
	if (v6->sa_family==AF_INET6){
		struct sockaddr_in6 *in6=(struct sockaddr_in6*)v6;
		
		if (IN6_IS_ADDR_V4MAPPED(&in6->sin6_addr)){
			struct sockaddr_in *in=(struct sockaddr_in*)result;
			result->sa_family=AF_INET;
			in->sin_addr.s_addr = IN6_GET_ADDR_V4MAPPED(&in6->sin6_addr);
			in->sin_port=in6->sin6_port;
			*result_len=sizeof(struct sockaddr_in);
		}else{
			if (v6!=result) memcpy(result,v6,sizeof(struct sockaddr_in6));
			*result_len=sizeof(struct sockaddr_in6);
		}
		
	}else{
		*result_len=sizeof(struct sockaddr_in);
		if (v6!=result) memcpy(result,v6,sizeof(struct sockaddr_in));
	}
586
}
587 588 589