diff options
Diffstat (limited to 'doc/examples/ex-serv-srp.c')
-rw-r--r-- | doc/examples/ex-serv-srp.c | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/doc/examples/ex-serv-srp.c b/doc/examples/ex-serv-srp.c index c79ceb20c5..7686a8fc1c 100644 --- a/doc/examples/ex-serv-srp.c +++ b/doc/examples/ex-serv-srp.c @@ -22,6 +22,11 @@ #define CERTFILE "cert.pem" #define CAFILE "/etc/ssl/certs/ca-certificates.crt" +#define LOOP_CHECK(rval, cmd) \ + do { \ + rval = cmd; \ + } while(rval == GNUTLS_E_AGAIN || rval == GNUTLS_E_INTERRUPTED) + /* This is a sample TLS-SRP echo server. */ @@ -117,11 +122,7 @@ int main(void) gnutls_transport_set_int(session, sd); - do { - ret = gnutls_handshake(session); - } - while (ret < 0 && gnutls_error_is_fatal(ret) == 0); - + LOOP_CHECK(ret, gnutls_handshake(session)); if (ret < 0) { close(sd); gnutls_deinit(session); @@ -137,7 +138,7 @@ int main(void) /* print_info(session); */ for (;;) { - ret = gnutls_record_recv(session, buffer, MAX_BUF); + LOOP_CHECK(ret, gnutls_record_recv(session, buffer, MAX_BUF)); if (ret == 0) { printf @@ -160,7 +161,7 @@ int main(void) } printf("\n"); /* do not wait for the peer to close the connection. */ - gnutls_bye(session, GNUTLS_SHUT_WR); + LOOP_CHECK(ret, gnutls_bye(session, GNUTLS_SHUT_WR)); close(sd); gnutls_deinit(session); |