RTL8710_SDK_GCC_VERSION/component/common/api/wifi/wifi_conf.c
RtlduinoMan 905d81784e GCC SDK RTL8710 basic version (including the window platform cygwin installation and Ubuntu platform Linux Installation routines),
including cross compilation of the installation, compile, link, run, debug, and so on.
SDK implementation of the function:
1, WiFi connection settings (including AP mode and STA mode).
2, peripheral resource control (including GPIO, SPI, UART, IIC, etc.).
3, the user uses the sample method.
2016-09-08 18:11:26 +08:00

1859 lines
53 KiB
C

//----------------------------------------------------------------------------//
#include "FreeRTOS.h"
#include "task.h"
#include "semphr.h"
#include <lwip_netconf.h>
#include "main.h"
#include <dhcp/dhcps.h>
//#include <flash/stm32_flash.h>
#include <platform/platform_stdlib.h>
#include <wifi/wifi_conf.h>
#include <wifi/wifi_util.h>
#include <wifi/wifi_ind.h>
#include "tcpip.h"
#include <osdep_service.h>
#if CONFIG_EXAMPLE_WLAN_FAST_CONNECT
#include "wlan_fast_connect/example_wlan_fast_connect.h"
#endif
#if CONFIG_EXAMPLE_UART_ATCMD
#include "at_cmd/atcmd_wifi.h"
#endif
#if CONFIG_INIC_EN
extern int inic_start(void);
extern int inic_stop(void);
#endif
/******************************************************
* Constants
******************************************************/
#define SCAN_USE_SEMAPHORE 0
#define RTW_JOIN_TIMEOUT 15000
#define JOIN_ASSOCIATED (uint32_t)(1 << 0)
#define JOIN_AUTHENTICATED (uint32_t)(1 << 1)
#define JOIN_LINK_READY (uint32_t)(1 << 2)
#define JOIN_SECURITY_COMPLETE (uint32_t)(1 << 3)
#define JOIN_COMPLETE (uint32_t)(1 << 4)
#define JOIN_NO_NETWORKS (uint32_t)(1 << 5)
#define JOIN_WRONG_SECURITY (uint32_t)(1 << 6)
#define JOIN_HANDSHAKE_DONE (uint32_t)(1 << 7)
#define JOIN_SIMPLE_CONFIG (uint32_t)(1 << 8)
#define JOIN_AIRKISS (uint32_t)(1 << 9)
/******************************************************
* Type Definitions
******************************************************/
/******************************************************
* Variables Declarations
******************************************************/
extern struct netif xnetif[NET_IF_NUM];
/******************************************************
* Variables Definitions
******************************************************/
static internal_scan_handler_t scan_result_handler_ptr = {0, 0, 0, RTW_FALSE, 0, 0, 0, 0, 0};
static internal_join_result_t* join_user_data;
extern rtw_mode_t wifi_mode;
int error_flag = RTW_UNKNOWN;
uint32_t rtw_join_status;
#if ATCMD_VER == ATVER_2
extern unsigned char dhcp_mode_sta;
#endif
/******************************************************
* Variables Definitions
******************************************************/
#ifndef WLAN0_NAME
#define WLAN0_NAME "wlan0"
#endif
#ifndef WLAN1_NAME
#define WLAN1_NAME "wlan1"
#endif
/* Give default value if not defined */
#ifndef NET_IF_NUM
#ifdef CONFIG_CONCURRENT_MODE
#define NET_IF_NUM 2
#else
#define NET_IF_NUM 1
#endif
#endif
/*Static IP ADDRESS*/
#ifndef IP_ADDR0
#define IP_ADDR0 192
#define IP_ADDR1 168
#define IP_ADDR2 1
#define IP_ADDR3 80
#endif
/*NETMASK*/
#ifndef NETMASK_ADDR0
#define NETMASK_ADDR0 255
#define NETMASK_ADDR1 255
#define NETMASK_ADDR2 255
#define NETMASK_ADDR3 0
#endif
/*Gateway Address*/
#ifndef GW_ADDR0
#define GW_ADDR0 192
#define GW_ADDR1 168
#define GW_ADDR2 1
#define GW_ADDR3 1
#endif
/*Static IP ADDRESS*/
#ifndef AP_IP_ADDR0
#define AP_IP_ADDR0 192
#define AP_IP_ADDR1 168
#define AP_IP_ADDR2 43
#define AP_IP_ADDR3 1
#endif
/*NETMASK*/
#ifndef AP_NETMASK_ADDR0
#define AP_NETMASK_ADDR0 255
#define AP_NETMASK_ADDR1 255
#define AP_NETMASK_ADDR2 255
#define AP_NETMASK_ADDR3 0
#endif
/*Gateway Address*/
#ifndef AP_GW_ADDR0
#define AP_GW_ADDR0 192
#define AP_GW_ADDR1 168
#define AP_GW_ADDR2 43
#define AP_GW_ADDR3 1
#endif
/******************************************************
* Function Definitions
******************************************************/
#if CONFIG_WLAN
//----------------------------------------------------------------------------//
static int wifi_connect_local(rtw_network_info_t *pWifi)
{
int ret = 0;
if(is_promisc_enabled())
promisc_set(0, NULL, 0);
if(!pWifi) return -1;
switch(pWifi->security_type){
case RTW_SECURITY_OPEN:
ret = wext_set_key_ext(WLAN0_NAME, IW_ENCODE_ALG_NONE, NULL, 0, 0, 0, 0, NULL, 0);
break;
case RTW_SECURITY_WEP_PSK:
case RTW_SECURITY_WEP_SHARED:
ret = wext_set_auth_param(WLAN0_NAME, IW_AUTH_80211_AUTH_ALG, IW_AUTH_ALG_SHARED_KEY);
if(ret == 0)
ret = wext_set_key_ext(WLAN0_NAME, IW_ENCODE_ALG_WEP, NULL, pWifi->key_id, 1 /* set tx key */, 0, 0, pWifi->password, pWifi->password_len);
break;
case RTW_SECURITY_WPA_TKIP_PSK:
case RTW_SECURITY_WPA2_TKIP_PSK:
ret = wext_set_auth_param(WLAN0_NAME, IW_AUTH_80211_AUTH_ALG, IW_AUTH_ALG_OPEN_SYSTEM);
if(ret == 0)
ret = wext_set_key_ext(WLAN0_NAME, IW_ENCODE_ALG_TKIP, NULL, 0, 0, 0, 0, NULL, 0);
if(ret == 0)
ret = wext_set_passphrase(WLAN0_NAME, pWifi->password, pWifi->password_len);
break;
case RTW_SECURITY_WPA_AES_PSK:
case RTW_SECURITY_WPA2_AES_PSK:
case RTW_SECURITY_WPA2_MIXED_PSK:
case RTW_SECURITY_WPA_WPA2_MIXED:
ret = wext_set_auth_param(WLAN0_NAME, IW_AUTH_80211_AUTH_ALG, IW_AUTH_ALG_OPEN_SYSTEM);
if(ret == 0)
ret = wext_set_key_ext(WLAN0_NAME, IW_ENCODE_ALG_CCMP, NULL, 0, 0, 0, 0, NULL, 0);
if(ret == 0)
ret = wext_set_passphrase(WLAN0_NAME, pWifi->password, pWifi->password_len);
break;
default:
ret = -1;
printf("\n\rWIFICONF: security type(0x%x) is not supported.\n\r", pWifi->security_type);
break;
}
if(ret == 0)
ret = wext_set_ssid(WLAN0_NAME, pWifi->ssid.val, pWifi->ssid.len);
return ret;
}
static int wifi_connect_bssid_local(rtw_network_info_t *pWifi)
{
int ret = 0;
u8 bssid[12] = {0};
if(is_promisc_enabled())
promisc_set(0, NULL, 0);
if(!pWifi) return -1;
switch(pWifi->security_type){
case RTW_SECURITY_OPEN:
ret = wext_set_key_ext(WLAN0_NAME, IW_ENCODE_ALG_NONE, NULL, 0, 0, 0, 0, NULL, 0);
break;
case RTW_SECURITY_WEP_PSK:
case RTW_SECURITY_WEP_SHARED:
ret = wext_set_auth_param(WLAN0_NAME, IW_AUTH_80211_AUTH_ALG, IW_AUTH_ALG_SHARED_KEY);
if(ret == 0)
ret = wext_set_key_ext(WLAN0_NAME, IW_ENCODE_ALG_WEP, NULL, pWifi->key_id, 1 /* set tx key */, 0, 0, pWifi->password, pWifi->password_len);
break;
case RTW_SECURITY_WPA_TKIP_PSK:
case RTW_SECURITY_WPA2_TKIP_PSK:
ret = wext_set_auth_param(WLAN0_NAME, IW_AUTH_80211_AUTH_ALG, IW_AUTH_ALG_OPEN_SYSTEM);
if(ret == 0)
ret = wext_set_key_ext(WLAN0_NAME, IW_ENCODE_ALG_TKIP, NULL, 0, 0, 0, 0, NULL, 0);
if(ret == 0)
ret = wext_set_passphrase(WLAN0_NAME, pWifi->password, pWifi->password_len);
break;
case RTW_SECURITY_WPA_AES_PSK:
case RTW_SECURITY_WPA2_AES_PSK:
case RTW_SECURITY_WPA2_MIXED_PSK:
ret = wext_set_auth_param(WLAN0_NAME, IW_AUTH_80211_AUTH_ALG, IW_AUTH_ALG_OPEN_SYSTEM);
if(ret == 0)
ret = wext_set_key_ext(WLAN0_NAME, IW_ENCODE_ALG_CCMP, NULL, 0, 0, 0, 0, NULL, 0);
if(ret == 0)
ret = wext_set_passphrase(WLAN0_NAME, pWifi->password, pWifi->password_len);
break;
default:
ret = -1;
printf("\n\rWIFICONF: security type(0x%x) is not supported.\n\r", pWifi->security_type);
break;
}
if(ret == 0){
memcpy(bssid, pWifi->bssid.octet, ETH_ALEN);
if(pWifi->ssid.len){
bssid[ETH_ALEN] = '#';
bssid[ETH_ALEN + 1] = '@';
memcpy(bssid + ETH_ALEN + 2, &pWifi, sizeof(pWifi));
}
ret = wext_set_bssid(WLAN0_NAME, bssid);
}
return ret;
}
void wifi_rx_beacon_hdl( char* buf, int buf_len, int flags, void* userdata) {
//printf("Beacon!\n");
}
static void wifi_no_network_hdl(char* buf, int buf_len, int flags, void* userdata)
{
if(join_user_data!=NULL)
rtw_join_status = JOIN_NO_NETWORKS;
}
static void wifi_connected_hdl( char* buf, int buf_len, int flags, void* userdata)
{
#ifdef CONFIG_ENABLE_EAP
if(get_eap_phase()){
rtw_join_status = JOIN_COMPLETE | JOIN_SECURITY_COMPLETE | JOIN_ASSOCIATED | JOIN_AUTHENTICATED | JOIN_LINK_READY;
return;
}
#endif /* CONFIG_ENABLE_EAP */
if((join_user_data!=NULL)&&((join_user_data->network_info.security_type == RTW_SECURITY_OPEN) ||
(join_user_data->network_info.security_type == RTW_SECURITY_WEP_PSK))){
rtw_join_status = JOIN_COMPLETE | JOIN_SECURITY_COMPLETE | JOIN_ASSOCIATED | JOIN_AUTHENTICATED | JOIN_LINK_READY;
rtw_up_sema(&join_user_data->join_sema);
}else if((join_user_data!=NULL)&&((join_user_data->network_info.security_type == RTW_SECURITY_WPA2_AES_PSK) )){
rtw_join_status = JOIN_COMPLETE | JOIN_SECURITY_COMPLETE | JOIN_ASSOCIATED | JOIN_AUTHENTICATED | JOIN_LINK_READY;
}
}
static void wifi_handshake_done_hdl( char* buf, int buf_len, int flags, void* userdata)
{
rtw_join_status = JOIN_COMPLETE | JOIN_SECURITY_COMPLETE | JOIN_ASSOCIATED | JOIN_AUTHENTICATED | JOIN_LINK_READY|JOIN_HANDSHAKE_DONE;
if(join_user_data != NULL)
rtw_up_sema(&join_user_data->join_sema);
}
static void wifi_disconn_hdl( char* buf, int buf_len, int flags, void* userdata)
{
if(join_user_data != NULL){
if(join_user_data->network_info.security_type == RTW_SECURITY_OPEN){
if(rtw_join_status == JOIN_NO_NETWORKS)
error_flag = RTW_NONE_NETWORK;
}else if(join_user_data->network_info.security_type == RTW_SECURITY_WEP_PSK){
if(rtw_join_status == JOIN_NO_NETWORKS)
error_flag = RTW_NONE_NETWORK;
else if(rtw_join_status == 0)
error_flag = RTW_CONNECT_FAIL;
}else if(join_user_data->network_info.security_type == RTW_SECURITY_WPA2_AES_PSK){
if(rtw_join_status ==JOIN_NO_NETWORKS)
error_flag = RTW_NONE_NETWORK;
else if(rtw_join_status == 0)
error_flag = RTW_CONNECT_FAIL;
else if(rtw_join_status ==JOIN_COMPLETE | JOIN_SECURITY_COMPLETE | JOIN_ASSOCIATED | JOIN_AUTHENTICATED | JOIN_LINK_READY)
error_flag = RTW_WRONG_PASSWORD;
}
}else{
if(error_flag == RTW_NO_ERROR) //wifi_disconn_hdl will be dispatched one more time after join_user_data = NULL add by frankie
error_flag = RTW_UNKNOWN;
}
if(join_user_data != NULL)
rtw_up_sema(&join_user_data->join_sema);
//printf("\r\nWiFi Disconnect. Error flag is %d.\n", error_flag);
}
#if CONFIG_EXAMPLE_WLAN_FAST_CONNECT
#define WLAN0_NAME "wlan0"
void restore_wifi_info_to_flash()
{
struct wlan_fast_reconnect * data_to_flash;
u32 channel = 0;
u8 index = 0;
u8 *ifname[1] = {WLAN0_NAME};
rtw_wifi_setting_t setting;
//struct security_priv *psecuritypriv = &padapter->securitypriv;
//WLAN_BSSID_EX *pcur_bss = pmlmepriv->cur_network.network;
data_to_flash = (struct wlan_fast_reconnect *)rtw_zmalloc(sizeof(struct wlan_fast_reconnect));
if(data_to_flash && p_write_reconnect_ptr){
if(wifi_get_setting((const char*)ifname[0],&setting) || setting.mode == RTW_MODE_AP){
printf("\r\n %s():wifi_get_setting fail or ap mode", __func__);
return;
}
channel = setting.channel;
rtw_memset(psk_essid[index], 0, sizeof(psk_essid[index]));
strncpy(psk_essid[index], setting.ssid, strlen(setting.ssid));
switch(setting.security_type){
case RTW_SECURITY_OPEN:
rtw_memset(psk_passphrase[index], 0, sizeof(psk_passphrase[index]));
rtw_memset(wpa_global_PSK[index], 0, sizeof(wpa_global_PSK[index]));
data_to_flash->security_type = RTW_SECURITY_OPEN;
break;
case RTW_SECURITY_WEP_PSK:
channel |= (setting.key_idx) << 28;
rtw_memset(psk_passphrase[index], 0, sizeof(psk_passphrase[index]));
rtw_memset(wpa_global_PSK[index], 0, sizeof(wpa_global_PSK[index]));
rtw_memcpy(psk_passphrase[index], setting.password, sizeof(psk_passphrase[index]));
data_to_flash->security_type = RTW_SECURITY_WEP_PSK;
break;
case RTW_SECURITY_WPA_TKIP_PSK:
data_to_flash->security_type = RTW_SECURITY_WPA_TKIP_PSK;
break;
case RTW_SECURITY_WPA2_AES_PSK:
data_to_flash->security_type = RTW_SECURITY_WPA2_AES_PSK;
break;
default:
break;
}
memcpy(data_to_flash->psk_essid, psk_essid[index], sizeof(data_to_flash->psk_essid));
if (strlen(psk_passphrase64) == 64) {
memcpy(data_to_flash->psk_passphrase, psk_passphrase64, sizeof(data_to_flash->psk_passphrase));
} else {
memcpy(data_to_flash->psk_passphrase, psk_passphrase[index], sizeof(data_to_flash->psk_passphrase));
}
memcpy(data_to_flash->wpa_global_PSK, wpa_global_PSK[index], sizeof(data_to_flash->wpa_global_PSK));
memcpy(&(data_to_flash->channel), &channel, 4);
//call callback function in user program
p_write_reconnect_ptr((u8 *)data_to_flash, sizeof(struct wlan_fast_reconnect));
}
if(data_to_flash)
rtw_free(data_to_flash);
}
#endif
//----------------------------------------------------------------------------//
int wifi_connect(
char *ssid,
rtw_security_t security_type,
char *password,
int ssid_len,
int password_len,
int key_id,
void *semaphore)
{
xSemaphoreHandle join_semaphore;
rtw_result_t result = RTW_SUCCESS;
u8 wep_hex = 0;
u8 wep_pwd[14] = {0};
if(rtw_join_status & JOIN_SIMPLE_CONFIG || rtw_join_status & JOIN_AIRKISS){
return RTW_ERROR;
}
rtw_join_status = 0;//clear for last connect status
error_flag = RTW_UNKNOWN ;//clear for last connect status
if ( ( ( ( password_len > RTW_MAX_PSK_LEN ) ||
( password_len < RTW_MIN_PSK_LEN ) ) &&
( ( security_type == RTW_SECURITY_WPA_TKIP_PSK ) ||
( security_type == RTW_SECURITY_WPA_AES_PSK ) ||
( security_type == RTW_SECURITY_WPA2_AES_PSK ) ||
( security_type == RTW_SECURITY_WPA2_TKIP_PSK ) ||
( security_type == RTW_SECURITY_WPA2_MIXED_PSK ) ) )) {
error_flag = RTW_WRONG_PASSWORD;
return RTW_INVALID_KEY;
}
if ((security_type == RTW_SECURITY_WEP_PSK)||
(security_type ==RTW_SECURITY_WEP_SHARED)) {
if ((password_len != 5) && (password_len != 13) &&
(password_len != 10)&& (password_len != 26)) {
error_flag = RTW_WRONG_PASSWORD;
return RTW_INVALID_KEY;
} else {
if(password_len == 10) {
u32 p[5] = {0};
u8 i = 0;
sscanf((const char*)password, "%02x%02x%02x%02x%02x", &p[0], &p[1], &p[2], &p[3], &p[4]);
for(i=0; i< 5; i++)
wep_pwd[i] = (u8)p[i];
wep_pwd[5] = '\0';
password_len = 5;
wep_hex = 1;
} else if (password_len == 26) {
u32 p[13] = {0};
u8 i = 0;
sscanf((const char*)password, "%02x%02x%02x%02x%02x%02x%02x"\
"%02x%02x%02x%02x%02x%02x", &p[0], &p[1], &p[2], &p[3], &p[4],\
&p[5], &p[6], &p[7], &p[8], &p[9], &p[10], &p[11], &p[12]);
for(i=0; i< 13; i++)
wep_pwd[i] = (u8)p[i];
wep_pwd[13] = '\0';
password_len = 13;
wep_hex = 1;
}
}
}
internal_join_result_t *join_result = (internal_join_result_t *)rtw_zmalloc(sizeof(internal_join_result_t));
if(!join_result) {
return RTW_NOMEM;
}
join_result->network_info.ssid.len = ssid_len > 32 ? 32 : ssid_len;
rtw_memcpy(join_result->network_info.ssid.val, ssid, ssid_len);
join_result->network_info.password_len = password_len;
if(password_len) {
/* add \0 to the end */
join_result->network_info.password = rtw_zmalloc(password_len + 1);
if(!join_result->network_info.password) {
result = RTW_NOMEM;
goto error;
}
if (0 == wep_hex)
rtw_memcpy(join_result->network_info.password, password, password_len);
else
rtw_memcpy(join_result->network_info.password, wep_pwd, password_len);
}
join_result->network_info.security_type = security_type;
join_result->network_info.key_id = key_id;
if(semaphore == NULL) {
rtw_init_sema( &join_result->join_sema, 0 );
if(!join_result->join_sema){
result = RTW_NORESOURCE;
goto error;
}
join_semaphore = join_result->join_sema;
} else {
join_result->join_sema = semaphore;
}
wifi_reg_event_handler(WIFI_EVENT_NO_NETWORK,wifi_no_network_hdl,NULL);
wifi_reg_event_handler(WIFI_EVENT_CONNECT, wifi_connected_hdl, NULL);
wifi_reg_event_handler(WIFI_EVENT_DISCONNECT, wifi_disconn_hdl, NULL);
wifi_reg_event_handler(WIFI_EVENT_FOURWAY_HANDSHAKE_DONE, wifi_handshake_done_hdl, NULL);
wifi_connect_local(&join_result->network_info);
join_user_data = join_result;
if(semaphore == NULL) {
// for eap connection, timeout should be longer (default value in wpa_supplicant: 60s)
#ifdef CONFIG_ENABLE_EAP
if(get_eap_phase()){
if(rtw_down_timeout_sema( &join_result->join_sema, 60000 ) == RTW_FALSE) {
printf("RTW API: Join bss timeout\r\n");
if(password_len) {
rtw_free(join_result->network_info.password);
}
result = RTW_TIMEOUT;
goto error;
} else {
if(wifi_is_connected_to_ap( ) != RTW_SUCCESS) {
result = RTW_ERROR;
goto error;
}
}
}
else
#endif
if(rtw_down_timeout_sema( &join_result->join_sema, RTW_JOIN_TIMEOUT ) == RTW_FALSE) {
printf("RTW API: Join bss timeout\r\n");
if(password_len) {
rtw_free(join_result->network_info.password);
}
result = RTW_TIMEOUT;
goto error;
} else {
if(join_result->network_info.password_len) {
rtw_free(join_result->network_info.password);
}
if(wifi_is_connected_to_ap( ) != RTW_SUCCESS) {
result = RTW_ERROR;
goto error;
}
}
}
result = RTW_SUCCESS;
#if CONFIG_EXAMPLE_WLAN_FAST_CONNECT
restore_wifi_info_to_flash();
#endif
error:
if(semaphore == NULL){
rtw_free_sema( &join_semaphore);
}
join_user_data = NULL;
rtw_free((u8*)join_result);
wifi_unreg_event_handler(WIFI_EVENT_CONNECT, wifi_connected_hdl);
wifi_unreg_event_handler(WIFI_EVENT_NO_NETWORK,wifi_no_network_hdl);
wifi_unreg_event_handler(WIFI_EVENT_FOURWAY_HANDSHAKE_DONE, wifi_handshake_done_hdl);
return result;
}
int wifi_connect_bssid(
unsigned char bssid[ETH_ALEN],
char *ssid,
rtw_security_t security_type,
char *password,
int bssid_len,
int ssid_len,
int password_len,
int key_id,
void *semaphore)
{
xSemaphoreHandle join_semaphore;
rtw_result_t result = RTW_SUCCESS;
if(rtw_join_status & JOIN_SIMPLE_CONFIG || rtw_join_status & JOIN_AIRKISS){
return RTW_ERROR;
}
rtw_join_status = 0;//clear for last connect status
error_flag = RTW_UNKNOWN;//clear for last connect status
internal_join_result_t *join_result = (internal_join_result_t *)rtw_zmalloc(sizeof(internal_join_result_t));
if(!join_result) {
return RTW_NOMEM;
}
if(ssid_len && ssid){
join_result->network_info.ssid.len = ssid_len > 32 ? 32 : ssid_len;
rtw_memcpy(join_result->network_info.ssid.val, ssid, ssid_len);
}
rtw_memcpy(join_result->network_info.bssid.octet, bssid, bssid_len);
if ( ( ( ( password_len > RTW_MAX_PSK_LEN ) ||
( password_len < RTW_MIN_PSK_LEN ) ) &&
( ( security_type == RTW_SECURITY_WPA_TKIP_PSK ) ||
( security_type == RTW_SECURITY_WPA_AES_PSK ) ||
( security_type == RTW_SECURITY_WPA2_AES_PSK ) ||
( security_type == RTW_SECURITY_WPA2_TKIP_PSK ) ||
( security_type == RTW_SECURITY_WPA2_MIXED_PSK ) ) )||
(((password_len != 5)&& (password_len != 13))&&
((security_type == RTW_SECURITY_WEP_PSK)||
(security_type ==RTW_SECURITY_WEP_SHARED ) ))) {
return RTW_INVALID_KEY;
}
join_result->network_info.password_len = password_len;
if(password_len) {
/* add \0 to the end */
join_result->network_info.password = rtw_zmalloc(password_len + 1);
if(!join_result->network_info.password) {
return RTW_NOMEM;
}
rtw_memcpy(join_result->network_info.password, password, password_len);
}
join_result->network_info.security_type = security_type;
join_result->network_info.key_id = key_id;
if(semaphore == NULL) {
rtw_init_sema( &join_result->join_sema, 0 );
if(!join_result->join_sema){
return RTW_NORESOURCE;
}
join_semaphore = join_result->join_sema;
} else {
join_result->join_sema = semaphore;
}
wifi_reg_event_handler(WIFI_EVENT_NO_NETWORK,wifi_no_network_hdl,NULL);
wifi_reg_event_handler(WIFI_EVENT_CONNECT, wifi_connected_hdl, NULL);
wifi_reg_event_handler(WIFI_EVENT_DISCONNECT, wifi_disconn_hdl, NULL);
wifi_reg_event_handler(WIFI_EVENT_FOURWAY_HANDSHAKE_DONE, wifi_handshake_done_hdl, NULL);
wifi_connect_bssid_local(&join_result->network_info);
join_user_data = join_result;
if(semaphore == NULL) {
if(rtw_down_timeout_sema( &join_result->join_sema, RTW_JOIN_TIMEOUT ) == RTW_FALSE) {
printf("RTW API: Join bss timeout\r\n");
if(password_len) {
rtw_free(join_result->network_info.password);
}
rtw_free((u8*)join_result);
rtw_free_sema( &join_semaphore);
result = RTW_TIMEOUT;
goto error;
} else {
rtw_free_sema( &join_semaphore );
if(join_result->network_info.password_len) {
rtw_free(join_result->network_info.password);
}
rtw_free((u8*)join_result);
if( wifi_is_connected_to_ap( ) != RTW_SUCCESS) {
result = RTW_ERROR;
goto error;
}
}
}
result = RTW_SUCCESS;
#if CONFIG_EXAMPLE_WLAN_FAST_CONNECT
restore_wifi_info_to_flash();
#endif
error:
join_user_data = NULL;
wifi_unreg_event_handler(WIFI_EVENT_CONNECT, wifi_connected_hdl);
wifi_unreg_event_handler(WIFI_EVENT_NO_NETWORK,wifi_no_network_hdl);
wifi_unreg_event_handler(WIFI_EVENT_FOURWAY_HANDSHAKE_DONE, wifi_handshake_done_hdl);
return result;
}
int wifi_disconnect(void)
{
int ret = 0;
//set MAC address last byte to 1 since driver will filter the mac with all 0x00 or 0xff
//add extra 2 zero byte for check of #@ in wext_set_bssid()
const __u8 null_bssid[ETH_ALEN + 2] = {0, 0, 0, 0, 0, 1, 0, 0};
if (wext_set_bssid(WLAN0_NAME, null_bssid) < 0){
printf("\n\rWEXT: Failed to set bogus BSSID to disconnect");
ret = -1;
}
return ret;
}
//----------------------------------------------------------------------------//
int wifi_is_connected_to_ap( void )
{
return rltk_wlan_is_connected_to_ap();
}
//----------------------------------------------------------------------------//
int wifi_is_up(rtw_interface_t interface)
{
if(interface == RTW_AP_INTERFACE) {
if(wifi_mode == RTW_MODE_STA_AP) {
return rltk_wlan_running(WLAN1_IDX);
}
}
return rltk_wlan_running(WLAN0_IDX);
}
int wifi_is_ready_to_transceive(rtw_interface_t interface)
{
switch ( interface )
{
case RTW_AP_INTERFACE:
return ( wifi_is_up(interface) == RTW_TRUE ) ? RTW_SUCCESS : RTW_ERROR;
case RTW_STA_INTERFACE:
switch ( error_flag)
{
case RTW_NO_ERROR:
return RTW_SUCCESS;
default:
return RTW_ERROR;
}
default:
return RTW_ERROR;
}
}
//----------------------------------------------------------------------------//
int wifi_set_mac_address(char * mac)
{
char buf[13+17+1];
rtw_memset(buf, 0, sizeof(buf));
snprintf(buf, 13+17, "write_mac %s", mac);
return wext_private_command(WLAN0_NAME, buf, 0);
}
int wifi_get_mac_address(char * mac)
{
int ret = 0;
char buf[32];
rtw_memset(buf, 0, sizeof(buf));
rtw_memcpy(buf, "read_mac", 8);
ret = wext_private_command_with_retval(WLAN0_NAME, buf, buf, 32);
strcpy(mac, buf);
return ret;
}
//----------------------------------------------------------------------------//
int wifi_enable_powersave(void)
{
return wext_enable_powersave(WLAN0_NAME, 1, 1);
}
int wifi_disable_powersave(void)
{
return wext_disable_powersave(WLAN0_NAME);
}
#if 0 //Not ready
//----------------------------------------------------------------------------//
int wifi_get_txpower(int *poweridx)
{
int ret = 0;
char buf[11];
rtw_memset(buf, 0, sizeof(buf));
rtw_memcpy(buf, "txpower", 11);
ret = wext_private_command_with_retval(WLAN0_NAME, buf, buf, 11);
sscanf(buf, "%d", poweridx);
return ret;
}
int wifi_set_txpower(int poweridx)
{
int ret = 0;
char buf[24];
rtw_memset(buf, 0, sizeof(buf));
snprintf(buf, 24, "txpower patha=%d", poweridx);
ret = wext_private_command(WLAN0_NAME, buf, 0);
return ret;
}
#endif
//----------------------------------------------------------------------------//
int wifi_get_associated_client_list(void * client_list_buffer, uint16_t buffer_length)
{
const char * ifname = WLAN0_NAME;
int ret = 0;
char buf[25];
if(wifi_mode == RTW_MODE_STA_AP) {
ifname = WLAN1_NAME;
}
rtw_memset(buf, 0, sizeof(buf));
snprintf(buf, 25, "get_client_list %x", client_list_buffer);
ret = wext_private_command(ifname, buf, 0);
return ret;
}
//----------------------------------------------------------------------------//
int wifi_get_ap_info(rtw_bss_info_t * ap_info, rtw_security_t* security)
{
const char * ifname = WLAN0_NAME;
int ret = 0;
char buf[24];
if(wifi_mode == RTW_MODE_STA_AP) {
ifname = WLAN1_NAME;
}
rtw_memset(buf, 0, sizeof(buf));
snprintf(buf, 24, "get_ap_info %x", ap_info);
ret = wext_private_command(ifname, buf, 0);
snprintf(buf, 24, "get_security");
ret = wext_private_command_with_retval(ifname, buf, buf, 24);
sscanf(buf, "%d", security);
return ret;
}
int wifi_get_drv_ability(uint32_t *ability)
{
return wext_get_drv_ability(WLAN0_NAME, ability);
}
//----------------------------------------------------------------------------//
int wifi_set_country(rtw_country_code_t country_code)
{
int ret;
ret = wext_set_country(WLAN0_NAME, country_code);
return ret;
}
//----------------------------------------------------------------------------//
int wifi_set_channel_plan(uint8_t channel_plan)
{
const char * ifname = WLAN0_NAME;
int ret = 0;
char buf[24];
rtw_memset(buf, 0, sizeof(buf));
snprintf(buf, 24, "set_ch_plan %x", channel_plan);
ret = wext_private_command(ifname, buf, 0);
return ret;
}
//----------------------------------------------------------------------------//
int wifi_get_rssi(int *pRSSI)
{
return wext_get_rssi(WLAN0_NAME, pRSSI);
}
//----------------------------------------------------------------------------//
int wifi_set_channel(int channel)
{
return wext_set_channel(WLAN0_NAME, channel);
}
int wifi_get_channel(int *channel)
{
return wext_get_channel(WLAN0_NAME, (u8*)channel);
}
//----------------------------------------------------------------------------//
int wifi_register_multicast_address(rtw_mac_t *mac)
{
return wext_register_multicast_address(WLAN0_NAME, mac);
}
int wifi_unregister_multicast_address(rtw_mac_t *mac)
{
return wext_unregister_multicast_address(WLAN0_NAME, mac);
}
//----------------------------------------------------------------------------//
void wifi_set_mib(void)
{
// adaptivity
wext_set_adaptivity(RTW_ADAPTIVITY_DISABLE);
}
//----------------------------------------------------------------------------//
int wifi_rf_on(void)
{
int ret;
ret = rltk_wlan_rf_on();
return ret;
}
//----------------------------------------------------------------------------//
int wifi_rf_off(void)
{
int ret;
ret = rltk_wlan_rf_off();
return ret;
}
//----------------------------------------------------------------------------//
int wifi_on(rtw_mode_t mode)
{
int ret = 1;
int timeout = 20;
int idx;
int devnum = 1;
static int event_init = 0;
if(rltk_wlan_running(WLAN0_IDX)) {
printf("\n\rWIFI is already running");
return 1;
}
if(event_init == 0){
init_event_callback_list();
event_init = 1;
}
wifi_mode = mode;
if(mode == RTW_MODE_STA_AP)
devnum = 2;
// set wifi mib
wifi_set_mib();
printf("\n\rInitializing WIFI ...");
for(idx=0;idx<devnum;idx++){
ret = rltk_wlan_init(idx, mode);
if(ret <0)
return ret;
}
for(idx=0;idx<devnum;idx++)
rltk_wlan_start(idx);
while(1) {
if(rltk_wlan_running(devnum-1)) {
printf("\n\rWIFI initialized\n");
/*
* printf("set country code here\n");
* wifi_set_country(RTW_COUNTRY_US);
*/
break;
}
if(timeout == 0) {
printf("\n\rERROR: Init WIFI timeout!");
break;
}
vTaskDelay(1 * configTICK_RATE_HZ);
timeout --;
}
#if CONFIG_INIC_EN
inic_start();
#endif
return ret;
}
int wifi_off(void)
{
int ret = 0;
int timeout = 20;
if((rltk_wlan_running(WLAN0_IDX) == 0) &&
(rltk_wlan_running(WLAN1_IDX) == 0)) {
printf("\n\rWIFI is not running");
return 0;
}
#if CONFIG_LWIP_LAYER
dhcps_deinit();
LwIP_DHCP(0, DHCP_STOP);
#endif
#if defined(CONFIG_ENABLE_WPS_AP) && CONFIG_ENABLE_WPS_AP
if((wifi_mode == RTW_MODE_AP) || (wifi_mode == RTW_MODE_STA_AP))
wpas_wps_deinit();
#endif
printf("\n\rDeinitializing WIFI ...");
rltk_wlan_deinit();
while(1) {
if((rltk_wlan_running(WLAN0_IDX) == 0) &&
(rltk_wlan_running(WLAN1_IDX) == 0)) {
printf("\n\rWIFI deinitialized");
break;
}
if(timeout == 0) {
printf("\n\rERROR: Deinit WIFI timeout!");
break;
}
vTaskDelay(1 * configTICK_RATE_HZ);
timeout --;
}
wifi_mode = RTW_MODE_NONE;
#if CONFIG_INIC_EN
inic_stop();
#endif
return ret;
}
int wifi_set_power_mode(unsigned char ips_mode, unsigned char lps_mode)
{
return wext_enable_powersave(WLAN0_NAME, ips_mode, lps_mode);
}
int wifi_set_tdma_param(unsigned char slot_period, unsigned char rfon_period_len_1, unsigned char rfon_period_len_2, unsigned char rfon_period_len_3)
{
return wext_set_tdma_param(WLAN0_NAME, slot_period, rfon_period_len_1, rfon_period_len_2, rfon_period_len_3);
}
int wifi_set_lps_dtim(unsigned char dtim)
{
return wext_set_lps_dtim(WLAN0_NAME, dtim);
}
int wifi_get_lps_dtim(unsigned char *dtim)
{
return wext_get_lps_dtim(WLAN0_NAME, dtim);
}
//----------------------------------------------------------------------------//
static void wifi_ap_sta_assoc_hdl( char* buf, int buf_len, int flags, void* userdata)
{
//USER TODO
}
static void wifi_ap_sta_disassoc_hdl( char* buf, int buf_len, int flags, void* userdata)
{
//USER TODO
}
int wifi_get_last_error(void)
{
return error_flag;
}
#if defined(CONFIG_ENABLE_WPS_AP) && CONFIG_ENABLE_WPS_AP
int wpas_wps_init(const char* ifname);
#endif
int wifi_start_ap(
char *ssid,
rtw_security_t security_type,
char *password,
int ssid_len,
int password_len,
int channel)
{
const char *ifname = WLAN0_NAME;
int ret = 0;
if(wifi_mode == RTW_MODE_STA_AP) {
ifname = WLAN1_NAME;
}
if(is_promisc_enabled())
promisc_set(0, NULL, 0);
wifi_reg_event_handler(WIFI_EVENT_STA_ASSOC, wifi_ap_sta_assoc_hdl, NULL);
wifi_reg_event_handler(WIFI_EVENT_STA_DISASSOC, wifi_ap_sta_disassoc_hdl, NULL);
ret = wext_set_mode(ifname, IW_MODE_MASTER);
if(ret < 0) goto exit;
ret = wext_set_channel(ifname, channel); //Set channel before starting ap
if(ret < 0) goto exit;
switch(security_type) {
case RTW_SECURITY_OPEN:
break;
case RTW_SECURITY_WPA2_AES_PSK:
ret = wext_set_auth_param(ifname, IW_AUTH_80211_AUTH_ALG, IW_AUTH_ALG_OPEN_SYSTEM);
if(ret == 0)
ret = wext_set_key_ext(ifname, IW_ENCODE_ALG_CCMP, NULL, 0, 0, 0, 0, NULL, 0);
if(ret == 0)
ret = wext_set_passphrase(ifname, (u8*)password, password_len);
break;
default:
ret = -1;
printf("\n\rWIFICONF: security type is not supported");
break;
}
if(ret < 0) goto exit;
ret = wext_set_ap_ssid(ifname, (u8*)ssid, ssid_len);
#if defined(CONFIG_ENABLE_WPS_AP) && CONFIG_ENABLE_WPS_AP
wpas_wps_init(ifname);
#endif
exit:
return ret;
}
int wifi_start_ap_with_hidden_ssid(
char *ssid,
rtw_security_t security_type,
char *password,
int ssid_len,
int password_len,
int channel)
{
const char *ifname = WLAN0_NAME;
int ret = 0;
if(wifi_mode == RTW_MODE_STA_AP) {
ifname = WLAN1_NAME;
}
if(is_promisc_enabled())
promisc_set(0, NULL, 0);
wifi_reg_event_handler(WIFI_EVENT_STA_ASSOC, wifi_ap_sta_assoc_hdl, NULL);
wifi_reg_event_handler(WIFI_EVENT_STA_DISASSOC, wifi_ap_sta_disassoc_hdl, NULL);
ret = wext_set_mode(ifname, IW_MODE_MASTER);
if(ret < 0) goto exit;
ret = wext_set_channel(ifname, channel); //Set channel before starting ap
if(ret < 0) goto exit;
switch(security_type) {
case RTW_SECURITY_OPEN:
break;
case RTW_SECURITY_WPA2_AES_PSK:
ret = wext_set_auth_param(ifname, IW_AUTH_80211_AUTH_ALG, IW_AUTH_ALG_OPEN_SYSTEM);
if(ret == 0)
ret = wext_set_key_ext(ifname, IW_ENCODE_ALG_CCMP, NULL, 0, 0, 0, 0, NULL, 0);
if(ret == 0)
ret = wext_set_passphrase(ifname, (u8*)password, password_len);
break;
default:
ret = -1;
printf("\n\rWIFICONF: security type is not supported");
break;
}
if(ret < 0) goto exit;
ret = set_hidden_ssid(ifname, 1);
if(ret < 0) goto exit;
ret = wext_set_ap_ssid(ifname, (u8*)ssid, ssid_len);
#if defined(CONFIG_ENABLE_WPS_AP) && CONFIG_ENABLE_WPS_AP
wpas_wps_init(ifname);
#endif
exit:
return ret;
}
void wifi_scan_each_report_hdl( char* buf, int buf_len, int flags, void* userdata)
{
int i =0;
int j =0;
int insert_pos = 0;
rtw_scan_result_t** result_ptr = (rtw_scan_result_t**)buf;
rtw_scan_result_t* temp = NULL;
for(i=0; i<scan_result_handler_ptr.scan_cnt; i++){
if(CMP_MAC(scan_result_handler_ptr.pap_details[i]->BSSID.octet, (*result_ptr)->BSSID.octet)){
if((*result_ptr)->signal_strength > scan_result_handler_ptr.pap_details[i]->signal_strength){
temp = scan_result_handler_ptr.pap_details[i];
for(j = i-1; j >= 0; j--){
if(scan_result_handler_ptr.pap_details[j]->signal_strength >= (*result_ptr)->signal_strength)
break;
else
scan_result_handler_ptr.pap_details[j+1] = scan_result_handler_ptr.pap_details[j];
}
scan_result_handler_ptr.pap_details[j+1] = temp;
scan_result_handler_ptr.pap_details[j+1]->signal_strength = (*result_ptr)->signal_strength;
}
memset(*result_ptr, 0, sizeof(rtw_scan_result_t));
return;
}
}
scan_result_handler_ptr.scan_cnt++;
if(scan_result_handler_ptr.scan_cnt > scan_result_handler_ptr.max_ap_size){
scan_result_handler_ptr.scan_cnt = scan_result_handler_ptr.max_ap_size;
if((*result_ptr)->signal_strength > scan_result_handler_ptr.pap_details[scan_result_handler_ptr.max_ap_size-1]->signal_strength){
rtw_memcpy(scan_result_handler_ptr.pap_details[scan_result_handler_ptr.max_ap_size-1], *result_ptr, sizeof(rtw_scan_result_t));
temp = scan_result_handler_ptr.pap_details[scan_result_handler_ptr.max_ap_size -1];
}else
return;
}else{
rtw_memcpy(&scan_result_handler_ptr.ap_details[scan_result_handler_ptr.scan_cnt-1], *result_ptr, sizeof(rtw_scan_result_t));
}
for(i=0; i< scan_result_handler_ptr.scan_cnt-1; i++){
if((*result_ptr)->signal_strength > scan_result_handler_ptr.pap_details[i]->signal_strength)
break;
}
insert_pos = i;
for(i = scan_result_handler_ptr.scan_cnt-1; i>insert_pos; i--)
scan_result_handler_ptr.pap_details[i] = scan_result_handler_ptr.pap_details[i-1];
if(temp != NULL)
scan_result_handler_ptr.pap_details[insert_pos] = temp;
else
scan_result_handler_ptr.pap_details[insert_pos] = &scan_result_handler_ptr.ap_details[scan_result_handler_ptr.scan_cnt-1];
rtw_memset(*result_ptr, 0, sizeof(rtw_scan_result_t));
}
void wifi_scan_done_hdl( char* buf, int buf_len, int flags, void* userdata)
{
int i = 0;
rtw_scan_handler_result_t scan_result_report;
for(i=0; i<scan_result_handler_ptr.scan_cnt; i++){
rtw_memcpy(&scan_result_report.ap_details, scan_result_handler_ptr.pap_details[i], sizeof(rtw_scan_result_t));
scan_result_report.scan_complete = scan_result_handler_ptr.scan_complete;
scan_result_report.user_data = scan_result_handler_ptr.user_data;
(*scan_result_handler_ptr.gscan_result_handler)(&scan_result_report);
}
scan_result_handler_ptr.scan_complete = RTW_TRUE;
scan_result_report.scan_complete = RTW_TRUE;
(*scan_result_handler_ptr.gscan_result_handler)(&scan_result_report);
rtw_free(scan_result_handler_ptr.ap_details);
rtw_free(scan_result_handler_ptr.pap_details);
#if SCAN_USE_SEMAPHORE
rtw_up_sema(&scan_result_handler_ptr.scan_semaphore);
#else
scan_result_handler_ptr.scan_running = 0;
#endif
wifi_unreg_event_handler(WIFI_EVENT_SCAN_RESULT_REPORT, wifi_scan_each_report_hdl);
wifi_unreg_event_handler(WIFI_EVENT_SCAN_DONE, wifi_scan_done_hdl);
return;
}
//int rtk_wifi_scan(char *buf, int buf_len, xSemaphoreHandle * semaphore)
int wifi_scan(rtw_scan_type_t scan_type,
rtw_bss_type_t bss_type,
void* result_ptr)
{
int ret;
scan_buf_arg * pscan_buf;
u16 flags = scan_type | (bss_type << 8);
if(result_ptr != NULL){
pscan_buf = (scan_buf_arg *)result_ptr;
ret = wext_set_scan(WLAN0_NAME, (char*)pscan_buf->buf, pscan_buf->buf_len, flags);
}else{
wifi_reg_event_handler(WIFI_EVENT_SCAN_RESULT_REPORT, wifi_scan_each_report_hdl, NULL);
wifi_reg_event_handler(WIFI_EVENT_SCAN_DONE, wifi_scan_done_hdl, NULL);
ret = wext_set_scan(WLAN0_NAME, NULL, 0, flags);
}
if(ret == 0) {
if(result_ptr != NULL){
ret = wext_get_scan(WLAN0_NAME, pscan_buf->buf, pscan_buf->buf_len);
}
}
return ret;
}
int wifi_scan_networks_with_ssid(int (results_handler)(char*buf, int buflen, char *ssid, void *user_data),
OUT void* user_data, IN int scan_buflen, IN char* ssid, IN int ssid_len)
{
int scan_cnt = 0, add_cnt = 0;
scan_buf_arg scan_buf;
int ret;
scan_buf.buf_len = scan_buflen;
scan_buf.buf = (char*)pvPortMalloc(scan_buf.buf_len);
if(!scan_buf.buf){
printf("\n\rERROR: Can't malloc memory(%d)", scan_buf.buf_len);
return RTW_NOMEM;
}
//set ssid
memset(scan_buf.buf, 0, scan_buf.buf_len);
memcpy(scan_buf.buf, &ssid_len, sizeof(int));
memcpy(scan_buf.buf+sizeof(int), ssid, ssid_len);
//Scan channel
if(scan_cnt = (wifi_scan(RTW_SCAN_TYPE_ACTIVE, RTW_BSS_TYPE_ANY, &scan_buf)) < 0){
printf("\n\rERROR: wifi scan failed");
ret = RTW_ERROR;
}else{
if(NULL == results_handler)
{
int plen = 0;
while(plen < scan_buf.buf_len){
int len, rssi, ssid_len, i, security_mode;
int wps_password_id;
char *mac, *ssid;
//u8 *security_mode;
printf("\n\r");
// len
len = (int)*(scan_buf.buf + plen);
printf("len = %d,\t", len);
// check end
if(len == 0) break;
// mac
mac = scan_buf.buf + plen + 1;
printf("mac = ");
for(i=0; i<6; i++)
printf("%02x ", (u8)*(mac+i));
printf(",\t");
// rssi
rssi = *(int*)(scan_buf.buf + plen + 1 + 6);
printf(" rssi = %d,\t", rssi);
// security_mode
security_mode = (int)*(scan_buf.buf + plen + 1 + 6 + 4);
switch (security_mode) {
case IW_ENCODE_ALG_NONE:
printf("sec = open ,\t");
break;
case IW_ENCODE_ALG_WEP:
printf("sec = wep ,\t");
break;
case IW_ENCODE_ALG_CCMP:
printf("sec = wpa/wpa2,\t");
break;
}
// password id
wps_password_id = (int)*(scan_buf.buf + plen + 1 + 6 + 4 + 1);
printf("wps password id = %d,\t", wps_password_id);
printf("channel = %d,\t", *(scan_buf.buf + plen + 1 + 6 + 4 + 1 + 1));
// ssid
ssid_len = len - 1 - 6 - 4 - 1 - 1 - 1;
ssid = scan_buf.buf + plen + 1 + 6 + 4 + 1 + 1 + 1;
printf("ssid = ");
for(i=0; i<ssid_len; i++)
printf("%c", *(ssid+i));
plen += len;
add_cnt++;
}
printf("\n\rwifi_scan: add count = %d, scan count = %d", add_cnt, scan_cnt);
}
ret = RTW_SUCCESS;
}
if(results_handler)
results_handler(scan_buf.buf, scan_buf.buf_len, ssid, user_data);
if(scan_buf.buf)
vPortFree(scan_buf.buf);
return ret;
}
int wifi_scan_networks(rtw_scan_result_handler_t results_handler, void* user_data)
{
unsigned int max_ap_size = 64;
#if SCAN_USE_SEMAPHORE
rtw_bool_t result;
if(NULL == scan_result_handler_ptr.scan_semaphore)
rtw_init_sema(&scan_result_handler_ptr.scan_semaphore, 1);
scan_result_handler_ptr.scan_start_time = rtw_get_current_time();
/* Initialise the semaphore that will prevent simultaneous access - cannot be a mutex, since
* we don't want to allow the same thread to start a new scan */
result = (rtw_bool_t)rtw_down_timeout_sema(&scan_result_handler_ptr.scan_semaphore, SCAN_LONGEST_WAIT_TIME);
if ( result != RTW_TRUE )
{
/* Return error result, but set the semaphore to work the next time */
rtw_up_sema(&scan_result_handler_ptr.scan_semaphore);
return RTW_TIMEOUT;
}
#else
if(scan_result_handler_ptr.scan_running){
int count = 100;
while(scan_result_handler_ptr.scan_running && count > 0)
{
rtw_msleep_os(20);
count --;
}
if(count == 0){
printf("\n\r[%d]WiFi: Scan is running. Wait 2s timeout.", rtw_get_current_time());
return RTW_TIMEOUT;
}
}
scan_result_handler_ptr.scan_start_time = rtw_get_current_time();
scan_result_handler_ptr.scan_running = 1;
#endif
scan_result_handler_ptr.gscan_result_handler = results_handler;
scan_result_handler_ptr.max_ap_size = max_ap_size;
scan_result_handler_ptr.ap_details = (rtw_scan_result_t*)rtw_zmalloc(max_ap_size*sizeof(rtw_scan_result_t));
if(scan_result_handler_ptr.ap_details == NULL){
goto err_exit;
}
rtw_memset(scan_result_handler_ptr.ap_details, 0, max_ap_size*sizeof(rtw_scan_result_t));
scan_result_handler_ptr.pap_details = (rtw_scan_result_t**)rtw_zmalloc(max_ap_size*sizeof(rtw_scan_result_t*));
if(scan_result_handler_ptr.pap_details == NULL)
goto error2_with_result_ptr;
rtw_memset(scan_result_handler_ptr.pap_details, 0, max_ap_size);
scan_result_handler_ptr.scan_cnt = 0;
scan_result_handler_ptr.scan_complete = RTW_FALSE;
scan_result_handler_ptr.user_data = user_data;
if (wifi_scan( RTW_SCAN_COMMAMD<<4 | RTW_SCAN_TYPE_ACTIVE, RTW_BSS_TYPE_ANY, NULL) != RTW_SUCCESS)
{
goto error1_with_result_ptr;
}
return RTW_SUCCESS;
error1_with_result_ptr:
rtw_free((u8*)scan_result_handler_ptr.pap_details);
scan_result_handler_ptr.pap_details = NULL;
error2_with_result_ptr:
rtw_free((u8*)scan_result_handler_ptr.ap_details);
scan_result_handler_ptr.ap_details = NULL;
err_exit:
rtw_memset((void *)&scan_result_handler_ptr, 0, sizeof(scan_result_handler_ptr));
return RTW_ERROR;
}
//----------------------------------------------------------------------------//
int wifi_set_pscan_chan(__u8 * channel_list,__u8 * pscan_config, __u8 length)
{
if(channel_list)
return wext_set_pscan_channel(WLAN0_NAME, channel_list, pscan_config, length);
else
return -1;
}
//----------------------------------------------------------------------------//
int wifi_get_setting(const char *ifname, rtw_wifi_setting_t *pSetting)
{
int ret = 0;
int mode = 0;
unsigned short security = 0;
memset(pSetting, 0, sizeof(rtw_wifi_setting_t));
if(wext_get_mode(ifname, &mode) < 0)
ret = -1;
switch(mode) {
case IW_MODE_MASTER:
pSetting->mode = RTW_MODE_AP;
break;
case IW_MODE_INFRA:
default:
pSetting->mode = RTW_MODE_STA;
break;
//default:
//printf("\r\n%s(): Unknown mode %d\n", __func__, mode);
//break;
}
if(wext_get_ssid(ifname, pSetting->ssid) < 0)
ret = -1;
if(wext_get_channel(ifname, &pSetting->channel) < 0)
ret = -1;
if(wext_get_enc_ext(ifname, &security, &pSetting->key_idx, pSetting->password) < 0)
ret = -1;
switch(security){
case IW_ENCODE_ALG_NONE:
pSetting->security_type = RTW_SECURITY_OPEN;
break;
case IW_ENCODE_ALG_WEP:
pSetting->security_type = RTW_SECURITY_WEP_PSK;
break;
case IW_ENCODE_ALG_TKIP:
pSetting->security_type = RTW_SECURITY_WPA_TKIP_PSK;
break;
case IW_ENCODE_ALG_CCMP:
pSetting->security_type = RTW_SECURITY_WPA2_AES_PSK;
break;
default:
break;
}
if(security == IW_ENCODE_ALG_TKIP || security == IW_ENCODE_ALG_CCMP)
if(wext_get_passphrase(ifname, pSetting->password) < 0)
ret = -1;
return ret;
}
//----------------------------------------------------------------------------//
int wifi_show_setting(const char *ifname, rtw_wifi_setting_t *pSetting)
{
int ret = 0;
printf("\n\r\nWIFI %s Setting:",ifname);
printf("\n\r==============================");
switch(pSetting->mode) {
case RTW_MODE_AP:
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("\r\nAP,");
#endif
printf("\n\r MODE => AP");
break;
case RTW_MODE_STA:
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("\r\nSTA,");
#endif
printf("\n\r MODE => STATION");
break;
default:
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("\r\nUNKNOWN,");
#endif
printf("\n\r MODE => UNKNOWN");
}
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("%s,%d,", pSetting->ssid, pSetting->channel);
#endif
printf("\n\r SSID => %s", pSetting->ssid);
printf("\n\r CHANNEL => %d", pSetting->channel);
switch(pSetting->security_type) {
case RTW_SECURITY_OPEN:
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("OPEN,");
#endif
printf("\n\r SECURITY => OPEN");
break;
case RTW_SECURITY_WEP_PSK:
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("WEP,%d,", pSetting->key_idx);
#endif
printf("\n\r SECURITY => WEP");
printf("\n\r KEY INDEX => %d", pSetting->key_idx);
break;
case RTW_SECURITY_WPA_TKIP_PSK:
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("TKIP,");
#endif
printf("\n\r SECURITY => TKIP");
break;
case RTW_SECURITY_WPA2_AES_PSK:
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("AES,");
#endif
printf("\n\r SECURITY => AES");
break;
default:
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("UNKNOWN,");
#endif
printf("\n\r SECURITY => UNKNOWN");
}
#if CONFIG_EXAMPLE_UART_ATCMD
at_printf("%s,", pSetting->password);
#endif
printf("\n\r PASSWORD => %s", pSetting->password);
printf("\n\r");
return ret;
}
//----------------------------------------------------------------------------//
int wifi_set_network_mode(rtw_network_mode_t mode)
{
if((mode == RTW_NETWORK_B) || (mode == RTW_NETWORK_BG) || (mode == RTW_NETWORK_BGN))
return rltk_wlan_wireless_mode((unsigned char) mode);
return -1;
}
int wifi_set_wps_phase(unsigned char is_trigger_wps)
{
return rltk_wlan_set_wps_phase(is_trigger_wps);
}
//----------------------------------------------------------------------------//
int wifi_set_promisc(rtw_rcr_level_t enabled, void (*callback)(unsigned char*, unsigned int, void*), unsigned char len_used)
{
return promisc_set(enabled, callback, len_used);
}
void wifi_enter_promisc_mode(){
int mode = 0;
unsigned char ssid[33];
if(wifi_mode == RTW_MODE_STA_AP){
wifi_off();
vTaskDelay(20);
wifi_on(RTW_MODE_PROMISC);
}else{
wext_get_mode(WLAN0_NAME, &mode);
switch(mode) {
case IW_MODE_MASTER: //In AP mode
//rltk_wlan_deinit();
wifi_off();//modified by Chris Yang for iNIC
vTaskDelay(20);
//rltk_wlan_init(0, RTW_MODE_PROMISC);
//rltk_wlan_start(0);
wifi_on(RTW_MODE_PROMISC);
break;
case IW_MODE_INFRA: //In STA mode
if(wext_get_ssid(WLAN0_NAME, ssid) > 0)
wifi_disconnect();
}
}
}
int wifi_restart_ap(
unsigned char *ssid,
rtw_security_t security_type,
unsigned char *password,
int ssid_len,
int password_len,
int channel)
{
unsigned char idx = 0;
struct ip_addr ipaddr;
struct ip_addr netmask;
struct ip_addr gw;
struct netif * pnetif = &xnetif[0];
#ifdef CONFIG_CONCURRENT_MODE
rtw_wifi_setting_t setting;
int sta_linked = 0;
#endif
if(rltk_wlan_running(WLAN1_IDX)){
idx = 1;
}
// stop dhcp server
dhcps_deinit();
#ifdef CONFIG_CONCURRENT_MODE
if(idx > 0){
sta_linked = wifi_get_setting(WLAN0_NAME, &setting);
wifi_off();
vTaskDelay(20);
wifi_on(RTW_MODE_STA_AP);
}
else
#endif
{
IP4_ADDR(&ipaddr, GW_ADDR0, GW_ADDR1, GW_ADDR2, GW_ADDR3);
IP4_ADDR(&netmask, NETMASK_ADDR0, NETMASK_ADDR1 , NETMASK_ADDR2, NETMASK_ADDR3);
IP4_ADDR(&gw, GW_ADDR0, GW_ADDR1, GW_ADDR2, GW_ADDR3);
netif_set_addr(pnetif, &ipaddr, &netmask,&gw);
wifi_off();
vTaskDelay(20);
wifi_on(RTW_MODE_AP);
}
// start ap
if(wifi_start_ap((char*)ssid, security_type, (char*)password, ssid_len, password_len, channel) < 0) {
printf("\n\rERROR: Operation failed!");
return -1;
}
#if (INCLUDE_uxTaskGetStackHighWaterMark == 1)
printf("\r\nWebServer Thread: High Water Mark is %ld\n", uxTaskGetStackHighWaterMark(NULL));
#endif
#ifdef CONFIG_CONCURRENT_MODE
// connect to ap if wlan0 was linked with ap
if(idx > 0 && sta_linked == 0){
int ret;
printf("\r\nAP: ssid=%s", (char*)setting.ssid);
printf("\r\nAP: security_type=%d", setting.security_type);
printf("\r\nAP: password=%s", (char*)setting.password);
printf("\r\nAP: key_idx =%d\n", setting.key_idx);
ret = wifi_connect((char*)setting.ssid,
setting.security_type,
(char*)setting.password,
strlen((char*)setting.ssid),
strlen((char*)setting.password),
setting.key_idx,
NULL);
#if CONFIG_DHCP_CLIENT
if(ret == RTW_SUCCESS) {
/* Start DHCPClient */
LwIP_DHCP(0, DHCP_START);
}
#endif
}
#endif
#if (INCLUDE_uxTaskGetStackHighWaterMark == 1)
printf("\r\nWebServer Thread: High Water Mark is %ld\n", uxTaskGetStackHighWaterMark(NULL));
#endif
// start dhcp server
dhcps_init(&xnetif[idx]);
return 0;
}
#if CONFIG_AUTO_RECONNECT
extern void (*p_wlan_autoreconnect_hdl)(rtw_security_t, char*, int, char*, int, int);
struct wifi_autoreconnect_param {
rtw_security_t security_type;
char *ssid;
int ssid_len;
char *password;
int password_len;
int key_id;
};
static void wifi_autoreconnect_thread(void *param)
{
int ret = RTW_ERROR;
struct wifi_autoreconnect_param *reconnect_param = (struct wifi_autoreconnect_param *) param;
printf("\n\rauto reconnect ...\n");
ret = wifi_connect(reconnect_param->ssid, reconnect_param->security_type, reconnect_param->password,
reconnect_param->ssid_len, reconnect_param->password_len, reconnect_param->key_id, NULL);
#if CONFIG_LWIP_LAYER
if(ret == RTW_SUCCESS) {
#if ATCMD_VER == ATVER_2
if (dhcp_mode_sta == 2){
struct netif * pnetif = &xnetif[0];
LwIP_UseStaticIP(pnetif);
dhcps_init(pnetif);
}
else
#endif
{
LwIP_DHCP(0, DHCP_START);
#if LWIP_AUTOIP
uint8_t *ip = LwIP_GetIP(&xnetif[0]);
if((ip[0] == 0) && (ip[1] == 0) && (ip[2] == 0) && (ip[3] == 0)) {
printf("\n\nIPv4 AUTOIP ...");
LwIP_AUTOIP(&xnetif[0]);
}
#endif
}
}
#endif //#if CONFIG_LWIP_LAYER
vTaskDelete(NULL);
}
void wifi_autoreconnect_hdl(rtw_security_t security_type,
char *ssid, int ssid_len,
char *password, int password_len,
int key_id)
{
static struct wifi_autoreconnect_param param;
param.security_type = security_type;
param.ssid = ssid;
param.ssid_len = ssid_len;
param.password = password;
param.password_len = password_len;
param.key_id = key_id;
xTaskCreate(wifi_autoreconnect_thread, (const char *)"wifi_autoreconnect", 512, &param, tskIDLE_PRIORITY + 1, NULL);
}
int wifi_config_autoreconnect(__u8 mode, __u8 retyr_times, __u16 timeout)
{
p_wlan_autoreconnect_hdl = wifi_autoreconnect_hdl;
return wext_set_autoreconnect(WLAN0_NAME, mode, retyr_times, timeout);
}
int wifi_set_autoreconnect(__u8 mode)
{
p_wlan_autoreconnect_hdl = wifi_autoreconnect_hdl;
return wifi_config_autoreconnect(mode, 3, 5);//default retry 2 times(limit is 3), timeout 5 seconds
}
int wifi_get_autoreconnect(__u8 *mode)
{
return wext_get_autoreconnect(WLAN0_NAME, mode);
}
#endif
#ifdef CONFIG_CUSTOM_IE
/*
* Example for custom ie
*
* u8 test_1[] = {221, 2, 2, 2};
* u8 test_2[] = {221, 2, 1, 1};
* cus_ie buf[2] = {{test_1, PROBE_REQ},
* {test_2, PROBE_RSP | BEACON}};
* u8 buf_test2[] = {221, 2, 1, 3} ;
* cus_ie buf_update = {buf_test2, PROBE_REQ};
*
* add ie list
* static void cmd_add_ie(int argc, char **argv)
* {
* wifi_add_custom_ie((void *)buf, 2);
* }
*
* update current ie
* static void cmd_update_ie(int argc, char **argv)
* {
* wifi_update_custom_ie(&buf_update, 2);
* }
*
* delete all ie
* static void cmd_del_ie(int argc, char **argv)
* {
* wifi_del_custom_ie();
* }
*/
int wifi_add_custom_ie(void *cus_ie, int ie_num)
{
return wext_add_custom_ie(WLAN0_NAME, cus_ie, ie_num);
}
int wifi_update_custom_ie(void *cus_ie, int ie_index)
{
return wext_update_custom_ie(WLAN0_NAME, cus_ie, ie_index);
}
int wifi_del_custom_ie()
{
return wext_del_custom_ie(WLAN0_NAME);
}
#endif
#ifdef CONFIG_PROMISC
extern void promisc_init_packet_filter(void);
extern int promisc_add_packet_filter(u8 filter_id, rtw_packet_filter_pattern_t *patt, rtw_packet_filter_rule_e rule);
extern int promisc_enable_packet_filter(u8 filter_id);
extern int promisc_disable_packet_filter(u8 filter_id);
extern int promisc_remove_packet_filter(u8 filter_id);
void wifi_init_packet_filter()
{
promisc_init_packet_filter();
}
int wifi_add_packet_filter(unsigned char filter_id, rtw_packet_filter_pattern_t *patt, rtw_packet_filter_rule_e rule)
{
return promisc_add_packet_filter(filter_id, patt, rule);
}
int wifi_enable_packet_filter(unsigned char filter_id)
{
return promisc_enable_packet_filter(filter_id);
}
int wifi_disable_packet_filter(unsigned char filter_id)
{
return promisc_disable_packet_filter(filter_id);
}
int wifi_remove_packet_filter(unsigned char filter_id)
{
return promisc_remove_packet_filter(filter_id);
}
#endif
#ifdef CONFIG_AP_MODE
int wifi_enable_forwarding(void)
{
return wext_enable_forwarding(WLAN0_NAME);
}
int wifi_disable_forwarding(void)
{
return wext_disable_forwarding(WLAN0_NAME);
}
#endif
/* API to set flag for concurrent mode wlan1 issue_deauth when channel switched by wlan0
* usage: wifi_set_ch_deauth(0) -> wlan0 wifi_connect -> wifi_set_ch_deauth(1)
*/
#ifdef CONFIG_CONCURRENT_MODE
int wifi_set_ch_deauth(__u8 enable)
{
return wext_set_ch_deauth(WLAN1_NAME, enable);
}
#endif
//----------------------------------------------------------------------------//
#endif //#if CONFIG_WLAN