sync ssld with ratbox3 repository: now r25594 + charybdis related changes
This commit is contained in:
parent
a0ef73fa59
commit
4b6a4d479c
1 changed files with 43 additions and 15 deletions
58
ssld/ssld.c
58
ssld/ssld.c
|
@ -18,7 +18,7 @@
|
|||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
|
||||
* USA
|
||||
*
|
||||
* $Id: ssld.c 25221 2008-04-09 22:12:16Z jilles $
|
||||
* $Id: ssld.c 25594 2008-06-25 14:58:47Z androsyn $
|
||||
*/
|
||||
|
||||
|
||||
|
@ -38,41 +38,34 @@ static void setup_signals(void);
|
|||
static inline rb_int32_t buf_to_int32(char *buf)
|
||||
{
|
||||
rb_int32_t x;
|
||||
x = *buf << 24;
|
||||
x |= *(++buf) << 16;
|
||||
x |= *(++buf) << 8;
|
||||
x |= *(++buf);
|
||||
memcpy(&x, buf, sizeof(x));
|
||||
return x;
|
||||
}
|
||||
|
||||
static inline void int32_to_buf(char *buf, rb_int32_t x)
|
||||
{
|
||||
*(buf) = x >> 24 & 0xFF;
|
||||
*(++buf) = x >> 16 & 0xFF;
|
||||
*(++buf) = x >> 8 & 0xFF;
|
||||
*(++buf) = x & 0xFF;
|
||||
memcpy(buf, &x, sizeof(x));
|
||||
return;
|
||||
}
|
||||
|
||||
static inline rb_uint16_t buf_to_uint16(char *buf)
|
||||
{
|
||||
rb_uint16_t x;
|
||||
x = *(buf) << 8;
|
||||
x |= *(++buf);
|
||||
memcpy(&x, buf, sizeof(x));
|
||||
return x;
|
||||
}
|
||||
|
||||
static inline void uint16_to_buf(char *buf, rb_uint16_t x)
|
||||
{
|
||||
*(buf) = x >> 8 & 0xFF;
|
||||
*(++buf) = x & 0xFF;
|
||||
memcpy(buf, &x, sizeof(x));
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
|
||||
static char inbuf[READBUF_SIZE];
|
||||
#ifdef HAVE_LIBZ
|
||||
static char outbuf[READBUF_SIZE];
|
||||
#endif
|
||||
|
||||
typedef struct _mod_ctl_buf
|
||||
{
|
||||
|
@ -154,9 +147,13 @@ typedef struct _conn
|
|||
#define NO_WAIT 0x0
|
||||
#define WAIT_PLAIN 0x1
|
||||
|
||||
#define HASH_WALK_SAFE(i, max, ptr, next, table) for(i = 0; i < max; i++) { RB_DLINK_FOREACH_SAFE(ptr, next, table[i].head)
|
||||
#define HASH_WALK_END }
|
||||
#define CONN_HASH_SIZE 2000
|
||||
#define connid_hash(x) (&connid_hash_table[(x % CONN_HASH_SIZE)])
|
||||
|
||||
|
||||
|
||||
static rb_dlink_list connid_hash_table[CONN_HASH_SIZE];
|
||||
static rb_dlink_list dead_list;
|
||||
|
||||
|
@ -173,6 +170,9 @@ static int zlib_ok = 1;
|
|||
#else
|
||||
static int zlib_ok = 0;
|
||||
#endif
|
||||
|
||||
|
||||
#ifdef HAVE_LIBZ
|
||||
static void *
|
||||
ssld_alloc(void *unused, size_t count, size_t size)
|
||||
{
|
||||
|
@ -184,6 +184,7 @@ ssld_free(void *unused, void *ptr)
|
|||
{
|
||||
rb_free(ptr);
|
||||
}
|
||||
#endif
|
||||
|
||||
static conn_t *
|
||||
conn_find_by_id(rb_int32_t id)
|
||||
|
@ -212,12 +213,14 @@ free_conn(conn_t * conn)
|
|||
{
|
||||
rb_free_rawbuffer(conn->modbuf_out);
|
||||
rb_free_rawbuffer(conn->plainbuf_out);
|
||||
#ifdef HAVE_LIBZ
|
||||
if(IsZip(conn))
|
||||
{
|
||||
zlib_stream_t *stream = conn->stream;
|
||||
inflateEnd(&stream->instream);
|
||||
deflateEnd(&stream->outstream);
|
||||
}
|
||||
#endif
|
||||
rb_free(conn);
|
||||
}
|
||||
|
||||
|
@ -287,6 +290,30 @@ make_conn(mod_ctl_t *ctl, rb_fde_t * mod_fd, rb_fde_t * plain_fd)
|
|||
return conn;
|
||||
}
|
||||
|
||||
static void
|
||||
check_handshake_flood(void *unused)
|
||||
{
|
||||
conn_t *conn;
|
||||
rb_dlink_node *ptr, *next;
|
||||
unsigned int count;
|
||||
int i;
|
||||
HASH_WALK_SAFE(i, CONN_HASH_SIZE, ptr, next, connid_hash_table)
|
||||
{
|
||||
conn = ptr->data;
|
||||
if(!IsSSL(conn))
|
||||
continue;
|
||||
|
||||
count = rb_ssl_handshake_count(conn->mod_fd);
|
||||
/* nothing needs to do this more than twice in ten seconds i don't think */
|
||||
if(count > 2)
|
||||
close_conn(conn, WAIT_PLAIN, "Handshake flooding");
|
||||
else
|
||||
rb_ssl_clear_handshake_count(conn->mod_fd);
|
||||
}
|
||||
HASH_WALK_END
|
||||
|
||||
}
|
||||
|
||||
static void
|
||||
conn_mod_write_sendq(rb_fde_t * fd, void *data)
|
||||
{
|
||||
|
@ -1031,7 +1058,7 @@ main(int argc, char **argv)
|
|||
if(s_ctlfd == NULL || s_pipe == NULL)
|
||||
{
|
||||
fprintf(stderr, "This is ircd-ratbox ssld. You know you aren't supposed to run me directly?\n");
|
||||
fprintf(stderr, "You get an Id tag for this: $Id: ssld.c 25221 2008-04-09 22:12:16Z jilles $\n");
|
||||
fprintf(stderr, "You get an Id tag for this: $Id: ssld.c 25594 2008-06-25 14:58:47Z androsyn $\n");
|
||||
fprintf(stderr, "Have a nice life\n");
|
||||
exit(1);
|
||||
}
|
||||
|
@ -1070,6 +1097,7 @@ main(int argc, char **argv)
|
|||
rb_set_nb(mod_ctl->F);
|
||||
rb_set_nb(mod_ctl->F_pipe);
|
||||
rb_event_addish("clean_dead_conns", clean_dead_conns, NULL, 10);
|
||||
rb_event_add("check_handshake_flood", check_handshake_flood, NULL, 10);
|
||||
read_pipe_ctl(mod_ctl->F_pipe, NULL);
|
||||
mod_read_ctl(mod_ctl->F, mod_ctl);
|
||||
if(!zlib_ok && !ssl_ok)
|
||||
|
|
Loading…
Reference in a new issue