master-login.c revision 9ba5c6da815d5d4b43861387dd08fcea321a0423
e59faf65ce864fe95dc00f5d52b8323cdbd0608aTimo Sirainen/* Copyright (c) 2009-2010 Dovecot authors, see the included COPYING file */
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "lib.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "ioloop.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "ostream.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "fdpass.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "fd-close-on-exec.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "llist.h"
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen#include "str.h"
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen#include "strescape.h"
3ba9a079592f46e94ce846e5aa80e4d479cd5e41Timo Sirainen#include "master-service-private.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "master-login.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "master-login-auth.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include <sys/stat.h>
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include <unistd.h>
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen#define MASTER_LOGIN_POSTLOGIN_TIMEOUT_MSECS (60*1000)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen#define master_login_conn_is_closed(conn) \
8149f4a09d07464f33de747d8e1e620f6d9f7faeTimo Sirainen ((conn)->fd == -1)
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenstruct master_login_connection {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login_connection *prev, *next;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login *login;
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen int refcount;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen int fd;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct io *io;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct ostream *output;
baca06331782e2752734199486e51a26d7c93d75Timo Sirainen
baca06331782e2752734199486e51a26d7c93d75Timo Sirainen unsigned int login_success:1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen};
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainenstruct master_login_postlogin {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen struct master_login_client *client;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen int fd;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen struct io *io;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen struct timeout *to;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen string_t *input;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen};
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenstruct master_login {
3ba9a079592f46e94ce846e5aa80e4d479cd5e41Timo Sirainen struct master_service *service;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen master_login_callback_t *callback;
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen master_login_failure_callback_t *failure_callback;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login_connection *conns;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login_auth *auth;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen char *postlogin_socket_path;
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen unsigned int stopping:1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen};
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainenstatic void master_login_conn_close(struct master_login_connection *conn);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainenstatic void master_login_conn_unref(struct master_login_connection **_conn);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenstruct master_login *
3ba9a079592f46e94ce846e5aa80e4d479cd5e41Timo Sirainenmaster_login_init(struct master_service *service, const char *auth_socket_path,
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen const char *postlogin_socket_path,
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen master_login_callback_t *callback,
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen master_login_failure_callback_t *failure_callback)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen{
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login *login;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen login = i_new(struct master_login, 1);
3ba9a079592f46e94ce846e5aa80e4d479cd5e41Timo Sirainen login->service = service;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen login->callback = callback;
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen login->failure_callback = failure_callback;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen login->auth = master_login_auth_init(auth_socket_path);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen login->postlogin_socket_path = i_strdup(postlogin_socket_path);
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen i_assert(service->login == NULL);
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen service->login = login;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return login;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen}
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenvoid master_login_deinit(struct master_login **_login)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen{
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login *login = *_login;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen *_login = NULL;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen i_assert(login->service->login == login);
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen login->service->login = NULL;
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen master_login_auth_deinit(&login->auth);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen while (login->conns != NULL) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login_connection *conn = login->conns;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen master_login_conn_close(conn);
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen master_login_conn_unref(&conn);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_free(login->postlogin_socket_path);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_free(login);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen}
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenstatic int
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenmaster_login_conn_read_request(struct master_login_connection *conn,
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_auth_request *req_r,
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen unsigned char data[MASTER_AUTH_MAX_DATA_SIZE],
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen int *client_fd_r)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen{
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct stat st;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen ssize_t ret;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen *client_fd_r = -1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen ret = fd_read(conn->fd, req_r, sizeof(*req_r), client_fd_r);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (ret != sizeof(*req_r)) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (ret == 0) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen /* disconnected */
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen } else if (ret > 0) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen /* request wasn't fully read */
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_error("fd_read() partial input (%d/%d)",
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen (int)ret, (int)sizeof(*req_r));
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen } else {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (errno == EAGAIN)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return 0;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_error("fd_read() failed: %m");
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return -1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (req_r->data_size != 0) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (req_r->data_size > MASTER_AUTH_MAX_DATA_SIZE) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_error("Too large auth data_size sent");
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return -1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen /* @UNSAFE */
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen ret = read(conn->fd, data, req_r->data_size);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (ret != (ssize_t)req_r->data_size) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (ret == 0) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen /* disconnected */
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen } else if (ret > 0) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen /* request wasn't fully read */
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_error("Data read partially %d/%u",
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen (int)ret, req_r->data_size);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen } else {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_error("read(data) failed: %m");
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return -1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (*client_fd_r == -1) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_error("Auth request missing a file descriptor");
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return -1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (fstat(*client_fd_r, &st) < 0) {
47f1c273dfaea552c034ffbecdec096f7cbaef35Timo Sirainen i_error("fstat(fd_read client) failed: %m");
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return -1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (st.st_ino != req_r->ino) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_error("Auth request inode mismatch: %s != %s",
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen dec2str(st.st_ino), dec2str(req_r->ino));
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return -1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return 1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen}
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainenstatic void master_login_client_free(struct master_login_client **_client)
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen{
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen struct master_login_client *client = *_client;
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen *_client = NULL;
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen if (client->fd != -1) {
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen if (close(client->fd) < 0)
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen i_error("close(fd_read client) failed: %m");
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen }
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen /* FIXME: currently we create a separate connection for each request,
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen so close the connection after we're done with this client */
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen if (!master_login_conn_is_closed(client->conn))
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen master_login_conn_unref(&client->conn);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen master_login_conn_unref(&client->conn);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen i_free(client);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen}
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainenstatic void master_login_auth_finish(struct master_login_client *client,
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen const char *const *auth_args)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen{
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen struct master_login *login = client->conn->login;
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen struct master_service *service = login->service;
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen bool close_sockets;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen close_sockets = service->master_status.available_count == 0 &&
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen service->service_count_left == 1;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
baca06331782e2752734199486e51a26d7c93d75Timo Sirainen client->conn->login_success = TRUE;
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen login->callback(client, auth_args[0], auth_args+1);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen if (close_sockets) {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen /* we're dying as soon as this connection closes. */
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen i_assert(master_login_auth_request_count(login->auth) == 0);
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen master_login_auth_disconnect(login->auth);
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen master_service_close_config_fd(service);
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen } else if (login->stopping) {
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen /* try stopping again */
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen master_login_stop(login);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
f096367f0f7b0e481f8abe0969afdf2d8250057eTimo Sirainen
f096367f0f7b0e481f8abe0969afdf2d8250057eTimo Sirainen client->fd = -1;
f096367f0f7b0e481f8abe0969afdf2d8250057eTimo Sirainen master_login_client_free(&client);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen}
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainenstatic void master_login_postlogin_free(struct master_login_postlogin *pl)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen{
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen timeout_remove(&pl->to);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen io_remove(&pl->io);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (close(pl->fd) < 0)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_error("close(postlogin) failed: %m");
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen str_free(&pl->input);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_free(pl);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen}
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainenstatic void master_login_postlogin_input(struct master_login_postlogin *pl)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen{
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen struct master_login *login = pl->client->conn->login;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen char buf[1024];
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen const char **auth_args, **p;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen unsigned int len;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen ssize_t ret;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen int fd = -1;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen while ((ret = fd_read(pl->fd, buf, sizeof(buf), &fd)) > 0) {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (fd != -1) {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen /* post-login script replaced fd */
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (close(pl->client->fd) < 0)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_error("close(client) failed: %m");
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen pl->client->fd = fd;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen str_append_n(pl->input, buf, ret);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen len = str_len(pl->input);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (len > 0 && str_c(pl->input)[len-1] == '\n') {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen /* finished reading the input */
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen str_truncate(pl->input, len-1);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen } else {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (ret < 0) {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (errno == EAGAIN)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen return;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_error("fd_read(%s) failed: %m",
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen login->postlogin_socket_path);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen } else {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_error("fd_read(%s) failed: disconnected",
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen login->postlogin_socket_path);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen master_login_client_free(&pl->client);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen master_login_postlogin_free(pl);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen return;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen auth_args = t_strsplit(str_c(pl->input), "\t");
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen for (p = auth_args; *p != NULL; p++)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen *p = str_tabunescape(t_strdup_noconst(*p));
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen master_login_auth_finish(pl->client, auth_args);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen master_login_postlogin_free(pl);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen}
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainenstatic void master_login_postlogin_timeout(struct master_login_postlogin *pl)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen{
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen struct master_login *login = pl->client->conn->login;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_error("%s: Timeout waiting for post-login script to finish, aborting",
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen login->postlogin_socket_path);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen master_login_client_free(&pl->client);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen master_login_postlogin_free(pl);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen}
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainenstatic int master_login_postlogin(struct master_login_client *client,
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen const char *const *auth_args)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen{
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen struct master_login *login = client->conn->login;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen struct master_login_postlogin *pl;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen string_t *str;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen unsigned int i;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen int fd;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen ssize_t ret;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen fd = net_connect_unix_with_retries(login->postlogin_socket_path, 1000);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (fd == -1) {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_error("net_connect_unix(%s) failed: %m",
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen login->postlogin_socket_path);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen return -1;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen str = t_str_new(256);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen str_printfa(str, "%s\t%s", net_ip2addr(&client->auth_req.local_ip),
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen net_ip2addr(&client->auth_req.remote_ip));
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen for (i = 0; auth_args[i] != NULL; i++) {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen str_append_c(str, '\t');
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen str_tabescape_write(str, auth_args[i]);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen str_append_c(str, '\n');
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen ret = fd_send(fd, client->fd, str_data(str), str_len(str));
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (ret != (ssize_t)str_len(str)) {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (ret < 0) {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_error("write(%s) failed: %m",
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen login->postlogin_socket_path);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen } else {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen i_error("write(%s) failed: partial write",
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen login->postlogin_socket_path);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen (void)close(fd);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen return -1;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen }
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen net_set_nonblock(fd, TRUE);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen pl = i_new(struct master_login_postlogin, 1);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen pl->client = client;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen pl->fd = fd;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen pl->io = io_add(fd, IO_READ, master_login_postlogin_input, pl);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen pl->to = timeout_add(MASTER_LOGIN_POSTLOGIN_TIMEOUT_MSECS,
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen master_login_postlogin_timeout, pl);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen pl->input = str_new(default_pool, 512);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen return 0;
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen}
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenstatic void
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainenmaster_login_auth_callback(const char *const *auth_args, const char *errormsg,
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen void *context)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen{
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login_client *client = context;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_auth_reply reply;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen memset(&reply, 0, sizeof(reply));
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen reply.tag = client->auth_req.tag;
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen reply.status = errormsg == NULL ? MASTER_AUTH_STATUS_OK :
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen MASTER_AUTH_STATUS_INTERNAL_ERROR;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen reply.mail_pid = getpid();
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen o_stream_send(client->conn->output, &reply, sizeof(reply));
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen if (errormsg != NULL || auth_args[0] == NULL) {
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen if (auth_args != NULL) {
b71f152acb8a197d20b709ca74366e6d765bd200Timo Sirainen i_error("login client: Username missing from auth reply");
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen errormsg = MASTER_AUTH_ERRMSG_INTERNAL_FAILURE;
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen }
3b22894b8805b186c73d8b754001e8d7e944be85Timo Sirainen client->conn->login->failure_callback(client, errormsg);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen master_login_client_free(&client);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen if (client->conn->login->postlogin_socket_path == NULL)
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen master_login_auth_finish(client, auth_args);
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen else {
1d3b9fce06b466bcf64f9ab7b622f3a6e4e939baTimo Sirainen /* execute post-login scripts before finishing auth */
baca06331782e2752734199486e51a26d7c93d75Timo Sirainen if (master_login_postlogin(client, auth_args) < 0)
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen master_login_client_free(&client);
3ba9a079592f46e94ce846e5aa80e4d479cd5e41Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen}
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenstatic void master_login_conn_input(struct master_login_connection *conn)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen{
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_auth_request req;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login_client *client;
442232f2d1cfdf28f3a18aa00a5c19246d321036Timo Sirainen struct master_login *login = conn->login;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen unsigned char data[MASTER_AUTH_MAX_DATA_SIZE];
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen int ret, client_fd;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen ret = master_login_conn_read_request(conn, &req, data, &client_fd);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (ret <= 0) {
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen if (ret < 0) {
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen master_login_conn_close(conn);
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen master_login_conn_unref(&conn);
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (client_fd != -1) {
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (close(client_fd) < 0)
47f1c273dfaea552c034ffbecdec096f7cbaef35Timo Sirainen i_error("close(fd_read client) failed: %m");
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen return;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen }
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen fd_close_on_exec(client_fd, TRUE);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen /* @UNSAFE: we have a request. do userdb lookup for it. */
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen client = i_malloc(sizeof(struct master_login_client) + req.data_size);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen client->conn = conn;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen client->fd = client_fd;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen client->auth_req = req;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen memcpy(client->data, data, req.data_size);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen conn->refcount++;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
442232f2d1cfdf28f3a18aa00a5c19246d321036Timo Sirainen master_login_auth_request(login->auth, &req,
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen master_login_auth_callback, client);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen}
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainenvoid master_login_add(struct master_login *login, int fd)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen{
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen struct master_login_connection *conn;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen conn = i_new(struct master_login_connection, 1);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen conn->refcount = 1;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen conn->login = login;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen conn->fd = fd;
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen conn->io = io_add(conn->fd, IO_READ, master_login_conn_input, conn);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen conn->output = o_stream_create_fd(fd, (size_t)-1, FALSE);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen DLLIST_PREPEND(&login->conns, conn);
442232f2d1cfdf28f3a18aa00a5c19246d321036Timo Sirainen
baca06331782e2752734199486e51a26d7c93d75Timo Sirainen /* NOTE: currently there's a separate connection for each request. */
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen}
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainenstatic void master_login_conn_close(struct master_login_connection *conn)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen{
8149f4a09d07464f33de747d8e1e620f6d9f7faeTimo Sirainen if (master_login_conn_is_closed(conn))
f096367f0f7b0e481f8abe0969afdf2d8250057eTimo Sirainen return;
f096367f0f7b0e481f8abe0969afdf2d8250057eTimo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen DLLIST_REMOVE(&conn->login->conns, conn);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen
442232f2d1cfdf28f3a18aa00a5c19246d321036Timo Sirainen if (conn->io != NULL)
442232f2d1cfdf28f3a18aa00a5c19246d321036Timo Sirainen io_remove(&conn->io);
d795594a018d8dcf47b6b569b8ca20e53e275293Timo Sirainen o_stream_close(conn->output);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen if (close(conn->fd) < 0)
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen i_error("close(master login) failed: %m");
f096367f0f7b0e481f8abe0969afdf2d8250057eTimo Sirainen conn->fd = -1;
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen}
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainenstatic void master_login_conn_unref(struct master_login_connection **_conn)
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen{
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen struct master_login_connection *conn = *_conn;
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen i_assert(conn->refcount > 0);
4e56e6408815c04f2e5b904a648a366a2dcbd408Timo Sirainen
d795594a018d8dcf47b6b569b8ca20e53e275293Timo Sirainen if (--conn->refcount > 0)
d795594a018d8dcf47b6b569b8ca20e53e275293Timo Sirainen return;
d795594a018d8dcf47b6b569b8ca20e53e275293Timo Sirainen
d795594a018d8dcf47b6b569b8ca20e53e275293Timo Sirainen *_conn = NULL;
e871bfe83e8c5cc7768de30afe0127a3c4373adeTimo Sirainen master_login_conn_close(conn);
d795594a018d8dcf47b6b569b8ca20e53e275293Timo Sirainen o_stream_unref(&conn->output);
9ba5c6da815d5d4b43861387dd08fcea321a0423Timo Sirainen
9ba5c6da815d5d4b43861387dd08fcea321a0423Timo Sirainen if (!conn->login_success)
9ba5c6da815d5d4b43861387dd08fcea321a0423Timo Sirainen master_service_client_connection_destroyed(conn->login->service);
d795594a018d8dcf47b6b569b8ca20e53e275293Timo Sirainen i_free(conn);
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen}
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainenvoid master_login_stop(struct master_login *login)
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen{
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen login->stopping = TRUE;
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen if (master_login_auth_request_count(login->auth) == 0) {
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen master_login_auth_disconnect(login->auth);
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen master_service_close_config_fd(login->service);
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen }
7891c8e6debdcfec552cb1beea2a0230fe89957bTimo Sirainen}