Compare commits
19 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d8105ee224 | ||
|
|
670d67c024 | ||
|
|
3c3b19882f | ||
|
|
8c55356bd9 | ||
|
|
82d3b913bb | ||
|
|
8b62b5a3b7 | ||
|
|
3a2db664a8 | ||
|
|
0c454bfe3a | ||
|
|
d517cf2627 | ||
|
|
b8022f58a4 | ||
|
|
8162b4c35b | ||
|
|
9e578471f1 | ||
|
|
efa94cc4a4 | ||
|
|
040568b478 | ||
|
|
a96b51d7f4 | ||
|
|
7ad63551d6 | ||
|
|
4627bcd109 | ||
|
|
f0af2b95e3 | ||
|
|
f348d82167 |
2
.github/workflows/aunit.yml
vendored
2
.github/workflows/aunit.yml
vendored
@@ -20,7 +20,7 @@ jobs:
|
||||
git clone https://github.com/hsaturn/TinyConsole
|
||||
git clone https://github.com/hsaturn/EpoxyDuino
|
||||
git clone https://github.com/bxparks/AceRoutine
|
||||
git clone https://github.com/hsaturn/AUnit
|
||||
git clone https://github.com/bxparks/AUnit
|
||||
git clone https://github.com/bxparks/AceCommon
|
||||
git clone https://github.com/hsaturn/EspMock
|
||||
- name: Verify tests
|
||||
|
||||
12
README.md
12
README.md
@@ -10,18 +10,6 @@
|
||||
|
||||
TinyMqtt is a small, fast and capable Mqtt Broker and Client for Esp8266 / Esp32 / Esp WROOM
|
||||
|
||||
### Statuses of all unit tests of TinyMqtt and its dependencies
|
||||
|
||||
| Project | Unit tests result |
|
||||
| ----------- | ------------ |
|
||||
| TinyMqtt | [](https://github.com/hsaturn/TinyMqtt/actions/workflows/aunit.yml) |
|
||||
| Dependencies ||
|
||||
| TinyConsole | [](https://github.com/hsaturn/TinyConsole/actions/workflows/aunit.yml) |
|
||||
| EpoxyDuino | [](https://github.com/hsaturn/EpoxyDuino/actions/workflows/aunit_tests.yml) |
|
||||
| EspMock | [](https://github.com/hsaturn/EspMock/actions/workflows/aunit.yml) |
|
||||
| AUnit | [](https://github.com/hsaturn/AUnit/actions/workflows/aunit_tests.yml) |
|
||||
| AceRoutine | [](https://github.com/bxparks/AceRoutine/actions/workflows/aunit_tests.yml) |
|
||||
|
||||
## Features
|
||||
|
||||
- Very fast broker I saw it re-sent 1000 topics per second for two
|
||||
|
||||
@@ -1,30 +0,0 @@
|
||||
#!/bin/bash
|
||||
current_version=$(git describe --tags --abbrev=0)
|
||||
if [ "$1" == "" ]; then
|
||||
echo
|
||||
echo "Syntax: $0 {new_version}"
|
||||
echo
|
||||
echo " Current version: $current_version"
|
||||
echo
|
||||
else
|
||||
echo "Current version: ($current_version)"
|
||||
echo "New version : ($1)"
|
||||
echo -n "Do you want to proceed ? "
|
||||
read a
|
||||
if [ "$a" == "y" ]; then
|
||||
echo "Doing this..."
|
||||
grep $current_version library.properties
|
||||
if [ "$?" == "0" ]; then
|
||||
sed -i "s/$current_version/$1/" library.properties
|
||||
sed -i "s/$current_version/$1/" library.json
|
||||
git tag $1
|
||||
git add library.properties
|
||||
git add library.json
|
||||
git commit -m "Release $1"
|
||||
git push
|
||||
git push --tags
|
||||
else
|
||||
echo "Current version does not match library.property version, aborting"
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
@@ -15,8 +15,6 @@
|
||||
#include <string>
|
||||
#include <map>
|
||||
|
||||
using string = TinyString;
|
||||
|
||||
bool echo_on = true;
|
||||
auto green = TinyConsole::green;
|
||||
auto red = TinyConsole::red;
|
||||
@@ -34,13 +32,13 @@ const char* password = "";
|
||||
struct free_broker
|
||||
{
|
||||
public:
|
||||
free_broker(const char* s, uint16_t p, const char* /* comment */) : url(s), port(p) {}
|
||||
free_broker(const char* s, uint16_t p, const char* comment) : url(s), port(p) {}
|
||||
|
||||
string url;
|
||||
std::string url;
|
||||
uint16_t port;
|
||||
};
|
||||
|
||||
const std::map<string, free_broker> list =
|
||||
const std::map<std::string, free_broker> list =
|
||||
{
|
||||
{ "mqtthq", { "public.mqtthq.com" , 8083, "publish/subscribe" }},
|
||||
{ "hivemq", { "broker.hivemq.com", 1883, "" }}
|
||||
@@ -75,8 +73,8 @@ void onPublish(const MqttClient* srce, const Topic& topic, const char* payload,
|
||||
}
|
||||
}
|
||||
|
||||
std::map<string, MqttClient*> clients;
|
||||
std::map<string, MqttBroker*> brokers;
|
||||
std::map<std::string, MqttClient*> clients;
|
||||
std::map<std::string, MqttBroker*> brokers;
|
||||
|
||||
void setup()
|
||||
{
|
||||
@@ -128,12 +126,12 @@ void setup()
|
||||
if (Console.isTerm()) onCommand("every 333 view");
|
||||
}
|
||||
|
||||
string getword(string& str, const char* if_empty=nullptr, char sep=' ');
|
||||
std::string getword(std::string& str, const char* if_empty=nullptr, char sep=' ');
|
||||
|
||||
int getint(string& str, const int if_empty=0)
|
||||
int getint(std::string& str, const int if_empty=0)
|
||||
{
|
||||
string str2=str;
|
||||
string sword = getword(str);
|
||||
std::string str2=str;
|
||||
std::string sword = getword(str);
|
||||
if (sword[0] and isdigit(sword[0]))
|
||||
{
|
||||
int ret=atoi(sword.c_str());
|
||||
@@ -145,11 +143,11 @@ int getint(string& str, const int if_empty=0)
|
||||
return if_empty;
|
||||
}
|
||||
|
||||
string getword(string& str, const char* if_empty/*=nullptr*/, char sep/*=' '*/)
|
||||
std::string getword(std::string& str, const char* if_empty/*=nullptr*/, char sep/*=' '*/)
|
||||
{
|
||||
char quote=(str[0]=='"' or str[0]=='\'' ? str[0] : 0);
|
||||
if (quote) str.erase(0,1);
|
||||
string sword;
|
||||
std::string sword;
|
||||
while(str.length() and (str[0]!=sep or quote))
|
||||
{
|
||||
if (str[0]==quote)
|
||||
@@ -192,7 +190,7 @@ string getword(string& str, const char* if_empty/*=nullptr*/, char sep/*=' '*/)
|
||||
return sword;
|
||||
}
|
||||
|
||||
bool isaddr(string s)
|
||||
bool isaddr(std::string s)
|
||||
{
|
||||
if (s.length()==0 or s.length()>3) return false;
|
||||
for(char c: s)
|
||||
@@ -200,14 +198,14 @@ bool isaddr(string s)
|
||||
return true;
|
||||
}
|
||||
|
||||
string getip(string& str, const char* if_empty=nullptr, char sep=' ')
|
||||
std::string getip(std::string& str, const char* if_empty=nullptr, char sep=' ')
|
||||
{
|
||||
string addr=getword(str, if_empty, sep);
|
||||
string ip=addr;
|
||||
std::vector<string> build;
|
||||
std::string addr=getword(str, if_empty, sep);
|
||||
std::string ip=addr;
|
||||
std::vector<std::string> build;
|
||||
while(ip.length())
|
||||
{
|
||||
string b=getword(ip,nullptr,'.');
|
||||
std::string b=getword(ip,nullptr,'.');
|
||||
if (isaddr(b) && build.size()<4)
|
||||
{
|
||||
build.push_back(b);
|
||||
@@ -221,9 +219,9 @@ string getip(string& str, const char* if_empty=nullptr, char sep=' ')
|
||||
{
|
||||
std::stringstream b;
|
||||
b << (int)local[3-build.size()];
|
||||
build.insert(build.begin(), b.str().c_str());
|
||||
build.insert(build.begin(), b.str());
|
||||
}
|
||||
for(string s: build)
|
||||
for(std::string s: build)
|
||||
{
|
||||
if (addr.length()) addr += '.';
|
||||
addr += s;
|
||||
@@ -232,22 +230,22 @@ string getip(string& str, const char* if_empty=nullptr, char sep=' ')
|
||||
return addr;
|
||||
}
|
||||
|
||||
std::map<string, string> vars;
|
||||
std::map<std::string, std::string> vars;
|
||||
|
||||
std::set<string> commands = {
|
||||
std::set<std::string> commands = {
|
||||
"broker", "blink", "client", "connect",
|
||||
"create", "delete", "debug", "help", "interval",
|
||||
"list", "ls", "ip", "off", "on", "set",
|
||||
"publish", "reset", "subscribe", "unsubscribe", "view", "echo", "every"
|
||||
};
|
||||
|
||||
void convertToCommand(string& search)
|
||||
void convertToCommand(std::string& search)
|
||||
{
|
||||
while(search[0]==' ') search.erase(0,1);
|
||||
if (search.length()==0) return;
|
||||
string matches;
|
||||
std::string matches;
|
||||
int count=0;
|
||||
for(string cmd: commands)
|
||||
for(std::string cmd: commands)
|
||||
{
|
||||
if (cmd.substr(0, search.length()) == search)
|
||||
{
|
||||
@@ -265,7 +263,7 @@ void convertToCommand(string& search)
|
||||
}
|
||||
}
|
||||
|
||||
void replace(const char* d, string& str, string srch, string to)
|
||||
void replace(const char* d, std::string& str, std::string srch, std::string to)
|
||||
{
|
||||
if (d[0] && d[1])
|
||||
{
|
||||
@@ -273,7 +271,7 @@ void replace(const char* d, string& str, string srch, string to)
|
||||
to=d[0]+to+d[1];
|
||||
|
||||
size_t pos = 0;
|
||||
while((pos=str.find(srch, pos)) != string::npos)
|
||||
while((pos=str.find(srch, pos)) != std::string::npos)
|
||||
{
|
||||
str.erase(pos, srch.length());
|
||||
str.insert(pos, to);
|
||||
@@ -282,7 +280,7 @@ void replace(const char* d, string& str, string srch, string to)
|
||||
}
|
||||
}
|
||||
|
||||
void replaceVars(string& cmd)
|
||||
void replaceVars(std::string& cmd)
|
||||
{
|
||||
cmd = ' '+cmd+' ';
|
||||
|
||||
@@ -293,12 +291,12 @@ void replaceVars(string& cmd)
|
||||
replace(" .", cmd, it.first, it.second);
|
||||
replace(" ", cmd, it.first, it.second);
|
||||
}
|
||||
cmd.erase(0, cmd.find_first_not_of(' '));
|
||||
cmd.erase(cmd.find_last_not_of(' ')+1);
|
||||
cmd.erase(0, cmd.find_first_not_of(" "));
|
||||
cmd.erase(cmd.find_last_not_of(" ")+1);
|
||||
|
||||
}
|
||||
|
||||
bool compare(string s, const char* cmd)
|
||||
bool compare(std::string s, const char* cmd)
|
||||
{
|
||||
uint8_t p=0;
|
||||
while(s[p++]==*cmd++)
|
||||
@@ -309,11 +307,11 @@ bool compare(string s, const char* cmd)
|
||||
return false;
|
||||
}
|
||||
|
||||
using ClientFunction = void(*)(string& cmd, MqttClient* publish);
|
||||
using ClientFunction = void(*)(std::string& cmd, MqttClient* publish);
|
||||
|
||||
struct Every
|
||||
{
|
||||
string cmd;
|
||||
std::string cmd;
|
||||
uint32_t ms;
|
||||
uint32_t next;
|
||||
uint32_t underrun=0;
|
||||
@@ -343,19 +341,19 @@ int16_t blink;
|
||||
|
||||
std::vector<Every> everies;
|
||||
|
||||
void onCommand(const string& command)
|
||||
void onCommand(const std::string& command)
|
||||
{
|
||||
Console << endl;
|
||||
string cmd=command;
|
||||
std::string cmd=command;
|
||||
if (cmd.substr(0,3)!="set") replaceVars(cmd);
|
||||
eval(cmd);
|
||||
Console << endl;
|
||||
Console.prompt();
|
||||
}
|
||||
|
||||
void clientConnect(MqttClient* client, string& cmd)
|
||||
void clientConnect(MqttClient* client, std::string& cmd)
|
||||
{
|
||||
string remote = getword(cmd);
|
||||
std::string remote = getword(cmd);
|
||||
uint16_t port;
|
||||
auto it=list.find(remote);
|
||||
if (it != list.end())
|
||||
@@ -371,18 +369,18 @@ void clientConnect(MqttClient* client, string& cmd)
|
||||
Console << (client->connected() ? "connected." : "not connected") << endl;
|
||||
}
|
||||
|
||||
void eval(string& cmd)
|
||||
void eval(std::string& cmd)
|
||||
{
|
||||
while(cmd.length())
|
||||
{
|
||||
MqttError retval = MqttOk;
|
||||
|
||||
string s;
|
||||
std::string s;
|
||||
MqttBroker* broker = nullptr;
|
||||
MqttClient* client = nullptr;
|
||||
|
||||
// client.function notation
|
||||
if (cmd.find('.') != string::npos &&
|
||||
if (cmd.find('.') != std::string::npos &&
|
||||
cmd.find('.') < cmd.find(' '))
|
||||
{
|
||||
s=getword(cmd, nullptr, '.');
|
||||
@@ -634,7 +632,7 @@ void eval(string& cmd)
|
||||
}
|
||||
else if (compare(s, "broker"))
|
||||
{
|
||||
string id=getword(cmd);
|
||||
std::string id=getword(cmd);
|
||||
if (clients.find(id) != clients.end())
|
||||
{
|
||||
Console << "A client already have that name" << endl;
|
||||
@@ -665,7 +663,7 @@ void eval(string& cmd)
|
||||
}
|
||||
else if (compare(s, "client"))
|
||||
{
|
||||
string id=getword(cmd);
|
||||
std::string id=getword(cmd);
|
||||
if (brokers.find(id) != brokers.end())
|
||||
{
|
||||
Console << "A broker have that name" << endl;
|
||||
@@ -702,7 +700,7 @@ void eval(string& cmd)
|
||||
}
|
||||
else if (compare(s, "set"))
|
||||
{
|
||||
string name(getword(cmd));
|
||||
std::string name(getword(cmd));
|
||||
if (name.length()==0)
|
||||
{
|
||||
for(auto it: vars)
|
||||
@@ -814,7 +812,7 @@ void loop()
|
||||
if (not every.active) continue;
|
||||
if (every.ms && every.cmd.length() && ms > every.next)
|
||||
{
|
||||
string cmd(every.cmd);
|
||||
std::string cmd(every.cmd);
|
||||
eval(cmd);
|
||||
every.next += every.ms;
|
||||
if (ms > every.next and ms > every.underrun)
|
||||
@@ -847,6 +845,7 @@ void loop()
|
||||
out++;
|
||||
}
|
||||
|
||||
static long count;
|
||||
#if defined(ESP9266)
|
||||
MDNS.update();
|
||||
#endif
|
||||
|
||||
@@ -1,9 +1,9 @@
|
||||
name=TinyMqtt
|
||||
version=0.9.14
|
||||
version=0.9.11
|
||||
author=Francois BIOT, HSaturn, <hsaturn@gmail.com>
|
||||
maintainer=Francois BIOT <hsaturn@gmail.com>
|
||||
maintainer=Francois BIOT, HSaturn, <hsaturn@gmail.com>
|
||||
sentence=A tiny broker and client library for MQTT messaging.
|
||||
paragraph=MQTT is a lightweight messaging protocol. This library allows to host a broker or to use a mqtt client in your ESP 8266 or 32 WROOM. It does support MQTT 3.1.1 with QoS=0, and is intented to be as smallest as possible, keeping a good efficiency (about 1k msg per second on ESP8266 12F).
|
||||
paragraph=MQTT is a lightweight messaging protocol ideal for small devices. This library allows to send and receive MQTT messages and to host a broker in your ESP 8266 and 32 WROOM. It does support MQTT 3.1.1 with QoS=0.
|
||||
category=Communication
|
||||
url=https://github.com/hsaturn/TinyMqtt
|
||||
architectures=*
|
||||
|
||||
@@ -1,4 +1,3 @@
|
||||
#include "StringIndexer.h"
|
||||
|
||||
StringIndexer::Strings StringIndexer::strings;
|
||||
|
||||
std::map<StringIndexer::index_t, StringIndexer::StringCounter> StringIndexer::strings;
|
||||
|
||||
@@ -1,25 +1,18 @@
|
||||
// vim: ts=2 sw=2 expandtab
|
||||
#pragma once
|
||||
#include <assert.h>
|
||||
#include <map>
|
||||
#include <unordered_map>
|
||||
#include "TinyConsole.h"
|
||||
#include <string>
|
||||
#include <string.h>
|
||||
|
||||
using string = TinyConsole::string;
|
||||
|
||||
/***
|
||||
* Allows to store up to 255 different strings with one byte class
|
||||
* very memory efficient when one string is used many times.
|
||||
*/
|
||||
class StringIndexer
|
||||
{
|
||||
private:
|
||||
|
||||
class StringCounter
|
||||
{
|
||||
string str;
|
||||
std::string str;
|
||||
uint8_t used=0;
|
||||
friend class StringIndexer;
|
||||
|
||||
@@ -36,13 +29,13 @@ class StringIndexer
|
||||
public:
|
||||
using index_t = uint8_t;
|
||||
|
||||
static const string& str(const index_t& index)
|
||||
{
|
||||
static string dummy;
|
||||
const auto& it=strings.find(index);
|
||||
if (it == strings.end()) return dummy;
|
||||
return it->second.str;
|
||||
}
|
||||
static const std::string& str(const index_t& index)
|
||||
{
|
||||
static std::string dummy;
|
||||
const auto& it=strings.find(index);
|
||||
if (it == strings.end()) return dummy;
|
||||
return it->second.str;
|
||||
}
|
||||
|
||||
static void use(const index_t& index)
|
||||
{
|
||||
@@ -84,7 +77,7 @@ class StringIndexer
|
||||
{
|
||||
if (strings.find(index)==strings.end())
|
||||
{
|
||||
strings[index].str = string(str, len);
|
||||
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;
|
||||
@@ -93,9 +86,7 @@ class StringIndexer
|
||||
return 0; // TODO out of indexes
|
||||
}
|
||||
|
||||
using Strings = std::unordered_map<index_t, StringCounter>;
|
||||
|
||||
static Strings strings;
|
||||
static std::map<index_t, StringCounter> strings;
|
||||
};
|
||||
|
||||
class IndexedString
|
||||
@@ -112,7 +103,7 @@ class IndexedString
|
||||
index=StringIndexer::strToIndex(str, len);
|
||||
}
|
||||
|
||||
IndexedString(const string& str) : IndexedString(str.c_str(), str.length()) {};
|
||||
IndexedString(const std::string& str) : IndexedString(str.c_str(), str.length()) {};
|
||||
|
||||
~IndexedString() { StringIndexer::release(index); }
|
||||
|
||||
@@ -133,7 +124,7 @@ class IndexedString
|
||||
return i1.index == i2.index;
|
||||
}
|
||||
|
||||
const string& str() const { return StringIndexer::str(index); }
|
||||
const std::string& str() const { return StringIndexer::str(index); }
|
||||
|
||||
const StringIndexer::index_t& getIndex() const { return index; }
|
||||
|
||||
|
||||
202
src/TinyMqtt.cpp
202
src/TinyMqtt.cpp
@@ -17,7 +17,7 @@ int TinyMqtt::debug=2;
|
||||
|
||||
MqttBroker::MqttBroker(uint16_t port)
|
||||
{
|
||||
server = new TcpServer(port);
|
||||
server.reset(new TcpServer(port));
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
server->onClient(onClient, this);
|
||||
#endif
|
||||
@@ -27,41 +27,31 @@ MqttBroker::~MqttBroker()
|
||||
{
|
||||
while(clients.size())
|
||||
{
|
||||
auto client = clients[0];
|
||||
client->local_broker = nullptr;
|
||||
if (client->cltFlags & MqttClient::CltFlags::CltFlagToDelete)
|
||||
{
|
||||
// std::cout << "Deleting client" << std::endl;
|
||||
delete client;
|
||||
}
|
||||
clients.erase(clients.begin());
|
||||
delete clients[0];
|
||||
}
|
||||
delete server;
|
||||
}
|
||||
|
||||
// private constructor used by broker only
|
||||
MqttClient::MqttClient(MqttBroker* local_broker, TcpClient* new_client)
|
||||
: local_broker(local_broker)
|
||||
{
|
||||
dclass;
|
||||
connect(local_broker);
|
||||
debug("MqttClient private with broker");
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
tcp_client = new_client;
|
||||
tcp_client.reset(new_client);
|
||||
tcp_client->onData(onData, this);
|
||||
// client->onConnect() TODO
|
||||
// client->onDisconnect() TODO
|
||||
#else
|
||||
tcp_client = new WiFiClient(*new_client);
|
||||
#endif
|
||||
#ifdef EPOXY_DUINO
|
||||
alive = millis()+500000;
|
||||
#else
|
||||
alive = millis()+5000; // TODO MAGIC client expires after 5s if no CONNECT msg
|
||||
tcp_client.reset(new WiFiClient(*new_client));
|
||||
#endif
|
||||
alive = millis()+5000;
|
||||
}
|
||||
|
||||
MqttClient::MqttClient(MqttBroker* local_broker, const string& id)
|
||||
MqttClient::MqttClient(MqttBroker* local_broker, const std::string& id)
|
||||
: local_broker(local_broker), clientId(id)
|
||||
{
|
||||
dclass;
|
||||
alive = 0;
|
||||
keep_alive = 0;
|
||||
|
||||
@@ -70,15 +60,15 @@ MqttClient::MqttClient(MqttBroker* local_broker, const string& id)
|
||||
|
||||
MqttClient::~MqttClient()
|
||||
{
|
||||
dtor;
|
||||
close();
|
||||
delete tcp_client;
|
||||
debug("*** MqttClient delete()");
|
||||
}
|
||||
|
||||
void MqttClient::close(bool bSendDisconnect)
|
||||
{
|
||||
debug("close " << id().c_str());
|
||||
resetFlag(CltFlagConnected);
|
||||
mqtt_flags &= ~FlagConnected;
|
||||
if (tcp_client) // connected to a remote broker
|
||||
{
|
||||
if (bSendDisconnect and tcp_client->connected())
|
||||
@@ -100,18 +90,18 @@ void MqttClient::close(bool bSendDisconnect)
|
||||
void MqttClient::connect(MqttBroker* local)
|
||||
{
|
||||
debug("MqttClient::connect_local");
|
||||
alive = 0;
|
||||
close();
|
||||
local_broker = local;
|
||||
local_broker->addClient(this);
|
||||
clientAlive();
|
||||
}
|
||||
|
||||
void MqttClient::connect(string broker, uint16_t port, uint16_t ka)
|
||||
void MqttClient::connect(std::string broker, uint16_t port, uint16_t ka)
|
||||
{
|
||||
debug("MqttClient::connect_to_host " << broker << ':' << port);
|
||||
keep_alive = ka;
|
||||
close();
|
||||
if (tcp_client) delete tcp_client;
|
||||
tcp_client = new TcpClient;
|
||||
tcp_client.reset(new TcpClient);
|
||||
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
tcp_client->onData(onData, this);
|
||||
@@ -121,7 +111,7 @@ void MqttClient::connect(string broker, uint16_t port, uint16_t ka)
|
||||
if (tcp_client->connect(broker.c_str(), port))
|
||||
{
|
||||
debug("link established");
|
||||
onConnect(this, tcp_client);
|
||||
onConnect(this, tcp_client.get());
|
||||
}
|
||||
else
|
||||
{
|
||||
@@ -130,13 +120,7 @@ void MqttClient::connect(string broker, uint16_t port, uint16_t ka)
|
||||
#endif
|
||||
}
|
||||
|
||||
void MqttBroker::addClient(MqttClient* client)
|
||||
{
|
||||
debug("MqttBroker::addClient");
|
||||
clients.push_back(client);
|
||||
}
|
||||
|
||||
void MqttBroker::connect(const string& host, uint16_t port)
|
||||
void MqttBroker::connect(const std::string& host, uint16_t port)
|
||||
{
|
||||
debug("MqttBroker::connect");
|
||||
if (remote_broker == nullptr) remote_broker = new MqttClient;
|
||||
@@ -146,24 +130,13 @@ void MqttBroker::connect(const string& host, uint16_t port)
|
||||
|
||||
void MqttBroker::removeClient(MqttClient* remove)
|
||||
{
|
||||
debug("removeClient");
|
||||
for(auto it=clients.begin(); it!=clients.end(); it++)
|
||||
{
|
||||
auto client=*it;
|
||||
if (client==remove)
|
||||
local_clients.erase(remove);
|
||||
for(auto it = clients.begin(); it!=clients.end(); it++)
|
||||
if (*it == remove)
|
||||
{
|
||||
// TODO if this broker is connected to an external broker
|
||||
// we have to unsubscribe remove's topics.
|
||||
// (but doing this, check that other clients are not subscribed...)
|
||||
// Unless -> we could receive useless messages
|
||||
// -> we are using (memory) one IndexedString plus its string for nothing.
|
||||
debug("Remove " << clients.size());
|
||||
clients.erase(it);
|
||||
debug("Client removed " << clients.size());
|
||||
return;
|
||||
break;
|
||||
}
|
||||
}
|
||||
debug(red << "Error cannot remove client"); // TODO should not occur
|
||||
}
|
||||
|
||||
void MqttBroker::onClient(void* broker_ptr, TcpClient* client)
|
||||
@@ -171,16 +144,15 @@ void MqttBroker::onClient(void* broker_ptr, TcpClient* client)
|
||||
debug("MqttBroker::onClient");
|
||||
MqttBroker* broker = static_cast<MqttBroker*>(broker_ptr);
|
||||
|
||||
MqttClient* mqtt = new MqttClient(broker, client);
|
||||
mqtt->setFlag(MqttClient::CltFlags::CltFlagToDelete);
|
||||
broker->addClient(mqtt);
|
||||
broker->clients.push_back(new MqttClient(broker, client));
|
||||
debug("New client");
|
||||
}
|
||||
|
||||
void MqttBroker::loop()
|
||||
{
|
||||
#ifndef TINY_MQTT_ASYNC
|
||||
WiFiClient client = server->accept();
|
||||
if (not server) return;
|
||||
WiFiClient client = server->available();
|
||||
|
||||
if (client)
|
||||
{
|
||||
@@ -194,9 +166,12 @@ void MqttBroker::loop()
|
||||
remote_broker->loop();
|
||||
}
|
||||
|
||||
for(size_t i=0; i<clients.size(); i++)
|
||||
// keep track on size because loop can remove a client from containers
|
||||
// loop on remote clients (connected through network)
|
||||
auto size = clients.size();
|
||||
for(auto it = clients.begin(); it!=clients.end(); it++)
|
||||
{
|
||||
MqttClient* client = clients[i];
|
||||
MqttClient* client = *it;
|
||||
if (client->connected())
|
||||
{
|
||||
client->loop();
|
||||
@@ -205,9 +180,16 @@ void MqttBroker::loop()
|
||||
{
|
||||
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.
|
||||
delete client;
|
||||
break;
|
||||
}
|
||||
if (size != clients.size()) break;
|
||||
}
|
||||
|
||||
// loop on local clients (on same device as the broker's)
|
||||
size = local_clients.size();
|
||||
for(auto& client: local_clients)
|
||||
{
|
||||
client->loop();
|
||||
if (local_clients.size() != size) break;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -223,40 +205,28 @@ MqttError MqttBroker::subscribe(const Topic& topic, uint8_t qos)
|
||||
|
||||
MqttError MqttBroker::publish(const MqttClient* source, const Topic& topic, MqttMessage& msg) const
|
||||
{
|
||||
MqttError retval = MqttOk;
|
||||
MqttError retval = MqttOk; // TODO here retval is badly computed
|
||||
|
||||
debug("MqttBroker::publish");
|
||||
int i=0;
|
||||
for(auto client: clients)
|
||||
{
|
||||
i++;
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << __LINE__ << " broker:" << (remote_broker && remote_broker->connected() ? "linked" : "alone") <<
|
||||
" srce=" << (source->isLocal() ? "loc" : "rem") << " clt#" << i << ", local=" << client->isLocal() << ", con=" << client->connected() << endl;
|
||||
#endif
|
||||
bool doit = false;
|
||||
if (remote_broker && remote_broker->connected()) // this (MqttBroker) is connected (to a external broker)
|
||||
{
|
||||
// ext_broker -> clients or clients -> ext_broker
|
||||
if (source == remote_broker) // external broker -> internal clients
|
||||
doit = true;
|
||||
else // external clients -> this broker
|
||||
{
|
||||
// As this broker is connected to another broker, simply forward the msg
|
||||
MqttError ret = remote_broker->publishIfSubscribed(topic, msg);
|
||||
if (ret != MqttOk) retval = ret;
|
||||
}
|
||||
}
|
||||
else // Disconnected
|
||||
{
|
||||
doit = true;
|
||||
}
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << ", doit=" << doit << ' ';
|
||||
#endif
|
||||
|
||||
if (doit) retval = client->publishIfSubscribed(topic, msg);
|
||||
debug("");
|
||||
if (remote_broker == nullptr or source == remote_broker) // external broker -> internal clients
|
||||
{
|
||||
for(auto& client: clients)
|
||||
{
|
||||
retval = client->publishIfSubscribed(topic, msg);
|
||||
}
|
||||
for(auto& client: local_clients)
|
||||
{
|
||||
retval = client->publishIfSubscribed(topic, msg);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (remote_broker && remote_broker->connected())
|
||||
{
|
||||
MqttError ret = remote_broker->publishIfSubscribed(topic, msg);
|
||||
if (ret != MqttOk) retval = ret;
|
||||
}
|
||||
}
|
||||
return retval;
|
||||
}
|
||||
@@ -277,16 +247,12 @@ void MqttMessage::getString(const char* &buff, uint16_t& len)
|
||||
buff+=2;
|
||||
}
|
||||
|
||||
void MqttClient::clientAlive(uint32_t more_seconds)
|
||||
void MqttClient::clientAlive()
|
||||
{
|
||||
debug("MqttClient::clientAlive");
|
||||
if (keep_alive)
|
||||
{
|
||||
#ifdef EPOXY_DUINO
|
||||
alive=millis()+500000+0*more_seconds;
|
||||
#else
|
||||
alive=millis()+1000*(keep_alive+more_seconds);
|
||||
#endif
|
||||
alive=millis()+1000*(keep_alive+(local_broker ? TINY_MQTT_CLIENT_ALIVE_TOLERANCE : 0));
|
||||
}
|
||||
else
|
||||
alive=0;
|
||||
@@ -298,7 +264,7 @@ void MqttClient::loop()
|
||||
{
|
||||
if (local_broker)
|
||||
{
|
||||
debug(red << "timeout client");
|
||||
Serial << "timeout client " << clientId << endl;
|
||||
close();
|
||||
debug(red << "closed");
|
||||
}
|
||||
@@ -306,8 +272,9 @@ void MqttClient::loop()
|
||||
{
|
||||
debug("pingreq");
|
||||
uint16_t pingreq = MqttMessage::Type::PingReq;
|
||||
|
||||
tcp_client->write((const char*)(&pingreq), 2);
|
||||
clientAlive(0);
|
||||
clientAlive();
|
||||
|
||||
// TODO when many MqttClient passes through a local broker
|
||||
// there is no need to send one PingReq per instance.
|
||||
@@ -343,7 +310,7 @@ void MqttClient::onConnect(void *mqttclient_ptr, TcpClient*)
|
||||
msg.reset();
|
||||
debug("cnx: mqtt sent " << (dbg_ptr)mqtt->local_broker);
|
||||
|
||||
mqtt->clientAlive(0);
|
||||
mqtt->clientAlive();
|
||||
}
|
||||
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
@@ -450,13 +417,14 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
switch(mesg->type())
|
||||
{
|
||||
case MqttMessage::Type::Connect:
|
||||
if (mqtt_connected())
|
||||
if (mqtt_flags & FlagConnected)
|
||||
{
|
||||
debug("already connected");
|
||||
break;
|
||||
}
|
||||
payload = header+10;
|
||||
mqtt_flags = header[7];
|
||||
// Todo should check that reserved == 0 (spec)
|
||||
mqtt_flags = header[7] & ~FlagConnected;
|
||||
keep_alive = MqttMessage::getSize(header+8);
|
||||
if (strncmp("MQTT", header+2,4))
|
||||
{
|
||||
@@ -465,13 +433,13 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
}
|
||||
if (header[6]!=0x04)
|
||||
{
|
||||
debug("Unsupported MQTT version (" << (int) header[6] << "), only version=4 supported" << endl);
|
||||
debug("unknown level");
|
||||
break; // Level 3.1.1
|
||||
}
|
||||
|
||||
// ClientId
|
||||
mesg->getString(payload, len);
|
||||
clientId = string(payload, len);
|
||||
clientId = std::string(payload, len);
|
||||
payload += len;
|
||||
|
||||
if (mqtt_flags & FlagWill) // Will topic
|
||||
@@ -500,7 +468,7 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
Console << yellow << "Client " << clientId << " connected : keep alive=" << keep_alive << '.' << white << endl;
|
||||
#endif
|
||||
bclose = false;
|
||||
setFlag(CltFlagConnected);
|
||||
mqtt_flags |= FlagConnected;
|
||||
{
|
||||
MqttMessage msg(MqttMessage::Type::ConnAck);
|
||||
msg.add(0); // Session present (not implemented)
|
||||
@@ -510,14 +478,14 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::ConnAck:
|
||||
setFlag(CltFlagConnected);
|
||||
mqtt_flags |= FlagConnected;
|
||||
bclose = false;
|
||||
resubscribe();
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::SubAck:
|
||||
case MqttMessage::Type::PubAck:
|
||||
if (not mqtt_connected()) break;
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
// Ignore acks
|
||||
bclose = false;
|
||||
break;
|
||||
@@ -528,7 +496,7 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::PingReq:
|
||||
if (not mqtt_connected()) break;
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
if (tcp_client)
|
||||
{
|
||||
uint16_t pingreq = MqttMessage::Type::PingResp;
|
||||
@@ -545,15 +513,15 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
case MqttMessage::Type::Subscribe:
|
||||
case MqttMessage::Type::UnSubscribe:
|
||||
{
|
||||
if (not mqtt_connected()) break;
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
payload = header+2;
|
||||
|
||||
debug("un/subscribe loop");
|
||||
string qoss;
|
||||
std::string qoss;
|
||||
while(payload < mesg->end())
|
||||
{
|
||||
mesg->getString(payload, len); // Topic
|
||||
debug( " topic (" << string(payload, len) << ')');
|
||||
debug( " topic (" << std::string(payload, len) << ')');
|
||||
// subscribe(Topic(payload, len));
|
||||
Topic topic(payload, len);
|
||||
|
||||
@@ -589,15 +557,15 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::UnSuback:
|
||||
if (not mqtt_connected()) break;
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
bclose = false;
|
||||
break;
|
||||
|
||||
case MqttMessage::Type::Publish:
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << "publish " << mqtt_connected() << '/' << (long) tcp_client << endl;
|
||||
Console << "publish " << (mqtt_flags & FlagConnected) << '/' << (long) tcp_client.get() << endl;
|
||||
#endif
|
||||
if (mqtt_connected() or tcp_client == nullptr)
|
||||
if ((mqtt_flags & FlagConnected) or tcp_client == nullptr)
|
||||
{
|
||||
uint8_t qos = mesg->flags();
|
||||
payload = header;
|
||||
@@ -607,7 +575,7 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << "Received Publish (" << published.str().c_str() << ") size=" << (int)len << endl;
|
||||
#endif
|
||||
// << '(' << 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
|
||||
len=mesg->end()-payload;
|
||||
// TODO reset DUP
|
||||
@@ -638,8 +606,8 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
|
||||
case MqttMessage::Type::Disconnect:
|
||||
// TODO should discard any will msg
|
||||
if (not mqtt_connected()) break;
|
||||
resetFlag(CltFlagConnected);
|
||||
if (not (mqtt_flags & FlagConnected)) break;
|
||||
mqtt_flags &= ~FlagConnected;
|
||||
close(false);
|
||||
bclose=false;
|
||||
break;
|
||||
@@ -660,7 +628,7 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
}
|
||||
else
|
||||
{
|
||||
clientAlive(local_broker ? 5 : 0);
|
||||
clientAlive();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -899,7 +867,7 @@ void MqttMessage::hexdump(const char* prefix) const
|
||||
(void)prefix;
|
||||
#if TINY_MQTT_DEBUG
|
||||
if (TinyMqtt::debug<2) return;
|
||||
static std::map<Type, string> tts={
|
||||
static std::map<Type, std::string> tts={
|
||||
{ Connect, "Connect" },
|
||||
{ ConnAck, "Connack" },
|
||||
{ Publish, "Publish" },
|
||||
@@ -912,7 +880,7 @@ void MqttMessage::hexdump(const char* prefix) const
|
||||
{ PingResp, "Pingresp" },
|
||||
{ Disconnect, "Disconnect" }
|
||||
};
|
||||
string t("Unknown");
|
||||
std::string t("Unknown");
|
||||
Type typ=static_cast<Type>(buffer[0] & 0xF0);
|
||||
if (tts.find(typ) != tts.end())
|
||||
t=tts[typ];
|
||||
@@ -929,7 +897,7 @@ void MqttMessage::hexdump(const char* prefix) const
|
||||
const char* hex_to_str = " | ";
|
||||
const char* separator = hex_to_str;
|
||||
const char* half_sep = " - ";
|
||||
string ascii;
|
||||
std::string ascii;
|
||||
|
||||
Console << prefix << " size(" << buffer.size() << "), state=" << state << endl;
|
||||
|
||||
|
||||
@@ -4,6 +4,10 @@
|
||||
#ifndef TINY_MQTT_DEBUG
|
||||
#define TINY_MQTT_DEBUG 0
|
||||
#endif
|
||||
#ifndef TINY_MQTT_DEFAULT_ALIVE
|
||||
#define TINY_MQTT_DEFAULT_ALIVE 10
|
||||
#endif
|
||||
#define TINY_MQTT_CLIENT_ALIVE_TOLERANCE 5
|
||||
|
||||
// TODO Should add a AUnit with both TINY_MQTT_ASYNC and not TINY_MQTT_ASYNC
|
||||
// #define TINY_MQTT_ASYNC // Uncomment this to use ESPAsyncTCP instead of normal cnx
|
||||
@@ -53,6 +57,15 @@
|
||||
#define debug(what) {}
|
||||
#endif
|
||||
|
||||
#include <TinyConsole.h>
|
||||
#if 0
|
||||
#define dclass { Console << __LINE__ << ':' << __PRETTY_FUNCTION__ << ", this=" << (long)this << endl; }
|
||||
#define dtor { Console << __LINE__ << ": ~" << __PRETTY_FUNCTION__ << ", this=" << (long)this << endl; }
|
||||
#else
|
||||
#define dclass
|
||||
#define dtor
|
||||
#endif
|
||||
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
using TcpClient = AsyncClient;
|
||||
using TcpServer = AsyncServer;
|
||||
@@ -68,15 +81,12 @@ enum __attribute__((packed)) MqttError
|
||||
MqttInvalidMessage=2,
|
||||
};
|
||||
|
||||
using string = TinyConsole::string;
|
||||
|
||||
class Topic : public IndexedString
|
||||
{
|
||||
public:
|
||||
Topic(const string& m) : IndexedString(m){}
|
||||
Topic(const char* s, uint8_t len) : IndexedString(s,len){}
|
||||
Topic(const char* s) : Topic(s, strlen(s)) {}
|
||||
// Topic(const string s) : Topic(s.c_str(), s.length()){};
|
||||
Topic(const std::string s) : Topic(s.c_str(), s.length()){};
|
||||
|
||||
const char* c_str() const { return str().c_str(); }
|
||||
|
||||
@@ -125,7 +135,7 @@ class MqttMessage
|
||||
void incoming(char byte);
|
||||
void add(char byte) { incoming(byte); }
|
||||
void add(const char* p, size_t len, bool addLength=true );
|
||||
void add(const string& s) { add(s.c_str(), s.length()); }
|
||||
void add(const std::string& s) { add(s.c_str(), s.length()); }
|
||||
void add(const Topic& t) { add(t.str()); }
|
||||
const char* end() const { return &buffer[0]+buffer.size(); }
|
||||
const char* getVHeader() const { return &buffer[vheader]; }
|
||||
@@ -159,7 +169,7 @@ class MqttMessage
|
||||
private:
|
||||
void encodeLength();
|
||||
|
||||
string buffer;
|
||||
std::string buffer;
|
||||
uint8_t vheader;
|
||||
uint16_t size; // bytes left to receive
|
||||
State state;
|
||||
@@ -176,14 +186,9 @@ class MqttClient
|
||||
FlagWillQos = 16 | 8, // unsupported
|
||||
FlagWill = 4, // unsupported
|
||||
FlagCleanSession = 2, // unsupported
|
||||
FlagReserved = 1
|
||||
};
|
||||
|
||||
enum __attribute__((packed)) CltFlags
|
||||
{
|
||||
CltFlagNone = 0,
|
||||
CltFlagConnected = 1,
|
||||
CltFlagToDelete = 2
|
||||
FlagReserved = 1, // use reserved as connected (save 1 byte)
|
||||
FlagConnected = 1
|
||||
};
|
||||
public:
|
||||
|
||||
@@ -191,13 +196,13 @@ class MqttClient
|
||||
|
||||
/** Constructor. Broker is the adress of a local broker if not null
|
||||
If you want to connect elsewhere, leave broker null and use connect() **/
|
||||
MqttClient(MqttBroker* broker = nullptr, const string& id = TINY_MQTT_DEFAULT_CLIENT_ID);
|
||||
MqttClient(const string& id) : MqttClient(nullptr, id){}
|
||||
MqttClient(MqttBroker* broker = nullptr, const std::string& id = TINY_MQTT_DEFAULT_CLIENT_ID);
|
||||
MqttClient(const std::string& id) : MqttClient(nullptr, id){ dclass; }
|
||||
|
||||
~MqttClient();
|
||||
|
||||
void connect(MqttBroker* local_broker);
|
||||
void connect(string broker, uint16_t port, uint16_t keep_alive = 10);
|
||||
void connect(std::string broker, uint16_t port, uint16_t keep_alive = TINY_MQTT_DEFAULT_ALIVE);
|
||||
|
||||
// TODO it seems that connected returns true in tcp mode even if
|
||||
// no negociation occurred
|
||||
@@ -212,8 +217,8 @@ class MqttClient
|
||||
if (tcp_client) tcp_client->write(buf, length);
|
||||
}
|
||||
|
||||
const string& id() const { return clientId; }
|
||||
void id(const string& new_id) { clientId = new_id; }
|
||||
const std::string& id() const { return clientId; }
|
||||
void id(const std::string& new_id) { clientId = new_id; }
|
||||
|
||||
/** Should be called in main loop() */
|
||||
void loop();
|
||||
@@ -231,7 +236,7 @@ class MqttClient
|
||||
MqttError publish(const Topic&, const char* payload, size_t pay_length);
|
||||
MqttError publish(const Topic& t, const char* payload) { return publish(t, payload, strlen(payload)); }
|
||||
MqttError publish(const Topic& t, const String& s) { return publish(t, s.c_str(), s.length()); }
|
||||
MqttError publish(const Topic& t, const string& s) { return publish(t,s.c_str(),s.length());}
|
||||
MqttError publish(const Topic& t, const std::string& s) { return publish(t,s.c_str(),s.length());}
|
||||
MqttError publish(const Topic& t) { return publish(t, nullptr, 0);};
|
||||
|
||||
MqttError subscribe(Topic topic, uint8_t qos=0);
|
||||
@@ -242,7 +247,7 @@ class MqttClient
|
||||
// TODO seems to be useless
|
||||
bool isLocal() const { return tcp_client == nullptr; }
|
||||
|
||||
void dump(string indent="")
|
||||
void dump(std::string indent="")
|
||||
{
|
||||
(void)indent;
|
||||
#if TINY_MQTT_DEBUG
|
||||
@@ -279,9 +284,6 @@ class MqttClient
|
||||
uint32_t keepAlive() const { return keep_alive; }
|
||||
|
||||
private:
|
||||
bool mqtt_connected() const { return cltFlags & CltFlagConnected; }
|
||||
void setFlag(CltFlags f) { cltFlags |= f; }
|
||||
void resetFlag(CltFlags f) { cltFlags &= ~f; }
|
||||
|
||||
// event when tcp/ip link established (real or fake)
|
||||
static void onConnect(void * client_ptr, TcpClient*);
|
||||
@@ -296,22 +298,23 @@ class MqttClient
|
||||
// republish a received publish if topic matches any in subscriptions
|
||||
MqttError publishIfSubscribed(const Topic& topic, MqttMessage& msg);
|
||||
|
||||
void clientAlive(uint32_t more_seconds);
|
||||
void clientAlive();
|
||||
void processMessage(MqttMessage* message);
|
||||
|
||||
uint8_t cltFlags = CltFlagNone;
|
||||
char mqtt_flags;
|
||||
uint32_t keep_alive = 30;
|
||||
uint32_t alive;
|
||||
char mqtt_flags = 0;
|
||||
uint16_t keep_alive = 30;
|
||||
// for client connected to remote broker, PingReq is sent when millis() >= alive
|
||||
// for a client managed by a broker, disconnect it if millis() >= alive
|
||||
uint32_t alive; // PingReq if millis() > alive,
|
||||
MqttMessage message;
|
||||
|
||||
// connection to local broker, or link to the parent
|
||||
// when MqttBroker uses MqttClient for each external connexion
|
||||
MqttBroker* local_broker=nullptr;
|
||||
|
||||
TcpClient* tcp_client=nullptr; // connection to remote broker
|
||||
std::set<Topic> subscriptions;
|
||||
string clientId;
|
||||
std::unique_ptr<TcpClient> tcp_client; // connection to remote broker
|
||||
std::set<Topic> subscriptions;
|
||||
std::string clientId;
|
||||
CallBack callback = nullptr;
|
||||
};
|
||||
|
||||
@@ -321,30 +324,30 @@ class MqttBroker
|
||||
{
|
||||
Disconnected, // Also the initial state
|
||||
Connecting, // connect and sends a fake publish to avoid circular cnx
|
||||
Connected, // this->broker is connected and circular cnx avoided
|
||||
Connected, // this->broker is connected and circular cnx avoided
|
||||
};
|
||||
public:
|
||||
// TODO limit max number of clients
|
||||
MqttBroker(uint16_t port);
|
||||
~MqttBroker();
|
||||
|
||||
void begin() { server->begin(); }
|
||||
void begin() { if (server) server->begin(); }
|
||||
void loop();
|
||||
|
||||
/** Connect the broker to a parent broker */
|
||||
void connect(const string& host, uint16_t port=1883);
|
||||
/** returns true if connected to another broker */
|
||||
void connect(const std::string& host, uint16_t port=1883);
|
||||
bool connected() const { return state == Connected; }
|
||||
|
||||
size_t clientsCount() const { return clients.size(); }
|
||||
|
||||
void dump(string indent="")
|
||||
void dump(std::string indent="")
|
||||
{
|
||||
for(auto client: clients)
|
||||
for(auto& client: clients)
|
||||
client->dump(indent);
|
||||
}
|
||||
|
||||
const std::vector<MqttClient*> getClients() const { return clients; }
|
||||
size_t localClientsCount() const { return local_clients.size(); }
|
||||
using Clients = std::vector<MqttClient*>;
|
||||
const Clients& getClients() const { return clients; }
|
||||
|
||||
private:
|
||||
friend class MqttClient;
|
||||
@@ -362,14 +365,15 @@ class MqttBroker
|
||||
MqttError subscribe(const Topic& topic, uint8_t qos);
|
||||
|
||||
// For clients that are added not by the broker itself (local clients)
|
||||
void addClient(MqttClient* client);
|
||||
void addClient(MqttClient* local) { local_clients.insert(local); }
|
||||
void removeClient(MqttClient* client);
|
||||
|
||||
bool compareString(const char* good, const char* str, uint8_t str_len) const;
|
||||
std::vector<MqttClient*> clients;
|
||||
Clients clients;
|
||||
std::set<MqttClient*> local_clients;
|
||||
|
||||
private:
|
||||
TcpServer* server = nullptr;
|
||||
std::unique_ptr<TcpServer> server;
|
||||
|
||||
const char* auth_user = "guest";
|
||||
const char* auth_password = "guest";
|
||||
|
||||
@@ -1,22 +1,11 @@
|
||||
SUB=
|
||||
|
||||
all:runtests
|
||||
|
||||
tests:
|
||||
@set -e; \
|
||||
for i in $(SUB)*-tests/Makefile; do \
|
||||
for i in ${SUB}*-tests/Makefile; do \
|
||||
echo '==== Making:' $$(dirname $$i); \
|
||||
$(MAKE) -C $$(dirname $$i) -j; \
|
||||
done
|
||||
$(MAKE) -C compile-test
|
||||
|
||||
valgrind:
|
||||
@set -e; \
|
||||
$(MAKE) tests; \
|
||||
for i in $(SUB)*-tests/Makefile; do \
|
||||
echo '==== Running:' $$(dirname $$i); \
|
||||
valgrind $$(dirname $$i)/$$(dirname $$i).out; \
|
||||
done
|
||||
|
||||
debugtest:
|
||||
@set -e; \
|
||||
@@ -24,17 +13,18 @@ debugtest:
|
||||
$(MAKE) -C debug-mode -j; \
|
||||
debug-mode/debug-tests.out
|
||||
|
||||
runtests:
|
||||
runtests: debugtest
|
||||
@$(MAKE) clean
|
||||
@$(MAKE) tests
|
||||
@set -e; \
|
||||
$(MAKE) tests; \
|
||||
for i in $(SUB)*-tests/Makefile; do \
|
||||
for i in ${SUB}*-tests/Makefile; do \
|
||||
echo '==== Running:' $$(dirname $$i); \
|
||||
$$(dirname $$i)/$$(dirname $$i).out; \
|
||||
done
|
||||
|
||||
clean:
|
||||
@set -e; \
|
||||
for i in $(SUB)*-tests/Makefile; do \
|
||||
for i in ${SUB}*-tests/Makefile; do \
|
||||
echo '==== Cleaning:' $$(dirname $$i); \
|
||||
$(MAKE) -C $$(dirname $$i) clean; \
|
||||
done
|
||||
|
||||
@@ -1,11 +0,0 @@
|
||||
|
||||
# GCC
|
||||
# CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
CXXFLAGS=-D_GNU_SOURCE -Werror=return-type -std=gnu++17 -Wall -g3 -O0
|
||||
|
||||
# CLANG SANITIZE
|
||||
# CXX=clang
|
||||
# EXTRA_CXXFLAGS=-g3 -O0 -fsanitize=memory
|
||||
# LDFLAGS = -lpthread -lstdc++
|
||||
@@ -3,6 +3,9 @@
|
||||
|
||||
EXTRA_CXXFLAGS=-g3 -O0 -DTINY_MQTT_TESTS
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics -DEPOXY_TEST
|
||||
|
||||
APP_NAME := classbind-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
|
||||
@@ -17,10 +17,9 @@ class TestReceiver : public MqttClassBinder<TestReceiver>
|
||||
public:
|
||||
TestReceiver(const char* name) : MqttClassBinder(), name_(name) {}
|
||||
|
||||
void onPublish(const MqttClient* /* source */, const Topic& topic, const char* /* payload */, size_t /* length */)
|
||||
void onPublish(const MqttClient* /* source */, const Topic& topic, const char* payload, size_t /* length */)
|
||||
{
|
||||
(void) topic;
|
||||
// Serial << "--> routed message received by " << name_ << ':' << topic.c_str() << " = " << payload << endl;
|
||||
Serial << "--> routed message received by " << name_ << ':' << topic.c_str() << " = " << payload << endl;
|
||||
messages[name_]++;
|
||||
}
|
||||
|
||||
@@ -36,8 +35,7 @@ std::map<std::string, int> TestReceiver::messages;
|
||||
static int unrouted = 0;
|
||||
void onUnrouted(const MqttClient*, const Topic& topic, const char*, size_t)
|
||||
{
|
||||
(void) topic;
|
||||
// Serial << "--> unrouted: " << topic.c_str() << endl;
|
||||
Serial << "--> unrouted: " << topic.c_str() << endl;
|
||||
unrouted++;
|
||||
}
|
||||
|
||||
@@ -129,8 +127,9 @@ void reset_and_start_servers(int n, bool early_accept = true)
|
||||
}
|
||||
}
|
||||
|
||||
test(classbind_one_client_receives_the_message)
|
||||
test(classbind_two_subscribers_binded_one_sender_wildcard)
|
||||
{
|
||||
EpoxyTest::set_millis(0);
|
||||
reset_and_start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
@@ -140,7 +139,50 @@ test(classbind_one_client_receives_the_message)
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
MqttClient mqtt_a(&broker, "mqtt_a");
|
||||
MqttClient mqtt_b(&broker, "mqtt_a");
|
||||
MqttClient mqtt_sender(&broker, "sender");
|
||||
|
||||
broker.loop();
|
||||
|
||||
assertTrue(mqtt_a.connected());
|
||||
assertTrue(mqtt_b.connected());
|
||||
|
||||
TestReceiver receiver("receiver");
|
||||
MqttClassBinder<TestReceiver>::onPublish(&mqtt_a, &receiver);
|
||||
MqttClassBinder<TestReceiver>::onPublish(&mqtt_b, &receiver);
|
||||
|
||||
mqtt_a.subscribe("#");
|
||||
mqtt_b.subscribe("#");
|
||||
mqtt_sender.publish("a/b", "ab");
|
||||
|
||||
for (int i =0; i<10; i++)
|
||||
{
|
||||
EpoxyTest::add_millis(100);
|
||||
mqtt_a.loop();
|
||||
mqtt_b.loop();
|
||||
mqtt_sender.loop();
|
||||
broker.loop();
|
||||
}
|
||||
|
||||
assertEqual(TestReceiver::messages["receiver"], 2);
|
||||
assertEqual(unrouted, 0);
|
||||
EpoxyTest::set_real_time();
|
||||
}
|
||||
|
||||
test(classbind_one_client_receives_the_message)
|
||||
{
|
||||
EpoxyTest::set_millis(0);
|
||||
reset_and_start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress ip_broker = WiFi.localIP();
|
||||
|
||||
// We have a 2nd ESP in order to test through wifi (opposed to local)
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client("sender");
|
||||
client.connect(ip_broker.toString().c_str(), 1883);
|
||||
broker.loop();
|
||||
assertTrue(client.connected());
|
||||
@@ -153,12 +195,14 @@ test(classbind_one_client_receives_the_message)
|
||||
|
||||
for (int i =0; i<10; i++)
|
||||
{
|
||||
EpoxyTest::add_millis(100);
|
||||
client.loop();
|
||||
broker.loop();
|
||||
}
|
||||
|
||||
assertEqual(TestReceiver::messages["receiver"], 1);
|
||||
assertEqual(unrouted, 0);
|
||||
EpoxyTest::set_real_time();
|
||||
}
|
||||
|
||||
test(classbind_routes_should_be_empty_when_receiver_goes_out_of_scope)
|
||||
@@ -340,7 +384,7 @@ void setup() {
|
||||
while(!Serial);
|
||||
*/
|
||||
|
||||
Serial.println("=============[ TinyMqtt class-bind TESTS ]========================");
|
||||
Serial.println("=============[ CLASS BINDER TinyMqtt TESTS ]========================");
|
||||
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin("network", "password");
|
||||
|
||||
@@ -1,11 +0,0 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
include ../Makefile.opts
|
||||
|
||||
APP_NAME := tinymqtt-test
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsync ESP8266mDNS TinyConsole
|
||||
ARDUINO_LIB_DIRS := ../../../EspMock/libraries
|
||||
EPOXY_CORE := EPOXY_CORE_ESP8266
|
||||
DUMMY := 1
|
||||
include ../../../EpoxyDuino/EpoxyDuino.mk
|
||||
@@ -1,848 +0,0 @@
|
||||
// vim: ts=2 sw=2 expandtab smartindent
|
||||
#include <TinyConsole.h>
|
||||
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
||||
#include <TinyStreaming.h>
|
||||
|
||||
#include <sstream>
|
||||
#include <string>
|
||||
#include <map>
|
||||
|
||||
using string = TinyConsole::string;
|
||||
|
||||
bool echo_on = true;
|
||||
auto green = TinyConsole::green;
|
||||
auto red = TinyConsole::red;
|
||||
auto white = TinyConsole::white;
|
||||
auto cyan = TinyConsole::cyan;
|
||||
auto yellow = TinyConsole::yellow;
|
||||
auto magenta = TinyConsole::magenta;
|
||||
auto save_cursor = TinyConsole::save_cursor;
|
||||
auto restore_cursor = TinyConsole::restore_cursor;
|
||||
auto erase_to_end = TinyConsole::erase_to_end;
|
||||
|
||||
const char* ssid = "";
|
||||
const char* password = "";
|
||||
|
||||
struct free_broker
|
||||
{
|
||||
public:
|
||||
free_broker(const char* s, uint16_t p, const char* /* comment */) : url(s), port(p) {}
|
||||
|
||||
string url;
|
||||
uint16_t port;
|
||||
};
|
||||
|
||||
const std::map<string, free_broker> list =
|
||||
{
|
||||
{ "mqtthq", { "public.mqtthq.com" , 8083, "publish/subscribe" }},
|
||||
{ "hivemq", { "broker.hivemq.com", 1883, "" }}
|
||||
};
|
||||
|
||||
/** Very complex example
|
||||
* Console allowing to make any kind of test,
|
||||
* even some stress tests.
|
||||
*
|
||||
* Upload the sketch, the use the terminal.
|
||||
* Press H for mini help.
|
||||
*
|
||||
* tested with mqtt-spy-0.5.4
|
||||
* TODO examples of scripts
|
||||
*/
|
||||
|
||||
void onPublish(const MqttClient* srce, const Topic& topic, const char* payload, size_t length)
|
||||
{
|
||||
Console << cyan << "--> " << srce->id().c_str() << ": received " << topic.c_str() << white;
|
||||
if (payload)
|
||||
{
|
||||
Console << ", payload[" << length << "]=[";
|
||||
while(length--)
|
||||
{
|
||||
const char c=*payload++;
|
||||
if (c<32)
|
||||
Console << '?';
|
||||
else
|
||||
Console << c;
|
||||
}
|
||||
Console << ']' << endl;
|
||||
}
|
||||
}
|
||||
|
||||
std::map<string, MqttClient*> clients;
|
||||
std::map<string, MqttBroker*> brokers;
|
||||
std::map<string, string> vars;
|
||||
|
||||
void eval(string& cmd);
|
||||
|
||||
void replace(const char* d, string& str, string srch, string to)
|
||||
{
|
||||
if (d[0] && d[1])
|
||||
{
|
||||
srch=d[0]+srch+d[1];
|
||||
to=d[0]+to+d[1];
|
||||
|
||||
size_t pos = 0;
|
||||
while((pos=str.find(srch, pos)) != string::npos)
|
||||
{
|
||||
str.erase(pos, srch.length());
|
||||
str.insert(pos, to);
|
||||
pos += to.length()-1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void replaceVars(string& cmd)
|
||||
{
|
||||
cmd = ' '+cmd+' ';
|
||||
|
||||
for(auto it: vars)
|
||||
{
|
||||
replace("..", cmd, it.first, it.second);
|
||||
replace(". ", cmd, it.first, it.second);
|
||||
replace(" .", cmd, it.first, it.second);
|
||||
replace(" ", cmd, it.first, it.second);
|
||||
}
|
||||
cmd.erase(0, cmd.find_first_not_of(' '));
|
||||
cmd.erase(cmd.find_last_not_of(' ')+1);
|
||||
|
||||
}
|
||||
|
||||
void onCommand(const string& command)
|
||||
{
|
||||
Console << endl;
|
||||
string cmd=command;
|
||||
if (cmd.substr(0,3)!="set") replaceVars(cmd);
|
||||
eval(cmd);
|
||||
Console << endl;
|
||||
Console.prompt();
|
||||
}
|
||||
|
||||
void setup()
|
||||
{
|
||||
Serial.begin(115200);
|
||||
Console.begin(Serial);
|
||||
Console.setPrompt("> ");
|
||||
Console.setCallback(onCommand);
|
||||
delay(500);
|
||||
|
||||
Console.cls();
|
||||
Console << endl << endl;
|
||||
Console << yellow
|
||||
<< "***************************************************************" << endl;
|
||||
Console << "* Welcome to the TinyMqtt console" << endl;
|
||||
Console << "* The console allows to test all features of the libraries." << endl;
|
||||
Console << "* Enter help to view the list of commands." << endl;
|
||||
Console << "***************************************************************" << endl;
|
||||
Console << endl;
|
||||
if (strlen(ssid)==0)
|
||||
Console << red << "* ERROR: You must modify ssid/password in order" << endl
|
||||
<< " to be able to connect to your Wifi network." << endl;
|
||||
Console << endl << white;
|
||||
|
||||
Console << "Connecting to '" << ssid << "' ";
|
||||
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin(ssid, password);
|
||||
WiFi.mode(WIFI_STA);
|
||||
WiFi.begin(ssid, password);
|
||||
|
||||
while (WiFi.status() != WL_CONNECTED)
|
||||
{ Console << '-'; delay(500); }
|
||||
|
||||
Console << endl << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
|
||||
const char* name="tinytest";
|
||||
Console << "Starting MDNS, name= " << name;
|
||||
|
||||
MqttBroker* broker = new MqttBroker(1883);
|
||||
broker->begin();
|
||||
brokers["broker"] = broker;
|
||||
|
||||
if (Console.isTerm()) onCommand("every 333 view");
|
||||
}
|
||||
|
||||
string getword(string& str, const char* if_empty=nullptr, char sep=' ');
|
||||
|
||||
int getint(string& str, const int if_empty=0)
|
||||
{
|
||||
string str2=str;
|
||||
string sword = getword(str);
|
||||
if (sword[0] and isdigit(sword[0]))
|
||||
{
|
||||
int ret=atoi(sword.c_str());
|
||||
while(isdigit(sword[0]) or sword[0]==' ') sword.erase(0,1);
|
||||
if (sword.length()) str = sword+' '+str;
|
||||
return ret;
|
||||
}
|
||||
str=str2;
|
||||
return if_empty;
|
||||
}
|
||||
|
||||
string getword(string& str, const char* if_empty/*=nullptr*/, char sep/*=' '*/)
|
||||
{
|
||||
char quote=(str[0]=='"' or str[0]=='\'' ? str[0] : 0);
|
||||
if (quote) str.erase(0,1);
|
||||
string sword;
|
||||
while(str.length() and (str[0]!=sep or quote))
|
||||
{
|
||||
if (str[0]==quote)
|
||||
{
|
||||
str.erase(0,1);
|
||||
break;
|
||||
}
|
||||
sword += str[0];
|
||||
str.erase(0,1);
|
||||
}
|
||||
while(str[0]==sep) str.erase(0,1);
|
||||
if (if_empty and sword.length()==0) return if_empty;
|
||||
if (quote==false and sword.length()>=4 and sword.substr(0,3)=="rnd")
|
||||
{
|
||||
sword.erase(0,3);
|
||||
if (sword[0]=='(')
|
||||
{
|
||||
int to = 100;
|
||||
sword.erase(0,1);
|
||||
int from=getint(sword);
|
||||
if (sword[0]==',')
|
||||
{
|
||||
sword.erase(0,1);
|
||||
to = getint(sword);
|
||||
if (sword[0]!=')') Console << "Missing ')'" << endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
to=from;
|
||||
from=0;
|
||||
}
|
||||
return String(random(from,to)).c_str();
|
||||
}
|
||||
else
|
||||
{
|
||||
Console << "Missing '('" << endl;
|
||||
}
|
||||
}
|
||||
while(str[0]==' ') str.erase(0,1);
|
||||
return sword;
|
||||
}
|
||||
|
||||
bool isaddr(string s)
|
||||
{
|
||||
if (s.length()==0 or s.length()>3) return false;
|
||||
for(char c: s)
|
||||
if (c<'0' or c>'9') return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
string getip(string& str, const char* if_empty=nullptr, char sep=' ')
|
||||
{
|
||||
string addr=getword(str, if_empty, sep);
|
||||
string ip=addr;
|
||||
std::vector<string> build;
|
||||
while(ip.length())
|
||||
{
|
||||
string b=getword(ip,nullptr,'.');
|
||||
if (isaddr(b) && build.size()<4)
|
||||
{
|
||||
build.push_back(b);
|
||||
}
|
||||
else
|
||||
return addr;
|
||||
}
|
||||
IPAddress local=WiFi.localIP();
|
||||
addr.clear();
|
||||
while(build.size()!=4)
|
||||
{
|
||||
std::stringstream b;
|
||||
b << (int)local[3-build.size()];
|
||||
build.insert(build.begin(), b.str().c_str());
|
||||
}
|
||||
for(string s: build)
|
||||
{
|
||||
if (addr.length()) addr += '.';
|
||||
addr += s;
|
||||
}
|
||||
Console << "connect address: " << addr << endl;
|
||||
return addr;
|
||||
}
|
||||
|
||||
std::set<string> commands = {
|
||||
"broker", "blink", "client", "connect",
|
||||
"create", "delete", "debug", "help", "interval",
|
||||
"list", "ls", "ip", "off", "on", "set",
|
||||
"publish", "reset", "subscribe", "unsubscribe", "view", "echo", "every"
|
||||
};
|
||||
|
||||
void convertToCommand(string& search)
|
||||
{
|
||||
while(search[0]==' ') search.erase(0,1);
|
||||
if (search.length()==0) return;
|
||||
string matches;
|
||||
int count=0;
|
||||
for(string cmd: commands)
|
||||
{
|
||||
if (cmd.substr(0, search.length()) == search)
|
||||
{
|
||||
if (count) matches +=", ";
|
||||
count++;
|
||||
matches += cmd;
|
||||
}
|
||||
}
|
||||
if (count==1)
|
||||
search = matches;
|
||||
else if (count>1)
|
||||
{
|
||||
Console << "Ambiguous command: " << matches << endl;
|
||||
search.clear();
|
||||
}
|
||||
}
|
||||
|
||||
bool compare(string s, const char* cmd)
|
||||
{
|
||||
uint8_t p=0;
|
||||
while(s[p++]==*cmd++)
|
||||
{
|
||||
if (*cmd==0 or s[p]==0) return true;
|
||||
if (s[p]==' ') return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
using ClientFunction = void(*)(string& cmd, MqttClient* publish);
|
||||
|
||||
struct Every
|
||||
{
|
||||
string cmd;
|
||||
uint32_t ms;
|
||||
uint32_t next;
|
||||
uint32_t underrun=0;
|
||||
bool active=true;
|
||||
|
||||
void dump()
|
||||
{
|
||||
if (active)
|
||||
Console << green << "enabled";
|
||||
else
|
||||
Console << red << "disabled";
|
||||
|
||||
auto mill=millis();
|
||||
Console << white << ms << "ms [" << cmd << "] next in ";
|
||||
if (mill > next)
|
||||
Console << "now";
|
||||
else
|
||||
Console << next-mill << "ms";
|
||||
}
|
||||
};
|
||||
|
||||
uint32_t blink_ms_on[16];
|
||||
uint32_t blink_ms_off[16];
|
||||
uint32_t blink_next[16];
|
||||
bool blink_state[16];
|
||||
int16_t blink;
|
||||
|
||||
std::vector<Every> everies;
|
||||
|
||||
void clientConnect(MqttClient* client, string& cmd)
|
||||
{
|
||||
string remote = getword(cmd);
|
||||
uint16_t port;
|
||||
auto it=list.find(remote);
|
||||
if (it != list.end())
|
||||
{
|
||||
Console << "Connecting to free broker: " << remote << endl;
|
||||
remote = it->second.url;
|
||||
port=it->second.port;
|
||||
}
|
||||
else
|
||||
port=getint(cmd);
|
||||
|
||||
client->connect(remote.c_str(), port, getint(cmd, 60));
|
||||
Console << (client->connected() ? "connected." : "not connected") << endl;
|
||||
}
|
||||
|
||||
void eval(string& cmd)
|
||||
{
|
||||
while(cmd.length())
|
||||
{
|
||||
MqttError retval = MqttOk;
|
||||
|
||||
string s;
|
||||
MqttBroker* broker = nullptr;
|
||||
MqttClient* client = nullptr;
|
||||
|
||||
// client.function notation
|
||||
if (cmd.find('.') != string::npos &&
|
||||
cmd.find('.') < cmd.find(' '))
|
||||
{
|
||||
s=getword(cmd, nullptr, '.');
|
||||
|
||||
if (s.length())
|
||||
{
|
||||
if (clients.find(s) != clients.end())
|
||||
{
|
||||
client = clients[s];
|
||||
}
|
||||
else if (brokers.find(s) != brokers.end())
|
||||
{
|
||||
broker = brokers[s];
|
||||
}
|
||||
else
|
||||
{
|
||||
Console << red << "Unknown class (" << s.c_str() << ")" << white << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
s = getword(cmd);
|
||||
if (s.length()) convertToCommand(s);
|
||||
if (s.length()==0)
|
||||
{}
|
||||
else if (compare(s, "debug"))
|
||||
{
|
||||
#if TINY_MQTT_DEBUG
|
||||
TinyMqtt::debug = getint(cmd);
|
||||
#else
|
||||
Console << red << "TinyMqtt not compiled in debug" << endl;
|
||||
#endif
|
||||
}
|
||||
else if (compare(s, "list"))
|
||||
{
|
||||
Console << "List of free servers" << endl;
|
||||
for(const auto& fb: list)
|
||||
{
|
||||
Console << " " << fb.first << " : " << fb.second.url << ":" << fb.second.port << endl;
|
||||
}
|
||||
}
|
||||
else if (compare(s, "delete"))
|
||||
{
|
||||
if (client==nullptr && broker==nullptr)
|
||||
{
|
||||
s = getword(cmd);
|
||||
if (clients.find(s) != clients.end())
|
||||
{
|
||||
client = clients[s];
|
||||
}
|
||||
else if (brokers.find(s) != brokers.end())
|
||||
{
|
||||
broker = brokers[s];
|
||||
}
|
||||
else
|
||||
Console << red << "Unable to find (" << s.c_str() << ")" << white << endl;
|
||||
}
|
||||
if (client)
|
||||
{
|
||||
for (auto it: clients)
|
||||
{
|
||||
if (it.second != client) continue;
|
||||
Console << "deleted" << endl;
|
||||
delete (it.second);
|
||||
clients.erase(it.first);
|
||||
break;
|
||||
}
|
||||
cmd += " ls";
|
||||
}
|
||||
else if (broker)
|
||||
{
|
||||
for(auto it: brokers)
|
||||
{
|
||||
if (broker != it.second) continue;
|
||||
Console << "deleted" << endl;
|
||||
delete (it.second);
|
||||
brokers.erase(it.first);
|
||||
break;
|
||||
}
|
||||
cmd += " ls";
|
||||
}
|
||||
else
|
||||
Console << "Nothing to delete" << endl;
|
||||
}
|
||||
else if (broker)
|
||||
{
|
||||
if (compare(s,"connect"))
|
||||
{
|
||||
Console << "NYI" << endl;
|
||||
}
|
||||
else if (compare(s, "view"))
|
||||
{
|
||||
broker->dump();
|
||||
}
|
||||
else
|
||||
{
|
||||
Console << "Unknown broker command (" << s << ")" << endl;
|
||||
s.clear();
|
||||
}
|
||||
}
|
||||
else if (client)
|
||||
{
|
||||
if (compare(s,"connect"))
|
||||
{
|
||||
clientConnect(client, cmd);
|
||||
}
|
||||
else if (compare(s,"publish"))
|
||||
{
|
||||
retval = client->publish(getword(cmd), getword(cmd));
|
||||
}
|
||||
else if (compare(s,"subscribe"))
|
||||
{
|
||||
client->subscribe(getword(cmd));
|
||||
}
|
||||
else if (compare(s, "unsubscribe"))
|
||||
{
|
||||
client->unsubscribe(getword(cmd));
|
||||
}
|
||||
else if (compare(s, "view"))
|
||||
{
|
||||
client->dump();
|
||||
}
|
||||
else
|
||||
{
|
||||
Console << "Unknown client command (" << s << ")" << endl;
|
||||
s.clear();
|
||||
}
|
||||
}
|
||||
else if (compare(s, "on"))
|
||||
{
|
||||
uint8_t pin=getint(cmd, 2);
|
||||
pinMode(pin, OUTPUT);
|
||||
digitalWrite(pin, HIGH);
|
||||
}
|
||||
else if (compare(s, "off"))
|
||||
{
|
||||
uint8_t pin=getint(cmd, 2);
|
||||
pinMode(pin, OUTPUT);
|
||||
digitalWrite(pin, LOW);
|
||||
}
|
||||
else if (compare(s, "echo"))
|
||||
{
|
||||
s=getword(cmd);
|
||||
if (s=="on")
|
||||
echo_on = true;
|
||||
else if (s=="off")
|
||||
echo_on = false;
|
||||
else
|
||||
{
|
||||
Console << s << ' ';
|
||||
while(cmd.length())
|
||||
{
|
||||
Console << getword(cmd) << ' ';
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (compare(s, "every"))
|
||||
{
|
||||
uint32_t ms = getint(cmd, 0);
|
||||
if (ms)
|
||||
{
|
||||
if (cmd.length())
|
||||
{
|
||||
Every every;
|
||||
every.ms=ms;
|
||||
every.cmd=cmd;
|
||||
every.next=millis()+ms;
|
||||
everies.push_back(every);
|
||||
every.dump();
|
||||
Console << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else if (compare(cmd, "off") or compare(cmd, "on"))
|
||||
{
|
||||
bool active=getword(cmd)=="on";
|
||||
uint8_t ever=getint(cmd, 100);
|
||||
uint8_t count=0;
|
||||
for(auto& every: everies)
|
||||
{
|
||||
if (count==ever or (ever==100))
|
||||
{
|
||||
if (every.active != active)
|
||||
{
|
||||
Console << "every #" << count << (active ? " on" :" off") << endl;
|
||||
every.active = active;
|
||||
every.underrun = 0;
|
||||
}
|
||||
}
|
||||
count++;
|
||||
}
|
||||
}
|
||||
else if (compare(cmd, "list") or cmd.length()==0)
|
||||
{
|
||||
getword(cmd);
|
||||
Console << "List of everies (ms=" << millis() << ")" << endl;
|
||||
uint8_t count=0;
|
||||
for(auto& every: everies)
|
||||
{
|
||||
Console << count << ": ";
|
||||
every.dump();
|
||||
Console << endl;
|
||||
count++;
|
||||
}
|
||||
}
|
||||
else if (compare(cmd, "remove"))
|
||||
{
|
||||
Console << "Removing..." << endl;
|
||||
getword(cmd);
|
||||
int8_t every=getint(cmd, -1);
|
||||
if (every==-1 and compare(cmd, "last") and everies.size())
|
||||
{
|
||||
getword(cmd);
|
||||
everies.erase(everies.begin()+everies.size()-1);
|
||||
}
|
||||
else if (every==-1 and compare(cmd, "all"))
|
||||
{
|
||||
getword(cmd);
|
||||
everies.clear();
|
||||
}
|
||||
else if (everies.size() > (uint8_t)every)
|
||||
{
|
||||
everies.erase(everies.begin()+every);
|
||||
}
|
||||
else
|
||||
Console << "Bad colmmand" << endl;
|
||||
}
|
||||
else
|
||||
Console << "Bad command" << endl;
|
||||
}
|
||||
else if (compare(s, "blink"))
|
||||
{
|
||||
int8_t blink_nr = getint(cmd, -1);
|
||||
if (blink_nr >= 0)
|
||||
{
|
||||
blink_ms_on[blink_nr]=getint(cmd, blink_ms_on[blink_nr]);
|
||||
blink_ms_off[blink_nr]=getint(cmd, blink_ms_on[blink_nr]);
|
||||
pinMode(blink_nr, OUTPUT);
|
||||
blink_next[blink_nr] = millis();
|
||||
Console << "Blink " << blink_nr << ' ' << (blink_ms_on[blink_nr] ? "on" : "off") << endl;
|
||||
if (blink_ms_on[blink_nr])
|
||||
blink |= 1<< blink_nr;
|
||||
else
|
||||
{
|
||||
blink &= ~(1<< blink_nr);
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (compare(s, "broker"))
|
||||
{
|
||||
string id=getword(cmd);
|
||||
if (clients.find(id) != clients.end())
|
||||
{
|
||||
Console << "A client already have that name" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
else if (id.length() or brokers.find(id)!=brokers.end())
|
||||
{
|
||||
int port=getint(cmd, 0);
|
||||
if (port)
|
||||
{
|
||||
MqttBroker* broker = new MqttBroker(port);
|
||||
broker->begin();
|
||||
|
||||
brokers[id] = broker;
|
||||
Console << "new broker (" << id.c_str() << ")" << endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
Console << "Missing port" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
Console << "Missing or existing broker name (" << id.c_str() << ")" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else if (compare(s, "client"))
|
||||
{
|
||||
string id=getword(cmd);
|
||||
if (brokers.find(id) != brokers.end())
|
||||
{
|
||||
Console << "A broker have that name" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
else if (id.length() or clients.find(id)!=clients.end())
|
||||
{
|
||||
s=getword(cmd); // broker
|
||||
if (s=="" or brokers.find(s) != brokers.end() or list.find(s) != list.end())
|
||||
{
|
||||
MqttBroker* broker = nullptr;
|
||||
if (s.length()) broker = brokers[s];
|
||||
MqttClient* client = new MqttClient(broker, id);
|
||||
clients[id]=client;
|
||||
client->setCallback(onPublish);
|
||||
if (list.find(s) != list.end())
|
||||
{
|
||||
cmd=s+' '+cmd;
|
||||
clientConnect(client, cmd);
|
||||
}
|
||||
Console << "new client (" << id.c_str() << ", " << s.c_str() << ')' << endl;
|
||||
}
|
||||
else if (s.length())
|
||||
{
|
||||
Console << " not found." << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
Console << "Missing or existing client name" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
}
|
||||
else if (compare(s, "set"))
|
||||
{
|
||||
string name(getword(cmd));
|
||||
if (name.length()==0)
|
||||
{
|
||||
for(auto it: vars)
|
||||
{
|
||||
Console << " " << it.first << " -> " << it.second << endl;
|
||||
}
|
||||
}
|
||||
else if (commands.find(name) != commands.end())
|
||||
{
|
||||
Console << "Reserved keyword (" << name << ")" << endl;
|
||||
cmd.clear();
|
||||
}
|
||||
else
|
||||
{
|
||||
if (cmd.length())
|
||||
{
|
||||
vars[name] = cmd;
|
||||
cmd.clear();
|
||||
}
|
||||
else if (vars.find(name) != vars.end())
|
||||
vars.erase(vars.find(name));
|
||||
}
|
||||
}
|
||||
else if (compare(s, "ls") or compare(s, "view"))
|
||||
{
|
||||
bool view = compare(s, "view");
|
||||
if (view)
|
||||
{
|
||||
Console << save_cursor << magenta;
|
||||
Console.gotoxy(1,1);
|
||||
}
|
||||
Console << "--< " << '/' << clients.size() << " client/s. >--" << erase_to_end << endl;
|
||||
for(auto it: clients)
|
||||
{
|
||||
it.second->dump(" ");
|
||||
}
|
||||
|
||||
Console << "--< " << brokers.size() << " brokers/s. >--" << erase_to_end << endl;
|
||||
for(auto it: brokers)
|
||||
{
|
||||
Console << " +-- '" << it.first.c_str() << "' " << it.second->clientsCount() << " client/s."<< erase_to_end << endl;
|
||||
it.second->dump(" ");
|
||||
}
|
||||
if (view)
|
||||
{
|
||||
Console.bg(white);
|
||||
Console << erase_to_end << restore_cursor;
|
||||
}
|
||||
}
|
||||
else if (compare(s, "reset"))
|
||||
ESP.restart();
|
||||
else if (compare(s, "ip"))
|
||||
Console << "IP: " << WiFi.localIP() << endl;
|
||||
else if (compare(s,"help"))
|
||||
{
|
||||
Console << "syntax:" << endl;
|
||||
Console << " MqttBroker:" << endl;
|
||||
Console << " broker {broker_name} {port} : create a new broker" << endl;
|
||||
Console << " broker_name can be one of 'list'" << endl;
|
||||
Console << " broker_name.delete : delete a broker (buggy)" << endl;
|
||||
Console << " broker_name.view : dump a broker" << endl;
|
||||
Console << endl;
|
||||
Console << " MqttClient:" << endl;
|
||||
Console << " client {name} {broker} : create a client then" << endl;
|
||||
Console << " name.connect [ip] [port] [alive]" << endl;
|
||||
Console << " name.[un]subscribe topic" << endl;
|
||||
Console << " name.publish topic [payload]" << endl;
|
||||
Console << " name.view" << endl;
|
||||
Console << " name.delete" << endl;
|
||||
Console << endl;
|
||||
Console << " list : list of free brokers (debug 1 advised)" << endl;
|
||||
Console << " debug #" << endl;
|
||||
Console << " list : get list of free brokers" << endl;
|
||||
Console << " blink [Dx on_ms off_ms] : make pin blink" << endl;
|
||||
Console << " ls / ip / reset" << endl;
|
||||
Console << " set [name][value]" << endl;
|
||||
Console << " ! repeat last command" << endl;
|
||||
Console << endl;
|
||||
Console << " echo [on|off] or strings" << endl;
|
||||
Console << " every ms [command]; every list; every remove [nr|all]; every (on|off) [#]" << endl;
|
||||
Console << " on {output}; off {output}" << endl;
|
||||
Console << " $id : name of the client." << endl;
|
||||
Console << " rnd[(min[,max])] random number." << endl;
|
||||
Console << endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
while(s[0]==' ') s.erase(0,1);
|
||||
if (s.length())
|
||||
Console << "Unknown command (" << s.c_str() << ")" << endl;
|
||||
}
|
||||
|
||||
if (retval != MqttOk)
|
||||
{
|
||||
Console << "# MQTT ERROR " << retval << endl;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void loop()
|
||||
{
|
||||
auto ms=millis();
|
||||
int8_t out=0;
|
||||
int16_t blink_bits = blink;
|
||||
uint8_t e=0;
|
||||
|
||||
for(auto& every: everies)
|
||||
{
|
||||
if (not every.active) continue;
|
||||
if (every.ms && every.cmd.length() && ms > every.next)
|
||||
{
|
||||
string cmd(every.cmd);
|
||||
eval(cmd);
|
||||
every.next += every.ms;
|
||||
if (ms > every.next and ms > every.underrun)
|
||||
{
|
||||
every.next += every.ms;
|
||||
Console << yellow << "Underrun every #" << e << ", " << (ms - every.next) << "ms late" << endl;
|
||||
every.underrun = ms+5000;
|
||||
}
|
||||
}
|
||||
e++;
|
||||
}
|
||||
|
||||
while(blink_bits)
|
||||
{
|
||||
if (blink_ms_on[out] and ms > blink_next[out])
|
||||
{
|
||||
if (blink_state[out])
|
||||
{
|
||||
blink_next[out] += blink_ms_on[out];
|
||||
digitalWrite(out, LOW);
|
||||
}
|
||||
else
|
||||
{
|
||||
blink_next[out] += blink_ms_off[out];
|
||||
digitalWrite(abs(out), HIGH);
|
||||
}
|
||||
blink_state[out] = not blink_state[out];
|
||||
}
|
||||
blink_bits >>=1;
|
||||
out++;
|
||||
}
|
||||
|
||||
#if defined(ESP9266)
|
||||
MDNS.update();
|
||||
#endif
|
||||
|
||||
for(auto it: brokers)
|
||||
it.second->loop();
|
||||
|
||||
for(auto it: clients)
|
||||
it.second->loop();
|
||||
|
||||
Console.loop();
|
||||
}
|
||||
@@ -1,7 +1,7 @@
|
||||
# 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
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
|
||||
@@ -1,7 +1,10 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
include ../Makefile.opts
|
||||
EXTRA_CXXFLAGS=-g3 -O0 -DTINY_MQTT_DEFAULT_ALIVE=1 -DEPOXY_TEST
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
|
||||
APP_NAME := local-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
|
||||
@@ -12,11 +12,10 @@
|
||||
* Also, this will allow to mock and thus run Action on github
|
||||
**/
|
||||
|
||||
using string = TinyConsole::string;
|
||||
using namespace std;
|
||||
|
||||
MqttBroker broker(1883);
|
||||
|
||||
std::map<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;
|
||||
size_t lastLength;
|
||||
@@ -29,27 +28,36 @@ void onPublish(const MqttClient* srce, const Topic& topic, const char* payload,
|
||||
lastLength = length;
|
||||
}
|
||||
|
||||
test(local_not_connected_by_default)
|
||||
{
|
||||
MqttClient client;
|
||||
assertEqual(client.connected(), false);
|
||||
|
||||
MqttBroker broker(1883);
|
||||
assertEqual(broker.connected(), false);
|
||||
}
|
||||
|
||||
test(local_client_should_unregister_when_destroyed)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
MqttBroker broker(1883);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0); // Ensure client is not yet connected
|
||||
MqttClient client(&broker);
|
||||
assertEqual(broker.clientsCount(), (size_t)1); // Ensure client is now connected
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is now connected
|
||||
}
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
}
|
||||
|
||||
test(local_client_do_not_disconnect_after_publishing_and_long_inactivity)
|
||||
test(local_client_alive)
|
||||
{
|
||||
EpoxyTest::set_millis(0);
|
||||
MqttBroker broker(1883);
|
||||
MqttClient client(&broker);
|
||||
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is now connected
|
||||
|
||||
EpoxyTest::add_millis(TINY_MQTT_DEFAULT_ALIVE*1000/2);
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is still connected
|
||||
|
||||
EpoxyTest::add_seconds(TINY_MQTT_DEFAULT_ALIVE*5);
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is still connected
|
||||
}
|
||||
|
||||
test(local_wildcard_subscribe)
|
||||
{
|
||||
EpoxyTest::set_millis(0);
|
||||
MqttBroker broker(1883);
|
||||
@@ -60,82 +68,58 @@ test(local_client_do_not_disconnect_after_publishing_and_long_inactivity)
|
||||
client.subscribe("#");
|
||||
client.subscribe("test");
|
||||
client.setCallback(onPublish);
|
||||
assertEqual(broker.clientsCount(), (size_t)2);
|
||||
assertEqual(broker.localClientsCount(), (size_t)2);
|
||||
|
||||
sender.publish("test", "value");
|
||||
broker.loop();
|
||||
|
||||
EpoxyTest::add_seconds(600);
|
||||
assertEqual(published.size(), (size_t)1); // client has received something
|
||||
}
|
||||
|
||||
test(local_client_do_not_disconnect_after_publishing)
|
||||
{
|
||||
EpoxyTest::set_millis(0);
|
||||
MqttBroker broker(1883);
|
||||
MqttClient client(&broker, "client");
|
||||
MqttClient sender(&broker, "sender");
|
||||
broker.loop();
|
||||
|
||||
client.subscribe("#");
|
||||
client.subscribe("test");
|
||||
client.setCallback(onPublish);
|
||||
assertEqual(broker.localClientsCount(), (size_t)2);
|
||||
|
||||
sender.publish("test", "value");
|
||||
broker.loop();
|
||||
|
||||
EpoxyTest::add_seconds(60);
|
||||
client.loop();
|
||||
sender.loop();
|
||||
broker.loop();
|
||||
|
||||
sender.publish("test", "value");
|
||||
broker.loop();
|
||||
sender.loop();
|
||||
broker.loop();
|
||||
|
||||
assertEqual(broker.clientsCount(), (size_t)2);
|
||||
assertEqual(broker.localClientsCount(), (size_t)2);
|
||||
assertEqual(sender.connected(), true);
|
||||
assertEqual(client.connected(), true);
|
||||
|
||||
assertEqual(published.size(), (size_t)1); // client has received something
|
||||
}
|
||||
|
||||
test(local_connect)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient client(&broker);
|
||||
assertTrue(client.connected());
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
}
|
||||
|
||||
test(local_publish_to_nowhere)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient publisher;
|
||||
MqttError status = publisher.publish("a/b");
|
||||
assertEqual(status, MqttError::MqttNowhereToSend);
|
||||
}
|
||||
|
||||
test(local_publish_should_be_dispatched)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "");
|
||||
subscriber.subscribe("a/b");
|
||||
subscriber.subscribe("a/c");
|
||||
subscriber.setCallback(onPublish);
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("a/b");
|
||||
publisher.publish("a/c");
|
||||
publisher.publish("a/c");
|
||||
|
||||
assertEqual(published.size(), (size_t)1); // 1 client has received something
|
||||
assertEqual(published[""]["a/b"], 1);
|
||||
assertEqual(published[""]["a/c"], 2);
|
||||
}
|
||||
|
||||
#if 0
|
||||
test(local_publish_should_be_dispatched_to_local_clients)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber_a(&broker, "A");
|
||||
MqttClient subscriber_a("A");
|
||||
subscriber_a.setCallback(onPublish);
|
||||
subscriber_a.subscribe("a/b");
|
||||
subscriber_a.subscribe("a/c");
|
||||
|
||||
MqttClient subscriber_b(&broker, "B");
|
||||
MqttClient subscriber_b("B");
|
||||
subscriber_b.setCallback(onPublish);
|
||||
subscriber_b.subscribe("a/b");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
MqttClient publisher;
|
||||
publisher.publish("a/b");
|
||||
publisher.publish("a/c");
|
||||
|
||||
@@ -149,13 +133,13 @@ test(local_publish_should_be_dispatched_to_local_clients)
|
||||
test(local_unsubscribe)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "");
|
||||
MqttClient subscriber;
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("a/b");
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
MqttClient publisher;
|
||||
publisher.publish("a/b");
|
||||
|
||||
subscriber.unsubscribe("a/b");
|
||||
@@ -169,11 +153,11 @@ test(local_unsubscribe)
|
||||
test(local_nocallback_when_destroyed)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
MqttClient publisher;
|
||||
{
|
||||
MqttClient subscriber(&broker);
|
||||
MqttClient subscriber;
|
||||
subscriber.setCallback(onPublish);
|
||||
subscriber.subscribe("a/b");
|
||||
publisher.publish("a/b");
|
||||
@@ -183,6 +167,7 @@ test(local_nocallback_when_destroyed)
|
||||
|
||||
assertEqual(published.size(), (size_t)1); // Only one publish has been received
|
||||
}
|
||||
#endif
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// setup() and loop()
|
||||
|
||||
@@ -1,10 +1,10 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
include ../Makefile.opts
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
# CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics -DEPOXY_TEST
|
||||
|
||||
APP_NAME := network-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
|
||||
@@ -9,6 +9,16 @@
|
||||
#include <string>
|
||||
#include <iostream>
|
||||
|
||||
uint32_t getClientKeepAlive(MqttBroker& broker)
|
||||
{
|
||||
if (broker.getClients().size() == 1)
|
||||
for (auto& it : broker.getClients())
|
||||
return it->keepAlive();
|
||||
|
||||
return 9999;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* TinyMqtt network unit tests.
|
||||
*
|
||||
@@ -16,8 +26,6 @@
|
||||
* Checks with a local broker. Clients must connect to the local broker
|
||||
**/
|
||||
|
||||
using string = TinyConsole::string;
|
||||
|
||||
// if ascii_pos = 0, no ascii dump, else ascii dump starts after column ascii_pos
|
||||
std::string bufferToHexa(const uint8_t* buffer, size_t length, char sep = 0, size_t ascii_pos = 0)
|
||||
{
|
||||
@@ -80,7 +88,7 @@ String toString(const IPAddress& ip)
|
||||
|
||||
MqttBroker broker(1883);
|
||||
|
||||
std::map<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;
|
||||
size_t lastLength;
|
||||
@@ -144,6 +152,46 @@ test(suback)
|
||||
assertEqual(MqttClient::counters[MqttMessage::Type::SubAck], 1);
|
||||
}
|
||||
|
||||
test(network_client_alive)
|
||||
{
|
||||
const uint32_t keep_alive=1;
|
||||
start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
EpoxyTest::set_millis(0); // Enter simulated time
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress broker_ip = WiFi.localIP();
|
||||
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(broker_ip.toString().c_str(), 1883, keep_alive);
|
||||
broker.loop();
|
||||
client.loop();
|
||||
|
||||
assertTrue(broker.clientsCount() == 1);
|
||||
assertTrue(client.connected());
|
||||
|
||||
uint32_t ka = getClientKeepAlive(broker);
|
||||
assertEqual(ka, keep_alive);
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
|
||||
// All is going well if we call client.loop()
|
||||
// The client is able to send PingReq to the broker
|
||||
EpoxyTest::add_seconds(keep_alive);
|
||||
client.loop();
|
||||
broker.loop();
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
|
||||
// Now simulate that the client is frozen for
|
||||
// a too long time
|
||||
EpoxyTest::add_seconds(TINY_MQTT_CLIENT_ALIVE_TOLERANCE*2);
|
||||
broker.loop();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
EpoxyTest::set_real_time();
|
||||
}
|
||||
|
||||
test(network_client_keep_alive_high)
|
||||
{
|
||||
const uint32_t keep_alive=1000;
|
||||
@@ -174,7 +222,7 @@ test(network_client_keep_alive_high)
|
||||
uint32_t sz = broker.getClients().size();
|
||||
assertEqual(sz , (uint32_t)1);
|
||||
|
||||
uint32_t ka = broker.getClients()[0]->keepAlive();
|
||||
uint32_t ka = getClientKeepAlive(broker);
|
||||
assertEqual(ka, keep_alive);
|
||||
|
||||
}
|
||||
@@ -264,14 +312,14 @@ test(network_one_client_one_broker_hudge_publish_and_subscribe_through_network)
|
||||
assertEqual((unsigned int)lastLength, (unsigned int)sent.size());
|
||||
}
|
||||
|
||||
test(network_client_should_unregister_when_destroyed)
|
||||
test(network_local_client_should_unregister_when_destroyed)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
{
|
||||
MqttClient client(&broker);
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
assertEqual(broker.localClientsCount(), (size_t)1);
|
||||
}
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
}
|
||||
|
||||
|
||||
@@ -284,13 +332,13 @@ test(network_connect)
|
||||
|
||||
MqttClient client(&broker);
|
||||
assertTrue(client.connected());
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
assertEqual(broker.localClientsCount(), (size_t)1);
|
||||
}
|
||||
|
||||
test(network_publish_should_be_dispatched)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.subscribe("a/b");
|
||||
@@ -391,11 +439,12 @@ test(network_small_payload)
|
||||
|
||||
test(network_hudge_payload)
|
||||
{
|
||||
const char* payload="This payload is hudge, just because its length exceeds 127. Thus when encoding length, we have to encode it on two bytes at min. This should not prevent the message from being encoded and decoded successfully !";
|
||||
// const char* payload="This payload is hudge, just because its length exceeds 127. Thus when encoding length, we have to encode it on two bytes at min. This should not prevent the message from being encoded and decoded successfully !";
|
||||
const char* payload="This was decoded successfully !";
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
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) TODO
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
publisher.publish("a/b", payload); // This publish is received
|
||||
@@ -404,25 +453,13 @@ test(network_hudge_payload)
|
||||
assertEqual(payload, lastPayload);
|
||||
assertEqual(lastLength, strlen(payload));
|
||||
assertEqual(strcmp(payload, lastPayload), 0);
|
||||
}
|
||||
|
||||
test(disconnected_when_broker_is_deleted)
|
||||
{
|
||||
MqttBroker* broker = new MqttBroker(1883);
|
||||
broker->begin();
|
||||
|
||||
MqttClient client;
|
||||
client.connect(broker);
|
||||
assertEqual(client.connected(), true);
|
||||
client.publish("a", "b");
|
||||
|
||||
delete broker;
|
||||
assertEqual(client.connected(), false);
|
||||
std::cout << "payload : " << payload << std::endl;
|
||||
std::cout << "received: " << lastPayload << std::endl;
|
||||
}
|
||||
|
||||
test(connack)
|
||||
{
|
||||
const bool view = false;
|
||||
const bool view = true;
|
||||
|
||||
NetworkObserver check(
|
||||
[this](const WiFiClient*, const uint8_t* buffer, size_t length)
|
||||
@@ -479,5 +516,4 @@ void loop() {
|
||||
aunit::TestRunner::run();
|
||||
|
||||
if (Serial.available()) ESP.reset();
|
||||
published.clear(); // Avoid crash in unit tests due to exit handlers
|
||||
}
|
||||
|
||||
@@ -1,7 +1,10 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
include ../Makefile.opts
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
# Remove flto flag from EpoxyDuino (too many <optimized out>)
|
||||
CXXFLAGS = -Wextra -Wall -std=gnu++11 -fno-exceptions -fno-threadsafe-statics
|
||||
|
||||
APP_NAME := nowifi-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
|
||||
@@ -11,11 +11,11 @@
|
||||
* Checks with a local broker. Clients must connect to the local broker
|
||||
**/
|
||||
|
||||
using string = TinyConsole::string;
|
||||
using namespace std;
|
||||
|
||||
MqttBroker broker(1883);
|
||||
|
||||
std::map<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;
|
||||
size_t lastLength;
|
||||
@@ -32,27 +32,27 @@ void onPublish(const MqttClient* srce, const Topic& topic, const char* payload,
|
||||
|
||||
test(nowifi_client_should_unregister_when_destroyed)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
{
|
||||
MqttClient client(&broker);
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
assertEqual(broker.localClientsCount(), (size_t)1);
|
||||
}
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
}
|
||||
|
||||
test(nowifi_connect)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient client(&broker);
|
||||
assertTrue(client.connected());
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
assertEqual(broker.localClientsCount(), (size_t)1);
|
||||
}
|
||||
|
||||
test(nowifi_publish_should_be_dispatched)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.subscribe("a/b");
|
||||
@@ -72,7 +72,7 @@ test(nowifi_publish_should_be_dispatched)
|
||||
test(nowifi_publish_should_be_dispatched_to_clients)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber_a(&broker, "A");
|
||||
subscriber_a.setCallback(onPublish);
|
||||
@@ -97,7 +97,7 @@ test(nowifi_publish_should_be_dispatched_to_clients)
|
||||
test(nowifi_subscribe_with_star_wildcard)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
@@ -118,7 +118,7 @@ test(nowifi_subscribe_with_star_wildcard)
|
||||
test(nowifi_subscribe_with_plus_wildcard)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
@@ -139,7 +139,7 @@ test(nowifi_subscribe_with_plus_wildcard)
|
||||
test(nowifi_should_not_receive_sys_msg)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
@@ -154,7 +154,7 @@ test(nowifi_should_not_receive_sys_msg)
|
||||
test(nowifi_subscribe_with_mixed_wildcards)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
@@ -173,7 +173,7 @@ test(nowifi_subscribe_with_mixed_wildcards)
|
||||
test(nowifi_unsubscribe_with_wildcards)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "A");
|
||||
subscriber.setCallback(onPublish);
|
||||
@@ -195,7 +195,7 @@ test(nowifi_unsubscribe_with_wildcards)
|
||||
test(nowifi_unsubscribe)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker);
|
||||
subscriber.setCallback(onPublish);
|
||||
@@ -215,7 +215,7 @@ test(nowifi_unsubscribe)
|
||||
test(nowifi_nocallback_when_destroyed)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
|
||||
@@ -279,5 +279,4 @@ void loop() {
|
||||
aunit::TestRunner::run();
|
||||
|
||||
if (Serial.available()) ESP.reset();
|
||||
published.clear(); // Avoid crash at exit handlers
|
||||
}
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
include ../Makefile.opts
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
APP_NAME := string-indexer-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsync TinyConsole
|
||||
|
||||
@@ -9,7 +9,7 @@
|
||||
*
|
||||
**/
|
||||
|
||||
using string = TinyConsole::string;
|
||||
using namespace std;
|
||||
|
||||
test(indexer_empty)
|
||||
{
|
||||
@@ -84,7 +84,7 @@ test(indexer_indexed_operator_eq)
|
||||
|
||||
test(indexer_get_string)
|
||||
{
|
||||
string sone("one");
|
||||
std::string sone("one");
|
||||
IndexedString one(sone);
|
||||
|
||||
assertTrue(sone==one.str());
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
# See https://github.com/bxparks/EpoxyDuino for documentation about this
|
||||
# Makefile to compile and run Arduino programs natively on Linux or MacOS.
|
||||
|
||||
include ../Makefile.opts
|
||||
EXTRA_CXXFLAGS=-g3 -O0
|
||||
|
||||
APP_NAME := topic-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsync TinyConsole
|
||||
|
||||
@@ -5,22 +5,26 @@
|
||||
#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));
|
||||
std::cout << " " << ta.c_str() << ' ';
|
||||
cout << " " << ta.c_str() << ' ';
|
||||
if (match != expected)
|
||||
std::cout << (expected ? " should match " : " should not match ");
|
||||
cout << (expected ? " should match " : " should not match ");
|
||||
else
|
||||
std::cout << (expected ? " matches " : " unmatches ");
|
||||
std::cout << tb.c_str() << std::endl;
|
||||
cout << (expected ? " matches " : " unmatches ");
|
||||
cout << tb.c_str() << endl;
|
||||
return expected == match;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user