src/pop3/main.c

	main.c revision cef2be5fb553b05f421f86c1ef497f0dc29d069e
45312f52ff3a3d4c137447be4c7556500c2f8bf2Timo Sirainen/* Copyright (c) 2002-2009 Dovecot authors, see the included COPYING file */
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
08d6658a4e2ec8104cd1307f6baa75fdb07a24f8Mark Washenberger#include "pop3-common.h"
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen#include "ioloop.h"
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen#include "istream.h"
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen#include "buffer.h"
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen#include "base64.h"
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen#include "restrict-access.h"
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen#include "process-title.h"
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen#include "master-service.h"
00f5efa3156ab6a0b4f21e8c703d0eb816cf3091Timo Sirainen#include "var-expand.h"
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen#include "mail-storage-service.h"
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen#include <stdio.h>
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen#include <stdlib.h>
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen#include <unistd.h>
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen#define IS_STANDALONE() \
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen        (getenv("LOGGED_IN") == NULL)
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
8f1d14e3ada93a6d6ee64f73c6e6ae2364d8eba1Timo Sirainenvoid (*hook_client_created)(struct client **client) = NULL;
8f1d14e3ada93a6d6ee64f73c6e6ae2364d8eba1Timo Sirainen
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainenstatic struct io *log_io = NULL;
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainenstatic void log_error_callback(void *context ATTR_UNUSED)
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen{
2421fd34d51fed6da985c62b5e078d7e96640653Timo Sirainen    /* the log fd is closed, don't die when trying to log later */
2421fd34d51fed6da985c62b5e078d7e96640653Timo Sirainen    i_set_failure_ignore_errors(TRUE);
2421fd34d51fed6da985c62b5e078d7e96640653Timo Sirainen
4d4d6d4745682790c20d759ba93dbea46b812c5dTimo Sirainen    master_service_stop(master_service);
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen}
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen
1b04762685272a53643ac2179939537a44c7c044Timo Sirainenstatic bool main_init(const struct pop3_settings *set, struct mail_user *user)
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen{
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen    struct client *client;
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    const char *str;
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen    bool ret = TRUE;
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
a64adf62fa33f2463a86f990217b0c9078531a40Timo Sirainen    if (set->shutdown_clients) {
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen        /* If master dies, the log fd gets closed and we'll quit */
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen        log_io = io_add(STDERR_FILENO, IO_ERROR,
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen                log_error_callback, NULL);
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen    }
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen
a64adf62fa33f2463a86f990217b0c9078531a40Timo Sirainen    client = client_create(0, 1, user, set);
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen    if (client == NULL)
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen        return FALSE;
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen
7753eaa6a4275e074b4ce8428b85d9d04fc67f31Timo Sirainen    if (!IS_STANDALONE())
7753eaa6a4275e074b4ce8428b85d9d04fc67f31Timo Sirainen        client_send_line(client, "+OK Logged in.");
7753eaa6a4275e074b4ce8428b85d9d04fc67f31Timo Sirainen
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen    str = getenv("CLIENT_INPUT");
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen    if (str != NULL) T_BEGIN {
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen        buffer_t *buf = t_base64_decode_str(str);
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen        if (buf->used > 0) {
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen            if (!i_stream_add_data(client->input, buf->data,
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen                           buf->used))
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen                i_panic("Couldn't add client input to stream");
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen            ret = client_handle_input(client);
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen        }
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen    } T_END;
cd94aeaa294f7cc507206b4b2075852f00e14d61Timo Sirainen    return ret;
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen}
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainenstatic void main_deinit(void)
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen{
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen    if (log_io != NULL)
f9ef36afc131626754716d6f4874a2ad04da0ac4Timo Sirainen        io_remove(&log_io);
f13c9ae4f96d4d13b3996ec716a959cf3380896cTimo Sirainen    clients_destroy_all();
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen}
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainenstatic void client_connected(const struct master_service_connection *conn)
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen{
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen    /* we can't handle this yet */
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen    (void)close(conn->fd);
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen}
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen
1b04762685272a53643ac2179939537a44c7c044Timo Sirainenint main(int argc, char *argv[], char *envp[])
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen{
812883e7758a86f615b9508ef10df1339123da83Timo Sirainen    const struct setting_parser_info *set_roots[] = {
812883e7758a86f615b9508ef10df1339123da83Timo Sirainen        &pop3_setting_parser_info,
812883e7758a86f615b9508ef10df1339123da83Timo Sirainen        NULL
812883e7758a86f615b9508ef10df1339123da83Timo Sirainen    };
b935d08e155589103e674b6f3a865a89e44b9423Timo Sirainen    enum master_service_flags service_flags =
b935d08e155589103e674b6f3a865a89e44b9423Timo Sirainen        MASTER_SERVICE_FLAG_STD_CLIENT;
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    enum mail_storage_service_flags storage_service_flags =
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen        MAIL_STORAGE_SERVICE_FLAG_DISALLOW_ROOT;
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen    struct mail_storage_service_input input;
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    struct mail_user *mail_user;
a64adf62fa33f2463a86f990217b0c9078531a40Timo Sirainen    const struct pop3_settings *set;
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen    const char *value;
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    int c;
a64adf62fa33f2463a86f990217b0c9078531a40Timo Sirainen
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen    if (IS_STANDALONE() && getuid() == 0 &&
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen        net_getpeername(1, NULL, NULL) == 0) {
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen        printf("-ERR pop3 binary must not be started from "
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen               "inetd, use pop3-login instead.\n");
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen        return 1;
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen    }
4dbe08e1f7f1271299ada9338ff5015367efd0b7Timo Sirainen
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    if (IS_STANDALONE())
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen        service_flags |= MASTER_SERVICE_FLAG_STANDALONE;
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    else
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen        service_flags |= MAIL_STORAGE_SERVICE_FLAG_DISALLOW_ROOT;
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen
4d4d6d4745682790c20d759ba93dbea46b812c5dTimo Sirainen    master_service = master_service_init("pop3", service_flags, argc, argv);
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    while ((c = getopt(argc, argv, master_service_getopt_string())) > 0) {
4d4d6d4745682790c20d759ba93dbea46b812c5dTimo Sirainen        if (!master_service_parse_option(master_service, c, optarg))
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen            exit(FATAL_DEFAULT);
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    }
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen    memset(&input, 0, sizeof(input));
1358e2c58ce29231485a5cfa454756d429ad3d2cTimo Sirainen    input.module = "pop3";
1358e2c58ce29231485a5cfa454756d429ad3d2cTimo Sirainen    input.service = "pop3";
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen    input.username = getenv("USER");
cef2be5fb553b05f421f86c1ef497f0dc29d069eTimo Sirainen    if (input.username == NULL && IS_STANDALONE())
cef2be5fb553b05f421f86c1ef497f0dc29d069eTimo Sirainen        input.username = getlogin();
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen    if (input.username == NULL) {
cef2be5fb553b05f421f86c1ef497f0dc29d069eTimo Sirainen        if (getenv(MASTER_UID_ENV) == NULL)
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen            i_fatal("USER environment missing");
cef2be5fb553b05f421f86c1ef497f0dc29d069eTimo Sirainen        else {
cef2be5fb553b05f421f86c1ef497f0dc29d069eTimo Sirainen            i_fatal("login_executable setting must be pop3-login, "
cef2be5fb553b05f421f86c1ef497f0dc29d069eTimo Sirainen                "not pop3");
cef2be5fb553b05f421f86c1ef497f0dc29d069eTimo Sirainen        }
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    }
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen    if ((value = getenv("IP")) != NULL)
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen        net_addr2ip(value, &input.remote_ip);
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen    if ((value = getenv("LOCAL_IP")) != NULL)
d1fff80640050631b06bfab904a34b2ad24601e8Timo Sirainen        net_addr2ip(value, &input.local_ip);
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen
4d4d6d4745682790c20d759ba93dbea46b812c5dTimo Sirainen    mail_user = mail_storage_service_init_user(master_service,
4d4d6d4745682790c20d759ba93dbea46b812c5dTimo Sirainen                           &input, set_roots,
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen                           storage_service_flags);
4d4d6d4745682790c20d759ba93dbea46b812c5dTimo Sirainen    set = mail_storage_service_get_settings(master_service);
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    restrict_access_allow_coredumps(TRUE);
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen        process_title_init(argv, envp);
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    /* fake that we're running, so we know if client was destroyed
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen       while initializing */
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    io_loop_set_running(current_ioloop);
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    if (main_init(set, mail_user))
4d4d6d4745682790c20d759ba93dbea46b812c5dTimo Sirainen        master_service_run(master_service, client_connected);
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    main_deinit();
1b04762685272a53643ac2179939537a44c7c044Timo Sirainen    mail_storage_service_deinit_user();
4d4d6d4745682790c20d759ba93dbea46b812c5dTimo Sirainen    master_service_deinit(&master_service);
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen    return 0;
548f87789cf9865572b7b86f7be5a9bbfa132f3fTimo Sirainen}