mirror of
https://github.com/systemd/systemd.git
synced 2024-11-07 09:56:51 +03:00
919ce0b7af
also define noreturn w/o <stdnoreturn.h>
299 lines
9.0 KiB
C
299 lines
9.0 KiB
C
/*-*- 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 <unistd.h>
|
|
#include <fcntl.h>
|
|
|
|
#include <systemd/sd-journal.h>
|
|
|
|
#include "journal-file.h"
|
|
#include "journal-internal.h"
|
|
#include "journal-vacuum.h"
|
|
#include "util.h"
|
|
#include "log.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_meta(LOG_CRIT, 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;
|
|
struct iovec iovec[1];
|
|
|
|
dual_timestamp_get(&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);
|
|
|
|
assert_se(rm_rf_dangerous(t, false, true, false) >= 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(seqnum == 1);
|
|
append_number(one, 2, &seqnum);
|
|
printf("seqnum=%"PRIu64"\n", seqnum);
|
|
assert(seqnum == 2);
|
|
|
|
assert(one->header->state == STATE_ONLINE);
|
|
assert(!sd_id128_equal(one->header->file_id, one->header->machine_id));
|
|
assert(!sd_id128_equal(one->header->file_id, one->header->boot_id));
|
|
assert(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(two->header->state == STATE_ONLINE);
|
|
assert(!sd_id128_equal(two->header->file_id, one->header->file_id));
|
|
assert(sd_id128_equal(one->header->machine_id, one->header->machine_id));
|
|
assert(sd_id128_equal(one->header->boot_id, one->header->boot_id));
|
|
assert(sd_id128_equal(one->header->seqnum_id, one->header->seqnum_id));
|
|
|
|
append_number(two, 3, &seqnum);
|
|
printf("seqnum=%"PRIu64"\n", seqnum);
|
|
assert(seqnum == 3);
|
|
append_number(two, 4, &seqnum);
|
|
printf("seqnum=%"PRIu64"\n", seqnum);
|
|
assert(seqnum == 4);
|
|
|
|
test_close(two);
|
|
|
|
append_number(one, 5, &seqnum);
|
|
printf("seqnum=%"PRIu64"\n", seqnum);
|
|
assert(seqnum == 5);
|
|
|
|
append_number(one, 6, &seqnum);
|
|
printf("seqnum=%"PRIu64"\n", seqnum);
|
|
assert(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(sd_id128_equal(two->header->seqnum_id, seqnum_id));
|
|
|
|
append_number(two, 7, &seqnum);
|
|
printf("seqnum=%"PRIu64"\n", seqnum);
|
|
assert(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);
|
|
|
|
assert_se(rm_rf_dangerous(t, false, true, false) >= 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;
|
|
}
|