#include "includes.h"

#include "ntkresolv.h"
#include "andns_net.h"
#include "snsd_cache.h"
#include "crypto.h"
#include "common.h"

static ntkresolv_opts globopts;
static struct timeval time_start,time_stop;
uint8_t mode_compute_hash=0;
uint8_t mode_parsable_output=0;

void version(void)
{
        say("ntk-resolv version %s (Netsukuku tools)\n\n"
            "Copyright (C) 2006.\n"
            "This is free software.  You may redistribute copies of it under the terms of\n"
            "the GNU General Public License <http://www.gnu.org/licenses/gpl.html>.\n"
            "There is NO WARRANTY, to the extent permitted by law.\n\n"
	    "Report bugs and ideas to <%s>.\n",VERSION,NTK_RESOLV_MAIL_BUGS);
	ntkresolv_safe_exit(1);
}


void usage(void)
{
        say("Usage:\n"
                "\tntk-resolv [OPTIONS] host\n"
                "\tntk-resolv -H host\n\n"
                " -v --version          print version, then exit.\n"
                " -n --nameserver=ns    use nameserver `ns' instead of localhost.\n"
                " -P --port=port        nameserver port, default 53.\n"
                " -t --query-type=qt    query type (`-t help' shows more info).\n"
                " -r --realm=realm      realm to scan (`-r help' shows more info).\n"
                " -s --service=service  SNSD service (`-s help' shows more info).\n"
                " -p --protocolo=proto  SNSD protocol (`-p help' shows more info).\n"
                " -S --silent           ntk-resolv will be not loquacious.\n"
                " -b --block-recursion  set recursion OFF.\n"
                " -m --md5-hash         hostname specified is hash-ed.\n"
                " -H --compute-hash     print the hash'ed hostname.\n"
                " -l --parsable-output  print answers in a synthetic way.\n"
                " -h --help             display this help, then exit.\n\n"
		"Report bugs and ideas to <%s>.\n",NTK_RESOLV_MAIL_BUGS);
	ntkresolv_safe_exit(1);
}
void qt_usage(char *arg)
{
	if (arg)
		say("Bad Query Type %s\n\n",arg);
	else
		say("ntk-resolv Query Type Help.\n\n");
	say(
	    "Valid query types are:\n"
            " * snsd\t\thost:port -> ip\n"
            "   ptr\t\tip -> host\n"
            "   global\thostname -> all services ip\n"
            "   mx\t\thostname MX -> ip\n\n"
            "(you can also use univoque abbreviation)\n"
	    "Note: mx query is equivalent to --query-type="
	    "snsd AND --service=25\n\n");
	ntkresolv_safe_exit(1);
}
void realm_usage(char *arg)
{
	if (arg)
		say("Bad Realm %s\n\n",arg);
	else
		say("ntk-resolv Realm Help.\n\n");
	say(
	    "Valid realms are:\n"
            " * ntk\tnetsukuku realm\n"
            "   inet\tinternet realm\n\n"
            "(you can also use univoque abbreviation)\n\n");
	ntkresolv_safe_exit(1);
}
void proto_usage(char *arg)
{
	if (arg)
		say("Bad Protocol %s\n\n",arg);
	else
		say("ntk-resolv Protocol Help.\n\n");
	say(
	    "Valid protocols are:\n"
            " * tcp\n"
            "   udp\n"
            "(you can also use univoque abbreviation)\n"
	    "Note: you can also specify the protocol with option `-s'.\n" 
	    "To know more, type:\n"
	    "\tntk-resolv -s help\n\n");
	ntkresolv_safe_exit(1);
}
void service_and_proto_usage(char *arg)
{
	if (arg)
		say("Bad service/proto %s\n\n"
			"Use `ntk-resolv -s help` for more info on"
			" service and proto.\n"	,arg);
	else say(
		"ntk-resolv Service and Proto Help.\n\n"
		"The form to specify a service and a protocol are:\n"
		"  ntk-resolv -s service/proto\n"
		"  ntk-resolv -s service -p proto\n\n"
		"Valid protocols are:\n"
	        " * tcp\n"
            	"   udp\n\n"
		"Valid services are expressed in /etc/services.\n"
		"You can use numeric form too.\n\n"
		"As example, the next commands are equivalent and\n"
		"will return the IP of the hostname that offers\n"
		"webpages for the hostname \"some_hostname\":\n\n"
		"  ntk-resolv -s http -p tcp some_hostname\n"
		"  ntk-resolv -s http/tcp    some_hostname\n"
		"  ntk-resolv -s 80/tcp      some_hostname\n"
		"  ntk-resolv -s 80          some_hostname\n\n");
	ntkresolv_safe_exit(1);
}
		

double diff_time(struct timeval a,struct timeval b)
{
        double res;
        res=(double)(b.tv_sec-a.tv_sec);
	if (res<0.9 || b.tv_usec>=a.tv_usec)
		res+=(b.tv_usec-a.tv_usec)/TIME_SCALE;
	else {
		res-=1.0;
		res+=(TIME_SCALE+b.tv_usec-a.tv_usec)/TIME_SCALE;
	}
        return res;
}


void opts_init(void)
{
	memset(&GOP,0,NTKRESOLV_OPTS_SZ);
	strcpy(GOP.nsserver,LOCALHOST);
	GOP.port=NTKRESOLV_PORT;
	GQT=create_andns_pkt();
	GQT->nk=REALM_NTK;
	GQT->p=SNSD_PROTO_DEFAULT;
	GQT->service=SNSD_SERVICE_DEFAULT;
	GQT->r=1;
	xsrand();
}

void opts_set_silent(void)
{
	GOP.silent=1;
}

void opts_set_port(char *arg)
{
	int res;
	uint16_t port;

	res=atoi(arg);
	port=(uint16_t)res;

	if (port!=res) {
		say("Bad port %s.",arg);
		ntkresolv_safe_exit(1);
	}
	GOP.port=port;
}

void opts_set_ns(char *arg)
{
	int slen;

	slen=strlen(arg);
	if (slen>=MAX_HOSTNAME_LEN) {
		say("Server hostname too long.");
		ntkresolv_safe_exit(1);
	}
	strcpy(GOP.nsserver,arg);
	GOP.nsserver[slen]=0;
}

void opts_set_qt(char *arg)
{
	int res;

	if (!strcmp(arg,HELP_STR))
		qt_usage(NULL);
	res=QTFROMPREF(arg);
	if (res==-1) 
		qt_usage(arg);
	if (res==QTYPE_MX) {
		GQT->qtype=QTYPE_A;
		GQT->service=25;
		GQT->p=SNSD_PROTO_TCP;
	} else
		GQT->qtype=res;
}

void opts_set_realm(char *arg)
{
	uint8_t res;

	if (!strcmp(arg,HELP_STR))
		realm_usage(NULL);
	res=REALMFROMPREF(arg);
	if (!res) 
		realm_usage(arg);
	GQT->nk=res;
}

void opts_set_service_and_proto(char *arg)
{
	int ret;

	if (!strcmp(arg,HELP_STR))
		service_and_proto_usage(NULL);
	ret=str_to_snsd_service(arg, (int *)&GQT->service, &GQT->p);
/*	if(ret == -1)
		say("Bad service %s.",arg);
	else if(ret == -2)
		proto_usage(arg);*/
	if(ret < 0)
		service_and_proto_usage(arg);
	GQT->p-=1;
}
void opts_set_proto(char *arg) 
{
	int ret;

	if (!strcmp(arg,HELP_STR))
		proto_usage(NULL);
	ret=PROTOFROMPREF(arg);
	if (ret<0)
		proto_usage(arg);
	GQT->p=ret;
}
/* This is a complex set of function. */
void opts_set_recursion(void)
{
	GQT->r=0;
}
void opts_set_hash(void) 
{
	GOP.hash=1;
}
void opts_set_compute_hash(void)
{
	mode_compute_hash=1;
}
void opts_set_parsable_output(void)
{
	mode_parsable_output=1;
	AMISILENT=1;
}
void opts_set_question(char *arg)
{
	struct in_addr ia;
	struct in6_addr i6a;
	int res;
	
	strcpy(GOP.obj,arg);
	res=strlen(arg);
	
	switch(GQT->qtype) {
		case QTYPE_A:
			if (GQT->nk==REALM_NTK) {
				G_ALIGN(ANDNS_HASH_H);
				if (GOP.hash) {
					if (res!=2*ANDNS_HASH_H) {
						say("Malformed Hostname hash `%s'.\n",arg);
						ntkresolv_safe_exit(1);
					}
					NTK_RESOLV_STR_HASH(arg,GQT->qstdata);
				}
				else
					hash_md5((unsigned char*)arg,res,
						(unsigned char*)GQT->qstdata);
			} else {
				if (res>255) {
					say("Hostname %s is too long for DNS standard.",arg);
					ntkresolv_safe_exit(1);
				}
				G_ALIGN(res);
				strcpy(GQT->qstdata,arg);
			}
			return;
		case QTYPE_PTR:
			res=inet_pton(AF_INET,arg,&ia);
			if (res) {
				G_ALIGN(ANDNS_HASH_H);
				memcpy(GQT->qstdata,&ia.s_addr,4);
				return;
			}
			res=inet_pton(AF_INET6,arg,&i6a);
			if (!res) {
				say("Bad address `%s'\n",arg);
				ntkresolv_safe_exit(1);
			}
			G_ALIGN(16);
			memcpy(GQT->qstdata,&i6a.in6_u,16);
			GQT->ipv=ANDNS_IPV6;
			return;
		case QTYPE_G:
			if (GQT->nk!=REALM_NTK) {
				say("Global query type is valid only for the Ntk realm.");
				ntkresolv_safe_exit(1);
			}
			G_ALIGN(ANDNS_HASH_H);
			if (GOP.hash)
				NTK_RESOLV_STR_HASH(arg,GQT->qstdata);
			else
				hash_md5((unsigned char*)arg,res,
					(unsigned char*)GQT->qstdata);
			return;	
		default:
			say("Unknow Query Type.\n");
			return;
	}
}
void opts_finish(char *arg)
{
	int r;

	r=strlen(arg);
	if (r>NTKRESOLV_MAX_OBJ_LEN) {
		say("Object requested is too long: %s",arg);
		ntkresolv_safe_exit(1);
	}

	if (mode_compute_hash) { /* Do command here and exit */
		G_ALIGN(ANDNS_HASH_H);
		hash_md5((unsigned char*)arg,r,
			(unsigned char*)GQT->qstdata);
		NTK_RESOLV_HASH_STR(GQT->qstdata,GOP.obj);
		say("%s\n",GOP.obj);
		ntkresolv_safe_exit(0);
	}
	if (GOP.hash && GQT->qtype==AT_PTR) {
		say("Option `-m' is not usable with inverse queries.\n");
		ntkresolv_safe_exit(1);
	}
	r=rand();
	GQT->id=r>>16;
	opts_set_question(arg);
}

void print_headers()
{
	andns_pkt *ap=GQT;
	say("\n - Headers Section:\n"
		"\tid ~ %6d\tqr  ~ %4d\tqtype ~ %7s\n"
		"\tan ~ %6d\tipv ~ %s\trealm ~ %7s\n"
		"\tsv ~ %6s\tprt ~ %4s\trCode ~ %s\n"
		"\trc ~ %6d\n",
		ap->id,ap->qr,QTYPE_STR(ap),
		ap->ancount,IPV_STR(ap),NK_STR(ap),
		SERVICE_STR(ap),PROTO_STR(ap),
		RCODE_STR(ap),ap->r);
}
void print_question()
{
	say("\n - Question Section:\n"
		"\tObj ~ %s\n",GOP.obj);
}

void ip_bin_to_str(void *data,char *dst)
{
	int family;
	struct in_addr ia;
	struct in6_addr i6a;
	const void *via;
	const char *crow;

	family=GQT->ipv==ANDNS_IPV4?
		AF_INET:AF_INET6;
	switch(family) {
		case AF_INET:
			memcpy(&(ia.s_addr),data,4);
			via=(void*)(&ia);
			break;
		case AF_INET6:
			memcpy(&(i6a.in6_u),data,16);
			via=(void*)(&i6a);
			break;
		default:
			strcpy(dst,"Unprintable Object");
			return;
	}
	crow=inet_ntop(family,via,dst,NTKRESOLV_MAX_OBJ_LEN);
	if (!crow) 
		strcpy(dst,"Unprintable Object");
}

void answer_data_to_str(andns_pkt_data *apd,char *dst)
{
	if (GQT->qtype==AT_PTR)
			strcpy(dst,apd->rdata);
	else if (GQT->qtype==AT_G || GQT->qtype==AT_A) {
			if (apd->m&APD_IP)
				ip_bin_to_str(apd->rdata,dst);
			else 
				NTK_RESOLV_HASH_STR(apd->rdata,dst);
	} 
	else
		strcpy(dst,"Unprintable Object");
}
void print_answers()
{
	int i=0;
	int ancount=GQT->ancount;
	andns_pkt_data *apd;

	if (!ancount)
		return;

	say("\n - Answers Section:\n");

	apd=GQT->pkt_answ;
	while (apd) {
		i++;
		if (i>ancount) 
			say("Answer not declared in Headers Packet.\n");
		answer_data_to_str(apd,GOP.obj);
		say("\t ~ %s",GOP.obj);
		if (apd->m&APD_MAIN_IP)
			say(" *");
		else if (GQT->qtype!=AT_PTR && !(apd->m&APD_IP) && GQT->r)
			say("\t + Recursion Failed");
		say("\n");
		if (GQT->qtype==AT_A || GQT->qtype==AT_G) 
			say("\t\tPrio ~ %d  Weigth ~ %d\n",
				apd->prio,apd->wg);
		if (GQT->qtype==AT_G)
			say("\t\tService ~ %d  Proto ~ %s\n",
				apd->service,apd->m&APD_UDP?
				"udp":"tcp");
		say("\n");
		apd=apd->next;
	}
}

void print_parsable_answers(void)
{
	int i=0;
	int ancount=GQT->ancount;
	andns_pkt_data *apd;

	if (!ancount)
		return;

	apd=GQT->pkt_answ;
	while(apd) {
		i++;
		if (i>ancount) 
			say("Answer not declared in Headers Packet.\n");
		answer_data_to_str(apd,GOP.obj);
		if (GQT->qtype==AT_PTR || 
		   (GQT->qtype==AT_A && !GQT->service)) 
			say("%s %s\n",NTK_RESOLV_SYMBOL(apd),GOP.obj);
		else if (GQT->qtype==AT_A) 
			say("%s %s %d %d\n",NTK_RESOLV_SYMBOL(apd),
				GOP.obj,apd->prio,apd->wg);
		else 
			say("%s %s %d %s %d %d\n",NTK_RESOLV_SYMBOL(apd),
				GOP.obj,apd->service,
				apd->m&APD_UDP?"udp":"tcp",
				apd->prio,apd->wg);
		apd=apd->next;
	}
}

void print_results(void) 
{
	if (!AMISILENT) {
		print_headers();
		print_question();
	}
	if (mode_parsable_output)
		print_parsable_answers();
	else
		print_answers();
}

void do_command(void)
{
	char buf[ANDNS_MAX_SZ];
	char answer[ANDNS_MAX_PK_LEN];
	int res;

	memset(buf,0,ANDNS_MAX_SZ);
	GOP.id=GQT->id;
	res=a_p(GQT,buf);
	if (res==-1) {
		say("Error building question.\n");
		ntkresolv_exit(1);
	}
	res=hn_send_recv_close(GOP.nsserver,GOP.port,
			SOCK_DGRAM,buf,res,answer,
			ANDNS_MAX_PK_LEN,0,NTK_RESOLV_TIMEOUT);
	if (res==-1) {
		say("Communication failed with %s.\n",GOP.nsserver);
		ntkresolv_exit(1);
	}
	if (res==-2) {
		say("Unable to send() to %s.\n",GOP.nsserver);
		ntkresolv_exit(1);
	}
	if (res==-3) {
		say("Unable to recv() from %s.\n",GOP.nsserver);
		ntkresolv_exit(1);
	}
	res=a_u(answer,res,&GQT);
	if (res<=0) {
		say("Error interpreting server answer.\n");
		ntkresolv_exit(1);
	}
	if (GQT->id!=GOP.id) 
		say("Warning: ID query (%d) is mismatching ID answer (%d)!\n",GOP.id,GQT->id);

	print_results();
	destroy_andns_pkt(GQT);
}
void ntkresolv_exit(int i)
{
	exit(i);
}
void ntkresolv_safe_exit(int i)
{
	destroy_andns_pkt(GQT);
	ntkresolv_exit(i);
}
int main(int argc, char **argv)
{
        int c;
        extern int optind, opterr, optopt;
        extern char *optarg;

	log_init("",0,1);
	gettimeofday(&time_start,NULL);

	opts_init();
	struct option longopts[]= {
                {"version",0,0,'v'},
                {"nameserver",1,0,'n'},
                {"port",1,0,'P'},
                {"query-type",1,0,'t'},
                {"realm",1,0,'r'},
                {"service",1,0,'s'},
                {"proto",1,0,'p'},
                {"silent",0,0,'S'},
                {"block-recursion",0,0,'b'},
                {"md5-hash",0,0,'m'},
                {"compute-hash",0,0,'H'},
                {"parsable-output",0,0,'l'},
                {"help",0,0,'h'},
                {0,0,0,0}
        };

	while(1) {
		int oindex=0;
		c=getopt_long(argc, argv, 
			"vn:P:t:r:s:p:ShbmHl", longopts, &oindex);
		if (c==-1)
			break;
		switch(c) {
			case 'v':
				version();
			case 'n':
				opts_set_ns(optarg);
				break;
			case 'P':
				opts_set_port(optarg);
				break;
			case 't':
				opts_set_qt(optarg);
				break;
			case 'r':
				opts_set_realm(optarg);
				break;
			case 's':
				opts_set_service_and_proto(optarg);
				break;	
			case 'p':
				opts_set_proto(optarg);
				break;	
			case 'h':
				usage();
			case 'S':
				opts_set_silent();
				break;
			case 'b':
				opts_set_recursion();
				break;
			case 'm':
				opts_set_hash();
				break;
			case 'H':
				opts_set_compute_hash();
				break;
			case 'l':
				opts_set_parsable_output();
				break;
			default:
				usage();
		}
	}
	if (optind==argc)
		usage();
	opts_finish(argv[optind]);
	do_command();
	time_report;
	bye;
	return 0;
}