X-Git-Url: http://git.silcnet.org/gitweb/?a=blobdiff_plain;f=lib%2Fsilcutil%2Funix%2Fsilcunixnet.c;h=1d3364a5568158ead8997c8a166a5f6ae53d4ab6;hb=40f8443d8d3a6577336ee66d18e04d9ac4d956bb;hp=b753ac5118be0430f4e795e63adb38611527835b;hpb=59f21d19e81a313c2348eb124f601dbcb88cbfb1;p=silc.git diff --git a/lib/silcutil/unix/silcunixnet.c b/lib/silcutil/unix/silcunixnet.c index b753ac51..1d3364a5 100644 --- a/lib/silcutil/unix/silcunixnet.c +++ b/lib/silcutil/unix/silcunixnet.c @@ -4,13 +4,12 @@ Author: Pekka Riikonen - Copyright (C) 1997 - 2001 Pekka Riikonen + Copyright (C) 1997 - 2005 Pekka Riikonen 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; either version 2 of the License, or - (at your option) any later version. - + 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 @@ -19,7 +18,7 @@ */ /* $Id$ */ -#include "silcincludes.h" +#include "silc.h" #include "silcnet.h" #ifdef HAVE_IPV6 @@ -37,7 +36,7 @@ typedef union { #endif } SilcSockaddr; -static bool silc_net_set_sockaddr(SilcSockaddr *addr, const char *ip_addr, +static SilcBool silc_net_set_sockaddr(SilcSockaddr *addr, const char *ip_addr, int port) { int len; @@ -54,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; @@ -62,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; @@ -82,125 +81,275 @@ 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; + SilcNetServer server = context; + + if (status != SILC_SOCKET_OK) + return; + + server->callback(SILC_NET_OK, stream, server->context); +} + +/* Accept incoming connection and notify upper layer */ + +SILC_TASK_CALLBACK(silc_net_accept) +{ + SilcNetServer server = context; + int sock; + + SILC_LOG_DEBUG(("Accepting new connection")); + + sock = silc_net_accept_connection(fd); + if (sock < 0) + return; + + /* Set socket options */ + silc_net_set_socket_nonblock(sock); + silc_net_set_socket_opt(sock, SOL_SOCKET, SO_REUSEADDR, 1); + + /* Create socket stream */ + silc_socket_stream_create(sock, TRUE, server->require_fqdn, schedule, + silc_net_accept_stream, server); +} + +/* Create network listener */ + +SilcNetServer +silc_net_create_server(const char **local_ip_addr, SilcUInt32 local_ip_count, + int port, SilcBool require_fqdn, SilcSchedule schedule, + SilcNetCallback callback, void *context) +{ + SilcNetServer netserver = NULL; SilcSockaddr server; + int i, sock, rval; + const char *ipany = "0.0.0.0"; - SILC_LOG_DEBUG(("Creating a new server listener")); + SILC_LOG_DEBUG(("Creating new network listener")); - /* Set sockaddr for server */ - if (!silc_net_set_sockaddr(&server, ip_addr, port)) - return -1; + if (port < 1 || !schedule || !callback) + 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))); - return -1; + 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; + } - /* 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; + local_ip_count = 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))); - 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; + } - /* Specify that we are listenning */ - rval = listen(sock, 5); - if (rval < 0) { - SILC_LOG_ERROR(("Cannot set socket listenning: %s", strerror(errno))); - return -1; - } + /* 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; + } - /* Set the server socket to non-blocking mode */ - silc_net_set_socket_nonblock(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))); + goto err; + } - SILC_LOG_DEBUG(("Server listener created, fd=%d", sock)); + /* 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 the server socket to non-blocking mode */ + silc_net_set_socket_nonblock(sock); + + /* Schedule for incoming connections */ + silc_schedule_task_add_fd(schedule, sock, silc_net_accept, netserver); + + SILC_LOG_DEBUG(("Network listener created, fd=%d", sock)); + netserver->socks[i] = sock; + netserver->socks_count++; + } + + return netserver; - return sock; + err: + if (callback) + callback(SILC_NET_ERROR, NULL, context); + if (netserver) + silc_net_close_server(netserver); + return NULL; } -/* Closes the server by closing the socket connection. */ +/* Close network listener */ -void silc_net_close_server(int sock) +void silc_net_close_server(SilcNetServer server) { - shutdown(sock, 2); - close(sock); + int i; + + SILC_LOG_DEBUG(("Closing network listener")); - SILC_LOG_DEBUG(("Server socket closed")); + 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 blocks the process while trying to create - the connection. */ +/* 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(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; + SilcBool prefer_ipv6 = TRUE; - SILC_LOG_DEBUG(("Creating connection 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)); } + /* Set the 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) { - SILC_LOG_ERROR(("Cannot connect to remote host: %s", strerror(errno))); - shutdown(sock, 2); - close(sock); - return -1; + if (errno != EINPROGRESS) { + /* retry using an IPv4 adress, if IPv6 didn't work */ + if (prefer_ipv6 && silc_net_is_ip6(conn->ip_addr)) { + shutdown(sock, 2); + close(sock); + + prefer_ipv6 = FALSE; + goto retry; + } + + shutdown(sock, 2); + close(sock); + + /** 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; + } } /* Set appropriate options */ @@ -209,85 +358,201 @@ int silc_net_create_connection(const char *local_ip, int port, #endif silc_net_set_socket_opt(sock, SOL_SOCKET, SO_KEEPALIVE, 1); - SILC_LOG_DEBUG(("Connection created")); + 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; } -/* 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. */ +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); +} -int silc_net_create_connection_async(const char *local_ip, int port, - const char *host) +SILC_FSM_STATE(silc_net_connect_st_connected) { - int sock, rval; - char ip_addr[64]; - SilcSockaddr desthost; - bool prefer_ipv6 = TRUE; + 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; + } - SILC_LOG_DEBUG(("Creating connection (async) to host %s port %d", - host, port)); + ret = silc_net_get_socket_opt(conn->sock, SOL_SOCKET, SO_ERROR, + &opt, &optlen); - /* 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; + 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; } - /* Set sockaddr for this connection */ - if (!silc_net_set_sockaddr(&desthost, ip_addr, port)) - return -1; + /** 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))); +} - /* 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; - } +SILC_FSM_STATE(silc_net_connect_st_stream) +{ + SilcNetConnect conn = fsm_context; - SILC_LOG_ERROR(("Cannot create socket: %s", strerror(errno))); - return -1; + if (conn->aborted) { + /** Aborted */ + silc_fsm_next(fsm, silc_net_connect_st_finish); + return SILC_FSM_CONTINUE; } - /* Bind to the local address if provided */ - if (local_ip) { - SilcSockaddr local; + 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 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 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); } - /* Set the socket to non-blocking mode */ - silc_net_set_socket_nonblock(sock); + return SILC_FSM_FINISH; +} - /* Connect to the host */ - rval = connect(sock, &desthost.sa, SIZEOF_SOCKADDR(desthost)); - if (rval < 0) { - if (errno != EINPROGRESS) { - SILC_LOG_ERROR(("Cannot connect to remote host: %s", strerror(errno))); - shutdown(sock, 2); - close(sock); - return -1; - } +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; } - /* 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); + /* 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; + } - SILC_LOG_DEBUG(("Connection operation in progress")); + 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 sock; + return conn->op; } /* Closes the connection by closing the socket connection. */ @@ -307,7 +572,7 @@ int silc_net_set_socket_nonblock(int sock) /* Converts the IP number string from numbers-and-dots notation to binary form. */ -bool silc_net_addr2bin(const char *addr, void *bin, SilcUInt32 bin_len) +SilcBool silc_net_addr2bin(const char *addr, void *bin, SilcUInt32 bin_len) { int ret = 0; @@ -317,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 {