dict-client.c revision 1cfe369d1278a9e2e6cc2e46103db48c493e1f21
02c335c23bf5fa225a467c19f2c063fb0dc7b8c3Timo Sirainen/* Copyright (c) 2005-2016 Dovecot authors, see the included COPYING file */
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen#include "lib.h"
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen#include "array.h"
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen#include "llist.h"
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen#include "str.h"
cbe2d47fda811fc56ba59cb2811c3fc26033b03cTimo Sirainen#include "strescape.h"
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen#include "time-util.h"
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen#include "connection.h"
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen#include "ostream.h"
87e603100685f9eb7faf995d2600edd0a578dd0cTimo Sirainen#include "eacces-error.h"
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen#include "dict-private.h"
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen#include "dict-client.h"
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen#include <unistd.h>
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen#include <fcntl.h>
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen/* Disconnect from dict server after this many milliseconds of idling after
8f079b04c103e073e57fa8b85cf69b14b0260ea4Timo Sirainen sending a command. Because dict server does blocking dict accesses, it can
8f079b04c103e073e57fa8b85cf69b14b0260ea4Timo Sirainen handle only one client at a time. This is why the default timeout is zero,
8f079b04c103e073e57fa8b85cf69b14b0260ea4Timo Sirainen so that there won't be many dict processes just doing nothing. Zero means
8f079b04c103e073e57fa8b85cf69b14b0260ea4Timo Sirainen that the socket is disconnected immediately after returning to ioloop. */
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen#define DICT_CLIENT_DEFAULT_TIMEOUT_MSECS 0
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen
98da0024e7340e036f0aa9371e9400176df18ebfTimo Sirainen/* Abort dict lookup after this many seconds. */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen#define DICT_CLIENT_REQUEST_TIMEOUT_MSECS 30000
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen/* Log a warning if dict lookup takes longer than this many milliseconds. */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen#define DICT_CLIENT_REQUEST_WARN_TIMEOUT_MSECS 5000
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstruct client_dict_cmd {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen int refcount;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct timeval start_time;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen char *query;
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen char *commit_query_human;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen bool retry_errors;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen bool no_replies;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen bool unfinished;
701eb90460d6c57845dc4e0bf595a5d0b90b01c1Timo Sirainen bool background;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen void (*callback)(struct client_dict_cmd *cmd,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *line, const char *error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_iterate_context *iter;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_lookup_callback_t *lookup;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_transaction_commit_callback_t *commit;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen void *context;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } api_callback;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen};
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstruct dict_connection {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct connection conn;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen};
98da0024e7340e036f0aa9371e9400176df18ebfTimo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstruct client_dict {
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct dict dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_connection conn;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen char *uri, *username;
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen enum dict_data_type value_type;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
fc0c06ee7b1e5c29a3faabd05f40fa0d26785dffTimo Sirainen time_t last_failed_connect;
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen char *last_connect_error;
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct ioloop *ioloop, *prev_ioloop;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct timeout *to_requests;
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen struct timeout *to_idle;
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen unsigned int idle_msecs;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct timeval last_input;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ARRAY(struct client_dict_cmd *) cmds;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen struct client_dict_transaction_context *transactions;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen unsigned int transaction_id_counter;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen};
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstruct client_dict_iter_result {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *key, *value;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen};
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstruct client_dict_iterate_context {
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct dict_iterate_context ctx;
055389c58fa3915e12fb4e72ec86782ce77c5c72Timo Sirainen char *error;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen pool_t results_pool;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ARRAY(struct client_dict_iter_result) results;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen unsigned int result_idx;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen bool async;
9e7a30991415f714b609f650105596371a58216dTimo Sirainen bool finished;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen bool deinit;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen};
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstruct client_dict_transaction_context {
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct dict_transaction_context ctx;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen struct client_dict_transaction_context *prev, *next;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen char *first_query;
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen char *error;
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen unsigned int id;
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen unsigned int query_count;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
0dffa25d211be541ee3c953b23566a1a990789dfTimo Sirainen bool sent_begin:1;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen};
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic struct connection_list *dict_connections;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
b5052fbfdbc2678cc8f12899afe55c998f43b740Timo Sirainenstatic int client_dict_connect(struct client_dict *dict, const char **error_r);
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainenstatic void client_dict_disconnect(struct client_dict *dict, const char *reason);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic struct client_dict_cmd *
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_cmd_init(struct client_dict *dict, const char *query)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *cmd;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
38fa43690a1cdc1917245f5f834ec40a89e83d91Timo Sirainen io_loop_time_refresh();
38fa43690a1cdc1917245f5f834ec40a89e83d91Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd = i_new(struct client_dict_cmd, 1);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->refcount = 1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->dict = dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->query = i_strdup(query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->start_time = ioloop_timeval;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return cmd;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void client_dict_cmd_ref(struct client_dict_cmd *cmd)
4210fac11e936f7edd6f3deecd1b23466534ce6aTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_assert(cmd->refcount > 0);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->refcount++;
4210fac11e936f7edd6f3deecd1b23466534ce6aTimo Sirainen}
4210fac11e936f7edd6f3deecd1b23466534ce6aTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic bool client_dict_cmd_unref(struct client_dict_cmd *cmd)
684cbcce30cc97340ee43240de5c0d236c917d29Timo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_assert(cmd->refcount > 0);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (--cmd->refcount > 0)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return TRUE;
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen i_free(cmd->commit_query_human);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_free(cmd->query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_free(cmd);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return FALSE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
e080054d7df103a9b6be62a476cbee18261acf6cTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void dict_pre_api_callback(struct client_dict *dict)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->prev_ioloop != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* Don't let callback see that we've created our
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen internal ioloop in case it wants to add some ios
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen or timeouts. */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen current_ioloop = dict->prev_ioloop;
684cbcce30cc97340ee43240de5c0d236c917d29Timo Sirainen }
684cbcce30cc97340ee43240de5c0d236c917d29Timo Sirainen}
684cbcce30cc97340ee43240de5c0d236c917d29Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void dict_post_api_callback(struct client_dict *dict)
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->prev_ioloop != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen current_ioloop = dict->ioloop;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* stop client_dict_wait() */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_stop(dict->ioloop);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen}
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic bool
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainendict_cmd_callback_line(struct client_dict_cmd *cmd, const char *line)
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->unfinished = FALSE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->callback(cmd, line, NULL);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return !cmd->unfinished;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen}
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainenstatic void
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainendict_cmd_callback_error(struct client_dict_cmd *cmd, const char *error)
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->unfinished = FALSE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (cmd->callback != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->callback(cmd, NULL, error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_assert(!cmd->unfinished);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void client_dict_input_timeout(struct client_dict *dict)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen int diff = timeval_diff_msecs(&ioloop_timeval, &dict->last_input);
79b0f6ec3e5fde8f878bd85f4a1c3d669af66ebeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_disconnect(dict, t_strdup_printf(
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen "Timeout: No input from dict for %u.%03u secs",
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen diff/1000, diff%1000));
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen}
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic int
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_cmd_query_send(struct client_dict *dict, const char *query)
98da0024e7340e036f0aa9371e9400176df18ebfTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct const_iovec iov[2];
98da0024e7340e036f0aa9371e9400176df18ebfTimo Sirainen ssize_t ret;
98da0024e7340e036f0aa9371e9400176df18ebfTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen iov[0].iov_base = query;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen iov[0].iov_len = strlen(query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen iov[1].iov_base = "\n";
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen iov[1].iov_len = 1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ret = o_stream_sendv(dict->conn.conn.output, iov, 2);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ret < 0)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_assert((size_t)ret == iov[0].iov_len + 1);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return 0;
98da0024e7340e036f0aa9371e9400176df18ebfTimo Sirainen}
98da0024e7340e036f0aa9371e9400176df18ebfTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic bool
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_cmd_send(struct client_dict *dict, struct client_dict_cmd **_cmd,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char **error_r)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *cmd = *_cmd;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *error = NULL;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen bool retry = cmd->retry_errors;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen int ret;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen *_cmd = NULL;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* we're no longer idling. even with no_replies=TRUE we're going to
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen wait for COMMIT/ROLLBACK. */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->to_idle != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen timeout_remove(&dict->to_idle);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (client_dict_connect(dict, &error) < 0) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen retry = FALSE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ret = -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ret = client_dict_cmd_query_send(dict, cmd->query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ret < 0) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen error = t_strdup_printf("write(%s) failed: %s", dict->conn.conn.name,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen o_stream_get_error(dict->conn.conn.output));
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen }
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ret < 0 && retry) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* Reconnect and try again. */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_disconnect(dict, error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (client_dict_connect(dict, &error) < 0)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen else if (client_dict_cmd_query_send(dict, cmd->query) < 0) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen error = t_strdup_printf("write(%s) failed: %s", dict->conn.conn.name,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen o_stream_get_error(dict->conn.conn.output));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ret = 0;
9261dbf0675204898c6557591c7aa376e23a52b2Timo Sirainen }
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (cmd->no_replies) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* just send and forget */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_cmd_unref(cmd);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else if (ret < 0) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_assert(error != NULL);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_cmd_callback_error(cmd, error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_cmd_unref(cmd);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (error_r != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen *error_r = error;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return FALSE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->to_requests == NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->to_requests =
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen timeout_add(DICT_CLIENT_REQUEST_TIMEOUT_MSECS,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_input_timeout, dict);
9e7a30991415f714b609f650105596371a58216dTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen array_append(&dict->cmds, &cmd, 1);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return TRUE;
9e7a30991415f714b609f650105596371a58216dTimo Sirainen }
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainenstatic bool
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_transaction_send_begin(struct client_dict_transaction_context *ctx)
feac4d832101e7090a134ed4edb16610877c07c7Timo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict = (struct client_dict *)ctx->ctx.dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *cmd;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *query, *error;
feac4d832101e7090a134ed4edb16610877c07c7Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_assert(ctx->error == NULL);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->sent_begin = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* transactions commands don't have replies. only COMMIT has. */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen query = t_strdup_printf("%c%u", DICT_PROTOCOL_CMD_BEGIN, ctx->id);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd = client_dict_cmd_init(dict, query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->no_replies = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->retry_errors = TRUE;
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainen if (!client_dict_cmd_send(dict, &cmd, &error)) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->error = i_strdup(error);
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainen return FALSE;
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainen }
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainen return TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_send_transaction_query(struct client_dict_transaction_context *ctx,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *query)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict = (struct client_dict *)ctx->ctx.dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *cmd;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *error;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ctx->error != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainen if (!ctx->sent_begin) {
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainen if (!client_dict_transaction_send_begin(ctx))
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainen return;
e13bef0a49ee60e4886967b331a7f7d5c96377ccTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen ctx->query_count++;
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen if (ctx->first_query == NULL)
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen ctx->first_query = i_strdup(query);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd = client_dict_cmd_init(dict, query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->no_replies = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (!client_dict_cmd_send(dict, &cmd, &error))
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->error = i_strdup(error);
feac4d832101e7090a134ed4edb16610877c07c7Timo Sirainen}
feac4d832101e7090a134ed4edb16610877c07c7Timo Sirainen
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainenstatic bool client_dict_is_finished(struct client_dict *dict)
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return dict->transactions == NULL && array_count(&dict->cmds) == 0;
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen}
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainenstatic void client_dict_timeout(struct client_dict *dict)
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen{
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen if (client_dict_is_finished(dict))
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen client_dict_disconnect(dict, "Idle disconnection");
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen}
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainenstatic void client_dict_add_timeout(struct client_dict *dict)
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen{
fc0c06ee7b1e5c29a3faabd05f40fa0d26785dffTimo Sirainen if (dict->to_idle != NULL) {
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen if (dict->idle_msecs > 0)
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen timeout_reset(dict->to_idle);
fc0c06ee7b1e5c29a3faabd05f40fa0d26785dffTimo Sirainen } else if (client_dict_is_finished(dict)) {
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen dict->to_idle = timeout_add(dict->idle_msecs,
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen client_dict_timeout, dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->to_requests != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen timeout_remove(&dict->to_requests);
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen }
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen}
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic int dict_conn_input_line(struct connection *_conn, const char *line)
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_connection *conn = (struct dict_connection *)_conn;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict = conn->dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *const *cmds;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen unsigned int count;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen bool finished;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen int diff;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->last_input = ioloop_timeval;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->to_requests != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen timeout_reset(dict->to_requests);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmds = array_get(&conn->dict->cmds, &count);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (count == 0) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_error("%s: Received reply without pending commands: %s",
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->conn.conn.name, line);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_assert(!cmds[0]->no_replies);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_cmd_ref(cmds[0]);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen finished = dict_cmd_callback_line(cmds[0], line);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (!client_dict_cmd_unref(cmds[0])) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* disconnected during command handling */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (!finished) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* more lines needed for this command */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return 1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
701eb90460d6c57845dc4e0bf595a5d0b90b01c1Timo Sirainen diff = cmds[0]->background ? 0 :
701eb90460d6c57845dc4e0bf595a5d0b90b01c1Timo Sirainen timeval_diff_msecs(&ioloop_timeval, &cmds[0]->start_time);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen if (diff < DICT_CLIENT_REQUEST_WARN_TIMEOUT_MSECS)
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen ;
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen else if (cmds[0]->commit_query_human != NULL) {
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen i_warning("read(%s): dict commit took %u.%03u seconds: %s",
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->conn.conn.name, diff/1000, diff % 1000,
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen cmds[0]->commit_query_human);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen } else {
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen const char *type = cmds[0]->iter != NULL ?
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen "iteration" : "lookup";
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen i_warning("read(%s): dict %s took %u.%03u seconds: %s",
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen dict->conn.conn.name, type, diff/1000, diff % 1000,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmds[0]->query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_cmd_unref(cmds[0]);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen array_delete(&dict->cmds, 0, 1);
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen client_dict_add_timeout(dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return 1;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
b5052fbfdbc2678cc8f12899afe55c998f43b740Timo Sirainenstatic int client_dict_connect(struct client_dict *dict, const char **error_r)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen const char *query, *error;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->conn.conn.fd_in != -1)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return 0;
fc0c06ee7b1e5c29a3faabd05f40fa0d26785dffTimo Sirainen if (dict->last_failed_connect == ioloop_time) {
0b15bc85459ef4fea30ac1f8903af48dd6d5f8e2Timo Sirainen /* Try again later */
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen *error_r = dict->last_connect_error;
0b15bc85459ef4fea30ac1f8903af48dd6d5f8e2Timo Sirainen return -1;
0b15bc85459ef4fea30ac1f8903af48dd6d5f8e2Timo Sirainen }
0b15bc85459ef4fea30ac1f8903af48dd6d5f8e2Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (connection_client_connect(&dict->conn.conn) < 0) {
fc0c06ee7b1e5c29a3faabd05f40fa0d26785dffTimo Sirainen dict->last_failed_connect = ioloop_time;
87e603100685f9eb7faf995d2600edd0a578dd0cTimo Sirainen if (errno == EACCES) {
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen error = eacces_error_get("net_connect_unix",
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen dict->conn.conn.name);
87e603100685f9eb7faf995d2600edd0a578dd0cTimo Sirainen } else {
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen error = t_strdup_printf(
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen "net_connect_unix(%s) failed: %m", dict->conn.conn.name);
87e603100685f9eb7faf995d2600edd0a578dd0cTimo Sirainen }
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen i_free(dict->last_connect_error);
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen dict->last_connect_error = i_strdup(error);
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen *error_r = error;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen return -1;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen }
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen query = t_strdup_printf("%c%u\t%u\t%d\t%s\t%s\n",
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen DICT_PROTOCOL_CMD_HELLO,
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen DICT_CLIENT_PROTOCOL_MAJOR_VERSION,
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen DICT_CLIENT_PROTOCOL_MINOR_VERSION,
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen dict->value_type, dict->username, dict->uri);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen o_stream_nsend_str(dict->conn.conn.output, query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_add_timeout(dict);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen return 0;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_abort_commands(struct client_dict *dict, const char *reason)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ARRAY(struct client_dict_cmd *) cmds_copy;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *const *cmdp;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* abort all commands */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen t_array_init(&cmds_copy, array_count(&dict->cmds));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen array_append_array(&cmds_copy, &dict->cmds);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen array_clear(&dict->cmds);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen array_foreach(&cmds_copy, cmdp) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_cmd_callback_error(*cmdp, reason);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_cmd_unref(*cmdp);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainenstatic void client_dict_disconnect(struct client_dict *dict, const char *reason)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
b529524e924da0d86f8b7f0fa3c9e3e3f763a9f3Timo Sirainen struct client_dict_transaction_context *ctx, *next;
b529524e924da0d86f8b7f0fa3c9e3e3f763a9f3Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_abort_commands(dict, reason);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* all transactions that have sent BEGIN are no longer valid */
b529524e924da0d86f8b7f0fa3c9e3e3f763a9f3Timo Sirainen for (ctx = dict->transactions; ctx != NULL; ctx = next) {
b529524e924da0d86f8b7f0fa3c9e3e3f763a9f3Timo Sirainen next = ctx->next;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ctx->sent_begin && ctx->error == NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->error = i_strdup(reason);
b529524e924da0d86f8b7f0fa3c9e3e3f763a9f3Timo Sirainen }
b529524e924da0d86f8b7f0fa3c9e3e3f763a9f3Timo Sirainen
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen if (dict->to_idle != NULL)
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen timeout_remove(&dict->to_idle);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->to_requests != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen timeout_remove(&dict->to_requests);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen connection_disconnect(&dict->conn.conn);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void dict_conn_destroy(struct connection *_conn)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_connection *conn = (struct dict_connection *)_conn;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_disconnect(conn->dict, connection_disconnect_reason(_conn));
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic const struct connection_settings dict_conn_set = {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen .input_max_size = (size_t)-1,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen .output_max_size = (size_t)-1,
b11e9673b057d18c21c683ab17ee7350b9455ddeTimo Sirainen .unix_client_connect_msecs = 1000,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen .client = TRUE
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen};
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic const struct connection_vfuncs dict_conn_vfuncs = {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen .destroy = dict_conn_destroy,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen .input_line = dict_conn_input_line
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen};
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
10399559650f552a23949772be79eb6a80198c5aTimo Sirainenstatic int
419cf63077e755935ce105747d6ebc67b7d38a7fTimo Sirainenclient_dict_init(struct dict *driver, const char *uri,
39ea5717264668e2c7f9f7986eb821d21785f47fTimo Sirainen const struct dict_settings *set,
39ea5717264668e2c7f9f7986eb821d21785f47fTimo Sirainen struct dict **dict_r, const char **error_r)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct ioloop *old_ioloop = current_ioloop;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict *dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *p, *dest_uri, *path;
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen unsigned int idle_msecs = DICT_CLIENT_DEFAULT_TIMEOUT_MSECS;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen /* uri = [idle_msecs=<n>:] [<path>] ":" <uri> */
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen if (strncmp(uri, "idle_msecs=", 11) == 0) {
567ace1f60eca983acbc9b42d7569a5c16e8c3ecTimo Sirainen p = strchr(uri+11, ':');
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen if (p == NULL) {
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen *error_r = t_strdup_printf("Invalid URI: %s", uri);
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen return -1;
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen }
567ace1f60eca983acbc9b42d7569a5c16e8c3ecTimo Sirainen if (str_to_uint(t_strdup_until(uri+11, p), &idle_msecs) < 0) {
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen *error_r = "Invalid idle_msecs";
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen return -1;
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen }
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen uri = p+1;
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen }
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen dest_uri = strchr(uri, ':');
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen if (dest_uri == NULL) {
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen *error_r = t_strdup_printf("Invalid URI: %s", uri);
10399559650f552a23949772be79eb6a80198c5aTimo Sirainen return -1;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen }
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict_connections == NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_connections = connection_list_init(&dict_conn_set,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen &dict_conn_vfuncs);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict = i_new(struct client_dict, 1);
419cf63077e755935ce105747d6ebc67b7d38a7fTimo Sirainen dict->dict = *driver;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->conn.dict = dict;
39ea5717264668e2c7f9f7986eb821d21785f47fTimo Sirainen dict->value_type = set->value_type;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->username = i_strdup(set->username);
09cd60cebc01b2b25ada824f3d9f877d5efd8272Timo Sirainen dict->idle_msecs = idle_msecs;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_array_init(&dict->cmds, 32);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
02a0277d897dbf5172937158f3d828f981ff230dTimo Sirainen if (uri[0] == ':') {
02a0277d897dbf5172937158f3d828f981ff230dTimo Sirainen /* default path */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen path = t_strconcat(set->base_dir,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen "/"DEFAULT_DICT_SERVER_SOCKET_FNAME, NULL);
02a0277d897dbf5172937158f3d828f981ff230dTimo Sirainen } else if (uri[0] == '/') {
02a0277d897dbf5172937158f3d828f981ff230dTimo Sirainen /* absolute path */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen path = t_strdup_until(uri, dest_uri);
02a0277d897dbf5172937158f3d828f981ff230dTimo Sirainen } else {
02a0277d897dbf5172937158f3d828f981ff230dTimo Sirainen /* relative path to base_dir */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen path = t_strconcat(set->base_dir, "/",
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen t_strdup_until(uri, dest_uri), NULL);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen connection_init_client_unix(dict_connections, &dict->conn.conn, path);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->uri = i_strdup(dest_uri + 1);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->ioloop = io_loop_create();
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_set_current(old_ioloop);
10399559650f552a23949772be79eb6a80198c5aTimo Sirainen *dict_r = &dict->dict;
eca38954bcf972618f6b85932a3690acbd2b673aTimo Sirainen return 0;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstatic void client_dict_deinit(struct dict *_dict)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict *dict = (struct client_dict *)_dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct ioloop *old_ioloop = current_ioloop;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_disconnect(dict, "Deinit");
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen connection_deinit(&dict->conn.conn);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
29dcd138546179e55e4dbcd3c256ba2fd6726e27Timo Sirainen i_assert(dict->transactions == NULL);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_assert(array_count(&dict->cmds) == 0);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_set_current(dict->ioloop);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_destroy(&dict->ioloop);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_set_current(old_ioloop);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen array_free(&dict->cmds);
98b7ea1744aecc52750f1dfb0d8ed6f9646b4605Timo Sirainen i_free(dict->last_connect_error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_free(dict->username);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_free(dict->uri);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_free(dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict_connections->connections == NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen connection_list_deinit(&dict_connections);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
c4478af52de63804efef2055580adf1dfc8679c6Timo Sirainenstatic void client_dict_wait(struct dict *_dict)
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen{
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen struct client_dict *dict = (struct client_dict *)_dict;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (array_count(&dict->cmds) == 0)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return;
7e74fadc6668d85c2bfd630f215770808d73d1a3Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->prev_ioloop = current_ioloop;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_set_current(dict->ioloop);
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainen dict_switch_ioloop(_dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen while (array_count(&dict->cmds) > 0)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_run(dict->ioloop);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_set_current(dict->prev_ioloop);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->prev_ioloop = NULL;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainen dict_switch_ioloop(_dict);
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainen}
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainen
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainenstatic bool client_dict_switch_ioloop(struct dict *_dict)
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainen{
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainen struct client_dict *dict = (struct client_dict *)_dict;
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->to_idle != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->to_idle = io_loop_move_timeout(&dict->to_idle);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->to_requests != NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->to_requests = io_loop_move_timeout(&dict->to_requests);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen connection_switch_ioloop(&dict->conn.conn);
a5ec9755556e3d97d7e6d78cb1b53046370e6598Timo Sirainen return array_count(&dict->cmds) > 0;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_lookup_async_callback(struct client_dict_cmd *cmd, const char *line,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *error)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict = cmd->dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_lookup_result result;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen memset(&result, 0, sizeof(result));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (error != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.error = error;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else switch (*line) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case DICT_PROTOCOL_REPLY_OK:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.value = t_str_tabunescape(line + 1);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = 1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case DICT_PROTOCOL_REPLY_NOTFOUND:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = 0;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case DICT_PROTOCOL_REPLY_FAIL:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.error = line[1] == '\0' ? "dict-server returned failure" :
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen t_strdup_printf("dict-server returned failure: %s",
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen t_str_tabunescape(line+1));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen default:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.error = t_strdup_printf(
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen "dict-client: Invalid lookup '%s' reply: %s",
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->query, line);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_disconnect(dict, result.error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_pre_api_callback(dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->api_callback.lookup(&result, cmd->api_callback.context);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_post_api_callback(dict);
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen}
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_lookup_async(struct dict *_dict, const char *key,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_lookup_callback_t *callback, void *context)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict *dict = (struct client_dict *)_dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *cmd;
b5052fbfdbc2678cc8f12899afe55c998f43b740Timo Sirainen const char *query;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen query = t_strdup_printf("%c%s", DICT_PROTOCOL_CMD_LOOKUP,
cbe2d47fda811fc56ba59cb2811c3fc26033b03cTimo Sirainen str_tabescape(key));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd = client_dict_cmd_init(dict, query);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->callback = client_dict_lookup_async_callback;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->api_callback.lookup = callback;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->api_callback.context = context;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->retry_errors = TRUE;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_cmd_send(dict, &cmd, NULL);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void client_dict_lookup_callback(const struct dict_lookup_result *result,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen void *context)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_lookup_result *result_copy = context;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen *result_copy = *result;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic int client_dict_lookup(struct dict *_dict, pool_t pool, const char *key,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char **value_r, const char **error_r)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_lookup_result result;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen memset(&result, 0, sizeof(result));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = -2;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_lookup_async(_dict, key, client_dict_lookup_callback, &result);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (result.ret == -2)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_wait(_dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen switch (result.ret) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case -1:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen *error_r = result.error;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case 0:
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen *value_r = NULL;
7e74fadc6668d85c2bfd630f215770808d73d1a3Timo Sirainen return 0;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case 1:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen *value_r = p_strdup(pool, result.value);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return 1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_unreached();
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void client_dict_iterate_free(struct client_dict_iterate_context *ctx)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (!ctx->deinit || !ctx->finished)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_free(ctx->error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_free(ctx);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_iter_api_callback(struct client_dict_iterate_context *ctx,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ctx->deinit) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* iterator was already deinitialized */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ctx->ctx.async_callback != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_pre_api_callback(dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->ctx.async_callback(ctx->ctx.async_context);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_post_api_callback(dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* synchronous lookup */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_stop(dict->ioloop);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen}
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_iter_async_callback(struct client_dict_cmd *cmd, const char *line,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *error)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_iterate_context *ctx = cmd->iter;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict = cmd->dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_iter_result *result;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *key = NULL, *value = NULL;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
701eb90460d6c57845dc4e0bf595a5d0b90b01c1Timo Sirainen if (ctx->deinit)
701eb90460d6c57845dc4e0bf595a5d0b90b01c1Timo Sirainen cmd->background = TRUE;
701eb90460d6c57845dc4e0bf595a5d0b90b01c1Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (error != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* failed */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else switch (*line) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case '\0':
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* end of iteration */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->finished = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_iter_api_callback(ctx, dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_iterate_free(ctx);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case DICT_PROTOCOL_REPLY_OK:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* key \t value */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen key = line+1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen value = strchr(key, '\t');
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
7e74fadc6668d85c2bfd630f215770808d73d1a3Timo Sirainen case DICT_PROTOCOL_REPLY_FAIL:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen error = t_strdup_printf("dict-server returned failure: %s", line+1);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
7e74fadc6668d85c2bfd630f215770808d73d1a3Timo Sirainen default:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (value == NULL && error == NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* broken protocol */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen error = t_strdup_printf("dict client (%s) sent broken iterate reply: %s",
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict->conn.conn.name, line);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_disconnect(dict, error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (error != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ctx->error == NULL)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->error = i_strdup(error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->finished = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (dict->prev_ioloop != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* stop client_dict_wait() */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen io_loop_stop(dict->ioloop);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_iterate_free(ctx);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->unfinished = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ctx->deinit) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* iterator was already deinitialized */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen key = t_strdup_until(key, value++);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result = array_append_space(&ctx->results);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result->key = p_strdup(ctx->results_pool, t_str_tabunescape(key));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result->value = p_strdup(ctx->results_pool, t_str_tabunescape(value));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_iter_api_callback(ctx, dict);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstatic struct dict_iterate_context *
9c7f6dbf65ca01026e5f9c8c8b67c7e629c0b5e7Timo Sirainenclient_dict_iterate_init(struct dict *_dict, const char *const *paths,
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen enum dict_iterate_flags flags)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict *dict = (struct client_dict *)_dict;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict_iterate_context *ctx;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *cmd;
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen string_t *query = t_str_new(256);
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen unsigned int i;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen ctx = i_new(struct client_dict_iterate_context, 1);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen ctx->ctx.dict = _dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->results_pool = pool_alloconly_create("client dict iteration", 512);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->async = (flags & DICT_ITERATE_FLAG_ASYNC) != 0;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen i_array_init(&ctx->results, 64);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen str_printfa(query, "%c%d", DICT_PROTOCOL_CMD_ITERATE, flags);
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen for (i = 0; paths[i] != NULL; i++) {
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen str_append_c(query, '\t');
cbe2d47fda811fc56ba59cb2811c3fc26033b03cTimo Sirainen str_append(query, str_tabescape(paths[i]));
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd = client_dict_cmd_init(dict, str_c(query));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->iter = ctx;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->callback = client_dict_iter_async_callback;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->retry_errors = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_cmd_send(dict, &cmd, NULL);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen return &ctx->ctx;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
8d25b6ad05b99e75613cb045a121efd51e6afbb6Timo Sirainenstatic bool client_dict_iterate(struct dict_iterate_context *_ctx,
8d25b6ad05b99e75613cb045a121efd51e6afbb6Timo Sirainen const char **key_r, const char **value_r)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict_iterate_context *ctx =
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen (struct client_dict_iterate_context *)_ctx;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const struct client_dict_iter_result *results;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen unsigned int count;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ctx->error != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->ctx.has_more = FALSE;
8d25b6ad05b99e75613cb045a121efd51e6afbb6Timo Sirainen return FALSE;
8d25b6ad05b99e75613cb045a121efd51e6afbb6Timo Sirainen }
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen results = array_get(&ctx->results, &count);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (ctx->result_idx < count) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen *key_r = results[ctx->result_idx].key;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen *value_r = results[ctx->result_idx].value;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->ctx.has_more = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->result_idx++;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return TRUE;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->ctx.has_more = !ctx->finished;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->result_idx = 0;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen array_clear(&ctx->results);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen p_clear(ctx->results_pool);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (!ctx->async && ctx->ctx.has_more) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_wait(_ctx->dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return client_dict_iterate(_ctx, key_r, value_r);
e4383dc44cc122924e65a533b105fd64bf566ebfTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen return FALSE;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
055389c58fa3915e12fb4e72ec86782ce77c5c72Timo Sirainenstatic int client_dict_iterate_deinit(struct dict_iterate_context *_ctx,
055389c58fa3915e12fb4e72ec86782ce77c5c72Timo Sirainen const char **error_r)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict *dict = (struct client_dict *)_ctx->dict;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict_iterate_context *ctx =
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen (struct client_dict_iterate_context *)_ctx;
055389c58fa3915e12fb4e72ec86782ce77c5c72Timo Sirainen int ret = ctx->error != NULL ? -1 : 0;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen ctx->deinit = TRUE;
9e7a30991415f714b609f650105596371a58216dTimo Sirainen
055389c58fa3915e12fb4e72ec86782ce77c5c72Timo Sirainen *error_r = t_strdup(ctx->error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen array_free(&ctx->results);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen pool_unref(&ctx->results_pool);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_iterate_free(ctx);
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen client_dict_add_timeout(dict);
8d25b6ad05b99e75613cb045a121efd51e6afbb6Timo Sirainen return ret;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstatic struct dict_transaction_context *
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenclient_dict_transaction_init(struct dict *_dict)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict *dict = (struct client_dict *)_dict;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict_transaction_context *ctx;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen ctx = i_new(struct client_dict_transaction_context, 1);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen ctx->ctx.dict = _dict;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen ctx->id = ++dict->transaction_id_counter;
684cbcce30cc97340ee43240de5c0d236c917d29Timo Sirainen
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen DLLIST_PREPEND(&dict->transactions, ctx);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen return &ctx->ctx;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenstatic void
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainenclient_dict_transaction_commit_callback(struct client_dict_cmd *cmd,
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *line, const char *error)
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen{
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict *dict = cmd->dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_commit_result result = {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen .ret = -1, .error = NULL
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen };
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (error != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* failed */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.error = error;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else switch (*line) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case DICT_PROTOCOL_REPLY_OK:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = 1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case DICT_PROTOCOL_REPLY_NOTFOUND:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = 0;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen case DICT_PROTOCOL_REPLY_FAIL: {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *error = strchr(line+1, '\t');
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.error = t_strdup_printf("dict-server returned failure: %s",
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen error != NULL ? t_str_tabunescape(error) : "");
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
7e74fadc6668d85c2bfd630f215770808d73d1a3Timo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen default:
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.ret = -1;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen result.error = t_strdup_printf(
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen "dict-client: Invalid commit reply: %s", line);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_disconnect(dict, result.error);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen break;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_pre_api_callback(dict);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->api_callback.commit(&result, cmd->api_callback.context);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen dict_post_api_callback(dict);
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen}
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainenstatic void
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainenclient_dict_transaction_free(struct client_dict_transaction_context *ctx)
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen{
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen i_free(ctx->first_query);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen i_free(ctx->error);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen i_free(ctx);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen}
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen
c2a66e7950cb4d3fc4d68e4480ea8f39bdd7c871Timo Sirainenstatic void
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainenclient_dict_transaction_commit(struct dict_transaction_context *_ctx,
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen bool async,
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen dict_transaction_commit_callback_t *callback,
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen void *context)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict_transaction_context *ctx =
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen (struct client_dict_transaction_context *)_ctx;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict *dict = (struct client_dict *)_ctx->dict;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct client_dict_cmd *cmd;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen const char *query;
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen DLLIST_REMOVE(&dict->transactions, ctx);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
661998e2ccd772ad92a9d4a75cb712692a8c94b3Timo Sirainen if (ctx->sent_begin && ctx->error == NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen query = t_strdup_printf("%c%u", DICT_PROTOCOL_CMD_COMMIT, ctx->id);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd = client_dict_cmd_init(dict, query);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen cmd->commit_query_human = i_strdup_printf(
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen "%s (%u commands, first: %s)", query,
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen ctx->query_count, ctx->first_query);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->callback = client_dict_transaction_commit_callback;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->api_callback.commit = callback;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen cmd->api_callback.context = context;
701eb90460d6c57845dc4e0bf595a5d0b90b01c1Timo Sirainen if (callback == dict_transaction_commit_async_noop_callback)
701eb90460d6c57845dc4e0bf595a5d0b90b01c1Timo Sirainen cmd->background = TRUE;
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (client_dict_cmd_send(dict, &cmd, NULL)) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen if (!async)
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_wait(_ctx->dict);
4210fac11e936f7edd6f3deecd1b23466534ce6aTimo Sirainen }
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else if (ctx->error != NULL) {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* already failed */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_commit_result result = {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen .ret = -1, .error = ctx->error
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen };
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen callback(&result, context);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen } else {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen /* nothing changed */
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen struct dict_commit_result result = {
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen .ret = 1, .error = NULL
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen };
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen callback(&result, context);
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen }
e080054d7df103a9b6be62a476cbee18261acf6cTimo Sirainen
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen client_dict_transaction_free(ctx);
c2a66e7950cb4d3fc4d68e4480ea8f39bdd7c871Timo Sirainen client_dict_add_timeout(dict);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstatic void
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenclient_dict_transaction_rollback(struct dict_transaction_context *_ctx)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict_transaction_context *ctx =
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen (struct client_dict_transaction_context *)_ctx;
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen struct client_dict *dict = (struct client_dict *)_ctx->dict;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen if (ctx->sent_begin) {
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen const char *query;
eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen query = t_strdup_printf("%c%u", DICT_PROTOCOL_CMD_ROLLBACK,
4210fac11e936f7edd6f3deecd1b23466534ce6aTimo Sirainen ctx->id);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_send_transaction_query(ctx, query);
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen }
e080054d7df103a9b6be62a476cbee18261acf6cTimo Sirainen
3954326e793bdef1e94e0ad781ed6cc7e48beebbTimo Sirainen DLLIST_REMOVE(&dict->transactions, ctx);
1cfe369d1278a9e2e6cc2e46103db48c493e1f21Timo Sirainen client_dict_transaction_free(ctx);
28a311381d783cc06e56f9baf1cb9f25634cbfe4Timo Sirainen client_dict_add_timeout(dict);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstatic void client_dict_set(struct dict_transaction_context *_ctx,
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen const char *key, const char *value)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict_transaction_context *ctx =
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen (struct client_dict_transaction_context *)_ctx;
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen const char *query;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen query = t_strdup_printf("%c%u\t%s\t%s",
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen DICT_PROTOCOL_CMD_SET, ctx->id,
cbe2d47fda811fc56ba59cb2811c3fc26033b03cTimo Sirainen str_tabescape(key),
cbe2d47fda811fc56ba59cb2811c3fc26033b03cTimo Sirainen str_tabescape(value));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_send_transaction_query(ctx, query);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainenstatic void client_dict_unset(struct dict_transaction_context *_ctx,
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen const char *key)
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen{
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen struct client_dict_transaction_context *ctx =
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen (struct client_dict_transaction_context *)_ctx;
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen const char *query;
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen query = t_strdup_printf("%c%u\t%s",
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen DICT_PROTOCOL_CMD_UNSET, ctx->id,
cbe2d47fda811fc56ba59cb2811c3fc26033b03cTimo Sirainen str_tabescape(key));
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_send_transaction_query(ctx, query);
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen}
92d1458b00f4f236c4cec96a696253d3bbf8b05aTimo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainenstatic void client_dict_atomic_inc(struct dict_transaction_context *_ctx,
a1295ce772f4765320a11a572595ce8eff37c343Timo Sirainen const char *key, long long diff)
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen{
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen struct client_dict_transaction_context *ctx =
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen (struct client_dict_transaction_context *)_ctx;
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen const char *query;
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen query = t_strdup_printf("%c%u\t%s\t%lld",
8f9a18189f01448267100fa54c3b4bb8639a1a56Timo Sirainen DICT_PROTOCOL_CMD_ATOMIC_INC,
cbe2d47fda811fc56ba59cb2811c3fc26033b03cTimo Sirainen ctx->id, str_tabescape(key), diff);
a7e46c05358b059aad2b90f01e271ba6732c5eeeTimo Sirainen client_dict_send_transaction_query(ctx, query);
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen}
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
419cf63077e755935ce105747d6ebc67b7d38a7fTimo Sirainenstruct dict dict_driver_client = {
7bafda1813454621e03615e83d55bccfa7cc56bdTimo Sirainen .name = "proxy",
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen {
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .init = client_dict_init,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .deinit = client_dict_deinit,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .wait = client_dict_wait,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .lookup = client_dict_lookup,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .iterate_init = client_dict_iterate_init,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .iterate = client_dict_iterate,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .iterate_deinit = client_dict_iterate_deinit,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .transaction_init = client_dict_transaction_init,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .transaction_commit = client_dict_transaction_commit,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .transaction_rollback = client_dict_transaction_rollback,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .set = client_dict_set,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .unset = client_dict_unset,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .atomic_inc = client_dict_atomic_inc,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .lookup_async = client_dict_lookup_async,
ade5567577dadb0b275c840208d3ad21a9f00a36Timo Sirainen .switch_ioloop = client_dict_switch_ioloop
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen }
ff52f5c52ee6a4c8a9c79964d32b5d0ff0ae92e6Timo Sirainen};