add interactive shell

Shell.cc

@@ -0,0 +1,182 @@
+#include "Shell.hh"
+
+#include <readline/history.h>
+#include <readline/readline.h>
+#include <stdio.h>
+
+#include <phosg/Strings.hh>
+
+using namespace std;
+
+
+
+class exit_shell : public runtime_error {
+public:
+  exit_shell() : runtime_error("shell exited") { }
+  ~exit_shell() = default;
+};
+
+
+
+void execute_command(shared_ptr<ServerState> state, 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 server\n\
+  reload <item> ...\n\
+    reload data. <item> can be licenses, battle-params, level-table, or quests.\n\
+  add-license <parameters>\n\
+    add a license to the server. <parameters> is some subset of the following:\n\
+      username=<username> (bb username)\n\
+      bb-password=<password> (bb password)\n\
+      gc-password=<password> (gc password)\n\
+      access-key=<access-key> (gc/pc access key)\n\
+      serial=<serial-number> (gc/pc serial number; required for all licenses)\n\
+      privileges=<privilege-mask> (can be normal, mod, admin, root, or numeric)\n\
+  delete-license <serial-number>\n\
+    delete a license from the server\n\
+  list-licenses\n\
+    list all licenses registered on the server\n\
+");
+
+  } else if (command_name == "reload") {
+    auto types = split(command_args, ' ');
+    for (const string& type : types) {
+      if (type == "licenses") {
+        shared_ptr<LicenseManager> lm(new LicenseManager("system/licenses.nsi"));
+        state->license_manager = lm;
+      } else if (type == "battle-params") {
+        shared_ptr<BattleParamTable> bpt(new BattleParamTable("system/blueburst/BattleParamEntry"));
+        state->battle_params = bpt;
+      } else if (type == "level-table") {
+        shared_ptr<LevelTable> lt(new LevelTable("system/blueburst/PlyLevelTbl.prs", true));
+        state->level_table = lt;
+      } else if (type == "quests") {
+        shared_ptr<QuestIndex> qi(new QuestIndex("system/quests"));
+        state->quest_index = qi;
+      } else {
+        throw invalid_argument("incorrect data type");
+      }
+    }
+
+  } else if (command_name == "add-license") {
+    shared_ptr<License> l(new License());
+    memset(l.get(), 0, sizeof(License));
+
+    for (const string& token : split(command_args, ' ')) {
+      if (starts_with(token, "username=")) {
+        if (token.size() >= 29) {
+          throw invalid_argument("username too long");
+        }
+        strcpy(l->username, token.c_str() + 9);
+
+      } else if (starts_with(token, "bb-password=")) {
+        if (token.size() >= 32) {
+          throw invalid_argument("bb-password too long");
+        }
+        strcpy(l->bb_password, token.c_str() + 12);
+
+      } else if (starts_with(token, "gc-password=")) {
+        if (token.size() > 20) {
+          throw invalid_argument("gc-password too long");
+        }
+        strcpy(l->gc_password, token.c_str() + 12);
+
+      } else if (starts_with(token, "access-key=")) {
+        if (token.size() > 23) {
+          throw invalid_argument("access-key is too long bytes");
+        }
+        strcpy(l->gc_password, token.c_str() + 11);
+
+      } else if (starts_with(token, "serial=")) {
+        l->serial_number = stoul(token.substr(7));
+
+      } else if (starts_with(token, "privileges=")) {
+        string mask = token.substr(11);
+        if (mask == "normal") {
+          l->privileges = 0;
+        } else if (mask == "mod") {
+          l->privileges = Privilege::Moderator;
+        } else if (mask == "admin") {
+          l->privileges = Privilege::Administrator;
+        } else if (mask == "root") {
+          l->privileges = Privilege::Root;
+        } else {
+          l->privileges = stoul(mask);
+        }
+
+      } else {
+        throw invalid_argument("incorrect field");
+      }
+    }
+
+    if (!l->serial_number) {
+      throw invalid_argument("license does not contain serial number");
+    }
+
+    state->license_manager->add(l);
+    fprintf(stderr, "license added\n");
+
+  } else if (command_name == "delete-license") {
+    uint32_t serial_number = stoul(command_args);
+    state->license_manager->remove(serial_number);
+    fprintf(stderr, "license deleted\n");
+
+  } else if (command_name == "list-licenses") {
+    for (const auto& l : state->license_manager->snapshot()) {
+      string s = l.str();
+      fprintf(stderr, "%s\n", s.c_str());
+    }
+
+  } else {
+    throw invalid_argument("unknown command; try \'help\'");
+  }
+}
+
+void run_shell(shared_ptr<ServerState> state) {
+  // initialize history
+  using_history();
+  // string history_filename = get_user_home_directory() + "/.newserv_history";
+  // read_history(history_filename.c_str());
+  // stifle_history(HISTORY_FILE_LENGTH);
+
+  // read and execute commands
+  bool should_continue = true;
+  while (should_continue) {
+
+    // read the command
+    char* command = readline("newserv> ");
+    if (!command) {
+      fprintf(stderr, " -- exit\n");
+      break;
+    }
+
+    // if there's a command, add it to the history
+    const char* command_to_execute = command + skip_whitespace(command, 0);
+    if (command_to_execute && *command_to_execute) {
+      add_history(command);
+    }
+
+    // dispatch the command
+    try {
+      execute_command(state, command_to_execute);
+    } catch (const exit_shell&) {
+      should_continue = false;
+    } catch (const exception& e) {
+      fprintf(stderr, "FAILED: %s\n", e.what());
+    }
+    free(command);
+  }
+}