Initial code commit for Toolkit 1.1.
[silc.git] / lib / silcutil / unix / silcunixnet.c
index f997196dca340dae737e5ca26786281211f48be9..be2d876013064847998ed6733c3b49366b161a84 100644 (file)
@@ -9,7 +9,7 @@
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; version 2 of the License.
-  
+
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
@@ -53,7 +53,7 @@ static bool silc_net_set_sockaddr(SilcSockaddr *addr, const char *ip_addr,
     if (silc_net_is_ip4(ip_addr)) {
       /* IPv4 address */
       len = sizeof(addr->sin.sin_addr);
-      silc_net_addr2bin(ip_addr, 
+      silc_net_addr2bin(ip_addr,
                        (unsigned char *)&addr->sin.sin_addr.s_addr, len);
       addr->sin.sin_family = AF_INET;
       addr->sin.sin_port = port ? htons(port) : 0;
@@ -61,7 +61,7 @@ static bool silc_net_set_sockaddr(SilcSockaddr *addr, const char *ip_addr,
 #ifdef HAVE_IPV6
       /* IPv6 address */
       len = sizeof(addr->sin6.sin6_addr);
-      silc_net_addr2bin(ip_addr, 
+      silc_net_addr2bin(ip_addr,
                        (unsigned char *)&addr->sin6.sin6_addr, len);
       addr->sin6.sin6_family = AF_INET6;
       addr->sin6.sin6_port = port ? htons(port) : 0;
@@ -81,194 +81,248 @@ static bool silc_net_set_sockaddr(SilcSockaddr *addr, const char *ip_addr,
   return TRUE;
 }
 
-/* This function creates server or daemon or listener or what ever. This
-   does not fork a new process, it must be done by the caller if caller
-   wants to create a child process. This is used by the SILC server. 
-   If argument `ip_addr' is NULL `any' address will be used. Returns 
-   the created socket or -1 on error. */
+/* Deliver new stream to upper layer */
 
-int silc_net_create_server(int port, const char *ip_addr)
+static void silc_net_accept_stream(SilcSocketStreamStatus status,
+                                  SilcStream stream, void *context)
 {
-  int sock, rval;
-  SilcSockaddr server;
+  SilcNetServer server = context;
 
-  SILC_LOG_DEBUG(("Creating a new server listener"));
+  if (status != SILC_SOCKET_OK)
+    return;
 
-  /* Set sockaddr for server */
-  if (!silc_net_set_sockaddr(&server, ip_addr, port))
-    return -1;
+  server->callback(SILC_NET_OK, stream, server->context);
+}
 
-  /* Create the socket */
-  sock = socket(server.sin.sin_family, SOCK_STREAM, 0);
-  if (sock < 0) {
-    SILC_LOG_ERROR(("Cannot create socket: %s", strerror(errno)));
-    return -1;
-  }
+/* Accept incoming connection and notify upper layer */
 
-  /* Set the socket options */
-  rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEADDR, 1);
-  if (rval < 0) {
-    SILC_LOG_ERROR(("Cannot set socket options: %s", strerror(errno)));
-    return -1;
-  }
+SILC_TASK_CALLBACK(silc_net_accept)
+{
+  SilcNetServer server = context;
+  int sock;
 
-  /* Bind the server socket */
-  rval = bind(sock, &server.sa, SIZEOF_SOCKADDR(server));
-  if (rval < 0) {
-    SILC_LOG_DEBUG(("Cannot bind socket: %s", strerror(errno)));
-    return -1;
-  }
+  SILC_LOG_DEBUG(("Accepting new connection"));
 
-  /* Specify that we are listenning */
-  rval = listen(sock, 5);
-  if (rval < 0) {
-    SILC_LOG_ERROR(("Cannot set socket listenning: %s", strerror(errno)));
-    return -1;
-  }
+  sock = silc_net_accept_connection(fd);
+  if (sock < 0)
+    return;
 
-  /* Set the server socket to non-blocking mode */
+  /* Set socket options */
   silc_net_set_socket_nonblock(sock);
+  silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEADDR, 1);
 
-  SILC_LOG_DEBUG(("Server listener created, fd=%d", sock));
-
-  return sock;
+  /* Create socket stream */
+  silc_socket_stream_create(sock, TRUE, server->require_fqdn, schedule,
+                           silc_net_accept_stream, server);
 }
 
-/* Closes the server by closing the socket connection. */
+/* Create network listener */
 
-void silc_net_close_server(int sock)
+SilcNetServer
+silc_net_create_server(const char **local_ip_addr, SilcUInt32 local_ip_count,
+                      int port, bool require_fqdn, SilcSchedule schedule,
+                      SilcNetCallback callback, void *context)
 {
-  shutdown(sock, 2);
-  close(sock);
-
-  SILC_LOG_DEBUG(("Server socket closed"));
-}
+  SilcNetServer netserver = NULL;
+  SilcSockaddr server;
+  int i, sock, rval;
+  const char *ipany = "0.0.0.0";
 
-/* Creates a connection (TCP/IP) to a remote host. Returns the connection
-   socket or -1 on error. This blocks the process while trying to create
-   the connection. */
+  SILC_LOG_DEBUG(("Creating new network listener"));
 
-int silc_net_create_connection(const char *local_ip, int port, 
-                              const char *host)
-{
-  int sock, rval;
-  char ip_addr[64];
-  SilcSockaddr desthost;
-  bool prefer_ipv6 = TRUE;
+  if (port < 1 || !schedule || !callback)
+    goto err;
 
-  SILC_LOG_DEBUG(("Creating connection to host %s port %d", host, port));
+  netserver = silc_calloc(1, sizeof(*netserver));
+  if (!netserver) {
+    callback(SILC_NET_NO_MEMORY, NULL, context);
+    return NULL;
+  }
+  netserver->schedule = schedule;
+  netserver->callback = callback;
+  netserver->context = context;
+
+  if (local_ip_count > 0) {
+    netserver->socks = silc_calloc(local_ip_count, sizeof(*netserver->socks));
+    if (!netserver->socks) {
+      callback(SILC_NET_NO_MEMORY, NULL, context);
+      return NULL;
+    }
+  } else {
+    netserver->socks = silc_calloc(1, sizeof(*netserver->socks));
+    if (!netserver->socks) {
+      callback(SILC_NET_NO_MEMORY, NULL, context);
+      return NULL;
+    }
 
-  /* Do host lookup */
- retry:
-  if (!silc_net_gethostbyname(host, prefer_ipv6, ip_addr, sizeof(ip_addr))) {
-    SILC_LOG_ERROR(("Network (%s) unreachable: could not resolve the "
-                   "IP address", host));
-    return -1;
+    local_ip_count = 1;
   }
 
-  /* Set sockaddr for this connection */
-  if (!silc_net_set_sockaddr(&desthost, ip_addr, port))
-    return -1;
+  /* Bind to local addresses */
+  for (i = 0; i < local_ip_count; i++) {
+    SILC_LOG_DEBUG(("Binding to local address %s",
+                   local_ip_addr ? local_ip_addr[i] : ipany));
+
+    /* Set sockaddr for server */
+    if (!silc_net_set_sockaddr(&server,
+                              local_ip_addr ? local_ip_addr[i] : ipany,
+                              port))
+      goto err;
+
+    /* Create the socket */
+    sock = socket(server.sin.sin_family, SOCK_STREAM, 0);
+    if (sock < 0) {
+      SILC_LOG_ERROR(("Cannot create socket: %s", strerror(errno)));
+      goto err;
+    }
 
-  /* Create the connection socket */
-  sock = socket(desthost.sin.sin_family, SOCK_STREAM, 0);
-  if (sock < 0) {
-    /* If address is IPv6, then fallback to IPv4 and see whether we can do
-       better with that on socket creation. */
-    if (prefer_ipv6 && silc_net_is_ip6(ip_addr)) {
-      prefer_ipv6 = FALSE;
-      goto retry;
+    /* Set the socket options */
+    rval = silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEADDR, 1);
+    if (rval < 0) {
+      SILC_LOG_ERROR(("Cannot set socket options: %s", strerror(errno)));
+      goto err;
     }
 
-    SILC_LOG_ERROR(("Cannot create socket: %s", strerror(errno)));
-    return -1;
-  }
+    /* Bind the server socket */
+    rval = bind(sock, &server.sa, SIZEOF_SOCKADDR(server));
+    if (rval < 0) {
+      SILC_LOG_DEBUG(("Cannot bind socket: %s", strerror(errno)));
+      goto err;
+    }
 
-  /* Bind to the local address if provided */
-  if (local_ip) {
-    SilcSockaddr local;
+    /* Specify that we are listenning */
+    rval = listen(sock, 5);
+    if (rval < 0) {
+      SILC_LOG_ERROR(("Cannot set socket listenning: %s", strerror(errno)));
+      goto err;
+    }
 
-    /* Set sockaddr for local listener, and try to bind it. */
-    if (silc_net_set_sockaddr(&local, local_ip, 0))
-      bind(sock, &local.sa, SIZEOF_SOCKADDR(local));
-  }
+    /* Set the server socket to non-blocking mode */
+    silc_net_set_socket_nonblock(sock);
 
-  /* Connect to the host */
-  rval = connect(sock, &desthost.sa, SIZEOF_SOCKADDR(desthost));
-  if (rval < 0) {
-    /* retry using an IPv4 adress, if IPv6 didn't work */
-    if (prefer_ipv6 && silc_net_is_ip6(ip_addr)) {
-      shutdown(sock, 2);
-      close(sock);
+    /* Schedule for incoming connections */
+    silc_schedule_task_add_fd(schedule, sock, silc_net_accept, netserver);
 
-      prefer_ipv6 = FALSE;
-      goto retry;
-    }
-    SILC_LOG_ERROR(("Cannot connect to remote host: %s", strerror(errno)));
-    shutdown(sock, 2);
-    close(sock);
-    return -1;
+    SILC_LOG_DEBUG(("Network listener created, fd=%d", sock));
+    netserver->socks[i] = sock;
+    netserver->socks_count++;
   }
 
-  /* Set appropriate options */
-#if defined(TCP_NODELAY)
-  silc_net_set_socket_opt(sock, IPPROTO_TCP, TCP_NODELAY, 1);
-#endif
-  silc_net_set_socket_opt(sock, SOL_SOCKET, SO_KEEPALIVE, 1);
+  return netserver;
+
+ err:
+  if (callback)
+    callback(SILC_NET_ERROR, NULL, context);
+  if (netserver)
+    silc_net_close_server(netserver);
+  return NULL;
+}
 
-  SILC_LOG_DEBUG(("Connection created"));
+/* Close network listener */
 
-  return sock;
+void silc_net_close_server(SilcNetServer server)
+{
+  int i;
+
+  SILC_LOG_DEBUG(("Closing network listener"));
+
+  for (i = 0; i < server->socks_count; i++) {
+    silc_schedule_task_del_by_fd(server->schedule, server->socks[i]);
+    shutdown(server->socks[i], 2);
+    close(server->socks[i]);
+  }
+
+  silc_free(server->socks);
+  silc_free(server);
 }
 
-/* Creates a connection (TCP/IP) to a remote host. Returns the connection
-   socket or -1 on error. This creates non-blocking socket hence the
-   connection returns directly. To get the result of the connect() one
-   must select() the socket and read the result after it's ready. */
+/* Asynchronous TCP/IP connecting */
+
+typedef struct {
+  SilcNetStatus status;
+  SilcSocketStreamStatus stream_status;
+  SilcStream stream;
+  SilcFSMStruct fsm;
+  SilcFSMSemaStruct sema;
+  SilcAsyncOperation op;
+  SilcAsyncOperation sop;
+  char *local_ip;
+  char *remote;
+  char ip_addr[64];
+  int sock;
+  SilcNetCallback callback;
+  void *context;
+  unsigned int port     : 24;
+  unsigned int retry    : 7;
+  unsigned int aborted  : 1;
+} *SilcNetConnect;
+
+SILC_FSM_STATE(silc_net_connect_st_start);
+SILC_FSM_STATE(silc_net_connect_st_connected);
+SILC_FSM_STATE(silc_net_connect_st_stream);
+SILC_FSM_STATE(silc_net_connect_st_finish);
+
+SILC_TASK_CALLBACK(silc_net_connect_wait)
+{
+  SilcNetConnect conn = context;
+  SILC_FSM_SEMA_POST(&conn->sema);
+}
 
-int silc_net_create_connection_async(const char *local_ip, int port, 
-                                    const char *host)
+SILC_FSM_STATE(silc_net_connect_st_start)
 {
+  SilcNetConnect conn = fsm_context;
   int sock, rval;
-  char ip_addr[64];
   SilcSockaddr desthost;
   bool prefer_ipv6 = TRUE;
 
-  SILC_LOG_DEBUG(("Creating connection (async) to host %s port %d", 
-                 host, port));
+  if (conn->aborted) {
+    /** Aborted */
+    silc_fsm_next(fsm, silc_net_connect_st_finish);
+    return SILC_FSM_CONTINUE;
+  }
 
   /* Do host lookup */
  retry:
-  if (!silc_net_gethostbyname(host, prefer_ipv6, ip_addr, sizeof(ip_addr))) {
+  if (!silc_net_gethostbyname(conn->remote, prefer_ipv6,
+                             conn->ip_addr, sizeof(conn->ip_addr))) {
     SILC_LOG_ERROR(("Network (%s) unreachable: could not resolve the "
-                   "IP address", host));
-    return -1;
+                   "host", conn->remote));
+
+    /** Network unreachable */
+    conn->status = SILC_NET_HOST_UNREACHABLE;
+    silc_fsm_next(fsm, silc_net_connect_st_finish);
+    return SILC_FSM_CONTINUE;
   }
 
   /* Set sockaddr for this connection */
-  if (!silc_net_set_sockaddr(&desthost, ip_addr, port))
-    return -1;
+  if (!silc_net_set_sockaddr(&desthost, conn->ip_addr, conn->port)) {
+    /** Sockaddr failed */
+    silc_fsm_next(fsm, silc_net_connect_st_finish);
+    return SILC_FSM_CONTINUE;
+  }
 
   /* Create the connection socket */
   sock = socket(desthost.sin.sin_family, SOCK_STREAM, 0);
   if (sock < 0) {
     /* If address is IPv6, then fallback to IPv4 and see whether we can do
        better with that on socket creation. */
-    if (prefer_ipv6 && silc_net_is_ip6(ip_addr)) {
+    if (prefer_ipv6 && silc_net_is_ip6(conn->ip_addr)) {
       prefer_ipv6 = FALSE;
       goto retry;
     }
 
+    /** Cannot create socket */
     SILC_LOG_ERROR(("Cannot create socket: %s", strerror(errno)));
-    return -1;
+    silc_fsm_next(fsm, silc_net_connect_st_finish);
+    return SILC_FSM_CONTINUE;
   }
 
   /* Bind to the local address if provided */
-  if (local_ip) {
+  if (conn->local_ip) {
     SilcSockaddr local;
 
     /* Set sockaddr for local listener, and try to bind it. */
-    if (silc_net_set_sockaddr(&local, local_ip, 0))
+    if (silc_net_set_sockaddr(&local, conn->local_ip, 0))
       bind(sock, &local.sa, SIZEOF_SOCKADDR(local));
   }
 
@@ -278,9 +332,9 @@ int silc_net_create_connection_async(const char *local_ip, int port,
   /* Connect to the host */
   rval = connect(sock, &desthost.sa, SIZEOF_SOCKADDR(desthost));
   if (rval < 0) {
-    if (errno !=  EINPROGRESS) {
+    if (errno != EINPROGRESS) {
       /* retry using an IPv4 adress, if IPv6 didn't work */
-      if (prefer_ipv6 && silc_net_is_ip6(ip_addr)) {
+      if (prefer_ipv6 && silc_net_is_ip6(conn->ip_addr)) {
         shutdown(sock, 2);
         close(sock);
 
@@ -288,10 +342,13 @@ int silc_net_create_connection_async(const char *local_ip, int port,
         goto retry;
       }
 
-      SILC_LOG_ERROR(("Cannot connect to remote host: %s", strerror(errno)));
       shutdown(sock, 2);
       close(sock);
-      return -1;
+
+      /** Cannot connect to remote host */
+      SILC_LOG_ERROR(("Cannot connect to remote host: %s", strerror(errno)));
+      silc_fsm_next(fsm, silc_net_connect_st_finish);
+      return SILC_FSM_CONTINUE;
     }
   }
 
@@ -303,7 +360,199 @@ int silc_net_create_connection_async(const char *local_ip, int port,
 
   SILC_LOG_DEBUG(("Connection operation in progress"));
 
-  return sock;
+  conn->sock = sock;
+
+  /** Wait for connection */
+  silc_fsm_next(fsm, silc_net_connect_st_connected);
+  silc_fsm_sema_init(&conn->sema, fsm, 0);
+  silc_schedule_task_add_fd(silc_fsm_get_schedule(fsm), sock,
+                           silc_net_connect_wait, conn);
+  silc_schedule_set_listen_fd(silc_fsm_get_schedule(fsm), sock,
+                             SILC_TASK_WRITE, FALSE);
+  SILC_FSM_SEMA_WAIT(&conn->sema);
+  return SILC_FSM_CONTINUE;
+}
+
+static void silc_net_connect_wait_stream(SilcSocketStreamStatus status,
+                                        SilcStream stream, void *context)
+{
+  SilcNetConnect conn = context;
+  conn->stream_status = status;
+  conn->stream = stream;
+  SILC_FSM_CALL_CONTINUE(&conn->fsm);
+}
+
+SILC_FSM_STATE(silc_net_connect_st_connected)
+{
+  SilcNetConnect conn = fsm_context;
+  SilcSchedule schedule = silc_fsm_get_schedule(fsm);
+  int opt = EINVAL, optlen = sizeof(opt), ret;
+
+  if (conn->aborted) {
+    /** Aborted */
+    silc_fsm_next(fsm, silc_net_connect_st_finish);
+    return SILC_FSM_CONTINUE;
+  }
+
+  ret = silc_net_get_socket_opt(conn->sock, SOL_SOCKET, SO_ERROR,
+                               &opt, &optlen);
+
+  silc_schedule_task_del_by_fd(schedule, conn->sock);
+  silc_schedule_unset_listen_fd(schedule, conn->sock);
+
+  if (ret != 0 || opt != 0) {
+    if (conn->retry) {
+      /** Retry connecting */
+      SILC_LOG_DEBUG(("Retry connecting"));
+      conn->retry--;
+      silc_net_close_connection(conn->sock);
+      silc_fsm_next(fsm, silc_net_connect_st_start);
+      return SILC_FSM_CONTINUE;
+    }
+
+#if defined(ECONNREFUSED)
+    if (errno == ECONNREFUSED)
+      conn->status = SILC_NET_CONNECTION_REFUSED;
+#endif /* ECONNREFUSED */
+#if defined(ETIMEDOUT)
+    if (errno == ETIMEDOUT)
+      conn->status = SILC_NET_CONNECTION_TIMEOUT;
+#endif /* ETIMEDOUT */
+#if defined(ENETUNREACH)
+    if (errno == ENETUNREACH)
+      conn->status = SILC_NET_HOST_UNREACHABLE;
+#endif /* ENETUNREACH */
+
+    /** Connecting failed */
+    SILC_LOG_DEBUG(("Connecting failed"));
+    silc_fsm_next(fsm, silc_net_connect_st_finish);
+    return SILC_FSM_CONTINUE;
+  }
+
+  /** Connection created */
+  silc_fsm_next(fsm, silc_net_connect_st_stream);
+  SILC_FSM_CALL((conn->sop = silc_socket_stream_create(
+                                    conn->sock, FALSE, FALSE,
+                                    schedule,
+                                    silc_net_connect_wait_stream, conn)));
+}
+
+SILC_FSM_STATE(silc_net_connect_st_stream)
+{
+  SilcNetConnect conn = fsm_context;
+
+  if (conn->aborted) {
+    /** Aborted */
+    silc_fsm_next(fsm, silc_net_connect_st_finish);
+    return SILC_FSM_CONTINUE;
+  }
+
+  if (conn->stream_status != SILC_SOCKET_OK) {
+    /** Stream creation failed */
+    if (conn->stream_status == SILC_SOCKET_UNKNOWN_IP)
+      conn->status = SILC_NET_UNKNOWN_IP;
+    else if (conn->stream_status == SILC_SOCKET_UNKNOWN_HOST)
+      conn->status = SILC_NET_UNKNOWN_HOST;
+    else
+      conn->status = SILC_NET_ERROR;
+    silc_fsm_next(fsm, silc_net_connect_st_finish);
+    return SILC_FSM_CONTINUE;
+  }
+
+  /* Set stream information */
+  silc_socket_stream_set_info(conn->stream,
+                             !silc_net_is_ip(conn->remote) ? conn->remote :
+                             conn->ip_addr, conn->ip_addr, conn->port);
+
+  /** Stream created successfully */
+  SILC_LOG_DEBUG(("Connected successfully"));
+  conn->status = SILC_NET_OK;
+  silc_fsm_next(fsm, silc_net_connect_st_finish);
+  return SILC_FSM_CONTINUE;
+}
+
+SILC_FSM_STATE(silc_net_connect_st_finish)
+{
+  SilcNetConnect conn = fsm_context;
+
+  /* Deliver error or new stream */
+  if (!conn->aborted) {
+    conn->callback(conn->status, conn->stream, conn->context);
+    if (conn->op)
+      silc_async_free(conn->op);
+    if (conn->sop)
+      silc_async_free(conn->sop);
+  }
+
+  return SILC_FSM_FINISH;
+}
+
+static void silc_net_connect_abort(SilcAsyncOperation op, void *context)
+{
+  SilcNetConnect conn = context;
+  conn->aborted = TRUE;
+
+  /* Abort underlaying stream creation too */
+  if (conn->sop)
+    silc_async_abort(conn->op, NULL, NULL);
+}
+
+static void silc_net_connect_destructor(SilcFSM fsm, void *fsm_context,
+                                       void *destructor_context)
+{
+  SilcNetConnect conn = fsm_context;
+  silc_free(conn->local_ip);
+  silc_free(conn->remote);
+  silc_free(conn);
+}
+
+/* Create asynchronous TCP/IP connection. */
+
+SilcAsyncOperation silc_net_tcp_connect(const char *local_ip_addr,
+                                       const char *remote_ip_addr,
+                                       int remote_port,
+                                       SilcSchedule schedule,
+                                       SilcNetCallback callback,
+                                       void *context)
+{
+  SilcNetConnect conn;
+
+  if (!remote_ip_addr || remote_port < 1 || !schedule || !callback)
+    return NULL;
+
+  SILC_LOG_DEBUG(("Creating connection to host %s port %d",
+                 remote_ip_addr, remote_port));
+
+  conn = silc_calloc(1, sizeof(*conn));
+  if (!conn) {
+    callback(SILC_NET_NO_MEMORY, NULL, context);
+    return NULL;
+  }
+
+  /* Start async operation */
+  conn->op = silc_async_alloc(silc_net_connect_abort, NULL, conn);
+  if (!conn->op) {
+    callback(SILC_NET_NO_MEMORY, NULL, context);
+    return NULL;
+  }
+
+  if (local_ip_addr)
+    conn->local_ip = strdup(local_ip_addr);
+  conn->remote = strdup(remote_ip_addr);
+  if (!conn->remote) {
+    callback(SILC_NET_NO_MEMORY, NULL, context);
+    return NULL;
+  }
+  conn->port = remote_port;
+  conn->callback = callback;
+  conn->context = context;
+  conn->retry = 1;
+  conn->status = SILC_NET_ERROR;
+
+  silc_fsm_init(&conn->fsm, conn, silc_net_connect_destructor, NULL, schedule);
+  silc_fsm_start(&conn->fsm, silc_net_connect_st_start);
+
+  return conn->op;
 }
 
 /* Closes the connection by closing the socket connection. */
@@ -333,7 +582,7 @@ bool silc_net_addr2bin(const char *addr, void *bin, SilcUInt32 bin_len)
     ret = inet_aton(addr, &tmp);
     if (bin_len < 4)
       return FALSE;
-    
+
     memcpy(bin, (unsigned char *)&tmp.s_addr, 4);
 #ifdef HAVE_IPV6
   } else {