DcruBro
afe10bbb6e
This commit adds common units tests and CI sanitasion.
Additional context for commit b64d9c4498:
- Fixed macOS/Linux non-portable and unsafe shell usage by adding a posix_spawn helper and replacing system() calls in virtual_interface.cpp.
- Fixed SessionRegistry::erase() to remove mIPSessions and mSessionIPs entries in session_registry.cpp.
- Prevented message-length truncation in tcp_message_handler.cpp by rejecting payloads > 65535 bytes.
- Validated handshake message sizes and removed silent truncation in:
- tcp_connection.cpp
- tcp_client.cpp
- Canonicalized and validated config and whitelist paths in utils.cpp using std::filesystem.
- Hardened environment-provided config path handling in main.cpp.
- Validated UDP ciphertext lengths and fixed session ID endianness in udp_client.cpp.
- Scheduled periodic SessionRegistry::cleanupExpired() in main.cpp (every 5 minutes).
50 lines
1.7 KiB
C++
50 lines
1.7 KiB
C++
// Simple unit tests for SessionRegistry
|
|
#include <cassert>
|
|
#include <iostream>
|
|
#include <chrono>
|
|
|
|
#include <columnlynx/common/net/session_registry.hpp>
|
|
#include <columnlynx/common/libsodium_wrapper.hpp>
|
|
|
|
int main() {
|
|
using namespace ColumnLynx::Net;
|
|
using namespace ColumnLynx::Utils;
|
|
|
|
auto ® = SessionRegistry::getInstance();
|
|
|
|
// Use a unique session id to avoid colliding with any running instance
|
|
const uint32_t sid = 0xDEADBEEF;
|
|
// ensure clean state
|
|
reg.erase(sid);
|
|
|
|
auto key = LibSodiumWrapper::generateRandom256Bit();
|
|
auto state = std::make_shared<SessionState>(key, std::chrono::hours(24), 0xC0A80101 /*192.168.1.1*/, 0, sid);
|
|
|
|
reg.put(sid, state);
|
|
|
|
assert(reg.exists(sid) && "Session should exist after put()");
|
|
|
|
auto got = reg.get(sid);
|
|
assert(got && got->sessionID == sid && "get() should return stored session");
|
|
|
|
auto byip = reg.getByIP(0xC0A80101);
|
|
assert(byip && byip->sessionID == sid && "getByIP() should find session by client IP");
|
|
|
|
// Erase and verify removed
|
|
reg.erase(sid);
|
|
assert(!reg.exists(sid) && "Session should not exist after erase()");
|
|
assert(reg.getByIP(0xC0A80101) == nullptr && "getByIP() should return nullptr after erase");
|
|
|
|
// Test cleanupExpired: insert an already-expired session
|
|
const uint32_t sid2 = 0xFEEDBEEF;
|
|
reg.erase(sid2);
|
|
auto expiredState = std::make_shared<SessionState>(key, std::chrono::seconds(0), 0xC0A80102, 0, sid2);
|
|
reg.put(sid2, expiredState);
|
|
// Force cleanup
|
|
reg.cleanupExpired();
|
|
assert(!reg.exists(sid2) && "Expired session should be removed by cleanupExpired()");
|
|
|
|
std::cout << "SessionRegistry tests passed\n";
|
|
return 0;
|
|
}
|