103 lines
3.8 KiB
C++
103 lines
3.8 KiB
C++
// main.cpp - Client entry point for ColumnLynx
|
|
// Copyright (C) 2025 DcruBro
|
|
// Distributed under the terms of the GNU General Public License, either version 2 only or version 3. See LICENSES/ for details.
|
|
|
|
#include <asio/asio.hpp>
|
|
#include <csignal>
|
|
#include <iostream>
|
|
#include <columnlynx/common/utils.hpp>
|
|
#include <columnlynx/common/panic_handler.hpp>
|
|
#include <columnlynx/client/net/tcp/tcp_client.hpp>
|
|
#include <columnlynx/client/net/udp/udp_client.hpp>
|
|
#include <cxxopts/cxxopts.hpp>
|
|
|
|
using asio::ip::tcp;
|
|
using namespace ColumnLynx::Utils;
|
|
|
|
volatile sig_atomic_t done = 0;
|
|
|
|
void signalHandler(int signum) {
|
|
if (signum == SIGINT || signum == SIGTERM) {
|
|
log("Received termination signal. Shutting down client.");
|
|
done = 1;
|
|
}
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
// Capture SIGINT and SIGTERM for graceful shutdown
|
|
struct sigaction action;
|
|
memset(&action, 0, sizeof(struct sigaction));
|
|
action.sa_handler = signalHandler;
|
|
sigaction(SIGINT, &action, nullptr);
|
|
sigaction(SIGTERM, &action, nullptr);
|
|
|
|
PanicHandler::init();
|
|
|
|
cxxopts::Options options("columnlynx_client", "ColumnLynx Client Application");
|
|
|
|
options.add_options()
|
|
("h,help", "Print help")
|
|
("s,server", "Server address", cxxopts::value<std::string>()->default_value("127.0.0.1"))
|
|
("p,port", "Server port", cxxopts::value<uint16_t>()->default_value(std::to_string(serverPort())))
|
|
("allow-selfsigned", "Allow self-signed certificates", cxxopts::value<bool>()->default_value("false"));
|
|
|
|
bool insecureMode = options.parse(argc, argv).count("allow-selfsigned") > 0;
|
|
|
|
auto result = options.parse(argc, argv);
|
|
if (result.count("help")) {
|
|
std::cout << options.help() << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
auto host = result["server"].as<std::string>();
|
|
auto port = std::to_string(result["port"].as<uint16_t>());
|
|
|
|
try {
|
|
log("ColumnLynx Client, Version " + getVersion());
|
|
log("This software is licensed under the GPLv2 only OR the GPLv3. See LICENSES/ for details.");
|
|
|
|
LibSodiumWrapper sodiumWrapper = LibSodiumWrapper();
|
|
|
|
std::array<uint8_t, 32> aesKey = {0}; // Defualt zeroed state until modified by handshake
|
|
uint64_t sessionID = 0;
|
|
|
|
asio::io_context io;
|
|
auto client = std::make_shared<ColumnLynx::Net::TCP::TCPClient>(io, host, port, &sodiumWrapper, &aesKey, &sessionID, &insecureMode);
|
|
auto udpClient = std::make_shared<ColumnLynx::Net::UDP::UDPClient>(io, host, port, &aesKey, &sessionID);
|
|
|
|
client->start();
|
|
udpClient->start();
|
|
|
|
// Run the IO context in a separate thread
|
|
std::thread ioThread([&io]() {
|
|
io.run();
|
|
});
|
|
ioThread.detach();
|
|
|
|
log("Client connected to " + host + ":" + port);
|
|
|
|
// Client is running
|
|
// TODO: SIGINT or SIGTERM seems to not kill this instantly!
|
|
while ((client->isConnected() || !client->isHandshakeComplete()) && !done) {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(100)); // Temp wait
|
|
|
|
if (client->isHandshakeComplete()) {
|
|
// Send a test UDP message every 5 seconds after handshake is complete
|
|
static auto lastSendTime = std::chrono::steady_clock::now();
|
|
auto now = std::chrono::steady_clock::now();
|
|
if (std::chrono::duration_cast<std::chrono::seconds>(now - lastSendTime).count() >= 5) {
|
|
udpClient->sendMessage("Hello from UDP client!");
|
|
lastSendTime = now;
|
|
}
|
|
}
|
|
}
|
|
log("Client shutting down.");
|
|
udpClient->stop();
|
|
client->disconnect();
|
|
io.stop();
|
|
ioThread.join();
|
|
|
|
} catch (const std::exception& e) {
|
|
error("Client error: " + std::string(e.what()));
|
|
}
|
|
} |