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
23 * $Id: reject.c 25119 2008-03-13 16:57:05Z androsyn $
39 static rb_patricia_tree_t
*reject_tree
;
40 static rb_dlink_list delay_exit
;
41 static rb_dlink_list reject_list
;
42 static rb_dlink_list throttle_list
;
43 static rb_patricia_tree_t
*throttle_tree
;
44 static void throttle_expires(void *unused
);
47 typedef struct _reject_data
55 typedef struct _delay_data
61 typedef struct _throttle
69 delay_exit_length(void)
71 return rb_dlink_list_length(&delay_exit
);
75 reject_exit(void *unused
)
77 rb_dlink_node
*ptr
, *ptr_next
;
79 static const char *errbuf
= "ERROR :Closing Link: (*** Banned (cache))\r\n";
81 RB_DLINK_FOREACH_SAFE(ptr
, ptr_next
, delay_exit
.head
)
85 rb_write(ddata
->F
, errbuf
, strlen(errbuf
));
90 delay_exit
.head
= delay_exit
.tail
= NULL
;
91 delay_exit
.length
= 0;
95 reject_expires(void *unused
)
97 rb_dlink_node
*ptr
, *next
;
98 rb_patricia_node_t
*pnode
;
101 RB_DLINK_FOREACH_SAFE(ptr
, next
, reject_list
.head
)
106 if(rdata
->time
+ ConfigFileEntry
.reject_duration
> rb_current_time())
109 rb_dlinkDelete(ptr
, &reject_list
);
111 rb_patricia_remove(reject_tree
, pnode
);
118 reject_tree
= rb_new_patricia(PATRICIA_BITS
);
119 throttle_tree
= rb_new_patricia(PATRICIA_BITS
);
120 rb_event_add("reject_exit", reject_exit
, NULL
, DELAYED_EXIT_TIME
);
121 rb_event_add("reject_expires", reject_expires
, NULL
, 60);
122 rb_event_add("throttle_expires", throttle_expires
, NULL
, 10);
127 add_reject(struct Client
*client_p
, const char *mask1
, const char *mask2
)
129 rb_patricia_node_t
*pnode
;
133 /* Reject is disabled */
134 if(ConfigFileEntry
.reject_after_count
== 0 || ConfigFileEntry
.reject_duration
== 0)
139 hashv
^= fnv_hash_upper((const unsigned char *)mask1
, 32);
141 hashv
^= fnv_hash_upper((const unsigned char *)mask2
, 32);
143 if((pnode
= rb_match_ip(reject_tree
, (struct sockaddr
*)&client_p
->localClient
->ip
)) != NULL
)
146 rdata
->time
= rb_current_time();
153 if(GET_SS_FAMILY(&client_p
->localClient
->ip
) == AF_INET6
)
156 pnode
= make_and_lookup_ip(reject_tree
, (struct sockaddr
*)&client_p
->localClient
->ip
, bitlen
);
157 pnode
->data
= rdata
= rb_malloc(sizeof(reject_t
));
158 rb_dlinkAddTail(pnode
, &rdata
->rnode
, &reject_list
);
159 rdata
->time
= rb_current_time();
162 rdata
->mask_hashv
= hashv
;
166 check_reject(rb_fde_t
*F
, struct sockaddr
*addr
)
168 rb_patricia_node_t
*pnode
;
171 /* Reject is disabled */
172 if(ConfigFileEntry
.reject_after_count
== 0 || ConfigFileEntry
.reject_duration
== 0)
175 pnode
= rb_match_ip(reject_tree
, addr
);
180 rdata
->time
= rb_current_time();
181 if(rdata
->count
> (unsigned long)ConfigFileEntry
.reject_after_count
)
183 ddata
= rb_malloc(sizeof(delay_t
));
184 ServerStats
.is_rej
++;
185 rb_setselect(F
, RB_SELECT_WRITE
| RB_SELECT_READ
, NULL
, NULL
);
187 rb_dlinkAdd(ddata
, &ddata
->node
, &delay_exit
);
191 /* Caller does what it wants */
198 rb_dlink_node
*ptr
, *next
;
199 rb_patricia_node_t
*pnode
;
202 RB_DLINK_FOREACH_SAFE(ptr
, next
, reject_list
.head
)
206 rb_dlinkDelete(ptr
, &reject_list
);
208 rb_patricia_remove(reject_tree
, pnode
);
213 remove_reject_ip(const char *ip
)
215 rb_patricia_node_t
*pnode
;
217 /* Reject is disabled */
218 if(ConfigFileEntry
.reject_after_count
== 0 || ConfigFileEntry
.reject_duration
== 0)
221 if((pnode
= rb_match_string(reject_tree
, ip
)) != NULL
)
223 reject_t
*rdata
= pnode
->data
;
224 rb_dlinkDelete(&rdata
->rnode
, &reject_list
);
226 rb_patricia_remove(reject_tree
, pnode
);
233 remove_reject_mask(const char *mask1
, const char *mask2
)
235 rb_dlink_node
*ptr
, *next
;
236 rb_patricia_node_t
*pnode
;
243 hashv
^= fnv_hash_upper((const unsigned char *)mask1
, 32);
245 hashv
^= fnv_hash_upper((const unsigned char *)mask2
, 32);
246 RB_DLINK_FOREACH_SAFE(ptr
, next
, reject_list
.head
)
250 if (rdata
->mask_hashv
== hashv
)
252 rb_dlinkDelete(ptr
, &reject_list
);
254 rb_patricia_remove(reject_tree
, pnode
);
262 throttle_add(struct sockaddr
*addr
)
265 rb_patricia_node_t
*pnode
;
267 if((pnode
= rb_match_ip(throttle_tree
, addr
)) != NULL
)
271 if(t
->count
> ConfigFileEntry
.throttle_count
)
274 /* Stop penalizing them after they've been throttled */
275 t
->last
= rb_current_time();
281 if(GET_SS_FAMILY(addr
) == AF_INET6
)
284 t
= rb_malloc(sizeof(throttle_t
));
285 t
->last
= rb_current_time();
287 pnode
= make_and_lookup_ip(throttle_tree
, addr
, bitlen
);
289 rb_dlinkAdd(pnode
, &t
->node
, &throttle_list
);
295 throttle_expires(void *unused
)
297 rb_dlink_node
*ptr
, *next
;
298 rb_patricia_node_t
*pnode
;
301 RB_DLINK_FOREACH_SAFE(ptr
, next
, throttle_list
.head
)
306 if(t
->last
+ ConfigFileEntry
.throttle_duration
> rb_current_time())
309 rb_dlinkDelete(ptr
, &throttle_list
);
311 rb_patricia_remove(throttle_tree
, pnode
);