2 * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
3 * Copyright (C) 1999-2012 Hiroyuki Yamamoto and the Claws Mail team
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 3 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "claws-features.h"
28 #include <glib/gi18n.h>
41 static gint session_connect_cb (SockInfo *sock,
43 static gint session_close (Session *session);
45 static gboolean session_timeout_cb (gpointer data);
47 static gboolean session_recv_msg_idle_cb (gpointer data);
48 static gboolean session_recv_data_idle_cb (gpointer data);
50 static gboolean session_read_msg_cb (SockInfo *source,
51 GIOCondition condition,
53 static gboolean session_read_data_cb (SockInfo *source,
54 GIOCondition condition,
56 static gboolean session_write_msg_cb (SockInfo *source,
57 GIOCondition condition,
59 static gboolean session_write_data_cb (SockInfo *source,
60 GIOCondition condition,
64 void session_init(Session *session, const void *prefs_account, gboolean is_smtp)
66 session->type = SESSION_UNKNOWN;
68 session->server = NULL;
71 session->ssl_type = SSL_NONE;
72 session->use_tls_sni = TRUE;
74 session->nonblocking = TRUE;
75 session->state = SESSION_READY;
76 session->last_access_time = time(NULL);
78 session->tv_prev = g_date_time_new_now_local();
84 session->read_buf_p = session->read_buf;
85 session->read_buf_len = 0;
87 session->read_msg_buf = g_string_sized_new(1024);
88 session->read_data_buf = g_byte_array_new();
90 session->write_buf = NULL;
91 session->write_buf_p = NULL;
92 session->write_buf_len = 0;
94 session->write_data = NULL;
95 session->write_data_p = NULL;
96 session->write_data_len = 0;
98 session->timeout_tag = 0;
99 session->timeout_interval = 0;
101 session->data = NULL;
102 session->account = prefs_account;
103 session->is_smtp = is_smtp;
105 session->ping_tag = -1;
107 session->proxy_info = NULL;
111 *\brief Set up parent and child process
112 * Childloop: Read commands from parent,
113 * send to server, get answer, pass to parent
115 *\param session Contains session information
116 * server to connect to
120 * -1 : pipe / fork errors (parent)
121 * 1 : connection error (child)
123 gint session_connect(Session *session, const gchar *server, gushort port)
125 session->server = g_strdup(server);
126 session->port = port;
128 if (session->proxy_info) {
129 server = session->proxy_info->proxy_host;
130 port = session->proxy_info->proxy_port;
134 session->conn_id = sock_connect_async(server, port, session_connect_cb,
136 if (session->conn_id < 0) {
137 g_warning("can't connect to server.");
138 session_close(session);
139 if (session->connect_finished)
140 session->connect_finished(session, FALSE);
148 sock = sock_connect(server, port);
150 g_warning("can't connect to server.");
151 session_close(session);
152 if (session->connect_finished)
153 session->connect_finished(session, FALSE);
156 sock->is_smtp = session->is_smtp;
158 return session_connect_cb(sock, session);
162 static gint session_connect_cb(SockInfo *sock, gpointer data)
164 Session *session = SESSION(data);
166 session->conn_id = 0;
169 g_warning("can't connect to server.");
170 session->state = SESSION_ERROR;
171 if (session->connect_finished)
172 session->connect_finished(session, FALSE);
176 session->sock = sock;
177 sock->account = session->account;
178 sock->is_smtp = session->is_smtp;
179 sock->ssl_cert_auto_accept = session->ssl_cert_auto_accept;
181 if (session->proxy_info) {
182 debug_print("connecting through socks\n");
183 sock_set_nonblocking_mode(sock, FALSE);
184 if (proxy_connect(sock, session->server, session->port,
185 session->proxy_info) < 0) {
186 g_warning("can't establish SOCKS connection.");
187 session->state = SESSION_ERROR;
194 sock->gnutls_priority = session->gnutls_priority;
195 sock->use_tls_sni = session->use_tls_sni;
197 if (session->ssl_type == SSL_TUNNEL) {
198 sock_set_nonblocking_mode(sock, FALSE);
199 if (!ssl_init_socket(sock)) {
200 g_warning("can't initialize SSL/TLS.");
201 log_error(LOG_PROTOCOL, _("SSL/TLS handshake failed\n"));
202 session->state = SESSION_ERROR;
203 if (session->connect_finished)
204 session->connect_finished(session, FALSE);
210 /* we could have gotten a timeout while waiting for user input in
211 * an SSL certificate dialog */
212 if (session->state == SESSION_TIMEOUT) {
213 if (session->connect_finished)
214 session->connect_finished(session, FALSE);
218 sock_set_nonblocking_mode(sock, session->nonblocking);
220 debug_print("session (%p): connected\n", session);
222 session->state = SESSION_RECV;
223 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
227 if (session->connect_finished)
228 session->connect_finished(session, TRUE);
233 *\brief child and parent: send DISCONNECT message to other process
235 *\param session Contains session information
239 gint session_disconnect(Session *session)
241 session_close(session);
248 *\param session Contains session information
250 void session_destroy(Session *session)
252 cm_return_if_fail(session != NULL);
253 cm_return_if_fail(session->destroy != NULL);
255 session_register_ping(session, NULL);
257 session_close(session);
258 session->destroy(session);
259 g_free(session->server);
260 g_string_free(session->read_msg_buf, TRUE);
261 g_byte_array_free(session->read_data_buf, TRUE);
262 g_free(session->read_data_terminator);
263 g_free(session->write_buf);
265 g_free(session->gnutls_priority);
267 g_date_time_unref(session->tv_prev);
268 debug_print("session (%p): destroyed\n", session);
273 gboolean session_is_running(Session *session)
275 return (session->state == SESSION_READY ||
276 session->state == SESSION_SEND ||
277 session->state == SESSION_RECV);
280 gboolean session_is_connected(Session *session)
282 return (session->state == SESSION_SEND ||
283 session->state == SESSION_RECV);
286 void session_set_access_time(Session *session)
288 session->last_access_time = time(NULL);
291 void session_set_timeout(Session *session, guint interval)
293 if (session->timeout_tag > 0)
294 g_source_remove(session->timeout_tag);
296 session->timeout_interval = interval;
298 if (interval % 1000 == 0)
299 session->timeout_tag =
300 g_timeout_add_seconds(interval/1000, session_timeout_cb, session);
302 session->timeout_tag =
303 g_timeout_add(interval, session_timeout_cb, session);
305 session->timeout_tag = 0;
308 static gboolean session_timeout_cb(gpointer data)
310 Session *session = SESSION(data);
312 g_warning("session timeout.");
314 if (session->io_tag > 0) {
315 g_source_remove(session->io_tag);
319 session->timeout_tag = 0;
320 session->state = SESSION_TIMEOUT;
325 void session_set_recv_message_notify(Session *session,
326 RecvMsgNotify notify_func, gpointer data)
328 session->recv_msg_notify = notify_func;
329 session->recv_msg_notify_data = data;
332 void session_set_recv_data_progressive_notify
334 RecvDataProgressiveNotify notify_func,
337 session->recv_data_progressive_notify = notify_func,
338 session->recv_data_progressive_notify_data = data;
341 void session_set_recv_data_notify(Session *session, RecvDataNotify notify_func,
344 session->recv_data_notify = notify_func;
345 session->recv_data_notify_data = data;
348 void session_set_send_data_progressive_notify
350 SendDataProgressiveNotify notify_func,
353 session->send_data_progressive_notify = notify_func;
354 session->send_data_progressive_notify_data = data;
357 void session_set_send_data_notify(Session *session, SendDataNotify notify_func,
360 session->send_data_notify = notify_func;
361 session->send_data_notify_data = data;
365 *\brief child and parent cleanup (child closes first)
367 *\param session Contains session information
371 static gint session_close(Session *session)
373 cm_return_val_if_fail(session != NULL, -1);
376 if (session->conn_id > 0) {
377 sock_connect_async_cancel(session->conn_id);
378 session->conn_id = 0;
379 debug_print("session (%p): connection cancelled\n", session);
383 session_set_timeout(session, 0);
385 if (session->io_tag > 0) {
386 g_source_remove(session->io_tag);
391 sock_close(session->sock, TRUE);
392 session->sock = NULL;
393 session->state = SESSION_DISCONNECTED;
394 debug_print("session (%p): closed\n", session);
401 gint session_start_tls(Session *session)
405 nb_mode = sock_is_nonblocking_mode(session->sock);
407 session->sock->ssl_cert_auto_accept = session->ssl_cert_auto_accept;
408 session->sock->gnutls_priority = session->gnutls_priority;
409 session->sock->use_tls_sni = session->use_tls_sni;
412 sock_set_nonblocking_mode(session->sock, FALSE);
414 if (!ssl_init_socket(session->sock)) {
415 g_warning("couldn't start STARTTLS session.");
417 sock_set_nonblocking_mode(session->sock, session->nonblocking);
422 sock_set_nonblocking_mode(session->sock, session->nonblocking);
428 gint session_send_msg(Session *session, const gchar *msg)
432 cm_return_val_if_fail(session->write_buf == NULL, -1);
433 cm_return_val_if_fail(msg != NULL, -1);
435 session->state = SESSION_SEND;
436 session->write_buf = g_strconcat((strlen(msg) > 0 ? msg : ""), "\r\n", NULL);
437 session->write_buf_p = session->write_buf;
438 session->write_buf_len = strlen(msg) + 2;
440 ret = session_write_msg_cb(session->sock, G_IO_OUT, session);
443 session->io_tag = sock_add_watch(session->sock, G_IO_OUT,
444 session_write_msg_cb, session);
445 else if (session->state == SESSION_ERROR)
451 gint session_recv_msg(Session *session)
453 cm_return_val_if_fail(session->read_msg_buf->len == 0, -1);
455 session->state = SESSION_RECV;
457 if (session->read_buf_len > 0)
458 g_idle_add(session_recv_msg_idle_cb, session);
460 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
461 session_read_msg_cb, session);
466 static gboolean session_recv_msg_idle_cb(gpointer data)
468 Session *session = SESSION(data);
471 ret = session_read_msg_cb(session->sock, G_IO_IN, session);
474 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
475 session_read_msg_cb, session);
481 *\brief parent (child?): send data to other process
483 *\param session Contains session information
490 gint session_send_data(Session *session, const guchar *data, guint size)
494 cm_return_val_if_fail(session->write_data == NULL, -1);
495 cm_return_val_if_fail(data != NULL, -1);
496 cm_return_val_if_fail(size != 0, -1);
498 session->state = SESSION_SEND;
500 session->write_data = data;
501 session->write_data_p = session->write_data;
502 session->write_data_len = size;
503 g_date_time_unref(session->tv_prev);
504 session->tv_prev = g_date_time_new_now_local();
506 ret = session_write_data_cb(session->sock, G_IO_OUT, session);
509 session->io_tag = sock_add_watch(session->sock, G_IO_OUT,
510 session_write_data_cb,
512 else if (session->state == SESSION_ERROR)
518 gint session_recv_data(Session *session, guint size, const gchar *terminator)
520 cm_return_val_if_fail(session->read_data_buf->len == 0, -1);
522 session->state = SESSION_RECV;
524 g_free(session->read_data_terminator);
525 session->read_data_terminator = g_strdup(terminator);
526 g_date_time_unref(session->tv_prev);
527 session->tv_prev = g_date_time_new_now_local();
529 if (session->read_buf_len > 0)
530 g_idle_add(session_recv_data_idle_cb, session);
532 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
533 session_read_data_cb, session);
538 static gboolean session_recv_data_idle_cb(gpointer data)
540 Session *session = SESSION(data);
543 ret = session_read_data_cb(session->sock, G_IO_IN, session);
546 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
547 session_read_data_cb, session);
552 static gboolean session_read_msg_cb(SockInfo *source, GIOCondition condition,
555 Session *session = SESSION(data);
556 gchar buf[SESSION_BUFFSIZE];
562 cm_return_val_if_fail(condition == G_IO_IN, FALSE);
564 session_set_timeout(session, session->timeout_interval);
566 if (session->read_buf_len == 0) {
570 read_len = sock_read(session->sock, session->read_buf,
571 SESSION_BUFFSIZE - 1);
573 if (read_len == -1 && session->state == SESSION_DISCONNECTED) {
574 g_warning ("sock_read: session disconnected");
575 if (session->io_tag > 0) {
576 g_source_remove(session->io_tag);
583 g_warning("sock_read: received EOF");
584 session->state = SESSION_EOF;
593 g_warning("sock_read: %s", g_strerror(errno));
594 session->state = SESSION_ERROR;
599 session->read_buf_len = read_len;
602 if ((newline = memchr(session->read_buf_p, '\n', session->read_buf_len))
604 line_len = newline - session->read_buf_p + 1;
606 line_len = session->read_buf_len;
611 memcpy(buf, session->read_buf_p, line_len);
612 buf[line_len] = '\0';
614 g_string_append(session->read_msg_buf, buf);
616 session->read_buf_len -= line_len;
617 if (session->read_buf_len == 0)
618 session->read_buf_p = session->read_buf;
620 session->read_buf_p += line_len;
622 /* incomplete read */
623 if (buf[line_len - 1] != '\n')
627 if (session->io_tag > 0) {
628 g_source_remove(session->io_tag);
633 msg = g_strdup(session->read_msg_buf->str);
635 g_string_truncate(session->read_msg_buf, 0);
637 ret = session->recv_msg(session, msg);
638 session->recv_msg_notify(session, msg, session->recv_msg_notify_data);
643 session->state = SESSION_ERROR;
648 static gboolean session_read_data_cb(SockInfo *source, GIOCondition condition,
651 Session *session = SESSION(data);
652 GByteArray *data_buf;
654 gboolean complete = FALSE;
658 cm_return_val_if_fail(condition == G_IO_IN, FALSE);
660 session_set_timeout(session, session->timeout_interval);
662 if (session->read_buf_len == 0) {
665 read_len = sock_read(session->sock, session->read_buf,
669 g_warning("sock_read: received EOF");
670 session->state = SESSION_EOF;
679 g_warning("sock_read: %s", g_strerror(errno));
680 session->state = SESSION_ERROR;
685 session->read_buf_len = read_len;
688 data_buf = session->read_data_buf;
689 terminator_len = strlen(session->read_data_terminator);
691 if (session->read_buf_len == 0)
694 g_byte_array_append(data_buf, session->read_buf_p,
695 session->read_buf_len);
697 session->read_buf_len = 0;
698 session->read_buf_p = session->read_buf;
700 /* check if data is terminated */
701 if (data_buf->len >= terminator_len) {
702 if (memcmp(data_buf->data, session->read_data_terminator,
703 terminator_len) == 0)
705 else if (data_buf->len >= terminator_len + 2 &&
706 memcmp(data_buf->data + data_buf->len -
707 (terminator_len + 2), "\r\n", 2) == 0 &&
708 memcmp(data_buf->data + data_buf->len -
709 terminator_len, session->read_data_terminator,
710 terminator_len) == 0)
714 /* incomplete read */
716 GDateTime *tv_cur = g_date_time_new_now_local();
718 GTimeSpan ts = g_date_time_difference(tv_cur, session->tv_prev);
719 if (1000 - ts < 0 || ts > UI_REFRESH_INTERVAL) {
720 session->recv_data_progressive_notify
721 (session, data_buf->len, 0,
722 session->recv_data_progressive_notify_data);
723 g_date_time_unref(session->tv_prev);
724 session->tv_prev = g_date_time_new_now_local();
726 g_date_time_unref(tv_cur);
731 if (session->io_tag > 0) {
732 g_source_remove(session->io_tag);
736 data_len = data_buf->len - terminator_len;
739 ret = session->recv_data_finished(session, (gchar *)data_buf->data,
742 g_byte_array_set_size(data_buf, 0);
744 session->recv_data_notify(session, data_len,
745 session->recv_data_notify_data);
748 session->state = SESSION_ERROR;
753 static gint session_write_buf(Session *session)
758 cm_return_val_if_fail(session->write_buf != NULL, -1);
759 cm_return_val_if_fail(session->write_buf_p != NULL, -1);
760 cm_return_val_if_fail(session->write_buf_len > 0, -1);
762 to_write_len = session->write_buf_len -
763 (session->write_buf_p - session->write_buf);
764 to_write_len = MIN(to_write_len, SESSION_BUFFSIZE);
766 write_len = sock_write(session->sock, session->write_buf_p,
775 g_warning("sock_write: %s", g_strerror(errno));
776 session->state = SESSION_ERROR;
781 /* incomplete write */
782 if (session->write_buf_p - session->write_buf + write_len <
783 session->write_buf_len) {
784 session->write_buf_p += write_len;
788 g_free(session->write_buf);
789 session->write_buf = NULL;
790 session->write_buf_p = NULL;
791 session->write_buf_len = 0;
796 static gint session_write_data(Session *session)
801 cm_return_val_if_fail(session->write_data != NULL, -1);
802 cm_return_val_if_fail(session->write_data_p != NULL, -1);
803 cm_return_val_if_fail(session->write_data_len > 0, -1);
805 to_write_len = session->write_data_len -
806 (session->write_data_p - session->write_data);
807 to_write_len = MIN(to_write_len, SESSION_BUFFSIZE);
809 write_len = sock_write(session->sock, session->write_data_p,
818 g_warning("sock_write: %s", g_strerror(errno));
819 session->state = SESSION_ERROR;
824 /* incomplete write */
825 if (session->write_data_p - session->write_data + write_len <
826 session->write_data_len) {
827 session->write_data_p += write_len;
831 session->write_data = NULL;
832 session->write_data_p = NULL;
833 session->write_data_len = 0;
838 static gboolean session_write_msg_cb(SockInfo *source, GIOCondition condition,
841 Session *session = SESSION(data);
844 cm_return_val_if_fail(condition == G_IO_OUT, FALSE);
845 cm_return_val_if_fail(session->write_buf != NULL, FALSE);
846 cm_return_val_if_fail(session->write_buf_p != NULL, FALSE);
847 cm_return_val_if_fail(session->write_buf_len > 0, FALSE);
849 ret = session_write_buf(session);
852 session->state = SESSION_ERROR;
857 if (session->io_tag > 0) {
858 g_source_remove(session->io_tag);
862 session_recv_msg(session);
867 static gboolean session_write_data_cb(SockInfo *source,
868 GIOCondition condition, gpointer data)
870 Session *session = SESSION(data);
871 guint write_data_len;
874 cm_return_val_if_fail(condition == G_IO_OUT, FALSE);
875 cm_return_val_if_fail(session->write_data != NULL, FALSE);
876 cm_return_val_if_fail(session->write_data_p != NULL, FALSE);
877 cm_return_val_if_fail(session->write_data_len > 0, FALSE);
879 write_data_len = session->write_data_len;
881 ret = session_write_data(session);
884 session->state = SESSION_ERROR;
886 } else if (ret > 0) {
887 GDateTime *tv_cur = g_date_time_new_now_local();
889 GTimeSpan ts = g_date_time_difference(tv_cur, session->tv_prev);
890 if (1000 - ts < 0 || ts > UI_REFRESH_INTERVAL) {
891 session_set_timeout(session, session->timeout_interval);
892 session->send_data_progressive_notify
894 session->write_data_p - session->write_data,
896 session->send_data_progressive_notify_data);
897 g_date_time_unref(session->tv_prev);
898 session->tv_prev = g_date_time_new_now_local();
900 g_date_time_unref(tv_cur);
904 if (session->io_tag > 0) {
905 g_source_remove(session->io_tag);
910 ret = session->send_data_finished(session, write_data_len);
911 session->send_data_notify(session, write_data_len,
912 session->send_data_notify_data);
917 void session_register_ping(Session *session, gboolean (*ping_cb)(gpointer data))
921 if (session->ping_tag > -1)
922 g_source_remove(session->ping_tag);
924 session->ping_tag = -1;
927 session->ping_tag = g_timeout_add_seconds(60, ping_cb, session);