2 * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
3 * Copyright (C) 1999-2003 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_read_msg_cb (GIOChannel *source,
46 GIOCondition condition,
48 static gboolean session_read_data_cb (GIOChannel *source,
49 GIOCondition condition,
51 static gboolean session_write_msg_cb (GIOChannel *source,
52 GIOCondition condition,
54 static gboolean session_write_data_cb (GIOChannel *source,
55 GIOCondition condition,
59 void session_init(Session *session)
61 session->type = SESSION_UNKNOWN;
63 session->server = NULL;
66 session->ssl_type = SSL_NONE;
68 session->state = SESSION_READY;
69 session->last_access_time = time(NULL);
71 gettimeofday(&session->tv_prev, NULL);
75 session->sock_ch = NULL;
78 session->read_buf = g_string_sized_new(1024);
79 session->read_data_buf = g_byte_array_new();
80 session->write_buf = NULL;
81 session->write_buf_p = NULL;
82 session->write_buf_len = 0;
88 *\brief Set up parent and child process
89 * Childloop: Read commands from parent,
90 * send to server, get answer, pass to parent
92 *\param session Contains session information
93 * server to connect to
97 * -1 : pipe / fork errors (parent)
98 * 1 : connection error (child)
100 gint session_connect(Session *session, const gchar *server, gushort port)
102 session->server = g_strdup(server);
103 session->port = port;
105 session->conn_id = sock_connect_async(server, port, session_connect_cb,
107 if (session->conn_id < 0) {
108 g_warning("can't connect to server.");
109 session_close(session);
116 static gint session_connect_cb(SockInfo *sock, gpointer data)
118 Session *session = SESSION(data);
120 session->conn_id = 0;
123 g_warning("can't connect to server.");
124 session->state = SESSION_ERROR;
128 session->sock = sock;
131 sock_set_nonblocking_mode(sock, FALSE);
132 if (session->ssl_type == SSL_TUNNEL && !ssl_init_socket(sock)) {
133 g_warning("can't initialize SSL.");
134 session->state = SESSION_ERROR;
139 debug_print("session: child: connected\n");
141 debug_print("session: connected\n");
143 session->state = SESSION_RECV;
144 session->sock_ch = g_io_channel_unix_new(sock->sock);
145 session->io_tag = g_io_add_watch(session->sock_ch, G_IO_IN,
153 *\brief child and parent: send DISCONNECT message to other process
155 *\param session Contains session information
159 gint session_disconnect(Session *session)
161 session_close(session);
168 *\param session Contains session information
170 void session_destroy(Session *session)
172 g_return_if_fail(session != NULL);
173 g_return_if_fail(session->destroy != NULL);
175 debug_print("session: session_destroy()\n");
176 session_close(session);
177 session->destroy(session);
178 g_free(session->server);
179 g_string_free(session->read_buf, TRUE);
180 g_byte_array_free(session->read_data_buf, TRUE);
181 g_free(session->read_data_terminator);
182 g_free(session->write_buf);
186 void session_set_recv_message_notify(Session *session,
187 RecvMsgNotify notify_func, gpointer data)
189 session->recv_msg_notify = notify_func;
190 session->recv_msg_notify_data = data;
193 void session_set_recv_data_progressive_notify
195 RecvDataProgressiveNotify notify_func,
198 session->recv_data_progressive_notify = notify_func,
199 session->recv_data_progressive_notify_data = data;
202 void session_set_recv_data_notify(Session *session, RecvDataNotify notify_func,
205 session->recv_data_notify = notify_func;
206 session->recv_data_notify_data = data;
209 void session_set_send_data_progressive_notify
211 SendDataProgressiveNotify notify_func,
214 session->send_data_progressive_notify = notify_func;
215 session->send_data_progressive_notify_data = data;
218 void session_set_send_data_notify(Session *session, SendDataNotify notify_func,
221 session->send_data_notify = notify_func;
222 session->send_data_notify_data = data;
226 *\brief child and parent cleanup (child closes first)
228 *\param session Contains session information
232 static gint session_close(Session *session)
234 g_return_val_if_fail(session != NULL, -1);
236 debug_print("session_close\n");
238 if (session->conn_id > 0) {
239 sock_connect_async_cancel(session->conn_id);
240 session->conn_id = 0;
243 if (session->io_tag > 0) {
244 g_source_remove(session->io_tag);
248 if (session->sock_ch) {
249 g_io_channel_unref(session->sock_ch);
250 session->sock_ch = NULL;
254 sock_close(session->sock);
255 session->sock = NULL;
256 session->state = SESSION_DISCONNECTED;
263 gint session_start_tls(Session *session)
267 nb_mode = sock_is_nonblocking_mode(session->sock);
270 sock_set_nonblocking_mode(session->sock, FALSE);
272 if (!ssl_init_socket_with_method(session->sock, SSL_METHOD_TLSv1)) {
273 g_warning("can't start TLS session.\n");
275 sock_set_nonblocking_mode(session->sock, TRUE);
280 sock_set_nonblocking_mode(session->sock, TRUE);
286 gint session_send_msg(Session *session, SessionMsgType type, const gchar *msg)
290 g_return_val_if_fail(session->write_buf == NULL, -1);
291 g_return_val_if_fail(msg != NULL, -1);
292 g_return_val_if_fail(msg[0] != '\0', -1);
294 session->state = SESSION_SEND;
295 session->write_buf = g_strconcat(msg, "\r\n", NULL);
296 session->write_buf_p = session->write_buf;
297 session->write_buf_len = strlen(msg) + 2;
299 ret = session_write_msg_cb(session->sock_ch, G_IO_OUT, session);
302 session->io_tag = g_io_add_watch(session->sock_ch, G_IO_OUT,
303 session_write_msg_cb, session);
304 else if (session->state == SESSION_ERROR)
310 gint session_recv_msg(Session *session)
312 g_return_val_if_fail(session->read_buf->len == 0, -1);
314 session->state = SESSION_RECV;
316 session->io_tag = g_io_add_watch(session->sock_ch, G_IO_IN,
317 session_read_msg_cb, session);
323 *\brief parent (child?): send data to other process
325 *\param session Contains session information
332 gint session_send_data(Session *session, const guchar *data, guint size)
336 g_return_val_if_fail(session->write_buf == NULL, -1);
337 g_return_val_if_fail(data != NULL, -1);
338 g_return_val_if_fail(size != 0, -1);
340 session->state = SESSION_SEND;
342 session->write_buf = g_malloc(size);
343 session->write_buf_p = session->write_buf;
344 memcpy(session->write_buf, data, size);
345 session->write_buf_len = size;
346 gettimeofday(&session->tv_prev, NULL);
348 ret = session_write_data_cb(session->sock_ch, G_IO_OUT, session);
351 session->io_tag = g_io_add_watch(session->sock_ch, G_IO_OUT,
352 session_write_data_cb,
354 else if (session->state == SESSION_ERROR)
360 gint session_recv_data(Session *session, guint size, const gchar *terminator)
362 g_return_val_if_fail(session->read_data_buf->len == 0, -1);
364 session->state = SESSION_RECV;
366 g_free(session->read_data_terminator);
367 session->read_data_terminator = g_strdup(terminator);
368 gettimeofday(&session->tv_prev, NULL);
370 session->io_tag = g_io_add_watch(session->sock_ch, G_IO_IN,
371 session_read_data_cb, session);
376 static gboolean session_read_msg_cb(GIOChannel *source, GIOCondition condition,
379 Session *session = SESSION(data);
380 gchar buf[SESSION_BUFFSIZE];
387 g_return_val_if_fail(condition == G_IO_IN, FALSE);
389 read_len = sock_peek(session->sock, buf, sizeof(buf) - 1);
396 g_warning("sock_peek: %s\n", g_strerror(errno));
397 session->state = SESSION_ERROR;
402 if ((newline = memchr(buf, '\n', read_len)) != NULL)
403 to_read_len = newline - buf + 1;
405 to_read_len = read_len;
407 read_len = sock_read(session->sock, buf, to_read_len);
409 /* this should always succeed */
415 g_warning("sock_read: %s\n", g_strerror(errno));
416 session->state = SESSION_ERROR;
421 buf[read_len] = '\0';
423 /* incomplete read */
424 if (read_len == 0 || buf[read_len - 1] != '\n') {
425 g_string_append(session->read_buf, buf);
431 g_string_append(session->read_buf, buf);
433 if (session->io_tag > 0) {
434 g_source_remove(session->io_tag);
439 msg = g_strdup(session->read_buf->str);
440 g_string_truncate(session->read_buf, 0);
442 ret = session->recv_msg(session, msg);
443 session->recv_msg_notify(session, msg, session->recv_msg_notify_data);
448 session->state = SESSION_ERROR;
453 static gboolean session_read_data_cb(GIOChannel *source, GIOCondition condition,
456 Session *session = SESSION(data);
457 gchar buf[SESSION_BUFFSIZE];
458 GByteArray *data_buf;
461 gboolean complete = FALSE;
466 g_return_val_if_fail(condition == G_IO_IN, FALSE);
468 read_len = sock_read(session->sock, buf, sizeof(buf));
475 g_warning("sock_read: %s\n", g_strerror(errno));
476 session->state = SESSION_ERROR;
481 data_buf = session->read_data_buf;
483 g_byte_array_append(data_buf, buf, read_len);
484 terminator_len = strlen(session->read_data_terminator);
486 /* check if data is terminated */
487 if (read_len > 0 && data_buf->len >= terminator_len) {
488 if (memcmp(data_buf->data, session->read_data_terminator,
489 terminator_len) == 0)
491 else if (data_buf->len >= terminator_len + 2 &&
492 memcmp(data_buf->data + data_buf->len -
493 (terminator_len + 2), "\r\n", 2) == 0 &&
494 memcmp(data_buf->data + data_buf->len -
495 terminator_len, session->read_data_terminator,
496 terminator_len) == 0)
500 /* incomplete read */
502 struct timeval tv_cur;
504 gettimeofday(&tv_cur, NULL);
505 if (tv_cur.tv_sec - session->tv_prev.tv_sec > 0 ||
506 tv_cur.tv_usec - session->tv_prev.tv_usec >
507 UI_REFRESH_INTERVAL) {
508 session->recv_data_progressive_notify
509 (session, data_buf->len, 0,
510 session->recv_data_progressive_notify_data);
511 gettimeofday(&session->tv_prev, NULL);
517 if (session->io_tag > 0) {
518 g_source_remove(session->io_tag);
523 ret_data = g_malloc(data_buf->len);
524 memcpy(ret_data, data_buf->data, data_buf->len);
525 data_len = data_buf->len - terminator_len;
526 g_byte_array_set_size(data_buf, 0);
528 ret = session->recv_data_finished(session, ret_data, data_len);
529 session->recv_data_notify(session, data_len,
530 session->recv_data_notify_data);
535 session->state = SESSION_ERROR;
540 static gint session_write_buf(Session *session)
545 g_return_val_if_fail(session->write_buf != NULL, -1);
546 g_return_val_if_fail(session->write_buf_p != NULL, -1);
547 g_return_val_if_fail(session->write_buf_len > 0, -1);
549 to_write_len = session->write_buf_len -
550 (session->write_buf_p - session->write_buf);
551 to_write_len = MIN(to_write_len, SESSION_BUFFSIZE);
553 write_len = sock_write(session->sock, session->write_buf_p,
562 g_warning("sock_write: %s\n", g_strerror(errno));
563 session->state = SESSION_ERROR;
568 /* incomplete write */
569 if (session->write_buf_p - session->write_buf + write_len <
570 session->write_buf_len) {
571 session->write_buf_p += write_len;
575 g_free(session->write_buf);
576 session->write_buf = NULL;
577 session->write_buf_p = NULL;
578 session->write_buf_len = 0;
583 static gboolean session_write_msg_cb(GIOChannel *source, GIOCondition condition,
586 Session *session = SESSION(data);
589 g_return_val_if_fail(condition == G_IO_OUT, FALSE);
590 g_return_val_if_fail(session->write_buf != NULL, FALSE);
591 g_return_val_if_fail(session->write_buf_p != NULL, FALSE);
592 g_return_val_if_fail(session->write_buf_len > 0, FALSE);
594 ret = session_write_buf(session);
597 session->state = SESSION_ERROR;
602 if (session->io_tag > 0) {
603 g_source_remove(session->io_tag);
607 session_recv_msg(session);
612 static gboolean session_write_data_cb(GIOChannel *source,
613 GIOCondition condition, gpointer data)
615 Session *session = SESSION(data);
619 g_return_val_if_fail(condition == G_IO_OUT, FALSE);
620 g_return_val_if_fail(session->write_buf != NULL, FALSE);
621 g_return_val_if_fail(session->write_buf_p != NULL, FALSE);
622 g_return_val_if_fail(session->write_buf_len > 0, FALSE);
624 write_buf_len = session->write_buf_len;
626 ret = session_write_buf(session);
629 session->state = SESSION_ERROR;
631 } else if (ret > 0) {
632 struct timeval tv_cur;
634 gettimeofday(&tv_cur, NULL);
635 if (tv_cur.tv_sec - session->tv_prev.tv_sec > 0 ||
636 tv_cur.tv_usec - session->tv_prev.tv_usec >
637 UI_REFRESH_INTERVAL) {
638 session->send_data_progressive_notify
640 session->write_buf_p - session->write_buf,
642 session->send_data_progressive_notify_data);
643 gettimeofday(&session->tv_prev, NULL);
648 if (session->io_tag > 0) {
649 g_source_remove(session->io_tag);
654 ret = session->send_data_finished(session, write_buf_len);
655 session->send_data_notify(session, write_buf_len,
656 session->send_data_notify_data);