m1ngsama/TNT
1
0
Fork 0
mirror of https://github.com/m1ngsama/TNT.git synced 2026-02-08 00:54:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

refactor: optimize rendering, log loading, and restructure tests

Browse source
This commit is contained in:
m1ngsama 2026-02-07 14:24:50 +08:00
parent aa2b842d03
commit bd4695b329
15 changed files with 285 additions and 110 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
.github/workflows/ci.yml vendored
View file

@ -33,11 +33,13 @@ jobs:
- name: Build with AddressSanitizer
run: make asan
- name: Run basic tests
- name: Run comprehensive tests
run: |
timeout 10 ./tnt &
sleep 2
pkill tnt || true
make test
cd tests
./test_security_features.sh
# Skipping anonymous access test in CI as it requires interactive pty handling which might be flaky
# ./test_anonymous_access.sh
- name: Check for memory leaks
if: runner.os == 'Linux'

6
Makefile
View file

@ -39,6 +39,7 @@ $(OBJ_DIR):
clean:
rm -rf $(OBJ_DIR) $(TARGET)
rm -f tests/*.log tests/host_key* tests/messages.log
@echo "Clean complete"
install: $(TARGET)
@ -69,6 +70,11 @@ check:
@command -v cppcheck >/dev/null 2>&1 && cppcheck --enable=warning,performance --quiet src/ || echo "cppcheck not installed"
@command -v clang-tidy >/dev/null 2>&1 && clang-tidy src/*.c -- -Iinclude $(INCLUDES) || echo "clang-tidy not installed"
# Test
test: all
@echo "Running tests..."
@cd tests && ./test_basic.sh
# Show build info
info:
@echo "Compiler: $(CC)"

4
README.md
View file

@ -96,8 +96,8 @@ tnt.service systemd unit
## Test
```sh
./test_basic.sh # functional
./test_stress.sh 50 # 50 clients
make test # run comprehensive test suite
# Individual tests are in tests/ directory
```
## Docs

10
TODO.md Normal file
View file

@ -0,0 +1,10 @@
# TODO
## Maintenance
- [ ] Replace deprecated `libssh` functions in `src/ssh_server.c`:
- `ssh_message_auth_password` (deprecated in newer libssh)
- `ssh_message_channel_request_pty_width`
- `ssh_message_channel_request_pty_height`
## Future Features
- [ ] Implement robust command handling for non-interactive SSH exec requests.

3
include/common.h
View file

@ -9,6 +9,9 @@
#include <time.h>
#include <pthread.h>
/* Project Metadata */
#define TNT_VERSION "1.0.0"
/* Configuration constants */
#define DEFAULT_PORT 2222
#define MAX_MESSAGES 100

1
include/ssh_server.h
View file

@ -21,6 +21,7 @@ typedef struct client {
bool show_help;
char command_input[256];
char command_output[2048];
char exec_command[256];
pthread_t thread;
bool connected;
int ref_count; /* Reference count for safe cleanup */

3
include/utf8.h
View file

@ -24,6 +24,9 @@ int utf8_strlen(const char *str);
/* Remove last UTF-8 character from string */
void utf8_remove_last_char(char *str);
/* Remove last word from string (mimic Ctrl+W) */
void utf8_remove_last_word(char *str);
/* Validate a UTF-8 byte sequence */
bool utf8_is_valid_sequence(const char *bytes, int len);

95
src/message.c
View file

@ -8,7 +8,7 @@ void message_init(void) {
/* Nothing to initialize for now */
}
/* Load messages from log file */
/* Load messages from log file - Optimized for large files */
int message_load(message_t **messages, int max_messages) {
/* Always allocate the message array */
message_t *msg_array = calloc(max_messages, sizeof(message_t));
@ -23,56 +23,75 @@ int message_load(message_t **messages, int max_messages) {
return 0;
}
char line[2048];
int count = 0;
/* Use a ring buffer approach - keep only last max_messages */
/* First, count total lines and seek to appropriate position */
/* Use dynamic allocation to handle large log files */
long *file_pos = NULL;
int pos_capacity = 1000;
int line_count = 0;
int start_index = 0;
/* Allocate initial position array */
file_pos = malloc(pos_capacity * sizeof(long));
if (!file_pos) {
/* Seek to end */
if (fseek(fp, 0, SEEK_END) != 0) {
fclose(fp);
*messages = msg_array;
return 0;
}
/* Record file positions */
while (fgets(line, sizeof(line), fp)) {
/* Expand array if needed */
if (line_count >= pos_capacity) {
int new_capacity = pos_capacity * 2;
long *new_pos = realloc(file_pos, new_capacity * sizeof(long));
if (!new_pos) {
/* Out of memory, stop scanning */
break;
}
file_pos = new_pos;
pos_capacity = new_capacity;
long file_size = ftell(fp);
if (file_size == 0) {
fclose(fp);
*messages = msg_array;
return 0;
}
/* Scan backwards to find the start position */
int newlines_found = 0;
long pos = file_size - 1;
/* Skip the very last byte if it's a newline */
if (pos >= 0) {
/* Read last char */
fseek(fp, pos, SEEK_SET);
if (fgetc(fp) == '\n') {
pos--;
}
file_pos[line_count++] = ftell(fp) - strlen(line);
}
/* Determine where to start reading */
if (line_count > max_messages) {
start_index = line_count - max_messages;
fseek(fp, file_pos[start_index], SEEK_SET);
} else {
fseek(fp, 0, SEEK_SET);
start_index = 0;
/* Read backwards in chunks for performance */
#define CHUNK_SIZE 4096
char chunk[CHUNK_SIZE];
while (pos >= 0 && newlines_found < max_messages) {
long read_size = (pos >= CHUNK_SIZE) ? CHUNK_SIZE : (pos + 1);
long read_pos = pos - read_size + 1;
fseek(fp, read_pos, SEEK_SET);
if (fread(chunk, 1, read_size, fp) != (size_t)read_size) {
break;
}
/* Scan chunk backwards */
for (int i = read_size - 1; i >= 0; i--) {
if (chunk[i] == '\n') {
newlines_found++;
if (newlines_found >= max_messages) {
/* Found our start point: one char after this newline */
fseek(fp, read_pos + i + 1, SEEK_SET);
goto read_messages;
}
}
}
pos -= read_size;
}
/* Now read the messages */
/* If we got here, we reached start of file or didn't find enough newlines */
fseek(fp, 0, SEEK_SET);
read_messages:;
char line[2048];
int count = 0;
/* Now read forward */
while (fgets(line, sizeof(line), fp) && count < max_messages) {
/* Check for oversized lines */
size_t line_len = strlen(line);
if (line_len >= sizeof(line) - 1) {
fprintf(stderr, "Warning: Skipping oversized line in messages.log\n");
/* Skip remainder of line */
int c;
while ((c = fgetc(fp)) != '\n' && c != EOF);
continue;
}
@ -109,7 +128,6 @@ int message_load(message_t **messages, int max_messages) {
time_t msg_time = mktime(&tm);
time_t now = time(NULL);
if (msg_time > now + 86400 || msg_time < now - 31536000 * 10) {
/* Skip messages more than 1 day in future or 10 years in past */
continue;
}
@ -121,7 +139,6 @@ int message_load(message_t **messages, int max_messages) {
count++;
}
free(file_pos);
fclose(fp);
*messages = msg_array;
return count;

87
src/ssh_server.c
View file

@ -14,6 +14,9 @@
#include <stdarg.h>
#include <sys/stat.h>
/* Suppress libssh deprecation warnings for legacy server API */
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
/* Global SSH bind instance */
static ssh_bind g_sshbind = NULL;
@ -425,7 +428,7 @@ static int read_username(client_t *client) {
if (pos < MAX_USERNAME_LEN - 1) {
username[pos++] = b;
username[pos] = '\0';
client_send(client, &b, 1);
client_send(client, (char *)&b, 1);
}
} else {
/* UTF-8 multi-byte */
@ -558,6 +561,20 @@ static void execute_command(client_t *client) {
/* Handle client key press - returns true if key was consumed */
static bool handle_key(client_t *client, unsigned char key, char *input) {
/* Handle Ctrl+C (Exit or switch to NORMAL) */
if (key == 3) {
if (client->mode != MODE_NORMAL) {
client->mode = MODE_NORMAL;
client->command_input[0] = '\0';
client->show_help = false;
tui_render_screen(client);
} else {
/* In NORMAL mode, Ctrl+C exits */
client->connected = false;
}
return true;
}
/* Handle help screen */
if (client->show_help) {
if (key == 'q' || key == 27) {
@ -603,7 +620,7 @@ static bool handle_key(client_t *client, unsigned char key, char *input) {
client->scroll_pos = 0;
tui_render_screen(client);
return true; /* Key consumed */
} else if (key == '\r') { /* Enter */
} else if (key == '\r' || key == '\n') { /* Enter */
if (input[0] != '\0') {
message_t msg = {
.timestamp = time(NULL),
@ -622,6 +639,18 @@ static bool handle_key(client_t *client, unsigned char key, char *input) {
tui_render_input(client, input);
}
return true; /* Key consumed */
} else if (key == 23) { /* Ctrl+W (Delete Word) */
if (input[0] != '\0') {
utf8_remove_last_word(input);
tui_render_input(client, input);
}
return true;
} else if (key == 21) { /* Ctrl+U (Delete Line) */
if (input[0] != '\0') {
input[0] = '\0';
tui_render_input(client, input);
}
return true;
}
break;
@ -690,6 +719,18 @@ static bool handle_key(client_t *client, unsigned char key, char *input) {
tui_render_screen(client);
}
return true; /* Key consumed */
} else if (key == 23) { /* Ctrl+W (Delete Word) */
if (client->command_input[0] != '\0') {
utf8_remove_last_word(client->command_input);
tui_render_screen(client);
}
return true;
} else if (key == 21) { /* Ctrl+U (Delete Line) */
if (client->command_input[0] != '\0') {
client->command_input[0] = '\0';
tui_render_screen(client);
}
return true;
}
break;
@ -711,6 +752,20 @@ void* client_handle_session(void *arg) {
client->help_lang = LANG_ZH;
client->connected = true;
/* Check for exec command */
if (client->exec_command[0] != '\0') {
if (strcmp(client->exec_command, "exit") == 0) {
/* Just exit */
ssh_channel_request_send_exit_status(client->channel, 0);
goto cleanup;
} else {
/* Unknown command */
client_printf(client, "Command not supported: %s\r\nOnly 'exit' is supported in non-interactive mode.\r\n", client->exec_command);
ssh_channel_request_send_exit_status(client->channel, 1);
goto cleanup;
}
}
/* Read username */
if (read_username(client) < 0) {
goto cleanup;
@ -759,9 +814,6 @@ void* client_handle_session(void *arg) {
unsigned char b = buf[0];
/* Ctrl+C */
if (b == 3) break;
/* Handle special keys - returns true if key was consumed */
bool key_consumed = handle_key(client, b, input);
@ -933,7 +985,6 @@ static ssh_channel handle_channel_open(ssh_session session) {
/* Handle PTY request and get terminal size */
static int handle_pty_request(ssh_channel channel, client_t *client) {
ssh_message message;
int pty_received = 0;
int shell_received = 0;
do {
@ -952,9 +1003,8 @@ static int handle_pty_request(ssh_channel channel, client_t *client) {
ssh_message_channel_request_reply_success(message);
ssh_message_free(message);
pty_received = 1;
/* Don't return yet, wait for shell request */
/* Don't return yet, wait for shell/exec request */
if (shell_received) {
return 0;
}
@ -964,12 +1014,21 @@ static int handle_pty_request(ssh_channel channel, client_t *client) {
ssh_message_channel_request_reply_success(message);
ssh_message_free(message);
shell_received = 1;
/* Shell requested, we are done */
return 0;
/* If we got PTY, we're done */
if (pty_received) {
return 0;
} else if (ssh_message_subtype(message) == SSH_CHANNEL_REQUEST_EXEC) {
/* Handle exec request (e.g. "ssh user@host command") */
const char *cmd = ssh_message_channel_request_command(message);
if (cmd) {
strncpy(client->exec_command, cmd, sizeof(client->exec_command) - 1);
client->exec_command[sizeof(client->exec_command) - 1] = '\0';
}
continue;
/* For now, just allow it and treat like shell start */
ssh_message_channel_request_reply_success(message);
ssh_message_free(message);
shell_received = 1;
return 0;
} else if (ssh_message_subtype(message) == SSH_CHANNEL_REQUEST_WINDOW_CHANGE) {
/* Handle terminal resize - this should be handled during session, not here */
@ -982,9 +1041,9 @@ static int handle_pty_request(ssh_channel channel, client_t *client) {
ssh_message_reply_default(message);
ssh_message_free(message);
} while (!pty_received || !shell_received);
} while (!shell_received);
return (pty_received && shell_received) ? 0 : -1;
return 0;
}
/* Initialize SSH server */

34
src/tui.c
View file

@ -61,8 +61,8 @@ void tui_render_screen(client_t *client) {
/* Now render using snapshot (no lock held) */
/* Clear and move to top */
pos += snprintf(buffer + pos, sizeof(buffer) - pos, ANSI_CLEAR ANSI_HOME);
/* Move to top (Home) - Do NOT clear screen to prevent flicker */
pos += snprintf(buffer + pos, sizeof(buffer) - pos, ANSI_HOME);
/* Title bar */
const char *mode_str = (client->mode == MODE_INSERT) ? "INSERT" :
@ -82,22 +82,21 @@ void tui_render_screen(client_t *client) {
for (int i = 0; i < padding; i++) {
buffer[pos++] = ' ';
}
pos += snprintf(buffer + pos, sizeof(buffer) - pos, ANSI_RESET "\r\n");
pos += snprintf(buffer + pos, sizeof(buffer) - pos, ANSI_RESET "\033[K\r\n");
/* Render messages from snapshot */
if (msg_snapshot) {
for (int i = 0; i < snapshot_count; i++) {
char msg_line[1024];
message_format(&msg_snapshot[i], msg_line, sizeof(msg_line), client->width);
pos += snprintf(buffer + pos, sizeof(buffer) - pos, "%s\r\n", msg_line);
pos += snprintf(buffer + pos, sizeof(buffer) - pos, "%s\033[K\r\n", msg_line);
}
free(msg_snapshot);
}
/* Fill empty lines */
/* Fill empty lines and clear them */
for (int i = snapshot_count; i < msg_height; i++) {
buffer[pos++] = '\r';
buffer[pos++] = '\n';
pos += snprintf(buffer + pos, sizeof(buffer) - pos, "\033[K\r\n");
}
/* Separator - use box drawing character */
@ -107,21 +106,20 @@ void tui_render_screen(client_t *client) {
memcpy(buffer + pos, line_char, len);
pos += len;
}
buffer[pos++] = '\r';
buffer[pos++] = '\n';
pos += snprintf(buffer + pos, sizeof(buffer) - pos, "\033[K\r\n");
/* Status/Input line */
if (client->mode == MODE_INSERT) {
pos += snprintf(buffer + pos, sizeof(buffer) - pos, "> ");
pos += snprintf(buffer + pos, sizeof(buffer) - pos, "> \033[K");
} else if (client->mode == MODE_NORMAL) {
int total = msg_count;
int scroll_pos = client->scroll_pos + 1;
if (total == 0) scroll_pos = 0;
pos += snprintf(buffer + pos, sizeof(buffer) - pos,
"-- NORMAL -- (%d/%d)", scroll_pos, total);
"-- NORMAL -- (%d/%d)\033[K", scroll_pos, total);
} else if (client->mode == MODE_COMMAND) {
pos += snprintf(buffer + pos, sizeof(buffer) - pos,
":%s", client->command_input);
":%s\033[K", client->command_input);
}
client_send(client, buffer, pos);
@ -224,7 +222,9 @@ const char* tui_get_help_text(help_lang_t lang) {
" ESC - Enter NORMAL mode\n"
" Enter - Send message\n"
" Backspace - Delete character\n"
" Ctrl+C - Exit chat\n"
" Ctrl+W - Delete last word\n"
" Ctrl+U - Delete line\n"
" Ctrl+C - Enter NORMAL mode\n"
"\n"
"NORMAL MODE KEYS:\n"
" i - Return to INSERT mode\n"
@ -240,6 +240,8 @@ const char* tui_get_help_text(help_lang_t lang) {
" Enter - Execute command\n"
" ESC - Cancel, return to NORMAL\n"
" Backspace - Delete character\n"
" Ctrl+W - Delete last word\n"
" Ctrl+U - Delete line\n"
"\n"
"AVAILABLE COMMANDS:\n"
" list, users, who - Show online users\n"
@ -266,7 +268,9 @@ const char* tui_get_help_text(help_lang_t lang) {
" ESC - 进入 NORMAL 模式\n"
" Enter - 发送消息\n"
" Backspace - 删除字符\n"
" Ctrl+C - 退出聊天\n"
" Ctrl+W - 删除上个单词\n"
" Ctrl+U - 删除整行\n"
" Ctrl+C - 进入 NORMAL 模式\n"
"\n"
"NORMAL 模式按键:\n"
" i - 返回 INSERT 模式\n"
@ -282,6 +286,8 @@ const char* tui_get_help_text(help_lang_t lang) {
" Enter - 执行命令\n"
" ESC - 取消,返回 NORMAL 模式\n"
" Backspace - 删除字符\n"
" Ctrl+W - 删除上个单词\n"
" Ctrl+U - 删除整行\n"
"\n"
"可用命令:\n"
" list, users, who - 显示在线用户\n"

20
src/utf8.c
View file

@ -136,6 +136,26 @@ void utf8_remove_last_char(char *str) {
str[i] = '\0';
}
/* Remove last word from string (mimic Ctrl+W) */
void utf8_remove_last_word(char *str) {
int len = strlen(str);
if (len == 0) return;
int i = len;
/* Skip trailing spaces */
while (i > 0 && str[i - 1] == ' ') {
i--;
}
/* Skip non-spaces (the word) */
while (i > 0 && str[i - 1] != ' ') {
i--;
}
str[i] = '\0';
}
/* Validate a UTF-8 byte sequence */
bool utf8_is_valid_sequence(const char *bytes, int len) {
if (len <= 0 || len > 4 || !bytes) {

56
test_anonymous_access.sh → tests/test_anonymous_access.sh
View file

@ -1,20 +1,39 @@
#!/bin/bash
# Test anonymous SSH access
BIN="../tnt"
PORT=${PORT:-2222}
if [ ! -f "$BIN" ]; then
echo "Error: Binary $BIN not found."
exit 1
fi
echo "Starting TNT server on port $PORT..."
$BIN -p $PORT > /dev/null 2>&1 &
SERVER_PID=$!
sleep 2
cleanup() {
kill $SERVER_PID 2>/dev/null
wait 2>/dev/null
}
trap cleanup EXIT
echo "Testing anonymous SSH access to TNT server..."
echo ""
# Test 1: Connection with any username and password
echo "Test 1: Connection with any username (should succeed)"
timeout 5 expect -c '
spawn ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -p 2223 testuser@localhost
timeout 10 expect -c "
spawn ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -p $PORT testuser@localhost
expect {
"password:" {
send "anypassword\r"
\"password:\" {
send \"anypassword\r\"
expect {
"请输入用户名:" {
send "TestUser\r"
send "\003"
\"请输入用户名\" {
send \"TestUser\r\"
send \"\003\"
exit 0
}
timeout { exit 1 }
@ -22,27 +41,28 @@ expect {
}
timeout { exit 1 }
}
' 2>&1 | grep -q "请输入用户名"
" 2>&1 | grep -q "请输入用户名"
if [ $? -eq 0 ]; then
echo "✓ Test 1 PASSED: Can connect with any password"
else
echo "✗ Test 1 FAILED: Cannot connect with any password"
exit 1
fi
echo ""
# Test 2: Connection should work without special SSH options
# Test 2: Connection should work with empty password
echo "Test 2: Simple connection (standard SSH command)"
timeout 5 expect -c '
spawn ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -p 2223 anonymous@localhost
timeout 10 expect -c "
spawn ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -p $PORT anonymous@localhost
expect {
"password:" {
send "\r"
\"password:\" {
send \"\r\"
expect {
"请输入用户名:" {
send "\r"
send "\003"
\"请输入用户名\" {
send \"\r\"
send \"\003\"
exit 0
}
timeout { exit 1 }
@ -50,13 +70,15 @@ expect {
}
timeout { exit 1 }
}
' 2>&1 | grep -q "请输入用户名"
" 2>&1 | grep -q "请输入用户名"
if [ $? -eq 0 ]; then
echo "✓ Test 2 PASSED: Can connect with empty password"
else
echo "✗ Test 2 FAILED: Cannot connect with empty password"
exit 1
fi
echo ""
echo "Anonymous access test completed."
exit 0

12
test_basic.sh → tests/test_basic.sh
View file

@ -15,8 +15,16 @@ trap cleanup EXIT
echo "=== TNT Basic Tests ==="
# Path to binary
BIN="../tnt"
if [ ! -f "$BIN" ]; then
echo "Error: Binary $BIN not found. Run make first."
exit 1
fi
# Start server
./tnt -p $PORT >test.log 2>&1 &
$BIN -p $PORT >test.log 2>&1 &
SERVER_PID=$!
sleep 2
@ -41,7 +49,7 @@ else
fi
# Test 3: Message logging
echo "test message" | timeout 5 ssh -o StrictHostKeyChecking=no \
(echo "testuser"; echo "test message"; sleep 1) | timeout 5 ssh -o StrictHostKeyChecking=no \
-o UserKnownHostsFile=/dev/null -p $PORT localhost >/dev/null 2>&1 &
sleep 3
if [ -f messages.log ]; then

38
test_security_features.sh → tests/test_security_features.sh
View file

@ -27,7 +27,7 @@ fail() {
}
cleanup() {
pkill -f "^\./tnt" 2>/dev/null || true
pkill -f "^\.\./tnt" 2>/dev/null || true
sleep 1
}
@ -37,10 +37,16 @@ echo -e "${YELLOW}========================================${NC}"
echo -e "${YELLOW}TNT Security Features Test Suite${NC}"
echo -e "${YELLOW}========================================${NC}"
BIN="../tnt"
if [ ! -f "$BIN" ]; then
echo "Error: Binary $BIN not found."
exit 1
fi
# Test 1: 4096-bit RSA Key Generation
print_test "1. RSA 4096-bit Key Generation"
rm -f host_key
./tnt &
$BIN &
PID=$!
sleep 8 # Wait for key generation
kill $PID 2>/dev/null || true
@ -69,19 +75,19 @@ fi
print_test "2. Environment Variable Configuration"
# Test bind address
TNT_BIND_ADDR=127.0.0.1 timeout 3 ./tnt 2>&1 | grep -q "TNT chat server" && \
TNT_BIND_ADDR=127.0.0.1 timeout 3 $BIN 2>&1 | grep -q "TNT chat server" && \
pass "TNT_BIND_ADDR configuration works" || fail "TNT_BIND_ADDR not working"
# Test with access token set (just verify it starts)
TNT_ACCESS_TOKEN="test123" timeout 3 ./tnt 2>&1 | grep -q "TNT chat server" && \
TNT_ACCESS_TOKEN="test123" timeout 3 $BIN 2>&1 | grep -q "TNT chat server" && \
pass "TNT_ACCESS_TOKEN configuration accepted" || fail "TNT_ACCESS_TOKEN not working"
# Test max connections configuration
TNT_MAX_CONNECTIONS=10 timeout 3 ./tnt 2>&1 | grep -q "TNT chat server" && \
TNT_MAX_CONNECTIONS=10 timeout 3 $BIN 2>&1 | grep -q "TNT chat server" && \
pass "TNT_MAX_CONNECTIONS configuration accepted" || fail "TNT_MAX_CONNECTIONS not working"
# Test rate limit toggle
TNT_RATE_LIMIT=0 timeout 3 ./tnt 2>&1 | grep -q "TNT chat server" && \
TNT_RATE_LIMIT=0 timeout 3 $BIN 2>&1 | grep -q "TNT chat server" && \
pass "TNT_RATE_LIMIT configuration accepted" || fail "TNT_RATE_LIMIT not working"
sleep 1
@ -101,7 +107,7 @@ newline
EOF
# Start server and let it load messages
./tnt &
$BIN &
PID=$!
sleep 3
kill $PID 2>/dev/null || true
@ -120,8 +126,8 @@ print_test "4. UTF-8 Input Validation"
cat > test_utf8.c <<'EOF'
#include <stdio.h>
#include <stdbool.h>
#include "include/utf8.h"
#include "include/common.h"
#include "../include/utf8.h"
#include "../include/common.h"
int main() {
// Valid UTF-8 sequences
@ -154,7 +160,7 @@ int main() {
}
EOF
if gcc -I. -o test_utf8 test_utf8.c src/utf8.c 2>/dev/null; then
if gcc -I../include -o test_utf8 test_utf8.c ../src/utf8.c 2>/dev/null; then
if ./test_utf8; then
pass "UTF-8 validation function works correctly"
else
@ -168,12 +174,12 @@ rm -f test_utf8.c
# Test 5: Buffer Safety with AddressSanitizer
print_test "5. Buffer Overflow Protection (ASAN Build)"
if make clean >/dev/null 2>&1 && make asan >/dev/null 2>&1; then
if make -C .. clean >/dev/null 2>&1 && make -C .. asan >/dev/null 2>&1; then
# Just verify it compiles - actual ASAN testing needs runtime
if [ -f tnt ]; then
if [ -f ../tnt ]; then
pass "AddressSanitizer build successful"
# Restore normal build
make clean >/dev/null 2>&1 && make >/dev/null 2>&1
make -C .. clean >/dev/null 2>&1 && make -C .. >/dev/null 2>&1
else
fail "AddressSanitizer build failed"
fi
@ -184,8 +190,8 @@ fi
# Test 6: Concurrent Safety
print_test "6. Concurrency Safety (Data Structure Integrity)"
# This test verifies the code compiles with thread sanitizer flags
if gcc -fsanitize=thread -g -O1 -Iinclude -I/opt/homebrew/opt/libssh/include \
-c src/chat_room.c -o /tmp/test_tsan.o 2>/dev/null; then
if gcc -fsanitize=thread -g -O1 -I../include -I/opt/homebrew/opt/libssh/include \
-c ../src/chat_room.c -o /tmp/test_tsan.o 2>/dev/null; then
pass "Code compiles with ThreadSanitizer (concurrency checks enabled)"
rm -f /tmp/test_tsan.o
else
@ -200,7 +206,7 @@ for i in $(seq 1 2000); do
echo "2026-01-22T$(printf "%02d" $((i/100))):$(printf "%02d" $((i%60))):00Z|user$i|message $i" >> messages.log
done
./tnt &
$BIN &
PID=$!
sleep 4
kill $PID 2>/dev/null || true

16
test_stress.sh → tests/test_stress.sh
View file

@ -5,9 +5,15 @@
PORT=${PORT:-2222}
CLIENTS=${1:-10}
DURATION=${2:-30}
BIN="../tnt"
if [ ! -f "$BIN" ]; then
echo "Error: Binary $BIN not found."
exit 1
fi
echo "Starting TNT server on port $PORT..."
./tnt -p $PORT &
$BIN -p $PORT &
SERVER_PID=$!
sleep 2
@ -35,4 +41,10 @@ kill $SERVER_PID 2>/dev/null
wait
echo "Stress test complete"
ps aux | grep tnt | grep -v grep && echo "WARNING: tnt process still running"
if ps aux | grep tnt | grep -v grep > /dev/null; then
echo "WARNING: tnt process still running"
else
echo "Server shutdown confirmed."
fi
exit 0