diff options
Diffstat (limited to '')
| -rw-r--r-- | bounce.c | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/bounce.c b/bounce.c index 712b9c3..648a17b 100644 --- a/bounce.c +++ b/bounce.c @@ -91,10 +91,12 @@ int main(int argc, char *argv[]) { const char *user = NULL; const char *real = NULL; const char *join = NULL; + const char *away = "pounced :3"; int opt; - while (0 < (opt = getopt(argc, argv, "C:H:K:NTP:W:a:h:j:n:p:r:u:vw:"))) { + while (0 < (opt = getopt(argc, argv, "A:C:H:K:NTP:W:a:h:j:n:p:r:u:vw:"))) { switch (opt) { + break; case 'A': away = optarg; break; case 'C': strlcpy(certPath, optarg, sizeof(certPath)); break; case 'H': localHost = optarg; break; case 'K': strlcpy(privPath, optarg, sizeof(privPath)); @@ -147,6 +149,7 @@ int main(int argc, char *argv[]) { } eventAdd(server, NULL); + size_t clients = 0; while (0 < poll(event.fds, event.len, -1)) { for (size_t i = 0; i < event.len; ++i) { short revents = event.fds[i].revents; @@ -156,6 +159,7 @@ int main(int argc, char *argv[]) { int fd; struct tls *tls = listenAccept(&fd, event.fds[i].fd); eventAdd(fd, clientAlloc(tls)); + if (!clients++) serverFormat("AWAY\r\n"); continue; } @@ -172,10 +176,10 @@ int main(int argc, char *argv[]) { if (revents & POLLIN) clientRecv(client); if (revents & POLLOUT) clientConsume(client); if (clientError(client) || revents & (POLLHUP | POLLERR)) { - // TODO: Set AWAY if no more clients remain. clientFree(client); close(event.fds[i].fd); eventRemove(i); + if (!--clients) serverFormat("AWAY :%s\r\n", away); break; } } |