#include <string.h>

#include "cw/sock.h"
#include "cw/mlist.h"
#include "mavl.h"
#include "cw/mod.h"
#include "cw/dbg.h"

#include "discovery_cache.h"
struct cw_DiscoveryCacheElem{
	struct sockaddr_storage addr;
	struct cw_Mod *cmod, *bmod;
	uint32_t ctrlo, ctrhi;
};

struct cw_DiscoveryCache{
	int len;
	int qpos;
	int qdel;
	mavl_t byaddrp;
	mavl_t byaddr;
	struct cw_DiscoveryCacheElem * queue;
	uint32_t ctrlo, ctrhi;
	
};


void discovery_cache_destroy(struct cw_DiscoveryCache * c)
{
	if (c->byaddr)
		mavl_destroy(c->byaddr);
	
	if (c->byaddrp)
		mavl_destroy(c->byaddrp);

	if (c->queue)
		free(c->queue);
	
	free(c);
}


static int cmpaddrp ( const void * d1, const void *d2 )
{
	struct cw_DiscoveryCacheElem * c1 = * ( void ** ) d1 ;
	struct cw_DiscoveryCacheElem * c2 = * ( void ** ) d2 ;
	return sock_cmpaddr ( ( struct sockaddr* ) &c1->addr, ( struct sockaddr* ) &c2->addr, 1 );
}


static int cmpaddr ( const void * d1, const void *d2 )
{
	struct cw_DiscoveryCacheElem * c1 = * ( void ** ) d1 ;
	struct cw_DiscoveryCacheElem * c2 = * ( void ** ) d2 ;
	int rc;

	rc = sock_cmpaddr ( ( struct sockaddr* ) &c1->addr, ( struct sockaddr* ) &c2->addr, 0 );
	if (rc != 0)
		return rc;
	
	if (c1->ctrhi != c2->ctrhi)
		return c1->ctrhi-c2->ctrhi;
	
	return c1->ctrlo-c2->ctrlo;

}




struct cw_DiscoveryCache * discovery_cache_create(int len)
{
	
	int i;
	struct cw_DiscoveryCache * cache;
	
	cache = malloc (sizeof(struct cw_DiscoveryCache));
	if (cache==NULL)
		return NULL;
		
	memset(cache,0,sizeof(struct cw_DiscoveryCache));
	cache->queue = malloc(sizeof(struct cw_DiscoveryCacheElem)*len);
	if (cache->queue==NULL)
		goto errX;

	for (i=0; i<len; i++){
		cache->queue[i].cmod=NULL;
	}

	cache->byaddr = mavl_create_ptr(cmpaddr,NULL);
	if (cache->byaddr == NULL)
		goto errX;

	
	cache->byaddrp = mavl_create_ptr(cmpaddrp,NULL);
	if (cache->byaddrp == NULL)
		goto errX;

	cache->len=len;
	cache->qpos=0;
	
	
	return cache;
errX:
	discovery_cache_destroy(cache);
	return NULL;
}

void discovery_cache_add(struct cw_DiscoveryCache *cache, 
	struct sockaddr * addr, struct cw_Mod  * mod_capwap, struct cw_Mod * mod_bindings)
{

	{
		cw_dbg(DBG_X,"//////////////////////////   ADDD CAHCE ///////////////////////");
		char str[128];
	sock_addrtostr(addr,str,128,1);
	}

	struct cw_DiscoveryCacheElem * cur = cache->queue+cache->qpos;
	if (cur->cmod!=NULL){
		/* delete here */

		cw_dbg(DBG_X,"DOIN MAVL DEL ----- ///////////");

		void * ptr = &cur;
		mavl_del(cache->byaddr,ptr);
		mavl_del(cache->byaddrp,ptr);
	}

	cur->cmod=mod_capwap;
	cur->bmod=mod_bindings;
	sock_copyaddr(&cur->addr,addr);
	
	cur->ctrhi=cache->ctrhi;
	cur->ctrlo=cache->ctrlo;
	
	if ((++cache->ctrlo)==0){
		cache->ctrhi++;
	}
	
	mavl_insert_ptr(cache->byaddr,cur);
	mavl_insert_ptr(cache->byaddrp,cur);
	
	cache->qpos++;
	if (cache->qpos>=cache->len)
		cache->qpos=0;
	
}

int discovery_cache_get(struct cw_DiscoveryCache * cache,struct sockaddr *addr, 
			struct cw_Mod ** modcapwap, struct cw_Mod **modbindings)
{
	struct cw_DiscoveryCacheElem * elem, ** result, *search_ptr, search;
	sock_copyaddr(&search.addr,addr);
	search.ctrhi=search.ctrlo=0;
	
	{
	cw_dbg(DBG_X,"DISCOVERY CACHE ---------------------------- GET ENTER");
	char str[128];
	sock_addrtostr(addr,str,128,1);
	}

	search_ptr = &search;
	result = mavl_get(cache->byaddrp ,&search_ptr);
	if (result != NULL){
		{
		cw_dbg(DBG_X,"DISCOVERY CACHE ---------------------------- FOUND 1 with Port");
		char str[128];
		sock_addrtostr(addr,str,128,1);
		}


		elem = *result;
		if (elem != NULL){
			mavl_del(cache->byaddr,result);
			mavl_del(cache->byaddrp,result);
			*modcapwap=elem->cmod;
			*modbindings=elem->bmod;
			
			elem->cmod=NULL;
			return 1;
		}
	}


	search_ptr = &search;
	/*elem = *((struct cw_DiscoveryCacheElem **)
*/
	result = mavl_get_first(cache->byaddr, &search_ptr);
	if (result == NULL)
		return 0;
	{
	cw_dbg(DBG_X,"DISCOVERY CACHE ---------------------------- DOUND BY ADDR");
	char str[128];
	sock_addrtostr(addr,str,128,1);
	}


	elem = *result;
	if (elem != NULL){
		cw_dbg(DBG_X,"elem != NULL");
		
	{
	cw_dbg(DBG_X,"DISCOVERY CACHE ---------------------------- DOUND BY ADDR");
	char str[128];
	sock_addrtostr((struct sockaddr*)&elem->addr,str,128,1);
	printf("ELEM addr: %s\n",str);
	}


		


		if (sock_cmpaddr((struct sockaddr*)&elem->addr,addr,0)!=0)
			return 0;
			
		cw_dbg(DBG_X,"DOIN MAVL DEL HERO ----- ///////////");
		mavl_del(cache->byaddr,&elem);
		cw_dbg(DBG_X,"DOIN MAVL DEL HERXXXXXXO ----- ///////////");
		mavl_del(cache->byaddrp,&elem);
		cw_dbg(DBG_X,"DOIN MAVL DEL HERO 0000 ----- ///////////");
		
		*modcapwap=elem->cmod;
		*modbindings=elem->bmod;
		
		elem->cmod=NULL;
	}
	
	return 1;
	
}