diff --git a/Makefile.am b/Makefile.am index fef13fabcd..10839e922a 100644 --- a/Makefile.am +++ b/Makefile.am @@ -2470,7 +2470,8 @@ test_journal_importer_LDADD = \ libsystemd-shared.la EXTRA_DIST += \ - test/journal-data/journal-1.txt + test/journal-data/journal-1.txt \ + test/journal-data/journal-2.txt # ------------------------------------------------------------------------------ ## .PHONY so it always rebuilds it diff --git a/src/test/test-journal-importer.c b/src/test/test-journal-importer.c index fea5c1d51c..1f0684863e 100644 --- a/src/test/test-journal-importer.c +++ b/src/test/test-journal-importer.c @@ -44,7 +44,7 @@ static void test_basic_parsing(void) { do r = journal_importer_process_data(&imp); - while (r == 0); + while (r == 0 && !journal_importer_eof(&imp)); assert_se(r == 1); /* We read one entry, so we should get EOF on next read, but not yet */ @@ -64,11 +64,27 @@ static void test_basic_parsing(void) { assert_se(journal_importer_eof(&imp)); } +static void test_bad_input(void) { + _cleanup_(journal_importer_cleanup) JournalImporter imp = {}; + int r; + + imp.fd = open(TEST_DATA_DIR("/journal-data/journal-2.txt"), O_RDONLY|O_CLOEXEC); + assert_se(imp.fd >= 0); + + do + r = journal_importer_process_data(&imp); + while (!journal_importer_eof(&imp)); + assert_se(r == 0); /* If we don't have enough input, 0 is returned */ + + assert_se(journal_importer_eof(&imp)); +} + int main(int argc, char **argv) { log_set_max_level(LOG_DEBUG); log_parse_environment(); test_basic_parsing(); + test_bad_input(); return 0; } diff --git a/test/journal-data/journal-2.txt b/test/journal-data/journal-2.txt new file mode 100644 index 0000000000..4f582a0e88 Binary files /dev/null and b/test/journal-data/journal-2.txt differ