lib: test-lib - add a real fatal test for mempool-alloconly

Crazily huge memory allocation must fail.

Signed-off-by: Phil Carmody <phil@dovecot.fi>

diff --git a/src/lib/test-lib.c b/src/lib/test-lib.c
index 2a66fb51230178b8d6088ede64f3147886cea691..98e419077d57d9a33c6746affe0203bbbadea2ae 100644 (file)
--- a/src/lib/test-lib.c
+++ b/src/lib/test-lib.c
@@ -47,5 +47,9 @@ int main(void)
                test_var_expand,
                NULL
        };
-       return test_run(test_functions);
+       static enum fatal_test_state (*fatal_functions[])(int) = {
+               fatal_mempool,
+               NULL
+       };
+       return test_run_with_fatals(test_functions, fatal_functions);
 }

diff --git a/src/lib/test-lib.h b/src/lib/test-lib.h
index 8a919669db21f6d38fd67b60bc210eb515082cb1..07549078fccb79c8ea9a97555ff6f89c7e94de36 100644 (file)
--- a/src/lib/test-lib.h
+++ b/src/lib/test-lib.h
@@ -27,6 +27,7 @@ void test_istream_tee(void);
 void test_json_parser(void);
 void test_llist(void);
 void test_mempool_alloconly(void);
+enum fatal_test_state fatal_mempool(int);
 void test_network(void);
 void test_numpack(void);
 void test_ostream_file(void);

diff --git a/src/lib/test-mempool-alloconly.c b/src/lib/test-mempool-alloconly.c
index 2e4b0710f502611ed1b6873c3993b18f1181a497..f34dcac2368aaec1507cb3c9fa817b76750351dc 100644 (file)
--- a/src/lib/test-mempool-alloconly.c
+++ b/src/lib/test-mempool-alloconly.c
@@ -44,3 +44,35 @@ void test_mempool_alloconly(void)
        }
        test_out("mempool_alloconly", success);
 }
+
+enum fatal_test_state fatal_mempool(int stage)
+{
+       static pool_t pool;
+
+       switch(stage) {
+       case 0: /* forbidden size */
+               test_begin("fatal_mempool");
+               pool = pool_alloconly_create(MEMPOOL_GROWING"fatal", 1);
+               (void)p_malloc(pool, 0);
+               return FATAL_TEST_FAILURE;
+
+       case 1: /* logically impossible size */
+               (void)p_malloc(pool, SSIZE_T_MAX + 1ULL);
+               return FATAL_TEST_FAILURE;
+
+       case 2: /* physically impossible size */
+               (void)p_malloc(pool, SSIZE_T_MAX - (size_t)MEM_ALIGN(1));
+               return FATAL_TEST_FAILURE;
+
+       /* Continue with other tests as follows:
+       case 3:
+               something_fatal();
+               return FATAL_TEST_FAILURE;
+       */
+       }
+
+       /* Either our tests have finished, or the test suite has got confused. */
+       pool_unref(&pool);
+       test_end();
+       return FATAL_TEST_FINISHED;
+}