Compare commits
41 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
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 |
1
.github/workflows/aunit.yml
vendored
1
.github/workflows/aunit.yml
vendored
@@ -17,6 +17,7 @@ jobs:
|
|||||||
- name: Setup
|
- name: Setup
|
||||||
run: |
|
run: |
|
||||||
cd ..
|
cd ..
|
||||||
|
git clone https://github.com/hsaturn/TinyConsole
|
||||||
git clone https://github.com/bxparks/EpoxyDuino
|
git clone https://github.com/bxparks/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/bxparks/AUnit
|
||||||
|
|||||||
20
README.md
20
README.md
@@ -17,11 +17,20 @@ TinyMqtt is a small, fast and capable Mqtt Broker and Client for Esp8266 / Esp32
|
|||||||
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 +42,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 +69,7 @@ no need for having tons of clients (also RAM is the problem with many clients)
|
|||||||
* ~~MqttClient auto re-subscribe (::resubscribe works bad on broker.emqx.io)~~
|
* ~~MqttClient auto 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)
|
||||||
|
|
||||||
|
|||||||
@@ -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,4 +1,5 @@
|
|||||||
#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)
|
||||||
*
|
*
|
||||||
@@ -55,7 +56,7 @@ void setup()
|
|||||||
|
|
||||||
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,5 +1,5 @@
|
|||||||
name=TinyMqtt
|
name=TinyMqtt
|
||||||
version=0.8.0
|
version=0.9.9
|
||||||
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, <hsaturn@gmail.com>
|
||||||
sentence=A tiny broker and client library for MQTT messaging.
|
sentence=A tiny broker and client library for MQTT messaging.
|
||||||
@@ -7,5 +7,5 @@ paragraph=MQTT is a lightweight messaging protocol ideal for small devices. This
|
|||||||
category=Communication
|
category=Communication
|
||||||
url=https://github.com/hsaturn/TinyMqtt
|
url=https://github.com/hsaturn/TinyMqtt
|
||||||
architectures=*
|
architectures=*
|
||||||
|
depends=TinyConsole
|
||||||
includes=TinyMqtt.h
|
includes=TinyMqtt.h
|
||||||
depends=AsyncTCP
|
|
||||||
|
|||||||
@@ -1,413 +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
|
|
||||||
*/
|
|
||||||
#pragma once
|
|
||||||
|
|
||||||
#ifndef ARDUINO_STREAMING
|
|
||||||
#define ARDUINO_STREAMING
|
|
||||||
|
|
||||||
#if (defined(ARDUINO) && ARDUINO >= 100) || defined(EPOXY_DUINO)
|
|
||||||
#include "Arduino.h"
|
|
||||||
#else
|
|
||||||
#ifndef STREAMING_CONSOLE
|
|
||||||
#include "WProgram.h"
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <string>
|
|
||||||
|
|
||||||
#if defined(ARDUINO_ARCH_AVR) || defined(ARDUINO_ARCH_MEGAAVR)
|
|
||||||
// No stl library, so need trivial version of std::is_signed ...
|
|
||||||
namespace std {
|
|
||||||
template<typename T>
|
|
||||||
struct is_signed { static const bool value = false; };
|
|
||||||
template<>
|
|
||||||
struct is_signed<int8_t> { static const bool value = true; };
|
|
||||||
template<>
|
|
||||||
struct is_signed<int16_t> { static const bool value = true; };
|
|
||||||
template<>
|
|
||||||
struct is_signed<int32_t> { static const bool value = true; };
|
|
||||||
};
|
|
||||||
#else
|
|
||||||
#include <type_traits>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#define STREAMING_LIBRARY_VERSION 6
|
|
||||||
|
|
||||||
#if !defined(typeof)
|
|
||||||
#define typeof(x) __typeof__(x)
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// PrintBuffer implementation of Print, a small buffer to print in
|
|
||||||
// see its use with pad_float()
|
|
||||||
template <size_t N>
|
|
||||||
class PrintBuffer : public Print
|
|
||||||
{
|
|
||||||
size_t pos = 0;
|
|
||||||
char str[N] {};
|
|
||||||
public:
|
|
||||||
inline const char *operator() ()
|
|
||||||
{ return str; };
|
|
||||||
|
|
||||||
// inline void clear()
|
|
||||||
// { pos = 0; str[0] = '\0'; };
|
|
||||||
|
|
||||||
inline size_t write(uint8_t c)
|
|
||||||
{ return write(&c, 1); };
|
|
||||||
|
|
||||||
inline size_t write(const uint8_t *buffer, size_t size)
|
|
||||||
{
|
|
||||||
size_t s = std::min(size, N-1 - pos); // need a /0 left
|
|
||||||
if (s)
|
|
||||||
{
|
|
||||||
memcpy(&str[pos], buffer, s);
|
|
||||||
pos += s;
|
|
||||||
}
|
|
||||||
return s;
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
// Generic template
|
|
||||||
template<class T>
|
|
||||||
inline Print &operator <<(Print &stream, const T &arg)
|
|
||||||
{ stream.print(arg); return stream; }
|
|
||||||
|
|
||||||
// TODO sfinae maybe could do the trick ?
|
|
||||||
inline Print &operator <<(Print &stream, const std::string &str)
|
|
||||||
{ stream.print(str.c_str()); return stream; }
|
|
||||||
|
|
||||||
template<typename T>
|
|
||||||
struct _BASED
|
|
||||||
{
|
|
||||||
T val;
|
|
||||||
int base;
|
|
||||||
_BASED(T v, int b): val(v), base(b)
|
|
||||||
{}
|
|
||||||
};
|
|
||||||
|
|
||||||
#if ARDUINO >= 100
|
|
||||||
|
|
||||||
struct _BYTE_CODE
|
|
||||||
{
|
|
||||||
byte val;
|
|
||||||
_BYTE_CODE(byte v) : val(v)
|
|
||||||
{}
|
|
||||||
};
|
|
||||||
#define _BYTE(a) _BYTE_CODE(a)
|
|
||||||
|
|
||||||
inline Print &operator <<(Print &obj, const _BYTE_CODE &arg)
|
|
||||||
{ obj.write(arg.val); return obj; }
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
#define _BYTE(a) _BASED<typeof(a)>(a, BYTE)
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#define _HEX(a) _BASED<typeof(a)>(a, HEX)
|
|
||||||
#define _DEC(a) _BASED<typeof(a)>(a, DEC)
|
|
||||||
#define _OCT(a) _BASED<typeof(a)>(a, OCT)
|
|
||||||
#define _BIN(a) _BASED<typeof(a)>(a, BIN)
|
|
||||||
|
|
||||||
// Specialization for class _BASED
|
|
||||||
// Thanks to Arduino forum user Ben Combee who suggested this
|
|
||||||
// clever technique to allow for expressions like
|
|
||||||
// Serial << _HEX(a);
|
|
||||||
|
|
||||||
template<typename T>
|
|
||||||
inline Print &operator <<(Print &obj, const _BASED<T> &arg)
|
|
||||||
{ obj.print(arg.val, arg.base); return obj; }
|
|
||||||
|
|
||||||
#if ARDUINO >= 18 || defined(EPOXY_DUINO)
|
|
||||||
// Specialization for class _FLOAT
|
|
||||||
// Thanks to Michael Margolis for suggesting a way
|
|
||||||
// to accommodate Arduino 0018's floating point precision
|
|
||||||
// feature like this:
|
|
||||||
// Serial << _FLOAT(gps_latitude, 6); // 6 digits of precision
|
|
||||||
|
|
||||||
struct _FLOAT
|
|
||||||
{
|
|
||||||
double val; // only Print::print(double)
|
|
||||||
int digits;
|
|
||||||
_FLOAT(double v, int d): val(v), digits(d)
|
|
||||||
{}
|
|
||||||
};
|
|
||||||
|
|
||||||
inline Print &operator <<(Print &obj, const _FLOAT &arg)
|
|
||||||
{ obj.print(arg.val, arg.digits); return obj; }
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Specialization for enum _EndLineCode
|
|
||||||
// Thanks to Arduino forum user Paul V. who suggested this
|
|
||||||
// clever technique to allow for expressions like
|
|
||||||
// Serial << "Hello!" << endl;
|
|
||||||
|
|
||||||
enum _EndLineCode { endl };
|
|
||||||
|
|
||||||
inline Print &operator <<(Print &obj, _EndLineCode)
|
|
||||||
{ obj.println(); return obj; }
|
|
||||||
|
|
||||||
// Specialization for padding & filling, mainly utilized
|
|
||||||
// by the width printers
|
|
||||||
//
|
|
||||||
// Use like
|
|
||||||
// Serial << _PAD(10,' '); // Will output 10 spaces
|
|
||||||
// Serial << _PAD(4, '0'); // Will output 4 zeros
|
|
||||||
struct _PAD
|
|
||||||
{
|
|
||||||
int8_t width;
|
|
||||||
char chr;
|
|
||||||
_PAD(int8_t w, char c) : width(w), chr(c) {}
|
|
||||||
};
|
|
||||||
|
|
||||||
inline Print &operator <<(Print& stm, const _PAD &arg)
|
|
||||||
{
|
|
||||||
for(int8_t i = 0; i < arg.width; i++)
|
|
||||||
stm.print(arg.chr);
|
|
||||||
return stm;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Specialization for width printing
|
|
||||||
//
|
|
||||||
// Use like Result
|
|
||||||
// -------- ------
|
|
||||||
// Serial << _WIDTH(1,5) " 1"
|
|
||||||
// Serial << _WIDTH(10,5) " 10"
|
|
||||||
// Serial << _WIDTH(100,5) " 100"
|
|
||||||
// Serial << _WIDTHZ(1,5) "00001"
|
|
||||||
//
|
|
||||||
// Great for times & dates, or hex dumps
|
|
||||||
//
|
|
||||||
// Serial << _WIDTHZ(hour,2) << ':' << _WIDTHZ(min,2) << ':' << _WIDTHZ(sec,2)
|
|
||||||
//
|
|
||||||
// for(int index=0; index<byte_array_size; index++)
|
|
||||||
// Serial << _WIDTHZ(_HEX(byte_array[index]))
|
|
||||||
|
|
||||||
template<typename T>
|
|
||||||
struct __WIDTH
|
|
||||||
{
|
|
||||||
const T val;
|
|
||||||
int8_t width;
|
|
||||||
char pad;
|
|
||||||
__WIDTH(const T& v, int8_t w, char p) : val(v), width(w), pad(p) {}
|
|
||||||
};
|
|
||||||
|
|
||||||
// Count digits in an integer of specific base
|
|
||||||
template<typename T>
|
|
||||||
inline uint8_t digits(T v, int8_t base = 10)
|
|
||||||
{
|
|
||||||
uint8_t digits = 0;
|
|
||||||
if ( std::is_signed<T>::value )
|
|
||||||
{
|
|
||||||
if ( v < 0 )
|
|
||||||
{
|
|
||||||
digits++;
|
|
||||||
v = -v; // v needs to be postive for the digits counter to work
|
|
||||||
}
|
|
||||||
}
|
|
||||||
do
|
|
||||||
{
|
|
||||||
v /= base;
|
|
||||||
digits++;
|
|
||||||
} while( v > 0 );
|
|
||||||
return digits;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generic get the width of a value in base 10
|
|
||||||
template<typename T>
|
|
||||||
inline uint8_t get_value_width(T val)
|
|
||||||
{ return digits(val); }
|
|
||||||
|
|
||||||
inline uint8_t get_value_width(const char * val)
|
|
||||||
{ return strlen(val); }
|
|
||||||
|
|
||||||
#ifdef ARDUINO
|
|
||||||
inline uint8_t get_value_width(const __FlashStringHelper * val)
|
|
||||||
{ return strlen_P(reinterpret_cast<const char *>(val)); }
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// _BASED<T> get the width of a value
|
|
||||||
template<typename T>
|
|
||||||
inline uint8_t get_value_width(_BASED<T> b)
|
|
||||||
{ return digits(b.val, b.base); }
|
|
||||||
|
|
||||||
// Constructor wrapper to allow automatic template parameter deduction
|
|
||||||
template<typename T>
|
|
||||||
__WIDTH<T> _WIDTH(T val, int8_t width) { return __WIDTH<T>(val, width, ' '); }
|
|
||||||
template<typename T>
|
|
||||||
__WIDTH<T> _WIDTHZ(T val, int8_t width) { return __WIDTH<T>(val, width, '0'); }
|
|
||||||
|
|
||||||
|
|
||||||
// Operator overload to handle width printing.
|
|
||||||
template<typename T>
|
|
||||||
inline Print &operator <<(Print &stm, const __WIDTH<T> &arg)
|
|
||||||
{ stm << _PAD(arg.width - get_value_width(arg.val), arg.pad) << arg.val; return stm; }
|
|
||||||
|
|
||||||
// explicit Operator overload to handle width printing of _FLOAT, double and float
|
|
||||||
template<typename T>
|
|
||||||
inline Print &pad_float(Print &stm, const __WIDTH<T> &arg, const double val, const int digits = 2) // see Print::print(double, int = 2)
|
|
||||||
{
|
|
||||||
PrintBuffer<32> buf; // it's only ~45B on the stack, no allocation, leak or fragmentation
|
|
||||||
size_t size = buf.print(val, digits); // print in buf
|
|
||||||
return stm << _PAD(arg.width - size, arg.pad) << buf(); // pad and concat what's in buf
|
|
||||||
}
|
|
||||||
|
|
||||||
inline Print &operator <<(Print &stm, const __WIDTH<float> &arg)
|
|
||||||
{ return pad_float(stm, arg, arg.val); }
|
|
||||||
|
|
||||||
inline Print &operator <<(Print &stm, const __WIDTH<double> &arg)
|
|
||||||
{ return pad_float(stm, arg, arg.val); }
|
|
||||||
|
|
||||||
inline Print &operator <<(Print &stm, const __WIDTH<_FLOAT> &arg)
|
|
||||||
{ auto& f = arg.val; return pad_float(stm, arg, f.val, f.digits); }
|
|
||||||
|
|
||||||
// a less verbose _FLOATW for _WIDTH(_FLOAT)
|
|
||||||
#define _FLOATW(val, digits, width) _WIDTH<_FLOAT>(_FLOAT((val), (digits)), (width))
|
|
||||||
|
|
||||||
// Specialization for replacement formatting
|
|
||||||
//
|
|
||||||
// Designed to be similar to printf that everyone knows and loves/hates. But without
|
|
||||||
// the internal buffers and type agnosticism. This version only has placeholders in
|
|
||||||
// the format string, the actual values are supplied using the stream safe operators
|
|
||||||
// defined in this library.
|
|
||||||
//
|
|
||||||
// Use like this:
|
|
||||||
//
|
|
||||||
// Serial << FMT(F("Replace % with %"), 1, 2 )
|
|
||||||
// Serial << FMT("Time is %:%:%", _WIDTHZ(hours,2), _WIDTHZ(minutes,2), _WIDTHZ(seconds,2))
|
|
||||||
// Serial << FMT("Your score is %\\%", score); // Note the \\ to escape the % sign
|
|
||||||
|
|
||||||
// Ok, hold your hats. This is a foray into C++11's variadic template engine ...
|
|
||||||
|
|
||||||
inline char get_next_format_char(const char *& format_string)
|
|
||||||
{
|
|
||||||
char format_char = *format_string;
|
|
||||||
if ( format_char > 0 ) format_string++;
|
|
||||||
return format_char;
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef ARDUINO
|
|
||||||
inline char get_next_format_char(const __FlashStringHelper*& format_string)
|
|
||||||
{
|
|
||||||
char format_char = pgm_read_byte(format_string);
|
|
||||||
if ( format_char > 0 ) format_string = reinterpret_cast<const __FlashStringHelper*>(reinterpret_cast<const char *>(format_string)+1);
|
|
||||||
return format_char;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
template<typename Ft>
|
|
||||||
inline bool check_backslash(char& format_char, Ft& format_string)
|
|
||||||
{
|
|
||||||
if ( format_char == '\\')
|
|
||||||
{
|
|
||||||
format_char = get_next_format_char(format_string);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// The template tail printer helper
|
|
||||||
template<typename Ft, typename... Ts>
|
|
||||||
struct __FMT
|
|
||||||
{
|
|
||||||
Ft format_string;
|
|
||||||
__FMT(Ft f, Ts ... args) : format_string(f) {}
|
|
||||||
inline void tstreamf(Print& stm, Ft format) const
|
|
||||||
{
|
|
||||||
while(char c = get_next_format_char(format))
|
|
||||||
{
|
|
||||||
check_backslash(c, format);
|
|
||||||
if ( c )
|
|
||||||
stm.print(c);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// The variadic template helper
|
|
||||||
template<typename Ft, typename T, typename... Ts>
|
|
||||||
struct __FMT<Ft, T, Ts...> : __FMT<Ft, Ts...>
|
|
||||||
{
|
|
||||||
T val;
|
|
||||||
__FMT(Ft f, T t, Ts... ts) : __FMT<Ft, Ts...>(f, ts...), val(t) {}
|
|
||||||
inline void tstreamf(Print& stm, Ft format) const
|
|
||||||
{
|
|
||||||
while(char c = get_next_format_char(format))
|
|
||||||
{
|
|
||||||
if (!check_backslash(c, format))
|
|
||||||
{
|
|
||||||
if ( c == '%')
|
|
||||||
{
|
|
||||||
stm << val;
|
|
||||||
// Variadic recursion ... compiler rolls this out during
|
|
||||||
// template argument pack expansion
|
|
||||||
__FMT<Ft, Ts...>::tstreamf(stm, format);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (c)
|
|
||||||
stm.print(c);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// The actual operator should you only instanciate the FMT
|
|
||||||
// helper with a format string and no parameters
|
|
||||||
template<typename Ft, typename... Ts>
|
|
||||||
inline Print& operator <<(Print &stm, const __FMT<Ft, Ts...> &args)
|
|
||||||
{
|
|
||||||
args.tstreamf(stm, args.format_string);
|
|
||||||
return stm;
|
|
||||||
}
|
|
||||||
|
|
||||||
// The variadic stream helper
|
|
||||||
template<typename Ft, typename T, typename... Ts>
|
|
||||||
inline Print& operator <<(Print &stm, const __FMT<Ft, T, Ts...> &args)
|
|
||||||
{
|
|
||||||
args.tstreamf(stm, args.format_string);
|
|
||||||
return stm;
|
|
||||||
}
|
|
||||||
|
|
||||||
// As we don't have C++17, we can't get a constructor to use
|
|
||||||
// automatic argument deduction, but ... this little trick gets
|
|
||||||
// around that ...
|
|
||||||
template<typename Ft, typename... Ts>
|
|
||||||
__FMT<Ft, Ts...> _FMT(Ft format, Ts ... args) { return __FMT<Ft, Ts...>(format, args...); }
|
|
||||||
|
|
||||||
#endif
|
|
||||||
@@ -1,3 +1,4 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#pragma once
|
#pragma once
|
||||||
#include <map>
|
#include <map>
|
||||||
#include <string>
|
#include <string>
|
||||||
@@ -26,32 +27,9 @@ class StringIndexer
|
|||||||
#endif
|
#endif
|
||||||
};
|
};
|
||||||
public:
|
public:
|
||||||
using index_t=uint8_t;
|
using index_t = uint8_t;
|
||||||
|
|
||||||
static index_t strToIndex(const char* str, uint8_t len)
|
static const std::string& str(const index_t& index)
|
||||||
{
|
|
||||||
for(auto it=strings.begin(); it!=strings.end(); it++)
|
|
||||||
{
|
|
||||||
if (strncmp(it->second.str.c_str(), str, len)==0)
|
|
||||||
{
|
|
||||||
it->second.used++;
|
|
||||||
return it->first;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for(index_t index=1; index; index++)
|
|
||||||
{
|
|
||||||
if (strings.find(index)==strings.end())
|
|
||||||
{
|
|
||||||
strings[index].str = std::string(str, len);
|
|
||||||
strings[index].used++;
|
|
||||||
// Serial << "Creating index " << index << " for (" << strings[index].str.c_str() << ") len=" << len << endl;
|
|
||||||
return index;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return 0; // TODO out of indexes
|
|
||||||
}
|
|
||||||
|
|
||||||
static const std::string& str(const index_t& index)
|
|
||||||
{
|
{
|
||||||
static std::string dummy;
|
static std::string dummy;
|
||||||
const auto& it=strings.find(index);
|
const auto& it=strings.find(index);
|
||||||
@@ -82,6 +60,32 @@ class StringIndexer
|
|||||||
static uint16_t count() { return strings.size(); }
|
static uint16_t count() { return strings.size(); }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
friend class IndexedString;
|
||||||
|
|
||||||
|
// increment use of str or create a new index
|
||||||
|
static index_t strToIndex(const char* str, uint8_t len)
|
||||||
|
{
|
||||||
|
for(auto it=strings.begin(); it!=strings.end(); it++)
|
||||||
|
{
|
||||||
|
if (it->second.str.length() == len && strcmp(it->second.str.c_str(), str)==0)
|
||||||
|
{
|
||||||
|
it->second.used++;
|
||||||
|
return it->first;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for(index_t index=1; index; index++)
|
||||||
|
{
|
||||||
|
if (strings.find(index)==strings.end())
|
||||||
|
{
|
||||||
|
strings[index].str = std::string(str, len);
|
||||||
|
strings[index].used++;
|
||||||
|
// Serial << "Creating index " << index << " for (" << strings[index].str.c_str() << ") len=" << len << endl;
|
||||||
|
return index;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0; // TODO out of indexes
|
||||||
|
}
|
||||||
|
|
||||||
static std::map<index_t, StringCounter> strings;
|
static std::map<index_t, StringCounter> strings;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
289
src/TinyMqtt.cpp
289
src/TinyMqtt.cpp
@@ -1,6 +1,16 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#include "TinyMqtt.h"
|
#include "TinyMqtt.h"
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
|
|
||||||
|
#if TINY_MQTT_DEBUG
|
||||||
|
static auto cyan = TinyConsole::cyan;
|
||||||
|
static auto white = TinyConsole::white;
|
||||||
|
static auto red = TinyConsole::red;
|
||||||
|
static auto yellow = TinyConsole::yellow;
|
||||||
|
|
||||||
|
int TinyMqtt::debug=2;
|
||||||
|
#endif
|
||||||
|
|
||||||
#ifdef EPOXY_DUINO
|
#ifdef EPOXY_DUINO
|
||||||
std::map<MqttMessage::Type, int> MqttClient::counters;
|
std::map<MqttMessage::Type, int> MqttClient::counters;
|
||||||
#endif
|
#endif
|
||||||
@@ -8,7 +18,7 @@
|
|||||||
MqttBroker::MqttBroker(uint16_t port)
|
MqttBroker::MqttBroker(uint16_t port)
|
||||||
{
|
{
|
||||||
server = new TcpServer(port);
|
server = new TcpServer(port);
|
||||||
#ifdef TCP_ASYNC
|
#ifdef TINY_MQTT_ASYNC
|
||||||
server->onClient(onClient, this);
|
server->onClient(onClient, this);
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
@@ -23,10 +33,11 @@ MqttBroker::~MqttBroker()
|
|||||||
}
|
}
|
||||||
|
|
||||||
// private constructor used by broker only
|
// private constructor used by broker only
|
||||||
MqttClient::MqttClient(MqttBroker* parent, TcpClient* new_client)
|
MqttClient::MqttClient(MqttBroker* local_broker, TcpClient* new_client)
|
||||||
: parent(parent)
|
: local_broker(local_broker)
|
||||||
{
|
{
|
||||||
#ifdef TCP_ASYNC
|
debug("MqttClient private with broker");
|
||||||
|
#ifdef TINY_MQTT_ASYNC
|
||||||
client = new_client;
|
client = new_client;
|
||||||
client->onData(onData, this);
|
client->onData(onData, this);
|
||||||
// client->onConnect() TODO
|
// client->onConnect() TODO
|
||||||
@@ -41,12 +52,12 @@ MqttClient::MqttClient(MqttBroker* parent, TcpClient* new_client)
|
|||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
MqttClient::MqttClient(MqttBroker* parent, const std::string& id)
|
MqttClient::MqttClient(MqttBroker* local_broker, const std::string& id)
|
||||||
: parent(parent), clientId(id)
|
: local_broker(local_broker), clientId(id)
|
||||||
{
|
{
|
||||||
client = nullptr;
|
client = nullptr;
|
||||||
|
|
||||||
if (parent) parent->addClient(this);
|
if (local_broker) local_broker->addClient(this);
|
||||||
}
|
}
|
||||||
|
|
||||||
MqttClient::~MqttClient()
|
MqttClient::~MqttClient()
|
||||||
@@ -64,59 +75,68 @@ void MqttClient::close(bool bSendDisconnect)
|
|||||||
if (bSendDisconnect and client->connected())
|
if (bSendDisconnect and client->connected())
|
||||||
{
|
{
|
||||||
message.create(MqttMessage::Type::Disconnect);
|
message.create(MqttMessage::Type::Disconnect);
|
||||||
|
message.hexdump("close");
|
||||||
message.sendTo(this);
|
message.sendTo(this);
|
||||||
}
|
}
|
||||||
client->stop();
|
client->stop();
|
||||||
}
|
}
|
||||||
|
|
||||||
if (parent)
|
if (local_broker)
|
||||||
{
|
{
|
||||||
parent->removeClient(this);
|
local_broker->removeClient(this);
|
||||||
parent = nullptr;
|
local_broker = nullptr;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void MqttClient::connect(MqttBroker* parentBroker)
|
void MqttClient::connect(MqttBroker* local)
|
||||||
{
|
{
|
||||||
|
debug("MqttClient::connect_local");
|
||||||
close();
|
close();
|
||||||
parent = parentBroker;
|
local_broker = local;
|
||||||
}
|
}
|
||||||
|
|
||||||
void MqttClient::connect(std::string broker, uint16_t port, uint16_t ka)
|
void MqttClient::connect(std::string broker, uint16_t port, uint16_t ka)
|
||||||
{
|
{
|
||||||
debug("MqttClient::connect");
|
debug("MqttClient::connect_to_host " << broker << ':' << port);
|
||||||
keep_alive = ka;
|
keep_alive = ka;
|
||||||
close();
|
close();
|
||||||
if (client) delete client;
|
if (client) delete client;
|
||||||
client = new TcpClient;
|
client = new TcpClient;
|
||||||
|
|
||||||
debug("Trying to connect to " << broker.c_str() << ':' << port);
|
#ifdef TINY_MQTT_ASYNC
|
||||||
#ifdef TCP_ASYNC
|
|
||||||
client->onData(onData, this);
|
client->onData(onData, this);
|
||||||
client->onConnect(onConnect, this);
|
client->onConnect(onConnect, this);
|
||||||
client->connect(broker.c_str(), port);
|
client->connect(broker.c_str(), port, ka);
|
||||||
#else
|
#else
|
||||||
if (client->connect(broker.c_str(), port))
|
if (client->connect(broker.c_str(), port))
|
||||||
{
|
{
|
||||||
|
debug("link established");
|
||||||
onConnect(this, client);
|
onConnect(this, client);
|
||||||
}
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
debug("unable to connect.");
|
||||||
|
}
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
void MqttBroker::addClient(MqttClient* client)
|
void MqttBroker::addClient(MqttClient* client)
|
||||||
{
|
{
|
||||||
clients.push_back(client);
|
debug("MqttBroker::addClient");
|
||||||
|
clients.push_back(client);
|
||||||
}
|
}
|
||||||
|
|
||||||
void MqttBroker::connect(const std::string& host, uint16_t port)
|
void MqttBroker::connect(const std::string& host, uint16_t port)
|
||||||
{
|
{
|
||||||
|
debug("MqttBroker::connect");
|
||||||
if (broker == nullptr) broker = new MqttClient;
|
if (broker == nullptr) broker = new MqttClient;
|
||||||
broker->connect(host, port);
|
broker->connect(host, port);
|
||||||
broker->parent = this; // Because connect removed the link
|
broker->local_broker = this; // Because connect removed the link
|
||||||
}
|
}
|
||||||
|
|
||||||
void MqttBroker::removeClient(MqttClient* remove)
|
void MqttBroker::removeClient(MqttClient* remove)
|
||||||
{
|
{
|
||||||
|
debug("removeClient");
|
||||||
for(auto it=clients.begin(); it!=clients.end(); it++)
|
for(auto it=clients.begin(); it!=clients.end(); it++)
|
||||||
{
|
{
|
||||||
auto client=*it;
|
auto client=*it;
|
||||||
@@ -133,11 +153,12 @@ void MqttBroker::removeClient(MqttClient* remove)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
debug("Error cannot remove client"); // TODO should not occur
|
debug(red << "Error cannot remove client"); // TODO should not occur
|
||||||
}
|
}
|
||||||
|
|
||||||
void MqttBroker::onClient(void* broker_ptr, TcpClient* client)
|
void MqttBroker::onClient(void* broker_ptr, TcpClient* client)
|
||||||
{
|
{
|
||||||
|
debug("MqttBroker::onClient");
|
||||||
MqttBroker* broker = static_cast<MqttBroker*>(broker_ptr);
|
MqttBroker* broker = static_cast<MqttBroker*>(broker_ptr);
|
||||||
|
|
||||||
broker->addClient(new MqttClient(broker, client));
|
broker->addClient(new MqttClient(broker, client));
|
||||||
@@ -146,7 +167,7 @@ void MqttBroker::onClient(void* broker_ptr, TcpClient* client)
|
|||||||
|
|
||||||
void MqttBroker::loop()
|
void MqttBroker::loop()
|
||||||
{
|
{
|
||||||
#ifndef TCP_ASYNC
|
#ifndef TINY_MQTT_ASYNC
|
||||||
WiFiClient client = server->available();
|
WiFiClient client = server->available();
|
||||||
|
|
||||||
if (client)
|
if (client)
|
||||||
@@ -173,7 +194,7 @@ void MqttBroker::loop()
|
|||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
debug("Client " << client->id().c_str() << " Disconnected, parent=" << (dbg_ptr)client->parent);
|
debug("Client " << client->id().c_str() << " Disconnected, local_broker=" << (dbg_ptr)client->local_broker);
|
||||||
// Note: deleting a client not added by the broker itself will probably crash later.
|
// Note: deleting a client not added by the broker itself will probably crash later.
|
||||||
delete client;
|
delete client;
|
||||||
break;
|
break;
|
||||||
@@ -183,6 +204,7 @@ void MqttBroker::loop()
|
|||||||
|
|
||||||
MqttError MqttBroker::subscribe(const Topic& topic, uint8_t qos)
|
MqttError MqttBroker::subscribe(const Topic& topic, uint8_t qos)
|
||||||
{
|
{
|
||||||
|
debug("MqttBroker::subscribe");
|
||||||
if (broker && broker->connected())
|
if (broker && broker->connected())
|
||||||
{
|
{
|
||||||
return broker->subscribe(topic, qos);
|
return broker->subscribe(topic, qos);
|
||||||
@@ -194,13 +216,13 @@ MqttError MqttBroker::publish(const MqttClient* source, const Topic& topic, Mqtt
|
|||||||
{
|
{
|
||||||
MqttError retval = MqttOk;
|
MqttError retval = MqttOk;
|
||||||
|
|
||||||
debug("publish ");
|
debug("MqttBroker::publish");
|
||||||
int i=0;
|
int i=0;
|
||||||
for(auto client: clients)
|
for(auto client: clients)
|
||||||
{
|
{
|
||||||
i++;
|
i++;
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << "brk_" << (broker && broker->connected() ? "con" : "dis") <<
|
Console << __LINE__ << " broker:" << (broker && broker->connected() ? "linked" : "alone") <<
|
||||||
" srce=" << (source->isLocal() ? "loc" : "rem") << " clt#" << i << ", local=" << client->isLocal() << ", con=" << client->connected() << endl;
|
" srce=" << (source->isLocal() ? "loc" : "rem") << " clt#" << i << ", local=" << client->isLocal() << ", con=" << client->connected() << endl;
|
||||||
#endif
|
#endif
|
||||||
bool doit = false;
|
bool doit = false;
|
||||||
@@ -220,8 +242,8 @@ MqttError MqttBroker::publish(const MqttClient* source, const Topic& topic, Mqtt
|
|||||||
{
|
{
|
||||||
doit = true;
|
doit = true;
|
||||||
}
|
}
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << ", doit=" << doit << ' ';
|
Console << ", doit=" << doit << ' ';
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
if (doit) retval = client->publishIfSubscribed(topic, msg);
|
if (doit) retval = client->publishIfSubscribed(topic, msg);
|
||||||
@@ -242,16 +264,17 @@ bool MqttBroker::compareString(
|
|||||||
|
|
||||||
void MqttMessage::getString(const char* &buff, uint16_t& len)
|
void MqttMessage::getString(const char* &buff, uint16_t& len)
|
||||||
{
|
{
|
||||||
len = (buff[0]<<8)|(buff[1]);
|
len = getSize(buff);
|
||||||
buff+=2;
|
buff+=2;
|
||||||
}
|
}
|
||||||
|
|
||||||
void MqttClient::clientAlive(uint32_t more_seconds)
|
void MqttClient::clientAlive(uint32_t more_seconds)
|
||||||
{
|
{
|
||||||
|
debug("MqttClient::clientAlive");
|
||||||
if (keep_alive)
|
if (keep_alive)
|
||||||
{
|
{
|
||||||
#ifdef EPOXY_DUINO
|
#ifdef EPOXY_DUINO
|
||||||
alive=millis()+500000;
|
alive=millis()+500000+0*more_seconds;
|
||||||
#else
|
#else
|
||||||
alive=millis()+1000*(keep_alive+more_seconds);
|
alive=millis()+1000*(keep_alive+more_seconds);
|
||||||
#endif
|
#endif
|
||||||
@@ -264,11 +287,11 @@ void MqttClient::loop()
|
|||||||
{
|
{
|
||||||
if (alive && (millis() > alive))
|
if (alive && (millis() > alive))
|
||||||
{
|
{
|
||||||
if (parent)
|
if (local_broker)
|
||||||
{
|
{
|
||||||
debug("timeout client");
|
debug(red << "timeout client");
|
||||||
close();
|
close();
|
||||||
debug("closed");
|
debug(red << "closed");
|
||||||
}
|
}
|
||||||
else if (client && client->connected())
|
else if (client && client->connected())
|
||||||
{
|
{
|
||||||
@@ -281,7 +304,7 @@ void MqttClient::loop()
|
|||||||
// there is no need to send one PingReq per instance.
|
// there is no need to send one PingReq per instance.
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#ifndef TCP_ASYNC
|
#ifndef TINY_MQTT_ASYNC
|
||||||
while(client && client->available()>0)
|
while(client && client->available()>0)
|
||||||
{
|
{
|
||||||
message.incoming(client->read());
|
message.incoming(client->read());
|
||||||
@@ -297,24 +320,24 @@ void MqttClient::loop()
|
|||||||
void MqttClient::onConnect(void *mqttclient_ptr, TcpClient*)
|
void MqttClient::onConnect(void *mqttclient_ptr, TcpClient*)
|
||||||
{
|
{
|
||||||
MqttClient* mqtt = static_cast<MqttClient*>(mqttclient_ptr);
|
MqttClient* mqtt = static_cast<MqttClient*>(mqttclient_ptr);
|
||||||
debug("cnx: connecting");
|
debug("MqttClient::onConnect");
|
||||||
MqttMessage msg(MqttMessage::Type::Connect);
|
MqttMessage msg(MqttMessage::Type::Connect);
|
||||||
msg.add("MQTT",4);
|
msg.add("MQTT",4);
|
||||||
msg.add(0x4); // Mqtt protocol version 3.1.1
|
msg.add(0x4); // Mqtt protocol version 3.1.1
|
||||||
msg.add(0x0); // Connect flags TODO user / name
|
msg.add(0x0); // Connect flags TODO user / name
|
||||||
|
|
||||||
msg.add(0x00); // keep_alive
|
msg.add((char)(mqtt->keep_alive >> 8)); // keep_alive
|
||||||
msg.add((char)mqtt->keep_alive);
|
msg.add((char)(mqtt->keep_alive & 0xFF));
|
||||||
msg.add(mqtt->clientId);
|
msg.add(mqtt->clientId);
|
||||||
debug("cnx: mqtt connecting");
|
debug("cnx: mqtt connecting");
|
||||||
msg.sendTo(mqtt);
|
msg.sendTo(mqtt);
|
||||||
msg.reset();
|
msg.reset();
|
||||||
debug("cnx: mqtt sent " << (dbg_ptr)mqtt->parent);
|
debug("cnx: mqtt sent " << (dbg_ptr)mqtt->local_broker);
|
||||||
|
|
||||||
mqtt->clientAlive(0);
|
mqtt->clientAlive(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef TCP_ASYNC
|
#ifdef TINY_MQTT_ASYNC
|
||||||
void MqttClient::onData(void* client_ptr, TcpClient*, void* data, size_t len)
|
void MqttClient::onData(void* client_ptr, TcpClient*, void* data, size_t len)
|
||||||
{
|
{
|
||||||
char* char_ptr = static_cast<char*>(data);
|
char* char_ptr = static_cast<char*>(data);
|
||||||
@@ -354,29 +377,30 @@ void MqttClient::resubscribe()
|
|||||||
|
|
||||||
MqttError MqttClient::subscribe(Topic topic, uint8_t qos)
|
MqttError MqttClient::subscribe(Topic topic, uint8_t qos)
|
||||||
{
|
{
|
||||||
debug("subsribe(" << topic.c_str() << ")");
|
debug("MqttClient::subsribe(" << topic.c_str() << ")");
|
||||||
MqttError ret = MqttOk;
|
MqttError ret = MqttOk;
|
||||||
|
|
||||||
subscriptions.insert(topic);
|
subscriptions.insert(topic);
|
||||||
|
|
||||||
if (parent==nullptr) // remote broker
|
if (local_broker==nullptr) // remote broker
|
||||||
{
|
{
|
||||||
return sendTopic(topic, MqttMessage::Type::Subscribe, qos);
|
return sendTopic(topic, MqttMessage::Type::Subscribe, qos);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
return parent->subscribe(topic, qos);
|
return local_broker->subscribe(topic, qos);
|
||||||
}
|
}
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
MqttError MqttClient::unsubscribe(Topic topic)
|
MqttError MqttClient::unsubscribe(Topic topic)
|
||||||
{
|
{
|
||||||
|
debug("MqttClient::unsubscribe");
|
||||||
auto it=subscriptions.find(topic);
|
auto it=subscriptions.find(topic);
|
||||||
if (it != subscriptions.end())
|
if (it != subscriptions.end())
|
||||||
{
|
{
|
||||||
subscriptions.erase(it);
|
subscriptions.erase(it);
|
||||||
if (parent==nullptr) // remote broker
|
if (local_broker==nullptr) // remote broker
|
||||||
{
|
{
|
||||||
return sendTopic(topic, MqttMessage::Type::UnSubscribe, 0);
|
return sendTopic(topic, MqttMessage::Type::UnSubscribe, 0);
|
||||||
}
|
}
|
||||||
@@ -386,6 +410,7 @@ MqttError MqttClient::unsubscribe(Topic topic)
|
|||||||
|
|
||||||
MqttError MqttClient::sendTopic(const Topic& topic, MqttMessage::Type type, uint8_t qos)
|
MqttError MqttClient::sendTopic(const Topic& topic, MqttMessage::Type type, uint8_t qos)
|
||||||
{
|
{
|
||||||
|
debug("MqttClient::sendTopic");
|
||||||
MqttMessage msg(type, 2);
|
MqttMessage msg(type, 2);
|
||||||
|
|
||||||
// TODO manage packet identifier
|
// TODO manage packet identifier
|
||||||
@@ -401,17 +426,8 @@ MqttError MqttClient::sendTopic(const Topic& topic, MqttMessage::Type type, uint
|
|||||||
|
|
||||||
void MqttClient::processMessage(MqttMessage* mesg)
|
void MqttClient::processMessage(MqttMessage* mesg)
|
||||||
{
|
{
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::Type::PingResp)
|
|
||||||
{
|
|
||||||
#ifdef NOT_ESP_CORE
|
|
||||||
Serial << "---> INCOMING " << _HEX(mesg->type()) << " client(" << (dbg_ptr)client << ':' << clientId << ") mem=" << " ESP.getFreeHeap() "<< endl;
|
|
||||||
#else
|
|
||||||
Serial << "---> INCOMING " << _HEX(mesg->type()) << " client(" << (dbg_ptr)client << ':' << clientId << ") mem=" << ESP.getFreeHeap() << endl;
|
|
||||||
#endif
|
|
||||||
// mesg->hexdump("Incoming");
|
|
||||||
mesg->hexdump("Incoming");
|
mesg->hexdump("Incoming");
|
||||||
}
|
|
||||||
#endif
|
#endif
|
||||||
auto header = mesg->getVHeader();
|
auto header = mesg->getVHeader();
|
||||||
const char* payload;
|
const char* payload;
|
||||||
@@ -432,7 +448,7 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
|||||||
}
|
}
|
||||||
payload = header+10;
|
payload = header+10;
|
||||||
mqtt_flags = header[7];
|
mqtt_flags = header[7];
|
||||||
keep_alive = (header[8]<<8)|(header[9]);
|
keep_alive = MqttMessage::getSize(header+8);
|
||||||
if (strncmp("MQTT", header+2,4))
|
if (strncmp("MQTT", header+2,4))
|
||||||
{
|
{
|
||||||
debug("bad mqtt header");
|
debug("bad mqtt header");
|
||||||
@@ -461,18 +477,18 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
|||||||
if (mqtt_flags & FlagUserName)
|
if (mqtt_flags & FlagUserName)
|
||||||
{
|
{
|
||||||
mesg->getString(payload, len);
|
mesg->getString(payload, len);
|
||||||
if (!parent->checkUser(payload, len)) break;
|
if (not local_broker->checkUser(payload, len)) break;
|
||||||
payload += len;
|
payload += len;
|
||||||
}
|
}
|
||||||
if (mqtt_flags & FlagPassword)
|
if (mqtt_flags & FlagPassword)
|
||||||
{
|
{
|
||||||
mesg->getString(payload, len);
|
mesg->getString(payload, len);
|
||||||
if (!parent->checkPassword(payload, len)) break;
|
if (not local_broker->checkPassword(payload, len)) break;
|
||||||
payload += len;
|
payload += len;
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << "Connected client:" << clientId.c_str() << ", keep alive=" << keep_alive << '.' << endl;
|
Console << yellow << "Client " << clientId << " connected : keep alive=" << keep_alive << '.' << white << endl;
|
||||||
#endif
|
#endif
|
||||||
bclose = false;
|
bclose = false;
|
||||||
mqtt_connected=true;
|
mqtt_connected=true;
|
||||||
@@ -507,12 +523,13 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
|||||||
if (client)
|
if (client)
|
||||||
{
|
{
|
||||||
uint16_t pingreq = MqttMessage::Type::PingResp;
|
uint16_t pingreq = MqttMessage::Type::PingResp;
|
||||||
|
debug(cyan << "Ping response to client ");
|
||||||
client->write((const char*)(&pingreq), 2);
|
client->write((const char*)(&pingreq), 2);
|
||||||
bclose = false;
|
bclose = false;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
debug("internal pingreq ?");
|
debug(red << "internal pingreq ?");
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
|
|
||||||
@@ -568,8 +585,8 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
|||||||
break;
|
break;
|
||||||
|
|
||||||
case MqttMessage::Type::Publish:
|
case MqttMessage::Type::Publish:
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << "publish " << mqtt_connected << '/' << (long) client << endl;
|
Console << "publish " << mqtt_connected << '/' << (long) client << endl;
|
||||||
#endif
|
#endif
|
||||||
if (mqtt_connected or client == nullptr)
|
if (mqtt_connected or client == nullptr)
|
||||||
{
|
{
|
||||||
@@ -578,28 +595,33 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
|||||||
mesg->getString(payload, len);
|
mesg->getString(payload, len);
|
||||||
Topic published(payload, len);
|
Topic published(payload, len);
|
||||||
payload += len;
|
payload += len;
|
||||||
// Serial << "Received Publish (" << published.str().c_str() << ") size=" << (int)len
|
#if TINY_MQTT_DEBUG
|
||||||
|
Console << "Received Publish (" << published.str().c_str() << ") size=" << (int)len << endl;
|
||||||
|
#endif
|
||||||
// << '(' << std::string(payload, len).c_str() << ')' << " msglen=" << mesg->length() << endl;
|
// << '(' << std::string(payload, len).c_str() << ')' << " msglen=" << mesg->length() << endl;
|
||||||
if (qos) payload+=2; // ignore packet identifier if any
|
if (qos) payload+=2; // ignore packet identifier if any
|
||||||
len=mesg->end()-payload;
|
len=mesg->end()-payload;
|
||||||
// TODO reset DUP
|
// TODO reset DUP
|
||||||
// TODO reset RETAIN
|
// TODO reset RETAIN
|
||||||
|
|
||||||
if (parent==nullptr or client==nullptr) // internal MqttClient receives publish
|
if (local_broker==nullptr or client==nullptr) // internal MqttClient receives publish
|
||||||
{
|
{
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << (isSubscribedTo(published) ? "not" : "") << " subscribed.\n";
|
if (TinyMqtt::debug >= 2)
|
||||||
Serial << "has " << (callback ? "" : "no ") << " callback.\n";
|
{
|
||||||
|
Console << (isSubscribedTo(published) ? "not" : "") << " subscribed.\n";
|
||||||
|
Console << "has " << (callback ? "" : "no ") << " callback.\n";
|
||||||
|
}
|
||||||
#endif
|
#endif
|
||||||
if (callback and isSubscribedTo(published))
|
if (callback and isSubscribedTo(published))
|
||||||
{
|
{
|
||||||
callback(this, published, payload, len); // TODO send the real payload
|
callback(this, published, payload, len); // TODO send the real payload
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else if (parent) // from outside to inside
|
else if (local_broker) // from outside to inside
|
||||||
{
|
{
|
||||||
debug("publishing to parent");
|
debug("publishing to local_broker");
|
||||||
parent->publish(this, published, *mesg);
|
local_broker->publish(this, published, *mesg);
|
||||||
}
|
}
|
||||||
bclose = false;
|
bclose = false;
|
||||||
}
|
}
|
||||||
@@ -619,27 +641,83 @@ if (mesg->type() != MqttMessage::Type::PingReq && mesg->type() != MqttMessage::T
|
|||||||
};
|
};
|
||||||
if (bclose)
|
if (bclose)
|
||||||
{
|
{
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << "*************** Error msg 0x" << _HEX(mesg->type());
|
Console << red << "*************** Error msg 0x" << _HEX(mesg->type());
|
||||||
mesg->hexdump("-------ERROR ------");
|
mesg->hexdump("-------ERROR ------");
|
||||||
dump();
|
dump();
|
||||||
Serial << endl;
|
Console << white << endl;
|
||||||
#endif
|
#endif
|
||||||
close();
|
close();
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
clientAlive(parent ? 5 : 0);
|
clientAlive(local_broker ? 5 : 0);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Topic::matches(const Topic& topic) const
|
bool Topic::matches(const Topic& topic) const
|
||||||
{
|
{
|
||||||
if (getIndex() == topic.getIndex()) return true;
|
if (getIndex() == topic.getIndex()) return true;
|
||||||
if (str() == topic.str()) return true;
|
const char* p1 = c_str();
|
||||||
return false;
|
const char* p2 = topic.c_str();
|
||||||
|
|
||||||
|
if (p1 == p2) return true;
|
||||||
|
if (*p2 == '$' and *p1 != '$') return false;
|
||||||
|
|
||||||
|
while(*p1 and *p2)
|
||||||
|
{
|
||||||
|
if (*p1 == '+')
|
||||||
|
{
|
||||||
|
++p1;
|
||||||
|
if (*p1 and *p1!='/') return false;
|
||||||
|
if (*p1) ++p1;
|
||||||
|
while(*p2 and *p2++!='/');
|
||||||
|
}
|
||||||
|
else if (*p1 == '#')
|
||||||
|
{
|
||||||
|
if (*++p1==0) return true;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else if (*p1 == '*')
|
||||||
|
{
|
||||||
|
const char c=*(p1+1);
|
||||||
|
if (c==0) return true;
|
||||||
|
if (c!='/') return false;
|
||||||
|
const char*p = p1+2;
|
||||||
|
while(*p and *p2)
|
||||||
|
{
|
||||||
|
if (*p == *p2)
|
||||||
|
{
|
||||||
|
if (*p==0) return true;
|
||||||
|
if (*p=='/')
|
||||||
|
{
|
||||||
|
p1=p;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
while(*p2 and *p2++!='/');
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
++p;
|
||||||
|
++p2;
|
||||||
|
}
|
||||||
|
if (*p==0) return true;
|
||||||
|
}
|
||||||
|
else if (*p1 == *p2)
|
||||||
|
{
|
||||||
|
++p1;
|
||||||
|
++p2;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (*p1=='/' and p1[1]=='#' and p1[2]==0) return true;
|
||||||
|
return *p1==0 and *p2==0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// publish from local client
|
// publish from local client
|
||||||
MqttError MqttClient::publish(const Topic& topic, const char* payload, size_t pay_length)
|
MqttError MqttClient::publish(const Topic& topic, const char* payload, size_t pay_length)
|
||||||
{
|
{
|
||||||
@@ -648,9 +726,9 @@ MqttError MqttClient::publish(const Topic& topic, const char* payload, size_t pa
|
|||||||
msg.add(payload, pay_length, false);
|
msg.add(payload, pay_length, false);
|
||||||
msg.complete();
|
msg.complete();
|
||||||
|
|
||||||
if (parent)
|
if (local_broker)
|
||||||
{
|
{
|
||||||
return parent->publish(this, topic, msg);
|
return local_broker->publish(this, topic, msg);
|
||||||
}
|
}
|
||||||
else if (client)
|
else if (client)
|
||||||
return msg.sendTo(this);
|
return msg.sendTo(this);
|
||||||
@@ -663,7 +741,7 @@ MqttError MqttClient::publishIfSubscribed(const Topic& topic, MqttMessage& msg)
|
|||||||
{
|
{
|
||||||
MqttError retval=MqttOk;
|
MqttError retval=MqttOk;
|
||||||
|
|
||||||
debug("mqttclient publish " << subscriptions.size());
|
debug("mqttclient publishIfSubscribed " << topic.c_str() << ' ' << subscriptions.size());
|
||||||
if (isSubscribedTo(topic))
|
if (isSubscribedTo(topic))
|
||||||
{
|
{
|
||||||
if (client)
|
if (client)
|
||||||
@@ -672,8 +750,8 @@ MqttError MqttClient::publishIfSubscribed(const Topic& topic, MqttMessage& msg)
|
|||||||
{
|
{
|
||||||
processMessage(&msg);
|
processMessage(&msg);
|
||||||
|
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << "Should call the callback ?\n";
|
Console << "Should call the callback ?\n";
|
||||||
#endif
|
#endif
|
||||||
// callback(this, topic, nullptr, 0); // TODO Payload
|
// callback(this, topic, nullptr, 0); // TODO Payload
|
||||||
}
|
}
|
||||||
@@ -741,8 +819,8 @@ void MqttMessage::incoming(char in_byte)
|
|||||||
break;
|
break;
|
||||||
case Complete:
|
case Complete:
|
||||||
default:
|
default:
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << "Spurious " << _HEX(in_byte) << endl;
|
Console << red << "Spurious " << _HEX(in_byte) << white << endl;
|
||||||
hexdump("spurious");
|
hexdump("spurious");
|
||||||
#endif
|
#endif
|
||||||
reset();
|
reset();
|
||||||
@@ -768,6 +846,7 @@ void MqttMessage::add(const char* p, size_t len, bool addLength)
|
|||||||
|
|
||||||
void MqttMessage::encodeLength()
|
void MqttMessage::encodeLength()
|
||||||
{
|
{
|
||||||
|
debug("encodingLength");
|
||||||
if (state != Complete)
|
if (state != Complete)
|
||||||
{
|
{
|
||||||
int length = buffer.size()-3; // 3 = 1 byte for header + 2 bytes for pre-reserved length field.
|
int length = buffer.size()-3; // 3 = 1 byte for header + 2 bytes for pre-reserved length field.
|
||||||
@@ -793,14 +872,14 @@ MqttError MqttMessage::sendTo(MqttClient* client)
|
|||||||
{
|
{
|
||||||
if (buffer.size())
|
if (buffer.size())
|
||||||
{
|
{
|
||||||
debug("sending " << buffer.size() << " bytes");
|
debug(cyan << "sending " << buffer.size() << " bytes to " << client->id());
|
||||||
encodeLength();
|
encodeLength();
|
||||||
// hexdump("snd");
|
hexdump("Sending ");
|
||||||
client->write(&buffer[0], buffer.size());
|
client->write(&buffer[0], buffer.size());
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
debug("??? Invalid send");
|
debug(red << "??? Invalid send");
|
||||||
return MqttInvalidMessage;
|
return MqttInvalidMessage;
|
||||||
}
|
}
|
||||||
return MqttOk;
|
return MqttOk;
|
||||||
@@ -809,7 +888,33 @@ MqttError MqttMessage::sendTo(MqttClient* client)
|
|||||||
void MqttMessage::hexdump(const char* prefix) const
|
void MqttMessage::hexdump(const char* prefix) const
|
||||||
{
|
{
|
||||||
(void)prefix;
|
(void)prefix;
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
|
if (TinyMqtt::debug<2) return;
|
||||||
|
static std::map<Type, std::string> tts={
|
||||||
|
{ Connect, "Connect" },
|
||||||
|
{ ConnAck, "Connack" },
|
||||||
|
{ Publish, "Publish" },
|
||||||
|
{ PubAck, "Puback" },
|
||||||
|
{ Subscribe, "Subscribe" },
|
||||||
|
{ SubAck, "Suback" },
|
||||||
|
{ UnSubscribe, "Unsubscribe" },
|
||||||
|
{ UnSuback, "Unsuback" },
|
||||||
|
{ PingReq, "Pingreq" },
|
||||||
|
{ PingResp, "Pingresp" },
|
||||||
|
{ Disconnect, "Disconnect" }
|
||||||
|
};
|
||||||
|
std::string t("Unknown");
|
||||||
|
Type typ=static_cast<Type>(buffer[0] & 0xF0);
|
||||||
|
if (tts.find(typ) != tts.end())
|
||||||
|
t=tts[typ];
|
||||||
|
Console.fg(cyan);
|
||||||
|
#ifdef NOT_ESP_CORE
|
||||||
|
Console << "---> MESSAGE " << t << ' ' << _HEX(typ) << ' ' << " mem=???" << endl;
|
||||||
|
#else
|
||||||
|
Console << "---> MESSAGE " << t << ' ' << _HEX(typ) << ' ' << " mem=" << ESP.getFreeHeap() << endl;
|
||||||
|
#endif
|
||||||
|
Console.fg(white);
|
||||||
|
|
||||||
uint16_t addr=0;
|
uint16_t addr=0;
|
||||||
const int bytes_per_row = 8;
|
const int bytes_per_row = 8;
|
||||||
const char* hex_to_str = " | ";
|
const char* hex_to_str = " | ";
|
||||||
@@ -817,19 +922,19 @@ void MqttMessage::hexdump(const char* prefix) const
|
|||||||
const char* half_sep = " - ";
|
const char* half_sep = " - ";
|
||||||
std::string ascii;
|
std::string ascii;
|
||||||
|
|
||||||
Serial << prefix << " size(" << buffer.size() << "), state=" << state << endl;
|
Console << prefix << " size(" << buffer.size() << "), state=" << state << endl;
|
||||||
|
|
||||||
for(const char chr: buffer)
|
for(const char chr: buffer)
|
||||||
{
|
{
|
||||||
if ((addr % bytes_per_row) == 0)
|
if ((addr % bytes_per_row) == 0)
|
||||||
{
|
{
|
||||||
if (ascii.length()) Serial << hex_to_str << ascii << separator << endl;
|
if (ascii.length()) Console << hex_to_str << ascii << separator << endl;
|
||||||
if (prefix) Serial << prefix << separator;
|
if (prefix) Console << prefix << separator;
|
||||||
ascii.clear();
|
ascii.clear();
|
||||||
}
|
}
|
||||||
addr++;
|
addr++;
|
||||||
if (chr<16) Serial << '0';
|
if (chr<16) Console << '0';
|
||||||
Serial << _HEX(chr) << ' ';
|
Console << _HEX(chr) << ' ';
|
||||||
|
|
||||||
ascii += (chr<32 ? '.' : chr);
|
ascii += (chr<32 ? '.' : chr);
|
||||||
if (ascii.length() == (bytes_per_row/2)) ascii += half_sep;
|
if (ascii.length() == (bytes_per_row/2)) ascii += half_sep;
|
||||||
@@ -838,12 +943,12 @@ void MqttMessage::hexdump(const char* prefix) const
|
|||||||
{
|
{
|
||||||
while(ascii.length() < bytes_per_row+strlen(half_sep))
|
while(ascii.length() < bytes_per_row+strlen(half_sep))
|
||||||
{
|
{
|
||||||
Serial << " "; // spaces per hexa byte
|
Console << " "; // spaces per hexa byte
|
||||||
ascii += ' ';
|
ascii += ' ';
|
||||||
}
|
}
|
||||||
Serial << hex_to_str << ascii << separator;
|
Console << hex_to_str << ascii << separator;
|
||||||
}
|
}
|
||||||
|
|
||||||
Serial << endl;
|
Console << endl;
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|||||||
122
src/TinyMqtt.h
122
src/TinyMqtt.h
@@ -1,17 +1,19 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#pragma once
|
#pragma once
|
||||||
|
#define TINY_MQTT_DEBUG 0
|
||||||
|
|
||||||
// TODO Should add a AUnit with both TCP_ASYNC and not TCP_ASYNC
|
// TODO Should add a AUnit with both TINY_MQTT_ASYNC and not TINY_MQTT_ASYNC
|
||||||
// #define TCP_ASYNC // Uncomment this to use ESPAsyncTCP instead of normal cnx
|
// #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
|
||||||
@@ -20,21 +22,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 << (int)__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
|
||||||
@@ -81,6 +97,7 @@ class MqttMessage
|
|||||||
PingResp = 0xD0,
|
PingResp = 0xD0,
|
||||||
Disconnect = 0xE0
|
Disconnect = 0xE0
|
||||||
};
|
};
|
||||||
|
|
||||||
enum __attribute__((packed)) State
|
enum __attribute__((packed)) State
|
||||||
{
|
{
|
||||||
FixedHeader=0,
|
FixedHeader=0,
|
||||||
@@ -92,6 +109,11 @@ class MqttMessage
|
|||||||
Create=6
|
Create=6
|
||||||
};
|
};
|
||||||
|
|
||||||
|
static inline uint32_t getSize(const char* buffer)
|
||||||
|
{
|
||||||
|
const unsigned char* bun = (const unsigned char*)buffer;
|
||||||
|
return (*bun << 8) | bun[1]; }
|
||||||
|
|
||||||
MqttMessage() { reset(); }
|
MqttMessage() { reset(); }
|
||||||
MqttMessage(Type t, uint8_t bits_d3_d0=0) { create(t); buffer[0] |= bits_d3_d0; }
|
MqttMessage(Type t, uint8_t bits_d3_d0=0) { create(t); buffer[0] |= bits_d3_d0; }
|
||||||
void incoming(char byte);
|
void incoming(char byte);
|
||||||
@@ -145,33 +167,37 @@ class MqttClient
|
|||||||
{
|
{
|
||||||
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:
|
public:
|
||||||
/** Constructor. If broker is not null, this is the adress of a local broker.
|
/** Constructor. Broker is the adress of a local broker if not null
|
||||||
If you want to connect elsewhere, leave broker null and use connect() **/
|
If you want to connect elsewhere, leave broker null and use connect() **/
|
||||||
MqttClient(MqttBroker* broker = nullptr, const std::string& id="");
|
MqttClient(MqttBroker* broker = nullptr, const std::string& id = TINY_MQTT_DEFAULT_CLIENT_ID);
|
||||||
MqttClient(const std::string& id) : MqttClient(nullptr, id){}
|
MqttClient(const std::string& id) : MqttClient(nullptr, id){}
|
||||||
|
|
||||||
~MqttClient();
|
~MqttClient();
|
||||||
|
|
||||||
void connect(MqttBroker* parent);
|
void connect(MqttBroker* local_broker);
|
||||||
void connect(std::string broker, uint16_t port, uint16_t keep_alive = 10);
|
void connect(std::string broker, uint16_t port, uint16_t keep_alive = 10);
|
||||||
|
|
||||||
// TODO it seems that connected returns true in tcp mode even if
|
// TODO it seems that connected returns true in tcp mode even if
|
||||||
// no negociation occured (only if tcp link is established)
|
// no negociation occurred
|
||||||
bool connected() { return
|
bool connected()
|
||||||
(parent!=nullptr and client==nullptr) or
|
{
|
||||||
(client and client->connected()); }
|
return (local_broker!=nullptr and client==nullptr) or (client and client->connected());
|
||||||
|
}
|
||||||
|
|
||||||
void write(const char* buf, size_t length)
|
void write(const char* buf, size_t length)
|
||||||
{ if (client) client->write(buf, length); }
|
{
|
||||||
|
if (client) client->write(buf, length);
|
||||||
|
}
|
||||||
|
|
||||||
const std::string& id() const { return clientId; }
|
const std::string& id() const { return clientId; }
|
||||||
void id(std::string& new_id) { clientId = new_id; }
|
void id(const std::string& new_id) { clientId = new_id; }
|
||||||
|
|
||||||
/** Should be called in main loop() */
|
/** Should be called in main loop() */
|
||||||
void loop();
|
void loop();
|
||||||
@@ -179,8 +205,8 @@ class MqttClient
|
|||||||
void setCallback(CallBack fun)
|
void setCallback(CallBack fun)
|
||||||
{
|
{
|
||||||
callback=fun;
|
callback=fun;
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
Serial << "Callback set to " << (long)fun << endl;
|
Console << TinyConsole::magenta << "Callback set to " << (long)fun << TinyConsole::white << endl;
|
||||||
if (callback) callback(this, "test/topic", "value", 5);
|
if (callback) callback(this, "test/topic", "value", 5);
|
||||||
#endif
|
#endif
|
||||||
};
|
};
|
||||||
@@ -203,44 +229,51 @@ class MqttClient
|
|||||||
void dump(std::string indent="")
|
void dump(std::string indent="")
|
||||||
{
|
{
|
||||||
(void)indent;
|
(void)indent;
|
||||||
#ifdef TINY_MQTT_DEBUG
|
#if TINY_MQTT_DEBUG
|
||||||
uint32_t ms=millis();
|
uint32_t ms=millis();
|
||||||
Serial << indent << "+-- " << '\'' << clientId.c_str() << "' " << (connected() ? " ON " : " OFF");
|
Console << indent << "+-- " << '\'' << clientId.c_str() << "' " << (connected() ? " ON " : " OFF");
|
||||||
Serial << ", alive=" << alive << '/' << ms << ", ka=" << keep_alive << ' ';
|
Console << ", alive=" << alive << '/' << ms << ", ka=" << keep_alive << ' ';
|
||||||
Serial << (client && client->connected() ? "" : "dis") << "connected";
|
if (client)
|
||||||
|
{
|
||||||
|
if (client->connected())
|
||||||
|
Console << TinyConsole::green << "connected";
|
||||||
|
else
|
||||||
|
Console << TinyConsole::red << "disconnected";
|
||||||
|
Console << TinyConsole::white;
|
||||||
|
}
|
||||||
if (subscriptions.size())
|
if (subscriptions.size())
|
||||||
{
|
{
|
||||||
bool c = false;
|
bool c = false;
|
||||||
Serial << " [";
|
Console << " [";
|
||||||
for(auto s: subscriptions)
|
for(auto s: subscriptions)
|
||||||
(void)indent;
|
|
||||||
{
|
{
|
||||||
if (c) Serial << ", ";
|
if (c) Console << ", ";
|
||||||
Serial << s.str().c_str();
|
Console << s.str().c_str();
|
||||||
c=true;
|
c=true;
|
||||||
}
|
}
|
||||||
Serial << ']';
|
Console << ']';
|
||||||
}
|
}
|
||||||
Serial << endl;
|
Console << TinyConsole::erase_to_end << endl;
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef EPOXY_DUINO
|
#ifdef EPOXY_DUINO
|
||||||
static std::map<MqttMessage::Type, int> counters; // Number of processed messages
|
static std::map<MqttMessage::Type, int> counters; // Number of processed messages
|
||||||
#endif
|
#endif
|
||||||
|
uint32_t keepAlive() const { return keep_alive; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|
||||||
// event when tcp/ip link established (real or fake)
|
// event when tcp/ip link established (real or fake)
|
||||||
static void onConnect(void * client_ptr, TcpClient*);
|
static void onConnect(void * client_ptr, TcpClient*);
|
||||||
#ifdef TCP_ASYNC
|
#ifdef TINY_MQTT_ASYNC
|
||||||
static void onData(void* client_ptr, TcpClient*, void* data, size_t len);
|
static void onData(void* client_ptr, TcpClient*, void* data, size_t len);
|
||||||
#endif
|
#endif
|
||||||
MqttError sendTopic(const Topic& topic, MqttMessage::Type type, uint8_t qos);
|
MqttError sendTopic(const Topic& topic, MqttMessage::Type type, uint8_t qos);
|
||||||
void resubscribe();
|
void resubscribe();
|
||||||
|
|
||||||
friend class MqttBroker;
|
friend class MqttBroker;
|
||||||
MqttClient(MqttBroker* parent, TcpClient* client);
|
MqttClient(MqttBroker* local_broker, TcpClient* client);
|
||||||
// republish a received publish if topic matches any in subscriptions
|
// republish a received publish if topic matches any in subscriptions
|
||||||
MqttError publishIfSubscribed(const Topic& topic, MqttMessage& msg);
|
MqttError publishIfSubscribed(const Topic& topic, MqttMessage& msg);
|
||||||
|
|
||||||
@@ -249,14 +282,13 @@ class MqttClient
|
|||||||
|
|
||||||
bool mqtt_connected = false;
|
bool mqtt_connected = false;
|
||||||
char mqtt_flags;
|
char mqtt_flags;
|
||||||
uint32_t keep_alive = 60;
|
uint32_t keep_alive = 30;
|
||||||
uint32_t alive;
|
uint32_t alive;
|
||||||
MqttMessage message;
|
MqttMessage message;
|
||||||
|
|
||||||
// TODO having a pointer on MqttBroker may produce larger binaries
|
// connection to local broker, or link to the parent
|
||||||
// due to unecessary function linked if ever parent is not used
|
// when MqttBroker uses MqttClient for each external connexion
|
||||||
// (this is the case when MqttBroker isn't used except here)
|
MqttBroker* local_broker=nullptr;
|
||||||
MqttBroker* parent=nullptr; // connection to local broker
|
|
||||||
|
|
||||||
TcpClient* client=nullptr; // connection to remote broker
|
TcpClient* client=nullptr; // connection to remote broker
|
||||||
std::set<Topic> subscriptions;
|
std::set<Topic> subscriptions;
|
||||||
@@ -291,6 +323,8 @@ class MqttBroker
|
|||||||
client->dump(indent);
|
client->dump(indent);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const std::vector<MqttClient*> getClients() const { return clients; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
friend class MqttClient;
|
friend class MqttClient;
|
||||||
|
|
||||||
@@ -306,17 +340,19 @@ class MqttBroker
|
|||||||
|
|
||||||
MqttError subscribe(const Topic& topic, uint8_t qos);
|
MqttError subscribe(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;
|
|
||||||
|
private:
|
||||||
|
TcpServer* server = nullptr;
|
||||||
|
|
||||||
const char* auth_user = "guest";
|
const char* auth_user = "guest";
|
||||||
const char* auth_password = "guest";
|
const char* auth_password = "guest";
|
||||||
State state = Disconnected;
|
|
||||||
|
|
||||||
MqttClient* broker = nullptr;
|
MqttClient* broker = nullptr;
|
||||||
|
|
||||||
|
State state = Disconnected;
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -1,2 +0,0 @@
|
|||||||
const char *ssid = "YOUR-SSID-HERE";
|
|
||||||
const char *password = "YOUR-PASSWORD-HERE";
|
|
||||||
@@ -5,7 +5,15 @@ tests:
|
|||||||
$(MAKE) -C $$(dirname $$i) -j; \
|
$(MAKE) -C $$(dirname $$i) -j; \
|
||||||
done
|
done
|
||||||
|
|
||||||
runtests: tests
|
debugtest:
|
||||||
|
set -e; \
|
||||||
|
$(MAKE) clean; \
|
||||||
|
$(MAKE) -C debug-mode -j; \
|
||||||
|
debug-mode/debug-tests.out
|
||||||
|
|
||||||
|
runtests: debugtest
|
||||||
|
$(MAKE) clean
|
||||||
|
$(MAKE) tests
|
||||||
set -e; \
|
set -e; \
|
||||||
for i in *-tests/Makefile; do \
|
for i in *-tests/Makefile; do \
|
||||||
echo '==== Running:' $$(dirname $$i); \
|
echo '==== Running:' $$(dirname $$i); \
|
||||||
|
|||||||
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();
|
||||||
|
}
|
||||||
@@ -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
|
||||||
|
|||||||
@@ -1,53 +1,54 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#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 namespace std;
|
||||||
|
|
||||||
MqttBroker broker(1883);
|
MqttBroker broker(1883);
|
||||||
|
|
||||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
std::map<std::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(length_decode_greater_than_127)
|
test(length_decode_greater_than_127)
|
||||||
{
|
{
|
||||||
// TODO WRITE THIS TEST
|
// TODO WRITE THIS TEST
|
||||||
// The test should verify than a mqtt message with more than 127 bytes is correctly decoded
|
// The test should verify than a mqtt message with more than 127 bytes is correctly decoded
|
||||||
assertEqual(1,2);
|
assertEqual(1,2);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
//----------------------------------------------------------------------------
|
//----------------------------------------------------------------------------
|
||||||
// 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();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ EXTRA_CXXFLAGS=-g3 -O0
|
|||||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||||
|
|
||||||
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
|
||||||
|
|||||||
@@ -4,148 +4,148 @@
|
|||||||
#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 namespace std;
|
||||||
|
|
||||||
MqttBroker broker(1883);
|
MqttBroker broker(1883);
|
||||||
|
|
||||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
std::map<std::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_client_should_unregister_when_destroyed)
|
test(local_client_should_unregister_when_destroyed)
|
||||||
{
|
{
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
{
|
{
|
||||||
assertEqual(broker.clientsCount(), (size_t)0); // Ensure client is not yet connected
|
assertEqual(broker.clientsCount(), (size_t)0); // Ensure client is not yet connected
|
||||||
MqttClient client(&broker);
|
MqttClient client(&broker);
|
||||||
assertEqual(broker.clientsCount(), (size_t)1); // Ensure client is now connected
|
assertEqual(broker.clientsCount(), (size_t)1); // Ensure client is now connected
|
||||||
}
|
}
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
}
|
}
|
||||||
|
|
||||||
#if 0
|
#if 0
|
||||||
test(local_connect)
|
test(local_connect)
|
||||||
{
|
{
|
||||||
assertEqual(broker.clientsCount(), (size_t)0);
|
assertEqual(broker.clientsCount(), (size_t)0);
|
||||||
|
|
||||||
MqttClient client;
|
MqttClient client;
|
||||||
assertTrue(client.connected());
|
assertTrue(client.connected());
|
||||||
assertEqual(broker.clientsCount(), (size_t)1);
|
assertEqual(broker.clientsCount(), (size_t)1);
|
||||||
}
|
}
|
||||||
|
|
||||||
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;
|
||||||
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;
|
||||||
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[""]["a/b"], 1);
|
||||||
assertEqual(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("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("B");
|
||||||
subscriber_b.setCallback(onPublish);
|
subscriber_b.setCallback(onPublish);
|
||||||
subscriber_b.subscribe("a/b");
|
subscriber_b.subscribe("a/b");
|
||||||
|
|
||||||
MqttClient publisher;
|
MqttClient publisher;
|
||||||
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
|
||||||
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(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;
|
||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
subscriber.subscribe("a/b");
|
subscriber.subscribe("a/b");
|
||||||
|
|
||||||
MqttClient publisher;
|
MqttClient publisher;
|
||||||
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");
|
||||||
|
|
||||||
assertEqual(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;
|
||||||
{
|
{
|
||||||
MqttClient subscriber;
|
MqttClient subscriber;
|
||||||
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
|
#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("=============[ LOCAL 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();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ EXTRA_CXXFLAGS=-g3 -O0
|
|||||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||||
|
|
||||||
APP_NAME := network-tests
|
APP_NAME := network-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,3 +1,4 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#include <Arduino.h>
|
#include <Arduino.h>
|
||||||
#include <AUnit.h>
|
#include <AUnit.h>
|
||||||
#include <TinyMqtt.h>
|
#include <TinyMqtt.h>
|
||||||
@@ -6,13 +7,14 @@
|
|||||||
#include <iomanip>
|
#include <iomanip>
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* TinyMqtt network unit tests.
|
* TinyMqtt network 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
|
||||||
**/
|
**/
|
||||||
|
|
||||||
// if ascii_pos = 0, no ascii dump, else ascii dump starts after column ascii_pos
|
// 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::string bufferToHexa(const uint8_t* buffer, size_t length, char sep = 0, size_t ascii_pos = 0)
|
||||||
@@ -76,7 +78,7 @@ String toString(const IPAddress& ip)
|
|||||||
|
|
||||||
MqttBroker broker(1883);
|
MqttBroker broker(1883);
|
||||||
|
|
||||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
||||||
|
|
||||||
char* lastPayload = nullptr;
|
char* lastPayload = nullptr;
|
||||||
size_t lastLength;
|
size_t lastLength;
|
||||||
@@ -95,12 +97,12 @@ void start_servers(int n, bool early_accept = true)
|
|||||||
|
|
||||||
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(network_single_broker_begin)
|
test(network_single_broker_begin)
|
||||||
@@ -140,6 +142,41 @@ test(suback)
|
|||||||
assertEqual(MqttClient::counters[MqttMessage::Type::SubAck], 1);
|
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)
|
test(network_client_to_broker_connexion)
|
||||||
{
|
{
|
||||||
start_servers(2, true);
|
start_servers(2, true);
|
||||||
@@ -227,12 +264,12 @@ test(network_one_client_one_broker_hudge_publish_and_subscribe_through_network)
|
|||||||
|
|
||||||
test(network_client_should_unregister_when_destroyed)
|
test(network_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);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -241,113 +278,113 @@ test(network_client_should_unregister_when_destroyed)
|
|||||||
|
|
||||||
test(network_connect)
|
test(network_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(network_publish_should_be_dispatched)
|
test(network_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(network_publish_should_be_dispatched_to_clients)
|
test(network_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(network_unsubscribe)
|
test(network_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(network_nocallback_when_destroyed)
|
test(network_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(network_small_payload)
|
test(network_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(network_hudge_payload)
|
test(network_hudge_payload)
|
||||||
@@ -358,13 +395,13 @@ test(network_hudge_payload)
|
|||||||
subscriber.setCallback(onPublish);
|
subscriber.setCallback(onPublish);
|
||||||
subscriber.subscribe("a/b"); // Note -> this does not send any byte .... (nowhere to send)
|
subscriber.subscribe("a/b"); // Note -> this does not send any byte .... (nowhere to send)
|
||||||
|
|
||||||
MqttClient publisher(&broker);
|
MqttClient publisher(&broker);
|
||||||
publisher.publish("a/b", payload); // This publish is received
|
publisher.publish("a/b", payload); // This publish is received
|
||||||
|
|
||||||
// onPublish should have filled lastPayload and lastLength
|
// onPublish should have filled lastPayload and lastLength
|
||||||
assertEqual(payload, lastPayload);
|
assertEqual(payload, lastPayload);
|
||||||
assertEqual(lastLength, strlen(payload));
|
assertEqual(lastLength, strlen(payload));
|
||||||
assertEqual(strcmp(payload, lastPayload), 0);
|
assertEqual(strcmp(payload, lastPayload), 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
test(connack)
|
test(connack)
|
||||||
@@ -411,19 +448,19 @@ test(connack)
|
|||||||
//----------------------------------------------------------------------------
|
//----------------------------------------------------------------------------
|
||||||
// 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("=============[ FAKE NETWORK TinyMqtt TESTS ]========================");
|
Serial.println("=============[ FAKE NETWORK TinyMqtt TESTS ]========================");
|
||||||
|
|
||||||
WiFi.mode(WIFI_STA);
|
WiFi.mode(WIFI_STA);
|
||||||
WiFi.begin("network", "password");
|
WiFi.begin("network", "password");
|
||||||
}
|
}
|
||||||
|
|
||||||
void loop() {
|
void loop() {
|
||||||
aunit::TestRunner::run();
|
aunit::TestRunner::run();
|
||||||
|
|
||||||
if (Serial.available()) ESP.reset();
|
if (Serial.available()) ESP.reset();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ EXTRA_CXXFLAGS=-g3 -O0
|
|||||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||||
|
|
||||||
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,3 +1,4 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#include <Arduino.h>
|
#include <Arduino.h>
|
||||||
#include <AUnit.h>
|
#include <AUnit.h>
|
||||||
#include <TinyMqtt.h>
|
#include <TinyMqtt.h>
|
||||||
@@ -5,149 +6,247 @@
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* 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 namespace std;
|
||||||
|
|
||||||
MqttBroker broker(1883);
|
MqttBroker broker(1883);
|
||||||
|
|
||||||
std::map<std::string, std::map<Topic, int>> published; // map[client_id] => map[topic] = count
|
std::map<std::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_small_payload)
|
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)
|
test(nowifi_hudge_payload)
|
||||||
@@ -158,26 +257,26 @@ test(nowifi_hudge_payload)
|
|||||||
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); // This publish is received
|
publisher.publish("a/b", payload); // This publish is received
|
||||||
|
|
||||||
// onPublish should have filled lastPayload and lastLength
|
// onPublish should have filled lastPayload and lastLength
|
||||||
assertEqual(payload, lastPayload);
|
assertEqual(payload, lastPayload);
|
||||||
assertEqual(lastLength, strlen(payload));
|
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();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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 := 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,12 +1,13 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
#include <Arduino.h>
|
#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 namespace std;
|
||||||
|
|
||||||
@@ -18,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");
|
std::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.
|
||||||
|
|
||||||
|
EXTRA_CXXFLAGS=-g3 -O0
|
||||||
|
|
||||||
|
APP_NAME := topic-tests
|
||||||
|
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsync TinyConsole
|
||||||
|
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||||
|
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||||
|
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||||
86
tests/topic-tests/topic-tests.ino
Normal file
86
tests/topic-tests/topic-tests.ino
Normal file
@@ -0,0 +1,86 @@
|
|||||||
|
// vim: ts=2 sw=2 expandtab
|
||||||
|
#include <Arduino.h>
|
||||||
|
#include <AUnit.h>
|
||||||
|
#include <TinyMqtt.h>
|
||||||
|
#include <map>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
#define endl "\n"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TinyMqtt / StringIndexer unit tests.
|
||||||
|
*
|
||||||
|
**/
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
bool testTopicMatch(const char* a, const char* b, bool expected)
|
||||||
|
{
|
||||||
|
Topic ta(a);
|
||||||
|
Topic tb(b);
|
||||||
|
bool match(ta.matches(tb));
|
||||||
|
cout << " " << ta.c_str() << ' ';
|
||||||
|
if (match != expected)
|
||||||
|
cout << (expected ? " should match " : " should not match ");
|
||||||
|
else
|
||||||
|
cout << (expected ? " matches " : " unmatches ");
|
||||||
|
cout << tb.c_str() << endl;
|
||||||
|
return expected == match;
|
||||||
|
}
|
||||||
|
|
||||||
|
test(topic_matches)
|
||||||
|
{
|
||||||
|
// matching
|
||||||
|
assertTrue(testTopicMatch("a/b/c" , "a/b/c" , true));
|
||||||
|
assertTrue(testTopicMatch("a/*/c" , "a/xyz/c" , true));
|
||||||
|
assertTrue(testTopicMatch("a/*/e" , "a/b/c/d/e" , true));
|
||||||
|
assertTrue(testTopicMatch("a/*" , "a/b/c/d/e" , true));
|
||||||
|
assertTrue(testTopicMatch("*/c" , "a/b/c" , true));
|
||||||
|
assertTrue(testTopicMatch("/*/c" , "/a/b/c" , true));
|
||||||
|
assertTrue(testTopicMatch("a/*" , "a/b/c/d" , true));
|
||||||
|
assertTrue(testTopicMatch("a/+/c" , "a/b/c" , true));
|
||||||
|
assertTrue(testTopicMatch("a/+/c/+/e", "a/b/c/d/e" , true));
|
||||||
|
assertTrue(testTopicMatch("a/*/c/+/e", "a/b/c/d/e" , true));
|
||||||
|
assertTrue(testTopicMatch("/+/b" , "/a/b" , true));
|
||||||
|
assertTrue(testTopicMatch("+" , "a" , true));
|
||||||
|
assertTrue(testTopicMatch("a/b/#" , "a/b/c/d" , true));
|
||||||
|
assertTrue(testTopicMatch("a/b/#" , "a/b" , true));
|
||||||
|
assertTrue(testTopicMatch("a/*/c" , "a/*/c" , true));
|
||||||
|
|
||||||
|
// not matching
|
||||||
|
assertTrue(testTopicMatch("a/b/c" , "a/b/d" , false));
|
||||||
|
assertTrue(testTopicMatch("a/b/c" , "a/b/d" , false));
|
||||||
|
assertTrue(testTopicMatch("a/*/e" , "a/b/c/d/f" , false));
|
||||||
|
assertTrue(testTopicMatch("a/+" , "a" , false));
|
||||||
|
assertTrue(testTopicMatch("a/+" , "a/b/d" , false));
|
||||||
|
assertTrue(testTopicMatch("a/+/" , "a/" , false));
|
||||||
|
|
||||||
|
// $SYS topics
|
||||||
|
assertTrue(testTopicMatch("+/any" , "$SYS/any" , false));
|
||||||
|
assertTrue(testTopicMatch("*/any" , "$SYS/any" , false));
|
||||||
|
assertTrue(testTopicMatch("$SYS/any" , "$SYS/any" , true));
|
||||||
|
assertTrue(testTopicMatch("$SYS/+/y" , "$SYS/a/y" , true));
|
||||||
|
assertTrue(testTopicMatch("$SYS/#" , "$SYS/a/y" , true));
|
||||||
|
|
||||||
|
// not valid
|
||||||
|
assertTrue(testTopicMatch("a/#/b" , "a/x/b" , false));
|
||||||
|
assertTrue(testTopicMatch("a+" , "a/b/d" , false));
|
||||||
|
assertTrue(testTopicMatch("a/b/#/d" , "a/b/c/d" , false));
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
//----------------------------------------------------------------------------
|
||||||
|
// setup() and loop()
|
||||||
|
void setup() {
|
||||||
|
delay(1000);
|
||||||
|
Serial.begin(115200);
|
||||||
|
while(!Serial);
|
||||||
|
|
||||||
|
Serial.println("=============[ TinyMqtt StringIndexer TESTS ]========================");
|
||||||
|
}
|
||||||
|
|
||||||
|
void loop() {
|
||||||
|
aunit::TestRunner::run();
|
||||||
|
|
||||||
|
// if (Serial.available()) ESP.reset();
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user