Martin Michelsen
72 lines
2.6 KiB
C++
72 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <event2/event.h>
|
|
|
|
#include <map>
|
|
#include <unordered_set>
|
|
#include <vector>
|
|
#include <string>
|
|
#include <memory>
|
|
|
|
#include "PSOEncryption.hh"
|
|
#include "PSOProtocol.hh"
|
|
|
|
|
|
|
|
class ProxyServer {
|
|
public:
|
|
ProxyServer() = delete;
|
|
ProxyServer(const ProxyServer&) = delete;
|
|
ProxyServer(ProxyServer&&) = delete;
|
|
ProxyServer(std::shared_ptr<struct event_base> base,
|
|
const struct sockaddr_storage& initial_destination, GameVersion version);
|
|
virtual ~ProxyServer() = default;
|
|
|
|
void listen(int port);
|
|
|
|
void send_to_client(const std::string& data);
|
|
void send_to_server(const std::string& data);
|
|
|
|
private:
|
|
std::shared_ptr<struct event_base> base;
|
|
std::map<int, std::unique_ptr<struct evconnlistener, void(*)(struct evconnlistener*)>> listeners;
|
|
std::unique_ptr<struct bufferevent, void(*)(struct bufferevent*)> client_bev;
|
|
std::unique_ptr<struct bufferevent, void(*)(struct bufferevent*)> server_bev;
|
|
struct sockaddr_storage next_destination;
|
|
int listen_port;
|
|
GameVersion version;
|
|
|
|
size_t header_size;
|
|
PSOCommandHeader client_input_header;
|
|
PSOCommandHeader server_input_header;
|
|
std::shared_ptr<PSOEncryption> client_input_crypt;
|
|
std::shared_ptr<PSOEncryption> client_output_crypt;
|
|
std::shared_ptr<PSOEncryption> server_input_crypt;
|
|
std::shared_ptr<PSOEncryption> server_output_crypt;
|
|
|
|
void send_to_end(const std::string& data, bool to_server);
|
|
|
|
static void dispatch_on_listen_accept(struct evconnlistener* listener,
|
|
evutil_socket_t fd, struct sockaddr *address, int socklen, void* ctx);
|
|
static void dispatch_on_listen_error(struct evconnlistener* listener, void* ctx);
|
|
static void dispatch_on_client_input(struct bufferevent* bev, void* ctx);
|
|
static void dispatch_on_client_error(struct bufferevent* bev, short events,
|
|
void* ctx);
|
|
static void dispatch_on_server_input(struct bufferevent* bev, void* ctx);
|
|
static void dispatch_on_server_error(struct bufferevent* bev, short events,
|
|
void* ctx);
|
|
|
|
void on_listen_accept(struct evconnlistener* listener, evutil_socket_t fd,
|
|
struct sockaddr *address, int socklen);
|
|
void on_listen_error(struct evconnlistener* listener);
|
|
void on_client_input(struct bufferevent* bev);
|
|
void on_client_error(struct bufferevent* bev, short events);
|
|
void on_server_input(struct bufferevent* bev);
|
|
void on_server_error(struct bufferevent* bev, short events);
|
|
|
|
size_t get_size_field(const PSOCommandHeader* header);
|
|
size_t get_command_field(const PSOCommandHeader* header);
|
|
|
|
void receive_and_process_commands(bool from_server);
|
|
};
|