/* Copyright (C) 2019 June McEnroe * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . * * Additional permission under GNU GPL version 3 section 7: * * If you modify this Program, or any covered work, by linking or * combining it with OpenSSL (or a modified version of that library), * containing parts covered by the terms of the OpenSSL License and the * original SSLeay license, the licensors of this Program grant you * additional permission to convey the resulting work. Corresponding * Source for a non-source form of such a combination shall include the * source code for the parts of OpenSSL used as well as that of the * covered work. */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "bounce.h" enum Cap clientCaps = 0 | CapConsumer | CapPassive | CapReadMarker | CapSTS | CapSelfMessage | CapServerTime; char *clientOrigin; char *clientPass; char *clientAway; char *clientQuit; static size_t active; static void activeIncr(const struct Client *client) { if (client->need) return; if (client->caps & CapPassive) return; if (!active++) { serverEnqueue("AWAY\r\n"); } } static void activeDecr(const struct Client *client) { if (client->need) return; if (client->caps & CapPassive) return; if (!--active && !stateAway) { serverEnqueue("AWAY :%s\r\n", clientAway); } } struct Client *clientAlloc(int sock, struct tls *tls) { struct Client *client = calloc(1, sizeof(*client)); if (!client) err(EX_OSERR, "calloc"); fcntl(sock, F_SETFL, O_NONBLOCK); client->sock = sock; client->tls = tls; client->time = time(NULL); client->idle = client->time; client->need = NeedHandshake | NeedNick | NeedUser; if (clientPass) client->need |= NeedPass; return client; } static void clientHandshake(struct Client *client) { int error = tls_handshake(client->tls); if (error == TLS_WANT_POLLIN || error == TLS_WANT_POLLOUT) return; if (error) { warnx("client tls_handshake: %s", tls_error(client->tls)); client->remove = true; return; } client->need &= ~NeedHandshake; if ((clientCaps & CapSASL) && tls_peer_cert_provided(client->tls)) { client->need &= ~NeedPass; } } void clientFree(struct Client *client) { activeDecr(client); tls_close(client->tls); tls_free(client->tls); free(client); } void clientSend(struct Client *client, const char *ptr, size_t len) { verboseLog("<-", ptr, len); fcntl(client->sock, F_SETFL, 0); while (len) { ssize_t ret = tls_write(client->tls, ptr, len); if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) continue; if (ret < 0) { warnx("client tls_write: %s", tls_error(client->tls)); client->remove = true; break; } ptr += ret; len -= ret; } fcntl(client->sock, F_SETFL, O_NONBLOCK); client->idle = time(NULL); } void clientFormat(struct Client *client, const char *format, ...) { char buf[MessageCap]; va_list ap; va_start(ap, format); int len = vsnprintf(buf, sizeof(buf), format, ap); va_end(ap); assert((size_t)len < sizeof(buf)); clientSend(client, buf, len); } static void passRequired(struct Client *client) { clientFormat( client, ":%s 464 * :Password incorrect\r\n" "ERROR :Password incorrect\r\n", clientOrigin ); client->remove = true; } static void maybeSync(struct Client *client) { if (client->need == NeedPass) passRequired(client); if (!client->need) { stateSync(client); if (client->setPos) ringSet(client->consumer, client->setPos); activeIncr(client); } } typedef void Handler(struct Client *client, struct Message *msg); static void handleNick(struct Client *client, struct Message *msg) { (void)msg; client->need &= ~NeedNick; maybeSync(client); } static void handleUser(struct Client *client, struct Message *msg) { if (!msg->params[0]) { client->remove = true; return; } if (client->need & NeedPass) { passRequired(client); } else { client->need &= ~NeedUser; client->consumer = ringConsumer(msg->params[0]); if (msg->params[0][0] == '-') client->caps |= CapPassive; maybeSync(client); } } static void handlePass(struct Client *client, struct Message *msg) { if (!clientPass) return; if (!msg->params[0]) { client->remove = true; return; } #ifdef __OpenBSD__ int error = crypt_checkpass(msg->params[0], clientPass); #else int error = strcmp(crypt(msg->params[0], clientPass), clientPass); #endif explicit_bzero(msg->params[0], strlen(msg->params[0])); if (error) { passRequired(client); } else { client->need &= ~NeedPass; maybeSync(client); } } static void handleCap(struct Client *client, struct Message *msg) { if (!msg->params[0]) msg->params[0] = ""; enum Cap avail = clientCaps | (stateCaps & ~CapSASL); const char *values[CapBits] = { [CapSASLBit] = "EXTERNAL", [CapSTSBit] = "duration=2147483647", }; if (!strcmp(msg->params[0], "END")) { if (!client->need) return; client->need &= ~NeedCapEnd; maybeSync(client); } else if (!strcmp(msg->params[0], "LS")) { if (client->need) client->need |= NeedCapEnd; int version = 0; if (msg->params[1]) version = strtol(msg->params[1], NULL, 10); if (version >= 302) { if (avail & CapCapNotify) client->caps |= CapCapNotify; clientFormat( client, ":%s CAP * LS :%s\r\n", clientOrigin, capList(avail, values) ); } else { clientFormat( client, ":%s CAP * LS :%s\r\n", clientOrigin, capList(avail, NULL) ); } } else if (!strcmp(msg->params[0], "REQ") && msg->params[1]) { if (client->need) client->need |= NeedCapEnd; enum Cap caps = capParse(msg->params[1], values); if (caps == (avail & caps)) { if (caps & CapConsumer && values[CapConsumerBit]) { client->setPos = strtoull(values[CapConsumerBit], NULL, 10); } if (caps & CapPassive && !(client->caps & CapPassive)) { activeDecr(client); } client->caps |= caps; clientFormat( client, ":%s CAP * ACK :%s\r\n", clientOrigin, msg->params[1] ); } else { clientFormat( client, ":%s CAP * NAK :%s\r\n", clientOrigin, msg->params[1] ); } } else if (!strcmp(msg->params[0], "LIST")) { clientFormat( client, ":%s CAP * LIST :%s\r\n", clientOrigin, capList(client->caps, NULL) ); } else { clientFormat( client, ":%s 410 * :Invalid CAP subcommand\r\n", clientOrigin ); } } static void handleAuthenticate(struct Client *client, struct Message *msg) { if (!msg->params[0]) msg->params[0] = ""; bool cert = (clientCaps & CapSASL) && tls_peer_cert_provided(client->tls); if (cert && !strcmp(msg->params[0], "EXTERNAL")) { clientFormat(client, "AUTHENTICATE +\r\n"); } else if (cert && !strcmp(msg->params[0], "+")) { const char *account = (stateAccount ? stateAccount : "*"); clientFormat( client, ":%s 900 * %s %s :You are now logged in as %s\r\n", clientOrigin, stateEcho(), account, account ); clientFormat( client, ":%s 903 * :SASL authentication successful\r\n", clientOrigin ); } else { clientFormat( client, ":%s 904 * :SASL authentication failed\r\n", clientOrigin ); } } static void handleJoin(struct Client *client, struct Message *msg) { (void)client; (void)msg; // irssi intentionally sends an invalid JOIN command, at // an invalid time (during client registration), on every // connection. Utterly mind-boggling. Ignore it so the // connection doesn't just get dropped like it deserves to be. } static void handleQuit(struct Client *client, struct Message *msg) { const char *mesg = msg->params[0]; if (mesg && !strncmp(mesg, "$pounce", 7) && (!mesg[7] || mesg[7] == ' ')) { mesg += 7; mesg += strspn(mesg, " "); clientQuit = strdup(mesg); if (!clientQuit) err(EX_OSERR, "strdup"); } else { clientFormat(client, "ERROR :Detaching\r\n"); client->remove = true; } } static bool hasTag(const char *tags, const char *tag) { if (!tags) return false; size_t len = strlen(tag); bool val = strchr(tag, '='); while (*tags && *tags != ' ') { if ( !strncmp(tags, tag, len) && (!tags[len] || strchr((val ? "; " : "=; "), tags[len])) ) return true; tags += strcspn(tags, "; "); tags += (*tags == ';'); } return false; } static const char *synthLabel(struct Client *client) { enum { LabelCap = 64 }; static char buf[sizeof("label=") + LabelCap]; snprintf(buf, sizeof(buf), "label=pounce~%zu", client->consumer); return buf; } static void reserialize( char *buf, size_t cap, const char *origin, const struct Message *msg ) { char *ptr = buf, *end = &buf[cap]; if (msg->tags) { ptr = seprintf(ptr, end, "@%s ", msg->tags); } if (origin || msg->origin) { ptr = seprintf(ptr, end, ":%s ", (origin ? origin : msg->origin)); } ptr = seprintf(ptr, end, "%s", msg->cmd); for (size_t i = 0; i < ParamCap && msg->params[i]; ++i) { if (i + 1 == ParamCap || !msg->params[i + 1]) { ptr = seprintf(ptr, end, " :%s", msg->params[i]); } else { ptr = seprintf(ptr, end, " %s", msg->params[i]); } } } static void clientProduce(struct Client *client, const char *line) { size_t diff = ringDiff(client->consumer); ringProduce(line); if (!diff && !(client->caps & CapEchoMessage)) { ringConsume(NULL, client->consumer); } } static void handlePrivmsg(struct Client *client, struct Message *msg) { if (!msg->params[0]) return; char buf[MessageCap]; bool self = !strcmp(msg->params[0], stateNick()); if (!(stateCaps & CapEchoMessage) || self) { reserialize(buf, sizeof(buf), stateEcho(), msg); clientProduce(client, buf); } if (self) return; reserialize(buf, sizeof(buf), NULL, msg); if (stateCaps & CapEchoMessage && !hasTag(msg->tags, "label")) { serverFormat( "@%s%c%s\r\n", synthLabel(client), (buf[0] == '@' ? ';' : ' '), (buf[0] == '@' ? &buf[1] : buf) ); } else { serverFormat("%s\r\n", buf); } } static void handlePalaver(struct Client *client, struct Message *msg) { if (client->need & NeedPass) return; char buf[MessageCap]; reserialize(buf, sizeof(buf), NULL, msg); clientProduce(client, buf); } struct Marker { char *target; char *timestamp; }; static struct { struct Marker *ptr; size_t cap, len; } markers; void clientGetMarker(struct Client *client, const char *target) { for (size_t i = 0; i < markers.len; ++i) { struct Marker marker = markers.ptr[i]; if (strcasecmp(marker.target, target)) continue; clientFormat( client, ":%s MARKREAD %s timestamp=%s\r\n", clientOrigin, target, marker.timestamp ); return; } clientFormat(client, ":%s MARKREAD %s *\r\n", clientOrigin, target); } static void clientSetMarker( struct Client *client, const char *target, const char *timestamp ) { struct Marker *marker = NULL; for (size_t i = 0; i < markers.len; ++i) { marker = &markers.ptr[i]; if (strcasecmp(marker->target, target)) continue; if (strcmp(timestamp, marker->timestamp) < 0) { clientFormat( client, ":%s MARKREAD %s timestamp=%s\r\n", clientOrigin, target, marker->timestamp ); return; } set(&marker->timestamp, timestamp); goto notify; } if (markers.len == markers.cap) { markers.cap = (markers.cap ? markers.cap * 2 : 8); markers.ptr = realloc(markers.ptr, sizeof(*markers.ptr) * markers.cap); if (!markers.ptr) err(EX_OSERR, "realloc"); } marker = &markers.ptr[markers.len++]; *marker = (struct Marker) {0}; set(&marker->target, target); set(&marker->timestamp, timestamp); notify:; char buf[512]; snprintf( buf, sizeof(buf), ":%s MARKREAD %s timestamp=%s", clientOrigin, marker->target, marker->timestamp ); ringProduce(buf); } static regex_t *TimestampRegex(void) { static const char *Pattern = { #define R2D "[0-9]{2}" "^timestamp=[0-9]{4,}-" R2D "-" R2D "T" R2D ":" R2D ":" R2D "[.][0-9]{3}Z$" #undef R2D }; static bool compiled; static regex_t regex; if (!compiled) { int error = regcomp(®ex, Pattern, REG_EXTENDED | REG_NOSUB); assert(!error); } compiled = true; return ®ex; } static void handleMarkRead(struct Client *client, struct Message *msg) { if (!msg->params[0]) { clientFormat( client, "FAIL MARKREAD NEED_MORE_PARAMS :Missing parameters\r\n" ); } else if (!msg->params[1]) { clientGetMarker(client, msg->params[0]); } else if (regexec(TimestampRegex(), msg->params[1], 0, NULL, 0)) { clientFormat( client, "FAIL MARKREAD INVALID_PARAMS %s :Invalid parameters\r\n", msg->params[1] ); } else { clientSetMarker(client, msg->params[0], &msg->params[1][10]); } } static void handlePong(struct Client *client, struct Message *msg) { (void)client; (void)msg; } static const struct { bool intercept; bool need; const char *cmd; Handler *fn; } Handlers[] = { { false, false, "AUTHENTICATE", handleAuthenticate }, { false, false, "JOIN", handleJoin }, { false, false, "NICK", handleNick }, { false, false, "PASS", handlePass }, { false, false, "USER", handleUser }, { true, false, "CAP", handleCap }, { true, false, "PALAVER", handlePalaver }, { true, false, "PONG", handlePong }, { true, true, "MARKREAD", handleMarkRead }, { true, true, "NOTICE", handlePrivmsg }, { true, true, "PRIVMSG", handlePrivmsg }, { true, true, "QUIT", handleQuit }, { true, true, "TAGMSG", handlePrivmsg }, }; static void clientParse(struct Client *client, char *line) { struct Message msg = parse(line); if (!msg.cmd) return; for (size_t i = 0; i < ARRAY_LEN(Handlers); ++i) { if (strcmp(msg.cmd, Handlers[i].cmd)) continue; if (Handlers[i].need && client->need) break; Handlers[i].fn(client, &msg); return; } client->remove = true; } static bool intercept(const char *line, size_t len) { if (line[0] == '@') { const char *sp = memchr(line, ' ', len); if (!sp) return false; sp++; len -= sp - line; line = sp; } for (size_t i = 0; i < ARRAY_LEN(Handlers); ++i) { if (!Handlers[i].intercept) continue; size_t n = strlen(Handlers[i].cmd); if (len < n) continue; if (memcmp(line, Handlers[i].cmd, n)) continue; if (len == n || line[n] == ' ' || line[n] == '\r') return true; } return false; } void clientRecv(struct Client *client) { if (client->need & NeedHandshake) { clientHandshake(client); return; } ssize_t read = tls_read( client->tls, &client->buf[client->len], sizeof(client->buf) - client->len ); if (read == TLS_WANT_POLLIN || read == TLS_WANT_POLLOUT) return; if (read <= 0) { if (read < 0) warnx("client tls_read: %s", tls_error(client->tls)); client->remove = true; return; } client->len += read; char *lf; char *line = client->buf; for (;;) { lf = memchr(line, '\n', &client->buf[client->len] - line); if (!lf) break; verboseLog("->", line, lf - line); if (client->need || intercept(line, lf - line)) { lf[0] = '\0'; if (lf - line && lf[-1] == '\r') lf[-1] = '\0'; clientParse(client, line); } else { serverSend(line, lf + 1 - line); } line = lf + 1; } client->len -= line - client->buf; memmove(client->buf, line, client->len); client->idle = time(NULL); } static int wordcmp(const char *line, size_t i, const char *word) { if (line[0] == '@') { line += strcspn(line, " "); if (*line) line++; } if (line[0] == ':') { line += strcspn(line, " "); if (*line) line++; } while (i--) { line += strcspn(line, " "); if (*line) line++; } size_t len = strcspn(line, " "); return len == strlen(word) ? strncmp(line, word, len) : (int)len - (int)strlen(word); } // s/..(..)../\1/g static char * snip(char *dst, size_t cap, const char *src, const regex_t *regex) { char *ptr = dst, *end = &dst[cap]; regmatch_t match[2]; assert(regex->re_nsub); for (; *src; src += match[0].rm_eo) { if (regexec(regex, src, 2, match, 0)) break; ptr = seprintf( ptr, end, "%.*s%.*s", (int)match[0].rm_so, src, (int)(match[1].rm_eo - match[1].rm_so), &src[match[1].rm_so] ); } ptr = seprintf(ptr, end, "%s", src); return (ptr == end ? NULL : dst); } static regex_t *compile(regex_t *regex, const char *pattern) { if (regex->re_nsub) return regex; int error = regcomp(regex, pattern, REG_EXTENDED); if (error) { char buf[256]; regerror(error, regex, buf, sizeof(buf)); errx(EX_SOFTWARE, "regcomp: %s: %s", buf, pattern); } return regex; } typedef const char *Filter(const char *line); static const char *filterAccountNotify(const char *line) { return (wordcmp(line, 0, "ACCOUNT") ? line : NULL); } static const char *filterAwayNotify(const char *line) { return (wordcmp(line, 0, "AWAY") ? line : NULL); } static const char *filterBatch(const char *line) { return (wordcmp(line, 0, "BATCH") ? line : NULL); } static const char *filterCapNotify(const char *line) { if (wordcmp(line, 0, "CAP")) return line; if (!wordcmp(line, 1, "NEW")) return NULL; if (!wordcmp(line, 1, "DEL")) return NULL; return line; } static const char *filterChghost(const char *line) { return (wordcmp(line, 0, "CHGHOST") ? line : NULL); } static const char *filterExtendedJoin(const char *line) { if (wordcmp(line, 0, "JOIN")) return line; static regex_t regex; static char buf[MessageCap]; return snip(buf, sizeof(buf), line, compile(®ex, "(JOIN [^ ]+).+")); } static const char *filterInviteNotify(const char *line) { if (wordcmp(line, 0, "INVITE")) return line; return (wordcmp(line, 1, stateNick()) ? NULL : line); } static const char *filterLabeledResponse(const char *line) { return (wordcmp(line, 0, "ACK") ? line : NULL); } static const char *filterMessageTags(const char *line) { return (wordcmp(line, 0, "TAGMSG") ? line : NULL); } static const char *filterMultiPrefix(const char *line) { static char buf[MessageCap]; if (!wordcmp(line, 0, "352")) { static regex_t regex; return snip( buf, sizeof(buf), line, compile(®ex, "( [HG][*]?[~!@%&+])[~!@%&+]+") ); } else if (!wordcmp(line, 0, "353")) { static regex_t regex; return snip( buf, sizeof(buf), line, compile(®ex, "( :?[~!@%&+])[~!@%&+]+") ); } else { return line; } } static const char *filterReadMarker(const char *line) { return (wordcmp(line, 0, "MARKREAD") ? line : NULL); } static const char *filterPalaverApp(const char *line) { return (wordcmp(line, 0, "PALAVER") ? line : NULL); } static const char *filterSetname(const char *line) { return (wordcmp(line, 0, "SETNAME") ? line : NULL); } static const char *filterUserhostInNames(const char *line) { if (wordcmp(line, 0, "353")) return line; static regex_t regex; static char buf[MessageCap]; return snip( buf, sizeof(buf), line, compile(®ex, "( :?[^!]+)![^ ]+") ); } static Filter *Filters[CapBits] = { [CapAccountNotifyBit] = filterAccountNotify, [CapAwayNotifyBit] = filterAwayNotify, [CapBatchBit] = filterBatch, [CapCapNotifyBit] = filterCapNotify, [CapChghostBit] = filterChghost, [CapExtendedJoinBit] = filterExtendedJoin, [CapInviteNotifyBit] = filterInviteNotify, [CapLabeledResponseBit] = filterLabeledResponse, [CapMessageTagsBit] = filterMessageTags, [CapMultiPrefixBit] = filterMultiPrefix, [CapPalaverAppBit] = filterPalaverApp, [CapReadMarkerBit] = filterReadMarker, [CapSetnameBit] = filterSetname, [CapUserhostInNamesBit] = filterUserhostInNames, }; static const char *filterEchoMessage(struct Client *client, const char *line) { if (line[0] != '@') return line; if (!hasTag(&line[1], synthLabel(client))) return line; return NULL; } static const char *filterTags(const char *line) { if (line[0] != '@') return line; const char *sp = strchr(line, ' '); return (sp ? sp + 1 : NULL); } void clientConsume(struct Client *client) { struct timeval time; const char *line = ringPeek(&time, client->consumer); if (!line) return; enum Cap diff = client->caps ^ (clientCaps | stateCaps); if (diff & CapEchoMessage) { line = filterEchoMessage(client, line); } if (line && stateCaps & TagCaps && !(client->caps & TagCaps)) { line = filterTags(line); } for (size_t i = 0; line && i < ARRAY_LEN(Filters); ++i) { if (!Filters[i]) continue; if (diff & (1 << i)) line = Filters[i](line); } if (!line) { ringConsume(NULL, client->consumer); return; } if ( client->caps & CapServerTime && (line[0] != '@' || !hasTag(&line[1], "time")) ) { char ts[sizeof("YYYY-MM-DDThh:mm:ss")]; struct tm *tm = gmtime(&time.tv_sec); strftime(ts, sizeof(ts), "%FT%T", tm); clientFormat( client, "@time=%s.%03dZ;causal.agency/pos=%zu%c%s\r\n", ts, (int)(time.tv_usec / 1000), ringPos(client->consumer) + 1, (line[0] == '@' ? ';' : ' '), (line[0] == '@' ? &line[1] : line) ); } else if (client->caps & CapConsumer) { clientFormat( client, "@causal.agency/pos=%zu%c%s\r\n", ringPos(client->consumer) + 1, (line[0] == '@' ? ';' : ' '), (line[0] == '@' ? &line[1] : line) ); } else { clientFormat(client, "%s\r\n", line); } if (!client->remove) ringConsume(NULL, client->consumer); }