resolve.c
1b71270a
 /* $Id$*/
7dd0b342
 /*
  *
  * Copyright (C) 2001-2003 Fhg Fokus
  *
  * This file is part of ser, a free SIP server.
  *
  * ser 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 2 of the License, or
  * (at your option) any later version
  *
  * For a license to use the ser software under conditions
  * other than those described here, or to purchase support for this
  * software, please contact iptel.org by e-mail at the following addresses:
  *    info@iptel.org
  *
  * ser 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, write to the Free Software 
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */
d531a5d5
 /*
  * History:
  * -------
  *  2003-02-13  added proto to sip_resolvehost, for SRV lookups (andrei)
3f601bd0
  *  2003-07-03  default port value set according to proto (andrei)
d531a5d5
  */ 
7dd0b342
 
1b71270a
 
53531528
 #include <sys/types.h>
1b71270a
 #include <netinet/in.h>
53531528
 #include <arpa/nameser.h>
1b71270a
 #include <resolv.h>
 #include <string.h>
 
 #include "resolve.h"
 #include "dprint.h"
 #include "mem/mem.h"
4c9f3fde
 #include "ip_addr.h"
1b71270a
 
 
 
 /* mallocs for local stuff */
 #define local_malloc pkg_malloc
 #define local_free   pkg_free
 
 
 
  /*  skips over a domain name in a dns message
  *  (it can be  a sequence of labels ending in \0, a pointer or
  *   a sequence of labels ending in a pointer -- see rfc1035
  *   returns pointer after the domain name or null on error*/
 unsigned char* dns_skipname(unsigned char* p, unsigned char* end)
 {
 	while(p<end){
 		/* check if \0 (root label length) */
 		if (*p==0){
 			p+=1;
 			break;
 		}
 		/* check if we found a pointer */
 		if (((*p)&0xc0)==0xc0){
 			/* if pointer skip over it (2 bytes) & we found the end */
 			p+=2;
 			break;
 		}
 		/* normal label */
 		p+=*p+1;
 	}
a2acf4e8
 	return (p>=end)?0:p;
1b71270a
 }
 
 
 
 /* parses the srv record into a srv_rdata structure
  *   msg   - pointer to the dns message
  *   end   - pointer to the end of the message
  *   rdata - pointer  to the rdata part of the srv answer
  * returns 0 on error, or a dyn. alloc'ed srv_rdata structure */
 /* SRV rdata format:
  *            111111
  *  0123456789012345
  * +----------------+
  * |     priority   |
  * |----------------|
  * |     weight     |
  * |----------------|
  * |   port number  |
  * |----------------|
  * |                |
  * ~      name      ~
  * |                |
  * +----------------+
  */
 struct srv_rdata* dns_srv_parser( unsigned char* msg, unsigned char* end,
 								  unsigned char* rdata)
 {
 	struct srv_rdata* srv;
 	int len;
 	
 	srv=0;
a2acf4e8
 	if ((rdata+6)>=end) goto error;
1b71270a
 	srv=(struct srv_rdata*)local_malloc(sizeof(struct srv_rdata));
 	if (srv==0){
4c55b6cf
 		LOG(L_ERR, "ERROR: dns_srv_parser: out of memory\n");
1b71270a
 		goto error;
 	}
 	
 	memcpy((void*)&srv->priority, rdata, 2);
 	memcpy((void*)&srv->weight,   rdata+2, 2);
 	memcpy((void*)&srv->port,     rdata+4, 2);
 	rdata+=6;
 	srv->priority=ntohs(srv->priority);
 	srv->weight=ntohs(srv->weight);
 	srv->port=ntohs(srv->port);
4c55b6cf
 	if ((len=dn_expand(msg, end, rdata, srv->name, MAX_DNS_NAME-1))==-1)
1b71270a
 		goto error;
4c55b6cf
 	/* add terminating 0 ? (warning: len=compressed name len) */
1b71270a
 	return srv;
 error:
 	if (srv) local_free(srv);
 	return 0;
 }
 
 
a2acf4e8
 /* parses the naptr record into a naptr_rdata structure
  *   msg   - pointer to the dns message
  *   end   - pointer to the end of the message
  *   rdata - pointer  to the rdata part of the naptr answer
  * returns 0 on error, or a dyn. alloc'ed naptr_rdata structure */
 /* NAPTR rdata format:
  *            111111
  *  0123456789012345
  * +----------------+
  * |      order     |
  * |----------------|
  * |   preference   |
  * |----------------|
  * ~     flags      ~
  * |   (string)     |
  * |----------------|
  * ~    services    ~
  * |   (string)     |
  * |----------------|
  * ~    regexp      ~
  * |   (string)     |
  * |----------------|
  * ~  replacement   ~
    |    (name)      |
  * +----------------+
  */
 struct naptr_rdata* dns_naptr_parser( unsigned char* msg, unsigned char* end,
 								  unsigned char* rdata)
 {
 	struct naptr_rdata* naptr;
 	int len;
 	
 	naptr = 0;
 	if ((rdata + 7) >= end) goto error;
 	naptr=(struct naptr_rdata*)local_malloc(sizeof(struct naptr_rdata));
 	if (naptr == 0){
 		LOG(L_ERR, "ERROR: dns_naptr_parser: out of memory\n");
 		goto error;
 	}
 	
 	memcpy((void*)&naptr->order, rdata, 2);
 	naptr->order=ntohs(naptr->order);
 	memcpy((void*)&naptr->pref, rdata + 2, 2);
 	naptr->pref=ntohs(naptr->pref);
 	naptr->flags_len = (int)rdata[4];
 	if ((rdata + 7 +  naptr->flags_len) >= end) goto error;
 	memcpy((void*)&naptr->flags, rdata + 5, naptr->flags_len);
 	naptr->services_len = (int)rdata[5 + naptr->flags_len];
 	if ((rdata + 7 + naptr->flags_len + naptr->services_len) >= end) goto error;
 	memcpy((void*)&naptr->services, rdata + 6 + naptr->flags_len, naptr->services_len);
 	naptr->regexp_len = (int)rdata[6 + naptr->flags_len + naptr->services_len];
 	if ((rdata + 7 + naptr->flags_len + naptr->services_len +
 					naptr->regexp_len) >= end) goto error;
 	memcpy((void*)&naptr->regexp, rdata + 7 + naptr->flags_len +
 				naptr->services_len, naptr->regexp_len);
 	rdata = rdata + 7 + naptr->flags_len + naptr->services_len + 
 			naptr->regexp_len;
 	if ((len=dn_expand(msg, end, rdata, naptr->repl, MAX_DNS_NAME-1)) == -1)
 		goto error;
 	/* add terminating 0 ? (warning: len=compressed name len) */
 	return naptr;
 error:
 	if (naptr) local_free(naptr);
 	return 0;
 }
 
 
1b71270a
 
4c55b6cf
 /* parses a CNAME record into a cname_rdata structure */
 struct cname_rdata* dns_cname_parser( unsigned char* msg, unsigned char* end,
 									  unsigned char* rdata)
 {
 	struct cname_rdata* cname;
 	int len;
 	
 	cname=0;
 	cname=(struct cname_rdata*)local_malloc(sizeof(struct cname_rdata));
 	if(cname==0){
 		LOG(L_ERR, "ERROR: dns_cname_parser: out of memory\n");
 		goto error;
 	}
 	if ((len=dn_expand(msg, end, rdata, cname->name, MAX_DNS_NAME-1))==-1)
 		goto error;
 	return cname;
 error:
 	if (cname) local_free(cname);
 	return 0;
 }
 
 
 
1b71270a
 /* parses an A record rdata into an a_rdata structure
  * returns 0 on error or a dyn. alloc'ed a_rdata struct
  */
 struct a_rdata* dns_a_parser(unsigned char* rdata, unsigned char* end)
 {
 	struct a_rdata* a;
 	
a2acf4e8
 	if (rdata+4>=end) goto error;
1b71270a
 	a=(struct a_rdata*)local_malloc(sizeof(struct a_rdata));
 	if (a==0){
 		LOG(L_ERR, "ERROR: dns_a_parser: out of memory\n");
 		goto error;
 	}
 	memcpy(a->ip, rdata, 4);
 	return a;
 error:
 	return 0;
 }
 
 
 
 /* parses an AAAA (ipv6) record rdata into an aaaa_rdata structure
  * returns 0 on error or a dyn. alloc'ed aaaa_rdata struct */
 struct aaaa_rdata* dns_aaaa_parser(unsigned char* rdata, unsigned char* end)
 {
 	struct aaaa_rdata* aaaa;
 	
a2acf4e8
 	if (rdata+16>=end) goto error;
1b71270a
 	aaaa=(struct aaaa_rdata*)local_malloc(sizeof(struct aaaa_rdata));
 	if (aaaa==0){
 		LOG(L_ERR, "ERROR: dns_aaaa_parser: out of memory\n");
 		goto error;
 	}
 	memcpy(aaaa->ip6, rdata, 16);
 	return aaaa;
 error:
 	return 0;
 }
 
 
 
 /* frees completely a struct rdata list */
 void free_rdata_list(struct rdata* head)
 {
 	struct rdata* l;
 	for(l=head; l; l=l->next){
 		/* free the parsed rdata*/
 		if (l->rdata) local_free(l->rdata);
 		local_free(l);
 	}
 }
 
 
 
 /* gets the DNS records for name:type
  * returns a dyn. alloc'ed struct rdata linked list with the parsed responses
  * or 0 on error
  * see rfc1035 for the query/response format */
 struct rdata* get_record(char* name, int type)
 {
 	int size;
 	int qno, answers_no;
5bc22c81
 	int r;
1b71270a
 	int ans_len;
 	static union dns_query buff;
 	unsigned char* p;
 	unsigned char* t;
 	unsigned char* end;
 	static unsigned char answer[ANS_SIZE];
 	unsigned short rtype, class, rdlength;
 	unsigned int ttl;
 	struct rdata* head;
 	struct rdata** crt;
e90526cd
 	struct rdata** last;
1b71270a
 	struct rdata* rd;
 	struct srv_rdata* srv_rd;
 	struct srv_rdata* crt_srv;
 	
 	size=res_search(name, C_IN, type, buff.buff, sizeof(buff));
072cead0
 	if (size<0) {
e90526cd
 		DBG("get_record: lookup(%s, %d) failed\n", name, type);
 		goto not_found;
072cead0
 	}
1b71270a
 	else if (size > sizeof(buff)) size=sizeof(buff);
e90526cd
 	head=rd=0;
 	last=crt=&head;
1b71270a
 	
 	p=buff.buff+DNS_HDR_SIZE;
 	end=buff.buff+size;
a2acf4e8
 	if (p>=end) goto error_boundary;
1b71270a
 	qno=ntohs((unsigned short)buff.hdr.qdcount);
 
 	for (r=0; r<qno; r++){
 		/* skip the name of the question */
072cead0
 		if ((p=dns_skipname(p, end))==0) {
 			LOG(L_ERR, "ERROR: get_record: skipname==0\n");
 			goto error;
 		}
1b71270a
 		p+=2+2; /* skip QCODE & QCLASS */
 	#if 0
 		for (;(p<end && (*p)); p++);
 		p+=1+2+2; /* skip the ending  '\0, QCODE and QCLASS */
 	#endif
a2acf4e8
 		if (p>=end) {
 			LOG(L_ERR, "ERROR: get_record: p>=end\n");
072cead0
 			goto error;
 		}
1b71270a
 	};
 	answers_no=ntohs((unsigned short)buff.hdr.ancount);
 	ans_len=ANS_SIZE;
 	t=answer;
 	for (r=0; (r<answers_no) && (p<end); r++){
 		/*  ignore it the default domain name */
072cead0
 		if ((p=dns_skipname(p, end))==0) {
 			LOG(L_ERR, "ERROR: get_record: skip_name=0 (#2)\n");
 			goto error;
 		}
1b71270a
 		/*
 		skip=dn_expand(buff.buff, end, p, t, ans_len);
 		p+=skip;
 		*/
a2acf4e8
 		/* check if enough space is left for type, class, ttl & size */
 		if ((p+2+2+4+2)>=end) goto error_boundary;
1b71270a
 		/* get type */
 		memcpy((void*) &rtype, (void*)p, 2);
 		rtype=ntohs(rtype);
 		p+=2;
 		/* get  class */
 		memcpy((void*) &class, (void*)p, 2);
 		class=ntohs(class);
 		p+=2;
 		/* get ttl*/
 		memcpy((void*) &ttl, (void*)p, 4);
 		ttl=ntohl(ttl);
 		p+=4;
 		/* get size */
 		memcpy((void*)&rdlength, (void*)p, 2);
 		rdlength=ntohs(rdlength);
 		p+=2;
 		/* check for type */
4c55b6cf
 		/*
1b71270a
 		if (rtype!=type){
5415e40b
 			LOG(L_ERR, "WARNING: get_record: wrong type in answer (%d!=%d)\n",
 					rtype, type);
1b71270a
 			p+=rdlength;
 			continue;
 		}
 		*/
4c55b6cf
 		/* expand the "type" record  (rdata)*/
 		
1b71270a
 		rd=(struct rdata*) local_malloc(sizeof(struct rdata));
 		if (rd==0){
 			LOG(L_ERR, "ERROR: get_record: out of memory\n");
 			goto error;
 		}
 		rd->type=rtype;
 		rd->class=class;
 		rd->ttl=ttl;
4c55b6cf
 		rd->next=0;
 		switch(rtype){
1b71270a
 			case T_SRV:
 				srv_rd= dns_srv_parser(buff.buff, end, p);
 				rd->rdata=(void*)srv_rd;
 				if (srv_rd==0) goto error_parse;
 				
e90526cd
 				/* insert sorted into the list */
1b71270a
 				for (crt=&head; *crt; crt= &((*crt)->next)){
 					crt_srv=(struct srv_rdata*)(*crt)->rdata;
 					if ((srv_rd->priority <  crt_srv->priority) ||
 					   ( (srv_rd->priority == crt_srv->priority) && 
 							 (srv_rd->weight > crt_srv->weight) ) ){
 						/* insert here */
e90526cd
 						goto skip;
1b71270a
 					}
 				}
e90526cd
 				last=&(rd->next); /*end of for => this will be the last elem*/
 			skip:
1b71270a
 				/* insert here */
 				rd->next=*crt;
 				*crt=rd;
 				
 				break;
 			case T_A:
 				rd->rdata=(void*) dns_a_parser(p,end);
 				if (rd->rdata==0) goto error_parse;
e90526cd
 				*last=rd; /* last points to the last "next" or the list head*/
 				last=&(rd->next);
1b71270a
 				break;
 			case T_AAAA:
 				rd->rdata=(void*) dns_aaaa_parser(p,end);
 				if (rd->rdata==0) goto error_parse;
e90526cd
 				*last=rd;
 				last=&(rd->next);
1b71270a
 				break;
4c55b6cf
 			case T_CNAME:
 				rd->rdata=(void*) dns_cname_parser(buff.buff, end, p);
 				if(rd->rdata==0) goto error_parse;
e90526cd
 				*last=rd;
 				last=&(rd->next);
4c55b6cf
 				break;
a2acf4e8
 			case T_NAPTR:
 				rd->rdata=(void*) dns_naptr_parser(buff.buff, end, p);
 				if(rd->rdata==0) goto error_parse;
 				*last=rd;
 				last=&(rd->next);
 				break;
1b71270a
 			default:
4c55b6cf
 				LOG(L_ERR, "WARNING: get_record: unknown type %d\n", rtype);
1b71270a
 				rd->rdata=0;
e90526cd
 				*last=rd;
 				last=&(rd->next);
1b71270a
 		}
 		
 		p+=rdlength;
 		
 	}
 	return head;
 error_boundary:
 		LOG(L_ERR, "ERROR: get_record: end of query buff reached\n");
 		return 0;
 error_parse:
 		LOG(L_ERR, "ERROR: get_record: rdata parse error \n");
 		if (rd) local_free(rd); /* rd->rdata=0 & rd is not linked yet into
 								   the list */
 error:
 		LOG(L_ERR, "ERROR: get_record \n");
 		if (head) free_rdata_list(head);
e90526cd
 not_found:
1b71270a
 	return 0;
 }
 
5bc22c81
 
 
 /* resolves a host name trying SRV lookup if *port==0 or normal A/AAAA lookup
  * if *port!=0.
d531a5d5
  * when performing SRV lookup (*port==0) it will use proto to look for
  * tcp or udp hosts, otherwise proto is unused; if proto==0 => no SRV lookup
5bc22c81
  * returns: hostent struct & *port filled with the port from the SRV record;
  *  0 on error
  */
d531a5d5
 struct hostent* sip_resolvehost(str* name, unsigned short* port, int proto)
5bc22c81
 {
 	struct hostent* he;
 	struct rdata* head;
 	struct rdata* l;
 	struct srv_rdata* srv;
4c9f3fde
 	struct ip_addr* ip;
5bc22c81
 	static char tmp[MAX_DNS_NAME]; /* tmp. buff. for SRV lookups */
 
 	/* try SRV if no port specified (draft-ietf-sip-srv-06) */
 	if ((port)&&(*port==0)){
3f601bd0
 		*port=(proto==PROTO_TLS)?SIPS_PORT:SIP_PORT; /* just in case we don't
 														find another */
5b9386d1
 		if ((name->len+SRV_MAX_PREFIX_LEN+1)>MAX_DNS_NAME){
5bc22c81
 			LOG(L_WARN, "WARNING: sip_resolvehost: domain name too long (%d),"
a6982b85
 						" unable to perform SRV lookup\n", name->len);
5bc22c81
 		}else{
4c9f3fde
 			/* check if it's an ip address */
a6982b85
 			if ( ((ip=str2ip(name))!=0)
4c9f3fde
 #ifdef	USE_IPV6
a6982b85
 				  || ((ip=str2ip6(name))!=0)
4c9f3fde
 #endif
 				){
 				/* we are lucky, this is an ip address */
a6982b85
 				return ip_addr2he(name,ip);
4c9f3fde
 			}
 			
d531a5d5
 			switch(proto){
 				case PROTO_NONE: /* no proto specified, use udp */
 					goto skip_srv;
 				case PROTO_UDP:
5b9386d1
 					memcpy(tmp, SRV_UDP_PREFIX, SRV_UDP_PREFIX_LEN);
 					memcpy(tmp+SRV_UDP_PREFIX_LEN, name->s, name->len);
 					tmp[SRV_UDP_PREFIX_LEN + name->len] = '\0';
d531a5d5
 					break;
 				case PROTO_TCP:
5b9386d1
 					memcpy(tmp, SRV_TCP_PREFIX, SRV_TCP_PREFIX_LEN);
 					memcpy(tmp+SRV_TCP_PREFIX_LEN, name->s, name->len);
 					tmp[SRV_TCP_PREFIX_LEN + name->len] = '\0';
8e807134
 					break;
 				case PROTO_TLS:
5b9386d1
 					memcpy(tmp, SRV_TLS_PREFIX, SRV_TLS_PREFIX_LEN);
 					memcpy(tmp+SRV_TLS_PREFIX_LEN, name->s, name->len);
 					tmp[SRV_TLS_PREFIX_LEN + name->len] = '\0';
d531a5d5
 					break;
 				default:
 					LOG(L_CRIT, "BUG: sip_resolvehost: unknown proto %d\n",
 							proto);
 					return 0;
 			}
a6982b85
 
5bc22c81
 			head=get_record(tmp, T_SRV);
 			for(l=head; l; l=l->next){
 				if (l->type!=T_SRV) continue; /*should never happen*/
 				srv=(struct srv_rdata*) l->rdata;
 				if (srv==0){
 					LOG(L_CRIT, "sip_resolvehost: BUG: null rdata\n");
 					free_rdata_list(head);
 					break;
 				}
 				he=resolvehost(srv->name);
 				if (he!=0){
 					/* we found it*/
 					DBG("sip_resolvehost: SRV(%s) = %s:%d\n",
 							tmp, srv->name, srv->port);
 					*port=srv->port;
 					free_rdata_list(head); /*clean up*/
 					return he;
 				}
 			}
6d44652c
 			if (head) free_rdata_list(head); /*clean up*/
75e2f1a6
 			DBG("sip_resolvehost: no SRV record found for %.*s," 
a6982b85
 					" trying 'normal' lookup...\n", name->len, name->s);
5bc22c81
 		}
 	}
d531a5d5
 skip_srv:
a6982b85
 	if (name->len >= MAX_DNS_NAME) {
 		LOG(L_ERR, "sip_resolvehost: domain name too long\n");
 		return 0;
 	}
 	memcpy(tmp, name->s, name->len);
 	tmp[name->len] = '\0';
 	he=resolvehost(tmp);
5bc22c81
 	return he;
 }