mirror of
https://github.com/Relintai/pandemonium_engine.git
synced 2024-12-25 05:07:12 +01:00
379 lines
7.1 KiB
C++
379 lines
7.1 KiB
C++
#include "request.h"
|
|
|
|
#include "web/http/cookie.h"
|
|
#include "web_server.h"
|
|
|
|
#include "http_session.h"
|
|
|
|
#include "session_manager.h"
|
|
#include "web/http/web_root.h"
|
|
|
|
#include "web_permission.h"
|
|
|
|
Ref<HTTPSession> Request::get_or_create_session() {
|
|
if (session.is_valid()) {
|
|
return session;
|
|
}
|
|
|
|
session = SessionManager::get_singleton()->create_session();
|
|
|
|
return session;
|
|
}
|
|
|
|
bool Request::can_view() const {
|
|
return (permissions & WebPermission::WEB_PERMISSION_VIEW) != 0;
|
|
}
|
|
bool Request::can_create() const {
|
|
return (permissions & WebPermission::WEB_PERMISSION_CREATE) != 0;
|
|
}
|
|
bool Request::can_edit() const {
|
|
return (permissions & WebPermission::WEB_PERMISSION_EDIT) != 0;
|
|
}
|
|
bool Request::can_delete() const {
|
|
return (permissions & WebPermission::WEB_PERMISSION_DELETE) != 0;
|
|
}
|
|
|
|
bool Request::has_csrf_token() {
|
|
if (!session.is_valid()) {
|
|
return false;
|
|
}
|
|
|
|
return session->has("csrf_token");
|
|
}
|
|
|
|
String Request::get_csrf_token() {
|
|
if (!session.is_valid()) {
|
|
return "";
|
|
}
|
|
|
|
const Variant &val = session->get_const("csrf_token");
|
|
|
|
if (val.is_simple_type()) {
|
|
return val.to_string();
|
|
}
|
|
|
|
return "";
|
|
}
|
|
|
|
void Request::set_csrf_token(const String &value) {
|
|
if (session.is_valid()) {
|
|
session->add("csrf_token", value);
|
|
|
|
SessionManager::get_singleton()->save_session(session);
|
|
}
|
|
}
|
|
|
|
bool Request::validate_csrf_token() {
|
|
String param_token = get_parameter("csrf_token");
|
|
param_token.trim();
|
|
|
|
if (param_token == "") {
|
|
return false;
|
|
}
|
|
|
|
String token = get_csrf_token();
|
|
|
|
if (token == "") {
|
|
return false;
|
|
}
|
|
|
|
return param_token == token;
|
|
}
|
|
|
|
const String Request::get_cookie(const String &key) {
|
|
static String str(0);
|
|
return str;
|
|
}
|
|
|
|
void Request::add_cookie(const ::Cookie &cookie) {
|
|
}
|
|
|
|
void Request::remove_cookie(const String &key) {
|
|
}
|
|
|
|
HTTPMethod Request::get_method() const {
|
|
return HTTP_METHOD_GET;
|
|
}
|
|
|
|
void Request::parse_files() {
|
|
}
|
|
int Request::get_file_count() const {
|
|
return 0;
|
|
}
|
|
int Request::get_file_length(const int index) const {
|
|
return 0;
|
|
}
|
|
const uint8_t *Request::get_file_data(const int index) const {
|
|
return nullptr;
|
|
}
|
|
|
|
const String Request::get_parameter(const String &key) const {
|
|
static String str(0);
|
|
return str;
|
|
}
|
|
|
|
HTTPStatusCode Request::get_status_code() const {
|
|
return _status_code;
|
|
}
|
|
void Request::set_status_code(const HTTPStatusCode status_code) {
|
|
_status_code = status_code;
|
|
}
|
|
|
|
void Request::send_redirect(const String &location, const HTTPStatusCode status_code) {
|
|
}
|
|
|
|
void Request::compile_body() {
|
|
compiled_body.ensure_capacity(body.size() + head.size() + 15 + 13 + 14 + 15 + 1);
|
|
|
|
// 15
|
|
compiled_body += "<!DOCTYPE html>";
|
|
|
|
// 13
|
|
compiled_body += "<html>"
|
|
"<head>";
|
|
|
|
compiled_body += head;
|
|
|
|
// 14
|
|
compiled_body += "</head>"
|
|
"<body>";
|
|
|
|
compiled_body += body;
|
|
compiled_body += footer;
|
|
|
|
// 15
|
|
compiled_body += "</body>"
|
|
"</html>";
|
|
|
|
// response->setBody(compiled_body);
|
|
}
|
|
|
|
void Request::compile_and_send_body() {
|
|
compile_body();
|
|
send();
|
|
}
|
|
|
|
void Request::send() {
|
|
// if (connection_closed) {
|
|
// RequestPool::return_request(this);
|
|
// return;
|
|
// }
|
|
|
|
// RequestPool::return_request(this);
|
|
}
|
|
|
|
void Request::send_file(const String &p_file_path) {
|
|
// RequestPool::return_request(this);
|
|
}
|
|
|
|
void Request::send_error(int error_code) {
|
|
server->get_web_root()->handle_error_send_request(this, error_code);
|
|
}
|
|
|
|
void Request::reset() {
|
|
session = nullptr;
|
|
server = nullptr;
|
|
_path_stack.clear();
|
|
_path_stack_pointer = 0;
|
|
file_size = 0;
|
|
current_file_progress = 0;
|
|
connection_closed = false;
|
|
_full_path = "";
|
|
_status_code = HTTP_STATUS_CODE_200_OK;
|
|
// Maybe set NONE or only VIEW as default?
|
|
permissions = WebPermission::WEB_PERMISSION_ALL;
|
|
active_permission.unref();
|
|
|
|
head.clear();
|
|
body.clear();
|
|
footer.clear();
|
|
compiled_body.clear();
|
|
|
|
data.clear();
|
|
reference_data.clear();
|
|
}
|
|
|
|
String Request::parser_get_path() {
|
|
return "";
|
|
}
|
|
|
|
void Request::setup_url_stack() {
|
|
_full_path = parser_get_path();
|
|
String path = parser_get_path();
|
|
|
|
size_t pos = 0;
|
|
String st;
|
|
while ((pos = path.find('/')) != -1) {
|
|
st = path.substr(0, pos);
|
|
|
|
if (st.size() != 0) {
|
|
_path_stack.push_back(st);
|
|
}
|
|
|
|
path.erase(0, pos + 1);
|
|
}
|
|
|
|
if (path.size() != 0) {
|
|
_path_stack.push_back(path);
|
|
}
|
|
}
|
|
|
|
String Request::get_path(const bool beginning_slash, const bool end_slash) const {
|
|
String path;
|
|
|
|
if (beginning_slash) {
|
|
path += '/';
|
|
}
|
|
|
|
for (uint32_t i = _path_stack_pointer; i < _path_stack.size(); ++i) {
|
|
path += _path_stack[i];
|
|
path += '/';
|
|
}
|
|
|
|
if (!end_slash && path.size() > 1) {
|
|
path.pop_back();
|
|
}
|
|
|
|
return path;
|
|
}
|
|
|
|
const String &Request::get_path_full() const {
|
|
return _full_path;
|
|
}
|
|
|
|
const String &Request::get_path_segment(const uint32_t i) const {
|
|
return _path_stack[i];
|
|
}
|
|
|
|
const String &Request::get_current_path_segment() const {
|
|
if (_path_stack_pointer >= _path_stack.size()) {
|
|
// for convenience
|
|
static const String e_str = "";
|
|
return e_str;
|
|
}
|
|
|
|
return _path_stack[_path_stack_pointer];
|
|
}
|
|
|
|
const String &Request::get_next_path_segment() const {
|
|
int pst = _path_stack_pointer + 1;
|
|
|
|
if (pst >= _path_stack.size()) {
|
|
// for convenience
|
|
static const String e_str = "";
|
|
return e_str;
|
|
}
|
|
|
|
return _path_stack[pst];
|
|
}
|
|
|
|
uint32_t Request::get_path_segment_count() const {
|
|
return _path_stack.size();
|
|
}
|
|
|
|
uint32_t Request::get_current_segment_index() const {
|
|
return _path_stack_pointer;
|
|
}
|
|
|
|
uint32_t Request::get_remaining_segment_count() const {
|
|
if (_path_stack_pointer > _path_stack.size()) {
|
|
return 0;
|
|
}
|
|
|
|
return _path_stack.size() - _path_stack_pointer;
|
|
}
|
|
|
|
void Request::pop_path() {
|
|
_path_stack_pointer -= 1;
|
|
}
|
|
|
|
void Request::push_path() {
|
|
_path_stack_pointer += 1;
|
|
}
|
|
|
|
String Request::get_url_root_parent(const int parent) const {
|
|
String path = "/";
|
|
|
|
for (uint32_t i = 0; i < _path_stack_pointer - parent; ++i) {
|
|
path += _path_stack[i];
|
|
path += "/";
|
|
}
|
|
|
|
return path;
|
|
}
|
|
|
|
String Request::get_url_root() const {
|
|
int pst = _path_stack_pointer + 1;
|
|
|
|
if (pst > _path_stack.size()) {
|
|
pst = _path_stack.size();
|
|
}
|
|
|
|
String path = "/";
|
|
|
|
for (uint32_t i = 0; i < _path_stack_pointer; ++i) {
|
|
path += _path_stack[i];
|
|
path += "/";
|
|
}
|
|
|
|
return path;
|
|
}
|
|
|
|
String Request::get_url_root_current() const {
|
|
int pst = _path_stack_pointer + 1;
|
|
|
|
if (pst > _path_stack.size()) {
|
|
pst = _path_stack.size();
|
|
}
|
|
|
|
String path = "/";
|
|
|
|
for (uint32_t i = 0; i < pst; ++i) {
|
|
path += _path_stack[i];
|
|
path += "/";
|
|
}
|
|
|
|
return path;
|
|
}
|
|
|
|
String Request::get_url_site() const {
|
|
String path = get_host();
|
|
|
|
for (uint32_t i = _path_stack_pointer; i < _path_stack.size(); ++i) {
|
|
path += _path_stack[i];
|
|
path += "/";
|
|
}
|
|
|
|
return path;
|
|
}
|
|
|
|
String Request::get_url_root_parent(const String &add) const {
|
|
return get_url_root_parent() + add;
|
|
}
|
|
String Request::get_url_root(const String &add) const {
|
|
return get_url_root() + add;
|
|
}
|
|
String Request::get_url_site(const String &add) const {
|
|
return get_url_site() + add;
|
|
}
|
|
|
|
String Request::get_host() const {
|
|
return "";
|
|
}
|
|
|
|
void Request::update() {
|
|
}
|
|
|
|
void Request::pool() {
|
|
}
|
|
|
|
Request::Request() {
|
|
// This value will need benchmarks, 2 MB seems to be just as fast for me as 4 MB, but 1MB is slower
|
|
// It is a tradeoff on server memory though, as every active download will consume this amount of memory
|
|
// where the file is bigger than this number
|
|
file_chunk_size = 1 << 21; // 2MB
|
|
}
|
|
|
|
Request::~Request() {
|
|
}
|