//----------------------------------------------------------------------------// #include #include #include "FreeRTOS.h" #include "task.h" #include "semphr.h" #include #include #include #if 1 #include "drv_types.h" // or #include "wlan_lib.h" #else #include "wifi_constants.h" #include "wifi_structures.h" //#include "wlan_lib.h" // or #include "drv_types.h" #endif #include #include #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 #if CONFIG_DEBUG_LOG > 0 #undef printf #define printf(...) rtl_printf(__VA_ARGS__) #else #undef printf #define printf(...) #endif //#define sscanf _sscanf #define SHOW_PRIVATE_OUT 0 // =0 - off, = 1 On /****************************************************** * Constants ******************************************************/ #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 ******************************************************/ internal_scan_handler_t scan_result_handler_ptr; static internal_join_result_t* join_user_data; unsigned char wifi_mode = RTW_MODE_NONE; // rtw_mode_t //extern rtw_mode_t wifi_mode; char 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 #define DD_WIFI_CONN 0 // pvvx //----------------------------------------------------------------------start-patch// #include "freertos/wrapper.h" #include "skbuff.h" extern unsigned char is_promisc_enabled(void); extern int promisc_set(rtw_rcr_level_t enabled, void (*callback)(unsigned char*, unsigned int, void*), unsigned char len_used); //------------------------------------------------------------------------end-patch// 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; error_printf("%s: security type(0x%x) is not supported!\n", pWifi->security_type, __func__); break; } 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); #if CONFIG_DEBUG_LOG > 4 info_printf("%s: Error flag is %d!\n", __func__, error_flag); #endif } #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(" %s():wifi_get_setting fail or ap mode\n", __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 wext_set_bssid(const char *ifname, const __u8 *bssid); //----------------------------------------------------------------------------// int wifi_connect( unsigned char bssid[ETH_ALEN], char use_bssid, // flag char *ssid, rtw_security_t security_type, char *password, int key_id, void *semaphore) { int ssid_len = 0; int password_len = 0; // int bssid_len = 6; 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; } if(ssid) { ssid_len = rtl_strlen(ssid); if(ssid_len > NDIS_802_11_LENGTH_SSID) ssid_len = NDIS_802_11_LENGTH_SSID; } if(password) { password_len = rtl_strlen(password); if(password_len > IW_PASSPHRASE_MAX_SIZE) password_len = IW_PASSPHRASE_MAX_SIZE; } 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) { #if CONFIG_DEBUG_LOG > 3 error_printf("%s: Can't malloc memory!\n", __func__); #endif 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); } if(bssid) rtw_memcpy(join_result->network_info.bssid.octet, bssid, ETH_ALEN); 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) { unsigned int g[5] = { 0 }; u8 i = 0; sscanf((const char*) password, "%02x%02x%02x%02x%02x", &g[0], &g[1], &g[2], &g[3], &g[4]); for (i = 0; i < 5; i++) wep_pwd[i] = (u8) g[i]; wep_pwd[5] = '\0'; password_len = 5; wep_hex = 1; } else if (password_len == 26) { unsigned int g[13] = { 0 }; u8 i = 0; sscanf((const char*) password, "%02x%02x%02x%02x%02x%02x%02x" "%02x%02x%02x%02x%02x%02x", &g[0], &g[1], &g[2], &g[3], &g[4], &g[5], &g[6], &g[7], &g[8], &g[9], &g[10], &g[11], &g[12]); for (i = 0; i < 13; i++) wep_pwd[i] = (u8) g[i]; wep_pwd[13] = '\0'; password_len = 13; wep_hex = 1; } } } 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; #if CONFIG_DEBUG_LOG > 3 error_printf("%s: Can't malloc memory!\n", __func__); #endif 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); rtw_network_info_t *pWifi = &join_result->network_info; if (wifi_connect_local(pWifi) == 0) { uint16 flg = 0; if(use_bssid) { struct { u8 bssid[ETH_ALEN + 2]; void * p; } bs = { 0 }; memcpy(bs.bssid, pWifi->bssid.octet, ETH_ALEN); for(int i = 0; i < ETH_ALEN; i++) { flg += bs.bssid[i]; } if(flg == 0x5FA || flg == 0) { // 0x5FA = 6*0xff use_bssid = 0; flg = 0; } else { use_bssid = 1; wext_set_bssid(WLAN0_NAME, bs.bssid); } } if(!use_bssid) { wext_set_ssid(WLAN0_NAME, join_result->network_info.ssid.val, join_result->network_info.ssid.len); }; } 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\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) { warning_printf("RTW API: Join bss timeout\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_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) { error_printf("WEXT: Failed to set bogus BSSID to disconnect\n"); 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 && 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: if (error_flag == RTW_NO_ERROR) return RTW_SUCCESS; break; } 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, SHOW_PRIVATE_OUT); } 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); #if SHOW_PRIVATE_OUT rtl_printf("%s\n", buf); #endif 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; // char buf[11]; char buf[64]; rtw_memset(buf, 0, sizeof(buf)); rtw_memcpy(buf, "txpower", 11); ret = wext_private_command_with_retval(WLAN0_NAME, buf, buf, sizeof(buf)); #if SHOW_PRIVATE_OUT rtl_printf("%s\n", buf); #endif sscanf(buf, "%d", poweridx); return ret; } int wifi_set_txpower(int poweridx) { int ret; char buf[24]; rtw_memset(buf, 0, sizeof(buf)); snprintf(buf, 24, "txpower patha=%d,pathb=%d", poweridx, poweridx); // patha=%d,pathb=%d ? ret = wext_private_command(WLAN0_NAME, buf, SHOW_PRIVATE_OUT); 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, SHOW_PRIVATE_OUT); 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, SHOW_PRIVATE_OUT); snprintf(buf, 24, "get_security"); ret = wext_private_command_with_retval(ifname, buf, buf, 24); #if SHOW_PRIVATE_OUT rtl_printf("%s\n", buf); #endif sscanf(buf, "%d", security); return ret; } extern int wext_get_drv_ability(const char *ifname, __u32 *ability); int wifi_get_drv_ability(uint32_t *ability) { return wext_get_drv_ability(WLAN0_NAME, (__u32 *)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, SHOW_PRIVATE_OUT); 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 = 0; int timeout = wifi_test_timeout_ms / wifi_test_timeout_step_ms; int idx; int devnum = 1; static int event_init = 0; if (rltk_wlan_running(WLAN0_IDX)) { warning_printf("WIFI is already running\n"); return 0; } 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("Initializing WIFI ...\n"); 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("WIFI initialized\n"); /* * printf("set country code here\n"); * wifi_set_country(RTW_COUNTRY_US); */ break; } if (timeout == 0) { printf("ERROR: Init WIFI timeout!\n"); break; } // vTaskDelay(1 * configTICK_RATE_HZ); vTaskDelay(wifi_test_timeout_step_ms / portTICK_RATE_MS); timeout--; } #if CONFIG_INIC_EN inic_start(); #endif return ret; } extern void dhcps_deinit(void); //#include "dhcps.h" int wifi_off(void) { // int ret = 0; uint32 timeout = xTaskGetTickCount(); #if CONFIG_LWIP_LAYER dhcps_deinit(); LwIP_DHCP(0, DHCP_STOP); LwIP_DHCP(1, DHCP_STOP); #endif if ((rltk_wlan_running(WLAN0_IDX) == 0) && (rltk_wlan_running(WLAN1_IDX) == 0)) { info_printf("WIFI is not running\n"); return 0; } info_printf("Deinitializing WIFI ...\n"); #if defined(CONFIG_ENABLE_WPS_AP) && CONFIG_ENABLE_WPS_AP // @todo wpas_wps_deinit(); #endif rltk_wlan_deinit(); while (1) { if ((rltk_wlan_running(WLAN0_IDX) == 0) && (rltk_wlan_running(WLAN1_IDX) == 0)) { // info_printf("WIFI deinitialized\n"); info_printf("WIFI deinitialized (%d ms)\n", xTaskGetTickCount() - timeout); break; } if(xTaskGetTickCount() - timeout > wifi_test_timeout_ms/portTICK_RATE_MS) { error_printf("WIFI deinitialized timeout!\n"); break; } vTaskDelay(10 / portTICK_RATE_MS); } wifi_mode = RTW_MODE_NONE; #if CONFIG_INIC_EN inic_stop(); #endif return 1; } 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 debug_printf("ap_sta_assoc\n"); } static void wifi_ap_sta_disassoc_hdl(char* buf, int buf_len, int flags, void* userdata) { //USER TODO debug_printf("ap_sta_disassoc\n"); } */ 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 extern int set_hidden_ssid(const char *ifname, uint8_t value); int wifi_start_ap(char *ssid, rtw_security_t security_type, char *password, int channel, char ssid_hidden) { const char *ifname = WLAN0_NAME; int ssid_len = 0; int password_len = 0; int ret = 0; if(ssid) { ssid_len = rtl_strlen(ssid); if(ssid_len > NDIS_802_11_LENGTH_SSID) ssid_len = NDIS_802_11_LENGTH_SSID; } if(password) { password_len = rtl_strlen(password); if(password_len > IW_PASSPHRASE_MAX_SIZE) password_len = IW_PASSPHRASE_MAX_SIZE; } if (wifi_mode == RTW_MODE_STA_AP) { ifname = WLAN1_NAME; } if (is_promisc_enabled()) promisc_set(0, NULL, 0); /* wifi_unreg_event_handler(WIFI_EVENT_STA_ASSOC, wifi_ap_sta_assoc_hdl); wifi_reg_event_handler(WIFI_EVENT_STA_ASSOC, wifi_ap_sta_assoc_hdl, NULL); wifi_unreg_event_handler(WIFI_EVENT_STA_DISASSOC, wifi_ap_sta_disassoc_hdl); 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; if(channel < 1 || channel > 14) channel = 1; ret = wext_set_channel(ifname, channel); //Set channel before starting ap if (ret < 0) goto exit; if(security_type != RTW_SECURITY_OPEN) { // case RTW_SECURITY_WPA2_AES_PSK: security_type = 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); } if (ret < 0) goto exit; if(ssid_hidden) { 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("ERROR: Can't malloc memory(%d)\n", 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("ERROR: wifi scan failed\n"); 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"); // 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("\nwifi_scan: add count = %d, scan count = %d\n", 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("WiFi: Scan is running. Wait 2s timeout.\n"); 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; } //----------------------------------------------------------------------------// extern char str_rom_57ch3Dch0A[]; // "=========================================================\n" 57 шт int wifi_show_setting(const char *ifname, rtw_wifi_setting_t *pSetting) { int ret = 0; printf("\nWIFI '%s' Setting:\n", ifname); printf(&str_rom_57ch3Dch0A[25]); // "================================\n" switch (pSetting->mode) { case RTW_MODE_AP: #if CONFIG_EXAMPLE_UART_ATCMD at_printf("AP:"); #endif printf("\tMODE => AP\n"); break; case RTW_MODE_STA: #if CONFIG_EXAMPLE_UART_ATCMD at_printf("STA:"); #endif printf("\tMODE => STATION\n"); break; default: #if CONFIG_EXAMPLE_UART_ATCMD at_printf("UNKNOWN,"); #endif printf("\tMODE => UNKNOWN\n"); } #if CONFIG_EXAMPLE_UART_ATCMD at_printf("%s,%d,", pSetting->ssid, pSetting->channel); #endif printf("\tSSID => %s\n", pSetting->ssid); printf("\tCHANNEL => %d\n", pSetting->channel); switch (pSetting->security_type) { case RTW_SECURITY_OPEN: #if CONFIG_EXAMPLE_UART_ATCMD at_printf("OPEN,"); #endif printf("\tSECURITY => OPEN\n"); break; case RTW_SECURITY_WEP_PSK: #if CONFIG_EXAMPLE_UART_ATCMD at_printf("WEP,%d,", pSetting->key_idx); #endif printf("\tSECURITY => WEP\n"); printf("\tKEY INDEX => %d\n", pSetting->key_idx); break; case RTW_SECURITY_WPA_TKIP_PSK: #if CONFIG_EXAMPLE_UART_ATCMD at_printf("TKIP,"); #endif printf("\tSECURITY => TKIP\n"); break; case RTW_SECURITY_WPA2_AES_PSK: #if CONFIG_EXAMPLE_UART_ATCMD at_printf("AES,"); #endif printf("\tSECURITY => AES\n"); break; default: #if CONFIG_EXAMPLE_UART_ATCMD at_printf("UNKNOWN,"); #endif printf("\tSECURITY => UNKNOWN\n"); } #if CONFIG_EXAMPLE_UART_ATCMD at_printf("%s,", pSetting->password); #endif printf("\tPASSWORD => %s\n", pSetting->password); // printf("\n"); 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(wifi_test_timeout_step_ms / portTICK_RATE_MS); 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(wifi_test_timeout_step_ms / portTICK_RATE_MS); //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(); } } } extern void dhcps_init(struct netif * pnetif); int wifi_restart_ap(unsigned char *ssid, rtw_security_t security_type, unsigned char *password, 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(wifi_test_timeout_step_ms / portTICK_RATE_MS); 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(wifi_test_timeout_step_ms / portTICK_RATE_MS); wifi_on(RTW_MODE_AP); } // start ap if (wifi_start_ap((char*) ssid, security_type, (char*) password, channel, 0) < 0) { printf("ERROR: Operation failed!\n"); return -1; } #if (INCLUDE_uxTaskGetStackHighWaterMark == 1) printf("WebServer 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("AP: ssid=%s\n", (char* )setting.ssid); printf("AP: security_type=%d\n", setting.security_type); printf("AP: password=%s\n", (char* )setting.password); printf("AP: key_idx =%d\n", setting.key_idx); ret = wifi_connect(NULL, 0 , (char*) setting.ssid, setting.security_type, (char*) setting.password, setting.key_idx, NULL); if (ret == RTW_SUCCESS) { #if CONFIG_DHCP_CLIENT /* Start DHCPClient */ LwIP_DHCP(0, DHCP_START); #endif #if CONFIG_WLAN_CONNECT_CB extern void connect_start(void); connect_start(); #endif } } #endif #if (INCLUDE_uxTaskGetStackHighWaterMark == 1) printf("WebServer 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("auto reconnect ...\n"); ret = wifi_connect( NULL, 0, reconnect_param->ssid, reconnect_param->security_type, reconnect_param->password, reconnect_param->key_id, NULL); if (ret == RTW_SUCCESS) { #if CONFIG_LWIP_LAYER #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("IPv4 AUTOIP ...\n"); LwIP_AUTOIP(&xnetif[0]); } #endif } #endif //#if CONFIG_LWIP_LAYER #if CONFIG_WLAN_CONNECT_CB extern void connect_start(void); connect_start(); #endif } 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 * )"st_recon", 400, ¶m, 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 // CONFIG_PROMISC #ifdef CONFIG_AP_MODE extern int wext_enable_forwarding(const char *ifname); int wifi_enable_forwarding(void) { return wext_enable_forwarding(WLAN0_NAME); } extern int wext_disable_forwarding(const char *ifname); int wifi_disable_forwarding(void) { return wext_disable_forwarding(WLAN0_NAME); } #endif // CONFIG_AP_MODE /* 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 extern int wext_set_ch_deauth(const char *ifname, __u8 enable); int wifi_set_ch_deauth(__u8 enable) { return wext_set_ch_deauth(WLAN1_NAME, enable); } #endif //----------------------------------------------------------------------------// #endif //#if CONFIG_WLAN