eden-emu/mbedtls
20
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix ssl_read() and close_notify error handling in programs

Browse source
This commit is contained in:
Manuel Pégourié-Gonnard 2014-08-16 11:28:40 +02:00
parent 67686c42e6
commit e08660e612
2 changed files with 49 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

31
programs/ssl/ssl_server2.c
View file

@ -1514,7 +1514,8 @@ reset:
memset( buf, 0, sizeof( buf ) );
ret = ssl_read( &ssl, buf, len );
if( ret == POLARSSL_ERR_NET_WANT_READ || ret == POLARSSL_ERR_NET_WANT_WRITE )
if( ret == POLARSSL_ERR_NET_WANT_READ ||
ret == POLARSSL_ERR_NET_WANT_WRITE )
continue;
if( ret <= 0 )
@ -1523,18 +1524,18 @@ reset:
{
case POLARSSL_ERR_SSL_PEER_CLOSE_NOTIFY:
printf( " connection was closed gracefully\n" );
break;
goto close_notify;
case 0:
case POLARSSL_ERR_NET_CONN_RESET:
printf( " connection was reset by peer\n" );
break;
ret = POLARSSL_ERR_NET_CONN_RESET;
goto reset;
default:
printf( " ssl_read returned -0x%x\n", -ret );
break;
goto close_notify;
}
break;
}
if( ssl_get_bytes_avail( &ssl ) == 0 )
@ -1658,10 +1659,22 @@ reset:
printf( " ok\n" );
}
/*
* 8. Close the connection cleanly
*/
close_notify:
printf( " . Closing the connection..." );
while( ( ret = ssl_close_notify( &ssl ) ) < 0 )
{
printf( " ret = %d (-0x%04X)", ret, -ret );
if( ret == POLARSSL_ERR_NET_CONN_RESET )
{
printf( " connection was reset by peer\n" );
ret = 0;
goto reset;
}
if( ret != POLARSSL_ERR_NET_WANT_READ &&
ret != POLARSSL_ERR_NET_WANT_WRITE )
{
@ -1671,12 +1684,12 @@ reset:
}
printf( " ok\n" );
ret = 0;
goto reset;
/*
* Cleanup and exit
*/
exit:
#ifdef POLARSSL_ERROR_C
if( ret != 0 )
{