]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/users.cpp
Add <connect:maxchans> as per feature bug #338 - combined with the last feature,...
[user/henk/code/inspircd.git] / src / users.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
6  * See: http://www.inspircd.org/wiki/index.php/Credits
7  *
8  * This program is free but copyrighted software; see
9  *            the file COPYING for details.
10  *
11  * ---------------------------------------------------
12  */
13
14 #include "inspircd.h"
15 #include "configreader.h"
16 #include "channels.h"
17 #include "users.h"
18 #include <stdarg.h>
19 #include "socketengine.h"
20 #include "wildcard.h"
21 #include "xline.h"
22 #include "commands/cmd_whowas.h"
23
24 static unsigned long already_sent[MAX_DESCRIPTORS] = {0};
25
26 /* XXX: Used for speeding up WriteCommon operations */
27 unsigned long uniq_id = 0;
28
29 bool InitTypes(ServerConfig* conf, const char* tag)
30 {
31         if (conf->opertypes.size())
32         {
33                 for (opertype_t::iterator n = conf->opertypes.begin(); n != conf->opertypes.end(); n++)
34                 {
35                         if (n->second)
36                                 delete[] n->second;
37                 }
38         }
39
40         conf->opertypes.clear();
41         return true;
42 }
43
44 bool InitClasses(ServerConfig* conf, const char* tag)
45 {
46         if (conf->operclass.size())
47         {
48                 for (operclass_t::iterator n = conf->operclass.begin(); n != conf->operclass.end(); n++)
49                 {
50                         if (n->second)
51                                 delete[] n->second;
52                 }
53         }
54
55         conf->operclass.clear();
56         return true;
57 }
58
59 bool DoType(ServerConfig* conf, const char* tag, char** entries, ValueList &values, int* types)
60 {
61         const char* TypeName = values[0].GetString();
62         const char* Classes = values[1].GetString();
63
64         conf->opertypes[TypeName] = strnewdup(Classes);
65         return true;
66 }
67
68 bool DoClass(ServerConfig* conf, const char* tag, char** entries, ValueList &values, int* types)
69 {
70         const char* ClassName = values[0].GetString();
71         const char* CommandList = values[1].GetString();
72
73         conf->operclass[ClassName] = strnewdup(CommandList);
74         return true;
75 }
76
77 bool DoneClassesAndTypes(ServerConfig* conf, const char* tag)
78 {
79         return true;
80 }
81
82 std::string userrec::ProcessNoticeMasks(const char *sm)
83 {
84         bool adding = true, oldadding = false;
85         const char *c = sm;
86         std::string output;
87
88         while (c && *c)
89         {
90                 switch (*c)
91                 {
92                         case '+':
93                                 adding = true;
94                         break;
95                         case '-':
96                                 adding = false;
97                         break;
98                         case '*':
99                                 for (unsigned char d = 'A'; d <= 'z'; d++)
100                                 {
101                                         if (ServerInstance->SNO->IsEnabled(d))
102                                         {
103                                                 if ((!IsNoticeMaskSet(d) && adding) || (IsNoticeMaskSet(d) && !adding))
104                                                 {
105                                                         if ((oldadding != adding) || (!output.length()))
106                                                                 output += (adding ? '+' : '-');
107
108                                                         this->SetNoticeMask(d, adding);
109
110                                                         output += d;
111                                                 }
112                                         }
113                                         oldadding = adding;
114                                 }
115                         break;
116                         default:
117                                 if ((*c >= 'A') && (*c <= 'z') && (ServerInstance->SNO->IsEnabled(*c)))
118                                 {
119                                         if ((!IsNoticeMaskSet(*c) && adding) || (IsNoticeMaskSet(*c) && !adding))
120                                         {
121                                                 if ((oldadding != adding) || (!output.length()))
122                                                         output += (adding ? '+' : '-');
123
124                                                 this->SetNoticeMask(*c, adding);
125
126                                                 output += *c;
127                                         }
128                                 }
129                                 oldadding = adding;
130                         break;
131                 }
132
133                 *c++;
134         }
135
136         return output;
137 }
138
139 void userrec::StartDNSLookup()
140 {
141         try
142         {
143                 bool cached;
144                 const char* ip = this->GetIPString();
145
146                 /* Special case for 4in6 (Have i mentioned i HATE 4in6?) */
147                 if (!strncmp(ip, "0::ffff:", 8))
148                         res_reverse = new UserResolver(this->ServerInstance, this, ip + 8, DNS_QUERY_PTR4, cached);
149                 else
150                         res_reverse = new UserResolver(this->ServerInstance, this, ip, this->GetProtocolFamily() == AF_INET ? DNS_QUERY_PTR4 : DNS_QUERY_PTR6, cached);
151
152                 this->ServerInstance->AddResolver(res_reverse, cached);
153         }
154         catch (CoreException& e)
155         {
156                 ServerInstance->Log(DEBUG,"Error in resolver: %s",e.GetReason());
157         }
158 }
159
160 UserResolver::UserResolver(InspIRCd* Instance, userrec* user, std::string to_resolve, QueryType qt, bool &cache) :
161         Resolver(Instance, to_resolve, qt, cache), bound_user(user)
162 {
163         this->fwd = (qt == DNS_QUERY_A || qt == DNS_QUERY_AAAA);
164         this->bound_fd = user->GetFd();
165 }
166
167 void UserResolver::OnLookupComplete(const std::string &result, unsigned int ttl, bool cached, int resultnum)
168 {
169         /* We are only interested in the first matching result */
170         if (resultnum)
171                 return;
172
173         if ((!this->fwd) && (ServerInstance->SE->GetRef(this->bound_fd) == this->bound_user))
174         {
175                 this->bound_user->stored_host = result;
176                 try
177                 {
178                         /* Check we didnt time out */
179                         if (this->bound_user->registered != REG_ALL)
180                         {
181                                 bool cached;
182 #ifdef IPV6
183                                 if (this->bound_user->GetProtocolFamily() == AF_INET6)
184                                 {
185                                         /* IPV6 forward lookup (with possibility of 4in6) */
186                                         const char* ip = this->bound_user->GetIPString();
187                                         bound_user->res_forward = new UserResolver(this->ServerInstance, this->bound_user, result, (!strncmp(ip, "0::ffff:", 8) ? DNS_QUERY_A : DNS_QUERY_AAAA), cached);
188                                 }
189                                 else
190                                         /* IPV4 lookup (mixed protocol mode) */
191 #endif
192                                 /* IPV4 lookup (ipv4 only mode) */
193                                 bound_user->res_forward = new UserResolver(this->ServerInstance, this->bound_user, result, DNS_QUERY_A, cached);
194                                 this->ServerInstance->AddResolver(bound_user->res_forward, cached);
195                         }
196                 }
197                 catch (CoreException& e)
198                 {
199                         ServerInstance->Log(DEBUG,"Error in resolver: %s",e.GetReason());
200                 }
201         }
202         else if ((this->fwd) && (ServerInstance->SE->GetRef(this->bound_fd) == this->bound_user))
203         {
204                 /* Both lookups completed */
205                 std::string result2("0::ffff:");
206                 result2.append(result);
207                 if (this->bound_user->GetIPString() == result || this->bound_user->GetIPString() == result2)
208                 {
209                         std::string hostname = this->bound_user->stored_host;
210                         if (hostname.length() < 65)
211                         {
212                                 /* Check we didnt time out */
213                                 if ((this->bound_user->registered != REG_ALL) && (!this->bound_user->dns_done))
214                                 {
215                                         /* Hostnames starting with : are not a good thing (tm) */
216                                         if (*(hostname.c_str()) == ':')
217                                                 hostname.insert(0, "0");
218
219                                         this->bound_user->WriteServ("NOTICE Auth :*** Found your hostname (%s)%s", hostname.c_str(), (cached ? " -- cached" : ""));
220                                         this->bound_user->dns_done = true;
221                                         strlcpy(this->bound_user->dhost, hostname.c_str(),64);
222                                         strlcpy(this->bound_user->host, hostname.c_str(),64);
223                                         /* Invalidate cache */
224                                         this->bound_user->InvalidateCache();
225                                 }
226                         }
227                         else
228                         {
229                                 if (!this->bound_user->dns_done)
230                                 {
231                                         this->bound_user->WriteServ("NOTICE Auth :*** Your hostname is longer than the maximum of 64 characters, using your IP address (%s) instead.", this->bound_user->GetIPString());
232                                         this->bound_user->dns_done = true;
233                                 }
234                         }
235                 }
236                 else
237                 {
238                         if (!this->bound_user->dns_done)
239                         {
240                                 this->bound_user->WriteServ("NOTICE Auth :*** Your hostname does not match up with your IP address. Sorry, using your IP address (%s) instead.", this->bound_user->GetIPString());
241                                 this->bound_user->dns_done = true;
242                         }
243                 }
244         }
245 }
246
247 void UserResolver::OnError(ResolverError e, const std::string &errormessage)
248 {
249         if (ServerInstance->SE->GetRef(this->bound_fd) == this->bound_user)
250         {
251                 /* Since dns timeout is implemented outside of the resolver, this was a race condition that could result in this message being sent *after*
252                  * the user was fully connected. This check fixes that issue  - Special */
253                 if (!this->bound_user->dns_done)
254                 {
255                         /* Error message here */
256                         this->bound_user->WriteServ("NOTICE Auth :*** Could not resolve your hostname: %s; using your IP address (%s) instead.", errormessage.c_str(), this->bound_user->GetIPString());
257                         this->bound_user->dns_done = true;
258                 }
259         }
260 }
261
262
263 bool userrec::IsNoticeMaskSet(unsigned char sm)
264 {
265         return (snomasks[sm-65]);
266 }
267
268 void userrec::SetNoticeMask(unsigned char sm, bool value)
269 {
270         snomasks[sm-65] = value;
271 }
272
273 const char* userrec::FormatNoticeMasks()
274 {
275         static char data[MAXBUF];
276         int offset = 0;
277
278         for (int n = 0; n < 64; n++)
279         {
280                 if (snomasks[n])
281                         data[offset++] = n+65;
282         }
283
284         data[offset] = 0;
285         return data;
286 }
287
288
289
290 bool userrec::IsModeSet(unsigned char m)
291 {
292         return (modes[m-65]);
293 }
294
295 void userrec::SetMode(unsigned char m, bool value)
296 {
297         modes[m-65] = value;
298 }
299
300 const char* userrec::FormatModes()
301 {
302         static char data[MAXBUF];
303         int offset = 0;
304         for (int n = 0; n < 64; n++)
305         {
306                 if (modes[n])
307                         data[offset++] = n+65;
308         }
309         data[offset] = 0;
310         return data;
311 }
312
313 void userrec::DecrementModes()
314 {
315         for (int n = 0; n < 64; n++)
316         {
317                 if (modes[n])
318                 {
319                         ModeHandler* mh = ServerInstance->Modes->FindMode(n+65, MODETYPE_USER);
320                         if (mh)
321                                 mh->ChangeCount(-1);
322                 }
323         }
324 }
325
326 userrec::userrec(InspIRCd* Instance) : ServerInstance(Instance)
327 {
328         // the PROPER way to do it, AVOID bzero at *ALL* costs
329         *password = *nick = *ident = *host = *dhost = *fullname = *awaymsg = *oper = 0;
330         server = (char*)Instance->FindServerNamePtr(Instance->Config->ServerName);
331         reset_due = ServerInstance->Time();
332         age = ServerInstance->Time(true);
333         lines_in = lastping = signon = idle_lastmsg = nping = registered = 0;
334         ChannelCount = timeout = flood = bytes_in = bytes_out = cmds_in = cmds_out = 0;
335         muted = exempt = haspassed = dns_done = false;
336         fd = -1;
337         recvq.clear();
338         sendq.clear();
339         WriteError.clear();
340         res_forward = res_reverse = NULL;
341         Visibility = NULL;
342         ip = NULL;
343         chans.clear();
344         invites.clear();
345         memset(modes,0,sizeof(modes));
346         memset(snomasks,0,sizeof(snomasks));
347         /* Invalidate cache */
348         operquit = cached_fullhost = cached_hostip = cached_makehost = cached_fullrealhost = NULL;
349 }
350
351 void userrec::RemoveCloneCounts()
352 {
353         clonemap::iterator x = ServerInstance->local_clones.find(this->GetIPString());
354         if (x != ServerInstance->local_clones.end())
355         {
356                 x->second--;
357                 if (!x->second)
358                 {
359                         ServerInstance->local_clones.erase(x);
360                 }
361         }
362         
363         clonemap::iterator y = ServerInstance->global_clones.find(this->GetIPString());
364         if (y != ServerInstance->global_clones.end())
365         {
366                 y->second--;
367                 if (!y->second)
368                 {
369                         ServerInstance->global_clones.erase(y);
370                 }
371         }
372 }
373
374 userrec::~userrec()
375 {
376         this->InvalidateCache();
377         this->DecrementModes();
378         if (operquit)
379                 free(operquit);
380         if (ip)
381         {
382                 this->RemoveCloneCounts();
383
384                 if (this->GetProtocolFamily() == AF_INET)
385                 {
386                         delete (sockaddr_in*)ip;
387                 }
388 #ifdef SUPPORT_IP6LINKS
389                 else
390                 {
391                         delete (sockaddr_in6*)ip;
392                 }
393 #endif
394         }
395 }
396
397 char* userrec::MakeHost()
398 {
399         if (this->cached_makehost)
400                 return this->cached_makehost;
401
402         char nhost[MAXBUF];
403         /* This is much faster than snprintf */
404         char* t = nhost;
405         for(char* n = ident; *n; n++)
406                 *t++ = *n;
407         *t++ = '@';
408         for(char* n = host; *n; n++)
409                 *t++ = *n;
410         *t = 0;
411
412         this->cached_makehost = strdup(nhost);
413
414         return this->cached_makehost;
415 }
416
417 char* userrec::MakeHostIP()
418 {
419         if (this->cached_hostip)
420                 return this->cached_hostip;
421
422         char ihost[MAXBUF];
423         /* This is much faster than snprintf */
424         char* t = ihost;
425         for(char* n = ident; *n; n++)
426                 *t++ = *n;
427         *t++ = '@';
428         for(const char* n = this->GetIPString(); *n; n++)
429                 *t++ = *n;
430         *t = 0;
431
432         this->cached_hostip = strdup(ihost);
433
434         return this->cached_hostip;
435 }
436
437 void userrec::CloseSocket()
438 {
439         shutdown(this->fd,2);
440         close(this->fd);
441 }
442
443 char* userrec::GetFullHost()
444 {
445         if (this->cached_fullhost)
446                 return this->cached_fullhost;
447
448         char result[MAXBUF];
449         char* t = result;
450         for(char* n = nick; *n; n++)
451                 *t++ = *n;
452         *t++ = '!';
453         for(char* n = ident; *n; n++)
454                 *t++ = *n;
455         *t++ = '@';
456         for(char* n = dhost; *n; n++)
457                 *t++ = *n;
458         *t = 0;
459
460         this->cached_fullhost = strdup(result);
461
462         return this->cached_fullhost;
463 }
464
465 char* userrec::MakeWildHost()
466 {
467         static char nresult[MAXBUF];
468         char* t = nresult;
469         *t++ = '*';     *t++ = '!';
470         *t++ = '*';     *t++ = '@';
471         for(char* n = dhost; *n; n++)
472                 *t++ = *n;
473         *t = 0;
474         return nresult;
475 }
476
477 int userrec::ReadData(void* buffer, size_t size)
478 {
479         if (IS_LOCAL(this))
480         {
481 #ifndef WIN32
482                 return read(this->fd, buffer, size);
483 #else
484                 return recv(this->fd, (char*)buffer, size, 0);
485 #endif
486         }
487         else
488                 return 0;
489 }
490
491
492 char* userrec::GetFullRealHost()
493 {
494         if (this->cached_fullrealhost)
495                 return this->cached_fullrealhost;
496
497         char fresult[MAXBUF];
498         char* t = fresult;
499         for(char* n = nick; *n; n++)
500                 *t++ = *n;
501         *t++ = '!';
502         for(char* n = ident; *n; n++)
503                 *t++ = *n;
504         *t++ = '@';
505         for(char* n = host; *n; n++)
506                 *t++ = *n;
507         *t = 0;
508
509         this->cached_fullrealhost = strdup(fresult);
510
511         return this->cached_fullrealhost;
512 }
513
514 bool userrec::IsInvited(const irc::string &channel)
515 {
516         for (InvitedList::iterator i = invites.begin(); i != invites.end(); i++)
517         {
518                 if (channel == *i)
519                 {
520                         return true;
521                 }
522         }
523         return false;
524 }
525
526 InvitedList* userrec::GetInviteList()
527 {
528         return &invites;
529 }
530
531 void userrec::InviteTo(const irc::string &channel)
532 {
533         invites.push_back(channel);
534 }
535
536 void userrec::RemoveInvite(const irc::string &channel)
537 {
538         for (InvitedList::iterator i = invites.begin(); i != invites.end(); i++)
539         {
540                 if (channel == *i)
541                 {
542                         invites.erase(i);
543                         return;
544                 }
545         }
546 }
547
548 bool userrec::HasPermission(const std::string &command)
549 {
550         char* mycmd;
551         char* savept;
552         char* savept2;
553
554         /*
555          * users on remote servers can completely bypass all permissions based checks.
556          * This prevents desyncs when one server has different type/class tags to another.
557          * That having been said, this does open things up to the possibility of source changes
558          * allowing remote kills, etc - but if they have access to the src, they most likely have
559          * access to the conf - so it's an end to a means either way.
560          */
561         if (!IS_LOCAL(this))
562                 return true;
563
564         // are they even an oper at all?
565         if (IS_OPER(this))
566         {
567                 opertype_t::iterator iter_opertype = ServerInstance->Config->opertypes.find(this->oper);
568                 if (iter_opertype != ServerInstance->Config->opertypes.end())
569                 {
570                         char* Classes = strdup(iter_opertype->second);
571                         char* myclass = strtok_r(Classes," ",&savept);
572                         while (myclass)
573                         {
574                                 operclass_t::iterator iter_operclass = ServerInstance->Config->operclass.find(myclass);
575                                 if (iter_operclass != ServerInstance->Config->operclass.end())
576                                 {
577                                         char* CommandList = strdup(iter_operclass->second);
578                                         mycmd = strtok_r(CommandList," ",&savept2);
579                                         while (mycmd)
580                                         {
581                                                 if ((!strcasecmp(mycmd,command.c_str())) || (*mycmd == '*'))
582                                                 {
583                                                         free(Classes);
584                                                         free(CommandList);
585                                                         return true;
586                                                 }
587                                                 mycmd = strtok_r(NULL," ",&savept2);
588                                         }
589                                         free(CommandList);
590                                 }
591                                 myclass = strtok_r(NULL," ",&savept);
592                         }
593                         free(Classes);
594                 }
595         }
596         return false;
597 }
598
599 /** NOTE: We cannot pass a const reference to this method.
600  * The string is changed by the workings of the method,
601  * so that if we pass const ref, we end up copying it to
602  * something we can change anyway. Makes sense to just let
603  * the compiler do that copy for us.
604  */
605 bool userrec::AddBuffer(std::string a)
606 {
607         try
608         {
609                 std::string::size_type i = a.rfind('\r');
610
611                 while (i != std::string::npos)
612                 {
613                         a.erase(i, 1);
614                         i = a.rfind('\r');
615                 }
616
617                 if (a.length())
618                         recvq.append(a);
619
620                 if (recvq.length() > (unsigned)this->recvqmax)
621                 {
622                         this->SetWriteError("RecvQ exceeded");
623                         ServerInstance->WriteOpers("*** User %s RecvQ of %d exceeds connect class maximum of %d",this->nick,recvq.length(),this->recvqmax);
624                         return false;
625                 }
626
627                 return true;
628         }
629
630         catch (...)
631         {
632                 ServerInstance->Log(DEBUG,"Exception in userrec::AddBuffer()");
633                 return false;
634         }
635 }
636
637 bool userrec::BufferIsReady()
638 {
639         return (recvq.find('\n') != std::string::npos);
640 }
641
642 void userrec::ClearBuffer()
643 {
644         recvq.clear();
645 }
646
647 std::string userrec::GetBuffer()
648 {
649         try
650         {
651                 if (!recvq.length())
652                         return "";
653
654                 /* Strip any leading \r or \n off the string.
655                  * Usually there are only one or two of these,
656                  * so its is computationally cheap to do.
657                  */
658                 std::string::iterator t = recvq.begin();
659                 while (t != recvq.end() && (*t == '\r' || *t == '\n'))
660                 {
661                         recvq.erase(t);
662                         t = recvq.begin();
663                 }
664
665                 for (std::string::iterator x = recvq.begin(); x != recvq.end(); x++)
666                 {
667                         /* Find the first complete line, return it as the
668                          * result, and leave the recvq as whats left
669                          */
670                         if (*x == '\n')
671                         {
672                                 std::string ret = std::string(recvq.begin(), x);
673                                 recvq.erase(recvq.begin(), x + 1);
674                                 return ret;
675                         }
676                 }
677                 return "";
678         }
679
680         catch (...)
681         {
682                 ServerInstance->Log(DEBUG,"Exception in userrec::GetBuffer()");
683                 return "";
684         }
685 }
686
687 void userrec::AddWriteBuf(const std::string &data)
688 {
689         if (*this->GetWriteError())
690                 return;
691
692         if (sendq.length() + data.length() > (unsigned)this->sendqmax)
693         {
694                 /*
695                  * Fix by brain - Set the error text BEFORE calling writeopers, because
696                  * if we dont it'll recursively  call here over and over again trying
697                  * to repeatedly add the text to the sendq!
698                  */
699                 this->SetWriteError("SendQ exceeded");
700                 ServerInstance->WriteOpers("*** User %s SendQ of %d exceeds connect class maximum of %d",this->nick,sendq.length() + data.length(),this->sendqmax);
701                 return;
702         }
703
704         try
705         {
706                 if (data.length() > MAXBUF - 2) /* MAXBUF has a value of 514, to account for line terminators */
707                         sendq.append(data.substr(0,MAXBUF - 4)).append("\r\n"); /* MAXBUF-4 = 510 */
708                 else
709                         sendq.append(data);
710         }
711         catch (...)
712         {
713                 this->SetWriteError("SendQ exceeded");
714                 ServerInstance->WriteOpers("*** User %s SendQ got an exception",this->nick);
715         }
716 }
717
718 // send AS MUCH OF THE USERS SENDQ as we are able to (might not be all of it)
719 void userrec::FlushWriteBuf()
720 {
721         try
722         {
723                 if ((this->fd == FD_MAGIC_NUMBER) || (*this->GetWriteError()))
724                 {
725                         sendq.clear();
726                 }
727                 if ((sendq.length()) && (this->fd != FD_MAGIC_NUMBER))
728                 {
729                         int old_sendq_length = sendq.length();
730 #ifndef WIN32
731                 int n_sent = write(this->fd, this->sendq.data(), this->sendq.length());
732 #else
733                 int n_sent = send(this->fd, (const char*)this->sendq.data(), this->sendq.length(), 0);
734 #endif
735                         if (n_sent == -1)
736                         {
737                                 if (errno == EAGAIN)
738                                 {
739                                         /* The socket buffer is full. This isnt fatal,
740                                          * try again later.
741                                          */
742                                         this->ServerInstance->SE->WantWrite(this);
743                                 }
744                                 else
745                                 {
746                                         /* Fatal error, set write error and bail
747                                          */
748                                         this->SetWriteError(errno ? strerror(errno) : "EOF from client");
749                                         return;
750                                 }
751                         }
752                         else
753                         {
754                                 /* advance the queue */
755                                 if (n_sent)
756                                         this->sendq = this->sendq.substr(n_sent);
757                                 /* update the user's stats counters */
758                                 this->bytes_out += n_sent;
759                                 this->cmds_out++;
760                                 if (n_sent != old_sendq_length)
761                                         this->ServerInstance->SE->WantWrite(this);
762                         }
763                 }
764         }
765
766         catch (...)
767         {
768                 ServerInstance->Log(DEBUG,"Exception in userrec::FlushWriteBuf()");
769         }
770
771         if (this->sendq.empty())
772         {
773                 FOREACH_MOD(I_OnBufferFlushed,OnBufferFlushed(this));
774         }
775 }
776
777 void userrec::SetWriteError(const std::string &error)
778 {
779         try
780         {
781                 // don't try to set the error twice, its already set take the first string.
782                 if (this->WriteError.empty())
783                         this->WriteError = error;
784         }
785
786         catch (...)
787         {
788                 ServerInstance->Log(DEBUG,"Exception in userrec::SetWriteError()");
789         }
790 }
791
792 const char* userrec::GetWriteError()
793 {
794         return this->WriteError.c_str();
795 }
796
797 void userrec::Oper(const std::string &opertype)
798 {
799         try
800         {
801                 this->modes[UM_OPERATOR] = 1;
802                 this->WriteServ("MODE %s :+o", this->nick);
803                 FOREACH_MOD(I_OnOper, OnOper(this, opertype));
804                 ServerInstance->Log(DEFAULT,"OPER: %s!%s@%s opered as type: %s", this->nick, this->ident, this->host, opertype.c_str());
805                 strlcpy(this->oper, opertype.c_str(), NICKMAX - 1);
806                 ServerInstance->all_opers.push_back(this);
807                 FOREACH_MOD(I_OnPostOper,OnPostOper(this, opertype));
808         }
809
810         catch (...)
811         {
812                 ServerInstance->Log(DEBUG,"Exception in userrec::Oper()");
813         }
814 }
815
816 void userrec::UnOper()
817 {
818         try
819         {
820                 if (IS_OPER(this))
821                 {
822                         // unset their oper type (what IS_OPER checks), and remove +o
823                         *this->oper = 0;
824                         this->modes[UM_OPERATOR] = 0;
825
826                         // remove them from the opers list.
827                         for (std::vector<userrec*>::iterator a = ServerInstance->all_opers.begin(); a < ServerInstance->all_opers.end(); a++)
828                         {
829                                 if (*a == this)
830                                 {
831                                         ServerInstance->all_opers.erase(a);
832                                         return;
833                                 }
834                         }
835                 }
836         }
837
838         catch (...)
839         {
840                 ServerInstance->Log(DEBUG,"Exception in userrec::UnOper()");
841         }
842 }
843
844 void userrec::QuitUser(InspIRCd* Instance, userrec *user, const std::string &quitreason, const char* operreason)
845 {
846         user->muted = true;
847         Instance->GlobalCulls.AddItem(user, quitreason.c_str(), operreason);
848 }
849
850 /* adds or updates an entry in the whowas list */
851 void userrec::AddToWhoWas()
852 {
853         command_t* whowas_command = ServerInstance->Parser->GetHandler("WHOWAS");
854         if (whowas_command)
855         {
856                 std::deque<classbase*> params;
857                 params.push_back(this);
858                 whowas_command->HandleInternal(WHOWAS_ADD, params);
859         }
860 }
861
862 /* add a client connection to the sockets list */
863 void userrec::AddClient(InspIRCd* Instance, int socket, int port, bool iscached, int socketfamily, sockaddr* ip)
864 {
865         std::string tempnick = ConvToStr(socket) + "-unknown";
866         user_hash::iterator iter = Instance->clientlist->find(tempnick);
867         char ipaddr[MAXBUF];
868 #ifdef IPV6
869         if (socketfamily == AF_INET6)
870                 inet_ntop(AF_INET6, &((const sockaddr_in6*)ip)->sin6_addr, ipaddr, sizeof(ipaddr));
871         else
872 #endif
873         inet_ntop(AF_INET, &((const sockaddr_in*)ip)->sin_addr, ipaddr, sizeof(ipaddr));
874         userrec* New;
875         int j = 0;
876
877         Instance->unregistered_count++;
878
879         /*
880          * fix by brain.
881          * as these nicknames are 'RFC impossible', we can be sure nobody is going to be
882          * using one as a registered connection. As they are per fd, we can also safely assume
883          * that we wont have collisions. Therefore, if the nick exists in the list, its only
884          * used by a dead socket, erase the iterator so that the new client may reclaim it.
885          * this was probably the cause of 'server ignores me when i hammer it with reconnects'
886          * issue in earlier alphas/betas
887          */
888         if (iter != Instance->clientlist->end())
889         {
890                 userrec* goner = iter->second;
891                 DELETE(goner);
892                 Instance->clientlist->erase(iter);
893         }
894
895         New = new userrec(Instance);
896         (*(Instance->clientlist))[tempnick] = New;
897         New->fd = socket;
898         strlcpy(New->nick,tempnick.c_str(),NICKMAX-1);
899
900         New->server = Instance->FindServerNamePtr(Instance->Config->ServerName);
901         /* We don't need range checking here, we KNOW 'unknown\0' will fit into the ident field. */
902         strcpy(New->ident, "unknown");
903
904         New->registered = REG_NONE;
905         New->signon = Instance->Time() + Instance->Config->dns_timeout;
906         New->lastping = 1;
907
908         New->SetSockAddr(socketfamily, ipaddr, port);
909
910         /* Smarter than your average bear^H^H^H^Hset of strlcpys. */
911         for (const char* temp = New->GetIPString(); *temp && j < 64; temp++, j++)
912                 New->dhost[j] = New->host[j] = *temp;
913         New->dhost[j] = New->host[j] = 0;
914
915         Instance->AddLocalClone(New);
916         Instance->AddGlobalClone(New);
917
918         /*
919          * First class check. We do this again in FullConnect after DNS is done, and NICK/USER is recieved.
920          * See my note down there for why this is required. DO NOT REMOVE. :) -- w00t
921          */
922         ConnectClass* i = New->GetClass();
923
924         if (!i)
925         {
926                 userrec::QuitUser(Instance, New, "Access denied by configuration");
927                 return;
928         }
929
930         /*
931          * Check connect class settings and initialise settings into userrec.
932          * This will be done again after DNS resolution. -- w00t
933          */
934         New->CheckClass();
935
936         Instance->local_users.push_back(New);
937
938         if ((Instance->local_users.size() > Instance->Config->SoftLimit) || (Instance->local_users.size() >= MAXCLIENTS))
939         {
940                 Instance->WriteOpers("*** Warning: softlimit value has been reached: %d clients", Instance->Config->SoftLimit);
941                 userrec::QuitUser(Instance, New,"No more connections allowed");
942                 return;
943         }
944
945         /*
946          * XXX -
947          * this is done as a safety check to keep the file descriptors within range of fd_ref_table.
948          * its a pretty big but for the moment valid assumption:
949          * file descriptors are handed out starting at 0, and are recycled as theyre freed.
950          * therefore if there is ever an fd over 65535, 65536 clients must be connected to the
951          * irc server at once (or the irc server otherwise initiating this many connections, files etc)
952          * which for the time being is a physical impossibility (even the largest networks dont have more
953          * than about 10,000 users on ONE server!)
954          */
955 #ifndef WINDOWS
956         if ((unsigned int)socket >= MAX_DESCRIPTORS)
957         {
958                 userrec::QuitUser(Instance, New, "Server is full");
959                 return;
960         }
961 #endif
962
963         New->exempt = (Instance->XLines->matches_exception(New) != NULL);
964         if (!New->exempt)
965         {
966                 ZLine* r = Instance->XLines->matches_zline(ipaddr);
967                 if (r)
968                 {
969                         char reason[MAXBUF];
970                         if (*Instance->Config->MoronBanner)
971                                 New->WriteServ("NOTICE %s :*** %s", New->nick, Instance->Config->MoronBanner);
972                         snprintf(reason,MAXBUF,"Z-Lined: %s",r->reason);
973                         userrec::QuitUser(Instance, New, reason);
974                         return;
975                 }
976         }
977
978         if (socket > -1)
979         {
980                 if (!Instance->SE->AddFd(New))
981                 {
982                         userrec::QuitUser(Instance, New, "Internal error handling connection");
983                         return;
984                 }
985         }
986
987         /* NOTE: even if dns lookups are *off*, we still need to display this.
988          * BOPM and other stuff requires it.
989          */
990         New->WriteServ("NOTICE Auth :*** Looking up your hostname...");
991 }
992
993 unsigned long userrec::GlobalCloneCount()
994 {
995         clonemap::iterator x = ServerInstance->global_clones.find(this->GetIPString());
996         if (x != ServerInstance->global_clones.end())
997                 return x->second;
998         else
999                 return 0;
1000 }
1001
1002 unsigned long userrec::LocalCloneCount()
1003 {
1004         clonemap::iterator x = ServerInstance->local_clones.find(this->GetIPString());
1005         if (x != ServerInstance->local_clones.end())
1006                 return x->second;
1007         else
1008                 return 0;
1009 }
1010
1011 /*
1012  * Check class restrictions
1013  */
1014 void userrec::CheckClass(const std::string &explicit_class)
1015 {
1016         ConnectClass* a = this->GetClass(explicit_class);
1017
1018         if ((!a) || (a->GetType() == CC_DENY))
1019         {
1020                 userrec::QuitUser(ServerInstance, this, "Unauthorised connection");
1021                 return;
1022         }
1023         else if ((a->GetMaxLocal()) && (this->LocalCloneCount() > a->GetMaxLocal()))
1024         {
1025                 userrec::QuitUser(ServerInstance, this, "No more connections allowed from your host via this connect class (local)");
1026                 ServerInstance->WriteOpers("*** WARNING: maximum LOCAL connections (%ld) exceeded for IP %s", a->GetMaxLocal(), this->GetIPString());
1027                 return;
1028         }
1029         else if ((a->GetMaxGlobal()) && (this->GlobalCloneCount() > a->GetMaxGlobal()))
1030         {
1031                 userrec::QuitUser(ServerInstance, this, "No more connections allowed from your host via this connect class (global)");
1032                 ServerInstance->WriteOpers("*** WARNING: maximum GLOBAL connections (%ld) exceeded for IP %s", a->GetMaxGlobal(), this->GetIPString());
1033                 return;
1034         }
1035
1036         this->pingmax = a->GetPingTime();
1037         this->nping = ServerInstance->Time() + a->GetPingTime() + ServerInstance->Config->dns_timeout;
1038         this->timeout = ServerInstance->Time() + a->GetRegTimeout();
1039         this->flood = a->GetFlood();
1040         this->threshold = a->GetThreshold();
1041         this->sendqmax = a->GetSendqMax();
1042         this->recvqmax = a->GetRecvqMax();
1043         this->MaxChans = a->GetMaxChans();
1044 }
1045
1046 void userrec::FullConnect()
1047 {
1048         ServerInstance->stats->statsConnects++;
1049         this->idle_lastmsg = ServerInstance->Time();
1050
1051         /*
1052          * You may be thinking "wtf, we checked this in userrec::AddClient!" - and yes, we did, BUT.
1053          * At the time AddClient is called, we don't have a resolved host, by here we probably do - which
1054          * may put the user into a totally seperate class with different restrictions! so we *must* check again.
1055          * Don't remove this! -- w00t
1056          */
1057         this->CheckClass();
1058         
1059         /* Check the password, if one is required by the user's connect class.
1060          * This CANNOT be in CheckClass(), because that is called prior to PASS as well!
1061          */
1062         if ((!this->GetClass()->GetPass().empty()) && (!this->haspassed))
1063         {
1064                 userrec::QuitUser(ServerInstance, this, "Invalid password");
1065                 return;
1066         }
1067         
1068         if (!this->exempt)
1069         {
1070                 GLine* r = ServerInstance->XLines->matches_gline(this);
1071
1072                 if (r)
1073                 {
1074                         this->muted = true;
1075                         char reason[MAXBUF];
1076                         if (*ServerInstance->Config->MoronBanner)
1077                                 this->WriteServ("NOTICE %s :*** %s", this->nick, ServerInstance->Config->MoronBanner);
1078                         snprintf(reason,MAXBUF,"G-Lined: %s",r->reason);
1079                         userrec::QuitUser(ServerInstance, this, reason);
1080                         return;
1081                 }
1082
1083                 KLine* n = ServerInstance->XLines->matches_kline(this);
1084
1085                 if (n)
1086                 {
1087                         this->muted = true;
1088                         char reason[MAXBUF];
1089                         if (*ServerInstance->Config->MoronBanner)
1090                                 this->WriteServ("NOTICE %s :*** %s", this, ServerInstance->Config->MoronBanner);
1091                         snprintf(reason,MAXBUF,"K-Lined: %s",n->reason);
1092                         userrec::QuitUser(ServerInstance, this, reason);
1093                         return;
1094                 }
1095         }
1096
1097         this->WriteServ("NOTICE Auth :Welcome to \002%s\002!",ServerInstance->Config->Network);
1098         this->WriteServ("001 %s :Welcome to the %s IRC Network %s!%s@%s",this->nick, ServerInstance->Config->Network, this->nick, this->ident, this->host);
1099         this->WriteServ("002 %s :Your host is %s, running version %s",this->nick,ServerInstance->Config->ServerName,VERSION);
1100         this->WriteServ("003 %s :This server was created %s %s", this->nick, __TIME__, __DATE__);
1101         this->WriteServ("004 %s %s %s %s %s %s", this->nick, ServerInstance->Config->ServerName, VERSION, ServerInstance->Modes->UserModeList().c_str(), ServerInstance->Modes->ChannelModeList().c_str(), ServerInstance->Modes->ParaModeList().c_str());
1102
1103         ServerInstance->Config->Send005(this);
1104
1105         this->ShowMOTD();
1106
1107         /* Now registered */
1108         if (ServerInstance->unregistered_count)
1109                 ServerInstance->unregistered_count--;
1110
1111         /* Trigger LUSERS output, give modules a chance too */
1112         int MOD_RESULT = 0;
1113         FOREACH_RESULT(I_OnPreCommand, OnPreCommand("LUSERS", NULL, 0, this, true, "LUSERS"));
1114         if (!MOD_RESULT)
1115                 ServerInstance->CallCommandHandler("LUSERS", NULL, 0, this);
1116
1117         /*
1118          * fix 3 by brain, move registered = 7 below these so that spurious modes and host
1119          * changes dont go out onto the network and produce 'fake direction'.
1120          */
1121         FOREACH_MOD(I_OnUserConnect,OnUserConnect(this));
1122
1123         this->registered = REG_ALL;
1124
1125         FOREACH_MOD(I_OnPostConnect,OnPostConnect(this));
1126
1127         ServerInstance->SNO->WriteToSnoMask('c',"Client connecting on port %d: %s!%s@%s [%s] [%s]", this->GetPort(), this->nick, this->ident, this->host, this->GetIPString(), this->fullname);
1128 }
1129
1130 /** userrec::UpdateNick()
1131  * re-allocates a nick in the user_hash after they change nicknames,
1132  * returns a pointer to the new user as it may have moved
1133  */
1134 userrec* userrec::UpdateNickHash(const char* New)
1135 {
1136         try
1137         {
1138                 //user_hash::iterator newnick;
1139                 user_hash::iterator oldnick = ServerInstance->clientlist->find(this->nick);
1140
1141                 if (!strcasecmp(this->nick,New))
1142                         return oldnick->second;
1143
1144                 if (oldnick == ServerInstance->clientlist->end())
1145                         return NULL; /* doesnt exist */
1146
1147                 userrec* olduser = oldnick->second;
1148                 (*(ServerInstance->clientlist))[New] = olduser;
1149                 ServerInstance->clientlist->erase(oldnick);
1150                 return olduser;
1151         }
1152
1153         catch (...)
1154         {
1155                 ServerInstance->Log(DEBUG,"Exception in userrec::UpdateNickHash()");
1156                 return NULL;
1157         }
1158 }
1159
1160 void userrec::InvalidateCache()
1161 {
1162         /* Invalidate cache */
1163         if (cached_fullhost)
1164                 free(cached_fullhost);
1165         if (cached_hostip)
1166                 free(cached_hostip);
1167         if (cached_makehost)
1168                 free(cached_makehost);
1169         if (cached_fullrealhost)
1170                 free(cached_fullrealhost);
1171         cached_fullhost = cached_hostip = cached_makehost = cached_fullrealhost = NULL;
1172 }
1173
1174 bool userrec::ForceNickChange(const char* newnick)
1175 {
1176         try
1177         {
1178                 int MOD_RESULT = 0;
1179
1180                 this->InvalidateCache();
1181
1182                 FOREACH_RESULT(I_OnUserPreNick,OnUserPreNick(this, newnick));
1183
1184                 if (MOD_RESULT)
1185                 {
1186                         ServerInstance->stats->statsCollisions++;
1187                         return false;
1188                 }
1189
1190                 if (ServerInstance->XLines->matches_qline(newnick))
1191                 {
1192                         ServerInstance->stats->statsCollisions++;
1193                         return false;
1194                 }
1195
1196                 if (this->registered == REG_ALL)
1197                 {
1198                         return (ServerInstance->Parser->CallHandler("NICK", &newnick, 1, this) == CMD_SUCCESS);
1199                 }
1200                 return false;
1201         }
1202
1203         catch (...)
1204         {
1205                 ServerInstance->Log(DEBUG,"Exception in userrec::ForceNickChange()");
1206                 return false;
1207         }
1208 }
1209
1210 void userrec::SetSockAddr(int protocol_family, const char* ip, int port)
1211 {
1212         switch (protocol_family)
1213         {
1214 #ifdef SUPPORT_IP6LINKS
1215                 case AF_INET6:
1216                 {
1217                         sockaddr_in6* sin = new sockaddr_in6;
1218                         sin->sin6_family = AF_INET6;
1219                         sin->sin6_port = port;
1220                         inet_pton(AF_INET6, ip, &sin->sin6_addr);
1221                         this->ip = (sockaddr*)sin;
1222                 }
1223                 break;
1224 #endif
1225                 case AF_INET:
1226                 {
1227                         sockaddr_in* sin = new sockaddr_in;
1228                         sin->sin_family = AF_INET;
1229                         sin->sin_port = port;
1230                         inet_pton(AF_INET, ip, &sin->sin_addr);
1231                         this->ip = (sockaddr*)sin;
1232                 }
1233                 break;
1234                 default:
1235                         ServerInstance->Log(DEBUG,"Ut oh, I dont know protocol %d to be set on '%s'!", protocol_family, this->nick);
1236                 break;
1237         }
1238 }
1239
1240 int userrec::GetPort()
1241 {
1242         if (this->ip == NULL)
1243                 return 0;
1244
1245         switch (this->GetProtocolFamily())
1246         {
1247 #ifdef SUPPORT_IP6LINKS
1248                 case AF_INET6:
1249                 {
1250                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1251                         return sin->sin6_port;
1252                 }
1253                 break;
1254 #endif
1255                 case AF_INET:
1256                 {
1257                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1258                         return sin->sin_port;
1259                 }
1260                 break;
1261                 default:
1262                 break;
1263         }
1264         return 0;
1265 }
1266
1267 int userrec::GetProtocolFamily()
1268 {
1269         if (this->ip == NULL)
1270                 return 0;
1271
1272         sockaddr_in* sin = (sockaddr_in*)this->ip;
1273         return sin->sin_family;
1274 }
1275
1276 const char* userrec::GetIPString()
1277 {
1278         static char buf[1024];
1279
1280         if (this->ip == NULL)
1281                 return "";
1282
1283         switch (this->GetProtocolFamily())
1284         {
1285 #ifdef SUPPORT_IP6LINKS
1286                 case AF_INET6:
1287                 {
1288                         static char temp[1024];
1289
1290                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1291                         inet_ntop(sin->sin6_family, &sin->sin6_addr, buf, sizeof(buf));
1292                         /* IP addresses starting with a : on irc are a Bad Thing (tm) */
1293                         if (*buf == ':')
1294                         {
1295                                 strlcpy(&temp[1], buf, sizeof(temp) - 1);
1296                                 *temp = '0';
1297                                 return temp;
1298                         }
1299                         return buf;
1300                 }
1301                 break;
1302 #endif
1303                 case AF_INET:
1304                 {
1305                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1306                         inet_ntop(sin->sin_family, &sin->sin_addr, buf, sizeof(buf));
1307                         return buf;
1308                 }
1309                 break;
1310                 default:
1311                 break;
1312         }
1313         return "";
1314 }
1315
1316 const char* userrec::GetIPString(char* buf)
1317 {
1318         if (this->ip == NULL)
1319         {
1320                 *buf = 0;
1321                 return buf;
1322         }
1323
1324         switch (this->GetProtocolFamily())
1325         {
1326 #ifdef SUPPORT_IP6LINKS
1327                 case AF_INET6:
1328                 {
1329                         static char temp[1024];
1330
1331                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1332                         inet_ntop(sin->sin6_family, &sin->sin6_addr, buf, sizeof(buf));
1333                         /* IP addresses starting with a : on irc are a Bad Thing (tm) */
1334                         if (*buf == ':')
1335                         {
1336                                 strlcpy(&temp[1], buf, sizeof(temp) - 1);
1337                                 *temp = '0';
1338                                 strlcpy(buf, temp, sizeof(temp));
1339                         }
1340                         return buf;
1341                 }
1342                 break;
1343 #endif
1344                 case AF_INET:
1345                 {
1346                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1347                         inet_ntop(sin->sin_family, &sin->sin_addr, buf, sizeof(buf));
1348                         return buf;
1349                 }
1350                 break;
1351
1352                 default:
1353                 break;
1354         }
1355         return "";
1356 }
1357
1358 /** NOTE: We cannot pass a const reference to this method.
1359  * The string is changed by the workings of the method,
1360  * so that if we pass const ref, we end up copying it to
1361  * something we can change anyway. Makes sense to just let
1362  * the compiler do that copy for us.
1363  */
1364 void userrec::Write(std::string text)
1365 {
1366         if (!ServerInstance->SE->BoundsCheckFd(this))
1367                 return;
1368
1369         try
1370         {
1371                 /* ServerInstance->Log(DEBUG,"C[%d] <- %s", this->GetFd(), text.c_str());
1372                  * WARNING: The above debug line is VERY loud, do NOT
1373                  * enable it till we have a good way of filtering it
1374                  * out of the logs (e.g. 1.2 would be good).
1375                  */
1376                 text.append("\r\n");
1377         }
1378         catch (...)
1379         {
1380                 ServerInstance->Log(DEBUG,"Exception in userrec::Write() std::string::append");
1381                 return;
1382         }
1383
1384         if (ServerInstance->Config->GetIOHook(this->GetPort()))
1385         {
1386                 try
1387                 {
1388                         /* XXX: The lack of buffering here is NOT a bug, modules implementing this interface have to
1389                          * implement their own buffering mechanisms
1390                          */
1391                         ServerInstance->Config->GetIOHook(this->GetPort())->OnRawSocketWrite(this->fd, text.data(), text.length());
1392                 }
1393                 catch (CoreException& modexcept)
1394                 {
1395                         ServerInstance->Log(DEBUG, "%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
1396                 }
1397         }
1398         else
1399         {
1400                 this->AddWriteBuf(text);
1401         }
1402         ServerInstance->stats->statsSent += text.length();
1403         this->ServerInstance->SE->WantWrite(this);
1404 }
1405
1406 /** Write()
1407  */
1408 void userrec::Write(const char *text, ...)
1409 {
1410         va_list argsPtr;
1411         char textbuffer[MAXBUF];
1412
1413         va_start(argsPtr, text);
1414         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1415         va_end(argsPtr);
1416
1417         this->Write(std::string(textbuffer));
1418 }
1419
1420 void userrec::WriteServ(const std::string& text)
1421 {
1422         char textbuffer[MAXBUF];
1423
1424         snprintf(textbuffer,MAXBUF,":%s %s",ServerInstance->Config->ServerName,text.c_str());
1425         this->Write(std::string(textbuffer));
1426 }
1427
1428 /** WriteServ()
1429  *  Same as Write(), except `text' is prefixed with `:server.name '.
1430  */
1431 void userrec::WriteServ(const char* text, ...)
1432 {
1433         va_list argsPtr;
1434         char textbuffer[MAXBUF];
1435
1436         va_start(argsPtr, text);
1437         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1438         va_end(argsPtr);
1439
1440         this->WriteServ(std::string(textbuffer));
1441 }
1442
1443
1444 void userrec::WriteFrom(userrec *user, const std::string &text)
1445 {
1446         char tb[MAXBUF];
1447
1448         snprintf(tb,MAXBUF,":%s %s",user->GetFullHost(),text.c_str());
1449
1450         this->Write(std::string(tb));
1451 }
1452
1453
1454 /* write text from an originating user to originating user */
1455
1456 void userrec::WriteFrom(userrec *user, const char* text, ...)
1457 {
1458         va_list argsPtr;
1459         char textbuffer[MAXBUF];
1460
1461         va_start(argsPtr, text);
1462         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1463         va_end(argsPtr);
1464
1465         this->WriteFrom(user, std::string(textbuffer));
1466 }
1467
1468
1469 /* write text to an destination user from a source user (e.g. user privmsg) */
1470
1471 void userrec::WriteTo(userrec *dest, const char *data, ...)
1472 {
1473         char textbuffer[MAXBUF];
1474         va_list argsPtr;
1475
1476         va_start(argsPtr, data);
1477         vsnprintf(textbuffer, MAXBUF, data, argsPtr);
1478         va_end(argsPtr);
1479
1480         this->WriteTo(dest, std::string(textbuffer));
1481 }
1482
1483 void userrec::WriteTo(userrec *dest, const std::string &data)
1484 {
1485         dest->WriteFrom(this, data);
1486 }
1487
1488
1489 void userrec::WriteCommon(const char* text, ...)
1490 {
1491         char textbuffer[MAXBUF];
1492         va_list argsPtr;
1493
1494         if (this->registered != REG_ALL)
1495                 return;
1496
1497         va_start(argsPtr, text);
1498         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1499         va_end(argsPtr);
1500
1501         this->WriteCommon(std::string(textbuffer));
1502 }
1503
1504 void userrec::WriteCommon(const std::string &text)
1505 {
1506         try
1507         {
1508                 bool sent_to_at_least_one = false;
1509                 char tb[MAXBUF];
1510
1511                 if (this->registered != REG_ALL)
1512                         return;
1513
1514                 uniq_id++;
1515
1516                 /* We dont want to be doing this n times, just once */
1517                 snprintf(tb,MAXBUF,":%s %s",this->GetFullHost(),text.c_str());
1518                 std::string out = tb;
1519
1520                 for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1521                 {
1522                         CUList* ulist = v->first->GetUsers();
1523                         for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1524                         {
1525                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1526                                 {
1527                                         already_sent[i->first->fd] = uniq_id;
1528                                         i->first->Write(out);
1529                                         sent_to_at_least_one = true;
1530                                 }
1531                         }
1532                 }
1533
1534                 /*
1535                  * if the user was not in any channels, no users will receive the text. Make sure the user
1536                  * receives their OWN message for WriteCommon
1537                  */
1538                 if (!sent_to_at_least_one)
1539                 {
1540                         this->Write(std::string(tb));
1541                 }
1542         }
1543
1544         catch (...)
1545         {
1546                 ServerInstance->Log(DEBUG,"Exception in userrec::WriteCommon()");
1547         }
1548 }
1549
1550
1551 /* write a formatted string to all users who share at least one common
1552  * channel, NOT including the source user e.g. for use in QUIT
1553  */
1554
1555 void userrec::WriteCommonExcept(const char* text, ...)
1556 {
1557         char textbuffer[MAXBUF];
1558         va_list argsPtr;
1559
1560         va_start(argsPtr, text);
1561         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1562         va_end(argsPtr);
1563
1564         this->WriteCommonExcept(std::string(textbuffer));
1565 }
1566
1567 void userrec::WriteCommonQuit(const std::string &normal_text, const std::string &oper_text)
1568 {
1569         char tb1[MAXBUF];
1570         char tb2[MAXBUF];
1571
1572         if (this->registered != REG_ALL)
1573                 return;
1574
1575         uniq_id++;
1576         snprintf(tb1,MAXBUF,":%s QUIT :%s",this->GetFullHost(),normal_text.c_str());
1577         snprintf(tb2,MAXBUF,":%s QUIT :%s",this->GetFullHost(),oper_text.c_str());
1578         std::string out1 = tb1;
1579         std::string out2 = tb2;
1580
1581         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1582         {
1583                 CUList *ulist = v->first->GetUsers();
1584                 for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1585                 {
1586                         if (this != i->first)
1587                         {
1588                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1589                                 {
1590                                         already_sent[i->first->fd] = uniq_id;
1591                                         i->first->Write(IS_OPER(i->first) ? out2 : out1);
1592                                 }
1593                         }
1594                 }
1595         }
1596 }
1597
1598 void userrec::WriteCommonExcept(const std::string &text)
1599 {
1600         char tb1[MAXBUF];
1601         std::string out1;
1602
1603         if (this->registered != REG_ALL)
1604                 return;
1605
1606         uniq_id++;
1607         snprintf(tb1,MAXBUF,":%s %s",this->GetFullHost(),text.c_str());
1608         out1 = tb1;
1609
1610         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1611         {
1612                 CUList *ulist = v->first->GetUsers();
1613                 for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1614                 {
1615                         if (this != i->first)
1616                         {
1617                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1618                                 {
1619                                         already_sent[i->first->fd] = uniq_id;
1620                                         i->first->Write(out1);
1621                                 }
1622                         }
1623                 }
1624         }
1625
1626 }
1627
1628 void userrec::WriteWallOps(const std::string &text)
1629 {
1630         if (!IS_OPER(this) && IS_LOCAL(this))
1631                 return;
1632
1633         std::string wallop("WALLOPS :");
1634         wallop.append(text);
1635
1636         for (std::vector<userrec*>::const_iterator i = ServerInstance->local_users.begin(); i != ServerInstance->local_users.end(); i++)
1637         {
1638                 userrec* t = *i;
1639                 if (t->IsModeSet('w'))
1640                         this->WriteTo(t,wallop);
1641         }
1642 }
1643
1644 void userrec::WriteWallOps(const char* text, ...)
1645 {
1646         char textbuffer[MAXBUF];
1647         va_list argsPtr;
1648
1649         va_start(argsPtr, text);
1650         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1651         va_end(argsPtr);
1652
1653         this->WriteWallOps(std::string(textbuffer));
1654 }
1655
1656 /* return 0 or 1 depending if users u and u2 share one or more common channels
1657  * (used by QUIT, NICK etc which arent channel specific notices)
1658  *
1659  * The old algorithm in 1.0 for this was relatively inefficient, iterating over
1660  * the first users channels then the second users channels within the outer loop,
1661  * therefore it was a maximum of x*y iterations (upon returning 0 and checking
1662  * all possible iterations). However this new function instead checks against the
1663  * channel's userlist in the inner loop which is a std::map<userrec*,userrec*>
1664  * and saves us time as we already know what pointer value we are after.
1665  * Don't quote me on the maths as i am not a mathematician or computer scientist,
1666  * but i believe this algorithm is now x+(log y) maximum iterations instead.
1667  */
1668 bool userrec::SharesChannelWith(userrec *other)
1669 {
1670         if ((!other) || (this->registered != REG_ALL) || (other->registered != REG_ALL))
1671                 return false;
1672
1673         /* Outer loop */
1674         for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1675         {
1676                 /* Eliminate the inner loop (which used to be ~equal in size to the outer loop)
1677                  * by replacing it with a map::find which *should* be more efficient
1678                  */
1679                 if (i->first->HasUser(other))
1680                         return true;
1681         }
1682         return false;
1683 }
1684
1685 bool userrec::ChangeName(const char* gecos)
1686 {
1687         if (!strcmp(gecos, this->fullname))
1688                 return true;
1689
1690         if (IS_LOCAL(this))
1691         {
1692                 int MOD_RESULT = 0;
1693                 FOREACH_RESULT(I_OnChangeLocalUserGECOS,OnChangeLocalUserGECOS(this,gecos));
1694                 if (MOD_RESULT)
1695                         return false;
1696                 FOREACH_MOD(I_OnChangeName,OnChangeName(this,gecos));
1697         }
1698         strlcpy(this->fullname,gecos,MAXGECOS+1);
1699
1700         return true;
1701 }
1702
1703 bool userrec::ChangeDisplayedHost(const char* host)
1704 {
1705         if (!strcmp(host, this->dhost))
1706                 return true;
1707
1708         if (IS_LOCAL(this))
1709         {
1710                 int MOD_RESULT = 0;
1711                 FOREACH_RESULT(I_OnChangeLocalUserHost,OnChangeLocalUserHost(this,host));
1712                 if (MOD_RESULT)
1713                         return false;
1714                 FOREACH_MOD(I_OnChangeHost,OnChangeHost(this,host));
1715         }
1716         if (this->ServerInstance->Config->CycleHosts)
1717                 this->WriteCommonExcept("QUIT :Changing hosts");
1718
1719         /* Fix by Om: userrec::dhost is 65 long, this was truncating some long hosts */
1720         strlcpy(this->dhost,host,64);
1721
1722         this->InvalidateCache();
1723
1724         if (this->ServerInstance->Config->CycleHosts)
1725         {
1726                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1727                 {
1728                         i->first->WriteAllExceptSender(this, false, 0, "JOIN %s", i->first->name);
1729                         std::string n = this->ServerInstance->Modes->ModeString(this, i->first);
1730                         if (n.length() > 0)
1731                                 i->first->WriteAllExceptSender(this, true, 0, "MODE %s +%s", i->first->name, n.c_str());
1732                 }
1733         }
1734
1735         if (IS_LOCAL(this))
1736                 this->WriteServ("396 %s %s :is now your displayed host",this->nick,this->dhost);
1737
1738         return true;
1739 }
1740
1741 bool userrec::ChangeIdent(const char* newident)
1742 {
1743         if (!strcmp(newident, this->ident))
1744                 return true;
1745
1746         if (this->ServerInstance->Config->CycleHosts)
1747                 this->WriteCommonExcept("%s","QUIT :Changing ident");
1748
1749         strlcpy(this->ident, newident, IDENTMAX+2);
1750
1751         this->InvalidateCache();
1752
1753         if (this->ServerInstance->Config->CycleHosts)
1754         {
1755                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1756                 {
1757                         i->first->WriteAllExceptSender(this, false, 0, "JOIN %s", i->first->name);
1758                         std::string n = this->ServerInstance->Modes->ModeString(this, i->first);
1759                         if (n.length() > 0)
1760                                 i->first->WriteAllExceptSender(this, true, 0, "MODE %s +%s", i->first->name, n.c_str());
1761                 }
1762         }
1763
1764         return true;
1765 }
1766
1767 void userrec::SendAll(const char* command, char* text, ...)
1768 {
1769         char textbuffer[MAXBUF];
1770         char formatbuffer[MAXBUF];
1771         va_list argsPtr;
1772
1773         va_start(argsPtr, text);
1774         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1775         va_end(argsPtr);
1776
1777         snprintf(formatbuffer,MAXBUF,":%s %s $* :%s", this->GetFullHost(), command, textbuffer);
1778         std::string fmt = formatbuffer;
1779
1780         for (std::vector<userrec*>::const_iterator i = ServerInstance->local_users.begin(); i != ServerInstance->local_users.end(); i++)
1781         {
1782                 (*i)->Write(fmt);
1783         }
1784 }
1785
1786
1787 std::string userrec::ChannelList(userrec* source)
1788 {
1789         try
1790         {
1791                 std::string list;
1792                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1793                 {
1794                         /* If the target is the same as the sender, let them see all their channels.
1795                          * If the channel is NOT private/secret OR the user shares a common channel
1796                          * If the user is an oper, and the <options:operspywhois> option is set.
1797                          */
1798                         if ((source == this) || (IS_OPER(source) && ServerInstance->Config->OperSpyWhois) || (((!i->first->IsModeSet('p')) && (!i->first->IsModeSet('s'))) || (i->first->HasUser(source))))
1799                         {
1800                                 list.append(i->first->GetPrefixChar(this)).append(i->first->name).append(" ");
1801                         }
1802                 }
1803                 return list;
1804         }
1805         catch (...)
1806         {
1807                 ServerInstance->Log(DEBUG,"Exception in userrec::ChannelList()");
1808                 return "";
1809         }
1810 }
1811
1812 void userrec::SplitChanList(userrec* dest, const std::string &cl)
1813 {
1814         std::string line;
1815         std::ostringstream prefix;
1816         std::string::size_type start, pos, length;
1817
1818         try
1819         {
1820                 prefix << this->nick << " " << dest->nick << " :";
1821                 line = prefix.str();
1822                 int namelen = strlen(ServerInstance->Config->ServerName) + 6;
1823
1824                 for (start = 0; (pos = cl.find(' ', start)) != std::string::npos; start = pos+1)
1825                 {
1826                         length = (pos == std::string::npos) ? cl.length() : pos;
1827
1828                         if (line.length() + namelen + length - start > 510)
1829                         {
1830                                 ServerInstance->SendWhoisLine(this, dest, 319, "%s", line.c_str());
1831                                 line = prefix.str();
1832                         }
1833
1834                         if(pos == std::string::npos)
1835                         {
1836                                 line.append(cl.substr(start, length - start));
1837                                 break;
1838                         }
1839                         else
1840                         {
1841                                 line.append(cl.substr(start, length - start + 1));
1842                         }
1843                 }
1844
1845                 if (line.length())
1846                 {
1847                         ServerInstance->SendWhoisLine(this, dest, 319, "%s", line.c_str());
1848                 }
1849         }
1850
1851         catch (...)
1852         {
1853                 ServerInstance->Log(DEBUG,"Exception in userrec::SplitChanList()");
1854         }
1855 }
1856
1857 unsigned int userrec::GetMaxChans()
1858 {
1859         return this->MaxChans;
1860 }
1861
1862 /* looks up a users password for their connection class (<ALLOW>/<DENY> tags)
1863  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
1864  * then their ip will be taken as 'priority' anyway, so for example,
1865  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
1866  */
1867 ConnectClass* userrec::GetClass(const std::string &explicit_name)
1868 {
1869         if (!explicit_name.empty())
1870         {
1871                 for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
1872                 {
1873                         if (explicit_name == i->GetName())
1874                                 return &(*i);
1875                 }
1876         }
1877         else
1878         {
1879                 for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
1880                 {
1881                         if (((match(this->GetIPString(),i->GetHost().c_str(),true)) || (match(this->host,i->GetHost().c_str()))))
1882                         {
1883                                 if (i->GetPort())
1884                                 {
1885                                         if (this->GetPort() == i->GetPort())
1886                                                 return &(*i);
1887                                         else
1888                                                 continue;
1889                                 }
1890                                 else
1891                                         return &(*i);
1892                         }
1893                 }
1894         }
1895         return NULL;
1896 }
1897
1898 void userrec::PurgeEmptyChannels()
1899 {
1900         std::vector<chanrec*> to_delete;
1901
1902         // firstly decrement the count on each channel
1903         for (UCListIter f = this->chans.begin(); f != this->chans.end(); f++)
1904         {
1905                 f->first->RemoveAllPrefixes(this);
1906                 if (f->first->DelUser(this) == 0)
1907                 {
1908                         /* No users left in here, mark it for deletion */
1909                         try
1910                         {
1911                                 to_delete.push_back(f->first);
1912                         }
1913                         catch (...)
1914                         {
1915                                 ServerInstance->Log(DEBUG,"Exception in userrec::PurgeEmptyChannels to_delete.push_back()");
1916                         }
1917                 }
1918         }
1919
1920         for (std::vector<chanrec*>::iterator n = to_delete.begin(); n != to_delete.end(); n++)
1921         {
1922                 chanrec* thischan = *n;
1923                 chan_hash::iterator i2 = ServerInstance->chanlist->find(thischan->name);
1924                 if (i2 != ServerInstance->chanlist->end())
1925                 {
1926                         FOREACH_MOD(I_OnChannelDelete,OnChannelDelete(i2->second));
1927                         DELETE(i2->second);
1928                         ServerInstance->chanlist->erase(i2);
1929                         this->chans.erase(*n);
1930                 }
1931         }
1932
1933         this->UnOper();
1934 }
1935
1936 void userrec::ShowMOTD()
1937 {
1938         if (!ServerInstance->Config->MOTD.size())
1939         {
1940                 this->WriteServ("422 %s :Message of the day file is missing.",this->nick);
1941                 return;
1942         }
1943         this->WriteServ("375 %s :%s message of the day", this->nick, ServerInstance->Config->ServerName);
1944
1945         for (file_cache::iterator i = ServerInstance->Config->MOTD.begin(); i != ServerInstance->Config->MOTD.end(); i++)
1946                 this->WriteServ("372 %s :- %s",this->nick,i->c_str());
1947
1948         this->WriteServ("376 %s :End of message of the day.", this->nick);
1949 }
1950
1951 void userrec::ShowRULES()
1952 {
1953         if (!ServerInstance->Config->RULES.size())
1954         {
1955                 this->WriteServ("434 %s :RULES File is missing",this->nick);
1956                 return;
1957         }
1958
1959         this->WriteServ("308 %s :- %s Server Rules -",this->nick,ServerInstance->Config->ServerName);
1960
1961         for (file_cache::iterator i = ServerInstance->Config->RULES.begin(); i != ServerInstance->Config->RULES.end(); i++)
1962                 this->WriteServ("232 %s :- %s",this->nick,i->c_str());
1963
1964         this->WriteServ("309 %s :End of RULES command.",this->nick);
1965 }
1966
1967 void userrec::HandleEvent(EventType et, int errornum)
1968 {
1969         /* WARNING: May delete this user! */
1970         int thisfd = this->GetFd();
1971
1972         try
1973         {
1974                 switch (et)
1975                 {
1976                         case EVENT_READ:
1977                                 ServerInstance->ProcessUser(this);
1978                         break;
1979                         case EVENT_WRITE:
1980                                 this->FlushWriteBuf();
1981                         break;
1982                         case EVENT_ERROR:
1983                                 /** This should be safe, but dont DARE do anything after it -- Brain */
1984                                 this->SetWriteError(errornum ? strerror(errornum) : "EOF from client");
1985                         break;
1986                 }
1987         }
1988         catch (...)
1989         {
1990                 ServerInstance->Log(DEBUG,"Exception in userrec::HandleEvent intercepted");
1991         }
1992
1993         /* If the user has raised an error whilst being processed, quit them now we're safe to */
1994         if ((ServerInstance->SE->GetRef(thisfd) == this))
1995         {
1996                 if (!WriteError.empty())
1997                 {
1998                         userrec::QuitUser(ServerInstance, this, GetWriteError());
1999                 }
2000         }
2001 }
2002
2003 void userrec::SetOperQuit(const std::string &oquit)
2004 {
2005         if (operquit)
2006                 return;
2007
2008         operquit = strdup(oquit.c_str());
2009 }
2010
2011 const char* userrec::GetOperQuit()
2012 {
2013         return operquit ? operquit : "";
2014 }
2015
2016 VisData::VisData()
2017 {
2018 }
2019
2020 VisData::~VisData()
2021 {
2022 }
2023
2024 bool VisData::VisibleTo(userrec* user)
2025 {
2026         return true;
2027 }
2028