594 lines
14 KiB
C
594 lines
14 KiB
C
|
/*
|
||
|
* Copyright (c) 2006-2021, RT-Thread Development Team
|
||
|
*
|
||
|
* SPDX-License-Identifier: Apache-2.0
|
||
|
*
|
||
|
* Change Logs:
|
||
|
* Date Author Notes
|
||
|
* 2018-08-13 tyx the first version
|
||
|
*/
|
||
|
|
||
|
#include <rtthread.h>
|
||
|
#include <wlan_mgnt.h>
|
||
|
#include <wlan_cfg.h>
|
||
|
#include <wlan_prot.h>
|
||
|
|
||
|
#if defined(RT_WLAN_MANAGE_ENABLE) && defined(RT_WLAN_MSH_CMD_ENABLE)
|
||
|
|
||
|
struct wifi_cmd_des
|
||
|
{
|
||
|
const char *cmd;
|
||
|
int (*fun)(int argc, char *argv[]);
|
||
|
};
|
||
|
|
||
|
static int wifi_help(int argc, char *argv[]);
|
||
|
static int wifi_scan(int argc, char *argv[]);
|
||
|
static int wifi_status(int argc, char *argv[]);
|
||
|
static int wifi_join(int argc, char *argv[]);
|
||
|
static int wifi_ap(int argc, char *argv[]);
|
||
|
static int wifi_list_sta(int argc, char *argv[]);
|
||
|
static int wifi_disconnect(int argc, char *argv[]);
|
||
|
static int wifi_ap_stop(int argc, char *argv[]);
|
||
|
|
||
|
#ifdef RT_WLAN_CMD_DEBUG
|
||
|
/* just for debug */
|
||
|
static int wifi_debug(int argc, char *argv[]);
|
||
|
static int wifi_debug_save_cfg(int argc, char *argv[]);
|
||
|
static int wifi_debug_dump_cfg(int argc, char *argv[]);
|
||
|
static int wifi_debug_clear_cfg(int argc, char *argv[]);
|
||
|
static int wifi_debug_dump_prot(int argc, char *argv[]);
|
||
|
static int wifi_debug_set_mode(int argc, char *argv[]);
|
||
|
static int wifi_debug_set_prot(int argc, char *argv[]);
|
||
|
static int wifi_debug_set_autoconnect(int argc, char *argv[]);
|
||
|
#endif
|
||
|
|
||
|
/* cmd table */
|
||
|
static const struct wifi_cmd_des cmd_tab[] =
|
||
|
{
|
||
|
{"scan", wifi_scan},
|
||
|
{"help", wifi_help},
|
||
|
{"status", wifi_status},
|
||
|
{"join", wifi_join},
|
||
|
{"ap", wifi_ap},
|
||
|
{"list_sta", wifi_list_sta},
|
||
|
{"disc", wifi_disconnect},
|
||
|
{"ap_stop", wifi_ap_stop},
|
||
|
{"smartconfig", RT_NULL},
|
||
|
#ifdef RT_WLAN_CMD_DEBUG
|
||
|
{"-d", wifi_debug},
|
||
|
#endif
|
||
|
};
|
||
|
|
||
|
#ifdef RT_WLAN_CMD_DEBUG
|
||
|
/* debug cmd table */
|
||
|
static const struct wifi_cmd_des debug_tab[] =
|
||
|
{
|
||
|
{"save_cfg", wifi_debug_save_cfg},
|
||
|
{"dump_cfg", wifi_debug_dump_cfg},
|
||
|
{"clear_cfg", wifi_debug_clear_cfg},
|
||
|
{"dump_prot", wifi_debug_dump_prot},
|
||
|
{"mode", wifi_debug_set_mode},
|
||
|
{"prot", wifi_debug_set_prot},
|
||
|
{"auto", wifi_debug_set_autoconnect},
|
||
|
};
|
||
|
#endif
|
||
|
|
||
|
static int wifi_help(int argc, char *argv[])
|
||
|
{
|
||
|
rt_kprintf("wifi\n");
|
||
|
rt_kprintf("wifi help\n");
|
||
|
rt_kprintf("wifi scan [SSID]\n");
|
||
|
rt_kprintf("wifi join [SSID] [PASSWORD]\n");
|
||
|
rt_kprintf("wifi ap SSID [PASSWORD]\n");
|
||
|
rt_kprintf("wifi disc\n");
|
||
|
rt_kprintf("wifi ap_stop\n");
|
||
|
rt_kprintf("wifi status\n");
|
||
|
rt_kprintf("wifi smartconfig\n");
|
||
|
#ifdef RT_WLAN_CMD_DEBUG
|
||
|
rt_kprintf("wifi -d debug command\n");
|
||
|
#endif
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_status(int argc, char *argv[])
|
||
|
{
|
||
|
int rssi;
|
||
|
struct rt_wlan_info info;
|
||
|
|
||
|
if (argc > 2)
|
||
|
return -1;
|
||
|
|
||
|
if (rt_wlan_is_connected() == 1)
|
||
|
{
|
||
|
rssi = rt_wlan_get_rssi();
|
||
|
rt_wlan_get_info(&info);
|
||
|
rt_kprintf("Wi-Fi STA Info\n");
|
||
|
rt_kprintf("SSID : %-.32s\n", &info.ssid.val[0]);
|
||
|
rt_kprintf("MAC Addr: %02x:%02x:%02x:%02x:%02x:%02x\n", info.bssid[0],
|
||
|
info.bssid[1],
|
||
|
info.bssid[2],
|
||
|
info.bssid[3],
|
||
|
info.bssid[4],
|
||
|
info.bssid[5]);
|
||
|
rt_kprintf("Channel: %d\n", info.channel);
|
||
|
rt_kprintf("DataRate: %dMbps\n", info.datarate / 1000000);
|
||
|
rt_kprintf("RSSI: %d\n", rssi);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("wifi disconnected!\n");
|
||
|
}
|
||
|
|
||
|
if (rt_wlan_ap_is_active() == 1)
|
||
|
{
|
||
|
rt_wlan_ap_get_info(&info);
|
||
|
rt_kprintf("Wi-Fi AP Info\n");
|
||
|
rt_kprintf("SSID : %-.32s\n", &info.ssid.val[0]);
|
||
|
rt_kprintf("MAC Addr: %02x:%02x:%02x:%02x:%02x:%02x\n", info.bssid[0],
|
||
|
info.bssid[1],
|
||
|
info.bssid[2],
|
||
|
info.bssid[3],
|
||
|
info.bssid[4],
|
||
|
info.bssid[5]);
|
||
|
rt_kprintf("Channel: %d\n", info.channel);
|
||
|
rt_kprintf("DataRate: %dMbps\n", info.datarate / 1000000);
|
||
|
rt_kprintf("hidden: %s\n", info.hidden ? "Enable" : "Disable");
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("wifi ap not start!\n");
|
||
|
}
|
||
|
rt_kprintf("Auto Connect status:%s!\n", (rt_wlan_get_autoreconnect_mode() ? "Enable" : "Disable"));
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_scan(int argc, char *argv[])
|
||
|
{
|
||
|
struct rt_wlan_scan_result *scan_result = RT_NULL;
|
||
|
struct rt_wlan_info *info = RT_NULL;
|
||
|
struct rt_wlan_info filter;
|
||
|
|
||
|
if (argc > 3)
|
||
|
return -1;
|
||
|
|
||
|
if (argc == 3)
|
||
|
{
|
||
|
INVALID_INFO(&filter);
|
||
|
SSID_SET(&filter, argv[2]);
|
||
|
info = &filter;
|
||
|
}
|
||
|
|
||
|
/* clean scan result */
|
||
|
rt_wlan_scan_result_clean();
|
||
|
/* scan ap info */
|
||
|
scan_result = rt_wlan_scan_with_info(info);
|
||
|
if (scan_result)
|
||
|
{
|
||
|
int index, num;
|
||
|
char *security;
|
||
|
|
||
|
num = scan_result->num;
|
||
|
rt_kprintf(" SSID MAC security rssi chn Mbps\n");
|
||
|
rt_kprintf("------------------------------- ----------------- -------------- ---- --- ----\n");
|
||
|
for (index = 0; index < num; index ++)
|
||
|
{
|
||
|
rt_kprintf("%-32.32s", &scan_result->info[index].ssid.val[0]);
|
||
|
rt_kprintf("%02x:%02x:%02x:%02x:%02x:%02x ",
|
||
|
scan_result->info[index].bssid[0],
|
||
|
scan_result->info[index].bssid[1],
|
||
|
scan_result->info[index].bssid[2],
|
||
|
scan_result->info[index].bssid[3],
|
||
|
scan_result->info[index].bssid[4],
|
||
|
scan_result->info[index].bssid[5]
|
||
|
);
|
||
|
switch (scan_result->info[index].security)
|
||
|
{
|
||
|
case SECURITY_OPEN:
|
||
|
security = "OPEN";
|
||
|
break;
|
||
|
case SECURITY_WEP_PSK:
|
||
|
security = "WEP_PSK";
|
||
|
break;
|
||
|
case SECURITY_WEP_SHARED:
|
||
|
security = "WEP_SHARED";
|
||
|
break;
|
||
|
case SECURITY_WPA_TKIP_PSK:
|
||
|
security = "WPA_TKIP_PSK";
|
||
|
break;
|
||
|
case SECURITY_WPA_AES_PSK:
|
||
|
security = "WPA_AES_PSK";
|
||
|
break;
|
||
|
case SECURITY_WPA2_AES_PSK:
|
||
|
security = "WPA2_AES_PSK";
|
||
|
break;
|
||
|
case SECURITY_WPA2_TKIP_PSK:
|
||
|
security = "WPA2_TKIP_PSK";
|
||
|
break;
|
||
|
case SECURITY_WPA2_MIXED_PSK:
|
||
|
security = "WPA2_MIXED_PSK";
|
||
|
break;
|
||
|
case SECURITY_WPS_OPEN:
|
||
|
security = "WPS_OPEN";
|
||
|
break;
|
||
|
case SECURITY_WPS_SECURE:
|
||
|
security = "WPS_SECURE";
|
||
|
break;
|
||
|
default:
|
||
|
security = "UNKNOWN";
|
||
|
break;
|
||
|
}
|
||
|
rt_kprintf("%-14.14s ", security);
|
||
|
rt_kprintf("%-4d ", scan_result->info[index].rssi);
|
||
|
rt_kprintf("%3d ", scan_result->info[index].channel);
|
||
|
rt_kprintf("%4d\n", scan_result->info[index].datarate / 1000000);
|
||
|
}
|
||
|
rt_wlan_scan_result_clean();
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rt_kprintf("wifi scan result is null\n");
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_join(int argc, char *argv[])
|
||
|
{
|
||
|
const char *ssid = RT_NULL;
|
||
|
const char *key = RT_NULL;
|
||
|
struct rt_wlan_cfg_info cfg_info;
|
||
|
|
||
|
rt_memset(&cfg_info, 0, sizeof(cfg_info));
|
||
|
if (argc == 2)
|
||
|
{
|
||
|
#ifdef RT_WLAN_CFG_ENABLE
|
||
|
/* get info to connect */
|
||
|
if (rt_wlan_cfg_read_index(&cfg_info, 0) == 1)
|
||
|
{
|
||
|
ssid = (char *)(&cfg_info.info.ssid.val[0]);
|
||
|
if (cfg_info.key.len)
|
||
|
key = (char *)(&cfg_info.key.val[0]);
|
||
|
}
|
||
|
else
|
||
|
#endif
|
||
|
{
|
||
|
rt_kprintf("not find connect info\n");
|
||
|
}
|
||
|
}
|
||
|
else if (argc == 3)
|
||
|
{
|
||
|
/* ssid */
|
||
|
ssid = argv[2];
|
||
|
}
|
||
|
else if (argc == 4)
|
||
|
{
|
||
|
ssid = argv[2];
|
||
|
/* password */
|
||
|
key = argv[3];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
rt_wlan_connect(ssid, key);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_ap(int argc, char *argv[])
|
||
|
{
|
||
|
const char *ssid = RT_NULL;
|
||
|
const char *key = RT_NULL;
|
||
|
|
||
|
if (argc == 3)
|
||
|
{
|
||
|
ssid = argv[2];
|
||
|
}
|
||
|
else if (argc == 4)
|
||
|
{
|
||
|
ssid = argv[2];
|
||
|
key = argv[3];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
rt_wlan_start_ap(ssid, key);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_list_sta(int argc, char *argv[])
|
||
|
{
|
||
|
struct rt_wlan_info *sta_info;
|
||
|
int num, i;
|
||
|
|
||
|
if (argc > 2)
|
||
|
return -1;
|
||
|
num = rt_wlan_ap_get_sta_num();
|
||
|
sta_info = rt_malloc(sizeof(struct rt_wlan_info) * num);
|
||
|
if (sta_info == RT_NULL)
|
||
|
{
|
||
|
rt_kprintf("num:%d\n", num);
|
||
|
return 0;
|
||
|
}
|
||
|
rt_wlan_ap_get_sta_info(sta_info, num);
|
||
|
rt_kprintf("num:%d\n", num);
|
||
|
for (i = 0; i < num; i++)
|
||
|
{
|
||
|
rt_kprintf("sta mac %02x:%02x:%02x:%02x:%02x:%02x\n",
|
||
|
sta_info[i].bssid[0], sta_info[i].bssid[1], sta_info[i].bssid[2],
|
||
|
sta_info[i].bssid[3], sta_info[i].bssid[4], sta_info[i].bssid[5]);
|
||
|
}
|
||
|
rt_free(sta_info);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_disconnect(int argc, char *argv[])
|
||
|
{
|
||
|
if (argc != 2)
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
rt_wlan_disconnect();
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_ap_stop(int argc, char *argv[])
|
||
|
{
|
||
|
if (argc != 2)
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
rt_wlan_ap_stop();
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
#ifdef RT_WLAN_CMD_DEBUG
|
||
|
/* just for debug */
|
||
|
static int wifi_debug_help(int argc, char *argv[])
|
||
|
{
|
||
|
rt_kprintf("save_cfg ssid [password]\n");
|
||
|
rt_kprintf("dump_cfg\n");
|
||
|
rt_kprintf("clear_cfg\n");
|
||
|
rt_kprintf("dump_prot\n");
|
||
|
rt_kprintf("mode sta/ap dev_name\n");
|
||
|
rt_kprintf("prot lwip dev_name\n");
|
||
|
rt_kprintf("auto enable/disable\n");
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_debug_save_cfg(int argc, char *argv[])
|
||
|
{
|
||
|
struct rt_wlan_cfg_info cfg_info;
|
||
|
int len;
|
||
|
char *ssid = RT_NULL, *password = RT_NULL;
|
||
|
|
||
|
rt_memset(&cfg_info, 0, sizeof(cfg_info));
|
||
|
INVALID_INFO(&cfg_info.info);
|
||
|
if (argc == 2)
|
||
|
{
|
||
|
ssid = argv[1];
|
||
|
}
|
||
|
else if (argc == 3)
|
||
|
{
|
||
|
ssid = argv[1];
|
||
|
password = argv[2];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
if (ssid != RT_NULL)
|
||
|
{
|
||
|
len = rt_strlen(ssid);
|
||
|
if (len > RT_WLAN_SSID_MAX_LENGTH)
|
||
|
{
|
||
|
rt_kprintf("ssid is to long");
|
||
|
return 0;
|
||
|
}
|
||
|
rt_memcpy(&cfg_info.info.ssid.val[0], ssid, len);
|
||
|
cfg_info.info.ssid.len = len;
|
||
|
}
|
||
|
|
||
|
if (password != RT_NULL)
|
||
|
{
|
||
|
len = rt_strlen(password);
|
||
|
if (len > RT_WLAN_PASSWORD_MAX_LENGTH)
|
||
|
{
|
||
|
rt_kprintf("password is to long");
|
||
|
return 0;
|
||
|
}
|
||
|
rt_memcpy(&cfg_info.key.val[0], password, len);
|
||
|
cfg_info.key.len = len;
|
||
|
}
|
||
|
#ifdef RT_WLAN_CFG_ENABLE
|
||
|
rt_wlan_cfg_save(&cfg_info);
|
||
|
#endif
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_debug_dump_cfg(int argc, char *argv[])
|
||
|
{
|
||
|
if (argc == 1)
|
||
|
{
|
||
|
#ifdef RT_WLAN_CFG_ENABLE
|
||
|
rt_wlan_cfg_dump();
|
||
|
#endif
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_debug_clear_cfg(int argc, char *argv[])
|
||
|
{
|
||
|
if (argc == 1)
|
||
|
{
|
||
|
#ifdef RT_WLAN_CFG_ENABLE
|
||
|
rt_wlan_cfg_delete_all();
|
||
|
rt_wlan_cfg_cache_save();
|
||
|
#endif
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_debug_dump_prot(int argc, char *argv[])
|
||
|
{
|
||
|
if (argc == 1)
|
||
|
{
|
||
|
rt_wlan_prot_dump();
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_debug_set_mode(int argc, char *argv[])
|
||
|
{
|
||
|
rt_wlan_mode_t mode;
|
||
|
|
||
|
if (argc != 3)
|
||
|
return -1;
|
||
|
|
||
|
if (rt_strcmp("sta", argv[1]) == 0)
|
||
|
{
|
||
|
mode = RT_WLAN_STATION;
|
||
|
}
|
||
|
else if (rt_strcmp("ap", argv[1]) == 0)
|
||
|
{
|
||
|
mode = RT_WLAN_AP;
|
||
|
}
|
||
|
else if (rt_strcmp("none", argv[1]) == 0)
|
||
|
{
|
||
|
mode = RT_WLAN_NONE;
|
||
|
}
|
||
|
else
|
||
|
return -1;
|
||
|
|
||
|
rt_wlan_set_mode(argv[2], mode);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_debug_set_prot(int argc, char *argv[])
|
||
|
{
|
||
|
if (argc != 3)
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
rt_wlan_prot_attach(argv[2], argv[1]);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_debug_set_autoconnect(int argc, char *argv[])
|
||
|
{
|
||
|
if (argc == 2)
|
||
|
{
|
||
|
if (rt_strcmp(argv[1], "enable") == 0)
|
||
|
rt_wlan_config_autoreconnect(RT_TRUE);
|
||
|
else if (rt_strcmp(argv[1], "disable") == 0)
|
||
|
rt_wlan_config_autoreconnect(RT_FALSE);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int wifi_debug(int argc, char *argv[])
|
||
|
{
|
||
|
int i, result = 0;
|
||
|
const struct wifi_cmd_des *run_cmd = RT_NULL;
|
||
|
|
||
|
if (argc < 3)
|
||
|
{
|
||
|
wifi_debug_help(0, RT_NULL);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < sizeof(debug_tab) / sizeof(debug_tab[0]); i++)
|
||
|
{
|
||
|
if (rt_strcmp(debug_tab[i].cmd, argv[2]) == 0)
|
||
|
{
|
||
|
run_cmd = &debug_tab[i];
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (run_cmd == RT_NULL)
|
||
|
{
|
||
|
wifi_debug_help(0, RT_NULL);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
if (run_cmd->fun != RT_NULL)
|
||
|
{
|
||
|
result = run_cmd->fun(argc - 2, &argv[2]);
|
||
|
}
|
||
|
|
||
|
if (result)
|
||
|
{
|
||
|
wifi_debug_help(argc - 2, &argv[2]);
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
#endif
|
||
|
|
||
|
static int wifi_msh(int argc, char *argv[])
|
||
|
{
|
||
|
int i, result = 0;
|
||
|
const struct wifi_cmd_des *run_cmd = RT_NULL;
|
||
|
|
||
|
if (argc == 1)
|
||
|
{
|
||
|
wifi_help(argc, argv);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/* find fun */
|
||
|
for (i = 0; i < sizeof(cmd_tab) / sizeof(cmd_tab[0]); i++)
|
||
|
{
|
||
|
if (rt_strcmp(cmd_tab[i].cmd, argv[1]) == 0)
|
||
|
{
|
||
|
run_cmd = &cmd_tab[i];
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* not find fun, print help */
|
||
|
if (run_cmd == RT_NULL)
|
||
|
{
|
||
|
wifi_help(argc, argv);
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
/* run fun */
|
||
|
if (run_cmd->fun != RT_NULL)
|
||
|
{
|
||
|
result = run_cmd->fun(argc, argv);
|
||
|
}
|
||
|
|
||
|
if (result)
|
||
|
{
|
||
|
wifi_help(argc, argv);
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
#if defined(RT_USING_FINSH)
|
||
|
MSH_CMD_EXPORT_ALIAS(wifi_msh, wifi, wifi command);
|
||
|
#endif
|
||
|
|
||
|
#endif
|