1 files changed, 5 insertions, 4 deletions
diff --git a/src/server/comms.c b/src/server/comms.c
index cc5a0ce..67e42a2 100644
--- a/src/server/comms.c
+++ b/src/server/comms.c
@@ -55,13 +55,14 @@ struct strs handle_input(mqd_t logging, mqd_t tx, mqd_t rx, char* payload, int l
     return output;
   }
 
-  int n = get_num_messages(rx);
-  while (n > 0) {
+  while (1) {
     char* buf = read_queue(rx);
+    if (strcmp(buf, "END-TRANSMISSION") == 0) {
+      free(buf);
+      break;
+    }
     add_str(&output, buf);
     free(buf);
-    n--;
-    if (n == 0) n = get_num_messages(rx);
   }
   return output;
 }