X-Git-Url: https://git.armaanb.net/?p=sic.git;a=blobdiff_plain;f=sic.c;h=059ad9ace17e67bc9dae4b13bd10d9079a4ae90f;hp=ec66f19c4d4af5aae80b2f64a0cc7c2a731d9281;hb=8a58b25cb7f239efa2f3df7c8aa088030f12aeda;hpb=75f0af7c42c1ccdf5bb87975c4ed6f52cd937f94 diff --git a/sic.c b/sic.c index ec66f19..059ad9a 100644 --- a/sic.c +++ b/sic.c @@ -1,7 +1,6 @@ -/* (C)opyright MMV-MMVI Anselm R. Garbe - * (C)opyright MMV-MMVI Nico Golde - * See LICENSE file for license details. - */ +/* © 2005-2008 Anselm R Garbe + * © 2005 Nico Golde + * See LICENSE file for license details. */ #include #include #include @@ -15,21 +14,27 @@ #include #define PINGTIMEOUT 300 -#define MAXMSG 4096 +#define MAXMSG 4096 + +static void die(const char *errstr, ...); +static void printl(char *channel, char *msg); +static void privmsg(char *channel, char *msg); +static void parsein(char *msg); +static void parsesrv(char *msg); +static int readl(int fd, unsigned int len, char *buf); static char *host = "irc.oftc.net"; static unsigned short port = 6667; -static char *nick = NULL; -static char *fullname = NULL; static char *password = NULL; +static char nick[32]; static char bufin[MAXMSG], bufout[MAXMSG]; static char channel[256]; static int srv; static time_t trespond; -static void -eprint(const char *errstr, ...) { +void +die(const char *errstr, ...) { va_list ap; va_start(ap, errstr); @@ -38,23 +43,8 @@ eprint(const char *errstr, ...) { exit(EXIT_FAILURE); } -static int -getline(int fd, unsigned int len, char *buf) { - unsigned int i = 0; - char c; - - do { - if(read(fd, &c, sizeof(char)) != sizeof(char)) - return -1; - buf[i++] = c; - } - while(c != '\n' && i < len); - buf[i - 1] = 0; - return 0; -} - -static void -pout(char *channel, char *msg) { +void +printl(char *channel, char *msg) { static char timestr[18]; time_t t = time(0); @@ -62,17 +52,17 @@ pout(char *channel, char *msg) { fprintf(stdout, "%-12.12s: %s %s\n", channel, timestr, msg); } -static void +void privmsg(char *channel, char *msg) { if(channel[0] == 0) return; snprintf(bufout, sizeof bufout, "<%s> %s", nick, msg); - pout(channel, bufout); + printl(channel, bufout); snprintf(bufout, sizeof bufout, "PRIVMSG %s :%s\r\n", channel, msg); write(srv, bufout, strlen(bufout)); } -static void +void parsein(char *msg) { char *p; @@ -96,32 +86,30 @@ parsein(char *msg) { strncpy(channel, msg + 3, sizeof channel); return; } - else if(!strncmp(msg + 1, "t ", 2)) { - if((p = strchr(msg + 3, ' '))) - *(p++) = 0; - snprintf(bufout, sizeof bufout, "TOPIC %s :%s\r\n", msg + 3, p); - } else snprintf(bufout, sizeof bufout, "%s\r\n", msg + 1); write(srv, bufout, strlen(bufout)); } -static void +void parsesrv(char *msg) { char *chan, *cmd, *p, *txt, *usr; txt = NULL; + usr = host; if(!msg || !(*msg)) return; if(msg[0] != ':') - return; /* don't handle prefix-less server commands */ - if(!(p = strchr(msg, ' '))) - return; - usr = msg + 1; - *p = 0; - cmd = ++p; - if((p = strchr(usr, '!'))) + cmd = msg; + else { + if(!(p = strchr(msg, ' '))) + return; *p = 0; + usr = msg + 1; + cmd = ++p; + if((p = strchr(usr, '!'))) + *p = 0; + } for(p = cmd; *p; p++) /* remove CRLFs */ if(*p == '\r' || *p == '\n') *p = 0; @@ -139,7 +127,7 @@ parsesrv(char *msg) { for(; *p && *p != ' '; p++); *p = 0; snprintf(bufout, sizeof bufout, "<%s> %s", usr, txt); - pout(chan, bufout); + printl(chan, bufout); } else if(!strncmp("PING", cmd, 4) && txt) { snprintf(bufout, sizeof bufout, "PONG %s\r\n", txt); @@ -147,10 +135,28 @@ parsesrv(char *msg) { } else { snprintf(bufout, sizeof bufout, ">< %s: %s", cmd, txt ? txt : ""); - pout(usr, bufout); + printl(usr, bufout); + if(!strncmp("NICK", cmd, 4) && !strncmp(usr, nick, sizeof nick) && txt) + strncpy(nick, txt, sizeof nick); } } +int +readl(int fd, unsigned int len, char *buf) { + unsigned int i = 0; + char c; + + do { + if(read(fd, &c, sizeof(char)) != sizeof(char)) + return -1; + buf[i++] = c; + } + while(c != '\n' && i < len); + buf[i - 1] = 0; + return 0; +} + + int main(int argc, char *argv[]) { int i; @@ -160,7 +166,7 @@ main(int argc, char *argv[]) { char ping[256]; fd_set rd; - nick = fullname = getenv("USER"); + strncpy(nick, getenv("USER"), sizeof nick); for(i = 1; i < argc; i++) if(!strncmp(argv[i], "-h", 3)) { if(++i < argc) host = argv[i]; @@ -169,40 +175,36 @@ main(int argc, char *argv[]) { if(++i < argc) port = (unsigned short)atoi(argv[i]); } else if(!strncmp(argv[i], "-n", 3)) { - if(++i < argc) nick = argv[i]; + if(++i < argc) strncpy(nick, argv[i], sizeof nick); } else if(!strncmp(argv[i], "-k", 3)) { if(++i < argc) password = argv[i]; } - else if(!strncmp(argv[i], "-f", 3)) { - if(++i < argc) fullname = argv[i]; - } else if(!strncmp(argv[i], "-v", 3)) - eprint("sic-"VERSION", (C)opyright MMVI Anselm R. Garbe\n"); + die("sic-"VERSION", © 2005-2008 Anselm R Garbe, Nico Golde\n"); else - eprint("usage: sic [-h host] [-p port] [-n nick]" - " [-k keyword] [-f fullname] [-v]\n"); + die("usage: sic [-h host] [-p port] [-n nick] [-k keyword] [-v]\n"); /* init */ if((srv = socket(AF_INET, SOCK_STREAM, 0)) < 0) - eprint("sic: cannot connect host '%s'\n", host); + die("error: cannot connect host '%s'\n", host); if(NULL == (hp = gethostbyname(host))) - eprint("sic: cannot resolve hostname '%s'\n", host); + die("error: cannot resolve hostname '%s'\n", host); addr.sin_family = AF_INET; addr.sin_port = htons(port); memcpy(&addr.sin_addr, hp->h_addr, hp->h_length); if(connect(srv, (struct sockaddr *) &addr, sizeof(struct sockaddr_in))) { close(srv); - eprint("sic: cannot connect host '%s'\n", host); + die("error: cannot connect host '%s'\n", host); } /* login */ if(password) snprintf(bufout, sizeof bufout, - "PASS %s\r\nNICK %s\r\nUSER %s localhost %s :%s\r\n", - password, nick, nick, host, fullname); + "PASS %s\r\nNICK %s\r\nUSER %s localhost %s :%s\r\n", + password, nick, nick, host, nick); else snprintf(bufout, sizeof bufout, "NICK %s\r\nUSER %s localhost %s :%s\r\n", - nick, nick, host, fullname); + nick, nick, host, nick); write(srv, bufout, strlen(bufout)); snprintf(ping, sizeof ping, "PING %s\r\n", host); channel[0] = 0; @@ -218,23 +220,23 @@ main(int argc, char *argv[]) { if(i < 0) { if(errno == EINTR) continue; - eprint("sic: error on select()"); + die("error: error on select()"); } else if(i == 0) { if(time(NULL) - trespond >= PINGTIMEOUT) - eprint("sic shutting down: parse timeout"); + die("error: sic shutting down: parse timeout"); write(srv, ping, strlen(ping)); continue; } if(FD_ISSET(srv, &rd)) { - if(getline(srv, sizeof bufin, bufin) == -1) - eprint("sic: remote host closed connection"); + if(readl(srv, sizeof bufin, bufin) == -1) + die("error: remote host closed connection"); parsesrv(bufin); trespond = time(NULL); } if(FD_ISSET(0, &rd)) { - if(getline(0, sizeof bufin, bufin) == -1) - eprint("sic: broken pipe"); + if(readl(0, sizeof bufin, bufin) == -1) + die("error: broken pipe"); parsein(bufin); } }