src/proto-bahamut.c

   1 /* proto-bahamut.c - IRC protocol output
   2  * Copyright 2000-2004 srvx Development Team
   3  *
   4  * This file is part of srvx.
   5  *
   6  * srvx is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with srvx; if not, write to the Free Software Foundation,
  18  * Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
  19  */
  20
  21 #include "proto-common.c"
  22
  23 #define CAPAB_TS3       0x01
  24 #define CAPAB_NOQUIT    0x02
  25 #define CAPAB_SSJOIN    0x04
  26 #define CAPAB_BURST     0x08
  27 #define CAPAB_UNCONNECT 0x10
  28 #define CAPAB_NICKIP    0x20
  29 #define CAPAB_TSMODE    0x40
  30 #define CAPAB_ZIP       0x80
  31
  32 struct service_message_info {
  33     privmsg_func_t on_privmsg;
  34     privmsg_func_t on_notice;
  35 };
  36
  37 static dict_t service_msginfo_dict; /* holds service_message_info structs */
  38 static int uplink_capab;
  39 static void privmsg_user_helper(struct userNode *un, void *data);
  40
  41 void irc_svsmode(struct userNode *target, char *modes, unsigned long stamp);
  42
  43 struct server *
  44 AddServer(struct server *uplink, const char *name, int hops, time_t boot, time_t link, UNUSED_ARG(const char *numeric), const char *description) {
  45     struct server* sNode;
  46
  47     sNode = calloc(1, sizeof(*sNode));
  48     sNode->uplink = uplink;
  49     safestrncpy(sNode->name, name, sizeof(sNode->name));
  50     sNode->hops = hops;
  51     sNode->boot = boot;
  52     sNode->link = link;
  53     sNode->users = dict_new();
  54     safestrncpy(sNode->description, description, sizeof(sNode->description));
  55     serverList_init(&sNode->children);
  56     if (sNode->uplink) {
  57         /* uplink may be NULL if we're just building ourself */
  58         serverList_append(&sNode->uplink->children, sNode);
  59     }
  60     dict_insert(servers, sNode->name, sNode);
  61
  62     if (hops && !self->burst) {
  63         unsigned int n;
  64         for (n=0; n<slf_used; n++) {
  65             slf_list[n](sNode);
  66         }
  67     }
  68
  69     return sNode;
  70 }
  71
  72 void
  73 DelServer(struct server* serv, int announce, const char *message) {
  74     unsigned int nn;
  75     dict_iterator_t it, next;
  76
  77     if (!serv) return;
  78     if (announce && (serv->uplink == self) && (serv != self->uplink)) {
  79         irc_squit(serv, message, NULL);
  80     }
  81     for (nn=serv->children.used; nn>0;) {
  82         if (serv->children.list[--nn] != self) {
  83             DelServer(serv->children.list[nn], false, "uplink delinking");
  84         }
  85     }
  86     for (it=dict_first(serv->users); it; it=next) {
  87         next = iter_next(it);
  88         DelUser(iter_data(it), NULL, false, "server delinking");
  89     }
  90     if (serv->uplink) serverList_remove(&serv->uplink->children, serv);
  91     if (serv == self->uplink) self->uplink = NULL;
  92     dict_remove(servers, serv->name);
  93     serverList_clean(&serv->children);
  94     dict_delete(serv->users);
  95     free(serv);
  96 }
  97
  98 int
  99 is_valid_nick(const char *nick) {
 100     /* IRC has some of The Most Fucked-Up ideas about character sets
 101      * in the world.. */
 102     if ((*nick < 'A') || (*nick >= '~')) return 0;
 103     for (++nick; *nick; ++nick) {
 104         if (!((*nick >= 'A') && (*nick < '~'))
 105             && !isdigit(*nick)
 106             && (*nick != '-')) {
 107             return 0;
 108         }
 109     }
 110     if (strlen(nick) > nicklen) return 0;
 111     return 1;
 112 }
 113
 114 struct userNode *
 115 AddUser(struct server* uplink, const char *nick, const char *ident, const char *hostname, const char *modes, const char *userinfo, time_t timestamp, struct in_addr realip, const char *stamp) {
 116     struct userNode *uNode, *oldUser;
 117     unsigned int nn;
 118
 119     if (!uplink) {
 120         log_module(MAIN_LOG, LOG_WARNING, "AddUser(%p, %s, ...): server does not exist!", uplink, nick);
 121         return NULL;
 122     }
 123
 124     if (!is_valid_nick(nick)) {
 125         log_module(MAIN_LOG, LOG_WARNING, "AddUser(%p, %s, ...): invalid nickname detected.", uplink, nick);
 126         return NULL;
 127     }
 128
 129     if ((oldUser = GetUserH(nick))) {
 130         if (IsLocal(oldUser) && IsService(oldUser)) {
 131             /* The service should collide the new user off. */
 132             oldUser->timestamp = timestamp - 1;
 133             irc_user(oldUser);
 134         }
 135         if (oldUser->timestamp > timestamp) {
 136             /* "Old" user is really newer; remove them */
 137             DelUser(oldUser, 0, 1, "Overruled by older nick");
 138         } else {
 139             /* User being added is too new */
 140             return NULL;
 141         }
 142     }
 143
 144     uNode = calloc(1, sizeof(*uNode));
 145     uNode->nick = strdup(nick);
 146     safestrncpy(uNode->ident, ident, sizeof(uNode->ident));
 147     safestrncpy(uNode->info, userinfo, sizeof(uNode->info));
 148     safestrncpy(uNode->hostname, hostname, sizeof(uNode->hostname));
 149     uNode->ip = realip;
 150     uNode->timestamp = timestamp;
 151     modeList_init(&uNode->channels);
 152     uNode->uplink = uplink;
 153     dict_insert(uplink->users, uNode->nick, uNode);
 154     if (++uNode->uplink->clients > uNode->uplink->max_clients) {
 155         uNode->uplink->max_clients = uNode->uplink->clients;
 156     }
 157
 158     dict_insert(clients, uNode->nick, uNode);
 159     if (dict_size(clients) > max_clients) {
 160         max_clients = dict_size(clients);
 161         max_clients_time = now;
 162     }
 163
 164     mod_usermode(uNode, modes);
 165     if (stamp) call_account_func(uNode, stamp);
 166     if (IsLocal(uNode)) irc_user(uNode);
 167     for (nn=0; nn<nuf_used; nn++) {
 168         if (nuf_list[nn](uNode)) break;
 169     }
 170     return uNode;
 171 }
 172
 173 struct userNode *
 174 AddService(const char *nick, const char *modes, const char *desc, const char *hostname) {
 175     time_t timestamp = now;
 176     struct userNode *old_user = GetUserH(nick);
 177     struct in_addr ipaddr = { INADDR_LOOPBACK };
 178     if (old_user) {
 179         if (IsLocal(old_user))
 180             return old_user;
 181         timestamp = old_user->timestamp - 1;
 182     }
 183     if (!hostname)
 184         hostname = self->name;
 185     return AddUser(self, nick, nick, hostname, modes ? modes : "+oikr", desc, timestamp, ipaddr, 0);
 186 }
 187
 188 struct userNode *
 189 AddClone(const char *nick, const char *ident, const char *hostname, const char *desc) {
 190     time_t timestamp = now;
 191     struct userNode *old_user = GetUserH(nick);
 192     struct in_addr ipaddr = { INADDR_LOOPBACK };
 193     if (old_user) {
 194         if (IsLocal(old_user))
 195             return old_user;
 196         timestamp = old_user->timestamp - 1;
 197     }
 198     return AddUser(self, nick, ident, hostname, "+ir", desc, timestamp, ipaddr, 0);
 199 }
 200
 201 void
 202 free_user(struct userNode *user)
 203 {
 204     free(user->nick);
 205     free(user);
 206 }
 207
 208 void
 209 DelUser(struct userNode* user, struct userNode *killer, int announce, const char *why) {
 210     unsigned int nn;
 211
 212     for (nn=user->channels.used; nn>0;) {
 213         DelChannelUser(user, user->channels.list[--nn]->channel, false, 0);
 214     }
 215     for (nn=duf_used; nn>0; ) duf_list[--nn](user, killer, why);
 216     user->uplink->clients--;
 217     dict_remove(user->uplink->users, user->nick);
 218     if (IsOper(user)) userList_remove(&curr_opers, user);
 219     if (IsInvisible(user)) invis_clients--;
 220     if (user == dict_find(clients, user->nick, NULL)) dict_remove(clients, user->nick);
 221     if (announce) {
 222         if (IsLocal(user)) {
 223             irc_quit(user, why);
 224         } else {
 225             irc_kill(killer, user, why);
 226         }
 227     }
 228     modeList_clean(&user->channels);
 229     user->dead = 1;
 230     if (dead_users.size) {
 231         userList_append(&dead_users, user);
 232     } else {
 233         free_user(user);
 234     }
 235 }
 236
 237 void
 238 irc_server(struct server *srv) {
 239     if (srv == self) {
 240         putsock("SERVER %s %d :%s", srv->name, srv->hops, srv->description);
 241     } else {
 242         putsock(":%s SERVER %s %d :%s", self->name, srv->name, srv->hops, srv->description);
 243     }
 244 }
 245
 246 void
 247 irc_user(struct userNode *user) {
 248     char modes[32];
 249     int modelen = 0;
 250     if (IsOper(user)) modes[modelen++] = 'o';
 251     if (IsInvisible(user)) modes[modelen++] = 'i';
 252     if (IsWallOp(user)) modes[modelen++] = 'w';
 253     if (IsService(user)) modes[modelen++] = 'k';
 254     if (IsServNotice(user)) modes[modelen++] = 's';
 255     if (IsDeaf(user)) modes[modelen++] = 'd';
 256     if (IsReggedNick(user)) modes[modelen++] = 'r';
 257     if (IsGlobal(user)) modes[modelen++] = 'g';
 258     modes[modelen] = 0;
 259     putsock("NICK %s %d "FMT_TIME_T" +%s %s %s %s %d %u :%s",
 260             user->nick, user->uplink->hops+2, user->timestamp, modes,
 261             user->ident, user->hostname, user->uplink->name, 0, ntohl(user->ip.s_addr), user->info);
 262 }
 263
 264 void
 265 irc_account(struct userNode *user, const char *stamp)
 266 {
 267     if (IsReggedNick(user)) {
 268         irc_svsmode(user, "+rd", base64toint(stamp, IDLEN));
 269     } else {
 270         irc_svsmode(user, "+d", base64toint(stamp, IDLEN));
 271     }
 272 }
 273
 274 void
 275 irc_fakehost(UNUSED_ARG(struct userNode *user), UNUSED_ARG(const char *host))
 276 {
 277     /* not supported in bahamut */
 278 }
 279
 280 void
 281 irc_regnick(struct userNode *user)
 282 {
 283     if (IsReggedNick(user)) {
 284         irc_svsmode(user, "+r", 0);
 285     } else {
 286         irc_svsmode(user, "-r", 0);
 287     }
 288 }
 289
 290 void
 291 irc_nick(struct userNode *user, const char *old_nick) {
 292     if (user->uplink == self) {
 293         /* update entries in PRIVMSG/NOTICE handlers (if they exist) */
 294         struct service_message_info *smi = dict_find(service_msginfo_dict, user->nick, NULL);
 295         if (smi) {
 296             dict_remove2(service_msginfo_dict, old_nick, 1);
 297             dict_insert(service_msginfo_dict, user->nick, smi);
 298         }
 299     }
 300     putsock(":%s NICK %s :"FMT_TIME_T, old_nick, user->nick, user->timestamp);
 301 }
 302
 303 void
 304 irc_pass(const char *passwd) {
 305     putsock("PASS %s :TS", passwd);
 306 }
 307
 308 void
 309 irc_capab() {
 310     putsock("CAPAB TS3 NOQUIT SSJOIN BURST UNCONNECT NICKIP TSMODE");
 311 }
 312
 313 void
 314 irc_svinfo() {
 315     putsock("SVINFO 3 3 0 :"FMT_TIME_T, now);
 316 }
 317
 318 void
 319 irc_introduce(const char *passwd) {
 320     extern time_t burst_begin;
 321
 322     irc_pass(passwd);
 323     irc_capab();
 324     irc_server(self);
 325     irc_svinfo();
 326     burst_length = 0;
 327     burst_begin = now;
 328     timeq_add(now + ping_freq, timed_send_ping, 0);
 329 }
 330
 331 void
 332 irc_ping(const char *something) {
 333     putsock("PING :%s", something);
 334 }
 335
 336 void
 337 irc_pong(const char *who, const char *data) {
 338     putsock(":%s PONG %s :%s", self->name, who, data);
 339 }
 340
 341 void
 342 irc_quit(struct userNode *user, const char *message) {
 343     putsock(":%s QUIT :%s", user->nick, message);
 344 }
 345
 346 void
 347 irc_squit(struct server *srv, const char *message, const char *service_message) {
 348     if (!service_message) service_message = message;
 349     /* If we're leaving the network, QUIT all our clients. */
 350     if ((srv == self) && (cManager.uplink->state == CONNECTED)) {
 351         dict_iterator_t it;
 352         for (it = dict_first(srv->users); it; it = iter_next(it)) {
 353             irc_quit(iter_data(it), service_message);
 354         }
 355     }
 356     putsock(":%s SQUIT %s 0 :%s", self->name, srv->name, message);
 357     if (srv == self) {
 358         /* Reconnect to the currently selected server. */
 359         cManager.uplink->tries = 0;
 360         log_module(MAIN_LOG, LOG_INFO, "Squitting from uplink: %s", message);
 361         close_socket();
 362     }
 363 }
 364
 365 void
 366 irc_privmsg(struct userNode *from, const char *to, const char *message) {
 367     putsock(":%s PRIVMSG %s :%s", from->nick, to, message);
 368 }
 369
 370 void
 371 irc_notice(struct userNode *from, const char *to, const char *message) {
 372     putsock(":%s NOTICE %s :%s", from->nick, to, message);
 373 }
 374
 375 void
 376 irc_notice_user(struct userNode *from, struct userNode *to, const char *message) {
 377     putsock(":%s NOTICE %s :%s", from->nick, to->nick, message);
 378 }
 379
 380 void
 381 irc_wallchops(UNUSED_ARG(struct userNode *from), UNUSED_ARG(const char *to), UNUSED_ARG(const char *message)) {
 382 }
 383
 384 void
 385 irc_join(struct userNode *who, struct chanNode *what) {
 386     if (what->members.used == 1) {
 387         putsock(":%s SJOIN "FMT_TIME_T" %s + :@%s", self->name, what->timestamp, what->name, who->nick);
 388     } else {
 389         putsock(":%s SJOIN "FMT_TIME_T" %s", who->nick, what->timestamp, what->name);
 390     }
 391 }
 392
 393 void
 394 irc_invite(struct userNode *from, struct userNode *who, struct chanNode *to) {
 395     putsock(":%s INVITE %s %s", from->nick, who->nick, to->name);
 396 }
 397
 398 void
 399 irc_mode(struct userNode *who, struct chanNode *target, const char *modes) {
 400     putsock(":%s MODE %s "FMT_TIME_T" %s", who->nick, target->name, target->timestamp, modes);
 401 }
 402
 403 void
 404 irc_svsmode(struct userNode *target, char *modes, unsigned long stamp) {
 405     extern struct userNode *nickserv;
 406     if (stamp) {
 407         putsock(":%s SVSMODE %s "FMT_TIME_T" %s %lu", nickserv->nick, target->nick, target->timestamp, modes, stamp);
 408     } else {
 409         putsock(":%s SVSMODE %s "FMT_TIME_T" %s", nickserv->nick, target->nick, target->timestamp, modes);
 410     }
 411 }
 412
 413 void
 414 irc_kick(struct userNode *who, struct userNode *target, struct chanNode *from, const char *msg) {
 415     putsock(":%s KICK %s %s :%s", who->nick, from->name, target->nick, msg);
 416     ChannelUserKicked(who, target, from);
 417 }
 418
 419 void
 420 irc_part(struct userNode *who, struct chanNode *what, const char *reason) {
 421     if (reason) {
 422         putsock(":%s PART %s :%s", who->nick, what->name, reason);
 423     } else {
 424         putsock(":%s PART %s", who->nick, what->name);
 425     }
 426 }
 427
 428 void
 429 irc_topic(struct userNode *who, struct chanNode *what, const char *topic) {
 430     putsock(":%s TOPIC %s :%s", who->nick, what->name, topic);
 431 }
 432
 433 void
 434 irc_fetchtopic(struct userNode *from, const char *to) {
 435     if (!from || !to) return;
 436     putsock(":%s TOPIC %s", from->nick, to);
 437 }
 438
 439 void
 440 irc_gline(struct server *srv, struct gline *gline) {
 441     char host[HOSTLEN+1], ident[USERLEN+1], *sep;
 442     unsigned int len;
 443     if (srv) {
 444         log_module(MAIN_LOG, LOG_WARNING, "%s tried to send a targeted G-line for %s (not supported by protocol!)", gline->issuer, gline->target);
 445         return;
 446     }
 447     if (!(sep = strchr(gline->target, '@'))) {
 448         log_module(MAIN_LOG, LOG_ERROR, "%s tried to add G-line with bad mask %s", gline->issuer, gline->target);
 449         return;
 450     }
 451     len = sep - gline->target + 1;
 452     if (len > ArrayLength(ident)) len = ArrayLength(ident);
 453     safestrncpy(ident, gline->target, len);
 454     safestrncpy(host, sep+1, ArrayLength(host));
 455     putsock(":%s AKILL %s %s "FMT_TIME_T" %s "FMT_TIME_T" :%s", self->name, host, ident, gline->expires-gline->issued, gline->issuer, gline->issued, gline->reason);
 456 }
 457
 458 void
 459 irc_settime(UNUSED_ARG(const char *srv_name_mask), UNUSED_ARG(time_t new_time))
 460 {
 461     /* Bahamut has nothing like this, so ignore it. */
 462 }
 463
 464 void
 465 irc_ungline(const char *mask) {
 466     char host[HOSTLEN+1], ident[USERLEN+1], *sep;
 467     unsigned int len;
 468     if (!(sep = strchr(mask, '@'))) {
 469         log_module(MAIN_LOG, LOG_ERROR, "Tried to remove G-line with bad mask %s", mask);
 470         return;
 471     }
 472     len = sep - mask + 1;
 473     if (len > ArrayLength(ident)) len = ArrayLength(ident);
 474     safestrncpy(ident, mask, len);
 475     safestrncpy(host, sep+1, ArrayLength(host));
 476     putsock(":%s RAKILL %s %s", self->name, host, ident);
 477 }
 478
 479 void
 480 irc_error(const char *to, const char *message) {
 481     if (to) {
 482         putsock("%s ERROR :%s", to, message);
 483     } else {
 484         putsock(":%s ERROR :%s", self->name, message);
 485     }
 486 }
 487
 488 void
 489 irc_kill(struct userNode *from, struct userNode *target, const char *message) {
 490     if (from) {
 491         putsock(":%s KILL %s :%s!%s (%s)", from->nick, target->nick, self->name, from->nick, message);
 492     } else {
 493         putsock(":%s KILL %s :%s (%s)", self->name, target->nick, self->name, message);
 494     }
 495 }
 496
 497 void
 498 irc_raw(const char *what) {
 499     putsock("%s", what);
 500 }
 501
 502 void
 503 irc_stats(struct userNode *from, struct server *target, char type) {
 504     putsock(":%s STATS %c :%s", from->nick, type, target->name);
 505 }
 506
 507 void
 508 irc_svsnick(struct userNode *from, struct userNode *target, const char *newnick)
 509 {
 510     putsock(":%s SVSNICK %s %s :"FMT_TIME_T, from->nick, target->nick, newnick, now);
 511 }
 512
 513 void
 514 irc_numeric(struct userNode *user, unsigned int num, const char *format, ...) {
 515     va_list arg_list;
 516     char buffer[MAXLEN];
 517     va_start(arg_list, format);
 518     vsnprintf(buffer, MAXLEN-2, format, arg_list);
 519     buffer[MAXLEN-1] = 0;
 520     putsock(":%s %03d %s %s", self->name, num, user->nick, buffer);
 521 }
 522
 523 static void
 524 parse_foreach(char *target_list, foreach_chanfunc cf, foreach_nonchan nc, foreach_userfunc uf, foreach_nonuser nu, void *data) {
 525     char *j, old;
 526     do {
 527         j = target_list;
 528         while (*j != 0 && *j != ',') j++;
 529         old = *j;
 530         *j = 0;
 531         if (IsChannelName(target_list)) {
 532             struct chanNode *chan = GetChannel(target_list);
 533             if (chan) {
 534                 if (cf) cf(chan, data);
 535             } else {
 536                 if (nc) nc(target_list, data);
 537             }
 538         } else {
 539             struct userNode *user;
 540             struct privmsg_desc *pd = data;
 541
 542             pd->is_qualified = 0;
 543             if (*target_list == '@') {
 544                 user = NULL;
 545             } else if (strchr(target_list, '@')) {
 546                 struct server *server;
 547
 548                 pd->is_qualified = 1;
 549                 user = GetUserH(strtok(target_list, "@"));
 550                 server = GetServerH(strtok(NULL, "@"));
 551
 552                 if (user && (user->uplink != server)) {
 553                     /* Don't attempt to index into any arrays
 554                        using a user's numeric on another server. */
 555                     user = NULL;
 556                 }
 557             } else {
 558                 user = GetUserH(target_list);
 559             }
 560
 561             if (user) {
 562                 if (uf) uf(user, data);
 563             } else {
 564                 if (nu) nu(target_list, data);
 565             }
 566         }
 567         target_list = j+1;
 568     } while (old == ',');
 569 }
 570
 571 static CMD_FUNC(cmd_notice) {
 572     struct privmsg_desc pd;
 573     if ((argc < 3) || !origin) return 0;
 574     if (!(pd.user = GetUserH(origin))) return 1;
 575     if (IsGagged(pd.user) && !IsOper(pd.user)) return 1;
 576     pd.is_notice = 1;
 577     pd.text = argv[2];
 578     parse_foreach(argv[1], privmsg_chan_helper, NULL, privmsg_user_helper, privmsg_invalid, &pd);
 579     return 1;
 580 }
 581
 582 static CMD_FUNC(cmd_privmsg) {
 583     struct privmsg_desc pd;
 584     if ((argc < 3) || !origin) return 0;
 585     if (!(pd.user = GetUserH(origin))) return 1;
 586     if (IsGagged(pd.user) && !IsOper(pd.user)) return 1;
 587     pd.is_notice = 0;
 588     pd.text = argv[2];
 589     parse_foreach(argv[1], privmsg_chan_helper, NULL, privmsg_user_helper, privmsg_invalid, &pd);
 590     return 1;
 591 }
 592
 593 static CMD_FUNC(cmd_whois) {
 594     struct userNode *from;
 595     struct userNode *who;
 596
 597     if (argc < 3)
 598         return 0;
 599     if (!(from = GetUserH(origin))) {
 600         log_module(MAIN_LOG, LOG_ERROR, "Could not find WHOIS origin user %s", origin);
 601         return 0;
 602     }
 603     if(!(who = GetUserH(argv[2]))) {
 604         irc_numeric(from, ERR_NOSUCHNICK, "%s@%s :No such nick", argv[2], self->name);
 605         return 1;
 606     }
 607     if (IsHiddenHost(who) && !IsOper(from)) {
 608         /* Just stay quiet. */
 609         return 1;
 610     }
 611     irc_numeric(from, RPL_WHOISUSER, "%s %s %s * :%s", who->nick, who->ident, who->hostname, who->info);
 612 #ifdef WITH_PROTOCOL_P10
 613     if (his_servername && his_servercomment)
 614       irc_numeric(from, RPL_WHOISSERVER, "%s %s :%s", who->nick, his_servername, his_servercomment);
 615     else
 616 #endif
 617     irc_numeric(from, RPL_WHOISSERVER, "%s %s :%s", who->nick, who->uplink->name, who->uplink->description);
 618
 619     if (IsOper(who)) {
 620         irc_numeric(from, RPL_WHOISOPERATOR, "%s :is a megalomaniacal power hungry tyrant", who->nick);
 621     }
 622     irc_numeric(from, RPL_ENDOFWHOIS, "%s :End of /WHOIS list", who->nick);
 623     return 1;
 624 }
 625
 626 static CMD_FUNC(cmd_capab) {
 627     static const struct {
 628         const char *name;
 629         unsigned int mask;
 630     } capabs[] = {
 631         { "TS3", CAPAB_TS3 },
 632         { "NOQUIT", CAPAB_NOQUIT },
 633         { "SSJOIN", CAPAB_SSJOIN },
 634         { "BURST", CAPAB_BURST },
 635         { "UNCONNECT", CAPAB_UNCONNECT },
 636         { "NICKIP", CAPAB_NICKIP },
 637         { "TSMODE", CAPAB_TSMODE },
 638         { "ZIP", CAPAB_ZIP },
 639         { NULL, 0 }
 640     };
 641     unsigned int nn, mm;
 642
 643     uplink_capab = 0;
 644     for(nn=1; nn<argc; nn++) {
 645         for (mm=0; capabs[mm].name && irccasecmp(capabs[mm].name, argv[nn]); mm++) ;
 646         if (capabs[mm].name) {
 647             uplink_capab |= capabs[mm].mask;
 648         } else {
 649             log_module(MAIN_LOG, LOG_INFO, "Saw unrecognized/unhandled capability %s.  Please notify srvx developers so they can add it.", argv[nn]);
 650         }
 651     }
 652     return 1;
 653 }
 654
 655 static void burst_channel(struct chanNode *chan) {
 656     char line[510];
 657     int pos, base_len, len, queued;
 658     unsigned int nn;
 659
 660     if (!chan->members.used) return;
 661     /* send list of users in the channel.. */
 662     base_len = sprintf(line, ":%s SJOIN "FMT_TIME_T" %s ", self->name, chan->timestamp, chan->name);
 663     len = irc_make_chanmode(chan, line+base_len);
 664     pos = base_len + len;
 665     line[pos++] = ' ';
 666     line[pos++] = ':';
 667     for (nn=0; nn<chan->members.used; nn++) {
 668         struct modeNode *mn = chan->members.list[nn];
 669         len = strlen(mn->user->nick);
 670         if (pos + len > 500) {
 671             line[pos-1] = 0;
 672             putsock("%s", line);
 673             pos = base_len;
 674             line[pos++] = '0';
 675             line[pos++] = ' ';
 676             line[pos++] = ':';
 677         }
 678         if (mn->modes & MODE_CHANOP) line[pos++] = '@';
 679         if (mn->modes & MODE_VOICE) line[pos++] = '+';
 680         memcpy(line+pos, mn->user->nick, len);
 681         pos = pos + len;
 682         line[pos++] = ' ';
 683     }
 684     /* print the last line */
 685     line[pos] = 0;
 686     putsock("%s", line);
 687     /* now send the bans.. */
 688     base_len = sprintf(line, ":%s MODE "FMT_TIME_T" %s +", self->name, chan->timestamp, chan->name);
 689     pos = sizeof(line)-1;
 690     line[pos] = 0;
 691     for (nn=queued=0; nn<chan->banlist.used; nn++) {
 692         struct banNode *bn = chan->banlist.list[nn];
 693         len = strlen(bn->ban);
 694         if (pos < base_len+queued+len+4) {
 695             while (queued) {
 696                 line[--pos] = 'b';
 697                 queued--;
 698             }
 699             putsock("%s%s", line, line+pos);
 700             pos = sizeof(line)-1;
 701         }
 702         pos -= len;
 703         memcpy(line+pos, bn->ban, len);
 704         line[--pos] = ' ';
 705         queued++;
 706     }
 707     if (queued) {
 708         while (queued) {
 709             line[--pos] = 'b';
 710             queued--;
 711         }
 712         putsock("%s%s", line, line+pos);
 713     }
 714 }
 715
 716 static void send_burst() {
 717     dict_iterator_t it;
 718     for (it = dict_first(servers); it; it = iter_next(it)) {
 719         struct server *serv = iter_data(it);
 720         if ((serv != self) && (serv != self->uplink)) irc_server(serv);
 721     }
 722     putsock("BURST");
 723     for (it = dict_first(clients); it; it = iter_next(it)) {
 724         irc_user(iter_data(it));
 725     }
 726     for (it = dict_first(channels); it; it = iter_next(it)) {
 727         burst_channel(iter_data(it));
 728     }
 729     /* Uplink will do AWAY and TOPIC bursts before sending BURST 0, but we don't */
 730     putsock("BURST 0");
 731 }
 732
 733 static CMD_FUNC(cmd_server) {
 734     if (argc < 4) return 0;
 735     if (origin) {
 736         AddServer(GetServerH(origin), argv[1], atoi(argv[2]), 0, now, 0, argv[3]);
 737     } else {
 738         self->uplink = AddServer(self, argv[1], atoi(argv[2]), 0, now, 0, argv[3]);
 739     }
 740     return 1;
 741 }
 742
 743 static CMD_FUNC(cmd_svinfo) {
 744     if (argc < 5) return 0;
 745     if ((atoi(argv[1]) < 3) || (atoi(argv[2]) > 3)) return 0;
 746     /* TODO: something with the timestamp we get from the other guy */
 747     send_burst();
 748     return 1;
 749 }
 750
 751 static CMD_FUNC(cmd_ping)
 752 {
 753     irc_pong(self->name, argc > 1 ? argv[1] : origin);
 754     timeq_del(0, timed_send_ping, 0, TIMEQ_IGNORE_WHEN|TIMEQ_IGNORE_DATA);
 755     timeq_del(0, timed_ping_timeout, 0, TIMEQ_IGNORE_WHEN|TIMEQ_IGNORE_DATA);
 756     timeq_add(now + ping_freq, timed_send_ping, 0);
 757     received_ping();
 758     return 1;
 759 }
 760
 761 static CMD_FUNC(cmd_burst) {
 762     struct server *sender = GetServerH(origin);
 763     if (!sender) return 0;
 764     if (argc == 1) return 1;
 765     if (sender == self->uplink) {
 766         cManager.uplink->state = CONNECTED;
 767     }
 768     sender->self_burst = 0;
 769     recalc_bursts(sender);
 770     return 1;
 771 }
 772
 773 static CMD_FUNC(cmd_nick) {
 774     struct userNode *un;
 775     if ((un = GetUserH(origin))) {
 776         /* nick change */
 777         if (argc < 2) return 0;
 778         NickChange(un, argv[1], 1);
 779     } else {
 780         /* new nick from a server */
 781         char id[8];
 782         unsigned long stamp;
 783         struct in_addr ip;
 784
 785         if (argc < 10) return 0;
 786         stamp = strtoul(argv[8], NULL, 0);
 787         if (stamp) inttobase64(id, stamp, IDLEN);
 788         ip.s_addr = (argc > 10) ? atoi(argv[9]) : 0;
 789         un = AddUser(GetServerH(argv[7]), argv[1], argv[5], argv[6], argv[4], argv[argc-1], atoi(argv[3]), ip, (stamp ? id : 0));
 790     }
 791     return 1;
 792 }
 793
 794 static CMD_FUNC(cmd_sjoin) {
 795     struct chanNode *cNode;
 796     struct userNode *uNode;
 797     struct modeNode *mNode;
 798     unsigned int next = 4, last;
 799     char *nick, *nickend;
 800
 801     if ((argc == 3) && (uNode = GetUserH(origin))) {
 802         /* normal JOIN */
 803         if (!(cNode = GetChannel(argv[2]))) {
 804             log_module(MAIN_LOG, LOG_ERROR, "Unable to find SJOIN target %s", argv[2]);
 805             return 0;
 806         }
 807         AddChannelUser(uNode, cNode);
 808         return 1;
 809     }
 810     if (argc < 5) return 0;
 811     if (argv[3][0] == '+') {
 812         char modes[MAXLEN], *pos;
 813         int n_modes;
 814         for (pos = argv[3], n_modes = 1; *pos; pos++) {
 815             if ((*pos == 'k') || (*pos == 'l')) n_modes++;
 816         }
 817         unsplit_string(argv+3, n_modes, modes);
 818         cNode = AddChannel(argv[2], atoi(argv[1]), modes, NULL);
 819     } else if (argv[3][0] == '0') {
 820         cNode = GetChannel(argv[2]);
 821     } else {
 822         log_module(MAIN_LOG, LOG_ERROR, "Unsure how to handle SJOIN when arg 3 is %s", argv[3]);
 823         return 0;
 824     }
 825
 826     /* argv[next] is now the space-delimited, @+-prefixed list of
 827      * nicks in the channel.  Split it and add the users. */
 828     for (last = 0, nick = argv[next]; !last; nick = nickend + 1) {
 829         int mode = 0;
 830         for (nickend = nick; *nickend && (*nickend != ' '); nickend++) ;
 831         if (!*nickend) last = 1;
 832         *nickend = 0;
 833         if (*nick == '@') { mode |= MODE_CHANOP; nick++; }
 834         if (*nick == '+') { mode |= MODE_VOICE; nick++; }
 835         if ((uNode = GetUserH(nick)) && (mNode = AddChannelUser(uNode, cNode))) {
 836             mNode->modes = mode;
 837         }
 838     }
 839     return 1;
 840 }
 841
 842 static CMD_FUNC(cmd_mode) {
 843     struct userNode *un;
 844
 845     if (argc < 2) {
 846         log_module(MAIN_LOG, LOG_ERROR, "Illegal MODE from %s (no arguments).", origin);
 847         return 0;
 848     } else if (IsChannelName(argv[1])) {
 849         struct chanNode *cn;
 850         struct modeNode *mn;
 851
 852         if (!(cn = GetChannel(argv[1]))) {
 853             log_module(MAIN_LOG, LOG_ERROR, "Unable to find channel %s whose mode is changing", argv[1]);
 854             return 0;
 855         }
 856
 857         if ((un = GetUserH(origin))) {
 858             /* Update idle time for the user */
 859             if ((mn = GetUserMode(cn, un)))
 860                 mn->idle_since = now;
 861         } else {
 862             /* Must be a server in burst or something.  Make sure we're using the right timestamp. */
 863             cn->timestamp = atoi(argv[2]);
 864         }
 865
 866         return mod_chanmode(un, cn, argv+3, argc-3, MCP_ALLOW_OVB|MCP_FROM_SERVER|MC_ANNOUNCE);
 867     } else if ((un = GetUserH(argv[1]))) {
 868         mod_usermode(un, argv[2]);
 869         return 1;
 870     } else {
 871         log_module(MAIN_LOG, LOG_ERROR, "Not sure what MODE %s is affecting (not a channel name and no such user)", argv[1]);
 872         return 0;
 873     }
 874 }
 875
 876 static CMD_FUNC(cmd_topic) {
 877     struct chanNode *cn;
 878     if (argc < 5) return 0;
 879     if (!(cn = GetChannel(argv[1]))) {
 880         log_module(MAIN_LOG, LOG_ERROR, "Unable to find channel %s whose topic is being set", argv[1]);
 881         return 0;
 882     }
 883     if (irccasecmp(origin, argv[2])) {
 884         /* coming from a topic burst; the origin is a server */
 885         safestrncpy(cn->topic, argv[4], sizeof(cn->topic));
 886         safestrncpy(cn->topic_nick, argv[2], sizeof(cn->topic_nick));
 887         cn->topic_time = atoi(argv[3]);
 888     } else {
 889         SetChannelTopic(cn, GetUserH(argv[2]), argv[4], 0);
 890     }
 891     return 1;
 892 }
 893
 894 static CMD_FUNC(cmd_part) {
 895     if (argc < 2) return 0;
 896     parse_foreach(argv[1], part_helper, NULL, NULL, NULL, GetUserH(origin));
 897     return 1;
 898 }
 899
 900 static CMD_FUNC(cmd_away) {
 901     struct userNode *un;
 902
 903     if (!(un = GetUserH(origin))) {
 904         log_module(MAIN_LOG, LOG_ERROR, "Unable to find user %s sending AWAY", origin);
 905         return 0;
 906     }
 907     if (argc > 1) {
 908         un->modes |= FLAGS_AWAY;
 909     } else {
 910         un->modes &= ~FLAGS_AWAY;
 911     }
 912     return 1;
 913 }
 914
 915 static CMD_FUNC(cmd_kick) {
 916     if (argc < 3) return 0;
 917     ChannelUserKicked(GetUserH(origin), GetUserH(argv[2]), GetChannel(argv[1]));
 918     return 1;
 919 }
 920
 921 static CMD_FUNC(cmd_kill) {
 922     struct userNode *user;
 923     if (argc < 3) return 0;
 924     if (!(user = GetUserH(argv[1]))) {
 925         log_module(MAIN_LOG, LOG_ERROR, "Unable to find kill victim %s", argv[1]);
 926         return 0;
 927     }
 928     if (IsLocal(user) && IsService(user)) {
 929         ReintroduceUser(user);
 930     } else {
 931         DelUser(user, GetUserH(origin), false, ((argc >= 3) ? argv[2] : NULL));
 932     }
 933     return 1;
 934 }
 935
 936 static CMD_FUNC(cmd_pong)
 937 {
 938     if (argc < 3) return 0;
 939     if (!strcmp(argv[2], self->name)) {
 940         timeq_del(0, timed_send_ping, 0, TIMEQ_IGNORE_WHEN|TIMEQ_IGNORE_DATA);
 941         timeq_del(0, timed_ping_timeout, 0, TIMEQ_IGNORE_WHEN|TIMEQ_IGNORE_DATA);
 942         timeq_add(now + ping_freq, timed_send_ping, 0);
 943         received_ping();
 944     }
 945     return 1;
 946 }
 947
 948 static CMD_FUNC(cmd_num_topic)
 949 {
 950     static struct chanNode *cn;
 951
 952     if (!argv[0])
 953         return 0; /* huh? */
 954     if (argv[2]) {
 955         cn = GetChannel(argv[2]);
 956         if (!cn) {
 957             log_module(MAIN_LOG, LOG_ERROR, "Unable to find channel %s in topic reply", argv[2]);
 958             return 0;
 959         }
 960     } else
 961         return 0;
 962
 963     switch (atoi(argv[0])) {
 964     case 331:
 965         cn->topic_time = 0;
 966         break;  /* no topic */
 967     case 332:
 968         if (argc < 4)
 969             return 0;
 970         safestrncpy(cn->topic, unsplit_string(argv+3, argc-3, NULL), sizeof(cn->topic));
 971         break;
 972     case 333:
 973         if (argc < 5)
 974             return 0;
 975         safestrncpy(cn->topic_nick, argv[3], sizeof(cn->topic_nick));
 976         cn->topic_time = atoi(argv[4]);
 977         break;
 978     default:
 979         return 0; /* should never happen */
 980     }
 981     return 1;
 982 }
 983
 984 static CMD_FUNC(cmd_quit)
 985 {
 986     struct userNode *user;
 987     if (argc < 2) return 0;
 988     /* Sometimes we get a KILL then a QUIT or the like, so we don't want to
 989      * call DelUser unless we have the user in our grasp. */
 990     if ((user = GetUserH(origin))) {
 991         DelUser(user, NULL, false, argv[1]);
 992     }
 993     return 1;
 994 }
 995
 996 static CMD_FUNC(cmd_squit)
 997 {
 998     struct server *server;
 999     if (argc < 3)
1000         return 0;
1001     if (!(server = GetServerH(argv[1])))
1002         return 0;
1003     if (server == self->uplink) {
1004         /* Force a reconnect to the currently selected server. */
1005         cManager.uplink->tries = 0;
1006         log_module(MAIN_LOG, LOG_INFO, "Squitting from uplink: %s", argv[3]);
1007         close_socket();
1008         return 1;
1009     }
1010
1011     DelServer(server, 0, argv[3]);
1012     return 1;
1013 }
1014
1015 static CMD_FUNC(cmd_svsnick)
1016 {
1017     struct userNode *target, *dest;
1018     if (argc < 4) return 0;
1019     if (!(target = GetUserH(argv[1]))) return 0;
1020     if (!IsLocal(target)) return 0;
1021     if ((dest = GetUserH(argv[2]))) return 0; /* Note: Bahamut will /KILL instead. */
1022     NickChange(target, argv[2], 0);
1023     return 1;
1024 }
1025
1026 static oper_func_t *of_list;
1027 static unsigned int of_size = 0, of_used = 0;
1028
1029 void parse_cleanup(void) {
1030     unsigned int nn;
1031     if (of_list) free(of_list);
1032     dict_delete(irc_func_dict);
1033     dict_delete(service_msginfo_dict);
1034     free(mcf_list);
1035     for (nn=0; nn<dead_users.used; nn++) free_user(dead_users.list[nn]);
1036     userList_clean(&dead_users);
1037 }
1038
1039 void init_parse(void) {
1040     const char *str, *desc;
1041
1042     str = conf_get_data("server/ping_freq", RECDB_QSTRING);
1043     ping_freq = str ? ParseInterval(str) : 120;
1044     str = conf_get_data("server/ping_timeout", RECDB_QSTRING);
1045     ping_timeout = str ? ParseInterval(str) : 30;
1046     str = conf_get_data("server/hostname", RECDB_QSTRING);
1047     desc = conf_get_data("server/description", RECDB_QSTRING);
1048     if (!str || !desc) {
1049         log_module(MAIN_LOG, LOG_ERROR, "No server/hostname entry in config file.");
1050         exit(1);
1051     }
1052     self = AddServer(NULL, str, 0, boot_time, now, NULL, desc);
1053
1054     str = conf_get_data("server/ping_freq", RECDB_QSTRING);
1055     ping_freq = str ? ParseInterval(str) : 120;
1056     str = conf_get_data("server/ping_timeout", RECDB_QSTRING);
1057     ping_timeout = str ? ParseInterval(str) : 30;
1058
1059     service_msginfo_dict = dict_new();
1060     dict_set_free_data(service_msginfo_dict, free);
1061     irc_func_dict = dict_new();
1062     dict_insert(irc_func_dict, "ADMIN", cmd_admin);
1063     dict_insert(irc_func_dict, "AWAY", cmd_away);
1064     dict_insert(irc_func_dict, "BURST", cmd_burst);
1065     dict_insert(irc_func_dict, "CAPAB", cmd_capab);
1066     dict_insert(irc_func_dict, "ERROR", cmd_error);
1067     dict_insert(irc_func_dict, "GNOTICE", cmd_dummy);
1068     dict_insert(irc_func_dict, "INVITE", cmd_dummy);
1069     dict_insert(irc_func_dict, "KICK", cmd_kick);
1070     dict_insert(irc_func_dict, "KILL", cmd_kill);
1071     dict_insert(irc_func_dict, "LUSERSLOCK", cmd_dummy);
1072     dict_insert(irc_func_dict, "MODE", cmd_mode);
1073     dict_insert(irc_func_dict, "NICK", cmd_nick);
1074     dict_insert(irc_func_dict, "NOTICE", cmd_notice);
1075     dict_insert(irc_func_dict, "PART", cmd_part);
1076     dict_insert(irc_func_dict, "PASS", cmd_pass);
1077     dict_insert(irc_func_dict, "PING", cmd_ping);
1078     dict_insert(irc_func_dict, "PONG", cmd_pong);
1079     dict_insert(irc_func_dict, "PRIVMSG", cmd_privmsg);
1080     dict_insert(irc_func_dict, "QUIT", cmd_quit);
1081     dict_insert(irc_func_dict, "SERVER", cmd_server);
1082     dict_insert(irc_func_dict, "SJOIN", cmd_sjoin);
1083     dict_insert(irc_func_dict, "SQUIT", cmd_squit);
1084     dict_insert(irc_func_dict, "STATS", cmd_stats);
1085     dict_insert(irc_func_dict, "SVSNICK", cmd_svsnick);
1086     dict_insert(irc_func_dict, "SVINFO", cmd_svinfo);
1087     dict_insert(irc_func_dict, "TOPIC", cmd_topic);
1088     dict_insert(irc_func_dict, "VERSION", cmd_version);
1089     dict_insert(irc_func_dict, "WHOIS", cmd_whois);
1090     dict_insert(irc_func_dict, "331", cmd_num_topic);
1091     dict_insert(irc_func_dict, "332", cmd_num_topic);
1092     dict_insert(irc_func_dict, "333", cmd_num_topic);
1093     dict_insert(irc_func_dict, "413", cmd_num_topic);
1094
1095     userList_init(&dead_users);
1096     reg_exit_func(parse_cleanup);
1097 }
1098
1099 int parse_line(char *line, int recursive) {
1100     char *argv[MAXNUMPARAMS];
1101     int argc, cmd, res;
1102     cmd_func_t *func;
1103
1104     argc = split_line(line, true, ArrayLength(argv), argv);
1105     cmd = line[0] == ':';
1106     if ((argc > cmd) && (func = dict_find(irc_func_dict, argv[cmd], NULL))) {
1107         char *origin;
1108         if (cmd) {
1109             origin = argv[0] + 1;
1110         } else if (self->uplink) {
1111             origin = self->uplink->name;
1112         } else {
1113             origin = NULL;
1114         }
1115         res = func(origin, argc-cmd, argv+cmd);
1116     } else {
1117         res = 0;
1118     }
1119     if (!res) {
1120         log_module(MAIN_LOG, LOG_ERROR, "PARSE ERROR on line: %s", unsplit_string(argv, argc, NULL));
1121     } else if (!recursive) {
1122         unsigned int i;
1123         for (i=0; i<dead_users.used; i++) {
1124             free_user(dead_users.list[i]);
1125         }
1126         dead_users.used = 0;
1127     }
1128     return res;
1129 }
1130
1131 static void
1132 privmsg_user_helper(struct userNode *un, void *data)
1133 {
1134     struct privmsg_desc *pd = data;
1135     struct service_message_info *info = dict_find(service_msginfo_dict, un->nick, 0);
1136     if (info) {
1137         if (pd->is_notice) {
1138             if (info->on_notice) info->on_notice(pd->user, un, pd->text, pd->is_qualified);
1139         } else {
1140             if (info->on_privmsg) info->on_privmsg(pd->user, un, pd->text, pd->is_qualified);
1141         }
1142     }
1143 }
1144
1145 void
1146 reg_privmsg_func(struct userNode *user, privmsg_func_t handler) {
1147     struct service_message_info *info = dict_find(service_msginfo_dict, user->nick, NULL);
1148     if (!info) {
1149         info = calloc(1, sizeof(*info));
1150         dict_insert(service_msginfo_dict, user->nick, info);
1151     }
1152     info->on_privmsg = handler;
1153 }
1154
1155 void
1156 reg_notice_func(struct userNode *user, privmsg_func_t handler) {
1157     struct service_message_info *info = dict_find(service_msginfo_dict, user->nick, NULL);
1158     if (!info) {
1159         info = calloc(1, sizeof(*info));
1160         dict_insert(service_msginfo_dict, user->nick, info);
1161     }
1162     info->on_notice = handler;
1163 }
1164
1165 void
1166 reg_oper_func(oper_func_t handler)
1167 {
1168     if (of_used == of_size) {
1169         if (of_size) {
1170             of_size <<= 1;
1171             of_list = realloc(of_list, of_size*sizeof(oper_func_t));
1172         } else {
1173             of_size = 8;
1174             of_list = malloc(of_size*sizeof(oper_func_t));
1175         }
1176     }
1177     of_list[of_used++] = handler;
1178 }
1179
1180 static void
1181 call_oper_funcs(struct userNode *user)
1182 {
1183     unsigned int n;
1184     if (IsLocal(user)) return;
1185     for (n=0; n<of_used; n++)
1186     {
1187         of_list[n](user);
1188     }
1189 }
1190
1191 void mod_usermode(struct userNode *user, const char *mode_change) {
1192     int add = 1;
1193
1194     if (!user || !mode_change) return;
1195     while (1) {
1196 #define do_user_mode(FLAG) do { if (add) user->modes |= FLAG; else user->modes &= ~FLAG; } while (0)
1197         switch (*mode_change++) {
1198         case 0: return;
1199         case '+': add = 1; break;
1200         case '-': add = 0; break;
1201         case 'o':
1202             do_user_mode(FLAGS_OPER);
1203             if (add) {
1204                 userList_append(&curr_opers, user);
1205                 call_oper_funcs(user);
1206             } else {
1207                 userList_remove(&curr_opers, user);
1208             }
1209             break;
1210         case 'O': do_user_mode(FLAGS_LOCOP); break;
1211         case 'i': do_user_mode(FLAGS_INVISIBLE);
1212             if (add) invis_clients++; else invis_clients--;
1213             break;
1214         case 'w': do_user_mode(FLAGS_WALLOP); break;
1215         case 's': do_user_mode(FLAGS_SERVNOTICE); break;
1216         case 'd': do_user_mode(FLAGS_DEAF); break;
1217         case 'r': do_user_mode(FLAGS_REGNICK); break;
1218         case 'k': do_user_mode(FLAGS_SERVICE); break;
1219         case 'g': do_user_mode(FLAGS_GLOBAL); break;
1220         case 'h': do_user_mode(FLAGS_HELPER); break;
1221         }
1222 #undef do_user_mode
1223     }
1224 }
1225
1226 struct mod_chanmode *
1227 mod_chanmode_parse(struct chanNode *channel, char **modes, unsigned int argc, unsigned int flags)
1228 {
1229     struct mod_chanmode *change;
1230     unsigned int ii, in_arg, ch_arg, add;
1231
1232     if (argc == 0)
1233         return NULL;
1234     if (!(change = mod_chanmode_alloc(argc)))
1235         return NULL;
1236
1237     for (ii = ch_arg = 0, in_arg = add = 1;
1238          (modes[0][ii] != '\0') && (modes[0][ii] != ' ');
1239          ++ii) {
1240         switch (modes[0][ii]) {
1241         case '+':
1242             add = 1;
1243             break;
1244         case '-':
1245             add = 0;
1246             break;
1247 #define do_chan_mode(FLAG) do { if (add) change->modes_set |= FLAG, change->modes_clear &= ~FLAG; else change->modes_clear |= FLAG, change->modes_set &= ~FLAG; } while(0)
1248         case 'R': do_chan_mode(MODE_REGONLY); break;
1249         case 'r': do_chan_mode(MODE_REGISTERED); break;
1250         case 'D': do_chan_mode(MODE_DELAYJOINS); break;
1251         case 'c': do_chan_mode(MODE_NOCOLORS); break;
1252         case 'i': do_chan_mode(MODE_INVITEONLY); break;
1253         case 'm': do_chan_mode(MODE_MODERATED); break;
1254         case 'n': do_chan_mode(MODE_NOPRIVMSGS); break;
1255         case 'p': do_chan_mode(MODE_PRIVATE); break;
1256         case 's': do_chan_mode(MODE_SECRET); break;
1257         case 't': do_chan_mode(MODE_TOPICLIMIT); break;
1258         case 'r':
1259             if (!(flags & MCP_REGISTERED)) {
1260              do_chan_mode(MODE_REGISTERED);
1261             } else {
1262              mod_chanmode_free(change);
1263              return NULL;
1264             }
1265             break;
1266 #undef do_chan_mode
1267         case 'l':
1268             if (add) {
1269                 if (in_arg >= argc)
1270                     goto error;
1271                 change->modes_set |= MODE_LIMIT;
1272                 change->new_limit = atoi(modes[in_arg++]);
1273             } else {
1274                 change->modes_clear |= MODE_LIMIT;
1275             }
1276             break;
1277         case 'k':
1278             if (add) {
1279                 if (in_arg >= argc)
1280                     goto error;
1281                 change->modes_set |= MODE_KEY;
1282                 safestrncpy(change->new_key, modes[in_arg++], sizeof(change->new_key));
1283             } else {
1284                 change->modes_clear |= MODE_KEY;
1285                 if (!(flags & MCP_KEY_FREE)) {
1286                     if (in_arg >= argc)
1287                         goto error;
1288                     in_arg++;
1289                 }
1290             }
1291             break;
1292         case 'b':
1293             if (!(flags & MCP_ALLOW_OVB))
1294                 goto error;
1295             if (in_arg >= argc)
1296                 goto error;
1297             change->args[ch_arg].mode = MODE_BAN;
1298             if (!add)
1299                 change->args[ch_arg].mode |= MODE_REMOVE;
1300             change->args[ch_arg++].hostmask = modes[in_arg++];
1301             break;
1302         case 'o': case 'v':
1303         {
1304             struct userNode *victim;
1305             if (!(flags & MCP_ALLOW_OVB))
1306                 goto error;
1307             if (in_arg >= argc)
1308                 goto error;
1309             change->args[ch_arg].mode = (modes[0][ii] == 'o') ? MODE_CHANOP : MODE_VOICE;
1310             if (!add)
1311                 change->args[ch_arg].mode |= MODE_REMOVE;
1312             victim = GetUserH(modes[in_arg++]);
1313             if (!victim)
1314                 continue;
1315             if ((change->args[ch_arg].member = GetUserMode(channel, victim)))
1316                 ch_arg++;
1317             break;
1318         }
1319         default:
1320             if (!(flags & MCP_FROM_SERVER))
1321                 goto error;
1322             break;
1323         }
1324     }
1325     change->argc = ch_arg; /* in case any turned out to be ignored */
1326     if (change->modes_set & MODE_SECRET) {
1327         change->modes_set &= ~(MODE_PRIVATE);
1328         change->modes_clear |= MODE_PRIVATE;
1329     } else if (change->modes_set & MODE_PRIVATE) {
1330         change->modes_set &= ~(MODE_SECRET);
1331         change->modes_clear |= MODE_SECRET;
1332     }
1333     return change;
1334   error:
1335     mod_chanmode_free(change);
1336     return NULL;
1337 }
1338
1339 struct chanmode_buffer {
1340     char modes[MAXLEN];
1341     char args[MAXLEN];
1342     struct chanNode *channel;
1343     struct userNode *actor;
1344     unsigned int modes_used;
1345     unsigned int args_used;
1346     size_t chname_len;
1347     unsigned int is_add : 1;
1348 };
1349
1350 static void
1351 mod_chanmode_append(struct chanmode_buffer *buf, char ch, const char *arg)
1352 {
1353     size_t arg_len = strlen(arg);
1354     if (buf->modes_used + buf->args_used + buf->chname_len + arg_len > 450) {
1355         memcpy(buf->modes + buf->modes_used, buf->args, buf->args_used);
1356         buf->modes[buf->modes_used + buf->args_used] = '\0';
1357         irc_mode(buf->actor, buf->channel, buf->modes);
1358         buf->modes[0] = buf->is_add ? '+' : '-';
1359         buf->modes_used = 1;
1360         buf->args_used = 0;
1361     }
1362     buf->modes[buf->modes_used++] = ch;
1363     buf->args[buf->args_used++] = ' ';
1364     memcpy(buf->args + buf->args_used, arg, arg_len);
1365     buf->args_used += arg_len;
1366 }
1367
1368 void
1369 mod_chanmode_announce(struct userNode *who, struct chanNode *channel, struct mod_chanmode *change)
1370 {
1371     struct chanmode_buffer chbuf;
1372     char int_buff[32];
1373     unsigned int arg;
1374
1375     assert(change->argc <= change->alloc_argc);
1376     memset(&chbuf, 0, sizeof(chbuf));
1377     chbuf.channel = channel;
1378     chbuf.actor = who;
1379     chbuf.chname_len = strlen(channel->name);
1380
1381     /* First remove modes */
1382     chbuf.is_add = 0;
1383     if (change->modes_clear) {
1384         chbuf.modes[chbuf.modes_used++] = '-';
1385 #define DO_MODE_CHAR(BIT, CHAR) if (change->modes_clear & MODE_##BIT) chbuf.modes[chbuf.modes_used++] = CHAR;
1386         DO_MODE_CHAR(PRIVATE, 'p');
1387         DO_MODE_CHAR(SECRET, 's');
1388         DO_MODE_CHAR(MODERATED, 'm');
1389         DO_MODE_CHAR(TOPICLIMIT, 't');
1390         DO_MODE_CHAR(INVITEONLY, 'i');
1391         DO_MODE_CHAR(NOPRIVMSGS, 'n');
1392         DO_MODE_CHAR(LIMIT, 'l');
1393         DO_MODE_CHAR(DELAYJOINS, 'D');
1394         DO_MODE_CHAR(REGONLY, 'R');
1395         DO_MODE_CHAR(NOCOLORS, 'c');
1396         DO_MODE_CHAR(REGISTERED, 'r');
1397 #undef DO_MODE_CHAR
1398         if (change->modes_clear & channel->modes & MODE_KEY)
1399             mod_chanmode_append(&chbuf, 'k', channel->key);
1400     }
1401     for (arg = 0; arg < change->argc; ++arg) {
1402         if (!(change->args[arg].mode & MODE_REMOVE))
1403             continue;
1404         switch (change->args[arg].mode & ~MODE_REMOVE) {
1405         case MODE_BAN:
1406             mod_chanmode_append(&chbuf, 'b', change->args[arg].hostmask);
1407             break;
1408         default:
1409             if (change->args[arg].mode & MODE_CHANOP)
1410                 mod_chanmode_append(&chbuf, 'o', change->args[arg].member->user->nick);
1411             if (change->args[arg].mode & MODE_VOICE)
1412                 mod_chanmode_append(&chbuf, 'v', change->args[arg].member->user->nick);
1413             break;
1414         }
1415     }
1416
1417     /* Then set them */
1418     chbuf.is_add = 1;
1419     if (change->modes_set) {
1420         chbuf.modes[chbuf.modes_used++] = '+';
1421 #define DO_MODE_CHAR(BIT, CHAR) if (change->modes_set & MODE_##BIT) chbuf.modes[chbuf.modes_used++] = CHAR;
1422         DO_MODE_CHAR(PRIVATE, 'p');
1423         DO_MODE_CHAR(SECRET, 's');
1424         DO_MODE_CHAR(MODERATED, 'm');
1425         DO_MODE_CHAR(TOPICLIMIT, 't');
1426         DO_MODE_CHAR(INVITEONLY, 'i');
1427         DO_MODE_CHAR(NOPRIVMSGS, 'n');
1428         DO_MODE_CHAR(DELAYJOINS, 'D');
1429         DO_MODE_CHAR(REGONLY, 'R');
1430         DO_MODE_CHAR(NOCOLORS, 'c');
1431         DO_MODE_CHAR(REGISTERED, 'r');
1432 #undef DO_MODE_CHAR
1433         if (change->modes_set & MODE_KEY)
1434             mod_chanmode_append(&chbuf, 'k', change->new_key);
1435         if (change->modes_set & MODE_LIMIT)
1436         {
1437             sprintf(int_buff, "%d", change->new_limit);
1438             mod_chanmode_append(&chbuf, 'l', int_buff);
1439         }
1440     }
1441     for (arg = 0; arg < change->argc; ++arg) {
1442         if (change->args[arg].mode & MODE_REMOVE)
1443             continue;
1444         switch (change->args[arg].mode) {
1445         case MODE_BAN:
1446             mod_chanmode_append(&chbuf, 'b', change->args[arg].hostmask);
1447             break;
1448         default:
1449             if (change->args[arg].mode & MODE_CHANOP)
1450                 mod_chanmode_append(&chbuf, 'o', change->args[arg].member->user->nick);
1451             if (change->args[arg].mode & MODE_VOICE)
1452                 mod_chanmode_append(&chbuf, 'v', change->args[arg].member->user->nick);
1453             break;
1454         }
1455     }
1456
1457     /* Flush the buffer and apply changes locally */
1458     if (chbuf.modes_used > 0) {
1459         memcpy(chbuf.modes + chbuf.modes_used, chbuf.args, chbuf.args_used);
1460         chbuf.modes[chbuf.modes_used + chbuf.args_used] = '\0';
1461         irc_mode(chbuf.actor, chbuf.channel, chbuf.modes);
1462     }
1463     mod_chanmode_apply(who, channel, change);
1464 }
1465
1466 char *
1467 mod_chanmode_format(struct mod_chanmode *change, char *outbuff)
1468 {
1469     unsigned int used = 0;
1470     assert(change->argc <= change->alloc_argc);
1471     if (change->modes_clear) {
1472         outbuff[used++] = '-';
1473 #define DO_MODE_CHAR(BIT, CHAR) if (change->modes_clear & MODE_##BIT) outbuff[used++] = CHAR
1474         DO_MODE_CHAR(PRIVATE, 'p');
1475         DO_MODE_CHAR(SECRET, 's');
1476         DO_MODE_CHAR(MODERATED, 'm');
1477         DO_MODE_CHAR(TOPICLIMIT, 't');
1478         DO_MODE_CHAR(INVITEONLY, 'i');
1479         DO_MODE_CHAR(NOPRIVMSGS, 'n');
1480         DO_MODE_CHAR(LIMIT, 'l');
1481         DO_MODE_CHAR(KEY, 'k');
1482         DO_MODE_CHAR(DELAYJOINS, 'D');
1483         DO_MODE_CHAR(REGONLY, 'R');
1484         DO_MODE_CHAR(NOCOLORS, 'c');
1485         DO_MODE_CHAR(REGISTERED, 'r');
1486 #undef DO_MODE_CHAR
1487     }
1488     if (change->modes_set) {
1489         outbuff[used++] = '+';
1490 #define DO_MODE_CHAR(BIT, CHAR) if (change->modes_set & MODE_##BIT) outbuff[used++] = CHAR
1491         DO_MODE_CHAR(PRIVATE, 'p');
1492         DO_MODE_CHAR(SECRET, 's');
1493         DO_MODE_CHAR(MODERATED, 'm');
1494         DO_MODE_CHAR(TOPICLIMIT, 't');
1495         DO_MODE_CHAR(INVITEONLY, 'i');
1496         DO_MODE_CHAR(NOPRIVMSGS, 'n');
1497         DO_MODE_CHAR(DELAYJOINS, 'D');
1498         DO_MODE_CHAR(REGONLY, 'R');
1499         DO_MODE_CHAR(NOCOLORS, 'c');
1500         DO_MODE_CHAR(REGISTERED, 'r');
1501 #undef DO_MODE_CHAR
1502         switch (change->modes_set & (MODE_KEY|MODE_LIMIT)) {
1503         case MODE_KEY|MODE_LIMIT:
1504             used += sprintf(outbuff+used, "lk %d %s", change->new_limit, change->new_key);
1505             break;
1506         case MODE_KEY:
1507             used += sprintf(outbuff+used, "k %s", change->new_key);
1508             break;
1509         case MODE_LIMIT:
1510             used += sprintf(outbuff+used, "l %d", change->new_limit);
1511             break;
1512         }
1513     }
1514     outbuff[used] = 0;
1515     return outbuff;
1516 }