1 files changed, 13 insertions, 5 deletions
diff --git a/src/client.c b/src/client.c
index 3d95287..5b87c37 100644
--- a/src/client.c
+++ b/src/client.c
@@ -52,12 +52,20 @@ int main()
 			err(EXIT_FAILURE, "failed writing to socket");
 		}
 
-		ret = read(tcp_sock, &cb, 1);
-		if (ret == -1) {
-			err(EXIT_FAILURE, "failed reading from socket");
-		}
+		if (cb == '\n') {
+			char buf[4096];
+			ret = read(tcp_sock, buf, 4096);
+			if (ret == -1) {
+				printf("Error\n");
+				break;
+			}
 
-		fprintf(stdout, "%c", cb);
+			if (ret == 0) {
+				printf("Closed socket\n");
+				break;
+			}
+			printf("%s\n", buf);
+		}
 	}
 
 	close(tcp_sock);