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;
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_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);
}
void wlan_emac_recv(struct netif *netif, int len)
{
struct eth_drv_sg sg_list[MAX_ETH_DRV_SG];
emac_stack_mem_t *buf;
struct pbuf *p;
int sg_len = 0;
if (!rltk_wlan_running(0)) {
return;
}
if (len > MAX_ETH_MSG || len < 0) {
len = MAX_ETH_MSG;
}
buf = emac_stack_mem_alloc(NULL, len, 0);
if (buf == NULL) {
return;
}
p = (struct pbuf *)buf;
for (; p != NULL && sg_len < MAX_ETH_DRV_SG; p = p->next) {
sg_list[sg_len].buf = (uint32_t) p->payload;
sg_list[sg_len].len = p->len;
sg_len++;
}
rltk_wlan_recv(0, sg_list, sg_len);
if (link_input_cb) {
link_input_cb(link_input_data, buf);
}
return;
}
bool RTW_EMAC::link_out(emac_mem_buf_t *buf)
{
struct eth_drv_sg *sg_list;
int sg_len = 0;
int tot_len;
emac_mem_buf_t *p;
bool ret = true;
if (!rltk_wlan_running(0)) {
memory_manager->free(buf);
return false;
}
sg_list = (struct eth_drv_sg *)malloc(sizeof(struct eth_drv_sg)*MAX_ETH_DRV_SG);
if (sg_list == 0) {
memory_manager->free(buf);
return false;
}
p = buf;
tot_len = memory_manager->get_total_len(p);
for (; p != NULL && sg_len < MAX_ETH_DRV_SG; p = memory_manager->get_next(p)) {
sg_list[sg_len].buf = (unsigned int)(static_cast<uint8_t *>(memory_manager->get_ptr(p)));
sg_list[sg_len].len = memory_manager->get_len(p);
sg_len++;
}
if (sg_len) {
if (rltk_wlan_send(0, sg_list, sg_len, tot_len) != 0) {
ret = false;
}
}
memory_manager->free(buf);
free(sg_list);
return ret;
}
void RTW_EMAC::wlan_emac_recv(void *param, struct netif *netif, uint32_t len)
{
struct eth_drv_sg sg_list[MAX_ETH_DRV_SG] = {0};
emac_mem_buf_t *buf;
RTW_EMAC *enet = static_cast<RTW_EMAC *>(param);
emac_mem_buf_t *p;
int sg_len = 0;
if (!rltk_wlan_running(0)) {
return;
}
if (len > MAX_ETH_MSG || len < 0) {
len = MAX_ETH_MSG;
}
buf = enet->memory_manager->alloc_heap(len, 0);
if (buf == NULL) {
return;
}
enet->memory_manager->set_len(buf, len);
p = buf;
for (; p != NULL && sg_len < MAX_ETH_DRV_SG; p = enet->memory_manager->get_next(p)) {
sg_list[sg_len].buf = (unsigned int)(static_cast<uint8_t *>(enet->memory_manager->get_ptr(p)));
sg_list[sg_len].len = enet->memory_manager->get_len(p);
sg_len++;
}
rltk_wlan_recv(0, sg_list, sg_len);
if (enet->emac_link_input_cb) {
enet->emac_link_input_cb(buf);
}
}
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
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;
}
static void LoadWifiSetting()
{
const char *ifname = WLAN0_NAME;
if(rltk_wlan_running(WLAN1_IDX))
{//STA_AP_MODE
ifname = WLAN1_NAME;
}
wifi_get_setting(ifname, &wifi_setting);
//printf("\r\nLoadWifiSetting(): wifi_setting.ssid=%s\n", wifi_setting.ssid);
//printf("\r\nLoadWifiSetting(): wifi_setting.channel=%d\n", wifi_setting.channel);
//printf("\r\nLoadWifiSetting(): wifi_setting.security_type=%d\n", wifi_setting.security_type);
//printf("\r\nLoadWifiSetting(): wifi_setting.password=%s\n", wifi_setting.password);
}
/*
cmd_ap_wps for AP WSC setting. command style:
cmd_ap_wps pbc or cmd_ap_wps pin 12345678
*/
void cmd_ap_wps(int argc, char **argv)
{
if(rltk_wlan_running(WLAN1_IDX)){
printf("\n\rNot support con-current softAP WSC!\n\r");
return;
}
if((argc == 2 || argc == 3) && (argv[1] != NULL)) {
if (strcmp(argv[1],"pin") == 0 ) {
unsigned int pin_val = 0;
if(argc == 3){
pin_val = atoi(argv[2]);
if (!wps_pin_valid(pin_val)) {
printf("\n\rWPS-AP: Device pin code is invalid. Not trigger WPS.\n\r");
return;
}
printf("\n\rWPS-AP: Start AP WPS PIN Keypad.\n");
ap_wps_start(WPS_CONFIG_KEYPAD, argv[2]);
}else{
char device_pin[10];
pin_val = wps_generate_pin();
sprintf(device_pin, "%08d", pin_val);
printf("\n\rWPS: Start WPS PIN Display. PIN: %s\n\r", device_pin);
ap_wps_start(WPS_CONFIG_DISPLAY, (char*)device_pin);
}
}else if (strcmp(argv[1],"pbc") == 0) {
printf("\n\rWPS-AP: Start AP WPS PBC\n");
ap_wps_start(WPS_CONFIG_PUSHBUTTON, NULL);
}else{
printf("\n\rWPS-AP Usage:\"wifi_ap_wps pin [pin_code]\" or \"wifi_ap_wps pbc\"\n");
return;
}
} else {
printf("\n\rWPS-AP Usage:\"wifi_ap_wps pin [pin_code]\" or \"wifi_ap_wps pbc\"\n");
}
return;
}
static bool wlan_link_out(emac_interface_t *emac, emac_stack_mem_t *buf)
{
struct eth_drv_sg * sg_list=0;
int sg_len = 0;
int tot_len;
struct pbuf *p;
bool ret = true;
if (!rltk_wlan_running(0)) {
return false;
}
sg_list = (struct eth_drv_sg *)malloc(sizeof(struct eth_drv_sg)*MAX_ETH_DRV_SG);
if(sg_list == 0){//malloc fail
return false;
}
emac_stack_mem_ref(emac, buf);
p = (struct pbuf *)buf;
tot_len = p->tot_len;
for (; p != NULL && sg_len < MAX_ETH_DRV_SG; p = p->next) {
sg_list[sg_len].buf = (uint32_t) p->payload;
sg_list[sg_len].len = p->len;
sg_len++;
}
if (sg_len) {
if (rltk_wlan_send(0, sg_list, sg_len, tot_len) != 0) {
ret = false;
}
}
emac_stack_mem_free(emac, buf);
free(sg_list);
return ret;
}
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;
}
void fATWx(void *arg){
int i = 0;
#if CONFIG_LWIP_LAYER
u8 *mac = LwIP_GetMAC(&xnetif[0]);
u8 *ip = LwIP_GetIP(&xnetif[0]);
u8 *gw = LwIP_GetGW(&xnetif[0]);
#endif
u8 *ifname[2] = {WLAN0_NAME,WLAN1_NAME};
rtw_wifi_setting_t setting;
printf("[ATW?]: _AT_WLAN_INFO_\n\r");
for(i=0;i<NET_IF_NUM;i++){
if(rltk_wlan_running(i)){
#if CONFIG_LWIP_LAYER
mac = LwIP_GetMAC(&xnetif[i]);
ip = LwIP_GetIP(&xnetif[i]);
gw = LwIP_GetGW(&xnetif[i]);
#endif
printf("\n\r\nWIFI %s Status: Running", ifname[i]);
printf("\n\r==============================");
rltk_wlan_statistic(i);
wifi_get_setting((const char*)ifname[i],&setting);
wifi_show_setting((const char*)ifname[i],&setting);
#if CONFIG_LWIP_LAYER
printf("\n\rInterface (%s)", ifname[i]);
printf("\n\r==============================");
printf("\n\r\tMAC => %02x:%02x:%02x:%02x:%02x:%02x", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]) ;
printf("\n\r\tIP => %d.%d.%d.%d", ip[0], ip[1], ip[2], ip[3]);
printf("\n\r\tGW => %d.%d.%d.%d\n\r", gw[0], gw[1], gw[2], gw[3]);
#endif
if(setting.mode == RTW_MODE_AP || i == 1)
{
int client_number;
struct {
int count;
rtw_mac_t mac_list[3];
} client_info;
client_info.count = 3;
wifi_get_associated_client_list(&client_info, sizeof(client_info));
printf("\n\rAssociated Client List:");
printf("\n\r==============================");
if(client_info.count == 0)
printf("\n\rClient Num: 0\n\r", client_info.count);
else
{
printf("\n\rClient Num: %d", client_info.count);
for( client_number=0; client_number < client_info.count; client_number++ )
{
printf("\n\rClient %d:", client_number + 1);
printf("\n\r\tMAC => "MAC_FMT"",
MAC_ARG(client_info.mac_list[client_number].octet));
}
printf("\n\r");
}
}
}
}
#if defined(configUSE_TRACE_FACILITY) && (configUSE_TRACE_FACILITY == 1) && (configUSE_STATS_FORMATTING_FUNCTIONS == 1)
{
signed char pcWriteBuffer[1024];
vTaskList((char*)pcWriteBuffer);
printf("\n\rTask List: \n\r%s", pcWriteBuffer);
}
#endif
}
