pandemonium_engine/modules/http_server_simple/http_server_simple.h

#ifndef HTTP_SERVER_SIMPLE_H
#define HTTP_SERVER_SIMPLE_H

/*************************************************************************/
/*  http_server_simple.h                                                 */
/*************************************************************************/
/*                         This file is part of:                         */
/*                          PANDEMONIUM ENGINE                           */
/*             https://github.com/Relintai/pandemonium_engine            */
/*************************************************************************/
/* Copyright (c) 2022-present Péter Magyar.                              */
/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md).   */
/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur.                 */
/*                                                                       */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the       */
/* "Software"), to deal in the Software without restriction, including   */
/* without limitation the rights to use, copy, modify, merge, publish,   */
/* distribute, sublicense, and/or sell copies of the Software, and to    */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions:                                             */
/*                                                                       */
/* The above copyright notice and this permission notice shall be        */
/* included in all copies or substantial portions of the Software.       */
/*                                                                       */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,       */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF    */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
/*************************************************************************/

#include "core/containers/list.h"
#include "core/containers/vector.h"
#include "core/io/image_loader.h"
#include "core/io/json.h"
#include "core/io/stream_peer_ssl.h"
#include "core/io/tcp_server.h"
#include "core/io/zip_io.h"
#include "core/os/rw_lock.h"
#include "core/os/semaphore.h"

#include "core/config/project_settings.h"

#include "modules/web/http/http_server_enums.h"

class HTTPParser;
class WebServerSimple;
class WebServerRequest;
class HTTPServerSimple;
class SimpleWebServerRequest;

class HTTPServerConnection : public Reference {
	GDCLASS(HTTPServerConnection, Reference);

public:
	void update();

	void send_redirect(Ref<WebServerRequest> request, const String &location, const HTTPServerEnums::HTTPStatusCode status_code);
	void send(Ref<WebServerRequest> request);
	void send_file(Ref<WebServerRequest> request, const String &p_file_path);

	void update_send_file(Ref<SimpleWebServerRequest> request);

	void close();
	bool closed();

	bool has_more_messages();

	void close_file(Ref<SimpleWebServerRequest> request);

	HTTPServerConnection();
	~HTTPServerConnection();

	WebServerSimple *_web_server;
	HTTPServerSimple *_http_server;

	uint64_t max_request_size;

	bool use_ssl = false;
	Ref<CryptoKey> key;

	Ref<StreamPeerTCP> tcp;
	Ref<StreamPeerSSL> ssl;
	Ref<StreamPeer> peer;

	Ref<HTTPParser> _http_parser;
	uint64_t time = 0;
	uint8_t req_buf[4096];

	Ref<SimpleWebServerRequest> _current_request;
	uint8_t _file_send_buffer[4096];
	uint64_t _file_buffer_start;
	uint64_t _file_buffer_end;

	uint64_t _file_start;
	uint64_t _file_end;
	uint64_t _file_length;

	int _file_buffer_send_max_consecutive_loops;

	uint64_t _timeout_usec;

	bool _closed;
};

class HTTPServerSimple : public Reference {
	GDCLASS(HTTPServerSimple, Reference);

public:
	void stop();

	Error listen(int p_port, IP_Address p_address, bool p_use_ssl, String p_ssl_key, String p_ssl_cert);
	bool is_listening() const;
	void poll();

	Dictionary unregister_connection_for_request(const Ref<WebServerRequest> &request);

	HTTPServerSimple();
	~HTTPServerSimple();

	WebServerSimple *_web_server;

	uint64_t max_request_size;

	RBMap<StringName, String> mimes;

	Ref<X509Certificate> cert;

	bool _use_worker_threads;
	int _thread_count;

	String _ssl_key_file;
	String _ssl_cert_file;

private:
	Ref<TCP_Server> server;

	Ref<CryptoKey> key;
	bool use_ssl = false;

	List<Ref<HTTPServerConnection>> _connections;
	RWLock _connections_lock;

	void _clear_clients();
	void _stop_workers();
	void _set_internal_certs(Ref<Crypto> p_crypto);

	void _wake_workers();

	struct ServerWorkerThread {
		Thread *thread;
		Semaphore *semaphore;
		Ref<HTTPServerSimple> server;
		Ref<HTTPServerConnection> current_connection;
		bool running;
		bool working;

		ServerWorkerThread() {
			thread = nullptr;
			semaphore = nullptr;
			running = false;
			working = false;
		}
	};

	Vector<ServerWorkerThread *> _threads;

	static void _worker_thread_func(void *data);
};

#endif