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).
44 lines
1.3 KiB
C++
44 lines
1.3 KiB
C++
// Tests for SessionRegistry IP allocation and lock/dealloc
|
|
#include <iostream>
|
|
#include <cassert>
|
|
|
|
#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();
|
|
|
|
const uint32_t sid = 0xABCDEF01;
|
|
reg.erase(sid);
|
|
|
|
auto key = LibSodiumWrapper::generateRandom256Bit();
|
|
auto state = std::make_shared<SessionState>(key, std::chrono::hours(24), 0, 0, sid);
|
|
reg.put(sid, state);
|
|
|
|
// Lock IP
|
|
uint32_t ip = 0xC0A80201; // 192.168.2.1
|
|
reg.lockIP(sid, ip);
|
|
|
|
auto byip = reg.getByIP(ip);
|
|
assert(byip && byip->sessionID == sid && "lockIP should populate mIPSessions");
|
|
|
|
// deallocIP
|
|
reg.deallocIP(sid);
|
|
assert(reg.getByIP(ip) == nullptr && "deallocIP should remove mapping");
|
|
|
|
// getFirstAvailableIP: choose a small /30 range to limit hosts
|
|
uint32_t base = 0x0A000000; // 10.0.0.0
|
|
uint8_t mask = 30; // 2 usable hosts
|
|
uint32_t first = reg.getFirstAvailableIP(base, mask);
|
|
assert(first != 0 && "Should find available IP in empty registry");
|
|
|
|
// cleanup
|
|
reg.erase(sid);
|
|
|
|
std::cout << "SessionRegistry IP tests passed\n";
|
|
return 0;
|
|
}
|