summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat
-rw-r--r--main/config.h9
-rw-r--r--main/network.c155
-rw-r--r--main/network.h2
-rw-r--r--main/sprayduck.c12
4 files changed, 165 insertions, 13 deletions
diff --git a/main/config.h b/main/config.h
new file mode 100644
index 0000000..29bae6b
--- /dev/null
+++ b/main/config.h
@@ -0,0 +1,9 @@
+#define THRESHOLD 2000
+#define FLIP_GPIO_PIN 23
+#define SENSOR_GPIO_PIN 22
+#define SENSOR_CHANNEL ADC_CHANNEL_7
+#define SSID "SSID"
+#define PASSWORD "PASSWORD"
+#define SERVER_IP "127.0.0.1"
+#define SERVER_PORT "80"
+
diff --git a/main/network.c b/main/network.c
index d45ca2f..b266654 100644
--- a/main/network.c
+++ b/main/network.c
@@ -1,23 +1,164 @@
#include "network.h"
+#include "esp_log.h"
#include "esp_log_level.h"
+#include "esp_netif.h"
+#include "esp_netif_ip_addr.h"
+#include "esp_netif_types.h"
+#include "esp_wifi_default.h"
#include "esp_wifi_types_generic.h"
+#include "config.h"
+#include "freertos/idf_additions.h"
+#include "freertos/projdefs.h"
+#include <netdb.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <string.h>
+#define TAG "sprayduck"
+static EventGroupHandle_t s_wifi_event_group;
+static esp_netif_ip_info_t* ip_address = NULL;
+
+
+
+void wifi_event_handler(void* handler_arg, esp_event_base_t base, int32_t id, void* event_data) {
+ if (base != WIFI_EVENT) {
+ return;
+ }
+ if (id == WIFI_EVENT_STA_START) {
+ // If station mode has started successfully, attempt a connection
+ ESP_LOGI(TAG, "Connecting...");
+ ESP_ERROR_CHECK(esp_wifi_connect());
+ ESP_LOGI(TAG, "CONNECTED...");
+ } else if (id == WIFI_EVENT_STA_DISCONNECTED) {
+ // If we disconnected, attempt a reconnect
+ ESP_LOGI(TAG, "Attempting reconnection...");
+ ESP_ERROR_CHECK(esp_wifi_connect());
+ // Mark connection as not ready
+ xEventGroupClearBits(s_wifi_event_group, BIT0);
+ ip_address = NULL;
+ }
+
+}
+
+void ip_event_handler(void* handler_arg, esp_event_base_t base, int32_t id, void* event_data) {
+ if (base != IP_EVENT) {
+ return;
+ }
+ if (id == IP_EVENT_STA_GOT_IP) {
+ // Got IP
+ ip_event_got_ip_t* event = (ip_event_got_ip_t*) event_data;
+ ESP_LOGI(TAG, "got ip:" IPSTR, IP2STR(&event->ip_info.ip));
+ // Mark connection as ready
+ xEventGroupSetBits(s_wifi_event_group, BIT0);
+ ip_address = &event->ip_info;
+ }
+
+}
void setup_network() {
+ // Event group to be notified when ready to connect
+ s_wifi_event_group = xEventGroupCreate();
+ xEventGroupClearBits(s_wifi_event_group, BIT0);
+
+ // Set logging level to verbose
esp_log_level_set("sprayduck", ESP_LOG_VERBOSE);
+
+ // Setup TCP/IP stack - create event loop which registers events from TC/IP
+ // stack
+ esp_netif_init();
+ // We must create the event loop before anything because the drivers need to
+ // register system handlers, which require the event loop
+ esp_event_loop_create_default();
+
+ // Setup Wi-fi driver, configured in Station (client) mode This step involves
+ // attaching system handlers to the above created default event loop. This
+ // must be done before attaching any user event handlers because this has
+ // higer priority
+ esp_netif_create_default_wifi_sta();
+
+ // Configuration for how the wifi will be setup
wifi_init_config_t config = WIFI_INIT_CONFIG_DEFAULT();
+ // Disable nvs memory, not needed
config.nvs_enable = 0;
-
+ // Initalize above declared wifi configuration This initalizes the wifi
+ // driver, and starts the driver task
ESP_ERROR_CHECK(esp_wifi_init(&config));
- esp_wifi_set_mode(WIFI_MODE_STA);
-
+ //Set to station mode
+ ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA));
+ // Connect to AP with provided SSID and PASSWORD
wifi_sta_config_t sta_config = {
- .ssid = "SSID",
- .password = "PASSWORD"
+ .ssid = SSID,
+ .password = PASSWORD,
};
+ // Apply above define configuration
ESP_ERROR_CHECK(esp_wifi_set_config(WIFI_IF_STA, (wifi_config_t*) &sta_config));
-
+ esp_event_handler_instance_t instance_any_id;
+ esp_event_handler_instance_t instance_got_ip;
+ // All Wifi related events are handled here
+ ESP_ERROR_CHECK(esp_event_handler_instance_register(WIFI_EVENT,
+ ESP_EVENT_ANY_ID,
+ &wifi_event_handler,
+ NULL,
+ &instance_any_id));
+ // All IP related events are handled here
+ ESP_ERROR_CHECK(esp_event_handler_instance_register(IP_EVENT,
+ IP_EVENT_STA_GOT_IP,
+ &ip_event_handler,
+ NULL,
+ &instance_got_ip));
+ // Start the client
ESP_ERROR_CHECK(esp_wifi_start());
- ESP_ERROR_CHECK(esp_wifi_connect());
+
+
+}
+
+void send_packet(char* data) {
+ ESP_LOGI(TAG, "Sending packet");
+ EventBits_t bits = xEventGroupGetBits(s_wifi_event_group);
+ // We are connected and have a valid ip_address
+ if (bits & BIT0 && ip_address != NULL) {
+ ESP_LOGI(TAG, "Have IP, Sending packet");
+ // char ip[20];
+ // sprintf(ip, IPSTR, IP2STR(&ip_address->ip));
+ int sock;
+ struct addrinfo hints, *p, *res;
+
+ memset(&hints, 0, sizeof hints);
+ hints.ai_family = AF_INET;
+ hints.ai_socktype = SOCK_STREAM;
+
+ if (getaddrinfo(SERVER_IP, SERVER_PORT, &hints, &res) != 0) {
+ return;
+ }
+
+ if ((sock = socket(res->ai_family, res->ai_socktype, res->ai_protocol)) ==
+ -1) {
+ fprintf(stderr, "%s: socket (%d): %s\n", TAG, errno, strerror(errno));
+ return ;
+ }
+
+ if (connect(sock, res->ai_addr, res->ai_addrlen) == -1) {
+ fprintf(stderr, "%s: connect (%d): %s\n", TAG, errno, strerror(errno));
+ return ;
+ }
+
+ int len = strlen(data);
+ int bytes_sent = 0;
+ int total_bytes_sent = 0;
+
+ while (1) {
+ if ((bytes_sent = send(sock, data, len, 0)) == -1) {
+ fprintf(stderr, "%s: send (%d): %s\n", TAG, errno, strerror(errno));
+ return ;
+ }
+ total_bytes_sent += bytes_sent;
+ if (bytes_sent == len) {
+ break;
+ }
+ }
+ close(sock);
+ freeaddrinfo(res);
+ return ;
+ }
}
diff --git a/main/network.h b/main/network.h
index f14569f..ce5cd12 100644
--- a/main/network.h
+++ b/main/network.h
@@ -4,4 +4,4 @@
void setup_network();
-
+void send_packet(char* data);
diff --git a/main/sprayduck.c b/main/sprayduck.c
index 60935b1..e1ba3ae 100644
--- a/main/sprayduck.c
+++ b/main/sprayduck.c
@@ -4,13 +4,11 @@
#include "esp_log.h"
#include "hal/gpio_types.h"
#include <driver/gpio.h>
+#include <stdio.h>
#include "hal/adc_types.h"
#include "esp_adc/adc_oneshot.h"
+#include "config.h"
-#define THRESHOLD 2000
-#define FLIP_GPIO_PIN 23
-#define SENSOR_GPIO_PIN 22
-#define SENSOR_CHANNEL ADC_CHANNEL_7
#define TAG "sprayduck"
@@ -63,7 +61,7 @@ void app_main(void)
while (1) {
//Turn on sensor
- // gpio_set_level((gpio_num_t) SENSOR_GPIO_PIN, 1);
+ gpio_set_level((gpio_num_t) SENSOR_GPIO_PIN, 1);
int value = read_sensor(handler);
ESP_LOGI("sprayduck", "Checking sensor value average %d", value);
if (value > THRESHOLD) {
@@ -76,6 +74,10 @@ void app_main(void)
}
// Turn off Sensor
gpio_set_level((gpio_num_t) SENSOR_GPIO_PIN, 0);
+
+ char data[4];
+ sprintf(data, "%d", value);
+ send_packet(data);
vTaskDelay(1000 / portTICK_PERIOD_MS);
}
generated by cgit (git 2.34.1) at 2025-09-03 17:17:55 +0000