46 lines
1.1 KiB
C++
46 lines
1.1 KiB
C++
#ifndef FIREWALL_H
|
|
#define FIREWALL_H
|
|
|
|
#include "Arduino.h"
|
|
#include "AsyncJson.h"
|
|
#include "ArduinoJson.h"
|
|
#ifdef ESP32
|
|
#include "WiFi.h"
|
|
#include "AsyncTCP.h"
|
|
#elif defined(ESP8266)
|
|
#include "ESP8266WiFi.h"
|
|
#include "ESPAsyncTCP.h"
|
|
#endif
|
|
#include "ESPAsyncWebServer.h"
|
|
|
|
#define PROTOCOL_LENGTH 4
|
|
#define TARGET_LENGTH 7
|
|
|
|
typedef struct firewall_rule
|
|
{
|
|
int key;
|
|
char source[IP4ADDR_STRLEN_MAX];
|
|
char destination[IP4ADDR_STRLEN_MAX];
|
|
char protocol[PROTOCOL_LENGTH];
|
|
char target[TARGET_LENGTH];
|
|
struct firewall_rule *next;
|
|
} firewall_rule_t;
|
|
|
|
class ESPFirewall
|
|
{
|
|
AsyncWebServer *firewall_api;
|
|
unsigned int amount_of_rules = 0;
|
|
struct firewall_rule *head = NULL;
|
|
|
|
void add_rule_to_firewall(firewall_rule_t *rule);
|
|
void get_firewall_handler(AsyncWebServerRequest *request);
|
|
void post_firewall_handler(AsyncWebServerRequest *request);
|
|
void not_found(AsyncWebServerRequest *request);
|
|
bool request_has_firewall_parameter(AsyncWebServerRequest *request);
|
|
void setup_routing();
|
|
|
|
public:
|
|
ESPFirewall(int port = 8080);
|
|
};
|
|
|
|
#endif
|