1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-24 21:34:56 +03:00
samba-mirror/ctdb/common/ctdb_logging.c
Amitay Isaacs 03a96f280f logging: Make sure ringbuffer messages are terminated with a newline
Signed-off-by: Amitay Isaacs <amitay@gmail.com>

(This used to be ctdb commit dbb7c550133c92292a7212bdcaaa79f399b0919b)
2013-05-24 09:15:08 +10:00

192 lines
4.6 KiB
C

/*
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 "lib/tdb/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) {
fprintf(f, "%s:%s %s", tbuf,
get_debug_by_level(log_entries[tmp_entry].level),
log_entries[tmp_entry].message);
}
}
fsize = ftell(f);
rewind(f);
data.dptr = talloc_size(NULL, fsize);
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) {
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;
}