eff19d1e28a70f4c9635221403984863a759e198
[srvx.git] / 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].u.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 void
242 assign_fakehost(struct userNode *user, const char *host, int announce)
243 {
244     safestrncpy(user->fakehost, host, sizeof(user->fakehost));
245     if (announce)
246         irc_fakehost(user, host);
247 }
248
249 static new_channel_func_t *ncf_list;
250 static unsigned int ncf_size = 0, ncf_used = 0;
251
252 void
253 reg_new_channel_func(new_channel_func_t handler)
254 {
255     if (ncf_used == ncf_size) {
256         if (ncf_size) {
257             ncf_size <<= 1;
258             ncf_list = realloc(ncf_list, ncf_size*sizeof(ncf_list[0]));
259         } else {
260             ncf_size = 8;
261             ncf_list = malloc(ncf_size*sizeof(ncf_list[0]));
262         }
263     }
264     ncf_list[ncf_used++] = handler;
265 }
266
267 static join_func_t *jf_list;
268 static unsigned int jf_size = 0, jf_used = 0;
269
270 void
271 reg_join_func(join_func_t handler)
272 {
273     if (jf_used == jf_size) {
274         if (jf_size) {
275             jf_size <<= 1;
276             jf_list = realloc(jf_list, jf_size*sizeof(join_func_t));
277         } else {
278             jf_size = 8;
279             jf_list = malloc(jf_size*sizeof(join_func_t));
280         }
281     }
282     jf_list[jf_used++] = handler;
283 }
284
285 int rel_age;
286
287 static void
288 wipeout_channel(struct chanNode *cNode, time_t new_time, char **modes, unsigned int modec) {
289     unsigned int orig_limit;
290     chan_mode_t orig_modes;
291     char orig_key[KEYLEN+1];
292     char orig_apass[KEYLEN+1];
293     char orig_upass[KEYLEN+1];
294     unsigned int nn, argc;
295
296     /* nuke old topic */
297     cNode->topic[0] = '\0';
298     cNode->topic_nick[0] = '\0';
299     cNode->topic_time = 0;
300
301     /* remember the old modes, and update them with the new */
302     orig_modes = cNode->modes;
303     orig_limit = cNode->limit;
304     strcpy(orig_key, cNode->key);
305     strcpy(orig_upass, cNode->upass);
306     strcpy(orig_apass, cNode->apass);
307     cNode->modes = 0;
308     mod_chanmode(NULL, cNode, modes, modec, 0);
309     cNode->timestamp = new_time;
310
311     /* remove our old ban list, replace it with the new one */
312     for (nn=0; nn<cNode->banlist.used; nn++)
313         free(cNode->banlist.list[nn]);
314     cNode->banlist.used = 0;
315
316     /* deop anybody in the channel now, but count services to reop */
317     for (nn=argc=0; nn<cNode->members.used; nn++) {
318         struct modeNode *mn = cNode->members.list[nn];
319         if ((mn->modes & MODE_CHANOP) && IsService(mn->user) && IsLocal(mn->user))
320             argc++;
321     }
322     if (argc) {
323         struct mod_chanmode *change;
324
325         change = mod_chanmode_alloc(argc);
326         change->modes_clear = 0;
327         change->modes_set = orig_modes;
328         change->new_limit = orig_limit;
329         strcpy(change->new_key, orig_key);
330         strcpy(change->new_upass, orig_upass);
331         strcpy(change->new_apass, orig_apass);
332         for (nn = argc = 0; nn < cNode->members.used; ++nn) {
333             struct modeNode *mn = cNode->members.list[nn];
334             if ((mn->modes & MODE_CHANOP) && IsService(mn->user) && IsLocal(mn->user)) {
335                 change->args[argc].mode = MODE_CHANOP;
336                 change->args[argc].u.member = mn;
337                 argc++;
338             }
339         }
340         assert(argc == change->argc);
341         change->args[0].u.member->modes &= ~MODE_CHANOP;
342         mod_chanmode_announce(change->args[0].u.member->user, cNode, change);
343         mod_chanmode_free(change);
344     }
345 }
346
347 struct chanNode *
348 AddChannel(const char *name, time_t time_, const char *modes, char *banlist)
349 {
350     struct chanNode *cNode;
351     char new_modes[MAXLEN], *argv[MAXNUMPARAMS];
352     unsigned int nn;
353
354     if (!IsChannelName(name)) {
355         log_module(MAIN_LOG, LOG_ERROR, "Somebody asked to add channel '%s', which isn't a channel name!", name);
356         return NULL;
357     }
358     if (!modes)
359         modes = "";
360
361     safestrncpy(new_modes, modes, sizeof(new_modes));
362     nn = split_line(new_modes, 0, ArrayLength(argv), argv);
363     if (!(cNode = GetChannel(name))) {
364         cNode = calloc(1, sizeof(*cNode) + strlen(name));
365         strcpy(cNode->name, name);
366         banList_init(&cNode->banlist);
367         modeList_init(&cNode->members);
368         mod_chanmode(NULL, cNode, argv, nn, 0);
369         dict_insert(channels, cNode->name, cNode);
370         cNode->timestamp = time_;
371         rel_age = 1;
372     } else if (cNode->timestamp > time_) {
373         wipeout_channel(cNode, time_, argv, nn);
374         rel_age = 1;
375     } else if (cNode->timestamp == time_) {
376         mod_chanmode(NULL, cNode, argv, nn, 0);
377         rel_age = 0;
378     } else {
379         rel_age = -1;
380     }
381
382     /* rel_age is the relative ages of our channel data versus what is
383      * in a BURST command.  1 means ours is younger, 0 means both are
384      * the same age, -1 means ours is older. */
385
386     /* if it's a new or updated channel, make callbacks */
387     if (rel_age > 0)
388         for (nn=0; nn<ncf_used; nn++)
389             ncf_list[nn](cNode);
390
391     /* go through list of bans and add each one */
392     if (banlist && (rel_age >= 0)) {
393         for (nn=0; banlist[nn];) {
394             char *ban = banlist + nn;
395             struct banNode *bn;
396             while (banlist[nn] != ' ' && banlist[nn])
397                 nn++;
398             while (banlist[nn] == ' ')
399                 banlist[nn++] = 0;
400             bn = calloc(1, sizeof(*bn));
401             safestrncpy(bn->ban, ban, sizeof(bn->ban));
402             safestrncpy(bn->who, "<unknown>", sizeof(bn->who));
403             bn->set = now;
404             banList_append(&cNode->banlist, bn);
405         }
406     }
407
408     return cNode;
409 }
410
411 static del_channel_func_t *dcf_list;
412 static unsigned int dcf_size = 0, dcf_used = 0;
413
414 void
415 reg_del_channel_func(del_channel_func_t handler)
416 {
417     if (dcf_used == dcf_size) {
418         if (dcf_size) {
419             dcf_size <<= 1;
420             dcf_list = realloc(dcf_list, dcf_size*sizeof(dcf_list[0]));
421         } else {
422             dcf_size = 8;
423             dcf_list = malloc(dcf_size*sizeof(dcf_list[0]));
424         }
425     }
426     dcf_list[dcf_used++] = handler;
427 }
428
429 static void
430 DelChannel(struct chanNode *channel)
431 {
432     unsigned int n;
433
434     verify(channel);
435     dict_remove(channels, channel->name);
436
437     if (channel->members.used || channel->locks) {
438         log_module(MAIN_LOG, LOG_ERROR, "Warning: deleting channel %s with %d users and %d locks remaining.", channel->name, channel->members.used, channel->locks);
439     }
440
441     /* go through all channel members and delete them from the channel */
442     for (n=channel->members.used; n>0; )
443         DelChannelUser(channel->members.list[--n]->user, channel, false, 1);
444
445     /* delete all channel bans */
446     for (n=channel->banlist.used; n>0; )
447         free(channel->banlist.list[--n]);
448     channel->banlist.used = 0;
449
450     for (n=0; n<dcf_used; n++)
451         dcf_list[n](channel);
452
453     modeList_clean(&channel->members);
454     banList_clean(&channel->banlist);
455     free(channel);
456 }
457
458 struct modeNode *
459 AddChannelUser(struct userNode *user, struct chanNode* channel)
460 {
461         struct modeNode *mNode;
462         unsigned int n;
463
464         mNode = GetUserMode(channel, user);
465         if (mNode)
466             return mNode;
467
468         mNode = malloc(sizeof(*mNode));
469
470         /* set up modeNode */
471         mNode->channel = channel;
472         mNode->user = user;
473         mNode->modes = 0;
474         mNode->idle_since = now;
475
476         /* Add modeNode to channel and to user.
477          * We have to do this before calling join funcs in case the
478          * modeNode is manipulated (e.g. chanserv ops the user).
479          */
480         modeList_append(&channel->members, mNode);
481         modeList_append(&user->channels, mNode);
482
483         if (channel->members.used == 1
484             && !(channel->modes & MODE_REGISTERED)
485             && !(channel->modes & MODE_APASS))
486             mNode->modes |= MODE_CHANOP;
487
488         for (n=0; n<jf_used; n++) {
489             /* Callbacks return true if they kick or kill the user,
490              * and we can continue without removing mNode. */
491             if (jf_list[n](mNode))
492                 return NULL;
493         }
494
495         if (IsLocal(user))
496             irc_join(user, channel);
497
498         return mNode;
499 }
500
501 static part_func_t *pf_list;
502 static unsigned int pf_size = 0, pf_used = 0;
503
504 void
505 reg_part_func(part_func_t handler)
506 {
507     if (pf_used == pf_size) {
508         if (pf_size) {
509             pf_size <<= 1;
510             pf_list = realloc(pf_list, pf_size*sizeof(part_func_t));
511         } else {
512             pf_size = 8;
513             pf_list = malloc(pf_size*sizeof(part_func_t));
514         }
515     }
516     pf_list[pf_used++] = handler;
517 }
518
519 void
520 unreg_part_func(part_func_t handler)
521 {
522     unsigned int i;
523     for (i=0; i<pf_used; i++)
524         if (pf_list[i] == handler)
525             break;
526     if (i == pf_used)
527         return;
528     memmove(pf_list+i, pf_list+i+1, (pf_used-i-1)*sizeof(pf_list[0]));
529     pf_used--;
530 }
531
532 void
533 LockChannel(struct chanNode* channel)
534 {
535     channel->locks++;
536 }
537
538 void
539 UnlockChannel(struct chanNode *channel)
540 {
541     assert(channel->locks > 0);
542     if (!--channel->locks && !channel->members.used)
543         DelChannel(channel);
544 }
545
546 void
547 DelChannelUser(struct userNode* user, struct chanNode* channel, const char *reason, int deleting)
548 {
549     struct modeNode* mNode;
550     unsigned int n;
551
552     if (reason)
553         irc_part(user, channel, reason);
554
555     mNode = GetUserMode(channel, user);
556
557     /* Sometimes we get a PART when the user has been KICKed.
558      * In this case, we get no usermode, and should not try to free it.
559      */
560     if (!mNode)
561         return;
562
563     /* remove modeNode from channel and user */
564     modeList_remove(&channel->members, mNode);
565     modeList_remove(&user->channels, mNode);
566
567     /* make callbacks */
568     for (n=0; n<pf_used; n++)
569         pf_list[n](mNode, reason);
570
571     /* free memory */
572     free(mNode);
573
574     /* A single check for APASS only should be enough here */
575     if (!deleting && !channel->members.used && !channel->locks 
576         && !(channel->modes & MODE_REGISTERED) && !(channel->modes & MODE_APASS))
577         DelChannel(channel);
578 }
579
580 void
581 KickChannelUser(struct userNode* target, struct chanNode* channel, struct userNode *kicker, const char *why)
582 {
583     if (!target || !channel || IsService(target) || !GetUserMode(channel, target))
584         return;
585     /* don't remove them from the channel, since the server will send a PART */
586     irc_kick(kicker, target, channel, why);
587
588     if (IsLocal(target))
589     {
590         /* NULL reason because we don't want a PART message to be
591            sent by DelChannelUser. */
592         DelChannelUser(target, channel, NULL, 0);
593     }
594 }
595
596 static kick_func_t *kf_list;
597 static unsigned int kf_size = 0, kf_used = 0;
598
599 void
600 reg_kick_func(kick_func_t handler)
601 {
602     if (kf_used == kf_size) {
603         if (kf_size) {
604             kf_size <<= 1;
605             kf_list = realloc(kf_list, kf_size*sizeof(kick_func_t));
606         } else {
607             kf_size = 8;
608             kf_list = malloc(kf_size*sizeof(kick_func_t));
609         }
610     }
611     kf_list[kf_used++] = handler;
612 }
613
614 void
615 ChannelUserKicked(struct userNode* kicker, struct userNode* victim, struct chanNode* channel)
616 {
617     unsigned int n;
618     struct modeNode *mn;
619
620     if (!victim || !channel || IsService(victim) || !GetUserMode(channel, victim))
621         return;
622
623     /* Update the kicker's idle time (kicker may be null if it was a server) */
624     if (kicker && (mn = GetUserMode(channel, kicker)))
625         mn->idle_since = now;
626
627     for (n=0; n<kf_used; n++)
628         kf_list[n](kicker, victim, channel);
629
630     DelChannelUser(victim, channel, 0, 0);
631
632     if (IsLocal(victim))
633         irc_part(victim, channel, NULL);
634 }
635
636 int ChannelBanExists(struct chanNode *channel, const char *ban)
637 {
638     unsigned int n;
639
640     for (n = 0; n < channel->banlist.used; n++)
641         if (match_ircglobs(channel->banlist.list[n]->ban, ban))
642             return 1;
643     return 0;
644 }
645
646 static topic_func_t *tf_list;
647 static unsigned int tf_size = 0, tf_used = 0;
648
649 void
650 reg_topic_func(topic_func_t handler)
651 {
652     if (tf_used == tf_size) {
653         if (tf_size) {
654             tf_size <<= 1;
655             tf_list = realloc(tf_list, tf_size*sizeof(topic_func_t));
656         } else {
657             tf_size = 8;
658             tf_list = malloc(tf_size*sizeof(topic_func_t));
659         }
660     }
661     tf_list[tf_used++] = handler;
662 }
663
664 void
665 SetChannelTopic(struct chanNode *channel, struct userNode *user, const char *topic, int announce)
666 {
667     unsigned int n;
668     struct modeNode *mn;
669     char old_topic[TOPICLEN+1];
670
671     safestrncpy(old_topic, channel->topic, sizeof(old_topic));
672     safestrncpy(channel->topic, topic, sizeof(channel->topic));
673     channel->topic_time = now;
674
675     if (user) {
676         safestrncpy(channel->topic_nick, user->nick, sizeof(channel->topic_nick));
677
678         /* Update the setter's idle time */
679         if ((mn = GetUserMode(channel, user)))
680             mn->idle_since = now;
681     }
682
683     if (announce) {
684         /* We don't really care if a local user messes with the topic,
685          * so don't call the tf_list functions. */
686         irc_topic(user, channel, topic);
687     } else {
688         for (n=0; n<tf_used; n++)
689             if (tf_list[n](user, channel, old_topic))
690                 break;
691     }
692 }
693
694 struct chanNode *
695 GetChannel(const char *name)
696 {
697     return dict_find(channels, name, NULL);
698 }
699
700 struct modeNode *
701 GetUserMode(struct chanNode *channel, struct userNode *user)
702 {
703     unsigned int n;
704     struct modeNode *mn = NULL;
705
706     verify(channel);
707     verify(channel->members.list);
708     verify(user);
709     verify(user->channels.list);
710     if (channel->members.used < user->channels.used) {
711         for (n=0; n<channel->members.used; n++) {
712             verify(channel->members.list[n]);
713             if (user == channel->members.list[n]->user) {
714                 mn = channel->members.list[n];
715                 break;
716             }
717         }
718     } else {
719         for (n=0; n<user->channels.used; n++) {
720             verify(user->channels.list[n]);
721             if (channel == user->channels.list[n]->channel) {
722                 mn = user->channels.list[n];
723                 break;
724             }
725         }
726     }
727     return mn;
728 }
729
730 DEFINE_LIST(userList, struct userNode*)
731 DEFINE_LIST(modeList, struct modeNode*)
732 DEFINE_LIST(banList, struct banNode*)
733 DEFINE_LIST(channelList, struct chanNode*)
734 DEFINE_LIST(serverList, struct server*)
735
736 static void
737 hash_cleanup(void)
738 {
739     dict_iterator_t it, next;
740
741     DelServer(self, 0, NULL);
742     for (it = dict_first(channels); it; it = next) {
743         next = iter_next(it);
744         DelChannel(iter_data(it));
745     }
746     dict_delete(channels);
747     dict_delete(clients);
748     dict_delete(servers);
749     userList_clean(&curr_opers);
750
751     free(slf_list);
752     free(nuf_list);
753     free(ncf2_list);
754     free(duf_list);
755     free(ncf_list);
756     free(jf_list);
757     free(dcf_list);
758     free(pf_list);
759     free(kf_list);
760     free(tf_list);
761 }