client-connection-http.c

/* Copyright (c) 2010-2017 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "compat.h"
#include "lib-signals.h"
#include "base64.h"
#include "ioloop.h"
#include "str.h"
#include "istream.h"
#include "ostream.h"
#include "strescape.h"
#include "settings-parser.h"
#include "iostream-ssl.h"
#include "iostream-temp.h"
#include "istream-seekable.h"
#include "master-service.h"
#include "master-service-ssl.h"
#include "master-service-settings.h"
#include "mail-storage-service.h"
#include "http-server.h"
#include "http-request.h"
#include "http-response.h"
#include "http-url.h"
#include "doveadm-util.h"
#include "doveadm-server.h"
#include "doveadm-mail.h"
#include "doveadm-print.h"
#include "doveadm-settings.h"
#include "client-connection-private.h"
#include "json-parser.h"

#include <unistd.h>
#include <ctype.h>

struct client_connection_http {
	struct client_connection conn;

	struct http_server_connection *http_conn;

	struct http_server_request *http_server_request;
	const struct http_request *http_request;
	struct http_server_response *http_response;

	struct io *io;
	struct istream *input;
	struct ostream *output;

	struct json_parser *json_parser;

	const struct doveadm_cmd_ver2 *cmd;
	struct doveadm_cmd_param *cmd_param;
	struct ioloop *ioloop;
	ARRAY_TYPE(doveadm_cmd_param_arr_t) pargv;
	int method_err;
	char *method_id;
	bool first_row;
	bool value_is_array;

	enum {
		JSON_STATE_INIT,
		JSON_STATE_COMMAND,
		JSON_STATE_COMMAND_NAME,
		JSON_STATE_COMMAND_PARAMETERS,
		JSON_STATE_COMMAND_PARAMETER_KEY,
		JSON_STATE_COMMAND_PARAMETER_VALUE,
		JSON_STATE_COMMAND_PARAMETER_VALUE_ARRAY,
		JSON_STATE_COMMAND_PARAMETER_VALUE_ISTREAM_CONSUME,
		JSON_STATE_COMMAND_ID,
		JSON_STATE_COMMAND_DONE,
		JSON_STATE_DONE
	} json_state;
};

typedef void doveadm_server_handler_t(struct client_connection_http *ctx);

struct doveadm_http_server_mount {
	const char *verb;
	const char *path;
	doveadm_server_handler_t *handler;
	bool auth;
};

static struct http_server *doveadm_http_server;

static void doveadm_http_server_handle_request(void *context, struct http_server_request *req);
static void doveadm_http_server_connection_destroy(void *context, const char *reason);

static const struct http_server_callbacks doveadm_http_callbacks = {
        .connection_destroy = doveadm_http_server_connection_destroy,
        .handle_request = doveadm_http_server_handle_request
};

static void doveadm_http_server_options_handler(struct client_connection_http *);
static void doveadm_http_server_print_mounts(struct client_connection_http *);
static void doveadm_http_server_send_api_v1(struct client_connection_http *);
static void doveadm_http_server_read_request_v1(struct client_connection_http *);

static struct doveadm_http_server_mount doveadm_http_server_mounts[] = {
{
	.verb = "OPTIONS",
	.path = NULL,
	.handler = doveadm_http_server_options_handler,
	.auth = FALSE
},{
	.verb = "GET",
	.path = "/",
	.handler = doveadm_http_server_print_mounts,
	.auth = TRUE
},{
	.verb = "GET",
	.path = "/doveadm/v1",
	.handler = doveadm_http_server_send_api_v1,
	.auth = TRUE
},{
	.verb = "POST",
	.path = "/doveadm/v1",
	.handler = doveadm_http_server_read_request_v1,
	.auth = TRUE
}
};

static void doveadm_http_server_send_response(void *context);

static void
client_connection_http_free(struct client_connection *_conn)
{
	struct client_connection_http *conn =
		(struct client_connection_http *)_conn;

	io_remove(&conn->io);
	o_stream_destroy(&conn->output);
	i_stream_destroy(&conn->input);

	if (conn->http_conn != NULL) {
		/* We're not in the lib-http/server's connection destroy callback. */
		http_server_connection_close(&conn->http_conn,
			"Server shutting down");
	}
}

struct client_connection *
client_connection_http_create(int fd, bool ssl)
{
	struct client_connection_http *conn;
	pool_t pool;

	pool = pool_alloconly_create("doveadm client", 1024*16);
	conn = p_new(pool, struct client_connection_http, 1);

	if (client_connection_init(&conn->conn,
		CLIENT_CONNECTION_TYPE_HTTP, pool, fd) < 0) {
		pool_unref(&conn->conn.pool);
		return NULL;
	}
	conn->conn.free = client_connection_http_free;

	conn->http_conn = http_server_connection_create(doveadm_http_server,
			fd, fd, ssl, &doveadm_http_callbacks, conn);
	return &conn->conn;
}

static void
doveadm_http_server_connection_destroy(void *context,
	const char *reason ATTR_UNUSED)
{
	struct client_connection_http *conn =
		(struct client_connection_http *)context;
	struct client_connection *bconn = &conn->conn;

	if (conn->http_conn == NULL) {
		/* already destroying client directly */
		return;
	}

	/* HTTP connection is destroyed already now */
	conn->http_conn = NULL;

	/* destroy the connection itself */
	client_connection_destroy(&bconn);
}

static void
doveadm_http_server_request_destroy(struct client_connection_http *conn)
{
	struct http_server_request *http_sreq = conn->http_server_request;
	const struct http_request *http_req = conn->http_request;
	struct http_server_response *http_resp =
		http_server_request_get_response(http_sreq);

	if (http_resp != NULL) {
		const char *agent, *url, *reason;
		uoff_t size;
		int status;

		http_server_response_get_status(http_resp, &status, &reason);
		size = http_server_response_get_total_size(http_resp);
		agent = http_request_header_get(http_req, "User-Agent");
		if (agent == NULL) agent = "";

		url = http_url_create(http_req->target.url);
		i_info("doveadm: %s %s %s \"%s %s "
		       "HTTP/%d.%d\" %d %"PRIuUOFF_T" \"%s\" \"%s\"",
		       net_ip2addr(&conn->conn.remote_ip), "-", "-",
		       http_req->method, http_req->target.url->path,
		       http_req->version_major, http_req->version_minor,
		       status, size, url, agent);
	}
	if (conn->json_parser != NULL) {
		const char *error ATTR_UNUSED;
		(void)json_parser_deinit(&conn->json_parser, &error);
		// we've already failed, ignore error
	}
	if (conn->output != NULL)
		o_stream_set_no_error_handling(conn->output, TRUE);

	http_server_request_unref(&(conn->http_server_request));
	http_server_switch_ioloop(doveadm_http_server);
}

static void doveadm_http_server_json_error(void *context, const char *error)
{
	struct client_connection_http *conn = context;
	struct ostream *output = conn->output;
	string_t *escaped;

	escaped = str_new(conn->conn.pool, 10);

	o_stream_nsend_str(output, "[\"error\",{\"type\":\"");
	json_append_escaped(escaped, error);
	o_stream_nsend_str(output, str_c(escaped));
	o_stream_nsend_str(output, "\", \"exitCode\":");
        str_truncate(escaped,0);
	str_printfa(escaped, "%d", doveadm_exit_code);
	o_stream_nsend_str(output, str_c(escaped));
	o_stream_nsend_str(output, "},\"");
	str_truncate(escaped,0);
	if (conn->method_id != NULL) {
	        json_append_escaped(escaped, conn->method_id);
		o_stream_nsend_str(output, str_c(escaped));
	}
	o_stream_nsend_str(output, "\"]");
}

static void doveadm_http_server_json_success(void *context, struct istream *result)
{
	struct client_connection_http *conn = context;
	struct ostream *output = conn->output;
	string_t *escaped;

	escaped = str_new(conn->conn.pool, 10);

	o_stream_nsend_str(output, "[\"doveadmResponse\",");
	o_stream_nsend_istream(output, result);
	o_stream_nsend_str(output, ",\"");
	if (conn->method_id != NULL) {
		json_append_escaped(escaped, conn->method_id);
		o_stream_nsend_str(output, str_c(escaped));
	}
	o_stream_nsend_str(output, "\"]");
}

static int doveadm_http_server_istream_read(struct client_connection_http *conn)
{
	struct istream *v_input = conn->cmd_param->value.v_istream;
	const unsigned char *data;
	size_t size;

	while (i_stream_read_more(v_input, &data, &size) > 0)
		i_stream_skip(v_input, size);
	if (!v_input->eof)
		return 0;

	if (v_input->stream_errno != 0) {
		i_error("read(%s) failed: %s",
			i_stream_get_name(v_input),
			i_stream_get_error(v_input));
		conn->method_err = 400;
		return -1;
	}
	return 1;
}

/**
 * this is to ensure we can handle arrays and other special parameter types
 */
static int doveadm_http_server_json_parse_next(struct client_connection_http *conn, enum json_type *type, const char **value)
{
	const char *tmp;
	int ret;

	switch (conn->json_state) {
	case JSON_STATE_COMMAND_PARAMETER_VALUE_ISTREAM_CONSUME:
		ret = doveadm_http_server_istream_read(conn);
		if (ret != 1)
			return ret;
		conn->json_state = JSON_STATE_COMMAND_PARAMETER_KEY;
		break;
	case JSON_STATE_COMMAND_PARAMETER_VALUE_ARRAY:
		/* reading through parameters in an array */
		while ((ret = json_parse_next(conn->json_parser,
					     type, value)) > 0) {
			if (*type == JSON_TYPE_ARRAY_END)
				break;
			if (*type != JSON_TYPE_STRING)
				return -2;
			tmp = p_strdup(conn->conn.pool,*value);
			array_append(&conn->cmd_param->value.v_array, &tmp, 1);
		}
		if (ret <= 0)
			return ret;
		conn->json_state = JSON_STATE_COMMAND_PARAMETER_KEY;
		break;
	case JSON_STATE_COMMAND_PARAMETER_VALUE:
		if (conn->cmd_param->type == CMD_PARAM_ISTREAM) {
			struct istream* is[2] = {0};

			ret = json_parse_next_stream(conn->json_parser, &is[0]);
			if (ret != 1)
				return ret;
			conn->cmd_param->value.v_istream = i_stream_create_seekable_path(is, IO_BLOCK_SIZE, "/tmp/doveadm.");
			i_stream_unref(&is[0]);
			conn->cmd_param->value_set = TRUE;
			conn->json_state = JSON_STATE_COMMAND_PARAMETER_VALUE_ISTREAM_CONSUME;
			return doveadm_http_server_json_parse_next(conn, type, value);
		}
		ret = json_parse_next(conn->json_parser, type, value);
		if (ret != 1)
			return ret;
		if (conn->cmd_param->type == CMD_PARAM_ARRAY) {
			p_array_init(&conn->cmd_param->value.v_array, conn->conn.pool, 1);
			conn->cmd_param->value_set = TRUE;
			if (*type == JSON_TYPE_ARRAY) {
				/* start of array */
				conn->value_is_array = TRUE;
				conn->json_state = JSON_STATE_COMMAND_PARAMETER_VALUE_ARRAY;
				return doveadm_http_server_json_parse_next(conn, type, value);
			}
			if (*type != JSON_TYPE_STRING) {
				/* FIXME: should handle other than string too */
				return -2;
			}
			tmp = p_strdup(conn->conn.pool, *value);
			array_append(&conn->cmd_param->value.v_array, &tmp, 1);
		} else {
			conn->cmd_param->value_set = TRUE;
			switch(conn->cmd_param->type) {
			case CMD_PARAM_BOOL:
				conn->cmd_param->value.v_bool = (strcmp(*value,"true") == 0);
				break;
			case CMD_PARAM_INT64:
				if (str_to_int64(*value, &conn->cmd_param->value.v_int64) != 0) {
					conn->method_err = 400;
				}
				break;
			case CMD_PARAM_IP:
				if (net_addr2ip(*value, &conn->cmd_param->value.v_ip) != 0) {
					conn->method_err = 400;
				}
				break;
			case CMD_PARAM_STR:
				conn->cmd_param->value.v_string = p_strdup(conn->conn.pool, *value);
				break;
			default:
				break;
			}
		}
		conn->json_state = JSON_STATE_COMMAND_PARAMETER_KEY;
		break;
	default:
		break;
	}

	return json_parse_next(conn->json_parser, type, value); /* just get next */
}

static void
doveadm_http_server_command_execute(struct client_connection_http *conn)
{
	struct doveadm_cmd_context cctx;
	struct istream *is;
	const char *user;
	struct ioloop *ioloop, *prev_ioloop;

	/* final preflight check */
	if (conn->method_err == 0 && !doveadm_client_is_allowed_command(conn->conn.set, conn->cmd->name))
		conn->method_err = 403;
	if (conn->method_err != 0) {
		if (conn->method_err == 404) {
			doveadm_http_server_json_error(conn, "unknownMethod");
		} else if (conn->method_err == 403) {
			doveadm_http_server_json_error(conn, "unAuthorized");
		} else if (conn->method_err == 400) {
			doveadm_http_server_json_error(conn, "invalidRequest");
		} else {
			doveadm_http_server_json_error(conn, "internalError");
		}
		return;
	}

	prev_ioloop = current_ioloop;
	i_zero(&cctx);
	cctx.conn_type = conn->conn.type;
	cctx.input = conn->input;
	cctx.output = conn->output;

	// create iostream
	doveadm_print_ostream = iostream_temp_create("/tmp/doveadm.", 0);
	cctx.cmd = conn->cmd;

	if ((cctx.cmd->flags & CMD_FLAG_NO_PRINT) == 0)
		doveadm_print_init(DOVEADM_PRINT_TYPE_JSON);

	/* then call it */
	doveadm_cmd_params_null_terminate_arrays(&conn->pargv);
	cctx.argv = array_get(&conn->pargv, (unsigned int*)&cctx.argc);
	ioloop = io_loop_create();
	doveadm_exit_code = 0;

	cctx.local_ip = conn->conn.local_ip;
	cctx.local_port = conn->conn.local_port;
	cctx.remote_ip = conn->conn.remote_ip;
	cctx.remote_port = conn->conn.remote_port;

	if (doveadm_cmd_param_str(&cctx, "user", &user))
		i_info("Executing command '%s' as '%s'", cctx.cmd->name, user);
	else
		i_info("Executing command '%s'", cctx.cmd->name);
	client_connection_set_proctitle(&conn->conn, cctx.cmd->name);
	cctx.cmd->cmd(&cctx);
	client_connection_set_proctitle(&conn->conn, "");

	io_loop_set_current(prev_ioloop);
	o_stream_switch_ioloop(conn->output);
	io_loop_set_current(ioloop);
	io_loop_destroy(&ioloop);

	if ((cctx.cmd->flags & CMD_FLAG_NO_PRINT) == 0)
		doveadm_print_deinit();
	if (o_stream_nfinish(doveadm_print_ostream) < 0) {
		i_info("Error writing output in command %s: %s",
		       conn->cmd->name,
		       o_stream_get_error(conn->output));
		doveadm_exit_code = EX_TEMPFAIL;
	}

	is = iostream_temp_finish(&doveadm_print_ostream, 4096);

	if (conn->first_row == TRUE)
		conn->first_row = FALSE;
	else
		o_stream_nsend_str(conn->output,",");

	if (doveadm_exit_code != 0) {
		if (doveadm_exit_code == 0 || doveadm_exit_code == EX_TEMPFAIL)
			i_error("Command %s failed", conn->cmd->name);
		doveadm_http_server_json_error(conn, "exitCode");
	} else {
		doveadm_http_server_json_success(conn, is);
	}
	i_stream_unref(&is);
}

static bool
doveadm_http_handle_json_v1(struct client_connection_http *conn,
			    enum json_type type, const char *value)
{
	const struct doveadm_cmd_ver2 *ccmd;
	struct doveadm_cmd_param *par;
	bool found;

	switch (conn->json_state) {
	case JSON_STATE_INIT:
		if (type != JSON_TYPE_ARRAY)
			return FALSE;
		conn->json_state = JSON_STATE_COMMAND;
		conn->first_row = TRUE;
		o_stream_nsend_str(conn->output,"[");
		return TRUE;
	case JSON_STATE_COMMAND:
		if (type == JSON_TYPE_ARRAY_END) {
			conn->json_state = JSON_STATE_DONE;
			return TRUE;
		}
		if (type != JSON_TYPE_ARRAY)
			return FALSE;
		conn->method_err = 0;
		p_free_and_null(conn->conn.pool, conn->method_id);
		conn->cmd = NULL;
		doveadm_cmd_params_clean(&conn->pargv);
		conn->json_state = JSON_STATE_COMMAND_NAME;
		return TRUE;
	case JSON_STATE_COMMAND_NAME:
		if (type != JSON_TYPE_STRING)
			return FALSE;
		/* see if we can find it */
		found = FALSE;
		array_foreach(&doveadm_cmds_ver2, ccmd) {
			if (i_strccdascmp(ccmd->name, value) == 0) {
				conn->cmd = ccmd;
				found = TRUE;
				break;
			}
		}
		if (!found) {
			json_parse_skip_next(conn->json_parser);
			conn->json_state = JSON_STATE_COMMAND_ID;
			conn->method_err = 404;
		} else {
			struct doveadm_cmd_param *param;
			int pargc;

			/* initialize pargv */
			for (pargc = 0; conn->cmd->parameters[pargc].name != NULL; pargc++) {
				param = array_append_space(&conn->pargv);
				*param = conn->cmd->parameters[pargc];
				param->value_set = FALSE;
			}
			conn->json_state = JSON_STATE_COMMAND_PARAMETERS;
		}
		return TRUE;
	case JSON_STATE_COMMAND_PARAMETERS:
		if (type == JSON_TYPE_OBJECT_END) {
			conn->json_state = JSON_STATE_COMMAND_ID;
			return TRUE;
		}
		if (type != JSON_TYPE_OBJECT)
			return FALSE;
		conn->json_state = JSON_STATE_COMMAND_PARAMETER_KEY;
		return TRUE;
	case JSON_STATE_COMMAND_PARAMETER_KEY:
		if (type == JSON_TYPE_OBJECT_END) {
			conn->json_state = JSON_STATE_COMMAND_ID;
			return TRUE;
		}
		// can happen...
		if (type != JSON_TYPE_OBJECT_KEY && type != JSON_TYPE_STRING)
			return FALSE;
		/* go hunting */
		found = FALSE;
		array_foreach_modifiable(&conn->pargv, par) {
			if (i_strccdascmp(par->name, value) == 0) {
				/* it's already set, cannot have same key twice in json */
				if (par->value_set)
					return FALSE;
				conn->cmd_param = par;
				found = TRUE;
				break;
			}
		}
		/* skip parameters if error has already occurred */
		if (!found || conn->method_err != 0) {
			json_parse_skip_next(conn->json_parser);
			conn->json_state = JSON_STATE_COMMAND_PARAMETER_KEY;
			conn->method_err = 400;
		} else {
			if (conn->cmd_param->value_set) {
				// FIXME: should be returned as error to client, not logged
				i_info("Parameter %s already set",
				       conn->cmd_param->name);
				return FALSE;
			}
			conn->value_is_array = FALSE;
			conn->json_state = JSON_STATE_COMMAND_PARAMETER_VALUE;
		}
		return TRUE;
	case JSON_STATE_COMMAND_ID:
		if (type != JSON_TYPE_STRING)
			return FALSE;
		conn->method_id = p_strdup(conn->conn.pool, value);
		conn->json_state = JSON_STATE_COMMAND_DONE;
		return TRUE;
	case JSON_STATE_COMMAND_DONE:
		/* should be end of array */
		if (type != JSON_TYPE_ARRAY_END)
			return FALSE;
		doveadm_http_server_command_execute(conn);
		conn->json_state = JSON_STATE_COMMAND;
		return TRUE;
	case JSON_STATE_DONE:
		// FIXME: should be returned as error to client, not logged
		i_info("Got unexpected elements in JSON data");
		return TRUE;
	default:
		break;
	}
	i_unreached();
}

static void
doveadm_http_server_read_request_v1(struct client_connection_http *conn)
{
	struct http_server_request *http_sreq = conn->http_server_request;
	enum json_type type;
	const char *value, *error;
	int ret;

	if (conn->json_parser == NULL) {
		conn->json_parser = json_parser_init_flags(conn->input, JSON_PARSER_NO_ROOT_OBJECT);
	}

	while ((ret = doveadm_http_server_json_parse_next(conn, &type, &value)) == 1) {
		if (!doveadm_http_handle_json_v1(conn, type, value))
			break;
	}

	if (!conn->input->eof && ret == 0)
		return;
	io_remove(&conn->io);

	doveadm_cmd_params_clean(&conn->pargv);

	if (ret == -2 || (ret == 1 && conn->json_state != JSON_STATE_DONE)) {
		/* this will happen if the parser above runs into unexpected element, but JSON is OK */
		http_server_request_fail_close(http_sreq, 400, "Unexpected element in input");
		// FIXME: should be returned as error to client, not logged
		i_info("unexpected element");
		return;
	}

	if (conn->input->stream_errno != 0) {
		http_server_request_fail_close(http_sreq, 400, "Client disconnected");
		i_info("read(client) failed: %s",
		       i_stream_get_error(conn->input));
		return;
	}

	if (json_parser_deinit(&conn->json_parser, &error) != 0) {
		// istream JSON parsing failures do not count as errors
		http_server_request_fail_close(http_sreq, 400, "Invalid JSON input");
		// FIXME: should be returned as error to client, not logged
		i_info("JSON parse error: %s", error);
		return;
	}
	o_stream_nsend_str(conn->output,"]");

	doveadm_http_server_send_response(conn);
}

static void doveadm_http_server_camelcase_value(string_t *value)
{
	size_t i, k;
	char *ptr = str_c_modifiable(value);

	for (i = 0, k = 0; i < strlen(ptr);) {
		if (ptr[i] == ' ' || ptr[i] == '-') {
			i++;
			ptr[k++] = i_toupper(ptr[i++]);
		} else {
			ptr[k++] = ptr[i++];
		}
	}
	str_truncate(value, k);
}

static void
doveadm_http_server_send_api_v1(struct client_connection_http *conn)
{
	struct ostream *output = conn->output;
	const struct doveadm_cmd_ver2 *cmd;
	const struct doveadm_cmd_param *par;
	unsigned int i, k;
	string_t *tmp;
	bool sent;

	tmp = str_new(conn->conn.pool, 8);

	o_stream_nsend_str(output,"[\n");
	for (i = 0; i < array_count(&doveadm_cmds_ver2); i++) {
		cmd = array_idx(&doveadm_cmds_ver2, i);
		if (i > 0)
			o_stream_nsend_str(output, ",\n");
		o_stream_nsend_str(output, "\t{\"command\":\"");
		json_append_escaped(tmp, cmd->name);
		doveadm_http_server_camelcase_value(tmp);
		o_stream_nsend_str(output, str_c(tmp));
		o_stream_nsend_str(output, "\", \"parameters\":[");

		sent = FALSE;
		for (k = 0; cmd->parameters[k].name != NULL; k++) {
			str_truncate(tmp, 0);
			par = &(cmd->parameters[k]);
			if ((par->flags & CMD_PARAM_FLAG_DO_NOT_EXPOSE) != 0)
				continue;
			if (sent)
				o_stream_nsend_str(output, ",\n");
			else
				o_stream_nsend_str(output, "\n");
			sent = TRUE;
			o_stream_nsend_str(output, "\t\t{\"name\":\"");
			json_append_escaped(tmp, par->name);
			doveadm_http_server_camelcase_value(tmp);
			o_stream_nsend_str(output, str_c(tmp));
			o_stream_nsend_str(output, "\",\"type\":\"");
			switch(par->type) {
			case CMD_PARAM_BOOL:
				o_stream_nsend_str(output, "boolean");
				break;
			case CMD_PARAM_INT64:
				o_stream_nsend_str(output, "integer");
				break;
			case CMD_PARAM_ARRAY:
				o_stream_nsend_str(output, "array");
				break;
			case CMD_PARAM_IP:
			case CMD_PARAM_ISTREAM:
			case CMD_PARAM_STR:
				o_stream_nsend_str(output, "string");
			}
			o_stream_nsend_str(output, "\"}");
		}
		if (k > 0)
			o_stream_nsend_str(output,"\n\t");
		o_stream_nsend_str(output,"]}");
		str_truncate(tmp, 0);
	}
	o_stream_nsend_str(output,"\n]");
	doveadm_http_server_send_response(conn);
}

static void
doveadm_http_server_options_handler(struct client_connection_http *conn)
{
	struct http_server_response *http_resp = conn->http_response;

	http_server_response_add_header(http_resp,
		"Access-Control-Allow-Origin", "*");
	http_server_response_add_header(http_resp,
		"Access-Control-Allow-Methods", "GET, POST, OPTIONS");
	http_server_response_add_header(http_resp,
		"Access-Control-Allow-Request-Headers",
		"Content-Type, X-API-Key, Authorization");
	http_server_response_add_header(http_resp,
		"Access-Control-Allow-Headers",
		"Content-Type, WWW-Authenticate");
	doveadm_http_server_send_response(conn);
}

static void
doveadm_http_server_print_mounts(struct client_connection_http *conn)
{
	struct ostream *output = conn->output;
	unsigned int i;

	o_stream_nsend_str(output, "[\n");
	for (i = 0; i < N_ELEMENTS(doveadm_http_server_mounts); i++) {
		if (i > 0)
			o_stream_nsend_str(output, ",\n");
		o_stream_nsend_str(output, "{\"method\":\"");
		if (doveadm_http_server_mounts[i].verb == NULL)
			o_stream_nsend_str(output, "*");
		else
			o_stream_nsend_str(output, doveadm_http_server_mounts[i].verb);
		o_stream_nsend_str(output, "\",\"path\":\"");
		if (doveadm_http_server_mounts[i].path == NULL)
			o_stream_nsend_str(output, "*");
		else
			o_stream_nsend_str(output, doveadm_http_server_mounts[i].path);
		o_stream_nsend_str(output, "\"}");
	}
	o_stream_nsend_str(output, "\n]");
	doveadm_http_server_send_response(conn);
}

static bool
doveadm_http_server_authorize_request(struct client_connection_http *conn)
{
	struct http_server_request *http_sreq = conn->http_server_request;
	bool auth = FALSE;
	struct http_auth_credentials creds;

	/* no authentication specified */
	if (doveadm_settings->doveadm_api_key[0] == '\0' &&
		*conn->conn.set->doveadm_password == '\0') {
		conn->http_response = http_server_response_create(http_sreq, 500, "Internal Server Error");
		i_error("No authentication defined in configuration. Add API key or password");
		return FALSE;
	}
	if (http_server_request_get_auth(http_sreq, &creds) == 1) {
		/* see if the mech is supported */
		if (strcasecmp(creds.scheme, "Basic") == 0 && *conn->conn.set->doveadm_password != '\0') {
			string_t *b64_value = str_new(conn->conn.pool, 32);
			char *value;

			value = p_strdup_printf(conn->conn.pool, "doveadm:%s", conn->conn.set->doveadm_password);
			base64_encode(value, strlen(value), b64_value);
			if (creds.data != NULL && strcmp(creds.data, str_c(b64_value)) == 0)
				auth = TRUE;
			else
				i_error("Invalid authentication attempt to HTTP API");
		} else if (strcasecmp(creds.scheme, "X-Dovecot-API") == 0 && doveadm_settings->doveadm_api_key[0] != '\0') {
			string_t *b64_value = str_new(conn->conn.pool, 32);

			base64_encode(doveadm_settings->doveadm_api_key, strlen(doveadm_settings->doveadm_api_key), b64_value);
			if (creds.data != NULL && strcmp(creds.data, str_c(b64_value)) == 0)
				auth = TRUE;
			else
				i_error("Invalid authentication attempt to HTTP API");
		} else {
			i_error("Unsupported authentication scheme to HTTP API");
		}
	}
	if (auth == FALSE) {
		conn->http_response = http_server_response_create(http_sreq, 401, "Authentication required");
		if (doveadm_settings->doveadm_api_key[0] != '\0') {
			http_server_response_add_header(conn->http_response,
				"WWW-Authenticate", "X-Dovecot-API"
			);
		}
		if (*conn->conn.set->doveadm_password != '\0') {
			http_server_response_add_header(conn->http_response,
				"WWW-Authenticate", "Basic Realm=\"doveadm\""
			);
		}
	}
	return auth;
}

static void
doveadm_http_server_handle_request(void *context, struct http_server_request *http_sreq)
{
	struct client_connection_http *conn = context;
	const struct http_request *http_req =
		http_server_request_get(http_sreq);
	struct doveadm_http_server_mount *ep = NULL;
	unsigned int i;

	conn->http_server_request = http_sreq;
	conn->http_request = http_req;

	http_server_request_set_destroy_callback(http_sreq, doveadm_http_server_request_destroy, conn);
	http_server_request_ref(conn->http_server_request);

	for (i = 0; i < N_ELEMENTS(doveadm_http_server_mounts); i++) {
		if (doveadm_http_server_mounts[i].verb == NULL ||
		    strcmp(http_req->method, doveadm_http_server_mounts[i].verb) == 0) {
			if (doveadm_http_server_mounts[i].path == NULL ||
                            strcmp(http_req->target.url->path, doveadm_http_server_mounts[i].path) == 0) {
				ep = &doveadm_http_server_mounts[i];
				break;
			}
		}
	}

	if (ep == NULL) {
		http_server_request_fail_close(http_sreq, 404, "Path Not Found");
		return;
	}

 	if (ep->auth == TRUE && !doveadm_http_server_authorize_request(conn)) {
		doveadm_http_server_send_response(conn);
		return;
	}

	conn->http_response = http_server_response_create(http_sreq, 200, "OK");
	http_server_response_add_header(conn->http_response, "Content-Type",
		"application/json; charset=utf-8");

	if (strcmp(http_req->method, "POST") == 0) {
		/* handle request */
		conn->input = http_req->payload;
		i_stream_set_name(conn->input, net_ip2addr(&conn->conn.remote_ip));
		i_stream_ref(conn->input);
		conn->io = io_add_istream(conn->input, *ep->handler, conn);
		conn->output = iostream_temp_create_named
			("/tmp/doveadm.", 0, net_ip2addr(&conn->conn.remote_ip));
		p_array_init(&conn->pargv, conn->conn.pool, 5);
		ep->handler(conn);
	} else {
		conn->output = iostream_temp_create_named
			("/tmp/doveadm.", 0, net_ip2addr(&conn->conn.remote_ip));
		ep->handler(conn);
	}
}

static void doveadm_http_server_send_response(void *context)
{
	struct client_connection_http *conn = context;
	struct http_server_response *http_resp = conn->http_response;

	if (conn->output != NULL) {
		if (o_stream_nfinish(conn->output) == -1) {
			i_info("error writing output: %s",
			       o_stream_get_error(conn->output));
			o_stream_destroy(&conn->output);
			http_server_response_update_status(http_resp, 500, "Internal server error");
		} else {
			// send the payload response
			struct istream *is;

			is = iostream_temp_finish(&conn->output, IO_BLOCK_SIZE);
			http_server_response_set_payload(http_resp, is);
			i_stream_unref(&is);
		}
	}
	// submit response
	http_server_response_submit_close(http_resp);
}

static struct http_server_settings http_server_set = {
	.max_client_idle_time_msecs = 0,
        .max_pipelined_requests = 0
};

void doveadm_http_server_init(void)
{
	http_server_set.rawlog_dir = doveadm_settings->doveadm_http_rawlog_dir;
	doveadm_http_server = http_server_init(&http_server_set);
}

void doveadm_http_server_deinit(void)
{
	http_server_deinit(&doveadm_http_server);
}