tests/mini-emsgsize-dtls: use gnutls_error_is_fatal and fail

Signed-off-by: Alexander Sosedkin <asosedkin@redhat.com>

diff --git a/tests/mini-emsgsize-dtls.c b/tests/mini-emsgsize-dtls.c
index 2bac761dcdb94e5b6293d8ab997a53896642cd03..7a3b8e2d98880b42a947fdd026cb2843ebb33bf6 100644 (file)
--- a/tests/mini-emsgsize-dtls.c
+++ b/tests/mini-emsgsize-dtls.c
@@ -125,7 +125,8 @@ void doit(void)
                "NONE:+VERS-DTLS1.2:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH",
                NULL);
        if (ret < 0)
-               exit(1);
+               fail("error in server priority_set_direct: %s\n",
+                    gnutls_strerror(ret));
        gnutls_credentials_set(server, GNUTLS_CRD_ANON, s_anoncred);
        gnutls_transport_set_push_function(server, server_push_300);
        gnutls_transport_set_pull_function(server, server_pull);
@@ -141,7 +142,8 @@ void doit(void)
                "NONE:+VERS-DTLS1.2:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH",
                NULL);
        if (cret < 0)
-               exit(1);
+               fail("error in client priority_set_direct: %s\n",
+                    gnutls_strerror(cret));
        gnutls_credentials_set(client, GNUTLS_CRD_ANON, c_anoncred);
        gnutls_transport_set_push_function(client, client_push_300);
        gnutls_transport_set_pull_function(client, client_pull);
@@ -152,10 +154,8 @@ void doit(void)
        handshake = 1;
        HANDSHAKE_DTLS(client, server);
 
-       if (gnutls_protocol_get_version(client) != GNUTLS_DTLS1_2) {
+       if (gnutls_protocol_get_version(client) != GNUTLS_DTLS1_2)
                fail("Error in negotiated version\n");
-               exit(1);
-       }
 
        handshake = 0;
        if (debug)
@@ -163,7 +163,9 @@ void doit(void)
 
        do {
                ret = gnutls_record_send(client, MSG, strlen(MSG));
-       } while (ret == GNUTLS_E_AGAIN);
+       } while (ret < 0 && !gnutls_error_is_fatal(ret));
+       if (ret < 0)
+               fail("error in record_send: %s\n", gnutls_strerror(ret));
 
        msglen = strlen(MSG);
        TRANSFER(client, server, MSG, msglen, buffer, MAX_BUF);