ref: c5df4bd968e0c643de952eac7287497094334b62
author: kemal <kemalinanc8@gmail.com>
date: Sun Oct 3 04:14:42 EDT 2021
init
--- /dev/null
+++ b/dat.h
@@ -1,0 +1,50 @@
+enum {
+ 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;
--- /dev/null
+++ b/fns.h
@@ -1,0 +1,20 @@
+int 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 *);
--- /dev/null
+++ b/irc.c
@@ -1,0 +1,100 @@
+#include <u.h>
+#include <libc.h>
+#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] == '#';
+}
--- /dev/null
+++ b/ircfmt.c
@@ -1,0 +1,140 @@
+#include <u.h>
+#include <libc.h>
+#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;
+ }
+}
--- /dev/null
+++ b/ircs.man
@@ -1,0 +1,141 @@
+.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 <del>; 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.
--- /dev/null
+++ b/irctime
@@ -1,0 +1,27 @@
+#!/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
+}
+' $*
--- /dev/null
+++ b/ircx
@@ -1,0 +1,317 @@
+#!/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='<del>, /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
--- /dev/null
+++ b/list.c
@@ -1,0 +1,60 @@
+#include <u.h>
+#include <libc.h>
+#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;
+}
--- /dev/null
+++ b/main.c
@@ -1,0 +1,1049 @@
+#include <u.h>
+#include <libc.h>
+#include <auth.h>
+#include <libsec.h>
+#include <thread.h>
+#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 :<nil>", 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);
+}
--- /dev/null
+++ b/mkfile
@@ -1,0 +1,19 @@
+</$objtype/mkfile
+
+BIN=$home/bin/$objtype
+TARG=ircs
+OFILES=\
+ irc.$O\
+ ircfmt.$O\
+ list.$O\
+ main.$O\
+ trie.$O\
+
+HFILES=\
+ dat.h\
+ fns.h\
+
+</sys/src/cmd/mkone
+
+install:V: $BIN/$TARG
+ cp ircx irctime nojqpn $home/bin/rc/
--- /dev/null
+++ b/nojqpn
@@ -1,0 +1,3 @@
+#!/bin/rc
+# nojqpn: filter out JOIN/QUIT/PART/NICK
+grep -v '^[0-9:]+ (JOIN|QUIT|PART|NICK)' $*
--- /dev/null
+++ b/trie.c
@@ -1,0 +1,232 @@
+#include <u.h>
+#include <libc.h>
+#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;
+}