Tests: dereference of NULL pointer on allocation failure

[babeltrace.git] / tests / lib / test_dwarf.c

diff --git a/tests/lib/test_dwarf.c b/tests/lib/test_dwarf.c
index 95fd9e117ad8dbcede3282966ff2c87294016e01..90c1441e3fe6929bae72e5833914780707cdad31 100644 (file)
--- a/tests/lib/test_dwarf.c
+++ b/tests/lib/test_dwarf.c
@@ -55,6 +55,9 @@ void test_bt_dwarf(const char *data_dir)
        ok(ret == 0, "bt_dwarf_cu_next successful");
        die = bt_dwarf_die_create(cu);
        ok(die != NULL, "bt_dwarf_die_create successful");
+       if (!die) {
+               exit(EXIT_FAILURE);
+       }
        /*
         * Test bt_dwarf_die_next twice, as the code path is different
         * for DIEs at depth 0 (just created) and other depths.
@@ -72,6 +75,10 @@ void test_bt_dwarf(const char *data_dir)
        /* Reset DIE to test dwarf_child */
        bt_dwarf_die_destroy(die);
        die = bt_dwarf_die_create(cu);
+       if (!die) {
+               diag("Failed to create bt_dwarf_die");
+               exit(EXIT_FAILURE);
+       }
 
        ret = bt_dwarf_die_child(die);
        ok(ret == 0, "bt_dwarf_die_child successful");