58 lines
1.7 KiB
Arduino
58 lines
1.7 KiB
Arduino
|
#include <PicoMQTT.h>
|
||
|
|
|
||
|
|
#if __has_include("config.h")
|
||
|
|
#include "config.h"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#ifndef WIFI_SSID
|
||
|
|
#define WIFI_SSID "WiFi SSID"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#ifndef WIFI_PASSWORD
|
||
|
|
#define WIFI_PASSWORD "password"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
class MQTT: public PicoMQTT::Server {
|
||
|
|
protected:
|
||
|
|
PicoMQTT::ConnectReturnCode auth(const char * client_id, const char * username, const char * password) override {
|
||
|
|
// only accept client IDs which are 3 chars or longer
|
||
|
|
if (String(client_id).length() < 3) { // client_id is never NULL
|
||
|
|
return PicoMQTT::CRC_IDENTIFIER_REJECTED;
|
||
|
|
}
|
||
|
|
|
||
|
|
// only accept connections if username and password are provided
|
||
|
|
if (!username || !password) { // username and password can be NULL
|
||
|
|
// no username or password supplied
|
||
|
|
return PicoMQTT::CRC_NOT_AUTHORIZED;
|
||
|
|
}
|
||
|
|
|
||
|
|
// accept two user/password combinations
|
||
|
|
if (
|
||
|
|
((String(username) == "alice") && (String(password) == "secret"))
|
||
|
|
|| ((String(username) == "bob") && (String(password) == "password"))) {
|
||
|
|
return PicoMQTT::CRC_ACCEPTED;
|
||
|
|
}
|
||
|
|
|
||
|
|
// reject all other credentials
|
||
|
|
return PicoMQTT::CRC_BAD_USERNAME_OR_PASSWORD;
|
||
|
|
}
|
||
|
|
} mqtt;
|
||
|
|
|
||
|
|
void setup() {
|
||
|
|
// Setup serial
|
||
|
|
Serial.begin(115200);
|
||
|
|
|
||
|
|
// Connect to WiFi
|
||
|
|
Serial.printf("Connecting to WiFi %s\n", WIFI_SSID);
|
||
|
|
WiFi.mode(WIFI_STA);
|
||
|
|
WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
|
||
|
|
while (WiFi.status() != WL_CONNECTED) { delay(1000); }
|
||
|
|
Serial.printf("WiFi connected, IP: %s\n", WiFi.localIP().toString().c_str());
|
||
|
|
|
||
|
|
mqtt.begin();
|
||
|
|
}
|
||
|
|
|
||
|
|
void loop() {
|
||
|
|
mqtt.loop();
|
||
|
|
}
|