switch to CMake

2021-12-29 11:51:17 -08:00
parent 6b5766449e
commit e8f23e4b2e
62 changed files with 253 additions and 326 deletions
@@ -0,0 +1,107 @@
+#include "ProxyShell.hh"
+
+#include <event2/event.h>
+#include <stdio.h>
+
+#include <phosg/Strings.hh>
+
+using namespace std;
+
+
+
+ProxyShell::ProxyShell(std::shared_ptr<struct event_base> base,
+    std::shared_ptr<ServerState> state,
+    std::shared_ptr<ProxyServer> proxy_server) : Shell(base, state),
+    proxy_server(proxy_server) { }
+
+void ProxyShell::execute_command(const string& command) {
+  // find the entry in the command table and run the command
+  size_t command_end = skip_non_whitespace(command, 0);
+  size_t args_begin = skip_whitespace(command, command_end);
+  string command_name = command.substr(0, command_end);
+  string command_args = command.substr(args_begin);
+
+  if (command_name == "exit") {
+    throw exit_shell();
+
+  } else if (command_name == "help") {
+    fprintf(stderr, "\
+commands:\n\
+  help\n\
+    you\'re reading it now\n\
+  exit (or ctrl+d)\n\
+    shut down the proxy\n\
+  sc <data>\n\
+    send a command to the client\n\
+  ss <data>\n\
+    send a command to the server\n\
+  chat <text>\n\
+    send a chat message to the server\n\
+  dchat <data>\n\
+    send a chat message to the server with arbitrary data in it\n\
+  marker <color-id>\n\
+    send a lobby marker message to the server\n\
+  event <event-id>\n\
+    send a lobby event update to yourself\n\
+");
+
+  } else if ((command_name == "sc") || (command_name == "ss")) {
+    bool to_client = (command_name[1] == 'c');
+    string data = parse_data_string(command_args);
+    if (data.size() & 3) {
+      throw invalid_argument("data size is not a multiple of 4");
+    }
+    if (data.size() == 0) {
+      throw invalid_argument("no data given");
+    }
+    uint16_t* size_field = reinterpret_cast<uint16_t*>(const_cast<char*>(data.data() + 2));
+    *size_field = data.size();
+
+    log(INFO, "%s (from proxy):", to_client ? "server" : "client");
+    print_data(stderr, data);
+
+    if (to_client) {
+      this->proxy_server->send_to_client(data);
+    } else {
+      this->proxy_server->send_to_server(data);
+    }
+
+  } else if ((command_name == "chat") || (command_name == "dchat")) {
+    string data(12, '\0');
+    data[0] = 0x06;
+    data.push_back('\x09');
+    data.push_back('E');
+    if (command_name == "dchat") {
+      data += parse_data_string(command_args);
+    } else {
+      data += command_args;
+    }
+    data.push_back('\0');
+    data.resize((data.size() + 3) & (~3));
+    uint16_t* size_field = reinterpret_cast<uint16_t*>(const_cast<char*>(data.data() + 2));
+    *size_field = data.size();
+
+    log(INFO, "client (from proxy):");
+    print_data(stderr, data);
+    this->proxy_server->send_to_server(data);
+
+  } else if (command_name == "marker") {
+    string data("\x89\x00\x04\x00", 4);
+    data[1] = stod(command_args);
+
+    log(INFO, "client (from proxy):");
+    print_data(stderr, data);
+    this->proxy_server->send_to_server(data);
+
+  } else if (command_name == "event") {
+    string data("\xDA\x00\x04\x00", 4);
+    data[1] = stod(command_args);
+
+    log(INFO, "server (from proxy):");
+    print_data(stderr, data);
+    this->proxy_server->send_to_client(data);
+
+  } else {
+    throw invalid_argument("unknown command; try \'help\'");
+  }
+}