Compare commits
123 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e41452edf0 | ||
|
|
bf84e29831 | ||
|
|
0c7c830a26 | ||
|
|
6e601228e6 | ||
|
|
46798ff0de | ||
|
|
45fedf84c9 | ||
|
|
f9c8dca1e5 | ||
|
|
7e1586c0b5 | ||
|
|
123c5a8fa5 | ||
|
|
21fb01848d | ||
|
|
66b1e71ee2 | ||
|
|
e5115087ea | ||
|
|
cc004875b5 | ||
|
|
58ae2436d3 | ||
|
|
cda94368a7 | ||
|
|
0126a39327 | ||
|
|
2587371457 | ||
|
|
a880a1ff2f | ||
|
|
143d57db2d | ||
|
|
2a8dbd09c5 | ||
|
|
00333ed805 | ||
|
|
9228408f57 | ||
|
|
1653b2e386 | ||
|
|
775fbc14ee | ||
|
|
c21b7b63fb | ||
|
|
e45af112c2 | ||
|
|
f9c539ff6a | ||
|
|
7dc23d322c | ||
|
|
4fb632ce3d | ||
|
|
c9130c7a24 | ||
|
|
31c51aeaff | ||
|
|
2eeda4ecab | ||
|
|
787cb77a26 | ||
|
|
d324a913ec | ||
|
|
ac1eedd72a | ||
|
|
0900e799a9 | ||
|
|
2086c7f0e7 | ||
|
|
72382bf351 | ||
|
|
3fec7b87a8 | ||
|
|
8641627742 | ||
|
|
3b2460572b | ||
|
|
a0435b2cfb | ||
|
|
bda041417d | ||
|
|
baffda8a6d | ||
|
|
09e3a3e45f | ||
|
|
f17ece3376 | ||
|
|
0db07df27b | ||
|
|
292592c3dd | ||
|
|
1f267c135b | ||
|
|
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 | ||
|
|
64a05bb60b | ||
|
|
bb89fc5284 | ||
|
|
c1fd1bc907 | ||
|
|
d919188eb0 | ||
|
|
88c7d552cb |
7
.github/workflows/aunit.yml
vendored
7
.github/workflows/aunit.yml
vendored
@@ -9,7 +9,7 @@ on: [push]
|
|||||||
jobs:
|
jobs:
|
||||||
build:
|
build:
|
||||||
|
|
||||||
runs-on: ubuntu-18.04
|
runs-on: ubuntu-20.04
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v2
|
- uses: actions/checkout@v2
|
||||||
@@ -17,9 +17,10 @@ jobs:
|
|||||||
- name: Setup
|
- name: Setup
|
||||||
run: |
|
run: |
|
||||||
cd ..
|
cd ..
|
||||||
git clone https://github.com/bxparks/EpoxyDuino
|
git clone https://github.com/hsaturn/TinyConsole
|
||||||
|
git clone https://github.com/hsaturn/EpoxyDuino
|
||||||
git clone https://github.com/bxparks/AceRoutine
|
git clone https://github.com/bxparks/AceRoutine
|
||||||
git clone https://github.com/bxparks/AUnit
|
git clone https://github.com/hsaturn/AUnit
|
||||||
git clone https://github.com/bxparks/AceCommon
|
git clone https://github.com/bxparks/AceCommon
|
||||||
git clone https://github.com/hsaturn/EspMock
|
git clone https://github.com/hsaturn/EspMock
|
||||||
- name: Verify tests
|
- name: Verify tests
|
||||||
|
|||||||
2
.github/workflows/ci.yml
vendored
2
.github/workflows/ci.yml
vendored
@@ -1,6 +1,6 @@
|
|||||||
name: "CI"
|
name: "CI"
|
||||||
on:
|
on:
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
ci:
|
ci:
|
||||||
runs-on: ubuntu-20.04
|
runs-on: ubuntu-20.04
|
||||||
|
|||||||
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
|
on: push
|
||||||
|
|
||||||
|
###############
|
||||||
|
# Set the Job #
|
||||||
|
###############
|
||||||
jobs:
|
jobs:
|
||||||
# Set the job key. The key is displayed as the job name
|
build:
|
||||||
# when a job name is not provided
|
|
||||||
super-lint:
|
|
||||||
# Name the Job
|
# Name the Job
|
||||||
name: Lint code base
|
name: Lint Code Base
|
||||||
# Set the type of machine to run on
|
# Set the agent to run on
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
##################
|
||||||
|
# Load all steps #
|
||||||
|
##################
|
||||||
steps:
|
steps:
|
||||||
# Checks out a copy of your repository on the ubuntu-latest machine
|
##########################
|
||||||
- name: Checkout code
|
# Checkout the code base #
|
||||||
uses: actions/checkout@v2
|
##########################
|
||||||
|
- 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
|
# Run Linter against code base #
|
||||||
uses: github/super-linter@v3
|
################################
|
||||||
|
- name: Lint Code Base
|
||||||
|
uses: github/super-linter/slim@v4
|
||||||
env:
|
env:
|
||||||
|
VALIDATE_ALL_CODEBASE: false
|
||||||
|
# Change to 'master' if your main branch differs
|
||||||
DEFAULT_BRANCH: main
|
DEFAULT_BRANCH: main
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
|||||||
35
README.md
35
README.md
@@ -10,18 +10,40 @@
|
|||||||
|
|
||||||
TinyMqtt is a small, fast and capable Mqtt Broker and Client for Esp8266 / Esp32 / Esp WROOM
|
TinyMqtt is a small, fast and capable Mqtt Broker and Client for Esp8266 / Esp32 / Esp WROOM
|
||||||
|
|
||||||
|
### Statuses of all unit tests of TinyMqtt and its dependencies
|
||||||
|
|
||||||
|
| Project | Unit tests result |
|
||||||
|
| ----------- | ------------ |
|
||||||
|
| TinyMqtt | [](https://github.com/hsaturn/TinyMqtt/actions/workflows/aunit.yml) |
|
||||||
|
| Dependencies ||
|
||||||
|
| TinyConsole | [](https://github.com/hsaturn/TinyConsole/actions/workflows/aunit.yml) |
|
||||||
|
| EpoxyDuino | [](https://github.com/hsaturn/EpoxyDuino/actions/workflows/aunit_tests.yml) |
|
||||||
|
| EspMock | [](https://github.com/hsaturn/EspMock/actions/workflows/aunit.yml) |
|
||||||
|
| AUnit | [](https://github.com/hsaturn/AUnit/actions/workflows/aunit_tests.yml) |
|
||||||
|
| AceRoutine | [](https://github.com/bxparks/AceRoutine/actions/workflows/aunit_tests.yml) |
|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
- Very (very !!) fast broker I saw it re-sent 1000 topics per second for two
|
- Async Wifi compatible (me-no-dev/ESPAsyncTCP@^1.2.2)
|
||||||
|
- 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.
|
clients that had subscribed (payload ~15 bytes ESP8266). No topic lost.
|
||||||
The max I've seen was 2k msg/s (1 client 1 subscription)
|
The max I've seen was 2k msg/s (1 client 1 subscription)
|
||||||
- Act as as a mqtt broker and/or a mqtt client
|
- Act as as a mqtt broker and/or a mqtt client
|
||||||
- Mqtt 3.1.1 / Qos 0 supported
|
- 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)
|
- Standalone (can work without WiFi) (degraded/local mode)
|
||||||
- Brokers can connect to another broker and becomes then a
|
- Brokers can connect to another broker and becomes then a
|
||||||
proxy for clients that are connected to it.
|
proxy for clients that are connected to it.
|
||||||
- zeroconf, this is a strange but very powerful mode where
|
- zeroconf, this is a strange but very powerful mode where
|
||||||
all brokers tries to connect together on the same local network.
|
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
|
## Quickstart
|
||||||
|
|
||||||
@@ -33,10 +55,11 @@ TinyMqtt is a small, fast and capable Mqtt Broker and Client for Esp8266 / Esp32
|
|||||||
|
|
||||||
| Example | Description |
|
| Example | Description |
|
||||||
| ------------------- | ------------------------------------------ |
|
| ------------------- | ------------------------------------------ |
|
||||||
| client-without-wifi | standalone example |
|
| [client-with-wifi](https://github.com/hsaturn/TinyMqtt/tree/main/examples/client-with-wifi/client-with-wifi.ino) | standalone example |
|
||||||
| simple-client | Connect the ESP to an external Mqtt broker |
|
| [client-without-wifi](https://github.com/hsaturn/TinyMqtt/tree/main/examples/client-without-wifi/client-without-wifi.ino) | standalone example |
|
||||||
| simple-broker | Simple Mqtt broker with your ESP |
|
| [simple-client](https://github.com/hsaturn/TinyMqtt/tree/main/examples/simple-client/simple-client.ino) | Connect the ESP to an external Mqtt broker |
|
||||||
| tinymqtt-test | Complex console example |
|
| [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
|
- tinymqtt-test : This is a complex sketch with a terminal console
|
||||||
that allows to add clients publish, connect etc with interpreted commands.
|
that allows to add clients publish, connect etc with interpreted commands.
|
||||||
@@ -59,7 +82,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 re-subscribe (::resubscribe works bad on broker.emqx.io)~~
|
||||||
* MqttClient auto reconnection
|
* MqttClient auto reconnection
|
||||||
* MqttClient user/password
|
* 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
|
* I suspect that MqttClient::parent could be removed and replaced with a simple boolean
|
||||||
(this'll need to rewrite a few functions)
|
(this'll need to rewrite a few functions)
|
||||||
|
|
||||||
|
|||||||
61
bump_version.sh
Executable file
61
bump_version.sh
Executable file
@@ -0,0 +1,61 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
current_version=$(git describe --tags --abbrev=0)
|
||||||
|
if [ "$1" == "-d" ]; then
|
||||||
|
do=0
|
||||||
|
shift
|
||||||
|
else
|
||||||
|
do=1
|
||||||
|
fi
|
||||||
|
if [ "$1" == "" ]; then
|
||||||
|
echo
|
||||||
|
echo "Syntax: $0 [-d] {new_version}"
|
||||||
|
echo
|
||||||
|
echo " -d : dry run, generate json and update properties but do not run git commands"
|
||||||
|
echo ""
|
||||||
|
echo " Current version: $current_version"
|
||||||
|
echo
|
||||||
|
else
|
||||||
|
echo "Current version: ($current_version)"
|
||||||
|
echo "New version : ($1)"
|
||||||
|
echo -n "Do you want to proceed ? "
|
||||||
|
read a
|
||||||
|
if [ "$a" == "y" ]; then
|
||||||
|
echo "Doing this..."
|
||||||
|
grep $current_version library.properties
|
||||||
|
if [ "$?" == "0" ]; then
|
||||||
|
sed -i "s/$current_version/$1/" library.properties
|
||||||
|
|
||||||
|
cp library.json.skeleton library.json
|
||||||
|
while ifs= read -r line; do
|
||||||
|
name=$(echo "$line" | sed "s/=.*//g")
|
||||||
|
value=$(echo "$line" | cut -d= -f 2 | sed 's/"//g')
|
||||||
|
echo " Replacing $name in json"
|
||||||
|
if [ "$name" == "depends" ]; then
|
||||||
|
depends=$(echo "$value" | sed "s/,/ /g")
|
||||||
|
echo " Depends=$depends"
|
||||||
|
fi
|
||||||
|
echo " " sed -i "s@#$name@$value@g" library.json
|
||||||
|
sed -i "s@#$name@$value@g" library.json
|
||||||
|
done < library.properties
|
||||||
|
deps=""
|
||||||
|
for depend in $depends; do
|
||||||
|
if [ "$deps" != "" ]; then
|
||||||
|
deps="$deps, "
|
||||||
|
fi
|
||||||
|
deps="$deps'$depend' : '*'"
|
||||||
|
done
|
||||||
|
sed -i "s@#dependencies@$deps@g" library.json
|
||||||
|
sed -i "s/'/\"/g" library.json
|
||||||
|
if [ "$do" == "1" ]; then
|
||||||
|
git tag $1
|
||||||
|
git add library.properties
|
||||||
|
git add library.json
|
||||||
|
git commit -m "Release $1"
|
||||||
|
git push
|
||||||
|
git push --tags
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "Current version does not match library.property version, aborting"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
fi
|
||||||
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()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,9 +1,9 @@
|
|||||||
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Local broker that accept connections and two local clients
|
* Local broker that accept connections and two local clients
|
||||||
*
|
*
|
||||||
*
|
*
|
||||||
* +-----------------------------+
|
* +-----------------------------+
|
||||||
* | ESP |
|
* | ESP |
|
||||||
* | +--------+ | 1883 <--- External client/s
|
* | +--------+ | 1883 <--- External client/s
|
||||||
@@ -12,14 +12,14 @@
|
|||||||
* | | ^ |
|
* | | ^ |
|
||||||
* | | | |
|
* | | | |
|
||||||
* | | | | -----
|
* | | | | -----
|
||||||
* | v v | ---
|
* | v v | ---
|
||||||
* | +----------+ +----------+ | -
|
* | +----------+ +----------+ | -
|
||||||
* | | internal | | internal | +-------* Wifi
|
* | | internal | | internal | +-------* Wifi
|
||||||
* | | client | | client | |
|
* | | client | | client | |
|
||||||
* | +----------+ +----------+ |
|
* | +----------+ +----------+ |
|
||||||
* | |
|
* | |
|
||||||
* +-----------------------------+
|
* +-----------------------------+
|
||||||
*
|
*
|
||||||
* pros - Reduces internal latency (when publish is received by the same ESP)
|
* pros - Reduces internal latency (when publish is received by the same ESP)
|
||||||
* - Reduces wifi traffic
|
* - Reduces wifi traffic
|
||||||
* - No need to have an external broker
|
* - No need to have an external broker
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
||||||
|
|
||||||
/** TinyMQTT allows a disconnected mode:
|
/** TinyMQTT allows a disconnected mode:
|
||||||
*
|
*
|
||||||
* +-----------------------------+
|
* +-----------------------------+
|
||||||
* | ESP |
|
* | ESP |
|
||||||
* | +--------+ |
|
* | +--------+ |
|
||||||
@@ -12,8 +12,8 @@
|
|||||||
* | v v |
|
* | v v |
|
||||||
* | +----------+ +----------+ |
|
* | +----------+ +----------+ |
|
||||||
* | | internal | | internal | |
|
* | | internal | | internal | |
|
||||||
* | | client | | client | |
|
* | | client | | client | |
|
||||||
* | +----------+ +----------+ |
|
* | +----------+ +----------+ |
|
||||||
* | |
|
* | |
|
||||||
* +-----------------------------+
|
* +-----------------------------+
|
||||||
*
|
*
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ MqttBroker broker(PORT);
|
|||||||
*
|
*
|
||||||
* +-----------------------------+
|
* +-----------------------------+
|
||||||
* | ESP |
|
* | ESP |
|
||||||
* | +--------+ |
|
* | +--------+ |
|
||||||
* | | broker | | 1883 <--- External client/s
|
* | | broker | | 1883 <--- External client/s
|
||||||
* | +--------+ |
|
* | +--------+ |
|
||||||
* | |
|
* | |
|
||||||
@@ -15,30 +15,30 @@ MqttBroker broker(PORT);
|
|||||||
*
|
*
|
||||||
* Your ESP will become a MqttBroker.
|
* Your ESP will become a MqttBroker.
|
||||||
* You can test it with any client such as mqtt-spy for example
|
* You can test it with any client such as mqtt-spy for example
|
||||||
*
|
*
|
||||||
*/
|
*/
|
||||||
|
|
||||||
const char* ssid = "";
|
const char* ssid = "";
|
||||||
const char* password = "";
|
const char* password = "";
|
||||||
|
|
||||||
void setup()
|
void setup()
|
||||||
{
|
{
|
||||||
Serial.begin(115200);
|
Serial.begin(115200);
|
||||||
|
|
||||||
if (strlen(ssid)==0)
|
if (strlen(ssid)==0)
|
||||||
Serial << "****** PLEASE MODIFY ssid/password *************" << endl;
|
Console << TinyConsole::red << "****** PLEASE MODIFY ssid/password *************" << endl;
|
||||||
|
|
||||||
WiFi.mode(WIFI_STA);
|
WiFi.mode(WIFI_STA);
|
||||||
WiFi.begin(ssid, password);
|
WiFi.begin(ssid, password);
|
||||||
|
|
||||||
while (WiFi.status() != WL_CONNECTED) {
|
while (WiFi.status() != WL_CONNECTED) {
|
||||||
Serial << '.';
|
Serial << '.';
|
||||||
delay(500);
|
delay(500);
|
||||||
}
|
}
|
||||||
Serial << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
Console << TinyConsole::green << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||||
|
|
||||||
broker.begin();
|
broker.begin();
|
||||||
Serial << "Broker ready : " << WiFi.localIP() << " on port " << PORT << endl;
|
Console << "Broker ready : " << WiFi.localIP() << " on port " << PORT << endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
void loop()
|
void loop()
|
||||||
|
|||||||
@@ -1,21 +1,22 @@
|
|||||||
#include "TinyMqtt.h" // https://github.com/hsaturn/TinyMqtt
|
#include "TinyMqtt.h" // https://github.com/hsaturn/TinyMqtt
|
||||||
|
#include "TinyStreaming.h" // https://github.com/hsaturn/TinyConsole
|
||||||
|
|
||||||
/** Simple Client (The simplest configuration)
|
/** Simple Client (The simplest configuration)
|
||||||
*
|
*
|
||||||
*
|
*
|
||||||
* +--------+
|
* +--------+
|
||||||
* +------>| broker |<--- < Other client
|
* +------>| broker |<--- < Other client
|
||||||
* | +--------+
|
* | +--------+
|
||||||
* |
|
* |
|
||||||
* +-----------------+
|
* +-----------------+
|
||||||
* | ESP | |
|
* | ESP | |
|
||||||
* | +----------+ |
|
* | +----------+ |
|
||||||
* | | internal | |
|
* | | internal | |
|
||||||
* | | client | |
|
* | | client | |
|
||||||
* | +----------+ |
|
* | +----------+ |
|
||||||
* | |
|
* | |
|
||||||
* +-----------------+
|
* +-----------------+
|
||||||
*
|
*
|
||||||
* 1 - change the ssid/password
|
* 1 - change the ssid/password
|
||||||
* 2 - change BROKER values (or keep emqx.io test broker)
|
* 2 - change BROKER values (or keep emqx.io test broker)
|
||||||
* 3 - you can use mqtt-spy to connect to the same broker and
|
* 3 - you can use mqtt-spy to connect to the same broker and
|
||||||
@@ -38,7 +39,7 @@ const char* password = "";
|
|||||||
static float temp=19;
|
static float temp=19;
|
||||||
static MqttClient client;
|
static MqttClient client;
|
||||||
|
|
||||||
void setup()
|
void setup()
|
||||||
{
|
{
|
||||||
Serial.begin(115200);
|
Serial.begin(115200);
|
||||||
delay(500);
|
delay(500);
|
||||||
@@ -49,13 +50,13 @@ void setup()
|
|||||||
|
|
||||||
WiFi.mode(WIFI_STA);
|
WiFi.mode(WIFI_STA);
|
||||||
WiFi.begin(ssid, password);
|
WiFi.begin(ssid, password);
|
||||||
|
|
||||||
while (WiFi.status() != WL_CONNECTED)
|
while (WiFi.status() != WL_CONNECTED)
|
||||||
{ delay(500); Serial << '.'; }
|
{ delay(500); Serial << '.'; }
|
||||||
|
|
||||||
Serial << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
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()
|
void loop()
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
@@ -1,12 +1,14 @@
|
|||||||
{
|
{
|
||||||
"name": "TinyMqtt",
|
"name": "TinyMqtt",
|
||||||
"keywords": "ethernet, mqtt, m2m, iot",
|
"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.",
|
"description": "MQTT is a lightweight messaging protocol. This library allows to host a broker or to use a mqtt client in your ESP 8266 or 32 WROOM. It does support MQTT 3.1.1 with QoS",
|
||||||
"repository": {
|
"repository": {
|
||||||
"type": "git",
|
"type": "git",
|
||||||
"url": "https://github.com/hsaturn/TinyMqtt.git"
|
"url": "https://github.com/hsaturn/TinyMqtt.git"
|
||||||
},
|
},
|
||||||
"version": "0.7.7",
|
"dependencies":
|
||||||
|
{ "hsaturn/TinyConsole" : "*" },
|
||||||
|
"version": "0.9.19",
|
||||||
"exclude": "",
|
"exclude": "",
|
||||||
"examples": "examples/*/*.ino",
|
"examples": "examples/*/*.ino",
|
||||||
"frameworks": "arduino",
|
"frameworks": "arduino",
|
||||||
|
|||||||
20
library.json.skeleton
Normal file
20
library.json.skeleton
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"name": "#name",
|
||||||
|
"keywords": "ethernet, mqtt, m2m, iot",
|
||||||
|
"description": "#paragraph",
|
||||||
|
"repository": {
|
||||||
|
"type": "git",
|
||||||
|
"url": "https://github.com/hsaturn/TinyMqtt.git"
|
||||||
|
},
|
||||||
|
"dependencies":
|
||||||
|
{ #dependencies },
|
||||||
|
"version": "#version",
|
||||||
|
"exclude": "",
|
||||||
|
"examples": "examples/*/*.ino",
|
||||||
|
"frameworks": "arduino",
|
||||||
|
"platforms": [
|
||||||
|
"atmelavr",
|
||||||
|
"espressif8266",
|
||||||
|
"espressif32"
|
||||||
|
]
|
||||||
|
}
|
||||||
@@ -1,11 +1,11 @@
|
|||||||
name=TinyMqtt
|
name=TinyMqtt
|
||||||
version=0.7.7
|
version=0.9.19
|
||||||
author=Francois BIOT, HSaturn, <hsaturn@gmail.com>
|
author=Francois BIOT, HSaturn, <hsaturn@gmail.com>
|
||||||
maintainer=Francois BIOT, HSaturn, <hsaturn@gmail.com>
|
maintainer=Francois BIOT <hsaturn@gmail.com>
|
||||||
sentence=A tiny broker and client library for MQTT messaging.
|
sentence=A tiny broker and client library for MQTT messaging.
|
||||||
paragraph=MQTT is a lightweight messaging protocol ideal for small devices. This library allows to send and receive MQTT messages and to host a broker in your ESP 8266 and 32 WROOM. It does support MQTT 3.1.1 with QoS=0.
|
paragraph=MQTT is a lightweight messaging protocol. This library allows to host a broker or to use a mqtt client in your ESP 8266 or 32 WROOM. It does support MQTT 3.1.1 with QoS=0, and is intented to be as smallest as possible, keeping a good efficiency (about 1k msg per second on ESP8266 12F).
|
||||||
category=Communication
|
category=Communication
|
||||||
url=https://github.com/hsaturn/TinyMqtt
|
url=https://github.com/hsaturn/TinyMqtt
|
||||||
architectures=*
|
architectures=*
|
||||||
|
depends=hsaturn/TinyConsole
|
||||||
includes=TinyMqtt.h
|
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
|
|
||||||
#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
|
|
||||||
// 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 @@
|
|||||||
#include "StringIndexer.h"
|
#include "StringIndexer.h"
|
||||||
|
|
||||||
std::map<StringIndexer::index_t, StringIndexer::StringCounter> StringIndexer::strings;
|
StringIndexer::Strings StringIndexer::strings;
|
||||||
|
|
||||||
|
|||||||
@@ -1,129 +1,144 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#pragma once
|
#pragma once
|
||||||
|
#include <assert.h>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
#include <unordered_map>
|
||||||
|
#include "TinyConsole.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
|
using string = TinyConsole::string;
|
||||||
|
|
||||||
/***
|
/***
|
||||||
* Allows to store up to 255 different strings with one byte class
|
* Allows to store up to 255 different strings with one byte class
|
||||||
* very memory efficient when one string is used many times.
|
* very memory efficient when one string is used many times.
|
||||||
*/
|
*/
|
||||||
class StringIndexer
|
class StringIndexer
|
||||||
{
|
{
|
||||||
class StringCounter
|
private:
|
||||||
{
|
|
||||||
std::string str;
|
|
||||||
uint8_t used=0;
|
|
||||||
friend class StringIndexer;
|
|
||||||
|
|
||||||
#if EPOXY_DUINO
|
class StringCounter
|
||||||
public:
|
{
|
||||||
// Workaround to avoid coredump in Indexer::release
|
string str;
|
||||||
// when destroying a Topic after the deletion of
|
uint8_t used=0;
|
||||||
// StringIndexer::strings map (which can occurs only with AUnit,
|
friend class StringIndexer;
|
||||||
// never in the ESP itself, because ESP never ends)
|
|
||||||
// (I hate static vars)
|
|
||||||
~StringCounter() { used=255; }
|
|
||||||
#endif
|
|
||||||
};
|
|
||||||
public:
|
|
||||||
using index_t=uint8_t;
|
|
||||||
|
|
||||||
static index_t strToIndex(const char* str, uint8_t len)
|
#if EPOXY_DUINO
|
||||||
{
|
public:
|
||||||
for(auto it=strings.begin(); it!=strings.end(); it++)
|
// Workaround to avoid coredump in Indexer::release
|
||||||
{
|
// when destroying a Topic after the deletion of
|
||||||
if (strncmp(it->second.str.c_str(), str, len)==0)
|
// StringIndexer::strings map (which can occurs only with AUnit,
|
||||||
{
|
// never in the ESP itself, because ESP never ends)
|
||||||
it->second.used++;
|
// (I hate static vars)
|
||||||
return it->first;
|
~StringCounter() { used=255; }
|
||||||
}
|
#endif
|
||||||
}
|
};
|
||||||
for(index_t index=1; index; index++)
|
public:
|
||||||
{
|
using index_t = uint8_t;
|
||||||
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 const string& str(const index_t& index)
|
||||||
{
|
{
|
||||||
static std::string dummy;
|
static string dummy;
|
||||||
const auto& it=strings.find(index);
|
const auto& it=strings.find(index);
|
||||||
if (it == strings.end()) return dummy;
|
if (it == strings.end()) return dummy;
|
||||||
return it->second.str;
|
return it->second.str;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void use(const index_t& index)
|
static void use(const index_t& index)
|
||||||
{
|
{
|
||||||
auto it=strings.find(index);
|
auto it=strings.find(index);
|
||||||
if (it != strings.end()) it->second.used++;
|
if (it != strings.end()) it->second.used++;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void release(const index_t& index)
|
static void release(const index_t& index)
|
||||||
{
|
{
|
||||||
auto it=strings.find(index);
|
auto it=strings.find(index);
|
||||||
if (it != strings.end())
|
if (it != strings.end())
|
||||||
{
|
{
|
||||||
it->second.used--;
|
it->second.used--;
|
||||||
if (it->second.used == 0)
|
if (it->second.used == 0)
|
||||||
{
|
{
|
||||||
strings.erase(it);
|
strings.erase(it);
|
||||||
// Serial << "Removing string(" << it->second.str.c_str() << ") size=" << strings.size() << endl;
|
// Serial << "Removing string(" << it->second.str.c_str() << ") size=" << strings.size() << endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static uint16_t count() { return strings.size(); }
|
static uint16_t count() { return strings.size(); }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
static std::map<index_t, StringCounter> strings;
|
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 = 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
|
||||||
|
}
|
||||||
|
|
||||||
|
using Strings = std::unordered_map<index_t, StringCounter>;
|
||||||
|
|
||||||
|
static Strings strings;
|
||||||
};
|
};
|
||||||
|
|
||||||
class IndexedString
|
class IndexedString
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
IndexedString(const IndexedString& source)
|
IndexedString(const IndexedString& source)
|
||||||
{
|
{
|
||||||
StringIndexer::use(source.index);
|
StringIndexer::use(source.index);
|
||||||
index = source.index;
|
index = source.index;
|
||||||
}
|
}
|
||||||
|
|
||||||
IndexedString(const char* str, uint8_t len)
|
IndexedString(IndexedString&& i) : index(i.index) {}
|
||||||
{
|
|
||||||
index=StringIndexer::strToIndex(str, len);
|
|
||||||
}
|
|
||||||
|
|
||||||
IndexedString(const std::string& str) : IndexedString(str.c_str(), str.length()) {};
|
IndexedString(const char* str, uint8_t len)
|
||||||
|
{
|
||||||
|
index=StringIndexer::strToIndex(str, len);
|
||||||
|
}
|
||||||
|
|
||||||
~IndexedString() { StringIndexer::release(index); }
|
IndexedString(const string& str) : IndexedString(str.c_str(), str.length()) {};
|
||||||
|
|
||||||
IndexedString& operator=(const IndexedString& source)
|
~IndexedString() { StringIndexer::release(index); }
|
||||||
{
|
|
||||||
StringIndexer::use(source.index);
|
|
||||||
index = source.index;
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
friend bool operator<(const IndexedString& i1, const IndexedString& i2)
|
IndexedString& operator=(const IndexedString& source)
|
||||||
{
|
{
|
||||||
return i1.index < i2.index;
|
StringIndexer::use(source.index);
|
||||||
}
|
index = source.index;
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
friend bool operator==(const IndexedString& i1, const IndexedString& i2)
|
friend bool operator<(const IndexedString& i1, const IndexedString& i2)
|
||||||
{
|
{
|
||||||
return i1.index == i2.index;
|
return i1.index < i2.index;
|
||||||
}
|
}
|
||||||
|
|
||||||
const std::string& str() const { return StringIndexer::str(index); }
|
friend bool operator==(const IndexedString& i1, const IndexedString& i2)
|
||||||
|
{
|
||||||
|
return i1.index == i2.index;
|
||||||
|
}
|
||||||
|
|
||||||
const StringIndexer::index_t& getIndex() const { return index; }
|
const string& str() const { return StringIndexer::str(index); }
|
||||||
|
|
||||||
private:
|
const StringIndexer::index_t& getIndex() const { return index; }
|
||||||
StringIndexer::index_t index;
|
|
||||||
|
private:
|
||||||
|
StringIndexer::index_t index;
|
||||||
};
|
};
|
||||||
|
|||||||
1414
src/TinyMqtt.cpp
1414
src/TinyMqtt.cpp
File diff suppressed because it is too large
Load Diff
523
src/TinyMqtt.h
523
src/TinyMqtt.h
@@ -1,18 +1,23 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
// TODO Should add a AUnit with both TCP_ASYNC and not TCP_ASYNC
|
#ifndef TINY_MQTT_DEBUG
|
||||||
// #define TCP_ASYNC // Uncomment this to use ESPAsyncTCP instead of normal cnx
|
#define TINY_MQTT_DEBUG 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// 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)
|
#if defined(ESP8266) || defined(EPOXY_DUINO)
|
||||||
#ifdef TCP_ASYNC
|
#ifdef TINY_MQTT_ASYNC
|
||||||
#include <ESPAsyncTCP.h>
|
#include <ESPAsyncTCP.h>
|
||||||
#else
|
#else
|
||||||
#include <ESP8266WiFi.h>
|
#include <ESP8266WiFi.h>
|
||||||
#endif
|
#endif
|
||||||
#elif defined(ESP32)
|
#elif defined(ESP32)
|
||||||
#include <WiFi.h>
|
#include <WiFi.h>
|
||||||
#ifdef TCP_ASYNC
|
#ifdef TINY_MQTT_ASYNC
|
||||||
#include <AsyncTCP.h> // https://github.com/me-no-dev/AsyncTCP
|
#include <AsyncTCP.h> // https://github.com/me-no-dev/AsyncTCP
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
#ifdef EPOXY_DUINO
|
#ifdef EPOXY_DUINO
|
||||||
@@ -20,21 +25,35 @@
|
|||||||
#else
|
#else
|
||||||
#define dbg_ptr uint32_t
|
#define dbg_ptr uint32_t
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#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 <vector>
|
#include <vector>
|
||||||
#include <set>
|
#include <set>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include "StringIndexer.h"
|
#include "StringIndexer.h"
|
||||||
#include <MqttStreaming.h>
|
|
||||||
|
|
||||||
// #define TINY_MQTT_DEBUG
|
#define TINY_MQTT_DEFAULT_CLIENT_ID "Tiny"
|
||||||
|
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#include <TinyStreaming.h>
|
||||||
#define debug(what) { Serial << __LINE__ << ' ' << what << endl; delay(100); }
|
#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(100); }
|
||||||
#else
|
#else
|
||||||
#define debug(what) {}
|
#define debug(what) {}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef TCP_ASYNC
|
#ifdef TINY_MQTT_ASYNC
|
||||||
using TcpClient = AsyncClient;
|
using TcpClient = AsyncClient;
|
||||||
using TcpServer = AsyncServer;
|
using TcpServer = AsyncServer;
|
||||||
#else
|
#else
|
||||||
@@ -42,263 +61,345 @@
|
|||||||
using TcpServer = WiFiServer;
|
using TcpServer = WiFiServer;
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
enum MqttError
|
enum __attribute__((packed)) MqttError
|
||||||
{
|
{
|
||||||
MqttOk = 0,
|
MqttOk = 0,
|
||||||
MqttNowhereToSend=1,
|
MqttNowhereToSend=1,
|
||||||
MqttInvalidMessage=2,
|
MqttInvalidMessage=2,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
using string = TinyConsole::string;
|
||||||
|
|
||||||
class Topic : public IndexedString
|
class Topic : public IndexedString
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
Topic(const char* s, uint8_t len) : IndexedString(s,len){}
|
Topic(const string& m) : IndexedString(m){}
|
||||||
Topic(const char* s) : Topic(s, strlen(s)) {}
|
Topic(const char* s, uint8_t len) : IndexedString(s,len){}
|
||||||
Topic(const std::string s) : Topic(s.c_str(), s.length()){};
|
Topic(const char* s) : Topic(s, strlen(s)) {}
|
||||||
|
// Topic(const string s) : Topic(s.c_str(), s.length()){};
|
||||||
|
|
||||||
const char* c_str() const { return str().c_str(); }
|
const char* c_str() const { return str().c_str(); }
|
||||||
|
|
||||||
bool matches(const Topic&) const;
|
bool matches(const Topic&) const;
|
||||||
};
|
};
|
||||||
|
|
||||||
class MqttClient;
|
class MqttClient;
|
||||||
class MqttMessage
|
class MqttMessage
|
||||||
{
|
{
|
||||||
const uint16_t MaxBufferLength = 4096; //hard limit: 16k
|
const uint16_t MaxBufferLength = 4096; //hard limit: 16k due to size decoding
|
||||||
public:
|
public:
|
||||||
enum Type
|
enum __attribute__((packed)) Type
|
||||||
{
|
{
|
||||||
Unknown = 0,
|
Unknown = 0,
|
||||||
Connect = 0x10,
|
Connect = 0x10,
|
||||||
ConnAck = 0x20,
|
ConnAck = 0x20,
|
||||||
Publish = 0x30,
|
Publish = 0x30,
|
||||||
PubAck = 0x40,
|
PubAck = 0x40,
|
||||||
Subscribe = 0x80,
|
Subscribe = 0x80,
|
||||||
SubAck = 0x90,
|
SubAck = 0x90,
|
||||||
UnSubscribe = 0xA0,
|
UnSubscribe = 0xA0,
|
||||||
UnSuback = 0xB0,
|
UnSuback = 0xB0,
|
||||||
PingReq = 0xC0,
|
PingReq = 0xC0,
|
||||||
PingResp = 0xD0,
|
PingResp = 0xD0,
|
||||||
Disconnect = 0xE0
|
Disconnect = 0xE0
|
||||||
};
|
};
|
||||||
enum State
|
|
||||||
{
|
|
||||||
FixedHeader=0,
|
|
||||||
Length=1,
|
|
||||||
VariableHeader=2,
|
|
||||||
PayLoad=3,
|
|
||||||
Complete=4,
|
|
||||||
Error=5,
|
|
||||||
Create=6
|
|
||||||
};
|
|
||||||
|
|
||||||
MqttMessage() { reset(); }
|
enum __attribute__((packed)) State
|
||||||
MqttMessage(Type t, uint8_t bits_d3_d0=0) { create(t); buffer[0] |= bits_d3_d0; }
|
{
|
||||||
void incoming(char byte);
|
FixedHeader=0,
|
||||||
void add(char byte) { incoming(byte); }
|
Length=1,
|
||||||
void add(const char* p, size_t len, bool addLength=true );
|
VariableHeader=2,
|
||||||
void add(const std::string& s) { add(s.c_str(), s.length()); }
|
PayLoad=3,
|
||||||
void add(const Topic& t) { add(t.str()); }
|
Complete=4,
|
||||||
const char* end() const { return &buffer[0]+buffer.size(); }
|
Error=5,
|
||||||
const char* getVHeader() const { return &buffer[vheader]; }
|
Create=6
|
||||||
uint16_t length() const { return buffer.size(); }
|
};
|
||||||
void complete();
|
|
||||||
|
|
||||||
void reset();
|
static inline uint32_t getSize(const char* buffer)
|
||||||
|
{
|
||||||
|
const unsigned char* bun = (const unsigned char*)buffer;
|
||||||
|
return (*bun << 8) | bun[1]; }
|
||||||
|
|
||||||
// buff is MSB/LSB/STRING
|
MqttMessage() { reset(); }
|
||||||
// output buff+=2, len=length(str)
|
MqttMessage(Type t, uint8_t bits_d3_d0=0) { create(t); buffer[0] |= (bits_d3_d0 & 0xF); }
|
||||||
static void getString(const char* &buff, uint16_t& len);
|
MqttMessage(const MqttMessage& m)
|
||||||
|
: buffer(m.buffer), vheader(m.vheader), size(m.size), state(m.state) {}
|
||||||
|
|
||||||
Type type() const
|
void incoming(char byte);
|
||||||
{
|
void add(char byte) { incoming(byte); }
|
||||||
return state == Complete ? static_cast<Type>(buffer[0]) : Unknown;
|
void add(const char* p, size_t len, bool addLength=true );
|
||||||
}
|
void add(const string& s) { add(s.c_str(), s.length()); }
|
||||||
|
void add(const Topic& t) { add(t.str()); }
|
||||||
|
const char* end() const { return &buffer[0]+buffer.size(); }
|
||||||
|
const char* getVHeader() const { return &buffer[vheader]; }
|
||||||
|
void complete() { encodeLength(); }
|
||||||
|
|
||||||
void create(Type type)
|
void reset();
|
||||||
{
|
|
||||||
buffer=(char)type;
|
|
||||||
buffer+='\0'; // reserved for msg length
|
|
||||||
vheader=2;
|
|
||||||
size=0;
|
|
||||||
state=Create;
|
|
||||||
}
|
|
||||||
MqttError sendTo(MqttClient*) const;
|
|
||||||
void hexdump(const char* prefix=nullptr) const;
|
|
||||||
|
|
||||||
private:
|
// buff is MSB/LSB/STRING
|
||||||
void encodeLength(char* msb, int length) const;
|
// output buff+=2, len=length(str)
|
||||||
|
static void getString(const char* &buff, uint16_t& len);
|
||||||
|
|
||||||
mutable std::string buffer; // mutable -> sendTo()
|
Type type() const
|
||||||
uint8_t vheader;
|
{
|
||||||
uint16_t size; // bytes left to receive
|
return state == Complete ? static_cast<Type>(buffer[0] & 0xF0) : Unknown;
|
||||||
State state;
|
}
|
||||||
|
|
||||||
|
uint8_t flags() const { return static_cast<uint8_t>(buffer[0] & 0x0F); }
|
||||||
|
|
||||||
|
void create(Type type)
|
||||||
|
{
|
||||||
|
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*);
|
||||||
|
void hexdump(const char* prefix=nullptr) const;
|
||||||
|
|
||||||
|
MqttMessage& operator = (MqttMessage&& m)
|
||||||
|
{
|
||||||
|
buffer = std::move(m.buffer);
|
||||||
|
vheader = m.vheader;
|
||||||
|
size = m.size;
|
||||||
|
state = m.state;
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
void encodeLength();
|
||||||
|
|
||||||
|
string buffer;
|
||||||
|
uint8_t vheader;
|
||||||
|
uint16_t size; // bytes left to receive
|
||||||
|
State state;
|
||||||
};
|
};
|
||||||
|
|
||||||
class MqttBroker;
|
class MqttBroker;
|
||||||
class MqttClient
|
class MqttClient
|
||||||
{
|
{
|
||||||
using CallBack = void (*)(const MqttClient* source, const Topic& topic, const char* payload, size_t payload_length);
|
enum __attribute__((packed)) Flags
|
||||||
enum Flags
|
{
|
||||||
{
|
FlagUserName = 128,
|
||||||
FlagUserName = 128,
|
FlagPassword = 64,
|
||||||
FlagPassword = 64,
|
FlagWillRetain = 32, // unsupported
|
||||||
FlagWillRetain = 32, // unsupported
|
FlagWillQos = 16 | 8, // unsupported
|
||||||
FlagWillQos = 16 | 8, // unsupported
|
FlagWill = 4, // unsupported
|
||||||
FlagWill = 4, // unsupported
|
FlagCleanSession = 2, // unsupported
|
||||||
FlagCleanSession = 2, // unsupported
|
FlagReserved = 1
|
||||||
FlagReserved = 1
|
};
|
||||||
};
|
|
||||||
public:
|
|
||||||
/** Constructor. If broker is not null, this is the adress of a local broker.
|
|
||||||
If you want to connect elsewhere, leave broker null and use connect() **/
|
|
||||||
MqttClient(MqttBroker* broker = nullptr, const std::string& id="");
|
|
||||||
MqttClient(const std::string& id) : MqttClient(nullptr, id){}
|
|
||||||
|
|
||||||
~MqttClient();
|
enum __attribute__((packed)) CltFlags
|
||||||
|
{
|
||||||
|
CltFlagNone = 0,
|
||||||
|
CltFlagConnected = 1,
|
||||||
|
CltFlagToDelete = 2
|
||||||
|
};
|
||||||
|
public:
|
||||||
|
|
||||||
void connect(MqttBroker* parent);
|
using CallBack = void (*)(const MqttClient* source, const Topic& topic, const char* payload, size_t payload_length);
|
||||||
void connect(std::string broker, uint16_t port, uint16_t keep_alive = 10);
|
|
||||||
|
|
||||||
bool connected() { return
|
/** Constructor. Broker is the adress of a local broker if not null
|
||||||
(parent!=nullptr and client==nullptr) or
|
If you want to connect elsewhere, leave broker null and use connect() **/
|
||||||
(client and client->connected()); }
|
MqttClient(MqttBroker* broker = nullptr, const string& id = TINY_MQTT_DEFAULT_CLIENT_ID);
|
||||||
void write(const char* buf, size_t length)
|
MqttClient(const string& id) : MqttClient(nullptr, id){}
|
||||||
{ if (client) client->write(buf, length); }
|
|
||||||
|
|
||||||
const std::string& id() const { return clientId; }
|
~MqttClient();
|
||||||
void id(std::string& new_id) { clientId = new_id; }
|
|
||||||
|
|
||||||
/** Should be called in main loop() */
|
void connect(MqttBroker* local_broker);
|
||||||
void loop();
|
void connect(string broker, uint16_t port, uint16_t keep_alive = 10);
|
||||||
void close(bool bSendDisconnect=true);
|
|
||||||
void setCallback(CallBack fun) {callback=fun; };
|
|
||||||
|
|
||||||
// Publish from client to the world
|
// TODO it seems that connected returns true in tcp mode even if
|
||||||
MqttError publish(const Topic&, const char* payload, size_t pay_length);
|
// no negociation occurred
|
||||||
MqttError publish(const Topic& t, const char* payload) { return publish(t, payload, strlen(payload)); }
|
bool connected()
|
||||||
MqttError publish(const Topic& t, const String& s) { return publish(t, s.c_str(), s.length()); }
|
{
|
||||||
MqttError publish(const Topic& t, const std::string& s) { return publish(t,s.c_str(),s.length());}
|
return (local_broker!=nullptr and tcp_client==nullptr)
|
||||||
MqttError publish(const Topic& t) { return publish(t, nullptr, 0);};
|
or (tcp_client and tcp_client->connected());
|
||||||
|
}
|
||||||
|
|
||||||
MqttError subscribe(Topic topic, uint8_t qos=0);
|
void write(const char* buf, size_t length)
|
||||||
MqttError unsubscribe(Topic topic);
|
{
|
||||||
bool isSubscribedTo(const Topic& topic) const;
|
if (tcp_client) tcp_client->write(buf, length);
|
||||||
|
}
|
||||||
|
|
||||||
// connected to local broker
|
const string& id() const { return clientId; }
|
||||||
// TODO seems to be useless
|
void id(const string& new_id) { clientId = new_id; }
|
||||||
bool isLocal() const { return client == nullptr; }
|
|
||||||
|
|
||||||
void dump(std::string indent="")
|
/** Should be called in main loop() */
|
||||||
{
|
void loop();
|
||||||
uint32_t ms=millis();
|
void close(bool bSendDisconnect=true);
|
||||||
Serial << indent << "+-- " << '\'' << clientId.c_str() << "' " << (connected() ? " ON " : " OFF");
|
void setCallback(CallBack fun)
|
||||||
Serial << ", alive=" << alive << '/' << ms << ", ka=" << keep_alive << ' ';
|
{
|
||||||
Serial << (client && client->connected() ? "" : "dis") << "connected";
|
callback=fun;
|
||||||
if (subscriptions.size())
|
#if TINY_MQTT_DEBUG
|
||||||
{
|
Console << TinyConsole::magenta << "Callback set to " << (long)fun << TinyConsole::white << endl;
|
||||||
bool c = false;
|
if (callback) callback(this, "test/topic", "value", 5);
|
||||||
Serial << " [";
|
#endif
|
||||||
for(auto s: subscriptions)
|
};
|
||||||
{
|
|
||||||
if (c) Serial << ", ";
|
|
||||||
Serial << s.str().c_str();
|
|
||||||
c=true;
|
|
||||||
}
|
|
||||||
Serial << ']';
|
|
||||||
}
|
|
||||||
Serial << endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Count the number of messages that have been sent **/
|
// Publish from client to the world
|
||||||
static long counter;
|
MqttError publish(const Topic&, const char* payload, size_t pay_length, bool retain=false);
|
||||||
|
MqttError publish(const Topic& t, const char* payload, bool retain=false) { return publish(t, payload, strlen(payload), retain); }
|
||||||
|
MqttError publish(const Topic& t, const String& s, bool retain=false) { return publish(t, s.c_str(), s.length(), retain); }
|
||||||
|
MqttError publish(const Topic& t, const string& s, bool retain=false) { return publish(t,s.c_str(),s.length(), retain);}
|
||||||
|
MqttError publish(const Topic& t, bool retain=false) { return publish(t, nullptr, 0, retain);};
|
||||||
|
|
||||||
private:
|
MqttError subscribe(Topic topic, uint8_t qos=0);
|
||||||
static void onConnect(void * client_ptr, TcpClient*);
|
MqttError unsubscribe(Topic topic);
|
||||||
#ifdef TCP_ASYNC
|
bool isSubscribedTo(const Topic& topic) const;
|
||||||
static void onData(void* client_ptr, TcpClient*, void* data, size_t len);
|
|
||||||
|
// connected to local broker
|
||||||
|
// TODO seems to be useless
|
||||||
|
bool isLocal() const { return tcp_client == nullptr; }
|
||||||
|
|
||||||
|
void dump(string indent="")
|
||||||
|
{
|
||||||
|
(void)indent;
|
||||||
|
#if TINY_MQTT_DEBUG
|
||||||
|
uint32_t ms=millis();
|
||||||
|
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;
|
||||||
|
Console << " [";
|
||||||
|
for(auto s: subscriptions)
|
||||||
|
{
|
||||||
|
if (c) Console << ", ";
|
||||||
|
Console << s.str().c_str();
|
||||||
|
c=true;
|
||||||
|
}
|
||||||
|
Console << ']';
|
||||||
|
}
|
||||||
|
Console << TinyConsole::erase_to_end << endl;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef EPOXY_DUINO
|
||||||
|
static std::map<MqttMessage::Type, int> counters; // Number of processed messages
|
||||||
#endif
|
#endif
|
||||||
MqttError sendTopic(const Topic& topic, MqttMessage::Type type, uint8_t qos);
|
uint32_t keepAlive() const { return keep_alive; }
|
||||||
void resubscribe();
|
|
||||||
|
|
||||||
friend class MqttBroker;
|
private:
|
||||||
MqttClient(MqttBroker* parent, TcpClient* client);
|
bool mqtt_connected() const { return cltFlags & CltFlagConnected; }
|
||||||
// republish a received publish if topic matches any in subscriptions
|
void setFlag(CltFlags f) { cltFlags |= f; }
|
||||||
MqttError publishIfSubscribed(const Topic& topic, const MqttMessage& msg);
|
void resetFlag(CltFlags f) { cltFlags &= ~f; }
|
||||||
|
|
||||||
void clientAlive(uint32_t more_seconds);
|
// event when tcp/ip link established (real or fake)
|
||||||
void processMessage(const MqttMessage* message);
|
static void onConnect(void * client_ptr, TcpClient*);
|
||||||
|
#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();
|
||||||
|
|
||||||
bool mqtt_connected = false;
|
friend class MqttBroker;
|
||||||
char mqtt_flags;
|
MqttClient(MqttBroker* local_broker, TcpClient* client);
|
||||||
uint32_t keep_alive = 60;
|
// republish a received publish if topic matches any in subscriptions
|
||||||
uint32_t alive;
|
MqttError publishIfSubscribed(const Topic& topic, MqttMessage& msg);
|
||||||
MqttMessage message;
|
|
||||||
|
|
||||||
// TODO having a pointer on MqttBroker may produce larger binaries
|
void clientAlive(uint32_t more_seconds);
|
||||||
// due to unecessary function linked if ever parent is not used
|
void processMessage(MqttMessage* message);
|
||||||
// (this is the case when MqttBroker isn't used except here)
|
|
||||||
MqttBroker* parent=nullptr; // connection to local broker
|
|
||||||
|
|
||||||
TcpClient* client=nullptr; // connection to mqtt client or to remote broker
|
uint8_t cltFlags = CltFlagNone;
|
||||||
std::set<Topic> subscriptions;
|
char mqtt_flags;
|
||||||
std::string clientId;
|
uint32_t keep_alive = 30;
|
||||||
CallBack callback = nullptr;
|
uint32_t alive;
|
||||||
|
MqttMessage message;
|
||||||
|
|
||||||
|
// connection to local broker, or link to the parent
|
||||||
|
// when MqttBroker uses MqttClient for each external connexion
|
||||||
|
MqttBroker* local_broker=nullptr;
|
||||||
|
|
||||||
|
TcpClient* tcp_client=nullptr; // connection to remote broker
|
||||||
|
std::set<Topic> subscriptions;
|
||||||
|
string clientId;
|
||||||
|
CallBack callback = nullptr;
|
||||||
};
|
};
|
||||||
|
|
||||||
class MqttBroker
|
class MqttBroker
|
||||||
{
|
{
|
||||||
enum State
|
public:
|
||||||
{
|
// TODO limit max number of clients
|
||||||
Disconnected, // Also the initial state
|
MqttBroker(uint16_t port, uint8_t retain_size=0);
|
||||||
Connecting, // connect and sends a fake publish to avoid circular cnx
|
~MqttBroker();
|
||||||
Connected, // this->broker is connected and circular cnx avoided
|
|
||||||
};
|
|
||||||
public:
|
|
||||||
// TODO limit max number of clients
|
|
||||||
MqttBroker(uint16_t port);
|
|
||||||
~MqttBroker();
|
|
||||||
|
|
||||||
void begin() { server->begin(); }
|
void begin() { server->begin(); }
|
||||||
void loop();
|
void loop();
|
||||||
|
|
||||||
void connect(const std::string& host, uint16_t port=1883);
|
/** Connect the broker to a parent broker */
|
||||||
bool connected() const { return state == Connected; }
|
void connect(const string& host, uint16_t port=1883);
|
||||||
|
/** returns true if connected to another broker */
|
||||||
|
bool connected() const { return remote_broker ? remote_broker->connected() : false; }
|
||||||
|
|
||||||
size_t clientsCount() const { return clients.size(); }
|
size_t clientsCount() const { return clients.size(); }
|
||||||
|
void retain(uint8_t size) { retain_size = size; }
|
||||||
|
|
||||||
void dump(std::string indent="")
|
void dump(string indent="")
|
||||||
{
|
{
|
||||||
for(auto client: clients)
|
for(auto client: clients)
|
||||||
client->dump(indent);
|
client->dump(indent);
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
const std::vector<MqttClient*> getClients() const { return clients; }
|
||||||
friend class MqttClient;
|
|
||||||
|
|
||||||
static void onClient(void*, TcpClient*);
|
private:
|
||||||
bool checkUser(const char* user, uint8_t len) const
|
friend class MqttClient;
|
||||||
{ return compareString(auth_user, user, len); }
|
|
||||||
|
|
||||||
bool checkPassword(const char* password, uint8_t len) const
|
static void onClient(void*, TcpClient*);
|
||||||
{ return compareString(auth_password, password, len); }
|
bool checkUser(const char* user, uint8_t len) const
|
||||||
|
{ return compareString(auth_user, user, len); }
|
||||||
|
|
||||||
|
bool checkPassword(const char* password, uint8_t len) const
|
||||||
|
{ 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);
|
||||||
|
|
||||||
MqttError subscribe(const Topic& topic, uint8_t qos);
|
MqttError subscribe(MqttClient*, const Topic& topic, uint8_t qos);
|
||||||
|
|
||||||
// For clients that are added not by the broker itself
|
// For clients that are added not by the broker itself (local clients)
|
||||||
void addClient(MqttClient* client);
|
void addClient(MqttClient* client);
|
||||||
void removeClient(MqttClient* client);
|
void removeClient(MqttClient* client);
|
||||||
|
|
||||||
bool compareString(const char* good, const char* str, uint8_t str_len) const;
|
bool compareString(const char* good, const char* str, uint8_t str_len) const;
|
||||||
std::vector<MqttClient*> clients;
|
std::vector<MqttClient*> clients;
|
||||||
TcpServer* server;
|
|
||||||
|
|
||||||
const char* auth_user = "guest";
|
private:
|
||||||
const char* auth_password = "guest";
|
TcpServer* server = nullptr;
|
||||||
State state = Disconnected;
|
|
||||||
|
|
||||||
MqttClient* broker = nullptr;
|
const char* auth_user = "guest";
|
||||||
|
const char* auth_password = "guest";
|
||||||
|
MqttClient* remote_broker = nullptr;
|
||||||
|
|
||||||
|
void retain(const Topic& topic, const MqttMessage& msg);
|
||||||
|
void retainDrop();
|
||||||
|
|
||||||
|
struct Retain
|
||||||
|
{
|
||||||
|
Retain(unsigned long ts, const MqttMessage& m) : timestamp(ts), msg(m) {}
|
||||||
|
Retain(Retain&& r) : timestamp(r.timestamp), msg(std::move(r.msg)) {}
|
||||||
|
|
||||||
|
Retain& operator=(Retain&& r)
|
||||||
|
{
|
||||||
|
timestamp = r.timestamp;
|
||||||
|
msg = std::move(r.msg);
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned long timestamp;
|
||||||
|
MqttMessage msg;
|
||||||
|
};
|
||||||
|
|
||||||
|
std::map<Topic, Retain> retained;
|
||||||
|
uint8_t retain_size;
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -1,2 +0,0 @@
|
|||||||
const char *ssid = "YOUR-SSID-HERE";
|
|
||||||
const char *password = "YOUR-PASSWORD-HERE";
|
|
||||||
@@ -1,20 +1,40 @@
|
|||||||
|
SUB=
|
||||||
|
|
||||||
|
all:runtests
|
||||||
|
|
||||||
tests:
|
tests:
|
||||||
set -e; \
|
@set -e; \
|
||||||
for i in *-tests/Makefile; do \
|
for i in $(SUB)*-tests/Makefile; do \
|
||||||
echo '==== Making:' $$(dirname $$i); \
|
echo '==== Making:' $$(dirname $$i); \
|
||||||
$(MAKE) -C $$(dirname $$i) -j; \
|
$(MAKE) -C $$(dirname $$i) -j; \
|
||||||
done
|
done
|
||||||
|
$(MAKE) -C compile-test
|
||||||
|
|
||||||
runtests: tests
|
valgrind:
|
||||||
set -e; \
|
@set -e; \
|
||||||
for i in *-tests/Makefile; do \
|
$(MAKE) tests; \
|
||||||
|
for i in $(SUB)*-tests/Makefile; do \
|
||||||
|
echo '==== Running:' $$(dirname $$i); \
|
||||||
|
valgrind $$(dirname $$i)/$$(dirname $$i).out; \
|
||||||
|
done
|
||||||
|
|
||||||
|
debugtest:
|
||||||
|
@set -e; \
|
||||||
|
$(MAKE) clean; \
|
||||||
|
$(MAKE) -C debug-mode -j; \
|
||||||
|
debug-mode/debug-tests.out
|
||||||
|
|
||||||
|
runtests:
|
||||||
|
@set -e; \
|
||||||
|
$(MAKE) tests; \
|
||||||
|
for i in $(SUB)*-tests/Makefile; do \
|
||||||
echo '==== Running:' $$(dirname $$i); \
|
echo '==== Running:' $$(dirname $$i); \
|
||||||
$$(dirname $$i)/$$(dirname $$i).out; \
|
$$(dirname $$i)/$$(dirname $$i).out; \
|
||||||
done
|
done
|
||||||
|
|
||||||
clean:
|
clean:
|
||||||
set -e; \
|
@set -e; \
|
||||||
for i in *-tests/Makefile; do \
|
for i in $(SUB)*-tests/Makefile; do \
|
||||||
echo '==== Cleaning:' $$(dirname $$i); \
|
echo '==== Cleaning:' $$(dirname $$i); \
|
||||||
$(MAKE) -C $$(dirname $$i) clean; \
|
$(MAKE) -C $$(dirname $$i) clean; \
|
||||||
done
|
done
|
||||||
|
|||||||
11
tests/Makefile.opts
Normal file
11
tests/Makefile.opts
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
|
||||||
|
# GCC
|
||||||
|
# CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||||
|
EXTRA_CXXFLAGS=-g3 -O0
|
||||||
|
|
||||||
|
CXXFLAGS=-D_GNU_SOURCE -Werror=return-type -std=gnu++17 -Wall -g3 -O0
|
||||||
|
|
||||||
|
# CLANG SANITIZE
|
||||||
|
# CXX=clang
|
||||||
|
# EXTRA_CXXFLAGS=-g3 -O0 -fsanitize=memory
|
||||||
|
# LDFLAGS = -lpthread -lstdc++
|
||||||
10
tests/classbind-tests/Makefile
Normal file
10
tests/classbind-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 -DTINY_MQTT_TESTS
|
||||||
|
|
||||||
|
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
|
||||||
353
tests/classbind-tests/classbind-tests.ino
Normal file
353
tests/classbind-tests/classbind-tests.ino
Normal file
@@ -0,0 +1,353 @@
|
|||||||
|
// 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 */)
|
||||||
|
{
|
||||||
|
(void) topic;
|
||||||
|
// 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)
|
||||||
|
{
|
||||||
|
(void) topic;
|
||||||
|
// 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("=============[ TinyMqtt class-bind TESTS ]========================");
|
||||||
|
|
||||||
|
WiFi.mode(WIFI_STA);
|
||||||
|
WiFi.begin("network", "password");
|
||||||
|
}
|
||||||
|
|
||||||
|
void loop() {
|
||||||
|
aunit::TestRunner::run();
|
||||||
|
|
||||||
|
if (Serial.available()) ESP.reset();
|
||||||
|
}
|
||||||
11
tests/compile-test/Makefile
Normal file
11
tests/compile-test/Makefile
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||||
|
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||||
|
|
||||||
|
include ../Makefile.opts
|
||||||
|
|
||||||
|
APP_NAME := tinymqtt-test
|
||||||
|
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsync ESP8266mDNS TinyConsole
|
||||||
|
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||||
|
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||||
|
DUMMY := 1
|
||||||
|
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||||
848
tests/compile-test/tinymqtt-test.ino
Normal file
848
tests/compile-test/tinymqtt-test.ino
Normal file
@@ -0,0 +1,848 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab smartindent
|
||||||
|
#include <TinyConsole.h>
|
||||||
|
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
||||||
|
#include <TinyStreaming.h>
|
||||||
|
|
||||||
|
#include <sstream>
|
||||||
|
#include <string>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
using string = TinyConsole::string;
|
||||||
|
|
||||||
|
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) {}
|
||||||
|
|
||||||
|
string url;
|
||||||
|
uint16_t port;
|
||||||
|
};
|
||||||
|
|
||||||
|
const std::map<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,
|
||||||
|
* even some stress tests.
|
||||||
|
*
|
||||||
|
* Upload the sketch, the use the terminal.
|
||||||
|
* Press H for mini help.
|
||||||
|
*
|
||||||
|
* tested with mqtt-spy-0.5.4
|
||||||
|
* TODO examples of scripts
|
||||||
|
*/
|
||||||
|
|
||||||
|
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t 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<32)
|
||||||
|
Console << '?';
|
||||||
|
else
|
||||||
|
Console << c;
|
||||||
|
}
|
||||||
|
Console << ']' << endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::map<string, MqttClient*> clients;
|
||||||
|
std::map<string, MqttBroker*> brokers;
|
||||||
|
std::map<string, string> vars;
|
||||||
|
|
||||||
|
void eval(string& cmd);
|
||||||
|
|
||||||
|
void replace(const char* d, string& str, string srch, string to)
|
||||||
|
{
|
||||||
|
if (d[0] && d[1])
|
||||||
|
{
|
||||||
|
srch=d[0]+srch+d[1];
|
||||||
|
to=d[0]+to+d[1];
|
||||||
|
|
||||||
|
size_t pos = 0;
|
||||||
|
while((pos=str.find(srch, pos)) != string::npos)
|
||||||
|
{
|
||||||
|
str.erase(pos, srch.length());
|
||||||
|
str.insert(pos, to);
|
||||||
|
pos += to.length()-1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void replaceVars(string& cmd)
|
||||||
|
{
|
||||||
|
cmd = ' '+cmd+' ';
|
||||||
|
|
||||||
|
for(auto it: vars)
|
||||||
|
{
|
||||||
|
replace("..", cmd, it.first, it.second);
|
||||||
|
replace(". ", cmd, it.first, it.second);
|
||||||
|
replace(" .", cmd, it.first, it.second);
|
||||||
|
replace(" ", cmd, it.first, it.second);
|
||||||
|
}
|
||||||
|
cmd.erase(0, cmd.find_first_not_of(' '));
|
||||||
|
cmd.erase(cmd.find_last_not_of(' ')+1);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
void onCommand(const string& command)
|
||||||
|
{
|
||||||
|
Console << endl;
|
||||||
|
string cmd=command;
|
||||||
|
if (cmd.substr(0,3)!="set") replaceVars(cmd);
|
||||||
|
eval(cmd);
|
||||||
|
Console << endl;
|
||||||
|
Console.prompt();
|
||||||
|
}
|
||||||
|
|
||||||
|
void setup()
|
||||||
|
{
|
||||||
|
Serial.begin(115200);
|
||||||
|
Console.begin(Serial);
|
||||||
|
Console.setPrompt("> ");
|
||||||
|
Console.setCallback(onCommand);
|
||||||
|
delay(500);
|
||||||
|
|
||||||
|
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)
|
||||||
|
Console << red << "* ERROR: You must modify ssid/password in order" << endl
|
||||||
|
<< " to be able to connect to your Wifi network." << endl;
|
||||||
|
Console << endl << white;
|
||||||
|
|
||||||
|
Console << "Connecting to '" << ssid << "' ";
|
||||||
|
|
||||||
|
WiFi.mode(WIFI_STA);
|
||||||
|
WiFi.begin(ssid, password);
|
||||||
|
WiFi.mode(WIFI_STA);
|
||||||
|
WiFi.begin(ssid, password);
|
||||||
|
|
||||||
|
while (WiFi.status() != WL_CONNECTED)
|
||||||
|
{ Console << '-'; delay(500); }
|
||||||
|
|
||||||
|
Console << endl << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||||
|
|
||||||
|
const char* name="tinytest";
|
||||||
|
Console << "Starting MDNS, name= " << name;
|
||||||
|
|
||||||
|
MqttBroker* broker = new MqttBroker(1883);
|
||||||
|
broker->begin();
|
||||||
|
brokers["broker"] = broker;
|
||||||
|
|
||||||
|
if (Console.isTerm()) onCommand("every 333 view");
|
||||||
|
}
|
||||||
|
|
||||||
|
string getword(string& str, const char* if_empty=nullptr, char sep=' ');
|
||||||
|
|
||||||
|
int getint(string& str, const int if_empty=0)
|
||||||
|
{
|
||||||
|
string str2=str;
|
||||||
|
string sword = getword(str);
|
||||||
|
if (sword[0] and isdigit(sword[0]))
|
||||||
|
{
|
||||||
|
int ret=atoi(sword.c_str());
|
||||||
|
while(isdigit(sword[0]) or sword[0]==' ') sword.erase(0,1);
|
||||||
|
if (sword.length()) str = sword+' '+str;
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
str=str2;
|
||||||
|
return if_empty;
|
||||||
|
}
|
||||||
|
|
||||||
|
string getword(string& str, const char* if_empty/*=nullptr*/, char sep/*=' '*/)
|
||||||
|
{
|
||||||
|
char quote=(str[0]=='"' or str[0]=='\'' ? str[0] : 0);
|
||||||
|
if (quote) str.erase(0,1);
|
||||||
|
string sword;
|
||||||
|
while(str.length() and (str[0]!=sep or quote))
|
||||||
|
{
|
||||||
|
if (str[0]==quote)
|
||||||
|
{
|
||||||
|
str.erase(0,1);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
sword += str[0];
|
||||||
|
str.erase(0,1);
|
||||||
|
}
|
||||||
|
while(str[0]==sep) str.erase(0,1);
|
||||||
|
if (if_empty and sword.length()==0) return if_empty;
|
||||||
|
if (quote==false and sword.length()>=4 and sword.substr(0,3)=="rnd")
|
||||||
|
{
|
||||||
|
sword.erase(0,3);
|
||||||
|
if (sword[0]=='(')
|
||||||
|
{
|
||||||
|
int to = 100;
|
||||||
|
sword.erase(0,1);
|
||||||
|
int from=getint(sword);
|
||||||
|
if (sword[0]==',')
|
||||||
|
{
|
||||||
|
sword.erase(0,1);
|
||||||
|
to = getint(sword);
|
||||||
|
if (sword[0]!=')') Console << "Missing ')'" << endl;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
to=from;
|
||||||
|
from=0;
|
||||||
|
}
|
||||||
|
return String(random(from,to)).c_str();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
Console << "Missing '('" << endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
while(str[0]==' ') str.erase(0,1);
|
||||||
|
return sword;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool isaddr(string s)
|
||||||
|
{
|
||||||
|
if (s.length()==0 or s.length()>3) return false;
|
||||||
|
for(char c: s)
|
||||||
|
if (c<'0' or c>'9') return false;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
string getip(string& str, const char* if_empty=nullptr, char sep=' ')
|
||||||
|
{
|
||||||
|
string addr=getword(str, if_empty, sep);
|
||||||
|
string ip=addr;
|
||||||
|
std::vector<string> build;
|
||||||
|
while(ip.length())
|
||||||
|
{
|
||||||
|
string b=getword(ip,nullptr,'.');
|
||||||
|
if (isaddr(b) && build.size()<4)
|
||||||
|
{
|
||||||
|
build.push_back(b);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
return addr;
|
||||||
|
}
|
||||||
|
IPAddress local=WiFi.localIP();
|
||||||
|
addr.clear();
|
||||||
|
while(build.size()!=4)
|
||||||
|
{
|
||||||
|
std::stringstream b;
|
||||||
|
b << (int)local[3-build.size()];
|
||||||
|
build.insert(build.begin(), b.str().c_str());
|
||||||
|
}
|
||||||
|
for(string s: build)
|
||||||
|
{
|
||||||
|
if (addr.length()) addr += '.';
|
||||||
|
addr += s;
|
||||||
|
}
|
||||||
|
Console << "connect address: " << addr << endl;
|
||||||
|
return addr;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::set<string> commands = {
|
||||||
|
"broker", "blink", "client", "connect",
|
||||||
|
"create", "delete", "debug", "help", "interval",
|
||||||
|
"list", "ls", "ip", "off", "on", "set",
|
||||||
|
"publish", "reset", "subscribe", "unsubscribe", "view", "echo", "every"
|
||||||
|
};
|
||||||
|
|
||||||
|
void convertToCommand(string& search)
|
||||||
|
{
|
||||||
|
while(search[0]==' ') search.erase(0,1);
|
||||||
|
if (search.length()==0) return;
|
||||||
|
string matches;
|
||||||
|
int count=0;
|
||||||
|
for(string cmd: commands)
|
||||||
|
{
|
||||||
|
if (cmd.substr(0, search.length()) == search)
|
||||||
|
{
|
||||||
|
if (count) matches +=", ";
|
||||||
|
count++;
|
||||||
|
matches += cmd;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (count==1)
|
||||||
|
search = matches;
|
||||||
|
else if (count>1)
|
||||||
|
{
|
||||||
|
Console << "Ambiguous command: " << matches << endl;
|
||||||
|
search.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool compare(string s, const char* cmd)
|
||||||
|
{
|
||||||
|
uint8_t p=0;
|
||||||
|
while(s[p++]==*cmd++)
|
||||||
|
{
|
||||||
|
if (*cmd==0 or s[p]==0) return true;
|
||||||
|
if (s[p]==' ') return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
using ClientFunction = void(*)(string& cmd, MqttClient* publish);
|
||||||
|
|
||||||
|
struct Every
|
||||||
|
{
|
||||||
|
string cmd;
|
||||||
|
uint32_t ms;
|
||||||
|
uint32_t next;
|
||||||
|
uint32_t underrun=0;
|
||||||
|
bool active=true;
|
||||||
|
|
||||||
|
void dump()
|
||||||
|
{
|
||||||
|
if (active)
|
||||||
|
Console << green << "enabled";
|
||||||
|
else
|
||||||
|
Console << red << "disabled";
|
||||||
|
|
||||||
|
auto mill=millis();
|
||||||
|
Console << white << ms << "ms [" << cmd << "] next in ";
|
||||||
|
if (mill > next)
|
||||||
|
Console << "now";
|
||||||
|
else
|
||||||
|
Console << next-mill << "ms";
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
uint32_t blink_ms_on[16];
|
||||||
|
uint32_t blink_ms_off[16];
|
||||||
|
uint32_t blink_next[16];
|
||||||
|
bool blink_state[16];
|
||||||
|
int16_t blink;
|
||||||
|
|
||||||
|
std::vector<Every> everies;
|
||||||
|
|
||||||
|
void clientConnect(MqttClient* client, string& cmd)
|
||||||
|
{
|
||||||
|
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(string& cmd)
|
||||||
|
{
|
||||||
|
while(cmd.length())
|
||||||
|
{
|
||||||
|
MqttError retval = MqttOk;
|
||||||
|
|
||||||
|
string s;
|
||||||
|
MqttBroker* broker = nullptr;
|
||||||
|
MqttClient* client = nullptr;
|
||||||
|
|
||||||
|
// client.function notation
|
||||||
|
if (cmd.find('.') != string::npos &&
|
||||||
|
cmd.find('.') < cmd.find(' '))
|
||||||
|
{
|
||||||
|
s=getword(cmd, nullptr, '.');
|
||||||
|
|
||||||
|
if (s.length())
|
||||||
|
{
|
||||||
|
if (clients.find(s) != clients.end())
|
||||||
|
{
|
||||||
|
client = clients[s];
|
||||||
|
}
|
||||||
|
else if (brokers.find(s) != brokers.end())
|
||||||
|
{
|
||||||
|
broker = brokers[s];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
Console << red << "Unknown class (" << s.c_str() << ")" << white << endl;
|
||||||
|
cmd.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
s = getword(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)
|
||||||
|
{
|
||||||
|
s = getword(cmd);
|
||||||
|
if (clients.find(s) != clients.end())
|
||||||
|
{
|
||||||
|
client = clients[s];
|
||||||
|
}
|
||||||
|
else if (brokers.find(s) != brokers.end())
|
||||||
|
{
|
||||||
|
broker = brokers[s];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
Console << red << "Unable to find (" << s.c_str() << ")" << white << endl;
|
||||||
|
}
|
||||||
|
if (client)
|
||||||
|
{
|
||||||
|
for (auto it: clients)
|
||||||
|
{
|
||||||
|
if (it.second != client) continue;
|
||||||
|
Console << "deleted" << endl;
|
||||||
|
delete (it.second);
|
||||||
|
clients.erase(it.first);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
cmd += " ls";
|
||||||
|
}
|
||||||
|
else if (broker)
|
||||||
|
{
|
||||||
|
for(auto it: brokers)
|
||||||
|
{
|
||||||
|
if (broker != it.second) continue;
|
||||||
|
Console << "deleted" << endl;
|
||||||
|
delete (it.second);
|
||||||
|
brokers.erase(it.first);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
cmd += " ls";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
Console << "Nothing to delete" << endl;
|
||||||
|
}
|
||||||
|
else if (broker)
|
||||||
|
{
|
||||||
|
if (compare(s,"connect"))
|
||||||
|
{
|
||||||
|
Console << "NYI" << endl;
|
||||||
|
}
|
||||||
|
else if (compare(s, "view"))
|
||||||
|
{
|
||||||
|
broker->dump();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
Console << "Unknown broker command (" << s << ")" << endl;
|
||||||
|
s.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (client)
|
||||||
|
{
|
||||||
|
if (compare(s,"connect"))
|
||||||
|
{
|
||||||
|
clientConnect(client, cmd);
|
||||||
|
}
|
||||||
|
else if (compare(s,"publish"))
|
||||||
|
{
|
||||||
|
retval = client->publish(getword(cmd), getword(cmd));
|
||||||
|
}
|
||||||
|
else if (compare(s,"subscribe"))
|
||||||
|
{
|
||||||
|
client->subscribe(getword(cmd));
|
||||||
|
}
|
||||||
|
else if (compare(s, "unsubscribe"))
|
||||||
|
{
|
||||||
|
client->unsubscribe(getword(cmd));
|
||||||
|
}
|
||||||
|
else if (compare(s, "view"))
|
||||||
|
{
|
||||||
|
client->dump();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
Console << "Unknown client command (" << s << ")" << endl;
|
||||||
|
s.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(s, "on"))
|
||||||
|
{
|
||||||
|
uint8_t pin=getint(cmd, 2);
|
||||||
|
pinMode(pin, OUTPUT);
|
||||||
|
digitalWrite(pin, HIGH);
|
||||||
|
}
|
||||||
|
else if (compare(s, "off"))
|
||||||
|
{
|
||||||
|
uint8_t pin=getint(cmd, 2);
|
||||||
|
pinMode(pin, OUTPUT);
|
||||||
|
digitalWrite(pin, LOW);
|
||||||
|
}
|
||||||
|
else if (compare(s, "echo"))
|
||||||
|
{
|
||||||
|
s=getword(cmd);
|
||||||
|
if (s=="on")
|
||||||
|
echo_on = true;
|
||||||
|
else if (s=="off")
|
||||||
|
echo_on = false;
|
||||||
|
else
|
||||||
|
{
|
||||||
|
Console << s << ' ';
|
||||||
|
while(cmd.length())
|
||||||
|
{
|
||||||
|
Console << getword(cmd) << ' ';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(s, "every"))
|
||||||
|
{
|
||||||
|
uint32_t ms = getint(cmd, 0);
|
||||||
|
if (ms)
|
||||||
|
{
|
||||||
|
if (cmd.length())
|
||||||
|
{
|
||||||
|
Every every;
|
||||||
|
every.ms=ms;
|
||||||
|
every.cmd=cmd;
|
||||||
|
every.next=millis()+ms;
|
||||||
|
everies.push_back(every);
|
||||||
|
every.dump();
|
||||||
|
Console << endl;
|
||||||
|
cmd.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(cmd, "off") or compare(cmd, "on"))
|
||||||
|
{
|
||||||
|
bool active=getword(cmd)=="on";
|
||||||
|
uint8_t ever=getint(cmd, 100);
|
||||||
|
uint8_t count=0;
|
||||||
|
for(auto& every: everies)
|
||||||
|
{
|
||||||
|
if (count==ever or (ever==100))
|
||||||
|
{
|
||||||
|
if (every.active != active)
|
||||||
|
{
|
||||||
|
Console << "every #" << count << (active ? " on" :" off") << endl;
|
||||||
|
every.active = active;
|
||||||
|
every.underrun = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
count++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(cmd, "list") or cmd.length()==0)
|
||||||
|
{
|
||||||
|
getword(cmd);
|
||||||
|
Console << "List of everies (ms=" << millis() << ")" << endl;
|
||||||
|
uint8_t count=0;
|
||||||
|
for(auto& every: everies)
|
||||||
|
{
|
||||||
|
Console << count << ": ";
|
||||||
|
every.dump();
|
||||||
|
Console << endl;
|
||||||
|
count++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(cmd, "remove"))
|
||||||
|
{
|
||||||
|
Console << "Removing..." << endl;
|
||||||
|
getword(cmd);
|
||||||
|
int8_t every=getint(cmd, -1);
|
||||||
|
if (every==-1 and compare(cmd, "last") and everies.size())
|
||||||
|
{
|
||||||
|
getword(cmd);
|
||||||
|
everies.erase(everies.begin()+everies.size()-1);
|
||||||
|
}
|
||||||
|
else if (every==-1 and compare(cmd, "all"))
|
||||||
|
{
|
||||||
|
getword(cmd);
|
||||||
|
everies.clear();
|
||||||
|
}
|
||||||
|
else if (everies.size() > (uint8_t)every)
|
||||||
|
{
|
||||||
|
everies.erase(everies.begin()+every);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
Console << "Bad colmmand" << endl;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
Console << "Bad command" << endl;
|
||||||
|
}
|
||||||
|
else if (compare(s, "blink"))
|
||||||
|
{
|
||||||
|
int8_t blink_nr = getint(cmd, -1);
|
||||||
|
if (blink_nr >= 0)
|
||||||
|
{
|
||||||
|
blink_ms_on[blink_nr]=getint(cmd, blink_ms_on[blink_nr]);
|
||||||
|
blink_ms_off[blink_nr]=getint(cmd, blink_ms_on[blink_nr]);
|
||||||
|
pinMode(blink_nr, OUTPUT);
|
||||||
|
blink_next[blink_nr] = millis();
|
||||||
|
Console << "Blink " << blink_nr << ' ' << (blink_ms_on[blink_nr] ? "on" : "off") << endl;
|
||||||
|
if (blink_ms_on[blink_nr])
|
||||||
|
blink |= 1<< blink_nr;
|
||||||
|
else
|
||||||
|
{
|
||||||
|
blink &= ~(1<< blink_nr);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(s, "broker"))
|
||||||
|
{
|
||||||
|
string id=getword(cmd);
|
||||||
|
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)
|
||||||
|
{
|
||||||
|
MqttBroker* broker = new MqttBroker(port);
|
||||||
|
broker->begin();
|
||||||
|
|
||||||
|
brokers[id] = broker;
|
||||||
|
Console << "new broker (" << id.c_str() << ")" << endl;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
Console << "Missing port" << endl;
|
||||||
|
cmd.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
Console << "Missing or existing broker name (" << id.c_str() << ")" << endl;
|
||||||
|
cmd.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(s, "client"))
|
||||||
|
{
|
||||||
|
string id=getword(cmd);
|
||||||
|
if (brokers.find(id) != 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, id);
|
||||||
|
clients[id]=client;
|
||||||
|
client->setCallback(onPublish);
|
||||||
|
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())
|
||||||
|
{
|
||||||
|
Console << " not found." << endl;
|
||||||
|
cmd.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
Console << "Missing or existing client name" << endl;
|
||||||
|
cmd.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(s, "set"))
|
||||||
|
{
|
||||||
|
string name(getword(cmd));
|
||||||
|
if (name.length()==0)
|
||||||
|
{
|
||||||
|
for(auto it: vars)
|
||||||
|
{
|
||||||
|
Console << " " << it.first << " -> " << it.second << endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (commands.find(name) != commands.end())
|
||||||
|
{
|
||||||
|
Console << "Reserved keyword (" << name << ")" << endl;
|
||||||
|
cmd.clear();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (cmd.length())
|
||||||
|
{
|
||||||
|
vars[name] = cmd;
|
||||||
|
cmd.clear();
|
||||||
|
}
|
||||||
|
else if (vars.find(name) != vars.end())
|
||||||
|
vars.erase(vars.find(name));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (compare(s, "ls") or compare(s, "view"))
|
||||||
|
{
|
||||||
|
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(" ");
|
||||||
|
}
|
||||||
|
|
||||||
|
Console << "--< " << brokers.size() << " brokers/s. >--" << erase_to_end << endl;
|
||||||
|
for(auto it: brokers)
|
||||||
|
{
|
||||||
|
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"))
|
||||||
|
Console << "IP: " << WiFi.localIP() << endl;
|
||||||
|
else if (compare(s,"help"))
|
||||||
|
{
|
||||||
|
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())
|
||||||
|
Console << "Unknown command (" << s.c_str() << ")" << endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (retval != MqttOk)
|
||||||
|
{
|
||||||
|
Console << "# MQTT ERROR " << retval << endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void loop()
|
||||||
|
{
|
||||||
|
auto ms=millis();
|
||||||
|
int8_t out=0;
|
||||||
|
int16_t blink_bits = blink;
|
||||||
|
uint8_t e=0;
|
||||||
|
|
||||||
|
for(auto& every: everies)
|
||||||
|
{
|
||||||
|
if (not every.active) continue;
|
||||||
|
if (every.ms && every.cmd.length() && ms > every.next)
|
||||||
|
{
|
||||||
|
string cmd(every.cmd);
|
||||||
|
eval(cmd);
|
||||||
|
every.next += every.ms;
|
||||||
|
if (ms > every.next and ms > every.underrun)
|
||||||
|
{
|
||||||
|
every.next += every.ms;
|
||||||
|
Console << yellow << "Underrun every #" << e << ", " << (ms - every.next) << "ms late" << endl;
|
||||||
|
every.underrun = ms+5000;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
e++;
|
||||||
|
}
|
||||||
|
|
||||||
|
while(blink_bits)
|
||||||
|
{
|
||||||
|
if (blink_ms_on[out] and ms > blink_next[out])
|
||||||
|
{
|
||||||
|
if (blink_state[out])
|
||||||
|
{
|
||||||
|
blink_next[out] += blink_ms_on[out];
|
||||||
|
digitalWrite(out, LOW);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
blink_next[out] += blink_ms_off[out];
|
||||||
|
digitalWrite(abs(out), HIGH);
|
||||||
|
}
|
||||||
|
blink_state[out] = not blink_state[out];
|
||||||
|
}
|
||||||
|
blink_bits >>=1;
|
||||||
|
out++;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if defined(ESP9266)
|
||||||
|
MDNS.update();
|
||||||
|
#endif
|
||||||
|
|
||||||
|
for(auto it: brokers)
|
||||||
|
it.second->loop();
|
||||||
|
|
||||||
|
for(auto it: clients)
|
||||||
|
it.second->loop();
|
||||||
|
|
||||||
|
Console.loop();
|
||||||
|
}
|
||||||
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
|
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||||
|
|
||||||
|
EXTRA_CXXFLAGS=-g3 -O0
|
||||||
|
|
||||||
APP_NAME := length-tests
|
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
|
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||||
54
tests/length-tests.todo/length-tests.ino
Normal file
54
tests/length-tests.todo/length-tests.ino
Normal file
@@ -0,0 +1,54 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
|
#include <AUnit.h>
|
||||||
|
#include <TinyMqtt.h>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TinyMqtt local unit tests.
|
||||||
|
*
|
||||||
|
* Clients are connected to pseudo remote broker
|
||||||
|
* The remote should be 127.0.0.1:1883 <--- But this does not work due to Esp network limitations
|
||||||
|
* We are using 127.0.0.1 because this is simpler to test with a single ESP
|
||||||
|
* Also, this will allow to mock and thus run Action on github
|
||||||
|
**/
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
MqttBroker broker(1883);
|
||||||
|
|
||||||
|
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
||||||
|
|
||||||
|
const char* lastPayload;
|
||||||
|
size_t lastLength;
|
||||||
|
|
||||||
|
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
||||||
|
{
|
||||||
|
if (srce)
|
||||||
|
published[srce->id()][topic]++;
|
||||||
|
lastPayload = payload;
|
||||||
|
lastLength = length;
|
||||||
|
}
|
||||||
|
|
||||||
|
test(length_decode_greater_than_127)
|
||||||
|
{
|
||||||
|
// TODO WRITE THIS TEST
|
||||||
|
// The test should verify than a mqtt message with more than 127 bytes is correctly decoded
|
||||||
|
assertEqual(1,2);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
//----------------------------------------------------------------------------
|
||||||
|
// setup() and loop()
|
||||||
|
void setup() {
|
||||||
|
delay(1000);
|
||||||
|
Serial.begin(115200);
|
||||||
|
while(!Serial);
|
||||||
|
|
||||||
|
Serial.println("=============[ NO WIFI CONNECTION TinyMqtt TESTS ]========================");
|
||||||
|
}
|
||||||
|
|
||||||
|
void loop() {
|
||||||
|
aunit::TestRunner::run();
|
||||||
|
|
||||||
|
if (Serial.available()) ESP.reset();
|
||||||
|
}
|
||||||
@@ -1,53 +0,0 @@
|
|||||||
#include <AUnit.h>
|
|
||||||
#include <TinyMqtt.h>
|
|
||||||
#include <map>
|
|
||||||
|
|
||||||
/**
|
|
||||||
* TinyMqtt local unit tests.
|
|
||||||
*
|
|
||||||
* Clients are connected to pseudo remote broker
|
|
||||||
* The remote should be 127.0.0.1:1883 <--- But this does not work due to Esp network limitations
|
|
||||||
* We are using 127.0.0.1 because this is simpler to test with a single ESP
|
|
||||||
* Also, this will allow to mock and thus run Action on github
|
|
||||||
**/
|
|
||||||
|
|
||||||
using namespace std;
|
|
||||||
|
|
||||||
MqttBroker broker(1883);
|
|
||||||
|
|
||||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
|
||||||
|
|
||||||
const char* lastPayload;
|
|
||||||
size_t lastLength;
|
|
||||||
|
|
||||||
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
|
||||||
{
|
|
||||||
if (srce)
|
|
||||||
published[srce->id()][topic]++;
|
|
||||||
lastPayload = payload;
|
|
||||||
lastLength = length;
|
|
||||||
}
|
|
||||||
|
|
||||||
test(length_decode_greater_than_127)
|
|
||||||
{
|
|
||||||
// TODO WRITE THIS TEST
|
|
||||||
// The test should verify than a mqtt message with more than 127 bytes is correctly decoded
|
|
||||||
assertEqual(1,2);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
//----------------------------------------------------------------------------
|
|
||||||
// setup() and loop()
|
|
||||||
void setup() {
|
|
||||||
delay(1000);
|
|
||||||
Serial.begin(115200);
|
|
||||||
while(!Serial);
|
|
||||||
|
|
||||||
Serial.println("=============[ NO WIFI CONNECTION TinyMqtt TESTS ]========================");
|
|
||||||
}
|
|
||||||
|
|
||||||
void loop() {
|
|
||||||
aunit::TestRunner::run();
|
|
||||||
|
|
||||||
if (Serial.available()) ESP.reset();
|
|
||||||
}
|
|
||||||
@@ -1,8 +1,10 @@
|
|||||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||||
|
|
||||||
|
include ../Makefile.opts
|
||||||
|
|
||||||
APP_NAME := local-tests
|
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
|
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||||
|
|||||||
@@ -1,152 +1,201 @@
|
|||||||
|
#include <Arduino.h>
|
||||||
#include <AUnit.h>
|
#include <AUnit.h>
|
||||||
#include <TinyMqtt.h>
|
#include <TinyMqtt.h>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* TinyMqtt local unit tests.
|
* TinyMqtt local unit tests.
|
||||||
*
|
*
|
||||||
* Clients are connected to pseudo remote broker
|
* Clients are connected to pseudo remote broker
|
||||||
* The remote should be 127.0.0.1:1883 <--- But this does not work due to Esp network limitations
|
* The remote should be 127.0.0.1:1883 <--- But this does not work due to Esp network limitations
|
||||||
* We are using 127.0.0.1 because this is simpler to test with a single ESP
|
* We are using 127.0.0.1 because this is simpler to test with a single ESP
|
||||||
* Also, this will allow to mock and thus run Action on github
|
* Also, this will allow to mock and thus run Action on github
|
||||||
**/
|
**/
|
||||||
|
|
||||||
using namespace std;
|
using string = TinyConsole::string;
|
||||||
|
|
||||||
MqttBroker broker(1883);
|
MqttBroker broker(1883);
|
||||||
|
|
||||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
std::map<string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
||||||
|
|
||||||
const char* lastPayload;
|
const char* lastPayload;
|
||||||
size_t lastLength;
|
size_t lastLength;
|
||||||
|
|
||||||
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
||||||
{
|
{
|
||||||
if (srce)
|
if (srce)
|
||||||
published[srce->id()][topic]++;
|
published[srce->id()][topic]++;
|
||||||
lastPayload = payload;
|
lastPayload = payload;
|
||||||
lastLength = length;
|
lastLength = length;
|
||||||
|
}
|
||||||
|
|
||||||
|
test(local_not_connected_by_default)
|
||||||
|
{
|
||||||
|
MqttClient client;
|
||||||
|
assertEqual(client.connected(), false);
|
||||||
|
|
||||||
|
MqttBroker broker(1883);
|
||||||
|
assertEqual(broker.connected(), false);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(local_client_should_unregister_when_destroyed)
|
test(local_client_should_unregister_when_destroyed)
|
||||||
{
|
{
|
||||||
return;
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
{
|
||||||
{
|
assertEqual(broker.clientsCount(), (size_t)0); // Ensure client is not yet connected
|
||||||
MqttClient client;
|
MqttClient client(&broker);
|
||||||
assertEqual(broker.clientsCount(), (size_t)0); // Ensure client is not yet connected
|
assertEqual(broker.clientsCount(), (size_t)1); // Ensure client is now connected
|
||||||
client.connect("127.0.0.1", 1883);
|
}
|
||||||
assertEqual(broker.clientsCount(), (size_t)1); // Ensure client is now connected
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
}
|
}
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
|
||||||
|
test(local_client_do_not_disconnect_after_publishing_and_long_inactivity)
|
||||||
|
{
|
||||||
|
EpoxyTest::set_millis(0);
|
||||||
|
MqttBroker broker(1883);
|
||||||
|
MqttClient client(&broker, "client");
|
||||||
|
MqttClient sender(&broker, "sender");
|
||||||
|
broker.loop();
|
||||||
|
|
||||||
|
client.subscribe("#");
|
||||||
|
client.subscribe("test");
|
||||||
|
client.setCallback(onPublish);
|
||||||
|
assertEqual(broker.clientsCount(), (size_t)2);
|
||||||
|
|
||||||
|
sender.publish("test", "value");
|
||||||
|
broker.loop();
|
||||||
|
|
||||||
|
EpoxyTest::add_seconds(600);
|
||||||
|
client.loop();
|
||||||
|
sender.loop();
|
||||||
|
broker.loop();
|
||||||
|
|
||||||
|
sender.publish("test", "value");
|
||||||
|
broker.loop();
|
||||||
|
sender.loop();
|
||||||
|
broker.loop();
|
||||||
|
|
||||||
|
assertEqual(broker.clientsCount(), (size_t)2);
|
||||||
|
assertEqual(sender.connected(), true);
|
||||||
|
assertEqual(client.connected(), true);
|
||||||
|
|
||||||
|
assertEqual(published.size(), (size_t)1); // client has received something
|
||||||
}
|
}
|
||||||
|
|
||||||
#if 0
|
|
||||||
test(local_connect)
|
test(local_connect)
|
||||||
{
|
{
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient client;
|
MqttClient client(&broker);
|
||||||
assertTrue(client.connected());
|
assertTrue(client.connected());
|
||||||
assertEqual(broker.clientsCount(), (size_t)1);
|
assertEqual(broker.clientsCount(), (size_t)1);
|
||||||
|
}
|
||||||
|
|
||||||
|
test(local_publish_to_nowhere)
|
||||||
|
{
|
||||||
|
published.clear();
|
||||||
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
|
MqttClient publisher;
|
||||||
|
MqttError status = publisher.publish("a/b");
|
||||||
|
assertEqual(status, MqttError::MqttNowhereToSend);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(local_publish_should_be_dispatched)
|
test(local_publish_should_be_dispatched)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient subscriber;
|
MqttClient subscriber(&broker, "");
|
||||||
subscriber.subscribe("a/b");
|
subscriber.subscribe("a/b");
|
||||||
subscriber.subscribe("a/c");
|
subscriber.subscribe("a/c");
|
||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
|
|
||||||
MqttClient publisher;
|
MqttClient publisher(&broker);
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
publisher.publish("a/c");
|
publisher.publish("a/c");
|
||||||
publisher.publish("a/c");
|
publisher.publish("a/c");
|
||||||
|
|
||||||
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
||||||
assertTrue(published[""]["a/b"] == 1);
|
assertEqual(published[""]["a/b"], 1);
|
||||||
assertTrue(published[""]["a/c"] == 2);
|
assertEqual(published[""]["a/c"], 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(local_publish_should_be_dispatched_to_local_clients)
|
test(local_publish_should_be_dispatched_to_local_clients)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient subscriber_a("A");
|
MqttClient subscriber_a(&broker, "A");
|
||||||
subscriber_a.setCallback(onPublish);
|
subscriber_a.setCallback(onPublish);
|
||||||
subscriber_a.subscribe("a/b");
|
subscriber_a.subscribe("a/b");
|
||||||
subscriber_a.subscribe("a/c");
|
subscriber_a.subscribe("a/c");
|
||||||
|
|
||||||
MqttClient subscriber_b("B");
|
MqttClient subscriber_b(&broker, "B");
|
||||||
subscriber_b.setCallback(onPublish);
|
subscriber_b.setCallback(onPublish);
|
||||||
subscriber_b.subscribe("a/b");
|
subscriber_b.subscribe("a/b");
|
||||||
|
|
||||||
MqttClient publisher;
|
MqttClient publisher(&broker);
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
publisher.publish("a/c");
|
publisher.publish("a/c");
|
||||||
|
|
||||||
assertEqual(published.size(), (size_t)2); // 2 clients have received something
|
assertEqual(published.size(), (size_t)2); // 2 clients have received something
|
||||||
assertTrue(published["A"]["a/b"] == 1);
|
assertEqual(published["A"]["a/b"], 1);
|
||||||
assertTrue(published["A"]["a/c"] == 1);
|
assertEqual(published["A"]["a/c"], 1);
|
||||||
assertTrue(published["B"]["a/b"] == 1);
|
assertEqual(published["B"]["a/b"], 1);
|
||||||
assertTrue(published["B"]["a/c"] == 0);
|
assertEqual(published["B"]["a/c"], 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(local_unsubscribe)
|
test(local_unsubscribe)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient subscriber;
|
MqttClient subscriber(&broker, "");
|
||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
subscriber.subscribe("a/b");
|
subscriber.subscribe("a/b");
|
||||||
|
|
||||||
MqttClient publisher;
|
MqttClient publisher(&broker);
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
|
|
||||||
subscriber.unsubscribe("a/b");
|
subscriber.unsubscribe("a/b");
|
||||||
|
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
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)
|
test(local_nocallback_when_destroyed)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient publisher;
|
MqttClient publisher(&broker);
|
||||||
{
|
{
|
||||||
MqttClient subscriber;
|
MqttClient subscriber(&broker);
|
||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
subscriber.subscribe("a/b");
|
subscriber.subscribe("a/b");
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
}
|
}
|
||||||
|
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
|
|
||||||
assertEqual(published.size(), (size_t)1); // Only one publish has been received
|
assertEqual(published.size(), (size_t)1); // Only one publish has been received
|
||||||
}
|
}
|
||||||
#endif
|
|
||||||
|
|
||||||
//----------------------------------------------------------------------------
|
//----------------------------------------------------------------------------
|
||||||
// setup() and loop()
|
// setup() and loop()
|
||||||
void setup() {
|
void setup() {
|
||||||
delay(1000);
|
delay(1000);
|
||||||
Serial.begin(115200);
|
Serial.begin(115200);
|
||||||
while(!Serial);
|
while(!Serial);
|
||||||
|
|
||||||
Serial.println("=============[ NO WIFI CONNECTION TinyMqtt TESTS ]========================");
|
Serial.println("=============[ LOCAL TinyMqtt TESTS ]========================");
|
||||||
}
|
}
|
||||||
|
|
||||||
void loop() {
|
void loop() {
|
||||||
aunit::TestRunner::run();
|
aunit::TestRunner::run();
|
||||||
|
|
||||||
if (Serial.available()) ESP.reset();
|
if (Serial.available()) ESP.reset();
|
||||||
}
|
}
|
||||||
|
|||||||
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.
|
||||||
|
|
||||||
|
include ../Makefile.opts
|
||||||
|
|
||||||
|
# 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
|
||||||
483
tests/network-tests/network-tests.ino
Normal file
483
tests/network-tests/network-tests.ino
Normal file
@@ -0,0 +1,483 @@
|
|||||||
|
// 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
|
||||||
|
**/
|
||||||
|
|
||||||
|
using string = TinyConsole::string;
|
||||||
|
|
||||||
|
// 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<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);
|
||||||
|
}
|
||||||
|
|
||||||
|
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 = broker.getClients()[0]->keepAlive();
|
||||||
|
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(disconnected_when_broker_is_deleted)
|
||||||
|
{
|
||||||
|
MqttBroker* broker = new MqttBroker(1883);
|
||||||
|
broker->begin();
|
||||||
|
|
||||||
|
MqttClient client;
|
||||||
|
client.connect(broker);
|
||||||
|
assertEqual(client.connected(), true);
|
||||||
|
client.publish("a", "b");
|
||||||
|
|
||||||
|
delete broker;
|
||||||
|
assertEqual(client.connected(), false);
|
||||||
|
}
|
||||||
|
|
||||||
|
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();
|
||||||
|
published.clear(); // Avoid crash in unit tests due to exit handlers
|
||||||
|
}
|
||||||
@@ -1,8 +1,10 @@
|
|||||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||||
|
|
||||||
|
include ../Makefile.opts
|
||||||
|
|
||||||
APP_NAME := nowifi-tests
|
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
|
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||||
|
|||||||
@@ -1,166 +1,283 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
|
#include <Arduino.h>
|
||||||
#include <AUnit.h>
|
#include <AUnit.h>
|
||||||
#include <TinyMqtt.h>
|
#include <TinyMqtt.h>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* TinyMqtt nowifi unit tests.
|
* TinyMqtt nowifi unit tests.
|
||||||
*
|
*
|
||||||
* No wifi connection unit tests.
|
* No wifi connection unit tests.
|
||||||
* Checks with a local broker. Clients must connect to the local broker
|
* Checks with a local broker. Clients must connect to the local broker
|
||||||
**/
|
**/
|
||||||
|
|
||||||
using namespace std;
|
using string = TinyConsole::string;
|
||||||
|
|
||||||
MqttBroker broker(1883);
|
MqttBroker broker(1883);
|
||||||
|
|
||||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
std::map<string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
||||||
|
|
||||||
char* lastPayload = nullptr;
|
char* lastPayload = nullptr;
|
||||||
size_t lastLength;
|
size_t lastLength;
|
||||||
|
|
||||||
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
||||||
{
|
{
|
||||||
if (srce)
|
if (srce)
|
||||||
published[srce->id()][topic]++;
|
published[srce->id()][topic]++;
|
||||||
|
|
||||||
if (lastPayload) free(lastPayload);
|
if (lastPayload) free(lastPayload);
|
||||||
lastPayload = strdup(payload);
|
lastPayload = strdup(payload);
|
||||||
lastLength = length;
|
lastLength = length;
|
||||||
}
|
}
|
||||||
|
|
||||||
test(nowifi_client_should_unregister_when_destroyed)
|
test(nowifi_client_should_unregister_when_destroyed)
|
||||||
{
|
{
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
{
|
{
|
||||||
MqttClient client(&broker);
|
MqttClient client(&broker);
|
||||||
assertEqual(broker.clientsCount(), (size_t)1);
|
assertEqual(broker.clientsCount(), (size_t)1);
|
||||||
}
|
}
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(nowifi_connect)
|
test(nowifi_connect)
|
||||||
{
|
{
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient client(&broker);
|
MqttClient client(&broker);
|
||||||
assertTrue(client.connected());
|
assertTrue(client.connected());
|
||||||
assertEqual(broker.clientsCount(), (size_t)1);
|
assertEqual(broker.clientsCount(), (size_t)1);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(nowifi_publish_should_be_dispatched)
|
test(nowifi_publish_should_be_dispatched)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient subscriber(&broker);
|
MqttClient subscriber(&broker);
|
||||||
subscriber.subscribe("a/b");
|
subscriber.subscribe("a/b");
|
||||||
subscriber.subscribe("a/c");
|
subscriber.subscribe("a/c");
|
||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
|
|
||||||
MqttClient publisher(&broker);
|
MqttClient publisher(&broker);
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
publisher.publish("a/c");
|
publisher.publish("a/c");
|
||||||
publisher.publish("a/c");
|
publisher.publish("a/c");
|
||||||
|
|
||||||
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
||||||
assertEqual(published[""]["a/b"], 1);
|
assertEqual(published[TINY_MQTT_DEFAULT_CLIENT_ID]["a/b"], 1);
|
||||||
assertEqual(published[""]["a/c"], 2);
|
assertEqual(published[TINY_MQTT_DEFAULT_CLIENT_ID]["a/c"], 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(nowifi_publish_should_be_dispatched_to_clients)
|
test(nowifi_publish_should_be_dispatched_to_clients)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient subscriber_a(&broker, "A");
|
MqttClient subscriber_a(&broker, "A");
|
||||||
subscriber_a.setCallback(onPublish);
|
subscriber_a.setCallback(onPublish);
|
||||||
subscriber_a.subscribe("a/b");
|
subscriber_a.subscribe("a/b");
|
||||||
subscriber_a.subscribe("a/c");
|
subscriber_a.subscribe("a/c");
|
||||||
|
|
||||||
MqttClient subscriber_b(&broker, "B");
|
MqttClient subscriber_b(&broker, "B");
|
||||||
subscriber_b.setCallback(onPublish);
|
subscriber_b.setCallback(onPublish);
|
||||||
subscriber_b.subscribe("a/b");
|
subscriber_b.subscribe("a/b");
|
||||||
|
|
||||||
MqttClient publisher(&broker);
|
MqttClient publisher(&broker);
|
||||||
publisher.publish("a/b"); // A and B should receive this
|
publisher.publish("a/b"); // A and B should receive this
|
||||||
publisher.publish("a/c"); // A should receive this
|
publisher.publish("a/c"); // A should receive this
|
||||||
|
|
||||||
assertEqual(published.size(), (size_t)2); // 2 clients have received something
|
assertEqual(published.size(), (size_t)2); // 2 clients have received something
|
||||||
assertEqual(published["A"]["a/b"], 1);
|
assertEqual(published["A"]["a/b"], 1);
|
||||||
assertEqual(published["A"]["a/c"], 1);
|
assertEqual(published["A"]["a/c"], 1);
|
||||||
assertEqual(published["B"]["a/b"], 1);
|
assertEqual(published["B"]["a/b"], 1);
|
||||||
assertEqual(published["B"]["a/c"], 0);
|
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)
|
test(nowifi_unsubscribe)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient subscriber(&broker);
|
MqttClient subscriber(&broker);
|
||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
subscriber.subscribe("a/b");
|
subscriber.subscribe("a/b");
|
||||||
|
|
||||||
MqttClient publisher(&broker);
|
MqttClient publisher(&broker);
|
||||||
publisher.publish("a/b"); // This publish is received
|
publisher.publish("a/b"); // This publish is received
|
||||||
|
|
||||||
subscriber.unsubscribe("a/b");
|
subscriber.unsubscribe("a/b");
|
||||||
|
|
||||||
publisher.publish("a/b"); // Those one, no (unsubscribed)
|
publisher.publish("a/b"); // Those one, no (unsubscribed)
|
||||||
publisher.publish("a/b");
|
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)
|
test(nowifi_nocallback_when_destroyed)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient publisher(&broker);
|
MqttClient publisher(&broker);
|
||||||
|
|
||||||
{
|
{
|
||||||
MqttClient subscriber(&broker);
|
MqttClient subscriber(&broker);
|
||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
subscriber.subscribe("a/b");
|
subscriber.subscribe("a/b");
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
}
|
}
|
||||||
|
|
||||||
publisher.publish("a/b");
|
publisher.publish("a/b");
|
||||||
|
|
||||||
assertEqual(published.size(), (size_t)1); // Only one publish has been received
|
assertEqual(published.size(), (size_t)1); // Only one publish has been received
|
||||||
}
|
}
|
||||||
|
|
||||||
test(nowifi_payload_nullptr)
|
test(nowifi_small_payload)
|
||||||
{
|
{
|
||||||
published.clear();
|
published.clear();
|
||||||
|
|
||||||
const char* payload="abcd";
|
const char* payload="abcd";
|
||||||
|
|
||||||
MqttClient subscriber(&broker);
|
MqttClient subscriber(&broker);
|
||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
subscriber.subscribe("a/b");
|
subscriber.subscribe("a/b");
|
||||||
|
|
||||||
MqttClient publisher(&broker);
|
MqttClient publisher(&broker);
|
||||||
publisher.publish("a/b", payload, strlen(payload)); // This publish is received
|
publisher.publish("a/b", payload, strlen(payload)); // This publish is received
|
||||||
|
|
||||||
// coming from MqttClient::publish(...)
|
// coming from MqttClient::publish(...)
|
||||||
assertEqual(payload, lastPayload);
|
assertEqual(payload, lastPayload);
|
||||||
assertEqual(lastLength, (size_t)4);
|
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()
|
// setup() and loop()
|
||||||
void setup() {
|
void setup() {
|
||||||
delay(1000);
|
delay(1000);
|
||||||
Serial.begin(115200);
|
Serial.begin(115200);
|
||||||
while(!Serial);
|
while(!Serial);
|
||||||
|
|
||||||
Serial.println("=============[ NO WIFI CONNECTION TinyMqtt TESTS ]========================");
|
Serial.println("=============[ NO WIFI CONNECTION TinyMqtt TESTS ]========================");
|
||||||
}
|
}
|
||||||
|
|
||||||
void loop() {
|
void loop() {
|
||||||
aunit::TestRunner::run();
|
aunit::TestRunner::run();
|
||||||
|
|
||||||
if (Serial.available()) ESP.reset();
|
if (Serial.available()) ESP.reset();
|
||||||
|
published.clear(); // Avoid crash at exit handlers
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,8 +1,10 @@
|
|||||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||||
|
|
||||||
|
include ../Makefile.opts
|
||||||
|
|
||||||
APP_NAME := string-indexer-tests
|
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
|
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||||
|
|||||||
@@ -1,13 +1,15 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
|
#include <Arduino.h>
|
||||||
#include <AUnit.h>
|
#include <AUnit.h>
|
||||||
#include <StringIndexer.h>
|
#include <StringIndexer.h>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* TinyMqtt / StringIndexer unit tests.
|
* TinyMqtt / StringIndexer unit tests.
|
||||||
*
|
*
|
||||||
**/
|
**/
|
||||||
|
|
||||||
using namespace std;
|
using string = TinyConsole::string;
|
||||||
|
|
||||||
test(indexer_empty)
|
test(indexer_empty)
|
||||||
{
|
{
|
||||||
@@ -17,93 +19,101 @@ test(indexer_empty)
|
|||||||
test(indexer_strings_deleted_should_empty_indexer)
|
test(indexer_strings_deleted_should_empty_indexer)
|
||||||
{
|
{
|
||||||
assertTrue(StringIndexer::count()==0);
|
assertTrue(StringIndexer::count()==0);
|
||||||
{
|
{
|
||||||
IndexedString one("one");
|
IndexedString one("one");
|
||||||
assertEqual(StringIndexer::count(), 1);
|
assertEqual(StringIndexer::count(), 1);
|
||||||
IndexedString two("two");
|
IndexedString two("two");
|
||||||
assertEqual(StringIndexer::count(), 2);
|
assertEqual(StringIndexer::count(), 2);
|
||||||
IndexedString three("three");
|
IndexedString three("three");
|
||||||
assertEqual(StringIndexer::count(), 3);
|
assertEqual(StringIndexer::count(), 3);
|
||||||
IndexedString four("four");
|
IndexedString four("four");
|
||||||
assertEqual(StringIndexer::count(), 4);
|
assertEqual(StringIndexer::count(), 4);
|
||||||
}
|
}
|
||||||
assertEqual(StringIndexer::count(), 0);
|
assertEqual(StringIndexer::count(), 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(indexer_same_strings_count_as_one)
|
test(indexer_same_strings_count_as_one)
|
||||||
{
|
{
|
||||||
IndexedString one ("one");
|
IndexedString one ("one");
|
||||||
IndexedString two ("one");
|
IndexedString two ("one");
|
||||||
IndexedString three("one");
|
IndexedString three("one");
|
||||||
IndexedString fourt("one");
|
IndexedString fourt("one");
|
||||||
|
|
||||||
assertEqual(StringIndexer::count(), 1);
|
assertEqual(StringIndexer::count(), 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(indexer_size_of_indexed_string)
|
test(indexer_size_of_indexed_string)
|
||||||
{
|
{
|
||||||
assertEqual(sizeof(IndexedString), (size_t)1);
|
assertEqual(sizeof(IndexedString), (size_t)1);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(indexer_different_strings_are_different)
|
test(indexer_different_strings_are_different)
|
||||||
{
|
{
|
||||||
IndexedString one("one");
|
IndexedString one("one");
|
||||||
IndexedString two("two");
|
IndexedString two("two");
|
||||||
|
|
||||||
assertFalse(one == two);
|
assertFalse(one == two);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(indexer_same_strings_should_equal)
|
test(indexer_same_strings_should_equal)
|
||||||
{
|
{
|
||||||
IndexedString one("one");
|
IndexedString one("one");
|
||||||
IndexedString two("one");
|
IndexedString two("one");
|
||||||
|
|
||||||
assertTrue(one == two);
|
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)
|
test(indexer_indexed_operator_eq)
|
||||||
{
|
{
|
||||||
IndexedString one("one");
|
IndexedString one("one");
|
||||||
{
|
{
|
||||||
IndexedString same = one;
|
IndexedString same = one;
|
||||||
assertTrue(one == same);
|
assertTrue(one == same);
|
||||||
assertEqual(StringIndexer::count(), 1);
|
assertEqual(StringIndexer::count(), 1);
|
||||||
}
|
}
|
||||||
assertEqual(StringIndexer::count(), 1);
|
assertEqual(StringIndexer::count(), 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(indexer_get_string)
|
test(indexer_get_string)
|
||||||
{
|
{
|
||||||
std::string sone("one");
|
string sone("one");
|
||||||
IndexedString one(sone);
|
IndexedString one(sone);
|
||||||
|
|
||||||
assertTrue(sone==one.str());
|
assertTrue(sone==one.str());
|
||||||
}
|
}
|
||||||
|
|
||||||
test(indexer_get_index)
|
test(indexer_get_index)
|
||||||
{
|
{
|
||||||
IndexedString one1("one");
|
IndexedString one1("one");
|
||||||
IndexedString one2("one");
|
IndexedString one2("one");
|
||||||
IndexedString two1("two");
|
IndexedString two1("two");
|
||||||
IndexedString two2("two");
|
IndexedString two2("two");
|
||||||
|
|
||||||
assertTrue(one1.getIndex() == one2.getIndex());
|
assertTrue(one1.getIndex() == one2.getIndex());
|
||||||
assertTrue(two1.getIndex() == two2.getIndex());
|
assertTrue(two1.getIndex() == two2.getIndex());
|
||||||
assertTrue(one1.getIndex() != two1.getIndex());
|
assertTrue(one1.getIndex() != two1.getIndex());
|
||||||
}
|
}
|
||||||
|
|
||||||
//----------------------------------------------------------------------------
|
//----------------------------------------------------------------------------
|
||||||
// setup() and loop()
|
// setup() and loop()
|
||||||
void setup() {
|
void setup() {
|
||||||
delay(1000);
|
delay(1000);
|
||||||
Serial.begin(115200);
|
Serial.begin(115200);
|
||||||
while(!Serial);
|
while(!Serial);
|
||||||
|
|
||||||
Serial.println("=============[ TinyMqtt StringIndexer TESTS ]========================");
|
Serial.println("=============[ TinyMqtt StringIndexer TESTS ]========================");
|
||||||
}
|
}
|
||||||
|
|
||||||
void loop() {
|
void loop() {
|
||||||
aunit::TestRunner::run();
|
aunit::TestRunner::run();
|
||||||
|
|
||||||
// if (Serial.available()) ESP.reset();
|
// if (Serial.available()) ESP.reset();
|
||||||
}
|
}
|
||||||
|
|||||||
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.
|
||||||
|
|
||||||
|
include ../Makefile.opts
|
||||||
|
|
||||||
|
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
|
||||||
82
tests/topic-tests/topic-tests.ino
Normal file
82
tests/topic-tests/topic-tests.ino
Normal file
@@ -0,0 +1,82 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
|
#include <Arduino.h>
|
||||||
|
#include <AUnit.h>
|
||||||
|
#include <TinyMqtt.h>
|
||||||
|
#include <map>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TinyMqtt / StringIndexer unit tests.
|
||||||
|
*
|
||||||
|
**/
|
||||||
|
|
||||||
|
bool testTopicMatch(const char* a, const char* b, bool expected)
|
||||||
|
{
|
||||||
|
Topic ta(a);
|
||||||
|
Topic tb(b);
|
||||||
|
bool match(ta.matches(tb));
|
||||||
|
std::cout << " " << ta.c_str() << ' ';
|
||||||
|
if (match != expected)
|
||||||
|
std::cout << (expected ? " should match " : " should not match ");
|
||||||
|
else
|
||||||
|
std::cout << (expected ? " matches " : " unmatches ");
|
||||||
|
std::cout << tb.c_str() << std::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