1776 lines
		
	
	
		
			49 KiB
		
	
	
	
		
			C
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			1776 lines
		
	
	
		
			49 KiB
		
	
	
	
		
			C
		
	
	
		
			Executable File
		
	
	
	
	
/*
 | 
						|
 * Copyright (c) 2006-2023, RT-Thread Development Team
 | 
						|
 *
 | 
						|
 * SPDX-License-Identifier: Apache-2.0
 | 
						|
 *
 | 
						|
 * Change Logs:
 | 
						|
 * Date           Author       Notes
 | 
						|
 * 2018-08-06     tyx          the first version
 | 
						|
 * 2023-12-12     Evlers       add the wlan join scan function
 | 
						|
 */
 | 
						|
 | 
						|
#include <rthw.h>
 | 
						|
#include <rtthread.h>
 | 
						|
#include <wlan_dev.h>
 | 
						|
#include <wlan_cfg.h>
 | 
						|
#include <wlan_mgnt.h>
 | 
						|
#include <wlan_prot.h>
 | 
						|
#include <wlan_workqueue.h>
 | 
						|
 | 
						|
// #define RT_WLAN_MGNT_DEBUG
 | 
						|
#define DBG_TAG "WLAN.mgnt"
 | 
						|
#ifdef RT_WLAN_MGNT_DEBUG
 | 
						|
#define DBG_LVL DBG_LOG
 | 
						|
#else
 | 
						|
#define DBG_LVL DBG_INFO
 | 
						|
#endif /* RT_WLAN_MGNT_DEBUG */
 | 
						|
#include <rtdbg.h>
 | 
						|
 | 
						|
#ifdef RT_WLAN_MANAGE_ENABLE
 | 
						|
 | 
						|
#ifndef RT_WLAN_DEVICE
 | 
						|
#define RT_WLAN_DEVICE(__device) ((struct rt_wlan_device *)__device)
 | 
						|
#endif
 | 
						|
 | 
						|
#define RT_WLAN_LOG_D(_fmt, ...) LOG_D("L:%d "_fmt"", __LINE__, ##__VA_ARGS__)
 | 
						|
#define RT_WLAN_LOG_I(...) LOG_I(__VA_ARGS__)
 | 
						|
#define RT_WLAN_LOG_W(_fmt, ...) LOG_W("F:%s L:%d "_fmt"", __FUNCTION__, __LINE__, ##__VA_ARGS__)
 | 
						|
#define RT_WLAN_LOG_E(_fmt, ...) LOG_E("F:%s L:%d "_fmt"", __FUNCTION__, __LINE__, ##__VA_ARGS__)
 | 
						|
 | 
						|
#define STA_DEVICE()  (_sta_mgnt.device)
 | 
						|
#define AP_DEVICE()  (_ap_mgnt.device)
 | 
						|
 | 
						|
#define STAINFO_LOCK()    (rt_mutex_take(&sta_info_mutex, RT_WAITING_FOREVER))
 | 
						|
#define STAINFO_UNLOCK()  (rt_mutex_release(&sta_info_mutex))
 | 
						|
 | 
						|
#define MGNT_LOCK()       (rt_mutex_take(&mgnt_mutex, RT_WAITING_FOREVER))
 | 
						|
#define MGNT_UNLOCK()     (rt_mutex_release(&mgnt_mutex))
 | 
						|
 | 
						|
#define COMPLETE_LOCK()       (rt_mutex_take(&complete_mutex, RT_WAITING_FOREVER))
 | 
						|
#define COMPLETE_UNLOCK()     (rt_mutex_release(&complete_mutex))
 | 
						|
 | 
						|
#ifdef RT_WLAN_AUTO_CONNECT_ENABLE
 | 
						|
#define TIME_STOP()    (rt_timer_stop(&reconnect_time))
 | 
						|
#define TIME_START()   (rt_timer_start(&reconnect_time))
 | 
						|
static rt_uint32_t id = 0;
 | 
						|
#else
 | 
						|
#define TIME_STOP()
 | 
						|
#define TIME_START()
 | 
						|
#endif
 | 
						|
 | 
						|
#if RT_WLAN_EBOX_NUM < 1
 | 
						|
#error "event box num Too few"
 | 
						|
#endif
 | 
						|
 | 
						|
struct rt_wlan_mgnt_des
 | 
						|
{
 | 
						|
    struct rt_wlan_device *device;
 | 
						|
    struct rt_wlan_info info;
 | 
						|
    struct rt_wlan_key key;
 | 
						|
    rt_uint8_t state;
 | 
						|
    rt_uint8_t flags;
 | 
						|
};
 | 
						|
 | 
						|
struct rt_wlan_event_desc
 | 
						|
{
 | 
						|
    rt_wlan_event_handler handler;
 | 
						|
    void *parameter;
 | 
						|
};
 | 
						|
 | 
						|
struct rt_wlan_sta_list
 | 
						|
{
 | 
						|
    struct rt_wlan_sta_list *next;
 | 
						|
    struct rt_wlan_info info;
 | 
						|
};
 | 
						|
 | 
						|
struct rt_wlan_sta_des
 | 
						|
{
 | 
						|
    int num;
 | 
						|
    struct rt_wlan_sta_list *node;
 | 
						|
};
 | 
						|
 | 
						|
struct rt_wlan_msg
 | 
						|
{
 | 
						|
    rt_int32_t event;
 | 
						|
    rt_int32_t len;
 | 
						|
    void *buff;
 | 
						|
};
 | 
						|
 | 
						|
struct rt_wlan_complete_des
 | 
						|
{
 | 
						|
    struct rt_event complete;
 | 
						|
    rt_uint32_t event_flag;
 | 
						|
    int index;
 | 
						|
};
 | 
						|
 | 
						|
static struct rt_mutex mgnt_mutex;
 | 
						|
 | 
						|
static struct rt_wlan_mgnt_des _sta_mgnt;
 | 
						|
static struct rt_wlan_mgnt_des _ap_mgnt;
 | 
						|
 | 
						|
 | 
						|
static struct rt_wlan_sta_des sta_info;
 | 
						|
static struct rt_mutex sta_info_mutex;
 | 
						|
 | 
						|
static struct rt_wlan_event_desc event_tab[RT_WLAN_EVT_MAX];
 | 
						|
 | 
						|
static struct rt_wlan_complete_des *complete_tab[5];
 | 
						|
static struct rt_mutex complete_mutex;
 | 
						|
 | 
						|
#ifdef RT_WLAN_AUTO_CONNECT_ENABLE
 | 
						|
static struct rt_timer reconnect_time;
 | 
						|
#endif
 | 
						|
 | 
						|
rt_inline int _sta_is_null(void)
 | 
						|
{
 | 
						|
    if (_sta_mgnt.device == RT_NULL)
 | 
						|
    {
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
rt_inline int _ap_is_null(void)
 | 
						|
{
 | 
						|
    if (_ap_mgnt.device == RT_NULL)
 | 
						|
    {
 | 
						|
        return 1;
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
rt_inline rt_bool_t _is_do_connect(void)
 | 
						|
{
 | 
						|
    if ((rt_wlan_get_autoreconnect_mode() == RT_FALSE) ||
 | 
						|
            (rt_wlan_is_connected() == RT_TRUE) ||
 | 
						|
            (_sta_mgnt.state & RT_WLAN_STATE_CONNECTING))
 | 
						|
    {
 | 
						|
        return RT_FALSE;
 | 
						|
    }
 | 
						|
    return RT_TRUE;
 | 
						|
}
 | 
						|
 | 
						|
#ifdef RT_WLAN_WORK_THREAD_ENABLE
 | 
						|
 | 
						|
static void rt_wlan_mgnt_work(void *parameter)
 | 
						|
{
 | 
						|
    struct rt_wlan_msg *msg = parameter;
 | 
						|
    void *user_parameter;
 | 
						|
    rt_wlan_event_handler handler = RT_NULL;
 | 
						|
    struct rt_wlan_buff user_buff = { 0 };
 | 
						|
    rt_base_t level;
 | 
						|
 | 
						|
    /* Get user callback */
 | 
						|
    if (msg->event < RT_WLAN_EVT_MAX)
 | 
						|
    {
 | 
						|
        level = rt_hw_interrupt_disable();
 | 
						|
        handler = event_tab[msg->event].handler;
 | 
						|
        user_parameter = event_tab[msg->event].parameter;
 | 
						|
        rt_hw_interrupt_enable(level);
 | 
						|
    }
 | 
						|
 | 
						|
    /* run user callback fun */
 | 
						|
    if (handler)
 | 
						|
    {
 | 
						|
        user_buff.data = msg->buff;
 | 
						|
        user_buff.len = msg->len;
 | 
						|
        RT_WLAN_LOG_D("wlan work thread run user callback, event:%d", msg->event);
 | 
						|
        handler(msg->event, &user_buff, user_parameter);
 | 
						|
    }
 | 
						|
 | 
						|
    switch (msg->event)
 | 
						|
    {
 | 
						|
    case RT_WLAN_EVT_STA_CONNECTED:
 | 
						|
    {
 | 
						|
        struct rt_wlan_cfg_info cfg_info;
 | 
						|
 | 
						|
        rt_memset(&cfg_info, 0, sizeof(cfg_info));
 | 
						|
        /* save config */
 | 
						|
        if (rt_wlan_is_connected() == RT_TRUE)
 | 
						|
        {
 | 
						|
            rt_enter_critical();
 | 
						|
            cfg_info.info = _sta_mgnt.info;
 | 
						|
            cfg_info.key = _sta_mgnt.key;
 | 
						|
            rt_exit_critical();
 | 
						|
            RT_WLAN_LOG_D("run save config! ssid:%s len%d", _sta_mgnt.info.ssid.val, _sta_mgnt.info.ssid.len);
 | 
						|
#ifdef RT_WLAN_CFG_ENABLE
 | 
						|
            rt_wlan_cfg_save(&cfg_info);
 | 
						|
#endif
 | 
						|
        }
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    default :
 | 
						|
        break;
 | 
						|
    }
 | 
						|
 | 
						|
    rt_free(msg);
 | 
						|
}
 | 
						|
 | 
						|
static rt_err_t rt_wlan_send_to_thread(rt_wlan_event_t event, void *buff, int len)
 | 
						|
{
 | 
						|
    struct rt_wlan_msg *msg;
 | 
						|
 | 
						|
    RT_WLAN_LOG_D("F:%s is run event:%d", __FUNCTION__, event);
 | 
						|
 | 
						|
    /* Event packing */
 | 
						|
    msg = rt_malloc(sizeof(struct rt_wlan_msg) + len);
 | 
						|
    if (msg == RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("wlan mgnt send msg err! No memory");
 | 
						|
        return -RT_ENOMEM;
 | 
						|
    }
 | 
						|
    rt_memset(msg, 0, sizeof(struct rt_wlan_msg) + len);
 | 
						|
    msg->event = event;
 | 
						|
    if (len != 0)
 | 
						|
    {
 | 
						|
        msg->buff = (void *)&msg[1];
 | 
						|
        rt_memcpy(msg->buff, buff, len);
 | 
						|
        msg->len = len;
 | 
						|
    }
 | 
						|
 | 
						|
    /* send event to wlan thread */
 | 
						|
    if (rt_wlan_workqueue_dowork(rt_wlan_mgnt_work, msg) != RT_EOK)
 | 
						|
    {
 | 
						|
        rt_free(msg);
 | 
						|
        RT_WLAN_LOG_E("wlan mgnt do work fail");
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
    return RT_EOK;
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
static rt_err_t rt_wlan_sta_info_add(struct rt_wlan_info *info, int timeout)
 | 
						|
{
 | 
						|
    struct rt_wlan_sta_list *sta_list;
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    if (_ap_is_null() || (info == RT_NULL)) return RT_EOK;
 | 
						|
 | 
						|
    err = rt_mutex_take(&sta_info_mutex, rt_tick_from_millisecond(timeout));
 | 
						|
    if (err == RT_EOK)
 | 
						|
    {
 | 
						|
        /* malloc memory */
 | 
						|
        sta_list = rt_malloc(sizeof(struct rt_wlan_sta_list));
 | 
						|
        if (sta_list == RT_NULL)
 | 
						|
        {
 | 
						|
            rt_mutex_release(&sta_info_mutex);
 | 
						|
            RT_WLAN_LOG_E("sta list malloc failed!");
 | 
						|
            return -RT_ENOMEM;
 | 
						|
        }
 | 
						|
        sta_list->next = RT_NULL;
 | 
						|
        sta_list->info = *info;
 | 
						|
 | 
						|
        /* Append sta info */
 | 
						|
        sta_list->next = sta_info.node;
 | 
						|
        sta_info.node = sta_list;
 | 
						|
        /* num++ */
 | 
						|
        sta_info.num ++;
 | 
						|
        rt_mutex_release(&sta_info_mutex);
 | 
						|
        RT_WLAN_LOG_I("sta associated mac:%02x:%02x:%02x:%02x:%02x:%02x",
 | 
						|
                      info->bssid[0], info->bssid[1], info->bssid[2],
 | 
						|
                      info->bssid[3], info->bssid[4], info->bssid[5]);
 | 
						|
    }
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
static rt_err_t rt_wlan_sta_info_del(struct rt_wlan_info *info, int timeout)
 | 
						|
{
 | 
						|
    struct rt_wlan_sta_list *sta_list, *sta_prve;
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    if (_ap_is_null() || (info == RT_NULL)) return RT_EOK;
 | 
						|
 | 
						|
    err = rt_mutex_take(&sta_info_mutex, rt_tick_from_millisecond(timeout));
 | 
						|
    if (err == RT_EOK)
 | 
						|
    {
 | 
						|
        /* traversing the list */
 | 
						|
        for (sta_list = sta_info.node, sta_prve = RT_NULL; sta_list != RT_NULL;
 | 
						|
                sta_prve = sta_list, sta_list = sta_list->next)
 | 
						|
        {
 | 
						|
            /* find mac addr */
 | 
						|
            if (rt_memcmp(&sta_list->info.bssid[0], &info->bssid[0], RT_WLAN_BSSID_MAX_LENGTH) == 0)
 | 
						|
            {
 | 
						|
                if (sta_prve == RT_NULL)
 | 
						|
                {
 | 
						|
                    sta_info.node = sta_list->next;
 | 
						|
                }
 | 
						|
                else
 | 
						|
                {
 | 
						|
                    sta_prve->next = sta_list->next;
 | 
						|
                }
 | 
						|
                sta_info.num --;
 | 
						|
                rt_free(sta_list);
 | 
						|
                break;
 | 
						|
            }
 | 
						|
        }
 | 
						|
        rt_mutex_release(&sta_info_mutex);
 | 
						|
        RT_WLAN_LOG_I("sta exit mac:%02x:%02x:%02x:%02x:%02x:%02x",
 | 
						|
                      info->bssid[0], info->bssid[1], info->bssid[2],
 | 
						|
                      info->bssid[3], info->bssid[4], info->bssid[5]);
 | 
						|
    }
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
static rt_err_t rt_wlan_sta_info_del_all(int timeout)
 | 
						|
{
 | 
						|
    struct rt_wlan_sta_list *sta_list, *sta_next;
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    err = rt_mutex_take(&sta_info_mutex, rt_tick_from_millisecond(timeout));
 | 
						|
    if (err == RT_EOK)
 | 
						|
    {
 | 
						|
        /* traversing the list */
 | 
						|
        for (sta_list = sta_info.node; sta_list != RT_NULL; sta_list = sta_next)
 | 
						|
        {
 | 
						|
            sta_next = sta_list->next;
 | 
						|
            sta_info.num --;
 | 
						|
            rt_free(sta_list);
 | 
						|
        }
 | 
						|
        rt_mutex_release(&sta_info_mutex);
 | 
						|
    }
 | 
						|
    if (sta_info.num != 0)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_W("\n\n!!!Program runing exception!!!\n\n");
 | 
						|
    }
 | 
						|
    sta_info.num = 0;
 | 
						|
    sta_info.node = RT_NULL;
 | 
						|
    return err;
 | 
						|
}
 | 
						|
#ifdef RT_WLAN_AUTO_CONNECT_ENABLE
 | 
						|
static void rt_wlan_auto_connect_run(struct rt_work *work, void *parameter)
 | 
						|
{
 | 
						|
    struct rt_wlan_cfg_info cfg_info;
 | 
						|
    char *password = RT_NULL;
 | 
						|
    rt_base_t level;
 | 
						|
 | 
						|
    RT_WLAN_LOG_D("F:%s is run", __FUNCTION__);
 | 
						|
 | 
						|
    if (rt_mutex_take(&mgnt_mutex, 0) != RT_EOK)
 | 
						|
        goto exit;
 | 
						|
 | 
						|
    /* auto connect status is disable or wifi is connect or connecting, exit */
 | 
						|
    if (_is_do_connect() == RT_FALSE)
 | 
						|
    {
 | 
						|
        id = 0;
 | 
						|
        RT_WLAN_LOG_D("not connection");
 | 
						|
        goto exit;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Read the next configuration */
 | 
						|
    rt_memset(&cfg_info, 0, sizeof(struct rt_wlan_cfg_info));
 | 
						|
    if (rt_wlan_cfg_read_index(&cfg_info, id ++) == 0)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("read cfg fail");
 | 
						|
        id = 0;
 | 
						|
        goto exit;
 | 
						|
    }
 | 
						|
 | 
						|
    if (id >= rt_wlan_cfg_get_num()) id = 0;
 | 
						|
 | 
						|
    if ((cfg_info.key.len > 0) && (cfg_info.key.len <= RT_WLAN_PASSWORD_MAX_LENGTH))
 | 
						|
    {
 | 
						|
        cfg_info.key.val[cfg_info.key.len] = '\0';
 | 
						|
        password = (char *)(&cfg_info.key.val[0]);
 | 
						|
    }
 | 
						|
    rt_wlan_connect((char *)cfg_info.info.ssid.val, password);
 | 
						|
exit:
 | 
						|
    rt_mutex_release(&mgnt_mutex);
 | 
						|
    level = rt_hw_interrupt_disable();
 | 
						|
    rt_memset(work, 0, sizeof(struct rt_work));
 | 
						|
    rt_hw_interrupt_enable(level);
 | 
						|
}
 | 
						|
 | 
						|
static void rt_wlan_cyclic_check(void *parameter)
 | 
						|
{
 | 
						|
    static struct rt_work work;
 | 
						|
    rt_base_t level;
 | 
						|
 | 
						|
    if ((_is_do_connect() == RT_TRUE) && (work.work_func == RT_NULL))
 | 
						|
    {
 | 
						|
        level = rt_hw_interrupt_disable();
 | 
						|
        rt_work_init(&work, rt_wlan_auto_connect_run, RT_NULL);
 | 
						|
        rt_hw_interrupt_enable(level);
 | 
						|
        if(rt_work_submit(&work,RT_TICK_PER_SECOND) != RT_EOK)
 | 
						|
        {
 | 
						|
            level = rt_hw_interrupt_disable();
 | 
						|
            rt_memset(&work, 0, sizeof(struct rt_work));
 | 
						|
            rt_hw_interrupt_enable(level);
 | 
						|
        }
 | 
						|
    }
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
static void rt_wlan_event_dispatch(struct rt_wlan_device *device, rt_wlan_dev_event_t event, struct rt_wlan_buff *buff, void *parameter)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_NULL;
 | 
						|
    rt_wlan_event_t user_event = RT_WLAN_EVT_MAX;
 | 
						|
    int i;
 | 
						|
    struct rt_wlan_buff user_buff = { 0 };
 | 
						|
 | 
						|
    if (buff)
 | 
						|
    {
 | 
						|
        user_buff = *buff;
 | 
						|
    }
 | 
						|
    /* Event Handle */
 | 
						|
    switch (event)
 | 
						|
    {
 | 
						|
    case RT_WLAN_DEV_EVT_CONNECT:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: CONNECT");
 | 
						|
#ifdef RT_WLAN_AUTO_CONNECT_ENABLE
 | 
						|
        id = 0;
 | 
						|
#endif
 | 
						|
        _sta_mgnt.state |= RT_WLAN_STATE_CONNECT;
 | 
						|
        _sta_mgnt.state &= ~RT_WLAN_STATE_CONNECTING;
 | 
						|
        user_event = RT_WLAN_EVT_STA_CONNECTED;
 | 
						|
        TIME_STOP();
 | 
						|
        user_buff.data = &_sta_mgnt.info;
 | 
						|
        user_buff.len = sizeof(struct rt_wlan_info);
 | 
						|
        RT_WLAN_LOG_I("wifi connect success ssid:%s", &_sta_mgnt.info.ssid.val[0]);
 | 
						|
 | 
						|
#ifdef RT_WLAN_CFG_ENABLE
 | 
						|
        {
 | 
						|
            struct rt_wlan_cfg_info cfg_info;
 | 
						|
            rt_memset(&cfg_info, 0, sizeof(cfg_info));
 | 
						|
            /* save config */
 | 
						|
            if (rt_wlan_is_connected() == RT_TRUE)
 | 
						|
            {
 | 
						|
                rt_enter_critical();
 | 
						|
                cfg_info.info = _sta_mgnt.info;
 | 
						|
                cfg_info.key = _sta_mgnt.key;
 | 
						|
                rt_exit_critical();
 | 
						|
                RT_WLAN_LOG_D("run save config! ssid:%s len%d", _sta_mgnt.info.ssid.val, _sta_mgnt.info.ssid.len);
 | 
						|
                rt_wlan_cfg_save(&cfg_info);
 | 
						|
            }
 | 
						|
        }
 | 
						|
#endif
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_CONNECT_FAIL:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: CONNECT_FAIL");
 | 
						|
        _sta_mgnt.state &= ~RT_WLAN_STATE_CONNECT;
 | 
						|
        _sta_mgnt.state &= ~RT_WLAN_STATE_CONNECTING;
 | 
						|
        _sta_mgnt.state &= ~RT_WLAN_STATE_READY;
 | 
						|
        user_event = RT_WLAN_EVT_STA_CONNECTED_FAIL;
 | 
						|
        user_buff.data = &_sta_mgnt.info;
 | 
						|
        user_buff.len = sizeof(struct rt_wlan_info);
 | 
						|
        if (rt_wlan_get_autoreconnect_mode())
 | 
						|
        {
 | 
						|
            TIME_START();
 | 
						|
        }
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_DISCONNECT:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: DISCONNECT");
 | 
						|
        _sta_mgnt.state &= ~RT_WLAN_STATE_CONNECT;
 | 
						|
        _sta_mgnt.state &= ~RT_WLAN_STATE_READY;
 | 
						|
        user_event = RT_WLAN_EVT_STA_DISCONNECTED;
 | 
						|
        user_buff.data = &_sta_mgnt.info;
 | 
						|
        user_buff.len = sizeof(struct rt_wlan_info);
 | 
						|
        if (rt_wlan_get_autoreconnect_mode())
 | 
						|
        {
 | 
						|
            TIME_START();
 | 
						|
        }
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_AP_START:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: AP_START");
 | 
						|
        _ap_mgnt.state |= RT_WLAN_STATE_ACTIVE;
 | 
						|
        user_event = RT_WLAN_EVT_AP_START;
 | 
						|
        user_buff.data = &_ap_mgnt.info;
 | 
						|
        user_buff.len = sizeof(struct rt_wlan_info);
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_AP_STOP:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: AP_STOP");
 | 
						|
        _ap_mgnt.state &= ~RT_WLAN_STATE_ACTIVE;
 | 
						|
        user_event = RT_WLAN_EVT_AP_STOP;
 | 
						|
        err = rt_wlan_sta_info_del_all(RT_WAITING_FOREVER);
 | 
						|
        if (err != RT_NULL)
 | 
						|
        {
 | 
						|
            RT_WLAN_LOG_W("AP_STOP event handle fail");
 | 
						|
        }
 | 
						|
        user_buff.data = &_ap_mgnt.info;
 | 
						|
        user_buff.len = sizeof(struct rt_wlan_info);
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_AP_ASSOCIATED:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: ASSOCIATED");
 | 
						|
        user_event = RT_WLAN_EVT_AP_ASSOCIATED;
 | 
						|
        if (user_buff.len != sizeof(struct rt_wlan_info))
 | 
						|
            break;
 | 
						|
        err = rt_wlan_sta_info_add(user_buff.data, RT_WAITING_FOREVER);
 | 
						|
        if (err != RT_EOK)
 | 
						|
        {
 | 
						|
            RT_WLAN_LOG_W("AP_ASSOCIATED event handle fail");
 | 
						|
        }
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_AP_DISASSOCIATED:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: DISASSOCIATED");
 | 
						|
        user_event = RT_WLAN_EVT_AP_DISASSOCIATED;
 | 
						|
        if (user_buff.len != sizeof(struct rt_wlan_info))
 | 
						|
            break;
 | 
						|
        err = rt_wlan_sta_info_del(user_buff.data, RT_WAITING_FOREVER);
 | 
						|
        if (err != RT_EOK)
 | 
						|
        {
 | 
						|
            RT_WLAN_LOG_W("AP_DISASSOCIATED event handle fail");
 | 
						|
        }
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_AP_ASSOCIATE_FAILED:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: AP_ASSOCIATE_FAILED");
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_SCAN_REPORT:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: SCAN_REPORT");
 | 
						|
        user_event = RT_WLAN_EVT_SCAN_REPORT;
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    case RT_WLAN_DEV_EVT_SCAN_DONE:
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: SCAN_DONE");
 | 
						|
        user_event = RT_WLAN_EVT_SCAN_DONE;
 | 
						|
        break;
 | 
						|
    }
 | 
						|
    default :
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("event: UNKNOWN");
 | 
						|
        return;
 | 
						|
    }
 | 
						|
    }
 | 
						|
 | 
						|
    /* send event */
 | 
						|
    COMPLETE_LOCK();
 | 
						|
    for (i = 0; i < sizeof(complete_tab) / sizeof(complete_tab[0]); i++)
 | 
						|
    {
 | 
						|
        if ((complete_tab[i] != RT_NULL))
 | 
						|
        {
 | 
						|
            complete_tab[i]->event_flag |= 0x1 << event;
 | 
						|
            rt_event_send(&complete_tab[i]->complete, 0x1 << event);
 | 
						|
            RT_WLAN_LOG_D("&complete_tab[i]->complete:0x%08x", &complete_tab[i]->complete);
 | 
						|
        }
 | 
						|
    }
 | 
						|
    COMPLETE_UNLOCK();
 | 
						|
#ifdef RT_WLAN_WORK_THREAD_ENABLE
 | 
						|
    rt_wlan_send_to_thread(user_event, user_buff.data, user_buff.len);
 | 
						|
#else
 | 
						|
    {
 | 
						|
        void *user_parameter;
 | 
						|
        rt_wlan_event_handler handler = RT_NULL;
 | 
						|
        rt_base_t level;
 | 
						|
        /* Get user callback */
 | 
						|
        if (user_event < RT_WLAN_EVT_MAX)
 | 
						|
        {
 | 
						|
            level = rt_hw_interrupt_disable();
 | 
						|
            handler = event_tab[user_event].handler;
 | 
						|
            user_parameter = event_tab[user_event].parameter;
 | 
						|
            rt_hw_interrupt_enable(level);
 | 
						|
        }
 | 
						|
 | 
						|
        /* run user callback fun */
 | 
						|
        if (handler)
 | 
						|
        {
 | 
						|
            RT_WLAN_LOG_D("unknown thread run user callback, event:%d", user_event);
 | 
						|
            handler(user_event, &user_buff, user_parameter);
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
static struct rt_wlan_complete_des *rt_wlan_complete_create(const char *name)
 | 
						|
{
 | 
						|
    struct rt_wlan_complete_des *complete;
 | 
						|
    int i;
 | 
						|
 | 
						|
    complete = rt_malloc(sizeof(struct rt_wlan_complete_des));
 | 
						|
    if (complete == RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("complete event create failed");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return complete;
 | 
						|
    }
 | 
						|
    rt_event_init(&complete->complete, name, RT_IPC_FLAG_FIFO);
 | 
						|
    complete->event_flag = 0;
 | 
						|
    //protect
 | 
						|
    COMPLETE_LOCK();
 | 
						|
    for (i = 0; i < sizeof(complete_tab) / sizeof(complete_tab[0]); i++)
 | 
						|
    {
 | 
						|
        if (complete_tab[i] == RT_NULL)
 | 
						|
        {
 | 
						|
            complete->index = i;
 | 
						|
            complete_tab[i] = complete;
 | 
						|
            break;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    COMPLETE_UNLOCK();
 | 
						|
 | 
						|
    if (i >= sizeof(complete_tab) / sizeof(complete_tab[0]))
 | 
						|
    {
 | 
						|
        rt_event_detach(&complete->complete);
 | 
						|
        rt_free(complete);
 | 
						|
        complete = RT_NULL;
 | 
						|
    }
 | 
						|
 | 
						|
    return complete;
 | 
						|
}
 | 
						|
 | 
						|
static rt_err_t rt_wlan_complete_wait(struct rt_wlan_complete_des *complete, rt_uint32_t event,
 | 
						|
                                      rt_uint32_t timeout, rt_uint32_t *recved)
 | 
						|
{
 | 
						|
    if (complete == RT_NULL)
 | 
						|
    {
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Check whether there is a waiting event */
 | 
						|
    if (complete->event_flag & event)
 | 
						|
    {
 | 
						|
        *recved = complete->event_flag;
 | 
						|
        return RT_EOK;
 | 
						|
    }
 | 
						|
    else
 | 
						|
    {
 | 
						|
        return rt_event_recv(&complete->complete, event, RT_EVENT_FLAG_OR,
 | 
						|
                             rt_tick_from_millisecond(timeout), recved);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
static void rt_wlan_complete_delete(struct rt_wlan_complete_des *complete)
 | 
						|
{
 | 
						|
    if (complete == RT_NULL)
 | 
						|
    {
 | 
						|
        return;
 | 
						|
    }
 | 
						|
    COMPLETE_LOCK();
 | 
						|
    complete_tab[complete->index] = RT_NULL;
 | 
						|
    COMPLETE_UNLOCK();
 | 
						|
    rt_event_detach(&complete->complete);
 | 
						|
    rt_free(complete);
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_set_mode(const char *dev_name, rt_wlan_mode_t mode)
 | 
						|
{
 | 
						|
    rt_device_t device = RT_NULL;
 | 
						|
    rt_err_t err;
 | 
						|
    rt_int8_t up_event_flag = 0;
 | 
						|
    rt_wlan_dev_event_handler handler = RT_NULL;
 | 
						|
 | 
						|
    if ((dev_name == RT_NULL) || (mode >= RT_WLAN_MODE_MAX))
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("Parameter Wrongful name:%s mode:%d", dev_name, mode);
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
 | 
						|
    RT_WLAN_LOG_D("%s is run dev_name:%s mode:%s%s%s", __FUNCTION__, dev_name,
 | 
						|
                  mode == RT_WLAN_NONE ? "NONE" : "",
 | 
						|
                  mode == RT_WLAN_STATION ? "STA" : "",
 | 
						|
                  mode == RT_WLAN_AP ? "AP" : ""
 | 
						|
                 );
 | 
						|
 | 
						|
    /* find device */
 | 
						|
    device = rt_device_find(dev_name);
 | 
						|
    if (device == RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("not find device, set mode failed! name:%s", dev_name);
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    MGNT_LOCK();
 | 
						|
    if (RT_WLAN_DEVICE(device)->mode == mode)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_D("L:%d this device mode is set");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return RT_EOK;
 | 
						|
    }
 | 
						|
 | 
						|
    if ((mode == RT_WLAN_STATION) &&
 | 
						|
            (RT_WLAN_DEVICE(device)->flags & RT_WLAN_FLAG_AP_ONLY))
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_I("this device ap mode only");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
    else if ((mode == RT_WLAN_AP) &&
 | 
						|
             (RT_WLAN_DEVICE(device)->flags & RT_WLAN_FLAG_STA_ONLY))
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_I("this device sta mode only");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
 | 
						|
    /*
 | 
						|
     * device == sta  and change to ap,  should deinit
 | 
						|
     * device == ap   and change to sta, should deinit
 | 
						|
    */
 | 
						|
    if (((mode == RT_WLAN_STATION) && (RT_WLAN_DEVICE(device) == AP_DEVICE())) ||
 | 
						|
            ((mode == RT_WLAN_AP) && (RT_WLAN_DEVICE(device) == STA_DEVICE())))
 | 
						|
    {
 | 
						|
        err = rt_wlan_set_mode(dev_name, RT_WLAN_NONE);
 | 
						|
        if (err != RT_EOK)
 | 
						|
        {
 | 
						|
            RT_WLAN_LOG_E("change mode failed!");
 | 
						|
            MGNT_UNLOCK();
 | 
						|
            return err;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /* init device */
 | 
						|
    err = rt_wlan_dev_init(RT_WLAN_DEVICE(device), mode);
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("F:%s L:%d wlan init failed", __FUNCTION__, __LINE__);
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
 | 
						|
    /* the mode is none */
 | 
						|
    if (mode == RT_WLAN_NONE)
 | 
						|
    {
 | 
						|
        if (_sta_mgnt.device == RT_WLAN_DEVICE(device))
 | 
						|
        {
 | 
						|
            _sta_mgnt.device = RT_NULL;
 | 
						|
            _sta_mgnt.state = 0;
 | 
						|
            up_event_flag = 1;
 | 
						|
            handler = RT_NULL;
 | 
						|
        }
 | 
						|
        else if (_ap_mgnt.device == RT_WLAN_DEVICE(device))
 | 
						|
        {
 | 
						|
            _ap_mgnt.state = 0;
 | 
						|
            _ap_mgnt.device = RT_NULL;
 | 
						|
            up_event_flag = 1;
 | 
						|
            handler = RT_NULL;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    /* save sta device */
 | 
						|
    else if (mode == RT_WLAN_STATION)
 | 
						|
    {
 | 
						|
        up_event_flag = 1;
 | 
						|
        handler = rt_wlan_event_dispatch;
 | 
						|
        _sta_mgnt.device = RT_WLAN_DEVICE(device);
 | 
						|
    }
 | 
						|
    /* save ap device */
 | 
						|
    else if (mode == RT_WLAN_AP)
 | 
						|
    {
 | 
						|
        up_event_flag = 1;
 | 
						|
        handler = rt_wlan_event_dispatch;
 | 
						|
        _ap_mgnt.device = RT_WLAN_DEVICE(device);
 | 
						|
    }
 | 
						|
 | 
						|
    /* update dev event handle */
 | 
						|
    if (up_event_flag == 1)
 | 
						|
    {
 | 
						|
        if (handler)
 | 
						|
        {
 | 
						|
            if (mode == RT_WLAN_STATION)
 | 
						|
            {
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_CONNECT, handler, RT_NULL);
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_CONNECT_FAIL, handler, RT_NULL);
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_DISCONNECT, handler, RT_NULL);
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_SCAN_REPORT, handler, RT_NULL);
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_SCAN_DONE, handler, RT_NULL);
 | 
						|
            }
 | 
						|
            else if (mode == RT_WLAN_AP)
 | 
						|
            {
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_AP_START, handler, RT_NULL);
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_AP_STOP, handler, RT_NULL);
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_AP_ASSOCIATED, handler, RT_NULL);
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_AP_DISASSOCIATED, handler, RT_NULL);
 | 
						|
                rt_wlan_dev_register_event_handler(RT_WLAN_DEVICE(device), RT_WLAN_DEV_EVT_AP_ASSOCIATE_FAILED, handler, RT_NULL);
 | 
						|
            }
 | 
						|
        }
 | 
						|
        else
 | 
						|
        {
 | 
						|
            rt_wlan_dev_event_t event;
 | 
						|
            handler = rt_wlan_event_dispatch;
 | 
						|
            for (event = RT_WLAN_DEV_EVT_INIT_DONE; event < RT_WLAN_DEV_EVT_MAX; event++)
 | 
						|
            {
 | 
						|
                rt_wlan_dev_unregister_event_handler(RT_WLAN_DEVICE(device), event, handler);
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
    MGNT_UNLOCK();
 | 
						|
 | 
						|
    /* Mount protocol */
 | 
						|
#if defined(RT_WLAN_PROT_ENABLE) && defined(RT_WLAN_DEFAULT_PROT)
 | 
						|
    if (err == RT_EOK)
 | 
						|
    {
 | 
						|
        rt_wlan_prot_attach(dev_name, RT_WLAN_DEFAULT_PROT);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_wlan_mode_t rt_wlan_get_mode(const char *dev_name)
 | 
						|
{
 | 
						|
    rt_device_t device = RT_NULL;
 | 
						|
    rt_wlan_mode_t mode;
 | 
						|
 | 
						|
    if (dev_name == RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("name is null");
 | 
						|
        return RT_WLAN_NONE;
 | 
						|
    }
 | 
						|
 | 
						|
    /* find device */
 | 
						|
    device = rt_device_find(dev_name);
 | 
						|
    if (device == RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("device not find! name:%s", dev_name);
 | 
						|
        return RT_WLAN_NONE;
 | 
						|
    }
 | 
						|
 | 
						|
    /* get mode */
 | 
						|
    mode = RT_WLAN_DEVICE(device)->mode;
 | 
						|
    RT_WLAN_LOG_D("%s is run dev_name:%s mode:%s%s%s", __FUNCTION__, dev_name,
 | 
						|
                  mode == RT_WLAN_NONE ? "NONE" : "",
 | 
						|
                  mode == RT_WLAN_STATION ? "STA" : "",
 | 
						|
                  mode == RT_WLAN_AP ? "AP" : "");
 | 
						|
 | 
						|
    return mode;
 | 
						|
}
 | 
						|
 | 
						|
#ifdef RT_WLAN_JOIN_SCAN_BY_MGNT
 | 
						|
static void rt_wlan_join_scan_callback(int event, struct rt_wlan_buff *buff, void *parameter)
 | 
						|
{
 | 
						|
    struct rt_wlan_info *info = RT_NULL;
 | 
						|
    struct rt_wlan_info *tgt_info = RT_NULL;
 | 
						|
 | 
						|
    RT_ASSERT(event == RT_WLAN_EVT_SCAN_REPORT);
 | 
						|
    RT_ASSERT(buff != RT_NULL);
 | 
						|
    RT_ASSERT(parameter != RT_NULL);
 | 
						|
 | 
						|
    info = (struct rt_wlan_info *)buff->data;
 | 
						|
    tgt_info = (struct rt_wlan_info *)parameter;
 | 
						|
 | 
						|
 | 
						|
    RT_WLAN_LOG_D("%s info len:%d tgt info len:%d", __FUNCTION__,info->ssid.len,tgt_info->ssid.len);
 | 
						|
    RT_WLAN_LOG_D("%s info ssid:%s tgt info ssid:%s", __FUNCTION__,&info->ssid.val[0],&tgt_info->ssid.val[0]);
 | 
						|
 | 
						|
    if(rt_memcmp(&info->ssid.val[0], &tgt_info->ssid.val[0], info->ssid.len) == 0 &&
 | 
						|
            info->ssid.len == tgt_info->ssid.len)
 | 
						|
    {
 | 
						|
        /*Get the rssi the max ap*/
 | 
						|
        if((info->rssi > tgt_info->rssi) || (tgt_info->rssi == 0))
 | 
						|
        {
 | 
						|
            tgt_info->security  = info->security;
 | 
						|
            tgt_info->band      = info->band;
 | 
						|
            tgt_info->datarate  = info->datarate;
 | 
						|
            tgt_info->channel   = info->channel;
 | 
						|
            tgt_info->rssi      = info->rssi;
 | 
						|
            tgt_info->hidden    = info->hidden;
 | 
						|
            /* hwaddr */
 | 
						|
            rt_memcpy(tgt_info->bssid,info->bssid,RT_WLAN_BSSID_MAX_LENGTH);
 | 
						|
        }
 | 
						|
    }
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
rt_err_t rt_wlan_connect(const char *ssid, const char *password)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
    int ssid_len = 0;
 | 
						|
    struct rt_wlan_info info;
 | 
						|
    struct rt_wlan_complete_des *complete;
 | 
						|
    rt_uint32_t set = 0, recved = 0;
 | 
						|
 | 
						|
    /* sta dev Can't be NULL */
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run ssid:%s password:%s", __FUNCTION__, ssid, password);
 | 
						|
    if (ssid == RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("ssid is null!");
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
    ssid_len = rt_strlen(ssid);
 | 
						|
    if (ssid_len > RT_WLAN_SSID_MAX_LENGTH)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("ssid is to long! ssid:%s len:%d", ssid, ssid_len);
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
 | 
						|
    if ((rt_wlan_is_connected() == RT_TRUE) &&
 | 
						|
            (rt_strcmp((char *)&_sta_mgnt.info.ssid.val[0], ssid) == 0))
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_I("wifi is connect ssid:%s", ssid);
 | 
						|
        return RT_EOK;
 | 
						|
    }
 | 
						|
    /* get info from cache */
 | 
						|
    INVALID_INFO(&info);
 | 
						|
    MGNT_LOCK();
 | 
						|
 | 
						|
    rt_memcpy(&info.ssid.val[0],ssid,rt_strlen(ssid));
 | 
						|
    info.ssid.len = rt_strlen(ssid);
 | 
						|
 | 
						|
#ifdef RT_WLAN_JOIN_SCAN_BY_MGNT
 | 
						|
    err = rt_wlan_register_event_handler(RT_WLAN_EVT_SCAN_REPORT,rt_wlan_join_scan_callback,&info);
 | 
						|
    if(err != RT_EOK)
 | 
						|
    {
 | 
						|
        LOG_E("Scan register user callback error:%d!\n",err);
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
 | 
						|
    err = rt_wlan_scan_with_info(&info);
 | 
						|
    if(err != RT_EOK)
 | 
						|
    {
 | 
						|
        LOG_E("Scan with info error:%d!\n",err);
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
 | 
						|
    if (info.channel <= 0)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_W("not find ap! ssid:%s,info.ssid.len=%d", ssid,info.ssid.len);
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
 | 
						|
    RT_WLAN_LOG_D("find best info ssid:%s mac: %02x %02x %02x %02x %02x %02x",
 | 
						|
                  info.ssid.val, info.bssid[0], info.bssid[1], info.bssid[2], info.bssid[3], info.bssid[4], info.bssid[5]);
 | 
						|
#endif
 | 
						|
 | 
						|
    /* create event wait complete */
 | 
						|
    complete = rt_wlan_complete_create("join");
 | 
						|
    if (complete == RT_NULL)
 | 
						|
    {
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ENOMEM;
 | 
						|
    }
 | 
						|
    /* run connect adv */
 | 
						|
    err = rt_wlan_connect_adv(&info, password);
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        rt_wlan_complete_delete(complete);
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Initializing events that need to wait */
 | 
						|
    set |= 0x1 << RT_WLAN_DEV_EVT_CONNECT;
 | 
						|
    set |= 0x1 << RT_WLAN_DEV_EVT_CONNECT_FAIL;
 | 
						|
    /* Check whether there is a waiting event */
 | 
						|
    rt_wlan_complete_wait(complete, set, RT_WLAN_CONNECT_WAIT_MS, &recved);
 | 
						|
    rt_wlan_complete_delete(complete);
 | 
						|
    /* check event */
 | 
						|
    set = 0x1 << RT_WLAN_DEV_EVT_CONNECT;
 | 
						|
    if (!(recved & set))
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_I("wifi connect failed!");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_connect_adv(struct rt_wlan_info *info, const char *password)
 | 
						|
{
 | 
						|
    int password_len = 0;
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    if (info == RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("info is null!");
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run ssid:%s password:%s", __FUNCTION__, info->ssid.val, password);
 | 
						|
    /* Parameter checking */
 | 
						|
    if (password != RT_NULL)
 | 
						|
    {
 | 
						|
        password_len = rt_strlen(password);
 | 
						|
        if (password_len > RT_WLAN_PASSWORD_MAX_LENGTH)
 | 
						|
        {
 | 
						|
            RT_WLAN_LOG_E("password is to long! password:%s len:%d", password, password_len);
 | 
						|
            return -RT_EINVAL;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    if (info->ssid.len == 0 || info->ssid.len > RT_WLAN_SSID_MAX_LENGTH)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("ssid is zero or to long! ssid:%s len:%d", info->ssid.val, info->ssid.len);
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
    /* is connect ? */
 | 
						|
    MGNT_LOCK();
 | 
						|
    if (rt_wlan_is_connected())
 | 
						|
    {
 | 
						|
        if ((_sta_mgnt.info.ssid.len == info->ssid.len) &&
 | 
						|
                (_sta_mgnt.key.len == password_len) &&
 | 
						|
                (rt_memcmp(&_sta_mgnt.info.ssid.val[0], &info->ssid.val[0], info->ssid.len) == 0) &&
 | 
						|
                (rt_memcmp(&_sta_mgnt.info.bssid[0], &info->bssid[0], RT_WLAN_BSSID_MAX_LENGTH) == 0) &&
 | 
						|
                (rt_memcmp(&_sta_mgnt.key.val[0], password, password_len) == 0))
 | 
						|
        {
 | 
						|
            RT_WLAN_LOG_I("wifi Already Connected");
 | 
						|
            MGNT_UNLOCK();
 | 
						|
            return RT_EOK;
 | 
						|
        }
 | 
						|
 | 
						|
        err = rt_wlan_disconnect();
 | 
						|
        if (err != RT_EOK)
 | 
						|
        {
 | 
						|
            MGNT_UNLOCK();
 | 
						|
            return err;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /* save info */
 | 
						|
    rt_enter_critical();
 | 
						|
    _sta_mgnt.info = *info;
 | 
						|
    rt_memcpy(&_sta_mgnt.key.val, password, password_len);
 | 
						|
    _sta_mgnt.key.len = password_len;
 | 
						|
    _sta_mgnt.key.val[password_len] = '\0';
 | 
						|
    rt_exit_critical();
 | 
						|
    /* run wifi connect */
 | 
						|
    _sta_mgnt.state |= RT_WLAN_STATE_CONNECTING;
 | 
						|
 | 
						|
    err = rt_wlan_dev_fast_connect(_sta_mgnt.device, info, password, password_len);
 | 
						|
    if(err != RT_EOK)
 | 
						|
    {
 | 
						|
        err = rt_wlan_dev_connect(_sta_mgnt.device, info, password, password_len);
 | 
						|
        if (err != RT_EOK)
 | 
						|
        {
 | 
						|
            rt_enter_critical();
 | 
						|
            rt_memset(&_sta_mgnt.info, 0, sizeof(struct rt_wlan_ssid));
 | 
						|
            rt_memset(&_sta_mgnt.key, 0, sizeof(struct rt_wlan_key));
 | 
						|
            rt_exit_critical();
 | 
						|
            _sta_mgnt.state &= ~RT_WLAN_STATE_CONNECTING;
 | 
						|
            MGNT_UNLOCK();
 | 
						|
            return err;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_disconnect(void)
 | 
						|
{
 | 
						|
    rt_err_t err;
 | 
						|
    struct rt_wlan_complete_des *complete;
 | 
						|
    rt_uint32_t recved = 0, set = 0;
 | 
						|
 | 
						|
    /* ap dev Can't be empty */
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
 | 
						|
    /* run disconnect */
 | 
						|
    MGNT_LOCK();
 | 
						|
    /* create event wait complete */
 | 
						|
    complete = rt_wlan_complete_create("disc");
 | 
						|
    if (complete == RT_NULL)
 | 
						|
    {
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ENOMEM;
 | 
						|
    }
 | 
						|
    err = rt_wlan_dev_disconnect(_sta_mgnt.device);
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("wifi disconnect fail");
 | 
						|
        rt_wlan_complete_delete(complete);
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
    /* Initializing events that need to wait */
 | 
						|
    set |= 0x1 << RT_WLAN_DEV_EVT_DISCONNECT;
 | 
						|
    /* Check whether there is a waiting event */
 | 
						|
    rt_wlan_complete_wait(complete, set, RT_WLAN_CONNECT_WAIT_MS, &recved);
 | 
						|
    rt_wlan_complete_delete(complete);
 | 
						|
    /* check event */
 | 
						|
    set = 0x1 << RT_WLAN_DEV_EVT_DISCONNECT;
 | 
						|
    if (!(recved & set))
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("disconnect failed!");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_I("disconnect success!");
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_bool_t rt_wlan_is_connected(void)
 | 
						|
{
 | 
						|
    rt_bool_t _connect;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return RT_FALSE;
 | 
						|
    }
 | 
						|
    _connect = _sta_mgnt.state & RT_WLAN_STATE_CONNECT ? RT_TRUE : RT_FALSE;
 | 
						|
    RT_WLAN_LOG_D("%s is run : %s", __FUNCTION__, _connect ? "connect" : "disconnect");
 | 
						|
    return _connect;
 | 
						|
}
 | 
						|
 | 
						|
rt_bool_t rt_wlan_is_ready(void)
 | 
						|
{
 | 
						|
    rt_bool_t _ready;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return RT_FALSE;
 | 
						|
    }
 | 
						|
    _ready = _sta_mgnt.state & RT_WLAN_STATE_READY ? RT_TRUE : RT_FALSE;
 | 
						|
    RT_WLAN_LOG_D("%s is run : %s", __FUNCTION__, _ready ? "ready" : "not ready");
 | 
						|
    return _ready;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_set_mac(rt_uint8_t mac[6])
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run mac: %02x:%02x:%02x:%02x:%02x:%02x",
 | 
						|
                  __FUNCTION__, mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
 | 
						|
 | 
						|
    MGNT_LOCK();
 | 
						|
    err = rt_wlan_dev_set_mac(STA_DEVICE(), mac);
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("set sta mac addr fail");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_get_mac(rt_uint8_t mac[6])
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    MGNT_LOCK();
 | 
						|
    err = rt_wlan_dev_get_mac(STA_DEVICE(), mac);
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("get sta mac addr fail");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run mac: %02x:%02x:%02x:%02x:%02x:%02x",
 | 
						|
                  __FUNCTION__, mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_get_info(struct rt_wlan_info *info)
 | 
						|
{
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
 | 
						|
    if (rt_wlan_is_connected() == RT_TRUE)
 | 
						|
    {
 | 
						|
        *info = _sta_mgnt.info;
 | 
						|
        info->rssi = rt_wlan_get_rssi();
 | 
						|
        return RT_EOK;
 | 
						|
    }
 | 
						|
    return -RT_ERROR;
 | 
						|
}
 | 
						|
 | 
						|
int rt_wlan_get_rssi(void)
 | 
						|
{
 | 
						|
    int rssi = 0;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    MGNT_LOCK();
 | 
						|
    rssi = rt_wlan_dev_get_rssi(STA_DEVICE());
 | 
						|
    RT_WLAN_LOG_D("%s is run rssi:%d", __FUNCTION__, rssi);
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return rssi;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_start_ap(const char *ssid, const char *password)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
    int ssid_len = 0;
 | 
						|
    struct rt_wlan_info info;
 | 
						|
    struct rt_wlan_complete_des *complete;
 | 
						|
    rt_uint32_t set = 0, recved = 0;
 | 
						|
 | 
						|
    if (_ap_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    if (ssid == RT_NULL) return -RT_EINVAL;
 | 
						|
 | 
						|
    rt_memset(&info, 0, sizeof(struct rt_wlan_info));
 | 
						|
    RT_WLAN_LOG_D("%s is run ssid:%s password:%s", __FUNCTION__, ssid, password);
 | 
						|
    if (password)
 | 
						|
    {
 | 
						|
        info.security = SECURITY_WPA2_AES_PSK;
 | 
						|
    }
 | 
						|
    ssid_len = rt_strlen(ssid);
 | 
						|
    if (ssid_len > RT_WLAN_SSID_MAX_LENGTH)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("ssid is to long! len:%d", ssid_len);
 | 
						|
    }
 | 
						|
 | 
						|
    /* copy info */
 | 
						|
    rt_memcpy(&info.ssid.val, ssid, ssid_len);
 | 
						|
    info.ssid.len = ssid_len;
 | 
						|
    info.channel = 6;
 | 
						|
    info.band = RT_802_11_BAND_2_4GHZ;
 | 
						|
 | 
						|
    /* Initializing events that need to wait */
 | 
						|
    MGNT_LOCK();
 | 
						|
    /* create event wait complete */
 | 
						|
    complete = rt_wlan_complete_create("start_ap");
 | 
						|
    if (complete == RT_NULL)
 | 
						|
    {
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ENOMEM;
 | 
						|
    }
 | 
						|
 | 
						|
    /* start ap */
 | 
						|
    err = rt_wlan_start_ap_adv(&info, password);
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        rt_wlan_complete_delete(complete);
 | 
						|
        RT_WLAN_LOG_I("start ap failed!");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Initializing events that need to wait */
 | 
						|
    set |= 0x1 << RT_WLAN_DEV_EVT_AP_START;
 | 
						|
    set |= 0x1 << RT_WLAN_DEV_EVT_AP_STOP;
 | 
						|
    /* Check whether there is a waiting event */
 | 
						|
    rt_wlan_complete_wait(complete, set, RT_WLAN_START_AP_WAIT_MS, &recved);
 | 
						|
    rt_wlan_complete_delete(complete);
 | 
						|
    /* check event */
 | 
						|
    set = 0x1 << RT_WLAN_DEV_EVT_AP_START;
 | 
						|
    if (!(recved & set))
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_I("start ap failed!");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_I("start ap successs!");
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_start_ap_adv(struct rt_wlan_info *info, const char *password)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
    int password_len = 0;
 | 
						|
 | 
						|
    if (_ap_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
    if (password != RT_NULL)
 | 
						|
    {
 | 
						|
        password_len = rt_strlen(password);
 | 
						|
    }
 | 
						|
    if (password_len > RT_WLAN_PASSWORD_MAX_LENGTH)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("key is to long! len:%d", password_len);
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
    /* is start up ? */
 | 
						|
    MGNT_LOCK();
 | 
						|
    if (rt_wlan_ap_is_active())
 | 
						|
    {
 | 
						|
        if ((_ap_mgnt.info.ssid.len == info->ssid.len) &&
 | 
						|
                (_ap_mgnt.info.security == info->security) &&
 | 
						|
                (_ap_mgnt.info.channel == info->channel) &&
 | 
						|
                (_ap_mgnt.info.hidden == info->hidden) &&
 | 
						|
                (_ap_mgnt.key.len == password_len) &&
 | 
						|
                (rt_memcmp(&_ap_mgnt.info.ssid.val[0], &info->ssid.val[0], info->ssid.len) == 0) &&
 | 
						|
                (rt_memcmp(&_ap_mgnt.key.val[0], password, password_len)))
 | 
						|
        {
 | 
						|
            RT_WLAN_LOG_D("wifi Already Start");
 | 
						|
            MGNT_UNLOCK();
 | 
						|
            return RT_EOK;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    err = rt_wlan_dev_ap_start(AP_DEVICE(), info, password, password_len);
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
    rt_memcpy(&_ap_mgnt.info, info, sizeof(struct rt_wlan_info));
 | 
						|
    rt_memcpy(&_ap_mgnt.key.val, password, password_len);
 | 
						|
    _ap_mgnt.key.len = password_len;
 | 
						|
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_bool_t rt_wlan_ap_is_active(void)
 | 
						|
{
 | 
						|
    rt_bool_t _active = RT_FALSE;
 | 
						|
 | 
						|
    if (_ap_is_null())
 | 
						|
    {
 | 
						|
        return RT_FALSE;
 | 
						|
    }
 | 
						|
 | 
						|
    _active = _ap_mgnt.state & RT_WLAN_STATE_ACTIVE ? RT_TRUE : RT_FALSE;
 | 
						|
    RT_WLAN_LOG_D("%s is run active:%s", __FUNCTION__, _active ? "Active" : "Inactive");
 | 
						|
    return _active;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_ap_stop(void)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
    struct rt_wlan_complete_des *complete;
 | 
						|
    rt_uint32_t set = 0, recved = 0;
 | 
						|
 | 
						|
    if (_ap_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
 | 
						|
    MGNT_LOCK();
 | 
						|
    /* create event wait complete */
 | 
						|
    complete = rt_wlan_complete_create("stop_ap");
 | 
						|
    if (complete == RT_NULL)
 | 
						|
    {
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ENOMEM;
 | 
						|
    }
 | 
						|
    err = rt_wlan_dev_ap_stop(AP_DEVICE());
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("ap stop fail");
 | 
						|
        rt_wlan_complete_delete(complete);
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
    /* Initializing events that need to wait */
 | 
						|
    set |= 0x1 << RT_WLAN_DEV_EVT_AP_STOP;
 | 
						|
    /* Check whether there is a waiting event */
 | 
						|
    rt_wlan_complete_wait(complete, set, RT_WLAN_START_AP_WAIT_MS, &recved);
 | 
						|
    rt_wlan_complete_delete(complete);
 | 
						|
    /* check event */
 | 
						|
    set = 0x1 << RT_WLAN_DEV_EVT_AP_STOP;
 | 
						|
    if (!(recved & set))
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_I("ap stop failed!");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_I("ap stop success!");
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_ap_get_info(struct rt_wlan_info *info)
 | 
						|
{
 | 
						|
    if (_ap_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
 | 
						|
    if (rt_wlan_ap_is_active() == RT_TRUE)
 | 
						|
    {
 | 
						|
        *info = _ap_mgnt.info;
 | 
						|
        return RT_EOK;
 | 
						|
    }
 | 
						|
    return -RT_ERROR;
 | 
						|
}
 | 
						|
 | 
						|
/* get sta number  */
 | 
						|
int rt_wlan_ap_get_sta_num(void)
 | 
						|
{
 | 
						|
    int sta_num = 0;
 | 
						|
 | 
						|
    STAINFO_LOCK();
 | 
						|
    sta_num = sta_info.num;
 | 
						|
    STAINFO_UNLOCK();
 | 
						|
    RT_WLAN_LOG_D("%s is run num:%d", __FUNCTION__, sta_num);
 | 
						|
    return sta_num;
 | 
						|
}
 | 
						|
 | 
						|
/* get sta info */
 | 
						|
int rt_wlan_ap_get_sta_info(struct rt_wlan_info *info, int num)
 | 
						|
{
 | 
						|
    int sta_num = 0, i = 0;
 | 
						|
    struct rt_wlan_sta_list *sta_list;
 | 
						|
 | 
						|
    STAINFO_LOCK();
 | 
						|
    /* sta_num = min(sta_info.num, num) */
 | 
						|
    sta_num = sta_info.num > num ? num : sta_info.num;
 | 
						|
    for (sta_list = sta_info.node; sta_list != RT_NULL && i < sta_num; sta_list = sta_list->next)
 | 
						|
    {
 | 
						|
        info[i] = sta_list->info;
 | 
						|
        i ++;
 | 
						|
    }
 | 
						|
    STAINFO_UNLOCK();
 | 
						|
    RT_WLAN_LOG_D("%s is run num:%d", __FUNCTION__, i);
 | 
						|
    return i;
 | 
						|
}
 | 
						|
 | 
						|
/* deauth sta */
 | 
						|
rt_err_t rt_wlan_ap_deauth_sta(rt_uint8_t *mac)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
    struct rt_wlan_sta_list *sta_list;
 | 
						|
    rt_bool_t find_flag = RT_FALSE;
 | 
						|
 | 
						|
    if (_ap_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run mac: %02x:%02x:%02x:%02x:%02x:%02x:%d",
 | 
						|
                  __FUNCTION__, mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
 | 
						|
 | 
						|
    if (mac == RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("mac addr is null");
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
 | 
						|
    MGNT_LOCK();
 | 
						|
    if (sta_info.node == RT_NULL || sta_info.num == 0)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("No AP");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
 | 
						|
    STAINFO_LOCK();
 | 
						|
    /* Search for MAC address from sta list */
 | 
						|
    for (sta_list = sta_info.node; sta_list != RT_NULL; sta_list = sta_list->next)
 | 
						|
    {
 | 
						|
        if (rt_memcmp(&sta_list->info.bssid[0], &mac[0], RT_WLAN_BSSID_MAX_LENGTH) == 0)
 | 
						|
        {
 | 
						|
            find_flag = RT_TRUE;
 | 
						|
            break;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    STAINFO_UNLOCK();
 | 
						|
 | 
						|
    /* No MAC address was found. return */
 | 
						|
    if (find_flag != RT_TRUE)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("Not find mac addr");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Kill STA */
 | 
						|
    err = rt_wlan_dev_ap_deauth(AP_DEVICE(), mac);
 | 
						|
    if (err != RT_NULL)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("deauth sta failed");
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return err;
 | 
						|
    }
 | 
						|
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_ap_set_country(rt_country_code_t country_code)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    if (_ap_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run country:%d", __FUNCTION__, country_code);
 | 
						|
    MGNT_LOCK();
 | 
						|
    err = rt_wlan_dev_set_country(AP_DEVICE(), country_code);
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_country_code_t rt_wlan_ap_get_country(void)
 | 
						|
{
 | 
						|
    rt_country_code_t country_code = RT_COUNTRY_UNKNOWN;
 | 
						|
 | 
						|
    if (_ap_is_null())
 | 
						|
    {
 | 
						|
        return country_code;
 | 
						|
    }
 | 
						|
    MGNT_LOCK();
 | 
						|
    country_code = rt_wlan_dev_get_country(AP_DEVICE());
 | 
						|
    RT_WLAN_LOG_D("%s is run country:%d", __FUNCTION__, country_code);
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return country_code;
 | 
						|
}
 | 
						|
 | 
						|
void rt_wlan_config_autoreconnect(rt_bool_t enable)
 | 
						|
{
 | 
						|
#ifdef RT_WLAN_AUTO_CONNECT_ENABLE
 | 
						|
    RT_WLAN_LOG_D("%s is run enable:%d", __FUNCTION__, enable);
 | 
						|
 | 
						|
    MGNT_LOCK();
 | 
						|
    if (enable)
 | 
						|
    {
 | 
						|
        TIME_START();
 | 
						|
        _sta_mgnt.flags |= RT_WLAN_STATE_AUTOEN;
 | 
						|
    }
 | 
						|
    else
 | 
						|
    {
 | 
						|
        TIME_STOP();
 | 
						|
        _sta_mgnt.flags &= ~RT_WLAN_STATE_AUTOEN;
 | 
						|
    }
 | 
						|
    MGNT_UNLOCK();
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
rt_bool_t rt_wlan_get_autoreconnect_mode(void)
 | 
						|
{
 | 
						|
#ifdef RT_WLAN_AUTO_CONNECT_ENABLE
 | 
						|
    rt_bool_t enable = 0;
 | 
						|
 | 
						|
    enable = _sta_mgnt.flags & RT_WLAN_STATE_AUTOEN ? 1 : 0;
 | 
						|
    RT_WLAN_LOG_D("%s is run enable:%d", __FUNCTION__, enable);
 | 
						|
    return enable;
 | 
						|
#else
 | 
						|
    return RT_FALSE;
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
/* Call the underlying scan function, which is asynchronous.
 | 
						|
The hotspots scanned are returned by callbacks */
 | 
						|
rt_err_t rt_wlan_scan(void)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
 | 
						|
    MGNT_LOCK();
 | 
						|
    err = rt_wlan_dev_scan(STA_DEVICE(), RT_NULL);
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_scan_with_info(struct rt_wlan_info *info)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
    struct rt_wlan_complete_des *complete;
 | 
						|
    rt_uint32_t set = 0, recved = 0;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
    if (info != RT_NULL && info->ssid.len > RT_WLAN_SSID_MAX_LENGTH)
 | 
						|
    {
 | 
						|
        RT_WLAN_LOG_E("ssid is to long!");
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Create an event that needs to wait. */
 | 
						|
    MGNT_LOCK();
 | 
						|
    complete = rt_wlan_complete_create("scan");
 | 
						|
    if (complete == RT_NULL)
 | 
						|
    {
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
 | 
						|
    /* run scan */
 | 
						|
    err = rt_wlan_dev_scan(STA_DEVICE(), info);
 | 
						|
    if (err != RT_EOK)
 | 
						|
    {
 | 
						|
        rt_wlan_complete_delete(complete);
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        RT_WLAN_LOG_E("scan sync fail");
 | 
						|
        return -RT_ERROR;
 | 
						|
    }
 | 
						|
 | 
						|
    /* Initializing events that need to wait */
 | 
						|
    set |= 0x1 << RT_WLAN_DEV_EVT_SCAN_DONE;
 | 
						|
    /* Check whether there is a waiting event */
 | 
						|
    rt_wlan_complete_wait(complete, set, RT_WLAN_CONNECT_WAIT_MS, &recved);
 | 
						|
    rt_wlan_complete_delete(complete);
 | 
						|
    /* check event */
 | 
						|
    set = 0x1 << RT_WLAN_DEV_EVT_SCAN_DONE;
 | 
						|
    if (!(recved & set))
 | 
						|
    {
 | 
						|
        MGNT_UNLOCK();
 | 
						|
        RT_WLAN_LOG_E("scan wait timeout!");
 | 
						|
        return -RT_ETIMEOUT;
 | 
						|
    }
 | 
						|
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return RT_EOK;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_set_powersave(int level)
 | 
						|
{
 | 
						|
    rt_err_t err = RT_EOK;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -RT_EIO;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
    MGNT_LOCK();
 | 
						|
    err = rt_wlan_dev_set_powersave(STA_DEVICE(), level);
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return err;
 | 
						|
}
 | 
						|
 | 
						|
int rt_wlan_get_powersave(void)
 | 
						|
{
 | 
						|
    int level;
 | 
						|
 | 
						|
    if (_sta_is_null())
 | 
						|
    {
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run", __FUNCTION__);
 | 
						|
    MGNT_LOCK();
 | 
						|
    level = rt_wlan_dev_get_powersave(STA_DEVICE());
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return level;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_register_event_handler(rt_wlan_event_t event, rt_wlan_event_handler handler, void *parameter)
 | 
						|
{
 | 
						|
    rt_base_t level;
 | 
						|
 | 
						|
    if (event >= RT_WLAN_EVT_MAX)
 | 
						|
    {
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run event:%d", __FUNCTION__, event);
 | 
						|
 | 
						|
    MGNT_LOCK();
 | 
						|
    /* Registering Callbacks */
 | 
						|
    level = rt_hw_interrupt_disable();
 | 
						|
    event_tab[event].handler = handler;
 | 
						|
    event_tab[event].parameter = parameter;
 | 
						|
    rt_hw_interrupt_enable(level);
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return RT_EOK;
 | 
						|
}
 | 
						|
 | 
						|
rt_err_t rt_wlan_unregister_event_handler(rt_wlan_event_t event)
 | 
						|
{
 | 
						|
    rt_base_t level;
 | 
						|
 | 
						|
    if (event >= RT_WLAN_EVT_MAX)
 | 
						|
    {
 | 
						|
        return -RT_EINVAL;
 | 
						|
    }
 | 
						|
    RT_WLAN_LOG_D("%s is run event:%d", __FUNCTION__, event);
 | 
						|
    MGNT_LOCK();
 | 
						|
    /* unregister*/
 | 
						|
    level = rt_hw_interrupt_disable();
 | 
						|
    event_tab[event].handler = RT_NULL;
 | 
						|
    event_tab[event].parameter = RT_NULL;
 | 
						|
    rt_hw_interrupt_enable(level);
 | 
						|
    MGNT_UNLOCK();
 | 
						|
    return RT_EOK;
 | 
						|
}
 | 
						|
 | 
						|
void rt_wlan_mgnt_lock(void)
 | 
						|
{
 | 
						|
    MGNT_LOCK();
 | 
						|
}
 | 
						|
 | 
						|
void rt_wlan_mgnt_unlock(void)
 | 
						|
{
 | 
						|
    MGNT_UNLOCK();
 | 
						|
}
 | 
						|
 | 
						|
int rt_wlan_prot_ready_event(struct rt_wlan_device *wlan, struct rt_wlan_buff *buff)
 | 
						|
{
 | 
						|
    rt_base_t level;
 | 
						|
 | 
						|
    if ((wlan == RT_NULL) || (_sta_mgnt.device != wlan) ||
 | 
						|
            (!(_sta_mgnt.state & RT_WLAN_STATE_CONNECT)))
 | 
						|
    {
 | 
						|
        return -1;
 | 
						|
    }
 | 
						|
    if (_sta_mgnt.state & RT_WLAN_STATE_READY)
 | 
						|
    {
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
    level = rt_hw_interrupt_disable();
 | 
						|
    _sta_mgnt.state |= RT_WLAN_STATE_READY;
 | 
						|
    rt_hw_interrupt_enable(level);
 | 
						|
#ifdef RT_WLAN_WORK_THREAD_ENABLE
 | 
						|
    rt_wlan_send_to_thread(RT_WLAN_EVT_READY, buff->data, buff->len);
 | 
						|
#else
 | 
						|
    {
 | 
						|
        void *user_parameter;
 | 
						|
        rt_wlan_event_handler handler = RT_NULL;
 | 
						|
 | 
						|
        level = rt_hw_interrupt_disable();
 | 
						|
        handler = event_tab[RT_WLAN_EVT_READY].handler;
 | 
						|
        user_parameter = event_tab[RT_WLAN_EVT_READY].parameter;
 | 
						|
        rt_hw_interrupt_enable(level);
 | 
						|
        if (handler)
 | 
						|
        {
 | 
						|
            handler(RT_WLAN_EVT_READY, buff, user_parameter);
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int rt_wlan_init(void)
 | 
						|
{
 | 
						|
    static rt_int8_t _init_flag = 0;
 | 
						|
 | 
						|
    /* Execute only once */
 | 
						|
    if (_init_flag == 0)
 | 
						|
    {
 | 
						|
        rt_memset(&_sta_mgnt, 0, sizeof(struct rt_wlan_mgnt_des));
 | 
						|
        rt_memset(&_ap_mgnt, 0, sizeof(struct rt_wlan_mgnt_des));
 | 
						|
        rt_memset(&sta_info, 0, sizeof(struct rt_wlan_sta_des));
 | 
						|
        rt_mutex_init(&mgnt_mutex, "mgnt", RT_IPC_FLAG_FIFO);
 | 
						|
        rt_mutex_init(&sta_info_mutex, "sta", RT_IPC_FLAG_FIFO);
 | 
						|
        rt_mutex_init(&complete_mutex, "complete", RT_IPC_FLAG_FIFO);
 | 
						|
#ifdef RT_WLAN_AUTO_CONNECT_ENABLE
 | 
						|
        rt_timer_init(&reconnect_time, "wifi_tim", rt_wlan_cyclic_check, RT_NULL,
 | 
						|
                      rt_tick_from_millisecond(AUTO_CONNECTION_PERIOD_MS),
 | 
						|
                      RT_TIMER_FLAG_PERIODIC | RT_TIMER_FLAG_SOFT_TIMER);
 | 
						|
#endif
 | 
						|
        _init_flag = 1;
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
INIT_PREV_EXPORT(rt_wlan_init);
 | 
						|
 | 
						|
#endif
 |