2024-10-10 20:11:57 +08:00
|
|
|
#include "log.h"
|
2024-10-15 01:36:02 +08:00
|
|
|
#include <stdio.h>
|
2024-11-12 19:22:55 +08:00
|
|
|
#include <string.h>
|
2024-10-15 01:36:02 +08:00
|
|
|
#include <stdlib.h>
|
|
|
|
#include <stdarg.h>
|
|
|
|
#include <time.h>
|
2024-10-10 20:11:57 +08:00
|
|
|
|
|
|
|
const char* get_current_time()
|
|
|
|
{
|
|
|
|
static char buffer[20];
|
|
|
|
time_t now = time(NULL);
|
|
|
|
struct tm* tm_info = localtime(&now);
|
|
|
|
strftime(buffer, sizeof(buffer), "%Y-%m-%d %H:%M:%S", tm_info);
|
|
|
|
return buffer;
|
|
|
|
}
|
|
|
|
|
2024-10-15 01:36:02 +08:00
|
|
|
void log_message(LogLevel level, const char* format, ...)
|
2024-10-10 20:11:57 +08:00
|
|
|
{
|
|
|
|
const char* level_strings[] = { "INFO", "WARNING", "ERROR" };
|
2024-11-12 19:22:55 +08:00
|
|
|
static char* last_message = NULL;
|
|
|
|
static unsigned int repeat_count = 0;
|
|
|
|
static unsigned int exp_count = 1;
|
|
|
|
char* message = NULL;
|
2024-10-10 20:11:57 +08:00
|
|
|
|
|
|
|
// Open the log file in append mode
|
|
|
|
FILE* log_file = fopen("logfile.txt", "a");
|
|
|
|
if (log_file == NULL) {
|
|
|
|
fprintf(stderr, "Could not open log file for writing.\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2024-10-15 01:36:02 +08:00
|
|
|
va_list args;
|
|
|
|
|
|
|
|
va_start(args, format);
|
2024-11-12 19:22:55 +08:00
|
|
|
int message_size = vsnprintf(NULL, 0, format, args) + 1;
|
|
|
|
va_end(args);
|
|
|
|
|
|
|
|
va_start(args, format);
|
|
|
|
message = malloc(message_size);
|
|
|
|
vsnprintf(message, message_size, format, args);
|
|
|
|
va_end(args);
|
|
|
|
|
|
|
|
if (!last_message) goto output;
|
|
|
|
// If current and last messages different
|
|
|
|
if (strcmp(message, last_message)) {
|
|
|
|
repeat_count = 0;
|
|
|
|
exp_count = 1;
|
|
|
|
goto output;
|
|
|
|
}
|
|
|
|
repeat_count++;
|
|
|
|
if (repeat_count < LOG_MAX_REPEAT) goto output;
|
|
|
|
|
|
|
|
unsigned int bit_width = sizeof(exp_count) * 8; // Calculate the number of bits in an int
|
|
|
|
unsigned int leftmost_bit_mask = 1 << (bit_width - 1); // Create a mask with the
|
|
|
|
while (exp_count < repeat_count && exp_count < leftmost_bit_mask) exp_count = exp_count << 1;
|
|
|
|
if (repeat_count != exp_count && repeat_count != LOG_MAX_REPEAT) goto clean;
|
|
|
|
|
|
|
|
fprintf(log_file, "[%s] [%s] %s !! x%u !!\n", get_current_time(), level_strings[level], last_message, repeat_count);
|
|
|
|
printf("[%s] %s !! x%u !!\n", level_strings[level], last_message, repeat_count);
|
|
|
|
|
|
|
|
goto clean;
|
|
|
|
output:
|
|
|
|
// Write the log message to the file
|
2024-10-15 01:36:02 +08:00
|
|
|
fprintf(log_file, "[%s] [%s] ", get_current_time(), level_strings[level]);
|
2024-11-12 19:22:55 +08:00
|
|
|
fprintf(log_file, "%s", message);
|
2024-10-15 01:36:02 +08:00
|
|
|
fprintf(log_file, "\n");
|
|
|
|
// Write to stdout
|
|
|
|
printf("[%s] ", level_strings[level]);
|
2024-11-12 19:22:55 +08:00
|
|
|
printf("%s", message);
|
2024-10-15 01:36:02 +08:00
|
|
|
printf("\n");
|
2024-11-12 19:22:55 +08:00
|
|
|
// Record last message
|
|
|
|
if (last_message) free(last_message);
|
|
|
|
last_message = malloc(message_size);
|
|
|
|
strcpy(last_message, message);
|
|
|
|
clean:
|
|
|
|
free(message);
|
2024-10-10 20:11:57 +08:00
|
|
|
fclose(log_file);
|
|
|
|
}
|