85 lines
2.2 KiB
C++
85 lines
2.2 KiB
C++
#ifndef FIREWALL_H
|
|
#define FIREWALL_H
|
|
|
|
#include "Arduino.h"
|
|
#include "AsyncJson.h"
|
|
#include "ArduinoJson.h"
|
|
#include "EEPROM.h"
|
|
#ifdef ESP32
|
|
#include "WiFi.h"
|
|
#include "AsyncTCP.h"
|
|
#elif defined(ESP8266)
|
|
#include "ESP8266WiFi.h"
|
|
#include "ESPAsyncTCP.h"
|
|
#endif
|
|
#include "ESPAsyncWebServer.h"
|
|
|
|
#define eeprom_start_firewall_rules 4
|
|
|
|
typedef enum firewall_target : uint8_t
|
|
{
|
|
FW_REJECT = 0,
|
|
FW_DROP = 1,
|
|
FW_ACCEPT = 2,
|
|
} firewall_target_t;
|
|
|
|
typedef enum firewall_protocol : uint8_t
|
|
{
|
|
FW_TCP = 0,
|
|
FW_UDP = 1,
|
|
FW_ALL = 255,
|
|
} firewall_protocol_t;
|
|
|
|
typedef struct firewall_rule
|
|
{
|
|
uint8_t key;
|
|
char source[IP4ADDR_STRLEN_MAX];
|
|
char destination[IP4ADDR_STRLEN_MAX];
|
|
firewall_protocol_t protocol;
|
|
firewall_target_t target;
|
|
struct firewall_rule *next;
|
|
} firewall_rule_t;
|
|
|
|
class ESPFirewall
|
|
{
|
|
uint16_t eeprom_size = 512;
|
|
uint8_t amount_of_rules = 0;
|
|
int eeprom_settings_head = 0;
|
|
int eeprom_rules_head = eeprom_start_firewall_rules;
|
|
struct firewall_rule *head = NULL;
|
|
|
|
AsyncWebServer *firewall_api;
|
|
|
|
// Protocol / Target conversion
|
|
String protocol_to_string(firewall_protocol_t &);
|
|
firewall_protocol_t string_to_protocol(String &);
|
|
String target_to_string(firewall_target_t &);
|
|
firewall_target_t string_to_target(String &);
|
|
|
|
// EEPROM
|
|
void setup_eeprom();
|
|
void eeprom_write_firewall_rule(firewall_rule_t *rule);
|
|
void eeprom_read_firewall_rules();
|
|
|
|
// Firewall Actions
|
|
void add_rule_to_firewall(firewall_rule_t *);
|
|
firewall_rule_t *get_rule_from_firewall(int);
|
|
bool delete_rule_from_firewall(int);
|
|
|
|
// Firewall-API Actions
|
|
void setup_routing();
|
|
void json_message_response(AsyncWebServerRequest *, String, int);
|
|
String construct_json_firewall_rule(firewall_rule_t *);
|
|
String construct_json_firewall();
|
|
void not_found(AsyncWebServerRequest *);
|
|
void get_firewall_rule_handler(AsyncWebServerRequest *);
|
|
void get_firewall_rules_handler(AsyncWebServerRequest *);
|
|
bool request_has_firewall_parameter(AsyncWebServerRequest *);
|
|
void post_firewall_handler(AsyncWebServerRequest *);
|
|
void delete_firewall_handler(AsyncWebServerRequest *);
|
|
|
|
public:
|
|
ESPFirewall(int port = 8080);
|
|
};
|
|
|
|
#endif
|