src/common/session.c

   1 /*
   2  * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
   3  * Copyright (C) 1999-2012 Hiroyuki Yamamoto and the Claws Mail team
   4  *
   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.
   9  *
  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.
  14  *
  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/>.
  17  *
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #  include "config.h"
  22 #include "claws-features.h"
  23 #endif
  24
  25 #include "defs.h"
  26
  27 #include <glib.h>
  28 #include <glib/gi18n.h>
  29
  30 #include <stdio.h>
  31 #include <stdlib.h>
  32 #include <string.h>
  33 #include <unistd.h>
  34 #include <time.h>
  35 #include <errno.h>
  36
  37 #include "session.h"
  38 #include "utils.h"
  39 #include "log.h"
  40
  41 static gint session_connect_cb          (SockInfo       *sock,
  42                                          gpointer        data);
  43 static gint session_close               (Session        *session);
  44
  45 static gboolean session_timeout_cb      (gpointer        data);
  46
  47 static gboolean session_recv_msg_idle_cb        (gpointer        data);
  48 static gboolean session_recv_data_idle_cb       (gpointer        data);
  49
  50 static gboolean session_read_msg_cb     (SockInfo       *source,
  51                                          GIOCondition    condition,
  52                                          gpointer        data);
  53 static gboolean session_read_data_cb    (SockInfo       *source,
  54                                          GIOCondition    condition,
  55                                          gpointer        data);
  56 static gboolean session_write_msg_cb    (SockInfo       *source,
  57                                          GIOCondition    condition,
  58                                          gpointer        data);
  59 static gboolean session_write_data_cb   (SockInfo       *source,
  60                                          GIOCondition    condition,
  61                                          gpointer        data);
  62
  63
  64 void session_init(Session *session, const void *prefs_account, gboolean is_smtp)
  65 {
  66         session->type = SESSION_UNKNOWN;
  67         session->sock = NULL;
  68         session->server = NULL;
  69         session->port = 0;
  70 #ifdef USE_GNUTLS
  71         session->ssl_type = SSL_NONE;
  72         session->use_tls_sni = TRUE;
  73 #endif
  74         session->nonblocking = TRUE;
  75         session->state = SESSION_READY;
  76         session->last_access_time = time(NULL);
  77
  78         session->tv_prev = g_date_time_new_now_local();
  79
  80         session->conn_id = 0;
  81
  82         session->io_tag = 0;
  83
  84         session->read_buf_p = session->read_buf;
  85         session->read_buf_len = 0;
  86
  87         session->read_msg_buf = g_string_sized_new(1024);
  88         session->read_data_buf = g_byte_array_new();
  89
  90         session->write_buf = NULL;
  91         session->write_buf_p = NULL;
  92         session->write_buf_len = 0;
  93
  94         session->write_data = NULL;
  95         session->write_data_p = NULL;
  96         session->write_data_len = 0;
  97
  98         session->timeout_tag = 0;
  99         session->timeout_interval = 0;
 100
 101         session->data = NULL;
 102         session->account = prefs_account;
 103         session->is_smtp = is_smtp;
 104
 105         session->ping_tag = -1;
 106
 107         session->proxy_info = NULL;
 108 }
 109
 110 /*!
 111  *\brief        Set up parent and child process
 112  *              Childloop: Read commands from parent,
 113  *              send to server, get answer, pass to parent
 114  *
 115  *\param        session Contains session information
 116  *              server to connect to
 117  *              port to connect to
 118  *
 119  *\return        0 : success
 120  *              -1 : pipe / fork errors (parent)
 121  *               1 : connection error (child)
 122  */
 123 gint session_connect(Session *session, const gchar *server, gushort port)
 124 {
 125         session->server = g_strdup(server);
 126         session->port = port;
 127
 128         if (session->proxy_info) {
 129                 server = session->proxy_info->proxy_host;
 130                 port = session->proxy_info->proxy_port;
 131         }
 132
 133 #ifdef G_OS_UNIX
 134         session->conn_id = sock_connect_async(server, port, session_connect_cb,
 135                                               session);
 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);
 141                 return -1;
 142         }
 143
 144         return 0;
 145 #else
 146         SockInfo *sock;
 147
 148         sock = sock_connect(server, port);
 149         if (sock == NULL) {
 150                 g_warning("can't connect to server.");
 151                 session_close(session);
 152                 if (session->connect_finished)
 153                         session->connect_finished(session, FALSE);
 154                 return -1;
 155         }
 156         sock->is_smtp = session->is_smtp;
 157
 158         return session_connect_cb(sock, session);
 159 #endif
 160 }
 161
 162 static gint session_connect_cb(SockInfo *sock, gpointer data)
 163 {
 164         Session *session = SESSION(data);
 165
 166         session->conn_id = 0;
 167
 168         if (!sock) {
 169                 g_warning("can't connect to server.");
 170                 session->state = SESSION_ERROR;
 171                 if (session->connect_finished)
 172                         session->connect_finished(session, FALSE);
 173                 return -1;
 174         }
 175
 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;
 180
 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;
 188                         return -1;
 189                 }
 190         }
 191
 192
 193 #ifdef USE_GNUTLS
 194         sock->gnutls_priority = session->gnutls_priority;
 195         sock->use_tls_sni = session->use_tls_sni;
 196
 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);
 205                         return -1;
 206                 }
 207         }
 208 #endif
 209
 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);
 215                 return -1;
 216         }
 217
 218         sock_set_nonblocking_mode(sock, session->nonblocking);
 219
 220         debug_print("session (%p): connected\n", session);
 221
 222         session->state = SESSION_RECV;
 223         session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 224                                          session_read_msg_cb,
 225                                          session);
 226
 227         if (session->connect_finished)
 228                 session->connect_finished(session, TRUE);
 229         return 0;
 230 }
 231
 232 /*!
 233  *\brief        child and parent: send DISCONNECT message to other process
 234  *
 235  *\param        session Contains session information
 236  *
 237  *\return        0 : success
 238  */
 239 gint session_disconnect(Session *session)
 240 {
 241         session_close(session);
 242         return 0;
 243 }
 244
 245 /*!
 246  *\brief        parent ?
 247  *
 248  *\param        session Contains session information
 249  */
 250 void session_destroy(Session *session)
 251 {
 252         cm_return_if_fail(session != NULL);
 253         cm_return_if_fail(session->destroy != NULL);
 254
 255         session_register_ping(session, NULL);
 256
 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);
 264 #ifdef USE_GNUTLS
 265         g_free(session->gnutls_priority);
 266 #endif
 267         g_date_time_unref(session->tv_prev);
 268         debug_print("session (%p): destroyed\n", session);
 269
 270         g_free(session);
 271 }
 272
 273 gboolean session_is_running(Session *session)
 274 {
 275         return (session->state == SESSION_READY ||
 276                 session->state == SESSION_SEND ||
 277                 session->state == SESSION_RECV);
 278 }
 279
 280 gboolean session_is_connected(Session *session)
 281 {
 282         return (session->state == SESSION_SEND ||
 283                 session->state == SESSION_RECV);
 284 }
 285
 286 void session_set_access_time(Session *session)
 287 {
 288         session->last_access_time = time(NULL);
 289 }
 290
 291 void session_set_timeout(Session *session, guint interval)
 292 {
 293         if (session->timeout_tag > 0)
 294                 g_source_remove(session->timeout_tag);
 295
 296         session->timeout_interval = interval;
 297         if (interval > 0) {
 298                 if (interval % 1000 == 0)
 299                         session->timeout_tag =
 300                                 g_timeout_add_seconds(interval/1000, session_timeout_cb, session);
 301                 else
 302                         session->timeout_tag =
 303                                 g_timeout_add(interval, session_timeout_cb, session);
 304         } else
 305                 session->timeout_tag = 0;
 306 }
 307
 308 static gboolean session_timeout_cb(gpointer data)
 309 {
 310         Session *session = SESSION(data);
 311
 312         g_warning("session timeout.");
 313
 314         if (session->io_tag > 0) {
 315                 g_source_remove(session->io_tag);
 316                 session->io_tag = 0;
 317         }
 318
 319         session->timeout_tag = 0;
 320         session->state = SESSION_TIMEOUT;
 321
 322         return FALSE;
 323 }
 324
 325 void session_set_recv_message_notify(Session *session,
 326                                      RecvMsgNotify notify_func, gpointer data)
 327 {
 328         session->recv_msg_notify = notify_func;
 329         session->recv_msg_notify_data = data;
 330 }
 331
 332 void session_set_recv_data_progressive_notify
 333                                         (Session *session,
 334                                          RecvDataProgressiveNotify notify_func,
 335                                          gpointer data)
 336 {
 337         session->recv_data_progressive_notify = notify_func,
 338         session->recv_data_progressive_notify_data = data;
 339 }
 340
 341 void session_set_recv_data_notify(Session *session, RecvDataNotify notify_func,
 342                                   gpointer data)
 343 {
 344         session->recv_data_notify = notify_func;
 345         session->recv_data_notify_data = data;
 346 }
 347
 348 void session_set_send_data_progressive_notify
 349                                         (Session *session,
 350                                          SendDataProgressiveNotify notify_func,
 351                                          gpointer data)
 352 {
 353         session->send_data_progressive_notify = notify_func;
 354         session->send_data_progressive_notify_data = data;
 355 }
 356
 357 void session_set_send_data_notify(Session *session, SendDataNotify notify_func,
 358                                   gpointer data)
 359 {
 360         session->send_data_notify = notify_func;
 361         session->send_data_notify_data = data;
 362 }
 363
 364 /*!
 365  *\brief        child and parent cleanup (child closes first)
 366  *
 367  *\param        session Contains session information
 368  *
 369  *\return        0 : success
 370  */
 371 static gint session_close(Session *session)
 372 {
 373         cm_return_val_if_fail(session != NULL, -1);
 374
 375 #ifdef G_OS_UNIX
 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);
 380         }
 381 #endif
 382
 383         session_set_timeout(session, 0);
 384
 385         if (session->io_tag > 0) {
 386                 g_source_remove(session->io_tag);
 387                 session->io_tag = 0;
 388         }
 389
 390         if (session->sock) {
 391                 sock_close(session->sock, TRUE);
 392                 session->sock = NULL;
 393                 session->state = SESSION_DISCONNECTED;
 394                 debug_print("session (%p): closed\n", session);
 395         }
 396
 397         return 0;
 398 }
 399
 400 #ifdef USE_GNUTLS
 401 gint session_start_tls(Session *session)
 402 {
 403         gboolean nb_mode;
 404
 405         nb_mode = sock_is_nonblocking_mode(session->sock);
 406
 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;
 410
 411         if (nb_mode)
 412                 sock_set_nonblocking_mode(session->sock, FALSE);
 413
 414         if (!ssl_init_socket(session->sock)) {
 415                 g_warning("couldn't start STARTTLS session.");
 416                 if (nb_mode)
 417                         sock_set_nonblocking_mode(session->sock, session->nonblocking);
 418                 return -1;
 419         }
 420
 421         if (nb_mode)
 422                 sock_set_nonblocking_mode(session->sock, session->nonblocking);
 423
 424         return 0;
 425 }
 426 #endif
 427
 428 gint session_send_msg(Session *session, const gchar *msg)
 429 {
 430         gboolean ret;
 431
 432         cm_return_val_if_fail(session->write_buf == NULL, -1);
 433         cm_return_val_if_fail(msg != NULL, -1);
 434
 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;
 439
 440         ret = session_write_msg_cb(session->sock, G_IO_OUT, session);
 441
 442         if (ret == TRUE)
 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)
 446                 return -1;
 447
 448         return 0;
 449 }
 450
 451 gint session_recv_msg(Session *session)
 452 {
 453         cm_return_val_if_fail(session->read_msg_buf->len == 0, -1);
 454
 455         session->state = SESSION_RECV;
 456
 457         if (session->read_buf_len > 0)
 458                 g_idle_add(session_recv_msg_idle_cb, session);
 459         else
 460                 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 461                                                  session_read_msg_cb, session);
 462
 463         return 0;
 464 }
 465
 466 static gboolean session_recv_msg_idle_cb(gpointer data)
 467 {
 468         Session *session = SESSION(data);
 469         gboolean ret;
 470
 471         ret = session_read_msg_cb(session->sock, G_IO_IN, session);
 472
 473         if (ret == TRUE)
 474                 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 475                                                  session_read_msg_cb, session);
 476
 477         return FALSE;
 478 }
 479
 480 /*!
 481  *\brief        parent (child?): send data to other process
 482  *
 483  *\param        session Contains session information
 484  *              data Data to send
 485  *              size Bytes to send
 486  *
 487  *\return        0 : success
 488  *              -1 : error
 489  */
 490 gint session_send_data(Session *session, const guchar *data, guint size)
 491 {
 492         gboolean ret;
 493
 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);
 497
 498         session->state = SESSION_SEND;
 499
 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();
 505
 506         ret = session_write_data_cb(session->sock, G_IO_OUT, session);
 507
 508         if (ret == TRUE)
 509                 session->io_tag = sock_add_watch(session->sock, G_IO_OUT,
 510                                                  session_write_data_cb,
 511                                                  session);
 512         else if (session->state == SESSION_ERROR)
 513                 return -1;
 514
 515         return 0;
 516 }
 517
 518 gint session_recv_data(Session *session, guint size, const gchar *terminator)
 519 {
 520         cm_return_val_if_fail(session->read_data_buf->len == 0, -1);
 521
 522         session->state = SESSION_RECV;
 523
 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();
 528
 529         if (session->read_buf_len > 0)
 530                 g_idle_add(session_recv_data_idle_cb, session);
 531         else
 532                 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 533                                                  session_read_data_cb, session);
 534
 535         return 0;
 536 }
 537
 538 static gboolean session_recv_data_idle_cb(gpointer data)
 539 {
 540         Session *session = SESSION(data);
 541         gboolean ret;
 542
 543         ret = session_read_data_cb(session->sock, G_IO_IN, session);
 544
 545         if (ret == TRUE)
 546                 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 547                                                  session_read_data_cb, session);
 548
 549         return FALSE;
 550 }
 551
 552 static gboolean session_read_msg_cb(SockInfo *source, GIOCondition condition,
 553                                     gpointer data)
 554 {
 555         Session *session = SESSION(data);
 556         gchar buf[SESSION_BUFFSIZE];
 557         gint line_len;
 558         gchar *newline;
 559         gchar *msg;
 560         gint ret;
 561
 562         cm_return_val_if_fail(condition == G_IO_IN, FALSE);
 563
 564         session_set_timeout(session, session->timeout_interval);
 565
 566         if (session->read_buf_len == 0) {
 567                 gint read_len = -1;
 568
 569                 if (session->sock)
 570                         read_len = sock_read(session->sock, session->read_buf,
 571                                      SESSION_BUFFSIZE - 1);
 572
 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);
 577                                 session->io_tag = 0;
 578                         }
 579                         return FALSE;
 580                 }
 581
 582                 if (read_len == 0) {
 583                         g_warning("sock_read: received EOF");
 584                         session->state = SESSION_EOF;
 585                         return FALSE;
 586                 }
 587
 588                 if (read_len < 0) {
 589                         switch (errno) {
 590                         case EAGAIN:
 591                                 return TRUE;
 592                         default:
 593                                 g_warning("sock_read: %s", g_strerror(errno));
 594                                 session->state = SESSION_ERROR;
 595                                 return FALSE;
 596                         }
 597                 }
 598
 599                 session->read_buf_len = read_len;
 600         }
 601
 602         if ((newline = memchr(session->read_buf_p, '\n', session->read_buf_len))
 603                 != NULL)
 604                 line_len = newline - session->read_buf_p + 1;
 605         else
 606                 line_len = session->read_buf_len;
 607
 608         if (line_len == 0)
 609                 return TRUE;
 610
 611         memcpy(buf, session->read_buf_p, line_len);
 612         buf[line_len] = '\0';
 613
 614         g_string_append(session->read_msg_buf, buf);
 615
 616         session->read_buf_len -= line_len;
 617         if (session->read_buf_len == 0)
 618                 session->read_buf_p = session->read_buf;
 619         else
 620                 session->read_buf_p += line_len;
 621
 622         /* incomplete read */
 623         if (buf[line_len - 1] != '\n')
 624                 return TRUE;
 625
 626         /* complete */
 627         if (session->io_tag > 0) {
 628                 g_source_remove(session->io_tag);
 629                 session->io_tag = 0;
 630         }
 631
 632         /* callback */
 633         msg = g_strdup(session->read_msg_buf->str);
 634         strretchomp(msg);
 635         g_string_truncate(session->read_msg_buf, 0);
 636
 637         ret = session->recv_msg(session, msg);
 638         session->recv_msg_notify(session, msg, session->recv_msg_notify_data);
 639
 640         g_free(msg);
 641
 642         if (ret < 0)
 643                 session->state = SESSION_ERROR;
 644
 645         return FALSE;
 646 }
 647
 648 static gboolean session_read_data_cb(SockInfo *source, GIOCondition condition,
 649                                      gpointer data)
 650 {
 651         Session *session = SESSION(data);
 652         GByteArray *data_buf;
 653         gint terminator_len;
 654         gboolean complete = FALSE;
 655         guint data_len;
 656         gint ret;
 657
 658         cm_return_val_if_fail(condition == G_IO_IN, FALSE);
 659
 660         session_set_timeout(session, session->timeout_interval);
 661
 662         if (session->read_buf_len == 0) {
 663                 gint read_len;
 664
 665                 read_len = sock_read(session->sock, session->read_buf,
 666                                      SESSION_BUFFSIZE);
 667
 668                 if (read_len == 0) {
 669                         g_warning("sock_read: received EOF");
 670                         session->state = SESSION_EOF;
 671                         return FALSE;
 672                 }
 673
 674                 if (read_len < 0) {
 675                         switch (errno) {
 676                         case EAGAIN:
 677                                 return TRUE;
 678                         default:
 679                                 g_warning("sock_read: %s", g_strerror(errno));
 680                                 session->state = SESSION_ERROR;
 681                                 return FALSE;
 682                         }
 683                 }
 684
 685                 session->read_buf_len = read_len;
 686         }
 687
 688         data_buf = session->read_data_buf;
 689         terminator_len = strlen(session->read_data_terminator);
 690
 691         if (session->read_buf_len == 0)
 692                 return TRUE;
 693
 694         g_byte_array_append(data_buf, session->read_buf_p,
 695                             session->read_buf_len);
 696
 697         session->read_buf_len = 0;
 698         session->read_buf_p = session->read_buf;
 699
 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)
 704                         complete = TRUE;
 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)
 711                         complete = TRUE;
 712         }
 713
 714         /* incomplete read */
 715         if (!complete) {
 716                 GDateTime *tv_cur = g_date_time_new_now_local();
 717
 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();
 725                 }
 726                 g_date_time_unref(tv_cur);
 727                 return TRUE;
 728         }
 729
 730         /* complete */
 731         if (session->io_tag > 0) {
 732                 g_source_remove(session->io_tag);
 733                 session->io_tag = 0;
 734         }
 735
 736         data_len = data_buf->len - terminator_len;
 737
 738         /* callback */
 739         ret = session->recv_data_finished(session, (gchar *)data_buf->data,
 740                                           data_len);
 741
 742         g_byte_array_set_size(data_buf, 0);
 743
 744         session->recv_data_notify(session, data_len,
 745                                   session->recv_data_notify_data);
 746
 747         if (ret < 0)
 748                 session->state = SESSION_ERROR;
 749
 750         return FALSE;
 751 }
 752
 753 static gint session_write_buf(Session *session)
 754 {
 755         gint write_len;
 756         gint to_write_len;
 757
 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);
 761
 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);
 765
 766         write_len = sock_write(session->sock, session->write_buf_p,
 767                                to_write_len);
 768
 769         if (write_len < 0) {
 770                 switch (errno) {
 771                 case EAGAIN:
 772                         write_len = 0;
 773                         break;
 774                 default:
 775                         g_warning("sock_write: %s", g_strerror(errno));
 776                         session->state = SESSION_ERROR;
 777                         return -1;
 778                 }
 779         }
 780
 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;
 785                 return 1;
 786         }
 787
 788         g_free(session->write_buf);
 789         session->write_buf = NULL;
 790         session->write_buf_p = NULL;
 791         session->write_buf_len = 0;
 792
 793         return 0;
 794 }
 795
 796 static gint session_write_data(Session *session)
 797 {
 798         gint write_len;
 799         gint to_write_len;
 800
 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);
 804
 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);
 808
 809         write_len = sock_write(session->sock, session->write_data_p,
 810                                to_write_len);
 811
 812         if (write_len < 0) {
 813                 switch (errno) {
 814                 case EAGAIN:
 815                         write_len = 0;
 816                         break;
 817                 default:
 818                         g_warning("sock_write: %s", g_strerror(errno));
 819                         session->state = SESSION_ERROR;
 820                         return -1;
 821                 }
 822         }
 823
 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;
 828                 return 1;
 829         }
 830
 831         session->write_data = NULL;
 832         session->write_data_p = NULL;
 833         session->write_data_len = 0;
 834
 835         return 0;
 836 }
 837
 838 static gboolean session_write_msg_cb(SockInfo *source, GIOCondition condition,
 839                                      gpointer data)
 840 {
 841         Session *session = SESSION(data);
 842         gint ret;
 843
 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);
 848
 849         ret = session_write_buf(session);
 850
 851         if (ret < 0) {
 852                 session->state = SESSION_ERROR;
 853                 return FALSE;
 854         } else if (ret > 0)
 855                 return TRUE;
 856
 857         if (session->io_tag > 0) {
 858                 g_source_remove(session->io_tag);
 859                 session->io_tag = 0;
 860         }
 861
 862         session_recv_msg(session);
 863
 864         return FALSE;
 865 }
 866
 867 static gboolean session_write_data_cb(SockInfo *source,
 868                                       GIOCondition condition, gpointer data)
 869 {
 870         Session *session = SESSION(data);
 871         guint write_data_len;
 872         gint ret;
 873
 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);
 878
 879         write_data_len = session->write_data_len;
 880
 881         ret = session_write_data(session);
 882
 883         if (ret < 0) {
 884                 session->state = SESSION_ERROR;
 885                 return FALSE;
 886         } else if (ret > 0) {
 887                 GDateTime *tv_cur = g_date_time_new_now_local();
 888
 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
 893                                 (session,
 894                                  session->write_data_p - session->write_data,
 895                                  write_data_len,
 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();
 899                 }
 900                 g_date_time_unref(tv_cur);
 901                 return TRUE;
 902         }
 903
 904         if (session->io_tag > 0) {
 905                 g_source_remove(session->io_tag);
 906                 session->io_tag = 0;
 907         }
 908
 909         /* callback */
 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);
 913
 914         return FALSE;
 915 }
 916
 917 void session_register_ping(Session *session, gboolean (*ping_cb)(gpointer data))
 918 {
 919         if (!session)
 920                 return;
 921         if (session->ping_tag > -1)
 922                 g_source_remove(session->ping_tag);
 923
 924         session->ping_tag = -1;
 925
 926         if (ping_cb != NULL)
 927                 session->ping_tag = g_timeout_add_seconds(60, ping_cb, session);
 928 }