lib/silcutil/unix/silcunixnet.c

   1 /*
   2
   3   silcunixnet.c
   4
   5   Author: Pekka Riikonen <priikone@silcnet.org>
   6
   7   Copyright (C) 1997 - 2007 Pekka Riikonen
   8
   9   This program is free software; you can redistribute it and/or modify
  10   it under the terms of the GNU General Public License as published by
  11   the Free Software Foundation; version 2 of the License.
  12
  13   This program is distributed in the hope that it will be useful,
  14   but WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16   GNU General Public License for more details.
  17
  18 */
  19 /* $Id$ */
  20
  21 #include "silc.h"
  22 #include "silcnet.h"
  23
  24 /************************** Types and definitions ***************************/
  25
  26 #ifdef HAVE_IPV6
  27 #define SIZEOF_SOCKADDR(so) ((so).sa.sa_family == AF_INET6 ?    \
  28   sizeof(so.sin6) : sizeof(so.sin))
  29 #else
  30 #define SIZEOF_SOCKADDR(so) (sizeof(so.sin))
  31 #endif
  32
  33 typedef union {
  34   struct sockaddr sa;
  35   struct sockaddr_in sin;
  36 #ifdef HAVE_IPV6
  37   struct sockaddr_in6 sin6;
  38 #endif
  39 } SilcSockaddr;
  40
  41 /************************ Static utility functions **************************/
  42
  43 static SilcBool silc_net_set_sockaddr(SilcSockaddr *addr, const char *ip_addr,
  44                                       int port)
  45 {
  46   int len;
  47
  48   memset(addr, 0, sizeof(*addr));
  49
  50   /* Check for IPv4 and IPv6 addresses */
  51   if (ip_addr) {
  52     if (!silc_net_is_ip(ip_addr)) {
  53       SILC_LOG_ERROR(("%s is not IP address", ip_addr));
  54       silc_set_errno_reason(SILC_ERR_BAD_IP, "%s is not an IP address",
  55                             ip_addr);
  56       return FALSE;
  57     }
  58
  59     if (silc_net_is_ip4(ip_addr)) {
  60       /* IPv4 address */
  61       len = sizeof(addr->sin.sin_addr);
  62       if (!silc_net_addr2bin(ip_addr,
  63                              (unsigned char *)&addr->sin.sin_addr.s_addr, len))
  64         return FALSE;
  65       addr->sin.sin_family = AF_INET;
  66       addr->sin.sin_port = port ? htons(port) : 0;
  67     } else {
  68 #ifdef HAVE_IPV6
  69       /* IPv6 address */
  70       len = sizeof(addr->sin6.sin6_addr);
  71       if (!silc_net_addr2bin(ip_addr,
  72                              (unsigned char *)&addr->sin6.sin6_addr, len))
  73         return FALSE;
  74       addr->sin6.sin6_family = AF_INET6;
  75       addr->sin6.sin6_port = port ? htons(port) : 0;
  76 #else
  77       SILC_LOG_ERROR(("IPv6 support is not compiled in"));
  78       silc_set_errno(SILC_ERR_NOT_SUPPORTED);
  79       return FALSE;
  80 #endif
  81     }
  82   } else {
  83     /* Any address */
  84     addr->sin.sin_family = AF_INET;
  85     addr->sin.sin_addr.s_addr = INADDR_ANY;
  86     if (port)
  87       addr->sin.sin_port = htons(port);
  88   }
  89
  90   return TRUE;
  91 }
  92
  93 /****************************** TCP Listener ********************************/
  94
  95 /* Deliver new stream to upper layer */
  96
  97 static void silc_net_accept_stream(SilcResult status,
  98                                    SilcStream stream, void *context)
  99 {
 100   SilcNetListener listener = context;
 101
 102   if (status != SILC_OK)
 103     return;
 104
 105   listener->callback(SILC_OK, stream, listener->context);
 106 }
 107
 108 /* Accept incoming connection and notify upper layer */
 109
 110 SILC_TASK_CALLBACK(silc_net_accept)
 111 {
 112   SilcNetListener listener = context;
 113   int sock;
 114
 115   SILC_LOG_DEBUG(("Accepting new connection"));
 116
 117   sock = silc_net_accept_connection(fd);
 118   if (sock < 0)
 119     return;
 120
 121   /* Set socket options */
 122   silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEADDR, 1);
 123
 124   /* Create socket stream */
 125   silc_socket_tcp_stream_create(sock, listener->lookup,
 126                                 listener->require_fqdn, schedule,
 127                                 silc_net_accept_stream, listener);
 128 }
 129
 130 /* Create TCP network listener */
 131
 132 SilcNetListener
 133 silc_net_tcp_create_listener(const char **local_ip_addr,
 134                              SilcUInt32 local_ip_count, int port,
 135                              SilcBool lookup, SilcBool require_fqdn,
 136                              SilcSchedule schedule,
 137                              SilcNetCallback callback, void *context)
 138 {
 139   SilcNetListener listener = NULL;
 140   SilcSockaddr server;
 141   int i, sock, rval;
 142   const char *ipany = "0.0.0.0";
 143
 144   SILC_LOG_DEBUG(("Creating TCP listener"));
 145
 146   if (port < 0 || !schedule || !callback) {
 147     silc_set_errno(SILC_ERR_INVALID_ARGUMENT);
 148     goto err;
 149   }
 150
 151   listener = silc_calloc(1, sizeof(*listener));
 152   if (!listener)
 153     return NULL;
 154   listener->schedule = schedule;
 155   listener->callback = callback;
 156   listener->context = context;
 157   listener->require_fqdn = require_fqdn;
 158   listener->lookup = lookup;
 159
 160   if (local_ip_count > 0) {
 161     listener->socks = silc_calloc(local_ip_count, sizeof(*listener->socks));
 162     if (!listener->socks)
 163       return NULL;
 164   } else {
 165     listener->socks = silc_calloc(1, sizeof(*listener->socks));
 166     if (!listener->socks)
 167       return NULL;
 168
 169     local_ip_count = 1;
 170   }
 171
 172   /* Bind to local addresses */
 173   for (i = 0; i < local_ip_count; i++) {
 174     SILC_LOG_DEBUG(("Binding to local address %s:%d",
 175                     local_ip_addr ? local_ip_addr[i] : ipany, port));
 176
 177     /* Set sockaddr for server */
 178     if (!silc_net_set_sockaddr(&server,
 179                                local_ip_addr ? local_ip_addr[i] : ipany,
 180                                port))
 181       goto err;
 182
 183     /* Create the socket */
 184     sock = socket(server.sin.sin_family, SOCK_STREAM, 0);
 185     if (sock < 0) {
 186       silc_set_errno_posix(errno);
 187       SILC_LOG_ERROR(("Cannot create socket: %s",
 188                       silc_errno_string(silc_errno)));
 189       goto err;
 190     }
 191
 192     /* Set the socket options */
 193     rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEADDR, 1);
 194     if (rval < 0) {
 195       SILC_LOG_ERROR(("Cannot set socket options: %s", strerror(errno)));
 196       close(sock);
 197       goto err;
 198     }
 199
 200     /* Bind the listener socket */
 201     rval = bind(sock, &server.sa, SIZEOF_SOCKADDR(server));
 202     if (rval < 0) {
 203       silc_set_errno_posix(errno);
 204       SILC_LOG_ERROR(("Cannot bind socket: %s",
 205                       silc_errno_string(silc_errno)));
 206       close(sock);
 207       goto err;
 208     }
 209
 210     /* Specify that we are listenning */
 211     rval = listen(sock, 64);
 212     if (rval < 0) {
 213       silc_set_errno_posix(errno);
 214       SILC_LOG_ERROR(("Cannot set socket listenning: %s",
 215                       silc_errno_string(silc_errno)));
 216       close(sock);
 217       goto err;
 218     }
 219
 220     /* Set the server socket to non-blocking mode */
 221     silc_net_set_socket_nonblock(sock);
 222
 223     /* Schedule for incoming connections */
 224     silc_schedule_task_add_fd(schedule, sock, silc_net_accept, listener);
 225
 226     SILC_LOG_DEBUG(("TCP listener created, fd=%d", sock));
 227     listener->socks[i] = sock;
 228     listener->socks_count++;
 229   }
 230
 231   return listener;
 232
 233  err:
 234   if (listener)
 235     silc_net_close_listener(listener);
 236   return NULL;
 237 }
 238
 239 /* Create TCP listener, multiple ports */
 240
 241 SilcNetListener
 242 silc_net_tcp_create_listener2(const char *local_ip_addr, int *ports,
 243                               SilcUInt32 port_count,
 244                               SilcBool ignore_port_error,
 245                               SilcBool lookup, SilcBool require_fqdn,
 246                               SilcSchedule schedule,
 247                               SilcNetCallback callback, void *context)
 248 {
 249   SilcNetListener listener = NULL;
 250   SilcSockaddr server;
 251   int i, sock, rval;
 252   const char *ipany = "0.0.0.0";
 253
 254   SILC_LOG_DEBUG(("Creating TCP listener"));
 255
 256   if (!schedule || !callback) {
 257     silc_set_errno(SILC_ERR_INVALID_ARGUMENT);
 258     goto err;
 259   }
 260
 261   listener = silc_calloc(1, sizeof(*listener));
 262   if (!listener)
 263     return NULL;
 264   listener->schedule = schedule;
 265   listener->callback = callback;
 266   listener->context = context;
 267   listener->require_fqdn = require_fqdn;
 268   listener->lookup = lookup;
 269
 270   if (port_count > 0) {
 271     listener->socks = silc_calloc(port_count, sizeof(*listener->socks));
 272     if (!listener->socks)
 273       return NULL;
 274   } else {
 275     listener->socks = silc_calloc(1, sizeof(*listener->socks));
 276     if (!listener->socks)
 277       return NULL;
 278
 279     port_count = 1;
 280   }
 281
 282   /* Bind to ports */
 283   for (i = 0; i < port_count; i++) {
 284     SILC_LOG_DEBUG(("Binding to local address %s:%d",
 285                     local_ip_addr ? local_ip_addr : ipany,
 286                     ports ? ports[i] : 0));
 287
 288     /* Set sockaddr for server */
 289     if (!silc_net_set_sockaddr(&server,
 290                                local_ip_addr ? local_ip_addr : ipany,
 291                                ports ? ports[i] : 0)) {
 292       if (ignore_port_error)
 293         continue;
 294       goto err;
 295     }
 296
 297     /* Create the socket */
 298     sock = socket(server.sin.sin_family, SOCK_STREAM, 0);
 299     if (sock < 0) {
 300       if (ignore_port_error)
 301         continue;
 302       silc_set_errno_posix(errno);
 303       SILC_LOG_ERROR(("Cannot create socket: %s",
 304                       silc_errno_string(silc_errno)));
 305       goto err;
 306     }
 307
 308     /* Set the socket options */
 309     rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEADDR, 1);
 310     if (rval < 0) {
 311       close(sock);
 312       if (ignore_port_error)
 313         continue;
 314       SILC_LOG_ERROR(("Cannot set socket options: %s",
 315                       silc_errno_string(silc_errno)));
 316       goto err;
 317     }
 318
 319     /* Bind the listener socket */
 320     rval = bind(sock, &server.sa, SIZEOF_SOCKADDR(server));
 321     if (rval < 0) {
 322       close(sock);
 323       if (ignore_port_error)
 324         continue;
 325       silc_set_errno_posix(errno);
 326       SILC_LOG_ERROR(("Cannot bind socket: %s",
 327                       silc_errno_string(silc_errno)));
 328       goto err;
 329     }
 330
 331     /* Specify that we are listenning */
 332     rval = listen(sock, 64);
 333     if (rval < 0) {
 334       close(sock);
 335       if (ignore_port_error)
 336         continue;
 337       silc_set_errno_posix(errno);
 338       SILC_LOG_ERROR(("Cannot set socket listenning: %s",
 339                       silc_errno_string(silc_errno)));
 340       goto err;
 341     }
 342
 343     /* Set the server socket to non-blocking mode */
 344     silc_net_set_socket_nonblock(sock);
 345
 346     /* Schedule for incoming connections */
 347     silc_schedule_task_add_fd(schedule, sock, silc_net_accept, listener);
 348
 349     SILC_LOG_DEBUG(("TCP listener created, fd=%d", sock));
 350     listener->socks[i] = sock;
 351     listener->socks_count++;
 352   }
 353
 354   if (ignore_port_error && !listener->socks_count)
 355     goto err;
 356
 357   return listener;
 358
 359  err:
 360   if (listener)
 361     silc_net_close_listener(listener);
 362   return NULL;
 363 }
 364
 365 /* Close network listener */
 366
 367 void silc_net_close_listener(SilcNetListener listener)
 368 {
 369   int i;
 370
 371   SILC_LOG_DEBUG(("Closing network listener"));
 372
 373   for (i = 0; i < listener->socks_count; i++) {
 374     silc_schedule_task_del_by_fd(listener->schedule, listener->socks[i]);
 375     shutdown(listener->socks[i], 2);
 376     close(listener->socks[i]);
 377   }
 378
 379   silc_free(listener->socks);
 380   silc_free(listener);
 381 }
 382
 383 /******************************* UDP Stream *********************************/
 384
 385 /* Create UDP stream */
 386
 387 SilcStream
 388 silc_net_udp_connect(const char *local_ip_addr, int local_port,
 389                      const char *remote_ip_addr, int remote_port,
 390                      SilcSchedule schedule)
 391 {
 392   SilcStream stream;
 393   SilcSockaddr server;
 394   int sock = -1, rval;
 395   const char *ipany = "0.0.0.0";
 396
 397   SILC_LOG_DEBUG(("Creating UDP stream"));
 398
 399   if (!schedule) {
 400     silc_set_errno(SILC_ERR_INVALID_ARGUMENT);
 401     goto err;
 402   }
 403
 404   /* Bind to local addresses */
 405   SILC_LOG_DEBUG(("Binding to local address %s",
 406                   local_ip_addr ? local_ip_addr : ipany));
 407
 408   /* Set sockaddr for server */
 409   if (!silc_net_set_sockaddr(&server, local_ip_addr ? local_ip_addr : ipany,
 410                              local_port))
 411     goto err;
 412
 413   /* Create the socket */
 414   sock = socket(server.sin.sin_family, SOCK_DGRAM, 0);
 415   if (sock < 0) {
 416     silc_set_errno_posix(errno);
 417     SILC_LOG_ERROR(("Cannot create socket: %s",
 418                     silc_errno_string(silc_errno)));
 419     goto err;
 420   }
 421
 422   /* Set the socket options */
 423   rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEADDR, 1);
 424   if (rval < 0) {
 425     SILC_LOG_ERROR(("Cannot set socket options: %s",
 426                     silc_errno_string(silc_errno)));
 427     goto err;
 428   }
 429 #ifdef SO_REUSEPORT
 430   rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEPORT, 1);
 431   if (rval < 0) {
 432     SILC_LOG_ERROR(("Cannot set socket options: %s",
 433                     silc_errno_string(silc_errno)));
 434     goto err;
 435   }
 436 #endif /* SO_REUSEPORT */
 437
 438   /* Bind the listener socket */
 439   rval = bind(sock, &server.sa, SIZEOF_SOCKADDR(server));
 440   if (rval < 0) {
 441     silc_set_errno_posix(errno);
 442     SILC_LOG_DEBUG(("Cannot bind socket: %s", silc_errno_string(silc_errno)));
 443     goto err;
 444   }
 445
 446   /* Set to connected state if remote address is provided. */
 447   if (remote_ip_addr && remote_port) {
 448     if (!silc_net_set_sockaddr(&server, remote_ip_addr, remote_port))
 449       goto err;
 450
 451     rval = connect(sock, &server.sa, SIZEOF_SOCKADDR(server));
 452     if (rval < 0) {
 453       silc_set_errno_posix(errno);
 454       SILC_LOG_DEBUG(("Cannot connect UDP stream: %s",
 455                       silc_errno_string(silc_errno)));
 456       goto err;
 457     }
 458   }
 459
 460   /* Set send and receive buffer size */
 461 #ifdef SO_SNDBUF
 462   rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_SNDBUF, 765535);
 463   if (rval < 0) {
 464     rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_SNDBUF, 65535);
 465     if (rval < 0) {
 466       SILC_LOG_ERROR(("Cannot set socket options: %s", strerror(errno)));
 467       goto err;
 468     }
 469   }
 470 #endif /* SO_SNDBUF */
 471 #ifdef SO_RCVBUF
 472   rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_RCVBUF, 765535);
 473   if (rval < 0) {
 474     rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_RCVBUF, 65535);
 475     if (rval < 0) {
 476       SILC_LOG_ERROR(("Cannot set socket options: %s", strerror(errno)));
 477       goto err;
 478     }
 479   }
 480 #endif /* SO_RCVBUF */
 481
 482   /* Encapsulate into socket stream */
 483   stream =
 484     silc_socket_udp_stream_create(sock, local_ip_addr ?
 485                                   silc_net_is_ip6(local_ip_addr) : FALSE,
 486                                   remote_ip_addr ? TRUE : FALSE, schedule);
 487   if (!stream)
 488     goto err;
 489
 490   SILC_LOG_DEBUG(("UDP stream created, fd=%d", sock));
 491   return stream;
 492
 493  err:
 494   if (sock != -1)
 495     close(sock);
 496   return NULL;
 497 }
 498
 499 /* Receive UDP packet */
 500
 501 int silc_net_udp_receive(SilcStream stream, char *remote_ip_addr,
 502                          SilcUInt32 remote_ip_addr_size, int *remote_port,
 503                          unsigned char *ret_data, SilcUInt32 data_size)
 504 {
 505   SilcSocketStream sock = stream;
 506   SilcSockaddr s;
 507   struct sockaddr *from;
 508   int len, flen;
 509
 510   SILC_LOG_DEBUG(("Reading data from UDP socket %d", sock->sock));
 511
 512   if (remote_ip_addr && remote_port) {
 513     if (sock->ipv6) {
 514 #ifdef HAVE_IPV6
 515       from = (struct sockaddr *)&s.sin6;
 516       flen = sizeof(s.sin6);
 517 #endif /* HAVE_IPV6 */
 518     } else {
 519       from = (struct sockaddr *)&s.sin;
 520       flen = sizeof(s.sin);
 521     }
 522     len = recvfrom(sock->sock, ret_data, data_size, 0, from, &flen);
 523   } else
 524     len = recv(sock->sock, ret_data, data_size, 0);
 525
 526   if (len < 0) {
 527     silc_set_errno_posix(errno);
 528     if (errno == EAGAIN || errno == EINTR) {
 529       SILC_LOG_DEBUG(("Could not read immediately, will do it later"));
 530       silc_schedule_set_listen_fd(sock->schedule, sock->sock,
 531                                   SILC_TASK_READ, FALSE);
 532       return -1;
 533     }
 534     SILC_LOG_DEBUG(("Cannot read from UDP socket: %d:%s",
 535                     sock->sock, strerror(errno)));
 536     silc_schedule_unset_listen_fd(sock->schedule, sock->sock);
 537     return -2;
 538   }
 539
 540   SILC_LOG_DEBUG(("Read %d bytes", len));
 541
 542   if (!len)
 543     silc_schedule_unset_listen_fd(sock->schedule, sock->sock);
 544
 545   /* Return remote address */
 546   if (remote_ip_addr && remote_port) {
 547     if (sock->ipv6) {
 548 #ifdef HAVE_IPV6
 549       *remote_port = ntohs(s.sin6.sin6_port);
 550       inet_ntop(AF_INET6, &s.sin6.sin6_addr, remote_ip_addr,
 551                 remote_ip_addr_size);
 552 #else
 553       *remote_port = 0;
 554 #endif /* HAVE_IPV6 */
 555     } else {
 556       *remote_port = ntohs(s.sin.sin_port);
 557       inet_ntop(AF_INET, &s.sin.sin_addr, remote_ip_addr,
 558                 remote_ip_addr_size);
 559     }
 560
 561     SILC_LOG_DEBUG(("UDP packet from %s:%d", remote_ip_addr, *remote_port));
 562   }
 563
 564   return len;
 565 }
 566
 567 /* Send UDP packet */
 568
 569 int silc_net_udp_send(SilcStream stream,
 570                       const char *remote_ip_addr, int remote_port,
 571                       const unsigned char *data, SilcUInt32 data_len)
 572 {
 573   SilcSocketStream sock = stream;
 574   SilcSockaddr remote;
 575   int ret;
 576
 577   SILC_LOG_DEBUG(("Sending data to UDP socket %d", sock->sock));
 578
 579   /* Set sockaddr */
 580   if (!silc_net_set_sockaddr(&remote, remote_ip_addr, remote_port))
 581     return -2;
 582
 583   /* Send */
 584   ret = sendto(sock->sock, data, data_len, 0, &remote.sa,
 585                SIZEOF_SOCKADDR(remote));
 586   if (ret < 0) {
 587     silc_set_errno_posix(errno);
 588     if (errno == EAGAIN || errno == EINTR) {
 589       SILC_LOG_DEBUG(("Could not send immediately, will do it later"));
 590       silc_schedule_set_listen_fd(sock->schedule, sock->sock,
 591                                   SILC_TASK_READ | SILC_TASK_WRITE, FALSE);
 592       return -1;
 593     }
 594     SILC_LOG_DEBUG(("Cannot send to UDP socket: %s", strerror(errno)));
 595     silc_schedule_unset_listen_fd(sock->schedule, sock->sock);
 596     return -2;
 597   }
 598
 599   SILC_LOG_DEBUG(("Sent data %d bytes", ret));
 600   if (silc_schedule_get_fd_events(sock->schedule, sock->sock) &
 601       SILC_TASK_WRITE)
 602     silc_schedule_set_listen_fd(sock->schedule, sock->sock,
 603                                 SILC_TASK_READ, FALSE);
 604
 605   return ret;
 606 }
 607
 608 /******************************* TCP Stream *********************************/
 609
 610 /* Asynchronous TCP/IP connecting */
 611
 612 typedef struct {
 613   SilcResult status;
 614   SilcStream stream;
 615   SilcFSMStruct fsm;
 616   SilcFSMEventStruct event;
 617   SilcAsyncOperation op;
 618   SilcAsyncOperation sop;
 619   char *local_ip;
 620   char *remote;
 621   char ip_addr[64];
 622   int sock;
 623   SilcNetCallback callback;
 624   void *context;
 625   unsigned int port     : 24;
 626   unsigned int retry    : 7;
 627   unsigned int aborted  : 1;
 628 } *SilcNetConnect;
 629
 630 SILC_FSM_STATE(silc_net_connect_st_start);
 631 SILC_FSM_STATE(silc_net_connect_st_connected);
 632 SILC_FSM_STATE(silc_net_connect_st_stream);
 633 SILC_FSM_STATE(silc_net_connect_st_finish);
 634
 635 SILC_TASK_CALLBACK(silc_net_connect_wait)
 636 {
 637   SilcNetConnect conn = context;
 638   SILC_FSM_EVENT_SIGNAL(&conn->event);
 639 }
 640
 641 SILC_FSM_STATE(silc_net_connect_st_start)
 642 {
 643   SilcNetConnect conn = fsm_context;
 644   int sock, rval;
 645   SilcSockaddr desthost;
 646   SilcBool prefer_ipv6 = TRUE;
 647
 648   if (conn->aborted) {
 649     /** Aborted */
 650     silc_fsm_next(fsm, silc_net_connect_st_finish);
 651     return SILC_FSM_CONTINUE;
 652   }
 653
 654   /* Do host lookup */
 655  retry:
 656   if (!silc_net_gethostbyname(conn->remote, prefer_ipv6,
 657                               conn->ip_addr, sizeof(conn->ip_addr))) {
 658     SILC_LOG_ERROR(("Network (%s) unreachable: could not resolve the "
 659                     "host", conn->remote));
 660
 661     /** Network unreachable */
 662     conn->status = SILC_ERR_UNREACHABLE;
 663     silc_fsm_next(fsm, silc_net_connect_st_finish);
 664     return SILC_FSM_CONTINUE;
 665   }
 666
 667   /* Set sockaddr for this connection */
 668   if (!silc_net_set_sockaddr(&desthost, conn->ip_addr, conn->port)) {
 669     /** Sockaddr failed */
 670     silc_fsm_next(fsm, silc_net_connect_st_finish);
 671     return SILC_FSM_CONTINUE;
 672   }
 673
 674   /* Create the connection socket */
 675   sock = socket(desthost.sin.sin_family, SOCK_STREAM, 0);
 676   if (sock < 0) {
 677     /* If address is IPv6, then fallback to IPv4 and see whether we can do
 678        better with that on socket creation. */
 679     if (prefer_ipv6 && silc_net_is_ip6(conn->ip_addr)) {
 680       prefer_ipv6 = FALSE;
 681       goto retry;
 682     }
 683
 684     /** Cannot create socket */
 685     silc_set_errno_posix(errno);
 686     SILC_LOG_ERROR(("Cannot create socket: %s", strerror(errno)));
 687     silc_fsm_next(fsm, silc_net_connect_st_finish);
 688     return SILC_FSM_CONTINUE;
 689   }
 690
 691   /* Bind to the local address if provided */
 692   if (conn->local_ip) {
 693     SilcSockaddr local;
 694
 695     /* Set sockaddr for local listener, and try to bind it. */
 696     if (silc_net_set_sockaddr(&local, conn->local_ip, 0))
 697       bind(sock, &local.sa, SIZEOF_SOCKADDR(local));
 698   }
 699
 700   /* Set the socket to non-blocking mode */
 701   silc_net_set_socket_nonblock(sock);
 702
 703   /* Connect to the host */
 704   rval = connect(sock, &desthost.sa, SIZEOF_SOCKADDR(desthost));
 705   if (rval < 0) {
 706     if (errno != EINPROGRESS) {
 707       silc_set_errno_posix(errno);
 708       shutdown(sock, 2);
 709       close(sock);
 710
 711       /* Retry using an IPv4 adress, if IPv6 didn't work */
 712       if (prefer_ipv6 && silc_net_is_ip6(conn->ip_addr)) {
 713         prefer_ipv6 = FALSE;
 714         goto retry;
 715       }
 716
 717       /** Cannot connect to remote host */
 718       SILC_LOG_ERROR(("Cannot connect to remote host: %s",
 719                       silc_errno_string(silc_errno)));
 720       silc_fsm_next(fsm, silc_net_connect_st_finish);
 721       return SILC_FSM_CONTINUE;
 722     }
 723   }
 724
 725   /* Set appropriate options */
 726 #if defined(TCP_NODELAY)
 727   silc_net_set_socket_opt(sock, IPPROTO_TCP, TCP_NODELAY, 1);
 728 #endif
 729   silc_net_set_socket_opt(sock, SOL_SOCKET, SO_KEEPALIVE, 1);
 730
 731   SILC_LOG_DEBUG(("Connection operation in progress"));
 732
 733   conn->sock = sock;
 734
 735   /** Wait for connection */
 736   silc_fsm_next(fsm, silc_net_connect_st_connected);
 737   silc_fsm_event_init(&conn->event, fsm);
 738   silc_schedule_task_add_fd(silc_fsm_get_schedule(fsm), sock,
 739                             silc_net_connect_wait, conn);
 740   silc_schedule_set_listen_fd(silc_fsm_get_schedule(fsm), sock,
 741                               SILC_TASK_WRITE, FALSE);
 742   SILC_FSM_EVENT_WAIT(&conn->event);
 743   return SILC_FSM_CONTINUE;
 744 }
 745
 746 static void silc_net_connect_wait_stream(SilcResult status,
 747                                          SilcStream stream, void *context)
 748 {
 749   SilcNetConnect conn = context;
 750   conn->sop = NULL;
 751   conn->status = status;
 752   conn->stream = stream;
 753   SILC_FSM_CALL_CONTINUE(&conn->fsm);
 754 }
 755
 756 SILC_FSM_STATE(silc_net_connect_st_connected)
 757 {
 758   SilcNetConnect conn = fsm_context;
 759   SilcSchedule schedule = silc_fsm_get_schedule(fsm);
 760   int opt = EINVAL, optlen = sizeof(opt), ret;
 761
 762   if (conn->aborted) {
 763     /** Aborted */
 764     silc_schedule_unset_listen_fd(schedule, conn->sock);
 765     silc_schedule_task_del_by_fd(schedule, conn->sock);
 766     silc_fsm_next(fsm, silc_net_connect_st_finish);
 767     return SILC_FSM_CONTINUE;
 768   }
 769
 770   ret = silc_net_get_socket_opt(conn->sock, SOL_SOCKET, SO_ERROR,
 771                                 &opt, &optlen);
 772
 773   silc_schedule_unset_listen_fd(schedule, conn->sock);
 774   silc_schedule_task_del_by_fd(schedule, conn->sock);
 775
 776   if (ret != 0 || opt != 0) {
 777     if (conn->retry) {
 778       /** Retry connecting */
 779       SILC_LOG_DEBUG(("Retry connecting"));
 780       conn->retry--;
 781       silc_net_close_connection(conn->sock);
 782       silc_fsm_next(fsm, silc_net_connect_st_start);
 783       return SILC_FSM_CONTINUE;
 784     }
 785
 786     /* Set error */
 787     silc_set_errno_posix(opt);
 788     conn->status = silc_errno;
 789
 790     /** Connecting failed */
 791     SILC_LOG_DEBUG(("Connecting failed, error %s",
 792                     silc_errno_string(silc_errno)));
 793     silc_fsm_next(fsm, silc_net_connect_st_finish);
 794     return SILC_FSM_CONTINUE;
 795   }
 796
 797   SILC_LOG_DEBUG(("TCP connection established"));
 798
 799   /** Connection created */
 800   silc_fsm_next(fsm, silc_net_connect_st_stream);
 801   SILC_FSM_CALL((conn->sop = silc_socket_tcp_stream_create(
 802                                      conn->sock, TRUE, FALSE, schedule,
 803                                      silc_net_connect_wait_stream, conn)));
 804 }
 805
 806 SILC_FSM_STATE(silc_net_connect_st_stream)
 807 {
 808   SilcNetConnect conn = fsm_context;
 809
 810   if (conn->aborted) {
 811     /** Aborted */
 812     silc_fsm_next(fsm, silc_net_connect_st_finish);
 813     return SILC_FSM_CONTINUE;
 814   }
 815
 816   if (conn->status != SILC_OK) {
 817     /** Stream creation failed */
 818     silc_fsm_next(fsm, silc_net_connect_st_finish);
 819     return SILC_FSM_CONTINUE;
 820   }
 821
 822   /** Stream created successfully */
 823   SILC_LOG_DEBUG(("Connected successfully, sock %d", conn->sock));
 824   conn->status = SILC_OK;
 825   silc_fsm_next(fsm, silc_net_connect_st_finish);
 826   return SILC_FSM_CONTINUE;
 827 }
 828
 829 SILC_FSM_STATE(silc_net_connect_st_finish)
 830 {
 831   SilcNetConnect conn = fsm_context;
 832
 833   /* Deliver error or new stream */
 834   if (!conn->aborted) {
 835     conn->callback(conn->status, conn->stream, conn->context);
 836     if (conn->op)
 837       silc_async_free(conn->op);
 838   }
 839
 840   return SILC_FSM_FINISH;
 841 }
 842
 843 static void silc_net_connect_abort(SilcAsyncOperation op, void *context)
 844 {
 845   SilcNetConnect conn = context;
 846   conn->aborted = TRUE;
 847
 848   /* Abort underlaying stream creation too */
 849   if (conn->sop) {
 850     silc_async_abort(conn->sop, NULL, NULL);
 851     conn->sop = NULL;
 852   }
 853 }
 854
 855 static void silc_net_connect_destructor(SilcFSM fsm, void *fsm_context,
 856                                         void *destructor_context)
 857 {
 858   SilcNetConnect conn = fsm_context;
 859   silc_free(conn->local_ip);
 860   silc_free(conn->remote);
 861   silc_free(conn);
 862 }
 863
 864 /* Create asynchronous TCP/IP connection. */
 865
 866 SilcAsyncOperation silc_net_tcp_connect(const char *local_ip_addr,
 867                                         const char *remote_ip_addr,
 868                                         int remote_port,
 869                                         SilcSchedule schedule,
 870                                         SilcNetCallback callback,
 871                                         void *context)
 872 {
 873   SilcNetConnect conn;
 874
 875   if (!remote_ip_addr || remote_port < 1 || !schedule || !callback)
 876     return NULL;
 877
 878   SILC_LOG_DEBUG(("Creating connection to host %s port %d",
 879                   remote_ip_addr, remote_port));
 880
 881   conn = silc_calloc(1, sizeof(*conn));
 882   if (!conn) {
 883     callback(silc_errno, NULL, context);
 884     return NULL;
 885   }
 886
 887   /* Start async operation */
 888   conn->op = silc_async_alloc(silc_net_connect_abort, NULL, conn);
 889   if (!conn->op) {
 890     silc_free(conn);
 891     callback(silc_errno, NULL, context);
 892     return NULL;
 893   }
 894
 895   if (local_ip_addr)
 896     conn->local_ip = silc_strdup(local_ip_addr);
 897   conn->remote = silc_strdup(remote_ip_addr);
 898   if (!conn->remote) {
 899     silc_async_free(conn->op);
 900     silc_free(conn->local_ip);
 901     silc_free(conn);
 902     callback(silc_errno, NULL, context);
 903     return NULL;
 904   }
 905   conn->port = remote_port;
 906   conn->callback = callback;
 907   conn->context = context;
 908   conn->retry = 1;
 909   conn->status = SILC_ERR;
 910
 911   silc_fsm_init(&conn->fsm, conn, silc_net_connect_destructor, NULL, schedule);
 912   silc_fsm_start(&conn->fsm, silc_net_connect_st_start);
 913
 914   return conn->op;
 915 }
 916
 917 /* Closes the connection by closing the socket connection. */
 918
 919 void silc_net_close_connection(int sock)
 920 {
 921   SILC_LOG_DEBUG(("Closing sock %d", sock));
 922   close(sock);
 923 }
 924
 925 /* Set's the socket to non-blocking mode. */
 926
 927 int silc_net_set_socket_nonblock(SilcSocket sock)
 928 {
 929   int ret = fcntl((int)sock, F_SETFL, fcntl(sock, F_GETFL, 0) | O_NONBLOCK);
 930   if (ret)
 931     silc_set_errno_posix(errno);
 932   return ret;
 933 }
 934
 935 /* Converts the IP number string from numbers-and-dots notation to
 936    binary form. */
 937
 938 SilcBool silc_net_addr2bin(const char *addr, void *bin, SilcUInt32 bin_len)
 939 {
 940   int ret = 0;
 941
 942   if (silc_net_is_ip4(addr)) {
 943     /* IPv4 address */
 944     struct in_addr tmp;
 945
 946     if (bin_len < 4) {
 947       silc_set_errno(SILC_ERR_INVALID_ARGUMENT);
 948       return FALSE;
 949     }
 950
 951     ret = inet_aton(addr, &tmp);
 952     if (!ret) {
 953       silc_set_errno_posix(errno);
 954       return FALSE;
 955     }
 956
 957     memcpy(bin, (unsigned char *)&tmp.s_addr, 4);
 958 #ifdef HAVE_IPV6
 959   } else {
 960     struct addrinfo hints, *ai;
 961     SilcSockaddr *s;
 962
 963     /* IPv6 address */
 964     if (bin_len < 16) {
 965       silc_set_errno(SILC_ERR_INVALID_ARGUMENT);
 966       return FALSE;
 967     }
 968
 969     memset(&hints, 0, sizeof(hints));
 970     hints.ai_family = AF_INET6;
 971     if (getaddrinfo(addr, NULL, &hints, &ai))
 972       return FALSE;
 973
 974     if (ai) {
 975       s = (SilcSockaddr *)ai->ai_addr;
 976       memcpy(bin, &s->sin6.sin6_addr, sizeof(s->sin6.sin6_addr));
 977       freeaddrinfo(ai);
 978     }
 979
 980     ret = TRUE;
 981 #endif /* HAVE_IPV6 */
 982   }
 983
 984   return ret != 0;
 985 }