ipv4_allow.h

Go to the documentation of this file.

#pragma once
 
#include "headers.h"
#include "socket.h"
#include "esphome/core/hal.h"
 
#include <cstddef>
#include <cstdint>
 
namespace esphome::socket {
 
struct Ipv4AllowEntry {
  uint32_t addr;
  uint32_t mask;
};
 
class Ipv4Allow {
 public:
  void set(const Ipv4AllowEntry *entries, size_t count) {
    this->entries_ = entries;
    this->count_ = count;
  }
 
  bool allows(const struct sockaddr *peer) const {
    if (this->count_ == 0) {
      return true;
    }
    uint32_t addr;
    return sockaddr_to_ipv4(peer, &addr) && this->allows(addr);
  }
 
  bool allows(uint32_t addr) const {
    if (this->count_ == 0) {
      return true;
    }
    for (size_t i = 0; i != this->count_; i++) {
      Ipv4AllowEntry e = this->entry(i);
      if ((addr & e.mask) == e.addr) {
        return true;
      }
    }
    return false;
  }
 
  size_t size() const { return this->count_; }
  Ipv4AllowEntry entry(size_t i) const {
    Ipv4AllowEntry e;
    progmem_memcpy(&e, &this->entries_[i], sizeof(e));
    return e;
  }
 
 private:
  const Ipv4AllowEntry *entries_{nullptr};
  size_t count_{0};
};
 
}  // namespace esphome::socket