auth-master-connection.c revision 647eed98cb795ee9e42911750402dab720b57514
e59faf65ce864fe95dc00f5d52b8323cdbd0608aTimo Sirainen/* Copyright (c) 2002-2010 Dovecot authors, see the included COPYING file */
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
08d6658a4e2ec8104cd1307f6baa75fdb07a24f8Mark Washenberger#include "auth-common.h"
16f816d3f3c32ae3351834253f52ddd0212bcbf3Timo Sirainen#include "array.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "hash.h"
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen#include "str.h"
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen#include "strescape.h"
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen#include "str-sanitize.h"
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen#include "hostpid.h"
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen#include "hex-binary.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "ioloop.h"
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen#include "network.h"
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen#include "istream.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "ostream.h"
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen#include "master-service.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "userdb.h"
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen#include "userdb-blocking.h"
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen#include "master-interface.h"
657afb33796f8216c568ad813627da89970760beTimo Sirainen#include "auth-request-handler.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "auth-client-connection.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "auth-master-connection.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen#include <unistd.h>
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen#include <stdlib.h>
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen#define MAX_INBUF_SIZE 1024
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#define MAX_OUTBUF_SIZE (1024*50)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainenstruct master_userdb_request {
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen struct auth_master_connection *conn;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen unsigned int id;
6c2c5f20760b06bfb4a40b0ee2ef5ab016bc41f0Timo Sirainen struct auth_request *auth_request;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen};
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainenstruct master_list_iter_ctx {
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen struct auth_master_connection *conn;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen struct auth_userdb *userdb;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen struct userdb_iterate_context *iter;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen unsigned int id;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen bool failed;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen};
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainenstatic void master_input(struct auth_master_connection *conn);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo SirainenARRAY_TYPE(auth_master_connections) auth_master_connections;
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen
226259ee6fb9830dafc1a5ba1e95bf5a4345b406Timo Sirainenvoid auth_master_request_callback(struct auth_stream_reply *reply,
226259ee6fb9830dafc1a5ba1e95bf5a4345b406Timo Sirainen void *context)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen{
657afb33796f8216c568ad813627da89970760beTimo Sirainen struct auth_master_connection *conn = context;
657afb33796f8216c568ad813627da89970760beTimo Sirainen struct const_iovec iov[2];
226259ee6fb9830dafc1a5ba1e95bf5a4345b406Timo Sirainen const char *reply_str;
226259ee6fb9830dafc1a5ba1e95bf5a4345b406Timo Sirainen
226259ee6fb9830dafc1a5ba1e95bf5a4345b406Timo Sirainen reply_str = auth_stream_reply_export(reply);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen if (conn->auth->set->debug)
e5acc283bf030b0b5c79ca4e52d315c516a299faPascal Volk i_debug("master out: %s", reply_str);
9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen
226259ee6fb9830dafc1a5ba1e95bf5a4345b406Timo Sirainen iov[0].iov_base = reply_str;
226259ee6fb9830dafc1a5ba1e95bf5a4345b406Timo Sirainen iov[0].iov_len = strlen(reply_str);
657afb33796f8216c568ad813627da89970760beTimo Sirainen iov[1].iov_base = "\n";
657afb33796f8216c568ad813627da89970760beTimo Sirainen iov[1].iov_len = 1;
a7bee3930831a9261fa6180d02af29c484d862e9Timo Sirainen
657afb33796f8216c568ad813627da89970760beTimo Sirainen (void)o_stream_sendv(conn->output, iov, 2);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen}
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
6ef7e31619edfaa17ed044b45861d106a86191efTimo Sirainenstatic bool
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainenmaster_input_request(struct auth_master_connection *conn, const char *args)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen{
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen struct auth_client_connection *client_conn;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen const char *const *list;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen unsigned int id, client_pid, client_id;
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen uint8_t cookie[MASTER_AUTH_COOKIE_SIZE];
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen buffer_t buf;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen /* <id> <client-pid> <client-id> <cookie> */
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen list = t_strsplit(args, "\t");
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen if (str_array_length(list) < 4) {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen i_error("BUG: Master sent broken REQUEST");
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return FALSE;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen }
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen id = (unsigned int)strtoul(list[0], NULL, 10);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen client_pid = (unsigned int)strtoul(list[1], NULL, 10);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen client_id = (unsigned int)strtoul(list[2], NULL, 10);
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen buffer_create_data(&buf, cookie, sizeof(cookie));
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen if (hex_to_binary(list[3], &buf) < 0) {
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen i_error("BUG: Master sent broken REQUEST cookie");
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen return FALSE;
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen }
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen client_conn = auth_client_connection_lookup(client_pid);
657afb33796f8216c568ad813627da89970760beTimo Sirainen if (client_conn == NULL) {
657afb33796f8216c568ad813627da89970760beTimo Sirainen i_error("Master requested auth for nonexisting client %u",
657afb33796f8216c568ad813627da89970760beTimo Sirainen client_pid);
657afb33796f8216c568ad813627da89970760beTimo Sirainen (void)o_stream_send_str(conn->output,
647eed98cb795ee9e42911750402dab720b57514Timo Sirainen t_strdup_printf("FAIL\t%u\n", id));
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen } else if (memcmp(client_conn->cookie, cookie, sizeof(cookie)) != 0) {
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen i_error("Master requested auth for client %u with invalid cookie",
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen client_pid);
419baa2c17c63ae516b2df6cc5695f15aaccbff8Timo Sirainen (void)o_stream_send_str(conn->output,
647eed98cb795ee9e42911750402dab720b57514Timo Sirainen t_strdup_printf("FAIL\t%u\n", id));
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen } else {
657afb33796f8216c568ad813627da89970760beTimo Sirainen auth_request_handler_master_request(
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen client_conn->request_handler, conn, id, client_id);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen }
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return TRUE;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen}
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainenstatic int
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainenmaster_input_auth_request(struct auth_master_connection *conn, const char *args,
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen const char *cmd, struct auth_request **request_r,
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen const char **error_r)
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen{
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen struct auth_request *auth_request;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen const char *const *list, *name, *arg;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen /* <id> <userid> [<parameters>] */
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen list = t_strsplit(args, "\t");
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (list[0] == NULL || list[1] == NULL) {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen i_error("BUG: Master sent broken %s", cmd);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen return -1;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen }
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request = auth_request_new_dummy(conn->auth);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request->id = (unsigned int)strtoul(list[0], NULL, 10);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request->context = conn;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_master_connection_ref(conn);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (!auth_request_set_username(auth_request, list[1], error_r)) {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen *request_r = auth_request;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen return 0;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen }
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen for (list += 2; *list != NULL; list++) {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen arg = strchr(*list, '=');
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (arg == NULL) {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen name = *list;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen arg = "";
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen } else {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen name = t_strdup_until(*list, arg);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen arg++;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen }
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen (void)auth_request_import(auth_request, name, arg);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen }
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (auth_request->service == NULL) {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen i_error("BUG: Master sent %s request without service", cmd);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_master_connection_unref(&conn);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request_unref(&auth_request);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen return -1;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen }
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen *request_r = auth_request;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen return 1;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen}
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainenstatic void
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainenuser_callback(enum userdb_result result,
137ea7ca34005345aa2304a940149b7f3774d727Timo Sirainen struct auth_request *auth_request)
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen{
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen struct auth_master_connection *conn = auth_request->context;
6fabfb7bbfd88d0c1de66981e52850f26067623bTimo Sirainen struct auth_stream_reply *reply = auth_request->userdb_reply;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen string_t *str;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen
6fabfb7bbfd88d0c1de66981e52850f26067623bTimo Sirainen if (auth_request->userdb_lookup_failed)
6fabfb7bbfd88d0c1de66981e52850f26067623bTimo Sirainen result = USERDB_RESULT_INTERNAL_FAILURE;
6fabfb7bbfd88d0c1de66981e52850f26067623bTimo Sirainen
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen str = t_str_new(128);
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainen switch (result) {
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainen case USERDB_RESULT_INTERNAL_FAILURE:
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainen str_printfa(str, "FAIL\t%u", auth_request->id);
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainen break;
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainen case USERDB_RESULT_USER_UNKNOWN:
3ee2da6133bd7773961d1f3f3ac531448a5158b9Timo Sirainen str_printfa(str, "NOTFOUND\t%u", auth_request->id);
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainen break;
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainen case USERDB_RESULT_OK:
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen str_printfa(str, "USER\t%u\t", auth_request->id);
137ea7ca34005345aa2304a940149b7f3774d727Timo Sirainen str_append(str, auth_stream_reply_export(reply));
0f39a57760d93cddbce3ca43096d78e0fe2f42fdTimo Sirainen break;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen }
397165dbd694272487e29c8d48d619822da52abcTimo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen if (conn->auth->set->debug)
e5acc283bf030b0b5c79ca4e52d315c516a299faPascal Volk i_debug("master out: %s", str_c(str));
397165dbd694272487e29c8d48d619822da52abcTimo Sirainen
3ee2da6133bd7773961d1f3f3ac531448a5158b9Timo Sirainen str_append_c(str, '\n');
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen (void)o_stream_send(conn->output, str_data(str), str_len(str));
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen auth_request_unref(&auth_request);
451254b8f080792f1e6f11639a9e8be663a108c0Timo Sirainen auth_master_connection_unref(&conn);
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen}
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen
6ef7e31619edfaa17ed044b45861d106a86191efTimo Sirainenstatic bool
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainenmaster_input_user(struct auth_master_connection *conn, const char *args)
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen{
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen struct auth_request *auth_request;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen const char *error;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen int ret;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen ret = master_input_auth_request(conn, args, "USER",
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen &auth_request, &error);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (ret <= 0) {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (ret < 0)
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen return FALSE;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request_log_info(auth_request, "userdb", "%s", error);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen user_callback(USERDB_RESULT_USER_UNKNOWN, auth_request);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen } else {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request->state = AUTH_REQUEST_STATE_USERDB;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request_lookup_user(auth_request, user_callback);
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen }
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen return TRUE;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen}
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainenstatic void
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainenpass_callback(enum passdb_result result,
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen const unsigned char *credentials ATTR_UNUSED,
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen size_t size ATTR_UNUSED,
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen struct auth_request *auth_request)
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen{
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen struct auth_master_connection *conn = auth_request->context;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen struct auth_stream_reply *reply = auth_request->extra_fields;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen string_t *str;
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen str = t_str_new(128);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen switch (result) {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen case PASSDB_RESULT_OK:
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen str_printfa(str, "PASS\t%u\t", auth_request->id);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (reply != NULL)
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen str_append(str, auth_stream_reply_export(reply));
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen break;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen case PASSDB_RESULT_USER_UNKNOWN:
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen case PASSDB_RESULT_USER_DISABLED:
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen case PASSDB_RESULT_PASS_EXPIRED:
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen str_printfa(str, "NOTFOUND\t%u", auth_request->id);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen break;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen case PASSDB_RESULT_PASSWORD_MISMATCH:
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen case PASSDB_RESULT_INTERNAL_FAILURE:
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen case PASSDB_RESULT_SCHEME_NOT_AVAILABLE:
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen str_printfa(str, "FAIL\t%u", auth_request->id);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen break;
4f17af486518a4a7526dc6c4eed702178dc0bee9Timo Sirainen }
4f17af486518a4a7526dc6c4eed702178dc0bee9Timo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (conn->auth->set->debug)
e5acc283bf030b0b5c79ca4e52d315c516a299faPascal Volk i_debug("master out: %s", str_c(str));
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen str_append_c(str, '\n');
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen (void)o_stream_send(conn->output, str_data(str), str_len(str));
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request_unref(&auth_request);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_master_connection_unref(&conn);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen}
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainenstatic bool
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainenmaster_input_pass(struct auth_master_connection *conn, const char *args)
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen{
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen struct auth_request *auth_request;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen const char *error;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen int ret;
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen ret = master_input_auth_request(conn, args, "PASS",
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen &auth_request, &error);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (ret <= 0) {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen if (ret < 0)
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen return FALSE;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request_log_info(auth_request, "passdb", "%s", error);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen pass_callback(PASSDB_RESULT_USER_UNKNOWN,
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen NULL, 0, auth_request);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen } else {
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request->state = AUTH_REQUEST_STATE_MECH_CONTINUE;
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen auth_request_lookup_credentials(auth_request, "",
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen pass_callback);
1862352e2ef8ed1ef824368d9c0c2c0fce89962eTimo Sirainen }
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen return TRUE;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen}
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainenstatic void master_input_list_finish(struct master_list_iter_ctx *ctx)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen{
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->conn->io = io_add(ctx->conn->fd, IO_READ, master_input, ctx->conn);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (ctx->iter != NULL)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen (void)userdb_blocking_iter_deinit(&ctx->iter);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen o_stream_unset_flush_callback(ctx->conn->output);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen i_free(ctx);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen}
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainenstatic int master_output_list(struct master_list_iter_ctx *ctx)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen{
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen int ret;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if ((ret = o_stream_flush(ctx->conn->output)) < 0) {
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen master_input_list_finish(ctx);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return 1;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen }
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (ret > 0)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen userdb_blocking_iter_next(ctx->iter);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return 1;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen}
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainenstatic void master_input_list_callback(const char *user, void *context)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen{
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen struct master_list_iter_ctx *ctx = context;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen int ret;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (user == NULL) {
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (userdb_blocking_iter_deinit(&ctx->iter) < 0)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->failed = TRUE;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen do {
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->userdb = ctx->userdb->next;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen } while (ctx->userdb != NULL &&
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->userdb->userdb->iface->iterate_init == NULL);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (ctx->userdb == NULL) {
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen /* iteration is finished */
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen const char *str;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen str = t_strdup_printf("DONE\t%u\t%s\n", ctx->id,
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->failed ? "fail" : "");
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen (void)o_stream_send_str(ctx->conn->output, str);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen master_input_list_finish(ctx);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen }
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen /* continue iterating next userdb */
550034b94a58491a08a620c8e392fd3087551e7aTimo Sirainen ctx->iter = userdb_blocking_iter_init(ctx->userdb,
550034b94a58491a08a620c8e392fd3087551e7aTimo Sirainen master_input_list_callback, ctx);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen userdb_blocking_iter_next(ctx->iter);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen }
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen T_BEGIN {
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen const char *str;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen str = t_strdup_printf("LIST\t%u\t%s\n", ctx->id,
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen str_tabescape(user));
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ret = o_stream_send_str(ctx->conn->output, str);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen } T_END;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (ret < 0) {
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen /* disconnected, don't bother finishing */
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen master_input_list_finish(ctx);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen }
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (o_stream_get_buffer_used_size(ctx->conn->output) == 0)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen userdb_blocking_iter_next(ctx->iter);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen}
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainenstatic bool
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainenmaster_input_list(struct auth_master_connection *conn, const char *args)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen{
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen struct auth_userdb *userdb = conn->auth->userdbs;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen struct master_list_iter_ctx *ctx;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen const char *str;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen unsigned int id;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen /* <id> */
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (*args == '\0') {
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen i_error("BUG: Master sent broken LIST");
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return FALSE;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen }
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen id = strtoul(args, NULL, 10);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen while (userdb != NULL && userdb->userdb->iface->iterate_init == NULL)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen userdb = userdb->next;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen if (userdb == NULL) {
3278289d240da169166de8aa785273c5df904e64Timo Sirainen i_error("Trying to iterate users, but userdbs don't support it");
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen str = t_strdup_printf("DONE\t%u\tfail", id);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen (void)o_stream_send_str(conn->output, str);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return TRUE;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen }
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx = i_new(struct master_list_iter_ctx, 1);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->conn = conn;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->userdb = userdb;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->id = id;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen io_remove(&conn->io);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen o_stream_set_flush_callback(conn->output, master_output_list, ctx);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen ctx->iter = userdb_blocking_iter_init(ctx->userdb,
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen master_input_list_callback, ctx);
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return TRUE;
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen}
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainenstatic bool
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainenauth_master_input_line(struct auth_master_connection *conn, const char *line)
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen{
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen if (conn->auth->set->debug)
e5acc283bf030b0b5c79ca4e52d315c516a299faPascal Volk i_debug("master in: %s", line);
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen
3278289d240da169166de8aa785273c5df904e64Timo Sirainen if (strncmp(line, "USER\t", 5) == 0)
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen return master_input_user(conn, line + 5);
3278289d240da169166de8aa785273c5df904e64Timo Sirainen if (strncmp(line, "LIST\t", 5) == 0)
43d3ea2780b5f8557ede7b4c039e8f56cb8d357dTimo Sirainen return master_input_list(conn, line + 5);
0bb8c135a1c065655fc4d2d89b74b4f958fe4f02Timo Sirainen if (strncmp(line, "PASS\t", 5) == 0)
0bb8c135a1c065655fc4d2d89b74b4f958fe4f02Timo Sirainen return master_input_pass(conn, line + 5);
3278289d240da169166de8aa785273c5df904e64Timo Sirainen
3278289d240da169166de8aa785273c5df904e64Timo Sirainen if (!conn->userdb_only) {
3278289d240da169166de8aa785273c5df904e64Timo Sirainen if (strncmp(line, "REQUEST\t", 8) == 0)
3278289d240da169166de8aa785273c5df904e64Timo Sirainen return master_input_request(conn, line + 8);
3278289d240da169166de8aa785273c5df904e64Timo Sirainen if (strncmp(line, "CPID\t", 5) == 0) {
3278289d240da169166de8aa785273c5df904e64Timo Sirainen i_error("Authentication client trying to connect to "
3278289d240da169166de8aa785273c5df904e64Timo Sirainen "master socket");
3278289d240da169166de8aa785273c5df904e64Timo Sirainen return FALSE;
3278289d240da169166de8aa785273c5df904e64Timo Sirainen }
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen }
3278289d240da169166de8aa785273c5df904e64Timo Sirainen
0bb8c135a1c065655fc4d2d89b74b4f958fe4f02Timo Sirainen i_error("BUG: Unknown command in %s socket: %s",
0bb8c135a1c065655fc4d2d89b74b4f958fe4f02Timo Sirainen conn->userdb_only ? "userdb" : "master",
3278289d240da169166de8aa785273c5df904e64Timo Sirainen str_sanitize(line, 80));
3278289d240da169166de8aa785273c5df904e64Timo Sirainen return FALSE;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen}
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen
2cfe9983ce7a6280636ee12beccc2e865111967bTimo Sirainenstatic void master_input(struct auth_master_connection *conn)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen{
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen char *line;
6ef7e31619edfaa17ed044b45861d106a86191efTimo Sirainen bool ret;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen switch (i_stream_read(conn->input)) {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen case 0:
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen case -1:
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen /* disconnected */
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen auth_master_connection_destroy(&conn);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen case -2:
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen /* buffer full */
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen i_error("BUG: Master sent us more than %d bytes",
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen (int)MAX_INBUF_SIZE);
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen auth_master_connection_destroy(&conn);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen return;
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen }
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen if (!conn->version_received) {
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen line = i_stream_next_line(conn->input);
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen if (line == NULL)
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen return;
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen /* make sure the major version matches */
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen if (strncmp(line, "VERSION\t", 8) != 0 ||
4b8c92b4773677a7b4064816e469eeafc976ba75Timo Sirainen atoi(t_strcut(line + 8, '\t')) !=
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen AUTH_MASTER_PROTOCOL_MAJOR_VERSION) {
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen i_error("Master not compatible with this server "
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen "(mixed old and new binaries?)");
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen auth_master_connection_destroy(&conn);
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen return;
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen }
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen conn->version_received = TRUE;
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen }
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen while ((line = i_stream_next_line(conn->input)) != NULL) {
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen T_BEGIN {
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen ret = auth_master_input_line(conn, line);
19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen } T_END;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen if (!ret) {
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen auth_master_connection_destroy(&conn);
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen return;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen }
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen }
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen}
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2cfe9983ce7a6280636ee12beccc2e865111967bTimo Sirainenstatic int master_output(struct auth_master_connection *conn)
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen{
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen int ret;
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen if ((ret = o_stream_flush(conn->output)) < 0) {
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen /* transmit error, probably master died */
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen auth_master_connection_destroy(&conn);
211ed7806d8715ec2280ffbf5d10f0d6e4f1beb2Timo Sirainen return 1;
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen }
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen if (o_stream_get_buffer_used_size(conn->output) <= MAX_OUTBUF_SIZE/2) {
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen /* allow input again */
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen conn->io = io_add(conn->fd, IO_READ, master_input, conn);
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen }
211ed7806d8715ec2280ffbf5d10f0d6e4f1beb2Timo Sirainen return 1;
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen}
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainenstruct auth_master_connection *
3278289d240da169166de8aa785273c5df904e64Timo Sirainenauth_master_connection_create(struct auth *auth, int fd, bool userdb_only)
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen{
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen struct auth_master_connection *conn;
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen const char *line;
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen conn = i_new(struct auth_master_connection, 1);
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen conn->refcount = 1;
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen conn->fd = fd;
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen conn->auth = auth;
2526d52441ef368215ab6bf04fd0356d3b09d235Timo Sirainen conn->input = i_stream_create_fd(fd, MAX_INBUF_SIZE, FALSE);
93fa87cf1a96c4f279ec4f5c311820313ba12c34Timo Sirainen conn->output = o_stream_create_fd(fd, (size_t)-1, FALSE);
2767104d81e97a109f0aa9758792bfa1da325a97Timo Sirainen o_stream_set_flush_callback(conn->output, master_output, conn);
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen conn->io = io_add(fd, IO_READ, master_input, conn);
3278289d240da169166de8aa785273c5df904e64Timo Sirainen conn->userdb_only = userdb_only;
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen line = t_strdup_printf("VERSION\t%u\t%u\nSPID\t%s\n",
657afb33796f8216c568ad813627da89970760beTimo Sirainen AUTH_MASTER_PROTOCOL_MAJOR_VERSION,
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen AUTH_MASTER_PROTOCOL_MINOR_VERSION,
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen my_pid);
657afb33796f8216c568ad813627da89970760beTimo Sirainen (void)o_stream_send_str(conn->output, line);
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen array_append(&auth_master_connections, &conn, 1);
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen return conn;
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen}
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainenvoid auth_master_connection_destroy(struct auth_master_connection **_conn)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen{
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen struct auth_master_connection *conn = *_conn;
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen struct auth_master_connection *const *masters;
7bd72e4deca3cbf757dd1ea298486d9f3bc24226Timo Sirainen unsigned int idx;
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen *_conn = NULL;
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen if (conn->destroyed)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen return;
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen conn->destroyed = TRUE;
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen
7bd72e4deca3cbf757dd1ea298486d9f3bc24226Timo Sirainen array_foreach(&auth_master_connections, masters) {
7bd72e4deca3cbf757dd1ea298486d9f3bc24226Timo Sirainen if (*masters == conn) {
7bd72e4deca3cbf757dd1ea298486d9f3bc24226Timo Sirainen idx = array_foreach_idx(&auth_master_connections,
7bd72e4deca3cbf757dd1ea298486d9f3bc24226Timo Sirainen masters);
7bd72e4deca3cbf757dd1ea298486d9f3bc24226Timo Sirainen array_delete(&auth_master_connections, idx, 1);
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen break;
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen }
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen }
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen if (conn->input != NULL)
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen i_stream_close(conn->input);
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen if (conn->output != NULL)
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen o_stream_close(conn->output);
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen if (conn->io != NULL)
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen io_remove(&conn->io);
62d1665ff7842b61bc99be238ede3921ec08ccc5Timo Sirainen if (conn->fd != -1) {
62d1665ff7842b61bc99be238ede3921ec08ccc5Timo Sirainen if (close(conn->fd) < 0)
62d1665ff7842b61bc99be238ede3921ec08ccc5Timo Sirainen i_error("close(): %m");
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen conn->fd = -1;
62d1665ff7842b61bc99be238ede3921ec08ccc5Timo Sirainen }
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
cf0ad1a0bddb0787f3d7b408a96d721a8b2a98a3Timo Sirainen master_service_client_connection_destroyed(master_service);
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen auth_master_connection_unref(&conn);
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen}
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainenvoid auth_master_connection_ref(struct auth_master_connection *conn)
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen{
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen i_assert(conn->refcount > 0);
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen conn->refcount++;
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen}
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainenvoid auth_master_connection_unref(struct auth_master_connection **_conn)
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen{
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen struct auth_master_connection *conn = *_conn;
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen *_conn = NULL;
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen i_assert(conn->refcount > 0);
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen if (--conn->refcount > 0)
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen return;
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen if (conn->input != NULL)
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen i_stream_unref(&conn->input);
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen if (conn->output != NULL)
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen o_stream_unref(&conn->output);
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen i_free(conn);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen}
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainenvoid auth_master_connections_init(void)
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen{
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen i_array_init(&auth_master_connections, 16);
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen}
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainenvoid auth_master_connections_deinit(void)
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen{
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen struct auth_master_connection **masters;
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen unsigned int i, count;
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen masters = array_get_modifiable(&auth_master_connections, &count);
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen for (i = count; i > 0; i--)
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen auth_master_connection_destroy(&masters[i-1]);
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen array_free(&auth_master_connections);
d176f84ce5ca2073f4dfbafb457b9c74f6bf0d76Timo Sirainen}