Merge branch 'master' of ctdb into 'master' of samba

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Signed-off-by: Michael Adam <obnox@samba.org>

1 files changed, 200 insertions, 0 deletions

diff --git a/ctdb/common/ctdb_logging.c b/ctdb/common/ctdb_logging.c
new file mode 100644
index 0000000000..ba3e8610c5
--- /dev/null
+++ b/ctdb/common/ctdb_logging.c
@@ -0,0 +1,200 @@
+/* 
+   ctdb logging code
+
+   Copyright (C) Ronnie Sahlberg 2009
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3 of the License, or
+   (at your option) any later version.
+   
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+   
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "includes.h"
+#include "tdb.h"
+#include "system/time.h"
+#include "../include/ctdb_private.h"
+#include "../include/ctdb_client.h"
+
+int log_ringbuf_size;
+
+#define MAX_LOG_SIZE 128
+
+static int first_entry = 0;
+static int ringbuf_count = 0;
+
+struct ctdb_log_entry {
+	int32_t level;
+	struct timeval t;
+	char message[MAX_LOG_SIZE];
+};
+
+
+static struct ctdb_log_entry *log_entries;
+
+/*
+ * this function logs all messages for all levels to a ringbuffer
+ */
+static void log_ringbuffer_v(const char *format, va_list ap)
+{
+	int ret;
+	int next_entry;
+
+	if (log_entries == NULL && log_ringbuf_size != 0) {
+		/* Hope this works. We cant log anything if it doesnt anyway */
+		log_entries = malloc(sizeof(struct ctdb_log_entry) * log_ringbuf_size);
+	}
+	if (log_entries == NULL) {
+		return;
+	}
+
+	next_entry = (first_entry + ringbuf_count) % log_ringbuf_size;
+
+	if (ringbuf_count > 0 && first_entry == next_entry) {
+		first_entry = (first_entry + 1) % log_ringbuf_size;
+	}
+
+	log_entries[next_entry].message[0] = '\0';
+
+	ret = vsnprintf(&log_entries[next_entry].message[0], MAX_LOG_SIZE, format, ap);
+	if (ret == -1) {
+		return;
+	}
+	/* Log messages longer than MAX_LOG_SIZE are truncated to MAX_LOG_SIZE-1
+	 * bytes.  In that case, add a newline.
+	 */
+	if (ret >= MAX_LOG_SIZE) {
+		log_entries[next_entry].message[MAX_LOG_SIZE-2] = '\n';
+	}
+
+	log_entries[next_entry].level = this_log_level;
+	log_entries[next_entry].t = timeval_current();
+
+	if (ringbuf_count < log_ringbuf_size) {
+		ringbuf_count++;
+	}
+}
+
+void log_ringbuffer(const char *format, ...)
+{
+	va_list ap;
+
+	va_start(ap, format);
+	log_ringbuffer_v(format, ap);
+	va_end(ap);
+}
+
+void ctdb_log_ringbuffer_free(void)
+{
+	if (log_entries != NULL) {
+		free(log_entries);
+		log_entries = NULL;
+	}
+	log_ringbuf_size = 0;
+}
+
+void ctdb_collect_log(struct ctdb_context *ctdb, struct ctdb_get_log_addr *log_addr)
+{
+	TDB_DATA data;
+	FILE *f;
+	long fsize;
+	int tmp_entry;
+	struct tm *tm;
+	char tbuf[100];
+	int i;
+
+	DEBUG(DEBUG_ERR,("Marshalling %d log entries\n", ringbuf_count));
+
+	/* dump to a file, then send the file as a blob */
+	f = tmpfile();
+	if (f == NULL) {
+		DEBUG(DEBUG_ERR,(__location__ " Unable to open tmpfile - %s\n", strerror(errno)));
+		return;
+	}
+
+	for (i=0; i<ringbuf_count; i++) {
+		tmp_entry = (first_entry + i) % log_ringbuf_size;
+
+		if (log_entries[tmp_entry].level > log_addr->level) {
+		 	continue;
+		}
+
+		tm = localtime(&log_entries[tmp_entry].t.tv_sec);
+		strftime(tbuf, sizeof(tbuf)-1,"%Y/%m/%d %H:%M:%S", tm);
+
+		if (log_entries[tmp_entry].message[0] != '\0') {
+			fprintf(f, "%s:%s %s", tbuf,
+				get_debug_by_level(log_entries[tmp_entry].level),
+				log_entries[tmp_entry].message);
+		}
+	}
+
+	fsize = ftell(f);
+	if (fsize < 0) {
+		fclose(f);
+		DEBUG(DEBUG_ERR, ("Cannot get file size for log entries\n"));
+		return;
+	}
+	rewind(f);
+	data.dptr = talloc_size(NULL, fsize);
+	if (data.dptr == NULL) {
+		fclose(f);
+		CTDB_NO_MEMORY_VOID(ctdb, data.dptr);
+	}
+	data.dsize = fread(data.dptr, 1, fsize, f);
+	fclose(f);
+
+	DEBUG(DEBUG_ERR,("Marshalling log entries into a blob of %d bytes\n", (int)data.dsize));
+
+	DEBUG(DEBUG_ERR,("Send log to %d:%d\n", (int)log_addr->pnn, (int)log_addr->srvid));
+	ctdb_client_send_message(ctdb, log_addr->pnn, log_addr->srvid, data);
+
+	talloc_free(data.dptr);
+}
+
+int32_t ctdb_control_get_log(struct ctdb_context *ctdb, TDB_DATA addr)
+{
+	struct ctdb_get_log_addr *log_addr = (struct ctdb_get_log_addr *)addr.dptr;
+	pid_t child;
+
+	/* spawn a child process to marshall the huge log blob and send it back
+	   to the ctdb tool using a MESSAGE
+	*/
+	child = ctdb_fork_no_free_ringbuffer(ctdb);
+	if (child == (pid_t)-1) {
+		DEBUG(DEBUG_ERR,("Failed to fork a log collector child\n"));
+		return -1;
+	}
+
+	if (child == 0) {
+		ctdb_set_process_name("ctdb_log_collector");
+		if (switch_from_server_to_client(ctdb, "log-collector") != 0) {
+			DEBUG(DEBUG_CRIT, (__location__ "ERROR: failed to switch log collector child into client mode.\n"));
+			_exit(1);
+		}
+		ctdb_collect_log(ctdb, log_addr);
+		_exit(0);
+	}
+
+	return 0;
+}
+
+void ctdb_clear_log(struct ctdb_context *ctdb)
+{
+	first_entry = 0;
+	ringbuf_count  = 0;
+}
+
+int32_t ctdb_control_clear_log(struct ctdb_context *ctdb)
+{
+	ctdb_clear_log(ctdb);
+
+	return 0;
+}