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 61b102ebfe991778238262226a182d6f3f60e4b0..90c1441e3fe6929bae72e5833914780707cdad31 100644 (file)
--- a/tests/lib/test_dwarf.c
+++ b/tests/lib/test_dwarf.c
@@ -44,6 +44,9 @@ void test_bt_dwarf(const char *data_dir)
 
        fd = open(path, O_RDONLY);
        ok(fd >= 0, "Open DWARF file %s", path);
+       if (fd < 0) {
+               exit(EXIT_FAILURE);
+       }
        dwarf_info = dwarf_begin(fd, DWARF_C_READ);
        ok(dwarf_info != NULL, "dwarf_begin successful");
        cu = bt_dwarf_cu_create(dwarf_info);
@@ -52,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.
@@ -69,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");