src/imap/main.c

	main.c revision a73052375a6696e47dbb0bd492974cece1b5fde7
c25356d5978632df6203437e1953bcb29e0c736fTimo Sirainen/* Copyright (c) 2002-2018 Dovecot authors, see the included COPYING file */
c25356d5978632df6203437e1953bcb29e0c736fTimo Sirainen
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen#include "imap-common.h"
0c909e3461607eadcd66f4eac69b7f34e37fccf1Timo Sirainen#include "ioloop.h"
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen#include "istream.h"
4d4d585520538a752e9f0a4a1c019a2918f52e56Timo Sirainen#include "ostream.h"
4d4d585520538a752e9f0a4a1c019a2918f52e56Timo Sirainen#include "path-util.h"
105addcb709523868418cc3e3baad7ad3453a91eTimo Sirainen#include "str.h"
aef92409cf369afdd2ecd81a4f80083cd4082f46Timo Sirainen#include "base64.h"
105addcb709523868418cc3e3baad7ad3453a91eTimo Sirainen#include "process-title.h"
105addcb709523868418cc3e3baad7ad3453a91eTimo Sirainen#include "randgen.h"
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen#include "restrict-access.h"
105addcb709523868418cc3e3baad7ad3453a91eTimo Sirainen#include "write-full.h"
105addcb709523868418cc3e3baad7ad3453a91eTimo Sirainen#include "settings-parser.h"
105addcb709523868418cc3e3baad7ad3453a91eTimo Sirainen#include "master-interface.h"
105addcb709523868418cc3e3baad7ad3453a91eTimo Sirainen#include "master-service.h"
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen#include "master-login.h"
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen#include "mail-user.h"
0add8c99ca65e56dbf613595fc37c41aafff3f7fTimo Sirainen#include "mail-storage-service.h"
40ef82c46f6652412b068ebcdac7c3e74840a284Timo Sirainen#include "smtp-submit-settings.h"
105addcb709523868418cc3e3baad7ad3453a91eTimo Sirainen#include "imap-master-client.h"
49d4afbb76f47c8904537d087bc81e43f1c0aa25Timo Sirainen#include "imap-resp-code.h"
aef92409cf369afdd2ecd81a4f80083cd4082f46Timo Sirainen#include "imap-commands.h"
aef92409cf369afdd2ecd81a4f80083cd4082f46Timo Sirainen#include "imap-fetch.h"
aef92409cf369afdd2ecd81a4f80083cd4082f46Timo Sirainen
b44faf865da16ac4d18eecd85a55b3fab6b9e63aTimo Sirainen#include <stdio.h>
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen#include <unistd.h>
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen#define IS_STANDALONE() \
1175f27441385a7011629f295f42708f9a3a4ffcTimo Sirainen        (getenv(MASTER_IS_PARENT_ENV) == NULL)
1175f27441385a7011629f295f42708f9a3a4ffcTimo Sirainen
1175f27441385a7011629f295f42708f9a3a4ffcTimo Sirainen#define IMAP_DIE_IDLE_SECS 10
56f45b3f3ae20e5c933701f4657dda5ef1916855Timo Sirainen
7797aa2479e99aeb71057b7a2584b2cb72e4d3f8Timo Sirainenstatic bool verbose_proctitle = FALSE;
7797aa2479e99aeb71057b7a2584b2cb72e4d3f8Timo Sirainenstatic struct mail_storage_service_ctx *storage_service;
7797aa2479e99aeb71057b7a2584b2cb72e4d3f8Timo Sirainenstatic struct master_login *master_login = NULL;
7797aa2479e99aeb71057b7a2584b2cb72e4d3f8Timo Sirainen
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainenimap_client_created_func_t *hook_client_created = NULL;
faed8babca9914257f34fb2e603d74016d563b2dTimo Sirainenbool imap_debug = FALSE;
c5794838af9995f50bfecb06a3cd4f9a0ac77858Timo Sirainen
0c909e3461607eadcd66f4eac69b7f34e37fccf1Timo Sirainenstruct event_category event_category_imap = {
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen    .name = "imap",
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainen};
651fc0f1e43fef3e02e0e7b5f498973b05f641d7Timo Sirainen
8872e5c991430f96138a46e36b7f3c2c40d8e5c2Timo Sirainenimap_client_created_func_t *
e188bab0b830136d04a1dd8b55e9afefae20d930Timo Sirainenimap_client_created_hook_set(imap_client_created_func_t *new_hook)
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen{
5a3f143585110536a7f72628e66a42b998bfc5fcTimo Sirainen    imap_client_created_func_t *old_hook = hook_client_created;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
51795bfe9d05d92fe942cb451aec2b9d16d32a11Timo Sirainen    hook_client_created = new_hook;
51795bfe9d05d92fe942cb451aec2b9d16d32a11Timo Sirainen    return old_hook;
041d312b44f8d41f0c9a5762c23e4d146ef7302bTimo Sirainen}
041d312b44f8d41f0c9a5762c23e4d146ef7302bTimo Sirainen
51795bfe9d05d92fe942cb451aec2b9d16d32a11Timo Sirainenvoid imap_refresh_proctitle(void)
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen{
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen#define IMAP_PROCTITLE_PREFERRED_LEN 80
0c909e3461607eadcd66f4eac69b7f34e37fccf1Timo Sirainen    struct client *client;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    struct client_command_context *cmd;
1175f27441385a7011629f295f42708f9a3a4ffcTimo Sirainen    string_t *title = t_str_new(128);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    bool wait_output;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
1175f27441385a7011629f295f42708f9a3a4ffcTimo Sirainen    if (!verbose_proctitle)
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        return;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    str_append_c(title, '[');
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    switch (imap_client_count) {
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    case 0:
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        str_append(title, "idling");
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        break;
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen    case 1:
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen        client = imap_clients;
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen        str_append(title, client->user->username);
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen        if (client->user->conn.remote_ip != NULL) {
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen            str_append_c(title, ' ');
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen            str_append(title,
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen                   net_ip2addr(client->user->conn.remote_ip));
ad48319996942463675b53877092ab7e13a7a75aTimo Sirainen        }
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        wait_output = FALSE;
659fe5d24825b160cae512538088020d97a60239Timo Sirainen        for (cmd = client->command_queue; cmd != NULL; cmd = cmd->next) {
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen            if (cmd->name == NULL)
0c909e3461607eadcd66f4eac69b7f34e37fccf1Timo Sirainen                continue;
0c909e3461607eadcd66f4eac69b7f34e37fccf1Timo Sirainen
de62ce819d59a529530da4b57be1b8d6dad13d6bTimo Sirainen            if (str_len(title) < IMAP_PROCTITLE_PREFERRED_LEN) {
0c909e3461607eadcd66f4eac69b7f34e37fccf1Timo Sirainen                str_append_c(title, ' ');
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen                str_append(title, cmd->name);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen            }
659fe5d24825b160cae512538088020d97a60239Timo Sirainen            if (cmd->state == CLIENT_COMMAND_STATE_WAIT_OUTPUT)
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen                wait_output = TRUE;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        }
232d5bef3c709e90e24f0874a36854b92187bb6cTimo Sirainen        if (wait_output) {
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen            str_printfa(title, " - %"PRIuSIZE_T" bytes waiting",
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen                    o_stream_get_buffer_used_size(client->output));
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen            if (o_stream_is_corked(client->output))
faed8babca9914257f34fb2e603d74016d563b2dTimo Sirainen                str_append(title, " corked");
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen        }
faed8babca9914257f34fb2e603d74016d563b2dTimo Sirainen        if (client->destroyed)
faed8babca9914257f34fb2e603d74016d563b2dTimo Sirainen            str_append(title, " (deinit)");
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen        break;
16c89b1260c9d07c01c83a9219424d3727069b2eTimo Sirainen    default:
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        str_printfa(title, "%u connections", imap_client_count);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        break;
faed8babca9914257f34fb2e603d74016d563b2dTimo Sirainen    }
faed8babca9914257f34fb2e603d74016d563b2dTimo Sirainen    str_append_c(title, ']');
faed8babca9914257f34fb2e603d74016d563b2dTimo Sirainen    process_title_set(str_c(title));
faed8babca9914257f34fb2e603d74016d563b2dTimo Sirainen}
e06c0b65c16ccce69bbee009ead14d7d3d17a256Timo Sirainen
e06c0b65c16ccce69bbee009ead14d7d3d17a256Timo Sirainenstatic void client_kill_idle(struct client *client)
e06c0b65c16ccce69bbee009ead14d7d3d17a256Timo Sirainen{
a928e7efabb1672b1476e597106d4b4b81ac6f3cTimo Sirainen    if (client->output_cmd_lock != NULL)
e06c0b65c16ccce69bbee009ead14d7d3d17a256Timo Sirainen        return;
e06c0b65c16ccce69bbee009ead14d7d3d17a256Timo Sirainen
838f56174b963779a88083a0d0e85b30d2d846e7Timo Sirainen    client_send_line(client, "* BYE Server shutting down.");
894987bf45718f8849cc3898afdfb1ac3cfa2445Timo Sirainen    mail_storage_service_io_activate_user(client->service_user);
894987bf45718f8849cc3898afdfb1ac3cfa2445Timo Sirainen    client_destroy(client, "Server shutting down.");
894987bf45718f8849cc3898afdfb1ac3cfa2445Timo Sirainen}
838f56174b963779a88083a0d0e85b30d2d846e7Timo Sirainen
838f56174b963779a88083a0d0e85b30d2d846e7Timo Sirainenstatic void imap_die(void)
7797aa2479e99aeb71057b7a2584b2cb72e4d3f8Timo Sirainen{
d143077bd518de129b8d446fb58e003903e50867Timo Sirainen    struct client *client, *next;
d143077bd518de129b8d446fb58e003903e50867Timo Sirainen    time_t last_io, now = time(NULL);
d143077bd518de129b8d446fb58e003903e50867Timo Sirainen    time_t stop_timestamp = now - IMAP_DIE_IDLE_SECS;
6a19e109ee8c5a6f688da83a86a7f6abeb71abddTimo Sirainen    unsigned int stop_msecs;
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen    for (client = imap_clients; client != NULL; client = next) {
bbf796c17f02538058d7559bfe96d677e5b55015Timo Sirainen        next = client->next;
838f56174b963779a88083a0d0e85b30d2d846e7Timo Sirainen
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen        last_io = I_MAX(client->last_input, client->last_output);
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen        if (last_io <= stop_timestamp)
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen            client_kill_idle(client);
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen        else {
6a19e109ee8c5a6f688da83a86a7f6abeb71abddTimo Sirainen            timeout_remove(&client->to_idle);
6a19e109ee8c5a6f688da83a86a7f6abeb71abddTimo Sirainen            stop_msecs = (last_io - stop_timestamp) * 1000;
6e07b4251bf6a3cf34019c351a32a65c08392e58Timo Sirainen            client->to_idle = timeout_add(stop_msecs,
6e07b4251bf6a3cf34019c351a32a65c08392e58Timo Sirainen                              client_kill_idle, client);
6a19e109ee8c5a6f688da83a86a7f6abeb71abddTimo Sirainen        }
6a19e109ee8c5a6f688da83a86a7f6abeb71abddTimo Sirainen    }
d143077bd518de129b8d446fb58e003903e50867Timo Sirainen}
7797aa2479e99aeb71057b7a2584b2cb72e4d3f8Timo Sirainen
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainenstruct client_input {
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen    const char *tag;
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen    const unsigned char *input;
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainen    size_t input_size;
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainen    bool send_untagged_capability;
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainen};
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainen
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainenstatic void
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainenclient_parse_input(const unsigned char *data, size_t len,
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainen           struct client_input *input_r)
e76f5e07be5bec4e5ca99c3e093ff7f11edbe1b7Timo Sirainen{
5a07b37a9df398b5189c14872a600384208ab74bTimo Sirainen    size_t taglen;
7797aa2479e99aeb71057b7a2584b2cb72e4d3f8Timo Sirainen
1175f27441385a7011629f295f42708f9a3a4ffcTimo Sirainen    i_assert(len > 0);
a928e7efabb1672b1476e597106d4b4b81ac6f3cTimo Sirainen
1175f27441385a7011629f295f42708f9a3a4ffcTimo Sirainen    i_zero(input_r);
c5794838af9995f50bfecb06a3cd4f9a0ac77858Timo Sirainen
c5794838af9995f50bfecb06a3cd4f9a0ac77858Timo Sirainen    if (data[0] == '1')
c5794838af9995f50bfecb06a3cd4f9a0ac77858Timo Sirainen        input_r->send_untagged_capability = TRUE;
c5794838af9995f50bfecb06a3cd4f9a0ac77858Timo Sirainen    data++; len--;
1175f27441385a7011629f295f42708f9a3a4ffcTimo Sirainen
e188bab0b830136d04a1dd8b55e9afefae20d930Timo Sirainen    input_r->tag = t_strndup(data, len);
e188bab0b830136d04a1dd8b55e9afefae20d930Timo Sirainen    taglen = strlen(input_r->tag) + 1;
e188bab0b830136d04a1dd8b55e9afefae20d930Timo Sirainen
e188bab0b830136d04a1dd8b55e9afefae20d930Timo Sirainen    if (len > taglen) {
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen        input_r->input = data + taglen;
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen        input_r->input_size = len - taglen;
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen    }
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen}
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen
77bc2bda5b781c4ffddc8a74b175cf32e9e2c2ecTimo Sirainenstatic void
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainenclient_add_input_capability(struct client *client, const unsigned char *client_input,
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen                size_t client_input_size)
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen{
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen    struct ostream *output;
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen    struct client_input input;
0d70a702dec63d22535684fec6a7247c5f153208Timo Sirainen
0d70a702dec63d22535684fec6a7247c5f153208Timo Sirainen    if (client_input_size > 0) {
0d70a702dec63d22535684fec6a7247c5f153208Timo Sirainen        client_parse_input(client_input, client_input_size, &input);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        if (input.input_size > 0 &&
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen            !i_stream_add_data(client->input, input.input,
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen                       input.input_size))
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen            i_panic("Couldn't add client input to stream");
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen    } else {
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        /* IMAPLOGINTAG environment is compatible with mailfront */
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        i_zero(&input);
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        input.tag = getenv("IMAPLOGINTAG");
94aa90d2d17a7aebcda5a4193a62e80ddbb169b7Timo Sirainen    }
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen    /* cork/uncork around the OK reply to minimize latency */
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen    output = client->output;
2d49f150b4bce6f2f59a84e268e4777901c3e42cTimo Sirainen    o_stream_ref(output);
2d49f150b4bce6f2f59a84e268e4777901c3e42cTimo Sirainen    o_stream_cork(output);
2d49f150b4bce6f2f59a84e268e4777901c3e42cTimo Sirainen    if (input.tag == NULL) {
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        client_send_line(client, t_strconcat(
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen            "* PREAUTH [CAPABILITY ",
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen            str_c(client->capability_string), "] "
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen            "Logged in as ", client->user->username, NULL));
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen    } else if (input.send_untagged_capability) {
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        /* client doesn't seem to understand tagged capabilities. send
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen           untagged instead and hope that it works. */
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        client_send_line(client, t_strconcat("* CAPABILITY ",
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen            str_c(client->capability_string), NULL));
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        client_send_line(client,
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen                 t_strconcat(input.tag, " OK Logged in", NULL));
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen    } else {
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        client_send_line(client, t_strconcat(
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen            input.tag, " OK [CAPABILITY ",
0d0451206a91e9f96e522075dce28a89adc2325dTimo Sirainen            str_c(client->capability_string), "] Logged in", NULL));
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    }
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    o_stream_uncork(output);
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen    o_stream_unref(&output);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen}
c2378fc4d82577009b0b5125f381387e7d74ce46Timo Sirainen
c2378fc4d82577009b0b5125f381387e7d74ce46Timo Sirainenstatic void
ae8817f05005f57bba32479a610b52d083e2b6ebTimo Sirainenclient_add_input_finalize(struct client *client)
ae8817f05005f57bba32479a610b52d083e2b6ebTimo Sirainen{
ae8817f05005f57bba32479a610b52d083e2b6ebTimo Sirainen    struct ostream *output;
ae8817f05005f57bba32479a610b52d083e2b6ebTimo Sirainen
ae8817f05005f57bba32479a610b52d083e2b6ebTimo Sirainen    /* try to condense any responses into as few packets as possible */
096953143c4032bad154637f687551856f7946cbTimo Sirainen    output = client->output;
096953143c4032bad154637f687551856f7946cbTimo Sirainen    o_stream_ref(output);
c680a6b35b459045e92814778908da5a93922107Timo Sirainen    o_stream_cork(output);
c680a6b35b459045e92814778908da5a93922107Timo Sirainen    (void)client_handle_input(client);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    o_stream_uncork(output);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    o_stream_unref(&output);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    /* we could have already handled LOGOUT, or we might need to continue
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen       pending ambiguous commands. */
5f78b33aa505b17e23cdf27b071a24e127b3db54Timo Sirainen    if (client->disconnected)
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen        client_destroy(client, NULL);
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen    else
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen        client_continue_pending_input(client);
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen}
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainenint client_create_from_input(const struct mail_storage_service_input *input,
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen                 int fd_in, int fd_out,
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen                 struct client **client_r, const char **error_r)
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen{
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    struct mail_storage_service_input service_input;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    struct mail_storage_service_user *user;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    struct mail_user *mail_user;
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen    struct client *client;
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen    struct imap_settings *imap_set;
95a1a5195d56f3cf5d1e529aad668f87ad3b979bTimo Sirainen    struct smtp_submit_settings *smtp_set;
c9c24293550541307f1bb41bba4a0fdfe2fa59e0Timo Sirainen    struct event *event;
c9c24293550541307f1bb41bba4a0fdfe2fa59e0Timo Sirainen    const char *errstr;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    event = event_create(NULL);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    event_add_category(event, &event_category_imap);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    event_add_fields(event, (const struct event_add_field []){
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        { .key = "user", .value = input->username },
43d32cbe60fdaef2699d99f1ca259053e9350411Timo Sirainen        { .key = "session", .value = input->session_id },
6ef7e31619edfaa17ed044b45861d106a86191efTimo Sirainen        { .key = NULL }
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    });
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
40a5aeebf6b4858b93f0ddff0bf12fba769cf903Timo Sirainen    service_input = *input;
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen    service_input.parent_event = event;
77bc2bda5b781c4ffddc8a74b175cf32e9e2c2ecTimo Sirainen    if (mail_storage_service_lookup_next(storage_service, &service_input,
77bc2bda5b781c4ffddc8a74b175cf32e9e2c2ecTimo Sirainen                         &user, &mail_user, error_r) <= 0) {
40a5aeebf6b4858b93f0ddff0bf12fba769cf903Timo Sirainen        event_unref(&event);
1279090ba03f9c176976a69ab7718f0ed77b19afTimo Sirainen        return -1;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    }
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    restrict_access_allow_coredumps(TRUE);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    smtp_set = mail_storage_service_user_get_set(user)[1];
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    imap_set = mail_storage_service_user_get_set(user)[2];
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    if (imap_set->verbose_proctitle)
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        verbose_proctitle = TRUE;
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    if (settings_var_expand(&smtp_submit_setting_parser_info, smtp_set,
3ee5f5427b36ea30a01561b35f4002232db7b061Timo Sirainen                mail_user->pool, mail_user_var_expand_table(mail_user),
3ee5f5427b36ea30a01561b35f4002232db7b061Timo Sirainen                &errstr) <= 0 ||
3ee5f5427b36ea30a01561b35f4002232db7b061Timo Sirainen            settings_var_expand(&imap_setting_parser_info, imap_set,
a050ca9def13949dbaa67bd6574a41c4f397ae26Timo Sirainen                mail_user->pool, mail_user_var_expand_table(mail_user),
6ef7e31619edfaa17ed044b45861d106a86191efTimo Sirainen                &errstr) <= 0) {
6ef7e31619edfaa17ed044b45861d106a86191efTimo Sirainen        *error_r = t_strdup_printf("Failed to expand settings: %s", errstr);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen        mail_user_unref(&mail_user);
902222fb0928d1701f20a384b73f327b1d9a15ddTimo Sirainen        mail_storage_service_user_unref(&user);
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen        event_unref(&event);
902222fb0928d1701f20a384b73f327b1d9a15ddTimo Sirainen        return -1;
6c2ddb9f586e6392552ddfb82ab55e57fcfc4110Timo Sirainen    }
6c2ddb9f586e6392552ddfb82ab55e57fcfc4110Timo Sirainen
6c2ddb9f586e6392552ddfb82ab55e57fcfc4110Timo Sirainen    client = client_create(fd_in, fd_out, input->session_id,
6c2ddb9f586e6392552ddfb82ab55e57fcfc4110Timo Sirainen                   event, mail_user, user, imap_set, smtp_set);
1be964ec6d835f95b4fdebf02add9265d58ad290Timo Sirainen    client->userdb_fields = input->userdb_fields == NULL ? NULL :
1be964ec6d835f95b4fdebf02add9265d58ad290Timo Sirainen        p_strarray_dup(client->pool, input->userdb_fields);
2a6af811ea3de3cf9e2f15e446674dd21b0705f3Timo Sirainen    event_unref(&event);
0cb2e8eb55e70f8ebe1e8349bdf49e4cbe5d8834Timo Sirainen    *client_r = client;
    return 0;
}

static void main_stdio_run(const char *username)
{
    struct client *client;
    struct mail_storage_service_input input;
    const char *value, *error, *input_base64;

    i_zero(&input);
    input.module = input.service = "imap";
    input.username = username != NULL ? username : getenv("USER");
    if (input.username == NULL && IS_STANDALONE())
        input.username = getlogin();
    if (input.username == NULL)
        i_fatal("USER environment missing");
    if ((value = getenv("IP")) != NULL)
        (void)net_addr2ip(value, &input.remote_ip);
    if ((value = getenv("LOCAL_IP")) != NULL)
        (void)net_addr2ip(value, &input.local_ip);

    if (client_create_from_input(&input, STDIN_FILENO, STDOUT_FILENO,
                     &client, &error) < 0)
        i_fatal("%s", error);

    input_base64 = getenv("CLIENT_INPUT");
    if (input_base64 == NULL)
        client_add_input_capability(client, NULL, 0);
    else {
        const buffer_t *input_buf = t_base64_decode_str(input_base64);
        client_add_input_capability(client, input_buf->data, input_buf->used);
    }

    if (mail_namespaces_init(client->user, &error) < 0)
        i_fatal("%s", error);
    client_add_input_finalize(client);
    /* client may be destroyed now */
}

static void
login_client_connected(const struct master_login_client *login_client,
               const char *username, const char *const *extra_fields)
{
#define MSG_BYE_INTERNAL_ERROR "* BYE "MAIL_ERRSTR_CRITICAL_MSG"\r\n"
    struct mail_storage_service_input input;
    struct client *client;
    enum mail_auth_request_flags flags = login_client->auth_req.flags;
    const char *error;

    i_zero(&input);
    input.module = input.service = "imap";
    input.local_ip = login_client->auth_req.local_ip;
    input.remote_ip = login_client->auth_req.remote_ip;
    input.local_port = login_client->auth_req.local_port;
    input.remote_port = login_client->auth_req.remote_port;
    input.username = username;
    input.userdb_fields = extra_fields;
    input.session_id = login_client->session_id;
    if ((flags & MAIL_AUTH_REQUEST_FLAG_CONN_SECURED) != 0)
        input.conn_secured = TRUE;
    if ((flags & MAIL_AUTH_REQUEST_FLAG_CONN_SSL_SECURED) != 0)
        input.conn_ssl_secured = TRUE;

    if (client_create_from_input(&input, login_client->fd, login_client->fd,
                     &client, &error) < 0) {
        int fd = login_client->fd;

        if (write(fd, MSG_BYE_INTERNAL_ERROR,
              strlen(MSG_BYE_INTERNAL_ERROR)) < 0) {
            if (errno != EAGAIN && errno != EPIPE)
                i_error("write(client) failed: %m");
        }
        i_error("%s", error);
        i_close_fd(&fd);
        master_service_client_connection_destroyed(master_service);
        return;
    }
    if ((flags & MAIL_AUTH_REQUEST_FLAG_TLS_COMPRESSION) != 0)
        client->tls_compression = TRUE;
    client_add_input_capability(client, login_client->data,
             login_client->auth_req.data_size);

    /* finish initializing the user (see comment in main()) */
    if (mail_namespaces_init(client->user, &error) < 0) {
        if (write_full(login_client->fd, MSG_BYE_INTERNAL_ERROR,
                   strlen(MSG_BYE_INTERNAL_ERROR)) < 0)
            if (errno != EAGAIN && errno != EPIPE)
                i_error("write_full(client) failed: %m");

        i_error("%s", error);
        client_destroy(client, error);
        return;
    }

    client_add_input_finalize(client);
    /* client may be destroyed now */
}

static void login_client_failed(const struct master_login_client *client,
                const char *errormsg)
{
    struct client_input input;
    const char *msg;

    client_parse_input(client->data, client->auth_req.data_size, &input);
    msg = t_strdup_printf("%s NO ["IMAP_RESP_CODE_UNAVAILABLE"] %s\r\n",
                  input.tag, errormsg);
    if (write(client->fd, msg, strlen(msg)) < 0) {
        /* ignored */
    }
}

static void client_connected(struct master_service_connection *conn)
{
    /* when running standalone, we shouldn't even get here */
    i_assert(master_login != NULL);

    master_service_client_connection_accept(conn);
    if (strcmp(conn->name, "imap-master") == 0) {
        /* restoring existing IMAP connection (e.g. from imap-idle) */
        imap_master_client_create(conn->fd);
    } else {
        master_login_add(master_login, conn->fd);
    }
}

int main(int argc, char *argv[])
{
    static const struct setting_parser_info *set_roots[] = {
        &smtp_submit_setting_parser_info,
        &imap_setting_parser_info,
        NULL
    };
    struct master_login_settings login_set;
    enum master_service_flags service_flags =
        MASTER_SERVICE_FLAG_SEND_STATS;
    enum mail_storage_service_flags storage_service_flags =
        /*
         * We include MAIL_STORAGE_SERVICE_FLAG_NO_NAMESPACES so
         * that the mail_user initialization is fast and we can
         * quickly send back the OK response to LOGIN/AUTHENTICATE.
         * Otherwise we risk a very slow namespace initialization to
         * cause client timeouts on login.
         */
        MAIL_STORAGE_SERVICE_FLAG_NO_NAMESPACES;
    const char *username = NULL, *auth_socket_path = "auth-master";
    int c;

    i_zero(&login_set);
    login_set.postlogin_timeout_secs = MASTER_POSTLOGIN_TIMEOUT_DEFAULT;
    login_set.request_auth_token = TRUE;

    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;
    }

    if (IS_STANDALONE()) {
        service_flags |= MASTER_SERVICE_FLAG_STANDALONE |
            MASTER_SERVICE_FLAG_STD_CLIENT;
    } else {
        service_flags |= MASTER_SERVICE_FLAG_KEEP_CONFIG_OPEN;
    }

    master_service = master_service_init("imap", service_flags,
                         &argc, &argv, "a:Dt:u:");
    while ((c = master_getopt(master_service)) > 0) {
        switch (c) {
        case 'a':
            auth_socket_path = optarg;
            break;
        case 't':
            if (str_to_uint(optarg, &login_set.postlogin_timeout_secs) < 0 ||
                login_set.postlogin_timeout_secs == 0)
                i_fatal("Invalid -t parameter: %s", optarg);
            break;
        case 'u':
            storage_service_flags |=
                MAIL_STORAGE_SERVICE_FLAG_USERDB_LOOKUP;
            username = optarg;
            break;
        case 'D':
            imap_debug = TRUE;
            break;
        default:
            return FATAL_DEFAULT;
        }
    }

    master_service_set_die_callback(master_service, imap_die);

    /* plugins may want to add commands, so this needs to be called early */
    commands_init();
    imap_fetch_handlers_init();
    imap_master_clients_init();

    const char *error;
    if (t_abspath(auth_socket_path, &login_set.auth_socket_path, &error) < 0)
        i_fatal("t_abspath(%s) failed: %s", auth_socket_path, error);

    if (argv[optind] != NULL) {
        if (t_abspath(argv[optind], &login_set.postlogin_socket_path, &error) < 0)
            i_fatal("t_abspath(%s) failed: %s", argv[optind], error);
    }
    login_set.callback = login_client_connected;
    login_set.failure_callback = login_client_failed;

    if (!IS_STANDALONE())
        master_login = master_login_init(master_service, &login_set);

    storage_service =
        mail_storage_service_init(master_service,
                      set_roots, storage_service_flags);
    master_service_init_finish(master_service);
    /* NOTE: login_set.*_socket_path are now invalid due to data stack
       having been freed */

    /* fake that we're running, so we know if client was destroyed
       while handling its initial input */
    io_loop_set_running(current_ioloop);

    if (IS_STANDALONE()) {
        T_BEGIN {
            main_stdio_run(username);
        } T_END;
    } else {
        io_loop_set_running(current_ioloop);
    }

    if (io_loop_is_running(current_ioloop))
        master_service_run(master_service, client_connected);
    clients_destroy_all();

    if (master_login != NULL)
        master_login_deinit(&master_login);
    mail_storage_service_deinit(&storage_service);

    imap_fetch_handlers_deinit();
    commands_deinit();
    imap_master_clients_deinit();

    master_service_deinit(&master_service);
    return 0;
}