]> jfr.im git - irc/rqf/shadowircd.git/blame - src/packet.c
Oups, change for last commit
[irc/rqf/shadowircd.git] / src / packet.c
CommitLineData
21c9d815
VY
1/*
2 * ircd-ratbox: A slightly useful ircd.
3 * packet.c: Packet handlers.
4 *
5 * Copyright (C) 1990 Jarkko Oikarinen and University of Oulu, Co Center
6 * Copyright (C) 1996-2002 Hybrid Development Team
7 * Copyright (C) 2002-2005 ircd-ratbox development team
8 *
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
18 *
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
22 * USA
23 *
24 * $Id: packet.c 3446 2007-05-14 22:21:16Z jilles $
25 */
26#include "stdinc.h"
27#include "s_conf.h"
28#include "s_serv.h"
29#include "client.h"
30#include "common.h"
31#include "ircd.h"
32#include "parse.h"
33#include "packet.h"
34#include "irc_string.h"
35#include "hook.h"
36#include "send.h"
37
38static char readBuf[READBUF_SIZE];
39static void client_dopacket(struct Client *client_p, char *buffer, size_t length);
40
41
42/*
43 * parse_client_queued - parse client queued messages
44 */
45static void
46parse_client_queued(struct Client *client_p)
47{
48 int dolen = 0;
49 int checkflood = 1;
50
51 if(IsAnyDead(client_p))
52 return;
53
54 if(IsUnknown(client_p))
55 {
56 for (;;)
57 {
58 if(client_p->localClient->sent_parsed >= client_p->localClient->allow_read)
59 break;
60
61 dolen = linebuf_get(&client_p->localClient->
62 buf_recvq, readBuf, READBUF_SIZE,
63 LINEBUF_COMPLETE, LINEBUF_PARSED);
64
65 if(dolen <= 0 || IsDead(client_p))
66 break;
67
68 client_dopacket(client_p, readBuf, dolen);
69 client_p->localClient->sent_parsed++;
70
71 /* He's dead cap'n */
72 if(IsAnyDead(client_p))
73 return;
74 /* if theyve dropped out of the unknown state, break and move
75 * to the parsing for their appropriate status. --fl
76 */
77 if(!IsUnknown(client_p))
78 {
79 /* reset their flood limits, they're now
80 * graced to flood
81 */
82 client_p->localClient->sent_parsed = 0;
83 break;
84 }
85
86 }
87 }
88
89 if(IsAnyServer(client_p) || IsExemptFlood(client_p))
90 {
91 while (!IsAnyDead(client_p) && (dolen = linebuf_get(&client_p->localClient->buf_recvq,
92 readBuf, READBUF_SIZE, LINEBUF_COMPLETE,
93 LINEBUF_PARSED)) > 0)
94 {
95 client_dopacket(client_p, readBuf, dolen);
96 }
97 }
98 else if(IsClient(client_p))
99 {
100
101 if(IsOper(client_p) && ConfigFileEntry.no_oper_flood)
102 checkflood = 0;
103 /*
104 * Handle flood protection here - if we exceed our flood limit on
105 * messages in this loop, we simply drop out of the loop prematurely.
106 * -- adrian
107 */
108 for (;;)
109 {
110 /* This flood protection works as follows:
111 *
112 * A client is given allow_read lines to send to the server. Every
113 * time a line is parsed, sent_parsed is increased. sent_parsed
114 * is decreased by 1 every time flood_recalc is called.
115 *
116 * Thus a client can 'burst' allow_read lines to the server, any
117 * excess lines will be parsed one per flood_recalc() call.
118 *
119 * Therefore a client will be penalised more if they keep flooding,
120 * as sent_parsed will always hover around the allow_read limit
121 * and no 'bursts' will be permitted.
122 */
123 if(checkflood)
124 {
125 if(client_p->localClient->sent_parsed >= client_p->localClient->allow_read)
126 break;
127 }
128
129 /* allow opers 4 times the amount of messages as users. why 4?
130 * why not. :) --fl_
131 */
132 else if(client_p->localClient->sent_parsed >= (4 * client_p->localClient->allow_read))
133 break;
134
135 dolen = linebuf_get(&client_p->localClient->
136 buf_recvq, readBuf, READBUF_SIZE,
137 LINEBUF_COMPLETE, LINEBUF_PARSED);
138
139 if(!dolen)
140 break;
141
142 client_dopacket(client_p, readBuf, dolen);
143 if(IsAnyDead(client_p))
144 return;
145 client_p->localClient->sent_parsed++;
146 }
147 }
148}
149
150/* flood_endgrace()
151 *
152 * marks the end of the clients grace period
153 */
154void
155flood_endgrace(struct Client *client_p)
156{
157 SetFloodDone(client_p);
158
159 /* Drop their flood limit back down */
160 client_p->localClient->allow_read = MAX_FLOOD;
161
162 /* sent_parsed could be way over MAX_FLOOD but under MAX_FLOOD_BURST,
163 * so reset it.
164 */
165 client_p->localClient->sent_parsed = 0;
166}
167
168/*
169 * flood_recalc
170 *
171 * recalculate the number of allowed flood lines. this should be called
172 * once a second on any given client. We then attempt to flush some data.
173 */
174void
175flood_recalc(int fd, void *data)
176{
177 struct Client *client_p = data;
178 struct LocalUser *lclient_p = client_p->localClient;
179
180 /* This can happen in the event that the client detached. */
181 if(!lclient_p)
182 return;
183
184 /* allow a bursting client their allocation per second, allow
185 * a client whos flooding an extra 2 per second
186 */
187 if(IsFloodDone(client_p))
188 lclient_p->sent_parsed -= 2;
189 else
190 lclient_p->sent_parsed = 0;
191
192 if(lclient_p->sent_parsed < 0)
193 lclient_p->sent_parsed = 0;
194
195 if(--lclient_p->actually_read < 0)
196 lclient_p->actually_read = 0;
197
198 parse_client_queued(client_p);
199
200 if(IsAnyDead(client_p))
201 return;
202
203 /* and finally, reset the flood check */
204 rb_setflush(fd, 1000, flood_recalc, client_p);
205}
206
207/*
208 * read_ctrl_packet - Read a 'packet' of data from a servlink control
209 * link and process it.
210 */
211void
212read_ctrl_packet(int fd, void *data)
213{
214 struct Client *server = data;
215 struct LocalUser *lserver = server->localClient;
216 struct SlinkRpl *reply;
217 int length = 0;
218 unsigned char tmp[2];
219 unsigned char *len = tmp;
220 struct SlinkRplDef *replydef;
221#ifdef USE_IODEBUG_HOOKS
222 hook_data_int hdata;
223#endif
224
225 s_assert(lserver != NULL);
226 if(IsAnyDead(server))
227 return;
228
229 reply = &lserver->slinkrpl;
230
231
232 if(!reply->command)
233 {
234 reply->gotdatalen = 0;
235 reply->readdata = 0;
236 reply->data = NULL;
237
238 length = read(fd, tmp, 1);
239
240 if(length <= 0)
241 {
242 if((length == -1) && ignoreErrno(errno))
243 goto nodata;
244 error_exit_client(server, length);
245 return;
246 }
247
248 reply->command = tmp[0];
249 }
250
251 for (replydef = slinkrpltab; replydef->handler; replydef++)
252 {
253 if((int)replydef->replyid == reply->command)
254 break;
255 }
256
257 /* we should be able to trust a local slink process...
258 * and if it sends an invalid command, that's a bug.. */
259 s_assert(replydef->handler);
260
261 if((replydef->flags & SLINKRPL_FLAG_DATA) && (reply->gotdatalen < 2))
262 {
263 /* we need a datalen u16 which we don't have yet... */
264 length = read(fd, len, (2 - reply->gotdatalen));
265 if(length <= 0)
266 {
267 if((length == -1) && ignoreErrno(errno))
268 goto nodata;
269 error_exit_client(server, length);
270 return;
271 }
272
273 if(reply->gotdatalen == 0)
274 {
275 reply->datalen = *len << 8;
276 reply->gotdatalen++;
277 length--;
278 len++;
279 }
280 if(length && (reply->gotdatalen == 1))
281 {
282 reply->datalen |= *len;
283 reply->gotdatalen++;
284 if(reply->datalen > 0)
285 reply->data = rb_malloc(reply->datalen);
286 }
287
288 if(reply->gotdatalen < 2)
289 return; /* wait for more data */
290 }
291
292 if(reply->readdata < reply->datalen) /* try to get any remaining data */
293 {
294 length = read(fd, (reply->data + reply->readdata),
295 (reply->datalen - reply->readdata));
296 if(length <= 0)
297 {
298 if((length == -1) && ignoreErrno(errno))
299 goto nodata;
300 error_exit_client(server, length);
301 return;
302 }
303
304 reply->readdata += length;
305 if(reply->readdata < reply->datalen)
306 return; /* wait for more data */
307 }
308
309#ifdef USE_IODEBUG_HOOKS
310 hdata.client = server;
311 hdata.arg1 = NULL;
312 hdata.arg2 = reply->command;
313 hdata.data = NULL;
314 call_hook(h_iorecvctrl_id, &hdata);
315#endif
316
317 /* we now have the command and any data, pass it off to the handler */
318 (*replydef->handler) (reply->command, reply->datalen, reply->data, server);
319
320 /* reset SlinkRpl */
321 if(reply->datalen > 0)
322 rb_free(reply->data);
323 reply->command = 0;
324
325 if(IsAnyDead(server))
326 return;
327
328 nodata:
329 /* If we get here, we need to register for another COMM_SELECT_READ */
330 rb_setselect(fd, FDLIST_SERVER, COMM_SELECT_READ, read_ctrl_packet, server, 0);
331}
332
333/*
334 * read_packet - Read a 'packet' of data from a connection and process it.
335 */
336void
337read_packet(int fd, void *data)
338{
339 struct Client *client_p = data;
340 struct LocalUser *lclient_p = client_p->localClient;
341 int length = 0;
342 int lbuf_len;
343
344 int binary = 0;
345#ifdef USE_IODEBUG_HOOKS
346 hook_data_int hdata;
347#endif
348 if(IsAnyDead(client_p))
349 return;
350
351 /*
352 * Read some data. We *used to* do anti-flood protection here, but
353 * I personally think it makes the code too hairy to make sane.
354 * -- adrian
355 */
356 length = client_p->localClient->F->read_impl(client_p->localClient->F, readBuf, READBUF_SIZE);
357
358 if(length <= 0)
359 {
360 if((length == -1) && ignoreErrno(errno))
361 {
362 rb_setselect(client_p->localClient->F->fd, FDLIST_IDLECLIENT,
363 COMM_SELECT_READ, read_packet, client_p, 0);
364 return;
365 }
366 error_exit_client(client_p, length);
367 return;
368 }
369
370#ifdef USE_IODEBUG_HOOKS
371 hdata.client = client_p;
372 hdata.arg1 = readBuf;
373 hdata.arg2 = length;
374 call_hook(h_iorecv_id, &hdata);
375#endif
376
377 if(client_p->localClient->lasttime < rb_current_time())
378 client_p->localClient->lasttime = rb_current_time();
379 client_p->flags &= ~FLAGS_PINGSENT;
380
381 /*
382 * Before we even think of parsing what we just read, stick
383 * it on the end of the receive queue and do it when its
384 * turn comes around.
385 */
386 if(IsHandshake(client_p) || IsUnknown(client_p))
387 binary = 1;
388
389 lbuf_len = linebuf_parse(&client_p->localClient->buf_recvq, readBuf, length, binary);
390
391 lclient_p->actually_read += lbuf_len;
392
393 if(IsAnyDead(client_p))
394 return;
395
396 /* Attempt to parse what we have */
397 parse_client_queued(client_p);
398
399 if(IsAnyDead(client_p))
400 return;
401
402 /* Check to make sure we're not flooding */
403 if(!IsAnyServer(client_p) &&
404 (linebuf_alloclen(&client_p->localClient->buf_recvq) > ConfigFileEntry.client_flood))
405 {
406 if(!(ConfigFileEntry.no_oper_flood && IsOper(client_p)))
407 {
408 exit_client(client_p, client_p, client_p, "Excess Flood");
409 return;
410 }
411 }
412
413 /* If we get here, we need to register for another COMM_SELECT_READ */
414 if(PARSE_AS_SERVER(client_p))
415 {
416 rb_setselect(client_p->localClient->F->fd, FDLIST_SERVER, COMM_SELECT_READ,
417 read_packet, client_p, 0);
418 }
419 else
420 {
421 rb_setselect(client_p->localClient->F->fd, FDLIST_IDLECLIENT,
422 COMM_SELECT_READ, read_packet, client_p, 0);
423 }
424}
425
426/*
427 * client_dopacket - copy packet to client buf and parse it
428 * client_p - pointer to client structure for which the buffer data
429 * applies.
430 * buffer - pointr to the buffer containing the newly read data
431 * length - number of valid bytes of data in the buffer
432 *
433 * Note:
434 * It is implicitly assumed that dopacket is called only
435 * with client_p of "local" variation, which contains all the
436 * necessary fields (buffer etc..)
437 */
438void
439client_dopacket(struct Client *client_p, char *buffer, size_t length)
440{
441 s_assert(client_p != NULL);
442 s_assert(buffer != NULL);
443
444 if(client_p == NULL || buffer == NULL)
445 return;
446 if(IsAnyDead(client_p))
447 return;
448 /*
449 * Update messages received
450 */
451 ++me.localClient->receiveM;
452 ++client_p->localClient->receiveM;
453
454 /*
455 * Update bytes received
456 */
457 client_p->localClient->receiveB += length;
458
459 if(client_p->localClient->receiveB > 1023)
460 {
461 client_p->localClient->receiveK += (client_p->localClient->receiveB >> 10);
462 client_p->localClient->receiveB &= 0x03ff; /* 2^10 = 1024, 3ff = 1023 */
463 }
464
465 me.localClient->receiveB += length;
466
467 if(me.localClient->receiveB > 1023)
468 {
469 me.localClient->receiveK += (me.localClient->receiveB >> 10);
470 me.localClient->receiveB &= 0x03ff;
471 }
472
473 parse(client_p, buffer, buffer + length);
474}