83 lines
2.2 KiB
C
83 lines
2.2 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <mutex>
|
||
|
|
#include <string>
|
||
|
|
#include <thread>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include <winsock2.h>
|
||
|
|
|
||
|
|
#include "util/rc4.h"
|
||
|
|
|
||
|
|
#include "module.h"
|
||
|
|
#include "websocket.h"
|
||
|
|
#include "serial.h"
|
||
|
|
|
||
|
|
namespace api {
|
||
|
|
|
||
|
|
struct ClientState {
|
||
|
|
SOCKADDR_IN address;
|
||
|
|
SOCKET socket;
|
||
|
|
bool close = false;
|
||
|
|
std::vector<Module*> modules;
|
||
|
|
std::string password;
|
||
|
|
bool password_change = false;
|
||
|
|
util::RC4 *cipher = nullptr;
|
||
|
|
};
|
||
|
|
|
||
|
|
class Controller {
|
||
|
|
private:
|
||
|
|
|
||
|
|
// configuration
|
||
|
|
const static int server_backlog = 16;
|
||
|
|
const static int server_receive_buffer_size = 64 * 1024;
|
||
|
|
const static int server_message_buffer_max_size = 64 * 1024;
|
||
|
|
const static int server_worker_count = 2;
|
||
|
|
const static int server_connection_limit = 4096;
|
||
|
|
|
||
|
|
// settings
|
||
|
|
unsigned short port;
|
||
|
|
std::string password;
|
||
|
|
bool pretty;
|
||
|
|
|
||
|
|
// server
|
||
|
|
WebSocketController *websocket;
|
||
|
|
std::vector<SerialController *> serial;
|
||
|
|
std::vector<std::thread> server_workers;
|
||
|
|
std::vector<std::thread> server_handlers;
|
||
|
|
std::mutex server_handlers_m;
|
||
|
|
std::vector<api::ClientState *> client_states;
|
||
|
|
std::mutex client_states_m;
|
||
|
|
SOCKET server;
|
||
|
|
void server_worker();
|
||
|
|
void connection_handler(ClientState client_state);
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
// state
|
||
|
|
bool server_running;
|
||
|
|
|
||
|
|
// constructor / destructor
|
||
|
|
Controller(unsigned short port, std::string password, bool pretty);
|
||
|
|
~Controller();
|
||
|
|
|
||
|
|
void listen_serial(std::string port, DWORD baud);
|
||
|
|
|
||
|
|
bool process_request(ClientState *state, std::vector<char> *in, std::vector<char> *out);
|
||
|
|
bool process_request(ClientState *state, const char *in, size_t in_size, std::vector<char> *out);
|
||
|
|
static void process_password_change(ClientState *state);
|
||
|
|
|
||
|
|
void init_state(ClientState *state);
|
||
|
|
static void free_state(ClientState *state);
|
||
|
|
|
||
|
|
void free_socket();
|
||
|
|
void obtain_client_states(std::vector<ClientState> *output);
|
||
|
|
|
||
|
|
std::string get_ip_address(sockaddr_in addr);
|
||
|
|
|
||
|
|
inline const std::string &get_password() const {
|
||
|
|
return this->password;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|