]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/users.cpp
Flexible SendQ
[user/henk/code/inspircd.git] / src / users.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
6  * See: http://wiki.inspircd.org/Credits
7  *
8  * This program is free but copyrighted software; see
9  *            the file COPYING for details.
10  *
11  * ---------------------------------------------------
12  */
13
14 /* $Core */
15
16 #include "inspircd.h"
17 #include <stdarg.h>
18 #include "socketengine.h"
19 #include "xline.h"
20 #include "bancache.h"
21 #include "commands/cmd_whowas.h"
22
23 /* XXX: Used for speeding up WriteCommon operations */
24 unsigned long uniq_id = 1;
25
26 static unsigned long* already_sent = NULL;
27
28 LocalIntExt User::NICKForced("NICKForced", NULL);
29 LocalStringExt User::OperQuit("OperQuit", NULL);
30
31 void InitializeAlreadySent(SocketEngine* SE)
32 {
33         already_sent = new unsigned long[SE->GetMaxFds()];
34         memset(already_sent, 0, SE->GetMaxFds() * sizeof(unsigned long));
35 }
36
37 std::string User::ProcessNoticeMasks(const char *sm)
38 {
39         bool adding = true, oldadding = false;
40         const char *c = sm;
41         std::string output;
42
43         while (c && *c)
44         {
45                 switch (*c)
46                 {
47                         case '+':
48                                 adding = true;
49                         break;
50                         case '-':
51                                 adding = false;
52                         break;
53                         case '*':
54                                 for (unsigned char d = 'A'; d <= 'z'; d++)
55                                 {
56                                         if (ServerInstance->SNO->IsEnabled(d))
57                                         {
58                                                 if ((!IsNoticeMaskSet(d) && adding) || (IsNoticeMaskSet(d) && !adding))
59                                                 {
60                                                         if ((oldadding != adding) || (!output.length()))
61                                                                 output += (adding ? '+' : '-');
62
63                                                         this->SetNoticeMask(d, adding);
64
65                                                         output += d;
66                                                 }
67                                         }
68                                         oldadding = adding;
69                                 }
70                         break;
71                         default:
72                                 if ((*c >= 'A') && (*c <= 'z') && (ServerInstance->SNO->IsEnabled(*c)))
73                                 {
74                                         if ((!IsNoticeMaskSet(*c) && adding) || (IsNoticeMaskSet(*c) && !adding))
75                                         {
76                                                 if ((oldadding != adding) || (!output.length()))
77                                                         output += (adding ? '+' : '-');
78
79                                                 this->SetNoticeMask(*c, adding);
80
81                                                 output += *c;
82                                         }
83                                 }
84                                 else
85                                         this->WriteNumeric(ERR_UNKNOWNSNOMASK, "%s %c :is unknown snomask char to me", this->nick.c_str(), *c);
86
87                                 oldadding = adding;
88                         break;
89                 }
90
91                 *c++;
92         }
93
94         std::string s = this->FormatNoticeMasks();
95         if (s.length() == 0)
96         {
97                 this->modes[UM_SNOMASK] = false;
98         }
99
100         return output;
101 }
102
103 void User::StartDNSLookup()
104 {
105         try
106         {
107                 bool cached = false;
108                 const char* sip = this->GetIPString();
109                 UserResolver *res_reverse;
110
111                 QueryType resolvtype = this->client_sa.sa.sa_family == AF_INET6 ? DNS_QUERY_PTR6 : DNS_QUERY_PTR4;
112                 res_reverse = new UserResolver(this, sip, resolvtype, cached);
113
114                 ServerInstance->AddResolver(res_reverse, cached);
115         }
116         catch (CoreException& e)
117         {
118                 ServerInstance->Logs->Log("USERS", DEBUG,"Error in resolver: %s",e.GetReason());
119         }
120 }
121
122 bool User::IsNoticeMaskSet(unsigned char sm)
123 {
124         if (!isalpha(sm))
125                 return false;
126         return (snomasks[sm-65]);
127 }
128
129 void User::SetNoticeMask(unsigned char sm, bool value)
130 {
131         if (!isalpha(sm))
132                 return;
133         snomasks[sm-65] = value;
134 }
135
136 const char* User::FormatNoticeMasks()
137 {
138         static char data[MAXBUF];
139         int offset = 0;
140
141         for (int n = 0; n < 64; n++)
142         {
143                 if (snomasks[n])
144                         data[offset++] = n+65;
145         }
146
147         data[offset] = 0;
148         return data;
149 }
150
151 bool User::IsModeSet(unsigned char m)
152 {
153         if (!isalpha(m))
154                 return false;
155         return (modes[m-65]);
156 }
157
158 void User::SetMode(unsigned char m, bool value)
159 {
160         if (!isalpha(m))
161                 return;
162         modes[m-65] = value;
163 }
164
165 const char* User::FormatModes(bool showparameters)
166 {
167         static char data[MAXBUF];
168         std::string params;
169         int offset = 0;
170
171         for (unsigned char n = 0; n < 64; n++)
172         {
173                 if (modes[n])
174                 {
175                         data[offset++] = n + 65;
176                         ModeHandler* mh = ServerInstance->Modes->FindMode(n + 65, MODETYPE_USER);
177                         if (showparameters && mh && mh->GetNumParams(true))
178                         {
179                                 std::string p = mh->GetUserParameter(this);
180                                 if (p.length())
181                                         params.append(" ").append(p);
182                         }
183                 }
184         }
185         data[offset] = 0;
186         strlcat(data, params.c_str(), MAXBUF);
187         return data;
188 }
189
190 void User::DecrementModes()
191 {
192         ServerInstance->Logs->Log("USERS", DEBUG, "DecrementModes()");
193         for (unsigned char n = 'A'; n <= 'z'; n++)
194         {
195                 if (modes[n-65])
196                 {
197                         ServerInstance->Logs->Log("USERS", DEBUG,"DecrementModes() found mode %c", n);
198                         ModeHandler* mh = ServerInstance->Modes->FindMode(n, MODETYPE_USER);
199                         if (mh)
200                         {
201                                 ServerInstance->Logs->Log("USERS", DEBUG,"Found handler %c and call ChangeCount", n);
202                                 mh->ChangeCount(-1);
203                         }
204                 }
205         }
206 }
207
208 User::User(const std::string &uid)
209 {
210         server = ServerInstance->FindServerNamePtr(ServerInstance->Config->ServerName);
211         age = ServerInstance->Time();
212         Penalty = 0;
213         lastping = signon = idle_lastmsg = nping = registered = 0;
214         bytes_in = bytes_out = cmds_in = cmds_out = 0;
215         quietquit = quitting = exempt = haspassed = dns_done = false;
216         fd = -1;
217         server_sa.sa.sa_family = AF_UNSPEC;
218         client_sa.sa.sa_family = AF_UNSPEC;
219         MyClass = NULL;
220         AllowedPrivs = AllowedOperCommands = NULL;
221
222         if (uid.empty())
223                 uuid.assign(ServerInstance->GetUID(), 0, UUID_LENGTH - 1);
224         else
225                 uuid.assign(uid, 0, UUID_LENGTH - 1);
226
227         ServerInstance->Logs->Log("USERS", DEBUG,"New UUID for user: %s (%s)", uuid.c_str(), uid.empty() ? "allocated new" : "used remote");
228
229         user_hash::iterator finduuid = ServerInstance->Users->uuidlist->find(uuid);
230         if (finduuid == ServerInstance->Users->uuidlist->end())
231                 (*ServerInstance->Users->uuidlist)[uuid] = this;
232         else
233                 throw CoreException("Duplicate UUID "+std::string(uuid)+" in User constructor");
234 }
235
236 User::~User()
237 {
238         ServerInstance->Logs->Log("USERS", DEBUG, "User destructor for %s", uuid.c_str());
239         /* NULL for remote users :) */
240         if (this->MyClass)
241         {
242                 this->MyClass->RefCount--;
243                 ServerInstance->Logs->Log("USERS", DEBUG, "User destructor -- connect refcount now: %lu", this->MyClass->RefCount);
244                 if (MyClass->RefCount == 0)
245                         delete MyClass;
246         }
247
248         if (this->AllowedOperCommands)
249         {
250                 delete AllowedOperCommands;
251                 AllowedOperCommands = NULL;
252         }
253
254         if (this->AllowedPrivs)
255         {
256                 delete AllowedPrivs;
257                 AllowedPrivs = NULL;
258         }
259
260         this->InvalidateCache();
261         this->DecrementModes();
262
263         if (client_sa.sa.sa_family != AF_UNSPEC)
264                 ServerInstance->Users->RemoveCloneCounts(this);
265
266         ServerInstance->Users->uuidlist->erase(uuid);
267 }
268
269 const std::string& User::MakeHost()
270 {
271         if (!this->cached_makehost.empty())
272                 return this->cached_makehost;
273
274         char nhost[MAXBUF];
275         /* This is much faster than snprintf */
276         char* t = nhost;
277         for(const char* n = ident.c_str(); *n; n++)
278                 *t++ = *n;
279         *t++ = '@';
280         for(const char* n = host.c_str(); *n; n++)
281                 *t++ = *n;
282         *t = 0;
283
284         this->cached_makehost.assign(nhost);
285
286         return this->cached_makehost;
287 }
288
289 const std::string& User::MakeHostIP()
290 {
291         if (!this->cached_hostip.empty())
292                 return this->cached_hostip;
293
294         char ihost[MAXBUF];
295         /* This is much faster than snprintf */
296         char* t = ihost;
297         for(const char* n = ident.c_str(); *n; n++)
298                 *t++ = *n;
299         *t++ = '@';
300         for(const char* n = this->GetIPString(); *n; n++)
301                 *t++ = *n;
302         *t = 0;
303
304         this->cached_hostip = ihost;
305
306         return this->cached_hostip;
307 }
308
309 const std::string User::GetFullHost()
310 {
311         if (!this->cached_fullhost.empty())
312                 return this->cached_fullhost;
313
314         char result[MAXBUF];
315         char* t = result;
316         for(const char* n = nick.c_str(); *n; n++)
317                 *t++ = *n;
318         *t++ = '!';
319         for(const char* n = ident.c_str(); *n; n++)
320                 *t++ = *n;
321         *t++ = '@';
322         for(const char* n = dhost.c_str(); *n; n++)
323                 *t++ = *n;
324         *t = 0;
325
326         this->cached_fullhost = result;
327
328         return this->cached_fullhost;
329 }
330
331 char* User::MakeWildHost()
332 {
333         static char nresult[MAXBUF];
334         char* t = nresult;
335         *t++ = '*';     *t++ = '!';
336         *t++ = '*';     *t++ = '@';
337         for(const char* n = dhost.c_str(); *n; n++)
338                 *t++ = *n;
339         *t = 0;
340         return nresult;
341 }
342
343 const std::string User::GetFullRealHost()
344 {
345         if (!this->cached_fullrealhost.empty())
346                 return this->cached_fullrealhost;
347
348         char fresult[MAXBUF];
349         char* t = fresult;
350         for(const char* n = nick.c_str(); *n; n++)
351                 *t++ = *n;
352         *t++ = '!';
353         for(const char* n = ident.c_str(); *n; n++)
354                 *t++ = *n;
355         *t++ = '@';
356         for(const char* n = host.c_str(); *n; n++)
357                 *t++ = *n;
358         *t = 0;
359
360         this->cached_fullrealhost = fresult;
361
362         return this->cached_fullrealhost;
363 }
364
365 bool User::IsInvited(const irc::string &channel)
366 {
367         time_t now = ServerInstance->Time();
368         InvitedList::iterator safei;
369         for (InvitedList::iterator i = invites.begin(); i != invites.end(); ++i)
370         {
371                 if (channel == i->first)
372                 {
373                         if (i->second != 0 && now > i->second)
374                         {
375                                 /* Expired invite, remove it. */
376                                 safei = i;
377                                 --i;
378                                 invites.erase(safei);
379                                 continue;
380                         }
381                         return true;
382                 }
383         }
384         return false;
385 }
386
387 InvitedList* User::GetInviteList()
388 {
389         time_t now = ServerInstance->Time();
390         /* Weed out expired invites here. */
391         InvitedList::iterator safei;
392         for (InvitedList::iterator i = invites.begin(); i != invites.end(); ++i)
393         {
394                 if (i->second != 0 && now > i->second)
395                 {
396                         /* Expired invite, remove it. */
397                         safei = i;
398                         --i;
399                         invites.erase(safei);
400                 }
401         }
402         return &invites;
403 }
404
405 void User::InviteTo(const irc::string &channel, time_t invtimeout)
406 {
407         time_t now = ServerInstance->Time();
408         if (invtimeout != 0 && now > invtimeout) return; /* Don't add invites that are expired from the get-go. */
409         for (InvitedList::iterator i = invites.begin(); i != invites.end(); ++i)
410         {
411                 if (channel == i->first)
412                 {
413                         if (i->second != 0 && invtimeout > i->second)
414                         {
415                                 i->second = invtimeout;
416                         }
417
418                         return;
419                 }
420         }
421         invites.push_back(std::make_pair(channel, invtimeout));
422 }
423
424 void User::RemoveInvite(const irc::string &channel)
425 {
426         for (InvitedList::iterator i = invites.begin(); i != invites.end(); i++)
427         {
428                 if (channel == i->first)
429                 {
430                         invites.erase(i);
431                         return;
432                 }
433         }
434 }
435
436 bool User::HasModePermission(unsigned char mode, ModeType type)
437 {
438         if (!IS_LOCAL(this))
439                 return true;
440
441         if (!IS_OPER(this))
442                 return false;
443
444         if (mode < 'A' || mode > ('A' + 64)) return false;
445
446         return ((type == MODETYPE_USER ? AllowedUserModes : AllowedChanModes))[(mode - 'A')];
447
448 }
449
450 bool User::HasPermission(const std::string &command)
451 {
452         /*
453          * users on remote servers can completely bypass all permissions based checks.
454          * This prevents desyncs when one server has different type/class tags to another.
455          * That having been said, this does open things up to the possibility of source changes
456          * allowing remote kills, etc - but if they have access to the src, they most likely have
457          * access to the conf - so it's an end to a means either way.
458          */
459         if (!IS_LOCAL(this))
460                 return true;
461
462         // are they even an oper at all?
463         if (!IS_OPER(this))
464         {
465                 return false;
466         }
467
468         if (!AllowedOperCommands)
469                 return false;
470
471         if (AllowedOperCommands->find(command) != AllowedOperCommands->end())
472                 return true;
473         else if (AllowedOperCommands->find("*") != AllowedOperCommands->end())
474                 return true;
475
476         return false;
477 }
478
479
480 bool User::HasPrivPermission(const std::string &privstr, bool noisy)
481 {
482         if (!IS_LOCAL(this))
483         {
484                 ServerInstance->Logs->Log("PRIVS", DEBUG, "Remote (yes)");
485                 return true;
486         }
487
488         if (!IS_OPER(this))
489         {
490                 if (noisy)
491                         this->WriteServ("NOTICE %s :You are not an oper", this->nick.c_str());
492                 return false;
493         }
494
495         if (!AllowedPrivs)
496         {
497                 if (noisy)
498                         this->WriteServ("NOTICE %s :Privset empty(!?)", this->nick.c_str());
499                 return false;
500         }
501
502         if (AllowedPrivs->find(privstr) != AllowedPrivs->end())
503         {
504                 return true;
505         }
506         else if (AllowedPrivs->find("*") != AllowedPrivs->end())
507         {
508                 return true;
509         }
510
511         if (noisy)
512                 this->WriteServ("NOTICE %s :Oper type %s does not have access to priv %s", this->nick.c_str(), this->oper.c_str(), privstr.c_str());
513         return false;
514 }
515
516 void User::OnDataReady()
517 {
518         if (quitting)
519                 return;
520
521         if (MyClass && recvq.length() > MyClass->GetRecvqMax() && !HasPrivPermission("users/flood/increased-buffers"))
522         {
523                 ServerInstance->Users->QuitUser(this, "RecvQ exceeded");
524                 ServerInstance->SNO->WriteToSnoMask('a', "User %s RecvQ of %lu exceeds connect class maximum of %lu",
525                         nick.c_str(), (unsigned long)recvq.length(), MyClass->GetRecvqMax());
526         }
527         unsigned long sendqmax = ULONG_MAX;
528         if (MyClass && !HasPrivPermission("users/flood/increased-buffers"))
529                 sendqmax = MyClass->GetSendqSoftMax();
530
531         while (Penalty < 10 && getSendQSize() < sendqmax)
532         {
533                 std::string line;
534                 line.reserve(MAXBUF);
535                 std::string::size_type qpos = 0;
536                 while (qpos < recvq.length())
537                 {
538                         char c = recvq[qpos++];
539                         switch (c)
540                         {
541                         case '\0':
542                                 c = ' ';
543                                 break;
544                         case '\r':
545                                 continue;
546                         case '\n':
547                                 goto eol_found;
548                         }
549                         if (line.length() < MAXBUF - 2)
550                                 line.push_back(c);
551                 }
552                 // if we got here, the recvq ran out before we found a newline
553                 return;
554 eol_found:
555                 // just found a newline. Terminate the string, and pull it out of recvq
556                 recvq = recvq.substr(qpos);
557
558                 // TODO should this be moved to when it was inserted in recvq?
559                 ServerInstance->stats->statsRecv += qpos;
560                 this->bytes_in += qpos;
561                 this->cmds_in++;
562
563                 ServerInstance->Parser->ProcessBuffer(line, this);
564         }
565         // Add pseudo-penalty so that we continue processing after sendq recedes
566         if (Penalty == 0 && getSendQSize() >= sendqmax)
567                 Penalty++;
568 }
569
570 void User::AddWriteBuf(const std::string &data)
571 {
572         // Don't bother sending text to remote users!
573         if (IS_REMOTE(this))
574                 return;
575         if (!quitting && MyClass && getSendQSize() + data.length() > MyClass->GetSendqHardMax() && !HasPrivPermission("users/flood/increased-buffers"))
576         {
577                 /*
578                  * Fix by brain - Set the error text BEFORE calling, because
579                  * if we dont it'll recursively  call here over and over again trying
580                  * to repeatedly add the text to the sendq!
581                  */
582                 ServerInstance->Users->QuitUser(this, "SendQ exceeded");
583                 ServerInstance->SNO->WriteToSnoMask('a', "User %s SendQ of %lu exceeds connect class maximum of %lu",
584                         nick.c_str(), (unsigned long)getSendQSize() + data.length(), MyClass->GetSendqHardMax());
585                 return;
586         }
587
588         // We still want to append data to the sendq of a quitting user,
589         // e.g. their ERROR message that says 'closing link'
590
591         WriteData(data);
592 }
593
594 void User::OnError(BufferedSocketError)
595 {
596         ServerInstance->Users->QuitUser(this, getError());
597 }
598
599 void User::cull()
600 {
601         if (!quitting)
602                 ServerInstance->Users->QuitUser(this, "Culled without QuitUser");
603         if (IS_LOCAL(this))
604         {
605                 if (fd != INT_MAX)
606                         Close();
607
608                 std::vector<User*>::iterator x = find(ServerInstance->Users->local_users.begin(),ServerInstance->Users->local_users.end(),this);
609                 if (x != ServerInstance->Users->local_users.end())
610                         ServerInstance->Users->local_users.erase(x);
611                 else
612                         ServerInstance->Logs->Log("USERS", DEBUG, "Failed to remove user from vector");
613         }
614 }
615
616 void User::Oper(const std::string &opertype, const std::string &opername)
617 {
618         if (this->IsModeSet('o'))
619                 this->UnOper();
620
621         this->modes[UM_OPERATOR] = 1;
622         this->WriteServ("MODE %s :+o", this->nick.c_str());
623         FOREACH_MOD(I_OnOper, OnOper(this, opertype));
624
625         ServerInstance->SNO->WriteToSnoMask('o',"%s (%s@%s) is now an IRC operator of type %s (using oper '%s')", this->nick.c_str(), this->ident.c_str(), this->host.c_str(), irc::Spacify(opertype.c_str()), opername.c_str());
626         this->WriteNumeric(381, "%s :You are now %s %s", this->nick.c_str(), strchr("aeiouAEIOU", *opertype.c_str()) ? "an" : "a", irc::Spacify(opertype.c_str()));
627
628         ServerInstance->Logs->Log("OPER", DEFAULT, "%s!%s@%s opered as type: %s", this->nick.c_str(), this->ident.c_str(), this->host.c_str(), opertype.c_str());
629         this->oper.assign(opertype, 0, 512);
630         ServerInstance->Users->all_opers.push_back(this);
631
632         /*
633          * This might look like it's in the wrong place.
634          * It is *not*!
635          *
636          * For multi-network servers, we may not have the opertypes of the remote server, but we still want to mark the user as an oper of that type.
637          * -- w00t
638          */
639         opertype_t::iterator iter_opertype = ServerInstance->Config->opertypes.find(this->oper.c_str());
640         if (iter_opertype != ServerInstance->Config->opertypes.end())
641         {
642                 if (AllowedOperCommands)
643                         AllowedOperCommands->clear();
644                 else
645                         AllowedOperCommands = new std::set<std::string>;
646
647                 if (AllowedPrivs)
648                         AllowedPrivs->clear();
649                 else
650                         AllowedPrivs = new std::set<std::string>;
651
652                 AllowedUserModes.reset();
653                 AllowedChanModes.reset();
654                 this->AllowedUserModes['o' - 'A'] = true; // Call me paranoid if you want.
655
656                 std::string myclass, mycmd, mypriv;
657                 irc::spacesepstream Classes(iter_opertype->second.c_str());
658                 while (Classes.GetToken(myclass))
659                 {
660                         operclass_t::iterator iter_operclass = ServerInstance->Config->operclass.find(myclass.c_str());
661                         if (iter_operclass != ServerInstance->Config->operclass.end())
662                         {
663                                 /* Process commands */
664                                 irc::spacesepstream CommandList(iter_operclass->second.commandlist);
665                                 while (CommandList.GetToken(mycmd))
666                                 {
667                                         this->AllowedOperCommands->insert(mycmd);
668                                 }
669
670                                 irc::spacesepstream PrivList(iter_operclass->second.privs);
671                                 while (PrivList.GetToken(mypriv))
672                                 {
673                                         this->AllowedPrivs->insert(mypriv);
674                                 }
675
676                                 for (unsigned char* c = (unsigned char*)iter_operclass->second.umodelist.c_str(); *c; ++c)
677                                 {
678                                         if (*c == '*')
679                                         {
680                                                 this->AllowedUserModes.set();
681                                         }
682                                         else
683                                         {
684                                                 this->AllowedUserModes[*c - 'A'] = true;
685                                         }
686                                 }
687
688                                 for (unsigned char* c = (unsigned char*)iter_operclass->second.cmodelist.c_str(); *c; ++c)
689                                 {
690                                         if (*c == '*')
691                                         {
692                                                 this->AllowedChanModes.set();
693                                         }
694                                         else
695                                         {
696                                                 this->AllowedChanModes[*c - 'A'] = true;
697                                         }
698                                 }
699                         }
700                 }
701         }
702
703         FOREACH_MOD(I_OnPostOper,OnPostOper(this, opertype, opername));
704 }
705
706 void User::UnOper()
707 {
708         if (IS_OPER(this))
709         {
710                 /*
711                  * unset their oper type (what IS_OPER checks).
712                  * note, order is important - this must come before modes as -o attempts
713                  * to call UnOper. -- w00t
714                  */
715                 this->oper.clear();
716
717
718                 /* Remove all oper only modes from the user when the deoper - Bug #466*/
719                 std::string moderemove("-");
720
721                 for (unsigned char letter = 'A'; letter <= 'z'; letter++)
722                 {
723                         ModeHandler* mh = ServerInstance->Modes->FindMode(letter, MODETYPE_USER);
724                         if (mh && mh->NeedsOper())
725                                 moderemove += letter;
726                 }
727
728
729                 std::vector<std::string> parameters;
730                 parameters.push_back(this->nick);
731                 parameters.push_back(moderemove);
732
733                 ServerInstance->Parser->CallHandler("MODE", parameters, this);
734
735                 /* remove the user from the oper list. Will remove multiple entries as a safeguard against bug #404 */
736                 ServerInstance->Users->all_opers.remove(this);
737
738                 if (AllowedOperCommands)
739                 {
740                         delete AllowedOperCommands;
741                         AllowedOperCommands = NULL;
742                 }
743
744                 if (AllowedPrivs)
745                 {
746                         delete AllowedPrivs;
747                         AllowedPrivs = NULL;
748                 }
749
750                 AllowedUserModes.reset();
751                 AllowedChanModes.reset();
752                 this->modes[UM_OPERATOR] = 0;
753         }
754 }
755
756 /* adds or updates an entry in the whowas list */
757 void User::AddToWhoWas()
758 {
759         Module* whowas = ServerInstance->Modules->Find("cmd_whowas.so");
760         if (whowas)
761         {
762                 WhowasRequest req(NULL, whowas, WhowasRequest::WHOWAS_ADD);
763                 req.user = this;
764                 req.Send();
765         }
766 }
767
768 /*
769  * Check class restrictions
770  */
771 void User::CheckClass()
772 {
773         ConnectClass* a = this->MyClass;
774
775         if ((!a) || (a->type == CC_DENY))
776         {
777                 ServerInstance->Users->QuitUser(this, "Unauthorised connection");
778                 return;
779         }
780         else if ((a->GetMaxLocal()) && (ServerInstance->Users->LocalCloneCount(this) > a->GetMaxLocal()))
781         {
782                 ServerInstance->Users->QuitUser(this, "No more connections allowed from your host via this connect class (local)");
783                 ServerInstance->SNO->WriteToSnoMask('a', "WARNING: maximum LOCAL connections (%ld) exceeded for IP %s", a->GetMaxLocal(), this->GetIPString());
784                 return;
785         }
786         else if ((a->GetMaxGlobal()) && (ServerInstance->Users->GlobalCloneCount(this) > a->GetMaxGlobal()))
787         {
788                 ServerInstance->Users->QuitUser(this, "No more connections allowed from your host via this connect class (global)");
789                 ServerInstance->SNO->WriteToSnoMask('a', "WARNING: maximum GLOBAL connections (%ld) exceeded for IP %s", a->GetMaxGlobal(), this->GetIPString());
790                 return;
791         }
792
793         this->nping = ServerInstance->Time() + a->GetPingTime() + ServerInstance->Config->dns_timeout;
794 }
795
796 bool User::CheckLines(bool doZline)
797 {
798         const char* check[] = { "G" , "K", (doZline) ? "Z" : NULL, NULL };
799
800         if (!this->exempt)
801         {
802                 for (int n = 0; check[n]; ++n)
803                 {
804                         XLine *r = ServerInstance->XLines->MatchesLine(check[n], this);
805
806                         if (r)
807                         {
808                                 r->Apply(this);
809                                 return true;
810                         }
811                 }
812         }
813
814         return false;
815 }
816
817 void User::FullConnect()
818 {
819         ServerInstance->stats->statsConnects++;
820         this->idle_lastmsg = ServerInstance->Time();
821
822         /*
823          * You may be thinking "wtf, we checked this in User::AddClient!" - and yes, we did, BUT.
824          * At the time AddClient is called, we don't have a resolved host, by here we probably do - which
825          * may put the user into a totally seperate class with different restrictions! so we *must* check again.
826          * Don't remove this! -- w00t
827          */
828         this->SetClass();
829
830         /* Check the password, if one is required by the user's connect class.
831          * This CANNOT be in CheckClass(), because that is called prior to PASS as well!
832          */
833         if (this->MyClass && !this->MyClass->GetPass().empty() && !this->haspassed)
834         {
835                 ServerInstance->Users->QuitUser(this, "Invalid password");
836                 return;
837         }
838
839         if (this->CheckLines())
840                 return;
841
842         this->WriteServ("NOTICE Auth :Welcome to \002%s\002!",ServerInstance->Config->Network);
843         this->WriteNumeric(RPL_WELCOME, "%s :Welcome to the %s IRC Network %s!%s@%s",this->nick.c_str(), ServerInstance->Config->Network, this->nick.c_str(), this->ident.c_str(), this->host.c_str());
844         this->WriteNumeric(RPL_YOURHOSTIS, "%s :Your host is %s, running version InspIRCd-2.0",this->nick.c_str(),ServerInstance->Config->ServerName);
845         this->WriteNumeric(RPL_SERVERCREATED, "%s :This server was created %s %s", this->nick.c_str(), __TIME__, __DATE__);
846         this->WriteNumeric(RPL_SERVERVERSION, "%s %s InspIRCd-2.0 %s %s %s", this->nick.c_str(), ServerInstance->Config->ServerName, ServerInstance->Modes->UserModeList().c_str(), ServerInstance->Modes->ChannelModeList().c_str(), ServerInstance->Modes->ParaModeList().c_str());
847
848         ServerInstance->Config->Send005(this);
849         this->WriteNumeric(RPL_YOURUUID, "%s %s :your unique ID", this->nick.c_str(), this->uuid.c_str());
850
851
852         this->ShowMOTD();
853
854         /* Now registered */
855         if (ServerInstance->Users->unregistered_count)
856                 ServerInstance->Users->unregistered_count--;
857
858         /* Trigger LUSERS output, give modules a chance too */
859         ModResult MOD_RESULT;
860         std::string command("LUSERS");
861         std::vector<std::string> parameters;
862         FIRST_MOD_RESULT(OnPreCommand, MOD_RESULT, (command, parameters, this, true, "LUSERS"));
863         if (!MOD_RESULT)
864                 ServerInstance->CallCommandHandler(command, parameters, this);
865
866         /*
867          * We don't set REG_ALL until triggering OnUserConnect, so some module events don't spew out stuff
868          * for a user that doesn't exist yet.
869          */
870         FOREACH_MOD(I_OnUserConnect,OnUserConnect(this));
871
872         this->registered = REG_ALL;
873
874         FOREACH_MOD(I_OnPostConnect,OnPostConnect(this));
875
876         ServerInstance->SNO->WriteToSnoMask('c',"Client connecting on port %d: %s!%s@%s [%s] [%s]",
877                 this->GetServerPort(), this->nick.c_str(), this->ident.c_str(), this->host.c_str(), this->GetIPString(), this->fullname.c_str());
878         ServerInstance->Logs->Log("BANCACHE", DEBUG, "BanCache: Adding NEGATIVE hit for %s", this->GetIPString());
879         ServerInstance->BanCache->AddHit(this->GetIPString(), "", "");
880 }
881
882 /** User::UpdateNick()
883  * re-allocates a nick in the user_hash after they change nicknames,
884  * returns a pointer to the new user as it may have moved
885  */
886 User* User::UpdateNickHash(const char* New)
887 {
888         //user_hash::iterator newnick;
889         user_hash::iterator oldnick = ServerInstance->Users->clientlist->find(this->nick);
890
891         if (!irc::string(this->nick.c_str()).compare(New))
892                 return oldnick->second;
893
894         if (oldnick == ServerInstance->Users->clientlist->end())
895                 return NULL; /* doesnt exist */
896
897         User* olduser = oldnick->second;
898         ServerInstance->Users->clientlist->erase(oldnick);
899         (*(ServerInstance->Users->clientlist))[New] = olduser;
900         return olduser;
901 }
902
903 void User::InvalidateCache()
904 {
905         /* Invalidate cache */
906         cached_fullhost.clear();
907         cached_hostip.clear();
908         cached_makehost.clear();
909         cached_fullrealhost.clear();
910 }
911
912 bool User::ForceNickChange(const char* newnick)
913 {
914         ModResult MOD_RESULT;
915
916         this->InvalidateCache();
917
918         NICKForced.set(this, 1);
919         FIRST_MOD_RESULT(OnUserPreNick, MOD_RESULT, (this, newnick));
920         NICKForced.set(this, 0);
921
922         if (MOD_RESULT == MOD_RES_DENY)
923         {
924                 ServerInstance->stats->statsCollisions++;
925                 return false;
926         }
927
928         std::deque<classbase*> dummy;
929         Command* nickhandler = ServerInstance->Parser->GetHandler("NICK");
930         if (nickhandler) // wtfbbq, when would this not be here
931         {
932                 std::vector<std::string> parameters;
933                 parameters.push_back(newnick);
934                 NICKForced.set(this, 1);
935                 bool result = (ServerInstance->Parser->CallHandler("NICK", parameters, this) == CMD_SUCCESS);
936                 NICKForced.set(this, 0);
937                 return result;
938         }
939
940         // Unreachable, we hope
941         return false;
942 }
943
944 int User::GetServerPort()
945 {
946         switch (this->server_sa.sa.sa_family)
947         {
948                 case AF_INET6:
949                         return htons(this->server_sa.in6.sin6_port);
950                 case AF_INET:
951                         return htons(this->server_sa.in4.sin_port);
952         }
953         return 0;
954 }
955
956 const char* User::GetCIDRMask(int range)
957 {
958         static char buf[44];
959
960         if (range < 0)
961                 throw "Negative range, sorry, no.";
962
963         /*
964          * Original code written by Oliver Lupton (Om).
965          * Integrated by me. Thanks. :) -- w00t
966          */
967         switch (this->client_sa.sa.sa_family)
968         {
969                 case AF_INET6:
970                 {
971                         /* unsigned char s6_addr[16]; */
972                         struct in6_addr v6;
973                         int i, bytestozero, extrabits;
974                         char buffer[40];
975
976                         if(range > 128)
977                                 throw "CIDR mask width greater than address width (IPv6, 128 bit)";
978
979                         /* To create the CIDR mask we want to set all the bits after 'range' bits of the address
980                          * to zero. This means the last (128 - range) bits of the address must be set to zero.
981                          * Hence this number divided by 8 is the number of whole bytes from the end of the address
982                          * which must be set to zero.
983                          */
984                         bytestozero = (128 - range) / 8;
985
986                         /* Some of the least significant bits of the next most significant byte may also have to
987                          * be zeroed. The number of bits is the remainder of the above division.
988                          */
989                         extrabits = (128 - range) % 8;
990
991                         /* Populate our working struct with the parts of the user's IP which are required in the
992                          * final CIDR mask. Set all the subsequent bytes to zero.
993                          * (16 - bytestozero) is the number of bytes which must be populated with actual IP data.
994                          */
995                         for(i = 0; i < (16 - bytestozero); i++)
996                         {
997                                 v6.s6_addr[i] = client_sa.in6.sin6_addr.s6_addr[i];
998                         }
999
1000                         /* And zero all the remaining bytes in the IP. */
1001                         for(; i < 16; i++)
1002                         {
1003                                 v6.s6_addr[i] = 0;
1004                         }
1005
1006                         /* And finally, zero the extra bits required. */
1007                         v6.s6_addr[15 - bytestozero] = (v6.s6_addr[15 - bytestozero] >> extrabits) << extrabits;
1008
1009                         snprintf(buf, 44, "%s/%d", inet_ntop(AF_INET6, &v6, buffer, 40), range);
1010                         return buf;
1011                 }
1012                 break;
1013                 case AF_INET:
1014                 {
1015                         struct in_addr v4;
1016                         char buffer[16];
1017
1018                         if (range > 32)
1019                                 throw "CIDR mask width greater than address width (IPv4, 32 bit)";
1020
1021                         /* Users already have a sockaddr* pointer (User::ip) which contains either a v4 or v6 structure */
1022                         v4.s_addr = client_sa.in4.sin_addr.s_addr;
1023
1024                         /* To create the CIDR mask we want to set all the bits after 'range' bits of the address
1025                          * to zero. This means the last (32 - range) bits of the address must be set to zero.
1026                          * This is done by shifting the value right and then back left by (32 - range) bits.
1027                          */
1028                         if(range > 0)
1029                         {
1030                                 v4.s_addr = ntohl(v4.s_addr);
1031                                 v4.s_addr = (v4.s_addr >> (32 - range)) << (32 - range);
1032                                 v4.s_addr = htonl(v4.s_addr);
1033                         }
1034                         else
1035                         {
1036                                 /* a range of zero would cause a 32 bit value to be shifted by 32 bits.
1037                                  * this has undefined behaviour, but for CIDR purposes the resulting mask
1038                                  * from a.b.c.d/0 is 0.0.0.0/0
1039                                  */
1040                                 v4.s_addr = 0;
1041                         }
1042
1043                         snprintf(buf, 44, "%s/%d", inet_ntop(AF_INET, &v4, buffer, 16), range);
1044                         return buf;
1045                 }
1046                 break;
1047         }
1048
1049         return ""; // unused, but oh well
1050 }
1051
1052 std::string User::GetServerIP()
1053 {
1054         int port;
1055         std::string ip;
1056         irc::sockets::satoap(&server_sa, ip, port);
1057         return ip;
1058 }
1059
1060 const char* User::GetIPString()
1061 {
1062         int port;
1063         if (cachedip.empty())
1064         {
1065                 irc::sockets::satoap(&client_sa, cachedip, port);
1066                 /* IP addresses starting with a : on irc are a Bad Thing (tm) */
1067                 if (cachedip.c_str()[0] == ':')
1068                         cachedip.insert(0,1,'0');
1069         }
1070
1071         return cachedip.c_str();
1072 }
1073
1074 bool User::SetClientIP(const char* sip)
1075 {
1076         this->cachedip = "";
1077         return irc::sockets::aptosa(sip, 0, &client_sa);
1078 }
1079
1080 static std::string wide_newline("\r\n");
1081
1082 void User::Write(const std::string& text)
1083 {
1084         if (!ServerInstance->SE->BoundsCheckFd(this))
1085                 return;
1086
1087         if (text.length() > MAXBUF - 2)
1088         {
1089                 // this should happen rarely or never. Crop the string at 512 and try again.
1090                 std::string try_again = text.substr(0, MAXBUF - 2);
1091                 Write(try_again);
1092                 return;
1093         }
1094
1095         ServerInstance->Logs->Log("USEROUTPUT", DEBUG,"C[%d] O %s", this->GetFd(), text.c_str());
1096
1097         this->AddWriteBuf(text);
1098         this->AddWriteBuf(wide_newline);
1099
1100         ServerInstance->stats->statsSent += text.length() + 2;
1101         this->bytes_out += text.length() + 2;
1102         this->cmds_out++;
1103 }
1104
1105 /** Write()
1106  */
1107 void User::Write(const char *text, ...)
1108 {
1109         va_list argsPtr;
1110         char textbuffer[MAXBUF];
1111
1112         va_start(argsPtr, text);
1113         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1114         va_end(argsPtr);
1115
1116         this->Write(std::string(textbuffer));
1117 }
1118
1119 void User::WriteServ(const std::string& text)
1120 {
1121         this->Write(":%s %s",ServerInstance->Config->ServerName,text.c_str());
1122 }
1123
1124 /** WriteServ()
1125  *  Same as Write(), except `text' is prefixed with `:server.name '.
1126  */
1127 void User::WriteServ(const char* text, ...)
1128 {
1129         va_list argsPtr;
1130         char textbuffer[MAXBUF];
1131
1132         va_start(argsPtr, text);
1133         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1134         va_end(argsPtr);
1135
1136         this->WriteServ(std::string(textbuffer));
1137 }
1138
1139
1140 void User::WriteNumeric(unsigned int numeric, const char* text, ...)
1141 {
1142         va_list argsPtr;
1143         char textbuffer[MAXBUF];
1144
1145         va_start(argsPtr, text);
1146         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1147         va_end(argsPtr);
1148
1149         this->WriteNumeric(numeric, std::string(textbuffer));
1150 }
1151
1152 void User::WriteNumeric(unsigned int numeric, const std::string &text)
1153 {
1154         char textbuffer[MAXBUF];
1155         ModResult MOD_RESULT;
1156
1157         FIRST_MOD_RESULT(OnNumeric, MOD_RESULT, (this, numeric, text));
1158
1159         if (MOD_RESULT == MOD_RES_DENY)
1160                 return;
1161
1162         snprintf(textbuffer,MAXBUF,":%s %03u %s",ServerInstance->Config->ServerName, numeric, text.c_str());
1163         this->Write(std::string(textbuffer));
1164 }
1165
1166 void User::WriteFrom(User *user, const std::string &text)
1167 {
1168         char tb[MAXBUF];
1169
1170         snprintf(tb,MAXBUF,":%s %s",user->GetFullHost().c_str(),text.c_str());
1171
1172         this->Write(std::string(tb));
1173 }
1174
1175
1176 /* write text from an originating user to originating user */
1177
1178 void User::WriteFrom(User *user, const char* text, ...)
1179 {
1180         va_list argsPtr;
1181         char textbuffer[MAXBUF];
1182
1183         va_start(argsPtr, text);
1184         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1185         va_end(argsPtr);
1186
1187         this->WriteFrom(user, std::string(textbuffer));
1188 }
1189
1190
1191 /* write text to an destination user from a source user (e.g. user privmsg) */
1192
1193 void User::WriteTo(User *dest, const char *data, ...)
1194 {
1195         char textbuffer[MAXBUF];
1196         va_list argsPtr;
1197
1198         va_start(argsPtr, data);
1199         vsnprintf(textbuffer, MAXBUF, data, argsPtr);
1200         va_end(argsPtr);
1201
1202         this->WriteTo(dest, std::string(textbuffer));
1203 }
1204
1205 void User::WriteTo(User *dest, const std::string &data)
1206 {
1207         dest->WriteFrom(this, data);
1208 }
1209
1210
1211 void User::WriteCommon(const char* text, ...)
1212 {
1213         char textbuffer[MAXBUF];
1214         va_list argsPtr;
1215
1216         if (this->registered != REG_ALL)
1217                 return;
1218
1219         va_start(argsPtr, text);
1220         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1221         va_end(argsPtr);
1222
1223         this->WriteCommon(std::string(textbuffer));
1224 }
1225
1226 void User::WriteCommon(const std::string &text)
1227 {
1228         bool sent_to_at_least_one = false;
1229         char tb[MAXBUF];
1230
1231         if (this->registered != REG_ALL)
1232                 return;
1233
1234         uniq_id++;
1235
1236         if (!already_sent)
1237                 InitializeAlreadySent(ServerInstance->SE);
1238
1239         /* We dont want to be doing this n times, just once */
1240         snprintf(tb,MAXBUF,":%s %s",this->GetFullHost().c_str(),text.c_str());
1241         std::string out = tb;
1242
1243         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1244         {
1245                 const UserMembList* ulist = (*v)->GetUsers();
1246                 for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
1247                 {
1248                         if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1249                         {
1250                                 already_sent[i->first->fd] = uniq_id;
1251                                 i->first->Write(out);
1252                                 sent_to_at_least_one = true;
1253                         }
1254                 }
1255         }
1256
1257         /*
1258          * if the user was not in any channels, no users will receive the text. Make sure the user
1259          * receives their OWN message for WriteCommon
1260          */
1261         if (!sent_to_at_least_one)
1262         {
1263                 this->Write(std::string(tb));
1264         }
1265 }
1266
1267
1268 /* write a formatted string to all users who share at least one common
1269  * channel, NOT including the source user e.g. for use in QUIT
1270  */
1271
1272 void User::WriteCommonExcept(const char* text, ...)
1273 {
1274         char textbuffer[MAXBUF];
1275         va_list argsPtr;
1276
1277         va_start(argsPtr, text);
1278         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1279         va_end(argsPtr);
1280
1281         this->WriteCommonExcept(std::string(textbuffer));
1282 }
1283
1284 void User::WriteCommonQuit(const std::string &normal_text, const std::string &oper_text)
1285 {
1286         char tb1[MAXBUF];
1287         char tb2[MAXBUF];
1288
1289         if (this->registered != REG_ALL)
1290                 return;
1291
1292         uniq_id++;
1293
1294         if (!already_sent)
1295                 InitializeAlreadySent(ServerInstance->SE);
1296
1297         snprintf(tb1,MAXBUF,":%s QUIT :%s",this->GetFullHost().c_str(),normal_text.c_str());
1298         snprintf(tb2,MAXBUF,":%s QUIT :%s",this->GetFullHost().c_str(),oper_text.c_str());
1299         std::string out1 = tb1;
1300         std::string out2 = tb2;
1301
1302         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1303         {
1304                 const UserMembList* ulist = (*v)->GetUsers();
1305                 for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
1306                 {
1307                         if (this != i->first)
1308                         {
1309                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1310                                 {
1311                                         already_sent[i->first->fd] = uniq_id;
1312                                         i->first->Write(IS_OPER(i->first) ? out2 : out1);
1313                                 }
1314                         }
1315                 }
1316         }
1317 }
1318
1319 void User::WriteCommonExcept(const std::string &text)
1320 {
1321         char tb1[MAXBUF];
1322         std::string out1;
1323
1324         if (this->registered != REG_ALL)
1325                 return;
1326
1327         uniq_id++;
1328
1329         if (!already_sent)
1330                 InitializeAlreadySent(ServerInstance->SE);
1331
1332         snprintf(tb1,MAXBUF,":%s %s",this->GetFullHost().c_str(),text.c_str());
1333         out1 = tb1;
1334
1335         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1336         {
1337                 const UserMembList* ulist = (*v)->GetUsers();
1338                 for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
1339                 {
1340                         if (this != i->first)
1341                         {
1342                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1343                                 {
1344                                         already_sent[i->first->fd] = uniq_id;
1345                                         i->first->Write(out1);
1346                                 }
1347                         }
1348                 }
1349         }
1350
1351 }
1352
1353 void User::WriteWallOps(const std::string &text)
1354 {
1355         std::string wallop("WALLOPS :");
1356         wallop.append(text);
1357
1358         for (std::vector<User*>::const_iterator i = ServerInstance->Users->local_users.begin(); i != ServerInstance->Users->local_users.end(); i++)
1359         {
1360                 User* t = *i;
1361                 if (t->IsModeSet('w'))
1362                         this->WriteTo(t,wallop);
1363         }
1364 }
1365
1366 void User::WriteWallOps(const char* text, ...)
1367 {
1368         if (!IS_LOCAL(this))
1369                 return;
1370
1371         char textbuffer[MAXBUF];
1372         va_list argsPtr;
1373
1374         va_start(argsPtr, text);
1375         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1376         va_end(argsPtr);
1377
1378         this->WriteWallOps(std::string(textbuffer));
1379 }
1380
1381 /* return 0 or 1 depending if users u and u2 share one or more common channels
1382  * (used by QUIT, NICK etc which arent channel specific notices)
1383  *
1384  * The old algorithm in 1.0 for this was relatively inefficient, iterating over
1385  * the first users channels then the second users channels within the outer loop,
1386  * therefore it was a maximum of x*y iterations (upon returning 0 and checking
1387  * all possible iterations). However this new function instead checks against the
1388  * channel's userlist in the inner loop which is a std::map<User*,User*>
1389  * and saves us time as we already know what pointer value we are after.
1390  * Don't quote me on the maths as i am not a mathematician or computer scientist,
1391  * but i believe this algorithm is now x+(log y) maximum iterations instead.
1392  */
1393 bool User::SharesChannelWith(User *other)
1394 {
1395         if ((!other) || (this->registered != REG_ALL) || (other->registered != REG_ALL))
1396                 return false;
1397
1398         /* Outer loop */
1399         for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1400         {
1401                 /* Eliminate the inner loop (which used to be ~equal in size to the outer loop)
1402                  * by replacing it with a map::find which *should* be more efficient
1403                  */
1404                 if ((*i)->HasUser(other))
1405                         return true;
1406         }
1407         return false;
1408 }
1409
1410 bool User::ChangeName(const char* gecos)
1411 {
1412         if (!this->fullname.compare(gecos))
1413                 return true;
1414
1415         if (IS_LOCAL(this))
1416         {
1417                 ModResult MOD_RESULT;
1418                 FIRST_MOD_RESULT(OnChangeLocalUserGECOS, MOD_RESULT, (this,gecos));
1419                 if (MOD_RESULT == MOD_RES_DENY)
1420                         return false;
1421                 FOREACH_MOD(I_OnChangeName,OnChangeName(this,gecos));
1422         }
1423         this->fullname.assign(gecos, 0, ServerInstance->Config->Limits.MaxGecos);
1424
1425         return true;
1426 }
1427
1428 void User::DoHostCycle(const std::string &quitline)
1429 {
1430         char buffer[MAXBUF];
1431
1432         ModResult result = MOD_RES_PASSTHRU;
1433         FIRST_MOD_RESULT(OnHostCycle, result, (this));
1434
1435         if (result == MOD_RES_DENY)
1436                 return;
1437         if (result == MOD_RES_PASSTHRU && !ServerInstance->Config->CycleHosts)
1438                 return;
1439
1440         uniq_id++;
1441
1442         if (!already_sent)
1443                 InitializeAlreadySent(ServerInstance->SE);
1444
1445         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1446         {
1447                 Channel* c = *v;
1448                 snprintf(buffer, MAXBUF, ":%s JOIN %s", GetFullHost().c_str(), c->name.c_str());
1449                 std::string joinline(buffer);
1450                 std::string modeline = ServerInstance->Modes->ModeString(this, c);
1451                 if (modeline.length() > 0)
1452                 {
1453                         snprintf(buffer, MAXBUF, ":%s MODE %s +%s", GetFullHost().c_str(), c->name.c_str(), modeline.c_str());
1454                         modeline = buffer;
1455                 }
1456
1457                 const UserMembList *ulist = c->GetUsers();
1458                 for (UserMembList::const_iterator i = ulist->begin(); i != ulist->end(); i++)
1459                 {
1460                         User* u = i->first;
1461                         if (u == this || !IS_LOCAL(u))
1462                                 continue;
1463
1464                         if (already_sent[i->first->fd] != uniq_id)
1465                         {
1466                                 u->Write(quitline);
1467                                 already_sent[i->first->fd] = uniq_id;
1468                         }
1469                         u->Write(joinline);
1470                         if (modeline.length() > 0)
1471                                 u->Write(modeline);
1472                 }
1473         }
1474 }
1475
1476 bool User::ChangeDisplayedHost(const char* shost)
1477 {
1478         if (dhost == shost)
1479                 return true;
1480
1481         if (IS_LOCAL(this))
1482         {
1483                 ModResult MOD_RESULT;
1484                 FIRST_MOD_RESULT(OnChangeLocalUserHost, MOD_RESULT, (this,shost));
1485                 if (MOD_RESULT == MOD_RES_DENY)
1486                         return false;
1487         }
1488
1489         FOREACH_MOD(I_OnChangeHost, OnChangeHost(this,shost));
1490
1491         std::string quitstr = ":" + GetFullHost() + " QUIT :Changing host";
1492
1493         /* Fix by Om: User::dhost is 65 long, this was truncating some long hosts */
1494         this->dhost.assign(shost, 0, 64);
1495
1496         this->InvalidateCache();
1497
1498         this->DoHostCycle(quitstr);
1499
1500         if (IS_LOCAL(this))
1501                 this->WriteNumeric(RPL_YOURDISPLAYEDHOST, "%s %s :is now your displayed host",this->nick.c_str(),this->dhost.c_str());
1502
1503         return true;
1504 }
1505
1506 bool User::ChangeIdent(const char* newident)
1507 {
1508         if (this->ident == newident)
1509                 return true;
1510
1511         FOREACH_MOD(I_OnChangeIdent, OnChangeIdent(this,newident));
1512
1513         std::string quitstr = ":" + GetFullHost() + " QUIT :Changing ident";
1514
1515         this->ident.assign(newident, 0, ServerInstance->Config->Limits.IdentMax + 1);
1516
1517         this->InvalidateCache();
1518
1519         this->DoHostCycle(quitstr);
1520
1521         return true;
1522 }
1523
1524 void User::SendAll(const char* command, const char* text, ...)
1525 {
1526         char textbuffer[MAXBUF];
1527         char formatbuffer[MAXBUF];
1528         va_list argsPtr;
1529
1530         va_start(argsPtr, text);
1531         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1532         va_end(argsPtr);
1533
1534         snprintf(formatbuffer,MAXBUF,":%s %s $* :%s", this->GetFullHost().c_str(), command, textbuffer);
1535         std::string fmt = formatbuffer;
1536
1537         for (std::vector<User*>::const_iterator i = ServerInstance->Users->local_users.begin(); i != ServerInstance->Users->local_users.end(); i++)
1538         {
1539                 (*i)->Write(fmt);
1540         }
1541 }
1542
1543
1544 std::string User::ChannelList(User* source, bool spy)
1545 {
1546         std::string list;
1547
1548         for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1549         {
1550                 Channel* c = *i;
1551                 /* If the target is the sender, neither +p nor +s is set, or
1552                  * the channel contains the user, it is not a spy channel
1553                  */
1554                 if (spy != (source == this || !(c->IsModeSet('p') || c->IsModeSet('s')) || c->HasUser(source)))
1555                         list.append(c->GetPrefixChar(this)).append(c->name).append(" ");
1556         }
1557
1558         return list;
1559 }
1560
1561 void User::SplitChanList(User* dest, const std::string &cl)
1562 {
1563         std::string line;
1564         std::ostringstream prefix;
1565         std::string::size_type start, pos, length;
1566
1567         prefix << this->nick << " " << dest->nick << " :";
1568         line = prefix.str();
1569         int namelen = strlen(ServerInstance->Config->ServerName) + 6;
1570
1571         for (start = 0; (pos = cl.find(' ', start)) != std::string::npos; start = pos+1)
1572         {
1573                 length = (pos == std::string::npos) ? cl.length() : pos;
1574
1575                 if (line.length() + namelen + length - start > 510)
1576                 {
1577                         ServerInstance->SendWhoisLine(this, dest, 319, "%s", line.c_str());
1578                         line = prefix.str();
1579                 }
1580
1581                 if(pos == std::string::npos)
1582                 {
1583                         line.append(cl.substr(start, length - start));
1584                         break;
1585                 }
1586                 else
1587                 {
1588                         line.append(cl.substr(start, length - start + 1));
1589                 }
1590         }
1591
1592         if (line.length())
1593         {
1594                 ServerInstance->SendWhoisLine(this, dest, 319, "%s", line.c_str());
1595         }
1596 }
1597
1598 /*
1599  * Sets a user's connection class.
1600  * If the class name is provided, it will be used. Otherwise, the class will be guessed using host/ip/ident/etc.
1601  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
1602  * then their ip will be taken as 'priority' anyway, so for example,
1603  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
1604  */
1605 ConnectClass* User::SetClass(const std::string &explicit_name)
1606 {
1607         ConnectClass *found = NULL;
1608
1609         if (!IS_LOCAL(this))
1610                 return NULL;
1611
1612         ServerInstance->Logs->Log("CONNECTCLASS", DEBUG, "Setting connect class for UID %s", this->uuid.c_str());
1613
1614         if (!explicit_name.empty())
1615         {
1616                 for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
1617                 {
1618                         ConnectClass* c = *i;
1619
1620                         if (explicit_name == c->name)
1621                         {
1622                                 ServerInstance->Logs->Log("CONNECTCLASS", DEBUG, "Explicitly set to %s", explicit_name.c_str());
1623                                 found = c;
1624                         }
1625                 }
1626         }
1627         else
1628         {
1629                 for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
1630                 {
1631                         ConnectClass* c = *i;
1632
1633                         if (c->type == CC_ALLOW)
1634                         {
1635                                 ServerInstance->Logs->Log("CONNECTCLASS", DEBUG, "ALLOW %s %d %s", c->host.c_str(), c->GetPort(), c->GetName().c_str());
1636                         }
1637                         else
1638                         {
1639                                 ServerInstance->Logs->Log("CONNECTCLASS", DEBUG, "DENY %s %d %s", c->GetHost().c_str(), c->GetPort(), c->GetName().c_str());
1640                         }
1641
1642                         /* check if host matches.. */
1643                         if (c->GetHost().length() && !InspIRCd::MatchCIDR(this->GetIPString(), c->GetHost(), NULL) &&
1644                             !InspIRCd::MatchCIDR(this->host, c->GetHost(), NULL))
1645                         {
1646                                 ServerInstance->Logs->Log("CONNECTCLASS", DEBUG, "No host match (for %s)", c->GetHost().c_str());
1647                                 continue;
1648                         }
1649
1650                         /*
1651                          * deny change if change will take class over the limit check it HERE, not after we found a matching class,
1652                          * because we should attempt to find another class if this one doesn't match us. -- w00t
1653                          */
1654                         if (c->limit && (c->RefCount >= c->limit))
1655                         {
1656                                 ServerInstance->Logs->Log("CONNECTCLASS", DEBUG, "OOPS: Connect class limit (%lu) hit, denying", c->limit);
1657                                 continue;
1658                         }
1659
1660                         /* if it requires a port ... */
1661                         if (c->GetPort())
1662                         {
1663                                 ServerInstance->Logs->Log("CONNECTCLASS", DEBUG, "Requires port (%d)", c->GetPort());
1664
1665                                 /* and our port doesn't match, fail. */
1666                                 if (this->GetServerPort() != c->GetPort())
1667                                 {
1668                                         ServerInstance->Logs->Log("CONNECTCLASS", DEBUG, "Port match failed (%d)", this->GetServerPort());
1669                                         continue;
1670                                 }
1671                         }
1672
1673                         /* we stop at the first class that meets ALL critera. */
1674                         found = c;
1675                         break;
1676                 }
1677         }
1678
1679         /*
1680          * Okay, assuming we found a class that matches.. switch us into that class, keeping refcounts up to date.
1681          */
1682         if (found)
1683         {
1684                 /* only fiddle with refcounts if they are already in a class .. */
1685                 if (this->MyClass)
1686                 {
1687                         if (found == this->MyClass) // no point changing this shit :P
1688                                 return this->MyClass;
1689                         this->MyClass->RefCount--;
1690                         ServerInstance->Logs->Log("USERS", DEBUG, "Untying user from connect class -- refcount: %lu", this->MyClass->RefCount);
1691                         if (MyClass->RefCount == 0)
1692                                 delete MyClass;
1693                 }
1694
1695                 this->MyClass = found;
1696                 this->MyClass->RefCount++;
1697                 ServerInstance->Logs->Log("USERS", DEBUG, "User tied to new class -- connect refcount now: %lu", this->MyClass->RefCount);
1698         }
1699
1700         return this->MyClass;
1701 }
1702
1703 /* looks up a users password for their connection class (<ALLOW>/<DENY> tags)
1704  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
1705  * then their ip will be taken as 'priority' anyway, so for example,
1706  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
1707  */
1708 ConnectClass* User::GetClass()
1709 {
1710         return this->MyClass;
1711 }
1712
1713 void User::PurgeEmptyChannels()
1714 {
1715         std::vector<Channel*> to_delete;
1716
1717         // firstly decrement the count on each channel
1718         for (UCListIter f = this->chans.begin(); f != this->chans.end(); f++)
1719         {
1720                 Channel* c = *f;
1721                 c->RemoveAllPrefixes(this);
1722                 if (c->DelUser(this) == 0)
1723                 {
1724                         /* No users left in here, mark it for deletion */
1725                         try
1726                         {
1727                                 to_delete.push_back(c);
1728                         }
1729                         catch (...)
1730                         {
1731                                 ServerInstance->Logs->Log("USERS", DEBUG,"Exception in User::PurgeEmptyChannels to_delete.push_back()");
1732                         }
1733                 }
1734         }
1735
1736         for (std::vector<Channel*>::iterator n = to_delete.begin(); n != to_delete.end(); n++)
1737         {
1738                 Channel* thischan = *n;
1739                 chan_hash::iterator i2 = ServerInstance->chanlist->find(thischan->name);
1740                 if (i2 != ServerInstance->chanlist->end())
1741                 {
1742                         ModResult MOD_RESULT;
1743                         FIRST_MOD_RESULT(OnChannelPreDelete, MOD_RESULT, (i2->second));
1744                         if (MOD_RESULT == MOD_RES_DENY)
1745                                 continue; // delete halted by module
1746                         FOREACH_MOD(I_OnChannelDelete,OnChannelDelete(i2->second));
1747                         delete i2->second;
1748                         ServerInstance->chanlist->erase(i2);
1749                         this->chans.erase(*n);
1750                 }
1751         }
1752
1753         this->UnOper();
1754 }
1755
1756 void User::ShowMOTD()
1757 {
1758         if (!ServerInstance->Config->MOTD.size())
1759         {
1760                 this->WriteNumeric(ERR_NOMOTD, "%s :Message of the day file is missing.",this->nick.c_str());
1761                 return;
1762         }
1763         this->WriteNumeric(RPL_MOTDSTART, "%s :%s message of the day", this->nick.c_str(), ServerInstance->Config->ServerName);
1764
1765         for (file_cache::iterator i = ServerInstance->Config->MOTD.begin(); i != ServerInstance->Config->MOTD.end(); i++)
1766                 this->WriteNumeric(RPL_MOTD, "%s :- %s",this->nick.c_str(),i->c_str());
1767
1768         this->WriteNumeric(RPL_ENDOFMOTD, "%s :End of message of the day.", this->nick.c_str());
1769 }
1770
1771 void User::ShowRULES()
1772 {
1773         if (!ServerInstance->Config->RULES.size())
1774         {
1775                 this->WriteNumeric(ERR_NORULES, "%s :RULES File is missing",this->nick.c_str());
1776                 return;
1777         }
1778
1779         this->WriteNumeric(RPL_RULESTART, "%s :- %s Server Rules -",this->nick.c_str(),ServerInstance->Config->ServerName);
1780
1781         for (file_cache::iterator i = ServerInstance->Config->RULES.begin(); i != ServerInstance->Config->RULES.end(); i++)
1782                 this->WriteNumeric(RPL_RULES, "%s :- %s",this->nick.c_str(),i->c_str());
1783
1784         this->WriteNumeric(RPL_RULESEND, "%s :End of RULES command.",this->nick.c_str());
1785 }
1786
1787 void User::IncreasePenalty(int increase)
1788 {
1789         this->Penalty += increase;
1790 }
1791
1792 void User::DecreasePenalty(int decrease)
1793 {
1794         this->Penalty -= decrease;
1795 }
1796
1797 void FakeUser::SetFakeServer(std::string name)
1798 {
1799         this->nick = name;
1800         this->server = nick.c_str();
1801 }
1802
1803 const std::string FakeUser::GetFullHost()
1804 {
1805         if (*ServerInstance->Config->HideWhoisServer)
1806                 return ServerInstance->Config->HideWhoisServer;
1807         return nick;
1808 }
1809
1810 const std::string FakeUser::GetFullRealHost()
1811 {
1812         if (*ServerInstance->Config->HideWhoisServer)
1813                 return ServerInstance->Config->HideWhoisServer;
1814         return nick;
1815 }
1816
1817 ConnectClass::ConnectClass(char t, const std::string& mask)
1818         : type(t), name("unnamed"), registration_timeout(0), host(mask),
1819         pingtime(0), pass(""), hash(""), softsendqmax(0), hardsendqmax(0),
1820         recvqmax(0), maxlocal(0), maxglobal(0), maxchans(0), port(0), limit(0),
1821         RefCount(1)
1822 {
1823 }
1824
1825 ConnectClass::ConnectClass(char t, const std::string& mask, const ConnectClass& parent)
1826         : type(t), name("unnamed"),
1827         registration_timeout(parent.registration_timeout), host(mask),
1828         pingtime(parent.pingtime), pass(parent.pass), hash(parent.hash),
1829         softsendqmax(parent.softsendqmax), hardsendqmax(parent.hardsendqmax),
1830         recvqmax(parent.recvqmax), maxlocal(parent.maxlocal),
1831         maxglobal(parent.maxglobal), maxchans(parent.maxchans),
1832         port(parent.port), limit(parent.limit), RefCount(1)
1833 {
1834 }
1835
1836 void ConnectClass::Update(const ConnectClass* src)
1837 {
1838         name = src->name;
1839         registration_timeout = src->registration_timeout;
1840         host = src->host;
1841         pingtime = src->pingtime;
1842         pass = src->pass;
1843         hash = src->hash;
1844         softsendqmax = src->softsendqmax;
1845         hardsendqmax = src->hardsendqmax;
1846         recvqmax = src->recvqmax;
1847         maxlocal = src->maxlocal;
1848         maxglobal = src->maxglobal;
1849         limit = src->limit;
1850 }