]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/users.cpp
Allow support for multiple dns results per request. This is a significant change...
[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()
1015 {
1016         ConnectClass* a = this->GetClass();
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 }
1044
1045 void userrec::FullConnect()
1046 {
1047         ServerInstance->stats->statsConnects++;
1048         this->idle_lastmsg = ServerInstance->Time();
1049
1050         /*
1051          * You may be thinking "wtf, we checked this in userrec::AddClient!" - and yes, we did, BUT.
1052          * At the time AddClient is called, we don't have a resolved host, by here we probably do - which
1053          * may put the user into a totally seperate class with different restrictions! so we *must* check again.
1054          * Don't remove this! -- w00t
1055          */
1056         this->CheckClass();
1057         
1058         /* Check the password, if one is required by the user's connect class.
1059          * This CANNOT be in CheckClass(), because that is called prior to PASS as well!
1060          */
1061         if ((!this->GetClass()->GetPass().empty()) && (!this->haspassed))
1062         {
1063                 userrec::QuitUser(ServerInstance, this, "Invalid password");
1064                 return;
1065         }
1066         
1067         if (!this->exempt)
1068         {
1069                 GLine* r = ServerInstance->XLines->matches_gline(this);
1070
1071                 if (r)
1072                 {
1073                         this->muted = true;
1074                         char reason[MAXBUF];
1075                         if (*ServerInstance->Config->MoronBanner)
1076                                 this->WriteServ("NOTICE %s :*** %s", this->nick, ServerInstance->Config->MoronBanner);
1077                         snprintf(reason,MAXBUF,"G-Lined: %s",r->reason);
1078                         userrec::QuitUser(ServerInstance, this, reason);
1079                         return;
1080                 }
1081
1082                 KLine* n = ServerInstance->XLines->matches_kline(this);
1083
1084                 if (n)
1085                 {
1086                         this->muted = true;
1087                         char reason[MAXBUF];
1088                         if (*ServerInstance->Config->MoronBanner)
1089                                 this->WriteServ("NOTICE %s :*** %s", this, ServerInstance->Config->MoronBanner);
1090                         snprintf(reason,MAXBUF,"K-Lined: %s",n->reason);
1091                         userrec::QuitUser(ServerInstance, this, reason);
1092                         return;
1093                 }
1094         }
1095
1096         this->WriteServ("NOTICE Auth :Welcome to \002%s\002!",ServerInstance->Config->Network);
1097         this->WriteServ("001 %s :Welcome to the %s IRC Network %s!%s@%s",this->nick, ServerInstance->Config->Network, this->nick, this->ident, this->host);
1098         this->WriteServ("002 %s :Your host is %s, running version %s",this->nick,ServerInstance->Config->ServerName,VERSION);
1099         this->WriteServ("003 %s :This server was created %s %s", this->nick, __TIME__, __DATE__);
1100         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());
1101
1102         ServerInstance->Config->Send005(this);
1103
1104         this->ShowMOTD();
1105
1106         /* Now registered */
1107         if (ServerInstance->unregistered_count)
1108                 ServerInstance->unregistered_count--;
1109
1110         /* Trigger LUSERS output, give modules a chance too */
1111         int MOD_RESULT = 0;
1112         FOREACH_RESULT(I_OnPreCommand, OnPreCommand("LUSERS", NULL, 0, this, true, "LUSERS"));
1113         if (!MOD_RESULT)
1114                 ServerInstance->CallCommandHandler("LUSERS", NULL, 0, this);
1115
1116         /*
1117          * fix 3 by brain, move registered = 7 below these so that spurious modes and host
1118          * changes dont go out onto the network and produce 'fake direction'.
1119          */
1120         FOREACH_MOD(I_OnUserConnect,OnUserConnect(this));
1121
1122         this->registered = REG_ALL;
1123
1124         FOREACH_MOD(I_OnPostConnect,OnPostConnect(this));
1125
1126         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);
1127 }
1128
1129 /** userrec::UpdateNick()
1130  * re-allocates a nick in the user_hash after they change nicknames,
1131  * returns a pointer to the new user as it may have moved
1132  */
1133 userrec* userrec::UpdateNickHash(const char* New)
1134 {
1135         try
1136         {
1137                 //user_hash::iterator newnick;
1138                 user_hash::iterator oldnick = ServerInstance->clientlist->find(this->nick);
1139
1140                 if (!strcasecmp(this->nick,New))
1141                         return oldnick->second;
1142
1143                 if (oldnick == ServerInstance->clientlist->end())
1144                         return NULL; /* doesnt exist */
1145
1146                 userrec* olduser = oldnick->second;
1147                 (*(ServerInstance->clientlist))[New] = olduser;
1148                 ServerInstance->clientlist->erase(oldnick);
1149                 return olduser;
1150         }
1151
1152         catch (...)
1153         {
1154                 ServerInstance->Log(DEBUG,"Exception in userrec::UpdateNickHash()");
1155                 return NULL;
1156         }
1157 }
1158
1159 void userrec::InvalidateCache()
1160 {
1161         /* Invalidate cache */
1162         if (cached_fullhost)
1163                 free(cached_fullhost);
1164         if (cached_hostip)
1165                 free(cached_hostip);
1166         if (cached_makehost)
1167                 free(cached_makehost);
1168         if (cached_fullrealhost)
1169                 free(cached_fullrealhost);
1170         cached_fullhost = cached_hostip = cached_makehost = cached_fullrealhost = NULL;
1171 }
1172
1173 bool userrec::ForceNickChange(const char* newnick)
1174 {
1175         try
1176         {
1177                 int MOD_RESULT = 0;
1178
1179                 this->InvalidateCache();
1180
1181                 FOREACH_RESULT(I_OnUserPreNick,OnUserPreNick(this, newnick));
1182
1183                 if (MOD_RESULT)
1184                 {
1185                         ServerInstance->stats->statsCollisions++;
1186                         return false;
1187                 }
1188
1189                 if (ServerInstance->XLines->matches_qline(newnick))
1190                 {
1191                         ServerInstance->stats->statsCollisions++;
1192                         return false;
1193                 }
1194
1195                 if (this->registered == REG_ALL)
1196                 {
1197                         return (ServerInstance->Parser->CallHandler("NICK", &newnick, 1, this) == CMD_SUCCESS);
1198                 }
1199                 return false;
1200         }
1201
1202         catch (...)
1203         {
1204                 ServerInstance->Log(DEBUG,"Exception in userrec::ForceNickChange()");
1205                 return false;
1206         }
1207 }
1208
1209 void userrec::SetSockAddr(int protocol_family, const char* ip, int port)
1210 {
1211         switch (protocol_family)
1212         {
1213 #ifdef SUPPORT_IP6LINKS
1214                 case AF_INET6:
1215                 {
1216                         sockaddr_in6* sin = new sockaddr_in6;
1217                         sin->sin6_family = AF_INET6;
1218                         sin->sin6_port = port;
1219                         inet_pton(AF_INET6, ip, &sin->sin6_addr);
1220                         this->ip = (sockaddr*)sin;
1221                 }
1222                 break;
1223 #endif
1224                 case AF_INET:
1225                 {
1226                         sockaddr_in* sin = new sockaddr_in;
1227                         sin->sin_family = AF_INET;
1228                         sin->sin_port = port;
1229                         inet_pton(AF_INET, ip, &sin->sin_addr);
1230                         this->ip = (sockaddr*)sin;
1231                 }
1232                 break;
1233                 default:
1234                         ServerInstance->Log(DEBUG,"Ut oh, I dont know protocol %d to be set on '%s'!", protocol_family, this->nick);
1235                 break;
1236         }
1237 }
1238
1239 int userrec::GetPort()
1240 {
1241         if (this->ip == NULL)
1242                 return 0;
1243
1244         switch (this->GetProtocolFamily())
1245         {
1246 #ifdef SUPPORT_IP6LINKS
1247                 case AF_INET6:
1248                 {
1249                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1250                         return sin->sin6_port;
1251                 }
1252                 break;
1253 #endif
1254                 case AF_INET:
1255                 {
1256                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1257                         return sin->sin_port;
1258                 }
1259                 break;
1260                 default:
1261                 break;
1262         }
1263         return 0;
1264 }
1265
1266 int userrec::GetProtocolFamily()
1267 {
1268         if (this->ip == NULL)
1269                 return 0;
1270
1271         sockaddr_in* sin = (sockaddr_in*)this->ip;
1272         return sin->sin_family;
1273 }
1274
1275 const char* userrec::GetIPString()
1276 {
1277         static char buf[1024];
1278
1279         if (this->ip == NULL)
1280                 return "";
1281
1282         switch (this->GetProtocolFamily())
1283         {
1284 #ifdef SUPPORT_IP6LINKS
1285                 case AF_INET6:
1286                 {
1287                         static char temp[1024];
1288
1289                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1290                         inet_ntop(sin->sin6_family, &sin->sin6_addr, buf, sizeof(buf));
1291                         /* IP addresses starting with a : on irc are a Bad Thing (tm) */
1292                         if (*buf == ':')
1293                         {
1294                                 strlcpy(&temp[1], buf, sizeof(temp) - 1);
1295                                 *temp = '0';
1296                                 return temp;
1297                         }
1298                         return buf;
1299                 }
1300                 break;
1301 #endif
1302                 case AF_INET:
1303                 {
1304                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1305                         inet_ntop(sin->sin_family, &sin->sin_addr, buf, sizeof(buf));
1306                         return buf;
1307                 }
1308                 break;
1309                 default:
1310                 break;
1311         }
1312         return "";
1313 }
1314
1315 const char* userrec::GetIPString(char* buf)
1316 {
1317         if (this->ip == NULL)
1318         {
1319                 *buf = 0;
1320                 return buf;
1321         }
1322
1323         switch (this->GetProtocolFamily())
1324         {
1325 #ifdef SUPPORT_IP6LINKS
1326                 case AF_INET6:
1327                 {
1328                         static char temp[1024];
1329
1330                         sockaddr_in6* sin = (sockaddr_in6*)this->ip;
1331                         inet_ntop(sin->sin6_family, &sin->sin6_addr, buf, sizeof(buf));
1332                         /* IP addresses starting with a : on irc are a Bad Thing (tm) */
1333                         if (*buf == ':')
1334                         {
1335                                 strlcpy(&temp[1], buf, sizeof(temp) - 1);
1336                                 *temp = '0';
1337                                 strlcpy(buf, temp, sizeof(temp));
1338                         }
1339                         return buf;
1340                 }
1341                 break;
1342 #endif
1343                 case AF_INET:
1344                 {
1345                         sockaddr_in* sin = (sockaddr_in*)this->ip;
1346                         inet_ntop(sin->sin_family, &sin->sin_addr, buf, sizeof(buf));
1347                         return buf;
1348                 }
1349                 break;
1350
1351                 default:
1352                 break;
1353         }
1354         return "";
1355 }
1356
1357 /** NOTE: We cannot pass a const reference to this method.
1358  * The string is changed by the workings of the method,
1359  * so that if we pass const ref, we end up copying it to
1360  * something we can change anyway. Makes sense to just let
1361  * the compiler do that copy for us.
1362  */
1363 void userrec::Write(std::string text)
1364 {
1365         if (!ServerInstance->SE->BoundsCheckFd(this))
1366                 return;
1367
1368         try
1369         {
1370                 /* ServerInstance->Log(DEBUG,"C[%d] <- %s", this->GetFd(), text.c_str());
1371                  * WARNING: The above debug line is VERY loud, do NOT
1372                  * enable it till we have a good way of filtering it
1373                  * out of the logs (e.g. 1.2 would be good).
1374                  */
1375                 text.append("\r\n");
1376         }
1377         catch (...)
1378         {
1379                 ServerInstance->Log(DEBUG,"Exception in userrec::Write() std::string::append");
1380                 return;
1381         }
1382
1383         if (ServerInstance->Config->GetIOHook(this->GetPort()))
1384         {
1385                 try
1386                 {
1387                         /* XXX: The lack of buffering here is NOT a bug, modules implementing this interface have to
1388                          * implement their own buffering mechanisms
1389                          */
1390                         ServerInstance->Config->GetIOHook(this->GetPort())->OnRawSocketWrite(this->fd, text.data(), text.length());
1391                 }
1392                 catch (CoreException& modexcept)
1393                 {
1394                         ServerInstance->Log(DEBUG, "%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason());
1395                 }
1396         }
1397         else
1398         {
1399                 this->AddWriteBuf(text);
1400         }
1401         ServerInstance->stats->statsSent += text.length();
1402         this->ServerInstance->SE->WantWrite(this);
1403 }
1404
1405 /** Write()
1406  */
1407 void userrec::Write(const char *text, ...)
1408 {
1409         va_list argsPtr;
1410         char textbuffer[MAXBUF];
1411
1412         va_start(argsPtr, text);
1413         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1414         va_end(argsPtr);
1415
1416         this->Write(std::string(textbuffer));
1417 }
1418
1419 void userrec::WriteServ(const std::string& text)
1420 {
1421         char textbuffer[MAXBUF];
1422
1423         snprintf(textbuffer,MAXBUF,":%s %s",ServerInstance->Config->ServerName,text.c_str());
1424         this->Write(std::string(textbuffer));
1425 }
1426
1427 /** WriteServ()
1428  *  Same as Write(), except `text' is prefixed with `:server.name '.
1429  */
1430 void userrec::WriteServ(const char* text, ...)
1431 {
1432         va_list argsPtr;
1433         char textbuffer[MAXBUF];
1434
1435         va_start(argsPtr, text);
1436         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1437         va_end(argsPtr);
1438
1439         this->WriteServ(std::string(textbuffer));
1440 }
1441
1442
1443 void userrec::WriteFrom(userrec *user, const std::string &text)
1444 {
1445         char tb[MAXBUF];
1446
1447         snprintf(tb,MAXBUF,":%s %s",user->GetFullHost(),text.c_str());
1448
1449         this->Write(std::string(tb));
1450 }
1451
1452
1453 /* write text from an originating user to originating user */
1454
1455 void userrec::WriteFrom(userrec *user, const char* text, ...)
1456 {
1457         va_list argsPtr;
1458         char textbuffer[MAXBUF];
1459
1460         va_start(argsPtr, text);
1461         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1462         va_end(argsPtr);
1463
1464         this->WriteFrom(user, std::string(textbuffer));
1465 }
1466
1467
1468 /* write text to an destination user from a source user (e.g. user privmsg) */
1469
1470 void userrec::WriteTo(userrec *dest, const char *data, ...)
1471 {
1472         char textbuffer[MAXBUF];
1473         va_list argsPtr;
1474
1475         va_start(argsPtr, data);
1476         vsnprintf(textbuffer, MAXBUF, data, argsPtr);
1477         va_end(argsPtr);
1478
1479         this->WriteTo(dest, std::string(textbuffer));
1480 }
1481
1482 void userrec::WriteTo(userrec *dest, const std::string &data)
1483 {
1484         dest->WriteFrom(this, data);
1485 }
1486
1487
1488 void userrec::WriteCommon(const char* text, ...)
1489 {
1490         char textbuffer[MAXBUF];
1491         va_list argsPtr;
1492
1493         if (this->registered != REG_ALL)
1494                 return;
1495
1496         va_start(argsPtr, text);
1497         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1498         va_end(argsPtr);
1499
1500         this->WriteCommon(std::string(textbuffer));
1501 }
1502
1503 void userrec::WriteCommon(const std::string &text)
1504 {
1505         try
1506         {
1507                 bool sent_to_at_least_one = false;
1508                 char tb[MAXBUF];
1509
1510                 if (this->registered != REG_ALL)
1511                         return;
1512
1513                 uniq_id++;
1514
1515                 /* We dont want to be doing this n times, just once */
1516                 snprintf(tb,MAXBUF,":%s %s",this->GetFullHost(),text.c_str());
1517                 std::string out = tb;
1518
1519                 for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1520                 {
1521                         CUList* ulist = v->first->GetUsers();
1522                         for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1523                         {
1524                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1525                                 {
1526                                         already_sent[i->first->fd] = uniq_id;
1527                                         i->first->Write(out);
1528                                         sent_to_at_least_one = true;
1529                                 }
1530                         }
1531                 }
1532
1533                 /*
1534                  * if the user was not in any channels, no users will receive the text. Make sure the user
1535                  * receives their OWN message for WriteCommon
1536                  */
1537                 if (!sent_to_at_least_one)
1538                 {
1539                         this->Write(std::string(tb));
1540                 }
1541         }
1542
1543         catch (...)
1544         {
1545                 ServerInstance->Log(DEBUG,"Exception in userrec::WriteCommon()");
1546         }
1547 }
1548
1549
1550 /* write a formatted string to all users who share at least one common
1551  * channel, NOT including the source user e.g. for use in QUIT
1552  */
1553
1554 void userrec::WriteCommonExcept(const char* text, ...)
1555 {
1556         char textbuffer[MAXBUF];
1557         va_list argsPtr;
1558
1559         va_start(argsPtr, text);
1560         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1561         va_end(argsPtr);
1562
1563         this->WriteCommonExcept(std::string(textbuffer));
1564 }
1565
1566 void userrec::WriteCommonQuit(const std::string &normal_text, const std::string &oper_text)
1567 {
1568         char tb1[MAXBUF];
1569         char tb2[MAXBUF];
1570
1571         if (this->registered != REG_ALL)
1572                 return;
1573
1574         uniq_id++;
1575         snprintf(tb1,MAXBUF,":%s QUIT :%s",this->GetFullHost(),normal_text.c_str());
1576         snprintf(tb2,MAXBUF,":%s QUIT :%s",this->GetFullHost(),oper_text.c_str());
1577         std::string out1 = tb1;
1578         std::string out2 = tb2;
1579
1580         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1581         {
1582                 CUList *ulist = v->first->GetUsers();
1583                 for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1584                 {
1585                         if (this != i->first)
1586                         {
1587                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1588                                 {
1589                                         already_sent[i->first->fd] = uniq_id;
1590                                         i->first->Write(IS_OPER(i->first) ? out2 : out1);
1591                                 }
1592                         }
1593                 }
1594         }
1595 }
1596
1597 void userrec::WriteCommonExcept(const std::string &text)
1598 {
1599         char tb1[MAXBUF];
1600         std::string out1;
1601
1602         if (this->registered != REG_ALL)
1603                 return;
1604
1605         uniq_id++;
1606         snprintf(tb1,MAXBUF,":%s %s",this->GetFullHost(),text.c_str());
1607         out1 = tb1;
1608
1609         for (UCListIter v = this->chans.begin(); v != this->chans.end(); v++)
1610         {
1611                 CUList *ulist = v->first->GetUsers();
1612                 for (CUList::iterator i = ulist->begin(); i != ulist->end(); i++)
1613                 {
1614                         if (this != i->first)
1615                         {
1616                                 if ((IS_LOCAL(i->first)) && (already_sent[i->first->fd] != uniq_id))
1617                                 {
1618                                         already_sent[i->first->fd] = uniq_id;
1619                                         i->first->Write(out1);
1620                                 }
1621                         }
1622                 }
1623         }
1624
1625 }
1626
1627 void userrec::WriteWallOps(const std::string &text)
1628 {
1629         if (!IS_OPER(this) && IS_LOCAL(this))
1630                 return;
1631
1632         std::string wallop("WALLOPS :");
1633         wallop.append(text);
1634
1635         for (std::vector<userrec*>::const_iterator i = ServerInstance->local_users.begin(); i != ServerInstance->local_users.end(); i++)
1636         {
1637                 userrec* t = *i;
1638                 if (t->IsModeSet('w'))
1639                         this->WriteTo(t,wallop);
1640         }
1641 }
1642
1643 void userrec::WriteWallOps(const char* text, ...)
1644 {
1645         char textbuffer[MAXBUF];
1646         va_list argsPtr;
1647
1648         va_start(argsPtr, text);
1649         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1650         va_end(argsPtr);
1651
1652         this->WriteWallOps(std::string(textbuffer));
1653 }
1654
1655 /* return 0 or 1 depending if users u and u2 share one or more common channels
1656  * (used by QUIT, NICK etc which arent channel specific notices)
1657  *
1658  * The old algorithm in 1.0 for this was relatively inefficient, iterating over
1659  * the first users channels then the second users channels within the outer loop,
1660  * therefore it was a maximum of x*y iterations (upon returning 0 and checking
1661  * all possible iterations). However this new function instead checks against the
1662  * channel's userlist in the inner loop which is a std::map<userrec*,userrec*>
1663  * and saves us time as we already know what pointer value we are after.
1664  * Don't quote me on the maths as i am not a mathematician or computer scientist,
1665  * but i believe this algorithm is now x+(log y) maximum iterations instead.
1666  */
1667 bool userrec::SharesChannelWith(userrec *other)
1668 {
1669         if ((!other) || (this->registered != REG_ALL) || (other->registered != REG_ALL))
1670                 return false;
1671
1672         /* Outer loop */
1673         for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1674         {
1675                 /* Eliminate the inner loop (which used to be ~equal in size to the outer loop)
1676                  * by replacing it with a map::find which *should* be more efficient
1677                  */
1678                 if (i->first->HasUser(other))
1679                         return true;
1680         }
1681         return false;
1682 }
1683
1684 bool userrec::ChangeName(const char* gecos)
1685 {
1686         if (!strcmp(gecos, this->fullname))
1687                 return true;
1688
1689         if (IS_LOCAL(this))
1690         {
1691                 int MOD_RESULT = 0;
1692                 FOREACH_RESULT(I_OnChangeLocalUserGECOS,OnChangeLocalUserGECOS(this,gecos));
1693                 if (MOD_RESULT)
1694                         return false;
1695                 FOREACH_MOD(I_OnChangeName,OnChangeName(this,gecos));
1696         }
1697         strlcpy(this->fullname,gecos,MAXGECOS+1);
1698
1699         return true;
1700 }
1701
1702 bool userrec::ChangeDisplayedHost(const char* host)
1703 {
1704         if (!strcmp(host, this->dhost))
1705                 return true;
1706
1707         if (IS_LOCAL(this))
1708         {
1709                 int MOD_RESULT = 0;
1710                 FOREACH_RESULT(I_OnChangeLocalUserHost,OnChangeLocalUserHost(this,host));
1711                 if (MOD_RESULT)
1712                         return false;
1713                 FOREACH_MOD(I_OnChangeHost,OnChangeHost(this,host));
1714         }
1715         if (this->ServerInstance->Config->CycleHosts)
1716                 this->WriteCommonExcept("QUIT :Changing hosts");
1717
1718         /* Fix by Om: userrec::dhost is 65 long, this was truncating some long hosts */
1719         strlcpy(this->dhost,host,64);
1720
1721         this->InvalidateCache();
1722
1723         if (this->ServerInstance->Config->CycleHosts)
1724         {
1725                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1726                 {
1727                         i->first->WriteAllExceptSender(this, false, 0, "JOIN %s", i->first->name);
1728                         std::string n = this->ServerInstance->Modes->ModeString(this, i->first);
1729                         if (n.length() > 0)
1730                                 i->first->WriteAllExceptSender(this, true, 0, "MODE %s +%s", i->first->name, n.c_str());
1731                 }
1732         }
1733
1734         if (IS_LOCAL(this))
1735                 this->WriteServ("396 %s %s :is now your displayed host",this->nick,this->dhost);
1736
1737         return true;
1738 }
1739
1740 bool userrec::ChangeIdent(const char* newident)
1741 {
1742         if (!strcmp(newident, this->ident))
1743                 return true;
1744
1745         if (this->ServerInstance->Config->CycleHosts)
1746                 this->WriteCommonExcept("%s","QUIT :Changing ident");
1747
1748         strlcpy(this->ident, newident, IDENTMAX+2);
1749
1750         this->InvalidateCache();
1751
1752         if (this->ServerInstance->Config->CycleHosts)
1753         {
1754                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1755                 {
1756                         i->first->WriteAllExceptSender(this, false, 0, "JOIN %s", i->first->name);
1757                         std::string n = this->ServerInstance->Modes->ModeString(this, i->first);
1758                         if (n.length() > 0)
1759                                 i->first->WriteAllExceptSender(this, true, 0, "MODE %s +%s", i->first->name, n.c_str());
1760                 }
1761         }
1762
1763         return true;
1764 }
1765
1766 void userrec::SendAll(const char* command, char* text, ...)
1767 {
1768         char textbuffer[MAXBUF];
1769         char formatbuffer[MAXBUF];
1770         va_list argsPtr;
1771
1772         va_start(argsPtr, text);
1773         vsnprintf(textbuffer, MAXBUF, text, argsPtr);
1774         va_end(argsPtr);
1775
1776         snprintf(formatbuffer,MAXBUF,":%s %s $* :%s", this->GetFullHost(), command, textbuffer);
1777         std::string fmt = formatbuffer;
1778
1779         for (std::vector<userrec*>::const_iterator i = ServerInstance->local_users.begin(); i != ServerInstance->local_users.end(); i++)
1780         {
1781                 (*i)->Write(fmt);
1782         }
1783 }
1784
1785
1786 std::string userrec::ChannelList(userrec* source)
1787 {
1788         try
1789         {
1790                 std::string list;
1791                 for (UCListIter i = this->chans.begin(); i != this->chans.end(); i++)
1792                 {
1793                         /* If the target is the same as the sender, let them see all their channels.
1794                          * If the channel is NOT private/secret OR the user shares a common channel
1795                          * If the user is an oper, and the <options:operspywhois> option is set.
1796                          */
1797                         if ((source == this) || (IS_OPER(source) && ServerInstance->Config->OperSpyWhois) || (((!i->first->IsModeSet('p')) && (!i->first->IsModeSet('s'))) || (i->first->HasUser(source))))
1798                         {
1799                                 list.append(i->first->GetPrefixChar(this)).append(i->first->name).append(" ");
1800                         }
1801                 }
1802                 return list;
1803         }
1804         catch (...)
1805         {
1806                 ServerInstance->Log(DEBUG,"Exception in userrec::ChannelList()");
1807                 return "";
1808         }
1809 }
1810
1811 void userrec::SplitChanList(userrec* dest, const std::string &cl)
1812 {
1813         std::string line;
1814         std::ostringstream prefix;
1815         std::string::size_type start, pos, length;
1816
1817         try
1818         {
1819                 prefix << this->nick << " " << dest->nick << " :";
1820                 line = prefix.str();
1821                 int namelen = strlen(ServerInstance->Config->ServerName) + 6;
1822
1823                 for (start = 0; (pos = cl.find(' ', start)) != std::string::npos; start = pos+1)
1824                 {
1825                         length = (pos == std::string::npos) ? cl.length() : pos;
1826
1827                         if (line.length() + namelen + length - start > 510)
1828                         {
1829                                 ServerInstance->SendWhoisLine(this, dest, 319, "%s", line.c_str());
1830                                 line = prefix.str();
1831                         }
1832
1833                         if(pos == std::string::npos)
1834                         {
1835                                 line.append(cl.substr(start, length - start));
1836                                 break;
1837                         }
1838                         else
1839                         {
1840                                 line.append(cl.substr(start, length - start + 1));
1841                         }
1842                 }
1843
1844                 if (line.length())
1845                 {
1846                         ServerInstance->SendWhoisLine(this, dest, 319, "%s", line.c_str());
1847                 }
1848         }
1849
1850         catch (...)
1851         {
1852                 ServerInstance->Log(DEBUG,"Exception in userrec::SplitChanList()");
1853         }
1854 }
1855
1856
1857 /* looks up a users password for their connection class (<ALLOW>/<DENY> tags)
1858  * NOTE: If the <ALLOW> or <DENY> tag specifies an ip, and this user resolves,
1859  * then their ip will be taken as 'priority' anyway, so for example,
1860  * <connect allow="127.0.0.1"> will match joe!bloggs@localhost
1861  */
1862 ConnectClass* userrec::GetClass()
1863 {
1864         for (ClassVector::iterator i = ServerInstance->Config->Classes.begin(); i != ServerInstance->Config->Classes.end(); i++)
1865         {
1866                 if (((match(this->GetIPString(),i->GetHost().c_str(),true)) || (match(this->host,i->GetHost().c_str()))))
1867                 {
1868                         if (i->GetPort())
1869                         {
1870                                 if (this->GetPort() == i->GetPort())
1871                                         return &(*i);
1872                                 else
1873                                         continue;
1874                         }
1875                         else
1876                                 return &(*i);
1877                 }
1878         }
1879         return NULL;
1880 }
1881
1882 void userrec::PurgeEmptyChannels()
1883 {
1884         std::vector<chanrec*> to_delete;
1885
1886         // firstly decrement the count on each channel
1887         for (UCListIter f = this->chans.begin(); f != this->chans.end(); f++)
1888         {
1889                 f->first->RemoveAllPrefixes(this);
1890                 if (f->first->DelUser(this) == 0)
1891                 {
1892                         /* No users left in here, mark it for deletion */
1893                         try
1894                         {
1895                                 to_delete.push_back(f->first);
1896                         }
1897                         catch (...)
1898                         {
1899                                 ServerInstance->Log(DEBUG,"Exception in userrec::PurgeEmptyChannels to_delete.push_back()");
1900                         }
1901                 }
1902         }
1903
1904         for (std::vector<chanrec*>::iterator n = to_delete.begin(); n != to_delete.end(); n++)
1905         {
1906                 chanrec* thischan = *n;
1907                 chan_hash::iterator i2 = ServerInstance->chanlist->find(thischan->name);
1908                 if (i2 != ServerInstance->chanlist->end())
1909                 {
1910                         FOREACH_MOD(I_OnChannelDelete,OnChannelDelete(i2->second));
1911                         DELETE(i2->second);
1912                         ServerInstance->chanlist->erase(i2);
1913                         this->chans.erase(*n);
1914                 }
1915         }
1916
1917         this->UnOper();
1918 }
1919
1920 void userrec::ShowMOTD()
1921 {
1922         if (!ServerInstance->Config->MOTD.size())
1923         {
1924                 this->WriteServ("422 %s :Message of the day file is missing.",this->nick);
1925                 return;
1926         }
1927         this->WriteServ("375 %s :%s message of the day", this->nick, ServerInstance->Config->ServerName);
1928
1929         for (file_cache::iterator i = ServerInstance->Config->MOTD.begin(); i != ServerInstance->Config->MOTD.end(); i++)
1930                 this->WriteServ("372 %s :- %s",this->nick,i->c_str());
1931
1932         this->WriteServ("376 %s :End of message of the day.", this->nick);
1933 }
1934
1935 void userrec::ShowRULES()
1936 {
1937         if (!ServerInstance->Config->RULES.size())
1938         {
1939                 this->WriteServ("434 %s :RULES File is missing",this->nick);
1940                 return;
1941         }
1942
1943         this->WriteServ("308 %s :- %s Server Rules -",this->nick,ServerInstance->Config->ServerName);
1944
1945         for (file_cache::iterator i = ServerInstance->Config->RULES.begin(); i != ServerInstance->Config->RULES.end(); i++)
1946                 this->WriteServ("232 %s :- %s",this->nick,i->c_str());
1947
1948         this->WriteServ("309 %s :End of RULES command.",this->nick);
1949 }
1950
1951 void userrec::HandleEvent(EventType et, int errornum)
1952 {
1953         /* WARNING: May delete this user! */
1954         int thisfd = this->GetFd();
1955
1956         try
1957         {
1958                 switch (et)
1959                 {
1960                         case EVENT_READ:
1961                                 ServerInstance->ProcessUser(this);
1962                         break;
1963                         case EVENT_WRITE:
1964                                 this->FlushWriteBuf();
1965                         break;
1966                         case EVENT_ERROR:
1967                                 /** This should be safe, but dont DARE do anything after it -- Brain */
1968                                 this->SetWriteError(errornum ? strerror(errornum) : "EOF from client");
1969                         break;
1970                 }
1971         }
1972         catch (...)
1973         {
1974                 ServerInstance->Log(DEBUG,"Exception in userrec::HandleEvent intercepted");
1975         }
1976
1977         /* If the user has raised an error whilst being processed, quit them now we're safe to */
1978         if ((ServerInstance->SE->GetRef(thisfd) == this))
1979         {
1980                 if (!WriteError.empty())
1981                 {
1982                         userrec::QuitUser(ServerInstance, this, GetWriteError());
1983                 }
1984         }
1985 }
1986
1987 void userrec::SetOperQuit(const std::string &oquit)
1988 {
1989         if (operquit)
1990                 return;
1991
1992         operquit = strdup(oquit.c_str());
1993 }
1994
1995 const char* userrec::GetOperQuit()
1996 {
1997         return operquit ? operquit : "";
1998 }
1999
2000 VisData::VisData()
2001 {
2002 }
2003
2004 VisData::~VisData()
2005 {
2006 }
2007
2008 bool VisData::VisibleTo(userrec* user)
2009 {
2010         return true;
2011 }
2012