-
Notifications
You must be signed in to change notification settings - Fork 2.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
4 changed files
with
119 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
../../../mongoose.c |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
../../../mongoose.h |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,15 @@ | ||
#pragma once | ||
|
||
#include "Arduino.h" | ||
|
||
#include <errno.h> | ||
#include <stdarg.h> | ||
#include <stdio.h> | ||
#include <time.h> | ||
|
||
#define MG_ARCH MG_ARCH_CUSTOM | ||
#define MG_ENABLE_SOCKET 0 | ||
#define MG_ENABLE_TCPIP 1 | ||
#define mkdir(a, b) (-1) | ||
#define MG_IO_SIZE 128 | ||
//#define MG_ENABLE_LOG 0 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,102 @@ | ||
#include <SPI.h> | ||
#include "mongoose.h" | ||
|
||
#define MQTT_SERVER "mqtt://broker.hivemq.com:1883" | ||
struct mg_connection *mqtt_connection; | ||
const char *mqtt_subscribe_topic = "mg/rx"; | ||
const char *mqtt_publish_topic = "mg/tx"; | ||
|
||
#define SS_PIN 17 // Slave select pin | ||
struct mg_mgr mgr; // Mongoose event manager | ||
struct mg_tcpip_spi spi = { | ||
NULL, // SPI data | ||
[](void *) { digitalWrite(SS_PIN, LOW); }, // Begin transaction | ||
[](void *) { digitalWrite(SS_PIN, HIGH); }, // End transaction | ||
[](void *, uint8_t c) { return SPI.transfer(c); }, // Execute transaction | ||
}; | ||
struct mg_tcpip_if mif = {.mac = {2, 0, 1, 2, 3, 5}}; // network interface | ||
|
||
static void mqtt_publish(const char *message) { | ||
struct mg_mqtt_opts opts = {}; | ||
opts.topic = mg_str(mqtt_publish_topic); | ||
opts.message = mg_str(message); | ||
if (mqtt_connection) mg_mqtt_pub(mqtt_connection, &opts); | ||
} | ||
|
||
static void exec_command(const char *req, size_t req_len) { | ||
char res[100]; | ||
if (req_len == 2 && strncmp(req, "on", req_len) == 0) { | ||
digitalWrite(LED_BUILTIN, true); | ||
snprintf(res, sizeof(res), "LED on"); | ||
} else if (req_len == 3 && strncmp(req, "off", req_len) == 0) { | ||
digitalWrite(LED_BUILTIN, false); | ||
snprintf(res, sizeof(res), "LED off"); | ||
} else { | ||
snprintf(res, sizeof(res), "Unknown command: [%.*s]", (int) req_len, req); | ||
} | ||
Serial.println(res); | ||
mqtt_publish(res); | ||
} | ||
|
||
static void process_input(char c) { | ||
static char buf[100]; | ||
static size_t len = 0; | ||
char response[100]; | ||
if (c != '\n' && c != '\0') buf[len++] = c; // Append to the buffer | ||
if (len >= sizeof(buf)) len = 0; // On overflow, reset | ||
if (c == '\n' && len > 0) { | ||
exec_command(buf, len); | ||
len = 0; | ||
} | ||
} | ||
|
||
static void fn(struct mg_connection *c, int ev, void *ev_data, void *fn_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_subscribe_topic); | ||
mg_mqtt_sub(c, &opts); | ||
MG_INFO(("%lu SUBSCRIBED to %s", c->id, mqtt_subscribe_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.ptr, (int) mm->topic.len, mm->topic.ptr)); | ||
exec_command(mm->data.ptr, mm->data.len); | ||
} else if (ev == MG_EV_CLOSE) { | ||
MG_INFO(("%lu CLOSED", c->id)); | ||
mqtt_connection = NULL; | ||
} | ||
} | ||
|
||
static void timer_fn(void *arg) { | ||
struct mg_mgr *mgr = (struct mg_mgr *) arg; | ||
struct mg_mqtt_opts opts = {.clean = true}; | ||
if (mif.state == MG_TCPIP_STATE_READY && mqtt_connection == NULL) { | ||
mqtt_connection = mg_mqtt_connect(mgr, MQTT_SERVER, &opts, fn, NULL); | ||
} | ||
MG_INFO(("Eth: %d, MQTT: %p", mif.state, mqtt_connection)); | ||
} | ||
|
||
void setup() { | ||
Serial.begin(115200); | ||
SPI.begin(); | ||
pinMode(SS_PIN, OUTPUT); | ||
pinMode(LED_BUILTIN, OUTPUT); | ||
|
||
// Set Mongoose logging function to a serial print | ||
mg_log_set_fn([](char ch, void *) { Serial.print(ch); }, NULL); | ||
mg_mgr_init(&mgr); | ||
|
||
mif.driver = &mg_tcpip_driver_w5500; | ||
mif.driver_data = &spi; | ||
mg_tcpip_init(&mgr, &mif); | ||
|
||
// Start a timer that keeps MQTT connection alive | ||
mg_timer_add(&mgr, 3000, MG_TIMER_REPEAT, timer_fn, &mgr); | ||
} | ||
|
||
void loop() { | ||
if (Serial.available()) process_input(Serial.read()); | ||
mg_mgr_poll(&mgr, 1); | ||
} |