freewtp/src/dfa_run.c

415 lines
13 KiB
C

#include "wtp.h"
#include "capwap_dfa.h"
#include "element.h"
#include "dfa.h"
#include "radio.h"
#include "ieee80211.h"
/* ev timer callbacks */
static void wtp_dfa_state_run_echo_timeout(EV_P_ ev_timer *w, int revents);
static void wtp_dfa_state_run_keepalive_timeout(EV_P_ ev_timer *w, int revents);
static void wtp_dfa_state_run_keepalivedead_timeout(EV_P_ ev_timer *w, int revents);
/* */
static int send_echo_request(void)
{
int result = -1;
struct capwap_header_data capwapheader;
struct capwap_packet_txmng* txmngpacket;
struct capwap_vendor_travelping_wtp_timestamp_element timestamp;
gettimeofday(&timestamp.tv, NULL);
/* Build packet */
capwap_header_init(&capwapheader, CAPWAP_RADIOID_NONE, g_wtp.binding);
txmngpacket = capwap_packet_txmng_create_ctrl_message(&capwapheader, CAPWAP_ECHO_REQUEST,
g_wtp.localseqnumber, g_wtp.mtu);
/* Add message element */
capwap_packet_txmng_add_message_element(txmngpacket,
CAPWAP_ELEMENT_VENDOR_TRAVELPING_WTP_TIMESTAMP,
&timestamp);
/* Echo request complete, get fragment packets */
wtp_free_reference_last_request();
capwap_packet_txmng_get_fragment_packets(txmngpacket, g_wtp.requestfragmentpacket, g_wtp.fragmentid);
if (g_wtp.requestfragmentpacket->count > 1)
g_wtp.fragmentid++;
/* Free packets manager */
capwap_packet_txmng_free(txmngpacket);
/* Send echo request to AC */
if (!capwap_crypt_sendto_fragmentpacket(&g_wtp.dtls, g_wtp.requestfragmentpacket)) {
/* Error to send packets */
log_printf(LOG_DEBUG, "Warning: error to send echo request packet");
wtp_free_reference_last_request();
return result;
}
return 0;
}
/* */
static int receive_echo_response(struct capwap_parsed_packet* packet)
{
struct capwap_resultcode_element* resultcode;
struct capwap_vendor_travelping_wtp_timestamp_element *timestamp;
ASSERT(packet != NULL);
/* Check the success of the Request */
resultcode = (struct capwap_resultcode_element *)
capwap_get_message_element_data(packet, CAPWAP_ELEMENT_RESULTCODE);
if (resultcode && !CAPWAP_RESULTCODE_OK(resultcode->code)) {
log_printf(LOG_WARNING, "Receive Echo Response with error: %d", (int)resultcode->code);
return 1;
}
timestamp = (struct capwap_vendor_travelping_wtp_timestamp_element *)
capwap_get_message_element_data(packet, CAPWAP_ELEMENT_VENDOR_TRAVELPING_WTP_TIMESTAMP);
if (timestamp) {
struct timeval now;
gettimeofday(&now, NULL);
timersub(&now, &timestamp->tv, &g_wtp.echo_latency);
log_printf(LOG_DEBUG, "Echo Latency: %ld.%03ld ms",
g_wtp.echo_latency.tv_sec * 1000 + g_wtp.echo_latency.tv_usec / 1000,
g_wtp.echo_latency.tv_usec % 1000);
}
/* Valid packet, free request packet */
wtp_free_reference_last_request();
return 0;
}
/* */
static void receive_reset_request(struct capwap_parsed_packet* packet)
{
unsigned short binding;
struct capwap_header_data capwapheader;
struct capwap_packet_txmng* txmngpacket;
struct capwap_resultcode_element resultcode = { .code = CAPWAP_RESULTCODE_SUCCESS };
ASSERT(packet != NULL);
/* */
binding = GET_WBID_HEADER(packet->rxmngpacket->header);
if (binding != g_wtp.binding) {
log_printf(LOG_DEBUG, "Reset Request for invalid binding");
return;
}
if (!IS_SEQUENCE_SMALLER(g_wtp.remoteseqnumber, packet->rxmngpacket->ctrlmsg.seq)) {
log_printf(LOG_DEBUG, "Reset Request with invalid sequence (%d < %d)",
g_wtp.remoteseqnumber, packet->rxmngpacket->ctrlmsg.seq);
return;
}
/* Build packet */
capwap_header_init(&capwapheader, CAPWAP_RADIOID_NONE, g_wtp.binding);
txmngpacket = capwap_packet_txmng_create_ctrl_message(&capwapheader,
CAPWAP_RESET_RESPONSE,
packet->rxmngpacket->ctrlmsg.seq, g_wtp.mtu);
/* Add message element */
capwap_packet_txmng_add_message_element(txmngpacket, CAPWAP_ELEMENT_RESULTCODE, &resultcode);
/* CAPWAP_ELEMENT_VENDORPAYLOAD */ /* TODO */
/* Reset response complete, get fragment packets */
wtp_free_reference_last_response();
capwap_packet_txmng_get_fragment_packets(txmngpacket, g_wtp.responsefragmentpacket, g_wtp.fragmentid);
if (g_wtp.responsefragmentpacket->count > 1)
g_wtp.fragmentid++;
/* Free packets manager */
capwap_packet_txmng_free(txmngpacket);
/* Save remote sequence number */
g_wtp.remotetype = packet->rxmngpacket->ctrlmsg.type;
g_wtp.remoteseqnumber = packet->rxmngpacket->ctrlmsg.seq;
/* Send Reset response to AC */
if (!capwap_crypt_sendto_fragmentpacket(&g_wtp.dtls, g_wtp.responsefragmentpacket)) {
log_printf(LOG_DEBUG, "Warning: error to send reset response packet");
}
}
/* */
static void receive_station_configuration_request(struct capwap_parsed_packet* packet)
{
unsigned short binding;
struct capwap_header_data capwapheader;
struct capwap_packet_txmng* txmngpacket;
struct capwap_resultcode_element resultcode = { .code = CAPWAP_RESULTCODE_FAILURE };
ASSERT(packet != NULL);
/* */
binding = GET_WBID_HEADER(packet->rxmngpacket->header);
if (binding != g_wtp.binding) {
log_printf(LOG_DEBUG, "Station Configuration Request for invalid binding");
return;
}
if (!IS_SEQUENCE_SMALLER(g_wtp.remoteseqnumber, packet->rxmngpacket->ctrlmsg.seq)) {
log_printf(LOG_DEBUG, "Station Configuration Request with invalid sequence (%d < %d)",
g_wtp.remoteseqnumber, packet->rxmngpacket->ctrlmsg.seq);
return;
}
/* Parsing request message */
if (capwap_get_message_element(packet, CAPWAP_ELEMENT_ADDSTATION)) {
resultcode.code = wtp_radio_add_station(packet);
} else if (capwap_get_message_element(packet, CAPWAP_ELEMENT_DELETESTATION)) {
resultcode.code = wtp_radio_delete_station(packet);
}
/* Build packet */
capwap_header_init(&capwapheader, CAPWAP_RADIOID_NONE, g_wtp.binding);
txmngpacket = capwap_packet_txmng_create_ctrl_message(&capwapheader,
CAPWAP_STATION_CONFIGURATION_RESPONSE,
packet->rxmngpacket->ctrlmsg.seq, g_wtp.mtu);
/* Add message element */
capwap_packet_txmng_add_message_element(txmngpacket, CAPWAP_ELEMENT_RESULTCODE, &resultcode);
/* CAPWAP_ELEMENT_VENDORPAYLOAD */ /* TODO */
/* Station Configuration response complete, get fragment packets */
wtp_free_reference_last_response();
capwap_packet_txmng_get_fragment_packets(txmngpacket, g_wtp.responsefragmentpacket, g_wtp.fragmentid);
if (g_wtp.responsefragmentpacket->count > 1)
g_wtp.fragmentid++;
/* Free packets manager */
capwap_packet_txmng_free(txmngpacket);
/* Save remote sequence number */
g_wtp.remotetype = packet->rxmngpacket->ctrlmsg.type;
g_wtp.remoteseqnumber = packet->rxmngpacket->ctrlmsg.seq;
/* Send Station Configuration response to AC */
if (!capwap_crypt_sendto_fragmentpacket(&g_wtp.dtls, g_wtp.responsefragmentpacket)) {
log_printf(LOG_DEBUG, "Warning: error to send Station Configuration response packet");
}
}
/* */
static void receive_ieee80211_wlan_configuration_request(struct capwap_parsed_packet* packet)
{
unsigned short binding;
struct capwap_message_element_id action = {0, 0};
struct capwap_header_data capwapheader;
struct capwap_packet_txmng* txmngpacket;
struct capwap_80211_assignbssid_element bssid;
struct capwap_resultcode_element resultcode = { .code = CAPWAP_RESULTCODE_FAILURE };
ASSERT(packet != NULL);
/* */
binding = GET_WBID_HEADER(packet->rxmngpacket->header);
if (binding != g_wtp.binding) {
log_printf(LOG_DEBUG, "IEEE 802.11 WLAN Configuration Request for invalid binding");
return;
}
if (!IS_SEQUENCE_SMALLER(g_wtp.remoteseqnumber, packet->rxmngpacket->ctrlmsg.seq)) {
log_printf(LOG_DEBUG, "IEEE 802.11 WLAN Configuration Request with invalid sequence (%d < %d)",
g_wtp.remoteseqnumber, packet->rxmngpacket->ctrlmsg.seq);
return;
}
/* Parsing request message */
if (capwap_get_message_element(packet, CAPWAP_ELEMENT_80211_ADD_WLAN)) {
action = CAPWAP_ELEMENT_80211_ADD_WLAN;
resultcode.code = wtp_radio_create_wlan(packet, &bssid);
} else if (capwap_get_message_element(packet, CAPWAP_ELEMENT_80211_UPDATE_WLAN)) {
action = CAPWAP_ELEMENT_80211_UPDATE_WLAN;
resultcode.code = wtp_radio_update_wlan(packet);
} else if (capwap_get_message_element(packet, CAPWAP_ELEMENT_80211_DELETE_WLAN)) {
action = CAPWAP_ELEMENT_80211_DELETE_WLAN;
resultcode.code = wtp_radio_delete_wlan(packet);
}
/* Build packet */
capwap_header_init(&capwapheader, CAPWAP_RADIOID_NONE, g_wtp.binding);
txmngpacket = capwap_packet_txmng_create_ctrl_message(&capwapheader,
CAPWAP_IEEE80211_WLAN_CONFIGURATION_RESPONSE,
packet->rxmngpacket->ctrlmsg.seq, g_wtp.mtu);
/* Add message element */
capwap_packet_txmng_add_message_element(txmngpacket, CAPWAP_ELEMENT_RESULTCODE, &resultcode);
if (resultcode.code == CAPWAP_RESULTCODE_SUCCESS &&
message_element_id_eq(action, CAPWAP_ELEMENT_80211_ADD_WLAN))
capwap_packet_txmng_add_message_element(txmngpacket, CAPWAP_ELEMENT_80211_ASSIGN_BSSID, &bssid);
/* CAPWAP_ELEMENT_VENDORPAYLOAD */ /* TODO */
/* IEEE802.11 WLAN Configuration response complete, get fragment packets */
wtp_free_reference_last_response();
capwap_packet_txmng_get_fragment_packets(txmngpacket, g_wtp.responsefragmentpacket, g_wtp.fragmentid);
if (g_wtp.responsefragmentpacket->count > 1)
g_wtp.fragmentid++;
/* Free packets manager */
capwap_packet_txmng_free(txmngpacket);
/* Save remote sequence number */
g_wtp.remotetype = packet->rxmngpacket->ctrlmsg.type;
g_wtp.remoteseqnumber = packet->rxmngpacket->ctrlmsg.seq;
/* Send IEEE802.11 WLAN Configuration response to AC */
if (!capwap_crypt_sendto_fragmentpacket(&g_wtp.dtls, g_wtp.responsefragmentpacket)) {
log_printf(LOG_DEBUG, "Warning: error to send IEEE802.11 WLAN Configuration response packet");
}
}
/* */
static void wtp_dfa_state_run_echo_timeout(EV_P_ ev_timer *w, int revents)
{
log_printf(LOG_DEBUG, "Send Echo Request");
if (send_echo_request()) {
log_printf(LOG_ERR, "Unable to send Echo Request");
wtp_teardown_connection();
return;
}
g_wtp.retransmitcount = 0;
wtp_dfa_start_retransmition_timer();
}
/* */
static void wtp_dfa_state_run_keepalive_timeout(EV_P_ ev_timer *w, int revents)
{
log_printf(LOG_DEBUG, "Send Keep-Alive");
ev_timer_again(EV_A_ &g_wtp.timerkeepalivedead);
if (wtp_kmod_send_keepalive()) {
log_printf(LOG_ERR, "Unable to send Keep-Alive");
wtp_teardown_connection();
}
}
/* */
static void wtp_dfa_state_run_keepalivedead_timeout(EV_P_ ev_timer *w, int revents)
{
log_printf(LOG_INFO, "Keep-Alive timeout, teardown");
wtp_teardown_connection();
}
/* */
void wtp_recv_data_keepalive(void) {
log_printf(LOG_DEBUG, "Receive Keep-Alive");
/* Receive Data Keep-Alive, wait for next packet */
if (ev_is_active(&g_wtp.timerkeepalivedead))
ev_timer_stop(EV_DEFAULT_UC_ &g_wtp.timerkeepalivedead);
ev_timer_again(EV_DEFAULT_UC_ &g_wtp.timerkeepalive);
}
/* */
void wtp_recv_data(uint8_t* buffer, int length) {
int headersize;
struct capwap_header* header = (struct capwap_header*)buffer;
/* */
if (length < sizeof(struct capwap_header)) {
return;
}
/* */
headersize = GET_HLEN_HEADER(header) * 4;
if ((length - headersize) > 0) {
wtp_radio_receive_data_packet(GET_RID_HEADER(header), GET_WBID_HEADER(header),
(buffer + headersize), (length - headersize));
}
}
/* */
void wtp_dfa_state_run_enter()
{
ev_timer_init(&g_wtp.timerecho,
wtp_dfa_state_run_echo_timeout,
0., g_wtp.echointerval / 1000.0);
ev_timer_init(&g_wtp.timerkeepalivedead,
wtp_dfa_state_run_keepalivedead_timeout,
0., WTP_DATACHANNEL_KEEPALIVEDEAD / 1000.0);
ev_timer_init(&g_wtp.timerkeepalive,
wtp_dfa_state_run_keepalive_timeout,
0., WTP_DATACHANNEL_KEEPALIVE_INTERVAL / 1000.0);
ev_timer_again(EV_DEFAULT_UC_ &g_wtp.timerecho);
ev_timer_again(EV_DEFAULT_UC_ &g_wtp.timerkeepalivedead);
}
/* */
void wtp_dfa_state_run(struct capwap_parsed_packet* packet)
{
ASSERT(packet != NULL);
if (!capwap_is_request_type(packet->rxmngpacket->ctrlmsg.type) &&
g_wtp.localseqnumber != packet->rxmngpacket->ctrlmsg.seq)
return;
if (!capwap_is_request_type(packet->rxmngpacket->ctrlmsg.type)) {
wtp_dfa_stop_retransmition_timer();
/* Update sequence */
g_wtp.localseqnumber++;
}
/* Parsing message */
switch (packet->rxmngpacket->ctrlmsg.type) {
case CAPWAP_CONFIGURATION_UPDATE_REQUEST:
/* TODO */
break;
case CAPWAP_CHANGE_STATE_EVENT_RESPONSE:
/* TODO */
break;
case CAPWAP_ECHO_RESPONSE:
if (!receive_echo_response(packet)) {
log_printf(LOG_DEBUG, "Receive Echo Response");
g_wtp.timerecho.repeat = g_wtp.echointerval / 1000.0;
ev_timer_again(EV_DEFAULT_UC_ &g_wtp.timerecho);
}
break;
case CAPWAP_CLEAR_CONFIGURATION_REQUEST:
/* TODO */
break;
case CAPWAP_WTP_EVENT_RESPONSE:
/* TODO */
break;
case CAPWAP_DATA_TRANSFER_REQUEST:
/* TODO */
break;
case CAPWAP_DATA_TRANSFER_RESPONSE:
/* TODO */
break;
case CAPWAP_STATION_CONFIGURATION_REQUEST:
receive_station_configuration_request(packet);
break;
case CAPWAP_RESET_REQUEST:
receive_reset_request(packet);
wtp_dfa_change_state(CAPWAP_RESET_STATE);
break;
case CAPWAP_IEEE80211_WLAN_CONFIGURATION_REQUEST:
receive_ieee80211_wlan_configuration_request(packet);
break;
}
}