libs/rt-thread-official
libs
/
rt-thread-official
mirror of https://github.com/RT-Thread/rt-thread.git
4
0
Fork 0
Code Issues Releases Wiki Activity
rt-thread-official/components/drivers/wlan/wlan_cmd.c

538 lines
13 KiB
C
Raw Blame History

/*
* 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);
View Git Blame Copy Permalink