mirror of https://github.com/arendst/Tasmota.git
150 lines
5.4 KiB
C++
150 lines
5.4 KiB
C++
/*
|
|
support_udp.ino - Udp support for Tasmota
|
|
|
|
Copyright (C) 2020 Heiko Krupp and Theo Arends
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifdef USE_EMULATION
|
|
|
|
#define UDP_BUFFER_SIZE 200 // Max UDP buffer size needed for M-SEARCH message
|
|
#define UDP_MSEARCH_SEND_DELAY 1500 // Delay in ms before M-Search response is send
|
|
|
|
#include <Ticker.h>
|
|
Ticker TickerMSearch;
|
|
|
|
IPAddress udp_remote_ip; // M-Search remote IP address
|
|
uint16_t udp_remote_port; // M-Search remote port
|
|
|
|
bool udp_connected = false;
|
|
bool udp_response_mutex = false; // M-Search response mutex to control re-entry
|
|
|
|
/*********************************************************************************************\
|
|
* UPNP/SSDP search targets
|
|
\*********************************************************************************************/
|
|
|
|
const char URN_BELKIN_DEVICE[] PROGMEM = "urn:belkin:device:**";
|
|
const char URN_BELKIN_DEVICE_CAP[] PROGMEM = "urn:Belkin:device:**";
|
|
const char UPNP_ROOTDEVICE[] PROGMEM = "upnp:rootdevice";
|
|
const char SSDPSEARCH_ALL[] PROGMEM = "ssdpsearch:all";
|
|
const char SSDP_ALL[] PROGMEM = "ssdp:all";
|
|
|
|
/*********************************************************************************************\
|
|
* UDP support routines
|
|
\*********************************************************************************************/
|
|
|
|
bool UdpDisconnect(void)
|
|
{
|
|
if (udp_connected) {
|
|
PortUdp.flush();
|
|
WiFiUDP::stopAll();
|
|
AddLog_P(LOG_LEVEL_DEBUG, PSTR(D_LOG_UPNP D_MULTICAST_DISABLED));
|
|
udp_connected = false;
|
|
}
|
|
return udp_connected;
|
|
}
|
|
|
|
bool UdpConnect(void)
|
|
{
|
|
if (!udp_connected && !restart_flag) {
|
|
// Simple Service Discovery Protocol (SSDP)
|
|
if (PortUdp.beginMulticast(WiFi.localIP(), IPAddress(239,255,255,250), 1900)) {
|
|
AddLog_P(LOG_LEVEL_INFO, PSTR(D_LOG_UPNP D_MULTICAST_REJOINED));
|
|
udp_response_mutex = false;
|
|
udp_connected = true;
|
|
} else {
|
|
AddLog_P(LOG_LEVEL_INFO, PSTR(D_LOG_UPNP D_MULTICAST_JOIN_FAILED));
|
|
udp_connected = false;
|
|
}
|
|
}
|
|
return udp_connected;
|
|
}
|
|
|
|
void PollUdp(void)
|
|
{
|
|
if (udp_connected) {
|
|
while (PortUdp.parsePacket()) {
|
|
char packet_buffer[UDP_BUFFER_SIZE]; // buffer to hold incoming UDP/SSDP packet
|
|
|
|
int len = PortUdp.read(packet_buffer, UDP_BUFFER_SIZE -1);
|
|
packet_buffer[len] = 0;
|
|
|
|
AddLog_P2(LOG_LEVEL_DEBUG_MORE, PSTR("UDP: Packet (%d)"), len);
|
|
// AddLog_P2(LOG_LEVEL_DEBUG_MORE, PSTR("\n%s"), packet_buffer);
|
|
|
|
// Simple Service Discovery Protocol (SSDP)
|
|
if (Settings.flag2.emulation) {
|
|
#ifdef USE_SCRIPT_HUE
|
|
if (!udp_response_mutex && (strstr_P(packet_buffer, PSTR("M-SEARCH")) != nullptr)) {
|
|
#else
|
|
if (devices_present && !udp_response_mutex && (strstr_P(packet_buffer, PSTR("M-SEARCH")) != nullptr)) {
|
|
#endif
|
|
udp_response_mutex = true;
|
|
|
|
udp_remote_ip = PortUdp.remoteIP();
|
|
udp_remote_port = PortUdp.remotePort();
|
|
|
|
// AddLog_P2(LOG_LEVEL_DEBUG_MORE, PSTR("UDP: M-SEARCH Packet from %s:%d\n%s"),
|
|
// udp_remote_ip.toString().c_str(), udp_remote_port, packet_buffer);
|
|
|
|
uint32_t response_delay = UDP_MSEARCH_SEND_DELAY + ((millis() &0x7) * 100); // 1500 - 2200 msec
|
|
|
|
LowerCase(packet_buffer, packet_buffer);
|
|
RemoveSpace(packet_buffer);
|
|
|
|
#ifdef USE_EMULATION_WEMO
|
|
if (EMUL_WEMO == Settings.flag2.emulation) {
|
|
if (strstr_P(packet_buffer, URN_BELKIN_DEVICE) != nullptr) { // type1 echo dot 2g, echo 1g's
|
|
TickerMSearch.attach_ms(response_delay, WemoRespondToMSearch, 1);
|
|
return;
|
|
}
|
|
else if ((strstr_P(packet_buffer, UPNP_ROOTDEVICE) != nullptr) || // type2 Echo 2g (echo & echo plus)
|
|
(strstr_P(packet_buffer, SSDPSEARCH_ALL) != nullptr) ||
|
|
(strstr_P(packet_buffer, SSDP_ALL) != nullptr)) {
|
|
TickerMSearch.attach_ms(response_delay, WemoRespondToMSearch, 2);
|
|
return;
|
|
}
|
|
}
|
|
#endif // USE_EMULATION_WEMO
|
|
|
|
#ifdef USE_EMULATION_HUE
|
|
if (EMUL_HUE == Settings.flag2.emulation) {
|
|
if ((strstr_P(packet_buffer, PSTR(":device:basic:1")) != nullptr) ||
|
|
(strstr_P(packet_buffer, UPNP_ROOTDEVICE) != nullptr) ||
|
|
(strstr_P(packet_buffer, SSDPSEARCH_ALL) != nullptr) ||
|
|
(strstr_P(packet_buffer, SSDP_ALL) != nullptr)) {
|
|
TickerMSearch.attach_ms(response_delay, HueRespondToMSearch);
|
|
return;
|
|
}
|
|
}
|
|
#endif // USE_EMULATION_HUE
|
|
|
|
udp_response_mutex = false;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
#ifdef USE_DEVICE_GROUPS
|
|
if (Settings.flag4.device_groups_enabled && !strncmp_P(packet_buffer, kDeviceGroupMessage, sizeof(DEVICE_GROUP_MESSAGE) - 1)) {
|
|
ProcessDeviceGroupMessage(packet_buffer, len);
|
|
}
|
|
#endif // USE_DEVICE_GROUPS
|
|
}
|
|
optimistic_yield(100);
|
|
}
|
|
}
|
|
|
|
#endif // USE_EMULATION
|