auth-master-connection.c revision 2526d52441ef368215ab6bf04fd0356d3b09d235
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen/* Copyright (C) 2002 Timo Sirainen */
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "common.h"
16f816d3f3c32ae3351834253f52ddd0212bcbf3Timo Sirainen#include "array.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "buffer.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "hash.h"
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen#include "str.h"
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen#include "str-sanitize.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "ioloop.h"
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen#include "istream.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "ostream.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "network.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "userdb.h"
657afb33796f8216c568ad813627da89970760beTimo Sirainen#include "auth-request-handler.h"
b0df0e9a8ed8889ad4bf032043ab245ce8851fdeTimo Sirainen#include "auth-master-interface.h"
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen#include "auth-client-connection.h"
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen#include "auth-master-listener.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
657afb33796f8216c568ad813627da89970760beTimo Sirainenvoid auth_master_request_callback(const char *reply, void *context)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen{
657afb33796f8216c568ad813627da89970760beTimo Sirainen struct auth_master_connection *conn = context;
657afb33796f8216c568ad813627da89970760beTimo Sirainen struct const_iovec iov[2];
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen if (conn->listener->auth->verbose_debug)
9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen i_info("master out: %s", reply);
9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen
657afb33796f8216c568ad813627da89970760beTimo Sirainen iov[0].iov_base = reply;
657afb33796f8216c568ad813627da89970760beTimo Sirainen iov[0].iov_len = strlen(reply);
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;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen /* <id> <client-pid> <client-id> */
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen list = t_strsplit(args, "\t");
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen if (list[0] == NULL || list[1] == NULL || list[2] == NULL) {
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);
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen client_conn = auth_client_connection_lookup(conn->listener, 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,
657afb33796f8216c568ad813627da89970760beTimo Sirainen t_strdup_printf("NOTFOUND\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
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
397165dbd694272487e29c8d48d619822da52abcTimo Sirainen if (conn->listener->auth->verbose_debug)
397165dbd694272487e29c8d48d619822da52abcTimo Sirainen i_info("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);
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;
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen const char *const *list, *name, *arg;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen /* <id> <userid> [<parameters>] */
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen list = t_strsplit(args, "\t");
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen if (list[0] == NULL || list[1] == NULL) {
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen i_error("BUG: Master sent broken USER");
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen return FALSE;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen }
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen auth_request = auth_request_new_dummy(conn->listener->auth);
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen auth_request->id = (unsigned int)strtoul(list[0], NULL, 10);
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen auth_request->user = p_strdup(auth_request->pool, list[1]);
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen auth_request->context = conn;
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen for (list += 2; *list != NULL; list++) {
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen arg = strchr(*list, '=');
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen if (arg == NULL) {
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen name = *list;
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen arg = "";
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen } else {
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen name = t_strdup_until(*list, arg);
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen arg++;
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen }
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen (void)auth_request_import(auth_request, name, arg);
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen }
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen if (auth_request->service == NULL) {
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen i_error("BUG: Master sent USER request without service");
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen auth_request_unref(&auth_request);
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen return FALSE;
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen }
70c181da837ed85fc5b0426c010b65609bda5329Timo Sirainen
2703c3499e2d422d4f26e62720a4e8a43164fa57Timo Sirainen auth_request->state = AUTH_REQUEST_STATE_USERDB;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen auth_request_lookup_user(auth_request, user_callback);
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen return TRUE;
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen}
9398c0935613ba038cf2275ff66c43b25092cfd0Timo 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) {
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen if (conn->listener->auth->verbose_debug)
9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen i_info("master in: %s", line);
9f431ccfb6932746db56245c8a3d3415717ef545Timo Sirainen
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen t_push();
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen if (strncmp(line, "REQUEST\t", 8) == 0)
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen ret = master_input_request(conn, line + 8);
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen else if (strncmp(line, "USER\t", 5) == 0)
9398c0935613ba038cf2275ff66c43b25092cfd0Timo Sirainen ret = master_input_user(conn, line + 5);
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen else if (strncmp(line, "CPID\t", 5) == 0) {
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen i_error("Authentication client trying to connect to "
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen "master socket");
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen ret = FALSE;
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen } else {
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen /* ignore unknown command */
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen i_error("BUG: Unknown command in master socket: %s",
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen str_sanitize(line, 80));
5cda7e699876c1de203c30777924c899d2a20221Timo Sirainen ret = FALSE;
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen }
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen t_pop();
73bfdbe28c2ce6d143eadf0bab8ccfbe4cab0faeTimo Sirainen
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 *
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainenauth_master_connection_create(struct auth_master_listener *listener, int fd)
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen{
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen struct auth_master_connection *conn;
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen conn = i_new(struct auth_master_connection, 1);
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen conn->listener = listener;
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen conn->refcount = 1;
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen conn->fd = fd;
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);
6b46a500174ace25494b8f0547283eb60dc13756Timo Sirainen
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen array_append(&listener->masters, &conn, 1);
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen return conn;
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen}
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainenvoid auth_master_connection_send_handshake(struct auth_master_connection *conn)
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen{
657afb33796f8216c568ad813627da89970760beTimo Sirainen const char *line;
657afb33796f8216c568ad813627da89970760beTimo Sirainen
657afb33796f8216c568ad813627da89970760beTimo Sirainen if (conn->output == NULL)
657afb33796f8216c568ad813627da89970760beTimo Sirainen return;
657afb33796f8216c568ad813627da89970760beTimo Sirainen
657afb33796f8216c568ad813627da89970760beTimo Sirainen line = t_strdup_printf("VERSION\t%u\t%u\nSPID\t%u\n",
657afb33796f8216c568ad813627da89970760beTimo Sirainen AUTH_MASTER_PROTOCOL_MAJOR_VERSION,
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen AUTH_MASTER_PROTOCOL_MINOR_VERSION,
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen conn->listener->pid);
657afb33796f8216c568ad813627da89970760beTimo Sirainen (void)o_stream_send_str(conn->output, line);
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;
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen struct auth_master_connection *const *conns;
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen unsigned int i, count;
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen
d5cebe7f98e63d4e2822863ef2faa4971e8b3a5dTimo Sirainen *_conn = NULL;
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen if (conn->destroyed)
2a90d8a14b0e7cc1508814bc87d3dfa598ef46a8Timo Sirainen return;
76213404317a7ed17bec0beadb5137c82785d816Timo Sirainen conn->destroyed = TRUE;
76213404317a7ed17bec0beadb5137c82785d816Timo 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
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen conns = array_get(&conn->listener->masters, &count);
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen for (i = 0; i < count; i++) {
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen if (conns[i] == conn) {
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen array_delete(&conn->listener->masters, i, 1);
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen break;
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen }
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen }
ad49932dae8ba31e07544b66bbc4f4de707a751cTimo Sirainen if (!standalone && auth_master_listeners_masters_left() == 0)
645397402d1d0c755cd485ecf721b59b6babc874Timo Sirainen io_loop_stop(ioloop);
ecb1b2d6236942bf82f822e8d0167f0e160b206dTimo Sirainen
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}