/*
  * $Id$
  *
  * proxy list & assoc. functions
  *

  *
  * 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

  */

  /*
   * History:
   * -------

   *  2003-02-13  all *proxy functions are now proto aware (andrei)

   *  2003-03-19  replaced all mallocs/frees w/ pkg_malloc/pkg_free (andrei)

   */

 #include "config.h"

 #include "proxy.h"
 #include "error.h"

 #include "dprint.h"

 #include "mem/mem.h"

 
 #include <string.h>

 #include <stdlib.h>

 #include <sys/socket.h>

 #ifdef DNS_IP_HACK
 #include "ut.h"
 #endif
 

 #include "resolve.h"
 #include "ip_addr.h"

 #include "globals.h"

 
 struct proxy_l* proxies=0;
 
 
 

 /* searches for the proxy named 'name', on port 'port' with 
    proto 'proto'; if proto==0 => proto wildcard (will match any proto)

    returns: pointer to proxy_l on success or 0 if not found */ 

 static struct proxy_l* find_proxy(str *name, unsigned short port, int proto)

 {
 	struct proxy_l* t;
 	for(t=proxies; t; t=t->next)

 		if (((t->name.len == name->len) &&
 			 ((proto==PROTO_NONE)||(t->proto==proto))&&
 			(strncasecmp(t->name.s, name->s, name->len)==0)) &&
 				(t->port==port))

 			break;
 	return t;
 }
 
 
 
 /* copies a hostent structure*, returns 0 on success, <0 on error*/

 static int hostent_cpy(struct hostent *dst, struct hostent* src)

 {

 	unsigned len,len2;
 	int r,ret,i;

 
 	/* start copying the host entry.. */
 	/* copy h_name */
 	len=strlen(src->h_name)+1;

 	dst->h_name=(char*)pkg_malloc(sizeof(char) * len);

 	if (dst->h_name) strncpy(dst->h_name,src->h_name, len);
 	else{

 		ser_error=ret=E_OUT_OF_MEM;

 		goto error;
 	}
 
 	/* copy h_aliases */

 	len=0;
 	if (src->h_aliases)
 		for (;src->h_aliases[len];len++);

 	dst->h_aliases=(char**)pkg_malloc(sizeof(char*)*(len+1));

 	if (dst->h_aliases==0){

 		ser_error=ret=E_OUT_OF_MEM;

 		pkg_free(dst->h_name);

 		goto error;
 	}
 	memset((void*)dst->h_aliases, 0, sizeof(char*) * (len+1) );
 	for (i=0;i<len;i++){
 		len2=strlen(src->h_aliases[i])+1;

 		dst->h_aliases[i]=(char*)pkg_malloc(sizeof(char)*len2);

 		if (dst->h_aliases==0){

 			ser_error=ret=E_OUT_OF_MEM;

 			pkg_free(dst->h_name);
 			for(r=0; r<i; r++)	pkg_free(dst->h_aliases[r]);
 			pkg_free(dst->h_aliases);

 			goto error;
 		}
 		strncpy(dst->h_aliases[i], src->h_aliases[i], len2);
 	}
 	/* copy h_addr_list */

 	len=0;
 	if (src->h_addr_list)
 		for (;src->h_addr_list[len];len++);

 	dst->h_addr_list=(char**)pkg_malloc(sizeof(char*)*(len+1));

 	if (dst->h_addr_list==0){

 		ser_error=ret=E_OUT_OF_MEM;

 		pkg_free(dst->h_name);
 		for(r=0; dst->h_aliases[r]; r++)	pkg_free(dst->h_aliases[r]);
 		pkg_free(dst->h_aliases[r]);
 		pkg_free(dst->h_aliases);

 		goto error;
 	}

 	memset((void*)dst->h_addr_list, 0, sizeof(char*) * (len+1) );

 	for (i=0;i<len;i++){

 		dst->h_addr_list[i]=(char*)pkg_malloc(sizeof(char)*src->h_length);

 		if (dst->h_addr_list[i]==0){

 			ser_error=ret=E_OUT_OF_MEM;

 			pkg_free(dst->h_name);
 			for(r=0; dst->h_aliases[r]; r++)	pkg_free(dst->h_aliases[r]);
 			pkg_free(dst->h_aliases[r]);
 			pkg_free(dst->h_aliases);
 			for (r=0; r<i;r++) pkg_free(dst->h_addr_list[r]);
 			pkg_free(dst->h_addr_list);

 			goto error;
 		}
 		memcpy(dst->h_addr_list[i], src->h_addr_list[i], src->h_length);
 	}
 
 	/* copy h_addr_type & length */
 	dst->h_addrtype=src->h_addrtype;

 	dst->h_length=src->h_length;

 	/*finished hostent copy */
 	
 	return 0;
 
 error:
 	LOG(L_CRIT, "ERROR: hostent_cpy: memory allocation failure\n");
 	return ret;
 }
 
 
 

 void free_hostent(struct hostent *dst)
 {
 	int r;

 	if (dst->h_name) pkg_free(dst->h_name);

 	if (dst->h_aliases){

 		for(r=0; dst->h_aliases[r]; r++) {
 			pkg_free(dst->h_aliases[r]);
 		}

 		pkg_free(dst->h_aliases);

 	}
 	if (dst->h_addr_list){

 		for (r=0; dst->h_addr_list[r];r++) { 
 			pkg_free(dst->h_addr_list[r]);
 		}

 		pkg_free(dst->h_addr_list);

 	}
 }
 
 
 
 

 struct proxy_l* add_proxy(str* name, unsigned short port, int proto)

 {

 	struct proxy_l* p;

 	if ((p=find_proxy(name, port, proto))!=0) return p;
 	if ((p=mk_proxy(name, port, proto))==0) goto error;

 	/* add p to the proxy list */
 	p->next=proxies;
 	proxies=p;
 	return p;
 
 error:
 	return 0;
 }
 
 
 

 /* same as add_proxy, but it doesn't add the proxy to the list

  * uses also SRV if possible & port==0 (quick hack) */

 struct proxy_l* mk_proxy(str* name, unsigned short port, int proto)

 {
 	struct proxy_l* p;
 	struct hostent* he;

 	p=(struct proxy_l*) pkg_malloc(sizeof(struct proxy_l));

 	if (p==0){

 		ser_error=E_OUT_OF_MEM;

 		LOG(L_CRIT, "ERROR: mk_proxy: memory allocation failure\n");

 		goto error;
 	}

 	memset(p,0,sizeof(struct proxy_l));

 	p->name=*name;

 	p->port=port;

 	p->proto=proto;

 	DBG("DEBUG: mk_proxy: doing DNS lookup...\n");

 	he=sip_resolvehost(name, &(p->port), proto);

 	if (he==0){

 		ser_error=E_BAD_ADDRESS;

 		LOG(L_CRIT, "ERROR: mk_proxy: could not resolve hostname:"

 					" \"%.*s\"\n", name->len, name->s);

 		pkg_free(p);

 		goto error;
 	}
 	if (hostent_cpy(&(p->host), he)!=0){

 		pkg_free(p);

 		goto error;
 	}
 	p->ok=1;
 	return p;
 error:
 	return 0;
 }
 

 /* same as mk_proxy, but get the host as an ip*/

 struct proxy_l* mk_proxy_from_ip(struct ip_addr* ip, unsigned short port,
 									int proto)

 {
 	struct proxy_l* p;
 

 	p=(struct proxy_l*) pkg_malloc(sizeof(struct proxy_l));

 	if (p==0){
 		LOG(L_CRIT, "ERROR: mk_proxy_from_ip: memory allocation failure\n");
 		goto error;
 	}
 	memset(p,0,sizeof(struct proxy_l));
 
 	p->port=port;

 	p->proto=proto;

 	p->host.h_addrtype=ip->af;
 	p->host.h_length=ip->len;

 	p->host.h_addr_list=pkg_malloc(2*sizeof(char*));

 	if (p->host.h_addr_list==0) goto error;
 	p->host.h_addr_list[1]=0;

 	p->host.h_addr_list[0]=pkg_malloc(ip->len+1);

 	if (p->host.h_addr_list[0]==0){

 		pkg_free(p->host.h_addr_list);

 		goto error;
 	}
 

 	memcpy(p->host.h_addr_list[0], ip->u.addr, ip->len);
 	p->host.h_addr_list[0][ip->len]=0;

 
 	return p;
 
 error:
 	return 0;
 }
 
 
 
 

 void free_proxy(struct proxy_l* p)
 {
 	if (p) free_hostent(&p->host);
 }