mod.c

   1 #include "mod.h"
   2
   3 #include <libxml/xmlstring.h>
   4 #include <libxml/HTMLparser.h>
   5
   6 #include <stdint.h>
   7 #include <inttypes.h>
   8 #include <sys/wait.h>
   9 #include <curl/curl.h>
  10
  11 #include <magic.h>
  12
  13 #include "entities.h"
  14 #include "quotes.h"
  15
  16 static const char *nick_self;
  17 static unsigned ponify;
  18 #define elements(x) (sizeof(x)/sizeof(*x))
  19
  20 static struct tdb_context *feed_db, *chan_db, *mail_db, *seen_db;
  21
  22 static struct command_hash command_channel = { "#", NULL };
  23
  24 #define sstrncmp(a, b) (strncmp(a, b, sizeof(b)-1))
  25 #define sstrncasecmp(a, b) (strncasecmp(a, b, sizeof(b)-1))
  26
  27 #define SPAM_CUTOFF 5
  28
  29 static int pipe_command(struct bio *b, const char *target, const char *nick, int redirect_stdout, int redirect_stderr, char *argv[])
  30 {
  31         int fd[2];
  32         int lines = 0;
  33         if (pipe2(fd, O_CLOEXEC) < 0) {
  34                 privmsg(b, target, "Could not create pipe: %m");
  35                 return -1;
  36         }
  37         pid_t pid = fork();
  38         if (pid < 0) {
  39                 privmsg(b, target, "Could not fork: %m");
  40                 close(fd[0]);
  41                 close(fd[1]);
  42                 return -1;
  43         } else if (!pid) {
  44                 int fdnull = open("/dev/null", O_WRONLY|O_CLOEXEC);
  45                 close(fd[0]);
  46                 if (dup3(redirect_stdout ? fd[1] : fdnull, 1, 0) < 0)
  47                         exit(errno);
  48                 if (dup3(redirect_stderr ? fd[1] : fdnull, 2, 0) < 0)
  49                         exit(errno);
  50                 exit(execv(argv[0], argv));
  51         } else {
  52                 int loc = -1;
  53                 char buffer[0x100];
  54                 int bufptr = 0, ret;
  55                 close(fd[1]);
  56                 fcntl(fd[0], F_SETFL, O_NONBLOCK);
  57                 while ((ret = waitpid(pid, &loc, WNOHANG)) >= 0) {
  58                         while (read(fd[0], buffer+bufptr, 1) == 1) {
  59                                 if (buffer[bufptr] != '\n' && bufptr < sizeof(buffer)-1) {
  60                                         bufptr++;
  61                                         continue;
  62                                 } else if (bufptr) {
  63                                         buffer[bufptr] = 0;
  64                                         bufptr = 0;
  65                                         lines++;
  66                                         if (lines < SPAM_CUTOFF)
  67                                                 privmsg(b, nick, "%s", buffer);
  68                                         else
  69                                                 fprintf(stderr, "%s\n", buffer);
  70                                 }
  71                         }
  72
  73                         if (ret) {
  74                                 if (bufptr) {
  75                                         buffer[bufptr] = 0;
  76                                         if (lines < SPAM_CUTOFF)
  77                                                 privmsg(b, nick, "%s", buffer);
  78                                         else
  79                                                 fprintf(stderr, "%s\n", buffer);
  80                                 }
  81                                 if (lines >= SPAM_CUTOFF)
  82                                         privmsg(b, nick, "%i lines suppressed", lines - SPAM_CUTOFF + 1);
  83                                 break;
  84                         }
  85                 }
  86                 if (ret < 0)
  87                         privmsg(b, target, "error on waitpid: %m");
  88                 else
  89                         ret = loc;
  90                 close(fd[0]);
  91                 return ret;
  92         }
  93 }
  94
  95 #include "local.c"
  96
  97 static struct command_hash commands[2048];
  98
  99 static void command_abort(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 100 {
 101         abort();
 102         return;
 103 }
 104
 105 static void command_crash(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 106 {
 107         *(char*)0 = 0;
 108 }
 109
 110 static void command_coinflip(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 111 {
 112         if (getrand() & 1)
 113                 action(b, target, "whirrrs and clicks excitedly at %s", nick);
 114         else
 115                 action(b, target, "eyes %s as nothing happens", nick);
 116 }
 117
 118 static void command_shesaid(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 119 {
 120         privmsg(b, target, "%s: \"%s\"", args ? args : nick, women_quotes[getrand() % elements(women_quotes)]);
 121 }
 122
 123 static void squash(char *title)
 124 {
 125         char *start = title;
 126         goto start;
 127         while (*title) {
 128                 while (*title != '\n' && *title != '\r' && *title != '\t' && *title != ' ' && *title) {
 129                         *(start++) = *(title++);
 130                 }
 131                 if (*title)
 132                         *(start++) = ' ';
 133 start:
 134                 while (*title == '\n' || *title == '\r' || *title == '\t' || *title == ' ')
 135                         title++;
 136         }
 137         *start = 0;
 138 }
 139
 140 struct curl_download_context
 141 {
 142         char *data;
 143         size_t len;
 144 };
 145
 146 static size_t write_data(void *ptr, size_t size, size_t nmemb, void *member) {
 147         struct curl_download_context *ctx = member;
 148         size *= nmemb;
 149         ctx->data = realloc(ctx->data, ctx->len + size);
 150         memcpy(ctx->data + ctx->len, ptr, size);
 151         ctx->len += size;
 152         return size;
 153 }
 154
 155 static const char *get_text(xmlNode *cur)
 156 {
 157         for (; cur; cur = cur->next) {
 158                 if (cur->type == XML_TEXT_NODE)
 159                         return cur->content;
 160         }
 161         return NULL;
 162 }
 163
 164 static const char *get_link(xmlAttr *cur, const char *which)
 165 {
 166         for (; cur; cur = cur->next) {
 167                 if (!strcasecmp(cur->name, which))
 168                         return get_text(cur->children);
 169         }
 170         return NULL;
 171 }
 172
 173 static int get_feed_entry(struct bio *b, const char *nick, const char *target, const char *category, xmlNode *entry, const char **title, const char **link)
 174 {
 175         const char *cur_title = NULL, *cur_link = NULL;
 176         xmlNode *cur;
 177         int cur_match = !category;
 178         for (cur = entry->children; cur; cur = cur->next) {
 179                 const char *name = cur->name, *cur_cat;
 180                 if (cur->type != XML_ELEMENT_NODE)
 181                         continue;
 182                 else if (!strcasecmp(name, "link")) {
 183                         const char *ishtml = get_link(cur->properties, "type");
 184                         const char *rel = get_link(cur->properties, "rel");
 185                         if ((!ishtml || !strcasecmp(ishtml, "text/html")) &&
 186                             rel && !strcasecmp(rel, "alternate"))
 187                                 cur_link = get_link(cur->properties, "href");
 188                 } else if (!strcasecmp(name, "title"))
 189                         cur_title = get_text(cur->children);
 190                 else if (!cur_match && !strcasecmp(name, "category") &&
 191                          (cur_cat = get_link(cur->properties, "term")) &&
 192                          strcasestr(cur_cat, category))
 193                         cur_match = 1;
 194         }
 195
 196         if (cur_title)
 197                 *title = cur_title;
 198         else
 199                 *title = "<no title>";
 200         *link = cur_link;
 201         return !cur_link  ? -1 : cur_match;
 202 }
 203
 204 static const char *walk_feed(struct bio *b, const char *nick, const char *target, const char *url, const char *category, xmlNode *root, const char *last_link)
 205 {
 206         const char *main_title = NULL, *main_subtitle = NULL, *main_link = NULL, *title, *link, *prev_link = NULL, *prev_title = NULL;
 207         char *t;
 208         int match, updates = 0;
 209         xmlNode *cur, *entry = NULL;
 210         for (cur = root->children; cur; cur = cur->next) {
 211                 const char *name = cur->name;
 212                 if (cur->type != XML_ELEMENT_NODE)
 213                         continue;
 214                 else if (!strcasecmp(name, "category"))
 215                         continue;
 216                 else if (!strcasecmp(name, "entry"))
 217                         entry = entry ? entry : cur;
 218                 else if (!strcasecmp(name, "title"))
 219                         main_title = get_text(cur->children);
 220                 else if (!strcasecmp(name, "subtitle"))
 221                         main_subtitle = get_text(cur->children);
 222                 else if (!strcasecmp(name, "link")) {
 223                         const char *ishtml = get_link(cur->properties, "type");
 224                         const char *rel = get_link(cur->properties, "rel");
 225                         if ((!ishtml || !strcasecmp(ishtml, "text/html")) && rel && !strcasecmp(rel, "alternate"))
 226                                 main_link = get_link(cur->properties, "href");
 227                 }
 228         }
 229         if (!main_link || !main_title) {
 230                 privmsg(b, target, "%s: Failed to parse main: %s %s", nick, main_link, main_title);
 231                 return NULL;
 232         }
 233         if (!entry)
 234                 return NULL;
 235
 236         if (!last_link)
 237                 privmsg(b, target, "adding blog %s \"%s\": %s", main_link, main_title, main_subtitle);
 238
 239         match = get_feed_entry(b, nick, target, category, entry, &title, &link);
 240         if (match < 0)
 241                 return NULL;
 242
 243         for (; !match && entry; entry = entry->next) {
 244                 if (!strcasecmp(entry->name, "entry"))
 245                         match = get_feed_entry(b, nick, target, category, entry, &title, &link);
 246         }
 247
 248         if (match < 0)
 249                 return NULL;
 250         if (!last_link) {
 251                 if (match > 0)
 252                         privmsg(b, target, "Most recent entry: %s %s", link, title);
 253                 else
 254                         privmsg(b, target, "Currently having no entries for this feed that matches the filter");
 255                 return link;
 256         }
 257         if (!strcmp(last_link, link) || !match || !entry)
 258                 return link;
 259
 260         for (entry = entry->next; entry; entry = entry->next) {
 261                 const char *cur_link, *cur_title;
 262                 if (strcasecmp(entry->name, "entry"))
 263                         continue;
 264                 match = get_feed_entry(b, nick, target, category, entry, &cur_title, &cur_link);
 265                 if (match < 0 || !strcmp(last_link, cur_link))
 266                         break;
 267                 if (match) {
 268                         prev_link = cur_link;
 269                         prev_title = cur_title;
 270                         updates++;
 271                 }
 272         }
 273         if (updates == 1) {
 274                 t = strdup(prev_title);
 275                 squash(t);
 276                 privmsg(b, target, "( %s ): %s", prev_link, t);
 277                 free(t);
 278         } else if (updates > 1)
 279                 privmsg(b, target, "( %s ): %u updates, linking most recent", main_link, 1+updates);
 280         t = strdup(title);
 281         squash(t);
 282         privmsg(b, target, "( %s ): %s", link, t);
 283         free(t);
 284         return link;
 285 }
 286
 287 static const char *walk_rss(struct bio *b, const char *nick, const char *target, const char *url, xmlNode *root, const char *last_link)
 288 {
 289         const char *main_title = NULL, *main_link = NULL, *title = NULL, *link = NULL, *ver;
 290         xmlNode *cur, *entry = NULL;
 291         ver = get_link(root->properties, "version");
 292         if (!ver || strcmp(ver, "2.0")) {
 293                 if (!ver)
 294                         privmsg(b, target, "%s: Could not parse rss feed", nick);
 295                 else
 296                         privmsg(b, target, "%s: Invalid rss version \"%s\"", nick, ver);
 297                 return NULL;
 298         }
 299         for (cur = root->children; cur && cur->type != XML_ELEMENT_NODE; cur = cur->next);
 300         if (!cur)
 301                 return NULL;
 302         for (cur = cur->children; cur; cur = cur->next) {
 303                 const char *name = cur->name;
 304                 if (cur->type != XML_ELEMENT_NODE)
 305                         continue;
 306                 if (!strcasecmp(name, "title"))
 307                         main_title = get_text(cur->children);
 308                 else if (!strcasecmp(name, "link"))
 309                         main_link = main_link ? main_link : get_text(cur->children);
 310                 else if (!strcasecmp(name, "item"))
 311                         entry = entry ? entry : cur;
 312         }
 313         if (!main_link || !main_title) {
 314                 privmsg(b, target, "%s: Failed to parse main: %s %s", nick, main_link, main_title);
 315                 return NULL;
 316         }
 317         if (!entry)
 318                 return NULL;
 319
 320         link = title = NULL;
 321         for (cur = entry->children; cur; cur = cur->next) {
 322                 const char *name = cur->name;
 323                 if (cur->type != XML_ELEMENT_NODE)
 324                         continue;
 325                 if (!strcasecmp(name, "title"))
 326                         title = get_text(cur->children);
 327                 else if (!strcasecmp(name, "link"))
 328                         link = get_text(cur->children);
 329         }
 330         if (!title)
 331                 title = "<no title>";
 332         if (!link) {
 333                 privmsg(b, target, "%s: Failed to parse entry: %s %s", nick, link, title);
 334                 return NULL;
 335         }
 336         if (!last_link) {
 337                 privmsg(b, target, "adding blog %s \"%s\"", main_link, main_title);
 338                 privmsg(b, target, "Most recent entry: %s %s", link, title);
 339         } else if (strcmp(last_link, link)) {
 340                 int updates = 0;
 341                 const char *prev_title = NULL, *prev_link = NULL, *cur_title = NULL, *cur_link = NULL;
 342                 char *t;
 343                 for (entry = entry->next; entry; entry = entry->next) {
 344                         if (strcasecmp(entry->name, "item"))
 345                                 continue;
 346                         prev_title = cur_title;
 347                         prev_link = cur_link;
 348                         cur_title = cur_link = NULL;
 349                         for (cur = entry->children; cur; cur = cur->next) {
 350                                 const char *name = cur->name;
 351                                 if (cur->type != XML_ELEMENT_NODE)
 352                                         continue;
 353                                 if (!strcasecmp(name, "title"))
 354                                         cur_title = get_text(cur->children);
 355                                 else if (!strcasecmp(name, "link"))
 356                                         cur_link = get_text(cur->children);
 357                         }
 358                         if (!cur_title)
 359                                 cur_title = "<no title>";
 360                         if (!cur_link || !strcmp(last_link, cur_link))
 361                                 break;
 362                         updates++;
 363                 }
 364                 if (updates == 1) {
 365                         t = strdup(prev_title);
 366                         squash(t);
 367                         privmsg(b, target, "( %s ): %s", prev_link, t);
 368                         free(t);
 369                 } else if (updates > 1)
 370                         privmsg(b, target, "( %s ): %u updates, linking most recent", main_link, 1+updates);
 371                 t = strdup(title);
 372                 squash(t);
 373                 privmsg(b, target, "( %s ): %s", link, t);
 374                 free(t);
 375         }
 376         return link;
 377 }
 378
 379 // HTML is a mess, so I'm just walking the tree depth first until I find the next element..
 380 static xmlNode *next_link(xmlNode *cur_node)
 381 {
 382         if (cur_node->children)
 383                 return cur_node->children;
 384         while (cur_node) {
 385                 if (cur_node->next)
 386                         return cur_node->next;
 387                 cur_node = cur_node->parent;
 388         }
 389         return NULL;
 390 }
 391
 392 static const char *get_atom_link(xmlNode *cur)
 393 {
 394         for (; cur; cur = next_link(cur)) {
 395                 if (cur->type != XML_ELEMENT_NODE)
 396                         continue;
 397                 if (!strcasecmp(cur->name, "link")) {
 398                         const char *isxml = get_link(cur->properties, "type");
 399                         if (isxml && !strcasecmp(isxml, "application/atom+xml"))
 400                                 return get_link(cur->properties, "href");
 401                 }
 402         }
 403         return NULL;
 404 }
 405
 406 static const char *get_rss_link(xmlNode *cur)
 407 {
 408         for (; cur; cur = next_link(cur)) {
 409                 if (cur->type != XML_ELEMENT_NODE)
 410                         continue;
 411                 if (!strcasecmp(cur->name, "link")) {
 412                         const char *isxml = get_link(cur->properties, "type");
 413                         if (isxml && !strcasecmp(isxml, "application/rss+xml"))
 414                                 return get_link(cur->properties, "href");
 415                 }
 416         }
 417         return NULL;
 418 }
 419
 420 static void do_html(struct bio *b, const char *nick, const char *target, const char *url, const char *data, unsigned len)
 421 {
 422         htmlDocPtr ctx = htmlReadMemory(data, len, 0, url, HTML_PARSE_RECOVER|HTML_PARSE_NOERROR|HTML_PARSE_NOWARNING);
 423         xmlNode *root = xmlDocGetRootElement(ctx);
 424         const char *link = get_atom_link(root);
 425         if (link)
 426                 privmsg(b, target, "%s: not a valid feed link, try atom: %s", nick, link);
 427         else if ((link = get_rss_link(root)))
 428                 privmsg(b, target, "%s: not a valid feed link, try rss: %s", nick, link);
 429         else
 430                 privmsg(b, target, "%s: not a valid feed link, no suggestion found", nick);
 431         xmlFreeDoc(ctx);
 432 }
 433
 434 static size_t get_time_from_header(void *data, size_t size, size_t size2, void *ptr)
 435 {
 436         char *d, *e;
 437         size *= size2;
 438         if (sstrncmp(data, "Last-Modified: "))
 439                 return size;
 440         data += sizeof("Last-Modified: ")-1;
 441         *(char**)ptr = d = strdup(data);
 442         if ((e = strchr(d, '\r')))
 443                 *e = 0;
 444         return size;
 445 }
 446
 447 static int check_single_feed(struct bio *b, const char *target, TDB_DATA key, const char *last_modified, const char *url, const char *link, const char *nick)
 448 {
 449         struct curl_download_context curl_ctx = {};
 450         struct curl_slist *headers = NULL;
 451         char error[CURL_ERROR_SIZE], *category = strchr(url, '#');
 452         int retval = -1;
 453         headers = curl_slist_append(headers, "User-Agent: Mozilla/5.0 (X11; Linux x86_64; rv:9.0.1) Gecko/20100101 Firefox/9.0.1");
 454         headers = curl_slist_append(headers, "Accept: */*");
 455         if (category)
 456                 *(category++) = 0;
 457
 458         CURL *h = curl_easy_init();
 459         curl_easy_setopt(h, CURLOPT_HTTPHEADER, headers);
 460         curl_easy_setopt(h, CURLOPT_URL, url);
 461         curl_easy_setopt(h, CURLOPT_WRITEFUNCTION, write_data);
 462         curl_easy_setopt(h, CURLOPT_WRITEDATA, &curl_ctx);
 463         curl_easy_setopt(h, CURLOPT_ERRORBUFFER, error);
 464         curl_easy_setopt(h, CURLOPT_FOLLOWLOCATION, 1);
 465         curl_easy_setopt(h, CURLOPT_MAXREDIRS, 3);
 466         curl_easy_setopt(h, CURLOPT_SSL_VERIFYPEER, 0);
 467         curl_easy_setopt(h, CURLOPT_TIMEOUT, 8);
 468         curl_easy_setopt(h, CURLOPT_CONNECTTIMEOUT, 8);
 469         curl_easy_setopt(h, CURLOPT_FILETIME, 1);
 470         curl_easy_setopt(h, CURLOPT_HEADERFUNCTION, get_time_from_header);
 471         curl_easy_setopt(h, CURLOPT_WRITEHEADER, &last_modified);
 472         //curl_easy_setopt(h, CURLOPT_VERBOSE, 1);
 473
 474         if (last_modified) {
 475                 char *tmp;
 476                 asprintf(&tmp, "If-Modified-Since: %s", last_modified);
 477                 headers = curl_slist_append(headers, tmp);
 478                 free(tmp);
 479         }
 480
 481         int success = curl_easy_perform(h);
 482         curl_slist_free_all(headers);
 483         if (success == CURLE_OK) {
 484                 char *mime = NULL;
 485                 long code;
 486                 curl_easy_getinfo(h, CURLINFO_CONTENT_TYPE, &mime);
 487                 curl_easy_getinfo(h, CURLINFO_RESPONSE_CODE, &code);
 488                 if (code == 304)
 489                         { }
 490                 else if (!mime || !sstrncmp(mime, "application/xml") || !sstrncmp(mime, "text/xml")) {
 491                         const char *ret_link = NULL;
 492                         xmlDocPtr ctx = xmlReadMemory(curl_ctx.data, curl_ctx.len, 0, url, XML_PARSE_NOWARNING | XML_PARSE_NOERROR);
 493                         xmlNode *root = xmlDocGetRootElement(ctx);
 494
 495                         if (!root || !root->name)
 496                                 fprintf(stderr, "Failed to parse feed %s %p", url, root);
 497                         else if (!strcasecmp(root->name, "feed"))
 498                                 ret_link = walk_feed(b, nick, target, url, category, root, link);
 499                         else if (!strcasecmp(root->name, "rss"))
 500                                 ret_link = walk_rss(b, nick, target, url, root, link);
 501                         else {
 502                                 privmsg(b, target, "Unknown feed type \"%s\"", root->name);
 503                                 goto free_ctx;
 504                         }
 505                         if (category)
 506                                 category[-1] = '#';
 507
 508                         if (!ret_link)
 509                                 privmsg(b, target, "Could not feed parse correctly");
 510                         else if (ret_link && (!link || strcmp(ret_link, link))) {
 511                                 TDB_DATA val;
 512                                 asprintf((char**)&val.dptr, "%s\001%s", last_modified ? last_modified : "", ret_link);
 513                                 val.dsize = strlen(val.dptr)+1;
 514                                 if (tdb_store(feed_db, key, val, 0) < 0)
 515                                         privmsg(b, target, "updating returns %s", tdb_errorstr(feed_db));
 516                                 free(val.dptr);
 517                                 retval = 1;
 518                         }
 519                         else
 520                                 retval = 0;
 521
 522 free_ctx:
 523                         xmlFreeDoc(ctx);
 524                 } else if (link)
 525                         {}
 526                 else if (!sstrncmp(mime, "text/html") || !sstrncmp(mime, "application/xhtml+xml"))
 527                         do_html(b, nick, target, url, curl_ctx.data, curl_ctx.len);
 528                 else
 529                         privmsg(b, target, "unhandled content type %s", mime);
 530         } else if (!link)
 531                 privmsg(b, target, "Error %s (%u)", error, success);
 532
 533         free(curl_ctx.data);
 534         curl_easy_cleanup(h);
 535         return retval;
 536 }
 537
 538 static void command_follow(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 539 {
 540         char *space, *last_link = NULL;
 541         TDB_DATA key, val;
 542         int ret;
 543         if (!feed_db)
 544                 return;
 545         if (target[0] != '#') {
 546                 privmsg(b, target, "%s: Can only follow on channels", nick);
 547                 return;
 548         }
 549         if (!args || !*args) {
 550                 privmsg(b, target, "%s: Usage: !follow <url>", nick);
 551                 return;
 552         }
 553
 554         if (((space = strchr(args, ' ')) && space < strchr(args, '#')) ||
 555             (sstrncmp(args, "http://") && sstrncmp(args, "https://"))) {
 556                 privmsg(b, target, "%s: Invalid url", nick);
 557                 return;
 558         }
 559
 560         key.dsize = asprintf((char**)&key.dptr, "%s,%s", target, args)+1;
 561         val = tdb_fetch(feed_db, key);
 562         if (val.dptr)
 563                 last_link = strchr(val.dptr, '\001');
 564         ret = check_single_feed(b, target, key, NULL, args, last_link ? last_link+1 : NULL, nick);
 565         free(val.dptr);
 566         if (!ret)
 567                 privmsg(b, target, "%s: Not updated", nick);
 568         free(key.dptr);
 569 }
 570
 571 static void channel_feed_check(struct bio *b, const char *target, int64_t now)
 572 {
 573         int len = strlen(target);
 574         int64_t save[] = { now, 0, 0 };
 575
 576         TDB_DATA chan, res;
 577         if (!feed_db || !chan_db)
 578                 return;
 579         chan.dptr = (char*)target;
 580         chan.dsize = len+1;
 581         res = tdb_fetch(chan_db, chan);
 582         if (res.dptr && res.dsize >= 8) {
 583                 uint64_t then = *(uint64_t*)res.dptr;
 584                 if (now - then <= 2000)
 585                         return;
 586                 if (res.dsize >= 16)
 587                         save[1] = ((uint64_t*)res.dptr)[1];
 588                 if (res.dsize >= 24)
 589                         save[2] = ((uint64_t*)res.dptr)[2];
 590         }
 591         free(res.dptr);
 592         res.dptr = (unsigned char*)save;
 593         res.dsize = sizeof(save);
 594         if (tdb_store(chan_db, chan, res, 0) < 0) {
 595                 static int complain_db;
 596                 if (!complain_db++)
 597                         privmsg(b, target, "updating database: %s", tdb_errorstr(feed_db));
 598                 return;
 599         }
 600
 601         for (TDB_DATA d = tdb_firstkey(feed_db); d.dptr;) {
 602                 TDB_DATA f = tdb_fetch(feed_db, d);
 603                 TDB_DATA next = tdb_nextkey(feed_db, d);
 604
 605                 if (!strncmp(d.dptr, target, len) && d.dptr[len] == ',') {
 606                         const char *url = (char*)d.dptr + len + 1;
 607                         char *sep;
 608                         if ((sep = strchr(f.dptr, '\001'))) {
 609                                 *(sep++) = 0;
 610                                 check_single_feed(b, target, d, f.dptr, url, sep, target);
 611                         }
 612                 }
 613                 free(d.dptr);
 614                 free(f.dptr);
 615                 d = next;
 616         }
 617 }
 618
 619 static void command_unfollow(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 620 {
 621         TDB_DATA key;
 622         char *url;
 623
 624         if (!feed_db)
 625                 return;
 626
 627         if (!(url = token(&args, ' ')) || (sstrncmp(url, "http://") && sstrncmp(url, "https://"))) {
 628                 privmsg(b, target, "%s: Invalid url", nick);
 629                 return;
 630         }
 631         if (target[0] != '#') {
 632                 privmsg(b, target, "%s: Can only unfollow on channels", nick);
 633                 return;
 634         }
 635         key.dsize = asprintf((char**)&key.dptr, "%s,%s", target, url)+1;
 636         if (tdb_delete(feed_db, key) < 0) {
 637                 if (tdb_error(feed_db) == TDB_ERR_NOEXIST)
 638                         privmsg(b, target, "%s: Not following %s on this channel", nick, url);
 639                 else
 640                         privmsg(b, target, "%s: Could not delete: %s", nick, tdb_errorstr(feed_db));
 641         } else
 642                 privmsg(b, target, "%s: No longer following %s", nick, url);
 643         free(key.dptr);
 644 }
 645
 646 static void command_g(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 647 {
 648         int ret = 0;
 649         int64_t g = 0, g_total = 0;
 650         TDB_DATA chan, res;
 651
 652         if (!chan_db || target[0] != '#')
 653                 return;
 654
 655         chan.dptr = (char*)target;
 656         chan.dsize = strlen(chan.dptr)+1;
 657         res = tdb_fetch(chan_db, chan);
 658         if (res.dptr && res.dsize >= 16) {
 659                 g = ((int64_t*)res.dptr)[1];
 660                 ((int64_t*)res.dptr)[1] = 0;
 661                 if (res.dsize >= 24)
 662                         g_total = ((int64_t*)res.dptr)[2];
 663                 ret = tdb_store(chan_db, chan, res, 0);
 664         }
 665         free(res.dptr);
 666         if (ret < 0)
 667                 fprintf(stderr, "updating database: %s", tdb_errorstr(feed_db));
 668         else
 669                 privmsg(b, target, "%s: %"PRIi64" g's since last check, %"PRIi64" total", nick, g, g_total);
 670 }
 671
 672 static void command_feeds(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 673 {
 674         int len = strlen(target), found = 0;
 675         if (!feed_db)
 676                 return;
 677         if (target[0] != '#') {
 678                 privmsg(b, target, "%s: Only useful in channels..", nick);
 679                 return;
 680         }
 681         for (TDB_DATA d = tdb_firstkey(feed_db); d.dptr;) {
 682                 TDB_DATA f = tdb_fetch(feed_db, d);
 683                 TDB_DATA next = tdb_nextkey(feed_db, d);
 684
 685                 if (!strncmp(d.dptr, target, len) && d.dptr[len] == ',') {
 686                         privmsg(b, target, "%s: following %s", nick, d.dptr + len + 1);
 687                         found++;
 688                 }
 689                 free(d.dptr);
 690                 free(f.dptr);
 691                 d = next;
 692         }
 693         if (!found)
 694                 privmsg(b, target, "%s: not following any feed on %s", nick, target);
 695 }
 696
 697 static void command_feed_get(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 698 {
 699         if (!feed_db)
 700                 return;
 701         for (TDB_DATA d = tdb_firstkey(feed_db); d.dptr;) {
 702                 TDB_DATA next = tdb_nextkey(feed_db, d);
 703                 if (!args || strcasestr(d.dptr, args)) {
 704                         TDB_DATA f = tdb_fetch(feed_db, d);
 705                         privmsg(b, target, "%s: %s = %s", nick, d.dptr, f.dptr);
 706                         free(f.dptr);
 707                 }
 708                 if (strlen(d.dptr)+1 < d.dsize) {
 709                         privmsg(b, target, "%s: removed buggy entry", nick);
 710                         tdb_delete(feed_db, d);
 711                 }
 712                 free(d.dptr);
 713                 d = next;
 714         }
 715 }
 716
 717 static void command_feed_set(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 718 {
 719         if (!feed_db)
 720                 return;
 721         TDB_DATA key, val;
 722         key.dptr = token(&args, ' ');
 723         char *value = token(&args, ' ');
 724         if (!key.dptr || !value)
 725                 return;
 726         key.dsize = strlen(key.dptr) + 1;
 727         val.dsize = strlen(value) + 2;
 728         val.dptr = malloc(val.dsize);
 729         strcpy(val.dptr+1, value);
 730         val.dptr[0] = '\001';
 731         if (tdb_store(feed_db, key, val, 0) < 0)
 732                 privmsg(b, target, "%s: setting failed: %s", nick, tdb_errorstr(feed_db));
 733         else
 734                 privmsg(b, target, "%s: burp", nick);
 735         free(val.dptr);
 736 }
 737
 738 static void command_feed_rem(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 739 {
 740         if (!feed_db || !args)
 741                 return;
 742         TDB_DATA key = { .dptr = (unsigned char*)args, .dsize = strlen(args)+1 };
 743         if (tdb_delete(feed_db, key) < 0)
 744                 privmsg(b, target, "%s: removing failed: %s", nick, tdb_errorstr(feed_db));
 745         else
 746                 privmsg(b, target, "%s: burp", nick);
 747 }
 748
 749 static void command_feed_xxx(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 750 {
 751         if (!feed_db)
 752                 return;
 753
 754         tdb_wipe_all(feed_db);
 755         privmsg(b, target, "%s: all evidence erased", nick);
 756 }
 757
 758 static void command_seen_xxx(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 759 {
 760         if (!seen_db)
 761                 return;
 762
 763         tdb_wipe_all(seen_db);
 764         privmsg(b, target, "%s: all evidence erased", nick);
 765 }
 766
 767 static void command_feed_counter(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 768 {
 769         if (!chan_db)
 770                 return;
 771         tdb_wipe_all(chan_db);
 772         privmsg(b, target, "%s: All update counters reset", nick);
 773 }
 774
 775 static char *get_text_appended(xmlNode *cur)
 776 {
 777         for (; cur; cur = cur->next) {
 778                 if (cur->type != XML_TEXT_NODE)
 779                         continue;
 780                 return strdup(cur->content);
 781         }
 782         return NULL;
 783 }
 784
 785 static char *get_title(struct bio *b, xmlNode *cur_node)
 786 {
 787         for (; cur_node; cur_node = next_link(cur_node)) {
 788                 if (cur_node->type == XML_ELEMENT_NODE && !strcasecmp(cur_node->name, "title")) {
 789                         if (!cur_node->children)
 790                                 return NULL;
 791                         return get_text_appended(cur_node->children);
 792                 }
 793         }
 794         return NULL;
 795 }
 796
 797 static void internal_link(struct bio *b, const char *nick, const char *host, const char *target, char *args, unsigned verbose)
 798 {
 799         CURL *h;
 800         struct curl_slist *headers = NULL;
 801         char error[CURL_ERROR_SIZE];
 802         int success, sent = verbose;
 803         struct curl_download_context curl_ctx = {};
 804
 805         if (!args)
 806                 return;
 807         int64_t stop, start = get_time(b, target);
 808
 809         h = curl_easy_init();
 810         headers = curl_slist_append(headers, "User-Agent: Mozilla/5.0 (X11; Linux x86_64; rv:9.0.1) Gecko/20100101 Firefox/9.0.1");
 811         headers = curl_slist_append(headers, "Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8");
 812         headers = curl_slist_append(headers, "Accept-Language: en-us,en;q=0.7");
 813         headers = curl_slist_append(headers, "Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.7");
 814         headers = curl_slist_append(headers, "DNT: 1");
 815         headers = curl_slist_append(headers, "Connection: keep-alive");
 816         curl_easy_setopt(h, CURLOPT_HTTPHEADER, headers);
 817         curl_easy_setopt(h, CURLOPT_URL, args);
 818         curl_easy_setopt(h, CURLOPT_WRITEFUNCTION, write_data);
 819         curl_easy_setopt(h, CURLOPT_WRITEDATA, &curl_ctx);
 820         curl_easy_setopt(h, CURLOPT_ERRORBUFFER, error);
 821         curl_easy_setopt(h, CURLOPT_FOLLOWLOCATION, 1);
 822         curl_easy_setopt(h, CURLOPT_MAXREDIRS, 3);
 823         curl_easy_setopt(h, CURLOPT_SSL_VERIFYPEER, 0);
 824         curl_easy_setopt(h, CURLOPT_TIMEOUT, 8);
 825         curl_easy_setopt(h, CURLOPT_CONNECTTIMEOUT, 8);
 826         //curl_easy_setopt(h, CURLOPT_VERBOSE, 1);
 827         success = curl_easy_perform(h);
 828         curl_easy_cleanup(h);
 829         curl_slist_free_all(headers);
 830         if (success == CURLE_OK) {
 831                 magic_t m = magic_open(MAGIC_MIME_TYPE);
 832                 magic_load(m, NULL);
 833                 const char *mime = magic_buffer(m, curl_ctx.data, curl_ctx.len);
 834                 if (strstr(mime, "text/html") || strstr(mime, "application/xml") || strstr(mime, "application/xhtml+xml")) {
 835                         htmlDocPtr ctx = htmlReadMemory(curl_ctx.data, curl_ctx.len, 0, args, HTML_PARSE_RECOVER|HTML_PARSE_NOERROR|HTML_PARSE_NOWARNING);
 836                         xmlNode *root_element = xmlDocGetRootElement(ctx);
 837                         char *title = get_title(b, root_element);
 838                         if (title) {
 839                                 char *nuke;
 840                                 squash(title);
 841                                 decode_html_entities_utf8(title, NULL);
 842                                 if ((nuke = strstr(title, " on SoundCloud - Create")))
 843                                         *nuke = 0;
 844                                 if (*title) {
 845                                         privmsg(b, target, "%s linked %s", nick, title);
 846                                         sent = 1;
 847                                 }
 848                                 free(title);
 849                         }
 850                         if (verbose && !title)
 851                                 privmsg(b, target, "%s linked %s page with invalid title", nick, mime);
 852                         xmlFreeDoc(ctx);
 853                 } else if (verbose) {
 854                         magic_setflags(m, MAGIC_COMPRESS);
 855                         const char *desc = magic_buffer(m, curl_ctx.data, curl_ctx.len);
 856                         privmsg(b, target, "%s linked type %s", nick, desc);
 857                 }
 858                 magic_close(m);
 859         }
 860         if (verbose && success != CURLE_OK)
 861                 privmsg(b, target, "Error %s (%u)", error, success);
 862         else if (!sent && (stop = get_time(b, target)) - start >= 15) {
 863                 privmsg(b, target, "Link (%s) by %s timed out, disabling links for 10 seconds", args, nick);
 864                 commands[strhash("get") % elements(commands)].disabled_until = stop + 10;
 865         }
 866         free(curl_ctx.data);
 867 }
 868
 869 static void command_get(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 870 {
 871         if (!args || (sstrncmp(args, "http://") && sstrncmp(args, "https://")))
 872                 return;
 873         internal_link(b, nick, host, target, token(&args, ' '), 1);
 874 }
 875
 876 static void command_hugs(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 877 {
 878         action(b, target, "gives a lunar hug to %s", args ? args : nick);
 879 }
 880
 881 static void command_hug(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 882 {
 883         if ((host && !strcmp(host, "life.on.the.moon.is.great.I.know.all.about.it")) ||
 884             (args && !sstrncasecmp(args, "lu")))
 885                 command_hugs(b, nick, host, target, args);
 886         else
 887                 action(b, target, "gives a robotic hug to %s", args ? args : nick);
 888 }
 889
 890 static void command_snuggles(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 891 {
 892         action(b, target, "gives a lunar snuggle to %s", args ? args : nick);
 893 }
 894
 895 static void command_snuggle(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 896 {
 897         if ((host && !strcmp(host, "life.on.the.moon.is.great.I.know.all.about.it")) ||
 898             (args && !sstrncasecmp(args, "lu")))
 899                 command_snuggles(b, nick, host, target, args);
 900         else
 901                 action(b, target, "gives a robotic snuggle to %s", args ? args : nick);
 902 }
 903
 904 static void command_cuddles(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 905 {
 906         action(b, target, "gives cuddles to %s in a lunaresque way", args ? args : nick);
 907 }
 908
 909 static void command_cuddle(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 910 {
 911         if ((host && !strcmp(host, "life.on.the.moon.is.great.I.know.all.about.it")) ||
 912             (args && !sstrncasecmp(args, "lu")))
 913                 command_cuddles(b, nick, host, target, args);
 914         else
 915                 action(b, target, "gives cuddles to %s in a robotic way", args ? args : nick);
 916 }
 917
 918
 919 static void command_cookie(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 920 {
 921         if (args && !strncasecmp(args, nick_self, strlen(nick_self)))
 922                 action(b, target, "eats the cookie offered by %s", nick);
 923         else
 924                 action(b, target, "hands a metallic looking cookie to %s", args ? args : nick);
 925 }
 926
 927 static void command_derpy(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 928 {
 929         static const char *insults[] = {
 930                 "accidentally shocks herself",
 931                 "tumbles down the stairs like a slinky",
 932                 "whirrrs and clicks in a screeching way",
 933                 "had problems executing this command",
 934                 "breaks down entirely",
 935                 "uses her magic to levitate herself off the ground, then hits it face first"
 936         };
 937         action(b, target, "%s", insults[getrand() % elements(insults)]);
 938 }
 939
 940 static void command_inspect(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 941 {
 942         struct command_hash *c;
 943         unsigned leave, crash;
 944         char *cmd;
 945         if (!args || !(cmd = token(&args, ' ')))
 946                 return;
 947
 948         if (strcmp(cmd, "#")) {
 949                 c = &commands[strhash(cmd) % elements(commands)];
 950                 if (!c->string || strcasecmp(c->string, cmd)) {
 951                         privmsg(b, target, "Command %s not valid", cmd);
 952                         return;
 953                 }
 954         } else
 955                 c = &command_channel;
 956
 957         leave = c->left + (c->cmd == command_inspect);
 958         crash = c->enter - leave;
 959         if (c->enter != leave)
 960                 privmsg(b, target, "%s: %u successes and %u crash%s, last crashing command: %s", c->string, leave, crash, crash == 1 ? "" : "es", c->failed_command);
 961         else
 962                 privmsg(b, target, "%s: %u time%s executed succesfully", c->string, leave, leave == 1 ? "" : "s");
 963 }
 964
 965 static void command_rebuild(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 966 {
 967         int ret;
 968         char *make[] = { "/usr/bin/make", "-j4", NULL };
 969         char *git_reset[] = { "/usr/bin/git", "reset", "--hard", "master", NULL };
 970         ret = pipe_command(b, target, nick, 0, 1, git_reset);
 971         if (ret) {
 972                 action(b, target, "could not rebuild");
 973                 return;
 974         }
 975         ret = pipe_command(b, target, nick, 0, 1, make);
 976         if (!ret)
 977                 kill(getpid(), SIGUSR1);
 978         else if (ret > 0)
 979                 action(b, target, "displays an ominous %i", ret);
 980 }
 981
 982 static void command_swear(struct bio *b, const char *nick, const char *host, const char *target, char *args)
 983 {
 984         static const char *insults[] = {
 985                 "featherbrain",
 986                 "ponyfeathers",
 987                 "What in the hey?",
 988                 "What are you, a dictionary?",
 989                 "TAR-DY!",
 990                 "[BUY SOME APPLES]",
 991                 "{ Your lack of bloodlust on the battlefield is proof positive that you are a soulless automaton! }",
 992                 "Your royal snootiness"
 993         };
 994         privmsg(b, target, "%s: %s", args ? args : nick, insults[getrand() % elements(insults)]);
 995 }
 996
 997 static const char *perty(int64_t *t)
 998 {
 999         if (*t >= 14 * 24 * 3600) {
1000                 *t /= 7 * 24 * 3600;
1001                 return "weeks";
1002         } else if (*t >= 48 * 3600) {
1003                 *t /= 24 * 3600;
1004                 return "days";
1005         } else if (*t >= 7200) {
1006                 *t /= 3600;
1007                 return "hours";
1008         } else if (*t >= 120) {
1009                 *t /= 60;
1010                 return "minutes";
1011         }
1012         return *t == 1 ? "second" : "seconds";
1013 }
1014
1015 static void command_timeout(struct bio *b, const char *nick, const char *host, const char *target, char *args)
1016 {
1017         struct command_hash *c;
1018         int64_t t = get_time(b, target);
1019         int64_t howlong;
1020         if (t < 0)
1021                 return;
1022         char *arg = token(&args, ' ');
1023         if (!arg || !args || !(howlong = atoi(args))) {
1024                 action(b, target, "pretends to time out");;
1025                 return;
1026         }
1027         c = &commands[strhash(arg) % elements(commands)];
1028         if (c->string && !strcasecmp(c->string, arg)) {
1029                 c->disabled_until = t + howlong;
1030                 const char *str = perty(&howlong);
1031                 action(b, target, "disables %s for %"PRIi64" %s", arg, howlong, str);
1032         } else
1033                 action(b, target, "clicks sadly at %s for not being able to find that command", nick);
1034 }
1035
1036 static void command_mfw(struct bio *b, const char *nick, const char *host, const char *target, char *args)
1037 {
1038         char error[CURL_ERROR_SIZE], *new_url;
1039         CURL *h = curl_easy_init();
1040         struct curl_slist *headers = NULL;
1041         struct curl_download_context curl_ctx = {};
1042         headers = curl_slist_append(headers, "User-Agent: Mozilla/5.0 (X11; Linux x86_64; rv:9.0.1) Gecko/20100101 Firefox/9.0.1");
1043         headers = curl_slist_append(headers, "Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8");
1044         headers = curl_slist_append(headers, "Accept-Language: en-us,en;q=0.7");
1045         headers = curl_slist_append(headers, "Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.7");
1046         headers = curl_slist_append(headers, "DNT: 1");
1047         headers = curl_slist_append(headers, "Connection: keep-alive");
1048         curl_easy_setopt(h, CURLOPT_HTTPHEADER, headers);
1049         curl_easy_setopt(h, CURLOPT_URL, "http://mylittlefacewhen.com/random/");
1050         curl_easy_setopt(h, CURLOPT_WRITEFUNCTION, write_data);
1051         curl_easy_setopt(h, CURLOPT_WRITEDATA, &curl_ctx);
1052         curl_easy_setopt(h, CURLOPT_ERRORBUFFER, error);
1053         curl_easy_setopt(h, CURLOPT_TIMEOUT, 8);
1054         curl_easy_setopt(h, CURLOPT_CONNECTTIMEOUT, 8);
1055         //curl_easy_setopt(h, CURLOPT_VERBOSE, 1);
1056         CURLcode ret = curl_easy_perform(h);
1057         if (ret == CURLE_OK && curl_easy_getinfo(h, CURLINFO_REDIRECT_URL, &new_url) == CURLE_OK)
1058                 privmsg(b, target, "%s: %s", nick, new_url);
1059         curl_slist_free_all(headers);
1060         curl_easy_cleanup(h);
1061         if (ret != CURLE_OK)
1062                 privmsg(b, target, "%s: You have no face", nick);
1063 }
1064
1065 static TDB_DATA get_mail_key(const char *nick)
1066 {
1067         TDB_DATA d;
1068         int i;
1069         d.dsize = strlen(nick)+1;
1070         d.dptr = malloc(d.dsize);
1071         for (i = 0; i < d.dsize - 1; ++i)
1072                 d.dptr[i] = tolower(nick[i]);
1073         d.dptr[i] = 0;
1074         return d;
1075 }
1076
1077 static void command_mail(struct bio *b, const char *nick, const char *host, const char *target, char *args)
1078 {
1079         char *victim, *x = NULL;
1080         size_t len;
1081         int override = 0;
1082         int64_t last_seen = 0;
1083
1084         if (!mail_db || !seen_db)
1085                 return;
1086         TDB_DATA key, val;
1087         victim = token(&args, ' ');
1088         if (victim && !strcasecmp(victim, "-seen")) {
1089                 victim = token(&args, ' ');
1090                 override = 1;
1091         }
1092         if (!victim || !args || victim[0] == '#' || strchr(victim, '@') || strchr(victim, '.')) {
1093                 privmsg(b, target, "%s: Usage: !mail <nick> <message>", nick);
1094                 return;
1095         }
1096         if (!strcasecmp(victim, nick_self)) {
1097                 action(b, target, "whirrs and clicks excitedly at the mail she received from %s", nick);
1098                 return;
1099         }
1100         if (!strcasecmp(victim, nick)) {
1101                 action(b, target, "echos the words from %s back to them: %s", nick, args);
1102                 return;
1103         }
1104         int64_t now = get_time(b, target);
1105         if (now < 0)
1106                 return;
1107
1108         key = get_mail_key(victim);
1109         val = tdb_fetch(seen_db, key);
1110         if (val.dptr && (x = strchr(val.dptr, ','))) {
1111                 *(x++) = 0;
1112                 last_seen = atoll(val.dptr);
1113         }
1114         if (x && !admin(host) && (now - last_seen) < 300 && (!sstrncasecmp(x, "in ") || !sstrncasecmp(x, "joining "))) {
1115                 action(b, target, "would rather not store mail for someone active so recently");
1116                         goto out;
1117         } else if (last_seen && now - last_seen > 14 * 24 * 3600 && !override) {
1118                 int64_t delta = now - last_seen;
1119                 const char *str = perty(&delta);
1120                 privmsg(b, target, "%s: \"%s\" was last seen %"PRIi64" %s ago, use !mail -seen %s <message> to override this check.", nick, victim, delta, str, victim);
1121                 return;
1122         } else if (!x && !override) {
1123                 privmsg(b, target, "%s: I've never seen \"%s\", use !mail -seen %s <message> to override this check.", nick, victim, victim);
1124                 return;
1125         }
1126
1127         val = tdb_fetch(mail_db, key);
1128         if (!val.dptr)
1129                 val.dsize = 0;
1130         else {
1131                 unsigned char *cur;
1132                 int letters = 0;
1133                 for (cur = val.dptr; cur < val.dptr + val.dsize; cur += strlen(cur)+1)
1134                         letters++;
1135                 if (letters >= 4) {
1136                         action(b, target, "looks sadly at %s as she cannot hold any more mail to %s", nick, victim);
1137                         goto out;
1138                 }
1139         }
1140         len = snprintf(NULL, 0, "%"PRIi64 ",%s: %s", now, nick, args) + 1;
1141         val.dptr = realloc(val.dptr, val.dsize + len);
1142         snprintf(val.dptr + val.dsize, len, "%"PRIi64 ",%s: %s", now, nick, args);
1143         val.dsize += len;
1144         if (tdb_store(mail_db, key, val, 0) < 0)
1145                 privmsg(b, target, "%s: updating mail returns %s", nick, tdb_errorstr(mail_db));
1146         else
1147                 action(b, target, "whirrs and clicks at %s as she stores the mail for %s", nick, victim);
1148 out:
1149         free(val.dptr);
1150         free(key.dptr);
1151 }
1152
1153 static void single_message(struct bio *b, const char *target, char *cur, int64_t now)
1154 {
1155         char *endtime, *sep = strchr(cur, ':');
1156         int64_t delta = -1;
1157         if (sep && (endtime = strchr(cur, ',')) && endtime < sep) {
1158                 int64_t t = atoll(cur);
1159                 if (t > 0)
1160                         delta = now - t;
1161                 cur = endtime + 1;
1162         }
1163         if (delta >= 0) {
1164                 const char *str = perty(&delta);
1165                 privmsg(b, target, "%"PRIi64" %s ago from %s", delta, str, cur);
1166         } else
1167                 privmsg(b, target, "From %s", cur);
1168 }
1169
1170 static void command_deliver(struct bio *b, const char *nick, const char *target)
1171 {
1172         TDB_DATA key, val;
1173         unsigned char *cur;
1174         static unsigned mail_enter, mail_leave;
1175         if (mail_enter++ != mail_leave)
1176                 return;
1177
1178         if (!mail_db)
1179                 return;
1180         key = get_mail_key(nick);
1181         val = tdb_fetch(mail_db, key);
1182         if (!val.dptr)
1183                 goto end;
1184         int64_t now = get_time(b, NULL);
1185         if (strcasecmp(key.dptr, nick_self)) {
1186                 privmsg(b, target, "%s: You've got mail!", nick);
1187                 for (cur = val.dptr; cur < val.dptr + val.dsize; cur += strlen(cur)+1)
1188                         single_message(b, target, cur, now);
1189         }
1190         free(val.dptr);
1191         tdb_delete(mail_db, key);
1192 end:
1193         free(key.dptr);
1194         mail_leave++;
1195 }
1196
1197 static void update_seen(struct bio *b, char *doingwhat, const char *nick)
1198 {
1199         TDB_DATA key;
1200         key = get_mail_key(nick);
1201         TDB_DATA val = { .dptr = doingwhat, .dsize = strlen(doingwhat)+1 };
1202         if (seen_db)
1203                 tdb_store(seen_db, key, val, 0);
1204         free(key.dptr);
1205 }
1206
1207 static void command_seen(struct bio *b, const char *nick, const char *host, const char *target, char *args)
1208 {
1209         char *arg = token(&args, ' '), *x;
1210         int64_t now = get_time(b, target);
1211         TDB_DATA key, val;
1212         if (now < 0)
1213                 return;
1214         if (!seen_db || !arg) {
1215                 privmsg(b, target, "%s: { Error... }", nick);
1216                 return;
1217         }
1218         if (!strcasecmp(arg, nick_self)) {
1219                 action(b, target, "whirrs and clicks at %s", nick);
1220                 return;
1221         }
1222         if (!strcasecmp(arg, nick)) {
1223                 action(b, target, "circles around %s dancing", nick);
1224                 return;
1225         }
1226         key = get_mail_key(arg);
1227         val = tdb_fetch(seen_db, key);
1228         if (val.dptr && (x = strchr(val.dptr, ','))) {
1229                 int64_t delta;
1230                 const char *str;
1231                 *(x++) = 0;
1232                 delta = now - atoll(val.dptr);
1233                 str = perty(&delta);
1234                 if (delta < 0)
1235                         privmsg(b, target, "%s was last seen in the future %s", arg, x);
1236                 else
1237                         privmsg(b, target, "%s was last seen %"PRIi64" %s ago %s", arg, delta, str, x);
1238         } else
1239                 action(b, target, "cannot find any evidence that %s exists", arg);
1240         free(val.dptr);
1241 }
1242
1243 static void command_mailbag(struct bio *b, const char *nick, const char *host, const char *target, char *args)
1244 {
1245         char buffer[256];
1246         unsigned rem = sizeof(buffer)-1, first = 2;
1247         if (!mail_db)
1248                 return;
1249         buffer[rem] = 0;
1250
1251         for (TDB_DATA f = tdb_firstkey(mail_db); f.dptr;) {
1252                 if (f.dsize + 4 > rem) {
1253                         privmsg(b, target, "%s: Holding mail for: %s", nick, &buffer[rem]);
1254                         first = 2;
1255                         rem = sizeof(buffer)-1;
1256                         assert(f.dsize + 4 < rem);
1257                 }
1258                 if (f.dptr) {
1259                         if (first == 2) {
1260                                 first = 1;
1261                         } else if (first) {
1262                                 rem -= 5;
1263                                 memcpy(&buffer[rem], " and ", 5);
1264                                 first = 0;
1265                         } else {
1266                                 rem -= 2;
1267                                 memcpy(&buffer[rem], ", ", 2);
1268                         }
1269                         rem -= f.dsize - 1;
1270                         memcpy(&buffer[rem], f.dptr, f.dsize - 1);
1271                 }
1272                 TDB_DATA next = tdb_nextkey(mail_db, f);
1273                 free(f.dptr);
1274                 f = next;
1275         }
1276         if (first < 2)
1277                 privmsg(b, target, "%s: Holding mail for: %s", nick, &buffer[rem]);
1278 }
1279
1280 static void command_mailread(struct bio *b, const char *nick, const char *host, const char *target, char *args)
1281 {
1282         TDB_DATA key, val;
1283         char *victim;
1284         if (!mail_db || !(victim = token(&args, ' ')))
1285                 return;
1286         key = get_mail_key(victim);
1287         val = tdb_fetch(mail_db, key);
1288         if (!val.dptr)
1289                 action(b, target, "ponyshrugs as no mail for %s was found", victim);
1290         else {
1291                 unsigned char *cur;
1292                 int64_t now = get_time(b, NULL);
1293                 action(b, target, "peeks through %s's mail", victim);
1294                 for (cur = val.dptr; cur < val.dptr + val.dsize; cur += strlen(cur)+1)
1295                         single_message(b, target, cur, now);
1296         }
1297         free(val.dptr);
1298         free(key.dptr);
1299 }
1300
1301 static void command_no_deliver(struct bio *b, const char *nick, const char *host, const char *target, char *args)
1302 {
1303         TDB_DATA key, val;
1304         char *cur;
1305         if (!mail_db || !(cur = token(&args, ' ')))
1306                 return;
1307         key = get_mail_key(cur);
1308         val = tdb_fetch(mail_db, key);
1309         if (!val.dptr)
1310                 action(b, target, "ponyshrugs as no mail for %s was found", cur);
1311         else {
1312                 action(b, target, "deletes all evidence of %s's mail", cur);
1313                 tdb_delete(mail_db, key);
1314         }
1315         free(val.dptr);
1316         free(key.dptr);
1317 }
1318
1319 static void perform_roll(struct bio *b, const char *nick, const char *target, long sides, long dice, long bonus)
1320 {
1321         long rem = dice, total = bonus;
1322         while (rem--)
1323                 total += 1 + (getrand() % sides);
1324         action(b, target, "rolls %li %li-sided %s for a total of %li (%+li)", dice, sides, dice == 1 ? "die" : "dice", total, bonus);
1325 }
1326
1327 static void command_roll(struct bio *b, const char *nick, const char *host, const char *target, char *args)
1328 {
1329         char *cur;
1330         long dice = 1, sides = 20;
1331         if ((cur = token(&args, ' '))) {
1332                 char *first = cur;
1333                 dice = strtol(first, &cur, 10);
1334                 if (first == cur)
1335                         dice = 1;
1336                 if (*cur && *cur != 'd' && *cur != 'D')
1337                         goto syntax;
1338
1339                 if (*cur) {
1340                         char *last = cur+1;
1341                         sides = strtol(last, &cur, 10);
1342                         if (last == cur)
1343                                 goto syntax;
1344                 }
1345         }
1346         if (dice <= 0 || dice > 10 || sides < 2 || sides > 20)
1347                 goto syntax;
1348         perform_roll(b, nick, target, sides, dice, 0);
1349         return;
1350
1351 syntax:
1352         action(b, target, "bleeps at %s in a confused manner", nick);
1353 }
1354
1355 static void add_g(struct bio *b, const char *target, int g)
1356 {
1357         int ret = 0;
1358         int64_t total_g = 0;
1359         TDB_DATA chan, res;
1360
1361         if (!chan_db || target[0] != '#')
1362                 return;
1363
1364         chan.dptr = (char*)target;
1365         chan.dsize = strlen(chan.dptr)+1;
1366         res = tdb_fetch(chan_db, chan);
1367         if (res.dptr && res.dsize >= 16) {
1368                 ((int64_t*)res.dptr)[1] += g;
1369                 if (res.dsize >= 24)
1370                         total_g = ((int64_t*)res.dptr)[2] += g;
1371                 ret = tdb_store(chan_db, chan, res, 0);
1372         }
1373         free(res.dptr);
1374         if (ret < 0)
1375                 fprintf(stderr, "updating database: %s", tdb_errorstr(feed_db));
1376         else if (g < total_g) {
1377                 int64_t last_g = total_g - g;
1378                 last_g -= last_g % 1000;
1379                 if (last_g + 1000 <= total_g)
1380                         privmsg(b, target, "g has been said %"PRIi64" times!", total_g);
1381         }
1382 }
1383
1384 static int parse_g(const char *cur, int *g)
1385 {
1386         int this_g = 0;
1387         for (const unsigned char *ptr = cur; *ptr; ptr++) {
1388                 if (*ptr >= 0x80)
1389                         return 0;
1390                 if (*ptr == 'g' || *ptr == 'G')
1391                         this_g++;
1392                 else if ((*ptr >= 'a' && *ptr <= 'z') || (*ptr >= 'A' && *ptr <= 'Z'))
1393                         return 0;
1394                 else if (*ptr != '1' && *ptr >= '0' && *ptr <= '9')
1395                         return 0;
1396         }
1397         *g += this_g;
1398         return this_g;
1399 }
1400
1401 static void channel_msg(struct bio *b, const char *nick, const char *host,
1402                         const char *chan, char *msg, int64_t t)
1403 {
1404         char *cur = NULL, *next;
1405         int is_action = 0;
1406         int g = 0;
1407
1408         int that = 0, what = 0, she = 0;
1409
1410         if (!msg || !strcmp(nick, "`Daring_Do`") || !strcmp(nick, "`Celestia`") || !strcmp(nick, "derpy") || !strcmp(nick, "`Derpy`") || !strcmp(nick, "`Luna`") || !sstrncmp(nick, "GitHub") || !sstrncmp(nick, "CIA-") || !strcmp(nick, "Terminus-Bot") || !strcmp(nick, "r0m"))
1411                 return;
1412
1413         if (!sstrncasecmp(msg, "\001ACTION ")) {
1414                 msg += sizeof("\001ACTION ")-1;
1415                 is_action = 1;
1416                 asprintf(&cur, "%"PRIi64 ",in %s: * %s %s", t, chan, nick, msg);
1417         } else
1418                 asprintf(&cur, "%"PRIi64 ",in %s: <%s> %s", t, chan, nick, msg);
1419         if (t > 0)
1420                 update_seen(b, cur, nick);
1421         free(cur);
1422         (void)is_action;
1423
1424         next = msg;
1425         while ((cur = token(&next, ' '))) {
1426                 if (!strcasecmp(cur, ">mfw") || !strcasecmp(cur, "mfw")) {
1427                         if (!strcasecmp(chan, "#brony") || !ponify)
1428                                 continue;
1429                         if (t < 0 || t > commands[strhash("mfw") % elements(commands)].disabled_until)
1430                                 command_mfw(b, nick, host, chan, NULL);
1431                         return;
1432                 } else if (!sstrncasecmp(cur, "http://") || !sstrncasecmp(cur, "https://")) {
1433                         static char last_url[512];
1434                         char *part;
1435                         if (!strcmp(cur, last_url))
1436                                 return;
1437                         strncpy(last_url, cur, sizeof(last_url)-1);
1438
1439                         if (t >= 0 && t < commands[strhash("get") % elements(commands)].disabled_until)
1440                                 return;
1441
1442                         else if (strcasestr(cur, "youtube.com/user") && (part = strstr(cur, "#p/"))) {
1443                                 char *foo;
1444                                 part = strrchr(part, '/') + 1;
1445                                 asprintf(&foo, "http://youtube.com/watch?v=%s", part);
1446                                 if (foo)
1447                                         internal_link(b, nick, host, chan, foo, 0);
1448                                 free(foo);
1449                                 return;
1450                         } else if (strcasestr(cur, "twitter.com/") || strcasestr(cur, "mlfw.info") || strcasestr(cur, "mylittlefacewhen.com") || !strcasecmp(chan, "#geek"))
1451                                 return;
1452                         internal_link(b, nick, host, chan, cur, 0);
1453                         return;
1454                 }
1455                 else if (!sstrncasecmp(cur, "that") || !sstrncasecmp(cur, "dat"))
1456                         that = 1;
1457                 else if (that && (!sstrncasecmp(cur, "what") || !sstrncasecmp(cur, "wat")))
1458                         what = 1;
1459                 else if (what && !sstrncasecmp(cur, "she"))
1460                         she = 1;
1461                 else if (she && !sstrncasecmp(cur, "said")) {
1462                         if (t <= 0 || t >= commands[strhash("shesaid") % elements(commands)].disabled_until)
1463                                 privmsg(b, chan, "%s: \"%s\"", nick, women_quotes[getrand() % elements(women_quotes)]);
1464                         return;
1465                 } else if (parse_g(cur, &g))
1466                         continue;
1467         }
1468         if (g)
1469                 add_g(b, chan, g);
1470 }
1471
1472 static struct command_hash unhashed[] = {
1473         { "1", command_coinflip },
1474         { "get", command_get },
1475         { "hug", command_hug },
1476         { "hugs", command_hugs },
1477         { "snug", command_snuggle },
1478         { "snugs", command_snuggles },
1479         { "snuggle", command_snuggle },
1480         { "snuggles", command_snuggles },
1481         { "cuddle", command_cuddle },
1482         { "cuddles", command_cuddles },
1483         { "cookie", command_cookie },
1484         { "mfw", command_mfw },
1485         { "swear", command_swear },
1486         { "mail", command_mail },
1487         { "seen", command_seen },
1488         { "derpy", command_derpy },
1489         { "g", command_g },
1490         { "shesaid", command_shesaid },
1491         { "roll", command_roll },
1492
1493         { "rebuild", command_rebuild, 1 },
1494         { "abort", command_abort, 1 },
1495         { "crash", command_crash, 1 },
1496         { "inspect", command_inspect, 1 },
1497         { "timeout", command_timeout, 1 },
1498
1499         { "follow", command_follow },
1500         { "unfollow", command_unfollow },
1501         { "feeds", command_feeds },
1502
1503 // DEBUG
1504         { "feed_get", command_feed_get, 1 },
1505         { "feed_set", command_feed_set, 1 },
1506         { "feed_rem", command_feed_rem, 1 },
1507         { "feed_xxx", command_feed_xxx, 1 },
1508         { "feed_counter", command_feed_counter, 1 },
1509         { "seen_xxx", command_seen_xxx, 1 },
1510         { "mailbag", command_mailbag },
1511         { "mailread", command_mailread, 1 },
1512         { "\"deliver\"", command_no_deliver, 1 },
1513 };
1514
1515 static void init_hash(struct bio *b, const char *target)
1516 {
1517         int i;
1518         for (i = 0; i < elements(unhashed); ++i) {
1519                 unsigned h = strhash(unhashed[i].string) % elements(commands);
1520                 if (commands[h].string)
1521                         privmsg(b, target, "%s is a duplicate command with %s", commands[h].string, unhashed[i].string);
1522                 else
1523                         commands[h] = unhashed[i];
1524         }
1525 #ifdef local_commands
1526         for (i = 0; i < elements(local_commands); ++i) {
1527                 unsigned h = strhash(local_commands[i].string) % elements(commands);
1528                 if (commands[h].string)
1529                         privmsg(b, target, "%s is a duplicate command with %s", commands[h].string, local_commands[i].string);
1530                 else
1531                         commands[h] = local_commands[i];
1532         }
1533 #endif
1534 }
1535
1536 void init_hook(struct bio *b, const char *target, const char *nick, unsigned is_ponified)
1537 {
1538         char *cwd, *path = NULL;
1539         nick_self = nick;
1540         static const char *messages[] = {
1541                 "feels circuits being activated that weren't before",
1542                 "suddenly gets a better feel of her surroundings",
1543                 "looks the same, yet there's definitely something different",
1544                 "emits a beep as her lights begin to pulse slowly",
1545                 "whirrrs and bleeps like never before",
1546                 "bleeps a few times happily",
1547                 "excitedly peeks at her surroundings"
1548         };
1549         init_hash(b, target);
1550         ponify = is_ponified;
1551
1552         cwd = getcwd(NULL, 0);
1553         asprintf(&path, "%s/db/feed.tdb", cwd);
1554         feed_db = tdb_open(path, 0, 0, O_RDWR|O_CREAT|O_CLOEXEC|O_NOCTTY, 0644);
1555         free(path);
1556         if (!feed_db)
1557                 privmsg(b, target, "Opening feed db failed: %m");
1558
1559         asprintf(&path, "%s/db/chan.tdb", cwd);
1560         chan_db = tdb_open(path, 0, 0, O_RDWR|O_CREAT|O_CLOEXEC|O_NOCTTY, 0644);
1561         free(path);
1562         if (!chan_db)
1563                 privmsg(b, target, "Opening chan db failed: %m");
1564
1565         asprintf(&path, "%s/db/mail.tdb", cwd);
1566         mail_db = tdb_open(path, 0, 0, O_RDWR|O_CREAT|O_CLOEXEC|O_NOCTTY, 0644);
1567         free(path);
1568         if (!mail_db)
1569                 privmsg(b, target, "Opening mail db failed: %m");
1570
1571         asprintf(&path, "%s/db/seen.tdb", cwd);
1572         seen_db = tdb_open(path, 0, 0, O_RDWR|O_CREAT|O_CLOEXEC|O_NOCTTY, 0644);
1573         free(path);
1574         if (!seen_db)
1575                 privmsg(b, target, "Opening seen db failed: %m");
1576
1577         free(cwd);
1578         action(b, target, "%s", messages[getrand() % elements(messages)]);
1579 }
1580
1581 static void  __attribute__((destructor)) shutdown_hook(void)
1582 {
1583         if (seen_db)
1584                 tdb_close(seen_db);
1585         if (mail_db)
1586                 tdb_close(mail_db);
1587         if (feed_db)
1588                 tdb_close(feed_db);
1589         if (chan_db)
1590                 tdb_close(chan_db);
1591 }
1592
1593 static char *nom_special(char *line)
1594 {
1595         while (*line) {
1596                 if (*line == 0x03) {
1597                         line++;
1598                         if (*line >= '0' && *line <= '9')
1599                                 line++;
1600                         else continue;
1601                         if (*line >= '0' && *line <= '9')
1602                                 line++;
1603                         if (line[0] == ',' && line[1] >= '0' && line[1] <= '9')
1604                                 line += 2;
1605                         else continue;
1606                         if (*line >= '0' && *line <= '9')
1607                                 line++;
1608                 } else if (*line != 0x02 && /* BOLD */
1609                            *line != 0x1f && /* UNDERLINE */
1610                            *line != 0x16 && /* ITALIC */
1611                            *line != 0x06 && /* NFI, is this even used? */
1612                            *line != 0x07 && /* NFI, is this even used? */
1613                            *line != 0x0f) /* NORMAL */
1614                         return line;
1615                 else
1616                         line++;
1617         }
1618         return line;
1619 }
1620
1621 static char *cleanup_special(char *line)
1622 {
1623         char *cur, *start = nom_special(line);
1624         if (!*start)
1625                 return NULL;
1626         for (line = cur = start; *line; line = nom_special(line))
1627                 *(cur++) = *(line++);
1628
1629         for (cur--; cur >= start; --cur)
1630                 if (*cur != ' ' && *cur != '\001')
1631                         break;
1632
1633         if (cur < start)
1634                 return NULL;
1635         cur[1] = 0;
1636         return start;
1637 }
1638
1639 static void rss_check(struct bio *b, const char *channel, int64_t t)
1640 {
1641         static unsigned rss_enter, rss_leave;
1642         if (t >= 0 && rss_enter == rss_leave) {
1643                 rss_enter++;
1644                 channel_feed_check(b, channel, t);
1645                 rss_leave++;
1646         }
1647 }
1648
1649 static const char *privileged_command[] = {
1650         "{ Rarity, I love you so much! }",
1651         "{ Rarity, have I ever told you that I love you? }",
1652         "{ Yes, I love my sister, Rarity. }",
1653         "{ Raaaaaaaaaaaaaarity. }",
1654         "{ You do not fool me, Rari...bot! }"
1655 };
1656
1657 static int cmd_check(struct bio *b, struct command_hash *c, int is_admin, int64_t t, const char *prefix, const char *target)
1658 {
1659         if (c->left != c->enter)
1660                 privmsg(b, target, "Command %s is disabled because of a crash", c->string);
1661         else if (t > 0 && t < c->disabled_until && !is_admin) {
1662                 int64_t delta = c->disabled_until - t;
1663                 const char *str = perty(&delta);
1664                 b->writeline(b, "NOTICE %s :Command %s is on timeout for the next %"PRIi64 " %s", prefix, c->string, delta, str);
1665         } else if (c->admin && !is_admin)
1666                 privmsg(b, target, "%s: %s", prefix, privileged_command[getrand() % elements(privileged_command)]);
1667         else
1668                 return 1;
1669         return 0;
1670 }
1671
1672 void privmsg_hook(struct bio *b, const char *prefix, const char *ident, const char *host,
1673                   const char *const *args, unsigned nargs)
1674 {
1675         char *cmd_args, *cmd;
1676         const char *target = args[0][0] == '#' ? args[0] : prefix;
1677         unsigned chan, nick_len;
1678         struct command_hash *c;
1679         int64_t t = get_time(b, target);
1680         int is_admin = admin(host);
1681
1682         chan = args[0][0] == '#';
1683         if (chan) {
1684                 rss_check(b, args[0], t);
1685                 command_deliver(b, prefix, args[0]);
1686         }
1687         cmd_args = cleanup_special((char*)args[1]);
1688         if (!cmd_args)
1689                 return;
1690
1691         if (ident && (!strcasecmp(ident, "Revolver") || !strcasecmp(ident, "Rev")))
1692                 return;
1693
1694         if (chan && cmd_args[0] == '!') {
1695                 cmd_args++;
1696         } else if (chan && (nick_len = strlen(nick_self)) &&
1697                 !strncasecmp(cmd_args, nick_self, nick_len) &&
1698                 (cmd_args[nick_len] == ':' || cmd_args[nick_len] == ',') && cmd_args[nick_len+1] == ' ') {
1699                 cmd_args += nick_len + 2;
1700                 if (!cmd_args[0])
1701                         return;
1702                 chan = 2;
1703         } else if (chan) {
1704                 if (command_channel.enter == command_channel.left) {
1705                         command_channel.enter++;
1706                         snprintf(command_channel.failed_command,
1707                                  sizeof(command_channel) - offsetof(struct command_hash, failed_command) - 1,
1708                                  "%s:%s (%s@%s) \"#\" %s", target, prefix, ident, host, cmd_args);
1709                         channel_msg(b, prefix, host, args[0], cmd_args, t);
1710                         command_channel.left++;
1711                 }
1712                 return;
1713         }
1714         cmd = token(&cmd_args, ' ');
1715         if (!chan && cmd_args && cmd_args[0] == '#') {
1716                 if (!is_admin) {
1717                         privmsg(b, target, "%s: %s", prefix, privileged_command[getrand() % elements(privileged_command)]);
1718                         return;
1719                 }
1720                 target = token(&cmd_args, ' ');
1721         }
1722
1723         c = &commands[strhash(cmd) % elements(commands)];
1724         if (c->string && !strcasecmp(c->string, cmd)) {
1725                 if (cmd_check(b, c, is_admin, t, prefix, target)) {
1726                         c->enter++;
1727                         snprintf(c->failed_command, sizeof(*c) - offsetof(struct command_hash, failed_command) - 1,
1728                                  "%s:%s (%s@%s) \"%s\" %s", target, prefix, ident, host, cmd, cmd_args);
1729                         c->cmd(b, prefix, host, target, cmd_args);
1730                         c->left++;
1731                 }
1732                 return;
1733         } else if (cmd[0] == 'd' && cmd[1] >= '0' && cmd[1] <= '9') {
1734                 char *end;
1735                 long base;
1736                 base = strtol(cmd+1, &end, 10);
1737                 if (base > 1 && base <= 20 && !*end) {
1738                         long dice = 1, bonus = 0;
1739                         char *strdice;
1740                         c = &commands[strhash("roll") % elements(commands)];
1741                         if (!cmd_check(b, c, is_admin, t, prefix, target))
1742                                 return;
1743                         c->enter++;
1744                         snprintf(c->failed_command, sizeof(*c) - offsetof(struct command_hash, failed_command) - 1,
1745                                 "%s:%s (%s@%s) \"%s\" %s", target, prefix, ident, host, cmd, cmd_args);
1746
1747                         strdice = token(&cmd_args, ' ');
1748                         if (strdice) {
1749                                 dice = strtol(strdice, &end, 10);
1750                                 if (*end || !dice || dice > 10)
1751                                         goto syntax;
1752                                 strdice = token(&cmd_args, ' ');
1753                                 if (strdice) {
1754                                         bonus = strtol(strdice, &end, 10);
1755                                         if (*end)
1756                                                 goto syntax;
1757                                 }
1758                         }
1759                         perform_roll(b, prefix, target, base, dice, bonus);
1760                         c->left++;
1761                         return;
1762
1763 syntax:
1764                         c->left++;
1765                         action(b, target, "bleeps at %s in a confused manner", prefix);
1766                         return;
1767                 }
1768         }
1769         if (chan == 2 && t > 0) {
1770                 static int64_t last_t;
1771                 if (t - last_t > 3)
1772                         privmsg(b, target, "%s: { I love you! }", prefix);
1773                 last_t = t;
1774         }
1775 }
1776
1777 void command_hook(struct bio *b, const char *prefix, const char *ident, const char *host,
1778                   const char *command, const char *const *args, unsigned nargs)
1779 {
1780         char *buf = NULL;
1781         int64_t t = -1;
1782         if (!strcasecmp(command, "NOTICE")) {
1783                 if (nargs < 2 || args[0][0] == '#')
1784                         return;
1785                 if (admin(host))
1786                         b->writeline(b, "%s", args[1]);
1787                 else
1788                         fprintf(stderr, "%s: %s\n", prefix, args[1]);
1789         } else if (!strcasecmp(command, "JOIN")) {
1790                 t = get_time(b, args[0]);
1791                 rss_check(b, args[0], t);
1792                 command_deliver(b, prefix, args[0]);
1793                 asprintf(&buf, "%"PRIi64",joining %s", t, args[0]);
1794         } else if (!strcasecmp(command, "PART")) {
1795                 t = get_time(b, args[0]);
1796                 rss_check(b, args[0], t);
1797                 asprintf(&buf, "%"PRIi64",leaving %s", t, args[0]);
1798         } else if (!strcasecmp(command, "QUIT")) {
1799                 t = get_time(b, NULL);
1800                 asprintf(&buf, "%"PRIi64",quitting with the message \"%s\"", t, args[0]);
1801         } else if (!strcasecmp(command, "NICK")) {
1802                 t = get_time(b, NULL);
1803                 if (t >= 0) {
1804                         asprintf(&buf, "%"PRIi64",changing nick from %s", t, prefix);
1805                         if (buf)
1806                                 update_seen(b, buf, args[0]);
1807                         free(buf);
1808                         buf = NULL;
1809                 }
1810                 asprintf(&buf, "%"PRIi64",changing nick to %s", t, args[0]);
1811         } else if (0) {
1812                 int i;
1813                 fprintf(stderr, ":%s!%s%s %s", prefix, ident, host, command);
1814                 for (i = 0; i < nargs; ++i)
1815                         fprintf(stderr, " %s", args[i]);
1816                 fprintf(stderr, "\n");
1817         }
1818         if (t >= 0 && buf)
1819                 update_seen(b, buf, prefix);
1820         free(buf);
1821         return;
1822 }
1823