285 lines
9.8 KiB
C++
285 lines
9.8 KiB
C++
#include "API.hpp"
|
|
|
|
namespace fw
|
|
{
|
|
API::API(fw::Firewall *firewall, const char *cert, const char *key, const char *username, const char *password, const uint16_t port)
|
|
{
|
|
this->firewall = firewall;
|
|
this->api_ip = WiFi.localIP().toString();
|
|
this->api_port = port;
|
|
if (this->setup_auth(username, password) == ERROR)
|
|
endless_loop();
|
|
#ifdef ESP32
|
|
this->server = new WebServer(port);
|
|
#elif defined(ESP8266)
|
|
this->server = new ESP8266WebServerSecure(port);
|
|
this->serverCache = new ServerSessions(5);
|
|
#endif
|
|
this->setup_routing(cert, key);
|
|
Serial.printf("Starting server on port %i...\n", port);
|
|
this->server->begin();
|
|
}
|
|
|
|
API::~API()
|
|
{
|
|
this->server->stop();
|
|
}
|
|
|
|
void API::handle_client()
|
|
{
|
|
this->server->handleClient();
|
|
}
|
|
|
|
String API::get_url_base()
|
|
{
|
|
#ifdef ESP32
|
|
return "http://" + this->api_ip + ":" + this->api_port;
|
|
#elif defined(ESP8266)
|
|
return "https://" + this->api_ip + ":" + this->api_port;
|
|
#endif
|
|
}
|
|
|
|
ok_t API::setup_auth(const char *username, const char *password)
|
|
{
|
|
if (!username || *username == 0x00 || strlen(username) > CREDENTIALS_LENGTH)
|
|
{
|
|
Serial.println("Username too long or missing!");
|
|
return ERROR;
|
|
}
|
|
strncpy(credentials.username, username, CREDENTIALS_LENGTH);
|
|
if (!password || *password == 0x00 || strlen(password) > CREDENTIALS_LENGTH)
|
|
{
|
|
Serial.println("Password too long or missing!");
|
|
return ERROR;
|
|
}
|
|
strncpy(credentials.password, password, CREDENTIALS_LENGTH);
|
|
return SUCCESS;
|
|
}
|
|
|
|
auth_t API::check_auth()
|
|
{
|
|
if (server->authenticate(this->credentials.username, this->credentials.password))
|
|
{
|
|
return AUTHENTICATED;
|
|
}
|
|
else
|
|
{
|
|
this->json_message_response("please provide username and password", 403);
|
|
return DENIED;
|
|
}
|
|
}
|
|
|
|
void API::setup_routing(const char *cert, const char *key)
|
|
{
|
|
#ifdef ESP8266
|
|
this->server->getServer().setRSACert(new BearSSL::X509List(cert), new BearSSL::PrivateKey(key));
|
|
this->server->getServer().setCache(serverCache);
|
|
#endif
|
|
this->server->on("/firewall", HTTP_GET, std::bind(&API::get_firewall_rules_handler, this));
|
|
this->server->on(UriRegex("/firewall/([0-9]+)"), HTTP_GET, std::bind(&API::get_firewall_rule_handler, this));
|
|
this->server->on("/firewall", HTTP_POST, std::bind(&API::post_firewall_handler, this));
|
|
this->server->on(UriRegex("/firewall/([0-9]+)"), HTTP_DELETE, std::bind(&API::delete_firewall_handler, this));
|
|
this->server->on("/api", HTTP_GET, std::bind(&API::get_endpoint_list_handler, this));
|
|
this->server->onNotFound(std::bind(&API::not_found_handler, this));
|
|
|
|
add_endpoint_to_list("/firewall", "GET", "Get all Firewall Rules");
|
|
add_endpoint_to_list("/firewall/1", "GET", "Get Firewall Rule by key");
|
|
add_endpoint_to_list("/firewall", "POST", "Create Firewall Rule");
|
|
add_endpoint_to_list("/firewall/1", "DELETE", "Delete Firewall Rule by key");
|
|
}
|
|
|
|
void API::add_endpoint_to_list(const String uri, const char *method, const char *description)
|
|
{
|
|
api_endpoint_t *temp;
|
|
const String url = get_url_base() + uri;
|
|
|
|
api_endpoint_t *api_ptr = (api_endpoint_t *)malloc(sizeof(api_endpoint_t));
|
|
strncpy(api_ptr->uri, url.c_str(), sizeof(api_ptr->uri));
|
|
strncpy(api_ptr->method, method, sizeof(api_ptr->method));
|
|
strncpy(api_ptr->description, description, sizeof(api_ptr->description));
|
|
|
|
if (this->endpoint_head == NULL)
|
|
{
|
|
this->endpoint_head = api_ptr;
|
|
api_ptr->next = NULL;
|
|
return;
|
|
}
|
|
temp = this->endpoint_head;
|
|
while (temp->next != NULL)
|
|
{
|
|
temp = temp->next;
|
|
}
|
|
temp->next = api_ptr;
|
|
api_ptr->next = NULL;
|
|
return;
|
|
}
|
|
|
|
void API::not_found_handler()
|
|
{
|
|
this->json_message_response(
|
|
"see " + get_url_base() + "/api for available routes",
|
|
404);
|
|
}
|
|
|
|
void API::get_endpoint_list_handler()
|
|
{
|
|
this->json_generic_response(this->construct_json_api(), 200);
|
|
}
|
|
|
|
void API::get_firewall_rule_handler()
|
|
{
|
|
if (this->check_auth() == DENIED)
|
|
return;
|
|
String param = this->server->pathArg(0);
|
|
int rule_number = atoi(param.c_str());
|
|
firewall_rule_t *rule_ptr = firewall->get_rule_from_firewall(rule_number);
|
|
if (rule_ptr == NULL)
|
|
this->json_message_response("rule does not exist", 404);
|
|
else
|
|
this->json_generic_response(this->construct_json_firewall_rule(rule_ptr), 200);
|
|
}
|
|
|
|
void API::get_firewall_rules_handler()
|
|
{
|
|
if (this->check_auth() == DENIED)
|
|
return;
|
|
this->json_generic_response(this->construct_json_firewall(), 200);
|
|
}
|
|
|
|
void API::post_firewall_handler()
|
|
{
|
|
if (this->check_auth() == DENIED)
|
|
return;
|
|
if (request_has_all_firewall_parameter())
|
|
{
|
|
firewall_rule_t *rule_ptr = firewall->add_rule_to_firewall(
|
|
this->server->arg("source"),
|
|
this->server->arg("destination"),
|
|
this->server->arg("port_from"),
|
|
this->server->arg("port_to"),
|
|
this->server->arg("protocol"),
|
|
this->server->arg("target"));
|
|
this->json_generic_response(this->construct_json_firewall_rule(rule_ptr), 200);
|
|
}
|
|
else
|
|
{
|
|
this->json_message_response("not enough parameter provided", 400);
|
|
}
|
|
}
|
|
|
|
void API::delete_firewall_handler()
|
|
{
|
|
if (this->check_auth() == DENIED)
|
|
return;
|
|
String param = this->server->pathArg(0);
|
|
int rule_number = atoi(param.c_str());
|
|
if (firewall->delete_rule_from_firewall(rule_number) == SUCCESS)
|
|
this->json_message_response("firewall rule deleted", 200);
|
|
else
|
|
this->json_message_response("cannot delete firewall rule", 500);
|
|
}
|
|
|
|
bool API::request_has_all_firewall_parameter()
|
|
{
|
|
if (!this->server->args())
|
|
return false;
|
|
else
|
|
{
|
|
for (uint8_t i = 0; i < firewall_fields_amount; i++)
|
|
{
|
|
if (!this->server->hasArg(firewall_fields[i]))
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
String API::json_new_attribute(String key, String value, bool last)
|
|
{
|
|
String json_string;
|
|
json_string += "\"" + key + "\": \"" + value + "\"";
|
|
if (!last)
|
|
json_string += ",";
|
|
return json_string;
|
|
}
|
|
|
|
String API::json_new_attribute(String key, uint32_t value, bool last)
|
|
{
|
|
return json_new_attribute(key, String(value), last);
|
|
}
|
|
|
|
void API::json_generic_response(String serialized_string, const uint16_t response_code)
|
|
{
|
|
this->server->send(
|
|
response_code,
|
|
"application/json; charset=utf-8",
|
|
construct_json_begin(response_code) + "\"result\": [" + serialized_string + "]}");
|
|
}
|
|
|
|
void API::json_message_response(String message, const uint16_t response_code)
|
|
{
|
|
String serialized_string = construct_json_begin(response_code);
|
|
serialized_string += json_new_attribute("message", message, true);
|
|
serialized_string += "}";
|
|
this->server->send(response_code, "application/json; charset=utf-8", serialized_string);
|
|
}
|
|
|
|
String API::construct_json_firewall_rule(firewall_rule_t *rule_ptr)
|
|
{
|
|
String serialized_string = "{";
|
|
serialized_string += json_new_attribute("key", rule_ptr->key);
|
|
serialized_string += json_new_attribute("source", rule_ptr->source);
|
|
serialized_string += json_new_attribute("destination", rule_ptr->destination);
|
|
serialized_string += json_new_attribute("port_from", rule_ptr->port_from);
|
|
serialized_string += json_new_attribute("port_to", rule_ptr->port_to);
|
|
serialized_string += json_new_attribute("protocol", protocol_to_string(rule_ptr->protocol));
|
|
serialized_string += json_new_attribute("target", target_to_string(rule_ptr->target), true);
|
|
serialized_string += "}";
|
|
return serialized_string;
|
|
}
|
|
|
|
String API::construct_json_firewall()
|
|
{
|
|
firewall_rule_t *rule_ptr = firewall->get_rule_head();
|
|
String serialized_string;
|
|
while (rule_ptr != NULL)
|
|
{
|
|
serialized_string += construct_json_firewall_rule(rule_ptr);
|
|
rule_ptr = rule_ptr->next;
|
|
if (rule_ptr != NULL)
|
|
serialized_string += ",";
|
|
}
|
|
return serialized_string;
|
|
}
|
|
|
|
String API::construct_json_api_endpoint(api_endpoint_t *api_ptr)
|
|
{
|
|
String serialized_string = "{";
|
|
serialized_string += json_new_attribute("endpoint", api_ptr->uri);
|
|
serialized_string += json_new_attribute("description", api_ptr->description);
|
|
serialized_string += json_new_attribute("method", api_ptr->method, true);
|
|
serialized_string += "}";
|
|
return serialized_string;
|
|
}
|
|
|
|
String API::construct_json_api()
|
|
{
|
|
api_endpoint_t *api_ptr = this->endpoint_head;
|
|
String serialized_string;
|
|
while (api_ptr != NULL)
|
|
{
|
|
serialized_string += construct_json_api_endpoint(api_ptr);
|
|
api_ptr = api_ptr->next;
|
|
if (api_ptr != NULL)
|
|
serialized_string += ",";
|
|
}
|
|
return serialized_string;
|
|
}
|
|
|
|
String API::construct_json_begin(const uint16_t response_code)
|
|
{
|
|
String serialized_string = "{";
|
|
serialized_string += json_new_attribute("status", response_code_to_string(response_code));
|
|
return serialized_string;
|
|
}
|
|
}
|