|
@@ -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);
|