adsb-tools/adsbus/log.c

#include <assert.h>
#include <fcntl.h>
#include <signal.h>
#include <stdarg.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <time.h>
#include <unistd.h>

#include "log.h"
#include "opts.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 bool log_timestamps = false;
static opts_group log_opts;

static char log_module = 'L';

static void log_rotate() {
	if (!log_path) {
		LOG(log_id, "Received SIGHUP but logging to stderr; ignoring");
		return;
	}

	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);
}

static bool log_set_path(const char *path) {
	if (log_path) {
		return false;
	}
	log_path = strdup(path);
	assert(log_path);
	return true;
}

static bool log_enable_timestamps(const char __attribute__ ((unused)) *arg) {
	log_timestamps = true;
	return true;
}

void log_opts_add() {
	opts_add("log-file", "PATH", log_set_path, log_opts);
	opts_add("log-timestamps", NULL, log_enable_timestamps, log_opts);
}

void log_init() {
	opts_call(log_opts);
	if (log_path) {
		log_stream = fopen(log_path, "a");
	} else {
		int fd = fcntl(STDERR_FILENO, F_DUPFD_CLOEXEC, 0);
		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;
	}
}

void log_write(char type, const char *loc, const uint8_t *id, const char *fmt, ...) {
	va_list ap;
	va_start(ap, fmt);

	char datetime[64] = "";
	if (log_timestamps) {
		time_t now;
		assert(time(&now) >= 0);
		struct tm tmnow;
		assert(gmtime_r(&now, &tmnow));
		assert(strftime(datetime, sizeof(datetime), "%FT%TZ ", &tmnow) > 0);
	}

	assert(fprintf(log_stream, "%s[%18s] %c %s: ", datetime, loc, type, id) > 0);
	assert(vfprintf(log_stream, fmt, ap) > 0);
	assert(fprintf(log_stream, "\n") == 1);
	va_end(ap);
}