Compare commits
13 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4746abfc96 | ||
|
|
ef24107169 | ||
|
|
295e1a06d3 | ||
|
|
1a66d2c991 | ||
|
|
8162b4c35b | ||
|
|
9e578471f1 | ||
|
|
efa94cc4a4 | ||
|
|
040568b478 | ||
|
|
a96b51d7f4 | ||
|
|
7ad63551d6 | ||
|
|
4627bcd109 | ||
|
|
f0af2b95e3 | ||
|
|
f348d82167 |
4
.github/workflows/aunit.yml
vendored
4
.github/workflows/aunit.yml
vendored
@@ -18,9 +18,9 @@ jobs:
|
||||
run: |
|
||||
cd ..
|
||||
git clone https://github.com/hsaturn/TinyConsole
|
||||
git clone https://github.com/hsaturn/EpoxyDuino
|
||||
git clone https://github.com/bxparks/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
|
||||
|
||||
13
README.md
13
README.md
@@ -10,21 +10,8 @@
|
||||
|
||||
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
|
||||
|
||||
- Async Wifi compatible (me-no-dev/ESPAsyncTCP@^1.2.2)
|
||||
- Very fast broker I saw it re-sent 1000 topics per second for two
|
||||
clients that had subscribed (payload ~15 bytes ESP8266). No topic lost.
|
||||
The max I've seen was 2k msg/s (1 client 1 subscription)
|
||||
|
||||
@@ -1,61 +0,0 @@
|
||||
#!/bin/bash
|
||||
current_version=$(git describe --tags --abbrev=0)
|
||||
if [ "$1" == "-d" ]; then
|
||||
do=0
|
||||
shift
|
||||
else
|
||||
do=1
|
||||
fi
|
||||
if [ "$1" == "" ]; then
|
||||
echo
|
||||
echo "Syntax: $0 [-d] {new_version}"
|
||||
echo
|
||||
echo " -d : dry run, generate json and update properties but do not run git commands"
|
||||
echo ""
|
||||
echo " Current version: $current_version"
|
||||
echo
|
||||
else
|
||||
echo "Current version: ($current_version)"
|
||||
echo "New version : ($1)"
|
||||
echo -n "Do you want to proceed ? "
|
||||
read a
|
||||
if [ "$a" == "y" ]; then
|
||||
echo "Doing this..."
|
||||
grep $current_version library.properties
|
||||
if [ "$?" == "0" ]; then
|
||||
sed -i "s/$current_version/$1/" library.properties
|
||||
|
||||
cp library.json.skeleton library.json
|
||||
while ifs= read -r line; do
|
||||
name=$(echo "$line" | sed "s/=.*//g")
|
||||
value=$(echo "$line" | cut -d= -f 2 | sed 's/"//g')
|
||||
echo " Replacing $name in json"
|
||||
if [ "$name" == "depends" ]; then
|
||||
depends=$(echo "$value" | sed "s/,/ /g")
|
||||
echo " Depends=$depends"
|
||||
fi
|
||||
echo " " sed -i "s@#$name@$value@g" library.json
|
||||
sed -i "s@#$name@$value@g" library.json
|
||||
done < library.properties
|
||||
deps=""
|
||||
for depend in $depends; do
|
||||
if [ "$deps" != "" ]; then
|
||||
deps="$deps, "
|
||||
fi
|
||||
deps="$deps'$depend' : '*'"
|
||||
done
|
||||
sed -i "s@#dependencies@$deps@g" library.json
|
||||
sed -i "s/'/\"/g" library.json
|
||||
if [ "$do" == "1" ]; then
|
||||
git tag $1
|
||||
git add library.properties
|
||||
git add library.json
|
||||
git commit -m "Release $1"
|
||||
git push
|
||||
git push --tags
|
||||
fi
|
||||
else
|
||||
echo "Current version does not match library.property version, aborting"
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
@@ -1,14 +1,14 @@
|
||||
// vim: ts=2 sw=2 expandtab smartindent
|
||||
#include <TinyConsole.h>
|
||||
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
||||
#include <TinyMqtt.h> // https://github.com/hsaturn/TinyMqtt
|
||||
#include <TinyStreaming.h>
|
||||
#if defined(ESP8266)
|
||||
#include <ESP8266mDNS.h>
|
||||
#include <ESP8266mDNS.h>
|
||||
#elif defined(ESP32)
|
||||
#include <WiFi.h>
|
||||
#include <ESPmDNS.h>
|
||||
#include <WiFi.h>
|
||||
#include <ESPmDNS.h>
|
||||
#else
|
||||
#error Unsupported platform
|
||||
#error Unsupported platform
|
||||
#endif
|
||||
|
||||
#include <sstream>
|
||||
@@ -26,47 +26,45 @@ auto save_cursor = TinyConsole::save_cursor;
|
||||
auto restore_cursor = TinyConsole::restore_cursor;
|
||||
auto erase_to_end = TinyConsole::erase_to_end;
|
||||
|
||||
const char *ssid = "Freebox-786A2F";
|
||||
const char *password = "usurpavi8dalum64lumine?";
|
||||
|
||||
void onCommand(const string &command);
|
||||
void eval(string &cmd);
|
||||
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) {}
|
||||
public:
|
||||
free_broker(const char* s, uint16_t p, const char* comment) : url(s), port(p) {}
|
||||
|
||||
string url;
|
||||
uint16_t port;
|
||||
std::string url;
|
||||
uint16_t port;
|
||||
};
|
||||
|
||||
const std::map<string, free_broker> list =
|
||||
{
|
||||
{"mqtthq", {"public.mqtthq.com", 8083, "publish/subscribe"}},
|
||||
{"hivemq", {"broker.hivemq.com", 1883, ""}}};
|
||||
const std::map<std::string, free_broker> list =
|
||||
{
|
||||
{ "mqtthq", { "public.mqtthq.com" , 8083, "publish/subscribe" }},
|
||||
{ "hivemq", { "broker.hivemq.com", 1883, "" }}
|
||||
};
|
||||
|
||||
/** Very complex example
|
||||
* Console allowing to make any kind of test,
|
||||
* 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
|
||||
*/
|
||||
* 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)
|
||||
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--)
|
||||
while(length--)
|
||||
{
|
||||
const char c = *payload++;
|
||||
if (c < 32)
|
||||
const char c=*payload++;
|
||||
if (c<32)
|
||||
Console << '?';
|
||||
else
|
||||
Console << c;
|
||||
@@ -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()
|
||||
{
|
||||
@@ -88,8 +86,7 @@ void setup()
|
||||
delay(500);
|
||||
|
||||
Console.cls();
|
||||
Console << endl
|
||||
<< endl;
|
||||
Console << endl << endl;
|
||||
Console << yellow
|
||||
<< "***************************************************************" << endl;
|
||||
Console << "* Welcome to the TinyMqtt console" << endl;
|
||||
@@ -97,11 +94,10 @@ void setup()
|
||||
Console << "* Enter help to view the list of commands." << endl;
|
||||
Console << "***************************************************************" << endl;
|
||||
Console << endl;
|
||||
if (strlen(ssid) == 0)
|
||||
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;
|
||||
<< " to be able to connect to your Wifi network." << endl;
|
||||
Console << endl << white;
|
||||
|
||||
Console << "Connecting to '" << ssid << "' ";
|
||||
|
||||
@@ -111,234 +107,215 @@ void setup()
|
||||
WiFi.begin(ssid, password);
|
||||
|
||||
while (WiFi.status() != WL_CONNECTED)
|
||||
{
|
||||
Console << '-';
|
||||
delay(500);
|
||||
}
|
||||
{ Console << '-'; delay(500); }
|
||||
|
||||
Console << endl
|
||||
<< "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
Console << endl << "Connected to " << ssid << "IP address: " << WiFi.localIP() << endl;
|
||||
|
||||
const char *name = "tinytest";
|
||||
const char* name="tinytest";
|
||||
Console << "Starting MDNS, name= " << name;
|
||||
if (!MDNS.begin(name))
|
||||
Console << " error, not available." << endl;
|
||||
else
|
||||
Console << " ok." << endl;
|
||||
|
||||
MqttBroker *broker = new MqttBroker(1883);
|
||||
|
||||
MqttBroker* broker = new MqttBroker(1883);
|
||||
broker->begin();
|
||||
brokers["broker"] = broker;
|
||||
|
||||
if (Console.isTerm())
|
||||
onCommand("every 333 view");
|
||||
|
||||
Console.prompt();
|
||||
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());
|
||||
while (isdigit(sword[0]) or sword[0] == ' ')
|
||||
sword.erase(0, 1);
|
||||
if (sword.length())
|
||||
str = sword + ' ' + str;
|
||||
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;
|
||||
str=str2;
|
||||
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;
|
||||
while (str.length() and (str[0] != sep or quote))
|
||||
char quote=(str[0]=='"' or str[0]=='\'' ? str[0] : 0);
|
||||
if (quote) str.erase(0,1);
|
||||
std::string sword;
|
||||
while(str.length() and (str[0]!=sep or quote))
|
||||
{
|
||||
if (str[0] == quote)
|
||||
if (str[0]==quote)
|
||||
{
|
||||
str.erase(0, 1);
|
||||
str.erase(0,1);
|
||||
break;
|
||||
}
|
||||
sword += str[0];
|
||||
str.erase(0, 1);
|
||||
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")
|
||||
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] == '(')
|
||||
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);
|
||||
int from=getint(sword);
|
||||
if (sword[0]==',')
|
||||
{
|
||||
sword.erase(0, 1);
|
||||
sword.erase(0,1);
|
||||
to = getint(sword);
|
||||
if (sword[0] != ')')
|
||||
Console << "Missing ')'" << endl;
|
||||
if (sword[0]!=')') Console << "Missing ')'" << endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
to = from;
|
||||
from = 0;
|
||||
to=from;
|
||||
from=0;
|
||||
}
|
||||
return String(random(from, to)).c_str();
|
||||
return String(random(from,to)).c_str();
|
||||
}
|
||||
else
|
||||
{
|
||||
Console << "Missing '('" << endl;
|
||||
}
|
||||
}
|
||||
while (str[0] == ' ')
|
||||
str.erase(0, 1);
|
||||
while(str[0]==' ') str.erase(0,1);
|
||||
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)
|
||||
if (c < '0' or c > '9')
|
||||
return false;
|
||||
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 = ' ')
|
||||
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;
|
||||
while (ip.length())
|
||||
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, '.');
|
||||
if (isaddr(b) && build.size() < 4)
|
||||
std::string b=getword(ip,nullptr,'.');
|
||||
if (isaddr(b) && build.size()<4)
|
||||
{
|
||||
build.push_back(b);
|
||||
}
|
||||
else
|
||||
return addr;
|
||||
}
|
||||
IPAddress local = WiFi.localIP();
|
||||
IPAddress local=WiFi.localIP();
|
||||
addr.clear();
|
||||
while (build.size() != 4)
|
||||
while(build.size()!=4)
|
||||
{
|
||||
std::stringstream b;
|
||||
b << (int)local[3 - build.size()];
|
||||
build.insert(build.begin(), b.str().c_str());
|
||||
b << (int)local[3-build.size()];
|
||||
build.insert(build.begin(), b.str());
|
||||
}
|
||||
for (string s : build)
|
||||
for(std::string s: build)
|
||||
{
|
||||
if (addr.length())
|
||||
addr += '.';
|
||||
if (addr.length()) addr += '.';
|
||||
addr += s;
|
||||
}
|
||||
Console << "connect address: " << addr << endl;
|
||||
return addr;
|
||||
}
|
||||
|
||||
std::map<string, string> vars;
|
||||
std::map<std::string, std::string> vars;
|
||||
|
||||
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"};
|
||||
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;
|
||||
int count = 0;
|
||||
for (string cmd : commands)
|
||||
while(search[0]==' ') search.erase(0,1);
|
||||
if (search.length()==0) return;
|
||||
std::string matches;
|
||||
int count=0;
|
||||
for(std::string cmd: commands)
|
||||
{
|
||||
if (cmd.substr(0, search.length()) == search)
|
||||
{
|
||||
if (count)
|
||||
matches += ", ";
|
||||
if (count) matches +=", ";
|
||||
count++;
|
||||
matches += cmd;
|
||||
}
|
||||
}
|
||||
if (count == 1)
|
||||
if (count==1)
|
||||
search = matches;
|
||||
else if (count > 1)
|
||||
else if (count>1)
|
||||
{
|
||||
Console << "Ambiguous command: " << matches << endl;
|
||||
search.clear();
|
||||
}
|
||||
}
|
||||
|
||||
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])
|
||||
{
|
||||
srch = d[0] + srch + d[1];
|
||||
to = d[0] + to + 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)
|
||||
while((pos=str.find(srch, pos)) != std::string::npos)
|
||||
{
|
||||
str.erase(pos, srch.length());
|
||||
str.insert(pos, to);
|
||||
pos += to.length() - 1;
|
||||
pos += to.length()-1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void replaceVars(string &cmd)
|
||||
void replaceVars(std::string& cmd)
|
||||
{
|
||||
cmd = ' ' + cmd + ' ';
|
||||
cmd = ' '+cmd+' ';
|
||||
|
||||
for (auto it : vars)
|
||||
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);
|
||||
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++)
|
||||
uint8_t p=0;
|
||||
while(s[p++]==*cmd++)
|
||||
{
|
||||
if (*cmd == 0 or s[p] == 0)
|
||||
return true;
|
||||
if (s[p] == ' ')
|
||||
return true;
|
||||
if (*cmd==0 or s[p]==0) return true;
|
||||
if (s[p]==' ') return true;
|
||||
}
|
||||
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;
|
||||
bool active = true;
|
||||
uint32_t underrun=0;
|
||||
bool active=true;
|
||||
|
||||
void dump()
|
||||
{
|
||||
@@ -347,12 +324,12 @@ struct Every
|
||||
else
|
||||
Console << red << "disabled";
|
||||
|
||||
auto mill = millis();
|
||||
auto mill=millis();
|
||||
Console << white << ms << "ms [" << cmd << "] next in ";
|
||||
if (mill > next)
|
||||
Console << "now";
|
||||
else
|
||||
Console << next - mill << "ms";
|
||||
Console << next-mill << "ms";
|
||||
}
|
||||
};
|
||||
|
||||
@@ -364,49 +341,49 @@ int16_t blink;
|
||||
|
||||
std::vector<Every> everies;
|
||||
|
||||
void onCommand(const string &command)
|
||||
void onCommand(const std::string& command)
|
||||
{
|
||||
Console << endl;
|
||||
string cmd = command;
|
||||
if (cmd.substr(0, 3) != "set")
|
||||
replaceVars(cmd);
|
||||
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);
|
||||
auto it=list.find(remote);
|
||||
if (it != list.end())
|
||||
{
|
||||
Console << "Connecting to free broker: " << remote << endl;
|
||||
remote = it->second.url;
|
||||
port = it->second.port;
|
||||
port=it->second.port;
|
||||
}
|
||||
else
|
||||
port = getint(cmd);
|
||||
port=getint(cmd);
|
||||
|
||||
client->connect(remote.c_str(), port, getint(cmd, 60));
|
||||
Console << (client->connected() ? "connected." : "not connected") << endl;
|
||||
}
|
||||
|
||||
void eval(string &cmd)
|
||||
void eval(std::string& cmd)
|
||||
{
|
||||
while (cmd.length())
|
||||
while(cmd.length())
|
||||
{
|
||||
MqttError retval = MqttOk;
|
||||
|
||||
string s;
|
||||
MqttBroker *broker = nullptr;
|
||||
MqttClient *client = nullptr;
|
||||
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, '.');
|
||||
s=getword(cmd, nullptr, '.');
|
||||
|
||||
if (s.length())
|
||||
{
|
||||
@@ -427,11 +404,9 @@ void eval(string &cmd)
|
||||
}
|
||||
|
||||
s = getword(cmd);
|
||||
if (s.length())
|
||||
convertToCommand(s);
|
||||
if (s.length() == 0)
|
||||
{
|
||||
}
|
||||
if (s.length()) convertToCommand(s);
|
||||
if (s.length()==0)
|
||||
{}
|
||||
else if (compare(s, "debug"))
|
||||
{
|
||||
#if TINY_MQTT_DEBUG
|
||||
@@ -443,14 +418,14 @@ void eval(string &cmd)
|
||||
else if (compare(s, "list"))
|
||||
{
|
||||
Console << "List of free servers" << endl;
|
||||
for (const auto &fb : list)
|
||||
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)
|
||||
if (client==nullptr && broker==nullptr)
|
||||
{
|
||||
s = getword(cmd);
|
||||
if (clients.find(s) != clients.end())
|
||||
@@ -466,10 +441,9 @@ void eval(string &cmd)
|
||||
}
|
||||
if (client)
|
||||
{
|
||||
for (auto it : clients)
|
||||
for (auto it: clients)
|
||||
{
|
||||
if (it.second != client)
|
||||
continue;
|
||||
if (it.second != client) continue;
|
||||
Console << "deleted" << endl;
|
||||
delete (it.second);
|
||||
clients.erase(it.first);
|
||||
@@ -479,10 +453,9 @@ void eval(string &cmd)
|
||||
}
|
||||
else if (broker)
|
||||
{
|
||||
for (auto it : brokers)
|
||||
for(auto it: brokers)
|
||||
{
|
||||
if (broker != it.second)
|
||||
continue;
|
||||
if (broker != it.second) continue;
|
||||
Console << "deleted" << endl;
|
||||
delete (it.second);
|
||||
brokers.erase(it.first);
|
||||
@@ -495,7 +468,7 @@ void eval(string &cmd)
|
||||
}
|
||||
else if (broker)
|
||||
{
|
||||
if (compare(s, "connect"))
|
||||
if (compare(s,"connect"))
|
||||
{
|
||||
Console << "NYI" << endl;
|
||||
}
|
||||
@@ -511,15 +484,15 @@ void eval(string &cmd)
|
||||
}
|
||||
else if (client)
|
||||
{
|
||||
if (compare(s, "connect"))
|
||||
if (compare(s,"connect"))
|
||||
{
|
||||
clientConnect(client, cmd);
|
||||
}
|
||||
else if (compare(s, "publish"))
|
||||
else if (compare(s,"publish"))
|
||||
{
|
||||
retval = client->publish(getword(cmd), getword(cmd));
|
||||
}
|
||||
else if (compare(s, "subscribe"))
|
||||
else if (compare(s,"subscribe"))
|
||||
{
|
||||
client->subscribe(getword(cmd));
|
||||
}
|
||||
@@ -539,27 +512,27 @@ void eval(string &cmd)
|
||||
}
|
||||
else if (compare(s, "on"))
|
||||
{
|
||||
uint8_t pin = getint(cmd, 2);
|
||||
uint8_t pin=getint(cmd, 2);
|
||||
pinMode(pin, OUTPUT);
|
||||
digitalWrite(pin, HIGH);
|
||||
}
|
||||
else if (compare(s, "off"))
|
||||
{
|
||||
uint8_t pin = getint(cmd, 2);
|
||||
uint8_t pin=getint(cmd, 2);
|
||||
pinMode(pin, OUTPUT);
|
||||
digitalWrite(pin, LOW);
|
||||
}
|
||||
else if (compare(s, "echo"))
|
||||
{
|
||||
s = getword(cmd);
|
||||
if (s == "on")
|
||||
s=getword(cmd);
|
||||
if (s=="on")
|
||||
echo_on = true;
|
||||
else if (s == "off")
|
||||
else if (s=="off")
|
||||
echo_on = false;
|
||||
else
|
||||
{
|
||||
Console << s << ' ';
|
||||
while (cmd.length())
|
||||
while(cmd.length())
|
||||
{
|
||||
Console << getword(cmd) << ' ';
|
||||
}
|
||||
@@ -573,9 +546,9 @@ void eval(string &cmd)
|
||||
if (cmd.length())
|
||||
{
|
||||
Every every;
|
||||
every.ms = ms;
|
||||
every.cmd = cmd;
|
||||
every.next = millis() + ms;
|
||||
every.ms=ms;
|
||||
every.cmd=cmd;
|
||||
every.next=millis()+ms;
|
||||
everies.push_back(every);
|
||||
every.dump();
|
||||
Console << endl;
|
||||
@@ -584,16 +557,16 @@ void eval(string &cmd)
|
||||
}
|
||||
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)
|
||||
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 (count==ever or (ever==100))
|
||||
{
|
||||
if (every.active != active)
|
||||
{
|
||||
Console << "every #" << count << (active ? " on" : " off") << endl;
|
||||
Console << "every #" << count << (active ? " on" :" off") << endl;
|
||||
every.active = active;
|
||||
every.underrun = 0;
|
||||
}
|
||||
@@ -601,12 +574,12 @@ void eval(string &cmd)
|
||||
count++;
|
||||
}
|
||||
}
|
||||
else if (compare(cmd, "list") or cmd.length() == 0)
|
||||
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)
|
||||
uint8_t count=0;
|
||||
for(auto& every: everies)
|
||||
{
|
||||
Console << count << ": ";
|
||||
every.dump();
|
||||
@@ -618,20 +591,20 @@ void eval(string &cmd)
|
||||
{
|
||||
Console << "Removing..." << endl;
|
||||
getword(cmd);
|
||||
int8_t every = getint(cmd, -1);
|
||||
if (every == -1 and compare(cmd, "last") and everies.size())
|
||||
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);
|
||||
everies.erase(everies.begin()+everies.size()-1);
|
||||
}
|
||||
else if (every == -1 and compare(cmd, "all"))
|
||||
else if (every==-1 and compare(cmd, "all"))
|
||||
{
|
||||
getword(cmd);
|
||||
everies.clear();
|
||||
}
|
||||
else if (everies.size() > (uint8_t)every)
|
||||
{
|
||||
everies.erase(everies.begin() + every);
|
||||
everies.erase(everies.begin()+every);
|
||||
}
|
||||
else
|
||||
Console << "Bad colmmand" << endl;
|
||||
@@ -644,33 +617,33 @@ void eval(string &cmd)
|
||||
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]);
|
||||
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;
|
||||
blink |= 1<< blink_nr;
|
||||
else
|
||||
{
|
||||
blink &= ~(1 << blink_nr);
|
||||
blink &= ~(1<< blink_nr);
|
||||
}
|
||||
}
|
||||
}
|
||||
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;
|
||||
cmd.clear();
|
||||
}
|
||||
else if (id.length() or brokers.find(id) != brokers.end())
|
||||
else if (id.length() or brokers.find(id)!=brokers.end())
|
||||
{
|
||||
int port = getint(cmd, 0);
|
||||
int port=getint(cmd, 0);
|
||||
if (port)
|
||||
{
|
||||
MqttBroker *broker = new MqttBroker(port);
|
||||
MqttBroker* broker = new MqttBroker(port);
|
||||
broker->begin();
|
||||
|
||||
brokers[id] = broker;
|
||||
@@ -690,26 +663,25 @@ 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;
|
||||
cmd.clear();
|
||||
}
|
||||
else if (id.length() or clients.find(id) != clients.end())
|
||||
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())
|
||||
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;
|
||||
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;
|
||||
cmd=s+' '+cmd;
|
||||
clientConnect(client, cmd);
|
||||
}
|
||||
Console << "new client (" << id.c_str() << ", " << s.c_str() << ')' << endl;
|
||||
@@ -728,10 +700,10 @@ void eval(string &cmd)
|
||||
}
|
||||
else if (compare(s, "set"))
|
||||
{
|
||||
string name(getword(cmd));
|
||||
if (name.length() == 0)
|
||||
std::string name(getword(cmd));
|
||||
if (name.length()==0)
|
||||
{
|
||||
for (auto it : vars)
|
||||
for(auto it: vars)
|
||||
{
|
||||
Console << " " << it.first << " -> " << it.second << endl;
|
||||
}
|
||||
@@ -758,18 +730,18 @@ void eval(string &cmd)
|
||||
if (view)
|
||||
{
|
||||
Console << save_cursor << magenta;
|
||||
Console.gotoxy(1, 1);
|
||||
Console.gotoxy(1,1);
|
||||
}
|
||||
Console << "--< " << '/' << clients.size() << " client/s. >--" << erase_to_end << endl;
|
||||
for (auto it : clients)
|
||||
for(auto it: clients)
|
||||
{
|
||||
it.second->dump(" ");
|
||||
}
|
||||
|
||||
Console << "--< " << brokers.size() << " brokers/s. >--" << erase_to_end << endl;
|
||||
for (auto it : brokers)
|
||||
for(auto it: brokers)
|
||||
{
|
||||
Console << " +-- '" << it.first.c_str() << "' " << it.second->clientsCount() << " client/s." << erase_to_end << endl;
|
||||
Console << " +-- '" << it.first.c_str() << "' " << it.second->clientsCount() << " client/s."<< erase_to_end << endl;
|
||||
it.second->dump(" ");
|
||||
}
|
||||
if (view)
|
||||
@@ -782,7 +754,7 @@ void eval(string &cmd)
|
||||
ESP.restart();
|
||||
else if (compare(s, "ip"))
|
||||
Console << "IP: " << WiFi.localIP() << endl;
|
||||
else if (compare(s, "help"))
|
||||
else if (compare(s,"help"))
|
||||
{
|
||||
Console << "syntax:" << endl;
|
||||
Console << " MqttBroker:" << endl;
|
||||
@@ -816,8 +788,7 @@ void eval(string &cmd)
|
||||
}
|
||||
else
|
||||
{
|
||||
while (s[0] == ' ')
|
||||
s.erase(0, 1);
|
||||
while(s[0]==' ') s.erase(0,1);
|
||||
if (s.length())
|
||||
Console << "Unknown command (" << s.c_str() << ")" << endl;
|
||||
}
|
||||
@@ -831,32 +802,30 @@ void eval(string &cmd)
|
||||
|
||||
void loop()
|
||||
{
|
||||
auto ms = millis();
|
||||
int8_t out = 0;
|
||||
auto ms=millis();
|
||||
int8_t out=0;
|
||||
int16_t blink_bits = blink;
|
||||
uint8_t e = 0;
|
||||
uint8_t e=0;
|
||||
|
||||
for (auto &every : everies)
|
||||
for(auto& every: everies)
|
||||
{
|
||||
if (not every.active)
|
||||
continue;
|
||||
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)
|
||||
{
|
||||
every.next += every.ms;
|
||||
Console << yellow << "Underrun every #" << e << ", " << (ms - every.next) << "ms late" << endl;
|
||||
Console.fg(white);
|
||||
every.underrun = ms + 5000;
|
||||
Console << yellow << "Underrun every #" << e << ", " << (ms - every.next) << "ms late" << endl;
|
||||
every.underrun = ms+5000;
|
||||
}
|
||||
}
|
||||
e++;
|
||||
}
|
||||
|
||||
while (blink_bits)
|
||||
while(blink_bits)
|
||||
{
|
||||
if (blink_ms_on[out] and ms > blink_next[out])
|
||||
{
|
||||
@@ -872,18 +841,19 @@ void loop()
|
||||
}
|
||||
blink_state[out] = not blink_state[out];
|
||||
}
|
||||
blink_bits >>= 1;
|
||||
blink_bits >>=1;
|
||||
out++;
|
||||
}
|
||||
|
||||
static long count;
|
||||
#if defined(ESP9266)
|
||||
MDNS.update();
|
||||
#endif
|
||||
|
||||
for (auto it : brokers)
|
||||
for(auto it: brokers)
|
||||
it.second->loop();
|
||||
|
||||
for (auto it : clients)
|
||||
for(auto it: clients)
|
||||
it.second->loop();
|
||||
|
||||
Console.loop();
|
||||
|
||||
20
library.json
20
library.json
@@ -1,20 +0,0 @@
|
||||
{
|
||||
"name": "TinyMqtt",
|
||||
"keywords": "ethernet, mqtt, m2m, iot",
|
||||
"description": "MQTT is a lightweight messaging protocol. This library allows to host a broker or to use a mqtt client in your ESP 8266 or 32 WROOM. It does support MQTT 3.1.1 with QoS",
|
||||
"repository": {
|
||||
"type": "git",
|
||||
"url": "https://github.com/hsaturn/TinyMqtt.git"
|
||||
},
|
||||
"dependencies":
|
||||
{ "hsaturn/TinyConsole" : "*" },
|
||||
"version": "0.9.18",
|
||||
"exclude": "",
|
||||
"examples": "examples/*/*.ino",
|
||||
"frameworks": "arduino",
|
||||
"platforms": [
|
||||
"atmelavr",
|
||||
"espressif8266",
|
||||
"espressif32"
|
||||
]
|
||||
}
|
||||
@@ -1,20 +0,0 @@
|
||||
{
|
||||
"name": "#name",
|
||||
"keywords": "ethernet, mqtt, m2m, iot",
|
||||
"description": "#paragraph",
|
||||
"repository": {
|
||||
"type": "git",
|
||||
"url": "https://github.com/hsaturn/TinyMqtt.git"
|
||||
},
|
||||
"dependencies":
|
||||
{ #dependencies },
|
||||
"version": "#version",
|
||||
"exclude": "",
|
||||
"examples": "examples/*/*.ino",
|
||||
"frameworks": "arduino",
|
||||
"platforms": [
|
||||
"atmelavr",
|
||||
"espressif8266",
|
||||
"espressif32"
|
||||
]
|
||||
}
|
||||
@@ -1,11 +1,11 @@
|
||||
name=TinyMqtt
|
||||
version=0.9.18
|
||||
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=*
|
||||
depends=hsaturn/TinyConsole
|
||||
depends=TinyConsole
|
||||
includes=TinyMqtt.h
|
||||
|
||||
@@ -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; }
|
||||
|
||||
|
||||
190
src/TinyMqtt.cpp
190
src/TinyMqtt.cpp
@@ -17,32 +17,16 @@ int TinyMqtt::debug=2;
|
||||
|
||||
MqttBroker::MqttBroker(uint16_t port)
|
||||
{
|
||||
server = new TcpServer(port);
|
||||
server = std::unique_ptr<TcpServer>(new TcpServer(port));
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
server->onClient(onClient, this);
|
||||
#endif
|
||||
}
|
||||
|
||||
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 server;
|
||||
}
|
||||
|
||||
// private constructor used by broker only
|
||||
MqttClient::MqttClient(MqttBroker* local_broker, TcpClient* new_client)
|
||||
: local_broker(local_broker)
|
||||
{
|
||||
connect(local_broker);
|
||||
debug("MqttClient private with broker");
|
||||
#ifdef TINY_MQTT_ASYNC
|
||||
tcp_client = new_client;
|
||||
@@ -50,20 +34,15 @@ MqttClient::MqttClient(MqttBroker* local_broker, TcpClient* new_client)
|
||||
// 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)
|
||||
{
|
||||
alive = 0;
|
||||
keep_alive = 0;
|
||||
|
||||
if (local_broker) local_broker->addClient(this);
|
||||
}
|
||||
@@ -71,14 +50,13 @@ MqttClient::MqttClient(MqttBroker* local_broker, const string& id)
|
||||
MqttClient::~MqttClient()
|
||||
{
|
||||
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,28 +78,28 @@ 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);
|
||||
tcp_client->onConnect(onConnect, this);
|
||||
tcp_client->connect(broker.c_str(), port);
|
||||
tcp_client->connect(broker.c_str(), port, ka);
|
||||
#else
|
||||
if (tcp_client->connect(broker.c_str(), port))
|
||||
{
|
||||
debug("link established");
|
||||
onConnect(this, tcp_client);
|
||||
onConnect(this, tcp_client.get());
|
||||
}
|
||||
else
|
||||
{
|
||||
@@ -130,13 +108,13 @@ void MqttClient::connect(string broker, uint16_t port, uint16_t ka)
|
||||
#endif
|
||||
}
|
||||
|
||||
void MqttBroker::addClient(MqttClient* client)
|
||||
void MqttBroker::addClient(TcpClient* client)
|
||||
{
|
||||
debug("MqttBroker::addClient");
|
||||
clients.push_back(client);
|
||||
clients.insert(std::unique_ptr<MqttClient>(new MqttClient(this, 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 +124,7 @@ 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)
|
||||
{
|
||||
// TODO if this broker is connected to an external broker
|
||||
// we have to unsubscribe remove's topics.
|
||||
// (but doing this, check that other clients are not subscribed...)
|
||||
// Unless -> we could receive useless messages
|
||||
// -> we are using (memory) one IndexedString plus its string for nothing.
|
||||
debug("Remove " << clients.size());
|
||||
clients.erase(it);
|
||||
debug("Client removed " << clients.size());
|
||||
return;
|
||||
}
|
||||
}
|
||||
debug(red << "Error cannot remove client"); // TODO should not occur
|
||||
local_clients.erase(remove);
|
||||
}
|
||||
|
||||
void MqttBroker::onClient(void* broker_ptr, TcpClient* client)
|
||||
@@ -171,16 +132,14 @@ 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->addClient(client);
|
||||
debug("New client");
|
||||
}
|
||||
|
||||
void MqttBroker::loop()
|
||||
{
|
||||
#ifndef TINY_MQTT_ASYNC
|
||||
WiFiClient client = server->accept();
|
||||
WiFiClient client = server->available();
|
||||
|
||||
if (client)
|
||||
{
|
||||
@@ -194,21 +153,18 @@ void MqttBroker::loop()
|
||||
remote_broker->loop();
|
||||
}
|
||||
|
||||
for(size_t i=0; i<clients.size(); i++)
|
||||
// 200 bytes shorter than for(auto& client: clients) !
|
||||
for(auto it=clients.begin(); it!=clients.end(); it++)
|
||||
{
|
||||
MqttClient* client = clients[i];
|
||||
if (client->connected())
|
||||
it->get()->loop();
|
||||
if (not it->get()->connected())
|
||||
{
|
||||
client->loop();
|
||||
}
|
||||
else
|
||||
{
|
||||
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;
|
||||
clients.erase(it);
|
||||
break;
|
||||
}
|
||||
}
|
||||
for(const auto& client: local_clients)
|
||||
client->loop();
|
||||
}
|
||||
|
||||
MqttError MqttBroker::subscribe(const Topic& topic, uint8_t qos)
|
||||
@@ -226,14 +182,13 @@ MqttError MqttBroker::publish(const MqttClient* source, const Topic& topic, Mqtt
|
||||
MqttError retval = MqttOk;
|
||||
|
||||
debug("MqttBroker::publish");
|
||||
int i=0;
|
||||
for(auto client: clients)
|
||||
int clt_num = 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
|
||||
debug (" broker:" << (remote_broker && remote_broker->connected() ? "linked" : "alone")
|
||||
<< " srce=" << (source->isLocal() ? "loc" : "rem") << " clt#" << ++clt_num
|
||||
<< ", local=" << client->isLocal() << ", con=" << client->connected());
|
||||
|
||||
bool doit = false;
|
||||
if (remote_broker && remote_broker->connected()) // this (MqttBroker) is connected (to a external broker)
|
||||
{
|
||||
@@ -251,9 +206,8 @@ MqttError MqttBroker::publish(const MqttClient* source, const Topic& topic, Mqtt
|
||||
{
|
||||
doit = true;
|
||||
}
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << ", doit=" << doit << ' ';
|
||||
#endif
|
||||
|
||||
debug(" doit=" << doit << ' ');
|
||||
|
||||
if (doit) retval = client->publishIfSubscribed(topic, msg);
|
||||
debug("");
|
||||
@@ -277,16 +231,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;
|
||||
@@ -294,11 +244,11 @@ void MqttClient::clientAlive(uint32_t more_seconds)
|
||||
|
||||
void MqttClient::loop()
|
||||
{
|
||||
if (keep_alive && (millis() >= alive))
|
||||
if (alive && (millis() >= alive))
|
||||
{
|
||||
if (local_broker)
|
||||
{
|
||||
debug(red << "timeout client");
|
||||
Serial << "timeout client " << clientId << endl;
|
||||
close();
|
||||
debug(red << "closed");
|
||||
}
|
||||
@@ -306,8 +256,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 +294,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
|
||||
@@ -375,7 +326,7 @@ void MqttClient::resubscribe()
|
||||
msg.add(0);
|
||||
msg.add(0);
|
||||
|
||||
for(auto topic: subscriptions)
|
||||
for(const auto& topic: subscriptions)
|
||||
{
|
||||
msg.add(topic);
|
||||
msg.add(0); // TODO qos
|
||||
@@ -450,13 +401,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 +417,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
|
||||
@@ -496,11 +448,10 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
payload += len;
|
||||
}
|
||||
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << yellow << "Client " << clientId << " connected : keep alive=" << keep_alive << '.' << white << endl;
|
||||
#endif
|
||||
debug(yellow << "Client " << clientId << " connected : keep alive=" << keep_alive << '.' << white);
|
||||
|
||||
bclose = false;
|
||||
setFlag(CltFlagConnected);
|
||||
mqtt_flags |= FlagConnected;
|
||||
{
|
||||
MqttMessage msg(MqttMessage::Type::ConnAck);
|
||||
msg.add(0); // Session present (not implemented)
|
||||
@@ -510,14 +461,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 +479,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 +496,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,25 +540,21 @@ 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;
|
||||
#endif
|
||||
if (mqtt_connected() or tcp_client == nullptr)
|
||||
debug("publish " << (mqtt_flags & FlagConnected) << '/' << (long) tcp_client.get());
|
||||
if ((mqtt_flags & FlagConnected) or tcp_client == nullptr)
|
||||
{
|
||||
uint8_t qos = mesg->flags();
|
||||
payload = header;
|
||||
mesg->getString(payload, len);
|
||||
Topic published(payload, len);
|
||||
payload += len;
|
||||
#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;
|
||||
debug("Received Publish (" << published.str().c_str() << ") size=" << (int)len);
|
||||
// << '(' << std::string(payload, len).c_str() << ')' << " msglen=" << mesg->length() << endl;
|
||||
if (qos) payload+=2; // ignore packet identifier if any
|
||||
len=mesg->end()-payload;
|
||||
// TODO reset DUP
|
||||
@@ -638,8 +585,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 +607,7 @@ void MqttClient::processMessage(MqttMessage* mesg)
|
||||
}
|
||||
else
|
||||
{
|
||||
clientAlive(local_broker ? 5 : 0);
|
||||
clientAlive();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -758,10 +705,7 @@ MqttError MqttClient::publishIfSubscribed(const Topic& topic, MqttMessage& msg)
|
||||
else
|
||||
{
|
||||
processMessage(&msg);
|
||||
|
||||
#if TINY_MQTT_DEBUG
|
||||
Console << "Should call the callback ?\n";
|
||||
#endif
|
||||
debug("Should call the callback ?");
|
||||
// callback(this, topic, nullptr, 0); // TODO Payload
|
||||
}
|
||||
}
|
||||
@@ -899,7 +843,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 +856,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 +873,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
|
||||
@@ -33,7 +37,7 @@
|
||||
#include <rpcWiFi.h>
|
||||
#endif
|
||||
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
#include <set>
|
||||
#include <string>
|
||||
#include "StringIndexer.h"
|
||||
@@ -48,7 +52,7 @@
|
||||
static int debug;
|
||||
};
|
||||
|
||||
#define debug(what) { if (TinyMqtt::debug>=1) Console << (int)__LINE__ << ' ' << what << TinyConsole::white << endl; delay(100); }
|
||||
#define debug(what) { if (TinyMqtt::debug>=1) Console << (int)__LINE__ << ' ' << what << TinyConsole::white << endl; delay(10); }
|
||||
#else
|
||||
#define debug(what) {}
|
||||
#endif
|
||||
@@ -68,15 +72,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 +126,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 +160,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 +177,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 +187,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){}
|
||||
|
||||
~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 +208,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 +227,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 +238,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 +275,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 +289,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 +315,33 @@ 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 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(const auto& client: clients)
|
||||
client->dump(indent);
|
||||
}
|
||||
|
||||
const std::vector<MqttClient*> getClients() const { return clients; }
|
||||
using Clients = std::set<std::unique_ptr<MqttClient>>;
|
||||
using LocalClients = std::set<MqttClient*>;
|
||||
|
||||
const Clients& getClients() const { return clients; }
|
||||
const LocalClients& getLocalClients() const { return local_clients; }
|
||||
|
||||
size_t clientsCount() const { return clients.size(); }
|
||||
size_t localClientsCount() const { return local_clients.size(); }
|
||||
|
||||
private:
|
||||
friend class MqttClient;
|
||||
@@ -361,15 +358,17 @@ 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 removeClient(MqttClient* client);
|
||||
void addClient(MqttClient* local) { local_clients.insert(local); }
|
||||
void addClient(TcpClient* client);
|
||||
|
||||
void removeClient(MqttClient* local);
|
||||
|
||||
bool compareString(const char* good, const char* str, uint8_t str_len) const;
|
||||
std::vector<MqttClient*> clients;
|
||||
Clients clients;
|
||||
LocalClients local_clients;
|
||||
|
||||
private:
|
||||
TcpServer* server = nullptr;
|
||||
std::unique_ptr<TcpServer> server;
|
||||
|
||||
const char* auth_user = "guest";
|
||||
const char* auth_password = "guest";
|
||||
|
||||
142
src/make_unique.inc
Normal file
142
src/make_unique.inc
Normal file
@@ -0,0 +1,142 @@
|
||||
// Implementation of C++14's make_unique for C++11 compilers.
|
||||
//
|
||||
// This has been tested with:
|
||||
// - MSVC 11.0 (Visual Studio 2012)
|
||||
// - gcc 4.6.3
|
||||
// - Xcode 4.4 (with clang "4.0")
|
||||
//
|
||||
// It is based off an implementation proposed by Stephan T. Lavavej for
|
||||
// inclusion in the C++14 standard:
|
||||
// http://isocpp.org/files/papers/N3656.txt
|
||||
// Where appropriate, it borrows the use of MSVC's _VARIADIC_EXPAND_0X macro
|
||||
// machinery to compensate for lack of variadic templates.
|
||||
//
|
||||
// This file injects make_unique into the std namespace, which I acknowledge is
|
||||
// technically forbidden ([C++11: 17.6.4.2.2.1/1]), but is necessary in order
|
||||
// to have syntax compatibility with C++14.
|
||||
//
|
||||
// I perform compiler version checking for MSVC, gcc, and clang to ensure that
|
||||
// we don't add make_unique if it is already there (instead, we include
|
||||
// <memory> to get the compiler-provided one). You can override the compiler
|
||||
// version checking by defining the symbol COMPILER_SUPPORTS_MAKE_UNIQUE.
|
||||
//
|
||||
//
|
||||
// ===============================================================================
|
||||
// This file is released into the public domain. See LICENCE for more information.
|
||||
// ===============================================================================
|
||||
|
||||
#pragma once
|
||||
|
||||
// If user hasn't specified COMPILER_SUPPORTS_MAKE_UNIQUE then try to figure out
|
||||
// based on compiler version if std::make_unique is provided.
|
||||
#if !defined(COMPILER_SUPPORTS_MAKE_UNIQUE)
|
||||
#if defined(_MSC_VER)
|
||||
// std::make_unique was added in MSVC 12.0
|
||||
#if _MSC_VER >= 1800 // MSVC 12.0 (Visual Studio 2013)
|
||||
#define COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
#endif
|
||||
#elif defined(__clang__)
|
||||
// std::make_unique was added in clang 3.4, but not until Xcode 6.
|
||||
// Annoyingly, Apple makes the clang version defines match the version
|
||||
// of Xcode, not the version of clang.
|
||||
#define CLANG_VERSION (__clang_major__ * 10000 + __clang_minor__ * 100 + __clang_patchlevel__)
|
||||
#if defined(__APPLE__) && CLANG_VERSION >= 60000
|
||||
#define COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
#elif !defined(__APPLE__) && CLANG_VERSION >= 30400
|
||||
#define COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
#endif
|
||||
#elif defined(__GNUC__)
|
||||
// std::make_unique was added in gcc 4.9, for standards versions greater
|
||||
// than -std=c++11.
|
||||
#define GCC_VERSION (__GNUC__ * 10000 + __GNUC_MINOR__ * 100 + __GNUC_PATCHLEVEL__)
|
||||
#if GCC_VERSION >= 40900 && __cplusplus > 201103L
|
||||
#define COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
#endif
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#if defined(COMPILER_SUPPORTS_MAKE_UNIQUE)
|
||||
|
||||
// If the compiler supports std::make_unique, then pull in <memory> to get it.
|
||||
#include <memory>
|
||||
|
||||
#else
|
||||
|
||||
// Otherwise, the compiler doesn't provide it, so implement it ourselves.
|
||||
|
||||
#include <cstddef>
|
||||
#include <memory>
|
||||
#include <type_traits>
|
||||
#include <utility>
|
||||
|
||||
namespace std {
|
||||
|
||||
template<class _Ty> struct _Unique_if {
|
||||
typedef unique_ptr<_Ty> _Single_object;
|
||||
};
|
||||
|
||||
template<class _Ty> struct _Unique_if<_Ty[]> {
|
||||
typedef unique_ptr<_Ty[]> _Unknown_bound;
|
||||
};
|
||||
|
||||
template<class _Ty, size_t N> struct _Unique_if<_Ty[N]> {
|
||||
typedef void _Known_bound;
|
||||
};
|
||||
|
||||
//
|
||||
// template< class T, class... Args >
|
||||
// unique_ptr<T> make_unique( Args&&... args);
|
||||
//
|
||||
|
||||
#if defined(_MSC_VER) && (_MSC_VER < 1800)
|
||||
|
||||
// Macro machinery because MSVC 11.0 doesn't support variadic templates.
|
||||
// The _VARIADIC_EXPAND_0X stuff is defined in <xstddef>
|
||||
#define _MAKE_UNIQUE( \
|
||||
TEMPLATE_LIST, PADDING_LIST, LIST, COMMA, X1, X2, X3, X4) \
|
||||
template<class _Ty COMMA LIST(_CLASS_TYPE)> inline \
|
||||
typename _Unique_if<_Ty>::_Single_object make_unique(LIST(_TYPE_REFREF_ARG)) \
|
||||
{ \
|
||||
return unique_ptr<_Ty>(new _Ty(LIST(_FORWARD_ARG))); \
|
||||
} \
|
||||
|
||||
_VARIADIC_EXPAND_0X(_MAKE_UNIQUE, , , , )
|
||||
#undef _MAKE_UNIQUE
|
||||
|
||||
#else // not MSVC 11.0 or earlier
|
||||
|
||||
template<class _Ty, class... Args>
|
||||
typename _Unique_if<_Ty>::_Single_object
|
||||
make_unique(Args&&... args) {
|
||||
return unique_ptr<_Ty>(new _Ty(std::forward<Args>(args)...));
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
// template< class T >
|
||||
// unique_ptr<T> make_unique( std::size_t size );
|
||||
|
||||
template<class _Ty>
|
||||
typename _Unique_if<_Ty>::_Unknown_bound
|
||||
make_unique(size_t n) {
|
||||
typedef typename remove_extent<_Ty>::type U;
|
||||
return unique_ptr<_Ty>(new U[n]());
|
||||
}
|
||||
|
||||
// template< class T, class... Args >
|
||||
// /* unspecified */ make_unique( Args&&... args ) = delete;
|
||||
|
||||
// MSVC 11.0 doesn't support deleted functions, so the best we can do
|
||||
// is simply not define the function.
|
||||
#if !(defined(_MSC_VER) && (_MSC_VER < 1800))
|
||||
|
||||
template<class T, class... Args>
|
||||
typename _Unique_if<T>::_Known_bound
|
||||
make_unique(Args&&...) = delete;
|
||||
|
||||
#endif
|
||||
|
||||
} // namespace std
|
||||
|
||||
#endif // !COMPILER_SUPPORTS_MAKE_UNIQUE
|
||||
|
||||
@@ -1,40 +1,30 @@
|
||||
SUB=
|
||||
|
||||
all:runtests
|
||||
SUB=n
|
||||
|
||||
tests:
|
||||
@set -e; \
|
||||
for i in $(SUB)*-tests/Makefile; do \
|
||||
set -e; \
|
||||
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; \
|
||||
set -e; \
|
||||
$(MAKE) clean; \
|
||||
$(MAKE) -C debug-mode -j; \
|
||||
debug-mode/debug-tests.out
|
||||
|
||||
runtests:
|
||||
@set -e; \
|
||||
$(MAKE) tests; \
|
||||
for i in $(SUB)*-tests/Makefile; do \
|
||||
runtests: debugtest
|
||||
$(MAKE) clean
|
||||
$(MAKE) tests
|
||||
set -e; \
|
||||
for i in ${SUB}*-tests/Makefile; do \
|
||||
echo '==== Running:' $$(dirname $$i); \
|
||||
$$(dirname $$i)/$$(dirname $$i).out; \
|
||||
done
|
||||
|
||||
clean:
|
||||
@set -e; \
|
||||
for i in $(SUB)*-tests/Makefile; do \
|
||||
set -e; \
|
||||
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
|
||||
|
||||
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++;
|
||||
}
|
||||
|
||||
@@ -340,7 +338,7 @@ void setup() {
|
||||
while(!Serial);
|
||||
*/
|
||||
|
||||
Serial.println("=============[ TinyMqtt class-bind TESTS ]========================");
|
||||
Serial.println("=============[ FAKE NETWORK 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,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
|
||||
|
||||
# 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,89 +28,57 @@ 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);
|
||||
{
|
||||
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.clientsCount(), (size_t)0);
|
||||
}
|
||||
|
||||
test(local_client_do_not_disconnect_after_publishing_and_long_inactivity)
|
||||
{
|
||||
EpoxyTest::set_millis(0);
|
||||
MqttBroker broker(1883);
|
||||
MqttClient client(&broker, "client");
|
||||
MqttClient sender(&broker, "sender");
|
||||
broker.loop();
|
||||
|
||||
client.subscribe("#");
|
||||
client.subscribe("test");
|
||||
client.setCallback(onPublish);
|
||||
assertEqual(broker.clientsCount(), (size_t)2);
|
||||
|
||||
sender.publish("test", "value");
|
||||
broker.loop();
|
||||
|
||||
EpoxyTest::add_seconds(600);
|
||||
client.loop();
|
||||
sender.loop();
|
||||
broker.loop();
|
||||
|
||||
sender.publish("test", "value");
|
||||
broker.loop();
|
||||
sender.loop();
|
||||
broker.loop();
|
||||
|
||||
assertEqual(broker.clientsCount(), (size_t)2);
|
||||
assertEqual(sender.connected(), true);
|
||||
assertEqual(client.connected(), true);
|
||||
|
||||
assertEqual(published.size(), (size_t)1); // client has received something
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
{
|
||||
assertEqual(broker.localClientsCount(), (size_t)0); // Ensure client is not yet connected
|
||||
MqttClient client(&broker);
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is now connected
|
||||
}
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
}
|
||||
|
||||
test(local_client_alive)
|
||||
{
|
||||
set_millis(0);
|
||||
MqttBroker broker(1883);
|
||||
MqttClient client(&broker);
|
||||
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is now connected
|
||||
|
||||
add_millis(TINY_MQTT_DEFAULT_ALIVE*1000/2);
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is still connected
|
||||
|
||||
add_seconds(TINY_MQTT_DEFAULT_ALIVE*5);
|
||||
broker.loop();
|
||||
assertEqual(broker.localClientsCount(), (size_t)1); // Ensure client is still connected
|
||||
}
|
||||
|
||||
#if 0
|
||||
test(local_connect)
|
||||
{
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient client(&broker);
|
||||
MqttClient client;
|
||||
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);
|
||||
assertEqual(broker.localClientsCount(), (size_t)1);
|
||||
}
|
||||
|
||||
test(local_publish_should_be_dispatched)
|
||||
{
|
||||
published.clear();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
assertEqual(broker.localClientsCount(), (size_t)0);
|
||||
|
||||
MqttClient subscriber(&broker, "");
|
||||
MqttClient subscriber;
|
||||
subscriber.subscribe("a/b");
|
||||
subscriber.subscribe("a/c");
|
||||
subscriber.setCallback(onPublish);
|
||||
|
||||
MqttClient publisher(&broker);
|
||||
MqttClient publisher;
|
||||
publisher.publish("a/b");
|
||||
publisher.publish("a/c");
|
||||
publisher.publish("a/c");
|
||||
@@ -124,18 +91,18 @@ test(local_publish_should_be_dispatched)
|
||||
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 +116,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 +136,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 +150,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
|
||||
|
||||
APP_NAME := network-tests
|
||||
ARDUINO_LIBS := AUnit AceCommon AceTime TinyMqtt EspMock ESP8266WiFi ESPAsyncTCP TinyConsole
|
||||
|
||||
@@ -16,8 +16,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 +78,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 +142,55 @@ test(suback)
|
||||
assertEqual(MqttClient::counters[MqttMessage::Type::SubAck], 1);
|
||||
}
|
||||
|
||||
uint32_t getClientKeepAlive(MqttBroker& broker)
|
||||
{
|
||||
if (broker.getClients().size() == 1)
|
||||
for (auto& it : broker.getClients())
|
||||
return it->keepAlive();
|
||||
|
||||
return 9999;
|
||||
}
|
||||
|
||||
test(network_client_alive)
|
||||
{
|
||||
const uint32_t keep_alive=1;
|
||||
start_servers(2, true);
|
||||
assertEqual(WiFi.status(), WL_CONNECTED);
|
||||
set_millis(0); // Enter simulated time
|
||||
|
||||
MqttBroker broker(1883);
|
||||
broker.begin();
|
||||
IPAddress broker_ip = WiFi.localIP();
|
||||
|
||||
ESP8266WiFiClass::selectInstance(2);
|
||||
MqttClient client;
|
||||
client.connect(broker_ip.toString().c_str(), 1883, keep_alive);
|
||||
broker.loop();
|
||||
client.loop();
|
||||
|
||||
assertTrue(broker.clientsCount() == 1);
|
||||
assertTrue(client.connected());
|
||||
|
||||
uint32_t ka = getClientKeepAlive(broker);
|
||||
assertEqual(ka, keep_alive);
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
|
||||
// All is going well if we call client.loop()
|
||||
// The client is able to send PingReq to the broker
|
||||
add_seconds(keep_alive);
|
||||
client.loop();
|
||||
broker.loop();
|
||||
assertEqual(broker.clientsCount(), (size_t)1);
|
||||
|
||||
// Now simulate that the client is frozen for
|
||||
// a too long time
|
||||
add_seconds(TINY_MQTT_CLIENT_ALIVE_TOLERANCE*2);
|
||||
broker.loop();
|
||||
assertEqual(broker.clientsCount(), (size_t)0);
|
||||
|
||||
set_real_time();
|
||||
}
|
||||
|
||||
test(network_client_keep_alive_high)
|
||||
{
|
||||
const uint32_t keep_alive=1000;
|
||||
@@ -174,9 +221,8 @@ 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);
|
||||
|
||||
}
|
||||
|
||||
test(network_client_to_broker_connexion)
|
||||
@@ -406,20 +452,6 @@ test(network_hudge_payload)
|
||||
assertEqual(strcmp(payload, lastPayload), 0);
|
||||
}
|
||||
|
||||
test(disconnected_when_broker_is_deleted)
|
||||
{
|
||||
MqttBroker* broker = new MqttBroker(1883);
|
||||
broker->begin();
|
||||
|
||||
MqttClient client;
|
||||
client.connect(broker);
|
||||
assertEqual(client.connected(), true);
|
||||
client.publish("a", "b");
|
||||
|
||||
delete broker;
|
||||
assertEqual(client.connected(), false);
|
||||
}
|
||||
|
||||
test(connack)
|
||||
{
|
||||
const bool view = false;
|
||||
@@ -479,5 +511,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;
|
||||
@@ -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