diff options
| -rw-r--r-- | norns_shell.c | 8 |
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; |