Mercurial > dovecot > original-hg > dovecot-1.2
view src/auth/login-connection.c @ 532:3b53dd1280c6 HEAD
I/O buffers now use real blocking instead of setting up a sub-ioloop to
poll(). alarm() is called every 30 secs to send SIGHUP and break out of the
read/write calls, so the given timeout values aren't exact.
Also some other cleanups, like not including ioloop.h in [io]buffer.h which
broke several other files which hadn't included it itself..
author | Timo Sirainen <tss@iki.fi> |
---|---|
date | Mon, 28 Oct 2002 06:18:26 +0200 |
parents | 1f0e7229ee58 |
children | 50bb42557703 |
line wrap: on
line source
/* Copyright (C) 2002 Timo Sirainen */ #include "common.h" #include "ioloop.h" #include "ibuffer.h" #include "obuffer.h" #include "network.h" #include "login-connection.h" #include <stdlib.h> #include <syslog.h> #define MAX_INBUF_SIZE \ (sizeof(AuthContinuedRequestData) + AUTH_MAX_REQUEST_DATA_SIZE) #define MAX_OUTBUF_SIZE \ (10 * (sizeof(AuthReplyData) + AUTH_MAX_REPLY_DATA_SIZE)) struct _LoginConnection { LoginConnection *next; int fd; IO io; IBuffer *inbuf; OBuffer *outbuf; AuthRequestType type; }; static AuthInitData auth_init_data; static LoginConnection *connections; static void request_callback(AuthReplyData *reply, const unsigned char *data, void *context) { LoginConnection *conn = context; i_assert(reply->data_size <= AUTH_MAX_REPLY_DATA_SIZE); if (o_buffer_send(conn->outbuf, reply, sizeof(AuthReplyData)) < 0) login_connection_destroy(conn); else if (reply->data_size > 0) { if (o_buffer_send(conn->outbuf, data, reply->data_size) < 0) login_connection_destroy(conn); } } static void login_input(void *context, int fd __attr_unused__, IO io __attr_unused__) { LoginConnection *conn = context; const unsigned char *data; size_t size; switch (i_buffer_read(conn->inbuf)) { case 0: return; case -1: /* disconnected */ login_connection_destroy(conn); return; case -2: /* buffer full */ i_error("BUG: imap-login sent us more than %d bytes of data", (int)MAX_INBUF_SIZE); login_connection_destroy(conn); return; } data = i_buffer_get_data(conn->inbuf, &size); if (size < sizeof(AuthRequestType)) return; /* note that we can't directly cast the received data pointer into structures, as it may not be aligned properly. */ if (conn->type == AUTH_REQUEST_NONE) { /* get the request type */ memcpy(&conn->type, data, sizeof(AuthRequestType)); } if (conn->type == AUTH_REQUEST_INIT) { AuthInitRequestData request; if (size < sizeof(request)) return; memcpy(&request, data, sizeof(request)); i_buffer_skip(conn->inbuf, sizeof(request)); /* we have a full init request */ auth_init_request(&request, request_callback, conn); conn->type = AUTH_REQUEST_NONE; } else if (conn->type == AUTH_REQUEST_CONTINUE) { AuthContinuedRequestData request; if (size < sizeof(request)) return; memcpy(&request, data, sizeof(request)); if (size < sizeof(request) + request.data_size) return; i_buffer_skip(conn->inbuf, sizeof(request) + request.data_size); /* we have a full continued request */ auth_continue_request(&request, data + sizeof(request), request_callback, conn); conn->type = AUTH_REQUEST_NONE; } else { /* unknown request */ i_error("BUG: imap-login sent us unknown request %u", conn->type); login_connection_destroy(conn); } } LoginConnection *login_connection_create(int fd) { LoginConnection *conn; conn = i_new(LoginConnection, 1); conn->fd = fd; conn->inbuf = i_buffer_create_file(fd, default_pool, MAX_INBUF_SIZE, FALSE); conn->outbuf = o_buffer_create_file(fd, default_pool, MAX_OUTBUF_SIZE, IO_PRIORITY_DEFAULT, FALSE); conn->io = io_add(fd, IO_READ, login_input, conn); conn->type = AUTH_REQUEST_NONE; conn->next = connections; connections = conn; if (o_buffer_send(conn->outbuf, &auth_init_data, sizeof(auth_init_data)) < 0) { login_connection_destroy(conn); conn = NULL; } return conn; } void login_connection_destroy(LoginConnection *conn) { LoginConnection **pos; for (pos = &connections; *pos != NULL; pos = &(*pos)->next) { if (*pos == conn) { *pos = conn->next; break; } } i_buffer_unref(conn->inbuf); o_buffer_unref(conn->outbuf); io_remove(conn->io); net_disconnect(conn->fd); i_free(conn); } void login_connections_init(void) { const char *env; env = getenv("AUTH_PROCESS"); if (env == NULL) i_fatal("AUTH_PROCESS environment is unset"); memset(&auth_init_data, 0, sizeof(auth_init_data)); auth_init_data.auth_process = atoi(env); auth_init_data.auth_methods = auth_methods; connections = NULL; } void login_connections_deinit(void) { LoginConnection *next; while (connections != NULL) { next = connections->next; login_connection_destroy(connections); connections = next; } }