Browse Source

[DeviceDrivers] Add wifi device/cmd.

bernard 7 years ago
parent
commit
4b2a3d02f3

+ 8 - 0
components/drivers/wlan/SConscript

@@ -0,0 +1,8 @@
+from building import *
+
+cwd     = GetCurrentDir()
+src     = Glob('*.c')
+CPPPATH = [cwd]
+group   = DefineGroup('DeviceDrivers', src, depend = ['RT_USING_WIFI'], CPPPATH = CPPPATH)
+
+Return('group')

+ 537 - 0
components/drivers/wlan/wlan_cmd.c

@@ -0,0 +1,537 @@
+/*
+ * File      : wlan_cmd.c
+ *             Wi-Fi common commands
+ * This file is part of RT-Thread RTOS
+ * COPYRIGHT (C) 2006 - 2016, RT-Thread Development Team
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License along
+ *  with this program; if not, write to the Free Software Foundation, Inc.,
+ *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ * 2016-03-12     Bernard      first version
+ */
+
+#include <rtthread.h>
+#include <wlan_dev.h>
+
+#include <finsh.h>
+
+#include <lwip/dhcp.h>
+#include "wlan_cmd.h"
+
+#ifdef LWIP_USING_DHCPD
+#include <dhcp_server.h>
+#endif
+
+struct rt_wlan_info info;
+
+#ifndef WIFI_SETTING_FN
+#define WIFI_SETTING_FN     "/appfs/setting.json"
+#endif
+
+#ifndef WIFI_DEVICE_NAME
+#define WIFI_DEVICE_NAME    "w0"
+#endif
+
+#ifdef RT_USING_DFS
+#include <dfs_posix.h>
+#ifdef PKG_USING_CJSON
+#include <cJSON_util.h>
+#endif
+
+static char wifi_ssid[32]    = {0};
+static char wifi_key[32]     = {0};
+static int network_mode      = WIFI_STATION;
+
+int wifi_get_mode(void)
+{
+    return network_mode;
+}
+
+int wifi_set_mode(int mode)
+{
+    network_mode = mode;
+
+    return network_mode;
+}
+
+int wifi_set_setting(const char* ssid, const char* pwd)
+{
+    if (!ssid) return -1;
+
+    strncpy(wifi_ssid, ssid, sizeof(wifi_ssid));
+    wifi_ssid[sizeof(wifi_ssid) - 1] = '\0';
+
+    if (pwd)
+    {
+        strncpy(wifi_key, pwd, sizeof(wifi_key));
+        wifi_key[sizeof(wifi_key) - 1] = '\0';
+    }
+    else wifi_key[0] = '\0';
+
+    return 0;
+}
+
+#ifdef PKG_USING_CJSON
+int wifi_read_cfg(const char* filename)
+{
+    int fd;
+    cJSON *json = RT_NULL;
+
+    fd = open(filename,O_RDONLY, 0);
+    if(fd < 0)
+    {
+        /* no setting file */
+        return -1;
+    }
+
+    if (fd >= 0)
+    {
+        int length;
+
+        length = lseek(fd, 0, SEEK_END);
+        if (length)
+        {
+            char *json_str = (char *) rt_malloc (length);
+            if (json_str)
+            {
+                lseek(fd, 0, SEEK_SET);
+                read(fd, json_str, length);
+
+                json = cJSON_Parse(json_str);
+                rt_free(json_str);
+            }
+        }
+        close(fd);
+    }
+
+    if (json)
+    {
+        cJSON *wifi = cJSON_GetObjectItem(json, "wifi");
+        cJSON *ssid = cJSON_GetObjectItem(wifi, "SSID");
+        cJSON *key  = cJSON_GetObjectItem(wifi, "Key");
+        cJSON *mode = cJSON_GetObjectItem(wifi, "Mode");
+
+        if (ssid)
+        {
+            memset(wifi_ssid, 0x0, sizeof(wifi_ssid));
+            rt_strncpy(wifi_ssid, ssid->valuestring, sizeof(wifi_ssid) - 1);
+        }
+
+        if (key)
+        {
+            memset(wifi_key, 0x0, sizeof(wifi_key));
+            rt_strncpy(wifi_key, key->valuestring, sizeof(wifi_key) - 1);
+        }
+
+        if (mode)
+        {
+            network_mode = mode->valueint;
+        }
+
+        cJSON_Delete(json);
+    }
+
+    return 0;
+}
+
+int wifi_save_cfg(const char* filename)
+{
+    int fd;
+    cJSON *json = RT_NULL;
+
+    fd = open(filename, O_RDONLY, 0);
+    if (fd >= 0)
+    {
+        int length;
+
+        length = lseek(fd, 0, SEEK_END);
+        if (length)
+        {
+            char *json_str = (char *) rt_malloc (length);
+            if (json_str)
+            {
+                lseek(fd, 0, SEEK_SET);
+                read(fd, json_str, length);
+
+                json = cJSON_Parse(json_str);
+                rt_free(json_str);
+            }
+        }
+        close(fd);
+    }
+    else
+    {
+        /* create a new setting.json */
+        fd = open(filename, O_WRONLY | O_TRUNC, 0);
+        if (fd >= 0)
+        {
+            json = cJSON_CreateObject();
+            if (json)
+            {
+                cJSON *wifi = cJSON_CreateObject();
+
+                if (wifi)
+                {
+                    char *json_str;
+
+                    cJSON_AddItemToObject(json, "wifi", wifi);
+                    cJSON_AddStringToObject(wifi, "SSID", wifi_ssid);
+                    cJSON_AddStringToObject(wifi, "Key", wifi_key);
+                    cJSON_AddNumberToObject(wifi, "Mode", network_mode);
+
+                    json_str = cJSON_Print(json);
+                    if (json_str)
+                    {
+                        write(fd, json_str, rt_strlen(json_str));
+                        cJSON_free(json_str);
+                    }
+                }
+            }
+        }
+        close(fd);
+
+        return 0;
+    }
+
+    if (json)
+    {
+        cJSON *wifi = cJSON_GetObjectItem(json, "wifi");
+        if (!wifi)
+        {
+            wifi = cJSON_CreateObject();
+            cJSON_AddItemToObject(json, "wifi", wifi);
+        }
+
+        if (cJSON_GetObjectItem(wifi, "SSID"))cJSON_ReplaceItemInObject(wifi, "SSID", cJSON_CreateString(wifi_ssid));
+        else cJSON_AddStringToObject(wifi, "SSID", wifi_ssid);
+        if (cJSON_GetObjectItem(wifi, "Key")) cJSON_ReplaceItemInObject(wifi, "Key", cJSON_CreateString(wifi_key));
+        else cJSON_AddStringToObject(wifi, "Key", wifi_key);
+        if (cJSON_GetObjectItem(wifi, "Mode")) cJSON_ReplaceItemInObject(wifi, "Mode", cJSON_CreateNumber(network_mode));
+        else cJSON_AddNumberToObject(wifi, "Mode", network_mode);
+
+        fd = open(filename, O_WRONLY | O_TRUNC, 0);
+        if (fd >= 0)
+        {
+            char *json_str = cJSON_Print(json);
+            if (json_str)
+            {
+                write(fd, json_str, rt_strlen(json_str));
+                cJSON_free(json_str);
+            }
+            close(fd);
+        }
+        cJSON_Delete(json);
+    }
+
+    return 0;
+}
+#endif
+
+int wifi_save_setting(void)
+{
+    wifi_save_cfg(WIFI_SETTING_FN);
+
+    return 0;
+}
+#endif
+
+int wifi_softap_setup_netif(struct netif *netif)
+{
+    if (netif)
+    {
+        ip_addr_t *ip;
+        ip_addr_t addr;
+
+#ifdef RT_LWIP_DHCP
+        /* Stop DHCP Client */
+        dhcp_stop(netif);
+#endif
+
+        /* set ipaddr, gw, netmask */
+        ip = (ip_addr_t *)&addr;
+
+        /* set ip address */
+        if (ipaddr_aton("192.168.169.1", &addr))
+        {
+            netif_set_ipaddr(netif, ip);
+        }
+
+        /* set gateway address */
+        if ( ipaddr_aton("192.168.169.1", &addr))
+        {
+            netif_set_gw(netif, ip);
+        }
+
+        /* set netmask address */
+        if ( ipaddr_aton("255.255.255.0", &addr))
+        {
+            netif_set_netmask(netif, ip);
+        }
+
+        netif_set_up(netif);
+
+#ifdef LWIP_USING_DHCPD
+        {
+            char name[8];
+            memset(name, 0, sizeof(name));
+            strncpy(name, netif->name, sizeof(name)>sizeof(netif->name)? sizeof(netif->name) : sizeof(name));
+            dhcpd_start(name);
+        }
+#endif
+    }
+
+    return 0;
+}
+
+int wifi_default(void)
+{
+    int result = 0;
+    struct rt_wlan_device *wlan;
+
+    /* read default setting for wifi */
+    wifi_read_cfg(WIFI_SETTING_FN);
+
+    /* get wlan device */
+    wlan = (struct rt_wlan_device*)rt_device_find(WIFI_DEVICE_NAME);
+    if (!wlan)
+    {
+        rt_kprintf("no wlan:%s device\n", WIFI_DEVICE_NAME);
+        return -1;
+    }
+
+    if (network_mode == WIFI_STATION)
+    {
+        struct rt_wlan_info *info;
+
+        info = (struct rt_wlan_info *)rt_malloc (sizeof(struct rt_wlan_info));
+        if (!info)
+        {
+            rt_kprintf("wifi: out of memory\n");
+            return -1;
+        }
+
+        /* wifi station */
+        rt_wlan_info_init(info, WIFI_STATION, SECURITY_WPA2_MIXED_PSK, wifi_ssid);
+        result =rt_wlan_init(wlan, WIFI_STATION);
+        if (result == RT_EOK)
+        {
+            result = rt_wlan_connect(wlan, info, wifi_key);
+        }
+    }
+    else
+    {
+        /* wifi AP */
+        struct rt_wlan_info *info;
+
+        info = (struct rt_wlan_info *)rt_malloc (sizeof(struct rt_wlan_info));
+        if (!info)
+        {
+            rt_kprintf("wifi: out of memory\n");
+            return -1;
+        }
+
+        rt_wlan_info_init(info, WIFI_AP, SECURITY_WPA2_AES_PSK, wifi_ssid);
+        info->channel = 11;
+
+        /* wifi soft-AP */
+        result =rt_wlan_init(wlan, WIFI_AP);
+        if (result == RT_EOK)
+        {
+            result = rt_wlan_softap(wlan, info, wifi_key);
+        }
+    }
+
+    return result;
+}
+
+static void wifi_usage(void)
+{
+    rt_kprintf("wifi wlan_dev - do the default wifi action\n");
+    rt_kprintf("wifi wlan_dev join SSID PASSWORD\n");
+    rt_kprintf("wifi wlan_dev ap SSID [PASSWORD]\n");
+    rt_kprintf("wifi cfg SSID PASSWORD\n");
+    rt_kprintf("wifi wlan_dev up\n");
+    rt_kprintf("wifi wlan_dev down\n");
+    rt_kprintf("wifi wlan_dev rssi\n");
+}
+
+int wifi(int argc, char** argv)
+{
+    struct rt_wlan_device *wlan;
+
+    if (argc == 1)
+    {
+        wifi_default();
+        return 0;
+    }
+
+    if (strcmp(argv[1], "help") == 0)
+    {
+        wifi_usage();
+        return 0;
+    }
+
+    if (strcmp(argv[1], "cfg") == 0)
+    {
+        /* configure wifi setting */
+        memset(wifi_ssid, 0x0, sizeof(wifi_ssid));
+        rt_strncpy(wifi_ssid, argv[2], sizeof(wifi_ssid) - 1);
+
+        memset(wifi_key, 0x0, sizeof(wifi_key));
+        rt_strncpy(wifi_key, argv[3], sizeof(wifi_key) - 1);
+
+        network_mode = WIFI_STATION;
+
+        wifi_save_cfg(WIFI_SETTING_FN);
+
+        return 0;
+    }
+
+    /* get wlan device */
+    wlan = (struct rt_wlan_device*)rt_device_find(argv[1]);
+    if (!wlan)
+    {
+        rt_kprintf("no wlan:%s device\n");
+        return 0;
+    }
+
+    if (argc < 3)
+    {
+        wifi_usage();
+        return 0;
+    }
+
+    if (strcmp(argv[2], "join") == 0)
+    {
+        rt_wlan_init(wlan, WIFI_STATION);
+
+        /* TODO: use easy-join to replace */
+        rt_wlan_info_init(&info, WIFI_STATION, SECURITY_WPA2_MIXED_PSK, argv[3]);
+        rt_wlan_connect(wlan, &info, argv[4]);
+    }
+    else if (strcmp(argv[2], "up") == 0)
+    {
+        /* the key was saved in wlan device */
+        rt_wlan_connect(wlan, RT_NULL, wlan->key);
+    }
+    else if (strcmp(argv[2], "down") == 0)
+    {
+        rt_wlan_disconnect(wlan);
+    }
+    else if (strcmp(argv[2], "scan") == 0)
+    {
+        struct rt_wlan_info *infos;
+
+        infos = (struct rt_wlan_info*)rt_malloc(sizeof(struct rt_wlan_info) * 12);
+        if (infos)
+        {
+            int index, num;
+
+            memset(infos, 0x0, sizeof(struct rt_wlan_info) * 12);
+            num = rt_wlan_scan(wlan, infos, 12);
+
+            for (index = 0; index < num; index ++)
+            {
+                rt_kprintf("----Wi-Fi AP[%d] Information----\n", index);
+                rt_kprintf("SSID: %-.32s\n", infos[index].ssid);
+                rt_kprintf("rssi: %d\n", infos[index].rssi);
+                rt_kprintf(" chn: %d\n", infos[index].channel);
+                rt_kprintf("rate: %d\n", infos[index].datarate);
+                rt_kprintf("\n");
+            }
+
+            /* de-initialize info */
+            for (index = 0; index < num; index ++)
+            {
+                rt_wlan_info_deinit(&infos[index]);
+            }
+            rt_free(infos);
+        }
+    }
+    else if (strcmp(argv[2], "rssi") == 0)
+    {
+        int rssi;
+
+        rssi = rt_wlan_get_rssi(wlan);
+        rt_kprintf("rssi=%d\n", rssi);
+    }
+    else if (strcmp(argv[2], "ap") == 0)
+    {
+        rt_err_t result = RT_EOK;
+        struct rt_wlan_info *info;
+
+        info = (struct rt_wlan_info*)rt_malloc(sizeof(struct rt_wlan_info));
+        if (argc == 4)
+        {
+            // open soft-AP
+            rt_wlan_info_init(info, WIFI_AP, SECURITY_OPEN, argv[3]);
+            info->channel = 11;
+
+            /* start soft ap */
+            result = rt_wlan_softap(wlan, info, NULL);
+        }
+        else if (argc == 5)
+        {
+            // WPA2 with password
+            rt_wlan_info_init(info, WIFI_AP, SECURITY_WPA2_AES_PSK, argv[3]);
+            info->channel = 11;
+
+            /* start soft ap */
+            result = rt_wlan_softap(wlan, info, argv[4]);
+        }
+        else
+        {
+            /* release information */
+            rt_free(info);
+
+            wifi_usage();
+        }
+        
+        if (result != RT_EOK)
+        {
+            rt_kprintf("wifi start failed! result=%d\n", result);
+        }
+    }
+    else if (strcmp(argv[2], "status") == 0)
+    {
+        int rssi;
+
+        if (netif_is_link_up(wlan->parent.netif))
+        {
+            rssi = rt_wlan_get_rssi(wlan);
+
+            rt_kprintf("Wi-Fi AP: %-.32s\n", wlan->info->ssid);
+            rt_kprintf("MAC Addr: %02x:%02x:%02x:%02x:%02x:%02x\n", wlan->info->bssid[0],
+                    wlan->info->bssid[1],
+                    wlan->info->bssid[2],
+                    wlan->info->bssid[3],
+                    wlan->info->bssid[4],
+                    wlan->info->bssid[5]);
+            rt_kprintf(" Channel: %d\n", wlan->info->channel);
+            rt_kprintf("DataRate: %dMbps\n", wlan->info->datarate/1000);
+            rt_kprintf("    RSSI: %d\n", rssi);
+        }
+        else
+        {
+            rt_kprintf("wifi disconnected!\n");
+        }
+
+        return 0;
+    }
+
+    return 0;
+}
+MSH_CMD_EXPORT(wifi, wifi command);

+ 24 - 0
components/drivers/wlan/wlan_cmd.h

@@ -0,0 +1,24 @@
+#ifndef WLAN_CMD_H__
+#define WLAN_CMD_H__
+
+struct netif;
+
+int wifi_get_mode(void);
+int wifi_set_mode(int mode);
+
+/* do the wifi default action: read wifi setting and then join or start soft-AP */
+int wifi_default(void);
+/* setup netif for soft-ap */
+int wifi_softap_setup_netif(struct netif *netif);
+
+int wifi_set_setting(const char* ssid, const char* pwd);
+
+int wifi_read_cfg(const char* filename);
+int wifi_save_cfg(const char* filename);
+
+/* save wifi setting with default storage file */
+int wifi_save_setting(void);
+
+extern struct rt_wlan_info info;
+
+#endif

+ 246 - 0
components/drivers/wlan/wlan_dev.c

@@ -0,0 +1,246 @@
+/*
+ *  RT-Thread Wi-Fi Device
+ *
+ * COPYRIGHT (C) 2014 - 2015, Shanghai Real-Thread Technology Co., Ltd
+ *
+ *  This file is part of RT-Thread (http://www.rt-thread.org)
+ *
+ *  All rights reserved.
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License along
+ *  with this program; if not, write to the Free Software Foundation, Inc.,
+ *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ * 2014-09-11     Bernard      the first verion
+ */
+
+#include <rtthread.h>
+#include <rtdevice.h>
+
+#include <lwip/netifapi.h>
+
+#include "wlan_dev.h"
+#include "wlan_cmd.h"
+
+#define NIOCTL_SADDR    0x02
+
+void rt_wlan_info_init(struct rt_wlan_info* info, rt_wlan_mode_t mode, rt_wlan_security_t security, 
+    char *ssid)
+{
+    if (info == RT_NULL) return ;
+
+    memset(info, 0x0, sizeof(struct rt_wlan_info));
+    info->mode = mode;
+    info->security = security;
+    if (ssid)
+    {
+        info->ssid = rt_malloc(strlen((char*)ssid) + 1);
+        if (info->ssid)
+        {
+            strncpy((char*)info->ssid, (char*)ssid, strlen((char*)ssid) + 1);
+        }
+    }
+}
+
+void rt_wlan_info_deinit(struct rt_wlan_info* info)
+{
+    if (info->ssid)
+    {
+        rt_free(info->ssid);
+        info->ssid = RT_NULL;
+    }
+
+    memset(info, 0x0, sizeof(struct rt_wlan_info));
+}
+
+int rt_wlan_init(struct rt_wlan_device* device, rt_wlan_mode_t mode)
+{
+    int result;
+
+    if (device == RT_NULL) return 0;
+
+    result = rt_device_control(RT_DEVICE(device), WIFI_INIT, (void*)&mode);
+
+    return result;
+}
+
+int rt_wlan_connect(struct rt_wlan_device* device, struct rt_wlan_info* info, char *password)
+{
+    int result = 0;
+
+    if (device == RT_NULL) return -RT_EIO;
+
+    if (info != RT_NULL)
+    {
+        rt_wlan_set_info(device, info);
+    }
+
+    result = rt_device_control(RT_DEVICE(device), WIFI_EASYJOIN, (void*)password);
+    if (result == RT_EOK)
+    {
+        struct netif *netif = device->parent.netif;
+
+        netifapi_netif_set_up(netif);
+        eth_device_linkchange(&(device->parent), RT_TRUE);
+
+#ifdef RT_LWIP_DHCP
+        /* set DHCP flags */
+        // netif->flags |= NETIF_FLAG_DHCP;
+        /* start DHCP */
+        dhcp_start(netif);
+#endif
+
+        rt_strncpy((char*)device->key, (char*)password, sizeof(device->key) - 1);
+    }
+
+    return result;
+}
+
+int rt_wlan_softap(struct rt_wlan_device* device, struct rt_wlan_info* info, char *password)
+{
+    int result = RT_EOK;
+
+    if (device == RT_NULL) return -RT_EIO;
+
+    if (info != RT_NULL)
+    {
+        rt_wlan_set_info(device, info);
+    }
+
+    result = rt_device_control(RT_DEVICE(device), WIFI_SOFTAP, (void*)password);
+    if (result == RT_EOK)
+    {
+        rt_strncpy((char*)device->key, (char*)password, sizeof(device->key) - 1);
+
+        netifapi_netif_set_up(device->parent.netif);
+        eth_device_linkchange(&(device->parent), RT_TRUE);
+
+        wifi_softap_setup_netif(device->parent.netif);
+    }
+
+    return result;
+}
+
+int rt_wlan_disconnect(struct rt_wlan_device* device)
+{
+    int result = 0;
+
+    if (device == RT_NULL) return -RT_EIO;
+
+    /* save event handler */
+    result = rt_device_control(RT_DEVICE(device), WIFI_DISCONNECT, RT_NULL);
+    if (result == RT_EOK)
+    {
+        netifapi_netif_set_down(device->parent.netif);
+        eth_device_linkchange(&(device->parent), RT_FALSE);
+    }
+
+    return result;
+}
+
+int rt_wlan_set_info(struct rt_wlan_device* device, struct rt_wlan_info* info)
+{
+    if (device->info == info) return RT_EOK; /* same info */
+
+    if (device->info != RT_NULL)
+    {
+        rt_wlan_info_deinit(device->info);
+        rt_free(device->info);
+    }
+
+    device->info = info;
+
+    return RT_EOK;
+}
+
+struct rt_wlan_info *rt_wlan_get_info(struct rt_wlan_device* device)
+{
+    struct rt_wlan_info* info = RT_NULL;
+
+    if (device != RT_NULL)
+    {
+        info = device->info;
+    }
+
+    return info;
+}
+
+int rt_wlan_scan(struct rt_wlan_device* device, struct rt_wlan_info *infos, int item_sz)
+{
+    int result;
+    struct rt_wlan_info_request request;
+
+    if (device == RT_NULL) return 0;
+
+    request.req_number = item_sz;
+    request.rsp_number = 0;
+    request.infos = infos;
+
+    result = rt_device_control(RT_DEVICE(device), WIFI_SCAN, (void*)&request);
+    result = result; /* skip warning */
+
+    return request.rsp_number;
+}
+
+int rt_wlan_get_rssi(struct rt_wlan_device* device)
+{
+    int rssi;
+    int result;
+
+    if (device == RT_NULL) return 0;
+    result = rt_device_control(RT_DEVICE(device), WIFI_GET_RSSI, (void*)&rssi);
+
+    if (result == RT_EOK) return rssi;
+
+    return result;
+}
+
+int rt_wlan_get_mac(struct rt_wlan_device* device, rt_uint8_t hwaddr[6])
+{
+    int result;
+    if (device == RT_NULL) return 0;
+    result = rt_device_control(RT_DEVICE(device), NIOCTL_GADDR, (void*)hwaddr);
+    return result;
+}
+
+int rt_wlan_set_mac(struct rt_wlan_device* device, rt_uint8_t hwaddr[6])
+{
+    int result;
+    if (device == RT_NULL) return 0;
+    result = rt_device_control(RT_DEVICE(device), NIOCTL_SADDR, (void*)hwaddr);
+    return result;
+}
+
+int rt_wlan_enter_powersave(struct rt_wlan_device* device, int level)
+{
+    int result = 0;
+
+    if (device == RT_NULL) return -RT_EIO;
+
+    result = rt_device_control(RT_DEVICE(device), WIFI_ENTER_POWERSAVE, (void*)&level);
+
+    return result;
+}
+
+void rt_wlan_set_event_callback(struct rt_wlan_device* device, rt_wlan_event_handler handler, 
+    void *user_data)
+{
+    if (device == RT_NULL) return ;
+
+    device->handler = handler;
+    device->user_data = user_data;
+
+    return ;
+}

+ 175 - 0
components/drivers/wlan/wlan_dev.h

@@ -0,0 +1,175 @@
+/*
+ *  RT-Thread Wi-Fi Device
+ *
+ * COPYRIGHT (C) 2014 - 2015, Shanghai Real-Thread Technology Co., Ltd
+ *
+ *  This file is part of RT-Thread (http://www.rt-thread.org)
+ *
+ *  All rights reserved.
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License along
+ *  with this program; if not, write to the Free Software Foundation, Inc.,
+ *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ * 2014-09-11     Bernard      the first verion
+ */
+#ifndef WIFI_DEVICE_H__
+#define WIFI_DEVICE_H__
+
+#include <stdint.h>
+#include <netif/ethernetif.h>
+
+typedef enum
+{
+    WIFI_STATION,
+    WIFI_AP,
+} rt_wlan_mode_t;
+
+typedef enum
+{
+    WIFI_INIT = 0x10,
+    WIFI_SCAN,              /* trigger scanning (list cells) */
+    WIFI_JOIN,
+    WIFI_EASYJOIN,          /* join network with less information */
+    WIFI_SOFTAP,            /* start soft-AP */
+    WIFI_DISCONNECT,
+    WIFI_GET_RSSI,          /* get sensitivity (dBm) */
+    WIFI_ENTER_POWERSAVE,
+} rt_wlan_cmd_t;
+
+typedef enum
+{
+    WIFI_PWR_OFF,
+    WIFI_PWR_SLEEP,
+    WIFI_PWR_NORMAL
+} rt_wlan_powersave_t;
+
+#define SHARED_ENABLED  0x00008000
+#define WPA_SECURITY    0x00200000
+#define WPA2_SECURITY   0x00400000
+#define WPS_ENABLED     0x10000000
+#define WEP_ENABLED     0x0001
+#define TKIP_ENABLED    0x0002
+#define AES_ENABLED     0x0004
+#define WSEC_SWFLAG     0x0008
+
+#define KEY_ARRAY_SIZE  32
+
+/**
+ * Enumeration of Wi-Fi security modes
+ */
+typedef enum
+{
+    SECURITY_OPEN           = 0,                                                /**< Open security                           */
+    SECURITY_WEP_PSK        = WEP_ENABLED,                                      /**< WEP Security with open authentication   */
+    SECURITY_WEP_SHARED     = ( WEP_ENABLED | SHARED_ENABLED ),                 /**< WEP Security with shared authentication */
+    SECURITY_WPA_TKIP_PSK   = ( WPA_SECURITY  | TKIP_ENABLED ),                 /**< WPA Security with TKIP                  */
+    SECURITY_WPA_AES_PSK    = ( WPA_SECURITY  | AES_ENABLED ),                  /**< WPA Security with AES                   */
+    SECURITY_WPA2_AES_PSK   = ( WPA2_SECURITY | AES_ENABLED ),                  /**< WPA2 Security with AES                  */
+    SECURITY_WPA2_TKIP_PSK  = ( WPA2_SECURITY | TKIP_ENABLED ),                 /**< WPA2 Security with TKIP                 */
+    SECURITY_WPA2_MIXED_PSK = ( WPA2_SECURITY | AES_ENABLED | TKIP_ENABLED ),   /**< WPA2 Security with AES & TKIP           */
+    SECURITY_WPS_OPEN       = WPS_ENABLED,                                      /**< WPS with open security                  */
+    SECURITY_WPS_SECURE     = (WPS_ENABLED | AES_ENABLED),                      /**< WPS with AES security                   */
+    SECURITY_UNKNOWN        = -1,                                               /**< May be returned by scan function if security is unknown. 
+                                                                                     Do not pass this to the join function! */
+} rt_wlan_security_t;
+
+typedef enum
+{
+    WIFI_EVT_LINK_DOWN,
+    WIFI_EVT_LINK_UP,
+}rt_wlan_event_t;
+
+/* wifi network information */
+struct rt_wlan_info
+{
+    rt_wlan_mode_t mode;   /* wifi mode */
+    rt_wlan_security_t security;
+
+    char *ssid;
+    uint8_t bssid[6];
+
+    /* maximal data rate */
+    uint32_t datarate;
+    /* radio channel */
+    uint16_t channel;
+    /* signal strength */
+    int16_t  rssi;
+};
+
+struct rt_wlan_info_request
+{
+    uint16_t req_number;    /* the number of information item for request */
+    uint16_t rsp_number;    /* the number of information item for response */
+
+    struct rt_wlan_info *infos;/* the array of information to save response */
+};
+
+struct rt_wlan_device;
+typedef void (*rt_wlan_event_handler)(struct rt_wlan_device* device, rt_wlan_event_t event, void* user_data);
+
+struct rt_wlan_device
+{
+    struct eth_device parent;
+
+    struct rt_wlan_info* info;
+    char key[KEY_ARRAY_SIZE + 1];
+
+    rt_wlan_event_handler handler;
+    void* user_data;
+    int interface;
+};
+
+/*
+ * Wi-Fi Information APIs
+ */
+void rt_wlan_info_init(struct rt_wlan_info* info, rt_wlan_mode_t mode, rt_wlan_security_t security, 
+    char *ssid);
+void rt_wlan_info_deinit(struct rt_wlan_info* info);
+
+/*
+ * Wi-Fi Manager APIs
+ */
+int rt_wlan_init(struct rt_wlan_device* device, rt_wlan_mode_t mode);
+
+int rt_wlan_connect(struct rt_wlan_device* device, struct rt_wlan_info* info, 
+    char *password);
+int rt_wlan_disconnect(struct rt_wlan_device* device);
+
+int rt_wlan_softap(struct rt_wlan_device* device, struct rt_wlan_info* info, 
+    char *password);
+
+/* set wifi information for AP */
+int rt_wlan_set_info(struct rt_wlan_device* device, struct rt_wlan_info* info);
+/* get wifi information for AP */
+struct rt_wlan_info *rt_wlan_get_info(struct rt_wlan_device* device);
+
+/* get the AP result which were scaned in station */
+int rt_wlan_scan(struct rt_wlan_device* device, struct rt_wlan_info *infos, int item_sz);
+
+/* get rssi */
+int rt_wlan_get_rssi(struct rt_wlan_device* device);
+/* Get/Set MAC */
+int rt_wlan_get_mac(struct rt_wlan_device* device,rt_uint8_t hwaddr[6]);
+int rt_wlan_set_mac(struct rt_wlan_device* device,rt_uint8_t hwaddr[6]);
+
+/* enter power save level */
+int rt_wlan_enter_powersave(struct rt_wlan_device* device, int level);
+
+void rt_wlan_set_event_callback(struct rt_wlan_device* device, rt_wlan_event_handler handler, 
+    void *user_data);
+
+#endif
+