mirror of
https://github.com/TES3MP/openmw-tes3mp.git
synced 2025-01-16 19:19:56 +00:00
[Master] Add RestAPI
This commit is contained in:
parent
892960f913
commit
ec6614ba32
8 changed files with 901 additions and 19 deletions
|
@ -3,7 +3,7 @@ project(masterserver)
|
||||||
set(CMAKE_CXX_STANDARD 14)
|
set(CMAKE_CXX_STANDARD 14)
|
||||||
include_directories("./")
|
include_directories("./")
|
||||||
|
|
||||||
set(SOURCE_FILES main.cpp MasterServer.cpp MasterServer.hpp)
|
set(SOURCE_FILES main.cpp MasterServer.cpp MasterServer.hpp RestServer.cpp RestServer.hpp)
|
||||||
|
|
||||||
add_executable(masterserver ${SOURCE_FILES})
|
add_executable(masterserver ${SOURCE_FILES})
|
||||||
target_link_libraries(masterserver ${RakNet_LIBRARY} components)
|
target_link_libraries(masterserver ${RakNet_LIBRARY} components)
|
||||||
|
|
|
@ -208,3 +208,8 @@ void MasterServer::Wait()
|
||||||
tMasterThread.join();
|
tMasterThread.join();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
MasterServer::ServerMap *MasterServer::GetServers()
|
||||||
|
{
|
||||||
|
return &servers;
|
||||||
|
}
|
||||||
|
|
|
@ -13,6 +13,22 @@
|
||||||
class MasterServer
|
class MasterServer
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
struct Ban
|
||||||
|
{
|
||||||
|
RakNet::SystemAddress sa;
|
||||||
|
bool permanent;
|
||||||
|
struct Date
|
||||||
|
{
|
||||||
|
} date;
|
||||||
|
};
|
||||||
|
struct SServer : QueryData
|
||||||
|
{
|
||||||
|
std::chrono::steady_clock::time_point lastUpdate;
|
||||||
|
};
|
||||||
|
typedef std::map<RakNet::SystemAddress, SServer> ServerMap;
|
||||||
|
//typedef ServerMap::const_iterator ServerCIter;
|
||||||
|
typedef ServerMap::iterator ServerIter;
|
||||||
|
|
||||||
MasterServer(unsigned short maxConnections, unsigned short port);
|
MasterServer(unsigned short maxConnections, unsigned short port);
|
||||||
~MasterServer();
|
~MasterServer();
|
||||||
|
|
||||||
|
@ -21,24 +37,7 @@ public:
|
||||||
bool isRunning();
|
bool isRunning();
|
||||||
void Wait();
|
void Wait();
|
||||||
|
|
||||||
struct SServer : QueryData
|
ServerMap* GetServers();
|
||||||
{
|
|
||||||
std::chrono::steady_clock::time_point lastUpdate;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct Ban
|
|
||||||
{
|
|
||||||
RakNet::SystemAddress sa;
|
|
||||||
bool permanent;
|
|
||||||
struct Date
|
|
||||||
{
|
|
||||||
|
|
||||||
} date;
|
|
||||||
};
|
|
||||||
|
|
||||||
typedef std::map<RakNet::SystemAddress, SServer> ServerMap;
|
|
||||||
typedef ServerMap::const_iterator ServerCIter;
|
|
||||||
typedef ServerMap::iterator ServerIter;
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void Thread();
|
void Thread();
|
||||||
|
|
191
apps/master/RestServer.cpp
Normal file
191
apps/master/RestServer.cpp
Normal file
|
@ -0,0 +1,191 @@
|
||||||
|
//
|
||||||
|
// Created by koncord on 13.05.17.
|
||||||
|
//
|
||||||
|
|
||||||
|
#include "RestServer.hpp"
|
||||||
|
|
||||||
|
#include <boost/property_tree/ptree.hpp>
|
||||||
|
#include <boost/property_tree/json_parser.hpp>
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
using namespace chrono;
|
||||||
|
using namespace boost::property_tree;
|
||||||
|
|
||||||
|
static string response201 = "HTTP/1.1 201 Created\r\nContent-Length: 11\r\n\r\n201 Created";
|
||||||
|
static string response202 = "HTTP/1.1 202 Accepted\r\nContent-Length: 12\r\n\r\n202 Accepted";
|
||||||
|
static string response400 = "HTTP/1.1 400 Bad Request\r\nContent-Length: 15\r\n\r\n400 Bad Request";
|
||||||
|
|
||||||
|
inline void ResponseStr(HttpServer::Response &response, string content, string type = "", string code = "200 OK")
|
||||||
|
{
|
||||||
|
response << "HTTP/1.1 " << code << "\r\n";
|
||||||
|
if (!type.empty())
|
||||||
|
response << "Content-Type: " << type <<"\r\n";
|
||||||
|
response << "Content-Length: " << content.length() << "\r\n\r\n" << content;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline void ptreeToServer(boost::property_tree::ptree &pt, MasterServer::SServer &server)
|
||||||
|
{
|
||||||
|
server.SetName(pt.get<string>("hostname").c_str());
|
||||||
|
server.SetGameMode(pt.get<string>("modname").c_str());
|
||||||
|
server.SetVersion(pt.get<string>("version").c_str());
|
||||||
|
server.SetPassword(pt.get<bool>("passw"));
|
||||||
|
//server.query_port = pt.get<unsigned short>("query_port");
|
||||||
|
server.SetPlayers(pt.get<unsigned>("players"));
|
||||||
|
server.SetMaxPlayers(pt.get<unsigned>("max_players"));
|
||||||
|
}
|
||||||
|
|
||||||
|
inline void queryToStringStream(stringstream &ss, string addr, MasterServer::SServer &query)
|
||||||
|
{
|
||||||
|
ss <<"\"" << addr << "\":{";
|
||||||
|
ss << "\"modname\": \"" << query.GetGameMode() << "\"" << ", ";
|
||||||
|
ss << "\"passw\": " << (query.GetPassword() ? "true" : "false") << ", ";
|
||||||
|
ss << "\"hostname\": \"" << query.GetName() << "\"" << ", ";
|
||||||
|
ss << "\"query_port\": " << 0 << ", ";
|
||||||
|
ss << "\"last_update\": " << duration_cast<seconds>(steady_clock::now() - query.lastUpdate).count() << ", ";
|
||||||
|
ss << "\"players\": " << query.GetPlayers() << ", ";
|
||||||
|
ss << "\"version\": \"" << query.GetVersion() << "\"" << ", ";
|
||||||
|
ss << "\"max_players\": " << query.GetMaxPlayers();
|
||||||
|
ss << "}";
|
||||||
|
}
|
||||||
|
|
||||||
|
RestServer::RestServer(unsigned short port, MasterServer::ServerMap *pMap) : serverMap(pMap)
|
||||||
|
{
|
||||||
|
httpServer.config.port = port;
|
||||||
|
}
|
||||||
|
|
||||||
|
void RestServer::start()
|
||||||
|
{
|
||||||
|
static const string ValidIpAddressRegex = "(?:[0-9]{1,3}\\.){3}[0-9]{1,3}";
|
||||||
|
static const string ValidPortRegex = "(?:[0-9]{1,4}|[1-5][0-9]{4}|6[0-4][0-9]{3}|65[0-4][0-9]{2}|655[0-2][0-9]|6553[0-5])$";
|
||||||
|
static const string ServersRegex = "^/api/servers(?:/(" + ValidIpAddressRegex + "\\:" + ValidPortRegex + "))?";
|
||||||
|
|
||||||
|
httpServer.resource[ServersRegex]["GET"] = [this](auto response, auto request) {
|
||||||
|
if(request->path_match[1].length() > 0)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
stringstream ss;
|
||||||
|
ss << "{";
|
||||||
|
auto addr = request->path_match[1].str();
|
||||||
|
auto port = (unsigned short)stoi(&(addr[addr.find(':')]));
|
||||||
|
queryToStringStream(ss, "server", serverMap->at(RakNet::SystemAddress(addr.c_str(), port)));
|
||||||
|
ss << "}";
|
||||||
|
ResponseStr(*response, ss.str(), "application/json");
|
||||||
|
}
|
||||||
|
catch(out_of_range e)
|
||||||
|
{
|
||||||
|
*response << response400;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
static string str;
|
||||||
|
|
||||||
|
//if(updatedCache)
|
||||||
|
{
|
||||||
|
stringstream ss;
|
||||||
|
ss << "{";
|
||||||
|
ss << "\"list servers\":{";
|
||||||
|
for (auto query = serverMap->begin(); query != serverMap->end(); query++)
|
||||||
|
{
|
||||||
|
queryToStringStream(ss, query->first.ToString(true, ':'), query->second);
|
||||||
|
if (next(query) != serverMap->end())
|
||||||
|
ss << ", ";
|
||||||
|
}
|
||||||
|
ss << "}}";
|
||||||
|
ResponseStr(*response, ss.str(), "application/json");
|
||||||
|
updatedCache = false;
|
||||||
|
}
|
||||||
|
*response << str;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
//Add query for < 0.6 servers
|
||||||
|
httpServer.resource[ServersRegex]["POST"] = [this](auto response, auto request) {
|
||||||
|
try
|
||||||
|
{
|
||||||
|
ptree pt;
|
||||||
|
read_json(request->content, pt);
|
||||||
|
|
||||||
|
MasterServer::SServer server;
|
||||||
|
ptreeToServer(pt, server);
|
||||||
|
|
||||||
|
unsigned short port = pt.get<unsigned short>("query_port");
|
||||||
|
serverMap->insert({RakNet::SystemAddress(request->remote_endpoint_address.c_str(), port), server});
|
||||||
|
updatedCache = true;
|
||||||
|
|
||||||
|
*response << response201;
|
||||||
|
}
|
||||||
|
catch (exception& e)
|
||||||
|
{
|
||||||
|
cout << e.what() << endl;
|
||||||
|
*response << response400;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
//Update query for < 0.6 servers
|
||||||
|
httpServer.resource[ServersRegex]["PUT"] = [this](auto response, auto request) {
|
||||||
|
auto addr = request->path_match[1].str();
|
||||||
|
auto port = (unsigned short)stoi(&(addr[addr.find(':')]));
|
||||||
|
|
||||||
|
auto query = serverMap->find(RakNet::SystemAddress(addr.c_str(), port));
|
||||||
|
|
||||||
|
if(query == serverMap->end())
|
||||||
|
{
|
||||||
|
cout << request->remote_endpoint_address + ": Trying to update a non-existent server or without permissions." << endl;
|
||||||
|
*response << response400;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(request->content.size() != 0)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
ptree pt;
|
||||||
|
read_json(request->content, pt);
|
||||||
|
|
||||||
|
ptreeToServer(pt, query->second);
|
||||||
|
|
||||||
|
updatedCache = true;
|
||||||
|
}
|
||||||
|
catch(exception &e)
|
||||||
|
{
|
||||||
|
cout << e.what() << endl;
|
||||||
|
*response << response400;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
query->second.lastUpdate = steady_clock::now();
|
||||||
|
|
||||||
|
*response << response202;
|
||||||
|
};
|
||||||
|
|
||||||
|
httpServer.resource["/api/servers/info"]["GET"] = [this](auto response, auto /*request*/) {
|
||||||
|
stringstream ss;
|
||||||
|
ss << '{';
|
||||||
|
ss << "servers: " << serverMap->size();
|
||||||
|
unsigned int players = 0;
|
||||||
|
for(auto s : *serverMap)
|
||||||
|
players += s.second.GetPlayers();
|
||||||
|
ss << ", players: " << players;
|
||||||
|
ss << "}";
|
||||||
|
|
||||||
|
ResponseStr(*response, ss.str(), "application/json");
|
||||||
|
};
|
||||||
|
|
||||||
|
httpServer.default_resource["GET"]=[](auto response, auto /*request*/) {
|
||||||
|
*response << response400;
|
||||||
|
};
|
||||||
|
|
||||||
|
httpServer.start();
|
||||||
|
}
|
||||||
|
|
||||||
|
void RestServer::cacheUpdated()
|
||||||
|
{
|
||||||
|
updatedCache = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void RestServer::stop()
|
||||||
|
{
|
||||||
|
httpServer.stop();
|
||||||
|
}
|
30
apps/master/RestServer.hpp
Normal file
30
apps/master/RestServer.hpp
Normal file
|
@ -0,0 +1,30 @@
|
||||||
|
//
|
||||||
|
// Created by koncord on 13.05.17.
|
||||||
|
//
|
||||||
|
|
||||||
|
#ifndef NEWRESTAPI_RESTSERVER_HPP
|
||||||
|
#define NEWRESTAPI_RESTSERVER_HPP
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <unordered_map>
|
||||||
|
#include "MasterServer.hpp"
|
||||||
|
#include "SimpleWeb/http_server.hpp"
|
||||||
|
|
||||||
|
typedef SimpleWeb::Server<SimpleWeb::HTTP> HttpServer;
|
||||||
|
|
||||||
|
class RestServer
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
RestServer(unsigned short port, MasterServer::ServerMap *pMap);
|
||||||
|
void start();
|
||||||
|
void stop();
|
||||||
|
void cacheUpdated();
|
||||||
|
|
||||||
|
private:
|
||||||
|
HttpServer httpServer;
|
||||||
|
MasterServer::ServerMap *serverMap;
|
||||||
|
bool updatedCache = true;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
#endif //NEWRESTAPI_RESTSERVER_HPP
|
511
apps/master/SimpleWeb/base_server.hpp
Normal file
511
apps/master/SimpleWeb/base_server.hpp
Normal file
|
@ -0,0 +1,511 @@
|
||||||
|
#ifndef BASE_SERVER_HPP
|
||||||
|
#define BASE_SERVER_HPP
|
||||||
|
|
||||||
|
#include <boost/asio.hpp>
|
||||||
|
#include <boost/algorithm/string/predicate.hpp>
|
||||||
|
#include <boost/functional/hash.hpp>
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
#include <unordered_map>
|
||||||
|
#include <thread>
|
||||||
|
#include <functional>
|
||||||
|
#include <iostream>
|
||||||
|
#include <sstream>
|
||||||
|
#include <regex>
|
||||||
|
|
||||||
|
#ifndef CASE_INSENSITIVE_EQUALS_AND_HASH
|
||||||
|
#define CASE_INSENSITIVE_EQUALS_AND_HASH
|
||||||
|
|
||||||
|
//Based on http://www.boost.org/doc/libs/1_60_0/doc/html/unordered/hash_equality.html
|
||||||
|
struct case_insensitive_equals
|
||||||
|
{
|
||||||
|
bool operator()(const std::string &key1, const std::string &key2) const
|
||||||
|
{
|
||||||
|
return boost::algorithm::iequals(key1, key2);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct case_insensitive_hash
|
||||||
|
{
|
||||||
|
size_t operator()(const std::string &key) const
|
||||||
|
{
|
||||||
|
std::size_t seed = 0;
|
||||||
|
for (auto &c: key)
|
||||||
|
boost::hash_combine(seed, std::tolower(c));
|
||||||
|
return seed;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
namespace SimpleWeb
|
||||||
|
{
|
||||||
|
template<class socket_type>
|
||||||
|
class Server;
|
||||||
|
|
||||||
|
template<class socket_type>
|
||||||
|
class ServerBase
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
virtual ~ServerBase()
|
||||||
|
{}
|
||||||
|
|
||||||
|
class Response : public std::ostream
|
||||||
|
{
|
||||||
|
friend class ServerBase<socket_type>;
|
||||||
|
|
||||||
|
boost::asio::streambuf streambuf;
|
||||||
|
|
||||||
|
std::shared_ptr<socket_type> socket;
|
||||||
|
|
||||||
|
Response(const std::shared_ptr<socket_type> &socket) : std::ostream(&streambuf), socket(socket)
|
||||||
|
{}
|
||||||
|
|
||||||
|
public:
|
||||||
|
size_t size()
|
||||||
|
{
|
||||||
|
return streambuf.size();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// If true, force server to close the connection after the response have been sent.
|
||||||
|
///
|
||||||
|
/// This is useful when implementing a HTTP/1.0-server sending content
|
||||||
|
/// without specifying the content length.
|
||||||
|
bool close_connection_after_response = false;
|
||||||
|
};
|
||||||
|
|
||||||
|
class Content : public std::istream
|
||||||
|
{
|
||||||
|
friend class ServerBase<socket_type>;
|
||||||
|
|
||||||
|
public:
|
||||||
|
size_t size()
|
||||||
|
{
|
||||||
|
return streambuf.size();
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string string()
|
||||||
|
{
|
||||||
|
std::stringstream ss;
|
||||||
|
ss << rdbuf();
|
||||||
|
return ss.str();
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
boost::asio::streambuf &streambuf;
|
||||||
|
|
||||||
|
Content(boost::asio::streambuf &streambuf) : std::istream(&streambuf), streambuf(streambuf)
|
||||||
|
{}
|
||||||
|
};
|
||||||
|
|
||||||
|
class Request
|
||||||
|
{
|
||||||
|
friend class ServerBase<socket_type>;
|
||||||
|
|
||||||
|
friend class Server<socket_type>;
|
||||||
|
|
||||||
|
public:
|
||||||
|
std::string method, path, http_version;
|
||||||
|
|
||||||
|
Content content;
|
||||||
|
|
||||||
|
std::unordered_multimap<std::string, std::string, case_insensitive_hash, case_insensitive_equals> header;
|
||||||
|
|
||||||
|
std::smatch path_match;
|
||||||
|
|
||||||
|
std::string remote_endpoint_address;
|
||||||
|
unsigned short remote_endpoint_port;
|
||||||
|
|
||||||
|
private:
|
||||||
|
Request(const socket_type &socket) : content(streambuf)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
remote_endpoint_address = socket.lowest_layer().remote_endpoint().address().to_string();
|
||||||
|
remote_endpoint_port = socket.lowest_layer().remote_endpoint().port();
|
||||||
|
}
|
||||||
|
catch (...)
|
||||||
|
{}
|
||||||
|
}
|
||||||
|
|
||||||
|
boost::asio::streambuf streambuf;
|
||||||
|
};
|
||||||
|
|
||||||
|
class Config
|
||||||
|
{
|
||||||
|
friend class ServerBase<socket_type>;
|
||||||
|
|
||||||
|
Config(unsigned short port) : port(port)
|
||||||
|
{}
|
||||||
|
|
||||||
|
public:
|
||||||
|
/// Port number to use. Defaults to 80 for HTTP and 443 for HTTPS.
|
||||||
|
unsigned short port;
|
||||||
|
/// Number of threads that the server will use when start() is called. Defaults to 1 thread.
|
||||||
|
size_t thread_pool_size = 1;
|
||||||
|
/// Timeout on request handling. Defaults to 5 seconds.
|
||||||
|
size_t timeout_request = 5;
|
||||||
|
/// Timeout on content handling. Defaults to 300 seconds.
|
||||||
|
size_t timeout_content = 300;
|
||||||
|
/// IPv4 address in dotted decimal form or IPv6 address in hexadecimal notation.
|
||||||
|
/// If empty, the address will be any address.
|
||||||
|
std::string address;
|
||||||
|
/// Set to false to avoid binding the socket to an address that is already in use. Defaults to true.
|
||||||
|
bool reuse_address = true;
|
||||||
|
};
|
||||||
|
|
||||||
|
///Set before calling start().
|
||||||
|
Config config;
|
||||||
|
|
||||||
|
private:
|
||||||
|
class regex_orderable : public std::regex
|
||||||
|
{
|
||||||
|
std::string str;
|
||||||
|
public:
|
||||||
|
regex_orderable(const char *regex_cstr) : std::regex(regex_cstr), str(regex_cstr)
|
||||||
|
{}
|
||||||
|
|
||||||
|
regex_orderable(const std::string ®ex_str) : std::regex(regex_str), str(regex_str)
|
||||||
|
{}
|
||||||
|
|
||||||
|
bool operator<(const regex_orderable &rhs) const
|
||||||
|
{
|
||||||
|
return str < rhs.str;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
/// Warning: do not add or remove resources after start() is called
|
||||||
|
std::map<regex_orderable, std::map<std::string,
|
||||||
|
std::function<void(std::shared_ptr<typename ServerBase<socket_type>::Response>,
|
||||||
|
std::shared_ptr<typename ServerBase<socket_type>::Request>)>>>
|
||||||
|
resource;
|
||||||
|
|
||||||
|
std::map<std::string,
|
||||||
|
std::function<void(std::shared_ptr<typename ServerBase<socket_type>::Response>,
|
||||||
|
std::shared_ptr<typename ServerBase<socket_type>::Request>)>> default_resource;
|
||||||
|
|
||||||
|
std::function<
|
||||||
|
void(std::shared_ptr<typename ServerBase<socket_type>::Request>,
|
||||||
|
const boost::system::error_code &)>
|
||||||
|
on_error;
|
||||||
|
|
||||||
|
std::function<void(std::shared_ptr<socket_type> socket,
|
||||||
|
std::shared_ptr<typename ServerBase<socket_type>::Request>)> on_upgrade;
|
||||||
|
|
||||||
|
virtual void start()
|
||||||
|
{
|
||||||
|
if (!io_service)
|
||||||
|
io_service = std::make_shared<boost::asio::io_service>();
|
||||||
|
|
||||||
|
if (io_service->stopped())
|
||||||
|
io_service->reset();
|
||||||
|
|
||||||
|
boost::asio::ip::tcp::endpoint endpoint;
|
||||||
|
if (config.address.size() > 0)
|
||||||
|
endpoint = boost::asio::ip::tcp::endpoint(boost::asio::ip::address::from_string(config.address),
|
||||||
|
config.port);
|
||||||
|
else
|
||||||
|
endpoint = boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), config.port);
|
||||||
|
|
||||||
|
if (!acceptor)
|
||||||
|
acceptor = std::unique_ptr<boost::asio::ip::tcp::acceptor>(
|
||||||
|
new boost::asio::ip::tcp::acceptor(*io_service));
|
||||||
|
acceptor->open(endpoint.protocol());
|
||||||
|
acceptor->set_option(boost::asio::socket_base::reuse_address(config.reuse_address));
|
||||||
|
acceptor->bind(endpoint);
|
||||||
|
acceptor->listen();
|
||||||
|
|
||||||
|
accept();
|
||||||
|
|
||||||
|
//If thread_pool_size>1, start m_io_service.run() in (thread_pool_size-1) threads for thread-pooling
|
||||||
|
threads.clear();
|
||||||
|
for (size_t c = 1; c < config.thread_pool_size; c++)
|
||||||
|
{
|
||||||
|
threads.emplace_back([this]()
|
||||||
|
{
|
||||||
|
io_service->run();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
//Main thread
|
||||||
|
if (config.thread_pool_size > 0)
|
||||||
|
io_service->run();
|
||||||
|
|
||||||
|
//Wait for the rest of the threads, if any, to finish as well
|
||||||
|
for (auto &t: threads)
|
||||||
|
{
|
||||||
|
t.join();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void stop()
|
||||||
|
{
|
||||||
|
acceptor->close();
|
||||||
|
if (config.thread_pool_size > 0)
|
||||||
|
io_service->stop();
|
||||||
|
}
|
||||||
|
|
||||||
|
///Use this function if you need to recursively send parts of a longer message
|
||||||
|
void send(const std::shared_ptr<Response> &response,
|
||||||
|
const std::function<void(const boost::system::error_code &)> &callback = nullptr) const
|
||||||
|
{
|
||||||
|
boost::asio::async_write(*response->socket, response->streambuf, [this, response, callback]
|
||||||
|
(const boost::system::error_code &ec, size_t /*bytes_transferred*/)
|
||||||
|
{
|
||||||
|
if (callback)
|
||||||
|
callback(ec);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/// If you have your own boost::asio::io_service, store its pointer here before running start().
|
||||||
|
/// You might also want to set config.thread_pool_size to 0.
|
||||||
|
std::shared_ptr<boost::asio::io_service> io_service;
|
||||||
|
protected:
|
||||||
|
std::unique_ptr<boost::asio::ip::tcp::acceptor> acceptor;
|
||||||
|
std::vector<std::thread> threads;
|
||||||
|
|
||||||
|
ServerBase(unsigned short port) : config(port)
|
||||||
|
{}
|
||||||
|
|
||||||
|
virtual void accept()=0;
|
||||||
|
|
||||||
|
std::shared_ptr<boost::asio::deadline_timer>
|
||||||
|
get_timeout_timer(const std::shared_ptr<socket_type> &socket, long seconds)
|
||||||
|
{
|
||||||
|
if (seconds == 0)
|
||||||
|
return nullptr;
|
||||||
|
|
||||||
|
auto timer = std::make_shared<boost::asio::deadline_timer>(*io_service);
|
||||||
|
timer->expires_from_now(boost::posix_time::seconds(seconds));
|
||||||
|
timer->async_wait([socket](const boost::system::error_code &ec)
|
||||||
|
{
|
||||||
|
if (!ec)
|
||||||
|
{
|
||||||
|
boost::system::error_code ec;
|
||||||
|
socket->lowest_layer().shutdown(boost::asio::ip::tcp::socket::shutdown_both, ec);
|
||||||
|
socket->lowest_layer().close();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return timer;
|
||||||
|
}
|
||||||
|
|
||||||
|
void read_request_and_content(const std::shared_ptr<socket_type> &socket)
|
||||||
|
{
|
||||||
|
//Create new streambuf (Request::streambuf) for async_read_until()
|
||||||
|
//shared_ptr is used to pass temporary objects to the asynchronous functions
|
||||||
|
std::shared_ptr<Request> request(new Request(*socket));
|
||||||
|
|
||||||
|
//Set timeout on the following boost::asio::async-read or write function
|
||||||
|
auto timer = this->get_timeout_timer(socket, config.timeout_request);
|
||||||
|
|
||||||
|
boost::asio::async_read_until(*socket, request->streambuf, "\r\n\r\n", [this, socket, request, timer]
|
||||||
|
(const boost::system::error_code &ec,
|
||||||
|
size_t bytes_transferred)
|
||||||
|
{
|
||||||
|
if (timer)
|
||||||
|
timer->cancel();
|
||||||
|
if (!ec)
|
||||||
|
{
|
||||||
|
//request->streambuf.size() is not necessarily the same as bytes_transferred, from Boost-docs:
|
||||||
|
//"After a successful async_read_until operation, the streambuf may contain additional data beyond the delimiter"
|
||||||
|
//The chosen solution is to extract lines from the stream directly when parsing the header. What is left of the
|
||||||
|
//streambuf (maybe some bytes of the content) is appended to in the async_read-function below (for retrieving content).
|
||||||
|
size_t num_additional_bytes =
|
||||||
|
request->streambuf.size() - bytes_transferred;
|
||||||
|
|
||||||
|
if (!this->parse_request(request))
|
||||||
|
return;
|
||||||
|
|
||||||
|
//If content, read that as well
|
||||||
|
auto it = request->header.find("Content-Length");
|
||||||
|
if (it != request->header.end())
|
||||||
|
{
|
||||||
|
unsigned long long content_length;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
content_length = stoull(it->second);
|
||||||
|
}
|
||||||
|
catch (const std::exception &e)
|
||||||
|
{
|
||||||
|
if (on_error)
|
||||||
|
on_error(request, boost::system::error_code(
|
||||||
|
boost::system::errc::protocol_error,
|
||||||
|
boost::system::generic_category()));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (content_length > num_additional_bytes)
|
||||||
|
{
|
||||||
|
//Set timeout on the following boost::asio::async-read or write function
|
||||||
|
auto timer = this->get_timeout_timer(socket,
|
||||||
|
config.timeout_content);
|
||||||
|
boost::asio::async_read(*socket, request->streambuf,
|
||||||
|
boost::asio::transfer_exactly(
|
||||||
|
content_length -
|
||||||
|
num_additional_bytes),
|
||||||
|
[this, socket, request, timer]
|
||||||
|
(const boost::system::error_code &ec,
|
||||||
|
size_t /*bytes_transferred*/)
|
||||||
|
{
|
||||||
|
if (timer)
|
||||||
|
timer->cancel();
|
||||||
|
if (!ec)
|
||||||
|
this->find_resource(socket,
|
||||||
|
request);
|
||||||
|
else if (on_error)
|
||||||
|
on_error(request, ec);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
else
|
||||||
|
this->find_resource(socket, request);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
this->find_resource(socket, request);
|
||||||
|
}
|
||||||
|
else if (on_error)
|
||||||
|
on_error(request, ec);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
bool parse_request(const std::shared_ptr<Request> &request) const
|
||||||
|
{
|
||||||
|
std::string line;
|
||||||
|
getline(request->content, line);
|
||||||
|
size_t method_end;
|
||||||
|
if ((method_end = line.find(' ')) != std::string::npos)
|
||||||
|
{
|
||||||
|
size_t path_end;
|
||||||
|
if ((path_end = line.find(' ', method_end + 1)) != std::string::npos)
|
||||||
|
{
|
||||||
|
request->method = line.substr(0, method_end);
|
||||||
|
request->path = line.substr(method_end + 1, path_end - method_end - 1);
|
||||||
|
|
||||||
|
size_t protocol_end;
|
||||||
|
if ((protocol_end = line.find('/', path_end + 1)) != std::string::npos)
|
||||||
|
{
|
||||||
|
if (line.compare(path_end + 1, protocol_end - path_end - 1, "HTTP") != 0)
|
||||||
|
return false;
|
||||||
|
request->http_version = line.substr(protocol_end + 1, line.size() - protocol_end - 2);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
return false;
|
||||||
|
|
||||||
|
getline(request->content, line);
|
||||||
|
size_t param_end;
|
||||||
|
while ((param_end = line.find(':')) != std::string::npos)
|
||||||
|
{
|
||||||
|
size_t value_start = param_end + 1;
|
||||||
|
if ((value_start) < line.size())
|
||||||
|
{
|
||||||
|
if (line[value_start] == ' ')
|
||||||
|
value_start++;
|
||||||
|
if (value_start < line.size())
|
||||||
|
request->header.emplace(line.substr(0, param_end),
|
||||||
|
line.substr(value_start, line.size() - value_start - 1));
|
||||||
|
}
|
||||||
|
|
||||||
|
getline(request->content, line);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
return false;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void find_resource(const std::shared_ptr<socket_type> &socket, const std::shared_ptr<Request> &request)
|
||||||
|
{
|
||||||
|
//Upgrade connection
|
||||||
|
if (on_upgrade)
|
||||||
|
{
|
||||||
|
auto it = request->header.find("Upgrade");
|
||||||
|
if (it != request->header.end())
|
||||||
|
{
|
||||||
|
on_upgrade(socket, request);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//Find path- and method-match, and call write_response
|
||||||
|
for (auto ®ex_method: resource)
|
||||||
|
{
|
||||||
|
auto it = regex_method.second.find(request->method);
|
||||||
|
if (it != regex_method.second.end())
|
||||||
|
{
|
||||||
|
std::smatch sm_res;
|
||||||
|
if (std::regex_match(request->path, sm_res, regex_method.first))
|
||||||
|
{
|
||||||
|
request->path_match = std::move(sm_res);
|
||||||
|
write_response(socket, request, it->second);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
auto it = default_resource.find(request->method);
|
||||||
|
if (it != default_resource.end())
|
||||||
|
{
|
||||||
|
write_response(socket, request, it->second);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void write_response(const std::shared_ptr<socket_type> &socket, const std::shared_ptr<Request> &request,
|
||||||
|
std::function<void(std::shared_ptr<typename ServerBase<socket_type>::Response>,
|
||||||
|
std::shared_ptr<
|
||||||
|
typename ServerBase<socket_type>::Request>)> &resource_function)
|
||||||
|
{
|
||||||
|
//Set timeout on the following boost::asio::async-read or write function
|
||||||
|
auto timer = this->get_timeout_timer(socket, config.timeout_content);
|
||||||
|
|
||||||
|
auto response = std::shared_ptr<Response>(new Response(socket), [this, request, timer]
|
||||||
|
(Response *response_ptr)
|
||||||
|
{
|
||||||
|
auto response = std::shared_ptr<Response>(response_ptr);
|
||||||
|
this->send(response, [this, response, request, timer](
|
||||||
|
const boost::system::error_code &ec)
|
||||||
|
{
|
||||||
|
if (timer)
|
||||||
|
timer->cancel();
|
||||||
|
if (!ec)
|
||||||
|
{
|
||||||
|
if (response->close_connection_after_response)
|
||||||
|
return;
|
||||||
|
|
||||||
|
auto range = request->header.equal_range(
|
||||||
|
"Connection");
|
||||||
|
for (auto it = range.first; it != range.second; it++)
|
||||||
|
{
|
||||||
|
if (boost::iequals(it->second, "close"))
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else if (boost::iequals(it->second, "keep-alive"))
|
||||||
|
{
|
||||||
|
this->read_request_and_content(
|
||||||
|
response->socket);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (request->http_version >= "1.1")
|
||||||
|
this->read_request_and_content(response->socket);
|
||||||
|
}
|
||||||
|
else if (on_error)
|
||||||
|
on_error(request, ec);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
resource_function(response, request);
|
||||||
|
}
|
||||||
|
catch (const std::exception &e)
|
||||||
|
{
|
||||||
|
if (on_error)
|
||||||
|
on_error(request, boost::system::error_code(boost::system::errc::operation_canceled,
|
||||||
|
boost::system::generic_category()));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
#endif //BASE_SERVER_HPP
|
55
apps/master/SimpleWeb/http_server.hpp
Normal file
55
apps/master/SimpleWeb/http_server.hpp
Normal file
|
@ -0,0 +1,55 @@
|
||||||
|
/*
|
||||||
|
* https://github.com/eidheim/Simple-Web-Server/
|
||||||
|
*
|
||||||
|
* The MIT License (MIT)
|
||||||
|
* Copyright (c) 2014-2016 Ole Christian Eidheim
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef SERVER_HTTP_HPP
|
||||||
|
#define SERVER_HTTP_HPP
|
||||||
|
|
||||||
|
#include "base_server.hpp"
|
||||||
|
|
||||||
|
namespace SimpleWeb
|
||||||
|
{
|
||||||
|
|
||||||
|
template<class socket_type>
|
||||||
|
class Server : public ServerBase<socket_type> {};
|
||||||
|
|
||||||
|
typedef boost::asio::ip::tcp::socket HTTP;
|
||||||
|
|
||||||
|
template<>
|
||||||
|
class Server<HTTP> : public ServerBase<HTTP>
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
Server() : ServerBase<HTTP>::ServerBase(80)
|
||||||
|
{}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
virtual void accept()
|
||||||
|
{
|
||||||
|
//Create new socket for this connection
|
||||||
|
//Shared_ptr is used to pass temporary objects to the asynchronous functions
|
||||||
|
auto socket = std::make_shared<HTTP>(*io_service);
|
||||||
|
|
||||||
|
acceptor->async_accept(*socket, [this, socket](const boost::system::error_code &ec)
|
||||||
|
{
|
||||||
|
//Immediately start accepting a new connection (if io_service hasn't been stopped)
|
||||||
|
if (ec != boost::asio::error::operation_aborted)
|
||||||
|
accept();
|
||||||
|
|
||||||
|
if (!ec)
|
||||||
|
{
|
||||||
|
boost::asio::ip::tcp::no_delay option(true);
|
||||||
|
socket->set_option(option);
|
||||||
|
|
||||||
|
this->read_request_and_content(socket);
|
||||||
|
}
|
||||||
|
else if (on_error)
|
||||||
|
on_error(std::shared_ptr<Request>(new Request(*socket)), ec);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif //SERVER_HTTP_HPP
|
91
apps/master/SimpleWeb/https_server.hpp
Normal file
91
apps/master/SimpleWeb/https_server.hpp
Normal file
|
@ -0,0 +1,91 @@
|
||||||
|
#ifndef HTTPS_SERVER_HPP
|
||||||
|
#define HTTPS_SERVER_HPP
|
||||||
|
|
||||||
|
#include "base_server.hpp"
|
||||||
|
#include <boost/asio/ssl.hpp>
|
||||||
|
#include <openssl/ssl.h>
|
||||||
|
#include <algorithm>
|
||||||
|
|
||||||
|
namespace SimpleWeb
|
||||||
|
{
|
||||||
|
typedef boost::asio::ssl::stream<boost::asio::ip::tcp::socket> HTTPS;
|
||||||
|
|
||||||
|
template<>
|
||||||
|
class Server<HTTPS> : public ServerBase<HTTPS>
|
||||||
|
{
|
||||||
|
std::string session_id_context;
|
||||||
|
bool set_session_id_context = false;
|
||||||
|
public:
|
||||||
|
Server(const std::string &cert_file, const std::string &private_key_file,
|
||||||
|
const std::string &verify_file = std::string()) : ServerBase<HTTPS>::ServerBase(443),
|
||||||
|
context(boost::asio::ssl::context::tlsv12)
|
||||||
|
{
|
||||||
|
context.use_certificate_chain_file(cert_file);
|
||||||
|
context.use_private_key_file(private_key_file, boost::asio::ssl::context::pem);
|
||||||
|
|
||||||
|
if (verify_file.size() > 0)
|
||||||
|
{
|
||||||
|
context.load_verify_file(verify_file);
|
||||||
|
context.set_verify_mode(boost::asio::ssl::verify_peer | boost::asio::ssl::verify_fail_if_no_peer_cert |
|
||||||
|
boost::asio::ssl::verify_client_once);
|
||||||
|
set_session_id_context = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void start()
|
||||||
|
{
|
||||||
|
if (set_session_id_context)
|
||||||
|
{
|
||||||
|
// Creating session_id_context from address:port but reversed due to small SSL_MAX_SSL_SESSION_ID_LENGTH
|
||||||
|
session_id_context = std::to_string(config.port) + ':';
|
||||||
|
session_id_context.append(config.address.rbegin(), config.address.rend());
|
||||||
|
SSL_CTX_set_session_id_context(context.native_handle(),
|
||||||
|
reinterpret_cast<const unsigned char *>(session_id_context.data()),
|
||||||
|
std::min<size_t>(session_id_context.size(),
|
||||||
|
SSL_MAX_SSL_SESSION_ID_LENGTH));
|
||||||
|
}
|
||||||
|
ServerBase::start();
|
||||||
|
}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
boost::asio::ssl::context context;
|
||||||
|
|
||||||
|
virtual void accept()
|
||||||
|
{
|
||||||
|
//Create new socket for this connection
|
||||||
|
//Shared_ptr is used to pass temporary objects to the asynchronous functions
|
||||||
|
auto socket = std::make_shared<HTTPS>(*io_service, context);
|
||||||
|
|
||||||
|
acceptor->async_accept((*socket).lowest_layer(), [this, socket](const boost::system::error_code &ec)
|
||||||
|
{
|
||||||
|
//Immediately start accepting a new connection (if io_service hasn't been stopped)
|
||||||
|
if (ec != boost::asio::error::operation_aborted)
|
||||||
|
accept();
|
||||||
|
|
||||||
|
|
||||||
|
if (!ec)
|
||||||
|
{
|
||||||
|
boost::asio::ip::tcp::no_delay option(true);
|
||||||
|
socket->lowest_layer().set_option(option);
|
||||||
|
|
||||||
|
//Set timeout on the following boost::asio::ssl::stream::async_handshake
|
||||||
|
auto timer = get_timeout_timer(socket, config.timeout_request);
|
||||||
|
socket->async_handshake(boost::asio::ssl::stream_base::server, [this, socket, timer]
|
||||||
|
(const boost::system::error_code &ec)
|
||||||
|
{
|
||||||
|
if (timer)
|
||||||
|
timer->cancel();
|
||||||
|
if (!ec)
|
||||||
|
read_request_and_content(socket);
|
||||||
|
else if (on_error)
|
||||||
|
on_error(std::shared_ptr<Request>(new Request(*socket)), ec);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
else if (on_error)
|
||||||
|
on_error(std::shared_ptr<Request>(new Request(*socket)), ec);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif //HTTPS_SERVER_HPP
|
Loading…
Reference in a new issue