1 files changed, 113 insertions, 0 deletions
diff --git a/src/logging.c b/src/logging.c
new file mode 100644
index 0000000..9907e5f
--- /dev/null
+++ b/src/logging.c
@@ -0,0 +1,113 @@
+/*
+ * (C) 2014 by Pablo Neira Ayuso <pablo@netfilter.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ */
+
+#include <stdio.h>
+#include <stdarg.h>
+#include <string.h>
+#include <time.h>
+#include <syslog.h>
+
+#include "config.h"
+#include "logging.h"
+
+static struct {
+	const char	*text;
+	const char	*color;
+} logging[NFTS_LOG_MAX] = {
+	[NFTS_LOG_DEBUG] = {
+		.text	= "DEBUG",
+		.color	= "\033[1;33m",
+	},
+        [NFTS_LOG_INFO] = {
+		.text	= "INFO",
+		.color	= "\033[1;32m",
+	},
+        [NFTS_LOG_NOTICE] = {
+		.text	= "NOTICE",
+		.color	= "\033[1;36m",
+	},
+        [NFTS_LOG_ERROR] = {
+		.text	= "ERROR",
+		.color	= "\033[1;33m",
+	},
+        [NFTS_LOG_FATAL] = {
+		.text	= "FATAL",
+		.color	= "\033[1;31m",
+	},
+};
+
+int nft_sync_log_init(struct nft_sync_inst *inst)
+{
+	int ret = 0;
+
+	switch (inst->log.type) {
+	case NFTS_LOG_T_SYSLOG:
+		break;
+	case NFTS_LOG_T_FILE:
+		if (inst->log.fd == NULL)
+			inst->log.fd = stdout;
+		else {
+			inst->log.fd = fopen(inst->log.filename, "w+");
+			if (inst->log.fd == NULL)
+				return -1;
+		}
+		break;
+	}
+
+	return ret;
+}
+
+void nft_sync_log_fini(struct nft_sync_inst *inst)
+{
+	switch (inst->log.type) {
+	case NFTS_LOG_T_SYSLOG:
+		break;
+	case NFTS_LOG_T_FILE:
+		if (inst->log.fd != NULL)
+			fclose(inst->log.fd);
+		break;
+	}
+}
+
+void nft_sync_log(struct nft_sync_inst *inst, int prio,
+		  const char *format, ...)
+{
+	time_t t;
+	char *timebuf = NULL;
+	va_list args;
+
+	switch (inst->log.type) {
+	case NFTS_LOG_T_FILE:
+		t = time(NULL);
+		timebuf = ctime(&t);
+		timebuf[strlen(timebuf) - 1]='\0';
+		break;
+	case NFTS_LOG_T_SYSLOG:
+		break;
+	}
+
+	switch (inst->log.type) {
+	case NFTS_LOG_T_FILE:
+		va_start(args, format);
+		fprintf(inst->log.fd, "%s[%s] [%s] ",
+			inst->log.color ? logging[prio].color : "", timebuf,
+			logging[prio].text);
+		vfprintf(inst->log.fd, format, args);
+		va_end(args);
+		fprintf(inst->log.fd, "%s\n",
+			inst->log.color ? "\033[1;0m" : "");
+		fflush(inst->log.fd);
+		break;
+	case NFTS_LOG_T_SYSLOG:
+		va_start(args, format);
+		vsyslog(prio, format, args);
+		va_end(args);
+		break;
+	}
+}