122 lines
2.6 KiB
C
122 lines
2.6 KiB
C
|
|
#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;
|
|
};
|