Compare commits
82 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4746abfc96 | ||
|
|
ef24107169 | ||
|
|
295e1a06d3 | ||
|
|
1a66d2c991 | ||
|
|
8162b4c35b | ||
|
|
9e578471f1 | ||
|
|
efa94cc4a4 | ||
|
|
040568b478 | ||
|
|
a96b51d7f4 | ||
|
|
7ad63551d6 | ||
|
|
4627bcd109 | ||
|
|
f0af2b95e3 | ||
|
|
f348d82167 | ||
|
|
2b92833ea5 | ||
|
|
42fc054c94 | ||
|
|
7f12ecfd6d | ||
|
|
3ae1afec27 | ||
|
|
9608ed9fdf | ||
|
|
220e904ae9 | ||
|
|
49b696315c | ||
|
|
a9ebf31e6f | ||
|
|
4b4eb0b684 | ||
|
|
18ce34c458 | ||
|
|
70ca3787bb | ||
|
|
396e3fde95 | ||
|
|
c913bc61bb | ||
|
|
7af8e46b59 | ||
|
|
0569bc6000 | ||
|
|
708a2b41dc | ||
|
|
c4edfb6e40 | ||
|
|
cf724507e9 | ||
|
|
744a590467 | ||
|
|
5a3e9bd90e | ||
|
|
d8b24adef7 | ||
|
|
4726ff293c | ||
|
|
2a4e84d827 | ||
|
|
9ef47fa6a4 | ||
|
|
3358340319 | ||
|
|
1fff9fd0e1 | ||
|
|
d12096ef51 | ||
|
|
ea56d21190 | ||
|
|
c802c895b6 | ||
|
|
074bca971f | ||
|
|
7bd9c27b89 | ||
|
|
3e73673302 | ||
|
|
4b12aaa198 | ||
|
|
5f9cab8992 | ||
|
|
96766f7091 | ||
|
|
354aec239f | ||
|
|
3839a0a830 | ||
|
|
0444a4c348 | ||
|
|
73207e4745 | ||
|
|
b7d44445af | ||
|
|
cce6b2ecfc | ||
|
|
883f1e27e6 | ||
|
|
e7fc147424 | ||
|
|
2147b147fc | ||
|
|
f5e9a43461 | ||
|
|
cabb56fc8c | ||
|
|
58786eb6d9 | ||
|
|
776242b259 | ||
|
|
a9d19c3218 | ||
|
|
7bd299ec07 | ||
|
|
107469cd78 | ||
|
|
709e1fd567 | ||
|
|
4eb8f18ebf | ||
|
|
d5d27c8020 | ||
|
|
da3ec41d71 | ||
|
|
5f7b4537c8 | ||
|
|
cce4fecac6 | ||
|
|
737e217172 | ||
|
|
8fe3517894 | ||
|
|
710503663a | ||
|
|
a5b8afc0bd | ||
|
|
f1d3a15498 | ||
|
|
dfd5983715 | ||
|
|
4dcc6a6cf4 | ||
|
|
b58f3e3d67 | ||
|
|
a6b3540cb8 | ||
|
|
ccbf42f81b | ||
|
|
d39c58d8f5 | ||
|
|
36dde2c063 |
3
.github/workflows/aunit.yml
vendored
3
.github/workflows/aunit.yml
vendored
@@ -9,7 +9,7 @@ on: [push]
|
||||
jobs:
|
||||
build:
|
||||
|
||||
runs-on: ubuntu-18.04
|
||||
runs-on: ubuntu-20.04
|
||||
|
||||
steps:
|
||||
- uses: actions/checkout@v2
|
||||
@@ -17,6 +17,7 @@ jobs:
|
||||
- name: Setup
|
||||
run: |
|
||||
cd ..
|
||||
git clone https://github.com/hsaturn/TinyConsole
|
||||
git clone https://github.com/bxparks/EpoxyDuino
|
||||
git clone https://github.com/bxparks/AceRoutine
|
||||
git clone https://github.com/bxparks/AUnit
|
||||
|
||||
52
.github/workflows/superlinter.yml
vendored
52
.github/workflows/superlinter.yml
vendored
@@ -1,26 +1,52 @@
|
||||
name: Super-Linter
|
||||
---
|
||||
#################################
|
||||
#################################
|
||||
## Super Linter GitHub Actions ##
|
||||
#################################
|
||||
#################################
|
||||
name: Lint Code Base
|
||||
|
||||
# Run this workflow every time a new commit pushed to your repository
|
||||
#
|
||||
# Documentation:
|
||||
# https://docs.github.com/en/actions/learn-github-actions/workflow-syntax-for-github-actions
|
||||
#
|
||||
|
||||
#############################
|
||||
# Start the job on all push #
|
||||
#############################
|
||||
on: push
|
||||
|
||||
###############
|
||||
# Set the Job #
|
||||
###############
|
||||
jobs:
|
||||
# Set the job key. The key is displayed as the job name
|
||||
# when a job name is not provided
|
||||
super-lint:
|
||||
build:
|
||||
# Name the Job
|
||||
name: Lint code base
|
||||
# Set the type of machine to run on
|
||||
name: Lint Code Base
|
||||
# Set the agent to run on
|
||||
runs-on: ubuntu-latest
|
||||
|
||||
##################
|
||||
# Load all steps #
|
||||
##################
|
||||
steps:
|
||||
# Checks out a copy of your repository on the ubuntu-latest machine
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v2
|
||||
##########################
|
||||
# Checkout the code base #
|
||||
##########################
|
||||
- name: Checkout Code
|
||||
uses: actions/checkout@v3
|
||||
with:
|
||||
# Full git history is needed to get a proper
|
||||
# list of changed files within `super-linter`
|
||||
fetch-depth: 0
|
||||
|
||||
# Runs the Super-Linter action
|
||||
- name: Run Super-Linter
|
||||
uses: github/super-linter@v3
|
||||
################################
|
||||
# Run Linter against code base #
|
||||
################################
|
||||
- name: Lint Code Base
|
||||
uses: github/super-linter/slim@v4
|
||||
env:
|
||||
VALIDATE_ALL_CODEBASE: false
|
||||
# Change to 'master' if your main branch differs
|
||||
DEFAULT_BRANCH: main
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
|
||||
22
README.md
22
README.md
@@ -12,16 +12,25 @@ TinyMqtt is a small, fast and capable Mqtt Broker and Client for Esp8266 / Esp32
|
||||
|
||||
## Features
|
||||
|
||||
- Very (very !!) fast broker I saw it re-sent 1000 topics per second for two
|
||||
- Very fast broker I saw it re-sent 1000 topics per second for two
|
||||
clients that had subscribed (payload ~15 bytes ESP8266). No topic lost.
|
||||
The max I've seen was 2k msg/s (1 client 1 subscription)
|
||||
- Act as as a mqtt broker and/or a mqtt client
|
||||
- Mqtt 3.1.1 / Qos 0 supported
|
||||
- Wildcards supported (+ # $ and * (even if not part of the spec...))
|
||||
- Standalone (can work without WiFi) (degraded/local mode)
|
||||
- Brokers can connect to another broker and becomes then a
|
||||
proxy for clients that are connected to it.
|
||||
- zeroconf, this is a strange but very powerful mode where
|
||||
all brokers tries to connect together on the same local network.
|
||||
- small memory footprint (very efficient topic storage)
|
||||
- long messages are supported (>127 bytes)
|
||||
- TinyMQTT is largely unit tested, so once a bug is fixed, it is fixed forever
|
||||
|
||||
## Limitations
|
||||
|
||||
- Max of 255 different topics can be stored (change index_t type to allow more)
|
||||
- No Qos because messages are not queued but immediately sent to clients
|
||||
|
||||
## Quickstart
|
||||
|
||||
@@ -33,10 +42,11 @@ TinyMqtt is a small, fast and capable Mqtt Broker and Client for Esp8266 / Esp32
|
||||
|
||||
| Example | Description |
|
||||
| ------------------- | ------------------------------------------ |
|
||||
| client-without-wifi | standalone example |
|
||||
| simple-client | Connect the ESP to an external Mqtt broker |
|
||||
| simple-broker | Simple Mqtt broker with your ESP |
|
||||
| tinymqtt-test | Complex console example |
|
||||
| [client-with-wifi](https://github.com/hsaturn/TinyMqtt/tree/main/examples/client-with-wifi/client-with-wifi.ino) | standalone example |
|
||||
| [client-without-wifi](https://github.com/hsaturn/TinyMqtt/tree/main/examples/client-without-wifi/client-without-wifi.ino) | standalone example |
|
||||
| [simple-client](https://github.com/hsaturn/TinyMqtt/tree/main/examples/simple-client/simple-client.ino) | Connect the ESP to an external Mqtt broker |
|
||||
| [simple-broker](https://github.com/hsaturn/TinyMqtt/tree/main/examples/simple-broker/simple-broker.ino) | Simple Mqtt broker with your ESP |
|
||||
| [tinymqtt-test](https://github.com/hsaturn/TinyMqtt/tree/main/examples/tinymqtt-test/tinymqtt-test.ino) | Complex console example |
|
||||
|
||||
- tinymqtt-test : This is a complex sketch with a terminal console
|
||||
that allows to add clients publish, connect etc with interpreted commands.
|
||||
@@ -59,7 +69,7 @@ no need for having tons of clients (also RAM is the problem with many clients)
|
||||
* ~~MqttClient auto re-subscribe (::resubscribe works bad on broker.emqx.io)~~
|
||||
* MqttClient auto reconnection
|
||||
* MqttClient user/password
|
||||
* Wildcards (I may implement only # as I'm not interrested by a clever and cpu consuming matching)
|
||||
* ~~Wildcards (I may implement only # as I'm not interrested by a clever and cpu consuming matching)~~
|
||||
* I suspect that MqttClient::parent could be removed and replaced with a simple boolean
|
||||
(this'll need to rewrite a few functions)
|
||||
|
||||
|
||||
134
examples/advanced/mqtt_class_binder/mqtt_class_binder.ino
Normal file
134
examples/advanced/mqtt_class_binder/mqtt_class_binder.ino
Normal file
@@ -0,0 +1,134 @@
|
||||
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
||||
#include <MqttClassBinder.h>
|
||||
|
||||
/**
|
||||
* Example on how to bind a class:onPublish function
|
||||
*
|
||||
* Local broker that accept connections and two local clients
|
||||
*
|
||||
*
|
||||
* +-----------------------------+
|
||||
* | ESP |
|
||||
* | +--------+ | 1883 <--- External client/s
|
||||
* | +-------->| broker | | 1883 <--- External client/s
|
||||
* | | +--------+ |
|
||||
* | | ^ |
|
||||
* | | | |
|
||||
* | | | | -----
|
||||
* | v v | ---
|
||||
* | +----------+ +----------+ | -
|
||||
* | | internal | | internal | +-------* Wifi
|
||||
* | | client | | client | |
|
||||
* | +----------+ +----------+ |
|
||||
* | |
|
||||
* +-----------------------------+
|
||||
*
|
||||
* pros - Reduces internal latency (when publish is received by the same ESP)
|
||||
* - Reduces wifi traffic
|
||||
* - No need to have an external broker
|
||||
* - can still report to a 'main' broker (TODO see documentation that have to be written)
|
||||
* - accepts external clients
|
||||
* - MqttClassBinder allows to mix together many mqtt sources
|
||||
*
|
||||
* cons - Takes more memory (24 more bytes for the one MqttClassBinder<Class>
|
||||
* - a bit hard to understand
|
||||
*
|
||||
*/
|
||||
|
||||
const char *ssid = "";
|
||||
const char *password = "";
|
||||
|
||||
std::string topic_b="sensor/btemp";
|
||||
std::string topic_sender= "sensor/counter";
|
||||
|
||||
MqttBroker broker(1883);
|
||||
|
||||
MqttClient mqtt_a(&broker);
|
||||
MqttClient mqtt_b(&broker);
|
||||
MqttClient mqtt_sender(&broker);
|
||||
|
||||
class MqttReceiver: public MqttClassBinder<MqttReceiver>
|
||||
{
|
||||
public:
|
||||
|
||||
void onPublish(const MqttClient* source, const Topic& topic, const char* payload, size_t /* length */)
|
||||
{
|
||||
Serial
|
||||
<< " * MqttReceiver received topic (" << topic.c_str() << ")"
|
||||
<< " from (" << source->id() << "), "
|
||||
<< " payload: (" << payload << ')' << endl;
|
||||
}
|
||||
};
|
||||
|
||||
void setup()
|
||||
{
|
||||
Serial.begin(115200);
|
||||
delay(500);
|
||||
Serial << "Clients with wifi " << endl;
|
||||
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin(ssid, password);
|
||||
|
||||
while (WiFi.status() != WL_CONNECTED)
|
||||
{
|
||||
Serial << '-'; delay(500);
|
||||
if (strlen(ssid)==0)
|
||||
Serial << "****** PLEASE EDIT THE EXAMPLE AND MODIFY ssid/password *************" << endl;
|
||||
}
|
||||
|
||||
Serial << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
|
||||
broker.begin();
|
||||
|
||||
MqttReceiver* receiver = new MqttReceiver;
|
||||
|
||||
// receiver will receive both publication from two MqttClient
|
||||
// (that could be connected to two different brokers)
|
||||
MqttClassBinder<MqttReceiver>::onPublish(&mqtt_a, receiver);
|
||||
MqttClassBinder<MqttReceiver>::onPublish(&mqtt_b, receiver);
|
||||
|
||||
mqtt_a.id("mqtt_a");
|
||||
mqtt_b.id("mqtt_b");
|
||||
mqtt_sender.id("sender");
|
||||
|
||||
mqtt_a.subscribe(topic_b);
|
||||
mqtt_b.subscribe(topic_sender);
|
||||
|
||||
}
|
||||
|
||||
void loop()
|
||||
{
|
||||
broker.loop(); // Don't forget to add loop for every broker and clients
|
||||
|
||||
mqtt_a.loop();
|
||||
mqtt_b.loop();
|
||||
mqtt_sender.loop();
|
||||
|
||||
// ============= client A publish ================
|
||||
{
|
||||
static const int interval = 5000; // publishes every 5s (please avoid usage of delay())
|
||||
static uint32_t timer = millis() + interval;
|
||||
|
||||
if (millis() > timer)
|
||||
{
|
||||
static int counter = 0;
|
||||
Serial << "Sender is publishing " << topic_sender.c_str() << endl;
|
||||
timer += interval;
|
||||
mqtt_sender.publish(topic_sender, "sent by Sender, message #"+std::string(String(counter++).c_str()));
|
||||
}
|
||||
}
|
||||
|
||||
// ============= client B publish ================
|
||||
{
|
||||
static const int interval = 7000; // will send topic each 7s
|
||||
static uint32_t timer = millis() + interval;
|
||||
static int temperature;
|
||||
|
||||
if (millis() > timer)
|
||||
{
|
||||
Serial << "B is publishing " << topic_b.c_str() << endl;
|
||||
timer += interval;
|
||||
mqtt_b.publish(topic_b, "sent by B: temp="+std::string(String(16+temperature++%6).c_str()));
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -26,7 +26,7 @@ void setup()
|
||||
Serial.begin(115200);
|
||||
|
||||
if (strlen(ssid)==0)
|
||||
Serial << "****** PLEASE MODIFY ssid/password *************" << endl;
|
||||
Console << TinyConsole::red << "****** PLEASE MODIFY ssid/password *************" << endl;
|
||||
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin(ssid, password);
|
||||
@@ -35,10 +35,10 @@ void setup()
|
||||
Serial << '.';
|
||||
delay(500);
|
||||
}
|
||||
Serial << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
Console << TinyConsole::green << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
|
||||
broker.begin();
|
||||
Serial << "Broker ready : " << WiFi.localIP() << " on port " << PORT << endl;
|
||||
Console << "Broker ready : " << WiFi.localIP() << " on port " << PORT << endl;
|
||||
}
|
||||
|
||||
void loop()
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
#include "TinyMqtt.h" // https://github.com/hsaturn/TinyMqtt
|
||||
#include "TinyStreaming.h" // https://github.com/hsaturn/TinyConsole
|
||||
|
||||
/** Simple Client (The simplest configuration)
|
||||
*
|
||||
@@ -55,7 +56,7 @@ void setup()
|
||||
|
||||
Serial << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
|
||||
client.connect(BROKER, BROKER_PORT); // Put here your broker ip / port
|
||||
client.connect(BROKER, BROKER_PORT);
|
||||
}
|
||||
|
||||
void loop()
|
||||
|
||||
@@ -1,5 +1,7 @@
|
||||
// vim: ts=2 sw=2 expandtab smartindent
|
||||
#include <TinyConsole.h>
|
||||
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
||||
#include <MqttStreaming.h>
|
||||
#include <TinyStreaming.h>
|
||||
#if defined(ESP8266)
|
||||
#include <ESP8266mDNS.h>
|
||||
#elif defined(ESP32)
|
||||
@@ -10,9 +12,37 @@
|
||||
#endif
|
||||
|
||||
#include <sstream>
|
||||
#include <string>
|
||||
#include <map>
|
||||
|
||||
bool echo_on = true;
|
||||
auto green = TinyConsole::green;
|
||||
auto red = TinyConsole::red;
|
||||
auto white = TinyConsole::white;
|
||||
auto cyan = TinyConsole::cyan;
|
||||
auto yellow = TinyConsole::yellow;
|
||||
auto magenta = TinyConsole::magenta;
|
||||
auto save_cursor = TinyConsole::save_cursor;
|
||||
auto restore_cursor = TinyConsole::restore_cursor;
|
||||
auto erase_to_end = TinyConsole::erase_to_end;
|
||||
|
||||
const char* ssid = "";
|
||||
const char* password = "";
|
||||
|
||||
struct free_broker
|
||||
{
|
||||
public:
|
||||
free_broker(const char* s, uint16_t p, const char* comment) : url(s), port(p) {}
|
||||
|
||||
std::string url;
|
||||
uint16_t port;
|
||||
};
|
||||
|
||||
const std::map<std::string, free_broker> list =
|
||||
{
|
||||
{ "mqtthq", { "public.mqtthq.com" , 8083, "publish/subscribe" }},
|
||||
{ "hivemq", { "broker.hivemq.com", 1883, "" }}
|
||||
};
|
||||
|
||||
/** Very complex example
|
||||
* Console allowing to make any kind of test,
|
||||
@@ -25,22 +55,22 @@ bool echo_on = true;
|
||||
* TODO examples of scripts
|
||||
*/
|
||||
|
||||
const char* ssid = "";
|
||||
const char* password = "";
|
||||
|
||||
std::string topic="sensor/temperature";
|
||||
|
||||
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
||||
{
|
||||
Serial << "--> " << srce->id().c_str() << ": ======> received " << topic.c_str();
|
||||
if (payload) Serial << ", payload[" << length << "]=[";
|
||||
Console << cyan << "--> " << srce->id().c_str() << ": received " << topic.c_str() << white;
|
||||
if (payload)
|
||||
{
|
||||
Console << ", payload[" << length << "]=[";
|
||||
while(length--)
|
||||
{
|
||||
const char c=*payload++;
|
||||
if (c!=10 and c!=13 and c <32) Serial << '?';
|
||||
Serial << *payload++;
|
||||
if (c<32)
|
||||
Console << '?';
|
||||
else
|
||||
Console << c;
|
||||
}
|
||||
Console << ']' << endl;
|
||||
}
|
||||
Serial<< endl;
|
||||
}
|
||||
|
||||
std::map<std::string, MqttClient*> clients;
|
||||
@@ -50,23 +80,26 @@ void setup()
|
||||
{
|
||||
WiFi.persistent(false); // https://github.com/esp8266/Arduino/issues/1054
|
||||
Serial.begin(115200);
|
||||
Console.begin(Serial);
|
||||
Console.setPrompt("> ");
|
||||
Console.setCallback(onCommand);
|
||||
delay(500);
|
||||
|
||||
Serial << endl << endl;
|
||||
Serial << "***************************************************************" << endl;
|
||||
Serial << "* Welcome to the TinyMqtt console" << endl;
|
||||
Serial << endl;
|
||||
Serial << "* The console allows to test all features of the libraries." << endl;
|
||||
Serial << endl;
|
||||
Console.cls();
|
||||
Console << endl << endl;
|
||||
Console << yellow
|
||||
<< "***************************************************************" << endl;
|
||||
Console << "* Welcome to the TinyMqtt console" << endl;
|
||||
Console << "* The console allows to test all features of the libraries." << endl;
|
||||
Console << "* Enter help to view the list of commands." << endl;
|
||||
Console << "***************************************************************" << endl;
|
||||
Console << endl;
|
||||
if (strlen(ssid)==0)
|
||||
Serial << "* WARNING: You may want to modify ssid/password in order" << endl
|
||||
<< " to reflect your Wifi configuration." << endl;
|
||||
Serial << endl;
|
||||
Serial << "* Enter help to view the list of commands." << endl;
|
||||
Serial << "***************************************************************" << endl;
|
||||
Serial << endl;
|
||||
Console << red << "* ERROR: You must modify ssid/password in order" << endl
|
||||
<< " to be able to connect to your Wifi network." << endl;
|
||||
Console << endl << white;
|
||||
|
||||
Serial << "Connecting to '" << ssid << "' ";
|
||||
Console << "Connecting to '" << ssid << "' ";
|
||||
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin(ssid, password);
|
||||
@@ -74,21 +107,23 @@ void setup()
|
||||
WiFi.begin(ssid, password);
|
||||
|
||||
while (WiFi.status() != WL_CONNECTED)
|
||||
{ Serial << '-'; delay(500); }
|
||||
{ Console << '-'; delay(500); }
|
||||
|
||||
Serial << endl << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
Console << endl << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
|
||||
const char* name="tinytest";
|
||||
Serial << "Starting MDNS, name= " << name;
|
||||
Console << "Starting MDNS, name= " << name;
|
||||
if (!MDNS.begin(name))
|
||||
Serial << " error, not available." << endl;
|
||||
Console << " error, not available." << endl;
|
||||
else
|
||||
Serial << " ok." << endl;
|
||||
Console << " ok." << endl;
|
||||
|
||||
|
||||
MqttBroker* broker = new MqttBroker(1883);
|
||||
broker->begin();
|
||||
brokers["broker"] = broker;
|
||||
|
||||
if (Console.isTerm()) onCommand("every 333 view");
|
||||
}
|
||||
|
||||
std::string getword(std::string& str, const char* if_empty=nullptr, char sep=' ');
|
||||
@@ -137,7 +172,7 @@ std::string getword(std::string& str, const char* if_empty/*=nullptr*/, char sep
|
||||
{
|
||||
sword.erase(0,1);
|
||||
to = getint(sword);
|
||||
if (sword[0]!=')') Serial << "Missing ')'" << endl;
|
||||
if (sword[0]!=')') Console << "Missing ')'" << endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
@@ -148,7 +183,7 @@ std::string getword(std::string& str, const char* if_empty/*=nullptr*/, char sep
|
||||
}
|
||||
else
|
||||
{
|
||||
Serial << "Missing '('" << endl;
|
||||
Console << "Missing '('" << endl;
|
||||
}
|
||||
}
|
||||
while(str[0]==' ') str.erase(0,1);
|
||||
@@ -191,15 +226,16 @@ std::string getip(std::string& str, const char* if_empty=nullptr, char sep=' ')
|
||||
if (addr.length()) addr += '.';
|
||||
addr += s;
|
||||
}
|
||||
Console << "connect address: " << addr << endl;
|
||||
return addr;
|
||||
}
|
||||
|
||||
std::map<std::string, std::string> vars;
|
||||
|
||||
std::set<std::string> commands = {
|
||||
"auto", "broker", "blink", "client", "connect",
|
||||
"create", "delete", "help", "interval",
|
||||
"ls", "ip", "off", "on", "set",
|
||||
"broker", "blink", "client", "connect",
|
||||
"create", "delete", "debug", "help", "interval",
|
||||
"list", "ls", "ip", "off", "on", "set",
|
||||
"publish", "reset", "subscribe", "unsubscribe", "view", "echo", "every"
|
||||
};
|
||||
|
||||
@@ -222,7 +258,7 @@ void convertToCommand(std::string& search)
|
||||
search = matches;
|
||||
else if (count>1)
|
||||
{
|
||||
Serial << "Ambiguous command: " << matches << endl;
|
||||
Console << "Ambiguous command: " << matches << endl;
|
||||
search.clear();
|
||||
}
|
||||
}
|
||||
@@ -260,123 +296,6 @@ void replaceVars(std::string& cmd)
|
||||
|
||||
}
|
||||
|
||||
// publish at regular interval
|
||||
class automatic
|
||||
{
|
||||
public:
|
||||
automatic(MqttClient* clt, uint32_t intervl)
|
||||
: client(clt), topic_(::topic)
|
||||
{
|
||||
interval(intervl);
|
||||
autos[clt] = this;
|
||||
}
|
||||
|
||||
void interval(uint32_t new_interval)
|
||||
{
|
||||
interval_ = new_interval;
|
||||
if (interval_<1000) interval_=1000;
|
||||
timer_ = millis() + interval_;
|
||||
}
|
||||
|
||||
void loop_()
|
||||
{
|
||||
if (!bon) return;
|
||||
if (interval_ && millis() > timer_)
|
||||
{
|
||||
Serial << "AUTO PUBLISH " << interval_ << endl;
|
||||
timer_ += interval_;
|
||||
client->publish(topic_, std::string(String(15+millis()%10).c_str()));
|
||||
}
|
||||
}
|
||||
|
||||
void topic(std::string new_topic) { topic_ = new_topic; }
|
||||
|
||||
static void loop()
|
||||
{
|
||||
for(auto it: autos)
|
||||
it.second->loop_();
|
||||
}
|
||||
|
||||
static void command(MqttClient* who, std::string cmd)
|
||||
{
|
||||
automatic* autop = nullptr;
|
||||
if (autos.find(who) != autos.end())
|
||||
{
|
||||
autop=autos[who];
|
||||
}
|
||||
std::string s = getword(cmd);
|
||||
if (compare(s, "create"))
|
||||
{
|
||||
std::string seconds=getword(cmd, "10000");
|
||||
if (autop) delete autop;
|
||||
std::string top = getword(cmd, ::topic.c_str());
|
||||
autos[who] = new automatic(who, atol(seconds.c_str()));
|
||||
autos[who]->topic(top);
|
||||
autos[who]->bon=true;
|
||||
Serial << "New auto (" << seconds.c_str() << " topic:" << top.c_str() << ')' << endl;
|
||||
}
|
||||
else if (autop)
|
||||
{
|
||||
while(s.length())
|
||||
{
|
||||
if (s=="on")
|
||||
{
|
||||
autop->bon = true;
|
||||
autop->interval(autop->interval_);
|
||||
}
|
||||
else if (s=="off")
|
||||
autop->bon=false;
|
||||
else if (s=="interval")
|
||||
{
|
||||
int32_t i=getint(cmd);
|
||||
if (i)
|
||||
autop->interval(atol(s.c_str()));
|
||||
else
|
||||
Serial << "Bad value" << endl;
|
||||
}
|
||||
else if (s=="view")
|
||||
{
|
||||
Serial << " automatic "
|
||||
<< (int32_t)autop->client
|
||||
<< " interval " << autop->interval_
|
||||
<< (autop->bon ? " on" : " off") << endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
Serial << "Unknown auto command (" << s.c_str() << ")" << endl;
|
||||
break;
|
||||
}
|
||||
s=getword(cmd);
|
||||
}
|
||||
}
|
||||
else if (who==nullptr)
|
||||
{
|
||||
for(auto it: autos)
|
||||
command(it.first, s+' '+cmd);
|
||||
}
|
||||
else
|
||||
Serial << "what ? (" << s.c_str() << ")" << endl;
|
||||
}
|
||||
|
||||
static void help()
|
||||
{
|
||||
Serial << " auto [$id] on/off" << endl;
|
||||
Serial << " auto [$id] view" << endl;
|
||||
Serial << " auto [$id] interval [s]" << endl;
|
||||
Serial << " auto [$id] create [millis] [topic]" << endl;
|
||||
}
|
||||
|
||||
private:
|
||||
MqttClient* client;
|
||||
uint32_t interval_;
|
||||
uint32_t timer_;
|
||||
std::string topic_;
|
||||
bool bon=false;
|
||||
static std::map<MqttClient*, automatic*> autos;
|
||||
float temp=19;
|
||||
};
|
||||
std::map<MqttClient*, automatic*> automatic::autos;
|
||||
|
||||
bool compare(std::string s, const char* cmd)
|
||||
{
|
||||
uint8_t p=0;
|
||||
@@ -400,13 +319,17 @@ struct Every
|
||||
|
||||
void dump()
|
||||
{
|
||||
Serial << (active ? "enabled " : "disabled ");
|
||||
auto mill=millis();
|
||||
Serial << ms << "ms [" << cmd << "] next in ";
|
||||
if (mill > next)
|
||||
Serial << "now";
|
||||
if (active)
|
||||
Console << green << "enabled";
|
||||
else
|
||||
Serial << next-mill << "ms";
|
||||
Console << red << "disabled";
|
||||
|
||||
auto mill=millis();
|
||||
Console << white << ms << "ms [" << cmd << "] next in ";
|
||||
if (mill > next)
|
||||
Console << "now";
|
||||
else
|
||||
Console << next-mill << "ms";
|
||||
}
|
||||
};
|
||||
|
||||
@@ -418,6 +341,34 @@ int16_t blink;
|
||||
|
||||
std::vector<Every> everies;
|
||||
|
||||
void onCommand(const std::string& command)
|
||||
{
|
||||
Console << endl;
|
||||
std::string cmd=command;
|
||||
if (cmd.substr(0,3)!="set") replaceVars(cmd);
|
||||
eval(cmd);
|
||||
Console << endl;
|
||||
Console.prompt();
|
||||
}
|
||||
|
||||
void clientConnect(MqttClient* client, std::string& cmd)
|
||||
{
|
||||
std::string remote = getword(cmd);
|
||||
uint16_t port;
|
||||
auto it=list.find(remote);
|
||||
if (it != list.end())
|
||||
{
|
||||
Console << "Connecting to free broker: " << remote << endl;
|
||||
remote = it->second.url;
|
||||
port=it->second.port;
|
||||
}
|
||||
else
|
||||
port=getint(cmd);
|
||||
|
||||
client->connect(remote.c_str(), port, getint(cmd, 60));
|
||||
Console << (client->connected() ? "connected." : "not connected") << endl;
|
||||
}
|
||||
|
||||
void eval(std::string& cmd)
|
||||
{
|
||||
while(cmd.length())
|
||||
@@ -446,7 +397,7 @@ void eval(std::string& cmd)
|
||||
}
|
||||
else
|
||||
{
|
||||
Serial << "Unknown class (" << s.c_str() << ")" << endl;
|
||||
Console << red << "Unknown class (" << s.c_str() << ")" << white << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
@@ -456,6 +407,22 @@ void eval(std::string& cmd)
|
||||
if (s.length()) convertToCommand(s);
|
||||
if (s.length()==0)
|
||||
{}
|
||||
else if (compare(s, "debug"))
|
||||
{
|
||||
#if TINY_MQTT_DEBUG
|
||||
TinyMqtt::debug = getint(cmd);
|
||||
#else
|
||||
Console << red << "TinyMqtt not compiled in debug" << endl;
|
||||
#endif
|
||||
}
|
||||
else if (compare(s, "list"))
|
||||
{
|
||||
Console << "List of free servers" << endl;
|
||||
for(const auto& fb: list)
|
||||
{
|
||||
Console << " " << fb.first << " : " << fb.second.url << ":" << fb.second.port << endl;
|
||||
}
|
||||
}
|
||||
else if (compare(s, "delete"))
|
||||
{
|
||||
if (client==nullptr && broker==nullptr)
|
||||
@@ -470,14 +437,14 @@ void eval(std::string& cmd)
|
||||
broker = brokers[s];
|
||||
}
|
||||
else
|
||||
Serial << "Unable to find (" << s.c_str() << ")" << endl;
|
||||
Console << red << "Unable to find (" << s.c_str() << ")" << white << endl;
|
||||
}
|
||||
if (client)
|
||||
{
|
||||
for (auto it: clients)
|
||||
{
|
||||
if (it.second != client) continue;
|
||||
Serial << "deleted" << endl;
|
||||
Console << "deleted" << endl;
|
||||
delete (it.second);
|
||||
clients.erase(it.first);
|
||||
break;
|
||||
@@ -489,7 +456,7 @@ void eval(std::string& cmd)
|
||||
for(auto it: brokers)
|
||||
{
|
||||
if (broker != it.second) continue;
|
||||
Serial << "deleted" << endl;
|
||||
Console << "deleted" << endl;
|
||||
delete (it.second);
|
||||
brokers.erase(it.first);
|
||||
break;
|
||||
@@ -497,13 +464,13 @@ void eval(std::string& cmd)
|
||||
cmd += " ls";
|
||||
}
|
||||
else
|
||||
Serial << "Nothing to delete" << endl;
|
||||
Console << "Nothing to delete" << endl;
|
||||
}
|
||||
else if (broker)
|
||||
{
|
||||
if (compare(s,"connect"))
|
||||
{
|
||||
Serial << "NYI" << endl;
|
||||
Console << "NYI" << endl;
|
||||
}
|
||||
else if (compare(s, "view"))
|
||||
{
|
||||
@@ -511,7 +478,7 @@ void eval(std::string& cmd)
|
||||
}
|
||||
else
|
||||
{
|
||||
Serial << "Unknown broker command (" << s << ")" << endl;
|
||||
Console << "Unknown broker command (" << s << ")" << endl;
|
||||
s.clear();
|
||||
}
|
||||
}
|
||||
@@ -519,20 +486,19 @@ void eval(std::string& cmd)
|
||||
{
|
||||
if (compare(s,"connect"))
|
||||
{
|
||||
client->connect(getip(cmd,"192.168.1.40").c_str(), getint(cmd, 1883), getint(cmd, 60));
|
||||
Serial << (client->connected() ? "connected." : "not connected") << endl;
|
||||
clientConnect(client, cmd);
|
||||
}
|
||||
else if (compare(s,"publish"))
|
||||
{
|
||||
retval = client->publish(getword(cmd, topic.c_str()), getword(cmd));
|
||||
retval = client->publish(getword(cmd), getword(cmd));
|
||||
}
|
||||
else if (compare(s,"subscribe"))
|
||||
{
|
||||
client->subscribe(getword(cmd, topic.c_str()));
|
||||
client->subscribe(getword(cmd));
|
||||
}
|
||||
else if (compare(s, "unsubscribe"))
|
||||
{
|
||||
client->unsubscribe(getword(cmd, topic.c_str()));
|
||||
client->unsubscribe(getword(cmd));
|
||||
}
|
||||
else if (compare(s, "view"))
|
||||
{
|
||||
@@ -540,7 +506,7 @@ void eval(std::string& cmd)
|
||||
}
|
||||
else
|
||||
{
|
||||
Serial << "Unknown client command (" << s << ")" << endl;
|
||||
Console << "Unknown client command (" << s << ")" << endl;
|
||||
s.clear();
|
||||
}
|
||||
}
|
||||
@@ -565,10 +531,10 @@ void eval(std::string& cmd)
|
||||
echo_on = false;
|
||||
else
|
||||
{
|
||||
Serial << s << ' ';
|
||||
Console << s << ' ';
|
||||
while(cmd.length())
|
||||
{
|
||||
Serial << getword(cmd) << ' ';
|
||||
Console << getword(cmd) << ' ';
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -585,7 +551,7 @@ void eval(std::string& cmd)
|
||||
every.next=millis()+ms;
|
||||
everies.push_back(every);
|
||||
every.dump();
|
||||
Serial << endl;
|
||||
Console << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
@@ -600,7 +566,7 @@ void eval(std::string& cmd)
|
||||
{
|
||||
if (every.active != active)
|
||||
{
|
||||
Serial << "every #" << count << (active ? " on" :" off") << endl;
|
||||
Console << "every #" << count << (active ? " on" :" off") << endl;
|
||||
every.active = active;
|
||||
every.underrun = 0;
|
||||
}
|
||||
@@ -611,19 +577,19 @@ void eval(std::string& cmd)
|
||||
else if (compare(cmd, "list") or cmd.length()==0)
|
||||
{
|
||||
getword(cmd);
|
||||
Serial << "List of everies (ms=" << millis() << ")" << endl;
|
||||
Console << "List of everies (ms=" << millis() << ")" << endl;
|
||||
uint8_t count=0;
|
||||
for(auto& every: everies)
|
||||
{
|
||||
Serial << count << ": ";
|
||||
Console << count << ": ";
|
||||
every.dump();
|
||||
Serial << endl;
|
||||
Console << endl;
|
||||
count++;
|
||||
}
|
||||
}
|
||||
else if (compare(cmd, "remove"))
|
||||
{
|
||||
Serial << "Removing..." << endl;
|
||||
Console << "Removing..." << endl;
|
||||
getword(cmd);
|
||||
int8_t every=getint(cmd, -1);
|
||||
if (every==-1 and compare(cmd, "last") and everies.size())
|
||||
@@ -641,10 +607,10 @@ void eval(std::string& cmd)
|
||||
everies.erase(everies.begin()+every);
|
||||
}
|
||||
else
|
||||
Serial << "Bad colmmand" << endl;
|
||||
Console << "Bad colmmand" << endl;
|
||||
}
|
||||
else
|
||||
Serial << "Bad command" << endl;
|
||||
Console << "Bad command" << endl;
|
||||
}
|
||||
else if (compare(s, "blink"))
|
||||
{
|
||||
@@ -655,7 +621,7 @@ void eval(std::string& cmd)
|
||||
blink_ms_off[blink_nr]=getint(cmd, blink_ms_on[blink_nr]);
|
||||
pinMode(blink_nr, OUTPUT);
|
||||
blink_next[blink_nr] = millis();
|
||||
Serial << "Blink " << blink_nr << ' ' << (blink_ms_on[blink_nr] ? "on" : "off") << endl;
|
||||
Console << "Blink " << blink_nr << ' ' << (blink_ms_on[blink_nr] ? "on" : "off") << endl;
|
||||
if (blink_ms_on[blink_nr])
|
||||
blink |= 1<< blink_nr;
|
||||
else
|
||||
@@ -664,16 +630,15 @@ void eval(std::string& cmd)
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (compare(s, "auto"))
|
||||
{
|
||||
automatic::command(client, cmd);
|
||||
if (client == nullptr)
|
||||
cmd.clear();
|
||||
}
|
||||
else if (compare(s, "broker"))
|
||||
{
|
||||
std::string id=getword(cmd);
|
||||
if (id.length() or brokers.find(id)!=brokers.end())
|
||||
if (clients.find(id) != clients.end())
|
||||
{
|
||||
Console << "A client already have that name" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
else if (id.length() or brokers.find(id)!=brokers.end())
|
||||
{
|
||||
int port=getint(cmd, 0);
|
||||
if (port)
|
||||
@@ -682,40 +647,56 @@ void eval(std::string& cmd)
|
||||
broker->begin();
|
||||
|
||||
brokers[id] = broker;
|
||||
Serial << "new broker (" << id.c_str() << ")" << endl;
|
||||
Console << "new broker (" << id.c_str() << ")" << endl;
|
||||
}
|
||||
else
|
||||
Serial << "Missing port" << endl;
|
||||
{
|
||||
Console << "Missing port" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else
|
||||
Serial << "Missing or existing broker name (" << id.c_str() << ")" << endl;
|
||||
cmd+=" ls";
|
||||
{
|
||||
Console << "Missing or existing broker name (" << id.c_str() << ")" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else if (compare(s, "client"))
|
||||
{
|
||||
std::string id=getword(cmd);
|
||||
if (id.length() or clients.find(id)!=clients.end())
|
||||
if (brokers.find(id) != brokers.end())
|
||||
{
|
||||
s=getword(cmd); // broker name
|
||||
if (s=="" or brokers.find(s) != brokers.end())
|
||||
Console << "A broker have that name" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
else if (id.length() or clients.find(id)!=clients.end())
|
||||
{
|
||||
s=getword(cmd); // broker
|
||||
if (s=="" or brokers.find(s) != brokers.end() or list.find(s) != list.end())
|
||||
{
|
||||
MqttBroker* broker = nullptr;
|
||||
if (s.length()) broker = brokers[s];
|
||||
MqttClient* client = new MqttClient(broker);
|
||||
client->id(id);
|
||||
MqttClient* client = new MqttClient(broker, id);
|
||||
clients[id]=client;
|
||||
client->setCallback(onPublish);
|
||||
client->subscribe(topic);
|
||||
Serial << "new client (" << id.c_str() << ", " << s.c_str() << ')' << endl;
|
||||
if (list.find(s) != list.end())
|
||||
{
|
||||
cmd=s+' '+cmd;
|
||||
clientConnect(client, cmd);
|
||||
}
|
||||
Console << "new client (" << id.c_str() << ", " << s.c_str() << ')' << endl;
|
||||
}
|
||||
else if (s.length())
|
||||
{
|
||||
Serial << " not found." << endl;
|
||||
Console << " not found." << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else
|
||||
Serial << "Missing or existing client name" << endl;
|
||||
cmd+=" ls";
|
||||
{
|
||||
Console << "Missing or existing client name" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else if (compare(s, "set"))
|
||||
{
|
||||
@@ -724,12 +705,12 @@ void eval(std::string& cmd)
|
||||
{
|
||||
for(auto it: vars)
|
||||
{
|
||||
Serial << " " << it.first << " -> " << it.second << endl;
|
||||
Console << " " << it.first << " -> " << it.second << endl;
|
||||
}
|
||||
}
|
||||
else if (commands.find(name) != commands.end())
|
||||
{
|
||||
Serial << "Reserved keyword (" << name << ")" << endl;
|
||||
Console << "Reserved keyword (" << name << ")" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
else
|
||||
@@ -745,63 +726,76 @@ void eval(std::string& cmd)
|
||||
}
|
||||
else if (compare(s, "ls") or compare(s, "view"))
|
||||
{
|
||||
Serial << "--< " << clients.size() << " client/s. >--" << endl;
|
||||
bool view = compare(s, "view");
|
||||
if (view)
|
||||
{
|
||||
Console << save_cursor << magenta;
|
||||
Console.gotoxy(1,1);
|
||||
}
|
||||
Console << "--< " << '/' << clients.size() << " client/s. >--" << erase_to_end << endl;
|
||||
for(auto it: clients)
|
||||
{
|
||||
it.second->dump(" ");
|
||||
}
|
||||
|
||||
Serial << "--< " << brokers.size() << " brokers/s. >--" << endl;
|
||||
Console << "--< " << brokers.size() << " brokers/s. >--" << erase_to_end << endl;
|
||||
for(auto it: brokers)
|
||||
{
|
||||
Serial << " +-- '" << it.first.c_str() << "' " << it.second->clientsCount() << " client/s."<< endl;
|
||||
Console << " +-- '" << it.first.c_str() << "' " << it.second->clientsCount() << " client/s."<< erase_to_end << endl;
|
||||
it.second->dump(" ");
|
||||
}
|
||||
if (view)
|
||||
{
|
||||
Console.bg(white);
|
||||
Console << erase_to_end << restore_cursor;
|
||||
}
|
||||
}
|
||||
else if (compare(s, "reset"))
|
||||
ESP.restart();
|
||||
else if (compare(s, "ip"))
|
||||
Serial << "IP: " << WiFi.localIP() << endl;
|
||||
Console << "IP: " << WiFi.localIP() << endl;
|
||||
else if (compare(s,"help"))
|
||||
{
|
||||
Serial << "syntax:" << endl;
|
||||
Serial << " MqttBroker:" << endl;
|
||||
Serial << " broker {name} {port} : create a new broker" << endl;
|
||||
Serial << endl;
|
||||
Serial << " MqttClient:" << endl;
|
||||
Serial << " client {name} {parent broker} : create a client then" << endl;
|
||||
Serial << " name.connect [ip] [port] [alive]" << endl;
|
||||
Serial << " name.[un]subscribe [topic]" << endl;
|
||||
Serial << " name.publish [topic][payload]" << endl;
|
||||
Serial << " name.view" << endl;
|
||||
Serial << " name.delete" << endl;
|
||||
|
||||
automatic::help();
|
||||
Serial << endl;
|
||||
Serial << " help" << endl;
|
||||
Serial << " blink [Dx on_ms off_ms]" << endl;
|
||||
Serial << " ls / ip / reset" << endl;
|
||||
Serial << " set [name][value]" << endl;
|
||||
Serial << " ! repeat last command" << endl;
|
||||
Serial << endl;
|
||||
Serial << " echo [on|off] or strings" << endl;
|
||||
Serial << " every ms [command]; every list; every remove [nr|all], every (on|off) [#]" << endl;
|
||||
Serial << " on {output}; off {output}" << endl;
|
||||
Serial << " $id : name of the client." << endl;
|
||||
Serial << " rnd[(min[,max])] random number." << endl;
|
||||
Serial << " default topic is '" << topic.c_str() << "'" << endl;
|
||||
Serial << endl;
|
||||
Console << "syntax:" << endl;
|
||||
Console << " MqttBroker:" << endl;
|
||||
Console << " broker {broker_name} {port} : create a new broker" << endl;
|
||||
Console << " broker_name can be one of 'list'" << endl;
|
||||
Console << " broker_name.delete : delete a broker (buggy)" << endl;
|
||||
Console << " broker_name.view : dump a broker" << endl;
|
||||
Console << endl;
|
||||
Console << " MqttClient:" << endl;
|
||||
Console << " client {name} {broker} : create a client then" << endl;
|
||||
Console << " name.connect [ip] [port] [alive]" << endl;
|
||||
Console << " name.[un]subscribe topic" << endl;
|
||||
Console << " name.publish topic [payload]" << endl;
|
||||
Console << " name.view" << endl;
|
||||
Console << " name.delete" << endl;
|
||||
Console << endl;
|
||||
Console << " list : list of free brokers (debug 1 advised)" << endl;
|
||||
Console << " debug #" << endl;
|
||||
Console << " list : get list of free brokers" << endl;
|
||||
Console << " blink [Dx on_ms off_ms] : make pin blink" << endl;
|
||||
Console << " ls / ip / reset" << endl;
|
||||
Console << " set [name][value]" << endl;
|
||||
Console << " ! repeat last command" << endl;
|
||||
Console << endl;
|
||||
Console << " echo [on|off] or strings" << endl;
|
||||
Console << " every ms [command]; every list; every remove [nr|all]; every (on|off) [#]" << endl;
|
||||
Console << " on {output}; off {output}" << endl;
|
||||
Console << " $id : name of the client." << endl;
|
||||
Console << " rnd[(min[,max])] random number." << endl;
|
||||
Console << endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
while(s[0]==' ') s.erase(0,1);
|
||||
if (s.length())
|
||||
Serial << "Unknown command (" << s.c_str() << ")" << endl;
|
||||
Console << "Unknown command (" << s.c_str() << ")" << endl;
|
||||
}
|
||||
|
||||
if (retval != MqttOk)
|
||||
{
|
||||
Serial << "# MQTT ERROR " << retval << endl;
|
||||
Console << "# MQTT ERROR " << retval << endl;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -823,7 +817,8 @@ void loop()
|
||||
every.next += every.ms;
|
||||
if (ms > every.next and ms > every.underrun)
|
||||
{
|
||||
Serial << "Underrun every #" << e << ", " << (ms - every.next) << "ms late" << endl;
|
||||
every.next += every.ms;
|
||||
Console << yellow << "Underrun every #" << e << ", " << (ms - every.next) << "ms late" << endl;
|
||||
every.underrun = ms+5000;
|
||||
}
|
||||
}
|
||||
@@ -855,41 +850,11 @@ void loop()
|
||||
MDNS.update();
|
||||
#endif
|
||||
|
||||
if (MqttClient::counter != count)
|
||||
{
|
||||
Serial << "# " << MqttClient::counter << endl;
|
||||
count = MqttClient::counter;
|
||||
}
|
||||
for(auto it: brokers)
|
||||
it.second->loop();
|
||||
|
||||
for(auto it: clients)
|
||||
it.second->loop();
|
||||
|
||||
automatic::loop();
|
||||
|
||||
if (Serial.available())
|
||||
{
|
||||
static std::string cmd;
|
||||
char c=Serial.read();
|
||||
if (echo_on)
|
||||
Serial << c;
|
||||
|
||||
if (c==10 or c==13)
|
||||
{
|
||||
Serial << "----------------[ " << cmd.c_str() << " ]--------------" << endl;
|
||||
static std::string last_cmd;
|
||||
if (cmd=="!")
|
||||
cmd=last_cmd;
|
||||
else
|
||||
last_cmd=cmd;
|
||||
|
||||
if (cmd.substr(0,3)!="set") replaceVars(cmd);
|
||||
eval(cmd);
|
||||
}
|
||||
else
|
||||
{
|
||||
cmd=cmd+c;
|
||||
}
|
||||
}
|
||||
Console.loop();
|
||||
}
|
||||
|
||||
18
library.json
18
library.json
@@ -1,18 +0,0 @@
|
||||
{
|
||||
"name": "TinyMqtt",
|
||||
"keywords": "ethernet, mqtt, m2m, iot",
|
||||
"description": "MQTT is a lightweight messaging protocol ideal for small devices. This library allows to send and receive and host a broker for MQTT. It does support MQTT 3.1.1 with QOS=0 on ESP8266 and ESP32 WROOM platfrms.",
|
||||
"repository": {
|
||||
"type": "git",
|
||||
"url": "https://github.com/hsaturn/TinyMqtt.git"
|
||||
},
|
||||
"version": "0.7.9",
|
||||
"exclude": "",
|
||||
"examples": "examples/*/*.ino",
|
||||
"frameworks": "arduino",
|
||||
"platforms": [
|
||||
"atmelavr",
|
||||
"espressif8266",
|
||||
"espressif32"
|
||||
]
|
||||
}
|
||||
@@ -1,5 +1,5 @@
|
||||
name=TinyMqtt
|
||||
version=0.7.9
|
||||
version=0.9.11
|
||||
author=Francois BIOT, HSaturn, <hsaturn@gmail.com>
|
||||
maintainer=Francois BIOT, HSaturn, <hsaturn@gmail.com>
|
||||
sentence=A tiny broker and client library for MQTT messaging.
|
||||
@@ -7,5 +7,5 @@ paragraph=MQTT is a lightweight messaging protocol ideal for small devices. This
|
||||
category=Communication
|
||||
url=https://github.com/hsaturn/TinyMqtt
|
||||
architectures=*
|
||||
depends=TinyConsole
|
||||
includes=TinyMqtt.h
|
||||
depends=AsyncTCP
|
||||
|
||||
72
src/MqttClassBinder.h
Normal file
72
src/MqttClassBinder.h
Normal file
@@ -0,0 +1,72 @@
|
||||
// MqttReceiver must implement onPublish(...)
|
||||
template <class MqttReceiver>
|
||||
class MqttClassBinder
|
||||
{
|
||||
public:
|
||||
MqttClassBinder()
|
||||
{
|
||||
unregister(this);
|
||||
}
|
||||
~MqttClassBinder() { unregister(this); }
|
||||
|
||||
static void onUnpublished(MqttClient::CallBack handler)
|
||||
{
|
||||
unrouted_handler = handler;
|
||||
}
|
||||
|
||||
static void onPublish(MqttClient* client, MqttReceiver* dest)
|
||||
{
|
||||
routes.insert(std::pair<MqttClient*, MqttReceiver*>(client, dest));
|
||||
client->setCallback(onRoutePublish);
|
||||
}
|
||||
|
||||
void onPublish(const MqttClient* client, const Topic& topic, const char* payload, size_t length)
|
||||
{
|
||||
static_cast<MqttReceiver*>(this)->MqttReceiver::onPublish(client, topic, payload, length);
|
||||
}
|
||||
|
||||
static size_t size() { return routes.size(); }
|
||||
|
||||
static void reset() { routes.clear(); }
|
||||
|
||||
private:
|
||||
|
||||
static void onRoutePublish(const MqttClient* client, const Topic& topic, const char* payload, size_t length)
|
||||
{
|
||||
bool unrouted = true;
|
||||
auto receivers = routes.equal_range(client);
|
||||
for(auto it = receivers.first; it != receivers.second; ++it)
|
||||
{
|
||||
it->second->onPublish(client, topic, payload, length);
|
||||
unrouted = false;
|
||||
}
|
||||
|
||||
if (unrouted and unrouted_handler)
|
||||
{
|
||||
unrouted_handler(client, topic, payload, length);
|
||||
}
|
||||
}
|
||||
|
||||
private:
|
||||
void unregister(MqttClassBinder<MqttReceiver>* which)
|
||||
{
|
||||
if (routes.size()==0) return; // bug in map stl
|
||||
for(auto it=routes.begin(); it!=routes.end(); it++)
|
||||
if (it->second == which)
|
||||
{
|
||||
routes.erase(it);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
static std::multimap<const MqttClient*, MqttClassBinder<MqttReceiver>*> routes;
|
||||
static MqttClient::CallBack unrouted_handler;
|
||||
|
||||
};
|
||||
|
||||
template<class MqttReceiver>
|
||||
std::multimap<const MqttClient*, MqttClassBinder<MqttReceiver>*> MqttClassBinder<MqttReceiver>::routes;
|
||||
|
||||
template<class MqttReceiver>
|
||||
MqttClient::CallBack MqttClassBinder<MqttReceiver>::unrouted_handler = nullptr;
|
||||
|
||||
@@ -1,412 +0,0 @@
|
||||
/* MqttStreaming.h - Fork of Streaming.h adding std::string and with some minor fixes
|
||||
* (I have to speek to the author in order to include my changes to his library if possible)
|
||||
**/
|
||||
|
||||
/*
|
||||
Streaming.h - Arduino library for supporting the << streaming operator
|
||||
Copyright (c) 2010-2012 Mikal Hart. All rights reserved.
|
||||
|
||||
This library is free software; you can redistribute it and/or
|
||||
modify it under the terms of the GNU Lesser General Public
|
||||
License as published by the Free Software Foundation; either
|
||||
version 2.1 of the License, or (at your option) any later version.
|
||||
|
||||
This library is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
Lesser General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU Lesser General Public
|
||||
License along with this library; if not, write to the Free Software
|
||||
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
*/
|
||||
|
||||
/*
|
||||
Version 6 library changes
|
||||
Copyright (c) 2019 Gazoodle. All rights reserved.
|
||||
|
||||
1. _BASED moved to template to remove type conversion to long and
|
||||
sign changes which break int8_t and int16_t negative numbers.
|
||||
The print implementation still upscales to long for it's internal
|
||||
print routine.
|
||||
|
||||
2. _PAD added to allow padding & filling of characters to the stream
|
||||
|
||||
3. _WIDTH & _WIDTHZ added to allow width printing with space padding
|
||||
and zero padding for numerics
|
||||
|
||||
4. Simple _FMT mechanism ala printf, but without the typeunsafetyness
|
||||
and no internal buffers for replaceable stream printing
|
||||
*/
|
||||
|
||||
#ifndef ARDUINO_STREAMING
|
||||
#define ARDUINO_STREAMING
|
||||
|
||||
#if (defined(ARDUINO) && ARDUINO >= 100) || defined(EPOXY_DUINO)
|
||||
#include "Arduino.h"
|
||||
#else
|
||||
#ifndef STREAMING_CONSOLE
|
||||
#include "WProgram.h"
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#include <string>
|
||||
|
||||
#if defined(ARDUINO_ARCH_AVR) || defined(ARDUINO_ARCH_MEGAAVR)
|
||||
// No stl library, so need trivial version of std::is_signed ...
|
||||
namespace std {
|
||||
template<typename T>
|
||||
struct is_signed { static const bool value = false; };
|
||||
template<>
|
||||
struct is_signed<int8_t> { static const bool value = true; };
|
||||
template<>
|
||||
struct is_signed<int16_t> { static const bool value = true; };
|
||||
template<>
|
||||
struct is_signed<int32_t> { static const bool value = true; };
|
||||
};
|
||||
#else
|
||||
#include <type_traits>
|
||||
#endif
|
||||
|
||||
#define STREAMING_LIBRARY_VERSION 6
|
||||
|
||||
#if !defined(typeof)
|
||||
#define typeof(x) __typeof__(x)
|
||||
#endif
|
||||
|
||||
// PrintBuffer implementation of Print, a small buffer to print in
|
||||
// see its use with pad_float()
|
||||
template <size_t N>
|
||||
class PrintBuffer : public Print
|
||||
{
|
||||
size_t pos = 0;
|
||||
char str[N] {};
|
||||
public:
|
||||
inline const char *operator() ()
|
||||
{ return str; };
|
||||
|
||||
// inline void clear()
|
||||
// { pos = 0; str[0] = '\0'; };
|
||||
|
||||
inline size_t write(uint8_t c)
|
||||
{ return write(&c, 1); };
|
||||
|
||||
inline size_t write(const uint8_t *buffer, size_t size)
|
||||
{
|
||||
size_t s = std::min(size, N-1 - pos); // need a /0 left
|
||||
if (s)
|
||||
{
|
||||
memcpy(&str[pos], buffer, s);
|
||||
pos += s;
|
||||
}
|
||||
return s;
|
||||
};
|
||||
};
|
||||
|
||||
// Generic template
|
||||
template<class T>
|
||||
inline Print &operator <<(Print &stream, const T &arg)
|
||||
{ stream.print(arg); return stream; }
|
||||
|
||||
// TODO sfinae maybe could do the trick ?
|
||||
inline Print &operator <<(Print &stream, const std::string &str)
|
||||
{ stream.print(str.c_str()); return stream; }
|
||||
|
||||
template<typename T>
|
||||
struct _BASED
|
||||
{
|
||||
T val;
|
||||
int base;
|
||||
_BASED(T v, int b): val(v), base(b)
|
||||
{}
|
||||
};
|
||||
|
||||
#if ARDUINO >= 100
|
||||
|
||||
struct _BYTE_CODE
|
||||
{
|
||||
byte val;
|
||||
_BYTE_CODE(byte v) : val(v)
|
||||
{}
|
||||
};
|
||||
#define _BYTE(a) _BYTE_CODE(a)
|
||||
|
||||
inline Print &operator <<(Print &obj, const _BYTE_CODE &arg)
|
||||
{ obj.write(arg.val); return obj; }
|
||||
|
||||
#else
|
||||
|
||||
#define _BYTE(a) _BASED<typeof(a)>(a, BYTE)
|
||||
|
||||
#endif
|
||||
|
||||
#define _HEX(a) _BASED<typeof(a)>(a, HEX)
|
||||
#define _DEC(a) _BASED<typeof(a)>(a, DEC)
|
||||
#define _OCT(a) _BASED<typeof(a)>(a, OCT)
|
||||
#define _BIN(a) _BASED<typeof(a)>(a, BIN)
|
||||
|
||||
// Specialization for class _BASED
|
||||
// Thanks to Arduino forum user Ben Combee who suggested this
|
||||
// clever technique to allow for expressions like
|
||||
// Serial << _HEX(a);
|
||||
|
||||
template<typename T>
|
||||
inline Print &operator <<(Print &obj, const _BASED<T> &arg)
|
||||
{ obj.print(arg.val, arg.base); return obj; }
|
||||
|
||||
#if ARDUINO >= 18 || defined(EPOXY_DUINO)
|
||||
// Specialization for class _FLOAT
|
||||
// Thanks to Michael Margolis for suggesting a way
|
||||
// to accommodate Arduino 0018's floating point precision
|
||||
// feature like this:
|
||||
// Serial << _FLOAT(gps_latitude, 6); // 6 digits of precision
|
||||
|
||||
struct _FLOAT
|
||||
{
|
||||
double val; // only Print::print(double)
|
||||
int digits;
|
||||
_FLOAT(double v, int d): val(v), digits(d)
|
||||
{}
|
||||
};
|
||||
|
||||
inline Print &operator <<(Print &obj, const _FLOAT &arg)
|
||||
{ obj.print(arg.val, arg.digits); return obj; }
|
||||
#endif
|
||||
|
||||
// Specialization for enum _EndLineCode
|
||||
// Thanks to Arduino forum user Paul V. who suggested this
|
||||
// clever technique to allow for expressions like
|
||||
// Serial << "Hello!" << endl;
|
||||
|
||||
enum _EndLineCode { endl };
|
||||
|
||||
inline Print &operator <<(Print &obj, _EndLineCode)
|
||||
{ obj.println(); return obj; }
|
||||
|
||||
// Specialization for padding & filling, mainly utilized
|
||||
// by the width printers
|
||||
//
|
||||
// Use like
|
||||
// Serial << _PAD(10,' '); // Will output 10 spaces
|
||||
// Serial << _PAD(4, '0'); // Will output 4 zeros
|
||||
struct _PAD
|
||||
{
|
||||
int8_t width;
|
||||
char chr;
|
||||
_PAD(int8_t w, char c) : width(w), chr(c) {}
|
||||
};
|
||||
|
||||
inline Print &operator <<(Print& stm, const _PAD &arg)
|
||||
{
|
||||
for(int8_t i = 0; i < arg.width; i++)
|
||||
stm.print(arg.chr);
|
||||
return stm;
|
||||
}
|
||||
|
||||
// Specialization for width printing
|
||||
//
|
||||
// Use like Result
|
||||
// -------- ------
|
||||
// Serial << _WIDTH(1,5) " 1"
|
||||
// Serial << _WIDTH(10,5) " 10"
|
||||
// Serial << _WIDTH(100,5) " 100"
|
||||
// Serial << _WIDTHZ(1,5) "00001"
|
||||
//
|
||||
// Great for times & dates, or hex dumps
|
||||
//
|
||||
// Serial << _WIDTHZ(hour,2) << ':' << _WIDTHZ(min,2) << ':' << _WIDTHZ(sec,2)
|
||||
//
|
||||
// for(int index=0; index<byte_array_size; index++)
|
||||
// Serial << _WIDTHZ(_HEX(byte_array[index]))
|
||||
|
||||
template<typename T>
|
||||
struct __WIDTH
|
||||
{
|
||||
const T val;
|
||||
int8_t width;
|
||||
char pad;
|
||||
__WIDTH(const T& v, int8_t w, char p) : val(v), width(w), pad(p) {}
|
||||
};
|
||||
|
||||
// Count digits in an integer of specific base
|
||||
template<typename T>
|
||||
inline uint8_t digits(T v, int8_t base = 10)
|
||||
{
|
||||
uint8_t digits = 0;
|
||||
if ( std::is_signed<T>::value )
|
||||
{
|
||||
if ( v < 0 )
|
||||
{
|
||||
digits++;
|
||||
v = -v; // v needs to be postive for the digits counter to work
|
||||
}
|
||||
}
|
||||
do
|
||||
{
|
||||
v /= base;
|
||||
digits++;
|
||||
} while( v > 0 );
|
||||
return digits;
|
||||
}
|
||||
|
||||
// Generic get the width of a value in base 10
|
||||
template<typename T>
|
||||
inline uint8_t get_value_width(T val)
|
||||
{ return digits(val); }
|
||||
|
||||
inline uint8_t get_value_width(const char * val)
|
||||
{ return strlen(val); }
|
||||
|
||||
#ifdef ARDUINO
|
||||
inline uint8_t get_value_width(const __FlashStringHelper * val)
|
||||
{ return strlen_P(reinterpret_cast<const char *>(val)); }
|
||||
#endif
|
||||
|
||||
// _BASED<T> get the width of a value
|
||||
template<typename T>
|
||||
inline uint8_t get_value_width(_BASED<T> b)
|
||||
{ return digits(b.val, b.base); }
|
||||
|
||||
// Constructor wrapper to allow automatic template parameter deduction
|
||||
template<typename T>
|
||||
__WIDTH<T> _WIDTH(T val, int8_t width) { return __WIDTH<T>(val, width, ' '); }
|
||||
template<typename T>
|
||||
__WIDTH<T> _WIDTHZ(T val, int8_t width) { return __WIDTH<T>(val, width, '0'); }
|
||||
|
||||
|
||||
// Operator overload to handle width printing.
|
||||
template<typename T>
|
||||
inline Print &operator <<(Print &stm, const __WIDTH<T> &arg)
|
||||
{ stm << _PAD(arg.width - get_value_width(arg.val), arg.pad) << arg.val; return stm; }
|
||||
|
||||
// explicit Operator overload to handle width printing of _FLOAT, double and float
|
||||
template<typename T>
|
||||
inline Print &pad_float(Print &stm, const __WIDTH<T> &arg, const double val, const int digits = 2) // see Print::print(double, int = 2)
|
||||
{
|
||||
PrintBuffer<32> buf; // it's only ~45B on the stack, no allocation, leak or fragmentation
|
||||
size_t size = buf.print(val, digits); // print in buf
|
||||
return stm << _PAD(arg.width - size, arg.pad) << buf(); // pad and concat what's in buf
|
||||
}
|
||||
|
||||
inline Print &operator <<(Print &stm, const __WIDTH<float> &arg)
|
||||
{ return pad_float(stm, arg, arg.val); }
|
||||
|
||||
inline Print &operator <<(Print &stm, const __WIDTH<double> &arg)
|
||||
{ return pad_float(stm, arg, arg.val); }
|
||||
|
||||
inline Print &operator <<(Print &stm, const __WIDTH<_FLOAT> &arg)
|
||||
{ auto& f = arg.val; return pad_float(stm, arg, f.val, f.digits); }
|
||||
|
||||
// a less verbose _FLOATW for _WIDTH(_FLOAT)
|
||||
#define _FLOATW(val, digits, width) _WIDTH<_FLOAT>(_FLOAT((val), (digits)), (width))
|
||||
|
||||
// Specialization for replacement formatting
|
||||
//
|
||||
// Designed to be similar to printf that everyone knows and loves/hates. But without
|
||||
// the internal buffers and type agnosticism. This version only has placeholders in
|
||||
// the format string, the actual values are supplied using the stream safe operators
|
||||
// defined in this library.
|
||||
//
|
||||
// Use like this:
|
||||
//
|
||||
// Serial << FMT(F("Replace % with %"), 1, 2 )
|
||||
// Serial << FMT("Time is %:%:%", _WIDTHZ(hours,2), _WIDTHZ(minutes,2), _WIDTHZ(seconds,2))
|
||||
// Serial << FMT("Your score is %\\%", score); // Note the \\ to escape the % sign
|
||||
|
||||
// Ok, hold your hats. This is a foray into C++11's variadic template engine ...
|
||||
|
||||
inline char get_next_format_char(const char *& format_string)
|
||||
{
|
||||
char format_char = *format_string;
|
||||
if ( format_char > 0 ) format_string++;
|
||||
return format_char;
|
||||
}
|
||||
|
||||
#ifdef ARDUINO
|
||||
inline char get_next_format_char(const __FlashStringHelper*& format_string)
|
||||
{
|
||||
char format_char = pgm_read_byte(format_string);
|
||||
if ( format_char > 0 ) format_string = reinterpret_cast<const __FlashStringHelper*>(reinterpret_cast<const char *>(format_string)+1);
|
||||
return format_char;
|
||||
}
|
||||
#endif
|
||||
|
||||
template<typename Ft>
|
||||
inline bool check_backslash(char& format_char, Ft& format_string)
|
||||
{
|
||||
if ( format_char == '\\')
|
||||
{
|
||||
format_char = get_next_format_char(format_string);
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
// The template tail printer helper
|
||||
template<typename Ft, typename... Ts>
|
||||
struct __FMT
|
||||
{
|
||||
Ft format_string;
|
||||
__FMT(Ft f, Ts ... args) : format_string(f) {}
|
||||
inline void tstreamf(Print& stm, Ft format) const
|
||||
{
|
||||
while(char c = get_next_format_char(format))
|
||||
{
|
||||
check_backslash(c, format);
|
||||
if ( c )
|
||||
stm.print(c);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
// The variadic template helper
|
||||
template<typename Ft, typename T, typename... Ts>
|
||||
struct __FMT<Ft, T, Ts...> : __FMT<Ft, Ts...>
|
||||
{
|
||||
T val;
|
||||
__FMT(Ft f, T t, Ts... ts) : __FMT<Ft, Ts...>(f, ts...), val(t) {}
|
||||
inline void tstreamf(Print& stm, Ft format) const
|
||||
{
|
||||
while(char c = get_next_format_char(format))
|
||||
{
|
||||
if (!check_backslash(c, format))
|
||||
{
|
||||
if ( c == '%')
|
||||
{
|
||||
stm << val;
|
||||
// Variadic recursion ... compiler rolls this out during
|
||||
// template argument pack expansion
|
||||
__FMT<Ft, Ts...>::tstreamf(stm, format);
|
||||
return;
|
||||
}
|
||||
}
|
||||
if (c)
|
||||
stm.print(c);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
// The actual operator should you only instanciate the FMT
|
||||
// helper with a format string and no parameters
|
||||
template<typename Ft, typename... Ts>
|
||||
inline Print& operator <<(Print &stm, const __FMT<Ft, Ts...> &args)
|
||||
{
|
||||
args.tstreamf(stm, args.format_string);
|
||||
return stm;
|
||||
}
|
||||
|
||||
// The variadic stream helper
|
||||
template<typename Ft, typename T, typename... Ts>
|
||||
inline Print& operator <<(Print &stm, const __FMT<Ft, T, Ts...> &args)
|
||||
{
|
||||
args.tstreamf(stm, args.format_string);
|
||||
return stm;
|
||||
}
|
||||
|
||||
// As we don't have C++17, we can't get a constructor to use
|
||||
// automatic argument deduction, but ... this little trick gets
|
||||
// around that ...
|
||||
template<typename Ft, typename... Ts>
|
||||
__FMT<Ft, Ts...> _FMT(Ft format, Ts ... args) { return __FMT<Ft, Ts...>(format, args...); }
|
||||
|
||||
#endif
|
||||
@@ -1,3 +1,4 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#pragma once
|
||||
#include <map>
|
||||
#include <string>
|
||||
@@ -28,29 +29,6 @@ class StringIndexer
|
||||
public:
|
||||
using index_t = uint8_t;
|
||||
|
||||
static index_t strToIndex(const char* str, uint8_t len)
|
||||
{
|
||||
for(auto it=strings.begin(); it!=strings.end(); it++)
|
||||
{
|
||||
if (strncmp(it->second.str.c_str(), str, len)==0)
|
||||
{
|
||||
it->second.used++;
|
||||
return it->first;
|
||||
}
|
||||
}
|
||||
for(index_t index=1; index; index++)
|
||||
{
|
||||
if (strings.find(index)==strings.end())
|
||||
{
|
||||
strings[index].str = std::string(str, len);
|
||||
strings[index].used++;
|
||||
// Serial << "Creating index " << index << " for (" << strings[index].str.c_str() << ") len=" << len << endl;
|
||||
return index;
|
||||
}
|
||||
}
|
||||
return 0; // TODO out of indexes
|
||||
}
|
||||
|
||||
static const std::string& str(const index_t& index)
|
||||
{
|
||||
static std::string dummy;
|
||||
@@ -82,6 +60,32 @@ class StringIndexer
|
||||
static uint16_t count() { return strings.size(); }
|
||||
|
||||
private:
|
||||
friend class IndexedString;
|
||||
|
||||
// increment use of str or create a new index
|
||||
static index_t strToIndex(const char* str, uint8_t len)
|
||||
{
|
||||
for(auto it=strings.begin(); it!=strings.end(); it++)
|
||||
{
|
||||
if (it->second.str.length() == len && strcmp(it->second.str.c_str(), str)==0)
|
||||
{
|
||||
it->second.used++;
|
||||
return it->first;
|
||||
}
|
||||
}
|
||||
for(index_t index=1; index; index++)
|
||||
{
|
||||
if (strings.find(index)==strings.end())
|
||||
{
|
||||
strings[index].str = std::string(str, len);
|
||||
strings[index].used++;
|
||||
// Serial << "Creating index " << index << " for (" << strings[index].str.c_str() << ") len=" << len << endl;
|
||||
return index;
|
||||
}
|
||||
}
|
||||
return 0; // TODO out of indexes
|
||||
}
|
||||
|
||||
static std::map<index_t, StringCounter> strings;
|
||||
};
|
||||
|
||||
|
||||
539
src/TinyMqtt.cpp
539
src/TinyMqtt.cpp
@@ -1,146 +1,144 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#include "TinyMqtt.h"
|
||||
#include <sstream>
|
||||
|
||||
void outstring(const char* prefix, const char*p, uint16_t len)
|
||||
{
|
||||
return;
|
||||
Serial << prefix << "='";
|
||||
while(len--) Serial << (char)*p++;
|
||||
Serial << '\'' << endl;
|
||||
}
|
||||
#if TINY_MQTT_DEBUG
|
||||
static auto cyan = TinyConsole::cyan;
|
||||
static auto white = TinyConsole::white;
|
||||
static auto red = TinyConsole::red;
|
||||
static auto yellow = TinyConsole::yellow;
|
||||
|
||||
int TinyMqtt::debug=2;
|
||||
#endif
|
||||
|
||||
#ifdef EPOXY_DUINO
|
||||
std::map<MqttMessage::Type, int> MqttClient::counters;
|
||||
#endif
|
||||
|
||||
MqttBroker::MqttBroker(uint16_t port)
|
||||
{
|
||||
server = new TcpServer(port);
|
||||
#ifdef TCP_ASYNC
|
||||
server = std::unique_ptr<TcpServer>(new TcpServer(port));
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
server->onClient(onClient, this);
|
||||
#endif
|
||||
}
|
||||
|
||||
MqttBroker::~MqttBroker()
|
||||
{
|
||||
while(clients.size())
|
||||
{
|
||||
delete clients[0];
|
||||
}
|
||||
delete server;
|
||||
}
|
||||
|
||||
// private constructor used by broker only
|
||||
MqttClient::MqttClient(MqttBroker* parent, TcpClient* new_client)
|
||||
: parent(parent)
|
||||
MqttClient::MqttClient(MqttBroker* local_broker, TcpClient* new_client)
|
||||
{
|
||||
#ifdef TCP_ASYNC
|
||||
client = new_client;
|
||||
client->onData(onData, this);
|
||||
connect(local_broker);
|
||||
debug("MqttClient private with broker");
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
tcp_client = new_client;
|
||||
tcp_client->onData(onData, this);
|
||||
// client->onConnect() TODO
|
||||
// client->onDisconnect() TODO
|
||||
#else
|
||||
client = new WiFiClient(*new_client);
|
||||
tcp_client.reset(new WiFiClient(*new_client));
|
||||
#endif
|
||||
alive = millis()+5000; // client expires after 5s if no CONNECT msg
|
||||
alive = millis()+5000;
|
||||
}
|
||||
|
||||
MqttClient::MqttClient(MqttBroker* parent, const std::string& id)
|
||||
: parent(parent), clientId(id)
|
||||
MqttClient::MqttClient(MqttBroker* local_broker, const std::string& id)
|
||||
: local_broker(local_broker), clientId(id)
|
||||
{
|
||||
client = nullptr;
|
||||
alive = 0;
|
||||
|
||||
if (parent) parent->addClient(this);
|
||||
if (local_broker) local_broker->addClient(this);
|
||||
}
|
||||
|
||||
MqttClient::~MqttClient()
|
||||
{
|
||||
close();
|
||||
delete client;
|
||||
debug("*** MqttClient delete()");
|
||||
}
|
||||
|
||||
void MqttClient::close(bool bSendDisconnect)
|
||||
{
|
||||
debug("close " << id().c_str());
|
||||
mqtt_connected = false;
|
||||
if (client)
|
||||
mqtt_flags &= ~FlagConnected;
|
||||
if (tcp_client) // connected to a remote broker
|
||||
{
|
||||
if (bSendDisconnect and client->connected())
|
||||
if (bSendDisconnect and tcp_client->connected())
|
||||
{
|
||||
message.create(MqttMessage::Type::Disconnect);
|
||||
message.hexdump("close");
|
||||
message.sendTo(this);
|
||||
}
|
||||
client->stop();
|
||||
tcp_client->stop();
|
||||
}
|
||||
|
||||
if (parent)
|
||||
if (local_broker)
|
||||
{
|
||||
parent->removeClient(this);
|
||||
parent=nullptr;
|
||||
local_broker->removeClient(this);
|
||||
local_broker = nullptr;
|
||||
}
|
||||
}
|
||||
|
||||
void MqttClient::connect(MqttBroker* local)
|
||||
{
|
||||
debug("MqttClient::connect_local");
|
||||
alive = 0;
|
||||
close();
|
||||
local_broker = local;
|
||||
clientAlive();
|
||||
}
|
||||
|
||||
void MqttClient::connect(std::string broker, uint16_t port, uint16_t ka)
|
||||
{
|
||||
debug("cnx: closing");
|
||||
debug("MqttClient::connect_to_host " << broker << ':' << port);
|
||||
keep_alive = ka;
|
||||
close();
|
||||
if (client) delete client;
|
||||
client = new TcpClient;
|
||||
tcp_client.reset(new TcpClient);
|
||||
|
||||
debug("Trying to connect to " << broker.c_str() << ':' << port);
|
||||
#ifdef TCP_ASYNC
|
||||
client->onData(onData, this);
|
||||
client->onConnect(onConnect, this);
|
||||
client->connect(broker.c_str(), port);
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
tcp_client->onData(onData, this);
|
||||
tcp_client->onConnect(onConnect, this);
|
||||
tcp_client->connect(broker.c_str(), port, ka);
|
||||
#else
|
||||
if (client->connect(broker.c_str(), port))
|
||||
if (tcp_client->connect(broker.c_str(), port))
|
||||
{
|
||||
onConnect(this, client);
|
||||
debug("link established");
|
||||
onConnect(this, tcp_client.get());
|
||||
}
|
||||
else
|
||||
{
|
||||
debug("unable to connect.");
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
void MqttBroker::addClient(MqttClient* client)
|
||||
void MqttBroker::addClient(TcpClient* client)
|
||||
{
|
||||
clients.push_back(client);
|
||||
debug("MqttBroker::addClient");
|
||||
clients.insert(std::unique_ptr<MqttClient>(new MqttClient(this, client)));
|
||||
}
|
||||
|
||||
void MqttBroker::connect(const std::string& host, uint16_t port)
|
||||
{
|
||||
if (broker == nullptr) broker = new MqttClient;
|
||||
broker->connect(host, port);
|
||||
broker->parent = this; // Because connect removed the link
|
||||
debug("MqttBroker::connect");
|
||||
if (remote_broker == nullptr) remote_broker = new MqttClient;
|
||||
remote_broker->connect(host, port);
|
||||
remote_broker->local_broker = this; // Because connect removed the link
|
||||
}
|
||||
|
||||
void MqttBroker::removeClient(MqttClient* remove)
|
||||
{
|
||||
for(auto it=clients.begin(); it!=clients.end(); it++)
|
||||
{
|
||||
auto client=*it;
|
||||
if (client==remove)
|
||||
{
|
||||
// TODO if this broker is connected to an external broker
|
||||
// we have to unsubscribe remove's topics.
|
||||
// (but doing this, check that other clients are not subscribed...)
|
||||
// Unless -> we could receive useless messages
|
||||
// -> we are using (memory) one IndexedString plus its string for nothing.
|
||||
debug("Remove " << clients.size());
|
||||
clients.erase(it);
|
||||
debug("Client removed " << clients.size());
|
||||
return;
|
||||
}
|
||||
}
|
||||
debug("Error cannot remove client"); // TODO should not occur
|
||||
local_clients.erase(remove);
|
||||
}
|
||||
|
||||
void MqttBroker::onClient(void* broker_ptr, TcpClient* client)
|
||||
{
|
||||
debug("MqttBroker::onClient");
|
||||
MqttBroker* broker = static_cast<MqttBroker*>(broker_ptr);
|
||||
|
||||
broker->addClient(new MqttClient(broker, client));
|
||||
broker->addClient(client);
|
||||
debug("New client");
|
||||
}
|
||||
|
||||
void MqttBroker::loop()
|
||||
{
|
||||
#ifndef TCP_ASYNC
|
||||
#ifndef TINY_MQTT_ASYNC
|
||||
WiFiClient client = server->available();
|
||||
|
||||
if (client)
|
||||
@@ -148,65 +146,59 @@ void MqttBroker::loop()
|
||||
onClient(this, &client);
|
||||
}
|
||||
#endif
|
||||
if (broker)
|
||||
if (remote_broker)
|
||||
{
|
||||
// TODO should monitor broker's activity.
|
||||
// 1 When broker disconnect and reconnect we have to re-subscribe
|
||||
broker->loop();
|
||||
remote_broker->loop();
|
||||
}
|
||||
|
||||
|
||||
// for(auto it=clients.begin(); it!=clients.end(); it++)
|
||||
// use index because size can change during the loop
|
||||
for(size_t i=0; i<clients.size(); i++)
|
||||
// 200 bytes shorter than for(auto& client: clients) !
|
||||
for(auto it=clients.begin(); it!=clients.end(); it++)
|
||||
{
|
||||
auto client = clients[i];
|
||||
if (client->connected())
|
||||
it->get()->loop();
|
||||
if (not it->get()->connected())
|
||||
{
|
||||
client->loop();
|
||||
}
|
||||
else
|
||||
{
|
||||
debug("Client " << client->id().c_str() << " Disconnected, parent=" << (dbg_ptr)client->parent);
|
||||
// Note: deleting a client not added by the broker itself will probably crash later.
|
||||
delete client;
|
||||
clients.erase(it);
|
||||
break;
|
||||
}
|
||||
}
|
||||
for(const auto& client: local_clients)
|
||||
client->loop();
|
||||
}
|
||||
|
||||
MqttError MqttBroker::subscribe(const Topic& topic, uint8_t qos)
|
||||
{
|
||||
if (broker && broker->connected())
|
||||
debug("MqttBroker::subscribe");
|
||||
if (remote_broker && remote_broker->connected())
|
||||
{
|
||||
return broker->subscribe(topic, qos);
|
||||
return remote_broker->subscribe(topic, qos);
|
||||
}
|
||||
return MqttNowhereToSend;
|
||||
}
|
||||
|
||||
MqttError MqttBroker::publish(const MqttClient* source, const Topic& topic, const MqttMessage& msg) const
|
||||
MqttError MqttBroker::publish(const MqttClient* source, const Topic& topic, MqttMessage& msg) const
|
||||
{
|
||||
MqttError retval = MqttOk;
|
||||
|
||||
debug("publish ");
|
||||
int i=0;
|
||||
for(auto client: clients)
|
||||
debug("MqttBroker::publish");
|
||||
int clt_num = 0;
|
||||
for(auto& client: clients)
|
||||
{
|
||||
i++;
|
||||
#ifdef TINY_MQTT_DEBUG
|
||||
Serial << "brk_" << (broker && broker->connected() ? "con" : "dis") <<
|
||||
" srce=" << (source->isLocal() ? "loc" : "rem") << " clt#" << i << ", local=" << client->isLocal() << ", con=" << client->connected() << endl;
|
||||
#endif
|
||||
debug (" broker:" << (remote_broker && remote_broker->connected() ? "linked" : "alone")
|
||||
<< " srce=" << (source->isLocal() ? "loc" : "rem") << " clt#" << ++clt_num
|
||||
<< ", local=" << client->isLocal() << ", con=" << client->connected());
|
||||
|
||||
bool doit = false;
|
||||
if (broker && broker->connected()) // this (MqttBroker) is connected (to a external broker)
|
||||
if (remote_broker && remote_broker->connected()) // this (MqttBroker) is connected (to a external broker)
|
||||
{
|
||||
// ext_broker -> clients or clients -> ext_broker
|
||||
if (source == broker) // external broker -> internal clients
|
||||
if (source == remote_broker) // external broker -> internal clients
|
||||
doit = true;
|
||||
else // external clients -> this broker
|
||||
{
|
||||
// As this broker is connected to another broker, simply forward the msg
|
||||
MqttError ret = broker->publishIfSubscribed(topic, msg);
|
||||
MqttError ret = remote_broker->publishIfSubscribed(topic, msg);
|
||||
if (ret != MqttOk) retval = ret;
|
||||
}
|
||||
}
|
||||
@@ -214,9 +206,8 @@ MqttError MqttBroker::publish(const MqttClient* source, const Topic& topic, cons
|
||||
{
|
||||
doit = true;
|
||||
}
|
||||
#ifdef TINY_MQTT_DEBUG
|
||||
Serial << ", doit=" << doit << ' ';
|
||||
#endif
|
||||
|
||||
debug(" doit=" << doit << ' ');
|
||||
|
||||
if (doit) retval = client->publishIfSubscribed(topic, msg);
|
||||
debug("");
|
||||
@@ -236,15 +227,16 @@ bool MqttBroker::compareString(
|
||||
|
||||
void MqttMessage::getString(const char* &buff, uint16_t& len)
|
||||
{
|
||||
len = (buff[0]<<8)|(buff[1]);
|
||||
len = getSize(buff);
|
||||
buff+=2;
|
||||
}
|
||||
|
||||
void MqttClient::clientAlive(uint32_t more_seconds)
|
||||
void MqttClient::clientAlive()
|
||||
{
|
||||
debug("MqttClient::clientAlive");
|
||||
if (keep_alive)
|
||||
{
|
||||
alive=millis()+1000*(keep_alive+more_seconds);
|
||||
alive=millis()+1000*(keep_alive+(local_broker ? TINY_MQTT_CLIENT_ALIVE_TOLERANCE : 0));
|
||||
}
|
||||
else
|
||||
alive=0;
|
||||
@@ -252,29 +244,30 @@ void MqttClient::clientAlive(uint32_t more_seconds)
|
||||
|
||||
void MqttClient::loop()
|
||||
{
|
||||
if (alive && (millis() > alive))
|
||||
if (alive && (millis() >= alive))
|
||||
{
|
||||
if (parent)
|
||||
if (local_broker)
|
||||
{
|
||||
debug("timeout client");
|
||||
Serial << "timeout client " << clientId << endl;
|
||||
close();
|
||||
debug("closed");
|
||||
debug(red << "closed");
|
||||
}
|
||||
else if (client && client->connected())
|
||||
else if (tcp_client && tcp_client->connected())
|
||||
{
|
||||
debug("pingreq");
|
||||
uint16_t pingreq = MqttMessage::Type::PingReq;
|
||||
client->write((const char*)(&pingreq), 2);
|
||||
clientAlive(0);
|
||||
|
||||
tcp_client->write((const char*)(&pingreq), 2);
|
||||
clientAlive();
|
||||
|
||||
// TODO when many MqttClient passes through a local broker
|
||||
// there is no need to send one PingReq per instance.
|
||||
}
|
||||
}
|
||||
#ifndef TCP_ASYNC
|
||||
while(client && client->available()>0)
|
||||
#ifndef TINY_MQTT_ASYNC
|
||||
while(tcp_client && tcp_client->available()>0)
|
||||
{
|
||||
message.incoming(client->read());
|
||||
message.incoming(tcp_client->read());
|
||||
if (message.type())
|
||||
{
|
||||
processMessage(&message);
|
||||
@@ -287,24 +280,24 @@ void MqttClient::loop()
|
||||
void MqttClient::onConnect(void *mqttclient_ptr, TcpClient*)
|
||||
{
|
||||
MqttClient* mqtt = static_cast<MqttClient*>(mqttclient_ptr);
|
||||
debug("cnx: connecting");
|
||||
debug("MqttClient::onConnect");
|
||||
MqttMessage msg(MqttMessage::Type::Connect);
|
||||
msg.add("MQTT",4);
|
||||
msg.add(0x4); // Mqtt protocol version 3.1.1
|
||||
msg.add(0x0); // Connect flags TODO user / name
|
||||
|
||||
msg.add(0x00); // keep_alive
|
||||
msg.add((char)mqtt->keep_alive);
|
||||
msg.add((char)(mqtt->keep_alive >> 8)); // keep_alive
|
||||
msg.add((char)(mqtt->keep_alive & 0xFF));
|
||||
msg.add(mqtt->clientId);
|
||||
debug("cnx: mqtt connecting");
|
||||
msg.sendTo(mqtt);
|
||||
msg.reset();
|
||||
debug("cnx: mqtt sent " << (dbg_ptr)mqtt->parent);
|
||||
debug("cnx: mqtt sent " << (dbg_ptr)mqtt->local_broker);
|
||||
|
||||
mqtt->clientAlive(0);
|
||||
mqtt->clientAlive();
|
||||
}
|
||||
|
||||
#ifdef TCP_ASYNC
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
void MqttClient::onData(void* client_ptr, TcpClient*, void* data, size_t len)
|
||||
{
|
||||
char* char_ptr = static_cast<char*>(data);
|
||||
@@ -333,7 +326,7 @@ void MqttClient::resubscribe()
|
||||
msg.add(0);
|
||||
msg.add(0);
|
||||
|
||||
for(auto topic: subscriptions)
|
||||
for(const auto& topic: subscriptions)
|
||||
{
|
||||
msg.add(topic);
|
||||
msg.add(0); // TODO qos
|
||||
@@ -344,29 +337,30 @@ void MqttClient::resubscribe()
|
||||
|
||||
MqttError MqttClient::subscribe(Topic topic, uint8_t qos)
|
||||
{
|
||||
debug("subsribe(" << topic.c_str() << ")");
|
||||
debug("MqttClient::subsribe(" << topic.c_str() << ")");
|
||||
MqttError ret = MqttOk;
|
||||
|
||||
subscriptions.insert(topic);
|
||||
|
||||
if (parent==nullptr) // remote broker
|
||||
if (local_broker==nullptr) // remote broker
|
||||
{
|
||||
return sendTopic(topic, MqttMessage::Type::Subscribe, qos);
|
||||
}
|
||||
else
|
||||
{
|
||||
return parent->subscribe(topic, qos);
|
||||
return local_broker->subscribe(topic, qos);
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
MqttError MqttClient::unsubscribe(Topic topic)
|
||||
{
|
||||
debug("MqttClient::unsubscribe");
|
||||
auto it=subscriptions.find(topic);
|
||||
if (it != subscriptions.end())
|
||||
{
|
||||
subscriptions.erase(it);
|
||||
if (parent==nullptr) // remote broker
|
||||
if (local_broker==nullptr) // remote broker
|
||||
{
|
||||
return sendTopic(topic, MqttMessage::Type::UnSubscribe, 0);
|
||||
}
|
||||
@@ -376,6 +370,7 @@ MqttError MqttClient::unsubscribe(Topic topic)
|
||||
|
||||
MqttError MqttClient::sendTopic(const Topic& topic, MqttMessage::Type type, uint8_t qos)
|
||||
{
|
||||
debug("MqttClient::sendTopic");
|
||||
MqttMessage msg(type, 2);
|
||||
|
||||
// TODO manage packet identifier
|
||||
@@ -389,34 +384,32 @@ MqttError MqttClient::sendTopic(const Topic& topic, MqttMessage::Type type, uint
|
||||
return msg.sendTo(this);
|
||||
}
|
||||
|
||||
long MqttClient::counter=0;
|
||||
|
||||
void MqttClient::processMessage(const MqttMessage* mesg)
|
||||
void MqttClient::processMessage(MqttMessage* mesg)
|
||||
{
|
||||
counter++;
|
||||
#ifdef TINY_MQTT_DEBUG
|
||||
if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::Type::PingResp)
|
||||
{
|
||||
Serial << "---> INCOMING " << _HEX(mesg->type()) << " client(" << (dbg_ptr)client << ':' << clientId << ") mem=" << ESP.getFreeHeap() << endl;
|
||||
// mesg->hexdump("Incoming");
|
||||
}
|
||||
#if TINY_MQTT_DEBUG
|
||||
mesg->hexdump("Incoming");
|
||||
#endif
|
||||
auto header = mesg->getVHeader();
|
||||
const char* payload;
|
||||
uint16_t len;
|
||||
bool bclose=true;
|
||||
|
||||
switch(mesg->type() & 0XF0)
|
||||
#ifdef EPOXY_DUINO
|
||||
counters[mesg->type()]++;
|
||||
#endif
|
||||
|
||||
switch(mesg->type())
|
||||
{
|
||||
case MqttMessage::Type::Connect:
|
||||
if (mqtt_connected)
|
||||
if (mqtt_flags & FlagConnected)
|
||||
{
|
||||
debug("already connected");
|
||||
break;
|
||||
}
|
||||
payload = header+10;
|
||||
mqtt_flags = header[7];
|
||||
keep_alive = (header[8]<<8)|(header[9]);
|
||||
// Todo should check that reserved == 0 (spec)
|
||||
mqtt_flags = header[7] & ~FlagConnected;
|
||||
keep_alive = MqttMessage::getSize(header+8);
|
||||
if (strncmp("MQTT", header+2,4))
|
||||
{
|
||||
debug("bad mqtt header");
|
||||
@@ -436,30 +429,29 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
||||
if (mqtt_flags & FlagWill) // Will topic
|
||||
{
|
||||
mesg->getString(payload, len); // Will Topic
|
||||
outstring("WillTopic", payload, len);
|
||||
payload += len;
|
||||
|
||||
mesg->getString(payload, len); // Will Message
|
||||
outstring("WillMessage", payload, len);
|
||||
payload += len;
|
||||
}
|
||||
// FIXME forgetting credential is allowed (security hole)
|
||||
if (mqtt_flags & FlagUserName)
|
||||
{
|
||||
mesg->getString(payload, len);
|
||||
if (!parent->checkUser(payload, len)) break;
|
||||
if (not local_broker->checkUser(payload, len)) break;
|
||||
payload += len;
|
||||
}
|
||||
if (mqtt_flags & FlagPassword)
|
||||
{
|
||||
mesg->getString(payload, len);
|
||||
if (!parent->checkPassword(payload, len)) break;
|
||||
if (not local_broker->checkPassword(payload, len)) break;
|
||||
payload += len;
|
||||
}
|
||||
|
||||
Serial << "Connected client:" << clientId.c_str() << ", keep alive=" << keep_alive << '.' << endl;
|
||||
debug(yellow << "Client " << clientId << " connected : keep alive=" << keep_alive << '.' << white);
|
||||
|
||||
bclose = false;
|
||||
mqtt_connected=true;
|
||||
mqtt_flags |= FlagConnected;
|
||||
{
|
||||
MqttMessage msg(MqttMessage::Type::ConnAck);
|
||||
msg.add(0); // Session present (not implemented)
|
||||
@@ -469,14 +461,14 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::ConnAck:
|
||||
mqtt_connected = true;
|
||||
mqtt_flags |= FlagConnected;
|
||||
bclose = false;
|
||||
resubscribe();
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::SubAck:
|
||||
case MqttMessage::Type::PubAck:
|
||||
if (!mqtt_connected) break;
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
// Ignore acks
|
||||
bclose = false;
|
||||
break;
|
||||
@@ -487,38 +479,46 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::PingReq:
|
||||
if (!mqtt_connected) break;
|
||||
if (client)
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
if (tcp_client)
|
||||
{
|
||||
uint16_t pingreq = MqttMessage::Type::PingResp;
|
||||
client->write((const char*)(&pingreq), 2);
|
||||
debug(cyan << "Ping response to client ");
|
||||
tcp_client->write((const char*)(&pingreq), 2);
|
||||
bclose = false;
|
||||
}
|
||||
else
|
||||
{
|
||||
debug("internal pingreq ?");
|
||||
debug(red << "internal pingreq ?");
|
||||
}
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::Subscribe:
|
||||
case MqttMessage::Type::UnSubscribe:
|
||||
{
|
||||
if (!mqtt_connected) break;
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
payload = header+2;
|
||||
|
||||
debug("un/subscribe loop");
|
||||
std::string qoss;
|
||||
while(payload < mesg->end())
|
||||
{
|
||||
mesg->getString(payload, len); // Topic
|
||||
debug( " topic (" << std::string(payload, len) << ')');
|
||||
outstring(" un/subscribes", payload, len);
|
||||
// subscribe(Topic(payload, len));
|
||||
Topic topic(payload, len);
|
||||
|
||||
payload += len;
|
||||
if ((mesg->type() & 0XF0) == MqttMessage::Type::Subscribe)
|
||||
if (mesg->type() == MqttMessage::Type::Subscribe)
|
||||
{
|
||||
uint8_t qos = *payload++;
|
||||
if (qos != 0) debug("Unsupported QOS" << qos << endl);
|
||||
if (qos != 0)
|
||||
{
|
||||
debug("Unsupported QOS" << qos << endl);
|
||||
qoss.push_back(0x80);
|
||||
}
|
||||
else
|
||||
qoss.push_back(qos);
|
||||
subscriptions.insert(topic);
|
||||
}
|
||||
else
|
||||
@@ -530,41 +530,54 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
||||
}
|
||||
debug("end loop");
|
||||
bclose = false;
|
||||
// TODO SUBACK
|
||||
|
||||
MqttMessage ack(mesg->type() == MqttMessage::Type::Subscribe ? MqttMessage::Type::SubAck : MqttMessage::Type::UnSuback);
|
||||
ack.add(header[0]);
|
||||
ack.add(header[1]);
|
||||
ack.add(qoss.c_str(), qoss.size(), false);
|
||||
ack.sendTo(this);
|
||||
}
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::UnSuback:
|
||||
if (!mqtt_connected) break;
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
bclose = false;
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::Publish:
|
||||
if (mqtt_connected or client == nullptr)
|
||||
debug("publish " << (mqtt_flags & FlagConnected) << '/' << (long) tcp_client.get());
|
||||
if ((mqtt_flags & FlagConnected) or tcp_client == nullptr)
|
||||
{
|
||||
uint8_t qos = mesg->type() & 0x6;
|
||||
uint8_t qos = mesg->flags();
|
||||
payload = header;
|
||||
mesg->getString(payload, len);
|
||||
Topic published(payload, len);
|
||||
payload += len;
|
||||
// Serial << "Received Publish (" << published.str().c_str() << ") size=" << (int)len
|
||||
debug("Received Publish (" << published.str().c_str() << ") size=" << (int)len);
|
||||
// << '(' << std::string(payload, len).c_str() << ')' << " msglen=" << mesg->length() << endl;
|
||||
if (qos) payload+=2; // ignore packet identifier if any
|
||||
len=mesg->end()-payload;
|
||||
// TODO reset DUP
|
||||
// TODO reset RETAIN
|
||||
|
||||
if (client==nullptr) // internal MqttClient receives publish
|
||||
if (local_broker==nullptr or tcp_client==nullptr) // internal MqttClient receives publish
|
||||
{
|
||||
#if TINY_MQTT_DEBUG
|
||||
if (TinyMqtt::debug >= 2)
|
||||
{
|
||||
Console << (isSubscribedTo(published) ? "not" : "") << " subscribed.\n";
|
||||
Console << "has " << (callback ? "" : "no ") << " callback.\n";
|
||||
}
|
||||
#endif
|
||||
if (callback and isSubscribedTo(published))
|
||||
{
|
||||
callback(this, published, payload, len); // TODO send the real payload
|
||||
}
|
||||
}
|
||||
else if (parent) // from outside to inside
|
||||
else if (local_broker) // from outside to inside
|
||||
{
|
||||
debug("publishing to parent");
|
||||
parent->publish(this, published, *mesg);
|
||||
debug("publishing to local_broker");
|
||||
local_broker->publish(this, published, *mesg);
|
||||
}
|
||||
bclose = false;
|
||||
}
|
||||
@@ -572,8 +585,8 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
||||
|
||||
case MqttMessage::Type::Disconnect:
|
||||
// TODO should discard any will msg
|
||||
if (!mqtt_connected) break;
|
||||
mqtt_connected = false;
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
mqtt_flags &= ~FlagConnected;
|
||||
close(false);
|
||||
bclose=false;
|
||||
break;
|
||||
@@ -584,24 +597,82 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
||||
};
|
||||
if (bclose)
|
||||
{
|
||||
Serial << "*************** Error msg 0x" << _HEX(mesg->type());
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << red << "*************** Error msg 0x" << _HEX(mesg->type());
|
||||
mesg->hexdump("-------ERROR ------");
|
||||
dump();
|
||||
Serial << endl;
|
||||
Console << white << endl;
|
||||
#endif
|
||||
close();
|
||||
}
|
||||
else
|
||||
{
|
||||
clientAlive(parent ? 5 : 0);
|
||||
clientAlive();
|
||||
}
|
||||
}
|
||||
|
||||
bool Topic::matches(const Topic& topic) const
|
||||
{
|
||||
if (getIndex() == topic.getIndex()) return true;
|
||||
if (str() == topic.str()) return true;
|
||||
const char* p1 = c_str();
|
||||
const char* p2 = topic.c_str();
|
||||
|
||||
if (p1 == p2) return true;
|
||||
if (*p2 == '$' and *p1 != '$') return false;
|
||||
|
||||
while(*p1 and *p2)
|
||||
{
|
||||
if (*p1 == '+')
|
||||
{
|
||||
++p1;
|
||||
if (*p1 and *p1!='/') return false;
|
||||
if (*p1) ++p1;
|
||||
while(*p2 and *p2++!='/');
|
||||
}
|
||||
else if (*p1 == '#')
|
||||
{
|
||||
if (*++p1==0) return true;
|
||||
return false;
|
||||
}
|
||||
else if (*p1 == '*')
|
||||
{
|
||||
const char c=*(p1+1);
|
||||
if (c==0) return true;
|
||||
if (c!='/') return false;
|
||||
const char*p = p1+2;
|
||||
while(*p and *p2)
|
||||
{
|
||||
if (*p == *p2)
|
||||
{
|
||||
if (*p==0) return true;
|
||||
if (*p=='/')
|
||||
{
|
||||
p1=p;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
while(*p2 and *p2++!='/');
|
||||
break;
|
||||
}
|
||||
++p;
|
||||
++p2;
|
||||
}
|
||||
if (*p==0) return true;
|
||||
}
|
||||
else if (*p1 == *p2)
|
||||
{
|
||||
++p1;
|
||||
++p2;
|
||||
}
|
||||
else
|
||||
return false;
|
||||
}
|
||||
if (*p1=='/' and p1[1]=='#' and p1[2]==0) return true;
|
||||
return *p1==0 and *p2==0;
|
||||
}
|
||||
|
||||
|
||||
// publish from local client
|
||||
MqttError MqttClient::publish(const Topic& topic, const char* payload, size_t pay_length)
|
||||
@@ -610,29 +681,31 @@ MqttError MqttClient::publish(const Topic& topic, const char* payload, size_t pa
|
||||
msg.add(topic);
|
||||
msg.add(payload, pay_length, false);
|
||||
msg.complete();
|
||||
if (parent)
|
||||
|
||||
if (local_broker)
|
||||
{
|
||||
return parent->publish(this, topic, msg);
|
||||
return local_broker->publish(this, topic, msg);
|
||||
}
|
||||
else if (client)
|
||||
else if (tcp_client)
|
||||
return msg.sendTo(this);
|
||||
else
|
||||
return MqttNowhereToSend;
|
||||
}
|
||||
|
||||
// republish a received publish if it matches any in subscriptions
|
||||
MqttError MqttClient::publishIfSubscribed(const Topic& topic, const MqttMessage& msg)
|
||||
MqttError MqttClient::publishIfSubscribed(const Topic& topic, MqttMessage& msg)
|
||||
{
|
||||
MqttError retval=MqttOk;
|
||||
|
||||
debug("mqttclient publish " << subscriptions.size());
|
||||
debug("mqttclient publishIfSubscribed " << topic.c_str() << ' ' << subscriptions.size());
|
||||
if (isSubscribedTo(topic))
|
||||
{
|
||||
if (client)
|
||||
if (tcp_client)
|
||||
retval = msg.sendTo(this);
|
||||
else
|
||||
{
|
||||
processMessage(&msg);
|
||||
debug("Should call the callback ?");
|
||||
// callback(this, topic, nullptr, 0); // TODO Payload
|
||||
}
|
||||
}
|
||||
@@ -661,29 +734,23 @@ void MqttMessage::incoming(char in_byte)
|
||||
switch(state)
|
||||
{
|
||||
case FixedHeader:
|
||||
size=0;
|
||||
size=MaxBufferLength;
|
||||
state = Length;
|
||||
break;
|
||||
case Length:
|
||||
if (size==0)
|
||||
|
||||
if (size==MaxBufferLength)
|
||||
size = in_byte & 0x7F;
|
||||
else if (size<128)
|
||||
size += static_cast<uint16_t>(in_byte & 0x7F)<<7;
|
||||
else
|
||||
state = Error; // Really don't want to handle msg with length > 16k
|
||||
size += static_cast<uint16_t>(in_byte & 0x7F)<<7;
|
||||
|
||||
if (size > MaxBufferLength)
|
||||
{
|
||||
state = Error;
|
||||
}
|
||||
else if ((in_byte & 0x80) == 0)
|
||||
{
|
||||
vheader = buffer.length();
|
||||
if (size==0)
|
||||
state = Complete;
|
||||
else if (size > 500) // TODO magic
|
||||
{
|
||||
state = Error;
|
||||
}
|
||||
else
|
||||
{
|
||||
buffer.reserve(size);
|
||||
@@ -705,12 +772,14 @@ void MqttMessage::incoming(char in_byte)
|
||||
break;
|
||||
case Complete:
|
||||
default:
|
||||
Serial << "Spurious " << _HEX(in_byte) << endl;
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << red << "Spurious " << _HEX(in_byte) << white << endl;
|
||||
hexdump("spurious");
|
||||
#endif
|
||||
reset();
|
||||
break;
|
||||
}
|
||||
if (buffer.length() > MaxBufferLength) // TODO magic 256 ?
|
||||
if (buffer.length() > MaxBufferLength)
|
||||
{
|
||||
debug("Too long " << state);
|
||||
reset();
|
||||
@@ -721,42 +790,49 @@ void MqttMessage::add(const char* p, size_t len, bool addLength)
|
||||
{
|
||||
if (addLength)
|
||||
{
|
||||
buffer.reserve(buffer.length()+addLength+2);
|
||||
buffer.reserve(buffer.length()+2);
|
||||
incoming(len>>8);
|
||||
incoming(len & 0xFF);
|
||||
}
|
||||
while(len--) incoming(*p++);
|
||||
}
|
||||
|
||||
void MqttMessage::encodeLength(char* msb, int length) const
|
||||
void MqttMessage::encodeLength()
|
||||
{
|
||||
do
|
||||
debug("encodingLength");
|
||||
if (state != Complete)
|
||||
{
|
||||
uint8_t encoded(length & 0x7F);
|
||||
length >>=7;
|
||||
if (length) encoded |= 0x80;
|
||||
*msb++ = encoded;
|
||||
} while (length);
|
||||
};
|
||||
|
||||
void MqttMessage::complete()
|
||||
int length = buffer.size()-3; // 3 = 1 byte for header + 2 bytes for pre-reserved length field.
|
||||
if (length <= 0x7F)
|
||||
{
|
||||
encodeLength(&buffer[1], buffer.size()-2);
|
||||
state = Complete;
|
||||
buffer.erase(1,1);
|
||||
buffer[1] = length;
|
||||
vheader = 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
buffer[1] = 0x80 | (length & 0x7F);
|
||||
buffer[2] = (length >> 7);
|
||||
vheader = 3;
|
||||
}
|
||||
|
||||
MqttError MqttMessage::sendTo(MqttClient* client) const
|
||||
// We could check that buffer[2] < 128 (end of length encoding)
|
||||
state = Complete;
|
||||
}
|
||||
};
|
||||
|
||||
MqttError MqttMessage::sendTo(MqttClient* client)
|
||||
{
|
||||
if (buffer.size())
|
||||
{
|
||||
debug("sending " << buffer.size() << " bytes");
|
||||
encodeLength(&buffer[1], buffer.size()-2);
|
||||
// hexdump("snd");
|
||||
debug(cyan << "sending " << buffer.size() << " bytes to " << client->id());
|
||||
encodeLength();
|
||||
hexdump("Sending ");
|
||||
client->write(&buffer[0], buffer.size());
|
||||
}
|
||||
else
|
||||
{
|
||||
debug("??? Invalid send");
|
||||
debug(red << "??? Invalid send");
|
||||
return MqttInvalidMessage;
|
||||
}
|
||||
return MqttOk;
|
||||
@@ -764,6 +840,34 @@ MqttError MqttMessage::sendTo(MqttClient* client) const
|
||||
|
||||
void MqttMessage::hexdump(const char* prefix) const
|
||||
{
|
||||
(void)prefix;
|
||||
#if TINY_MQTT_DEBUG
|
||||
if (TinyMqtt::debug<2) return;
|
||||
static std::map<Type, std::string> tts={
|
||||
{ Connect, "Connect" },
|
||||
{ ConnAck, "Connack" },
|
||||
{ Publish, "Publish" },
|
||||
{ PubAck, "Puback" },
|
||||
{ Subscribe, "Subscribe" },
|
||||
{ SubAck, "Suback" },
|
||||
{ UnSubscribe, "Unsubscribe" },
|
||||
{ UnSuback, "Unsuback" },
|
||||
{ PingReq, "Pingreq" },
|
||||
{ PingResp, "Pingresp" },
|
||||
{ Disconnect, "Disconnect" }
|
||||
};
|
||||
std::string t("Unknown");
|
||||
Type typ=static_cast<Type>(buffer[0] & 0xF0);
|
||||
if (tts.find(typ) != tts.end())
|
||||
t=tts[typ];
|
||||
Console.fg(cyan);
|
||||
#ifdef NOT_ESP_CORE
|
||||
Console << "---> MESSAGE " << t << ' ' << _HEX(typ) << ' ' << " mem=???" << endl;
|
||||
#else
|
||||
Console << "---> MESSAGE " << t << ' ' << _HEX(typ) << ' ' << " mem=" << ESP.getFreeHeap() << endl;
|
||||
#endif
|
||||
Console.fg(white);
|
||||
|
||||
uint16_t addr=0;
|
||||
const int bytes_per_row = 8;
|
||||
const char* hex_to_str = " | ";
|
||||
@@ -771,19 +875,19 @@ void MqttMessage::hexdump(const char* prefix) const
|
||||
const char* half_sep = " - ";
|
||||
std::string ascii;
|
||||
|
||||
Serial << prefix << " size(" << buffer.size() << "), state=" << state << endl;
|
||||
Console << prefix << " size(" << buffer.size() << "), state=" << state << endl;
|
||||
|
||||
for(const char chr: buffer)
|
||||
{
|
||||
if ((addr % bytes_per_row) == 0)
|
||||
{
|
||||
if (ascii.length()) Serial << hex_to_str << ascii << separator << endl;
|
||||
if (prefix) Serial << prefix << separator;
|
||||
if (ascii.length()) Console << hex_to_str << ascii << separator << endl;
|
||||
if (prefix) Console << prefix << separator;
|
||||
ascii.clear();
|
||||
}
|
||||
addr++;
|
||||
if (chr<16) Serial << '0';
|
||||
Serial << _HEX(chr) << ' ';
|
||||
if (chr<16) Console << '0';
|
||||
Console << _HEX(chr) << ' ';
|
||||
|
||||
ascii += (chr<32 ? '.' : chr);
|
||||
if (ascii.length() == (bytes_per_row/2)) ascii += half_sep;
|
||||
@@ -792,11 +896,12 @@ void MqttMessage::hexdump(const char* prefix) const
|
||||
{
|
||||
while(ascii.length() < bytes_per_row+strlen(half_sep))
|
||||
{
|
||||
Serial << " "; // spaces per hexa byte
|
||||
Console << " "; // spaces per hexa byte
|
||||
ascii += ' ';
|
||||
}
|
||||
Serial << hex_to_str << ascii << separator;
|
||||
Console << hex_to_str << ascii << separator;
|
||||
}
|
||||
|
||||
Serial << endl;
|
||||
Console << endl;
|
||||
#endif
|
||||
}
|
||||
|
||||
220
src/TinyMqtt.h
220
src/TinyMqtt.h
@@ -1,17 +1,26 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#pragma once
|
||||
|
||||
// TODO Should add a AUnit with both TCP_ASYNC and not TCP_ASYNC
|
||||
// #define TCP_ASYNC // Uncomment this to use ESPAsyncTCP instead of normal cnx
|
||||
#ifndef TINY_MQTT_DEBUG
|
||||
#define TINY_MQTT_DEBUG 0
|
||||
#endif
|
||||
#ifndef TINY_MQTT_DEFAULT_ALIVE
|
||||
#define TINY_MQTT_DEFAULT_ALIVE 10
|
||||
#endif
|
||||
#define TINY_MQTT_CLIENT_ALIVE_TOLERANCE 5
|
||||
|
||||
// TODO Should add a AUnit with both TINY_MQTT_ASYNC and not TINY_MQTT_ASYNC
|
||||
// #define TINY_MQTT_ASYNC // Uncomment this to use ESPAsyncTCP instead of normal cnx
|
||||
|
||||
#if defined(ESP8266) || defined(EPOXY_DUINO)
|
||||
#ifdef TCP_ASYNC
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
#include <ESPAsyncTCP.h>
|
||||
#else
|
||||
#include <ESP8266WiFi.h>
|
||||
#endif
|
||||
#elif defined(ESP32)
|
||||
#include <WiFi.h>
|
||||
#ifdef TCP_ASYNC
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
#include <AsyncTCP.h> // https://github.com/me-no-dev/AsyncTCP
|
||||
#endif
|
||||
#endif
|
||||
@@ -20,21 +29,35 @@
|
||||
#else
|
||||
#define dbg_ptr uint32_t
|
||||
#endif
|
||||
#include <vector>
|
||||
|
||||
#ifdef WIO_TERMINAL
|
||||
// Uncommon board handling
|
||||
// If you have a problem with this line, just remove it.
|
||||
// Note: https://github.com/hsaturn/TinyMqtt/issues/41
|
||||
#include <rpcWiFi.h>
|
||||
#endif
|
||||
|
||||
#include <memory>
|
||||
#include <set>
|
||||
#include <string>
|
||||
#include "StringIndexer.h"
|
||||
#include <MqttStreaming.h>
|
||||
|
||||
// #define TINY_MQTT_DEBUG
|
||||
#define TINY_MQTT_DEFAULT_CLIENT_ID "Tiny"
|
||||
|
||||
#ifdef TINY_MQTT_DEBUG
|
||||
#define debug(what) { Serial << __LINE__ << ' ' << what << endl; delay(100); }
|
||||
#include <TinyStreaming.h>
|
||||
#if TINY_MQTT_DEBUG
|
||||
#include <TinyConsole.h> // https://github.com/hsaturn/TinyConsole
|
||||
struct TinyMqtt
|
||||
{
|
||||
static int debug;
|
||||
};
|
||||
|
||||
#define debug(what) { if (TinyMqtt::debug>=1) Console << (int)__LINE__ << ' ' << what << TinyConsole::white << endl; delay(10); }
|
||||
#else
|
||||
#define debug(what) {}
|
||||
#endif
|
||||
|
||||
#ifdef TCP_ASYNC
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
using TcpClient = AsyncClient;
|
||||
using TcpServer = AsyncServer;
|
||||
#else
|
||||
@@ -42,7 +65,7 @@
|
||||
using TcpServer = WiFiServer;
|
||||
#endif
|
||||
|
||||
enum MqttError
|
||||
enum __attribute__((packed)) MqttError
|
||||
{
|
||||
MqttOk = 0,
|
||||
MqttNowhereToSend=1,
|
||||
@@ -64,9 +87,9 @@ class Topic : public IndexedString
|
||||
class MqttClient;
|
||||
class MqttMessage
|
||||
{
|
||||
const uint16_t MaxBufferLength = 4096; //hard limit: 16k
|
||||
const uint16_t MaxBufferLength = 4096; //hard limit: 16k due to size decoding
|
||||
public:
|
||||
enum Type
|
||||
enum __attribute__((packed)) Type
|
||||
{
|
||||
Unknown = 0,
|
||||
Connect = 0x10,
|
||||
@@ -81,7 +104,8 @@ class MqttMessage
|
||||
PingResp = 0xD0,
|
||||
Disconnect = 0xE0
|
||||
};
|
||||
enum State
|
||||
|
||||
enum __attribute__((packed)) State
|
||||
{
|
||||
FixedHeader=0,
|
||||
Length=1,
|
||||
@@ -92,6 +116,11 @@ class MqttMessage
|
||||
Create=6
|
||||
};
|
||||
|
||||
static inline uint32_t getSize(const char* buffer)
|
||||
{
|
||||
const unsigned char* bun = (const unsigned char*)buffer;
|
||||
return (*bun << 8) | bun[1]; }
|
||||
|
||||
MqttMessage() { reset(); }
|
||||
MqttMessage(Type t, uint8_t bits_d3_d0=0) { create(t); buffer[0] |= bits_d3_d0; }
|
||||
void incoming(char byte);
|
||||
@@ -101,8 +130,7 @@ class MqttMessage
|
||||
void add(const Topic& t) { add(t.str()); }
|
||||
const char* end() const { return &buffer[0]+buffer.size(); }
|
||||
const char* getVHeader() const { return &buffer[vheader]; }
|
||||
uint16_t length() const { return buffer.size(); }
|
||||
void complete();
|
||||
void complete() { encodeLength(); }
|
||||
|
||||
void reset();
|
||||
|
||||
@@ -112,24 +140,27 @@ class MqttMessage
|
||||
|
||||
Type type() const
|
||||
{
|
||||
return state == Complete ? static_cast<Type>(buffer[0]) : Unknown;
|
||||
return state == Complete ? static_cast<Type>(buffer[0] & 0xF0) : Unknown;
|
||||
}
|
||||
|
||||
uint8_t flags() const { return static_cast<uint8_t>(buffer[0] & 0x0F); }
|
||||
|
||||
void create(Type type)
|
||||
{
|
||||
buffer=(char)type;
|
||||
buffer+='\0'; // reserved for msg length
|
||||
vheader=2;
|
||||
buffer=(decltype(buffer)::value_type)type;
|
||||
buffer+='\0'; // reserved for msg length byte 1/2
|
||||
buffer+='\0'; // reserved for msg length byte 2/2 (fixed)
|
||||
vheader=3; // Should never change
|
||||
size=0;
|
||||
state=Create;
|
||||
}
|
||||
MqttError sendTo(MqttClient*) const;
|
||||
MqttError sendTo(MqttClient*);
|
||||
void hexdump(const char* prefix=nullptr) const;
|
||||
|
||||
private:
|
||||
void encodeLength(char* msb, int length) const;
|
||||
void encodeLength();
|
||||
|
||||
mutable std::string buffer; // mutable -> sendTo()
|
||||
std::string buffer;
|
||||
uint8_t vheader;
|
||||
uint16_t size; // bytes left to receive
|
||||
State state;
|
||||
@@ -138,8 +169,7 @@ class MqttMessage
|
||||
class MqttBroker;
|
||||
class MqttClient
|
||||
{
|
||||
using CallBack = void (*)(const MqttClient* source, const Topic& topic, const char* payload, size_t payload_length);
|
||||
enum Flags
|
||||
enum __attribute__((packed)) Flags
|
||||
{
|
||||
FlagUserName = 128,
|
||||
FlagPassword = 64,
|
||||
@@ -147,32 +177,51 @@ class MqttClient
|
||||
FlagWillQos = 16 | 8, // unsupported
|
||||
FlagWill = 4, // unsupported
|
||||
FlagCleanSession = 2, // unsupported
|
||||
FlagReserved = 1
|
||||
|
||||
FlagReserved = 1, // use reserved as connected (save 1 byte)
|
||||
FlagConnected = 1
|
||||
};
|
||||
public:
|
||||
/** Constructor. If broker is not null, this is the adress of a local broker.
|
||||
|
||||
using CallBack = void (*)(const MqttClient* source, const Topic& topic, const char* payload, size_t payload_length);
|
||||
|
||||
/** Constructor. Broker is the adress of a local broker if not null
|
||||
If you want to connect elsewhere, leave broker null and use connect() **/
|
||||
MqttClient(MqttBroker* broker = nullptr, const std::string& id="");
|
||||
MqttClient(MqttBroker* broker = nullptr, const std::string& id = TINY_MQTT_DEFAULT_CLIENT_ID);
|
||||
MqttClient(const std::string& id) : MqttClient(nullptr, id){}
|
||||
|
||||
~MqttClient();
|
||||
|
||||
void connect(MqttBroker* parent);
|
||||
void connect(std::string broker, uint16_t port, uint16_t keep_alive = 10);
|
||||
void connect(MqttBroker* local_broker);
|
||||
void connect(std::string broker, uint16_t port, uint16_t keep_alive = TINY_MQTT_DEFAULT_ALIVE);
|
||||
|
||||
// TODO it seems that connected returns true in tcp mode even if
|
||||
// no negociation occurred
|
||||
bool connected()
|
||||
{
|
||||
return (local_broker!=nullptr and tcp_client==nullptr)
|
||||
or (tcp_client and tcp_client->connected());
|
||||
}
|
||||
|
||||
bool connected() { return
|
||||
(parent!=nullptr and client==nullptr) or
|
||||
(client and client->connected()); }
|
||||
void write(const char* buf, size_t length)
|
||||
{ if (client) client->write(buf, length); }
|
||||
{
|
||||
if (tcp_client) tcp_client->write(buf, length);
|
||||
}
|
||||
|
||||
const std::string& id() const { return clientId; }
|
||||
void id(std::string& new_id) { clientId = new_id; }
|
||||
void id(const std::string& new_id) { clientId = new_id; }
|
||||
|
||||
/** Should be called in main loop() */
|
||||
void loop();
|
||||
void close(bool bSendDisconnect=true);
|
||||
void setCallback(CallBack fun) {callback=fun; };
|
||||
void setCallback(CallBack fun)
|
||||
{
|
||||
callback=fun;
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << TinyConsole::magenta << "Callback set to " << (long)fun << TinyConsole::white << endl;
|
||||
if (callback) callback(this, "test/topic", "value", 5);
|
||||
#endif
|
||||
};
|
||||
|
||||
// Publish from client to the world
|
||||
MqttError publish(const Topic&, const char* payload, size_t pay_length);
|
||||
@@ -187,60 +236,74 @@ class MqttClient
|
||||
|
||||
// connected to local broker
|
||||
// TODO seems to be useless
|
||||
bool isLocal() const { return client == nullptr; }
|
||||
bool isLocal() const { return tcp_client == nullptr; }
|
||||
|
||||
void dump(std::string indent="")
|
||||
{
|
||||
(void)indent;
|
||||
#if TINY_MQTT_DEBUG
|
||||
uint32_t ms=millis();
|
||||
Serial << indent << "+-- " << '\'' << clientId.c_str() << "' " << (connected() ? " ON " : " OFF");
|
||||
Serial << ", alive=" << alive << '/' << ms << ", ka=" << keep_alive << ' ';
|
||||
Serial << (client && client->connected() ? "" : "dis") << "connected";
|
||||
Console << indent << "+-- " << '\'' << clientId.c_str() << "' " << (connected() ? " ON " : " OFF");
|
||||
Console << ", alive=" << alive << '/' << ms << ", ka=" << keep_alive << ' ';
|
||||
if (tcp_client)
|
||||
{
|
||||
if (tcp_client->connected())
|
||||
Console << TinyConsole::green << "connected";
|
||||
else
|
||||
Console << TinyConsole::red << "disconnected";
|
||||
Console << TinyConsole::white;
|
||||
}
|
||||
if (subscriptions.size())
|
||||
{
|
||||
bool c = false;
|
||||
Serial << " [";
|
||||
Console << " [";
|
||||
for(auto s: subscriptions)
|
||||
{
|
||||
if (c) Serial << ", ";
|
||||
Serial << s.str().c_str();
|
||||
if (c) Console << ", ";
|
||||
Console << s.str().c_str();
|
||||
c=true;
|
||||
}
|
||||
Serial << ']';
|
||||
Console << ']';
|
||||
}
|
||||
Serial << endl;
|
||||
Console << TinyConsole::erase_to_end << endl;
|
||||
#endif
|
||||
}
|
||||
|
||||
/** Count the number of messages that have been sent **/
|
||||
static long counter;
|
||||
#ifdef EPOXY_DUINO
|
||||
static std::map<MqttMessage::Type, int> counters; // Number of processed messages
|
||||
#endif
|
||||
uint32_t keepAlive() const { return keep_alive; }
|
||||
|
||||
private:
|
||||
|
||||
// event when tcp/ip link established (real or fake)
|
||||
static void onConnect(void * client_ptr, TcpClient*);
|
||||
#ifdef TCP_ASYNC
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
static void onData(void* client_ptr, TcpClient*, void* data, size_t len);
|
||||
#endif
|
||||
MqttError sendTopic(const Topic& topic, MqttMessage::Type type, uint8_t qos);
|
||||
void resubscribe();
|
||||
|
||||
friend class MqttBroker;
|
||||
MqttClient(MqttBroker* parent, TcpClient* client);
|
||||
MqttClient(MqttBroker* local_broker, TcpClient* client);
|
||||
// republish a received publish if topic matches any in subscriptions
|
||||
MqttError publishIfSubscribed(const Topic& topic, const MqttMessage& msg);
|
||||
MqttError publishIfSubscribed(const Topic& topic, MqttMessage& msg);
|
||||
|
||||
void clientAlive(uint32_t more_seconds);
|
||||
void processMessage(const MqttMessage* message);
|
||||
void clientAlive();
|
||||
void processMessage(MqttMessage* message);
|
||||
|
||||
bool mqtt_connected = false;
|
||||
char mqtt_flags;
|
||||
uint32_t keep_alive = 60;
|
||||
uint32_t alive;
|
||||
char mqtt_flags = 0;
|
||||
uint16_t keep_alive = 30;
|
||||
// for client connected to remote broker, PingReq is sent when millis() >= alive
|
||||
// for a client managed by a broker, disconnect it if millis() >= alive
|
||||
uint32_t alive; // PingReq if millis() > alive,
|
||||
MqttMessage message;
|
||||
|
||||
// TODO having a pointer on MqttBroker may produce larger binaries
|
||||
// due to unecessary function linked if ever parent is not used
|
||||
// (this is the case when MqttBroker isn't used except here)
|
||||
MqttBroker* parent=nullptr; // connection to local broker
|
||||
// connection to local broker, or link to the parent
|
||||
// when MqttBroker uses MqttClient for each external connexion
|
||||
MqttBroker* local_broker=nullptr;
|
||||
|
||||
TcpClient* client=nullptr; // connection to mqtt client or to remote broker
|
||||
std::unique_ptr<TcpClient> tcp_client; // connection to remote broker
|
||||
std::set<Topic> subscriptions;
|
||||
std::string clientId;
|
||||
CallBack callback = nullptr;
|
||||
@@ -248,7 +311,7 @@ class MqttClient
|
||||
|
||||
class MqttBroker
|
||||
{
|
||||
enum State
|
||||
enum __attribute__((packed)) State
|
||||
{
|
||||
Disconnected, // Also the initial state
|
||||
Connecting, // connect and sends a fake publish to avoid circular cnx
|
||||
@@ -257,7 +320,6 @@ class MqttBroker
|
||||
public:
|
||||
// TODO limit max number of clients
|
||||
MqttBroker(uint16_t port);
|
||||
~MqttBroker();
|
||||
|
||||
void begin() { server->begin(); }
|
||||
void loop();
|
||||
@@ -265,14 +327,22 @@ class MqttBroker
|
||||
void connect(const std::string& host, uint16_t port=1883);
|
||||
bool connected() const { return state == Connected; }
|
||||
|
||||
size_t clientsCount() const { return clients.size(); }
|
||||
|
||||
void dump(std::string indent="")
|
||||
{
|
||||
for(auto client: clients)
|
||||
for(const auto& client: clients)
|
||||
client->dump(indent);
|
||||
}
|
||||
|
||||
using Clients = std::set<std::unique_ptr<MqttClient>>;
|
||||
using LocalClients = std::set<MqttClient*>;
|
||||
|
||||
const Clients& getClients() const { return clients; }
|
||||
const LocalClients& getLocalClients() const { return local_clients; }
|
||||
|
||||
size_t clientsCount() const { return clients.size(); }
|
||||
size_t localClientsCount() const { return local_clients.size(); }
|
||||
|
||||
private:
|
||||
friend class MqttClient;
|
||||
|
||||
@@ -284,21 +354,25 @@ class MqttBroker
|
||||
{ return compareString(auth_password, password, len); }
|
||||
|
||||
|
||||
MqttError publish(const MqttClient* source, const Topic& topic, const MqttMessage& msg) const;
|
||||
MqttError publish(const MqttClient* source, const Topic& topic, MqttMessage& msg) const;
|
||||
|
||||
MqttError subscribe(const Topic& topic, uint8_t qos);
|
||||
|
||||
// For clients that are added not by the broker itself
|
||||
void addClient(MqttClient* client);
|
||||
void removeClient(MqttClient* client);
|
||||
void addClient(MqttClient* local) { local_clients.insert(local); }
|
||||
void addClient(TcpClient* client);
|
||||
|
||||
void removeClient(MqttClient* local);
|
||||
|
||||
bool compareString(const char* good, const char* str, uint8_t str_len) const;
|
||||
std::vector<MqttClient*> clients;
|
||||
TcpServer* server;
|
||||
Clients clients;
|
||||
LocalClients local_clients;
|
||||
|
||||
private:
|
||||
std::unique_ptr<TcpServer> server;
|
||||
|
||||
const char* auth_user = "guest";
|
||||
const char* auth_password = "guest";
|
||||
State state = Disconnected;
|
||||
MqttClient* remote_broker = nullptr;
|
||||
|
||||
MqttClient* broker = nullptr;
|
||||
State state = Disconnected;
|
||||
};
|
||||
|
||||
142
src/make_unique.inc
Normal file
142
src/make_unique.inc
Normal file
@@ -0,0 +1,142 @@
|
||||
// Implementation of C++14's make_unique for C++11 compilers.
|
||||
//
|
||||
// This has been tested with:
|
||||
// - MSVC 11.0 (Visual Studio 2012)
|
||||
// - gcc 4.6.3
|
||||
// - Xcode 4.4 (with clang "4.0")
|
||||
//
|
||||
// It is based off an implementation proposed by Stephan T. Lavavej for
|
||||
// inclusion in the C++14 standard:
|
||||
// http://isocpp.org/files/papers/N3656.txt
|
||||
// Where appropriate, it borrows the use of MSVC's _VARIADIC_EXPAND_0X macro
|
||||
// machinery to compensate for lack of variadic templates.
|
||||
//
|
||||
// This file injects make_unique into the std namespace, which I acknowledge is
|
||||
// technically forbidden ([C++11: 17.6.4.2.2.1/1]), but is necessary in order
|
||||
// to have syntax compatibility with C++14.
|
||||
//
|
||||
// I perform compiler version checking for MSVC, gcc, and clang to ensure that
|
||||
// we don't add make_unique if it is already there (instead, we include
|
||||
// <memory> to get the compiler-provided one). You can override the compiler
|
||||
// version checking by defining the symbol COMPILER_SUPPORTS_MAKE_UNIQUE.
|
||||
//
|
||||
//
|
||||
// ===============================================================================
|
||||
// This file is released into the public domain. See LICENCE for more information.
|
||||
// ===============================================================================
|
||||
|
||||
#pragma once
|
||||
|
||||
// If user hasn't specified COMPILER_SUPPORTS_MAKE_UNIQUE then try to figure out
|
||||
// based on compiler version if std::make_unique is provided.
|
||||
#if !defined(COMPILER_SUPPORTS_MAKE_UNIQUE)
|
||||
#if defined(_MSC_VER)
|
||||
// std::make_unique was added in MSVC 12.0
|
||||
#if _MSC_VER >= 1800 // MSVC 12.0 (Visual Studio 2013)
|
||||
#define COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
#endif
|
||||
#elif defined(__clang__)
|
||||
// std::make_unique was added in clang 3.4, but not until Xcode 6.
|
||||
// Annoyingly, Apple makes the clang version defines match the version
|
||||
// of Xcode, not the version of clang.
|
||||
#define CLANG_VERSION (__clang_major__ * 10000 + __clang_minor__ * 100 + __clang_patchlevel__)
|
||||
#if defined(__APPLE__) && CLANG_VERSION >= 60000
|
||||
#define COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
#elif !defined(__APPLE__) && CLANG_VERSION >= 30400
|
||||
#define COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
#endif
|
||||
#elif defined(__GNUC__)
|
||||
// std::make_unique was added in gcc 4.9, for standards versions greater
|
||||
// than -std=c++11.
|
||||
#define GCC_VERSION (__GNUC__ * 10000 + __GNUC_MINOR__ * 100 + __GNUC_PATCHLEVEL__)
|
||||
#if GCC_VERSION >= 40900 && __cplusplus > 201103L
|
||||
#define COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
#endif
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#if defined(COMPILER_SUPPORTS_MAKE_UNIQUE)
|
||||
|
||||
// If the compiler supports std::make_unique, then pull in <memory> to get it.
|
||||
#include <memory>
|
||||
|
||||
#else
|
||||
|
||||
// Otherwise, the compiler doesn't provide it, so implement it ourselves.
|
||||
|
||||
#include <cstddef>
|
||||
#include <memory>
|
||||
#include <type_traits>
|
||||
#include <utility>
|
||||
|
||||
namespace std {
|
||||
|
||||
template<class _Ty> struct _Unique_if {
|
||||
typedef unique_ptr<_Ty> _Single_object;
|
||||
};
|
||||
|
||||
template<class _Ty> struct _Unique_if<_Ty[]> {
|
||||
typedef unique_ptr<_Ty[]> _Unknown_bound;
|
||||
};
|
||||
|
||||
template<class _Ty, size_t N> struct _Unique_if<_Ty[N]> {
|
||||
typedef void _Known_bound;
|
||||
};
|
||||
|
||||
//
|
||||
// template< class T, class... Args >
|
||||
// unique_ptr<T> make_unique( Args&&... args);
|
||||
//
|
||||
|
||||
#if defined(_MSC_VER) && (_MSC_VER < 1800)
|
||||
|
||||
// Macro machinery because MSVC 11.0 doesn't support variadic templates.
|
||||
// The _VARIADIC_EXPAND_0X stuff is defined in <xstddef>
|
||||
#define _MAKE_UNIQUE( \
|
||||
TEMPLATE_LIST, PADDING_LIST, LIST, COMMA, X1, X2, X3, X4) \
|
||||
template<class _Ty COMMA LIST(_CLASS_TYPE)> inline \
|
||||
typename _Unique_if<_Ty>::_Single_object make_unique(LIST(_TYPE_REFREF_ARG)) \
|
||||
{ \
|
||||
return unique_ptr<_Ty>(new _Ty(LIST(_FORWARD_ARG))); \
|
||||
} \
|
||||
|
||||
_VARIADIC_EXPAND_0X(_MAKE_UNIQUE, , , , )
|
||||
#undef _MAKE_UNIQUE
|
||||
|
||||
#else // not MSVC 11.0 or earlier
|
||||
|
||||
template<class _Ty, class... Args>
|
||||
typename _Unique_if<_Ty>::_Single_object
|
||||
make_unique(Args&&... args) {
|
||||
return unique_ptr<_Ty>(new _Ty(std::forward<Args>(args)...));
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
// template< class T >
|
||||
// unique_ptr<T> make_unique( std::size_t size );
|
||||
|
||||
template<class _Ty>
|
||||
typename _Unique_if<_Ty>::_Unknown_bound
|
||||
make_unique(size_t n) {
|
||||
typedef typename remove_extent<_Ty>::type U;
|
||||
return unique_ptr<_Ty>(new U[n]());
|
||||
}
|
||||
|
||||
// template< class T, class... Args >
|
||||
// /* unspecified */ make_unique( Args&&... args ) = delete;
|
||||
|
||||
// MSVC 11.0 doesn't support deleted functions, so the best we can do
|
||||
// is simply not define the function.
|
||||
#if !(defined(_MSC_VER) && (_MSC_VER < 1800))
|
||||
|
||||
template<class T, class... Args>
|
||||
typename _Unique_if<T>::_Known_bound
|
||||
make_unique(Args&&...) = delete;
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace std
|
||||
|
||||
#endif // !COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
|
||||
@@ -1,2 +0,0 @@
|
||||
const char *ssid = "YOUR-SSID-HERE";
|
||||
const char *password = "YOUR-PASSWORD-HERE";
|
||||
@@ -1,20 +1,30 @@
|
||||
SUB=n
|
||||
|
||||
tests:
|
||||
set -e; \
|
||||
for i in *-tests/Makefile; do \
|
||||
for i in ${SUB}*-tests/Makefile; do \
|
||||
echo '==== Making:' $$(dirname $$i); \
|
||||
$(MAKE) -C $$(dirname $$i) -j; \
|
||||
done
|
||||
|
||||
runtests: tests
|
||||
debugtest:
|
||||
set -e; \
|
||||
for i in *-tests/Makefile; do \
|
||||
$(MAKE) clean; \
|
||||
$(MAKE) -C debug-mode -j; \
|
||||
debug-mode/debug-tests.out
|
||||
|
||||
runtests: debugtest
|
||||
$(MAKE) clean
|
||||
$(MAKE) tests
|
||||
set -e; \
|
||||
for i in ${SUB}*-tests/Makefile; do \
|
||||
echo '==== Running:' $$(dirname $$i); \
|
||||
$$(dirname $$i)/$$(dirname $$i).out; \
|
||||
done
|
||||
|
||||
clean:
|
||||
set -e; \
|
||||
for i in *-tests/Makefile; do \
|
||||
for i in ${SUB}*-tests/Makefile; do \
|
||||
echo '==== Cleaning:' $$(dirname $$i); \
|
||||
$(MAKE) -C $$(dirname $$i) clean; \
|
||||
done
|
||||
|
||||
13
tests/classbind-tests/Makefile
Normal file
13
tests/classbind-tests/Makefile
Normal file
@@ -0,0 +1,13 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0 -DTINY_MQTT_TESTS
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
|
||||
APP_NAME := classbind-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
351
tests/classbind-tests/classbind-tests.ino
Normal file
351
tests/classbind-tests/classbind-tests.ino
Normal file
@@ -0,0 +1,351 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#include <Arduino.h>
|
||||
#include <AUnit.h>
|
||||
#include <TinyMqtt.h>
|
||||
#include <MqttClassBinder.h>
|
||||
#include <map>
|
||||
#include <iostream>
|
||||
#include <iomanip>
|
||||
#include <sstream>
|
||||
#include <string>
|
||||
#include <iostream>
|
||||
|
||||
// --------------------- CUT HERE - MQTT MESSAGE ROUTER FILE ----------------------------
|
||||
|
||||
class TestReceiver : public MqttClassBinder<TestReceiver>
|
||||
{
|
||||
public:
|
||||
TestReceiver(const char* name) : MqttClassBinder(), name_(name) {}
|
||||
|
||||
void onPublish(const MqttClient* /* source */, const Topic& topic, const char* payload, size_t /* length */)
|
||||
{
|
||||
Serial << "--> routed message received by " << name_ << ':' << topic.c_str() << " = " << payload << endl;
|
||||
messages[name_]++;
|
||||
}
|
||||
|
||||
private:
|
||||
const std::string name_;
|
||||
|
||||
public:
|
||||
static std::map<std::string, int> messages;
|
||||
};
|
||||
|
||||
std::map<std::string, int> TestReceiver::messages;
|
||||
|
||||
static int unrouted = 0;
|
||||
void onUnrouted(const MqttClient*, const Topic& topic, const char*, size_t)
|
||||
{
|
||||
Serial << "--> unrouted: " << topic.c_str() << endl;
|
||||
unrouted++;
|
||||
}
|
||||
|
||||
|
||||
static std::string topic="sensor/temperature";
|
||||
|
||||
/**
|
||||
* TinyMqtt network unit tests.
|
||||
*
|
||||
* No wifi connection unit tests.
|
||||
* Checks with a local broker. Clients must connect to the local broker
|
||||
**/
|
||||
|
||||
// if ascii_pos = 0, no ascii dump, else ascii dump starts after column ascii_pos
|
||||
std::string bufferToHexa(const uint8_t* buffer, size_t length, char sep = 0, size_t ascii_pos = 0)
|
||||
{
|
||||
std::stringstream out;
|
||||
std::string ascii;
|
||||
std::string h("0123456789ABCDEF");
|
||||
for(size_t i=0; i<length; i++)
|
||||
{
|
||||
uint8_t c = buffer[i];
|
||||
out << h[ c >> 4] << h[ c & 0x0F ];
|
||||
if (sep) out << sep;
|
||||
if (ascii_pos)
|
||||
{
|
||||
if (c>=32)
|
||||
ascii += c;
|
||||
else
|
||||
ascii +='.';
|
||||
}
|
||||
}
|
||||
std::string ret(out.str());
|
||||
if (ascii_pos)
|
||||
{
|
||||
while(ret.length() < ascii_pos)
|
||||
ret += ' ';
|
||||
ret +='[' + ascii + ']';
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
void dumpMqttMessage(const uint8_t* buffer, size_t length)
|
||||
{
|
||||
std::map<int, std::string> pkt =
|
||||
{ { MqttMessage::Unknown , "Unknown " },
|
||||
{ MqttMessage::Connect , "Connect " },
|
||||
{ MqttMessage::ConnAck , "ConnAck " },
|
||||
{ MqttMessage::Publish , "Publish " },
|
||||
{ MqttMessage::PubAck , "PubAck " },
|
||||
{ MqttMessage::Subscribe , "Subscribe " },
|
||||
{ MqttMessage::SubAck , "SubAck " },
|
||||
{ MqttMessage::UnSubscribe , "Unsubscribe " },
|
||||
{ MqttMessage::UnSuback , "UnSubAck " },
|
||||
{ MqttMessage::PingReq , "PingReq " },
|
||||
{ MqttMessage::PingResp , "PingResp " },
|
||||
{ MqttMessage::Disconnect , "Disconnect " } };
|
||||
|
||||
std::cout << " | data sent " << std::setw(3) << length << " : ";
|
||||
auto it = pkt.find(buffer[0] & 0xF0);
|
||||
if (it == pkt.end())
|
||||
std::cout << pkt[MqttMessage::Unknown];
|
||||
else
|
||||
std::cout << it->second;
|
||||
|
||||
std::cout << bufferToHexa(buffer, length, ' ', 60) << std::endl;
|
||||
}
|
||||
|
||||
String toString(const IPAddress& ip)
|
||||
{
|
||||
return String(ip[0])+'.'+String(ip[1])+'.'+String(ip[2])+'.'+String(ip[3]);
|
||||
}
|
||||
|
||||
MqttBroker broker(1883);
|
||||
|
||||
void reset_and_start_servers(int n, bool early_accept = true)
|
||||
{
|
||||
MqttClassBinder<TestReceiver>::reset();
|
||||
TestReceiver::messages.clear();
|
||||
unrouted = 0;
|
||||
|
||||
ESP8266WiFiClass::resetInstances();
|
||||
ESP8266WiFiClass::earlyAccept = early_accept;
|
||||
while(n)
|
||||
{
|
||||
ESP8266WiFiClass::selectInstance(n--);
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin("fake_ssid", "fake_pwd");
|
||||
}
|
||||
}
|
||||
|
||||
test(classbind_one_client_receives_the_message)
|
||||
{
|
||||
reset_and_start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress ip_broker = WiFi.localIP();
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
assertTrue(client.connected());
|
||||
|
||||
TestReceiver receiver("receiver");
|
||||
MqttClassBinder<TestReceiver>::onPublish(&client, &receiver);
|
||||
|
||||
client.subscribe("a/b");
|
||||
client.publish("a/b", "ab");
|
||||
|
||||
for (int i =0; i<10; i++)
|
||||
{
|
||||
client.loop();
|
||||
broker.loop();
|
||||
}
|
||||
|
||||
assertEqual(TestReceiver::messages["receiver"], 1);
|
||||
assertEqual(unrouted, 0);
|
||||
}
|
||||
|
||||
test(classbind_routes_should_be_empty_when_receiver_goes_out_of_scope)
|
||||
{
|
||||
reset_and_start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress ip_broker = WiFi.localIP();
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
assertTrue(client.connected());
|
||||
|
||||
// Make a receiver going out of scope
|
||||
{
|
||||
TestReceiver receiver("receiver");
|
||||
MqttClassBinder<TestReceiver>::onPublish(&client, &receiver);
|
||||
assertEqual(MqttClassBinder<TestReceiver>::size(), (size_t)1);
|
||||
}
|
||||
|
||||
client.subscribe("a/b");
|
||||
client.publish("a/b", "ab");
|
||||
|
||||
for (int i =0; i<10; i++)
|
||||
{
|
||||
client.loop();
|
||||
broker.loop();
|
||||
}
|
||||
|
||||
assertEqual(TestReceiver::messages["receiver"], 0);
|
||||
assertEqual(MqttClassBinder<TestReceiver>::size(), (size_t)0);
|
||||
}
|
||||
|
||||
test(classbind_publish_should_be_dispatched_to_many_receivers)
|
||||
{
|
||||
reset_and_start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress ip_broker = WiFi.localIP();
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
assertTrue(client.connected());
|
||||
|
||||
TestReceiver receiver_1("receiver_1");
|
||||
TestReceiver receiver_2("receiver_2");
|
||||
|
||||
MqttClassBinder<TestReceiver>::onPublish(&client, &receiver_1);
|
||||
MqttClassBinder<TestReceiver>::onPublish(&client, &receiver_2);
|
||||
client.subscribe("a/b");
|
||||
client.publish("a/b", "ab");
|
||||
|
||||
for (int i =0; i<10; i++)
|
||||
{
|
||||
client.loop();
|
||||
broker.loop();
|
||||
}
|
||||
|
||||
assertEqual(TestReceiver::messages["receiver_1"], 1);
|
||||
assertEqual(TestReceiver::messages["receiver_2"], 1);
|
||||
}
|
||||
|
||||
test(classbind_register_to_many_clients)
|
||||
{
|
||||
reset_and_start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress ip_broker = WiFi.localIP();
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client_1;
|
||||
client_1.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
|
||||
MqttClient client_2;
|
||||
client_2.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
|
||||
assertTrue(client_1.connected());
|
||||
assertTrue(client_2.connected());
|
||||
|
||||
TestReceiver receiver("receiver");
|
||||
|
||||
MqttClassBinder<TestReceiver>::onPublish(&client_1, &receiver);
|
||||
MqttClassBinder<TestReceiver>::onPublish(&client_2, &receiver);
|
||||
|
||||
auto loop = [&client_1, &client_2, &broker]()
|
||||
{
|
||||
client_1.loop();
|
||||
client_2.loop();
|
||||
broker.loop();
|
||||
};
|
||||
|
||||
client_1.subscribe("a/b");
|
||||
client_2.subscribe("a/b");
|
||||
|
||||
// Ensure subscribptions are passed
|
||||
for (int i =0; i<5; i++) loop();
|
||||
|
||||
client_1.publish("a/b", "from 1");
|
||||
client_2.publish("a/b", "from 2");
|
||||
|
||||
// Ensure publishes are processed
|
||||
for (int i =0; i<5; i++) loop();
|
||||
|
||||
assertEqual(TestReceiver::messages["receiver"], 4);
|
||||
}
|
||||
|
||||
test(classbind_unrouted_fallback)
|
||||
{
|
||||
reset_and_start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress ip_broker = WiFi.localIP();
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
|
||||
assertTrue(client.connected());
|
||||
|
||||
MqttClassBinder<TestReceiver>::onUnpublished(onUnrouted);
|
||||
{
|
||||
TestReceiver receiver("receiver");
|
||||
MqttClassBinder<TestReceiver>::onPublish(&client, &receiver);
|
||||
}
|
||||
|
||||
client.subscribe("a/b");
|
||||
client.publish("a/b", "from 2");
|
||||
|
||||
// Ensure subscribptions are passed
|
||||
for (int i =0; i<5; i++)
|
||||
{
|
||||
client.loop();
|
||||
broker.loop();
|
||||
}
|
||||
|
||||
assertEqual(TestReceiver::messages["receiver"], 0);
|
||||
assertEqual(unrouted, 1);
|
||||
}
|
||||
|
||||
test(classbind_should_cleanup_when_MqttClient_dies)
|
||||
{
|
||||
reset_and_start_servers(2, true);
|
||||
TestReceiver receiver("receiver");
|
||||
|
||||
{
|
||||
MqttClient client;
|
||||
|
||||
MqttClassBinder<TestReceiver>::onPublish(&client, &receiver);
|
||||
assertEqual(MqttClassBinder<TestReceiver>::size(), (size_t)1);
|
||||
}
|
||||
assertEqual(MqttClassBinder<TestReceiver>::size(), (size_t)1);
|
||||
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// setup() and loop()
|
||||
void setup() {
|
||||
/* delay(1000);
|
||||
Serial.begin(115200);
|
||||
while(!Serial);
|
||||
*/
|
||||
|
||||
Serial.println("=============[ FAKE NETWORK TinyMqtt TESTS ]========================");
|
||||
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin("network", "password");
|
||||
}
|
||||
|
||||
void loop() {
|
||||
aunit::TestRunner::run();
|
||||
|
||||
if (Serial.available()) ESP.reset();
|
||||
}
|
||||
13
tests/debug-mode/Makefile
Normal file
13
tests/debug-mode/Makefile
Normal file
@@ -0,0 +1,13 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0 -DTINY_MQTT_DEBUG
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
|
||||
APP_NAME := debug-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
15
tests/debug-mode/debug-tests.ino
Normal file
15
tests/debug-mode/debug-tests.ino
Normal file
@@ -0,0 +1,15 @@
|
||||
#include <Arduino.h>
|
||||
#include <AUnit.h>
|
||||
#include <TinyMqtt.h>
|
||||
#include <map>
|
||||
#include <iostream>
|
||||
#include <iomanip>
|
||||
#include <sstream>
|
||||
#include <string>
|
||||
|
||||
// Only compilation check, so do nothing
|
||||
|
||||
void setup() {}
|
||||
void loop() {
|
||||
aunit::TestRunner::run();
|
||||
}
|
||||
7
tests/howto
Normal file
7
tests/howto
Normal file
@@ -0,0 +1,7 @@
|
||||
cd TinyMqtt/tests/../..
|
||||
git clone https://github.com/hsaturn/EspMock.git
|
||||
git clone https://github.com/bxparks/AUnit.git
|
||||
git clone https://github.com/bxparks/EpoxyDuino.git
|
||||
cd TinyMqtt/tests
|
||||
make
|
||||
make runtests
|
||||
@@ -1,8 +1,10 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
APP_NAME := length-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
|
||||
@@ -1,3 +1,4 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#include <AUnit.h>
|
||||
#include <TinyMqtt.h>
|
||||
#include <map>
|
||||
|
||||
@@ -1,8 +1,13 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0 -DTINY_MQTT_DEFAULT_ALIVE=1
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
|
||||
APP_NAME := local-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
|
||||
@@ -14,7 +14,6 @@
|
||||
|
||||
using namespace std;
|
||||
|
||||
MqttBroker broker(1883);
|
||||
|
||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
||||
|
||||
@@ -31,31 +30,48 @@ void onPublish(const MqttClient* srce, const Topic& topic, const char* payload,
|
||||
|
||||
test(local_client_should_unregister_when_destroyed)
|
||||
{
|
||||
return;
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
MqttBroker broker(1883);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
{
|
||||
MqttClient client;
|
||||
assertEqual(broker.clientsCount(), (size_t)0); // Ensure client is not yet connected
|
||||
client.connect("127.0.0.1", 1883);
|
||||
assertEqual(broker.clientsCount(), (size_t)1); // Ensure client is now connected
|
||||
assertEqual(broker.localClientsCount(), (size_t)0); // Ensure client is not yet connected
|
||||
MqttClient client(&broker);
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is now connected
|
||||
}
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
}
|
||||
|
||||
test(local_client_alive)
|
||||
{
|
||||
set_millis(0);
|
||||
MqttBroker broker(1883);
|
||||
MqttClient client(&broker);
|
||||
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is now connected
|
||||
|
||||
add_millis(TINY_MQTT_DEFAULT_ALIVE*1000/2);
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is still connected
|
||||
|
||||
add_seconds(TINY_MQTT_DEFAULT_ALIVE*5);
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is still connected
|
||||
}
|
||||
|
||||
#if 0
|
||||
test(local_connect)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient client;
|
||||
assertTrue(client.connected());
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
assertEqual(broker.localClientsCount(), (size_t)1);
|
||||
}
|
||||
|
||||
test(local_publish_should_be_dispatched)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber;
|
||||
subscriber.subscribe("a/b");
|
||||
@@ -68,14 +84,14 @@ test(local_publish_should_be_dispatched)
|
||||
publisher.publish("a/c");
|
||||
|
||||
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
||||
assertTrue(published[""]["a/b"] == 1);
|
||||
assertTrue(published[""]["a/c"] == 2);
|
||||
assertEqual(published[""]["a/b"], 1);
|
||||
assertEqual(published[""]["a/c"], 2);
|
||||
}
|
||||
|
||||
test(local_publish_should_be_dispatched_to_local_clients)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber_a("A");
|
||||
subscriber_a.setCallback(onPublish);
|
||||
@@ -91,16 +107,16 @@ test(local_publish_should_be_dispatched_to_local_clients)
|
||||
publisher.publish("a/c");
|
||||
|
||||
assertEqual(published.size(), (size_t)2); // 2 clients have received something
|
||||
assertTrue(published["A"]["a/b"] == 1);
|
||||
assertTrue(published["A"]["a/c"] == 1);
|
||||
assertTrue(published["B"]["a/b"] == 1);
|
||||
assertTrue(published["B"]["a/c"] == 0);
|
||||
assertEqual(published["A"]["a/b"], 1);
|
||||
assertEqual(published["A"]["a/c"], 1);
|
||||
assertEqual(published["B"]["a/b"], 1);
|
||||
assertEqual(published["B"]["a/c"], 0);
|
||||
}
|
||||
|
||||
test(local_unsubscribe)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber;
|
||||
subscriber.setCallback(onPublish);
|
||||
@@ -114,13 +130,13 @@ test(local_unsubscribe)
|
||||
publisher.publish("a/b");
|
||||
publisher.publish("a/b");
|
||||
|
||||
assertTrue(published[""]["a/b"] == 1); // Only one publish has been received
|
||||
assertEqual(published[""]["a/b"], 1); // Only one publish has been received
|
||||
}
|
||||
|
||||
test(local_nocallback_when_destroyed)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient publisher;
|
||||
{
|
||||
@@ -143,7 +159,7 @@ void setup() {
|
||||
Serial.begin(115200);
|
||||
while(!Serial);
|
||||
|
||||
Serial.println("=============[ NO WIFI CONNECTION TinyMqtt TESTS ]========================");
|
||||
Serial.println("=============[ LOCAL TinyMqtt TESTS ]========================");
|
||||
}
|
||||
|
||||
void loop() {
|
||||
|
||||
13
tests/network-tests/Makefile
Normal file
13
tests/network-tests/Makefile
Normal file
@@ -0,0 +1,13 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
|
||||
APP_NAME := network-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
514
tests/network-tests/network-tests.ino
Normal file
514
tests/network-tests/network-tests.ino
Normal file
@@ -0,0 +1,514 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#include <Arduino.h>
|
||||
#include <AUnit.h>
|
||||
#include <TinyMqtt.h>
|
||||
#include <map>
|
||||
#include <iostream>
|
||||
#include <iomanip>
|
||||
#include <sstream>
|
||||
#include <string>
|
||||
#include <iostream>
|
||||
|
||||
/**
|
||||
* TinyMqtt network unit tests.
|
||||
*
|
||||
* No wifi connection unit tests.
|
||||
* Checks with a local broker. Clients must connect to the local broker
|
||||
**/
|
||||
|
||||
// if ascii_pos = 0, no ascii dump, else ascii dump starts after column ascii_pos
|
||||
std::string bufferToHexa(const uint8_t* buffer, size_t length, char sep = 0, size_t ascii_pos = 0)
|
||||
{
|
||||
std::stringstream out;
|
||||
std::string ascii;
|
||||
std::string h("0123456789ABCDEF");
|
||||
for(size_t i=0; i<length; i++)
|
||||
{
|
||||
uint8_t c = buffer[i];
|
||||
out << h[ c >> 4] << h[ c & 0x0F ];
|
||||
if (sep) out << sep;
|
||||
if (ascii_pos)
|
||||
{
|
||||
if (c>=32)
|
||||
ascii += c;
|
||||
else
|
||||
ascii +='.';
|
||||
}
|
||||
}
|
||||
std::string ret(out.str());
|
||||
if (ascii_pos)
|
||||
{
|
||||
while(ret.length() < ascii_pos)
|
||||
ret += ' ';
|
||||
ret +='[' + ascii + ']';
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
void dumpMqttMessage(const uint8_t* buffer, size_t length)
|
||||
{
|
||||
std::map<int, std::string> pkt =
|
||||
{ { MqttMessage::Unknown , "Unknown " },
|
||||
{ MqttMessage::Connect , "Connect " },
|
||||
{ MqttMessage::ConnAck , "ConnAck " },
|
||||
{ MqttMessage::Publish , "Publish " },
|
||||
{ MqttMessage::PubAck , "PubAck " },
|
||||
{ MqttMessage::Subscribe , "Subscribe " },
|
||||
{ MqttMessage::SubAck , "SubAck " },
|
||||
{ MqttMessage::UnSubscribe , "Unsubscribe " },
|
||||
{ MqttMessage::UnSuback , "UnSubAck " },
|
||||
{ MqttMessage::PingReq , "PingReq " },
|
||||
{ MqttMessage::PingResp , "PingResp " },
|
||||
{ MqttMessage::Disconnect , "Disconnect " } };
|
||||
|
||||
std::cout << " | data sent " << std::setw(3) << length << " : ";
|
||||
auto it = pkt.find(buffer[0] & 0xF0);
|
||||
if (it == pkt.end())
|
||||
std::cout << pkt[MqttMessage::Unknown];
|
||||
else
|
||||
std::cout << it->second;
|
||||
|
||||
std::cout << bufferToHexa(buffer, length, ' ', 60) << std::endl;
|
||||
}
|
||||
|
||||
String toString(const IPAddress& ip)
|
||||
{
|
||||
return String(ip[0])+'.'+String(ip[1])+'.'+String(ip[2])+'.'+String(ip[3]);
|
||||
}
|
||||
|
||||
MqttBroker broker(1883);
|
||||
|
||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
||||
|
||||
char* lastPayload = nullptr;
|
||||
size_t lastLength;
|
||||
|
||||
void start_servers(int n, bool early_accept = true)
|
||||
{
|
||||
ESP8266WiFiClass::resetInstances();
|
||||
ESP8266WiFiClass::earlyAccept = early_accept;
|
||||
while(n)
|
||||
{
|
||||
ESP8266WiFiClass::selectInstance(n--);
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin("fake_ssid", "fake_pwd");
|
||||
}
|
||||
}
|
||||
|
||||
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
||||
{
|
||||
if (srce)
|
||||
published[srce->id()][topic]++;
|
||||
|
||||
if (lastPayload) free(lastPayload);
|
||||
lastPayload = strdup(payload);
|
||||
lastLength = length;
|
||||
}
|
||||
|
||||
test(network_single_broker_begin)
|
||||
{
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
|
||||
// TODO Nothing is tested here !
|
||||
}
|
||||
|
||||
test(suback)
|
||||
{
|
||||
start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress broker_ip = WiFi.localIP();
|
||||
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(broker_ip.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
|
||||
assertTrue(broker.clientsCount() == 1);
|
||||
assertTrue(client.connected());
|
||||
|
||||
MqttClient::counters[MqttMessage::Type::SubAck] = 0;
|
||||
client.subscribe("a/b");
|
||||
|
||||
// TODO how to avoid these loops ???
|
||||
broker.loop();
|
||||
client.loop();
|
||||
|
||||
assertEqual(MqttClient::counters[MqttMessage::Type::SubAck], 1);
|
||||
}
|
||||
|
||||
uint32_t getClientKeepAlive(MqttBroker& broker)
|
||||
{
|
||||
if (broker.getClients().size() == 1)
|
||||
for (auto& it : broker.getClients())
|
||||
return it->keepAlive();
|
||||
|
||||
return 9999;
|
||||
}
|
||||
|
||||
test(network_client_alive)
|
||||
{
|
||||
const uint32_t keep_alive=1;
|
||||
start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
set_millis(0); // Enter simulated time
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress broker_ip = WiFi.localIP();
|
||||
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(broker_ip.toString().c_str(), 1883, keep_alive);
|
||||
broker.loop();
|
||||
client.loop();
|
||||
|
||||
assertTrue(broker.clientsCount() == 1);
|
||||
assertTrue(client.connected());
|
||||
|
||||
uint32_t ka = getClientKeepAlive(broker);
|
||||
assertEqual(ka, keep_alive);
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
|
||||
// All is going well if we call client.loop()
|
||||
// The client is able to send PingReq to the broker
|
||||
add_seconds(keep_alive);
|
||||
client.loop();
|
||||
broker.loop();
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
|
||||
// Now simulate that the client is frozen for
|
||||
// a too long time
|
||||
add_seconds(TINY_MQTT_CLIENT_ALIVE_TOLERANCE*2);
|
||||
broker.loop();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
set_real_time();
|
||||
}
|
||||
|
||||
test(network_client_keep_alive_high)
|
||||
{
|
||||
const uint32_t keep_alive=1000;
|
||||
start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress broker_ip = WiFi.localIP();
|
||||
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(broker_ip.toString().c_str(), 1883, keep_alive);
|
||||
broker.loop();
|
||||
|
||||
assertTrue(broker.clientsCount() == 1);
|
||||
assertTrue(client.connected());
|
||||
|
||||
MqttClient::counters[MqttMessage::Type::SubAck] = 0;
|
||||
client.subscribe("a/b");
|
||||
|
||||
// TODO how to avoid these loops ???
|
||||
broker.loop();
|
||||
client.loop();
|
||||
|
||||
assertEqual(MqttClient::counters[MqttMessage::Type::SubAck], 1);
|
||||
|
||||
uint32_t sz = broker.getClients().size();
|
||||
assertEqual(sz , (uint32_t)1);
|
||||
|
||||
uint32_t ka = getClientKeepAlive(broker);
|
||||
assertEqual(ka, keep_alive);
|
||||
}
|
||||
|
||||
test(network_client_to_broker_connexion)
|
||||
{
|
||||
start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress broker_ip = WiFi.localIP();
|
||||
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(broker_ip.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
|
||||
assertTrue(broker.clientsCount() == 1);
|
||||
assertTrue(client.connected());
|
||||
}
|
||||
|
||||
test(network_one_client_one_broker_publish_and_subscribe_through_network)
|
||||
{
|
||||
start_servers(2, true);
|
||||
published.clear();
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress ip_broker = WiFi.localIP();
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
assertTrue(client.connected());
|
||||
|
||||
client.setCallback(onPublish);
|
||||
client.subscribe("a/b");
|
||||
client.publish("a/b", "ab");
|
||||
|
||||
for (int i =0; i<2; i++)
|
||||
{
|
||||
client.loop();
|
||||
broker.loop();
|
||||
}
|
||||
|
||||
assertEqual(published.size(), (size_t)1);
|
||||
assertEqual((int)lastLength, (int)2); // sizeof(ab)
|
||||
}
|
||||
|
||||
test(network_one_client_one_broker_hudge_publish_and_subscribe_through_network)
|
||||
{
|
||||
start_servers(2, true);
|
||||
published.clear();
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress ip_broker = WiFi.localIP();
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
assertTrue(client.connected());
|
||||
|
||||
std::string sent;
|
||||
|
||||
for(int i=0; i<200; i++)
|
||||
sent += char('0'+i%10);
|
||||
|
||||
client.setCallback(onPublish);
|
||||
client.subscribe("a/b");
|
||||
client.publish("a/b", sent.c_str());
|
||||
|
||||
for (int i =0; i<2; i++)
|
||||
{
|
||||
client.loop();
|
||||
broker.loop();
|
||||
}
|
||||
|
||||
assertEqual(published.size(), (size_t)1);
|
||||
assertEqual((unsigned int)lastLength, (unsigned int)sent.size());
|
||||
}
|
||||
|
||||
test(network_client_should_unregister_when_destroyed)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
{
|
||||
MqttClient client(&broker);
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
}
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
}
|
||||
|
||||
|
||||
// THESE TESTS ARE IN LOCAL MODE
|
||||
// WE HAVE TO CONVERT THEM TO WIFI MODE (pass through virtual TCP link)
|
||||
|
||||
test(network_connect)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient client(&broker);
|
||||
assertTrue(client.connected());
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
}
|
||||
|
||||
test(network_publish_should_be_dispatched)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.subscribe("a/b");
|
||||
subscriber.subscribe("a/c");
|
||||
subscriber.setCallback(onPublish);
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("a/b");
|
||||
publisher.publish("a/c");
|
||||
publisher.publish("a/c");
|
||||
|
||||
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
||||
assertEqual(published[TINY_MQTT_DEFAULT_CLIENT_ID]["a/b"], 1);
|
||||
assertEqual(published[TINY_MQTT_DEFAULT_CLIENT_ID]["a/c"], 2);
|
||||
}
|
||||
|
||||
test(network_publish_should_be_dispatched_to_clients)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber_a(&broker, "A");
|
||||
subscriber_a.setCallback(onPublish);
|
||||
subscriber_a.subscribe("a/b");
|
||||
subscriber_a.subscribe("a/c");
|
||||
|
||||
MqttClient subscriber_b(&broker, "B");
|
||||
subscriber_b.setCallback(onPublish);
|
||||
subscriber_b.subscribe("a/b");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("a/b"); // A and B should receive this
|
||||
publisher.publish("a/c"); // A should receive this
|
||||
|
||||
assertEqual(published.size(), (size_t)2); // 2 clients have received something
|
||||
assertEqual(published["A"]["a/b"], 1);
|
||||
assertEqual(published["A"]["a/c"], 1);
|
||||
assertEqual(published["B"]["a/b"], 1);
|
||||
assertEqual(published["B"]["a/c"], 0);
|
||||
}
|
||||
|
||||
test(network_unsubscribe)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("a/b");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("a/b"); // This publish is received
|
||||
|
||||
subscriber.unsubscribe("a/b");
|
||||
|
||||
publisher.publish("a/b"); // Those one, no (unsubscribed)
|
||||
publisher.publish("a/b");
|
||||
|
||||
assertEqual(published[TINY_MQTT_DEFAULT_CLIENT_ID]["a/b"], 1); // Only one publish has been received
|
||||
}
|
||||
|
||||
test(network_nocallback_when_destroyed)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
|
||||
{
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("a/b");
|
||||
publisher.publish("a/b");
|
||||
}
|
||||
|
||||
publisher.publish("a/b");
|
||||
|
||||
assertEqual(published.size(), (size_t)1); // Only one publish has been received
|
||||
}
|
||||
|
||||
test(network_small_payload)
|
||||
{
|
||||
published.clear();
|
||||
|
||||
const char* payload="abcd";
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("a/b");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("a/b", payload, strlen(payload)); // This publish is received
|
||||
|
||||
// coming from MqttClient::publish(...)
|
||||
assertEqual(payload, lastPayload);
|
||||
assertEqual(lastLength, (size_t)4);
|
||||
}
|
||||
|
||||
test(network_hudge_payload)
|
||||
{
|
||||
const char* payload="This payload is hudge, just because its length exceeds 127. Thus when encoding length, we have to encode it on two bytes at min. This should not prevent the message from being encoded and decoded successfully !";
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("a/b"); // Note -> this does not send any byte .... (nowhere to send)
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("a/b", payload); // This publish is received
|
||||
|
||||
// onPublish should have filled lastPayload and lastLength
|
||||
assertEqual(payload, lastPayload);
|
||||
assertEqual(lastLength, strlen(payload));
|
||||
assertEqual(strcmp(payload, lastPayload), 0);
|
||||
}
|
||||
|
||||
test(connack)
|
||||
{
|
||||
const bool view = false;
|
||||
|
||||
NetworkObserver check(
|
||||
[this](const WiFiClient*, const uint8_t* buffer, size_t length)
|
||||
{
|
||||
if (view) dumpMqttMessage(buffer, length);
|
||||
if (buffer[0] == MqttMessage::ConnAck)
|
||||
{
|
||||
std::string hex = bufferToHexa(buffer, length);
|
||||
assertStringCaseEqual(hex.c_str(), "20020000");
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress broker_ip = WiFi.localIP();
|
||||
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(broker_ip.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
|
||||
assertTrue(broker.clientsCount() == 1);
|
||||
assertTrue(client.connected());
|
||||
|
||||
MqttClient::counters[MqttMessage::Type::SubAck] = 0;
|
||||
client.subscribe("a/b");
|
||||
|
||||
// TODO how to avoid these loops ???
|
||||
broker.loop();
|
||||
client.loop();
|
||||
|
||||
assertEqual(MqttClient::counters[MqttMessage::Type::SubAck], 1);
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// setup() and loop()
|
||||
void setup() {
|
||||
/* delay(1000);
|
||||
Serial.begin(115200);
|
||||
while(!Serial);
|
||||
*/
|
||||
|
||||
Serial.println("=============[ FAKE NETWORK TinyMqtt TESTS ]========================");
|
||||
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin("network", "password");
|
||||
}
|
||||
|
||||
void loop() {
|
||||
aunit::TestRunner::run();
|
||||
|
||||
if (Serial.available()) ESP.reset();
|
||||
}
|
||||
@@ -1,8 +1,13 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
|
||||
APP_NAME := nowifi-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
|
||||
@@ -1,3 +1,4 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#include <Arduino.h>
|
||||
#include <AUnit.h>
|
||||
#include <TinyMqtt.h>
|
||||
@@ -64,8 +65,8 @@ test(nowifi_publish_should_be_dispatched)
|
||||
publisher.publish("a/c");
|
||||
|
||||
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
||||
assertEqual(published[""]["a/b"], 1);
|
||||
assertEqual(published[""]["a/c"], 2);
|
||||
assertEqual(published[TINY_MQTT_DEFAULT_CLIENT_ID]["a/b"], 1);
|
||||
assertEqual(published[TINY_MQTT_DEFAULT_CLIENT_ID]["a/c"], 2);
|
||||
}
|
||||
|
||||
test(nowifi_publish_should_be_dispatched_to_clients)
|
||||
@@ -93,6 +94,104 @@ test(nowifi_publish_should_be_dispatched_to_clients)
|
||||
assertEqual(published["B"]["a/c"], 0);
|
||||
}
|
||||
|
||||
test(nowifi_subscribe_with_star_wildcard)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("house/*/temp");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("house/bedroom/temp");
|
||||
publisher.publish("house/kitchen/temp");
|
||||
publisher.publish("house/living_room/tv/temp");
|
||||
publisher.publish("building/location1/bedroom/temp");
|
||||
|
||||
assertEqual(published["A"]["house/bedroom/temp"], 1);
|
||||
assertEqual(published["A"]["house/kitchen/temp"], 1);
|
||||
assertEqual(published["A"]["house/living_room/tv/temp"], 1);
|
||||
assertEqual(published["A"]["building/location1/bedroom/temp"], 0);
|
||||
}
|
||||
|
||||
test(nowifi_subscribe_with_plus_wildcard)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("house/+/temp");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("house/bedroom/temp");
|
||||
publisher.publish("house/kitchen/temp");
|
||||
publisher.publish("house/living_room/tv/temp");
|
||||
publisher.publish("building/location1/bedroom/temp");
|
||||
|
||||
assertEqual(published["A"]["house/bedroom/temp"], 1);
|
||||
assertEqual(published["A"]["house/kitchen/temp"], 1);
|
||||
assertEqual(published["A"]["house/living_room/tv/temp"], 0);
|
||||
assertEqual(published["A"]["building/location1/bedroom/temp"], 0);
|
||||
}
|
||||
|
||||
test(nowifi_should_not_receive_sys_msg)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("+/data");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("$SYS/data");
|
||||
|
||||
assertEqual(published["A"]["$SYS/data"], 0);
|
||||
}
|
||||
|
||||
test(nowifi_subscribe_with_mixed_wildcards)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("+/data/#");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("node1/data/update");
|
||||
publisher.publish("node2/data/delta");
|
||||
publisher.publish("node3/data");
|
||||
|
||||
assertEqual(published["A"]["node1/data/update"], 1);
|
||||
assertEqual(published["A"]["node2/data/delta"], 1);
|
||||
assertEqual(published["A"]["node3/data"], 1);
|
||||
}
|
||||
|
||||
test(nowifi_unsubscribe_with_wildcards)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("one/two/+");
|
||||
subscriber.subscribe("one/two/three");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("one/two/three");
|
||||
publisher.publish("one/two/four");
|
||||
|
||||
subscriber.unsubscribe("one/two/+");
|
||||
publisher.publish("one/two/five");
|
||||
|
||||
assertEqual(published["A"]["one/two/three"], 1);
|
||||
assertEqual(published["A"]["one/two/four"], 1);
|
||||
assertEqual(published["A"]["one/two/five"], 0);
|
||||
}
|
||||
|
||||
test(nowifi_unsubscribe)
|
||||
{
|
||||
published.clear();
|
||||
@@ -110,7 +209,7 @@ test(nowifi_unsubscribe)
|
||||
publisher.publish("a/b"); // Those one, no (unsubscribed)
|
||||
publisher.publish("a/b");
|
||||
|
||||
assertEqual(published[""]["a/b"], 1); // Only one publish has been received
|
||||
assertEqual(published[TINY_MQTT_DEFAULT_CLIENT_ID]["a/b"], 1); // Only one publish has been received
|
||||
}
|
||||
|
||||
test(nowifi_nocallback_when_destroyed)
|
||||
@@ -132,7 +231,7 @@ test(nowifi_nocallback_when_destroyed)
|
||||
assertEqual(published.size(), (size_t)1); // Only one publish has been received
|
||||
}
|
||||
|
||||
test(nowifi_payload_nullptr)
|
||||
test(nowifi_small_payload)
|
||||
{
|
||||
published.clear();
|
||||
|
||||
@@ -150,6 +249,22 @@ test(nowifi_payload_nullptr)
|
||||
assertEqual(lastLength, (size_t)4);
|
||||
}
|
||||
|
||||
test(nowifi_hudge_payload)
|
||||
{
|
||||
const char* payload="This payload is hudge, just because its length exceeds 127. Thus when encoding length, we have to encode it on two bytes at min. This should not prevent the message from being encoded and decoded successfully !";
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("a/b");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("a/b", payload); // This publish is received
|
||||
|
||||
// onPublish should have filled lastPayload and lastLength
|
||||
assertEqual(payload, lastPayload);
|
||||
assertEqual(lastLength, strlen(payload));
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// setup() and loop()
|
||||
void setup() {
|
||||
|
||||
@@ -1,8 +1,10 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
APP_NAME := string-indexer-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsync
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsync TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
|
||||
@@ -1,3 +1,4 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#include <Arduino.h>
|
||||
#include <AUnit.h>
|
||||
#include <StringIndexer.h>
|
||||
@@ -62,6 +63,14 @@ test(indexer_same_strings_should_equal)
|
||||
assertTrue(one == two);
|
||||
}
|
||||
|
||||
test(indexer_compare_strings_with_same_beginning)
|
||||
{
|
||||
IndexedString two("one_two");
|
||||
IndexedString one("one");
|
||||
|
||||
assertNotEqual(one.getIndex(), two.getIndex());
|
||||
}
|
||||
|
||||
test(indexer_indexed_operator_eq)
|
||||
{
|
||||
IndexedString one("one");
|
||||
|
||||
10
tests/topic-tests/Makefile
Normal file
10
tests/topic-tests/Makefile
Normal file
@@ -0,0 +1,10 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
APP_NAME := topic-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsync TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
86
tests/topic-tests/topic-tests.ino
Normal file
86
tests/topic-tests/topic-tests.ino
Normal file
@@ -0,0 +1,86 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#include <Arduino.h>
|
||||
#include <AUnit.h>
|
||||
#include <TinyMqtt.h>
|
||||
#include <map>
|
||||
#include <iostream>
|
||||
|
||||
#define endl "\n"
|
||||
|
||||
/**
|
||||
* TinyMqtt / StringIndexer unit tests.
|
||||
*
|
||||
**/
|
||||
|
||||
using namespace std;
|
||||
|
||||
bool testTopicMatch(const char* a, const char* b, bool expected)
|
||||
{
|
||||
Topic ta(a);
|
||||
Topic tb(b);
|
||||
bool match(ta.matches(tb));
|
||||
cout << " " << ta.c_str() << ' ';
|
||||
if (match != expected)
|
||||
cout << (expected ? " should match " : " should not match ");
|
||||
else
|
||||
cout << (expected ? " matches " : " unmatches ");
|
||||
cout << tb.c_str() << endl;
|
||||
return expected == match;
|
||||
}
|
||||
|
||||
test(topic_matches)
|
||||
{
|
||||
// matching
|
||||
assertTrue(testTopicMatch("a/b/c" , "a/b/c" , true));
|
||||
assertTrue(testTopicMatch("a/*/c" , "a/xyz/c" , true));
|
||||
assertTrue(testTopicMatch("a/*/e" , "a/b/c/d/e" , true));
|
||||
assertTrue(testTopicMatch("a/*" , "a/b/c/d/e" , true));
|
||||
assertTrue(testTopicMatch("*/c" , "a/b/c" , true));
|
||||
assertTrue(testTopicMatch("/*/c" , "/a/b/c" , true));
|
||||
assertTrue(testTopicMatch("a/*" , "a/b/c/d" , true));
|
||||
assertTrue(testTopicMatch("a/+/c" , "a/b/c" , true));
|
||||
assertTrue(testTopicMatch("a/+/c/+/e", "a/b/c/d/e" , true));
|
||||
assertTrue(testTopicMatch("a/*/c/+/e", "a/b/c/d/e" , true));
|
||||
assertTrue(testTopicMatch("/+/b" , "/a/b" , true));
|
||||
assertTrue(testTopicMatch("+" , "a" , true));
|
||||
assertTrue(testTopicMatch("a/b/#" , "a/b/c/d" , true));
|
||||
assertTrue(testTopicMatch("a/b/#" , "a/b" , true));
|
||||
assertTrue(testTopicMatch("a/*/c" , "a/*/c" , true));
|
||||
|
||||
// not matching
|
||||
assertTrue(testTopicMatch("a/b/c" , "a/b/d" , false));
|
||||
assertTrue(testTopicMatch("a/b/c" , "a/b/d" , false));
|
||||
assertTrue(testTopicMatch("a/*/e" , "a/b/c/d/f" , false));
|
||||
assertTrue(testTopicMatch("a/+" , "a" , false));
|
||||
assertTrue(testTopicMatch("a/+" , "a/b/d" , false));
|
||||
assertTrue(testTopicMatch("a/+/" , "a/" , false));
|
||||
|
||||
// $SYS topics
|
||||
assertTrue(testTopicMatch("+/any" , "$SYS/any" , false));
|
||||
assertTrue(testTopicMatch("*/any" , "$SYS/any" , false));
|
||||
assertTrue(testTopicMatch("$SYS/any" , "$SYS/any" , true));
|
||||
assertTrue(testTopicMatch("$SYS/+/y" , "$SYS/a/y" , true));
|
||||
assertTrue(testTopicMatch("$SYS/#" , "$SYS/a/y" , true));
|
||||
|
||||
// not valid
|
||||
assertTrue(testTopicMatch("a/#/b" , "a/x/b" , false));
|
||||
assertTrue(testTopicMatch("a+" , "a/b/d" , false));
|
||||
assertTrue(testTopicMatch("a/b/#/d" , "a/b/c/d" , false));
|
||||
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// setup() and loop()
|
||||
void setup() {
|
||||
delay(1000);
|
||||
Serial.begin(115200);
|
||||
while(!Serial);
|
||||
|
||||
Serial.println("=============[ TinyMqtt StringIndexer TESTS ]========================");
|
||||
}
|
||||
|
||||
void loop() {
|
||||
aunit::TestRunner::run();
|
||||
|
||||
// if (Serial.available()) ESP.reset();
|
||||
}
|
||||
Reference in New Issue
Block a user