]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/users.cpp
Customizable by-byte-size recvq stuff
[user/henk/code/inspircd.git] / src / users.cpp
1 /*       +------------------------------------+
2  *       | Inspire Internet Relay Chat Daemon |
3  *       +------------------------------------+
4  *
5  *  Inspire is copyright (C) 2002-2004 ChatSpike-Dev.
6  *                       E-mail:
7  *                <brain@chatspike.net>
8  *                <Craig@chatspike.net>
9  *     
10  * Written by Craig Edwards, Craig McLure, and others.
11  * This program is free but copyrighted software; see
12  *            the file COPYING for details.
13  *
14  * ---------------------------------------------------
15  */
16
17 #include "inspircd_config.h" 
18 #include "channels.h"
19 #include "users.h"
20 #include "inspircd.h"
21 #include <stdio.h>
22 #include "inspstring.h"
23
24 extern std::stringstream config_f;
25
26 extern time_t TIME;
27
28 userrec::userrec()
29 {
30         // the PROPER way to do it, AVOID bzero at *ALL* costs
31         strcpy(nick,"");
32         strcpy(ip,"127.0.0.1");
33         timeout = 0;
34         strcpy(ident,"");
35         strcpy(host,"");
36         strcpy(dhost,"");
37         strcpy(fullname,"");
38         strcpy(modes,"");
39         strcpy(inbuf,"");
40         strcpy(server,"");
41         strcpy(awaymsg,"");
42         strcpy(oper,"");
43         reset_due = TIME;
44         lines_in = 0;
45         fd = lastping = signon = idle_lastmsg = nping = registered = 0;
46         flood = port = bytes_in = bytes_out = cmds_in = cmds_out = 0;
47         haspassed = false;
48         dns_done = false;
49         recvq = "";
50         sendq = "";
51         strcpy(result,"");
52         for (int i = 0; i < MAXCHANS; i++)
53         {
54                 this->chans[i].channel = NULL;
55                 this->chans[i].uc_modes = 0;
56         }
57         invites.clear();
58 }
59
60
61  
62 char* userrec::GetFullHost()
63 {
64         snprintf(result,MAXBUF,"%s!%s@%s",nick,ident,dhost);
65         return result;
66 }
67
68
69 char* userrec::GetFullRealHost()
70 {
71         snprintf(result,MAXBUF,"%s!%s@%s",nick,ident,host);
72         return result;
73 }
74
75 bool userrec::IsInvited(char* channel)
76 {
77         for (InvitedList::iterator i = invites.begin(); i != invites.end(); i++)
78         {
79                 if (i->channel) {
80                         if (!strcasecmp(i->channel,channel))
81                         {
82                                 return true;
83                         }
84                 }
85         }
86         return false;
87 }
88
89 void userrec::InviteTo(char* channel)
90 {
91         Invited i;
92         strlcpy(i.channel,channel,CHANMAX);
93         invites.push_back(i);
94 }
95
96 void userrec::RemoveInvite(char* channel)
97 {
98         log(DEBUG,"Removing invites");
99         if (channel)
100         {
101                 if (invites.size())
102                 {
103                         for (InvitedList::iterator i = invites.begin(); i != invites.end(); i++)
104                         {
105                                 if (i->channel)
106                                 {
107                                         if (!strcasecmp(i->channel,channel))
108                                         {
109                                                 invites.erase(i);
110                                                 return;
111                                         }
112                                 }
113                         }
114                 }
115         }
116 }
117
118 bool userrec::HasPermission(char* command)
119 {
120         char TypeName[MAXBUF],Classes[MAXBUF],ClassName[MAXBUF],CommandList[MAXBUF];
121         char* mycmd;
122         char* savept;
123         char* savept2;
124         
125         // are they even an oper at all?
126         if (strchr(this->modes,'o'))
127         {
128                 log(DEBUG,"*** HasPermission: %s is an oper",this->nick);
129                 for (int j =0; j < ConfValueEnum("type",&config_f); j++)
130                 {
131                         ConfValue("type","name",j,TypeName,&config_f);
132                         if (!strcmp(TypeName,this->oper))
133                         {
134                                 log(DEBUG,"*** HasPermission: %s is an oper of type '%s'",this->nick,this->oper);
135                                 ConfValue("type","classes",j,Classes,&config_f);
136                                 char* myclass = strtok_r(Classes," ",&savept);
137                                 while (myclass)
138                                 {
139                                         log(DEBUG,"*** HasPermission: checking classtype '%s'",myclass);
140                                         for (int k =0; k < ConfValueEnum("class",&config_f); k++)
141                                         {
142                                                 ConfValue("class","name",k,ClassName,&config_f);
143                                                 if (!strcmp(ClassName,myclass))
144                                                 {
145                                                         ConfValue("class","commands",k,CommandList,&config_f);
146                                                         log(DEBUG,"*** HasPermission: found class named %s with commands: '%s'",ClassName,CommandList);
147                                                         
148                                                         
149                                                         mycmd = strtok_r(CommandList," ",&savept2);
150                                                         while (mycmd)
151                                                         {
152                                                                 if (!strcasecmp(mycmd,command))
153                                                                 {
154                                                                         log(DEBUG,"*** Command %s found, returning true",command);
155                                                                         return true;
156                                                                 }
157                                                                 mycmd = strtok_r(NULL," ",&savept2);
158                                                         }
159                                                 }
160                                         }
161                                         myclass = strtok_r(NULL," ",&savept);
162                                 }
163                         }
164                 }
165         }
166         return false;
167 }
168
169
170 bool userrec::AddBuffer(std::string a)
171 {
172         std::string b = "";
173         for (int i = 0; i < a.length(); i++)
174                 if ((a[i] != '\r') && (a[i] != '\0') && (a[i] != 7))
175                         b = b + a[i];
176         std::stringstream stream(recvq);
177         stream << b;
178         recvq = stream.str();
179         int i = 0;
180         // count the size of the first line in the buffer.
181         while (i < recvq.length())
182         {
183                 if (recvq[i++] == '\n')
184                         break;
185         }
186         if (recvq.length() > this->recvqmax)
187         {
188                 this->SetWriteError("RecvQ exceeded");
189                 WriteOpers("*** User %s RecvQ of %d exceeds connect class maximum of %d",this->nick,recvq.length(),this->recvqmax);
190         }
191         // return false if we've had more than 600 characters WITHOUT
192         // a carriage return (this is BAD, drop the socket)
193         return (i < 600);
194 }
195
196 bool userrec::BufferIsReady()
197 {
198         for (int i = 0; i < recvq.length(); i++)
199                 if (recvq[i] == '\n')
200                         return true;
201         return false;
202 }
203
204 void userrec::ClearBuffer()
205 {
206         recvq = "";
207 }
208
209 std::string userrec::GetBuffer()
210 {
211         if (recvq == "")
212                 return "";
213         char* line = (char*)recvq.c_str();
214         std::string ret = "";
215         while ((*line != '\n') && (strlen(line)))
216         {
217                 ret = ret + *line;
218                 line++;
219         }
220         if ((*line == '\n') || (*line == '\r'))
221                 line++;
222         recvq = line;
223         return ret;
224 }
225
226 void userrec::AddWriteBuf(std::string data)
227 {
228         if (this->GetWriteError() != "")
229                 return;
230         if (sendq.length() + data.length() > this->sendqmax)
231         {
232                 WriteOpers("*** User %s SendQ of %d exceeds connect class maximum of %d",this->nick,sendq.length() + data.length(),this->sendqmax);
233                 this->SetWriteError("SendQ exceeded");
234                 return;
235         }
236         std::stringstream stream;
237         stream << sendq << data;
238         sendq = stream.str();
239 }
240
241 // send AS MUCH OF THE USERS SENDQ as we are able to (might not be all of it)
242 void userrec::FlushWriteBuf()
243 {
244         if (sendq.length())
245         {
246                 char* tb = (char*)this->sendq.c_str();
247                 int n_sent = write(this->fd,tb,this->sendq.length());
248                 if (n_sent == -1)
249                 {
250                         this->SetWriteError(strerror(errno));
251                 }
252                 else
253                 {
254                         // advance the queue
255                         tb += n_sent;
256                         this->sendq = tb;
257                         // update the user's stats counters
258                         this->bytes_out += n_sent;
259                         this->cmds_out++;
260                 }
261         }
262 }
263
264 void userrec::SetWriteError(std::string error)
265 {
266         log(DEBUG,"Setting error string for %s to '%s'",this->nick,error.c_str());
267         // don't try to set the error twice, its already set take the first string.
268         if (this->WriteError == "")
269                 this->WriteError = error;
270 }
271
272 std::string userrec::GetWriteError()
273 {
274         return this->WriteError;
275 }