src/auth/main.c

	main.c revision 5a47b70946f83e80c9195d8cb71a616f62feb388
/* Copyright (C) 2002 Timo Sirainen */

#include "common.h"
#include "ioloop.h"
#include "network.h"
#include "lib-signals.h"
#include "restrict-access.h"
#include "randgen.h"
#include "auth.h"
#include "cookie.h"
#include "login-connection.h"
#include "userinfo.h"

#include <stdlib.h>
#include <syslog.h>

void master_init(void);
void master_deinit(void);

IOLoop ioloop;
static IO io_listen;

static void sig_quit(int signo __attr_unused__)
{
    io_loop_stop(ioloop);
}

static void auth_accept(void *context __attr_unused__, int listen_fd,
            IO io __attr_unused__)
{
    int fd;

    fd = net_accept(listen_fd, NULL, NULL);
    if (fd < 0) {
        if (fd < -1)
            i_fatal("accept() failed: %m");
    } else {
        (void)login_connection_create(fd);
    }
}

static void main_init(void)
{
    const char *logfile;

    lib_init_signals(sig_quit);

    logfile = getenv("IMAP_LOGFILE");
    if (logfile == NULL) {
        /* open the syslog immediately so chroot() won't
           break logging */
        openlog("imap-auth", LOG_NDELAY, LOG_MAIL);

        i_set_panic_handler(i_syslog_panic_handler);
        i_set_fatal_handler(i_syslog_fatal_handler);
        i_set_error_handler(i_syslog_error_handler);
        i_set_warning_handler(i_syslog_warning_handler);
    } else {
        /* log failures into specified log file */
        i_set_failure_file(logfile, "imap-auth");
        i_set_failure_timestamp_format(getenv("IMAP_LOGSTAMP"));
    }

    /* open /dev/urandom before chrooting */
    random_init();

    restrict_access_by_env();

    auth_init();
    cookies_init();
    login_connections_init();
    master_init();
    userinfo_init();

    io_listen = io_add(LOGIN_LISTEN_FD, IO_READ, auth_accept, NULL);
}

static void main_deinit(void)
{
        if (lib_signal_kill != 0)
        i_warning("Killed with signal %d", lib_signal_kill);

    io_remove(io_listen);

    userinfo_deinit();
    master_deinit();
    login_connections_deinit();
    cookies_deinit();
    auth_deinit();

    random_deinit();

    closelog();
}

int main(int argc __attr_unused__, char *argv[] __attr_unused__)
{
    /* NOTE: we start rooted, so keep the code minimal until
       restrict_access_by_env() is called */
    lib_init();
    ioloop = io_loop_create(system_pool);

    main_init();
        io_loop_run(ioloop);
    main_deinit();

    io_loop_destroy(ioloop);
    lib_deinit();

        return 0;
}