src/hash.c

   1 /* hash.c - IRC network state database
   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 "conf.h"
  22 #include "global.h"
  23 #include "hash.h"
  24 #include "log.h"
  25
  26 struct server *self;
  27 dict_t channels;
  28 dict_t clients;
  29 dict_t servers;
  30 unsigned int max_clients, invis_clients;
  31 time_t max_clients_time;
  32 struct userList curr_opers;
  33
  34 static void hash_cleanup(void);
  35
  36 void init_structs(void)
  37 {
  38     channels = dict_new();
  39     clients = dict_new();
  40     servers = dict_new();
  41     userList_init(&curr_opers);
  42     reg_exit_func(hash_cleanup);
  43 }
  44
  45 server_link_func_t *slf_list;
  46 unsigned int slf_size = 0, slf_used = 0;
  47
  48 void
  49 reg_server_link_func(server_link_func_t handler)
  50 {
  51     if (slf_used == slf_size) {
  52         if (slf_size) {
  53             slf_size <<= 1;
  54             slf_list = realloc(slf_list, slf_size*sizeof(server_link_func_t));
  55         } else {
  56             slf_size = 8;
  57             slf_list = malloc(slf_size*sizeof(server_link_func_t));
  58         }
  59     }
  60     slf_list[slf_used++] = handler;
  61 }
  62
  63 struct server*
  64 GetServerH(const char *name)
  65 {
  66     return dict_find(servers, name, NULL);
  67 }
  68
  69 new_user_func_t *nuf_list;
  70 unsigned int nuf_size = 0, nuf_used = 0;
  71
  72 void
  73 reg_new_user_func(new_user_func_t handler)
  74 {
  75     if (nuf_used == nuf_size) {
  76         if (nuf_size) {
  77             nuf_size <<= 1;
  78             nuf_list = realloc(nuf_list, nuf_size*sizeof(new_user_func_t));
  79         } else {
  80             nuf_size = 8;
  81             nuf_list = malloc(nuf_size*sizeof(new_user_func_t));
  82         }
  83     }
  84     nuf_list[nuf_used++] = handler;
  85 }
  86
  87 static nick_change_func_t *ncf2_list;
  88 static unsigned int ncf2_size = 0, ncf2_used = 0;
  89
  90 void
  91 reg_nick_change_func(nick_change_func_t handler)
  92 {
  93     if (ncf2_used == ncf2_size) {
  94         if (ncf2_size) {
  95             ncf2_size <<= 1;
  96             ncf2_list = realloc(ncf2_list, ncf2_size*sizeof(nick_change_func_t));
  97         } else {
  98             ncf2_size = 8;
  99             ncf2_list = malloc(ncf2_size*sizeof(nick_change_func_t));
 100         }
 101     }
 102     ncf2_list[ncf2_used++] = handler;
 103 }
 104
 105
 106 del_user_func_t *duf_list;
 107 unsigned int duf_size = 0, duf_used = 0;
 108
 109 void
 110 reg_del_user_func(del_user_func_t handler)
 111 {
 112     if (duf_used == duf_size) {
 113         if (duf_size) {
 114             duf_size <<= 1;
 115             duf_list = realloc(duf_list, duf_size*sizeof(del_user_func_t));
 116         } else {
 117             duf_size = 8;
 118             duf_list = malloc(duf_size*sizeof(del_user_func_t));
 119         }
 120     }
 121     duf_list[duf_used++] = handler;
 122 }
 123
 124 void
 125 unreg_del_user_func(del_user_func_t handler)
 126 {
 127     unsigned int i;
 128     for (i=0; i<duf_used; i++) {
 129         if (duf_list[i] == handler) break;
 130     }
 131     if (i == duf_used) return;
 132     memmove(duf_list+i, duf_list+i+1, (duf_used-i-1)*sizeof(duf_list[0]));
 133     duf_used--;
 134 }
 135
 136 /* reintroduces a user after it has been killed. */
 137 void
 138 ReintroduceUser(struct userNode *user)
 139 {
 140     struct mod_chanmode change;
 141     unsigned int n;
 142
 143     irc_user(user);
 144     mod_chanmode_init(&change);
 145     change.argc = 1;
 146     for (n = 0; n < user->channels.used; n++) {
 147         struct modeNode *mn = user->channels.list[n];
 148         irc_join(user, mn->channel);
 149         if (mn->modes) {
 150             change.args[0].mode = mn->modes;
 151             change.args[0].member = mn;
 152             mod_chanmode_announce(user, mn->channel, &change);
 153         }
 154     }
 155 }
 156
 157 void
 158 NickChange(struct userNode* user, const char *new_nick, int no_announce)
 159 {
 160     char *old_nick;
 161     unsigned int nn;
 162
 163     /* don't do anything if there's no change */
 164     old_nick = user->nick;
 165     if (!strncmp(new_nick, old_nick, NICKLEN))
 166         return;
 167
 168     /* remove old entry from clients dictionary */
 169     dict_remove(clients, old_nick);
 170 #if !defined(WITH_PROTOCOL_P10)
 171     /* Remove from uplink's clients dict */
 172     dict_remove(user->uplink->users, old_nick);
 173 #endif
 174     /* and reinsert */
 175     user->nick = strdup(new_nick);
 176     dict_insert(clients, user->nick, user);
 177 #if !defined(WITH_PROTOCOL_P10)
 178     dict_insert(user->uplink->users, user->nick, user);
 179 #endif
 180
 181     /* Make callbacks for nick changes.  Do this with new nick in
 182      * place because that is slightly more useful.
 183      */
 184     for (nn=0; nn<ncf2_used; nn++)
 185         ncf2_list[nn](user, old_nick);
 186     user->timestamp = now;
 187     if (IsLocal(user) && !no_announce)
 188         irc_nick(user, old_nick);
 189     free(old_nick);
 190 }
 191
 192 struct userNode *
 193 GetUserH(const char *nick)
 194 {
 195     return dict_find(clients, nick, NULL);
 196 }
 197
 198 static account_func_t account_func;
 199
 200 void
 201 reg_account_func(account_func_t handler)
 202 {
 203     if (account_func) {
 204         log_module(MAIN_LOG, LOG_WARNING, "Reregistering ACCOUNT handler.");
 205     }
 206     account_func = handler;
 207 }
 208
 209 void
 210 call_account_func(struct userNode *user, const char *stamp)
 211 {
 212     /* We've received an account stamp for a user; notify
 213        NickServ, which registers the sole account_func
 214        right now.
 215
 216        P10 Protocol violation if (user->modes & FLAGS_STAMPED) here.
 217     */
 218     if (account_func)
 219         account_func(user, stamp);
 220
 221 #ifdef WITH_PROTOCOL_P10
 222     /* Mark the user so we don't stamp it again. */
 223     user->modes |= FLAGS_STAMPED;
 224 #endif
 225 }
 226
 227 void
 228 StampUser(struct userNode *user, const char *stamp)
 229 {
 230 #ifdef WITH_PROTOCOL_P10
 231     /* The P10 protocol says we can't stamp users who already
 232        have a stamp. */
 233     if (IsStamped(user))
 234         return;
 235 #endif
 236
 237     irc_account(user, stamp);
 238     user->modes |= FLAGS_STAMPED;
 239 }
 240
 241 static new_channel_func_t *ncf_list;
 242 static unsigned int ncf_size = 0, ncf_used = 0;
 243
 244 void
 245 reg_new_channel_func(new_channel_func_t handler)
 246 {
 247     if (ncf_used == ncf_size) {
 248         if (ncf_size) {
 249             ncf_size <<= 1;
 250             ncf_list = realloc(ncf_list, ncf_size*sizeof(ncf_list[0]));
 251         } else {
 252             ncf_size = 8;
 253             ncf_list = malloc(ncf_size*sizeof(ncf_list[0]));
 254         }
 255     }
 256     ncf_list[ncf_used++] = handler;
 257 }
 258
 259 static join_func_t *jf_list;
 260 static unsigned int jf_size = 0, jf_used = 0;
 261
 262 void
 263 reg_join_func(join_func_t handler)
 264 {
 265     if (jf_used == jf_size) {
 266         if (jf_size) {
 267             jf_size <<= 1;
 268             jf_list = realloc(jf_list, jf_size*sizeof(join_func_t));
 269         } else {
 270             jf_size = 8;
 271             jf_list = malloc(jf_size*sizeof(join_func_t));
 272         }
 273     }
 274     jf_list[jf_used++] = handler;
 275 }
 276
 277 int rel_age;
 278
 279 static void
 280 wipeout_channel(struct chanNode *cNode, time_t new_time, char **modes, unsigned int modec) {
 281     unsigned int orig_limit;
 282     chan_mode_t orig_modes;
 283     char orig_key[KEYLEN+1];
 284     unsigned int nn, argc;
 285
 286     /* nuke old topic */
 287     cNode->topic[0] = '\0';
 288     cNode->topic_nick[0] = '\0';
 289     cNode->topic_time = 0;
 290
 291     /* remember the old modes, and update them with the new */
 292     orig_modes = cNode->modes;
 293     orig_limit = cNode->limit;
 294     strcpy(orig_key, cNode->key);
 295     cNode->modes = 0;
 296     mod_chanmode(NULL, cNode, modes, modec, 0);
 297     cNode->timestamp = new_time;
 298
 299     /* remove our old ban list, replace it with the new one */
 300     for (nn=0; nn<cNode->banlist.used; nn++)
 301         free(cNode->banlist.list[nn]);
 302     cNode->banlist.used = 0;
 303
 304     /* deop anybody in the channel now, but count services to reop */
 305     for (nn=argc=0; nn<cNode->members.used; nn++) {
 306         struct modeNode *mn = cNode->members.list[nn];
 307         if ((mn->modes & MODE_CHANOP) && IsService(mn->user) && IsLocal(mn->user))
 308             argc++;
 309     }
 310     if (argc) {
 311         extern struct userNode *opserv;
 312         struct mod_chanmode *change;
 313
 314         change = mod_chanmode_alloc(argc);
 315         change->modes_clear = 0;
 316         change->modes_set = orig_modes;
 317         change->new_limit = orig_limit;
 318         strcpy(change->new_key, orig_key);
 319         for (nn = argc = 0; nn < cNode->members.used; ++nn) {
 320             struct modeNode *mn = cNode->members.list[nn];
 321             if ((mn->modes & MODE_CHANOP) && IsService(mn->user) && IsLocal(mn->user)) {
 322                 change->args[argc].mode = MODE_CHANOP;
 323                 change->args[argc].member = mn;
 324                 argc++;
 325             }
 326         }
 327         assert(argc == change->argc);
 328         if (change->argc > 0)
 329             mod_chanmode_announce(change->args[0].member->user, cNode, change);
 330         else
 331             mod_chanmode_announce(opserv, cNode, change);
 332         mod_chanmode_free(change);
 333     }
 334 }
 335
 336 struct chanNode *
 337 AddChannel(const char *name, time_t time_, const char *modes, char *banlist)
 338 {
 339     struct chanNode *cNode;
 340     char new_modes[MAXLEN], *argv[MAXNUMPARAMS];
 341     unsigned int nn;
 342
 343     if (!IsChannelName(name)) {
 344         log_module(MAIN_LOG, LOG_ERROR, "Somebody asked to add channel '%s', which isn't a channel name!", name);
 345         return NULL;
 346     }
 347     if (!modes)
 348         modes = "";
 349
 350     safestrncpy(new_modes, modes, sizeof(new_modes));
 351     nn = split_line(new_modes, 0, ArrayLength(argv), argv);
 352     if (!(cNode = GetChannel(name))) {
 353         cNode = calloc(1, sizeof(*cNode) + strlen(name));
 354         strcpy(cNode->name, name);
 355         banList_init(&cNode->banlist);
 356         modeList_init(&cNode->members);
 357         mod_chanmode(NULL, cNode, argv, nn, 0);
 358         dict_insert(channels, cNode->name, cNode);
 359         cNode->timestamp = time_;
 360         rel_age = 1;
 361     } else if (cNode->timestamp > time_) {
 362         wipeout_channel(cNode, time_, argv, nn);
 363         rel_age = 1;
 364     } else if (cNode->timestamp == time_) {
 365         mod_chanmode(NULL, cNode, argv, nn, 0);
 366         rel_age = 0;
 367     } else {
 368         rel_age = -1;
 369     }
 370
 371     /* rel_age is the relative ages of our channel data versus what is
 372      * in a BURST command.  1 means ours is younger, 0 means both are
 373      * the same age, -1 means ours is older. */
 374
 375     /* if it's a new or updated channel, make callbacks */
 376     if (rel_age > 0)
 377         for (nn=0; nn<ncf_used; nn++)
 378             ncf_list[nn](cNode);
 379
 380     /* go through list of bans and add each one */
 381     if (banlist && (rel_age >= 0)) {
 382         for (nn=0; banlist[nn];) {
 383             char *ban = banlist + nn;
 384             struct banNode *bn;
 385             while (banlist[nn] != ' ' && banlist[nn])
 386                 nn++;
 387             while (banlist[nn] == ' ')
 388                 banlist[nn++] = 0;
 389             bn = calloc(1, sizeof(*bn));
 390             safestrncpy(bn->ban, ban, sizeof(bn->ban));
 391             safestrncpy(bn->who, "<unknown>", sizeof(bn->who));
 392             bn->set = now;
 393             banList_append(&cNode->banlist, bn);
 394         }
 395     }
 396
 397     return cNode;
 398 }
 399
 400 static del_channel_func_t *dcf_list;
 401 static unsigned int dcf_size = 0, dcf_used = 0;
 402
 403 void
 404 reg_del_channel_func(del_channel_func_t handler)
 405 {
 406     if (dcf_used == dcf_size) {
 407         if (dcf_size) {
 408             dcf_size <<= 1;
 409             dcf_list = realloc(dcf_list, dcf_size*sizeof(dcf_list[0]));
 410         } else {
 411             dcf_size = 8;
 412             dcf_list = malloc(dcf_size*sizeof(dcf_list[0]));
 413         }
 414     }
 415     dcf_list[dcf_used++] = handler;
 416 }
 417
 418 static void
 419 DelChannel(struct chanNode *channel)
 420 {
 421     unsigned int n;
 422
 423     dict_remove(channels, channel->name);
 424
 425     if (channel->members.used || channel->locks) {
 426         log_module(MAIN_LOG, LOG_ERROR, "Warning: deleting channel %s with %d users and %d locks remaining.", channel->name, channel->members.used, channel->locks);
 427     }
 428
 429     /* go through all channel members and delete them from the channel */
 430     for (n=channel->members.used; n>0; )
 431         DelChannelUser(channel->members.list[--n]->user, channel, false, 1);
 432
 433     /* delete all channel bans */
 434     for (n=channel->banlist.used; n>0; )
 435         free(channel->banlist.list[--n]);
 436     channel->banlist.used = 0;
 437
 438     for (n=0; n<dcf_used; n++)
 439         dcf_list[n](channel);
 440
 441     modeList_clean(&channel->members);
 442     banList_clean(&channel->banlist);
 443     free(channel);
 444 }
 445
 446 struct modeNode *
 447 AddChannelUser(struct userNode *user, struct chanNode* channel)
 448 {
 449         struct modeNode *mNode;
 450         unsigned int n;
 451
 452         mNode = GetUserMode(channel, user);
 453         if (mNode)
 454             return mNode;
 455
 456         mNode = malloc(sizeof(*mNode));
 457
 458         /* set up modeNode */
 459         mNode->channel = channel;
 460         mNode->user = user;
 461         mNode->modes = 0;
 462         mNode->idle_since = now;
 463
 464         /* Add modeNode to channel and to user.
 465          * We have to do this before calling join funcs in case the
 466          * modeNode is manipulated (e.g. chanserv ops the user).
 467          */
 468         modeList_append(&channel->members, mNode);
 469         modeList_append(&user->channels, mNode);
 470
 471         if (channel->members.used == 1)
 472             mNode->modes |= MODE_CHANOP;
 473
 474         for (n=0; n<jf_used; n++) {
 475             /* Callbacks return true if they kick or kill the user,
 476              * and we can continue without removing mNode. */
 477             if (jf_list[n](mNode))
 478                 return NULL;
 479         }
 480
 481         if (IsLocal(user))
 482             irc_join(user, channel);
 483
 484         return mNode;
 485 }
 486
 487 static part_func_t *pf_list;
 488 static unsigned int pf_size = 0, pf_used = 0;
 489
 490 void
 491 reg_part_func(part_func_t handler)
 492 {
 493     if (pf_used == pf_size) {
 494         if (pf_size) {
 495             pf_size <<= 1;
 496             pf_list = realloc(pf_list, pf_size*sizeof(part_func_t));
 497         } else {
 498             pf_size = 8;
 499             pf_list = malloc(pf_size*sizeof(part_func_t));
 500         }
 501     }
 502     pf_list[pf_used++] = handler;
 503 }
 504
 505 void
 506 unreg_part_func(part_func_t handler)
 507 {
 508     unsigned int i;
 509     for (i=0; i<pf_used; i++)
 510         if (pf_list[i] == handler)
 511             break;
 512     if (i == pf_used)
 513         return;
 514     memmove(pf_list+i, pf_list+i+1, (pf_used-i-1)*sizeof(pf_list[0]));
 515     pf_used--;
 516 }
 517
 518 void
 519 LockChannel(struct chanNode* channel)
 520 {
 521     channel->locks++;
 522 }
 523
 524 void
 525 UnlockChannel(struct chanNode *channel)
 526 {
 527     assert(channel->locks > 0);
 528     if (!--channel->locks && !channel->members.used)
 529         DelChannel(channel);
 530 }
 531
 532 void
 533 DelChannelUser(struct userNode* user, struct chanNode* channel, const char *reason, int deleting)
 534 {
 535     struct modeNode* mNode;
 536     unsigned int n;
 537
 538     if (reason) {
 539         irc_part(user, channel, reason);
 540     }
 541
 542     mNode = GetUserMode(channel, user);
 543
 544     /* Sometimes we get a PART when the user has been KICKed.
 545      * In this case, we get no usermode, and should not try to free it.
 546      */
 547     if (!mNode)
 548         return;
 549
 550     /* remove modeNode from channel and user */
 551     modeList_remove(&channel->members, mNode);
 552     modeList_remove(&user->channels, mNode);
 553     free(mNode);
 554
 555     for (n=0; n<pf_used; n++)
 556         pf_list[n](user, channel, reason);
 557
 558     if (!deleting && !channel->members.used && !channel->locks)
 559         DelChannel(channel);
 560 }
 561
 562 void
 563 KickChannelUser(struct userNode* target, struct chanNode* channel, struct userNode *kicker, const char *why)
 564 {
 565     if (!target || !channel || IsService(target) || !GetUserMode(channel, target))
 566         return;
 567     /* don't remove them from the channel, since the server will send a PART */
 568     irc_kick(kicker, target, channel, why);
 569
 570     if (IsLocal(target))
 571     {
 572         /* NULL reason because we don't want a PART message to be
 573            sent by DelChannelUser. */
 574         DelChannelUser(target, channel, NULL, 0);
 575     }
 576 }
 577
 578 static kick_func_t *kf_list;
 579 static unsigned int kf_size = 0, kf_used = 0;
 580
 581 void
 582 reg_kick_func(kick_func_t handler)
 583 {
 584     if (kf_used == kf_size) {
 585         if (kf_size) {
 586             kf_size <<= 1;
 587             kf_list = realloc(kf_list, kf_size*sizeof(kick_func_t));
 588         } else {
 589             kf_size = 8;
 590             kf_list = malloc(kf_size*sizeof(kick_func_t));
 591         }
 592     }
 593     kf_list[kf_used++] = handler;
 594 }
 595
 596 void
 597 ChannelUserKicked(struct userNode* kicker, struct userNode* victim, struct chanNode* channel)
 598 {
 599     unsigned int n;
 600     struct modeNode *mn;
 601
 602     if (!victim || !channel || IsService(victim) || !GetUserMode(channel, victim))
 603         return;
 604
 605     /* Update the kicker's idle time (kicker may be null if it was a server) */
 606     if (kicker && (mn = GetUserMode(channel, kicker)))
 607         mn->idle_since = now;
 608
 609     for (n=0; n<kf_used; n++)
 610         kf_list[n](kicker, victim, channel);
 611
 612     DelChannelUser(victim, channel, 0, 0);
 613
 614     if (IsLocal(victim))
 615         irc_part(victim, channel, NULL);
 616 }
 617
 618 int ChannelBanExists(struct chanNode *channel, const char *ban)
 619 {
 620     unsigned int n;
 621
 622     for (n = 0; n < channel->banlist.used; n++)
 623         if (match_ircglobs(channel->banlist.list[n]->ban, ban))
 624             return 1;
 625     return 0;
 626 }
 627
 628 static topic_func_t *tf_list;
 629 static unsigned int tf_size = 0, tf_used = 0;
 630
 631 void
 632 reg_topic_func(topic_func_t handler)
 633 {
 634     if (tf_used == tf_size) {
 635         if (tf_size) {
 636             tf_size <<= 1;
 637             tf_list = realloc(tf_list, tf_size*sizeof(topic_func_t));
 638         } else {
 639             tf_size = 8;
 640             tf_list = malloc(tf_size*sizeof(topic_func_t));
 641         }
 642     }
 643     tf_list[tf_used++] = handler;
 644 }
 645
 646 void
 647 SetChannelTopic(struct chanNode *channel, struct userNode *user, const char *topic, int announce)
 648 {
 649     unsigned int n;
 650     struct modeNode *mn;
 651     char old_topic[TOPICLEN+1];
 652
 653     safestrncpy(old_topic, channel->topic, sizeof(old_topic));
 654     safestrncpy(channel->topic, topic, sizeof(channel->topic));
 655     channel->topic_time = now;
 656
 657     if (user) {
 658         safestrncpy(channel->topic_nick, user->nick, sizeof(channel->topic_nick));
 659
 660         /* Update the setter's idle time */
 661         if ((mn = GetUserMode(channel, user)))
 662             mn->idle_since = now;
 663     }
 664
 665     if (announce) {
 666         /* We don't really care if a local user messes with the topic,
 667          * so don't call the tf_list functions. */
 668         irc_topic(user, channel, topic);
 669     } else {
 670         for (n=0; n<tf_used; n++)
 671             if (tf_list[n](user, channel, old_topic))
 672                 break;
 673     }
 674 }
 675
 676 struct chanNode *
 677 GetChannel(const char *name)
 678 {
 679     return dict_find(channels, name, NULL);
 680 }
 681
 682 struct modeNode *
 683 GetUserMode(struct chanNode *channel, struct userNode *user)
 684 {
 685     unsigned int n;
 686     struct modeNode *mn = NULL;
 687     if (channel->members.used < user->channels.used) {
 688         for (n=0; n<channel->members.used; n++) {
 689             if (user == channel->members.list[n]->user) {
 690                 mn = channel->members.list[n];
 691                 break;
 692             }
 693         }
 694     } else {
 695         for (n=0; n<user->channels.used; n++) {
 696             if (channel == user->channels.list[n]->channel) {
 697                 mn = user->channels.list[n];
 698                 break;
 699             }
 700         }
 701     }
 702     return mn;
 703 }
 704
 705 DEFINE_LIST(userList, struct userNode*)
 706 DEFINE_LIST(modeList, struct modeNode*)
 707 DEFINE_LIST(banList, struct banNode*)
 708 DEFINE_LIST(channelList, struct chanNode*)
 709 DEFINE_LIST(serverList, struct server*)
 710
 711 static void
 712 hash_cleanup(void)
 713 {
 714     DelServer(self, 0, NULL);
 715     dict_delete(channels);
 716     dict_delete(clients);
 717     dict_delete(servers);
 718     userList_clean(&curr_opers);
 719
 720     free(slf_list);
 721     free(nuf_list);
 722     free(ncf2_list);
 723     free(duf_list);
 724     free(ncf_list);
 725     free(jf_list);
 726     free(dcf_list);
 727     free(pf_list);
 728     free(kf_list);
 729     free(tf_list);
 730 }