forked from espressif/esp-idf
88 lines
2.5 KiB
C
88 lines
2.5 KiB
C
/*
|
|
* SPDX-FileCopyrightText: 2022 Espressif Systems (Shanghai) CO LTD
|
|
*
|
|
* SPDX-License-Identifier: Unlicense OR CC0-1.0
|
|
*/
|
|
|
|
|
|
#include <string.h>
|
|
#include "protocol_examples_common.h"
|
|
#include "example_common_private.h"
|
|
#include "esp_wifi.h"
|
|
#include "esp_log.h"
|
|
#include "esp_console.h"
|
|
#include "argtable3/argtable3.h"
|
|
|
|
|
|
static const char *TAG = "example_console";
|
|
|
|
typedef struct {
|
|
struct arg_str *ssid;
|
|
struct arg_str *password;
|
|
struct arg_int *channel;
|
|
struct arg_end *end;
|
|
} wifi_connect_args_t;
|
|
static wifi_connect_args_t connect_args;
|
|
|
|
static int cmd_do_wifi_connect(int argc, char **argv)
|
|
{
|
|
int nerrors = arg_parse(argc, argv, (void **) &connect_args);
|
|
|
|
if (nerrors != 0) {
|
|
arg_print_errors(stderr, connect_args.end, argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
wifi_config_t wifi_config = {
|
|
.sta = {
|
|
.scan_method = WIFI_ALL_CHANNEL_SCAN,
|
|
.sort_method = WIFI_CONNECT_AP_BY_SIGNAL,
|
|
},
|
|
};
|
|
if (connect_args.channel->count > 0) {
|
|
wifi_config.sta.channel = (uint8_t)(connect_args.channel->ival[0]);
|
|
}
|
|
const char *ssid = connect_args.ssid->sval[0];
|
|
const char *pass = connect_args.password->sval[0];
|
|
strlcpy((char *) wifi_config.sta.ssid, ssid, sizeof(wifi_config.sta.ssid));
|
|
if (pass) {
|
|
strlcpy((char *) wifi_config.sta.password, pass, sizeof(wifi_config.sta.password));
|
|
}
|
|
example_wifi_sta_do_connect(wifi_config, false);
|
|
return 0;
|
|
}
|
|
|
|
static int cmd_do_wifi_disconnect(int argc, char **argv)
|
|
{
|
|
example_wifi_sta_do_disconnect();
|
|
return 0;
|
|
}
|
|
|
|
void register_wifi_connect_commands(void)
|
|
{
|
|
ESP_LOGI(TAG, "Registering WiFi connect commands.");
|
|
example_wifi_start();
|
|
|
|
connect_args.ssid = arg_str1(NULL, NULL, "<ssid>", "SSID of AP");
|
|
connect_args.password = arg_str0(NULL, NULL, "<pass>", "password of AP");
|
|
connect_args.channel = arg_int0("n", "channel", "<channel>", "channel of AP");
|
|
connect_args.end = arg_end(2);
|
|
const esp_console_cmd_t wifi_connect_cmd = {
|
|
.command = "wifi_connect",
|
|
.help = "WiFi is station mode, join specified soft-AP",
|
|
.hint = NULL,
|
|
.func = &cmd_do_wifi_connect,
|
|
.argtable = &connect_args
|
|
};
|
|
ESP_ERROR_CHECK( esp_console_cmd_register(&wifi_connect_cmd) );
|
|
|
|
|
|
const esp_console_cmd_t wifi_disconnect_cmd = {
|
|
.command = "wifi_disconnect",
|
|
.help = "Do wifi disconnect",
|
|
.hint = NULL,
|
|
.func = &cmd_do_wifi_disconnect,
|
|
};
|
|
ESP_ERROR_CHECK( esp_console_cmd_register(&wifi_disconnect_cmd) );
|
|
}
|