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

#include "cw/action.h"

#include "mod_fortinet.h"
#include "../modload.h"

#include "cw/vendors.h"


extern int fortinet_register_actions80211_ac(struct cw_actiondef *def);
extern int fortinet_register_actions_ac(struct cw_actiondef *def);

static int register_actions(struct cw_actiondef *actions, int mode)
{
	switch (mode) {
		case CW_MOD_MODE_CAPWAP:
		{

			struct cw_Mod *cmod = modload_ac("capwap");
			if (!cmod) {
				cw_log(LOG_ERR,
				       "Can't initialize mod_fortinet, failed to load base mod mod_capwap");
				return 1;
			}
			cmod->register_actions(actions, CW_MOD_MODE_CAPWAP);
			int rc = fortinet_register_actions_ac(actions);
			cw_dbg(DBG_INFO, "Initialized mod fortinet with %d actions", rc);
			return 0;
		}
		case MOD_MODE_BINDINGS:
		{
			struct cw_Mod *cmod = modload_ac("capwap80211");
			if (!cmod) {
				cw_log(LOG_ERR,
				       "Can't initialize mod_fortinet, failed to load base mod mod_capwap80211");
				return 1;
			}
			cmod->register_actions(actions, MOD_MODE_BINDINGS);
			int rc = fortinet_register_actions80211_ac(actions);
			cw_dbg(DBG_INFO, "Initialized mod_fortinet 80211 with %d actions", rc);
			return 0;
		}


	}


	return 0;


}



static int init()
{
	cw_dbg(DBG_INFO, "Initialiazing mod_fortineto ...");
//      struct cw_Mod *cmod = modload_ac("capwap");
	return 1;
}


static int detect(struct 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 Fortinet AP we look for any vendor
	 * specific payload Fortinet 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_FORTINET) {
				//              conn->actions = &actions;
				if (mode == CW_MOD_MODE_CAPWAP) {
					cw_dbg(DBG_MOD, "Fortinet capwap detected: yes");
				} else {
					cw_dbg(DBG_MOD, "Fortinet bindings detected: yes");
				}

				return 1;

			}

		}

	}

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

	return 0;
}

static struct cw_Mod capwap_ac = {
	.name = "fortinet",
	.init = init,
	.detect = detect,
	.register_actions = register_actions
};

struct cw_Mod *mod_fortinet_ac()
{
	return &capwap_ac;
};