mirror of
https://github.com/m1ngsama/TNT.git
synced 2026-02-08 00:54:03 +00:00
m1ngsama
1913a00f27
Previous implementation: - Allocated MAX_MESSAGES * 10 (1000 messages) temporarily - Wasted ~100KB per server startup - Could fail if log file grows very large New implementation: - Track file positions of last 1000 lines - Seek to appropriate position before reading - Only allocate MAX_MESSAGES (100 messages) - Memory usage reduced by 90% Benefits: - Faster startup with large log files - Lower memory footprint - No risk of allocation failure - Same functionality maintained Uses fseek/ftell for efficient log file handling.
115 lines
3.2 KiB
C
115 lines
3.2 KiB
C
#include "message.h"
|
|
#include "utf8.h"
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
|
|
/* Initialize message subsystem */
|
|
void message_init(void) {
|
|
/* Nothing to initialize for now */
|
|
}
|
|
|
|
/* Load messages from log file */
|
|
int message_load(message_t **messages, int max_messages) {
|
|
/* Always allocate the message array */
|
|
message_t *msg_array = calloc(max_messages, sizeof(message_t));
|
|
if (!msg_array) {
|
|
return 0;
|
|
}
|
|
|
|
FILE *fp = fopen(LOG_FILE, "r");
|
|
if (!fp) {
|
|
/* File doesn't exist yet, no messages */
|
|
*messages = msg_array;
|
|
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 */
|
|
long file_pos[1000]; /* Track positions of last 1000 lines */
|
|
int line_count = 0;
|
|
int start_index = 0;
|
|
|
|
/* Record file positions */
|
|
while (fgets(line, sizeof(line), fp) && line_count < 1000) {
|
|
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;
|
|
}
|
|
|
|
/* Now read the messages */
|
|
while (fgets(line, sizeof(line), fp) && count < max_messages) {
|
|
/* Format: RFC3339_timestamp|username|content */
|
|
char line_copy[2048];
|
|
strncpy(line_copy, line, sizeof(line_copy) - 1);
|
|
line_copy[sizeof(line_copy) - 1] = '\0';
|
|
|
|
char *timestamp_str = strtok(line_copy, "|");
|
|
char *username = strtok(NULL, "|");
|
|
char *content = strtok(NULL, "\n");
|
|
|
|
if (!timestamp_str || !username || !content) {
|
|
continue;
|
|
}
|
|
|
|
/* Parse ISO 8601 timestamp */
|
|
struct tm tm = {0};
|
|
char *result = strptime(timestamp_str, "%Y-%m-%dT%H:%M:%S", &tm);
|
|
if (!result) {
|
|
continue;
|
|
}
|
|
|
|
msg_array[count].timestamp = mktime(&tm);
|
|
strncpy(msg_array[count].username, username, MAX_USERNAME_LEN - 1);
|
|
strncpy(msg_array[count].content, content, MAX_MESSAGE_LEN - 1);
|
|
count++;
|
|
}
|
|
|
|
fclose(fp);
|
|
*messages = msg_array;
|
|
return count;
|
|
}
|
|
|
|
/* Save a message to log file */
|
|
int message_save(const message_t *msg) {
|
|
FILE *fp = fopen(LOG_FILE, "a");
|
|
if (!fp) {
|
|
return -1;
|
|
}
|
|
|
|
/* Format timestamp as RFC3339 */
|
|
char timestamp[64];
|
|
struct tm tm_info;
|
|
gmtime_r(&msg->timestamp, &tm_info);
|
|
strftime(timestamp, sizeof(timestamp), "%Y-%m-%dT%H:%M:%SZ", &tm_info);
|
|
|
|
/* Write to file: timestamp|username|content */
|
|
fprintf(fp, "%s|%s|%s\n", timestamp, msg->username, msg->content);
|
|
|
|
fclose(fp);
|
|
return 0;
|
|
}
|
|
|
|
/* Format a message for display */
|
|
void message_format(const message_t *msg, char *buffer, size_t buf_size, int width) {
|
|
struct tm tm_info;
|
|
localtime_r(&msg->timestamp, &tm_info);
|
|
char time_str[64];
|
|
strftime(time_str, sizeof(time_str), "%Y-%m-%d %H:%M %Z", &tm_info);
|
|
|
|
snprintf(buffer, buf_size, "[%s] %s: %s", time_str, msg->username, msg->content);
|
|
|
|
/* Truncate if too long */
|
|
if (utf8_string_width(buffer) > width) {
|
|
utf8_truncate(buffer, width);
|
|
}
|
|
}
|