Mercurial > dovecot > core-2.2
view src/imap/main.c @ 3620:3360cc019737 HEAD
Implemented new signal handling framework, which makes handling signals much
easier.
author | Timo Sirainen <tss@iki.fi> |
---|---|
date | Sun, 25 Sep 2005 14:07:32 +0300 |
parents | 0e05687892dc |
children | 8b68997d6e34 |
line wrap: on
line source
/* Copyright (C) 2002-2003 Timo Sirainen */ #include "common.h" #include "ioloop.h" #include "network.h" #include "ostream.h" #include "str.h" #include "lib-signals.h" #include "restrict-access.h" #include "fd-close-on-exec.h" #include "process-title.h" #include "randgen.h" #include "module-dir.h" #include "mail-storage.h" #include "commands.h" #include "namespace.h" #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <syslog.h> #define IS_STANDALONE() \ (getenv("LOGGED_IN") == NULL && getenv("IMAPLOGINTAG") == NULL) struct client_workaround_list { const char *name; enum client_workarounds num; }; struct client_workaround_list client_workaround_list[] = { { "delay-newmail", WORKAROUND_DELAY_NEWMAIL }, { "outlook-idle", WORKAROUND_OUTLOOK_IDLE }, { "netscape-eoh", WORKAROUND_NETSCAPE_EOH }, { "tb-extra-mailbox-sep", WORKAROUND_TB_EXTRA_MAILBOX_SEP }, { NULL, 0 } }; struct ioloop *ioloop; unsigned int max_keyword_length; unsigned int imap_max_line_length; enum client_workarounds client_workarounds = 0; static struct module *modules; static char log_prefix[128]; /* syslog() needs this to be permanent */ static pool_t namespace_pool; void (*hook_mail_storage_created)(struct mail_storage *storage) = NULL; void (*hook_client_created)(struct client **client) = NULL; string_t *capability_string; static void sig_die(int signo, void *context __attr_unused__) { /* warn about being killed because of some signal, except SIGINT (^C) which is too common at least while testing :) */ if (signo != SIGINT) i_warning("Killed with signal %d", signo); io_loop_stop(ioloop); } static void parse_workarounds(void) { struct client_workaround_list *list; const char *env, *const *str; env = getenv("IMAP_CLIENT_WORKAROUNDS"); if (env == NULL) return; for (str = t_strsplit_spaces(env, " "); *str != NULL; str++) { list = client_workaround_list; for (; list->name != NULL; list++) { if (strcasecmp(*str, list->name) == 0) { client_workarounds |= list->num; break; } } if (list->name == NULL) i_fatal("Unknown client workaround: %s", *str); } } static void open_logfile(void) { const char *user; if (getenv("LOG_TO_MASTER") != NULL) { i_set_failure_internal(); return; } user = getenv("USER"); if (user == NULL) { if (IS_STANDALONE()) user = getlogin(); if (user == NULL) user = "??"; } if (strlen(user) >= sizeof(log_prefix)-6) { /* quite a long user name, cut it */ user = t_strndup(user, sizeof(log_prefix)-6-2); user = t_strconcat(user, "..", NULL); } i_snprintf(log_prefix, sizeof(log_prefix), "imap(%s)", user); if (getenv("USE_SYSLOG") != NULL) i_set_failure_syslog(log_prefix, LOG_NDELAY, LOG_MAIL); else { /* log to file or stderr */ i_set_failure_file(getenv("LOGFILE"), log_prefix); } if (getenv("INFOLOGFILE") != NULL) i_set_info_file(getenv("INFOLOGFILE")); i_set_failure_timestamp_format(getenv("LOGSTAMP")); } static void drop_privileges(void) { /* Log file or syslog opening probably requires roots */ open_logfile(); /* Most likely needed. Have to open /dev/urandom before possible chrooting. */ random_init(); restrict_access_by_env(!IS_STANDALONE()); } static void main_init(void) { struct client *client; const char *user, *str; lib_signals_init(); lib_signals_set_handler(SIGINT, TRUE, sig_die, NULL); lib_signals_set_handler(SIGTERM, TRUE, sig_die, NULL); lib_signals_set_handler(SIGPIPE, FALSE, NULL, NULL); user = getenv("USER"); if (user == NULL) { if (IS_STANDALONE()) user = getlogin(); if (user == NULL) i_fatal("USER environment missing"); } if (getenv("DEBUG") != NULL) { i_info("Effective uid=%s, gid=%s", dec2str(geteuid()), dec2str(getegid())); } capability_string = str_new(default_pool, sizeof(CAPABILITY_STRING)+32); str_append(capability_string, CAPABILITY_STRING); mail_storage_init(); mail_storage_register_all(); clients_init(); commands_init(); modules = getenv("MODULE_DIR") == NULL ? NULL : module_dir_load(getenv("MODULE_DIR"), TRUE); str = getenv("IMAP_MAX_LINE_LENGTH"); imap_max_line_length = str != NULL ? (unsigned int)strtoul(str, NULL, 10) : DEFAULT_IMAP_MAX_LINE_LENGTH; str = getenv("MAIL_MAX_KEYWORD_LENGTH"); max_keyword_length = str != NULL ? (unsigned int)strtoul(str, NULL, 10) : DEFAULT_MAX_KEYWORD_LENGTH; parse_workarounds(); namespace_pool = pool_alloconly_create("namespaces", 1024); client = client_create(0, 1, namespace_init(namespace_pool, user)); o_stream_cork(client->output); if (IS_STANDALONE()) { client_send_line(client, t_strconcat( "* PREAUTH [CAPABILITY "CAPABILITY_STRING"] " "Logged in as ", user, NULL)); } else if (getenv("IMAPLOGINTAG") != NULL) { /* Support for mailfront */ client_send_line(client, t_strconcat(getenv("IMAPLOGINTAG"), " OK Logged in.", NULL)); } o_stream_uncork(client->output); } static void main_deinit(void) { module_dir_unload(modules); commands_deinit(); clients_deinit(); mail_storage_deinit(); random_deinit(); pool_unref(namespace_pool); str_free(capability_string); lib_signals_deinit(); closelog(); } int main(int argc __attr_unused__, char *argv[], char *envp[]) { #ifdef DEBUG if (getenv("LOGGED_IN") != NULL && getenv("GDB") == NULL) fd_debug_verify_leaks(3, 1024); #endif if (IS_STANDALONE() && getuid() == 0 && net_getpeername(1, NULL, NULL) == 0) { printf("* BAD [ALERT] imap binary must not be started from " "inetd, use imap-login instead.\n"); return 1; } /* NOTE: we start rooted, so keep the code minimal until restrict_access_by_env() is called */ lib_init(); drop_privileges(); process_title_init(argv, envp); ioloop = io_loop_create(system_pool); main_init(); io_loop_run(ioloop); main_deinit(); io_loop_destroy(ioloop); lib_deinit(); return 0; }