Client.

Message encryption.
Packet handler.
Disconnect event handler.

2 files changed, 51 insertions, 0 deletions

diff --git a/server/src/util/xor.cpp b/server/src/util/xor.cpp
new file mode 100644
index 0000000..a00ecc9
--- /dev/null
+++ b/server/src/util/xor.cpp
@@ -0,0 +1,39 @@
+#include "../include.h"
+#include "xor.h"
+
+char enc::gen_key() {
+  std::random_device r;
+
+  std::default_random_engine e1(r());
+  std::uniform_real_distribution<> uniform_dist(0, 255);
+  return static_cast<char>(uniform_dist(e1));
+}
+
+void enc::encrypt_message(std::string &str) {
+  std::array<char, key_len> keys;
+  for (size_t i = 0; i < key_len; i++) {
+    keys[i] = gen_key();
+    str.insert(str.end(), keys[i]);
+  }
+
+  for (auto &key : keys) {
+    for (size_t i = 0; i < str.size() - key_len; i++) {
+      str[i] ^= key;
+    }
+  }
+}
+
+void enc::decrypt_message(std::string &str) {
+  if (str.size() <= key_len) return;
+
+  std::string keys = str.substr(0, key_len);
+  std::reverse(keys.begin(), keys.end());
+
+  for (auto &key : keys) {
+    for (size_t i = key_len; i < str.size(); i++) {
+      str[i] ^= key;
+    }
+  }
+
+  str.erase(str.begin(), str.begin() + key_len);
+}
\ No newline at end of file
diff --git a/server/src/util/xor.h b/server/src/util/xor.h
new file mode 100644
index 0000000..7180945
--- /dev/null
+++ b/server/src/util/xor.h
@@ -0,0 +1,12 @@
+#pragma once
+
+namespace enc {
+constexpr size_t key_len = 50;
+
+char gen_key();
+
+void encrypt_message(std::string &str);
+
+void decrypt_message(std::string &str);
+
+}  // namespace enc
\ No newline at end of file