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