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 #endif
  73         session->nonblocking = TRUE;
  74         session->state = SESSION_READY;
  75         session->last_access_time = time(NULL);
  76
  77         g_get_current_time(&session->tv_prev);
  78
  79         session->conn_id = 0;
  80
  81         session->io_tag = 0;
  82
  83         session->read_buf_p = session->read_buf;
  84         session->read_buf_len = 0;
  85
  86         session->read_msg_buf = g_string_sized_new(1024);
  87         session->read_data_buf = g_byte_array_new();
  88
  89         session->write_buf = NULL;
  90         session->write_buf_p = NULL;
  91         session->write_buf_len = 0;
  92
  93         session->write_data = NULL;
  94         session->write_data_p = NULL;
  95         session->write_data_len = 0;
  96
  97         session->timeout_tag = 0;
  98         session->timeout_interval = 0;
  99
 100         session->data = NULL;
 101         session->account = prefs_account;
 102         session->is_smtp = is_smtp;
 103
 104         session->ping_tag = -1;
 105 }
 106
 107 /*!
 108  *\brief        Set up parent and child process
 109  *              Childloop: Read commands from parent,
 110  *              send to server, get answer, pass to parent
 111  *
 112  *\param        session Contains session information
 113  *              server to connect to
 114  *              port to connect to
 115  *
 116  *\return        0 : success
 117  *              -1 : pipe / fork errors (parent)
 118  *               1 : connection error (child)
 119  */
 120 gint session_connect(Session *session, const gchar *server, gushort port)
 121 {
 122 #ifdef G_OS_UNIX
 123         session->server = g_strdup(server);
 124         session->port = port;
 125
 126         session->conn_id = sock_connect_async(server, port, session_connect_cb,
 127                                               session);
 128         if (session->conn_id < 0) {
 129                 g_warning("can't connect to server.");
 130                 session_close(session);
 131                 return -1;
 132         }
 133
 134         return 0;
 135 #else
 136         SockInfo *sock;
 137
 138         session->server = g_strdup(server);
 139         session->port = port;
 140
 141         sock = sock_connect(server, port);
 142         if (sock == NULL) {
 143                 g_warning("can't connect to server.");
 144                 session_close(session);
 145                 return -1;
 146         }
 147         sock->is_smtp = session->is_smtp;
 148
 149         return session_connect_cb(sock, session);
 150 #endif
 151 }
 152
 153 static gint session_connect_cb(SockInfo *sock, gpointer data)
 154 {
 155         Session *session = SESSION(data);
 156
 157         session->conn_id = 0;
 158
 159         if (!sock) {
 160                 g_warning("can't connect to server.");
 161                 session->state = SESSION_ERROR;
 162                 return -1;
 163         }
 164
 165         session->sock = sock;
 166         sock->account = session->account;
 167         sock->is_smtp = session->is_smtp;
 168 #ifdef USE_GNUTLS
 169         sock->gnutls_priority = session->gnutls_priority;
 170
 171         if (session->ssl_type == SSL_TUNNEL) {
 172                 sock_set_nonblocking_mode(sock, FALSE);
 173                 if (!ssl_init_socket(sock)) {
 174                         g_warning("can't initialize SSL.");
 175                         log_error(LOG_PROTOCOL, _("SSL handshake failed\n"));
 176                         session->state = SESSION_ERROR;
 177                         return -1;
 178                 }
 179         }
 180 #endif
 181
 182         /* we could have gotten a timeout while waiting for user input in
 183          * an SSL certificate dialog */
 184         if (session->state == SESSION_TIMEOUT)
 185                 return -1;
 186
 187         sock_set_nonblocking_mode(sock, session->nonblocking);
 188
 189         debug_print("session (%p): connected\n", session);
 190
 191         session->state = SESSION_RECV;
 192         session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 193                                          session_read_msg_cb,
 194                                          session);
 195
 196         return 0;
 197 }
 198
 199 /*!
 200  *\brief        child and parent: send DISCONNECT message to other process
 201  *
 202  *\param        session Contains session information
 203  *
 204  *\return        0 : success
 205  */
 206 gint session_disconnect(Session *session)
 207 {
 208         session_close(session);
 209         return 0;
 210 }
 211
 212 /*!
 213  *\brief        parent ?
 214  *
 215  *\param        session Contains session information
 216  */
 217 void session_destroy(Session *session)
 218 {
 219         cm_return_if_fail(session != NULL);
 220         cm_return_if_fail(session->destroy != NULL);
 221
 222         session_register_ping(session, NULL);
 223
 224         session_close(session);
 225         session->destroy(session);
 226         g_free(session->server);
 227         g_string_free(session->read_msg_buf, TRUE);
 228         g_byte_array_free(session->read_data_buf, TRUE);
 229         g_free(session->read_data_terminator);
 230         g_free(session->write_buf);
 231 #ifdef USE_GNUTLS
 232         g_free(session->gnutls_priority);
 233 #endif
 234
 235         debug_print("session (%p): destroyed\n", session);
 236
 237         g_free(session);
 238 }
 239
 240 gboolean session_is_running(Session *session)
 241 {
 242         return (session->state == SESSION_READY ||
 243                 session->state == SESSION_SEND ||
 244                 session->state == SESSION_RECV);
 245 }
 246
 247 gboolean session_is_connected(Session *session)
 248 {
 249         return (session->state == SESSION_SEND ||
 250                 session->state == SESSION_RECV);
 251 }
 252
 253 void session_set_access_time(Session *session)
 254 {
 255         session->last_access_time = time(NULL);
 256 }
 257
 258 void session_set_timeout(Session *session, guint interval)
 259 {
 260         if (session->timeout_tag > 0)
 261                 g_source_remove(session->timeout_tag);
 262
 263         session->timeout_interval = interval;
 264         if (interval > 0) {
 265 #if GLIB_CHECK_VERSION(2,14,0)
 266                 if (interval % 1000 == 0)
 267                         session->timeout_tag =
 268                                 g_timeout_add_seconds(interval/1000, session_timeout_cb, session);
 269                 else
 270 #endif
 271                 session->timeout_tag =
 272                         g_timeout_add(interval, session_timeout_cb, session);
 273         } else
 274                 session->timeout_tag = 0;
 275 }
 276
 277 static gboolean session_timeout_cb(gpointer data)
 278 {
 279         Session *session = SESSION(data);
 280
 281         g_warning("session timeout.\n");
 282
 283         if (session->io_tag > 0) {
 284                 g_source_remove(session->io_tag);
 285                 session->io_tag = 0;
 286         }
 287
 288         session->timeout_tag = 0;
 289         session->state = SESSION_TIMEOUT;
 290
 291         return FALSE;
 292 }
 293
 294 void session_set_recv_message_notify(Session *session,
 295                                      RecvMsgNotify notify_func, gpointer data)
 296 {
 297         session->recv_msg_notify = notify_func;
 298         session->recv_msg_notify_data = data;
 299 }
 300
 301 void session_set_recv_data_progressive_notify
 302                                         (Session *session,
 303                                          RecvDataProgressiveNotify notify_func,
 304                                          gpointer data)
 305 {
 306         session->recv_data_progressive_notify = notify_func,
 307         session->recv_data_progressive_notify_data = data;
 308 }
 309
 310 void session_set_recv_data_notify(Session *session, RecvDataNotify notify_func,
 311                                   gpointer data)
 312 {
 313         session->recv_data_notify = notify_func;
 314         session->recv_data_notify_data = data;
 315 }
 316
 317 void session_set_send_data_progressive_notify
 318                                         (Session *session,
 319                                          SendDataProgressiveNotify notify_func,
 320                                          gpointer data)
 321 {
 322         session->send_data_progressive_notify = notify_func;
 323         session->send_data_progressive_notify_data = data;
 324 }
 325
 326 void session_set_send_data_notify(Session *session, SendDataNotify notify_func,
 327                                   gpointer data)
 328 {
 329         session->send_data_notify = notify_func;
 330         session->send_data_notify_data = data;
 331 }
 332
 333 /*!
 334  *\brief        child and parent cleanup (child closes first)
 335  *
 336  *\param        session Contains session information
 337  *
 338  *\return        0 : success
 339  */
 340 static gint session_close(Session *session)
 341 {
 342         cm_return_val_if_fail(session != NULL, -1);
 343
 344 #ifdef G_OS_UNIX
 345         if (session->conn_id > 0) {
 346                 sock_connect_async_cancel(session->conn_id);
 347                 session->conn_id = 0;
 348                 debug_print("session (%p): connection cancelled\n", session);
 349         }
 350 #endif
 351
 352         session_set_timeout(session, 0);
 353
 354         if (session->io_tag > 0) {
 355                 g_source_remove(session->io_tag);
 356                 session->io_tag = 0;
 357         }
 358
 359         if (session->sock) {
 360                 sock_close(session->sock);
 361                 session->sock = NULL;
 362                 session->state = SESSION_DISCONNECTED;
 363                 debug_print("session (%p): closed\n", session);
 364         }
 365
 366         return 0;
 367 }
 368
 369 #ifdef USE_GNUTLS
 370 gint session_start_tls(Session *session)
 371 {
 372         gboolean nb_mode;
 373
 374         nb_mode = sock_is_nonblocking_mode(session->sock);
 375
 376         if (nb_mode)
 377                 sock_set_nonblocking_mode(session->sock, FALSE);
 378
 379         if (!ssl_init_socket_with_method(session->sock, SSL_METHOD_TLSv1)) {
 380                 g_warning("couldn't start TLS session.\n");
 381                 if (nb_mode)
 382                         sock_set_nonblocking_mode(session->sock, session->nonblocking);
 383                 return -1;
 384         }
 385
 386         if (nb_mode)
 387                 sock_set_nonblocking_mode(session->sock, session->nonblocking);
 388
 389         return 0;
 390 }
 391 #endif
 392
 393 gint session_send_msg(Session *session, SessionMsgType type, const gchar *msg)
 394 {
 395         gboolean ret;
 396
 397         cm_return_val_if_fail(session->write_buf == NULL, -1);
 398         cm_return_val_if_fail(msg != NULL, -1);
 399         cm_return_val_if_fail(msg[0] != '\0', -1);
 400
 401         session->state = SESSION_SEND;
 402         session->write_buf = g_strconcat(msg, "\r\n", NULL);
 403         session->write_buf_p = session->write_buf;
 404         session->write_buf_len = strlen(msg) + 2;
 405
 406         ret = session_write_msg_cb(session->sock, G_IO_OUT, session);
 407
 408         if (ret == TRUE)
 409                 session->io_tag = sock_add_watch(session->sock, G_IO_OUT,
 410                                                  session_write_msg_cb, session);
 411         else if (session->state == SESSION_ERROR)
 412                 return -1;
 413
 414         return 0;
 415 }
 416
 417 gint session_recv_msg(Session *session)
 418 {
 419         cm_return_val_if_fail(session->read_msg_buf->len == 0, -1);
 420
 421         session->state = SESSION_RECV;
 422
 423         if (session->read_buf_len > 0)
 424                 g_idle_add(session_recv_msg_idle_cb, session);
 425         else
 426                 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 427                                                  session_read_msg_cb, session);
 428
 429         return 0;
 430 }
 431
 432 static gboolean session_recv_msg_idle_cb(gpointer data)
 433 {
 434         Session *session = SESSION(data);
 435         gboolean ret;
 436
 437         ret = session_read_msg_cb(session->sock, G_IO_IN, session);
 438
 439         if (ret == TRUE)
 440                 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 441                                                  session_read_msg_cb, session);
 442
 443         return FALSE;
 444 }
 445
 446 /*!
 447  *\brief        parent (child?): send data to other process
 448  *
 449  *\param        session Contains session information
 450  *              data Data to send
 451  *              size Bytes to send
 452  *
 453  *\return        0 : success
 454  *              -1 : error
 455  */
 456 gint session_send_data(Session *session, const guchar *data, guint size)
 457 {
 458         gboolean ret;
 459
 460         cm_return_val_if_fail(session->write_data == NULL, -1);
 461         cm_return_val_if_fail(data != NULL, -1);
 462         cm_return_val_if_fail(size != 0, -1);
 463
 464         session->state = SESSION_SEND;
 465
 466         session->write_data = data;
 467         session->write_data_p = session->write_data;
 468         session->write_data_len = size;
 469         g_get_current_time(&session->tv_prev);
 470
 471         ret = session_write_data_cb(session->sock, G_IO_OUT, session);
 472
 473         if (ret == TRUE)
 474                 session->io_tag = sock_add_watch(session->sock, G_IO_OUT,
 475                                                  session_write_data_cb,
 476                                                  session);
 477         else if (session->state == SESSION_ERROR)
 478                 return -1;
 479
 480         return 0;
 481 }
 482
 483 gint session_recv_data(Session *session, guint size, const gchar *terminator)
 484 {
 485         cm_return_val_if_fail(session->read_data_buf->len == 0, -1);
 486
 487         session->state = SESSION_RECV;
 488
 489         g_free(session->read_data_terminator);
 490         session->read_data_terminator = g_strdup(terminator);
 491         g_get_current_time(&session->tv_prev);
 492
 493         if (session->read_buf_len > 0)
 494                 g_idle_add(session_recv_data_idle_cb, session);
 495         else
 496                 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 497                                                  session_read_data_cb, session);
 498
 499         return 0;
 500 }
 501
 502 static gboolean session_recv_data_idle_cb(gpointer data)
 503 {
 504         Session *session = SESSION(data);
 505         gboolean ret;
 506
 507         ret = session_read_data_cb(session->sock, G_IO_IN, session);
 508
 509         if (ret == TRUE)
 510                 session->io_tag = sock_add_watch(session->sock, G_IO_IN,
 511                                                  session_read_data_cb, session);
 512
 513         return FALSE;
 514 }
 515
 516 static gboolean session_read_msg_cb(SockInfo *source, GIOCondition condition,
 517                                     gpointer data)
 518 {
 519         Session *session = SESSION(data);
 520         gchar buf[SESSION_BUFFSIZE];
 521         gint line_len;
 522         gchar *newline;
 523         gchar *msg;
 524         gint ret;
 525
 526         cm_return_val_if_fail(condition == G_IO_IN, FALSE);
 527
 528         session_set_timeout(session, session->timeout_interval);
 529
 530         if (session->read_buf_len == 0) {
 531                 gint read_len;
 532
 533                 read_len = sock_read(session->sock, session->read_buf,
 534                                      SESSION_BUFFSIZE - 1);
 535
 536                 if (read_len == -1 && session->state == SESSION_DISCONNECTED) {
 537                         g_warning ("sock_read: session disconnected\n");
 538                         if (session->io_tag > 0) {
 539                                 g_source_remove(session->io_tag);
 540                                 session->io_tag = 0;
 541                         }
 542                         return FALSE;
 543                 }
 544
 545                 if (read_len == 0) {
 546                         g_warning("sock_read: received EOF\n");
 547                         session->state = SESSION_EOF;
 548                         return FALSE;
 549                 }
 550
 551                 if (read_len < 0) {
 552                         switch (errno) {
 553                         case EAGAIN:
 554                                 return TRUE;
 555                         default:
 556                                 g_warning("sock_read: %s\n", g_strerror(errno));
 557                                 session->state = SESSION_ERROR;
 558                                 return FALSE;
 559                         }
 560                 }
 561
 562                 session->read_buf_len = read_len;
 563         }
 564
 565         if ((newline = memchr(session->read_buf_p, '\n', session->read_buf_len))
 566                 != NULL)
 567                 line_len = newline - session->read_buf_p + 1;
 568         else
 569                 line_len = session->read_buf_len;
 570
 571         if (line_len == 0)
 572                 return TRUE;
 573
 574         memcpy(buf, session->read_buf_p, line_len);
 575         buf[line_len] = '\0';
 576
 577         g_string_append(session->read_msg_buf, buf);
 578
 579         session->read_buf_len -= line_len;
 580         if (session->read_buf_len == 0)
 581                 session->read_buf_p = session->read_buf;
 582         else
 583                 session->read_buf_p += line_len;
 584
 585         /* incomplete read */
 586         if (buf[line_len - 1] != '\n')
 587                 return TRUE;
 588
 589         /* complete */
 590         if (session->io_tag > 0) {
 591                 g_source_remove(session->io_tag);
 592                 session->io_tag = 0;
 593         }
 594
 595         /* callback */
 596         msg = g_strdup(session->read_msg_buf->str);
 597         strretchomp(msg);
 598         g_string_truncate(session->read_msg_buf, 0);
 599
 600         ret = session->recv_msg(session, msg);
 601         session->recv_msg_notify(session, msg, session->recv_msg_notify_data);
 602
 603         g_free(msg);
 604
 605         if (ret < 0)
 606                 session->state = SESSION_ERROR;
 607
 608         return FALSE;
 609 }
 610
 611 static gboolean session_read_data_cb(SockInfo *source, GIOCondition condition,
 612                                      gpointer data)
 613 {
 614         Session *session = SESSION(data);
 615         GByteArray *data_buf;
 616         gint terminator_len;
 617         gboolean complete = FALSE;
 618         guint data_len;
 619         gint ret;
 620
 621         cm_return_val_if_fail(condition == G_IO_IN, FALSE);
 622
 623         session_set_timeout(session, session->timeout_interval);
 624
 625         if (session->read_buf_len == 0) {
 626                 gint read_len;
 627
 628                 read_len = sock_read(session->sock, session->read_buf,
 629                                      SESSION_BUFFSIZE);
 630
 631                 if (read_len == 0) {
 632                         g_warning("sock_read: received EOF\n");
 633                         session->state = SESSION_EOF;
 634                         return FALSE;
 635                 }
 636
 637                 if (read_len < 0) {
 638                         switch (errno) {
 639                         case EAGAIN:
 640                                 return TRUE;
 641                         default:
 642                                 g_warning("sock_read: %s\n", g_strerror(errno));
 643                                 session->state = SESSION_ERROR;
 644                                 return FALSE;
 645                         }
 646                 }
 647
 648                 session->read_buf_len = read_len;
 649         }
 650
 651         data_buf = session->read_data_buf;
 652         terminator_len = strlen(session->read_data_terminator);
 653
 654         if (session->read_buf_len == 0)
 655                 return TRUE;
 656
 657         g_byte_array_append(data_buf, session->read_buf_p,
 658                             session->read_buf_len);
 659
 660         session->read_buf_len = 0;
 661         session->read_buf_p = session->read_buf;
 662
 663         /* check if data is terminated */
 664         if (data_buf->len >= terminator_len) {
 665                 if (memcmp(data_buf->data, session->read_data_terminator,
 666                            terminator_len) == 0)
 667                         complete = TRUE;
 668                 else if (data_buf->len >= terminator_len + 2 &&
 669                          memcmp(data_buf->data + data_buf->len -
 670                                 (terminator_len + 2), "\r\n", 2) == 0 &&
 671                          memcmp(data_buf->data + data_buf->len -
 672                                 terminator_len, session->read_data_terminator,
 673                                 terminator_len) == 0)
 674                         complete = TRUE;
 675         }
 676
 677         /* incomplete read */
 678         if (!complete) {
 679                 GTimeVal tv_cur;
 680
 681                 g_get_current_time(&tv_cur);
 682                 if (tv_cur.tv_sec - session->tv_prev.tv_sec > 0 ||
 683                     tv_cur.tv_usec - session->tv_prev.tv_usec >
 684                     UI_REFRESH_INTERVAL) {
 685                         session->recv_data_progressive_notify
 686                                 (session, data_buf->len, 0,
 687                                  session->recv_data_progressive_notify_data);
 688                         g_get_current_time(&session->tv_prev);
 689                 }
 690                 return TRUE;
 691         }
 692
 693         /* complete */
 694         if (session->io_tag > 0) {
 695                 g_source_remove(session->io_tag);
 696                 session->io_tag = 0;
 697         }
 698
 699         data_len = data_buf->len - terminator_len;
 700
 701         /* callback */
 702         ret = session->recv_data_finished(session, (gchar *)data_buf->data,
 703                                           data_len);
 704
 705         g_byte_array_set_size(data_buf, 0);
 706
 707         session->recv_data_notify(session, data_len,
 708                                   session->recv_data_notify_data);
 709
 710         if (ret < 0)
 711                 session->state = SESSION_ERROR;
 712
 713         return FALSE;
 714 }
 715
 716 static gint session_write_buf(Session *session)
 717 {
 718         gint write_len;
 719         gint to_write_len;
 720
 721         cm_return_val_if_fail(session->write_buf != NULL, -1);
 722         cm_return_val_if_fail(session->write_buf_p != NULL, -1);
 723         cm_return_val_if_fail(session->write_buf_len > 0, -1);
 724
 725         to_write_len = session->write_buf_len -
 726                 (session->write_buf_p - session->write_buf);
 727         to_write_len = MIN(to_write_len, SESSION_BUFFSIZE);
 728
 729         write_len = sock_write(session->sock, session->write_buf_p,
 730                                to_write_len);
 731
 732         if (write_len < 0) {
 733                 switch (errno) {
 734                 case EAGAIN:
 735                         write_len = 0;
 736                         break;
 737                 default:
 738                         g_warning("sock_write: %s\n", g_strerror(errno));
 739                         session->state = SESSION_ERROR;
 740                         return -1;
 741                 }
 742         }
 743
 744         /* incomplete write */
 745         if (session->write_buf_p - session->write_buf + write_len <
 746             session->write_buf_len) {
 747                 session->write_buf_p += write_len;
 748                 return 1;
 749         }
 750
 751         g_free(session->write_buf);
 752         session->write_buf = NULL;
 753         session->write_buf_p = NULL;
 754         session->write_buf_len = 0;
 755
 756         return 0;
 757 }
 758
 759 static gint session_write_data(Session *session)
 760 {
 761         gint write_len;
 762         gint to_write_len;
 763
 764         cm_return_val_if_fail(session->write_data != NULL, -1);
 765         cm_return_val_if_fail(session->write_data_p != NULL, -1);
 766         cm_return_val_if_fail(session->write_data_len > 0, -1);
 767
 768         to_write_len = session->write_data_len -
 769                 (session->write_data_p - session->write_data);
 770         to_write_len = MIN(to_write_len, SESSION_BUFFSIZE);
 771
 772         write_len = sock_write(session->sock, session->write_data_p,
 773                                to_write_len);
 774
 775         if (write_len < 0) {
 776                 switch (errno) {
 777                 case EAGAIN:
 778                         write_len = 0;
 779                         break;
 780                 default:
 781                         g_warning("sock_write: %s\n", g_strerror(errno));
 782                         session->state = SESSION_ERROR;
 783                         return -1;
 784                 }
 785         }
 786
 787         /* incomplete write */
 788         if (session->write_data_p - session->write_data + write_len <
 789             session->write_data_len) {
 790                 session->write_data_p += write_len;
 791                 return 1;
 792         }
 793
 794         session->write_data = NULL;
 795         session->write_data_p = NULL;
 796         session->write_data_len = 0;
 797
 798         return 0;
 799 }
 800
 801 static gboolean session_write_msg_cb(SockInfo *source, GIOCondition condition,
 802                                      gpointer data)
 803 {
 804         Session *session = SESSION(data);
 805         gint ret;
 806
 807         cm_return_val_if_fail(condition == G_IO_OUT, FALSE);
 808         cm_return_val_if_fail(session->write_buf != NULL, FALSE);
 809         cm_return_val_if_fail(session->write_buf_p != NULL, FALSE);
 810         cm_return_val_if_fail(session->write_buf_len > 0, FALSE);
 811
 812         ret = session_write_buf(session);
 813
 814         if (ret < 0) {
 815                 session->state = SESSION_ERROR;
 816                 return FALSE;
 817         } else if (ret > 0)
 818                 return TRUE;
 819
 820         if (session->io_tag > 0) {
 821                 g_source_remove(session->io_tag);
 822                 session->io_tag = 0;
 823         }
 824
 825         session_recv_msg(session);
 826
 827         return FALSE;
 828 }
 829
 830 static gboolean session_write_data_cb(SockInfo *source,
 831                                       GIOCondition condition, gpointer data)
 832 {
 833         Session *session = SESSION(data);
 834         guint write_data_len;
 835         gint ret;
 836
 837         cm_return_val_if_fail(condition == G_IO_OUT, FALSE);
 838         cm_return_val_if_fail(session->write_data != NULL, FALSE);
 839         cm_return_val_if_fail(session->write_data_p != NULL, FALSE);
 840         cm_return_val_if_fail(session->write_data_len > 0, FALSE);
 841
 842         write_data_len = session->write_data_len;
 843
 844         ret = session_write_data(session);
 845
 846         if (ret < 0) {
 847                 session->state = SESSION_ERROR;
 848                 return FALSE;
 849         } else if (ret > 0) {
 850                 GTimeVal tv_cur;
 851
 852                 g_get_current_time(&tv_cur);
 853                 if (tv_cur.tv_sec - session->tv_prev.tv_sec > 0 ||
 854                     tv_cur.tv_usec - session->tv_prev.tv_usec >
 855                     UI_REFRESH_INTERVAL) {
 856                         session_set_timeout(session, session->timeout_interval);
 857                         session->send_data_progressive_notify
 858                                 (session,
 859                                  session->write_data_p - session->write_data,
 860                                  write_data_len,
 861                                  session->send_data_progressive_notify_data);
 862                         g_get_current_time(&session->tv_prev);
 863                 }
 864                 return TRUE;
 865         }
 866
 867         if (session->io_tag > 0) {
 868                 g_source_remove(session->io_tag);
 869                 session->io_tag = 0;
 870         }
 871
 872         /* callback */
 873         ret = session->send_data_finished(session, write_data_len);
 874         session->send_data_notify(session, write_data_len,
 875                                   session->send_data_notify_data);
 876
 877         return FALSE;
 878 }
 879
 880 void session_register_ping(Session *session, gboolean (*ping_cb)(gpointer data))
 881 {
 882         if (!session)
 883                 return;
 884         if (session->ping_tag > -1)
 885                 g_source_remove(session->ping_tag);
 886
 887         session->ping_tag = -1;
 888
 889         if (ping_cb != NULL)
 890 #if GLIB_CHECK_VERSION(2,14,0)
 891                 session->ping_tag =
 892                         g_timeout_add_seconds(60, ping_cb, session);
 893 #else
 894                 session->ping_tag =
 895                         g_timeout_add(60*1000, ping_cb, session);
 896 #endif
 897 }