|
|
@@ -0,0 +1,173 @@
|
|
|
+/*
|
|
|
+ * SPDX-FileCopyrightText: 2021 Espressif Systems (Shanghai) CO LTD
|
|
|
+ *
|
|
|
+ * SPDX-License-Identifier: CC0-1.0
|
|
|
+ *
|
|
|
+ * OpenThread Border Router Example
|
|
|
+ *
|
|
|
+ * This example code is in the Public Domain (or CC0 licensed, at your option.)
|
|
|
+ *
|
|
|
+ * Unless required by applicable law or agreed to in writing, this
|
|
|
+ * software is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
|
+ * CONDITIONS OF ANY KIND, either express or implied.
|
|
|
+*/
|
|
|
+
|
|
|
+#include <stdio.h>
|
|
|
+#include <string.h>
|
|
|
+#include <stdlib.h>
|
|
|
+#include "esp_netif_types.h"
|
|
|
+#include "esp_openthread_border_router.h"
|
|
|
+#include "esp_check.h"
|
|
|
+#include "esp_event.h"
|
|
|
+#include "esp_openthread_lock.h"
|
|
|
+#include "esp_netif.h"
|
|
|
+#include "esp_netif_ip_addr.h"
|
|
|
+#include "esp_netif_net_stack.h"
|
|
|
+#include "esp_wifi.h"
|
|
|
+#include "freertos/event_groups.h"
|
|
|
+#include "freertos/FreeRTOS.h"
|
|
|
+#include "freertos/portmacro.h"
|
|
|
+#include "freertos/task.h"
|
|
|
+#include "openthread/cli.h"
|
|
|
+
|
|
|
+static EventGroupHandle_t wifi_event_group;
|
|
|
+const int CONNECTED_IP4_BIT = BIT0;
|
|
|
+const int CONNECTED_IP6_BIT = BIT1;
|
|
|
+static bool s_wifi_connected = false;
|
|
|
+
|
|
|
+void esp_ot_wifi_netif_init(void)
|
|
|
+{
|
|
|
+ esp_netif_t *esp_netif = esp_netif_create_default_wifi_sta();
|
|
|
+ assert(esp_netif);
|
|
|
+}
|
|
|
+
|
|
|
+static void event_handler(void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data)
|
|
|
+{
|
|
|
+ if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_DISCONNECTED) {
|
|
|
+ xEventGroupClearBits(wifi_event_group, CONNECTED_IP4_BIT);
|
|
|
+ xEventGroupClearBits(wifi_event_group, CONNECTED_IP6_BIT);
|
|
|
+ esp_wifi_connect();
|
|
|
+ } else if (event_base == IP_EVENT && event_id == IP_EVENT_STA_GOT_IP) {
|
|
|
+ xEventGroupSetBits(wifi_event_group, CONNECTED_IP4_BIT);
|
|
|
+ } else if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_CONNECTED) {
|
|
|
+ esp_netif_create_ip6_linklocal(esp_netif_get_handle_from_ifkey("WIFI_STA_DEF"));
|
|
|
+ } else if (event_base == IP_EVENT && event_id == IP_EVENT_GOT_IP6) {
|
|
|
+ xEventGroupSetBits(wifi_event_group, CONNECTED_IP6_BIT);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static void wifi_join(const char *ssid, const char *psk)
|
|
|
+{
|
|
|
+ static bool s_initialized = false;
|
|
|
+ if (!s_initialized) {
|
|
|
+ wifi_event_group = xEventGroupCreate();
|
|
|
+ wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
|
|
+ esp_wifi_init(&cfg);
|
|
|
+ ESP_ERROR_CHECK( esp_event_handler_register(WIFI_EVENT, WIFI_EVENT_STA_DISCONNECTED, &event_handler, NULL) );
|
|
|
+ ESP_ERROR_CHECK( esp_event_handler_register(IP_EVENT, IP_EVENT_STA_GOT_IP, &event_handler, NULL) );
|
|
|
+ ESP_ERROR_CHECK( esp_event_handler_register(WIFI_EVENT, WIFI_EVENT_STA_CONNECTED, &event_handler, NULL) );
|
|
|
+ ESP_ERROR_CHECK( esp_event_handler_register(IP_EVENT, IP_EVENT_GOT_IP6, &event_handler, NULL) );
|
|
|
+ esp_wifi_set_storage(WIFI_STORAGE_RAM);
|
|
|
+ esp_wifi_set_mode(WIFI_MODE_NULL);
|
|
|
+ s_initialized = true;
|
|
|
+ }
|
|
|
+ esp_wifi_start();
|
|
|
+ wifi_config_t wifi_config = { 0 };
|
|
|
+ strncpy((char *) wifi_config.sta.ssid, ssid, sizeof(wifi_config.sta.ssid));
|
|
|
+ if (psk) {
|
|
|
+ strncpy((char *) wifi_config.sta.password, psk, sizeof(wifi_config.sta.password));
|
|
|
+ }
|
|
|
+
|
|
|
+ ESP_ERROR_CHECK( esp_wifi_set_ps(WIFI_PS_NONE) );
|
|
|
+ ESP_ERROR_CHECK( esp_wifi_set_mode(WIFI_MODE_STA) );
|
|
|
+ ESP_ERROR_CHECK( esp_wifi_set_config(WIFI_IF_STA, &wifi_config) );
|
|
|
+ esp_wifi_connect();
|
|
|
+
|
|
|
+ int bits = xEventGroupWaitBits(wifi_event_group, CONNECTED_IP4_BIT | CONNECTED_IP6_BIT, pdFALSE, pdTRUE, 10000 / portTICK_PERIOD_MS);
|
|
|
+
|
|
|
+ if (((bits & CONNECTED_IP4_BIT) != 0) && ((bits & CONNECTED_IP6_BIT) != 0)) {
|
|
|
+ s_wifi_connected = true;
|
|
|
+ xEventGroupClearBits(wifi_event_group, CONNECTED_IP4_BIT);
|
|
|
+ xEventGroupClearBits(wifi_event_group, CONNECTED_IP6_BIT);
|
|
|
+ esp_event_handler_unregister(WIFI_EVENT, WIFI_EVENT_STA_DISCONNECTED, &event_handler);
|
|
|
+ esp_event_handler_unregister(WIFI_EVENT, WIFI_EVENT_STA_CONNECTED, &event_handler);
|
|
|
+ esp_event_handler_unregister(IP_EVENT, IP_EVENT_STA_GOT_IP, &event_handler);
|
|
|
+ esp_event_handler_unregister(IP_EVENT, IP_EVENT_GOT_IP6, &event_handler);
|
|
|
+ vEventGroupDelete(wifi_event_group);
|
|
|
+ s_initialized = false;
|
|
|
+ esp_openthread_lock_acquire(portMAX_DELAY);
|
|
|
+ esp_openthread_border_router_init();
|
|
|
+ esp_openthread_lock_release();
|
|
|
+ otCliOutputFormat("wifi sta is connected successfully\n");
|
|
|
+ } else {
|
|
|
+ esp_wifi_disconnect();
|
|
|
+ esp_wifi_stop();
|
|
|
+ ESP_LOGW("","Connection time out, please check your ssid & password, then retry.");
|
|
|
+ otCliOutputFormat("wifi sta connection is failed\n");
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+void esp_ot_process_wifi_sta(void *aContext, uint8_t aArgsLength, char *aArgs[])
|
|
|
+{
|
|
|
+ char ssid[100] = "";
|
|
|
+ char psk[100] = "";
|
|
|
+ (void)(aContext);
|
|
|
+ if (aArgsLength == 0) {
|
|
|
+ otCliOutputFormat("---wifi sta parameter---\n");
|
|
|
+ otCliOutputFormat("-s : wifi ssid\n");
|
|
|
+ otCliOutputFormat("-p : wifi psk\n");
|
|
|
+ otCliOutputFormat("---example---\n");
|
|
|
+ otCliOutputFormat("join a wifi:\nssid: threadcertAP \npsk: threadcertAP : sta -s threadcertAP -p threadcertAP\n");
|
|
|
+ } else {
|
|
|
+ for (int i = 0; i < aArgsLength; i++) {
|
|
|
+ if (strcmp(aArgs[i], "-s") == 0) {
|
|
|
+ i++;
|
|
|
+ strcpy(ssid, aArgs[i]);
|
|
|
+ otCliOutputFormat("ssid: %s\n", ssid);
|
|
|
+ } else if (strcmp(aArgs[i], "-p") == 0) {
|
|
|
+ i++;
|
|
|
+ strcpy(psk, aArgs[i]);
|
|
|
+ otCliOutputFormat("psk: %s\n", psk);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (!s_wifi_connected) {
|
|
|
+ wifi_join(ssid, psk);
|
|
|
+ } else {
|
|
|
+ otCliOutputFormat("wifi has already connected\n");
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+void esp_ot_process_get_wifi_info(void *aContext, uint8_t aArgsLength, char *aArgs[])
|
|
|
+{
|
|
|
+ (void)(aContext);
|
|
|
+ if (aArgsLength == 0) {
|
|
|
+ otCliOutputFormat("---get wifi informations---\n");
|
|
|
+ otCliOutputFormat("-i : get sta addr\n");
|
|
|
+ otCliOutputFormat("-s : get wifi state, disconnect or connect\n");
|
|
|
+ } else {
|
|
|
+ for (int i = 0; i < aArgsLength; i++) {
|
|
|
+ if (strcmp(aArgs[i], "-i") == 0) {
|
|
|
+ if (s_wifi_connected) {
|
|
|
+ esp_netif_ip_info_t local_ip;
|
|
|
+ char addr_str[128];
|
|
|
+ esp_netif_get_ip_info(esp_netif_get_handle_from_ifkey("WIFI_STA_DEF"), &local_ip);
|
|
|
+ ip4addr_ntoa_r((ip4_addr_t *)(&local_ip.ip), addr_str, sizeof(addr_str) - 1);
|
|
|
+ otCliOutputFormat("inet %s\n");
|
|
|
+ esp_ip6_addr_t if_ip6;
|
|
|
+ esp_netif_get_ip6_linklocal(esp_netif_get_handle_from_ifkey("WIFI_STA_DEF"), &if_ip6);
|
|
|
+ ip6addr_ntoa_r((ip6_addr_t *)(&if_ip6), addr_str, sizeof(addr_str) - 1);
|
|
|
+ otCliOutputFormat("inet6 %s\n", addr_str);
|
|
|
+ } else {
|
|
|
+ otCliOutputFormat("wifi is disconnected\n");
|
|
|
+ }
|
|
|
+ } else if (strcmp(aArgs[i], "-s") == 0) {
|
|
|
+ if (s_wifi_connected) {
|
|
|
+ otCliOutputFormat("connected\n");
|
|
|
+ } else {
|
|
|
+ otCliOutputFormat("disconnected\n");
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
zhangwenxu