More through error checks in set_up

Checks the return values of ssl_init_wbio_buffer() and ssl3_setup_buffers().

diff --git a/ssl/heartbeat_test.c b/ssl/heartbeat_test.c
index 35985f836c9b6e0206e3c07d3cc117c94060c202..23ae0532d56862df01bf662cd58c5ac3fac255a8 100644 (file)
--- a/ssl/heartbeat_test.c
+++ b/ssl/heartbeat_test.c
@@ -68,6 +68,7 @@ static HEARTBEAT_TEST_FIXTURE set_up(const char* const test_case_name,
        const SSL_METHOD* meth)
        {
        HEARTBEAT_TEST_FIXTURE fixture;
+       int setup_ok = 1;
        memset(&fixture, 0, sizeof(fixture));
        fixture.test_case_name = test_case_name;
 
@@ -75,7 +76,8 @@ static HEARTBEAT_TEST_FIXTURE set_up(const char* const test_case_name,
        if (!fixture.ctx)
                {
                fprintf(stderr, "Failed to allocate SSL_CTX for test: %s\n",
-                                               test_case_name);
+                       test_case_name);
+               setup_ok = 0;
                goto fail;
                }
 
@@ -83,14 +85,28 @@ static HEARTBEAT_TEST_FIXTURE set_up(const char* const test_case_name,
        if (!fixture.s)
                {
                fprintf(stderr, "Failed to allocate SSL for test: %s\n", test_case_name);
+               setup_ok = 0;
                goto fail;
                }
 
-       ssl_init_wbio_buffer(fixture.s, 1);
-       ssl3_setup_buffers(fixture.s);
+       if (!ssl_init_wbio_buffer(fixture.s, 1))
+               {
+               fprintf(stderr, "Failed to set up wbio buffer for test: %s\n",
+                       test_case_name);
+               setup_ok = 0;
+               goto fail;
+               }
+
+       if (!ssl3_setup_buffers(fixture.s))
+               {
+               fprintf(stderr, "Failed to setup buffers for test: %s\n",
+                       test_case_name);
+               setup_ok = 0;
+               goto fail;
+               }
 
        fail:
-       if (!fixture.s)
+       if (!setup_ok)
                {
                ERR_print_errors_fp(stderr);
                exit(EXIT_FAILURE);