101 lines
2.3 KiB
C
101 lines
2.3 KiB
C
#include <assert.h>
|
|
#include <signal.h>
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "log.h"
|
|
#include "peer.h"
|
|
#include "uuid.h"
|
|
|
|
#pragma GCC diagnostic ignored "-Wformat-nonliteral"
|
|
|
|
static FILE *log_stream = NULL;
|
|
static char *log_path = NULL;
|
|
static uint8_t log_id[UUID_LEN];
|
|
static int log_rotate_fd;
|
|
static struct peer log_rotate_peer;
|
|
|
|
static char log_module = 'L';
|
|
|
|
static void log_rotate() {
|
|
uint8_t old_log_id[UUID_LEN], new_log_id[UUID_LEN];
|
|
uuid_gen(new_log_id);
|
|
LOG(log_id, "Switching to new log with ID %s at: %s", new_log_id, log_path);
|
|
memcpy(old_log_id, log_id, UUID_LEN);
|
|
memcpy(log_id, new_log_id, UUID_LEN);
|
|
assert(!fclose(log_stream));
|
|
log_stream = fopen(log_path, "a");
|
|
assert(log_stream);
|
|
setlinebuf(log_stream);
|
|
LOG(log_id, "Log start after switch from log ID %s", old_log_id);
|
|
}
|
|
|
|
static void log_rotate_handler(struct peer *peer) {
|
|
char buf[1];
|
|
assert(read(peer->fd, buf, 1) == 1);
|
|
assert(buf[0] == 'L');
|
|
log_rotate();
|
|
}
|
|
|
|
static void log_rotate_signal(int __attribute__ ((unused)) signum) {
|
|
assert(write(log_rotate_fd, "L", 1) == 1);
|
|
}
|
|
|
|
void log_init() {
|
|
int fd = dup(STDERR_FILENO);
|
|
assert(fd >= 0);
|
|
log_stream = fdopen(fd, "a");
|
|
assert(log_stream);
|
|
setlinebuf(log_stream);
|
|
|
|
uuid_gen(log_id);
|
|
LOG(log_id, "Log start");
|
|
}
|
|
|
|
void log_init2() {
|
|
int rotate_fds[2];
|
|
assert(!socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0, rotate_fds));
|
|
log_rotate_peer.fd = rotate_fds[0];
|
|
assert(!shutdown(log_rotate_peer.fd, SHUT_WR));
|
|
log_rotate_peer.event_handler = log_rotate_handler;
|
|
peer_epoll_add(&log_rotate_peer, EPOLLIN);
|
|
|
|
log_rotate_fd = rotate_fds[1];
|
|
signal(SIGHUP, log_rotate_signal);
|
|
}
|
|
|
|
void log_cleanup() {
|
|
LOG(log_id, "Log end");
|
|
assert(!fclose(log_stream));
|
|
assert(!close(log_rotate_fd));
|
|
assert(!close(log_rotate_peer.fd));
|
|
if (log_path) {
|
|
free(log_path);
|
|
log_path = NULL;
|
|
}
|
|
}
|
|
|
|
bool log_reopen(const char *path) {
|
|
if (log_path) {
|
|
free(log_path);
|
|
}
|
|
log_path = strdup(path);
|
|
assert(log_path);
|
|
log_rotate();
|
|
return true;
|
|
}
|
|
|
|
void log_write(char type, const char *loc, const uint8_t *id, const char *fmt, ...) {
|
|
va_list ap;
|
|
va_start(ap, fmt);
|
|
assert(fprintf(log_stream, "%c [%20s] %s: ", type, loc, id) > 0);
|
|
assert(vfprintf(log_stream, fmt, ap) > 0);
|
|
assert(fprintf(log_stream, "\n") == 1);
|
|
va_end(ap);
|
|
}
|