From 57ae082c3c90dce3efc219447c2f2123f806411f Mon Sep 17 00:00:00 2001 From: "C. McEnroe" Date: Wed, 25 Dec 2019 12:50:36 -0500 Subject: Broadcast nick to user and host if they're unset --- litterbox.c | 34 ++++++++++++++++------------------ 1 file changed, 16 insertions(+), 18 deletions(-) diff --git a/litterbox.c b/litterbox.c index 7524f4e..baaf780 100644 --- a/litterbox.c +++ b/litterbox.c @@ -226,12 +226,9 @@ static struct Message parse(char *line) { char *origin = 1 + strsep(&line, " "); msg.nick = strsep(&origin, "!"); msg.user = strsep(&origin, "@"); - if (origin) { - msg.host = origin; - } else { - msg.host = msg.nick; - msg.nick = NULL; - } + msg.host = origin; + if (!msg.user) msg.user = msg.nick; + if (!msg.host) msg.host = msg.user; } msg.cmd = strsep(&line, " "); for (size_t i = 0; line && i < ParamCap; ++i) { @@ -244,7 +241,8 @@ static struct Message parse(char *line) { return msg; } -static void require(const struct Message *msg, size_t len) { +static void require(const struct Message *msg, bool nick, size_t len) { + if (nick && !msg->nick) errx(EX_PROTOCOL, "%s missing origin", msg->cmd); for (size_t i = 0; i < len; ++i) { if (msg->params[i]) continue; errx(EX_PROTOCOL, "%s missing parameter %zu", msg->cmd, 1 + i); @@ -271,7 +269,7 @@ static void handleCap(struct Message *msg) { } static void handleReplyWelcome(struct Message *msg) { - require(msg, 1); + require(msg, false, 1); set(&self, msg->params[0]); format("JOIN :%s\r\n", join); } @@ -294,8 +292,7 @@ static void handleReplyISupport(struct Message *msg) { } static void handlePrivmsg(struct Message *msg) { - require(msg, 2); - if (!msg->nick) return; + require(msg, true, 2); bool query = true; const char *context = msg->params[0]; @@ -319,7 +316,7 @@ static void handlePrivmsg(struct Message *msg) { } static void handleReplyNames(struct Message *msg) { - require(msg, 3); + require(msg, false, 3); char *names = msg->params[3]; while (names) { char *nick = strsep(&names, " "); @@ -329,7 +326,7 @@ static void handleReplyNames(struct Message *msg) { } static void handleJoin(struct Message *msg) { - require(msg, 1); + require(msg, true, 1); insertContext(msg->params[0], false); insertName(msg->nick, msg->user, msg->host); insertEvent( @@ -340,7 +337,7 @@ static void handleJoin(struct Message *msg) { } static void handlePart(struct Message *msg) { - require(msg, 1); + require(msg, true, 1); insertContext(msg->params[0], false); insertName(msg->nick, msg->user, msg->host); insertEvent( @@ -355,8 +352,7 @@ static void handlePart(struct Message *msg) { } static void handleKick(struct Message *msg) { - require(msg, 2); - // FIXME: Sometimes the server kicks people... + require(msg, true, 2); insertContext(msg->params[0], false); insertName(msg->nick, msg->user, msg->host); insertEvent( @@ -372,7 +368,7 @@ static void handleKick(struct Message *msg) { } static void handleNick(struct Message *msg) { - require(msg, 1); + require(msg, true, 1); if (!strcmp(msg->nick, self)) set(&self, msg->params[0]); insertName(msg->nick, msg->user, msg->host); insertEvents( @@ -383,6 +379,7 @@ static void handleNick(struct Message *msg) { } static void handleQuit(struct Message *msg) { + require(msg, true, 0); insertName(msg->nick, msg->user, msg->host); insertEvents( msg->time, Quit, @@ -392,7 +389,7 @@ static void handleQuit(struct Message *msg) { } static void handleTopic(struct Message *msg) { - require(msg, 1); + require(msg, true, 1); insertContext(msg->params[0], false); insertName(msg->nick, msg->user, msg->host); insertEvent( @@ -402,7 +399,7 @@ static void handleTopic(struct Message *msg) { } static void handlePing(struct Message *msg) { - require(msg, 1); + require(msg, false, 1); format("PONG :%s\r\n", msg->params[0]); } @@ -490,6 +487,7 @@ int main(int argc, char *argv[]) { } if (!host) errx(EX_USAGE, "host required"); + set(&self, "*"); set(&chanTypes, "#&"); set(&prefixes, "@+"); -- cgit 1.4.1