aboutsummaryrefslogtreecommitdiff
path: root/norns_shell.c
diff options
context:
space:
mode:
Diffstat (limited to 'norns_shell.c')
-rw-r--r--norns_shell.c8
1 files changed, 6 insertions, 2 deletions
diff --git a/norns_shell.c b/norns_shell.c
index fc7867d..10f10b0 100644
--- a/norns_shell.c
+++ b/norns_shell.c
@@ -89,7 +89,9 @@ int main (int argc, char **argv)
int received = 0;
struct timespec sent_sentinel = {0};
- /* TODO wait for responses before exiting after EOF */
+ /* we'll also wait for one message after EOF... */
+ int sent_last = 0;
+
struct nn_pollfd poll_s = { .fd = s, .events = NN_POLLIN | NN_POLLOUT };
while (!killed) {
switch (nn_poll(&poll_s, 1, 1000)) {
@@ -120,6 +122,8 @@ int main (int argc, char **argv)
fprintf(stderr, "Couldn't write: %m\n");
goto error;
}
+ if (sent_last)
+ goto finish;
}
nn_freemsg(buf);
received = 1;
@@ -155,7 +159,7 @@ int main (int argc, char **argv)
goto error;
}
}
- if (n == 0) goto finish;
+ if (n == 0) sent_last = 1;
/* treat EAGAIN as "read 0" */
if (n < 1) n = 0;
line_n += n;