conky/src/journal.cc

/* -*- mode: c++; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*-
 * vim: ts=4 sw=4 noet ai cindent syntax=cpp
 *
 * Conky, a system monitor, based on torsmo
 *
 * Any original torsmo code is licensed under the BSD license
 *
 * All code written since the fork of torsmo is licensed under the GPL
 *
 * Please see COPYING for details
 *
 * Copyright (c) 2004, Hannu Saransaari and Lauri Hakkarainen
 * Copyright (c) 2005-2012 Brenden Matthews, Philip Kovacs, et. al.
 *	(see AUTHORS)
 * All rights reserved.
 *
 * 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 "conky.h"
#include "config.h"
#include "common.h"
#include "text_object.h"
#include "logging.h"
#include <ctype.h>
#include <string.h>
#include <unistd.h>
#include <memory>
#include <systemd/sd-journal.h>

#define MAX_JOURNAL_LINES 200

struct journal {
	int wantedlines;
	int flags;

	journal()
		: wantedlines(0), flags(SD_JOURNAL_LOCAL_ONLY)
	{}

};

void free_journal(struct text_object *obj)
{
	struct journal *j = (struct journal *)obj->data.opaque;
	obj->data.opaque = NULL;
	delete j;
}

void init_journal(const char* type, const char* arg, struct text_object *obj, void* free_at_crash) {
	unsigned int args;
	struct journal *j = new journal;

	std::unique_ptr<char []> tmp(new char[DEFAULT_TEXT_BUFFER_SIZE]);
	memset(tmp.get(), 0, DEFAULT_TEXT_BUFFER_SIZE);

	args = sscanf(arg, "%d %6s", &j->wantedlines, tmp.get());
	if (args < 1 || args > 2) {
		free_journal(obj);
		CRIT_ERR(obj, free_at_crash, "%s a number of lines as 1st argument and optionally a journal type as 2nd argument", type);
	}
	if (j->wantedlines > 0 && j->wantedlines <= MAX_JOURNAL_LINES) {
		if(args > 1) {
			if(strcmp(tmp.get(), "system") == 0) {
				j->flags |= SD_JOURNAL_SYSTEM;
			} else if(strcmp(tmp.get(), "user") == 0) {
				j->flags |= SD_JOURNAL_CURRENT_USER;
			} else {
				free_journal(obj);
				CRIT_ERR(obj, free_at_crash, "invalid arg for %s, type must be 'system' or 'user'", type);
			}
		}

	} else {
		free_journal(obj);
		CRIT_ERR(obj, free_at_crash, "invalid arg for %s, number of lines must be between 1 and %d", type, MAX_JOURNAL_LINES);
	}
	obj->data.opaque = j;
}

static int print_field(sd_journal *jh, const char *field, char spacer, size_t *read, char *p, int p_max_size) {
	const void *get;
	size_t length;
	size_t fieldlen = strlen(field)+1;

	int ret = sd_journal_get_data(jh, field, &get, &length);
	if(ret == -ENOENT)
		goto out;

	if(ret < 0 || length + *read > p_max_size)
		return -1;

	memcpy(p + *read, (const char*)get+fieldlen, length-fieldlen);
	*read += length-fieldlen;

out:
	if(spacer)
		p[(*read)++] = spacer;
	return length ? length-fieldlen : 0;
}

void print_journal(struct text_object *obj, char *p, int p_max_size) {
	size_t read = 0, length;
	struct journal *j = (struct journal *)obj->data.opaque;
	sd_journal *jh = NULL;

	struct tm *tm;
	time_t time;
	uint64_t timestamp;

	if(sd_journal_open(&jh, j->flags) != 0) {
		NORM_ERR("unable to open journal");
		goto out;
	}
	if (!j)
		return;

	if(sd_journal_seek_tail(jh) < 0)
	{
		NORM_ERR("unable to seek to end of journal");
		goto out;
	}
	if(sd_journal_previous_skip(jh, j->wantedlines) < 0) {
		NORM_ERR("unable to seek back %d lines", j->wantedlines);
		goto out;
	}

	do {
		if(sd_journal_get_realtime_usec(jh, &timestamp) < 0)
			break;
		time = timestamp / 1000000;
		tm = localtime(&time);

		if((length = strftime(p+read, p_max_size-read, "%b %d %H:%M:%S", tm)) <= 0)
			break;
		read += length;
		p[read++] = ' ';

		if(print_field(jh, "_HOSTNAME", ' ', &read, p, p_max_size) < 0)
			break;

		if(print_field(jh, "SYSLOG_IDENTIFIER", '[', &read, p, p_max_size) < 0)
			break;

		if(print_field(jh, "_PID", ']', &read, p, p_max_size) < 0)
			break;

		p[read++] = ':';
		p[read++] = ' ';

		if(print_field(jh, "MESSAGE", '\n', &read, p, p_max_size) < 0)
			break;
	} while(sd_journal_next(jh));

out:
	if(jh)
		sd_journal_close(jh);
	p[read] = '\0';
	return;
}