test-socket-signal: close server when no more data

diff --git a/Utils/test-socket-signal/server.c b/Utils/test-socket-signal/server.c
index b2cd28f..a4b0028 100644 (file)
--- a/Utils/test-socket-signal/server.c
+++ b/Utils/test-socket-signal/server.c
@@ -61,15 +61,6 @@ static void print_buffer_meta(void)
                                curr_time_secs - sender_time_secs);
 }
 
-static void handle_data(int sockfd)
-{
-       ssize_t nbytes;
-
-       nbytes = receive_buffer(sockfd);
-       DIE(nbytes < 0, "receive_buffer");
-       print_buffer_meta();
-}
-
 int main(void)
 {
        int listenfd;
@@ -85,7 +76,14 @@ int main(void)
        DIE(sockfd < 0, "accept");
 
        while (1) {
-               handle_data(sockfd);
+               ssize_t nbytes;
+
+               nbytes = receive_buffer(sockfd);
+               DIE(nbytes < 0, "receive_buffer");
+               if (nbytes == 0)
+                       break;
+
+               print_buffer_meta();
        }
 
        close(sockfd);