Inial commit
This commit is contained in:
commit
bb9002b03e
6 changed files with 837 additions and 0 deletions
1
main/CMakeLists.txt
Normal file
1
main/CMakeLists.txt
Normal file
|
|
@ -0,0 +1 @@
|
|||
idf_component_register(SRCS main.c wifi.c INCLUDE_DIRS .)
|
||||
208
main/main.c
Normal file
208
main/main.c
Normal file
|
|
@ -0,0 +1,208 @@
|
|||
#include <freertos/FreeRTOS.h>
|
||||
#include <freertos/task.h>
|
||||
#include <freertos/event_groups.h>
|
||||
#include <esp_system.h>
|
||||
#include <esp_timer.h>
|
||||
#include <esp_log.h>
|
||||
#include <driver/gpio.h>
|
||||
|
||||
#include <lwip/err.h>
|
||||
#include <lwip/sys.h>
|
||||
#include <lwip/sockets.h>
|
||||
|
||||
#include "wifi.h"
|
||||
|
||||
#define HOST_IP_ADDR "10.0.0.1"
|
||||
#define HOST_UDP_PORT 40586
|
||||
|
||||
#define RI_PIN 5
|
||||
#define RI_LONG_TIME 3000
|
||||
#define RI_SHORT_TIME 1000
|
||||
|
||||
static const char *UDP_TAG = "udp";
|
||||
|
||||
static xQueueHandle ri_gpio_evt_queue;
|
||||
static xQueueHandle ri_message_queue;
|
||||
|
||||
static void udp_client_task(void *pvParameters)
|
||||
{
|
||||
char tx_buffer[64];
|
||||
char addr_str[128];
|
||||
int addr_family;
|
||||
int ip_protocol;
|
||||
|
||||
while(true)
|
||||
{
|
||||
struct sockaddr_in destAddr;
|
||||
destAddr.sin_addr.s_addr = inet_addr(HOST_IP_ADDR);
|
||||
destAddr.sin_family = AF_INET;
|
||||
destAddr.sin_port = htons(HOST_UDP_PORT);
|
||||
addr_family = AF_INET;
|
||||
ip_protocol = IPPROTO_IP;
|
||||
inet_ntoa_r(destAddr.sin_addr, addr_str, sizeof(addr_str) - 1);
|
||||
|
||||
int sock = socket(addr_family, SOCK_DGRAM, ip_protocol);
|
||||
if (sock < 0)
|
||||
{
|
||||
ESP_LOGE(UDP_TAG, "Unable to create socket: errno %d", errno);
|
||||
break;
|
||||
}
|
||||
ESP_LOGI(UDP_TAG, "Socket created");
|
||||
|
||||
while(true)
|
||||
{
|
||||
uint16_t ri_message;
|
||||
if(xQueueReceive(ri_message_queue, &ri_message, portMAX_DELAY))
|
||||
{
|
||||
snprintf(tx_buffer, sizeof(tx_buffer), "RI_MESSAGE: %u\n", ri_message);
|
||||
int err = sendto(sock, tx_buffer, strlen(tx_buffer), 0, (struct sockaddr *)&destAddr, sizeof(destAddr));
|
||||
if (err < 0)
|
||||
{
|
||||
ESP_LOGE(UDP_TAG, "Error occured during sending: errno %d", errno);
|
||||
break;
|
||||
}
|
||||
ESP_LOGI(UDP_TAG, "Message sent");
|
||||
gpio_set_level(2, true);
|
||||
}
|
||||
}
|
||||
|
||||
if(sock != -1)
|
||||
{
|
||||
ESP_LOGE(UDP_TAG, "Shutting down socket and restarting...");
|
||||
shutdown(sock, 0);
|
||||
close(sock);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
struct ri_gpio_event
|
||||
{
|
||||
bool level;
|
||||
uint64_t time;
|
||||
};
|
||||
|
||||
void ri_gpio_isr_handler(void* data)
|
||||
{
|
||||
struct ri_gpio_event event;
|
||||
event.level = gpio_get_level(RI_PIN);
|
||||
event.time = esp_timer_get_time();
|
||||
xQueueSendFromISR(ri_gpio_evt_queue, &event, NULL);
|
||||
}
|
||||
|
||||
void ri_reciver_setup(void)
|
||||
{
|
||||
gpio_config_t config;
|
||||
|
||||
config.intr_type = GPIO_INTR_ANYEDGE;
|
||||
config.mode = GPIO_MODE_INPUT;
|
||||
config.pin_bit_mask = 1 << RI_PIN;
|
||||
config.pull_down_en = false;
|
||||
config.pull_up_en = false;
|
||||
gpio_config(&config);
|
||||
|
||||
ri_gpio_evt_queue = xQueueCreate(10, sizeof(struct ri_gpio_event));
|
||||
ri_message_queue = xQueueCreate(10, sizeof(uint16_t));
|
||||
|
||||
gpio_install_isr_service(0);
|
||||
gpio_isr_handler_add(RI_PIN, ri_gpio_isr_handler, NULL);
|
||||
}
|
||||
|
||||
typedef enum
|
||||
{
|
||||
RI_STATE_SEARCH,
|
||||
RI_STATE_WAIT_SYNC_PULSE,
|
||||
RI_STATE_BIT_START_WAIT,
|
||||
RI_STATE_BIT_START,
|
||||
RI_STATE_BIT_DATA,
|
||||
} ri_state_t;
|
||||
|
||||
bool time_matches(uint64_t time, uint64_t referance, uint32_t tollerance)
|
||||
{
|
||||
return time + tollerance > referance && time - tollerance < referance;
|
||||
}
|
||||
|
||||
void ri_reciver_task(void *pvParameters)
|
||||
{
|
||||
(void)pvParameters;
|
||||
|
||||
ri_state_t state = RI_STATE_SEARCH;
|
||||
unsigned int bit = 0;
|
||||
uint64_t prev_time = 0;
|
||||
uint16_t data;
|
||||
|
||||
while(true)
|
||||
{
|
||||
struct ri_gpio_event event;
|
||||
if(xQueueReceive(ri_gpio_evt_queue, &event, portMAX_DELAY))
|
||||
{
|
||||
if(state == RI_STATE_SEARCH && event.level)
|
||||
{
|
||||
state = RI_STATE_WAIT_SYNC_PULSE;
|
||||
bit = 0;
|
||||
data = 0;
|
||||
}
|
||||
else if(state == RI_STATE_WAIT_SYNC_PULSE && !event.level)
|
||||
{
|
||||
uint64_t elapsed = event.time - prev_time;
|
||||
if(!time_matches(elapsed, RI_LONG_TIME, 500))
|
||||
state = RI_STATE_SEARCH;
|
||||
else
|
||||
state = RI_STATE_BIT_START_WAIT;
|
||||
}
|
||||
else if(state == RI_STATE_BIT_START_WAIT && event.level)
|
||||
{
|
||||
uint64_t elapsed = event.time - prev_time;
|
||||
if(!time_matches(elapsed, RI_SHORT_TIME, 150))
|
||||
state = RI_STATE_SEARCH;
|
||||
else
|
||||
state = RI_STATE_BIT_START;
|
||||
}
|
||||
else if(state == RI_STATE_BIT_START && !event.level)
|
||||
{
|
||||
uint64_t elapsed = event.time - prev_time;
|
||||
if(!time_matches(elapsed, RI_SHORT_TIME, 150))
|
||||
state = RI_STATE_SEARCH;
|
||||
else
|
||||
state = RI_STATE_BIT_DATA;
|
||||
}
|
||||
else if(state == RI_STATE_BIT_DATA && event.level)
|
||||
{
|
||||
uint64_t elapsed = event.time - prev_time;
|
||||
if(elapsed > RI_SHORT_TIME + 500)
|
||||
data |= 1 << bit;
|
||||
++bit;
|
||||
if(bit >= 12)
|
||||
{
|
||||
gpio_set_level(2, false);
|
||||
xQueueSendToBack(ri_message_queue, &data, portMAX_DELAY);
|
||||
state = RI_STATE_SEARCH;
|
||||
}
|
||||
else
|
||||
{
|
||||
state = RI_STATE_BIT_START;
|
||||
}
|
||||
}
|
||||
prev_time = event.time;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void app_main()
|
||||
{
|
||||
gpio_config_t config = {};
|
||||
config.intr_type = GPIO_INTR_DISABLE;
|
||||
config.mode = GPIO_MODE_OUTPUT;
|
||||
config.pin_bit_mask = 1 << 2;
|
||||
gpio_config(&config);
|
||||
gpio_set_level(2, false);
|
||||
|
||||
wifi_init_sta();
|
||||
|
||||
ri_reciver_setup();
|
||||
|
||||
xTaskCreate(ri_reciver_task, "ri_reciver_task", 2048, NULL, 10, NULL);
|
||||
|
||||
xTaskCreate(udp_client_task, "udp_client", 4096, NULL, 5, NULL);
|
||||
|
||||
gpio_set_level(2, true);
|
||||
}
|
||||
93
main/wifi.c
Normal file
93
main/wifi.c
Normal file
|
|
@ -0,0 +1,93 @@
|
|||
#include <esp_wifi.h>
|
||||
#include <freertos/event_groups.h>
|
||||
#include <esp_log.h>
|
||||
|
||||
#define WIFI_SSID "UvosServerNet24"
|
||||
#define WIFI_PASS "A159753654F"
|
||||
#define MAXIMUM_RETRY 8
|
||||
|
||||
static const char *WIFI_TAG = "wifi station";
|
||||
|
||||
static EventGroupHandle_t s_wifi_event_group;
|
||||
|
||||
#define WIFI_CONNECTED_BIT BIT0
|
||||
#define WIFI_FAIL_BIT BIT1
|
||||
|
||||
static int s_retry_num = 0;
|
||||
|
||||
static void net_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_START)
|
||||
{
|
||||
esp_wifi_connect();
|
||||
}
|
||||
else if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_DISCONNECTED)
|
||||
{
|
||||
if (s_retry_num < MAXIMUM_RETRY)
|
||||
{
|
||||
esp_wifi_connect();
|
||||
s_retry_num++;
|
||||
ESP_LOGI(WIFI_TAG, "retry to connect to the AP");
|
||||
}
|
||||
else
|
||||
{
|
||||
xEventGroupSetBits(s_wifi_event_group, WIFI_FAIL_BIT);
|
||||
}
|
||||
ESP_LOGI(WIFI_TAG,"connect to the AP fail");
|
||||
}
|
||||
else if (event_base == IP_EVENT && event_id == IP_EVENT_STA_GOT_IP)
|
||||
{
|
||||
ip_event_got_ip_t* event = (ip_event_got_ip_t*) event_data;
|
||||
ESP_LOGI(WIFI_TAG, "got ip: %s",
|
||||
ip4addr_ntoa(&event->ip_info.ip));
|
||||
s_retry_num = 0;
|
||||
xEventGroupSetBits(s_wifi_event_group, WIFI_CONNECTED_BIT);
|
||||
}
|
||||
}
|
||||
|
||||
void wifi_init_sta(void)
|
||||
{
|
||||
s_wifi_event_group = xEventGroupCreate();
|
||||
|
||||
tcpip_adapter_init();
|
||||
|
||||
ESP_ERROR_CHECK(esp_event_loop_create_default());
|
||||
|
||||
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
||||
ESP_ERROR_CHECK(esp_wifi_init(&cfg));
|
||||
|
||||
ESP_ERROR_CHECK(esp_event_handler_register(WIFI_EVENT, ESP_EVENT_ANY_ID, &net_event_handler, NULL));
|
||||
ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_STA_GOT_IP, &net_event_handler, NULL));
|
||||
|
||||
wifi_config_t wifi_config =
|
||||
{
|
||||
.sta = {
|
||||
.ssid = WIFI_SSID,
|
||||
.password = WIFI_PASS,
|
||||
.threshold.authmode = WIFI_AUTH_WPA2_PSK
|
||||
},
|
||||
};
|
||||
|
||||
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_LOGI(WIFI_TAG, "wifi_init_sta finished.");
|
||||
|
||||
EventBits_t bits = xEventGroupWaitBits(s_wifi_event_group,
|
||||
WIFI_CONNECTED_BIT | WIFI_FAIL_BIT,
|
||||
pdFALSE,
|
||||
pdFALSE,
|
||||
portMAX_DELAY);
|
||||
|
||||
if(bits & WIFI_CONNECTED_BIT)
|
||||
ESP_LOGI(WIFI_TAG, "connected to ap");
|
||||
else if(bits & WIFI_FAIL_BIT)
|
||||
ESP_LOGI(WIFI_TAG, "Failed to connect to ap");
|
||||
else
|
||||
ESP_LOGE(WIFI_TAG, "UNEXPECTED EVENT");
|
||||
|
||||
ESP_ERROR_CHECK(esp_event_handler_unregister(IP_EVENT, IP_EVENT_STA_GOT_IP, &net_event_handler));
|
||||
ESP_ERROR_CHECK(esp_event_handler_unregister(WIFI_EVENT, ESP_EVENT_ANY_ID, &net_event_handler));
|
||||
vEventGroupDelete(s_wifi_event_group);
|
||||
}
|
||||
3
main/wifi.h
Normal file
3
main/wifi.h
Normal file
|
|
@ -0,0 +1,3 @@
|
|||
#pragma once
|
||||
|
||||
void wifi_init_sta(void);
|
||||
Loading…
Add table
Add a link
Reference in a new issue