Mercurial > dovecot > core-2.2
view src/doveadm/server-connection.c @ 12629:02370eda76f8
lib-storage: Mailbox virtual names are now in UTF-8 format, not IMAP mUTF-7.
Plugins that use mailbox names in configuration now take them also as UTF-8
rather than mUTF-7.
author | Timo Sirainen <tss@iki.fi> |
---|---|
date | Tue, 01 Feb 2011 20:41:54 +0200 |
parents | b60e225386bf |
children | 447bce266022 |
line wrap: on
line source
/* Copyright (c) 2010 Dovecot authors, see the included COPYING file */ #include "lib.h" #include "array.h" #include "ioloop.h" #include "network.h" #include "istream.h" #include "ostream.h" #include "str.h" #include "strescape.h" #include "doveadm-print.h" #include "doveadm-util.h" #include "doveadm-server.h" #include "server-connection.h" #include <unistd.h> #define MAX_INBUF_SIZE (1024*32) enum server_reply_state { SERVER_REPLY_STATE_DONE = 0, SERVER_REPLY_STATE_PRINT, SERVER_REPLY_STATE_RET }; struct server_connection { struct doveadm_server *server; int fd; struct io *io; struct istream *input; struct ostream *output; server_cmd_callback_t *callback; void *context; enum server_reply_state state; unsigned int handshaked:1; unsigned int authenticated:1; unsigned int streaming:1; }; static struct server_connection *printing_conn = NULL; static void server_connection_input(struct server_connection *conn); static void print_connection_released(void) { struct doveadm_server *server = printing_conn->server; struct server_connection *const *conns; unsigned int i, count; printing_conn = NULL; conns = array_get(&server->connections, &count); for (i = 0; i < count; i++) { if (conns[i]->io != NULL) continue; conns[i]->io = io_add(conns[i]->fd, IO_READ, server_connection_input, conns[i]); server_connection_input(conns[i]); if (printing_conn != NULL) break; } } static void server_connection_callback(struct server_connection *conn, enum server_cmd_reply reply) { server_cmd_callback_t *callback = conn->callback; conn->callback = NULL; callback(reply, conn->context); } static void stream_data(string_t *str, const unsigned char *data, size_t size) { const char *text; str_truncate(str, 0); str_append_n(str, data, size); text = str_tabunescape(str_c_modifiable(str)); doveadm_print_stream(text, strlen(text)); } static void server_flush_field(struct server_connection *conn, string_t *str, const unsigned char *data, size_t size) { if (conn->streaming) { conn->streaming = FALSE; stream_data(str, data, size); doveadm_print_stream("", 0); } else { const char *text; str_truncate(str, 0); str_append_n(str, data, size); text = str_tabunescape(str_c_modifiable(str)); doveadm_print(text); } } static void server_handle_input(struct server_connection *conn, const unsigned char *data, size_t size) { string_t *str; size_t i, start; if (printing_conn == conn) { /* continue printing */ } else if (printing_conn == NULL) { printing_conn = conn; } else { /* someone else is printing. don't continue until it goes away */ io_remove(&conn->io); return; } if (data[size-1] == '\001') { /* last character is an escape */ size--; } str = t_str_new(128); for (i = start = 0; i < size; i++) { if (data[i] == '\n') { if (i != start) i_error("doveadm server sent broken input"); conn->state = SERVER_REPLY_STATE_RET; i_stream_skip(conn->input, i + 1); print_connection_released(); return; } if (data[i] == '\t') { server_flush_field(conn, str, data + start, i - start); start = i + 1; } } if (start != size) { conn->streaming = TRUE; stream_data(str, data + start, size - start); } i_stream_skip(conn->input, size); } static void server_connection_authenticate(struct server_connection *conn ATTR_UNUSED) { i_fatal("Authentication not supported yet"); } static void server_connection_input(struct server_connection *conn) { const unsigned char *data; size_t size; const char *line; if (!conn->handshaked) { if ((line = i_stream_read_next_line(conn->input)) == NULL) { if (conn->input->eof || conn->input->stream_errno != 0) server_connection_destroy(&conn); return; } conn->handshaked = TRUE; if (strcmp(line, "+") == 0) conn->authenticated = TRUE; else if (strcmp(line, "-") == 0) server_connection_authenticate(conn); else { i_error("doveadm server sent invalid handshake: %s", line); server_connection_destroy(&conn); return; } } if (i_stream_read(conn->input) == -1) { /* disconnected */ server_connection_destroy(&conn); return; } data = i_stream_get_data(conn->input, &size); if (size == 0) return; switch (conn->state) { case SERVER_REPLY_STATE_DONE: i_error("doveadm server sent unexpected input"); server_connection_destroy(&conn); return; case SERVER_REPLY_STATE_PRINT: server_handle_input(conn, data, size); if (conn->state != SERVER_REPLY_STATE_RET) break; /* fall through */ data = i_stream_get_data(conn->input, &size); case SERVER_REPLY_STATE_RET: if (size < 2) return; if (data[0] == '+' && data[1] == '\n') server_connection_callback(conn, SERVER_CMD_REPLY_OK); else if (data[0] == '-' && data[1] == '\n') server_connection_callback(conn, SERVER_CMD_REPLY_FAIL); else { i_error("doveadm server sent broken input"); server_connection_destroy(&conn); return; } break; } } struct server_connection * server_connection_create(struct doveadm_server *server) { #define DOVEADM_SERVER_HANDSHAKE "VERSION\tdoveadm-server\t1\t0\n" struct server_connection *conn; conn = i_new(struct server_connection, 1); conn->server = server; conn->fd = doveadm_connect(server->name); net_set_nonblock(conn->fd, TRUE); conn->io = io_add(conn->fd, IO_READ, server_connection_input, conn); conn->input = i_stream_create_fd(conn->fd, MAX_INBUF_SIZE, FALSE); conn->output = o_stream_create_fd(conn->fd, (size_t)-1, FALSE); conn->state = SERVER_REPLY_STATE_DONE; o_stream_send_str(conn->output, DOVEADM_SERVER_HANDSHAKE); array_append(&conn->server->connections, &conn, 1); return conn; } void server_connection_destroy(struct server_connection **_conn) { struct server_connection *conn = *_conn; struct server_connection *const *conns; unsigned int i, count; *_conn = NULL; conns = array_get(&conn->server->connections, &count); for (i = 0; i < count; i++) { if (conns[i] == conn) { array_delete(&conn->server->connections, i, 1); break; } } if (conn->callback != NULL) { server_connection_callback(conn, SERVER_CMD_REPLY_INTERNAL_FAILURE); } if (printing_conn == conn) print_connection_released(); i_stream_destroy(&conn->input); o_stream_destroy(&conn->output); if (conn->io != NULL) io_remove(&conn->io); if (close(conn->fd) < 0) i_error("close(server) failed: %m"); i_free(conn); } struct doveadm_server * server_connection_get_server(struct server_connection *conn) { return conn->server; } void server_connection_cmd(struct server_connection *conn, const char *line, server_cmd_callback_t *callback, void *context) { conn->state = SERVER_REPLY_STATE_PRINT; o_stream_send_str(conn->output, line); conn->callback = callback; conn->context = context; } bool server_connection_is_idle(struct server_connection *conn) { return conn->callback == NULL; }