2 * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
3 * Copyright (C) 1999-2004 Hiroyuki Yamamoto
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 2 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, write to the Free Software
17 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
31 #include <sys/types.h>
33 #include <sys/signal.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)
66 session->type = SESSION_UNKNOWN;
68 session->server = NULL;
71 session->ssl_type = SSL_NONE;
73 session->nonblocking = TRUE;
74 session->state = SESSION_READY;
75 session->last_access_time = time(NULL);
77 gettimeofday(&session->tv_prev, NULL);
83 session->read_buf_p = session->read_buf;
84 session->read_buf_len = 0;
86 session->read_msg_buf = g_string_sized_new(1024);
87 session->read_data_buf = g_byte_array_new();
89 session->write_buf = NULL;
90 session->write_buf_p = NULL;
91 session->write_buf_len = 0;
93 session->timeout_tag = 0;
94 session->timeout_interval = 0;
100 *\brief Set up parent and child process
101 * Childloop: Read commands from parent,
102 * send to server, get answer, pass to parent
104 *\param session Contains session information
105 * server to connect to
109 * -1 : pipe / fork errors (parent)
110 * 1 : connection error (child)
112 gint session_connect(Session *session, const gchar *server, gushort port)
114 session->server = g_strdup(server);
115 session->port = port;
117 session->conn_id = sock_connect_async(server, port, session_connect_cb,
119 if (session->conn_id < 0) {
120 g_warning("can't connect to server.");
121 session_close(session);
128 static gint session_connect_cb(SockInfo *sock, gpointer data)
130 Session *session = SESSION(data);
132 session->conn_id = 0;
135 g_warning("can't connect to server.");
136 session->state = SESSION_ERROR;
140 session->sock = sock;
143 if (session->ssl_type == SSL_TUNNEL) {
144 sock_set_nonblocking_mode(sock, FALSE);
145 if (!ssl_init_socket(sock)) {
146 g_warning("can't initialize SSL.");
147 session->state = SESSION_ERROR;
153 sock_set_nonblocking_mode(sock, session->nonblocking);
155 debug_print("session (%p): connected\n", session);
157 session->state = SESSION_RECV;
158 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
166 *\brief child and parent: send DISCONNECT message to other process
168 *\param session Contains session information
172 gint session_disconnect(Session *session)
174 session_close(session);
181 *\param session Contains session information
183 void session_destroy(Session *session)
185 g_return_if_fail(session != NULL);
186 g_return_if_fail(session->destroy != NULL);
188 session_close(session);
189 session->destroy(session);
190 g_free(session->server);
191 g_string_free(session->read_msg_buf, TRUE);
192 g_byte_array_free(session->read_data_buf, TRUE);
193 g_free(session->read_data_terminator);
194 g_free(session->write_buf);
196 debug_print("session (%p): destroyed\n", session);
201 gboolean session_is_connected(Session *session)
203 return (session->state == SESSION_READY ||
204 session->state == SESSION_SEND ||
205 session->state == SESSION_RECV);
208 void session_set_timeout(Session *session, guint interval)
210 if (session->timeout_tag > 0)
211 g_source_remove(session->timeout_tag);
213 session->timeout_interval = interval;
215 session->timeout_tag =
216 g_timeout_add(interval, session_timeout_cb, session);
218 session->timeout_tag = 0;
221 static gboolean session_timeout_cb(gpointer data)
223 Session *session = SESSION(data);
225 g_warning("session timeout.\n");
227 if (session->io_tag > 0) {
228 g_source_remove(session->io_tag);
232 session->timeout_tag = 0;
233 session->state = SESSION_TIMEOUT;
238 void session_set_recv_message_notify(Session *session,
239 RecvMsgNotify notify_func, gpointer data)
241 session->recv_msg_notify = notify_func;
242 session->recv_msg_notify_data = data;
245 void session_set_recv_data_progressive_notify
247 RecvDataProgressiveNotify notify_func,
250 session->recv_data_progressive_notify = notify_func,
251 session->recv_data_progressive_notify_data = data;
254 void session_set_recv_data_notify(Session *session, RecvDataNotify notify_func,
257 session->recv_data_notify = notify_func;
258 session->recv_data_notify_data = data;
261 void session_set_send_data_progressive_notify
263 SendDataProgressiveNotify notify_func,
266 session->send_data_progressive_notify = notify_func;
267 session->send_data_progressive_notify_data = data;
270 void session_set_send_data_notify(Session *session, SendDataNotify notify_func,
273 session->send_data_notify = notify_func;
274 session->send_data_notify_data = data;
278 *\brief child and parent cleanup (child closes first)
280 *\param session Contains session information
284 static gint session_close(Session *session)
286 g_return_val_if_fail(session != NULL, -1);
288 if (session->conn_id > 0) {
289 sock_connect_async_cancel(session->conn_id);
290 session->conn_id = 0;
291 debug_print("session (%p): connection cancelled\n", session);
294 session_set_timeout(session, 0);
296 if (session->io_tag > 0) {
297 g_source_remove(session->io_tag);
302 sock_close(session->sock);
303 session->sock = NULL;
304 session->state = SESSION_DISCONNECTED;
305 debug_print("session (%p): closed\n", session);
312 gint session_start_tls(Session *session)
316 nb_mode = sock_is_nonblocking_mode(session->sock);
319 sock_set_nonblocking_mode(session->sock, FALSE);
321 if (!ssl_init_socket_with_method(session->sock, SSL_METHOD_TLSv1)) {
322 g_warning("can't start TLS session.\n");
324 sock_set_nonblocking_mode(session->sock, TRUE);
329 sock_set_nonblocking_mode(session->sock, session->nonblocking);
335 gint session_send_msg(Session *session, SessionMsgType type, const gchar *msg)
339 g_return_val_if_fail(session->write_buf == NULL, -1);
340 g_return_val_if_fail(msg != NULL, -1);
341 g_return_val_if_fail(msg[0] != '\0', -1);
343 session->state = SESSION_SEND;
344 session->write_buf = g_strconcat(msg, "\r\n", NULL);
345 session->write_buf_p = session->write_buf;
346 session->write_buf_len = strlen(msg) + 2;
348 ret = session_write_msg_cb(session->sock, G_IO_OUT, session);
351 session->io_tag = sock_add_watch(session->sock, G_IO_OUT,
352 session_write_msg_cb, session);
353 else if (session->state == SESSION_ERROR)
359 gint session_recv_msg(Session *session)
361 g_return_val_if_fail(session->read_msg_buf->len == 0, -1);
363 session->state = SESSION_RECV;
365 if (session->read_buf_len > 0)
366 g_idle_add(session_recv_msg_idle_cb, session);
368 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
369 session_read_msg_cb, session);
374 static gboolean session_recv_msg_idle_cb(gpointer data)
376 Session *session = SESSION(data);
379 ret = session_read_msg_cb(session->sock, G_IO_IN, session);
382 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
383 session_read_msg_cb, session);
389 *\brief parent (child?): send data to other process
391 *\param session Contains session information
398 gint session_send_data(Session *session, const guchar *data, guint size)
402 g_return_val_if_fail(session->write_buf == NULL, -1);
403 g_return_val_if_fail(data != NULL, -1);
404 g_return_val_if_fail(size != 0, -1);
406 session->state = SESSION_SEND;
408 session->write_buf = g_malloc(size);
409 session->write_buf_p = session->write_buf;
410 memcpy(session->write_buf, data, size);
411 session->write_buf_len = size;
412 gettimeofday(&session->tv_prev, NULL);
414 ret = session_write_data_cb(session->sock, G_IO_OUT, session);
417 session->io_tag = sock_add_watch(session->sock, G_IO_OUT,
418 session_write_data_cb,
420 else if (session->state == SESSION_ERROR)
426 gint session_recv_data(Session *session, guint size, const gchar *terminator)
428 g_return_val_if_fail(session->read_data_buf->len == 0, -1);
430 session->state = SESSION_RECV;
432 g_free(session->read_data_terminator);
433 session->read_data_terminator = g_strdup(terminator);
434 gettimeofday(&session->tv_prev, NULL);
436 if (session->read_buf_len > 0)
437 g_idle_add(session_recv_data_idle_cb, session);
439 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
440 session_read_data_cb, session);
445 static gboolean session_recv_data_idle_cb(gpointer data)
447 Session *session = SESSION(data);
450 ret = session_read_data_cb(session->sock, G_IO_IN, session);
453 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
454 session_read_data_cb, session);
459 static gboolean session_read_msg_cb(SockInfo *source, GIOCondition condition,
462 Session *session = SESSION(data);
463 gchar buf[SESSION_BUFFSIZE];
469 g_return_val_if_fail(condition == G_IO_IN, FALSE);
471 session_set_timeout(session, session->timeout_interval);
473 if (session->read_buf_len == 0) {
476 read_len = sock_read(session->sock, session->read_buf,
477 SESSION_BUFFSIZE - 1);
479 if (read_len == -1 && session->state == SESSION_DISCONNECTED) {
480 g_warning ("sock_read: session disconnected\n");
481 if (session->io_tag > 0) {
482 g_source_remove(session->io_tag);
489 g_warning("sock_read: received EOF\n");
490 session->state = SESSION_EOF;
499 g_warning("sock_read: %s\n", g_strerror(errno));
500 session->state = SESSION_ERROR;
505 session->read_buf_len = read_len;
508 if ((newline = memchr(session->read_buf_p, '\n', session->read_buf_len))
510 line_len = newline - session->read_buf_p + 1;
512 line_len = session->read_buf_len;
517 memcpy(buf, session->read_buf_p, line_len);
518 buf[line_len] = '\0';
520 g_string_append(session->read_msg_buf, buf);
522 session->read_buf_len -= line_len;
523 if (session->read_buf_len == 0)
524 session->read_buf_p = session->read_buf;
526 session->read_buf_p += line_len;
528 /* incomplete read */
529 if (buf[line_len - 1] != '\n')
533 if (session->io_tag > 0) {
534 g_source_remove(session->io_tag);
539 msg = g_strdup(session->read_msg_buf->str);
541 g_string_truncate(session->read_msg_buf, 0);
543 ret = session->recv_msg(session, msg);
544 session->recv_msg_notify(session, msg, session->recv_msg_notify_data);
549 session->state = SESSION_ERROR;
554 static gboolean session_read_data_cb(SockInfo *source, GIOCondition condition,
557 Session *session = SESSION(data);
558 GByteArray *data_buf;
560 gboolean complete = FALSE;
564 g_return_val_if_fail(condition == G_IO_IN, FALSE);
566 session_set_timeout(session, session->timeout_interval);
568 if (session->read_buf_len == 0) {
571 read_len = sock_read(session->sock, session->read_buf,
575 g_warning("sock_read: received EOF\n");
576 session->state = SESSION_EOF;
585 g_warning("sock_read: %s\n", g_strerror(errno));
586 session->state = SESSION_ERROR;
591 session->read_buf_len = read_len;
594 data_buf = session->read_data_buf;
595 terminator_len = strlen(session->read_data_terminator);
597 if (session->read_buf_len == 0)
600 g_byte_array_append(data_buf, session->read_buf_p,
601 session->read_buf_len);
603 session->read_buf_len = 0;
604 session->read_buf_p = session->read_buf;
606 /* check if data is terminated */
607 if (data_buf->len >= terminator_len) {
608 if (memcmp(data_buf->data, session->read_data_terminator,
609 terminator_len) == 0)
611 else if (data_buf->len >= terminator_len + 2 &&
612 memcmp(data_buf->data + data_buf->len -
613 (terminator_len + 2), "\r\n", 2) == 0 &&
614 memcmp(data_buf->data + data_buf->len -
615 terminator_len, session->read_data_terminator,
616 terminator_len) == 0)
620 /* incomplete read */
622 struct timeval tv_cur;
624 gettimeofday(&tv_cur, NULL);
625 if (tv_cur.tv_sec - session->tv_prev.tv_sec > 0 ||
626 tv_cur.tv_usec - session->tv_prev.tv_usec >
627 UI_REFRESH_INTERVAL) {
628 session->recv_data_progressive_notify
629 (session, data_buf->len, 0,
630 session->recv_data_progressive_notify_data);
631 gettimeofday(&session->tv_prev, NULL);
637 if (session->io_tag > 0) {
638 g_source_remove(session->io_tag);
642 data_len = data_buf->len - terminator_len;
645 ret = session->recv_data_finished(session, (gchar *)data_buf->data,
648 g_byte_array_set_size(data_buf, 0);
650 session->recv_data_notify(session, data_len,
651 session->recv_data_notify_data);
654 session->state = SESSION_ERROR;
659 static gint session_write_buf(Session *session)
664 g_return_val_if_fail(session->write_buf != NULL, -1);
665 g_return_val_if_fail(session->write_buf_p != NULL, -1);
666 g_return_val_if_fail(session->write_buf_len > 0, -1);
668 to_write_len = session->write_buf_len -
669 (session->write_buf_p - session->write_buf);
670 to_write_len = MIN(to_write_len, SESSION_BUFFSIZE);
672 write_len = sock_write(session->sock, session->write_buf_p,
681 g_warning("sock_write: %s\n", g_strerror(errno));
682 session->state = SESSION_ERROR;
687 /* incomplete write */
688 if (session->write_buf_p - session->write_buf + write_len <
689 session->write_buf_len) {
690 session->write_buf_p += write_len;
694 g_free(session->write_buf);
695 session->write_buf = NULL;
696 session->write_buf_p = NULL;
697 session->write_buf_len = 0;
702 static gboolean session_write_msg_cb(SockInfo *source, GIOCondition condition,
705 Session *session = SESSION(data);
708 g_return_val_if_fail(condition == G_IO_OUT, FALSE);
709 g_return_val_if_fail(session->write_buf != NULL, FALSE);
710 g_return_val_if_fail(session->write_buf_p != NULL, FALSE);
711 g_return_val_if_fail(session->write_buf_len > 0, FALSE);
713 ret = session_write_buf(session);
716 session->state = SESSION_ERROR;
721 if (session->io_tag > 0) {
722 g_source_remove(session->io_tag);
726 session_recv_msg(session);
731 static gboolean session_write_data_cb(SockInfo *source,
732 GIOCondition condition, gpointer data)
734 Session *session = SESSION(data);
738 g_return_val_if_fail(condition == G_IO_OUT, FALSE);
739 g_return_val_if_fail(session->write_buf != NULL, FALSE);
740 g_return_val_if_fail(session->write_buf_p != NULL, FALSE);
741 g_return_val_if_fail(session->write_buf_len > 0, FALSE);
743 write_buf_len = session->write_buf_len;
745 ret = session_write_buf(session);
748 session->state = SESSION_ERROR;
750 } else if (ret > 0) {
751 struct timeval tv_cur;
753 gettimeofday(&tv_cur, NULL);
754 if (tv_cur.tv_sec - session->tv_prev.tv_sec > 0 ||
755 tv_cur.tv_usec - session->tv_prev.tv_usec >
756 UI_REFRESH_INTERVAL) {
757 session_set_timeout(session, session->timeout_interval);
758 session->send_data_progressive_notify
760 session->write_buf_p - session->write_buf,
762 session->send_data_progressive_notify_data);
763 gettimeofday(&session->tv_prev, NULL);
768 if (session->io_tag > 0) {
769 g_source_remove(session->io_tag);
774 ret = session->send_data_finished(session, write_buf_len);
775 session->send_data_notify(session, write_buf_len,
776 session->send_data_notify_data);