HS-Esslingen/esp-firewall
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add post request

Browse source
This commit is contained in:
Florian Hoss 2022-04-11 13:49:48 +02:00
parent bbea1428e5
commit 3b6c5e0561
2 changed files with 28 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
SourceCode/arduino/lib/Firewall/Firewall.cpp
View file

@ -13,7 +13,8 @@ ESPFirewall::ESPFirewall(int port)
void ESPFirewall::setup_routing() void ESPFirewall::setup_routing()
{ {
this->firewall_api->on("/api/v1/firewall", HTTP_GET, std::bind(&ESPFirewall::get_firewall_rules, this)); this->firewall_api->on("/api/v1/firewall", HTTP_GET, std::bind(&ESPFirewall::get_firewall_handler, this));
this->firewall_api->on("/api/v1/firewall", HTTP_POST, std::bind(&ESPFirewall::post_firewall_handler, this));
this->firewall_api->begin(); this->firewall_api->begin();
} }
@ -28,15 +29,15 @@ void ESPFirewall::custom_message_response(const char *message, int response_code
void ESPFirewall::prepare_firewall_json(cJSON *jsonResponse, firewall_rule_t *link) void ESPFirewall::prepare_firewall_json(cJSON *jsonResponse, firewall_rule_t *link)
{ {
cJSON_AddStringToObject(jsonResponse, "status", "success"); cJSON_AddBoolToObject(jsonResponse, "ok", true);
cJSON_AddNumberToObject(jsonResponse, "rule", link->key); cJSON_AddNumberToObject(jsonResponse, "number", link->key);
cJSON_AddStringToObject(jsonResponse, "source", link->source); cJSON_AddStringToObject(jsonResponse, "source", link->source);
cJSON_AddStringToObject(jsonResponse, "destination", link->destination); cJSON_AddStringToObject(jsonResponse, "destination", link->destination);
cJSON_AddStringToObject(jsonResponse, "protocol", link->protocol); cJSON_AddStringToObject(jsonResponse, "protocol", link->protocol);
cJSON_AddStringToObject(jsonResponse, "target", link->target); cJSON_AddStringToObject(jsonResponse, "target", link->target);
} }
firewall_rule_t *ESPFirewall::add_rule_to_firewall(char *source, char *destination, char *protocol, char *target) firewall_rule_t *ESPFirewall::add_rule_to_firewall(const char *source, const char *destination, const char *protocol, const char *target)
{ {
firewall_rule_t *temp; firewall_rule_t *temp;
firewall_rule_t *link = (firewall_rule_t *)malloc(sizeof(firewall_rule_t)); firewall_rule_t *link = (firewall_rule_t *)malloc(sizeof(firewall_rule_t));
@ -61,7 +62,26 @@ firewall_rule_t *ESPFirewall::add_rule_to_firewall(char *source, char *destinati
return link; return link;
} }
void ESPFirewall::get_firewall_rules() void ESPFirewall::post_firewall_handler()
{
if ((firewall_api->hasArg("source") || firewall_api->hasArg("destination") || firewall_api->hasArg("protocol") || firewall_api->hasArg("target")) == false)
{
this->custom_message_response("not enough erguments provided", 400);
}
else
{
const char *source = firewall_api->arg("source").c_str();
const char *destination = firewall_api->arg("destination").c_str();
const char *protocol = firewall_api->arg("protocol").c_str();
const char *target = firewall_api->arg("target").c_str();
firewall_rule_t *ptr = this->add_rule_to_firewall(source, destination, protocol, target);
cJSON *json_response = cJSON_CreateObject();
prepare_firewall_json(json_response, ptr);
this->firewall_api->send(200, "application/json", cJSON_Print(json_response));
}
}
void ESPFirewall::get_firewall_handler()
{ {
firewall_rule_t *ptr = head; firewall_rule_t *ptr = head;
cJSON *json_response = cJSON_CreateArray(); cJSON *json_response = cJSON_CreateArray();

5
SourceCode/arduino/lib/Firewall/Firewall.h
View file

@ -24,8 +24,9 @@ class ESPFirewall
void setup_routing(); void setup_routing();
void custom_message_response(const char *message, int response_code); void custom_message_response(const char *message, int response_code);
void prepare_firewall_json(cJSON *jsonResponse, firewall_rule_t *link); void prepare_firewall_json(cJSON *jsonResponse, firewall_rule_t *link);
firewall_rule_t *add_rule_to_firewall(char *source, char *destination, char *protocol, char *target); firewall_rule_t *add_rule_to_firewall(const char *source, const char *destination, const char *protocol, const char *target);
void get_firewall_rules(); void post_firewall_handler();
void get_firewall_handler();
public: public:
ESPFirewall(int port = 8080); ESPFirewall(int port = 8080);