[DeviceDrivers] Add wifi device/cmd.

1 files changed, 537 insertions, 0 deletions

diff --git a/components/drivers/wlan/wlan_cmd.c b/components/drivers/wlan/wlan_cmd.c
new file mode 100644
index 0000000000..21afd7020a
--- /dev/null
+++ b/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);