ircs-55/ 775 0 0 0 13221630333 76535ustar00jrmjrmircs-55/dat.h 664 0 0 1215 13042165550 10600ustar00jrmjrmenum { Bufsize = 512, }; typedef struct Ircmsg Ircmsg; typedef struct Ircpref Ircpref; typedef void (*Ircfmt)(Ircmsg *, char *, int); typedef struct List List; typedef struct Trie Trie; typedef struct Triewalk Triewalk; struct Ircpref { char *nick; char *user; char *host; char prebuf[128]; }; struct Ircmsg { char *pre; char *cmd; char *par[15]; char *trail; int npar; char buf[Bufsize]; Ircpref; }; struct List { void *val; List *next; }; struct Trie { Rune rune; Trie *parent; Trie *child; Trie *next; void *value; }; struct Triewalk { Trie *root; Trie *curr; Rune *key; int keysize; int depth; }; extern char *logdir; ircs-55/fns.h 664 0 0 1240 13042165270 10613ustar00jrmjrmint ircparse(Ircmsg *, char *); void ircprint(Ircmsg *); int ircischan(char *); int ircfmttime(char *, int, long, int *); void ircfmtjoin(Ircmsg *, char *, int); void ircfmtnick(Ircmsg *, char *, int); void ircfmtquit(Ircmsg *, char *, int); void ircfmtpart(Ircmsg *, char *, int); void ircfmtpriv(Ircmsg *, char *, int); void ircfmtnumeric(Ircmsg *, char *, int); void listfree(List **); int listadd(List **, void *); int listdel(List **, void *); Trie* triealloc(void); void triefree(Trie *); void triewalk(Trie *, Triewalk *, Rune *, int); void* trienext(Triewalk *); void* trieadd(Trie *, Rune *, void *); void* triedel(Trie *, Rune *); void* trieget(Trie *, Rune *); ircs-55/irc.c 664 0 0 2737 13042167007 10611ustar00jrmjrm#include #include #include "dat.h" static int ircprefix(Ircmsg *irc) { char *p; irc->nick = nil; irc->user = nil; irc->host = nil; irc->prebuf[0] = 0; if(irc->pre == nil) return -1; snprint(irc->prebuf, sizeof(irc->prebuf), "%s", irc->pre); irc->nick = irc->prebuf; for(p = irc->prebuf; *p != 0; p++){ if(*p == '@'){ *p = 0; irc->host = p+1; break; } if(irc->user == nil && *p == '!'){ *p = 0; irc->user = p+1; } } return 0; } int ircparse(Ircmsg *irc, char *msg) { int i; char *p; snprint(irc->buf, sizeof(irc->buf), "%s", msg); irc->pre = nil; irc->cmd = nil; for(i = 0; i < 15; i++) irc->par[i] = nil; irc->trail = nil; irc->npar = 0; i = 0; p = irc->buf; if(*p == ':') irc->pre = ++p; else irc->cmd = p; for(; *p != 0; p++){ if(*p == ' '){ *p = 0; if(irc->cmd == nil) irc->cmd = p+1; else { if(i == 14 || *(p+1) == ':'){ if(*(p+1) == ':') irc->trail = p+2; else irc->trail = p+1; irc->par[i++] = irc->trail; break; } irc->par[i++] = p+1; } } } assert(irc->trail == nil || irc->trail == irc->par[i-1]); irc->npar = i; ircprefix(irc); return 0; } void ircprint(Ircmsg *irc) { int i; fprint(2, "pre=%s cmd=%s", irc->pre, irc->cmd); for(i = 0; i < irc->npar; i++) fprint(2, " par[%d]=%s", i, irc->par[i]); fprint(2, " trail=%s\n", irc->trail != nil ? "y" : "n"); } int ircischan(char *target) { return target != nil && target[0] == '#'; } ircs-55/ircfmt.c 664 0 0 4604 13042166107 11313ustar00jrmjrm#include #include #include "dat.h" #include "fns.h" static char *wday[] = { "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat", }; static char *mon[] = { "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec", }; int ircfmttime(char *buf, int bufsize, long time, int *mday) { Tm *t; int n; n = 0; t = localtime(time); if(t->mday != *mday){ *mday = t->mday; n += snprint(buf, bufsize, "-- %s, %02d %s %d --\n", wday[t->wday], t->mday, mon[t->mon], t->year + 1900); bufsize -= n; buf += n; } n += snprint(buf, bufsize, "%02d:%02d:%02d ", t->hour, t->min, t->sec); return n; } void ircfmtjoin(Ircmsg *irc, char *buf, int bufsize) { if(irc->pre != nil){ if(logdir != nil) snprint(buf, bufsize, "JOIN (%s)", irc->nick); else snprint(buf, bufsize, "JOIN (%s)\t%s", irc->nick, irc->par[0]); } else snprint(buf, bufsize, "JOIN %s", irc->par[0]); } void ircfmtnick(Ircmsg *irc, char *buf, int bufsize) { if(irc->pre != nil){ snprint(buf, bufsize, "NICK %s → %s", irc->nick, irc->par[0]); } else snprint(buf, bufsize, "NICK %s", irc->par[0]); } void ircfmtquit(Ircmsg *irc, char *buf, int bufsize) { if(irc->pre != nil){ snprint(buf, bufsize, "QUIT (%s)\t%s", irc->nick, irc->trail); } else snprint(buf, bufsize, "QUIT\t%s", irc->trail); } void ircfmtpart(Ircmsg *irc, char *buf, int bufsize) { if(irc->pre != nil){ if(logdir != nil) snprint(buf, bufsize, "PART (%s)", irc->nick); else snprint(buf, bufsize, "PART (%s)\t%s", irc->nick, irc->par[0]); } else snprint(buf, bufsize, "PART %s", irc->par[0]); } void ircfmtpriv(Ircmsg *irc, char *buf, int bufsize) { if(irc->pre != nil){ if(logdir != nil && ircischan(irc->par[0])) snprint(buf, bufsize, "%s → %s", irc->nick, irc->trail); else snprint(buf, bufsize, "(%s) %s → %s", irc->par[0], irc->nick, irc->trail); } else { if(logdir != nil && ircischan(irc->par[0])) snprint(buf, bufsize, "⇐ %s", irc->trail); else snprint(buf, bufsize, "(%s) ⇐ %s", irc->par[0], irc->trail); } } void ircfmtnumeric(Ircmsg *irc, char *buf, int bufsize) { int i, n; *buf = 0; n = snprint(buf, bufsize, irc->cmd); bufsize -= n; buf += n; for(i = 1; i < irc->npar; i++){ n = snprint(buf, bufsize, " %s", irc->par[i]); bufsize -= n; buf += n; } } ircs-55/list.c 664 0 0 1372 13036630566 11011ustar00jrmjrm#include #include #include "dat.h" int listallocs; void listfree(List **lp) { List *l, *n; for(l = *lp; l != nil; l = n){ n = l->next; free(l); listallocs--; } *lp = nil; } int listadd(List **lp, void *val) { List *l; for(l = *lp; l != nil; l = l->next) if(l->val == val) break; if(l != nil) return 0; l = mallocz(sizeof(List), 1); if(l == nil) sysfatal("mallocz: %r"); listallocs++; l->val = val; l->next = *lp; *lp = l; return 1; } int listdel(List **lp, void *val) { List *l, *prev; prev = nil; for(l = *lp; l != nil; l = l->next){ if(l->val == val) break; prev = l; } if(l == nil) return 0; if(prev == nil) *lp = l->next; else prev->next = l->next; free(l); listallocs--; return 1; } ircs-55/main.c 664 0 0 41310 13131074712 10765ustar00jrmjrm#include #include #include #include #include #include "dat.h" #include "fns.h" enum { Stacksize = 2 * 8192, Chanmsgs = 10, }; enum { Init = 0, Ok, Fail, Quit, Connend, Connerr, Pipeend, Pipeerr, Reconn, Reconnok, }; typedef struct Log Log; typedef struct Ch Ch; typedef struct User User; struct Log { int fd; int mday; }; struct Ch { char name[64]; Log log; List *users; }; struct User { char nick[32]; List *channels; }; int mainstacksize = Stacksize; char *logdir; static char *service; static char *post; static char *file; static char *mynick; static char *user; static char *addr; static int cmdfd = -1; static int ircfd = -1; static int state = Init; static Channel *inic; static Channel *ctlc; static Ioproc *outio; static Ioproc *logio; static TLSconn tls; static int debug; static int timestamps = 1; static int usetls; static int passwd; static int rawlog; static int userdb = 1; static Log mainlog; static Trie *channels; static Trie *users; static void * emalloc(ulong n) { void *p; p = mallocz(n, 1); if(p == nil) sysfatal("mallocz: %r"); return p; } static char * estrdup(char *s) { s = strdup(s); if(s == nil) sysfatal("strdup: %r"); return s; } static int eiowrite(Ioproc *io, int fd, void *buf, long n) { if(iowrite(io, fd, buf, n) != n){ perror("iowrite"); return -1; } return 0; } static long _iosetname(va_list *arg) { char *name; name = va_arg(*arg, char*); threadsetname(name); return 0; } static void iosetname(Ioproc *io, char *name, ...) { va_list arg; char buf[Bufsize]; va_start(arg, name); vsnprint(buf, sizeof(buf), name, arg); va_end(arg); iocall(io, _iosetname, buf); } static Ch * challoc(char *name) { Ch *c; char buf[128]; c = emalloc(sizeof(Ch)); snprint(c->name, sizeof(c->name), "%s", name); if(logdir != nil){ snprint(buf, sizeof(buf), "%s/%s", logdir, name); c->log.fd = create(buf, OWRITE, 0600 | DMAPPEND); if(c->log.fd < 0) sysfatal("create: %r"); c->log.mday = 0; } else c->log.fd = 0; c->users = nil; return c; } static Ch * chget(char *name) { Ch *c; Rune key[64]; runesnprint(key, nelem(key), "%s", name); c = trieget(channels, key); if(c == nil){ c = challoc(name); trieadd(channels, key, c); } return c; } static void chfree(Ch *c) { Rune key[64]; runesnprint(key, nelem(key), "%s", c->name); triedel(channels, key); listfree(&c->users); if(c->log.fd > 0) close(c->log.fd); free(c); } static User * useralloc(char *nick) { User *u; u = emalloc(sizeof(User)); snprint(u->nick, sizeof(u->nick), "%s", nick); u->channels = nil; return u; } static User * userget(char *nick) { User *u; Rune key[32]; runesnprint(key, nelem(key), "%s", nick); u = trieget(users, key); if(u == nil){ u = useralloc(nick); trieadd(users, key, u); } return u; } static void userfree(User *u) { Rune key[32]; runesnprint(key, nelem(key), "%s", u->nick); triedel(users, key); listfree(&u->channels); free(u); } static void usernick(User *u, char *newnick) { Rune key[32]; runesnprint(key, nelem(key), "%s", u->nick); triedel(users, key); snprint(u->nick, sizeof(u->nick), "%s", newnick); runesnprint(key, nelem(key), "%s", u->nick); trieadd(users, key, u); } static void ircsend(char *msg) { long len; len = strlen(msg); if(debug) fprint(2, "ircsend: %s", msg); if(len > 0){ eiowrite(outio, ircfd, msg, len); if(msg[len-1] != '\n'){ if(debug) fprint(2, "\n"); eiowrite(outio, ircfd, "\r\n", 2); } } } static void logsend(char *msg, long time, Log *log) { long len, n; char buf[Bufsize]; len = strlen(msg); if(len > 0 && log->fd > 0){ if(timestamps){ if(rawlog) n = snprint(buf, sizeof(buf), "%ld ", time); else n = ircfmttime(buf, sizeof(buf), time, &log->mday); eiowrite(logio, log->fd, buf, n); } n = len; eiowrite(logio, log->fd, msg, n); if(msg[n-1] != '\n') eiowrite(logio, log->fd, "\n", 1); } } static void loginfo(char *msg) { Triewalk w; Rune key[64]; Ch *c; long t; t = time(0); logsend(msg, t, &mainlog); if(logdir != nil){ triewalk(channels, &w, key, nelem(key)); while((c = trienext(&w)) != nil) logsend(msg, t, &c->log); } } static void joinmsg(Ircmsg *irc, char *msg, long time) { Ch *c; Log *log; User *u; char *channel, *m, buf[Bufsize]; channel = irc->par[0]; if(rawlog) m = msg; else { ircfmtjoin(irc, buf, sizeof(buf)); m = buf; } if(logdir != nil){ c = chget(channel); log = &c->log; if(userdb && irc->pre != nil){ u = userget(irc->nick); listadd(&c->users, u); listadd(&u->channels, c); } } else { log = &mainlog; /* for rejoin() */ if(irc->pre != nil && strcmp(irc->nick, mynick) == 0) chget(channel); } logsend(m, time, log); } static void freeusers(Ch *c) { List *i; User *u; for(i = c->users; i != nil; i = i->next){ u = i->val; listdel(&u->channels, c); if(u->channels == nil) userfree(u); } listfree(&c->users); } static void partmsg(Ircmsg *irc, char *msg, long time) { Ch *c; User *u; char *channel, *m, buf[Bufsize]; channel = irc->par[0]; if(rawlog) m = msg; else { ircfmtpart(irc, buf, sizeof(buf)); m = buf; } if(logdir != nil){ c = chget(channel); if(userdb && irc->pre != nil){ u = userget(irc->nick); listdel(&c->users, u); listdel(&u->channels, c); if(u->channels == nil) userfree(u); } logsend(m, time, &c->log); if(irc->pre != nil && strcmp(irc->nick, mynick) == 0){ freeusers(c); chfree(c); } } else { logsend(m, time, &mainlog); /* for rejoin() */ if(irc->pre != nil && strcmp(irc->nick, mynick) == 0){ c = chget(channel); freeusers(c); chfree(c); } } } static void quitmsg(Ircmsg *irc, char *msg, long time) { User *u; List *i; Ch *c; char *m, buf[Bufsize]; if(rawlog) m = msg; else { ircfmtquit(irc, buf, sizeof(buf)); m = buf; } if(logdir != nil && userdb && irc->pre != nil){ u = userget(irc->nick); for(i = u->channels; i != nil; i = i->next){ c = i->val; logsend(m, time, &c->log); listdel(&c->users, u); } if(u->channels == nil) logsend(m, time, &mainlog); userfree(u); } else logsend(m, time, &mainlog); } static void nickmsg(Ircmsg *irc, char *msg, long time) { User *u; List *i; Ch *c; char *newnick, *m, buf[Bufsize]; newnick = irc->par[0]; if(rawlog) m = msg; else { ircfmtnick(irc, buf, sizeof(buf)); m = buf; } if(logdir != nil && userdb && irc->pre != nil){ u = userget(irc->nick); for(i = u->channels; i != nil; i = i->next){ c = i->val; logsend(m, time, &c->log); } if(u->channels == nil) logsend(m, time, &mainlog); usernick(u, newnick); } else logsend(m, time, &mainlog); if(irc->pre != nil && strcmp(irc->nick, mynick) == 0){ free(mynick); mynick = estrdup(newnick); } } static void namreply(Ircmsg *irc, char *msg, long time) { Ch *c; Log *log; User *u; char *channel, *nick, *m, *a[128], buf[Bufsize]; int i, n; if(rawlog) m = msg; else { ircfmtnumeric(irc, buf, sizeof(buf)); m = buf; } if(logdir != nil){ channel = irc->par[2]; c = chget(channel); log = &c->log; if(userdb){ n = getfields(irc->trail, a, nelem(a), 1, " "); for(i = 0; i < n; i++){ nick = a[i]; if(*nick == '@' || *nick == '+'){ nick++; if(*nick == 0) continue; } u = userget(nick); listadd(&c->users, u); listadd(&u->channels, c); } } } else log = &mainlog; logsend(m, time, log); } static void srvmsg(char *msg, long time) { Ircmsg irc; Ircfmt fmt; Ch *c; Log *log; char *target, *m, buf[Bufsize]; ircparse(&irc, msg); if(debug) ircprint(&irc); fmt = nil; target = nil; if(strcmp(irc.cmd, "PRIVMSG") == 0){ fmt = ircfmtpriv; target = irc.par[0]; } else if(strcmp(irc.cmd, "JOIN") == 0){ joinmsg(&irc, msg, time); return; } else if(strcmp(irc.cmd, "QUIT") == 0){ quitmsg(&irc, msg, time); return; } else if(strcmp(irc.cmd, "PART") == 0){ partmsg(&irc, msg, time); return; } else if(strcmp(irc.cmd, "NICK") == 0){ nickmsg(&irc, msg, time); return; } else if(strcmp(irc.cmd, "MODE") == 0 || strcmp(irc.cmd, "KICK") == 0 || strcmp(irc.cmd, "TOPIC") == 0){ target = irc.par[0]; } else if(strcmp(irc.cmd, "001") == 0){ /* welcome */ sendul(ctlc, Ok); } else if(strcmp(irc.cmd, "433") == 0){ /* nick in use */ sendul(ctlc, Fail); } else if(strcmp(irc.cmd, "332") == 0 || strcmp(irc.cmd, "333") == 0 || strcmp(irc.cmd, "366") == 0){ /* 332 = RPL_TOPIC */ /* 333 = RPL_TOPICWHOTIME */ /* 366 = RPL_ENDOFNAMES */ target = irc.par[1]; } else if(strcmp(irc.cmd, "353") == 0){ /* 353 = RPL_NAMREPLY */ namreply(&irc, msg, time); return; } if(irc.cmd[0] >= '0' && irc.cmd[0] <= '9') fmt = ircfmtnumeric; if(logdir != nil && ircischan(target)){ c = chget(target); log = &c->log; } else log = &mainlog; if(rawlog || fmt == nil) m = msg; else { fmt(&irc, buf, sizeof(buf)); m = buf; } logsend(m, time, log); } static void usrmsg(char *msg, long time) { Ircmsg irc; Ircfmt fmt; Ch *c; Log *log; char *target, *m, buf[Bufsize]; ircparse(&irc, msg); if(debug) ircprint(&irc); fmt = nil; target = nil; if(strcmp(irc.cmd, "PRIVMSG") == 0){ fmt = ircfmtpriv; target = irc.par[0]; } else if(strcmp(irc.cmd, "QUIT") == 0){ state = Quit; } if(logdir != nil && ircischan(target)){ c = chget(target); log = &c->log; } else log = &mainlog; if(rawlog || fmt == nil) m = msg; else { fmt(&irc, buf, sizeof(buf)); m = buf; } logsend(m, time, log); ircsend(msg); } static void touch(int fd, long time) { Dir d; nulldir(&d); d.mtime = time; if(dirfwstat(fd, &d) < 0) perror("dirfwstat"); } static void ircin(void *v) { Ioproc *io; char buf[Bufsize], *e, *p; long n, t; io = v; threadsetname("ircin"); iosetname(io, "%s net reader", service); p = buf; e = buf + sizeof(buf); while((n = ioread(io, ircfd, p, e - p)) > 0){ t = time(0); p += n; while((p > buf) && (e = memchr(buf, '\n', p - buf))){ if((e > buf) && (e[-1] == '\r')) e[-1] = 0; *e++ = 0; if(strncmp(buf, "PING", 4) == 0){ buf[1] = 'O'; ircsend(buf); touch(mainlog.fd, t); } else srvmsg(buf, t); p -= e - buf; if(p > buf) memmove(buf, e, p - buf); } e = buf + sizeof(buf); } if(n < 0){ perror("ircin: ioread"); loginfo("ircs: connection error"); sendul(ctlc, Connerr); } if(n == 0){ loginfo("ircs: connection eof"); sendul(ctlc, Connend); } closeioproc(io); threadexits(nil); } static void cmdin(void *v) { Ioproc *io; char buf[Bufsize]; long n, t; io = v; threadsetname("cmdin"); iosetname(io, "%s cmd reader", service); while((n = ioread(io, cmdfd, buf, sizeof(buf)-1)) > 0){ t = time(0); buf[n] = 0; if(debug) fprint(2, "cmdin: %s", buf); if(strncmp(buf, "quit", 4) == 0){ sendul(ctlc, Quit); break; } if(buf[n-1] == '\n') buf[n-1] = 0; if(n > 1 && buf[n-2] == '\r') buf[n-2] = 0; usrmsg(buf, t); } if(n < 0){ perror("cmdin: ioread"); loginfo("ircs: command pipe error"); sendul(ctlc, Pipeerr); } if(n == 0){ loginfo("ircs: command pipe eof"); sendul(ctlc, Pipeend); } closeioproc(io); threadexits(nil); } static void cleanup(void) { if(access(post, AEXIST) == 0) if(remove(post) < 0) perror("remove"); } static int note(void *, char *msg) { fprint(2, "%d: received note: %s\n", getpid(), msg); cleanup(); threadexitsall("note"); return 0; } static void usage(void) { fprint(2, "usage: %s [-deprTU] [-s srvname] [-f file] nick[!user] addr\n", argv0); threadexits("usage"); } static int connect(void) { UserPasswd *u; char buf[Bufsize]; long t; int fd; u = nil; if(passwd && usetls){ u = auth_getuserpasswd(auth_getkey, "proto=pass service=irc server=%q user=%q", addr, mynick); if(u == nil){ perror("auth_getuserpasswd"); return -1; } } if(ircfd >= 0) close(ircfd); ircfd = dial(netmkaddr(addr, "tcp", usetls ? "6697" : "6667"), nil, nil, nil); if(ircfd < 0){ perror("dial"); free(u); return -1; } if(usetls){ memset(&tls, 0, sizeof(tls)); fd = tlsClient(ircfd, &tls); if(fd < 0){ perror("tlsClient"); close(ircfd); free(u); return -1; } ircfd = fd; } t = time(0); if(u != nil){ snprint(buf, sizeof(buf), "PASS"); logsend(buf, t, &mainlog); snprint(buf+4, sizeof(buf)-4, " %s", u->passwd); ircsend(buf); free(u); } snprint(buf, sizeof(buf), "USER %s foo bar :", user); logsend(buf, t, &mainlog); ircsend(buf); snprint(buf, sizeof(buf), "NICK %s", mynick); logsend(buf, t, &mainlog); ircsend(buf); return 0; } static void freeallusers(void) { Triewalk w; Rune key[64]; Ch *c; triewalk(channels, &w, key, nelem(key)); while((c = trienext(&w)) != nil) freeusers(c); } static void rejoin(void) { Triewalk w; Rune key[64]; Ch *c; long t; char buf[Bufsize]; t = time(0); triewalk(channels, &w, key, nelem(key)); while((c = trienext(&w)) != nil){ snprint(buf, sizeof(buf), "JOIN %s", c->name); logsend(buf, t, &mainlog); ircsend(buf); } } static void reconnproc(void *) { int c, i; long sec[] = { 5, 30, 60, 300, 900 }; threadsetname("%s reconnect", service); c = 0; i = 0; while(connect() < 0){ c++; threadsetname("%s reconnect attempts=%d", service, c); sleep(sec[i] * 1000); if(i < nelem(sec)-1) i++; } sendul(ctlc, Reconnok); threadexits(nil); } static void mainproc(void *) { ulong msg; threadnotify(note, 1); threadsetname("%s %s %s %s", mynick, addr, post, logdir != nil ? logdir : file); outio = ioproc(); logio = ioproc(); iosetname(outio, "%s net writer", service); iosetname(logio, "%s log writer", service); if(connect() < 0){ sendul(inic, Fail); threadexits("no"); } threadcreate(cmdin, ioproc(), Stacksize); threadcreate(ircin, ioproc(), Stacksize); for(;;){ msg = recvul(ctlc); if(state == Init) sendul(inic, msg); switch(msg){ case Ok: state = Ok; break; case Connend: case Connerr: if(state == Quit) goto done; if(state != Init){ state = Reconn; if(userdb) freeallusers(); loginfo("ircs: reconnecting"); proccreate(reconnproc, nil, Stacksize); } break; case Reconnok: state = Reconnok; rejoin(); threadcreate(ircin, ioproc(), Stacksize); break; case Pipeend: case Pipeerr: case Fail: case Quit: if(state != Init) goto done; break; default: assert(0); } } done: cleanup(); threadexitsall(nil); } static void postpipe(void) { int fd, p[2]; assert(service != nil && *service != 0); post = smprint("/srv/%s", service); if(post == nil) sysfatal("smprint: %r"); fd = create(post, OWRITE, 0600); if(fd < 0) sysfatal("create: %r"); if(pipe(p) < 0) sysfatal("pipe: %r"); if(fprint(fd, "%d", p[1]) < 0) sysfatal("can't post: %r"); close(fd); close(p[1]); if(dup(p[0], 0) < 0) sysfatal("dup: %r"); close(p[0]); cmdfd = 0; } static void initlog(void) { int fd; if(file == nil){ assert(service != nil && *service != 0); logdir = smprint("/tmp/%s", service); if(logdir == nil) sysfatal("smprint: %r"); if(access(logdir, AEXIST) < 0){ fd = create(logdir, OREAD, DMDIR | 0700); if(fd < 0) sysfatal("create: %r"); close(fd); } file = smprint("%s/log", logdir); if(file == nil) sysfatal("smprint: %r"); } fd = create(file, OWRITE, 0600 | DMAPPEND); if(fd < 0) sysfatal("create: %r"); if(dup(fd, 1) < 0) sysfatal("dup: %r"); close(fd); mainlog.fd = 1; mainlog.mday = 0; } static void setnickuser(char *nickuser) { char *p; if(p = strchr(nickuser, '!')){ *p = 0; if(*(++p) != 0) user = estrdup(p); else user = estrdup(nickuser); } else user = estrdup(nickuser); if(*nickuser != 0) mynick = estrdup(nickuser); else sysfatal("empty nick"); } void threadmain(int argc, char *argv[]) { ARGBEGIN{ case 'd': debug++; break; case 'e': usetls++; break; case 'f': file = EARGF(usage()); break; case 'p': passwd++; break; case 'r': rawlog++; break; case 's': service = EARGF(usage()); break; case 'T': timestamps = 0; break; case 'U': userdb = 0; break; default: usage(); }ARGEND if(argc != 2) usage(); if(passwd && !usetls){ fprint(2, "-p requires -e, exiting\n"); threadexits("passwd"); } if(file != nil) userdb = 0; if(service == nil) service = estrdup("ircs"); else if(*service == 0) sysfatal("empty srvname"); setnickuser(argv[0]); addr = argv[1]; fprint(2, "initializing, please wait\n"); initlog(); postpipe(); threadnotify(note, 1); inic = chancreate(sizeof(ulong), Chanmsgs); ctlc = chancreate(sizeof(ulong), Chanmsgs); if(inic == nil || ctlc == nil) sysfatal("chancreate"); channels = triealloc(); users = triealloc(); procrfork(mainproc, nil, Stacksize, RFNOTEG); switch(recvul(inic)){ case Ok: break; default: fprint(2, "init failed, exiting\n"); cleanup(); threadexitsall("no"); } fprint(2, "init ok\n"); chanclose(inic); threadexits(nil); } ick); } } static void namreply(Ircmsg *irc, char *msg, long time) { Ch *c; Log *log; User *u; char *channel, *nick, *m, *a[128], buf[Bufsize]; int i, n; if(rawlog) m = msg; else { ircfmtnumeric(irc, buf, sizeof(buf)); m = buf; } if(logdir != nil){ channel = irc->par[2]; c = chget(channel);ircs-55/trie.c 664 0 0 6167 13052004337 10775ustar00jrmjrm#include #include #include "dat.h" int trieallocs; int triedebug; #define dprint if(triedebug) print Trie * triealloc(void) { Trie *t; t = mallocz(sizeof(Trie), 1); if(t == nil) sysfatal("mallocz: %r"); trieallocs++; return t; } void triefree(Trie *t) { Trie *c, *n; for(c = t->child; c != nil; c = n){ n = c->next; triefree(c); } free(t); trieallocs--; } void triewalk(Trie *t, Triewalk *w, Rune *key, int keysize) { w->root = t; w->curr = nil; w->key = key; w->keysize = keysize; w->depth = 0; } void * trienext(Triewalk *w) { Trie *t; int d; if(w->curr == nil){ w->curr = w->root; w->depth = 0; w->key[0] = 0; if(w->root->value != nil) return w->root->value; } t = w->curr; d = w->depth; do{ if(t->child != nil && d < w->keysize-1){ t = t->child; d++; } else if(t->next != nil) t = t->next; else { while(t->next == nil && t != w->root){ t = t->parent; d--; } if(t->next != nil) t = t->next; else { assert(t == w->root); w->curr = nil; return nil; } } w->key[d-1] = t->rune; } while(t->value == nil); w->key[d] = 0; w->curr = t; w->depth = d; return t->value; } static Trie * getchild(Trie *t, Rune r) { dprint(" getchild %C: ", r); for(t = t->child; t != nil; t = t->next) if(t->rune == r) break; dprint("%s\n", t != nil ? "found" : "not found"); return t; } static Trie * addchild(Trie *t, Rune r) { Trie *c, *n, *prev; dprint(" addchild %C\n", r); c = triealloc(); c->rune = r; c->parent = t; prev = nil; for(n = t->child; n != nil; n = n->next){ if((prev == nil || prev->rune < r) && r < n->rune) break; assert(r > n->rune); prev = n; } if(prev == nil) t->child = c; else prev->next = c; c->next = n; return c; } static int delchild(Trie *t, Rune r) { Trie *c, *prev; dprint(" delchild %C\n", r); prev = nil; for(c = t->child; c != nil; c = c->next){ if(c->rune == r) break; prev = c; } if(c == nil) return 0; if(c->child != nil) return 0; if(prev == nil) t->child = c->next; else prev->next = c->next; triefree(c); return 1; } void * trieadd(Trie *t, Rune *key, void *val) { Trie *c; void *v; if(val == nil) return nil; while(*key){ c = getchild(t, *key); if(c == nil) break; t = c; key++; } while(*key) t = addchild(t, *key++); v = t->value; t->value = val; return v; } static Trie * find(Trie *t, Rune *key) { Trie *c; while(*key){ c = getchild(t, *key); if(c == nil) break; t = c; key++; } if(*key) return nil; return t; } void * triedel(Trie *t, Rune *key) { Rune *r; void *v; t = find(t, key); if(t == nil) return nil; /* only non-leaves and root can have empty value */ if(t->value == nil){ assert(t->child != nil || t->parent == nil); return nil; } v = t->value; t->value = nil; if(t->child != nil || t->parent == nil) return v; r = runestrchr(key, 0); while(t->child == nil && t->value == nil && --r >= key){ t = t->parent; assert(t != nil); assert(delchild(t, *r)); } return v; } void * trieget(Trie *t, Rune *key) { t = find(t, key); if(t == nil) return nil; return t->value; } if(strncmp(buf, "quit", 4) == 0){ sendul(ctlc, Quit); break; } if(buf[n-1] == '\n') buf[n-1] = 0; if(n > 1 && buf[n-2] == '\r') buf[n-2] = 0; usrmsg(buf, t); } if(n < 0){ perror("cmdin: ioread"); loginfo("ircs: command pipe error"); sendul(ctlc, Pipeerr); } if(n == 0){ loginfo("ircs: command pipe eof"); sendul(ctlc, Pipeend); } closeioproc(io); threadexircs-55/ircs.man 664 0 0 12026 13221627747 11350ustar00jrmjrm.TH IRCS 1 .SH NAME ircs, ircx \- internet relay chat client and ui .SH SYNOPSIS ircs [ -deprTU ] [ -s srvname ] [ -f file ] nick[!user] addr ircx [ -eprTU ] [ -s srvname ] [ -f file ] [ -t target ] [ -b lines ] [ nick[!user] addr ] irctime [ file ... ] .SH DESCRIPTION Ircs is a persistent, logging IRC client inspired by the classic Plan 9 IRC client irc7. Its usage is similar to irc7's "ircsrv": it makes a connection to an IRC server, posts its command pipe in srv(3), and logs incoming and outgoing messages to a log directory, or to a given file. Ircs dials the IRC server at addr and uses nick as the IRC nickname. The addr is a network address understood by dial(2). The default port number is 6667 unless the -e option (enable TLS encryption) is given, in which case the default port is 6697. The username user can be specified by separating the nick and the user with an exclamation mark ('!'). If no user is given the nick is used also as the username. The user's (raw) IRC commands are read via a command pipe which by default is posted to service file /srv/ircs. The log is written by default to files under /tmp/ircs directory; server and private messages are written to the main log file /tmp/ircs/log, and messages related to channels are written to files named /tmp/ircs/#channel, where "#channel" is the name of the channel. The -s option changes the service file to /srv/srvname and the log directory to /tmp/srvname. The -f option causes all logs to be written to a given file instead of a directory. All messages are logged apart from PING/PONG; when a PING is received the modification time of the (main) log file is set to the current time. When logging to a directory, ircs attempts to keep track of all users (i.e. nicks) on joined channels in order to log incoming NICK/QUIT messages to related channel log files. When ircs cannot determine the channel(s) a particular user is on, or if the option -U (disable user tracking) is given, incoming NICK/QUIT messages are logged to the main log file. The -d option causes some debug info to be printed to standard error. The -p option causes a password to be sent to the server. The password is read via the authentication agent factotum(4) which must be mounted in the same namespace. The -p option can only be used with the -e (use TLS) option. By default the logs are formatted so that they can easily be processed with Plan 9 tools, like cat(1) or grep(1). The log lines are preceded with timestamps in a human-readable format. The -r option disables this formatting and enables raw logging with timestamps in the Unix time format. The -T option disables timestamps altogether. Normally a user need not deal with ircs directly, but can use the ircx wrapper script, whose usage is nearly identical to ircs. Ircx is a simple rc(1) user interface to ircs, similar to irc7's client program "irc". It makes use of the service provided by an ircs instance, and provides an interface for starting and stopping ircs and for sending and displaying messages. Unlike in irc7, where the most functionality is in the client part, in ircs the main functionality is intended to be in the service part. With nick and addr arguments ircx attempts to start an ircs instance; if the ircs starts successfully, ircx then runs the user interface: the related log file is printed to standard output as it accumulates, and user's commands are read from standard input. Without nick and addr arguments ircx assumes that the ircs instance is already running, and only the user interface is run. Ircx can be exited by pressing ; the /h command prints the list of known commands. The -t option sets the target (channel or nick) to which the input commands apply. If the target is a channel (the first character is '#'), the channel is joined. The -b option sets the number of lines that are printed from the end of the log file when ircx starts. An ircs instance is defined by its service name (and log file if -f option is used). Like ircs, ircx uses by default the service file /srv/ircs and log directory /tmp/ircs; the service name "ircs" can be changed by the -s option. When running multiple ircs instances, like when connecting to multiple IRC networks, the -s and (optionally) -f options should be used to differentiate between the instances. Irctime copies the given files (standard input by default) to its standard output, converting the default human-readable ircs timestamps to ISO 8601 format. .SH EXAMPLES Connect to freenode using nick and username "foo": .IP .EX ircx foo chat.freenode.net .EE .PP Join the channel "#cat-v" assuming ircs is already running: .IP .EX ircx -t '#cat-v' .EE .PP Follow server and private messages assuming ircs is already running: .IP .EX ircx .EE .PP Check what ircs instances are running: .IP .EX ps -a | grep ircs .EE .SH SOURCE http://www.plan9.fi/src/ircs.tgz .SH SEE ALSO https://tools.ietf.org/html/rfc2812 .SH BUGS The -t (set target) option does not affect the output of ircx when the -f (set log file) option is used; ircx prints out all messages from the log file regardless of the target. ic)){ case Ok: break; default: fprint(2, "init failed, exiting\n"); cleanup(); threadexitsall("no"); } fprint(2, "init ok\n"); chanclose(inic); threadexits(nil); } ick); } } static void namreply(Ircmsg *irc, char *msg, long time) { Ch *c; Log *log; User *u; char *channel, *nick, *m, *a[128], buf[Bufsize]; int i, n; if(rawlog) m = msg; else { ircfmtnumeric(irc, buf, sizeof(buf)); m = buf; } if(logdir != nil){ channel = irc->par[2]; c = chget(channel);ircs-55/ircs.txt 664 0 0 14004 13221630027 11373ustar00jrmjrm IRCS(1) IRCS(1) NAME ircs, ircx - internet relay chat client and ui SYNOPSIS ircs [ -deprTU ] [ -s srvname ] [ -f file ] nick[!user] addr ircx [ -eprTU ] [ -s srvname ] [ -f file ] [ -t target ] [ -b lines ] [ nick[!user] addr ] irctime [ file ... ] DESCRIPTION Ircs is a persistent, logging IRC client inspired by the classic Plan 9 IRC client irc7. Its usage is similar to irc7's "ircsrv": it makes a connection to an IRC server, posts its command pipe in srv(3), and logs incoming and outgoing messages to a log directory, or to a given file. Ircs dials the IRC server at addr and uses nick as the IRC nickname. The addr is a network address understood by dial(2). The default port number is 6667 unless the -e option (enable TLS encryption) is given, in which case the default port is 6697. The username user can be specified by separating the nick and the user with an exclamation mark ('!'). If no user is given the nick is used also as the username. The user's (raw) IRC commands are read via a command pipe which by default is posted to service file /srv/ircs. The log is written by default to files under /tmp/ircs directory; server and private messages are written to the main log file /tmp/ircs/log, and messages related to channels are written to files named /tmp/ircs/#channel, where "#channel" is the name of the channel. The -s option changes the service file to /srv/srvname and the log directory to /tmp/srvname. The -f option causes all logs to be written to a given file instead of a directory. All messages are logged apart from PING/PONG; when a PING is received the modification time of the (main) log file is set to the current time. When logging to a directory, ircs attempts to keep track of all users (i.e. nicks) on joined channels in order to log incoming NICK/QUIT messages to related channel log files. When ircs cannot determine the channel(s) a particular user is on, or if the option -U (disable user tracking) is given, incoming NICK/QUIT messages are logged to the main log file. The -d option causes some debug info to be printed to standard error. The -p option causes a password to be sent to the server. The password is read via the authentication agent factotum(4) which must be mounted in the same namespace. The -p option can only be used with the -e (use TLS) option. By default the logs are formatted so that they can easily be processed with Plan 9 tools, like cat(1) or grep(1). The log lines are preceded with timestamps in a human-readable format. The -r option disables this formatting and enables raw logging with timestamps in the Unix time format. The -T option disables timestamps altogether. Normally a user need not deal with ircs directly, but can use the ircx wrapper script, whose usage is nearly identical to ircs. Ircx is a simple rc(1) user interface to ircs, similar to irc7's client program "irc". It makes use of the service provided by an ircs instance, and provides an interface for starting and stopping ircs and for sending and displaying messages. Unlike in irc7, where the most functionality is in the client part, in ircs the main functionality is intended to be in the service part. With nick and addr arguments ircx attempts to start an ircs instance; if the ircs starts successfully, ircx then runs the user interface: the related log file is printed to standard output as it accumulates, and user's commands are read from standard input. Without nick and addr arguments ircx assumes that the ircs instance is already running, and only the user interface is run. Ircx can be exited by pressing ; the /h command prints the list of known commands. The -t option sets the target (channel or nick) to which the input commands apply. If the target is a channel (the first character is '#'), the channel is joined. The -b option sets the number of lines that are printed from the end of the log file when ircx starts. An ircs instance is defined by its service name (and log file if -f option is used). Like ircs, ircx uses by default the service file /srv/ircs and log directory /tmp/ircs; the service name "ircs" can be changed by the -s option. When running multiple ircs instances, like when connecting to multiple IRC networks, the -s and (optionally) -f options should be used to differentiate between the instances. Irctime copies the given files (standard input by default) to its standard output, converting the default human- readable ircs timestamps to ISO 8601 format. EXAMPLES Connect to freenode using nick and username "foo": ircx foo chat.freenode.net Join the channel "#cat-v" assuming ircs is already running: ircx -t '#cat-v' Follow server and private messages assuming ircs is already running: ircx Check what ircs instances are running: ps -a | grep ircs SOURCE http://www.plan9.fi/src/ircs.tgz SEE ALSO https://tools.ietf.org/html/rfc2812 BUGS The -t (set target) option does not affect the output of ircx when the -f (set log file) option is used; ircx prints out all messages from the log file regardless of the target. causes some debug info to be printed to standard error. The -p option causes a password to be sent to the server. The password is read via the authentication agent factotum(4) which must be mounted in the same namespace. The -p option can only be used with the -e (use TLS) option. By default the logs are formatted so that they can easily be processed with Plan 9 tools, like cat(1) or grep(1). The log lines are preceded with timestamps in a human-readable format. The -r option disables this formattircs-55/ircx 775 0 0 11462 13221626501 10574ustar00jrmjrm#!/bin/rc # ircx: ircs ui rfork e flagfmt='e,p,r,T,U,s srvname,f file,t target,b lines' args='[nick[!user] addr]' srv=ircs mainlog=() log=() opt=() lines=() target=() offset=() help=', /q quit ircx (leaves ircs running) /h print this help /j #channel join #channel and set it as target /n nick change nick /N list nicks on channel target /p [message] part from channel target /Q [message] quit (stops ircs) /r message send raw message to server /t target set target (channel or nick) /T [topic] get [set] topic of channel target /w mask send who query /W mask send whois query /x create new ircx window [/ ]message send message to target' fn usage{ aux/usage exit usage } fn fatal{ echo $* >[1=2] exit $"* } fn isnum{ echo $1 | grep -s '^[0-9]+$' } fn clear{ awk 'BEGIN { for(i = 0; i < '$1'; i++) printf "\x08" }' } fn runtail{ { tail -$2 while(sleep 1) cat } <$1 & tailpid=$apid } fn oldlabel{ if(! ~ $#oldlabel 0) label $oldlabel } fn killtail{ @{ echo kill >/proc/$tailpid/note } & } fn saveoffset{ if(~ $#offset 1 && test -r $log) wc -l <$log | tr -d ' ' >$offset } fn shutdown{ saveoffset killtail oldlabel exit 0 } fn sighup sigint sigterm{ shutdown } fn send{ if(test -e /srv/$srv) echo $* >>/srv/$srv if not{ echo /srv/$srv not found '(ircs stopped?)' shutdown } } fn settarget{ if(~ $1 ''){ target=() label $mainlog } if not{ target=$1 label $target } } fn newtarget{ if(~ $#flagf 1) settarget $newtarget if not{ if(~ $newtarget '#'*){ f=/tmp/$srv/$newtarget sleep 1 } if not{ settarget $newtarget f=$mainlog } if(! ~ $f $log){ if(test -r $f){ settarget $newtarget saveoffset offset=() killtail log=$f echo $log runtail $log 10 } if not echo can''''t access $f } } newtarget=() } fn lines{ if(~ $#offset 1 && test -r $offset && test -r $log){ n₀=`{cat $offset} n₁=`{wc -l <$log} δn=`{echo $n₁ - $n₀ + 10 | bc} echo $δn } if not echo 10 } if(! ifs=() eval `{aux/getflags $*} || ! ~ $#* 0 2) usage if(~ $#flagb 1){ if(! isnum $flagb) usage lines=$flagb } if(~ $#flagt 1) target=$flagt if(~ $#flags 1){ if(~ $flags '') fatal empty srvname srv=$flags opt=($opt -s $srv) } if(~ $#flagf 1){ mainlog=$flagf opt=($opt -f $mainlog) } if not{ mainlog=/tmp/$srv/log if(~ $#lines 0){ offset=/tmp/$srv/offsets/log mkdir -p /tmp/$srv/offsets } } if(~ $#flage 1) opt=($opt -e) if(~ $#flagp 1) opt=($opt -p) if(~ $#flagr 1) opt=($opt -r) if(~ $#flagT 1) opt=($opt -T) if(~ $#flagU 1) opt=($opt -U) switch($#*){ case 0 for(i in /srv/$srv $mainlog) if(! test -e $i) fatal $i not found case 2 if(test -e /srv/$srv) fatal ircs already running nick=$1 addr=$2 ircs $opt $nick $addr stat=$status if(! ~ $stat '') exit $stat case * usage } oldlabel=`{cat /dev/label} oifs=$ifs log=$mainlog if(~ $#target 1){ label $target if(~ $target '#'*){ send JOIN $target if(~ $#flagf 0){ log=/tmp/$srv/$target if(~ $#lines 0) offset=/tmp/$srv/offsets/$target if(! test -e $log) sleep 1 } } } if not label $log if(~ $#lines 0) lines=`{lines} echo $log -$lines runtail $log $lines while(ifs=() line=`{read | tr -d \xa}){ ifs=$oifs clear `{echo $line | wc -r} if(! ~ $line ''){ msg=() *=`{echo $line} cmd=$1 shift switch($cmd){ case /h echo $help case /j if(~ $#* 0) echo no channel if not{ msg='JOIN '^$"* newtarget=$1 } case /t if(~ $#* 0) echo no target if not newtarget=$1 case /N if(! ~ $target '#'*) echo no channel if not msg='NAMES '^$target case /T if(! ~ $target '#'*) echo no channel if not{ msg='TOPIC '^$target if(! ~ $#* 0) msg=$msg^' :'^$"* } case /p if(! ~ $target '#'*) echo no channel if not{ msg='PART '^$target if(! ~ $#* 0) msg=$msg^' :'^$"* newtarget='' } case /q shutdown case /Q if(~ $#* 0) send QUIT if not send 'QUIT :'^$"* sleep 1 shutdown case /r if(~ $#* 0) echo no message if not{ ifs=() msg=`{echo -n $line | sed 's!^[ ]*/r[ ]+!!'} ifs=$oifs } case /n if(~ $#* 0) echo no nick if not msg='NICK '^$1 case /w if(~ $#* 0) echo no mask if not msg='WHO '^$"* case /W if(~ $#* 0) echo no mask if not msg='WHOIS '^$"* case /x window -m ircx $opt case / if(~ $#target 0) echo no target if not{ ifs=() line=`{echo -n $line | sed 's!^[ ]*/[ ]*!!'} ifs=$oifs msg='PRIVMSG '^$target^' :'^$line } case /* echo unknown command case * if(~ $#target 0) echo no target if not msg='PRIVMSG '^$target^' :'^$line } if(! ~ $#msg 0) send $msg if(~ $#newtarget 1) newtarget } } shutdown a given file instead of a directory. All messages are logged apart from PING/PONG; when a PING is received the modification time of the (main) log file is set to the current ircs-55/mkfile 664 0 0 334 13111452350 11024ustar00jrmjrm[2]/dev/null}) dir=ircs-$rev if not dir=ircs mkdir $dir cp -x *.[ch] ircs.man ircs.txt ircx mkfile \ pack irctime nojqpn $dir tar cvzf ircs.tgz $dir rm -rf $dir ircs-55/irctime 775 0 0 621 13052302426 11215ustar00jrmjrm#!/bin/rc # irctime: convert ircs timestamps to ISO 8601 format awk ' BEGIN { mon["Jan"] = "01" mon["Feb"] = "02" mon["Mar"] = "03" mon["Apr"] = "04" mon["May"] = "05" mon["Jun"] = "06" mon["Jul"] = "07" mon["Aug"] = "08" mon["Sep"] = "09" mon["Oct"] = "10" mon["Nov"] = "11" mon["Dec"] = "12" } /^-- / { dd = $3 mm = mon[$4] yyyy = $5 next } { print yyyy "-" mm "-" dd "T" $0 } ' $* ircs-55/nojqpn 775 0 0 137 13111450743 11072ustar00jrmjrm#!/bin/rc # nojqpn: filter out JOIN/QUIT/PART/NICK grep -v '^[0-9:]+ (JOIN|QUIT|PART|NICK)' $*