Cleaning WTP DFA code
This commit is contained in:
vemax78
@ -87,231 +87,55 @@ static void wtp_send_invalid_request(struct capwap_packet_rxmng* rxmngpacket, st
|
||||
}
|
||||
|
||||
/* WTP Execute state */
|
||||
static int wtp_dfa_execute(struct capwap_parsed_packet* packet, struct timeout_control* timeout) {
|
||||
int action = WTP_DFA_NO_PACKET;
|
||||
|
||||
static void wtp_dfa_execute(struct capwap_parsed_packet* packet, struct timeout_control* timeout) {
|
||||
switch (g_wtp.dfa.state) {
|
||||
case CAPWAP_IDLE_STATE: {
|
||||
action = wtp_dfa_state_idle(packet, timeout);
|
||||
wtp_dfa_state_idle(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_IDLE_TO_DISCOVERY_STATE: {
|
||||
action = wtp_dfa_state_idle_to_discovery(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_IDLE_TO_DTLS_SETUP_STATE: {
|
||||
action = wtp_dfa_state_idle_to_dtlssetup(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DISCOVERY_STATE: {
|
||||
action = wtp_dfa_state_discovery(packet, timeout);
|
||||
wtp_dfa_state_discovery(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DISCOVERY_TO_SULKING_STATE: {
|
||||
action = wtp_dfa_state_discovery_to_sulking(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DISCOVERY_TO_DTLS_SETUP_STATE: {
|
||||
action = wtp_dfa_state_discovery_to_dtlssetup(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
|
||||
case CAPWAP_SULKING_STATE: {
|
||||
action = wtp_dfa_state_sulking(packet, timeout);
|
||||
wtp_dfa_state_sulking(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_SULKING_TO_IDLE_STATE: {
|
||||
action = wtp_dfa_state_sulking_to_idle(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_SETUP_STATE: {
|
||||
action = wtp_dfa_state_dtlssetup(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_SETUP_TO_IDLE_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_SETUP_TO_SULKING_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_SETUP_TO_AUTHORIZE_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_AUTHORIZE_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_AUTHORIZE_TO_DTLS_SETUP_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_AUTHORIZE_TO_DTLS_CONNECT_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_AUTHORIZE_TO_DTLS_TEARDOWN_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_CONNECT_STATE: {
|
||||
action = wtp_dfa_state_dtlsconnect(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_CONNECT_TO_DTLS_TEARDOWN_STATE: {
|
||||
action = wtp_dfa_state_dtlsconnect_to_dtlsteardown(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_CONNECT_TO_JOIN_STATE: {
|
||||
action = wtp_dfa_state_dtlsconnect_to_join(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_TEARDOWN_STATE: {
|
||||
action = wtp_dfa_state_dtlsteardown(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_TEARDOWN_TO_IDLE_STATE: {
|
||||
action = wtp_dfa_state_dtlsteardown_to_idle(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DTLS_TEARDOWN_TO_SULKING_STATE: {
|
||||
action = wtp_dfa_state_dtlsteardown_to_sulking(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_JOIN_STATE: {
|
||||
action = wtp_dfa_state_join(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_JOIN_TO_DTLS_TEARDOWN_STATE: {
|
||||
action = wtp_dfa_state_join_to_dtlsteardown(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_JOIN_TO_IMAGE_DATA_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_JOIN_TO_CONFIGURE_STATE: {
|
||||
action = wtp_dfa_state_join_to_configure(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_IMAGE_DATA_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_IMAGE_DATA_TO_RESET_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_IMAGE_DATA_TO_DTLS_TEARDOWN_STATE: {
|
||||
action = wtp_dfa_state_imagedata_to_dtlsteardown(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_CONFIGURE_STATE: {
|
||||
action = wtp_dfa_state_configure(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_CONFIGURE_TO_RESET_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_CONFIGURE_TO_DTLS_TEARDOWN_STATE: {
|
||||
action = wtp_dfa_state_configure_to_dtlsteardown(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_CONFIGURE_TO_DATA_CHECK_STATE: {
|
||||
action = wtp_dfa_state_configure_to_datacheck(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_RESET_STATE: {
|
||||
action = wtp_dfa_state_reset(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_RESET_TO_DTLS_TEARDOWN_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DATA_CHECK_STATE: {
|
||||
action = wtp_dfa_state_datacheck(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DATA_CHECK_TO_DTLS_TEARDOWN_STATE: {
|
||||
action = wtp_dfa_state_datacheck_to_dtlsteardown(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DATA_CHECK_TO_RUN_STATE: {
|
||||
action = wtp_dfa_state_datacheck_to_run(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_RUN_STATE: {
|
||||
action = wtp_dfa_state_run(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_RUN_TO_DTLS_TEARDOWN_STATE: {
|
||||
action = wtp_dfa_state_run_to_dtlsteardown(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_RUN_TO_RESET_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
wtp_teardown_connection(timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DEAD_STATE: {
|
||||
/* Never called with this state */
|
||||
ASSERT(0);
|
||||
case CAPWAP_DTLS_TEARDOWN_STATE: {
|
||||
wtp_dfa_state_dtlsteardown(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_JOIN_STATE: {
|
||||
wtp_dfa_state_join(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_CONFIGURE_STATE: {
|
||||
wtp_dfa_state_configure(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_RESET_STATE: {
|
||||
wtp_dfa_state_reset(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_DATA_CHECK_STATE: {
|
||||
wtp_dfa_state_datacheck(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
case CAPWAP_RUN_STATE: {
|
||||
wtp_dfa_state_run(packet, timeout);
|
||||
break;
|
||||
}
|
||||
|
||||
@ -320,15 +144,12 @@ static int wtp_dfa_execute(struct capwap_parsed_packet* packet, struct timeout_c
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return action;
|
||||
}
|
||||
|
||||
/* WTP state machine */
|
||||
int wtp_dfa_running(void) {
|
||||
int res;
|
||||
int result = CAPWAP_SUCCESSFUL;
|
||||
int action = WTP_DFA_NO_PACKET;
|
||||
struct timeout_control timeout;
|
||||
|
||||
char bufferencrypt[CAPWAP_MAX_PACKET_SIZE];
|
||||
@ -368,211 +189,213 @@ int wtp_dfa_running(void) {
|
||||
signal(SIGINT, wtp_signal_handler);
|
||||
signal(SIGTERM, wtp_signal_handler);
|
||||
|
||||
while (action != WTP_DFA_EXIT) {
|
||||
/* Init complete, start DFA */
|
||||
wtp_dfa_change_state(CAPWAP_IDLE_STATE);
|
||||
wtp_dfa_state_idle(NULL, &timeout);
|
||||
|
||||
/* */
|
||||
while (g_wtp.dfa.state != CAPWAP_DEAD_STATE) {
|
||||
/* If request wait packet from AC */
|
||||
isrecvpacket = 0;
|
||||
if ((action == WTP_DFA_ACCEPT_PACKET) || (action == WTP_DFA_DROP_PACKET)) {
|
||||
buffer = bufferencrypt;
|
||||
buffersize = CAPWAP_MAX_PACKET_SIZE;
|
||||
index = capwap_recvfrom(fds, fdscount, buffer, &buffersize, &recvfromaddr, &recvtoaddr, &timeout);
|
||||
if (!g_wtp.running) {
|
||||
capwap_logging_debug("Closing WTP, Teardown connection");
|
||||
buffer = bufferencrypt;
|
||||
buffersize = CAPWAP_MAX_PACKET_SIZE;
|
||||
index = capwap_recvfrom(fds, fdscount, buffer, &buffersize, &recvfromaddr, &recvtoaddr, &timeout);
|
||||
if (!g_wtp.running) {
|
||||
capwap_logging_debug("Closing WTP, Teardown connection");
|
||||
|
||||
/* Manual teardown */
|
||||
index = CAPWAP_RECV_ERROR_TIMEOUT;
|
||||
wtp_teardown_connection(&timeout);
|
||||
/* Manual teardown */
|
||||
index = CAPWAP_RECV_ERROR_TIMEOUT;
|
||||
wtp_teardown_connection(&timeout);
|
||||
|
||||
/* Wait RFC teardown timeout */
|
||||
capwap_wait_timeout(&timeout, CAPWAP_TIMER_CONTROL_CONNECTION);
|
||||
}
|
||||
/* Wait RFC teardown timeout */
|
||||
capwap_wait_timeout(&timeout, CAPWAP_TIMER_CONTROL_CONNECTION);
|
||||
}
|
||||
|
||||
if (index >= 0) {
|
||||
if ((action == WTP_DFA_DROP_PACKET) || g_wtp.teardown) {
|
||||
/* Drop packet */
|
||||
continue;
|
||||
} else {
|
||||
int check;
|
||||
if (index >= 0) {
|
||||
if (g_wtp.teardown) {
|
||||
/* Drop packet */
|
||||
continue;
|
||||
} else {
|
||||
int check;
|
||||
|
||||
/* Retrieve network information */
|
||||
capwap_get_network_socket(&g_wtp.net, &socket, fds[index].fd);
|
||||
/* Retrieve network information */
|
||||
capwap_get_network_socket(&g_wtp.net, &socket, fds[index].fd);
|
||||
|
||||
/* Check source */
|
||||
if (socket.isctrlsocket && (g_wtp.acctrladdress.ss_family != AF_UNSPEC)) {
|
||||
if (capwap_compare_ip(&g_wtp.acctrladdress, &recvfromaddr)) {
|
||||
/* Unknown source */
|
||||
continue;
|
||||
}
|
||||
} else if (!socket.isctrlsocket && (g_wtp.acdataaddress.ss_family != AF_UNSPEC)) {
|
||||
if (capwap_compare_ip(&g_wtp.acdataaddress, &recvfromaddr)) {
|
||||
/* Unknown source */
|
||||
continue;
|
||||
}
|
||||
/* Check source */
|
||||
if (socket.isctrlsocket && (g_wtp.acctrladdress.ss_family != AF_UNSPEC)) {
|
||||
if (capwap_compare_ip(&g_wtp.acctrladdress, &recvfromaddr)) {
|
||||
/* Unknown source */
|
||||
continue;
|
||||
}
|
||||
} else if (!socket.isctrlsocket && (g_wtp.acdataaddress.ss_family != AF_UNSPEC)) {
|
||||
if (capwap_compare_ip(&g_wtp.acdataaddress, &recvfromaddr)) {
|
||||
/* Unknown source */
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
/* Check of packet */
|
||||
check = capwap_sanity_check(socket.isctrlsocket, g_wtp.dfa.state, buffer, buffersize, g_wtp.ctrldtls.enable, g_wtp.datadtls.enable);
|
||||
if (check == CAPWAP_DTLS_PACKET) {
|
||||
struct capwap_dtls* dtls = (socket.isctrlsocket ? &g_wtp.ctrldtls : &g_wtp.datadtls);
|
||||
/* Check of packet */
|
||||
check = capwap_sanity_check(socket.isctrlsocket, g_wtp.dfa.state, buffer, buffersize, g_wtp.ctrldtls.enable, g_wtp.datadtls.enable);
|
||||
if (check == CAPWAP_DTLS_PACKET) {
|
||||
struct capwap_dtls* dtls = (socket.isctrlsocket ? &g_wtp.ctrldtls : &g_wtp.datadtls);
|
||||
|
||||
if (dtls->enable) {
|
||||
int oldaction = dtls->action;
|
||||
if (dtls->enable) {
|
||||
int oldaction = dtls->action;
|
||||
|
||||
/* Decrypt packet */
|
||||
buffersize = capwap_decrypt_packet(dtls, buffer, buffersize, bufferplain, CAPWAP_MAX_PACKET_SIZE);
|
||||
if (buffersize > 0) {
|
||||
buffer = bufferplain;
|
||||
check = CAPWAP_PLAIN_PACKET;
|
||||
} else if (buffersize == CAPWAP_ERROR_AGAIN) {
|
||||
/* Check is handshake complete */
|
||||
if ((oldaction == CAPWAP_DTLS_ACTION_HANDSHAKE) && (dtls->action == CAPWAP_DTLS_ACTION_DATA)) {
|
||||
if (socket.isctrlsocket) {
|
||||
if (g_wtp.dfa.state == CAPWAP_DTLS_CONNECT_STATE) {
|
||||
check = CAPWAP_NONE_PACKET;
|
||||
wtp_dfa_change_state(CAPWAP_DTLS_CONNECT_TO_JOIN_STATE);
|
||||
action = WTP_DFA_NO_PACKET;
|
||||
} else {
|
||||
/* TODO */ /* Connection error */
|
||||
check = CAPWAP_WRONG_PACKET;
|
||||
}
|
||||
/* Decrypt packet */
|
||||
buffersize = capwap_decrypt_packet(dtls, buffer, buffersize, bufferplain, CAPWAP_MAX_PACKET_SIZE);
|
||||
if (buffersize > 0) {
|
||||
buffer = bufferplain;
|
||||
check = CAPWAP_PLAIN_PACKET;
|
||||
} else if (buffersize == CAPWAP_ERROR_AGAIN) {
|
||||
/* Check is handshake complete */
|
||||
if ((oldaction == CAPWAP_DTLS_ACTION_HANDSHAKE) && (dtls->action == CAPWAP_DTLS_ACTION_DATA)) {
|
||||
if (socket.isctrlsocket) {
|
||||
if (g_wtp.dfa.state == CAPWAP_DTLS_CONNECT_STATE) {
|
||||
check = CAPWAP_NONE_PACKET;
|
||||
wtp_send_join(&timeout);
|
||||
} else {
|
||||
if (g_wtp.dfa.state == CAPWAP_DATA_CHECK_TO_RUN_STATE) {
|
||||
check = CAPWAP_NONE_PACKET;
|
||||
action = WTP_DFA_NO_PACKET;
|
||||
} else {
|
||||
/* TODO */ /* Connection error */
|
||||
check = CAPWAP_WRONG_PACKET;
|
||||
}
|
||||
check = CAPWAP_WRONG_PACKET;
|
||||
wtp_teardown_connection(&timeout);
|
||||
}
|
||||
} else {
|
||||
if (g_wtp.dfa.state == CAPWAP_DATA_CHECK_STATE) {
|
||||
check = CAPWAP_NONE_PACKET;
|
||||
wtp_start_datachannel(&timeout);
|
||||
} else {
|
||||
check = CAPWAP_WRONG_PACKET;
|
||||
wtp_teardown_connection(&timeout);
|
||||
}
|
||||
}
|
||||
|
||||
continue; /* Next packet */
|
||||
} else {
|
||||
if ((oldaction == CAPWAP_DTLS_ACTION_DATA) && (dtls->action == CAPWAP_DTLS_ACTION_SHUTDOWN)) {
|
||||
action = wtp_teardown_connection(&timeout);
|
||||
}
|
||||
|
||||
continue; /* Next packet */
|
||||
}
|
||||
|
||||
continue; /* Next packet */
|
||||
} else {
|
||||
continue; /* Drop packet */
|
||||
if ((oldaction == CAPWAP_DTLS_ACTION_DATA) && (dtls->action == CAPWAP_DTLS_ACTION_SHUTDOWN)) {
|
||||
wtp_teardown_connection(&timeout);
|
||||
}
|
||||
|
||||
continue; /* Next packet */
|
||||
}
|
||||
} else {
|
||||
continue; /* Drop packet */
|
||||
}
|
||||
} else if (check == CAPWAP_WRONG_PACKET) {
|
||||
capwap_logging_debug("Warning: sanity check failure");
|
||||
/* Drop packet */
|
||||
continue;
|
||||
}
|
||||
|
||||
/* */
|
||||
if (check == CAPWAP_PLAIN_PACKET) {
|
||||
struct capwap_packet_rxmng* rxmngpacket;
|
||||
|
||||
/* Detect local address */
|
||||
if (recvtoaddr.ss_family == AF_UNSPEC) {
|
||||
if (capwap_get_localaddress_by_remoteaddress(&recvtoaddr, &recvfromaddr, g_wtp.net.bind_interface, (!(g_wtp.net.bind_ctrl_flags & CAPWAP_IPV6ONLY_FLAG) ? 1 : 0))) {
|
||||
struct sockaddr_storage sockinfo;
|
||||
socklen_t sockinfolen = sizeof(struct sockaddr_storage);
|
||||
|
||||
memset(&sockinfo, 0, sizeof(struct sockaddr_storage));
|
||||
if (getsockname(fds[index].fd, (struct sockaddr*)&sockinfo, &sockinfolen) < 0) {
|
||||
break;
|
||||
}
|
||||
|
||||
CAPWAP_SET_NETWORK_PORT(&recvtoaddr, CAPWAP_GET_NETWORK_PORT(&sockinfo));
|
||||
}
|
||||
} else if (check == CAPWAP_WRONG_PACKET) {
|
||||
capwap_logging_debug("Warning: sanity check failure");
|
||||
/* Drop packet */
|
||||
continue;
|
||||
}
|
||||
|
||||
/* */
|
||||
if (check == CAPWAP_PLAIN_PACKET) {
|
||||
struct capwap_packet_rxmng* rxmngpacket;
|
||||
if (socket.isctrlsocket) {
|
||||
capwap_logging_debug("Receive control packet");
|
||||
} else {
|
||||
capwap_logging_debug("Receive data packet");
|
||||
}
|
||||
|
||||
/* Detect local address */
|
||||
if (recvtoaddr.ss_family == AF_UNSPEC) {
|
||||
if (capwap_get_localaddress_by_remoteaddress(&recvtoaddr, &recvfromaddr, g_wtp.net.bind_interface, (!(g_wtp.net.bind_ctrl_flags & CAPWAP_IPV6ONLY_FLAG) ? 1 : 0))) {
|
||||
struct sockaddr_storage sockinfo;
|
||||
socklen_t sockinfolen = sizeof(struct sockaddr_storage);
|
||||
/* Defragment management */
|
||||
rxmngpacket = wtp_get_packet_rxmng(socket.isctrlsocket);
|
||||
|
||||
memset(&sockinfo, 0, sizeof(struct sockaddr_storage));
|
||||
if (getsockname(fds[index].fd, (struct sockaddr*)&sockinfo, &sockinfolen) < 0) {
|
||||
break;
|
||||
}
|
||||
/* If request, defragmentation packet */
|
||||
check = capwap_packet_rxmng_add_recv_packet(rxmngpacket, buffer, buffersize);
|
||||
if (check == CAPWAP_REQUEST_MORE_FRAGMENT) {
|
||||
continue;
|
||||
} else if (check != CAPWAP_RECEIVE_COMPLETE_PACKET) {
|
||||
/* Discard fragments */
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
continue;
|
||||
}
|
||||
|
||||
CAPWAP_SET_NETWORK_PORT(&recvtoaddr, CAPWAP_GET_NETWORK_PORT(&sockinfo));
|
||||
/* Receive all fragment */
|
||||
memcpy(&connection.socket, &socket, sizeof(struct capwap_socket));
|
||||
memcpy(&connection.localaddr, &recvtoaddr, sizeof(struct sockaddr_storage));
|
||||
memcpy(&connection.remoteaddr, &recvfromaddr, sizeof(struct sockaddr_storage));
|
||||
|
||||
/* Check for already response to packet */
|
||||
if (socket.isctrlsocket) {
|
||||
if (capwap_recv_retrasmitted_request(&g_wtp.ctrldtls, rxmngpacket, &connection, g_wtp.lastrecvpackethash, g_wtp.responsefragmentpacket)) {
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
capwap_logging_debug("Retrasmitted packet");
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Check message type */
|
||||
res = capwap_check_message_type(rxmngpacket);
|
||||
if (res != VALID_MESSAGE_TYPE) {
|
||||
if (res == INVALID_REQUEST_MESSAGE_TYPE) {
|
||||
capwap_logging_warning("Unexpected Unrecognized Request, send Response Packet with error");
|
||||
wtp_send_invalid_request(rxmngpacket, &connection, CAPWAP_RESULTCODE_MSG_UNEXPECTED_UNRECOGNIZED_REQUEST);
|
||||
}
|
||||
|
||||
capwap_logging_debug("Invalid message type");
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
/* Parsing packet */
|
||||
res = capwap_parsing_packet(rxmngpacket, &connection, &packet);
|
||||
if (res != PARSING_COMPLETE) {
|
||||
if (socket.isctrlsocket && (res == UNRECOGNIZED_MESSAGE_ELEMENT) && capwap_is_request_type(rxmngpacket->ctrlmsg.type)) {
|
||||
capwap_logging_warning("Unrecognized Message Element, send Response Packet with error");
|
||||
wtp_send_invalid_request(rxmngpacket, &connection, CAPWAP_RESULTCODE_FAILURE_UNRECOGNIZED_MESSAGE_ELEMENT);
|
||||
/* TODO: add the unrecognized message element */
|
||||
}
|
||||
|
||||
/* */
|
||||
if (socket.isctrlsocket) {
|
||||
capwap_logging_debug("Receive control packet");
|
||||
} else {
|
||||
capwap_logging_debug("Receive data packet");
|
||||
}
|
||||
|
||||
/* Defragment management */
|
||||
rxmngpacket = wtp_get_packet_rxmng(socket.isctrlsocket);
|
||||
|
||||
/* If request, defragmentation packet */
|
||||
check = capwap_packet_rxmng_add_recv_packet(rxmngpacket, buffer, buffersize);
|
||||
if (check == CAPWAP_REQUEST_MORE_FRAGMENT) {
|
||||
continue;
|
||||
} else if (check != CAPWAP_RECEIVE_COMPLETE_PACKET) {
|
||||
/* Discard fragments */
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Receive all fragment */
|
||||
memcpy(&connection.socket, &socket, sizeof(struct capwap_socket));
|
||||
memcpy(&connection.localaddr, &recvtoaddr, sizeof(struct sockaddr_storage));
|
||||
memcpy(&connection.remoteaddr, &recvfromaddr, sizeof(struct sockaddr_storage));
|
||||
|
||||
/* Check for already response to packet */
|
||||
if (socket.isctrlsocket) {
|
||||
if (capwap_recv_retrasmitted_request(&g_wtp.ctrldtls, rxmngpacket, &connection, g_wtp.lastrecvpackethash, g_wtp.responsefragmentpacket)) {
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
capwap_logging_debug("Retrasmitted packet");
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Check message type */
|
||||
res = capwap_check_message_type(rxmngpacket);
|
||||
if (res != VALID_MESSAGE_TYPE) {
|
||||
if (res == INVALID_REQUEST_MESSAGE_TYPE) {
|
||||
capwap_logging_warning("Unexpected Unrecognized Request, send Response Packet with error");
|
||||
wtp_send_invalid_request(rxmngpacket, &connection, CAPWAP_RESULTCODE_MSG_UNEXPECTED_UNRECOGNIZED_REQUEST);
|
||||
}
|
||||
|
||||
capwap_logging_debug("Invalid message type");
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
/* Parsing packet */
|
||||
res = capwap_parsing_packet(rxmngpacket, &connection, &packet);
|
||||
if (res != PARSING_COMPLETE) {
|
||||
if (socket.isctrlsocket && (res == UNRECOGNIZED_MESSAGE_ELEMENT) && capwap_is_request_type(rxmngpacket->ctrlmsg.type)) {
|
||||
capwap_logging_warning("Unrecognized Message Element, send Response Packet with error");
|
||||
wtp_send_invalid_request(rxmngpacket, &connection, CAPWAP_RESULTCODE_FAILURE_UNRECOGNIZED_MESSAGE_ELEMENT);
|
||||
/* TODO: add the unrecognized message element */
|
||||
}
|
||||
|
||||
/* */
|
||||
capwap_free_parsed_packet(&packet);
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
capwap_logging_debug("Failed parsing packet");
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Validate packet */
|
||||
if (capwap_validate_parsed_packet(&packet, NULL)) {
|
||||
if (socket.isctrlsocket && capwap_is_request_type(rxmngpacket->ctrlmsg.type)) {
|
||||
capwap_logging_warning("Missing Mandatory Message Element, send Response Packet with error");
|
||||
wtp_send_invalid_request(rxmngpacket, &connection, CAPWAP_RESULTCODE_FAILURE_MISSING_MANDATORY_MSG_ELEMENT);
|
||||
}
|
||||
|
||||
/* */
|
||||
capwap_free_parsed_packet(&packet);
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
capwap_logging_debug("Failed validation parsed packet");
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Receive a complete packet */
|
||||
isrecvpacket = 1;
|
||||
capwap_free_parsed_packet(&packet);
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
capwap_logging_debug("Failed parsing packet");
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Validate packet */
|
||||
if (capwap_validate_parsed_packet(&packet, NULL)) {
|
||||
if (socket.isctrlsocket && capwap_is_request_type(rxmngpacket->ctrlmsg.type)) {
|
||||
capwap_logging_warning("Missing Mandatory Message Element, send Response Packet with error");
|
||||
wtp_send_invalid_request(rxmngpacket, &connection, CAPWAP_RESULTCODE_FAILURE_MISSING_MANDATORY_MSG_ELEMENT);
|
||||
}
|
||||
|
||||
/* */
|
||||
capwap_free_parsed_packet(&packet);
|
||||
wtp_free_packet_rxmng(socket.isctrlsocket);
|
||||
capwap_logging_debug("Failed validation parsed packet");
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Receive a complete packet */
|
||||
isrecvpacket = 1;
|
||||
}
|
||||
} else if (index == CAPWAP_RECV_ERROR_INTR) {
|
||||
/* Ignore recv */
|
||||
continue;
|
||||
} else if (index == CAPWAP_RECV_ERROR_SOCKET) {
|
||||
/* Socket close */
|
||||
break;
|
||||
}
|
||||
} else if (index == CAPWAP_RECV_ERROR_INTR) {
|
||||
/* Ignore recv */
|
||||
continue;
|
||||
} else if (index == CAPWAP_RECV_ERROR_SOCKET) {
|
||||
/* Socket close */
|
||||
break;
|
||||
}
|
||||
|
||||
/* Execute state */
|
||||
action = wtp_dfa_execute((isrecvpacket ? &packet : NULL), &timeout);
|
||||
wtp_dfa_execute((isrecvpacket ? &packet : NULL), &timeout);
|
||||
|
||||
/* Free memory */
|
||||
capwap_free_parsed_packet(&packet);
|
||||
|
||||
Reference in New Issue
Block a user