2 * ircd-ratbox: A slightly useful ircd
3 * reject.c: reject users with prejudice
5 * Copyright (C) 2003 Aaron Sethman <androsyn@ratbox.org>
6 * Copyright (C) 2003-2005 ircd-ratbox development team
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
37 static rb_patricia_tree_t
*reject_tree
;
38 static rb_dlink_list delay_exit
;
39 static rb_dlink_list reject_list
;
40 static rb_dlink_list throttle_list
;
41 static rb_patricia_tree_t
*throttle_tree
;
42 static void throttle_expires(void *unused
);
45 typedef struct _reject_data
53 typedef struct _delay_data
59 typedef struct _throttle
67 delay_exit_length(void)
69 return rb_dlink_list_length(&delay_exit
);
73 reject_exit(void *unused
)
75 rb_dlink_node
*ptr
, *ptr_next
;
77 static const char *errbuf
= "ERROR :Closing Link: (*** Banned (cache))\r\n";
79 RB_DLINK_FOREACH_SAFE(ptr
, ptr_next
, delay_exit
.head
)
83 rb_write(ddata
->F
, errbuf
, strlen(errbuf
));
88 delay_exit
.head
= delay_exit
.tail
= NULL
;
89 delay_exit
.length
= 0;
93 reject_expires(void *unused
)
95 rb_dlink_node
*ptr
, *next
;
96 rb_patricia_node_t
*pnode
;
99 RB_DLINK_FOREACH_SAFE(ptr
, next
, reject_list
.head
)
104 if(rdata
->time
+ ConfigFileEntry
.reject_duration
> rb_current_time())
107 rb_dlinkDelete(ptr
, &reject_list
);
109 rb_patricia_remove(reject_tree
, pnode
);
116 reject_tree
= rb_new_patricia(PATRICIA_BITS
);
117 throttle_tree
= rb_new_patricia(PATRICIA_BITS
);
118 rb_event_add("reject_exit", reject_exit
, NULL
, DELAYED_EXIT_TIME
);
119 rb_event_add("reject_expires", reject_expires
, NULL
, 60);
120 rb_event_add("throttle_expires", throttle_expires
, NULL
, 10);
128 rb_patricia_node_t
*pnode
;
132 RB_DLINK_FOREACH(ptr
, throttle_list
.head
)
136 if (t
->count
> ConfigFileEntry
.throttle_count
)
144 add_reject(struct Client
*client_p
, const char *mask1
, const char *mask2
)
146 rb_patricia_node_t
*pnode
;
150 /* Reject is disabled */
151 if(ConfigFileEntry
.reject_after_count
== 0 || ConfigFileEntry
.reject_duration
== 0)
156 hashv
^= fnv_hash_upper((const unsigned char *)mask1
, 32);
158 hashv
^= fnv_hash_upper((const unsigned char *)mask2
, 32);
160 if((pnode
= rb_match_ip(reject_tree
, (struct sockaddr
*)&client_p
->localClient
->ip
)) != NULL
)
163 rdata
->time
= rb_current_time();
169 if(GET_SS_FAMILY(&client_p
->localClient
->ip
) == AF_INET6
)
171 pnode
= make_and_lookup_ip(reject_tree
, (struct sockaddr
*)&client_p
->localClient
->ip
, bitlen
);
172 pnode
->data
= rdata
= rb_malloc(sizeof(reject_t
));
173 rb_dlinkAddTail(pnode
, &rdata
->rnode
, &reject_list
);
174 rdata
->time
= rb_current_time();
177 rdata
->mask_hashv
= hashv
;
181 check_reject(rb_fde_t
*F
, struct sockaddr
*addr
)
183 rb_patricia_node_t
*pnode
;
186 /* Reject is disabled */
187 if(ConfigFileEntry
.reject_after_count
== 0 || ConfigFileEntry
.reject_duration
== 0)
190 pnode
= rb_match_ip(reject_tree
, addr
);
195 rdata
->time
= rb_current_time();
196 if(rdata
->count
> (unsigned long)ConfigFileEntry
.reject_after_count
)
198 ddata
= rb_malloc(sizeof(delay_t
));
199 ServerStats
.is_rej
++;
200 rb_setselect(F
, RB_SELECT_WRITE
| RB_SELECT_READ
, NULL
, NULL
);
202 rb_dlinkAdd(ddata
, &ddata
->node
, &delay_exit
);
206 /* Caller does what it wants */
211 is_reject_ip(struct sockaddr
*addr
)
213 rb_patricia_node_t
*pnode
;
217 /* Reject is disabled */
218 if(ConfigFileEntry
.reject_after_count
== 0 || ConfigFileEntry
.reject_duration
== 0)
221 pnode
= rb_match_ip(reject_tree
, addr
);
226 if(rdata
->count
> (unsigned long)ConfigFileEntry
.reject_after_count
)
228 duration
= rdata
->time
+ ConfigFileEntry
.reject_duration
- rb_current_time();
229 return duration
> 0 ? duration
: 1;
238 rb_dlink_node
*ptr
, *next
;
239 rb_patricia_node_t
*pnode
;
242 RB_DLINK_FOREACH_SAFE(ptr
, next
, reject_list
.head
)
246 rb_dlinkDelete(ptr
, &reject_list
);
248 rb_patricia_remove(reject_tree
, pnode
);
253 remove_reject_ip(const char *ip
)
255 rb_patricia_node_t
*pnode
;
257 /* Reject is disabled */
258 if(ConfigFileEntry
.reject_after_count
== 0 || ConfigFileEntry
.reject_duration
== 0)
261 if((pnode
= rb_match_string(reject_tree
, ip
)) != NULL
)
263 reject_t
*rdata
= pnode
->data
;
264 rb_dlinkDelete(&rdata
->rnode
, &reject_list
);
266 rb_patricia_remove(reject_tree
, pnode
);
273 remove_reject_mask(const char *mask1
, const char *mask2
)
275 rb_dlink_node
*ptr
, *next
;
276 rb_patricia_node_t
*pnode
;
283 hashv
^= fnv_hash_upper((const unsigned char *)mask1
, 32);
285 hashv
^= fnv_hash_upper((const unsigned char *)mask2
, 32);
286 RB_DLINK_FOREACH_SAFE(ptr
, next
, reject_list
.head
)
290 if (rdata
->mask_hashv
== hashv
)
292 rb_dlinkDelete(ptr
, &reject_list
);
294 rb_patricia_remove(reject_tree
, pnode
);
302 throttle_add(struct sockaddr
*addr
)
305 rb_patricia_node_t
*pnode
;
307 if((pnode
= rb_match_ip(throttle_tree
, addr
)) != NULL
)
311 if(t
->count
> ConfigFileEntry
.throttle_count
)
313 ServerStats
.is_thr
++;
316 /* Stop penalizing them after they've been throttled */
317 t
->last
= rb_current_time();
322 if(GET_SS_FAMILY(addr
) == AF_INET6
)
324 t
= rb_malloc(sizeof(throttle_t
));
325 t
->last
= rb_current_time();
327 pnode
= make_and_lookup_ip(throttle_tree
, addr
, bitlen
);
329 rb_dlinkAdd(pnode
, &t
->node
, &throttle_list
);
335 is_throttle_ip(struct sockaddr
*addr
)
338 rb_patricia_node_t
*pnode
;
341 if((pnode
= rb_match_ip(throttle_tree
, addr
)) != NULL
)
344 if(t
->count
> ConfigFileEntry
.throttle_count
)
346 duration
= t
->last
+ ConfigFileEntry
.throttle_duration
- rb_current_time();
347 return duration
> 0 ? duration
: 1;
356 rb_dlink_node
*ptr
, *next
;
357 rb_patricia_node_t
*pnode
;
360 RB_DLINK_FOREACH_SAFE(ptr
, next
, throttle_list
.head
)
365 rb_dlinkDelete(ptr
, &throttle_list
);
367 rb_patricia_remove(throttle_tree
, pnode
);
372 throttle_expires(void *unused
)
374 rb_dlink_node
*ptr
, *next
;
375 rb_patricia_node_t
*pnode
;
378 RB_DLINK_FOREACH_SAFE(ptr
, next
, throttle_list
.head
)
383 if(t
->last
+ ConfigFileEntry
.throttle_duration
> rb_current_time())
386 rb_dlinkDelete(ptr
, &throttle_list
);
388 rb_patricia_remove(throttle_tree
, pnode
);