|
|
@@ -10,12 +10,13 @@
|
|
|
#include <sys/param.h>
|
|
|
#include "freertos/FreeRTOS.h"
|
|
|
#include "freertos/task.h"
|
|
|
-#include "freertos/event_groups.h"
|
|
|
#include "esp_system.h"
|
|
|
#include "esp_wifi.h"
|
|
|
-#include "esp_event_loop.h"
|
|
|
+#include "esp_event.h"
|
|
|
#include "esp_log.h"
|
|
|
#include "nvs_flash.h"
|
|
|
+#include "tcpip_adapter.h"
|
|
|
+#include "protocol_examples_common.h"
|
|
|
|
|
|
#include "lwip/err.h"
|
|
|
#include "lwip/sockets.h"
|
|
|
@@ -23,85 +24,10 @@
|
|
|
#include <lwip/netdb.h>
|
|
|
|
|
|
|
|
|
-/* The examples use simple WiFi configuration that you can set via
|
|
|
- 'make menuconfig'.
|
|
|
- If you'd rather not, just change the below entries to strings with
|
|
|
- the config you want - ie #define EXAMPLE_WIFI_SSID "mywifissid"
|
|
|
-*/
|
|
|
-#define EXAMPLE_WIFI_SSID CONFIG_WIFI_SSID
|
|
|
-#define EXAMPLE_WIFI_PASS CONFIG_WIFI_PASSWORD
|
|
|
-
|
|
|
#define PORT CONFIG_EXAMPLE_PORT
|
|
|
|
|
|
-/* FreeRTOS event group to signal when we are connected & ready to make a request */
|
|
|
-static EventGroupHandle_t wifi_event_group;
|
|
|
-
|
|
|
-const int IPV4_GOTIP_BIT = BIT0;
|
|
|
-const int IPV6_GOTIP_BIT = BIT1;
|
|
|
-
|
|
|
static const char *TAG = "example";
|
|
|
|
|
|
-static esp_err_t event_handler(void *ctx, system_event_t *event)
|
|
|
-{
|
|
|
- switch (event->event_id) {
|
|
|
- case SYSTEM_EVENT_STA_START:
|
|
|
- esp_wifi_connect();
|
|
|
- ESP_LOGI(TAG, "SYSTEM_EVENT_STA_START");
|
|
|
- break;
|
|
|
- case SYSTEM_EVENT_STA_CONNECTED:
|
|
|
- /* enable ipv6 */
|
|
|
- tcpip_adapter_create_ip6_linklocal(TCPIP_ADAPTER_IF_STA);
|
|
|
- break;
|
|
|
- case SYSTEM_EVENT_STA_GOT_IP:
|
|
|
- xEventGroupSetBits(wifi_event_group, IPV4_GOTIP_BIT);
|
|
|
- ESP_LOGI(TAG, "SYSTEM_EVENT_STA_GOT_IP");
|
|
|
- break;
|
|
|
- case SYSTEM_EVENT_STA_DISCONNECTED:
|
|
|
- /* This is a workaround as ESP32 WiFi libs don't currently auto-reassociate. */
|
|
|
- esp_wifi_connect();
|
|
|
- xEventGroupClearBits(wifi_event_group, IPV4_GOTIP_BIT);
|
|
|
- xEventGroupClearBits(wifi_event_group, IPV6_GOTIP_BIT);
|
|
|
- break;
|
|
|
- case SYSTEM_EVENT_AP_STA_GOT_IP6:
|
|
|
- xEventGroupSetBits(wifi_event_group, IPV6_GOTIP_BIT);
|
|
|
- ESP_LOGI(TAG, "SYSTEM_EVENT_STA_GOT_IP6");
|
|
|
-
|
|
|
- char *ip6 = ip6addr_ntoa(&event->event_info.got_ip6.ip6_info.ip);
|
|
|
- ESP_LOGI(TAG, "IPv6: %s", ip6);
|
|
|
- default:
|
|
|
- break;
|
|
|
- }
|
|
|
- return ESP_OK;
|
|
|
-}
|
|
|
-
|
|
|
-static void initialise_wifi(void)
|
|
|
-{
|
|
|
- tcpip_adapter_init();
|
|
|
- wifi_event_group = xEventGroupCreate();
|
|
|
- ESP_ERROR_CHECK( esp_event_loop_init(event_handler, NULL) );
|
|
|
- wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
|
|
- ESP_ERROR_CHECK( esp_wifi_init(&cfg) );
|
|
|
- ESP_ERROR_CHECK( esp_wifi_set_storage(WIFI_STORAGE_RAM) );
|
|
|
- wifi_config_t wifi_config = {
|
|
|
- .sta = {
|
|
|
- .ssid = EXAMPLE_WIFI_SSID,
|
|
|
- .password = EXAMPLE_WIFI_PASS,
|
|
|
- },
|
|
|
- };
|
|
|
- ESP_LOGI(TAG, "Setting WiFi configuration SSID %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() );
|
|
|
-}
|
|
|
-
|
|
|
-static void wait_for_ip()
|
|
|
-{
|
|
|
- uint32_t bits = IPV4_GOTIP_BIT | IPV6_GOTIP_BIT ;
|
|
|
-
|
|
|
- ESP_LOGI(TAG, "Waiting for AP connection...");
|
|
|
- xEventGroupWaitBits(wifi_event_group, bits, false, true, portMAX_DELAY);
|
|
|
- ESP_LOGI(TAG, "Connected to AP");
|
|
|
-}
|
|
|
|
|
|
static void tcp_server_task(void *pvParameters)
|
|
|
{
|
|
|
@@ -113,21 +39,21 @@ static void tcp_server_task(void *pvParameters)
|
|
|
while (1) {
|
|
|
|
|
|
#ifdef CONFIG_EXAMPLE_IPV4
|
|
|
- struct sockaddr_in destAddr;
|
|
|
- destAddr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
|
- destAddr.sin_family = AF_INET;
|
|
|
- destAddr.sin_port = htons(PORT);
|
|
|
+ struct sockaddr_in dest_addr;
|
|
|
+ dest_addr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
|
+ dest_addr.sin_family = AF_INET;
|
|
|
+ dest_addr.sin_port = htons(PORT);
|
|
|
addr_family = AF_INET;
|
|
|
ip_protocol = IPPROTO_IP;
|
|
|
- inet_ntoa_r(destAddr.sin_addr, addr_str, sizeof(addr_str) - 1);
|
|
|
+ inet_ntoa_r(dest_addr.sin_addr, addr_str, sizeof(addr_str) - 1);
|
|
|
#else // IPV6
|
|
|
- struct sockaddr_in6 destAddr;
|
|
|
- bzero(&destAddr.sin6_addr.un, sizeof(destAddr.sin6_addr.un));
|
|
|
- destAddr.sin6_family = AF_INET6;
|
|
|
- destAddr.sin6_port = htons(PORT);
|
|
|
+ struct sockaddr_in6 dest_addr;
|
|
|
+ bzero(&dest_addr.sin6_addr.un, sizeof(dest_addr.sin6_addr.un));
|
|
|
+ dest_addr.sin6_family = AF_INET6;
|
|
|
+ dest_addr.sin6_port = htons(PORT);
|
|
|
addr_family = AF_INET6;
|
|
|
ip_protocol = IPPROTO_IPV6;
|
|
|
- inet6_ntoa_r(destAddr.sin6_addr, addr_str, sizeof(addr_str) - 1);
|
|
|
+ inet6_ntoa_r(dest_addr.sin6_addr, addr_str, sizeof(addr_str) - 1);
|
|
|
#endif
|
|
|
|
|
|
int listen_sock = socket(addr_family, SOCK_STREAM, ip_protocol);
|
|
|
@@ -137,23 +63,23 @@ static void tcp_server_task(void *pvParameters)
|
|
|
}
|
|
|
ESP_LOGI(TAG, "Socket created");
|
|
|
|
|
|
- int err = bind(listen_sock, (struct sockaddr *)&destAddr, sizeof(destAddr));
|
|
|
+ int err = bind(listen_sock, (struct sockaddr *)&dest_addr, sizeof(dest_addr));
|
|
|
if (err != 0) {
|
|
|
ESP_LOGE(TAG, "Socket unable to bind: errno %d", errno);
|
|
|
break;
|
|
|
}
|
|
|
- ESP_LOGI(TAG, "Socket binded");
|
|
|
+ ESP_LOGI(TAG, "Socket bound, port %d", PORT);
|
|
|
|
|
|
err = listen(listen_sock, 1);
|
|
|
if (err != 0) {
|
|
|
- ESP_LOGE(TAG, "Error occured during listen: errno %d", errno);
|
|
|
+ ESP_LOGE(TAG, "Error occurred during listen: errno %d", errno);
|
|
|
break;
|
|
|
}
|
|
|
ESP_LOGI(TAG, "Socket listening");
|
|
|
|
|
|
- struct sockaddr_in6 sourceAddr; // Large enough for both IPv4 or IPv6
|
|
|
- uint addrLen = sizeof(sourceAddr);
|
|
|
- int sock = accept(listen_sock, (struct sockaddr *)&sourceAddr, &addrLen);
|
|
|
+ struct sockaddr_in6 source_addr; // Large enough for both IPv4 or IPv6
|
|
|
+ uint addr_len = sizeof(source_addr);
|
|
|
+ int sock = accept(listen_sock, (struct sockaddr *)&source_addr, &addr_len);
|
|
|
if (sock < 0) {
|
|
|
ESP_LOGE(TAG, "Unable to accept connection: errno %d", errno);
|
|
|
break;
|
|
|
@@ -162,7 +88,7 @@ static void tcp_server_task(void *pvParameters)
|
|
|
|
|
|
while (1) {
|
|
|
int len = recv(sock, rx_buffer, sizeof(rx_buffer) - 1, 0);
|
|
|
- // Error occured during receiving
|
|
|
+ // Error occurred during receiving
|
|
|
if (len < 0) {
|
|
|
ESP_LOGE(TAG, "recv failed: errno %d", errno);
|
|
|
break;
|
|
|
@@ -175,10 +101,10 @@ static void tcp_server_task(void *pvParameters)
|
|
|
// Data received
|
|
|
else {
|
|
|
// Get the sender's ip address as string
|
|
|
- if (sourceAddr.sin6_family == PF_INET) {
|
|
|
- inet_ntoa_r(((struct sockaddr_in *)&sourceAddr)->sin_addr.s_addr, addr_str, sizeof(addr_str) - 1);
|
|
|
- } else if (sourceAddr.sin6_family == PF_INET6) {
|
|
|
- inet6_ntoa_r(sourceAddr.sin6_addr, addr_str, sizeof(addr_str) - 1);
|
|
|
+ if (source_addr.sin6_family == PF_INET) {
|
|
|
+ inet_ntoa_r(((struct sockaddr_in *)&source_addr)->sin_addr.s_addr, addr_str, sizeof(addr_str) - 1);
|
|
|
+ } else if (source_addr.sin6_family == PF_INET6) {
|
|
|
+ inet6_ntoa_r(source_addr.sin6_addr, addr_str, sizeof(addr_str) - 1);
|
|
|
}
|
|
|
|
|
|
rx_buffer[len] = 0; // Null-terminate whatever we received and treat like a string
|
|
|
@@ -187,7 +113,7 @@ static void tcp_server_task(void *pvParameters)
|
|
|
|
|
|
int err = send(sock, rx_buffer, len, 0);
|
|
|
if (err < 0) {
|
|
|
- ESP_LOGE(TAG, "Error occured during sending: errno %d", errno);
|
|
|
+ ESP_LOGE(TAG, "Error occurred during sending: errno %d", errno);
|
|
|
break;
|
|
|
}
|
|
|
}
|
|
|
@@ -204,9 +130,15 @@ static void tcp_server_task(void *pvParameters)
|
|
|
|
|
|
void app_main()
|
|
|
{
|
|
|
- ESP_ERROR_CHECK( nvs_flash_init() );
|
|
|
- initialise_wifi();
|
|
|
- wait_for_ip();
|
|
|
+ ESP_ERROR_CHECK(nvs_flash_init());
|
|
|
+ tcpip_adapter_init();
|
|
|
+ ESP_ERROR_CHECK(esp_event_loop_create_default());
|
|
|
+
|
|
|
+ /* This helper function configures Wi-Fi or Ethernet, as selected in menuconfig.
|
|
|
+ * Read "Establishing Wi-Fi or Ethernet Connection" section in
|
|
|
+ * examples/protocols/README.md for more information about this function.
|
|
|
+ */
|
|
|
+ ESP_ERROR_CHECK(example_connect());
|
|
|
|
|
|
xTaskCreate(tcp_server_task, "tcp_server", 4096, NULL, 5, NULL);
|
|
|
}
|