#include <unistd.h>

#include <fcntl.h>

#include <systemd/sd-journal.h>

#include "journal-file.h"

#include "journal-internal.h"

#include "util.h"

#include "log.h"

#define N_ENTRIES 200

static void verify_contents(sd_journal *j, unsigned skip) {

unsigned i;

assert(j);

i = 0;

SD_JOURNAL_FOREACH(j) {

const void *d;

char *k, *c;

size_t l;

unsigned u;

assert_se(sd_journal_get_cursor(j, &k) >= 0);

printf("cursor: %s\n", k);

free(k);

assert_se(sd_journal_get_data(j, "MAGIC", &d, &l) >= 0);

printf("\t%.*s\n", (int) l, (const char*) d);

assert_se(sd_journal_get_data(j, "NUMBER", &d, &l) >= 0);

assert_se(k = strndup(d, l));

printf("\t%s\n", k);

if (skip > 0) {

assert_se(safe_atou(k + 7, &u) >= 0);

assert_se(i == u);

i += skip;

}

free(k);

assert_se(sd_journal_get_cursor(j, &c) >= 0);

assert_se(sd_journal_test_cursor(j, c) > 0);

free(c);

}

if (skip > 0)

assert_se(i == N_ENTRIES);

}

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

JournalFile *one, *two, *three;

char t[] = "/tmp/journal-stream-XXXXXX";

unsigned i;

sd_journal *j;

char *z;

log_set_max_level(LOG_DEBUG);

assert_se(mkdtemp(t));

assert_se(chdir(t) >= 0);

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

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

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

for (i = 0; i < N_ENTRIES; i++) {

char *p, *q;

dual_timestamp ts;

struct iovec iovec[2];

dual_timestamp_get(&ts);

assert_se(asprintf(&p, "NUMBER=%u", i) >= 0);

iovec[0].iov_base = p;

iovec[0].iov_len = strlen(p);

assert_se(asprintf(&q, "MAGIC=%s", i % 5 == 0 ? "quux" : "waldo") >= 0);

iovec[1].iov_base = q;

iovec[1].iov_len = strlen(q);

if (i % 10 == 0)

assert_se(journal_file_append_entry(three, &ts, iovec, 2, NULL, NULL, NULL) == 0);

else {

if (i % 3 == 0)

assert_se(journal_file_append_entry(two, &ts, iovec, 2, NULL, NULL, NULL) == 0);

assert_se(journal_file_append_entry(one, &ts, iovec, 2, NULL, NULL, NULL) == 0);

}

free(p);

free(q);

}

journal_file_close(one);

journal_file_close(two);

journal_file_close(three);

assert_se(sd_journal_open_directory(&j, t, 0) >= 0);

assert_se(sd_journal_add_match(j, "MAGIC=quux", 0) >= 0);

SD_JOURNAL_FOREACH_BACKWARDS(j) {

const void *d;

size_t l;

char *c;

assert_se(sd_journal_get_data(j, "NUMBER", &d, &l) >= 0);

printf("\t%.*s\n", (int) l, (const char*) d);

assert_se(sd_journal_get_cursor(j, &c) >= 0);

assert_se(sd_journal_test_cursor(j, c) > 0);

free(c);

}

SD_JOURNAL_FOREACH(j) {

const void *d;

size_t l;

char *c;

assert_se(sd_journal_get_data(j, "NUMBER", &d, &l) >= 0);

printf("\t%.*s\n", (int) l, (const char*) d);

assert_se(sd_journal_get_cursor(j, &c) >= 0);

assert_se(sd_journal_test_cursor(j, c) > 0);

free(c);

}

sd_journal_flush_matches(j);

verify_contents(j, 1);

printf("NEXT TEST\n");

assert_se(sd_journal_add_match(j, "MAGIC=quux", 0) >= 0);

assert_se(z = journal_make_match_string(j));

printf("resulting match expression is: %s\n", z);

free(z);

verify_contents(j, 5);

printf("NEXT TEST\n");

sd_journal_flush_matches(j);

assert_se(sd_journal_add_match(j, "MAGIC=waldo", 0) >= 0);

assert_se(sd_journal_add_match(j, "NUMBER=10", 0) >= 0);

assert_se(sd_journal_add_match(j, "NUMBER=11", 0) >= 0);

assert_se(sd_journal_add_match(j, "NUMBER=12", 0) >= 0);

assert_se(z = journal_make_match_string(j));

printf("resulting match expression is: %s\n", z);

free(z);

verify_contents(j, 0);

sd_journal_close(j);

assert_se(rm_rf_dangerous(t, false, true, false) >= 0);

return 0;

}