Adds chat session timeout

1 files changed, 14 insertions, 1 deletions

diff --git a/chat_acceptor.h b/chat_acceptor.h
index a67374b..83d0d51 100644
--- a/chat_acceptor.h
+++ b/chat_acceptor.h
@@ -13,6 +13,12 @@ void *receive_chat_messages(void *arg) {
 	char   separators[4] = ":";
 	char * tok;
 
+	// set the timeout of the socket
+	struct timeval tv;
+	tv.tv_sec  = SESSION_TIMEOUT;
+	tv.tv_usec = 0;
+	setsockopt(client_s, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(struct timeval));
+
 	retcode = recv(client_s, in_buf, sizeof(in_buf), 0);
 
 	while (strcmp(in_buf, "SESQ") != 0) {
@@ -27,6 +33,8 @@ void *receive_chat_messages(void *arg) {
 				fflush(stdout);
 			}
 
+		} else {
+			break;
 		}
 
 		strcpy(in_buf, "");
@@ -34,7 +42,12 @@ void *receive_chat_messages(void *arg) {
 		retcode = recv(client_s, in_buf, sizeof(in_buf), 0);
 	}
 
-	printf("\n*** Peer closed the session. *** \n");
+	// display appropriate error message
+	if (retcode == -1) {
+		printf("\n*** Session timed out. *** \n");
+	} else {
+		printf("\n*** Peer closed the session. *** \n");
+	}
 
 	// close the socket
 	close(client_s);