mirror of
https://github.com/espressif/esp-idf.git
synced 2024-10-05 20:47:46 -04:00
314 lines
11 KiB
C
314 lines
11 KiB
C
/* Common functions for protocol examples, to establish Wi-Fi or Ethernet connection.
|
|
|
|
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 <string.h>
|
|
#include "protocol_examples_common.h"
|
|
#include "sdkconfig.h"
|
|
#include "esp_event.h"
|
|
#include "esp_wifi.h"
|
|
#include "esp_wifi_default.h"
|
|
#if CONFIG_EXAMPLE_CONNECT_ETHERNET
|
|
#include "esp_eth.h"
|
|
#endif
|
|
#include "esp_log.h"
|
|
#include "esp_netif.h"
|
|
#include "driver/gpio.h"
|
|
#include "freertos/FreeRTOS.h"
|
|
#include "freertos/task.h"
|
|
#include "freertos/event_groups.h"
|
|
#include "lwip/err.h"
|
|
#include "lwip/sys.h"
|
|
|
|
#define GOT_IPV4_BIT BIT(0)
|
|
#define GOT_IPV6_BIT BIT(1)
|
|
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
#define CONNECTED_BITS (GOT_IPV4_BIT | GOT_IPV6_BIT)
|
|
#else
|
|
#define CONNECTED_BITS (GOT_IPV4_BIT)
|
|
#endif
|
|
|
|
static EventGroupHandle_t s_connect_event_group;
|
|
static esp_ip4_addr_t s_ip_addr;
|
|
static const char *s_connection_name;
|
|
static esp_netif_t *s_example_esp_netif = NULL;
|
|
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
static esp_ip6_addr_t s_ipv6_addr;
|
|
#endif
|
|
|
|
static const char *TAG = "example_connect";
|
|
|
|
/* set up connection, Wi-Fi or Ethernet */
|
|
static void start(void);
|
|
|
|
/* tear down connection, release resources */
|
|
static void stop(void);
|
|
|
|
static void on_got_ip(void *arg, esp_event_base_t event_base,
|
|
int32_t event_id, void *event_data)
|
|
{
|
|
ESP_LOGI(TAG, "Got IP event!");
|
|
ip_event_got_ip_t *event = (ip_event_got_ip_t *)event_data;
|
|
memcpy(&s_ip_addr, &event->ip_info.ip, sizeof(s_ip_addr));
|
|
xEventGroupSetBits(s_connect_event_group, GOT_IPV4_BIT);
|
|
}
|
|
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
|
|
static void on_got_ipv6(void *arg, esp_event_base_t event_base,
|
|
int32_t event_id, void *event_data)
|
|
{
|
|
ip_event_got_ip6_t *event = (ip_event_got_ip6_t *)event_data;
|
|
if (event->esp_netif != s_example_esp_netif) {
|
|
ESP_LOGD(TAG, "Got IPv6 from another netif: ignored");
|
|
return;
|
|
}
|
|
ESP_LOGI(TAG, "Got IPv6 event!");
|
|
memcpy(&s_ipv6_addr, &event->ip6_info.ip, sizeof(s_ipv6_addr));
|
|
xEventGroupSetBits(s_connect_event_group, GOT_IPV6_BIT);
|
|
}
|
|
|
|
#endif // CONFIG_EXAMPLE_CONNECT_IPV6
|
|
|
|
esp_err_t example_connect(void)
|
|
{
|
|
if (s_connect_event_group != NULL) {
|
|
return ESP_ERR_INVALID_STATE;
|
|
}
|
|
s_connect_event_group = xEventGroupCreate();
|
|
start();
|
|
ESP_ERROR_CHECK(esp_register_shutdown_handler(&stop));
|
|
ESP_LOGI(TAG, "Waiting for IP");
|
|
xEventGroupWaitBits(s_connect_event_group, CONNECTED_BITS, true, true, portMAX_DELAY);
|
|
ESP_LOGI(TAG, "Connected to %s", s_connection_name);
|
|
ESP_LOGI(TAG, "IPv4 address: " IPSTR, IP2STR(&s_ip_addr));
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
ESP_LOGI(TAG, "IPv6 address: " IPV6STR, IPV62STR(s_ipv6_addr));
|
|
#endif
|
|
return ESP_OK;
|
|
}
|
|
|
|
esp_err_t example_disconnect(void)
|
|
{
|
|
if (s_connect_event_group == NULL) {
|
|
return ESP_ERR_INVALID_STATE;
|
|
}
|
|
vEventGroupDelete(s_connect_event_group);
|
|
s_connect_event_group = NULL;
|
|
stop();
|
|
ESP_LOGI(TAG, "Disconnected from %s", s_connection_name);
|
|
s_connection_name = NULL;
|
|
ESP_ERROR_CHECK(esp_unregister_shutdown_handler(&stop));
|
|
return ESP_OK;
|
|
}
|
|
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_WIFI
|
|
|
|
static void on_wifi_disconnect(void *arg, esp_event_base_t event_base,
|
|
int32_t event_id, void *event_data)
|
|
{
|
|
ESP_LOGI(TAG, "Wi-Fi disconnected, trying to reconnect...");
|
|
esp_err_t err = esp_wifi_connect();
|
|
if (err == ESP_ERR_WIFI_NOT_STARTED) {
|
|
return;
|
|
}
|
|
ESP_ERROR_CHECK(err);
|
|
}
|
|
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
|
|
static void on_wifi_connect(void *esp_netif, esp_event_base_t event_base,
|
|
int32_t event_id, void *event_data)
|
|
{
|
|
esp_netif_create_ip6_linklocal(esp_netif);
|
|
}
|
|
|
|
#endif // CONFIG_EXAMPLE_CONNECT_IPV6
|
|
|
|
static void start(void)
|
|
{
|
|
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
|
ESP_ERROR_CHECK(esp_wifi_init(&cfg));
|
|
|
|
esp_netif_config_t netif_config = ESP_NETIF_DEFAULT_WIFI_STA();
|
|
|
|
esp_netif_t *netif = esp_netif_new(&netif_config);
|
|
|
|
assert(netif);
|
|
|
|
esp_netif_attach_wifi_station(netif);
|
|
esp_wifi_set_default_wifi_sta_handlers();
|
|
|
|
s_example_esp_netif = netif;
|
|
|
|
ESP_ERROR_CHECK(esp_event_handler_register(WIFI_EVENT, WIFI_EVENT_STA_DISCONNECTED, &on_wifi_disconnect, NULL));
|
|
ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_STA_GOT_IP, &on_got_ip, NULL));
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
ESP_ERROR_CHECK(esp_event_handler_register(WIFI_EVENT, WIFI_EVENT_STA_CONNECTED, &on_wifi_connect, netif));
|
|
ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_GOT_IP6, &on_got_ipv6, NULL));
|
|
#endif
|
|
|
|
ESP_ERROR_CHECK(esp_wifi_set_storage(WIFI_STORAGE_RAM));
|
|
wifi_config_t wifi_config = {
|
|
.sta = {
|
|
.ssid = CONFIG_EXAMPLE_WIFI_SSID,
|
|
.password = CONFIG_EXAMPLE_WIFI_PASSWORD,
|
|
},
|
|
};
|
|
ESP_LOGI(TAG, "Connecting to %s...", wifi_config.sta.ssid);
|
|
ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA));
|
|
ESP_ERROR_CHECK(esp_wifi_set_config(ESP_IF_WIFI_STA, &wifi_config));
|
|
ESP_ERROR_CHECK(esp_wifi_start());
|
|
esp_wifi_connect();
|
|
s_connection_name = CONFIG_EXAMPLE_WIFI_SSID;
|
|
}
|
|
|
|
static void stop(void)
|
|
{
|
|
ESP_ERROR_CHECK(esp_event_handler_unregister(WIFI_EVENT, WIFI_EVENT_STA_DISCONNECTED, &on_wifi_disconnect));
|
|
ESP_ERROR_CHECK(esp_event_handler_unregister(IP_EVENT, IP_EVENT_STA_GOT_IP, &on_got_ip));
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
ESP_ERROR_CHECK(esp_event_handler_unregister(IP_EVENT, IP_EVENT_GOT_IP6, &on_got_ipv6));
|
|
ESP_ERROR_CHECK(esp_event_handler_unregister(WIFI_EVENT, WIFI_EVENT_STA_CONNECTED, &on_wifi_connect));
|
|
#endif
|
|
esp_err_t err = esp_wifi_stop();
|
|
if (err == ESP_ERR_WIFI_NOT_INIT) {
|
|
return;
|
|
}
|
|
ESP_ERROR_CHECK(err);
|
|
ESP_ERROR_CHECK(esp_wifi_deinit());
|
|
ESP_ERROR_CHECK(esp_wifi_clear_default_wifi_driver_and_handlers(s_example_esp_netif));
|
|
esp_netif_destroy(s_example_esp_netif);
|
|
s_example_esp_netif = NULL;
|
|
}
|
|
#endif // CONFIG_EXAMPLE_CONNECT_WIFI
|
|
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_ETHERNET
|
|
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
|
|
/** Event handler for Ethernet events */
|
|
static void on_eth_event(void *esp_netif, esp_event_base_t event_base,
|
|
int32_t event_id, void *event_data)
|
|
{
|
|
switch (event_id) {
|
|
case ETHERNET_EVENT_CONNECTED:
|
|
ESP_LOGI(TAG, "Ethernet Link Up");
|
|
esp_netif_create_ip6_linklocal(esp_netif);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
#endif // CONFIG_EXAMPLE_CONNECT_IPV6
|
|
|
|
static esp_eth_handle_t s_eth_handle = NULL;
|
|
static esp_eth_mac_t *s_mac = NULL;
|
|
static esp_eth_phy_t *s_phy = NULL;
|
|
static void *s_eth_glue = NULL;
|
|
|
|
static void start(void)
|
|
{
|
|
esp_netif_config_t netif_config = ESP_NETIF_DEFAULT_ETH();
|
|
esp_netif_t *netif = esp_netif_new(&netif_config);
|
|
assert(netif);
|
|
s_example_esp_netif = netif;
|
|
// Set default handlers to process TCP/IP stuffs
|
|
ESP_ERROR_CHECK(esp_eth_set_default_handlers(netif));
|
|
// Register user defined event handers
|
|
ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_ETH_GOT_IP, &on_got_ip, NULL));
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
ESP_ERROR_CHECK(esp_event_handler_register(ETH_EVENT, ETHERNET_EVENT_CONNECTED, &on_eth_event, netif));
|
|
ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_GOT_IP6, &on_got_ipv6, NULL));
|
|
#endif
|
|
eth_mac_config_t mac_config = ETH_MAC_DEFAULT_CONFIG();
|
|
eth_phy_config_t phy_config = ETH_PHY_DEFAULT_CONFIG();
|
|
phy_config.phy_addr = CONFIG_EXAMPLE_ETH_PHY_ADDR;
|
|
phy_config.reset_gpio_num = CONFIG_EXAMPLE_ETH_PHY_RST_GPIO;
|
|
#if CONFIG_EXAMPLE_USE_INTERNAL_ETHERNET
|
|
mac_config.smi_mdc_gpio_num = CONFIG_EXAMPLE_ETH_MDC_GPIO;
|
|
mac_config.smi_mdio_gpio_num = CONFIG_EXAMPLE_ETH_MDIO_GPIO;
|
|
s_mac = esp_eth_mac_new_esp32(&mac_config);
|
|
#if CONFIG_EXAMPLE_ETH_PHY_IP101
|
|
s_phy = esp_eth_phy_new_ip101(&phy_config);
|
|
#elif CONFIG_EXAMPLE_ETH_PHY_RTL8201
|
|
s_phy = esp_eth_phy_new_rtl8201(&phy_config);
|
|
#elif CONFIG_EXAMPLE_ETH_PHY_LAN8720
|
|
s_phy = esp_eth_phy_new_lan8720(&phy_config);
|
|
#elif CONFIG_EXAMPLE_ETH_PHY_DP83848
|
|
s_phy = esp_eth_phy_new_dp83848(&phy_config);
|
|
#endif
|
|
#elif CONFIG_EXAMPLE_USE_DM9051
|
|
gpio_install_isr_service(0);
|
|
spi_device_handle_t spi_handle = NULL;
|
|
spi_bus_config_t buscfg = {
|
|
.miso_io_num = CONFIG_EXAMPLE_DM9051_MISO_GPIO,
|
|
.mosi_io_num = CONFIG_EXAMPLE_DM9051_MOSI_GPIO,
|
|
.sclk_io_num = CONFIG_EXAMPLE_DM9051_SCLK_GPIO,
|
|
.quadwp_io_num = -1,
|
|
.quadhd_io_num = -1,
|
|
};
|
|
ESP_ERROR_CHECK(spi_bus_initialize(CONFIG_EXAMPLE_DM9051_SPI_HOST, &buscfg, 1));
|
|
spi_device_interface_config_t devcfg = {
|
|
.command_bits = 1,
|
|
.address_bits = 7,
|
|
.mode = 0,
|
|
.clock_speed_hz = CONFIG_EXAMPLE_DM9051_SPI_CLOCK_MHZ * 1000 * 1000,
|
|
.spics_io_num = CONFIG_EXAMPLE_DM9051_CS_GPIO,
|
|
.queue_size = 20
|
|
};
|
|
ESP_ERROR_CHECK(spi_bus_add_device(CONFIG_EXAMPLE_DM9051_SPI_HOST, &devcfg, &spi_handle));
|
|
/* dm9051 ethernet driver is based on spi driver */
|
|
eth_dm9051_config_t dm9051_config = ETH_DM9051_DEFAULT_CONFIG(spi_handle);
|
|
dm9051_config.int_gpio_num = CONFIG_EXAMPLE_DM9051_INT_GPIO;
|
|
s_mac = esp_eth_mac_new_dm9051(&dm9051_config, &mac_config);
|
|
s_phy = esp_eth_phy_new_dm9051(&phy_config);
|
|
#elif CONFIG_EXAMPLE_USE_OPENETH
|
|
phy_config.autonego_timeout_ms = 100;
|
|
s_mac = esp_eth_mac_new_openeth(&mac_config);
|
|
s_phy = esp_eth_phy_new_dp83848(&phy_config);
|
|
#endif
|
|
|
|
// Install Ethernet driver
|
|
esp_eth_config_t config = ETH_DEFAULT_CONFIG(s_mac, s_phy);
|
|
ESP_ERROR_CHECK(esp_eth_driver_install(&config, &s_eth_handle));
|
|
// combine driver with netif
|
|
s_eth_glue = esp_eth_new_netif_glue(s_eth_handle);
|
|
esp_netif_attach(netif, s_eth_glue);
|
|
esp_eth_start(s_eth_handle);
|
|
s_connection_name = "Ethernet";
|
|
}
|
|
|
|
static void stop(void)
|
|
{
|
|
ESP_ERROR_CHECK(esp_event_handler_unregister(IP_EVENT, IP_EVENT_ETH_GOT_IP, &on_got_ip));
|
|
#ifdef CONFIG_EXAMPLE_CONNECT_IPV6
|
|
ESP_ERROR_CHECK(esp_event_handler_unregister(IP_EVENT, IP_EVENT_GOT_IP6, &on_got_ipv6));
|
|
ESP_ERROR_CHECK(esp_event_handler_unregister(ETH_EVENT, ETHERNET_EVENT_CONNECTED, &on_eth_event));
|
|
#endif
|
|
ESP_ERROR_CHECK(esp_eth_stop(s_eth_handle));
|
|
ESP_ERROR_CHECK(esp_eth_del_netif_glue(s_eth_glue));
|
|
ESP_ERROR_CHECK(esp_eth_clear_default_handlers(s_example_esp_netif));
|
|
ESP_ERROR_CHECK(esp_eth_driver_uninstall(s_eth_handle));
|
|
ESP_ERROR_CHECK(s_phy->del(s_phy));
|
|
ESP_ERROR_CHECK(s_mac->del(s_mac));
|
|
|
|
esp_netif_destroy(s_example_esp_netif);
|
|
s_example_esp_netif = NULL;
|
|
}
|
|
|
|
#endif // CONFIG_EXAMPLE_CONNECT_ETHERNET
|
|
|
|
esp_netif_t *get_example_netif(void)
|
|
{
|
|
return s_example_esp_netif;
|
|
}
|