src/journal/test-journal-interleaving.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <fcntl.h>
   4 #include <unistd.h>
   5
   6 #include "sd-journal.h"
   7
   8 #include "alloc-util.h"
   9 #include "journal-file.h"
  10 #include "journal-vacuum.h"
  11 #include "log.h"
  12 #include "parse-util.h"
  13 #include "rm-rf.h"
  14 #include "util.h"
  15 #include "tests.h"
  16
  17 /* This program tests skipping around in a multi-file journal. */
  18
  19 static bool arg_keep = false;
  20
  21 _noreturn_ static void log_assert_errno(const char *text, int error, const char *file, int line, const char *func) {
  22         log_internal(LOG_CRIT, error, file, line, func,
  23                      "'%s' failed at %s:%u (%s): %m", text, file, line, func);
  24         abort();
  25 }
  26
  27 #define assert_ret(expr)                                                \
  28         do {                                                            \
  29                 int _r_ = (expr);                                       \
  30                 if (_unlikely_(_r_ < 0))                                \
  31                         log_assert_errno(#expr, -_r_, __FILE__, __LINE__, __PRETTY_FUNCTION__); \
  32         } while (false)
  33
  34 static JournalFile *test_open(const char *name) {
  35         JournalFile *f;
  36         assert_ret(journal_file_open(-1, name, O_RDWR|O_CREAT, 0644, true, (uint64_t) -1, false, NULL, NULL, NULL, NULL, &f));
  37         return f;
  38 }
  39
  40 static void test_close(JournalFile *f) {
  41         (void) journal_file_close (f);
  42 }
  43
  44 static void append_number(JournalFile *f, int n, uint64_t *seqnum) {
  45         char *p;
  46         dual_timestamp ts;
  47         static dual_timestamp previous_ts = {};
  48         struct iovec iovec[1];
  49
  50         dual_timestamp_get(&ts);
  51
  52         if (ts.monotonic <= previous_ts.monotonic)
  53                 ts.monotonic = previous_ts.monotonic + 1;
  54
  55         if (ts.realtime <= previous_ts.realtime)
  56                 ts.realtime = previous_ts.realtime + 1;
  57
  58         previous_ts = ts;
  59
  60         assert_se(asprintf(&p, "NUMBER=%d", n) >= 0);
  61         iovec[0].iov_base = p;
  62         iovec[0].iov_len = strlen(p);
  63         assert_ret(journal_file_append_entry(f, &ts, NULL, iovec, 1, seqnum, NULL, NULL));
  64         free(p);
  65 }
  66
  67 static void test_check_number (sd_journal *j, int n) {
  68         const void *d;
  69         _cleanup_free_ char *k;
  70         size_t l;
  71         int x;
  72
  73         assert_ret(sd_journal_get_data(j, "NUMBER", &d, &l));
  74         assert_se(k = strndup(d, l));
  75         printf("%s\n", k);
  76
  77         assert_se(safe_atoi(k + 7, &x) >= 0);
  78         assert_se(n == x);
  79 }
  80
  81 static void test_check_numbers_down (sd_journal *j, int count) {
  82         int i;
  83
  84         for (i = 1; i <= count; i++) {
  85                 int r;
  86                 test_check_number(j, i);
  87                 assert_ret(r = sd_journal_next(j));
  88                 if (i == count)
  89                         assert_se(r == 0);
  90                 else
  91                         assert_se(r == 1);
  92         }
  93
  94 }
  95
  96 static void test_check_numbers_up (sd_journal *j, int count) {
  97         for (int i = count; i >= 1; i--) {
  98                 int r;
  99                 test_check_number(j, i);
 100                 assert_ret(r = sd_journal_previous(j));
 101                 if (i == 1)
 102                         assert_se(r == 0);
 103                 else
 104                         assert_se(r == 1);
 105         }
 106
 107 }
 108
 109 static void setup_sequential(void) {
 110         JournalFile *one, *two;
 111         one = test_open("one.journal");
 112         two = test_open("two.journal");
 113         append_number(one, 1, NULL);
 114         append_number(one, 2, NULL);
 115         append_number(two, 3, NULL);
 116         append_number(two, 4, NULL);
 117         test_close(one);
 118         test_close(two);
 119 }
 120
 121 static void setup_interleaved(void) {
 122         JournalFile *one, *two;
 123         one = test_open("one.journal");
 124         two = test_open("two.journal");
 125         append_number(one, 1, NULL);
 126         append_number(two, 2, NULL);
 127         append_number(one, 3, NULL);
 128         append_number(two, 4, NULL);
 129         test_close(one);
 130         test_close(two);
 131 }
 132
 133 static void test_skip(void (*setup)(void)) {
 134         char t[] = "/tmp/journal-skip-XXXXXX";
 135         sd_journal *j;
 136         int r;
 137
 138         assert_se(mkdtemp(t));
 139         assert_se(chdir(t) >= 0);
 140
 141         setup();
 142
 143         /* Seek to head, iterate down.
 144          */
 145         assert_ret(sd_journal_open_directory(&j, t, 0));
 146         assert_ret(sd_journal_seek_head(j));
 147         assert_ret(sd_journal_next(j));
 148         test_check_numbers_down(j, 4);
 149         sd_journal_close(j);
 150
 151         /* Seek to tail, iterate up.
 152          */
 153         assert_ret(sd_journal_open_directory(&j, t, 0));
 154         assert_ret(sd_journal_seek_tail(j));
 155         assert_ret(sd_journal_previous(j));
 156         test_check_numbers_up(j, 4);
 157         sd_journal_close(j);
 158
 159         /* Seek to tail, skip to head, iterate down.
 160          */
 161         assert_ret(sd_journal_open_directory(&j, t, 0));
 162         assert_ret(sd_journal_seek_tail(j));
 163         assert_ret(r = sd_journal_previous_skip(j, 4));
 164         assert_se(r == 4);
 165         test_check_numbers_down(j, 4);
 166         sd_journal_close(j);
 167
 168         /* Seek to head, skip to tail, iterate up.
 169          */
 170         assert_ret(sd_journal_open_directory(&j, t, 0));
 171         assert_ret(sd_journal_seek_head(j));
 172         assert_ret(r = sd_journal_next_skip(j, 4));
 173         assert_se(r == 4);
 174         test_check_numbers_up(j, 4);
 175         sd_journal_close(j);
 176
 177         log_info("Done...");
 178
 179         if (arg_keep)
 180                 log_info("Not removing %s", t);
 181         else {
 182                 journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);
 183
 184                 assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
 185         }
 186
 187         puts("------------------------------------------------------------");
 188 }
 189
 190 static void test_sequence_numbers(void) {
 191
 192         char t[] = "/tmp/journal-seq-XXXXXX";
 193         JournalFile *one, *two;
 194         uint64_t seqnum = 0;
 195         sd_id128_t seqnum_id;
 196
 197         assert_se(mkdtemp(t));
 198         assert_se(chdir(t) >= 0);
 199
 200         assert_se(journal_file_open(-1, "one.journal", O_RDWR|O_CREAT, 0644,
 201                                     true, (uint64_t) -1, false, NULL, NULL, NULL, NULL, &one) == 0);
 202
 203         append_number(one, 1, &seqnum);
 204         printf("seqnum=%"PRIu64"\n", seqnum);
 205         assert_se(seqnum == 1);
 206         append_number(one, 2, &seqnum);
 207         printf("seqnum=%"PRIu64"\n", seqnum);
 208         assert_se(seqnum == 2);
 209
 210         assert_se(one->header->state == STATE_ONLINE);
 211         assert_se(!sd_id128_equal(one->header->file_id, one->header->machine_id));
 212         assert_se(!sd_id128_equal(one->header->file_id, one->header->boot_id));
 213         assert_se(sd_id128_equal(one->header->file_id, one->header->seqnum_id));
 214
 215         memcpy(&seqnum_id, &one->header->seqnum_id, sizeof(sd_id128_t));
 216
 217         assert_se(journal_file_open(-1, "two.journal", O_RDWR|O_CREAT, 0644,
 218                                     true, (uint64_t) -1, false, NULL, NULL, NULL, one, &two) == 0);
 219
 220         assert_se(two->header->state == STATE_ONLINE);
 221         assert_se(!sd_id128_equal(two->header->file_id, one->header->file_id));
 222         assert_se(sd_id128_equal(one->header->machine_id, one->header->machine_id));
 223         assert_se(sd_id128_equal(one->header->boot_id, one->header->boot_id));
 224         assert_se(sd_id128_equal(one->header->seqnum_id, one->header->seqnum_id));
 225
 226         append_number(two, 3, &seqnum);
 227         printf("seqnum=%"PRIu64"\n", seqnum);
 228         assert_se(seqnum == 3);
 229         append_number(two, 4, &seqnum);
 230         printf("seqnum=%"PRIu64"\n", seqnum);
 231         assert_se(seqnum == 4);
 232
 233         test_close(two);
 234
 235         append_number(one, 5, &seqnum);
 236         printf("seqnum=%"PRIu64"\n", seqnum);
 237         assert_se(seqnum == 5);
 238
 239         append_number(one, 6, &seqnum);
 240         printf("seqnum=%"PRIu64"\n", seqnum);
 241         assert_se(seqnum == 6);
 242
 243         test_close(one);
 244
 245         /* restart server */
 246         seqnum = 0;
 247
 248         assert_se(journal_file_open(-1, "two.journal", O_RDWR, 0,
 249                                     true, (uint64_t) -1, false, NULL, NULL, NULL, NULL, &two) == 0);
 250
 251         assert_se(sd_id128_equal(two->header->seqnum_id, seqnum_id));
 252
 253         append_number(two, 7, &seqnum);
 254         printf("seqnum=%"PRIu64"\n", seqnum);
 255         assert_se(seqnum == 5);
 256
 257         /* So..., here we have the same seqnum in two files with the
 258          * same seqnum_id. */
 259
 260         test_close(two);
 261
 262         log_info("Done...");
 263
 264         if (arg_keep)
 265                 log_info("Not removing %s", t);
 266         else {
 267                 journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);
 268
 269                 assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
 270         }
 271 }
 272
 273 int main(int argc, char *argv[]) {
 274         test_setup_logging(LOG_DEBUG);
 275
 276         /* journal_file_open requires a valid machine id */
 277         if (access("/etc/machine-id", F_OK) != 0)
 278                 return log_tests_skipped("/etc/machine-id not found");
 279
 280         arg_keep = argc > 1;
 281
 282         test_skip(setup_sequential);
 283         test_skip(setup_interleaved);
 284
 285         test_sequence_numbers();
 286
 287         return 0;
 288 }