96 lines
3.5 KiB
C
96 lines
3.5 KiB
C
#include "wtp.h"
|
|
#include "capwap_dfa.h"
|
|
#include "wtp_dfa.h"
|
|
|
|
/* */
|
|
int wtp_dfa_state_idle(struct capwap_parsed_packet* packet, struct timeout_control* timeout) {
|
|
ASSERT(timeout != NULL);
|
|
ASSERT(packet == NULL);
|
|
|
|
/* Remove teardown flag */
|
|
g_wtp.teardown = 0;
|
|
|
|
if (!g_wtp.acdiscoveryrequest && (g_wtp.acpreferedarray->count > 0)) {
|
|
/* Found in configuration file the AC address */
|
|
memcpy(&g_wtp.acctrladdress, capwap_array_get_item_pointer(g_wtp.acpreferedarray, g_wtp.acpreferedselected), sizeof(struct sockaddr_storage));
|
|
memcpy(&g_wtp.acdataaddress, &g_wtp.acctrladdress, sizeof(struct sockaddr_storage));
|
|
CAPWAP_SET_NETWORK_PORT(&g_wtp.acdataaddress, CAPWAP_GET_NETWORK_PORT(&g_wtp.acdataaddress) + 1);
|
|
|
|
/* Configure socket */
|
|
capwap_get_network_socket(&g_wtp.net, &g_wtp.acctrlsock, capwap_get_socket(&g_wtp.net, g_wtp.acctrladdress.ss_family, IPPROTO_UDP, CAPWAP_CTRL_SOCKET));
|
|
capwap_get_network_socket(&g_wtp.net, &g_wtp.acdatasock, capwap_get_socket(&g_wtp.net, g_wtp.acdataaddress.ss_family, (g_wtp.transport.type == CAPWAP_UDP_TRANSPORT ? IPPROTO_UDP : IPPROTO_UDPLITE), CAPWAP_DATA_SOCKET));
|
|
|
|
/* */
|
|
g_wtp.acpreferedselected = (g_wtp.acpreferedselected + 1) % g_wtp.acpreferedarray->count;
|
|
|
|
/* Connect */
|
|
wtp_dfa_change_state(CAPWAP_IDLE_TO_DTLS_SETUP_STATE);
|
|
} else {
|
|
/* Search AC */
|
|
wtp_dfa_change_state(CAPWAP_IDLE_TO_DISCOVERY_STATE);
|
|
}
|
|
|
|
capwap_kill_timeout(timeout, CAPWAP_TIMER_CONTROL_CONNECTION);
|
|
return WTP_DFA_NO_PACKET;
|
|
}
|
|
|
|
/* Prepare to discovery AC */
|
|
int wtp_dfa_state_idle_to_discovery(struct capwap_parsed_packet* packet, struct timeout_control* timeout) {
|
|
ASSERT(timeout != NULL);
|
|
ASSERT(packet == NULL);
|
|
|
|
/* Set discovery interval */
|
|
g_wtp.dfa.rfcDiscoveryInterval = capwap_get_rand(g_wtp.dfa.rfcMaxDiscoveryInterval - WTP_MIN_DISCOVERY_INTERVAL) + WTP_MIN_DISCOVERY_INTERVAL;
|
|
g_wtp.dfa.rfcDiscoveryCount = 0;
|
|
|
|
/* Change state */
|
|
wtp_dfa_change_state(CAPWAP_DISCOVERY_STATE);
|
|
|
|
/* Wait before send Discovery Request */
|
|
capwap_set_timeout(g_wtp.dfa.rfcDiscoveryInterval, timeout, CAPWAP_TIMER_CONTROL_CONNECTION);
|
|
|
|
return WTP_DFA_DROP_PACKET;
|
|
}
|
|
|
|
/* Prepare to connect with AC */
|
|
int wtp_dfa_state_idle_to_dtlssetup(struct capwap_parsed_packet* packet, struct timeout_control* timeout) {
|
|
int status = WTP_DFA_ACCEPT_PACKET;
|
|
|
|
ASSERT(timeout != NULL);
|
|
ASSERT(packet == NULL);
|
|
|
|
/* Retrieve local address */
|
|
if (!capwap_get_localaddress_by_remoteaddress(&g_wtp.wtpctrladdress, &g_wtp.acctrladdress, g_wtp.net.bind_interface, (!(g_wtp.net.bind_ctrl_flags & CAPWAP_IPV6ONLY_FLAG) ? 1 : 0))) {
|
|
wtp_dfa_change_state(CAPWAP_IDLE_STATE);
|
|
status = WTP_DFA_NO_PACKET;
|
|
} else {
|
|
struct sockaddr_storage sockinfo;
|
|
socklen_t sockinfolen = sizeof(struct sockaddr_storage);
|
|
|
|
memset(&sockinfo, 0, sizeof(struct sockaddr_storage));
|
|
if (getsockname(g_wtp.acctrlsock.socket[g_wtp.acctrlsock.type], (struct sockaddr*)&sockinfo, &sockinfolen) < 0) {
|
|
wtp_dfa_change_state(CAPWAP_IDLE_STATE);
|
|
status = WTP_DFA_NO_PACKET;
|
|
} else {
|
|
CAPWAP_SET_NETWORK_PORT(&g_wtp.wtpctrladdress, CAPWAP_GET_NETWORK_PORT(&sockinfo));
|
|
|
|
/* */
|
|
memcpy(&g_wtp.wtpdataaddress, &g_wtp.wtpctrladdress, sizeof(struct sockaddr_storage));
|
|
CAPWAP_SET_NETWORK_PORT(&g_wtp.wtpdataaddress, CAPWAP_GET_NETWORK_PORT(&g_wtp.wtpdataaddress) + 1);
|
|
|
|
/* */
|
|
if (!g_wtp.enabledtls) {
|
|
/* Bypass DTLS connection */
|
|
wtp_dfa_change_state(CAPWAP_DTLS_CONNECT_TO_JOIN_STATE);
|
|
status = WTP_DFA_NO_PACKET;
|
|
} else {
|
|
/* Create DTLS connection */
|
|
wtp_dfa_change_state(CAPWAP_DTLS_SETUP_STATE);
|
|
status = WTP_DFA_NO_PACKET;
|
|
}
|
|
}
|
|
}
|
|
|
|
return status;
|
|
}
|