src/journal/test-journal-interleaving.c - systemd-stable - Rivoreo Source Code Repositories

 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/

 /***
   This file is part of systemd.

   Copyright 2013 Marius Vollmer
   Copyright 2013 Zbigniew Jędrzejewski-Szmek

   systemd is free software; you can redistribute it and/or modify it
   under the terms of the GNU Lesser General Public License as published by
   the Free Software Foundation; either version 2.1 of the License, or
   (at your option) any later version.

   systemd 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
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public License
   along with systemd; If not, see <http://www.gnu.org/licenses/>.
 ***/

 #include <fcntl.h>
 #include <unistd.h>

 #include "sd-journal.h"

 #include "alloc-util.h"
 #include "journal-file.h"
 #include "journal-vacuum.h"
 #include "log.h"
 #include "parse-util.h"
 #include "rm-rf.h"
 #include "util.h"

 /* This program tests skipping around in a multi-file journal.
  */

 static bool arg_keep = false;

 noreturn static void log_assert_errno(const char *text, int eno, const char *file, int line, const char *func) {
         log_internal(LOG_CRIT, 0, file, line, func,
                      "'%s' failed at %s:%u (%s): %s.",
                      text, file, line, func, strerror(eno));
         abort();
 }

 #define assert_ret(expr)                                                \
         do {                                                            \
                 int _r_ = (expr);                                       \
                 if (_unlikely_(_r_ < 0))                                \
                         log_assert_errno(#expr, -_r_, __FILE__, __LINE__, __PRETTY_FUNCTION__); \
         } while (false)

 static JournalFile *test_open(const char *name) {
         JournalFile *f;
         assert_ret(journal_file_open(name, O_RDWR|O_CREAT, 0644, true, false, NULL, NULL, NULL, &f));
         return f;
 }

 static void test_close(JournalFile *f) {
         journal_file_close (f);
 }

 static void append_number(JournalFile *f, int n, uint64_t *seqnum) {
         char *p;
         dual_timestamp ts;
         static dual_timestamp previous_ts = {};
         struct iovec iovec[1];

         dual_timestamp_get(&ts);

         if (ts.monotonic <= previous_ts.monotonic)
                 ts.monotonic = previous_ts.monotonic + 1;

         if (ts.realtime <= previous_ts.realtime)
                 ts.realtime = previous_ts.realtime + 1;

         previous_ts = ts;

         assert_se(asprintf(&p, "NUMBER=%d", n) >= 0);
         iovec[0].iov_base = p;
         iovec[0].iov_len = strlen(p);
         assert_ret(journal_file_append_entry(f, &ts, iovec, 1, seqnum, NULL, NULL));
         free(p);
 }

 static void test_check_number (sd_journal *j, int n) {
         const void *d;
         _cleanup_free_ char *k;
         size_t l;
         int x;

         assert_ret(sd_journal_get_data(j, "NUMBER", &d, &l));
         assert_se(k = strndup(d, l));
         printf("%s\n", k);

         assert_se(safe_atoi(k + 7, &x) >= 0);
         assert_se(n == x);
 }

 static void test_check_numbers_down (sd_journal *j, int count) {
         int i;

         for (i = 1; i <= count; i++) {
                 int r;
                 test_check_number(j, i);
                 assert_ret(r = sd_journal_next(j));
                 if (i == count)
                         assert_se(r == 0);
                 else
                         assert_se(r == 1);
         }

 }

 static void test_check_numbers_up (sd_journal *j, int count) {
         for (int i = count; i >= 1; i--) {
                 int r;
                 test_check_number(j, i);
                 assert_ret(r = sd_journal_previous(j));
                 if (i == 1)
                         assert_se(r == 0);
                 else
                         assert_se(r == 1);
         }

 }

 static void setup_sequential(void) {
         JournalFile *one, *two;
         one = test_open("one.journal");
         two = test_open("two.journal");
         append_number(one, 1, NULL);
         append_number(one, 2, NULL);
         append_number(two, 3, NULL);
         append_number(two, 4, NULL);
         test_close(one);
         test_close(two);
 }

 static void setup_interleaved(void) {
         JournalFile *one, *two;
         one = test_open("one.journal");
         two = test_open("two.journal");
         append_number(one, 1, NULL);
         append_number(two, 2, NULL);
         append_number(one, 3, NULL);
         append_number(two, 4, NULL);
         test_close(one);
         test_close(two);
 }

 static void test_skip(void (*setup)(void)) {
         char t[] = "/tmp/journal-skip-XXXXXX";
         sd_journal *j;
         int r;

         assert_se(mkdtemp(t));
         assert_se(chdir(t) >= 0);

         setup();

         /* Seek to head, iterate down.
          */
         assert_ret(sd_journal_open_directory(&j, t, 0));
         assert_ret(sd_journal_seek_head(j));
         assert_ret(sd_journal_next(j));
         test_check_numbers_down(j, 4);
         sd_journal_close(j);

         /* Seek to tail, iterate up.
          */
         assert_ret(sd_journal_open_directory(&j, t, 0));
         assert_ret(sd_journal_seek_tail(j));
         assert_ret(sd_journal_previous(j));
         test_check_numbers_up(j, 4);
         sd_journal_close(j);

         /* Seek to tail, skip to head, iterate down.
          */
         assert_ret(sd_journal_open_directory(&j, t, 0));
         assert_ret(sd_journal_seek_tail(j));
         assert_ret(r = sd_journal_previous_skip(j, 4));
         assert_se(r == 4);
         test_check_numbers_down(j, 4);
         sd_journal_close(j);

         /* Seek to head, skip to tail, iterate up.
          */
         assert_ret(sd_journal_open_directory(&j, t, 0));
         assert_ret(sd_journal_seek_head(j));
         assert_ret(r = sd_journal_next_skip(j, 4));
         assert_se(r == 4);
         test_check_numbers_up(j, 4);
         sd_journal_close(j);

         log_info("Done...");

         if (arg_keep)
                 log_info("Not removing %s", t);
         else {
                 journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);

                 assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
         }

         puts("------------------------------------------------------------");
 }

 static void test_sequence_numbers(void) {

         char t[] = "/tmp/journal-seq-XXXXXX";
         JournalFile *one, *two;
         uint64_t seqnum = 0;
         sd_id128_t seqnum_id;

         assert_se(mkdtemp(t));
         assert_se(chdir(t) >= 0);

         assert_se(journal_file_open("one.journal", O_RDWR|O_CREAT, 0644,
                                     true, false, NULL, NULL, NULL, &one) == 0);

         append_number(one, 1, &seqnum);
         printf("seqnum=%"PRIu64"\n", seqnum);
         assert_se(seqnum == 1);
         append_number(one, 2, &seqnum);
         printf("seqnum=%"PRIu64"\n", seqnum);
         assert_se(seqnum == 2);

         assert_se(one->header->state == STATE_ONLINE);
         assert_se(!sd_id128_equal(one->header->file_id, one->header->machine_id));
         assert_se(!sd_id128_equal(one->header->file_id, one->header->boot_id));
         assert_se(sd_id128_equal(one->header->file_id, one->header->seqnum_id));

         memcpy(&seqnum_id, &one->header->seqnum_id, sizeof(sd_id128_t));

         assert_se(journal_file_open("two.journal", O_RDWR|O_CREAT, 0644,
                                     true, false, NULL, NULL, one, &two) == 0);

         assert_se(two->header->state == STATE_ONLINE);
         assert_se(!sd_id128_equal(two->header->file_id, one->header->file_id));
         assert_se(sd_id128_equal(one->header->machine_id, one->header->machine_id));
         assert_se(sd_id128_equal(one->header->boot_id, one->header->boot_id));
         assert_se(sd_id128_equal(one->header->seqnum_id, one->header->seqnum_id));

         append_number(two, 3, &seqnum);
         printf("seqnum=%"PRIu64"\n", seqnum);
         assert_se(seqnum == 3);
         append_number(two, 4, &seqnum);
         printf("seqnum=%"PRIu64"\n", seqnum);
         assert_se(seqnum == 4);

         test_close(two);

         append_number(one, 5, &seqnum);
         printf("seqnum=%"PRIu64"\n", seqnum);
         assert_se(seqnum == 5);

         append_number(one, 6, &seqnum);
         printf("seqnum=%"PRIu64"\n", seqnum);
         assert_se(seqnum == 6);

         test_close(one);

         /* restart server */
         seqnum = 0;

         assert_se(journal_file_open("two.journal", O_RDWR, 0,
                                     true, false, NULL, NULL, NULL, &two) == 0);

         assert_se(sd_id128_equal(two->header->seqnum_id, seqnum_id));

         append_number(two, 7, &seqnum);
         printf("seqnum=%"PRIu64"\n", seqnum);
         assert_se(seqnum == 5);

         /* So..., here we have the same seqnum in two files with the
          * same seqnum_id. */

         test_close(two);

         log_info("Done...");

         if (arg_keep)
                 log_info("Not removing %s", t);
         else {
                 journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);

                 assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
         }
 }

 int main(int argc, char *argv[]) {
         log_set_max_level(LOG_DEBUG);

         /* journal_file_open requires a valid machine id */
         if (access("/etc/machine-id", F_OK) != 0)
                 return EXIT_TEST_SKIP;

         arg_keep = argc > 1;

         test_skip(setup_sequential);
         test_skip(setup_interleaved);

         test_sequence_numbers();

         return 0;
 }
	/-- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil --/

	/***
	This file is part of systemd.

	Copyright 2013 Marius Vollmer
	Copyright 2013 Zbigniew Jędrzejewski-Szmek

	systemd is free software; you can redistribute it and/or modify it
	under the terms of the GNU Lesser General Public License as published by
	the Free Software Foundation; either version 2.1 of the License, or
	(at your option) any later version.

	systemd 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
	Lesser General Public License for more details.

	You should have received a copy of the GNU Lesser General Public License
	along with systemd; If not, see <http://www.gnu.org/licenses/>.
	***/

	#include <fcntl.h>
	#include <unistd.h>

	#include "sd-journal.h"

	#include "alloc-util.h"
	#include "journal-file.h"
	#include "journal-vacuum.h"
	#include "log.h"
	#include "parse-util.h"
	#include "rm-rf.h"
	#include "util.h"

	/* This program tests skipping around in a multi-file journal.
	*/

	static bool arg_keep = false;

	noreturn static void log_assert_errno(const char text, int eno, const char file, int line, const char *func) {
	log_internal(LOG_CRIT, 0, file, line, func,
	"'%s' failed at %s:%u (%s): %s.",
	text, file, line, func, strerror(eno));
	abort();
	}

	#define assert_ret(expr) \
	do { \
	int _r_ = (expr); \
	if (_unlikely_(_r_ < 0)) \
	log_assert_errno(#expr, -_r_, __FILE__, __LINE__, __PRETTY_FUNCTION__); \
	} while (false)

	static JournalFile test_open(const char name) {
	JournalFile *f;
	assert_ret(journal_file_open(name, O_RDWR\|O_CREAT, 0644, true, false, NULL, NULL, NULL, &f));
	return f;
	}

	static void test_close(JournalFile *f) {
	journal_file_close (f);
	}

	static void append_number(JournalFile f, int n, uint64_t seqnum) {
	char *p;
	dual_timestamp ts;
	static dual_timestamp previous_ts = {};
	struct iovec iovec[1];

	dual_timestamp_get(&ts);

	if (ts.monotonic <= previous_ts.monotonic)
	ts.monotonic = previous_ts.monotonic + 1;

	if (ts.realtime <= previous_ts.realtime)
	ts.realtime = previous_ts.realtime + 1;

	previous_ts = ts;

	assert_se(asprintf(&p, "NUMBER=%d", n) >= 0);
	iovec[0].iov_base = p;
	iovec[0].iov_len = strlen(p);
	assert_ret(journal_file_append_entry(f, &ts, iovec, 1, seqnum, NULL, NULL));
	free(p);
	}

	static void test_check_number (sd_journal *j, int n) {
	const void *d;
	_cleanup_free_ char *k;
	size_t l;
	int x;

	assert_ret(sd_journal_get_data(j, "NUMBER", &d, &l));
	assert_se(k = strndup(d, l));
	printf("%s\n", k);

	assert_se(safe_atoi(k + 7, &x) >= 0);
	assert_se(n == x);
	}

	static void test_check_numbers_down (sd_journal *j, int count) {
	int i;

	for (i = 1; i <= count; i++) {
	int r;
	test_check_number(j, i);
	assert_ret(r = sd_journal_next(j));
	if (i == count)
	assert_se(r == 0);
	else
	assert_se(r == 1);
	}

	}

	static void test_check_numbers_up (sd_journal *j, int count) {
	for (int i = count; i >= 1; i--) {
	int r;
	test_check_number(j, i);
	assert_ret(r = sd_journal_previous(j));
	if (i == 1)
	assert_se(r == 0);
	else
	assert_se(r == 1);
	}

	}

	static void setup_sequential(void) {
	JournalFile one, two;
	one = test_open("one.journal");
	two = test_open("two.journal");
	append_number(one, 1, NULL);
	append_number(one, 2, NULL);
	append_number(two, 3, NULL);
	append_number(two, 4, NULL);
	test_close(one);
	test_close(two);
	}

	static void setup_interleaved(void) {
	JournalFile one, two;
	one = test_open("one.journal");
	two = test_open("two.journal");
	append_number(one, 1, NULL);
	append_number(two, 2, NULL);
	append_number(one, 3, NULL);
	append_number(two, 4, NULL);
	test_close(one);
	test_close(two);
	}

	static void test_skip(void (*setup)(void)) {
	char t[] = "/tmp/journal-skip-XXXXXX";
	sd_journal *j;
	int r;

	assert_se(mkdtemp(t));
	assert_se(chdir(t) >= 0);

	setup();

	/* Seek to head, iterate down.
	*/
	assert_ret(sd_journal_open_directory(&j, t, 0));
	assert_ret(sd_journal_seek_head(j));
	assert_ret(sd_journal_next(j));
	test_check_numbers_down(j, 4);
	sd_journal_close(j);

	/* Seek to tail, iterate up.
	*/
	assert_ret(sd_journal_open_directory(&j, t, 0));
	assert_ret(sd_journal_seek_tail(j));
	assert_ret(sd_journal_previous(j));
	test_check_numbers_up(j, 4);
	sd_journal_close(j);

	/* Seek to tail, skip to head, iterate down.
	*/
	assert_ret(sd_journal_open_directory(&j, t, 0));
	assert_ret(sd_journal_seek_tail(j));
	assert_ret(r = sd_journal_previous_skip(j, 4));
	assert_se(r == 4);
	test_check_numbers_down(j, 4);
	sd_journal_close(j);

	/* Seek to head, skip to tail, iterate up.
	*/
	assert_ret(sd_journal_open_directory(&j, t, 0));
	assert_ret(sd_journal_seek_head(j));
	assert_ret(r = sd_journal_next_skip(j, 4));
	assert_se(r == 4);
	test_check_numbers_up(j, 4);
	sd_journal_close(j);

	log_info("Done...");

	if (arg_keep)
	log_info("Not removing %s", t);
	else {
	journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);

	assert_se(rm_rf(t, REMOVE_ROOT\|REMOVE_PHYSICAL) >= 0);
	}

	puts("------------------------------------------------------------");
	}

	static void test_sequence_numbers(void) {

	char t[] = "/tmp/journal-seq-XXXXXX";
	JournalFile one, two;
	uint64_t seqnum = 0;
	sd_id128_t seqnum_id;

	assert_se(mkdtemp(t));
	assert_se(chdir(t) >= 0);

	assert_se(journal_file_open("one.journal", O_RDWR\|O_CREAT, 0644,
	true, false, NULL, NULL, NULL, &one) == 0);

	append_number(one, 1, &seqnum);
	printf("seqnum=%"PRIu64"\n", seqnum);
	assert_se(seqnum == 1);
	append_number(one, 2, &seqnum);
	printf("seqnum=%"PRIu64"\n", seqnum);
	assert_se(seqnum == 2);

	assert_se(one->header->state == STATE_ONLINE);
	assert_se(!sd_id128_equal(one->header->file_id, one->header->machine_id));
	assert_se(!sd_id128_equal(one->header->file_id, one->header->boot_id));
	assert_se(sd_id128_equal(one->header->file_id, one->header->seqnum_id));

	memcpy(&seqnum_id, &one->header->seqnum_id, sizeof(sd_id128_t));

	assert_se(journal_file_open("two.journal", O_RDWR\|O_CREAT, 0644,
	true, false, NULL, NULL, one, &two) == 0);

	assert_se(two->header->state == STATE_ONLINE);
	assert_se(!sd_id128_equal(two->header->file_id, one->header->file_id));
	assert_se(sd_id128_equal(one->header->machine_id, one->header->machine_id));
	assert_se(sd_id128_equal(one->header->boot_id, one->header->boot_id));
	assert_se(sd_id128_equal(one->header->seqnum_id, one->header->seqnum_id));

	append_number(two, 3, &seqnum);
	printf("seqnum=%"PRIu64"\n", seqnum);
	assert_se(seqnum == 3);
	append_number(two, 4, &seqnum);
	printf("seqnum=%"PRIu64"\n", seqnum);
	assert_se(seqnum == 4);

	test_close(two);

	append_number(one, 5, &seqnum);
	printf("seqnum=%"PRIu64"\n", seqnum);
	assert_se(seqnum == 5);

	append_number(one, 6, &seqnum);
	printf("seqnum=%"PRIu64"\n", seqnum);
	assert_se(seqnum == 6);

	test_close(one);

	/* restart server */
	seqnum = 0;

	assert_se(journal_file_open("two.journal", O_RDWR, 0,
	true, false, NULL, NULL, NULL, &two) == 0);

	assert_se(sd_id128_equal(two->header->seqnum_id, seqnum_id));

	append_number(two, 7, &seqnum);
	printf("seqnum=%"PRIu64"\n", seqnum);
	assert_se(seqnum == 5);

	/* So..., here we have the same seqnum in two files with the
	* same seqnum_id. */

	test_close(two);

	log_info("Done...");

	if (arg_keep)
	log_info("Not removing %s", t);
	else {
	journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);

	assert_se(rm_rf(t, REMOVE_ROOT\|REMOVE_PHYSICAL) >= 0);
	}
	}

	int main(int argc, char *argv[]) {
	log_set_max_level(LOG_DEBUG);

	/* journal_file_open requires a valid machine id */
	if (access("/etc/machine-id", F_OK) != 0)
	return EXIT_TEST_SKIP;

	arg_keep = argc > 1;

	test_skip(setup_sequential);
	test_skip(setup_interleaved);

	test_sequence_numbers();

	return 0;
	}