#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>

#include "wtp_interface.h"
#include "wtp_conf.h"

#include "cw/dtls.h"
#include "cw/log.h"
#include "cw/dbg.h"
#include "cw/sock.h"
#include "cw/dtls.h"
#include "cw/mavl.h"

#include "wtp.h"


/*
#define acinfo_log acinfo_log_

void acinfo_log_(int level,const struct ac_info *acinfo,const char * xstr)
{
	char str[8192];
	acinfo_print(str,acinfo);
	cw_dbg(DBG_CW_INFO,"%s",str);
//	cw_log_debug(level,"%s\n%s",xstr,str);
	return;
}




int join_state(struct conn * conn)
{
	struct wtpinfo * wtpinfo = get_wtpinfo();

	struct radioinfo ri;
	memset(&ri,0,sizeof(ri));
	int rc;


	struct radioinfo *rip = &(wtpinfo->radioinfo[0]);

//#ifdef WITH_CW_LOG_DEBUG	
//	char str[64];
//	sock_addrtostr(&conn->addr,str,64);
//	cw_log_debug0("Sending join request to %s",str);
//#endif	
	printf("Seqnum before = %i\n",conn->seqnum);
	rc = cwsend_join_request(conn,rip,wtpinfo);
	printf("Seqnum after = %i\n",conn->seqnum);

	
	struct cwrmsg * cwrmsg;
//	do {
int i;
for(int i=0; i<10; i++){
		
		cwrmsg = conn_get_message(conn);
		if ( cwrmsg)
			break;
}
	printf("Received %08p\n",cwrmsg);
//	}while(cwrmsg==0);


if(!cwrmsg) {
	exit(0);
}
	
	printf("SEQS %d %d\n",cwrmsg->type,conn->seqnum);

//	cw_log_debug0("Received message %i",cwrmsg->seqnum);

	if (cwrmsg->type != CWMSG_JOIN_RESPONSE || cwrmsg->seqnum != conn->seqnum){
		printf("Wrong message %d %d\n",cwrmsg->type,conn->seqnum);
		
	}

	struct ac_info acinfo;
	memset(&acinfo,0,sizeof(acinfo));

acinfo.result_code=99;

	cwread_join_response(&acinfo,cwrmsg->msgelems,cwrmsg->msgelems_len);

	acinfo_log(0,&acinfo,"Connectet to the following AC");

	printf("Result_code: %d\n",acinfo.result_code);

}

*/





int run_join_d(struct conn * conn, struct sockaddr *sa)
{
	char addrstr[SOCK_ADDR_BUFSIZE];
	int sockfd;
	
	int rsec,lsec;
	int rc;

printf("JOINFD: %s\n",sock_addr2str_p(sa,addrstr));

	lsec = cw_ktv_get_byte(conn->local_cfg,"ac-descriptor/security",0);
	rsec = cw_ktv_get_byte(conn->remote_cfg,"ac-descriptor/security",0);

	printf("Anding my and remote %d %d %d\n",lsec,rsec, rsec & lsec);
	
	if ((lsec & rsec) == 0){
		cw_log(LOG_ERR, "Can't establish DTLS with AC, my sec: %d, remote sec %d",lsec,rsec);
		return 0;
	}
	
/*	struct conn *conn = get_conn();*/

	conn->capwap_state = CAPWAP_STATE_JOIN;



	sockfd = socket(sa->sa_family, SOCK_DGRAM, 0);
	if (sockfd == -1) {
		cw_log(LOG_ERR, "Can't create socket: %s\n", strerror(errno));
		return -1;
	}
	sock_set_recvtimeout(sockfd, 1);
	conn->sock = sockfd;
	sock_copyaddr(&conn->addr, sa);


	/* we call connect to bind this socket to a local IP address,
	 * which we can later obtain by getsockname */
	rc = connect(sockfd, (struct sockaddr *) sa,
		     sock_addrlen((struct sockaddr *) sa));

/*	if (rc < 0) {
		cw_log(LOG_ERR, "Can't connect to %s: %s\n", sock_addr2str(sa),
		       strerror(errno));
		close(sockfd);
		return -1;
	}
*/

	cw_dbg(DBG_DTLS, "Establishing DTLS session with %s", sock_addr2str(sa, addrstr));
/*
	int dtls_conf_ok=0;

	if (conn->dtls_psk) {
		conn->dtls_psk = conf_dtls_psk;
		conn->dtls_psk_len = strlen(conn->dtls_psk);
		conn->dtls_cipher = conf_dtls_cipher;
		dtls_conf_ok=1;
	}

	if (conf_sslkeyfilename && conf_sslcertfilename) {

		conn->dtls_key_file = conf_sslkeyfilename;
		conn->dtls_cert_file = conf_sslcertfilename;
		conn->dtls_key_pass = conf_sslkeypass;
		conn->dtls_cipher = conf_dtls_cipher;
		dtls_conf_ok=1;
	}

	if (!dtls_conf_ok){
		cw_log(LOG_ERR,"Can't establish DTLS connection with %s, neither psk nor cert set in config",
			sock_addr2str(sa,addrstr));
		close(sockfd);
		return 0;
	}
*/




	rc = dtls_connect(conn);
	if (rc != 1) {
		dtls_shutdown(conn);
		cw_log(LOG_ERR, "Can't establish DTLS connection with %s",
		       sock_addr2str(sa,addrstr));
		close(sockfd);
		return 0;
	}


	cw_dbg(DBG_DTLS, "DTLS Connection successful established with %s",
	       sock_addr2str(sa,addrstr));



	return 1;
}






int run_join(struct conn *conn)
{
	char addrstr[SOCK_ADDR_BUFSIZE];
/*	
//      cw_init_request(conn, CW_MSG_JOIN_REQUEST);
//      if ( cw_put_msg(conn, conn->req_buffer) == -1 )
//              return 0;
//
//      conn_send_msg(conn, conn->req_buffer);
*/

/*	mbag_del_all(conn->incomming);*/

/*	//mbag_del (conn->incomming,CW_ITEM_RESULT_CODE);*/

	int rc = cw_send_request(conn, CAPWAP_MSG_JOIN_REQUEST);

	if (!cw_result_is_ok(rc)) {
		if (rc > 0) {
			cw_log(LOG_ERR, "Can't Join AC at %s, AC said: %d - %s.",
			       sock_addr2str(&conn->addr,addrstr), rc, cw_strerror(rc));

		} else {
			cw_log(LOG_ERR, "Can't Join AC at %s: %d - %s.",
			       sock_addr2str(&conn->addr,addrstr), errno, cw_strerror(rc));
		}
		return 0;
	}

	cw_dbg(DBG_ELEM_IN, "Joined AC at %s,  Join Result: %d - %s",
	       sock_addr2str(&conn->addr,addrstr), rc, cw_strresult(rc));

	return 1;
}

int join(struct conn * conn, struct cw_DiscoveryResult * dis)
{

	mavliter_t ii;
	mavliter_init(&ii,dis->prio_ip);

	mavliter_foreach(&ii){
		cw_KTV_t * val,*ac;
		mavl_t rcfg;
		char * rk;
		char ipstr[100];
		char ac_name[CAPWAP_MAX_AC_NAME_LEN];
		struct sockaddr_storage sockaddr;
		
		val = mavliter_get(&ii);
		rk = val->key;
		val = val->val.ptr;
		val->type->to_str(val, ipstr, 100);
		
		rcfg = cw_ktv_get_sysptr(dis->prio_ac,rk,NULL);
		
		ac=cw_ktv_get(rcfg,"ac-name",CW_TYPE_BSTR16);
		if (ac != NULL){
			ac->type->to_str(ac,ac_name,sizeof(ac_name));
		}
		else{
			strcpy(ac_name,"");
		}
		
		
		cw_dbg(DBG_INFO, "Going to join CAPWAP controller '%s' at %s.",ac_name,ipstr);
		
		conn->remote_cfg=rcfg;
		
		/*cw_dbg_ktv_dump(conn->local_cfg,DBG_INFO,"remopte ac","preifx**: ","bottom");
		*/
		
		sock_strtoaddr(ipstr,(struct sockaddr*)(&sockaddr));
		sock_setport((struct sockaddr*)&sockaddr,5246);
		run_join_d(conn,(struct sockaddr*)(&sockaddr));
	}


	/*mbag_del_all(conn->incomming);*/

/*	cw_aciplist_t iplist =
	    mbag_get_mavl(conn->local, CW_ITEM_CAPWAP_CONTROL_IP_ADDRESS_LIST);
	if (!iplist) {
		cw_log(LOG_ERR, "No IPs to join controller.");
		return 0;
	}

	if (!mavl_get_count(iplist)){
		cw_log(LOG_ERR, "No IPs to join controller. IP list is empty.");
		return 0;
	}
*/


	mavliter_foreach(&ii) {

/*		cw_acip_t *ip = avliter_get(&ii);*/


	/*	cw_dbg(DBG_INFO, "Going to join CAWAP controller on %s",
		       sock_addr2str_p(&ip->ip,addrstr));
*/

/*		int rc = run_join_d((struct sockaddr *) &ip->ip);

		if (rc<=0)
			continue;

		rc = run_join(conn);
		if (rc) {
			conn->capwap_state = CW_STATE_CONFIGURE;
			return 1;
		}
*/


	}


	return 0;

}