mirror of https://github.com/esphome/esphome.git
143 lines
4.1 KiB
C++
143 lines
4.1 KiB
C++
#pragma once
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <cstdio>
|
|
#include <array>
|
|
#include "esphome/core/macros.h"
|
|
#include "esphome/core/helpers.h"
|
|
|
|
#if defined(USE_ESP_IDF) || defined(USE_LIBRETINY) || USE_ARDUINO_VERSION_CODE > VERSION_CODE(3, 0, 0)
|
|
#include <lwip/ip_addr.h>
|
|
#endif
|
|
|
|
#if USE_ARDUINO
|
|
#include <Arduino.h>
|
|
#include <IPAddress.h>
|
|
#endif /* USE_ADRDUINO */
|
|
|
|
#ifdef USE_HOST
|
|
#include <arpa/inet.h>
|
|
using ip_addr_t = in_addr;
|
|
using ip4_addr_t = in_addr;
|
|
#define ipaddr_aton(x, y) inet_aton((x), (y))
|
|
#endif
|
|
|
|
#if USE_ESP32_FRAMEWORK_ARDUINO
|
|
#define arduino_ns Arduino_h
|
|
#elif USE_LIBRETINY
|
|
#define arduino_ns arduino
|
|
#elif USE_ARDUINO
|
|
#define arduino_ns
|
|
#endif
|
|
|
|
#ifdef USE_ESP32
|
|
#include <cstring>
|
|
#include <esp_netif.h>
|
|
#endif
|
|
|
|
namespace esphome {
|
|
namespace network {
|
|
|
|
struct IPAddress {
|
|
public:
|
|
#ifdef USE_HOST
|
|
IPAddress() { ip_addr_.s_addr = 0; }
|
|
IPAddress(uint8_t first, uint8_t second, uint8_t third, uint8_t fourth) {
|
|
this->ip_addr_.s_addr = htonl((first << 24) | (second << 16) | (third << 8) | fourth);
|
|
}
|
|
IPAddress(const std::string &in_address) { inet_aton(in_address.c_str(), &ip_addr_); }
|
|
IPAddress(const ip_addr_t *other_ip) { ip_addr_ = *other_ip; }
|
|
#else
|
|
IPAddress() { ip_addr_set_zero(&ip_addr_); }
|
|
IPAddress(uint8_t first, uint8_t second, uint8_t third, uint8_t fourth) {
|
|
IP_ADDR4(&ip_addr_, first, second, third, fourth);
|
|
}
|
|
IPAddress(const ip_addr_t *other_ip) { ip_addr_copy(ip_addr_, *other_ip); }
|
|
IPAddress(const std::string &in_address) { ipaddr_aton(in_address.c_str(), &ip_addr_); }
|
|
IPAddress(ip4_addr_t *other_ip) {
|
|
memcpy((void *) &ip_addr_, (void *) other_ip, sizeof(ip4_addr_t));
|
|
#if USE_ESP32 && LWIP_IPV6
|
|
ip_addr_.type = IPADDR_TYPE_V4;
|
|
#endif
|
|
}
|
|
#if USE_ARDUINO
|
|
IPAddress(const arduino_ns::IPAddress &other_ip) { ip_addr_set_ip4_u32(&ip_addr_, other_ip); }
|
|
#endif
|
|
#if LWIP_IPV6
|
|
IPAddress(ip6_addr_t *other_ip) {
|
|
memcpy((void *) &ip_addr_, (void *) other_ip, sizeof(ip6_addr_t));
|
|
ip_addr_.type = IPADDR_TYPE_V6;
|
|
}
|
|
#endif /* LWIP_IPV6 */
|
|
|
|
#ifdef USE_ESP32
|
|
#if LWIP_IPV6
|
|
IPAddress(esp_ip6_addr_t *other_ip) {
|
|
memcpy((void *) &ip_addr_.u_addr.ip6, (void *) other_ip, sizeof(esp_ip6_addr_t));
|
|
ip_addr_.type = IPADDR_TYPE_V6;
|
|
}
|
|
#endif /* LWIP_IPV6 */
|
|
IPAddress(esp_ip4_addr_t *other_ip) { memcpy((void *) &ip_addr_, (void *) other_ip, sizeof(esp_ip4_addr_t)); }
|
|
IPAddress(esp_ip_addr_t *other_ip) {
|
|
#if LWIP_IPV6
|
|
memcpy((void *) &ip_addr_, (void *) other_ip, sizeof(ip_addr_));
|
|
#else
|
|
memcpy((void *) &ip_addr_, (void *) &other_ip->u_addr.ip4, sizeof(ip_addr_));
|
|
#endif
|
|
}
|
|
operator esp_ip_addr_t() const {
|
|
esp_ip_addr_t tmp;
|
|
#if LWIP_IPV6
|
|
memcpy((void *) &tmp, (void *) &ip_addr_, sizeof(ip_addr_));
|
|
#else
|
|
memcpy((void *) &tmp.u_addr.ip4, (void *) &ip_addr_, sizeof(ip_addr_));
|
|
#endif /* LWIP_IPV6 */
|
|
return tmp;
|
|
}
|
|
operator esp_ip4_addr_t() const {
|
|
esp_ip4_addr_t tmp;
|
|
#if LWIP_IPV6
|
|
memcpy((void *) &tmp, (void *) &ip_addr_.u_addr.ip4, sizeof(esp_ip4_addr_t));
|
|
#else
|
|
memcpy((void *) &tmp, (void *) &ip_addr_, sizeof(ip_addr_));
|
|
#endif /* LWIP_IPV6 */
|
|
return tmp;
|
|
}
|
|
#endif /* USE_ESP32 */
|
|
|
|
operator ip_addr_t() const { return ip_addr_; }
|
|
#if LWIP_IPV6
|
|
operator ip4_addr_t() const { return *ip_2_ip4(&ip_addr_); }
|
|
#endif /* LWIP_IPV6 */
|
|
|
|
#if USE_ARDUINO
|
|
operator arduino_ns::IPAddress() const { return ip_addr_get_ip4_u32(&ip_addr_); }
|
|
#endif
|
|
|
|
bool is_set() { return !ip_addr_isany(&ip_addr_); }
|
|
bool is_ip4() { return IP_IS_V4(&ip_addr_); }
|
|
bool is_ip6() { return IP_IS_V6(&ip_addr_); }
|
|
std::string str() const { return str_lower_case(ipaddr_ntoa(&ip_addr_)); }
|
|
bool operator==(const IPAddress &other) const { return ip_addr_cmp(&ip_addr_, &other.ip_addr_); }
|
|
bool operator!=(const IPAddress &other) const { return !ip_addr_cmp(&ip_addr_, &other.ip_addr_); }
|
|
IPAddress &operator+=(uint8_t increase) {
|
|
if (IP_IS_V4(&ip_addr_)) {
|
|
#if LWIP_IPV6
|
|
(((u8_t *) (&ip_addr_.u_addr.ip4))[3]) += increase;
|
|
#else
|
|
(((u8_t *) (&ip_addr_.addr))[3]) += increase;
|
|
#endif /* LWIP_IPV6 */
|
|
}
|
|
return *this;
|
|
}
|
|
#endif
|
|
|
|
protected:
|
|
ip_addr_t ip_addr_;
|
|
};
|
|
|
|
using IPAddresses = std::array<IPAddress, 5>;
|
|
|
|
} // namespace network
|
|
} // namespace esphome
|