From 72305929b206e8d0e47c2cbbe53e49a36d490dcc Mon Sep 17 00:00:00 2001 From: ENGO150 Date: Wed, 15 Feb 2023 20:06:08 +0100 Subject: [PATCH] added invalid socket check to read_socket function --- src/chat/server/main.c | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/chat/server/main.c b/src/chat/server/main.c index 0a2a388..0c1e1c6 100644 --- a/src/chat/server/main.c +++ b/src/chat/server/main.c @@ -52,6 +52,8 @@ int main(void) received = read_socket(accepted); + if (received == NULL) continue; //FAILED + printf("Received:\n%s\n\n", received); why2_deallocate(received); @@ -65,6 +67,12 @@ int main(void) char *read_socket(int socket) { + if (socket == -1) + { + fprintf(stderr, "Reading socket failed."); + return NULL; + } + long content_size = 0; char *content = NULL; @@ -78,4 +86,4 @@ char *read_socket(int socket) if (recv(socket, content, content_size, 0) != content_size) why2_die("Reading socket failed!"); return content; -} \ No newline at end of file +}