#include "cw/cw.h"
#include "cw/mod.h"
#include "cw/log.h"
#include "cw/dbg.h"
#include "cw/dtls.h"
#include "cw/cfg.h"


/*#include "mod_cisco.h"*/
/*#include "../modload.h"*/

#include "cw/vendors.h"

#include "mod_cisco.h"
#include "capwap_cisco.h"

/*
extern int cisco_register_actions80211_ac(struct cw_actiondef *def);
extern int cisco_register_actions_ac(struct cw_actiondef *def);
*/

/*mbag_t cisco_config = NULL;*/

static struct cw_Mod * capwap_mod = NULL;
static struct cw_Mod * capwap80211_mod = NULL; 


static int write_header(struct cw_ElemHandler * handler, uint8_t * dst, int len)
{
	if (handler->proto == 0){
		if (handler->vendor)
			return len + cw_put_elem_vendor_hdr(dst, handler->vendor, handler->id, len);

		return  len + cw_put_elem_hdr(dst, handler->id, len);
	}
	/* put the lwap elem header */
	lw_set_dword(dst + 10, handler->vendor);
	lw_set_word(dst + 14, handler->id);
	return len + 6 + cw_put_elem_vendor_hdr(dst, handler->vendor, 
		CISCO_ELEM_SPAM_VENDOR_SPECIFIC, len+6);	
	
}

static int header_len(struct cw_ElemHandler * handler)
{
	if (handler->proto==0) 
		return handler->vendor ? 10 : 4;
	
	return 16;
}


static struct cw_MsgSet * register_messages(struct cw_MsgSet *set, int mode)
{
	cw_dbg(DBG_INFO,"CISCO: Register messages");
	switch (mode) {
		case CW_MOD_MODE_CAPWAP:
		{
			capwap_mod->register_messages(set, CW_MOD_MODE_CAPWAP);
			
			capwap80211_mod->register_messages(set, CW_MOD_MODE_BINDINGS);
			cisco_register_msg_set(set,CW_MOD_MODE_CAPWAP);

			set->write_header = write_header;
		      	set->header_len = header_len;	

		/*	cw_dbg(DBG_MOD,"Cisco: loading cisco message set");*/
			
/*			cw_dbg(DBG_INFO, "Initialized mod_cisco with %d messages", 7);*/
			break;
		}
		case CW_MOD_MODE_BINDINGS:
		{
			break;
		}


	}



	cw_dbg(DBG_INFO,"CISCO: Done register messages");
	return 0;


}

/*
static void errfunc(cfg_t *cfg, const char *fmt, va_list ap){
	
	if (cfg && cfg->filename && cfg->line)
		cw_log(LOG_ERR, "MOD Cisco cfg file in %s:%d: ", 
			cfg->filename, cfg->line);
	else if (cfg && cfg->filename)
		cw_log(LOG_ERR, "MOD Cisco cfg file in %s:", cfg->filename);
}
*/

static int init(struct cw_Mod *mod, mavl_t global_cfg, int role)
{
/*	uint8_t * str;*/
	static char * hardware_version; /*strdup(".x01000001");*/
	static char * software_version; /* = NULL; */
/*	cfg_t *cfg;*/
	
	int rc = 1;
/*	cfg_opt_t opts[] = {
		CFG_SIMPLE_STR("hardware_version", &hardware_version),
		CFG_SIMPLE_STR("software_version",&software_version),
		CFG_END()
	};
*/
	cw_dbg(DBG_INFO, "CISCO: Initialiazing mod_cisco ...");
	cw_dbg(DBG_MOD, "CISCO: Loading base module: capwap");
	
	capwap_mod = cw_mod_load("capwap",global_cfg,role);
	if (capwap_mod == NULL){
		cw_log(LOG_ERR, "CISCO: failed to load base module 'capwap");
	}

	capwap80211_mod = cw_mod_load("capwap80211", global_cfg,role);
	if (capwap_mod == NULL){
		cw_log(LOG_ERR, "CISCO: failed to load base module 'capwap80211");
	}
	
	/*cisco_config = mbag_create();*/


/*
	cfg = cfg_init(opts, CFGF_NONE);
	
	cfg_set_error_function(cfg, errfunc);
	
	char *filename = "cisco.conf";
	FILE * f = fopen(filename,"r");
	if (f){
		fclose(f);
		if (cfg_parse(cfg, filename)){
			rc = 0;
			goto errX;
		}
	}
*/


/*	
	str = bstr_create_from_cfgstr(hardware_version);
	mbag_set_bstrv(cisco_config, CW_ITEM_AC_HARDWARE_VERSION, 
		CW_VENDOR_ID_CISCO, 
		bstr_data(str),bstr_len(str)
		);
	free(str);

	if (software_version){
		str = bstr_create_from_cfgstr(software_version);
		mbag_set_bstrv(cisco_config, CW_ITEM_AC_SOFTWARE_VERSION, 
			CW_VENDOR_ID_CISCO, 
			bstr_data(str),bstr_len(str)
			);
		free(str);
	}
*/
/*errX:*/
	if (hardware_version)
		free (hardware_version);
	if (software_version)
		free(software_version);
	return rc;
}

static int detect(struct cw_Conn *conn, const uint8_t * rawmsg, int rawlen, int elems_len,
		  struct sockaddr *from, int mode)
{

	int offset = cw_get_hdr_msg_offset(rawmsg);
	const uint8_t *msg_ptr = rawmsg + offset;

	const uint8_t *elems_ptr = cw_get_msg_elems_ptr(msg_ptr);
	const uint8_t *elem;


	/* To detect a Cisco AP we look for any vendor 
	 * specific payload Cisco identifier */
	cw_foreach_elem(elem, elems_ptr, elems_len) {
		int id = cw_get_elem_id(elem);
		if (id == CAPWAP_ELEM_VENDOR_SPECIFIC_PAYLOAD) {
			uint32_t vendor_id = cw_get_dword(cw_get_elem_data(elem));
			if (vendor_id == CW_VENDOR_ID_CISCO) {
/*				//              conn->actions = &actions;*/
				if (mode == CW_MOD_MODE_CAPWAP) {
					cw_dbg(DBG_MOD, "CISCO capwap detected: yes");
				} else {
					cw_dbg(DBG_MOD, "CISCO bindings detected: yes");
				}

				return 1;

			}

		}

	}

	if (mode == CW_MOD_MODE_CAPWAP) {
		cw_dbg(DBG_MOD, "CISCO capwap detected: no");
	} else {
		cw_dbg(DBG_MOD, "CISCO bindings detected: no");
	}

	return 0;
}

/*
static struct cw_Mod capwap_ac = {
	.name = "cisco",
	.init = init,
	.detect = detect,
//	.register_actions = register_actions,
	.register_messages = register_messages
};
*/


int static setup_cfg(struct cw_Conn  * conn)
{
	int security;
	
//	conn->write_header=write_header;
//	conn->header_len=header_len;
	
	security = cw_setup_dtls(conn,conn->global_cfg,"cisco",CAPWAP_CIPHER);

	if (conn->role == CW_ROLE_AC){
		cw_cfg_set_int(conn->local_cfg,"ac-descriptor/security",security);
	
//		if (conn->default_cfg==NULL){
//			conn->default_cfg=cw_cfg_create();
//		}
	}
	


	return 0;
}


struct cw_Mod mod_cisco = {
	"cisco",			/* name */
	init,				/* init */
	detect,				/* detect */
	register_messages,		/* register_messages */
	NULL,				/* dll_handle */
	NULL,				/* data */
	setup_cfg			/* setup_cfg */
};





/*
struct cw_Mod *mod_cisco()
{
	return &cisco_data;
}
*/