src/journal/test-journal-flush.c

   1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
   2
   3 #include <fcntl.h>
   4 #include <unistd.h>
   5
   6 #include "sd-journal.h"
   7
   8 #include "alloc-util.h"
   9 #include "chattr-util.h"
  10 #include "journal-internal.h"
  11 #include "macro.h"
  12 #include "managed-journal-file.h"
  13 #include "path-util.h"
  14 #include "rm-rf.h"
  15 #include "string-util.h"
  16 #include "tmpfile-util.h"
  17
  18 static void test_journal_flush(int argc, char *argv[]) {
  19         _cleanup_(mmap_cache_unrefp) MMapCache *m = NULL;
  20         _cleanup_free_ char *fn = NULL;
  21         _cleanup_(rm_rf_physical_and_freep) char *dn = NULL;
  22         ManagedJournalFile *new_journal = NULL;
  23         sd_journal *j = NULL;
  24         unsigned n = 0;
  25         int r;
  26
  27         assert_se(m = mmap_cache_new());
  28         assert_se(mkdtemp_malloc("/var/tmp/test-journal-flush.XXXXXX", &dn) >= 0);
  29         (void) chattr_path(dn, FS_NOCOW_FL, FS_NOCOW_FL, NULL);
  30
  31         assert_se(fn = path_join(dn, "test.journal"));
  32
  33         r = managed_journal_file_open(-1, fn, O_CREAT|O_RDWR, 0, 0644, 0, NULL, m, NULL, NULL, &new_journal);
  34         assert_se(r >= 0);
  35
  36         if (argc > 1)
  37                 r = sd_journal_open_files(&j, (const char **) strv_skip(argv, 1), 0);
  38         else
  39                 r = sd_journal_open(&j, 0);
  40         assert_se(r == 0);
  41
  42         sd_journal_set_data_threshold(j, 0);
  43
  44         SD_JOURNAL_FOREACH(j) {
  45                 Object *o;
  46                 JournalFile *f;
  47
  48                 f = j->current_file;
  49                 assert_se(f && f->current_offset > 0);
  50
  51                 r = journal_file_move_to_object(f, OBJECT_ENTRY, f->current_offset, &o);
  52                 if (r < 0)
  53                         log_error_errno(r, "journal_file_move_to_object failed: %m");
  54                 assert_se(r >= 0);
  55
  56                 r = journal_file_copy_entry(f, new_journal->file, o, f->current_offset, NULL, NULL);
  57                 if (r < 0)
  58                         log_warning_errno(r, "journal_file_copy_entry failed: %m");
  59                 assert_se(r >= 0 ||
  60                           IN_SET(r, -EBADMSG,         /* corrupted file */
  61                                     -EPROTONOSUPPORT, /* unsupported compression */
  62                                     -EIO,             /* file rotated */
  63                                     -EREMCHG));       /* clock rollback */
  64
  65                 if (++n >= 10000)
  66                         break;
  67         }
  68
  69         sd_journal_close(j);
  70
  71         (void) managed_journal_file_close(new_journal);
  72 }
  73
  74 int main(int argc, char *argv[]) {
  75         assert_se(setenv("SYSTEMD_JOURNAL_COMPACT", "0", 1) >= 0);
  76         test_journal_flush(argc, argv);
  77
  78         assert_se(setenv("SYSTEMD_JOURNAL_COMPACT", "1", 1) >= 0);
  79         test_journal_flush(argc, argv);
  80
  81         return 0;
  82 }