From db49330e0852294f09434a037805fa69b4e2c6e7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Manuel=20P=C3=A9gouri=C3=A9-Gonnard?= <mpg@elzevir.fr>
Date: Thu, 14 Aug 2014 15:36:12 +0200
Subject: [PATCH] ssl_server2 aborts cleanly on SIGTERM

(while waiting for a new connection)
---
 programs/ssl/ssl_server2.c | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/programs/ssl/ssl_server2.c b/programs/ssl/ssl_server2.c
index e58099f10..a09aa3b0a 100644
--- a/programs/ssl/ssl_server2.c
+++ b/programs/ssl/ssl_server2.c
@@ -63,6 +63,7 @@ int main( int argc, char *argv[] )
 #include <string.h>
 #include <stdlib.h>
 #include <stdio.h>
+#include <signal.h>
 
 #include "polarssl/net.h"
 #include "polarssl/ssl.h"
@@ -564,10 +565,19 @@ int psk_callback( void *p_info, ssl_context *ssl,
 }
 #endif /* POLARSSL_KEY_EXCHANGE__SOME__PSK_ENABLED */
 
+/* Interruption handler to ensure clean exit (for valgrind testing) */
+static int listen_fd;
+static int received_sigterm = 0;
+void term_handler( int sig )
+{
+    ((void) sig);
+    received_sigterm = 1;
+    net_close( listen_fd ); /* causes net_accept() to abort */
+}
+
 int main( int argc, char *argv[] )
 {
     int ret = 0, len, written, frags;
-    int listen_fd;
     int client_fd = -1;
     int version_suites[4][2];
     unsigned char buf[IO_BUF_LEN];
@@ -635,6 +645,9 @@ int main( int argc, char *argv[] )
     memset( (void *) alpn_list, 0, sizeof( alpn_list ) );
 #endif
 
+    /* Abort cleanly on SIGTERM */
+    signal( SIGTERM, term_handler );
+
     if( argc == 0 )
     {
     usage:
@@ -1377,6 +1390,13 @@ reset:
 
     if( ( ret = net_accept( listen_fd, &client_fd, NULL ) ) != 0 )
     {
+        if( received_sigterm )
+        {
+            printf( " interrupted by SIGTERM\n" );
+            ret = 0;
+            goto exit;
+        }
+
         printf( " failed\n  ! net_accept returned -0x%x\n\n", -ret );
         goto exit;
     }