/* This file is part of Netsukuku
 * (c) Copyright 2005 Andrea Lo Pumo aka AlpT <alpt@freaknet.org>
 *
 * This source code 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.
 *
 * This source code 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.
 * Please refer to the GNU Public License for more details.
 *
 * You should have received a copy of the GNU Public License along with
 * this source code; if not, write to:
 * Free Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include "includes.h"
#include "request.h"
#include "xmalloc.h"
#include "log.h"

const static u_char request_str[][30]=
{ 
	{ "ECHO_ME" },
	{ "ECHO_REPLY" },
	{ "GET_FREE_NODES" },
	{ "GET_QSPN_ROUND" },

	{ "GET_INTERNET_GWS" },
	{ "SET_FOREIGN_ROUTE" },
	{ "DEL_FOREIGN_ROUTE"},
	{ "NEW_BACKROUTE"},
	{ "DELAYED_BROADCAST" },
	{ "SPLIT_ROUTE" },
	{ "SET_NO_IDENTITY" },
	{ "QSPN_CLOSE"},
	{ "QSPN_OPEN"},
	{ "QSPN_RFR"},
	{ "GET_DNODE_BLOCK" },
	{ "GET_DNODE_IP"},
	{ "TRACER_PKT" },
	{ "TRACER_PKT_CONNECT" },
	{ "DEL_SNODE" },
	{ "DEL_GNODE" },
	{ "GET_INT_MAP" },
	{ "GET_EXT_MAP" },
	{ "GET_BNODE_MAP" },

	{ "ANDNA_REGISTER_HNAME" },
	{ "ANDNA_CHECK_COUNTER"},
	{ "ANDNA_RESOLVE_HNAME"},
	{ "ANDNA_RESOLVE_IP"},
	{ "ANDNA_RESOLVE_MX"},
	{ "ANDNA_GET_ANDNA_CACHE" },
	{ "ANDNA_GET_SINGLE_ACACHE" },
	{ "ANDNA_SPREAD_SACACHE" },
	{ "ANDNA_GET_COUNT_CACHE"}

};

const static char 	unknown_reply[]="Unknow reply";
const static u_char	reply_str[][30]=
{
	{ "PUT_FREE_NODES" },
	{ "PUT_QSPN_ROUND" },
	{ "PUT_INTERNET_GWS" },
	{ "PUT_DNODEIP"	   },
	{ "EMPTY_REPLY_SLOT" },
	{ "EMPTY_REPLY_SLOT1" },
	{ "PUT_INT_MAP"	   },
	{ "PUT_EXT_MAP"     },
	{ "PUT_BNODE_MAP" },
	{ "ANDNA_RESOLVE_REPLY"   },
	{ "ANDNA_REV_RESOLVE_REPLY"   },
	{ "ANDNA_MX_RESOLVE_REPLY"   },
	{ "ANDNA_PUT_COUNT_CACHE" },
	{ "ANDNA_PUT_ANDNA_CACHE" },

	{ "ACK_AFFERMATIVE"},
	{ "ACK_NEGATIVE"   }
};

const static u_char unknown_error[]="Unknow error";
const static u_char error_str[][40]=
{	
	{ "Invalid request" },
	{ "Accept table full" },
	{ "Request table full" },
	{ "Quadro Group full" },
	{ "Netsukuku is full" },
	{ "Invalid signature" },
	{ "Cannot forward the pkt" },
	{ "Invalid hash_gnode" },
	{ "ANDNA cache queue full" },
	{ "Hostname update too early" },
	{ "Too many hostname registered" },
	{ "Hname updates counter mismatch" },
	{ "Inexistent host name" },
	{ "Counter check failed" },
	{ "Too many connection" },
};

/*Wait time*/
#define ECHO_ME_WAIT			5		/*(in seconds)*/
#define ECHO_REPLY_WAIT			5
#define GET_FREE_NODES_WAIT		10
#define GET_QSPN_ROUND_WAIT		10

#define GET_INTERNET_GWS_WAIT		5
#define SET_FOREIGN_ROUTE_WAIT		5
#define DEL_FOREIGN_ROUTE_WAIT		5
#define NEW_BACKROUTE_WAIT		10
#define DELAYED_BROADCAST_WAIT		5
#define SPLIT_ROUTE_WAIT		20
#define SET_NO_IDENTITY_WAIT		20

#define QSPN_CLOSE_WAIT			0
#define QSPN_OPEN_WAIT			0
#define QSPN_RFR_WAIT			5
#define GET_DNODEBLOCK_WAIT		20
#define GET_DNODEIP_WAIT	     	5
#define TRACER_PKT_WAIT			10
#define TRACER_PKT_CONNECT_WAIT		10

#define DEL_SNODE_WAIT			10
#define DEL_GNODE_WAIT			10

#define GET_INT_MAP_WAIT		10
#define GET_EXT_MAP_WAIT		10
#define GET_BNODE_MAP_WAIT		10

#define ANDNA_REGISTER_HNAME_WAIT	5
#define ANDNA_CHECK_COUNTER_WAIT	5
#define ANDNA_RESOLVE_HNAME_WAIT	2
#define ANDNA_RESOLVE_IP_WAIT		5
#define ANDNA_RESOLVE_MX_WAIT		5
#define ANDNA_GET_ANDNA_CACHE_WAIT	10
#define ANDNA_GET_SINGLE_ACACHE_WAIT	10
#define ANDNA_SPREAD_SACACHE_WAIT	10
#define	ANDNA_GET_COUNT_CACHE_WAIT	10


/*Max simultaneous requests*/ 
#define ECHO_ME_MAXRQ			0	/*NO LIMITS*/
#define ECHO_REPLY_MAXRQ		20
#define GET_FREE_NODES_MAXRQ		5
#define GET_QSPN_ROUND_MAXRQ		5

#define GET_INTERNET_GWS_MAXRQ		5
#define SET_FOREIGN_ROUTE_MAXRQ		30
#define DEL_FOREIGN_ROUTE_MAXRQ		30
#define NEW_BACKROUTE_MAXRQ		10
#define DELAYED_BROADCAST_MAXRQ		5
#define SPLIT_ROUTE_MAXRQ		1
#define SET_NO_IDENTITY_MAXRQ		1

#define QSPN_CLOSE_MAXRQ		0	/*NO LIMITS*/
#define QSPN_OPEN_MAXRQ			0	/*NO LIMITS*/
#define QSPN_RFR_MAXRQ			10
#define GET_DNODEBLOCK_MAXRQ		1
#define GET_DNODEIP_MAXRQ		10
#define TRACER_PKT_MAXRQ		20
#define TRACER_PKT_CONNECT_MAXRQ	10

#define DEL_SNODE_MAXRQ			20
#define DEL_GNODE_MAXRQ			5

#define GET_INT_MAP_MAXRQ		2
#define GET_EXT_MAP_MAXRQ		2
#define GET_BNODE_MAP_MAXRQ		2

#define ANDNA_REGISTER_HNAME_MAXRQ	30
#define ANDNA_CHECK_COUNTER_MAXRQ	0	/*NO LIMITS*/
#define ANDNA_RESOLVE_HNAME_MAXRQ	80
#define ANDNA_RESOLVE_IP_MAXRQ		40
#define ANDNA_RESOLVE_MX_MAXRQ		40
#define ANDNA_GET_ANDNA_CACHE_MAXRQ	5
#define ANDNA_GET_SINGLE_ACACHE_MAXRQ	10
#define ANDNA_SPREAD_SACACHE_MAXRQ	10
#define	ANDNA_GET_COUNT_CACHE_MAXRQ	5

const static u_char unknown_request[]="Unknow request";
const static u_char request_array[][2]=
{ 
	{ ECHO_ME_WAIT,        ECHO_ME_MAXRQ	    },
	{ ECHO_REPLY_WAIT,     ECHO_REPLY_MAXRQ	    },
	{ GET_FREE_NODES_WAIT, GET_FREE_NODES_MAXRQ },
	{ GET_QSPN_ROUND_WAIT, GET_QSPN_ROUND_MAXRQ },
	
	{ GET_INTERNET_GWS_WAIT,  GET_INTERNET_GWS_MAXRQ  },
	{ SET_FOREIGN_ROUTE_WAIT, SET_FOREIGN_ROUTE_MAXRQ },
	{ DEL_FOREIGN_ROUTE_WAIT, DEL_FOREIGN_ROUTE_MAXRQ },
	{ NEW_BACKROUTE_WAIT,     NEW_BACKROUTE_MAXRQ 	  },
	{ DELAYED_BROADCAST_WAIT, DELAYED_BROADCAST_MAXRQ },
	{ SPLIT_ROUTE_WAIT,       SPLIT_ROUTE_MAXRQ       },
	{ SET_NO_IDENTITY_WAIT,   SET_NO_IDENTITY_MAXRQ   },
	{ QSPN_CLOSE_WAIT,        QSPN_CLOSE_MAXRQ        },
	{ QSPN_OPEN_WAIT,         QSPN_OPEN_MAXRQ         },
	{ QSPN_RFR_WAIT,	  QSPN_RFR_MAXRQ	  },
	{ GET_DNODEBLOCK_WAIT,    GET_DNODEBLOCK_MAXRQ    },
	{ GET_DNODEIP_WAIT,       GET_DNODEIP_MAXRQ       },
	{ TRACER_PKT_WAIT,	  TRACER_PKT_MAXRQ	  },
	{ TRACER_PKT_CONNECT_WAIT,TRACER_PKT_CONNECT_MAXRQ},
	{ DEL_SNODE_WAIT,         DEL_SNODE_MAXRQ         },
	{ DEL_GNODE_WAIT,         DEL_GNODE_MAXRQ         },
	{ GET_INT_MAP_WAIT,	  GET_INT_MAP_MAXRQ	  },
	{ GET_EXT_MAP_WAIT,	  GET_EXT_MAP_MAXRQ	  },
	{ GET_BNODE_MAP_WAIT,	  GET_BNODE_MAP_MAXRQ	  },

	{ ANDNA_REGISTER_HNAME_WAIT,   ANDNA_REGISTER_HNAME_MAXRQ   },
	{ ANDNA_CHECK_COUNTER_WAIT,    ANDNA_CHECK_COUNTER_MAXRQ    },
	{ ANDNA_RESOLVE_HNAME_WAIT,    ANDNA_RESOLVE_HNAME_MAXRQ    },
	{ ANDNA_RESOLVE_IP_WAIT,       ANDNA_RESOLVE_IP_MAXRQ	    },
	{ ANDNA_RESOLVE_MX_WAIT,       ANDNA_RESOLVE_MX_MAXRQ	    },
	{ ANDNA_GET_ANDNA_CACHE_WAIT,  ANDNA_GET_ANDNA_CACHE_MAXRQ  },
	{ ANDNA_GET_SINGLE_ACACHE_WAIT,ANDNA_GET_SINGLE_ACACHE_MAXRQ},
	{ ANDNA_SPREAD_SACACHE_WAIT,   ANDNA_SPREAD_SACACHE_MAXRQ   },
	{ ANDNA_GET_COUNT_CACHE_WAIT,  ANDNA_GET_COUNT_CACHE_MAXRQ  }
};

/* 
 * Request_array indexes defines:
 * ex: request_array[SET_FOREIGN_ROUTE][RQ_WAIT]
 */
#define RQ_WAIT 	0
#define RQ_MAXRQ	1

void rq_wait_idx_init(int *rq_wait_idx)
{
	int e, idx;
	
	for(e=0, idx=0; e<TOTAL_REQUESTS; e++) {
		rq_wait_idx[e]=idx;
		idx+=request_array[e][RQ_MAXRQ];
	}
}

int op_verify(u_char op)
{
	return op >= TOTAL_OPS;
}

int rq_verify(u_char rq)
{
	return rq >= TOTAL_REQUESTS;
}

int re_verify(u_char re)
{
	return ((op_verify(re)) || (re < TOTAL_REQUESTS));
}

int err_verify(u_char err)
{
	return err >= TOTAL_ERRORS;
}

const u_char *rq_strerror(int err)
{
	if(err_verify(err))
		return unknown_error;
	return error_str[err];
}

const u_char *rq_to_str(u_char rq)
{
	if(rq_verify(rq))
		return unknown_request;
	return request_str[rq];
}

const u_char *re_to_str(u_char re)
{
	if(re_verify(re))
		return (const u_char*)unknown_reply;
	return reply_str[re-TOTAL_REQUESTS];
}

void update_rq_tbl(rq_tbl *tbl)
{
	u_char i=0,e=0, idx=0;
	time_t cur_t;

	if(update_rq_tbl_mutex)
		return;
	else
		update_rq_tbl_mutex=1;

	time(&cur_t);

	for(; i<TOTAL_REQUESTS; i++) {
		for(e=0; e < request_array[i][RQ_MAXRQ]; e++) {
			if(tbl->rq_wait[idx] && (tbl->rq_wait[idx]+request_array[i][RQ_WAIT]) <= cur_t) {
				tbl->rq_wait[idx]=0;
				tbl->rq[i]--;
			}
			idx++;
		}
	}

	update_rq_tbl_mutex=0;
}
	
int is_rq_full(u_char rq, rq_tbl *tbl)
{
	if(rq_verify(rq))
		return E_INVALID_REQUEST;
	
	update_rq_tbl(tbl);
	
	if(tbl->rq[rq] >= request_array[rq][RQ_MAXRQ] && request_array[rq][RQ_MAXRQ])
		return E_REQUEST_TBL_FULL;
	else if(!request_array[rq][RQ_MAXRQ])
		return -1; /* No limits */
	else
		return 0;
}



int find_free_rq_wait(u_char rq, rq_tbl *tbl)
{
	int e, idx;
	
	for(e=0; e < request_array[rq][RQ_MAXRQ]; e++) {
		idx = rq_wait_idx[rq] + e;
		if(!tbl->rq_wait[idx])
			return idx;
	}
	
	return -1;	/*This happens if the rq_tbl is full for the "rq" request*/
}

int add_rq(u_char rq, rq_tbl *tbl)
{
	int err;
	time_t cur_t;
	
	/* TODO: XXX: Activate it and test it!!! */
	return 0;
	/* TODO: XXX: Activate it and test it!!! */
	
	if((err=is_rq_full(rq, tbl)) > 0)
		return err;
	else if(err < 0)
		return 0; /* no limits */
	
	time(&cur_t);
	
	tbl->rq[rq]++;
	tbl->rq_wait[find_free_rq_wait(rq, tbl)]=cur_t;	
	return 0;
}

/*
 * op_filter_reset_re: resets all the replies
 */
void op_filter_reset_re(int bit)
{
	int i;
	for(i=TOTAL_REQUESTS; i<TOTAL_OPS; i++)
		if(bit)
			op_filter_set(i);
		else
			op_filter_clr(i);
}

/*
 * op_filter_reset_rq: resets all the requests
 */
void op_filter_reset_rq(int bit)
{
	int i;
	for(i=0; i<TOTAL_REQUESTS; i++)
		if(bit)
			op_filter_set(i);
		else
			op_filter_clr(i);
}