Skip to content

Commit

Permalink
Add esp32-mqtt
Browse files Browse the repository at this point in the history
  • Loading branch information
cpq committed Dec 30, 2024
1 parent 5479918 commit 6cf36b5
Show file tree
Hide file tree
Showing 4 changed files with 107 additions and 0 deletions.
102 changes: 102 additions & 0 deletions examples/arduino/esp32-mqtt/esp32-mqtt.ino
Original file line number Diff line number Diff line change
@@ -0,0 +1,102 @@
#include "WiFi.h"
#include "mongoose.h"

#define WIFI_SSID "wifi_network_name" // Change this
#define WIFI_PASS "wifi_password" // And this

#define MQTT_SERVER "mqtt://broker.hivemq.com:1883"
#define MQTT_SUB_TOPIC "mg/rx" // Subscribe to this topic
#define MQTT_PUB_TOPIC "mg/tx" // Publish to this topic

#define LED_PIN 0

struct mg_mgr mgr;
struct mg_connection *mqtt_connection;

void mqtt_publish(const char *message) {
struct mg_mqtt_opts opts = {};
opts.topic = mg_str(MQTT_PUB_TOPIC);
opts.message = mg_str(message);
if (mqtt_connection) mg_mqtt_pub(mqtt_connection, &opts);
}

// Crude function to get available RAM, for quick profiling
size_t getFreeRAM(void) {
size_t size = 0, increment = 100;
void *p;
while ((p = malloc(size)) != NULL) free(p), size += increment;
return size;
}

// Implement LED control over MQTT: "on" and "off" commands
void handle_command(struct mg_str msg) {
if (msg.len == 3 && memcmp(msg.buf, "off", 3) == 0) {
digitalWrite(LED_PIN, LOW);
mqtt_publish("done - off");
} else if (msg.len == 2 && memcmp(msg.buf, "on", 2) == 0) {
digitalWrite(LED_PIN, HIGH);
mqtt_publish("done - on");
}
MG_INFO(("Free RAM: %u bytes", getFreeRAM()));
}

static void mqtt_ev_handler(struct mg_connection *c, int ev, void *ev_data) {
if (ev == MG_EV_MQTT_OPEN) {
MG_INFO(("%lu CONNECTED to %s", c->id, MQTT_SERVER));
struct mg_mqtt_opts opts = {};
opts.topic = mg_str(MQTT_SUB_TOPIC);
mg_mqtt_sub(c, &opts);
MG_INFO(("%lu SUBSCRIBED to %s", c->id, MQTT_SUB_TOPIC));
} else if (ev == MG_EV_MQTT_MSG) {
// Received MQTT message
struct mg_mqtt_message *mm = (struct mg_mqtt_message *) ev_data;
MG_INFO(("%lu RECEIVED %.*s <- %.*s", c->id, (int) mm->data.len,
mm->data.buf, (int) mm->topic.len, mm->topic.buf));
handle_command(mm->data);
} else if (ev == MG_EV_CLOSE) {
MG_INFO(("%lu CLOSED", c->id));
mqtt_connection = NULL;
}
}

void reconnect_if_not_connected(void) {
if (mqtt_connection == NULL) {
struct mg_mqtt_opts opts = {};
opts.clean = true;
mqtt_connection =
mg_mqtt_connect(&mgr, MQTT_SERVER, &opts, mqtt_ev_handler, NULL);
}
}

void setup() {
pinMode(LED_PIN, OUTPUT);
Serial.begin(115200);
while (!Serial) delay(50);

WiFi.mode(WIFI_STA);
WiFi.begin(WIFI_SSID, WIFI_PASS);
while (WiFi.status() != WL_CONNECTED) Serial.print("."), delay(100);

Serial.println("\nConnected to the WiFi network");
Serial.print("IP Address: ");
Serial.println(WiFi.localIP());

mg_mgr_init(&mgr);
mg_log_set(MG_LL_DEBUG);
mg_log_set_fn([](char ch, void *) { Serial.print(ch); }, NULL);
}

void loop() {
mg_mgr_poll(&mgr, 1); // Process network events
reconnect_if_not_connected(); // Reconnect to MQTT server if needed
}

extern "C" int lwip_hook_ip6_input(struct pbuf *p, struct netif *inp)
__attribute__((weak));
extern "C" int lwip_hook_ip6_input(struct pbuf *p, struct netif *inp) {
if (ip6_addr_isany_val(inp->ip6_addr[0].u_addr.ip6)) {
pbuf_free(p);
return 1;
}
return 0;
}
1 change: 1 addition & 0 deletions examples/arduino/esp32-mqtt/mongoose.c
1 change: 1 addition & 0 deletions examples/arduino/esp32-mqtt/mongoose.h
3 changes: 3 additions & 0 deletions examples/arduino/esp32-mqtt/mongoose_config.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
#pragma once

#define MG_ARCH MG_ARCH_ESP32

0 comments on commit 6cf36b5

Please sign in to comment.