diff options
author | Nikos Mavrogiannopoulos <nmav@gnutls.org> | 2013-11-08 22:14:07 +0100 |
---|---|---|
committer | Nikos Mavrogiannopoulos <nmav@gnutls.org> | 2013-11-08 22:17:10 +0100 |
commit | 76c93d23c073ef8b885503b7d28a31ffe2add6d8 (patch) | |
tree | 1dd2d22a197bc40c5330e516969a7cb1ae9bc96f /tests/mini-dtls-record.c | |
parent | 559a144f6bbcbb611453f82e655dd7438c14d1a7 (diff) | |
download | gnutls-76c93d23c073ef8b885503b7d28a31ffe2add6d8.tar.gz |
reindented code
Diffstat (limited to 'tests/mini-dtls-record.c')
-rw-r--r-- | tests/mini-dtls-record.c | 607 |
1 files changed, 296 insertions, 311 deletions
diff --git a/tests/mini-dtls-record.c b/tests/mini-dtls-record.c index 5128d2ed9d..37d1ebfb46 100644 --- a/tests/mini-dtls-record.c +++ b/tests/mini-dtls-record.c @@ -24,17 +24,16 @@ #include <config.h> #endif -#define REL_LAYER +#define REL_LAYER #include <stdio.h> #include <stdlib.h> #if defined(_WIN32) -int -main () +int main() { - exit (77); + exit(77); } #else @@ -53,7 +52,7 @@ main () #include "utils.h" static int test_finished = 0; -static void terminate (void); +static void terminate(void); /* This program tests the rehandshake in DTLS */ @@ -66,16 +65,14 @@ tls_audit_log_func (gnutls_session_t session, const char *str) } */ -static void -server_log_func (int level, const char *str) +static void server_log_func(int level, const char *str) { - fprintf (stderr, "server|<%d>| %s", level, str); + fprintf(stderr, "server|<%d>| %s", level, str); } -static void -client_log_func (int level, const char *str) +static void client_log_func(int level, const char *str) { - fprintf (stderr, "client|<%d>| %s", level, str); + fprintf(stderr, "client|<%d>| %s", level, str); } /* These are global */ @@ -88,363 +85,351 @@ static pid_t child; #define MAX_SEQ 128 -static int msg_seq[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 10, 16, 5, 32, 11, 11, 11, 11, 12, 10, 13, 14, - 15, 16, 17, 19, 20, 18, 22, 24, 23, 25, 26, 27, 29, 28, 29, 29, 30, 31, 32, 33, 34, 35, 37, 36, 38, 39, - 42, 37, 40, 41, 41, -1}; +static int msg_seq[] = + { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 10, 16, 5, 32, 11, 11, 11, 11, 12, +10, 13, 14, + 15, 16, 17, 19, 20, 18, 22, 24, 23, 25, 26, 27, 29, 28, 29, 29, 30, + 31, 32, 33, 34, 35, 37, 36, 38, 39, + 42, 37, 40, 41, 41, -1 +}; + static unsigned int current = 0; static unsigned int pos = 0; -unsigned char* stored_messages[MAX_SEQ]; +unsigned char *stored_messages[MAX_SEQ]; unsigned int stored_sizes[MAX_SEQ]; static ssize_t -odd_push (gnutls_transport_ptr_t tr, const void *data, size_t len) +odd_push(gnutls_transport_ptr_t tr, const void *data, size_t len) { -ssize_t ret; -unsigned i; - - if (msg_seq[current] == -1 || test_finished != 0) - { - test_finished = 1; - return len; - } - - stored_messages[current] = malloc(len); - memcpy(stored_messages[current], data, len); - stored_sizes[current] = len; - - if (pos != current) - { - for (i=pos;i<=current;i++) - { - if (stored_messages[msg_seq[i]] != NULL) - { - do - { - ret = send((long int)tr, stored_messages[msg_seq[i]], stored_sizes[msg_seq[i]], 0); - } - while(ret == -1 && errno == EAGAIN); - pos++; - } - else - break; - } - } - else if (msg_seq[current] == (int)current) - { - do - { - ret = send((long int)tr, data, len, 0); - } - while(ret == -1 && errno == EAGAIN); - - current++; - pos++; - - return ret; - } - else if (stored_messages[msg_seq[current]] != NULL) - { - do - { - ret = send((long int)tr, stored_messages[msg_seq[current]], stored_sizes[msg_seq[current]], 0); - } - while(ret == -1 && errno == EAGAIN); - current++; - pos++; - return ret; - } - - current++; - - return len; + ssize_t ret; + unsigned i; + + if (msg_seq[current] == -1 || test_finished != 0) { + test_finished = 1; + return len; + } + + stored_messages[current] = malloc(len); + memcpy(stored_messages[current], data, len); + stored_sizes[current] = len; + + if (pos != current) { + for (i = pos; i <= current; i++) { + if (stored_messages[msg_seq[i]] != NULL) { + do { + ret = + send((long int) tr, + stored_messages[msg_seq + [i]], + stored_sizes[msg_seq[i]], + 0); + } + while (ret == -1 && errno == EAGAIN); + pos++; + } else + break; + } + } else if (msg_seq[current] == (int) current) { + do { + ret = send((long int) tr, data, len, 0); + } + while (ret == -1 && errno == EAGAIN); + + current++; + pos++; + + return ret; + } else if (stored_messages[msg_seq[current]] != NULL) { + do { + ret = + send((long int) tr, + stored_messages[msg_seq[current]], + stored_sizes[msg_seq[current]], 0); + } + while (ret == -1 && errno == EAGAIN); + current++; + pos++; + return ret; + } + + current++; + + return len; } static ssize_t -n_push (gnutls_transport_ptr_t tr, const void *data, size_t len) +n_push(gnutls_transport_ptr_t tr, const void *data, size_t len) { - return send((unsigned long)tr, data, len, 0); + return send((unsigned long) tr, data, len, 0); } /* The first five messages are handshake. Thus corresponds to msg_seq+5 */ -static int recv_msg_seq[] = { 1, 2, 3, 4, 5, 6, 12, 28, 7, 8, 9, 10, 11, 13, 15, 16, 14, 18, 20, 19, 21, 22, - 23, 25, 24, 26, 27, 29, 30, 31, 33, 32, 34, 35, 38, 36, 37, -1}; +static int recv_msg_seq[] = + { 1, 2, 3, 4, 5, 6, 12, 28, 7, 8, 9, 10, 11, 13, 15, 16, 14, 18, 20, +19, 21, 22, + 23, 25, 24, 26, 27, 29, 30, 31, 33, 32, 34, 35, 38, 36, 37, -1 +}; -static void -client (int fd) +static void client(int fd) { - gnutls_session_t session; - int ret; - char buffer[MAX_BUF + 1]; - gnutls_anon_client_credentials_t anoncred; - unsigned char seq[8]; - uint64_t useq; - unsigned current = 0; + gnutls_session_t session; + int ret; + char buffer[MAX_BUF + 1]; + gnutls_anon_client_credentials_t anoncred; + unsigned char seq[8]; + uint64_t useq; + unsigned current = 0; #ifndef REL_LAYER - struct timespec ts; + struct timespec ts; - ts.tv_sec = 0; - ts.tv_nsec = 100*1000*1000; + ts.tv_sec = 0; + ts.tv_nsec = 100 * 1000 * 1000; #endif - - memset(buffer, 0, sizeof(buffer)); - - /* Need to enable anonymous KX specifically. */ + + memset(buffer, 0, sizeof(buffer)); + + /* Need to enable anonymous KX specifically. */ /* gnutls_global_set_audit_log_function (tls_audit_log_func); */ - global_init (); - - if (debug) - { - gnutls_global_set_log_function (client_log_func); - gnutls_global_set_log_level (99); - } - - gnutls_anon_allocate_client_credentials (&anoncred); - - /* Initialize TLS session - */ - gnutls_init (&session, GNUTLS_CLIENT | GNUTLS_DATAGRAM); - gnutls_heartbeat_enable (session, GNUTLS_HB_PEER_ALLOWED_TO_SEND); - gnutls_dtls_set_mtu (session, 1500); - - /* Use default priorities */ - gnutls_priority_set_direct (session, - "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-ECDH:+CURVE-ALL", - NULL); - - /* put the anonymous credentials to the current session - */ - gnutls_credentials_set (session, GNUTLS_CRD_ANON, anoncred); - - gnutls_transport_set_int (session, fd); - - /* Perform the TLS handshake - */ - do - { - ret = gnutls_handshake (session); - } - while (ret < 0 && gnutls_error_is_fatal (ret) == 0); - - if (ret < 0) - { - fail ("client: Handshake failed\n"); - gnutls_perror (ret); - exit (1); - } - else - { - if (debug) - success ("client: Handshake was completed\n"); - } - - gnutls_record_send( session, buffer, 1); - - if (debug) - success ("client: DTLS version is: %s\n", - gnutls_protocol_get_name (gnutls_protocol_get_version - (session))); - do - { - ret = gnutls_record_recv_seq (session, buffer, sizeof (buffer), seq); - - if (ret > 0) - { - useq = seq[3] | (seq[2] << 8) | (seq[1] << 16) | (seq[0] << 24); - useq <<= 32; - useq |= seq[7] | (seq[6] << 8) | (seq[5] << 16) | (seq[4] << 24); - - if (recv_msg_seq[current] == -1) - { - fail("received message sequence differs\n"); - terminate(); - } - - if ((uint32_t)recv_msg_seq[current] != (uint32_t)useq) - { - fail("received message sequence differs\n"); - terminate(); - } - - current++; - } + global_init(); + + if (debug) { + gnutls_global_set_log_function(client_log_func); + gnutls_global_set_log_level(99); + } + + gnutls_anon_allocate_client_credentials(&anoncred); + + /* Initialize TLS session + */ + gnutls_init(&session, GNUTLS_CLIENT | GNUTLS_DATAGRAM); + gnutls_heartbeat_enable(session, GNUTLS_HB_PEER_ALLOWED_TO_SEND); + gnutls_dtls_set_mtu(session, 1500); + + /* Use default priorities */ + gnutls_priority_set_direct(session, + "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-ECDH:+CURVE-ALL", + NULL); + + /* put the anonymous credentials to the current session + */ + gnutls_credentials_set(session, GNUTLS_CRD_ANON, anoncred); + + gnutls_transport_set_int(session, fd); + + /* Perform the TLS handshake + */ + do { + ret = gnutls_handshake(session); + } + while (ret < 0 && gnutls_error_is_fatal(ret) == 0); + + if (ret < 0) { + fail("client: Handshake failed\n"); + gnutls_perror(ret); + exit(1); + } else { + if (debug) + success("client: Handshake was completed\n"); + } + + gnutls_record_send(session, buffer, 1); + + if (debug) + success("client: DTLS version is: %s\n", + gnutls_protocol_get_name + (gnutls_protocol_get_version(session))); + do { + ret = + gnutls_record_recv_seq(session, buffer, sizeof(buffer), + seq); + + if (ret > 0) { + useq = + seq[3] | (seq[2] << 8) | (seq[1] << 16) | + (seq[0] << 24); + useq <<= 32; + useq |= + seq[7] | (seq[6] << 8) | (seq[5] << 16) | + (seq[4] << 24); + + if (recv_msg_seq[current] == -1) { + fail("received message sequence differs\n"); + terminate(); + } + + if ((uint32_t) recv_msg_seq[current] != + (uint32_t) useq) { + fail("received message sequence differs\n"); + terminate(); + } + + current++; + } #ifndef REL_LAYER - nanosleep(&ts, NULL); + nanosleep(&ts, NULL); #endif - } - while ((ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED || ret > 0)); + } + while ((ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED + || ret > 0)); - gnutls_bye (session, GNUTLS_SHUT_WR); + gnutls_bye(session, GNUTLS_SHUT_WR); - close (fd); + close(fd); - gnutls_deinit (session); + gnutls_deinit(session); - gnutls_anon_free_client_credentials (anoncred); + gnutls_anon_free_client_credentials(anoncred); - gnutls_global_deinit (); + gnutls_global_deinit(); } -static void -terminate (void) +static void terminate(void) { - int status; + int status; - kill (child, SIGTERM); - wait (&status); - exit (1); + kill(child, SIGTERM); + wait(&status); + exit(1); } -static void -server (int fd) +static void server(int fd) { - int ret; - gnutls_session_t session; - gnutls_anon_server_credentials_t anoncred; - char c; + int ret; + gnutls_session_t session; + gnutls_anon_server_credentials_t anoncred; + char c; #ifndef REL_LAYER - struct timespec ts; - - ts.tv_sec = 0; - ts.tv_nsec = 100*1000*1000; + struct timespec ts; + + ts.tv_sec = 0; + ts.tv_nsec = 100 * 1000 * 1000; #endif - global_init (); - - if (debug) - { - gnutls_global_set_log_function (server_log_func); - gnutls_global_set_log_level (4711); - } - - gnutls_anon_allocate_server_credentials (&anoncred); - - gnutls_init (&session, GNUTLS_SERVER | GNUTLS_DATAGRAM); - gnutls_transport_set_push_function (session, odd_push); - gnutls_heartbeat_enable (session, GNUTLS_HB_PEER_ALLOWED_TO_SEND); - gnutls_dtls_set_mtu (session, 1500); - - /* avoid calling all the priority functions, since the defaults - * are adequate. - */ - gnutls_priority_set_direct (session, - "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-ECDH:+CURVE-ALL", - NULL); - gnutls_credentials_set (session, GNUTLS_CRD_ANON, anoncred); - - gnutls_transport_set_int (session, fd); - - do - { - ret = gnutls_handshake (session); - } - while (ret < 0 && gnutls_error_is_fatal (ret) == 0); - if (ret < 0) - { - close (fd); - gnutls_deinit (session); - fail ("server: Handshake has failed (%s)\n\n", - gnutls_strerror (ret)); - terminate (); - } - if (debug) - success ("server: Handshake was completed\n"); - - if (debug) - success ("server: TLS version is: %s\n", - gnutls_protocol_get_name (gnutls_protocol_get_version - (session))); - - gnutls_record_recv(session, &c, 1); - do - { - do - { - ret = gnutls_record_send( session, &c, 1); - } - while (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED); - - if (ret < 0) - { - fail ("send: %s\n", gnutls_strerror (ret)); - terminate (); - } + global_init(); + + if (debug) { + gnutls_global_set_log_function(server_log_func); + gnutls_global_set_log_level(4711); + } + + gnutls_anon_allocate_server_credentials(&anoncred); + + gnutls_init(&session, GNUTLS_SERVER | GNUTLS_DATAGRAM); + gnutls_transport_set_push_function(session, odd_push); + gnutls_heartbeat_enable(session, GNUTLS_HB_PEER_ALLOWED_TO_SEND); + gnutls_dtls_set_mtu(session, 1500); + + /* avoid calling all the priority functions, since the defaults + * are adequate. + */ + gnutls_priority_set_direct(session, + "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-ECDH:+CURVE-ALL", + NULL); + gnutls_credentials_set(session, GNUTLS_CRD_ANON, anoncred); + + gnutls_transport_set_int(session, fd); + + do { + ret = gnutls_handshake(session); + } + while (ret < 0 && gnutls_error_is_fatal(ret) == 0); + if (ret < 0) { + close(fd); + gnutls_deinit(session); + fail("server: Handshake has failed (%s)\n\n", + gnutls_strerror(ret)); + terminate(); + } + if (debug) + success("server: Handshake was completed\n"); + + if (debug) + success("server: TLS version is: %s\n", + gnutls_protocol_get_name + (gnutls_protocol_get_version(session))); + + gnutls_record_recv(session, &c, 1); + do { + do { + ret = gnutls_record_send(session, &c, 1); + } + while (ret == GNUTLS_E_AGAIN + || ret == GNUTLS_E_INTERRUPTED); + + if (ret < 0) { + fail("send: %s\n", gnutls_strerror(ret)); + terminate(); + } #ifndef REL_LAYER - nanosleep(&ts, NULL); + nanosleep(&ts, NULL); #endif - } - while (test_finished == 0); + } + while (test_finished == 0); - gnutls_transport_set_push_function (session, n_push); + gnutls_transport_set_push_function(session, n_push); #ifndef REL_LAYER - nanosleep(&ts, NULL); + nanosleep(&ts, NULL); #endif - do - { - ret = gnutls_bye (session, GNUTLS_SHUT_WR); - } - while (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED); + do { + ret = gnutls_bye(session, GNUTLS_SHUT_WR); + } + while (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED); - close (fd); - gnutls_deinit (session); + close(fd); + gnutls_deinit(session); - gnutls_anon_free_server_credentials (anoncred); + gnutls_anon_free_server_credentials(anoncred); - gnutls_global_deinit (); + gnutls_global_deinit(); - if (debug) - success ("server: finished\n"); + if (debug) + success("server: finished\n"); } -static void -start (void) +static void start(void) { - int fd[2]; - int ret; + int fd[2]; + int ret; #ifdef REL_LAYER - ret = socketpair (AF_UNIX, SOCK_STREAM, 0, fd); + ret = socketpair(AF_UNIX, SOCK_STREAM, 0, fd); #else - ret = socketpair (AF_UNIX, SOCK_DGRAM, 0, fd); + ret = socketpair(AF_UNIX, SOCK_DGRAM, 0, fd); #endif - if (ret < 0) - { - perror ("socketpair"); - exit (1); - } - - child = fork (); - if (child < 0) - { - perror ("fork"); - fail ("fork"); - exit (1); - } - - if (child) - { - int status; - /* parent */ - - server (fd[0]); - wait (&status); - if (WEXITSTATUS (status) != 0) - fail ("Child died with status %d\n", WEXITSTATUS (status)); - } - else - { - close (fd[0]); - client (fd[1]); - exit (0); - } + if (ret < 0) { + perror("socketpair"); + exit(1); + } + + child = fork(); + if (child < 0) { + perror("fork"); + fail("fork"); + exit(1); + } + + if (child) { + int status; + /* parent */ + + server(fd[0]); + wait(&status); + if (WEXITSTATUS(status) != 0) + fail("Child died with status %d\n", + WEXITSTATUS(status)); + } else { + close(fd[0]); + client(fd[1]); + exit(0); + } } -void -doit (void) +void doit(void) { - start (); + start(); } -#endif /* _WIN32 */ +#endif /* _WIN32 */ |