Firewall.hpp

#ifndef FIREWALL_HPP
#define FIREWALL_HPP
 
#include "Utils.hpp"
#include "Storage.hpp"
#include "WiFiClient.h"
#include "lwip/netif.h"
#include "lwip/pbuf.h"
#include "lwip/ip4.h"
#include "lwip/prot/udp.h"
#include "lwip/prot/tcp.h"
 
namespace fw
{
    class Firewall : public Storage
    {
    public:
        Firewall();
 
        ~Firewall();
 
        firewall_rule_t *get_rule_head();
 
        void add_rule_to_firewall(firewall_rule_t *rule_ptr, const bool save_in_eeprom = true);
 
        firewall_rule_t *add_rule_to_firewall(String *args);
 
        firewall_rule_t *update_rule_of_firewall(String *args, const uint8_t key);
 
        firewall_rule_t *get_rule_from_firewall(const uint8_t key);
 
        ok_t delete_rule_from_firewall(const uint8_t key);
 
        bool is_packet_allowed(struct pbuf *pbuf);
 
    protected:
        bool rule_allows_packet(firewall_rule_t *rule_ptr, my_packet_t *packet);
 
        my_packet_t *get_packet_information(struct pbuf *pbuf);
 
        uint8_t amount_of_rules = 0;
        firewall_rule_t *rule_head = NULL;
    };
}
 
#endif