Switched to C++23 as the project standard.
Added a basic parser for client_config and server_config, and added some basic authorization. Need to work on verification of the server.
This commit is contained in:
@@ -13,7 +13,7 @@ project(ColumnLynx
|
|||||||
# ---------------------------------------------------------
|
# ---------------------------------------------------------
|
||||||
# General C++ setup
|
# General C++ setup
|
||||||
# ---------------------------------------------------------
|
# ---------------------------------------------------------
|
||||||
set(CMAKE_CXX_STANDARD 20)
|
set(CMAKE_CXX_STANDARD 23)
|
||||||
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
||||||
set(CMAKE_CXX_EXTENSIONS OFF)
|
set(CMAKE_CXX_EXTENSIONS OFF)
|
||||||
#set(CMAKE_CXX_FLAGS_DEBUG "-g")
|
#set(CMAKE_CXX_FLAGS_DEBUG "-g")
|
||||||
|
|||||||
@@ -13,6 +13,7 @@
|
|||||||
#include <array>
|
#include <array>
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
#include <unordered_map>
|
||||||
#include <columnlynx/common/net/protocol_structs.hpp>
|
#include <columnlynx/common/net/protocol_structs.hpp>
|
||||||
#include <columnlynx/common/net/virtual_interface.hpp>
|
#include <columnlynx/common/net/virtual_interface.hpp>
|
||||||
|
|
||||||
@@ -42,7 +43,24 @@ namespace ColumnLynx::Net::TCP {
|
|||||||
mLastHeartbeatReceived(std::chrono::steady_clock::now()),
|
mLastHeartbeatReceived(std::chrono::steady_clock::now()),
|
||||||
mLastHeartbeatSent(std::chrono::steady_clock::now()),
|
mLastHeartbeatSent(std::chrono::steady_clock::now()),
|
||||||
mTun(tun)
|
mTun(tun)
|
||||||
{}
|
{
|
||||||
|
// Preload the config map
|
||||||
|
mRawClientConfig = Utils::getConfigMap("client_config");
|
||||||
|
|
||||||
|
if (!mRawClientConfig.empty()) {
|
||||||
|
Utils::debug("Loading the keys");
|
||||||
|
|
||||||
|
PrivateKey sk;
|
||||||
|
PublicKey pk;
|
||||||
|
std::copy_n(Utils::hexStringToBytes(mRawClientConfig.find("CLIENT_PRIVATE_KEY")->second).begin(), sk.size(), sk.begin()); // This is extremely stupid, but the C++ compiler has forced my hand (I would've just used to_array, but fucking asio decls)
|
||||||
|
std::copy_n(Utils::hexStringToBytes(mRawClientConfig.find("CLIENT_PUBLIC_KEY")->second).begin(), pk.size(), pk.begin());
|
||||||
|
|
||||||
|
mLibSodiumWrapper->setKeys(pk, sk);
|
||||||
|
|
||||||
|
Utils::debug("Newly-Loaded Public Key: " + Utils::bytesToHexString(mLibSodiumWrapper->getPublicKey(), 32));
|
||||||
|
Utils::debug("Newly-Loaded Private Key: " + Utils::bytesToHexString(mLibSodiumWrapper->getPrivateKey(), 64));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
void start();
|
void start();
|
||||||
void sendMessage(ClientMessageType type, const std::string& data = "");
|
void sendMessage(ClientMessageType type, const std::string& data = "");
|
||||||
@@ -76,5 +94,6 @@ namespace ColumnLynx::Net::TCP {
|
|||||||
bool mIsHostDomain;
|
bool mIsHostDomain;
|
||||||
Protocol::TunConfig mTunConfig;
|
Protocol::TunConfig mTunConfig;
|
||||||
std::shared_ptr<VirtualInterface> mTun = nullptr;
|
std::shared_ptr<VirtualInterface> mTun = nullptr;
|
||||||
|
std::unordered_map<std::string, std::string> mRawClientConfig;
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
@@ -40,6 +40,16 @@ namespace ColumnLynx::Utils {
|
|||||||
uint8_t* getXPublicKey() { return mXPublicKey.data(); }
|
uint8_t* getXPublicKey() { return mXPublicKey.data(); }
|
||||||
uint8_t* getXPrivateKey() { return mXPrivateKey.data(); }
|
uint8_t* getXPrivateKey() { return mXPrivateKey.data(); }
|
||||||
|
|
||||||
|
// Dangerous!
|
||||||
|
void setKeys(PublicKey pk, PrivateKey sk) {
|
||||||
|
mPublicKey = pk;
|
||||||
|
mPrivateKey = sk;
|
||||||
|
|
||||||
|
// Convert to Curve25519 keys for encryption
|
||||||
|
crypto_sign_ed25519_pk_to_curve25519(mXPublicKey.data(), mPublicKey.data());
|
||||||
|
crypto_sign_ed25519_sk_to_curve25519(mXPrivateKey.data(), mPrivateKey.data());
|
||||||
|
}
|
||||||
|
|
||||||
// Helper section
|
// Helper section
|
||||||
|
|
||||||
// Generates a random 256-bit (32-byte) array
|
// Generates a random 256-bit (32-byte) array
|
||||||
|
|||||||
@@ -11,6 +11,7 @@
|
|||||||
#include <sstream>
|
#include <sstream>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
|
#include <unordered_map>
|
||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
#include <winsock2.h>
|
#include <winsock2.h>
|
||||||
@@ -64,4 +65,7 @@ namespace ColumnLynx::Utils {
|
|||||||
inline constexpr uint64_t cbe64toh(uint64_t x) {
|
inline constexpr uint64_t cbe64toh(uint64_t x) {
|
||||||
return cbswap64(x); // big-endian -> host (for little-endian hosts)
|
return cbswap64(x); // big-endian -> host (for little-endian hosts)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Returns the config file in an unordered_map format. This purely reads the config file, you still need to parse it manually.
|
||||||
|
std::unordered_map<std::string, std::string> getConfigMap(std::string path);
|
||||||
};
|
};
|
||||||
@@ -31,6 +31,9 @@ namespace ColumnLynx::Net::TCP {
|
|||||||
mSodiumWrapper(sodiumWrapper),
|
mSodiumWrapper(sodiumWrapper),
|
||||||
mHostRunning(hostRunning)
|
mHostRunning(hostRunning)
|
||||||
{
|
{
|
||||||
|
// Preload the config map
|
||||||
|
mRawServerConfig = Utils::getConfigMap("server_config");
|
||||||
|
|
||||||
asio::error_code ec;
|
asio::error_code ec;
|
||||||
|
|
||||||
if (!ipv4Only) {
|
if (!ipv4Only) {
|
||||||
@@ -68,6 +71,7 @@ namespace ColumnLynx::Net::TCP {
|
|||||||
std::unordered_set<TCPConnection::pointer> mClients;
|
std::unordered_set<TCPConnection::pointer> mClients;
|
||||||
Utils::LibSodiumWrapper *mSodiumWrapper;
|
Utils::LibSodiumWrapper *mSodiumWrapper;
|
||||||
bool* mHostRunning;
|
bool* mHostRunning;
|
||||||
|
std::unordered_map<std::string, std::string> mRawServerConfig;
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
}
|
||||||
@@ -49,6 +49,9 @@ int main(int argc, char** argv) {
|
|||||||
auto result = options.parse(argc, argv);
|
auto result = options.parse(argc, argv);
|
||||||
if (result.count("help")) {
|
if (result.count("help")) {
|
||||||
std::cout << options.help() << std::endl;
|
std::cout << options.help() << std::endl;
|
||||||
|
std::cout << "This software is licensed under the GPLv2-only license OR the GPLv3 license.\n";
|
||||||
|
std::cout << "Copyright (C) 2025, The ColumnLynx Contributors.\n";
|
||||||
|
std::cout << "This software is provided under ABSOLUTELY NO WARRANTY, to the extent permitted by law.\n";
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -67,6 +70,8 @@ int main(int argc, char** argv) {
|
|||||||
log("Using virtual interface: " + tun->getName());
|
log("Using virtual interface: " + tun->getName());
|
||||||
|
|
||||||
LibSodiumWrapper sodiumWrapper = LibSodiumWrapper();
|
LibSodiumWrapper sodiumWrapper = LibSodiumWrapper();
|
||||||
|
debug("Public Key: " + Utils::bytesToHexString(sodiumWrapper.getPublicKey(), 32));
|
||||||
|
debug("Private Key: " + Utils::bytesToHexString(sodiumWrapper.getPrivateKey(), 64));
|
||||||
|
|
||||||
std::array<uint8_t, 32> aesKey = {0}; // Defualt zeroed state until modified by handshake
|
std::array<uint8_t, 32> aesKey = {0}; // Defualt zeroed state until modified by handshake
|
||||||
uint64_t sessionID = 0;
|
uint64_t sessionID = 0;
|
||||||
|
|||||||
@@ -33,9 +33,13 @@ namespace ColumnLynx::Net::TCP {
|
|||||||
std::vector<uint8_t> payload;
|
std::vector<uint8_t> payload;
|
||||||
payload.reserve(1 + crypto_box_PUBLICKEYBYTES);
|
payload.reserve(1 + crypto_box_PUBLICKEYBYTES);
|
||||||
payload.push_back(Utils::protocolVersion());
|
payload.push_back(Utils::protocolVersion());
|
||||||
payload.insert(payload.end(),
|
/*payload.insert(payload.end(),
|
||||||
mLibSodiumWrapper->getXPublicKey(),
|
mLibSodiumWrapper->getXPublicKey(),
|
||||||
mLibSodiumWrapper->getXPublicKey() + crypto_box_PUBLICKEYBYTES
|
mLibSodiumWrapper->getXPublicKey() + crypto_box_PUBLICKEYBYTES
|
||||||
|
);*/
|
||||||
|
payload.insert(payload.end(),
|
||||||
|
mLibSodiumWrapper->getPublicKey(),
|
||||||
|
mLibSodiumWrapper->getPublicKey() + crypto_sign_PUBLICKEYBYTES
|
||||||
);
|
);
|
||||||
|
|
||||||
mHandler->sendMessage(ClientMessageType::HANDSHAKE_INIT, Utils::uint8ArrayToString(payload.data(), payload.size()));
|
mHandler->sendMessage(ClientMessageType::HANDSHAKE_INIT, Utils::uint8ArrayToString(payload.data(), payload.size()));
|
||||||
|
|||||||
@@ -113,4 +113,34 @@ namespace ColumnLynx::Utils {
|
|||||||
|
|
||||||
return out;
|
return out;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::unordered_map<std::string, std::string> getConfigMap(std::string path) {
|
||||||
|
// TODO: Currently re-reads every time.
|
||||||
|
std::vector<std::string> readLines;
|
||||||
|
|
||||||
|
std::ifstream file(path);
|
||||||
|
std::string line;
|
||||||
|
|
||||||
|
while (std::getline(file, line)) {
|
||||||
|
readLines.push_back(line);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse them into the struct
|
||||||
|
std::unordered_map<std::string, std::string> config;
|
||||||
|
char delimiter = '=';
|
||||||
|
|
||||||
|
for (std::string str : readLines) {
|
||||||
|
std::stringstream ss(str);
|
||||||
|
|
||||||
|
std::string key;
|
||||||
|
std::string val;
|
||||||
|
|
||||||
|
std::getline(ss, key, delimiter);
|
||||||
|
std::getline(ss, val, delimiter);
|
||||||
|
|
||||||
|
config.insert({ key, val });
|
||||||
|
}
|
||||||
|
|
||||||
|
return config;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -50,6 +50,9 @@ int main(int argc, char** argv) {
|
|||||||
auto result = options.parse(argc, argv);
|
auto result = options.parse(argc, argv);
|
||||||
if (result.count("help")) {
|
if (result.count("help")) {
|
||||||
std::cout << options.help() << std::endl;
|
std::cout << options.help() << std::endl;
|
||||||
|
std::cout << "This software is licensed under the GPLv2-only license OR the GPLv3 license.\n";
|
||||||
|
std::cout << "Copyright (C) 2025, The ColumnLynx Contributors.\n";
|
||||||
|
std::cout << "This software is provided under ABSOLUTELY NO WARRANTY, to the extent permitted by law.\n";
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -114,15 +114,24 @@ namespace ColumnLynx::Net::TCP {
|
|||||||
|
|
||||||
Utils::log("Client protocol version " + std::to_string(clientProtoVer) + " accepted from " + reqAddr + ".");
|
Utils::log("Client protocol version " + std::to_string(clientProtoVer) + " accepted from " + reqAddr + ".");
|
||||||
|
|
||||||
std::memcpy(mConnectionPublicKey.data(), data.data() + 1, std::min(data.size() - 1, sizeof(mConnectionPublicKey))); // Store the client's public key (for identification)
|
PublicKey signPk;
|
||||||
|
std::memcpy(signPk.data(), data.data() + 1, std::min(data.size() - 1, sizeof(signPk))); // Store the client's public key (for identification)
|
||||||
|
|
||||||
|
crypto_sign_ed25519_pk_to_curve25519(mConnectionPublicKey.data(), signPk.data());
|
||||||
|
|
||||||
|
Utils::debug("Key attempted connect: " + Utils::bytesToHexString(signPk.data(), signPk.size()));
|
||||||
|
|
||||||
std::vector<std::string> whitelistedKeys = Utils::getWhitelistedKeys();
|
std::vector<std::string> whitelistedKeys = Utils::getWhitelistedKeys();
|
||||||
|
|
||||||
if (std::find(whitelistedKeys.begin(), whitelistedKeys.end(), Utils::bytesToHexString(mConnectionPublicKey.data(), mConnectionPublicKey.size())) == whitelistedKeys.end()) {
|
if (std::find(whitelistedKeys.begin(), whitelistedKeys.end(), Utils::bytesToHexString(signPk.data(), signPk.size())) == whitelistedKeys.end()) {
|
||||||
Utils::warn("Non-whitelisted client attempted to connect, terminating. Client IP: " + reqAddr);
|
Utils::warn("Non-whitelisted client attempted to connect, terminating. Client IP: " + reqAddr);
|
||||||
disconnect();
|
disconnect();
|
||||||
|
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Utils::debug("Client " + reqAddr + " passed authorized_keys");
|
||||||
|
|
||||||
mHandler->sendMessage(ServerMessageType::HANDSHAKE_IDENTIFY, Utils::uint8ArrayToString(mLibSodiumWrapper->getPublicKey(), crypto_sign_PUBLICKEYBYTES)); // This public key should always exist
|
mHandler->sendMessage(ServerMessageType::HANDSHAKE_IDENTIFY, Utils::uint8ArrayToString(mLibSodiumWrapper->getPublicKey(), crypto_sign_PUBLICKEYBYTES)); // This public key should always exist
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user