X-Git-Url: http://git.meshlink.io/?p=meshlink;a=blobdiff_plain;f=src%2Fnet_setup.c;h=238ae2d342dc316cd034f50289ad840c4a40881c;hp=b4a16c7e17fd782341db34bd256c9754c1c7fd8b;hb=b70b090a28ff4cbdce29a3ad030f7d51ce9079b2;hpb=4216a7e7a1897c0e34ce82e7c2c4cc82070c7b10 diff --git a/src/net_setup.c b/src/net_setup.c index b4a16c7e..238ae2d3 100644 --- a/src/net_setup.c +++ b/src/net_setup.c @@ -46,6 +46,7 @@ static bool node_get_config(meshlink_handle_t *mesh, node_t *n, config_t *config uint32_t version = packmsg_get_uint32(in); if(version != MESHLINK_CONFIG_VERSION) { + logger(mesh, MESHLINK_ERROR, "Invalid config file for node %s", n->name); config_free(config); return false; } @@ -54,6 +55,7 @@ static bool node_get_config(meshlink_handle_t *mesh, node_t *n, config_t *config uint32_t len = packmsg_get_str_raw(in, &name); if(len != strlen(n->name) || !name || strncmp(name, n->name, len)) { + logger(mesh, MESHLINK_ERROR, "Invalid config file for node %s", n->name); config_free(config); return false; } @@ -61,43 +63,6 @@ static bool node_get_config(meshlink_handle_t *mesh, node_t *n, config_t *config return true; } -/// Read device class, blacklist status and submesh from a host config file. Used at startup when reading all host config files. -bool node_read_partial(meshlink_handle_t *mesh, node_t *n) { - config_t config; - packmsg_input_t in; - - if(!node_get_config(mesh, n, &config, &in)) { - return false; - } - - char *submesh_name = packmsg_get_str_dup(&in); - - if(!strcmp(submesh_name, CORE_MESH)) { - free(submesh_name); - n->submesh = NULL; - } else { - n->submesh = lookup_or_create_submesh(mesh, submesh_name); - free(submesh_name); - - if(!n->submesh) { - config_free(&config); - return false; - } - } - - dev_class_t devclass = packmsg_get_int32(&in); - bool blacklisted = packmsg_get_bool(&in); - config_free(&config); - - if(!packmsg_input_ok(&in) || devclass < 0 || devclass >= DEV_CLASS_COUNT) { - return false; - } - - n->devclass = devclass; - n->status.blacklisted = blacklisted; - return true; -} - /// Read the public key from a host config file. Used whenever we need to start an SPTPS session. bool node_read_public_key(meshlink_handle_t *mesh, node_t *n) { if(ecdsa_active(n->ecdsa)) { @@ -135,7 +100,7 @@ bool node_read_public_key(meshlink_handle_t *mesh, node_t *n) { // Append any known addresses in the config file to the list we currently have uint32_t known_count = 0; - for(uint32_t i = 0; i < 5; i++) { + for(uint32_t i = 0; i < MAX_RECENT; i++) { if(n->recent[i].sa.sa_family) { known_count++; } @@ -143,14 +108,24 @@ bool node_read_public_key(meshlink_handle_t *mesh, node_t *n) { uint32_t count = packmsg_get_array(&in); - if(count > 5 - known_count) { - count = 5 - known_count; + for(uint32_t i = 0; i < count; i++) { + if(i < MAX_RECENT - known_count) { + n->recent[i + known_count] = packmsg_get_sockaddr(&in); + } else { + packmsg_skip_element(&in); + } } - for(uint32_t i = 0; i < count; i++) { - n->recent[i + known_count] = packmsg_get_sockaddr(&in); + time_t last_reachable = packmsg_get_int64(&in); + time_t last_unreachable = packmsg_get_int64(&in); + + if(!n->last_reachable) { + n->last_reachable = last_reachable; } + if(!n->last_unreachable) { + n->last_unreachable = last_unreachable; + } config_free(&config); return true; @@ -205,25 +180,30 @@ bool node_read_from_config(meshlink_handle_t *mesh, node_t *n, const config_t *c const void *key; uint32_t len = packmsg_get_bin_raw(&in, &key); - if(len != 32) { - return false; - } + if(len) { + if(len != 32) { + return false; + } - if(!ecdsa_active(n->ecdsa)) { - n->ecdsa = ecdsa_set_public_key(key); + if(!ecdsa_active(n->ecdsa)) { + n->ecdsa = ecdsa_set_public_key(key); + } } n->canonical_address = packmsg_get_str_dup(&in); uint32_t count = packmsg_get_array(&in); - if(count > 5) { - count = 5; - } - for(uint32_t i = 0; i < count; i++) { - n->recent[i] = packmsg_get_sockaddr(&in); + if(i < MAX_RECENT) { + n->recent[i] = packmsg_get_sockaddr(&in); + } else { + packmsg_skip_element(&in); + } } + n->last_reachable = packmsg_get_int64(&in); + n->last_unreachable = packmsg_get_int64(&in); + return packmsg_done(&in); } @@ -251,7 +231,7 @@ bool node_write_config(meshlink_handle_t *mesh, node_t *n) { uint32_t count = 0; - for(uint32_t i = 0; i < 5; i++) { + for(uint32_t i = 0; i < MAX_RECENT; i++) { if(n->recent[i].sa.sa_family) { count++; } else { @@ -265,14 +245,18 @@ bool node_write_config(meshlink_handle_t *mesh, node_t *n) { packmsg_add_sockaddr(&out, &n->recent[i]); } + packmsg_add_int64(&out, n->last_reachable); + packmsg_add_int64(&out, n->last_unreachable); + if(!packmsg_output_ok(&out)) { + meshlink_errno = MESHLINK_EINTERNAL; return false; } config_t config = {buf, packmsg_output_size(&out, buf)}; if(!config_write(mesh, "current", n->name, &config, mesh->config_key)) { - call_error_cb(mesh); + call_error_cb(mesh, MESHLINK_ESTORAGE); return false; } @@ -295,35 +279,143 @@ static bool load_node(meshlink_handle_t *mesh, const char *name, void *priv) { n = new_node(); n->name = xstrdup(name); - if(!node_read_partial(mesh, n)) { + config_t config; + packmsg_input_t in; + + if(!node_get_config(mesh, n, &config, &in)) { free_node(n); - return true; + return false; } + if(!node_read_from_config(mesh, n, &config)) { + logger(mesh, MESHLINK_ERROR, "Invalid config file for node %s", n->name); + config_free(&config); + free_node(n); + return false; + } + + config_free(&config); + node_add(mesh, n); return true; } -/* - Add listening sockets. -*/ -static bool add_listen_address(meshlink_handle_t *mesh, char *address, bool bindto) { - char *port = mesh->myport; +int setup_tcp_listen_socket(meshlink_handle_t *mesh, const struct addrinfo *aip) { + int nfd = socket(aip->ai_family, SOCK_STREAM, IPPROTO_TCP); + + if(nfd == -1) { + return -1; + } - if(address) { - char *space = strchr(address, ' '); +#ifdef FD_CLOEXEC + fcntl(nfd, F_SETFD, FD_CLOEXEC); +#endif - if(space) { - *space++ = 0; - port = space; - } + int option = 1; + setsockopt(nfd, SOL_SOCKET, SO_REUSEADDR, (void *)&option, sizeof(option)); - if(!strcmp(address, "*")) { - *address = 0; - } +#if defined(IPV6_V6ONLY) + + if(aip->ai_family == AF_INET6) { + setsockopt(nfd, IPPROTO_IPV6, IPV6_V6ONLY, (void *)&option, sizeof(option)); + } + +#else +#warning IPV6_V6ONLY not defined +#endif + + if(bind(nfd, aip->ai_addr, aip->ai_addrlen)) { + closesocket(nfd); + return -1; + } + + if(listen(nfd, 3)) { + logger(mesh, MESHLINK_ERROR, "System call `%s' failed: %s", "listen", sockstrerror(sockerrno)); + closesocket(nfd); + return -1; + } + + return nfd; +} + +int setup_udp_listen_socket(meshlink_handle_t *mesh, const struct addrinfo *aip) { + int nfd = socket(aip->ai_family, SOCK_DGRAM, IPPROTO_UDP); + + if(nfd == -1) { + return -1; + } + +#ifdef FD_CLOEXEC + fcntl(nfd, F_SETFD, FD_CLOEXEC); +#endif + +#ifdef O_NONBLOCK + int flags = fcntl(nfd, F_GETFL); + + if(fcntl(nfd, F_SETFL, flags | O_NONBLOCK) < 0) { + closesocket(nfd); + logger(mesh, MESHLINK_ERROR, "System call `%s' failed: %s", "fcntl", strerror(errno)); + return -1; + } + +#elif defined(WIN32) + unsigned long arg = 1; + + if(ioctlsocket(nfd, FIONBIO, &arg) != 0) { + closesocket(nfd); + logger(mesh, MESHLINK_ERROR, "Call to `%s' failed: %s", "ioctlsocket", sockstrerror(sockerrno)); + return -1; + } + +#endif + + int option = 1; + setsockopt(nfd, SOL_SOCKET, SO_REUSEADDR, (void *)&option, sizeof(option)); + setsockopt(nfd, SOL_SOCKET, SO_BROADCAST, (void *)&option, sizeof(option)); + +#if defined(IPV6_V6ONLY) + + if(aip->ai_family == AF_INET6) { + setsockopt(nfd, IPPROTO_IPV6, IPV6_V6ONLY, (void *)&option, sizeof(option)); } +#endif + +#if defined(IP_DONTFRAG) && !defined(IP_DONTFRAGMENT) +#define IP_DONTFRAGMENT IP_DONTFRAG +#endif + +#if defined(IP_MTU_DISCOVER) && defined(IP_PMTUDISC_DO) + option = IP_PMTUDISC_DO; + setsockopt(nfd, IPPROTO_IP, IP_MTU_DISCOVER, (void *)&option, sizeof(option)); +#elif defined(IP_DONTFRAGMENT) + option = 1; + setsockopt(nfd, IPPROTO_IP, IP_DONTFRAGMENT, (void *)&option, sizeof(option)); +#endif + + if(aip->ai_family == AF_INET6) { +#if defined(IPV6_MTU_DISCOVER) && defined(IPV6_PMTUDISC_DO) + option = IPV6_PMTUDISC_DO; + setsockopt(nfd, IPPROTO_IPV6, IPV6_MTU_DISCOVER, (void *)&option, sizeof(option)); +#elif defined(IPV6_DONTFRAG) + option = 1; + setsockopt(nfd, IPPROTO_IPV6, IPV6_DONTFRAG, (void *)&option, sizeof(option)); +#endif + } + + if(bind(nfd, aip->ai_addr, aip->ai_addrlen)) { + closesocket(nfd); + return -1; + } + + return nfd; +} + +/* + Add listening sockets. +*/ +static bool add_listen_sockets(meshlink_handle_t *mesh) { struct addrinfo *ai; struct addrinfo hint = { @@ -333,9 +425,7 @@ static bool add_listen_address(meshlink_handle_t *mesh, char *address, bool bind .ai_flags = AI_PASSIVE, }; - int err = getaddrinfo(address && *address ? address : NULL, port, &hint, &ai); - - free(address); + int err = getaddrinfo(NULL, mesh->myport, &hint, &ai); if(err || !ai) { logger(mesh, MESHLINK_ERROR, "System call `%s' failed: %s", "getaddrinfo", err == EAI_SYSTEM ? strerror(err) : gai_strerror(err)); @@ -348,11 +438,12 @@ static bool add_listen_address(meshlink_handle_t *mesh, char *address, bool bind // Ignore duplicate addresses bool found = false; - for(int i = 0; i < mesh->listen_sockets; i++) + for(int i = 0; i < mesh->listen_sockets; i++) { if(!memcmp(&mesh->listen_socket[i].sa, aip->ai_addr, aip->ai_addrlen)) { found = true; break; } + } if(found) { continue; @@ -363,29 +454,39 @@ static bool add_listen_address(meshlink_handle_t *mesh, char *address, bool bind return false; } - int tcp_fd = setup_listen_socket((sockaddr_t *) aip->ai_addr); + /* Try to bind to TCP */ - if(tcp_fd < 0) { - continue; + int tcp_fd = setup_tcp_listen_socket(mesh, aip); + + if(tcp_fd == -1) { + if(errno == EADDRINUSE) { + /* If this port is in use for any address family, avoid it. */ + success = false; + break; + } else { + continue; + } } - int udp_fd = setup_vpn_in_socket(mesh, (sockaddr_t *) aip->ai_addr); + /* If TCP worked, then we require that UDP works as well. */ - if(udp_fd < 0) { - close(tcp_fd); - continue; + int udp_fd = setup_udp_listen_socket(mesh, aip); + + if(udp_fd == -1) { + closesocket(tcp_fd); + success = false; + break; } io_add(&mesh->loop, &mesh->listen_socket[mesh->listen_sockets].tcp, handle_new_meta_connection, &mesh->listen_socket[mesh->listen_sockets], tcp_fd, IO_READ); io_add(&mesh->loop, &mesh->listen_socket[mesh->listen_sockets].udp, handle_incoming_vpn_data, &mesh->listen_socket[mesh->listen_sockets], udp_fd, IO_READ); - if(mesh->log_level >= MESHLINK_INFO) { + if(mesh->log_level <= MESHLINK_INFO) { char *hostname = sockaddr2hostname((sockaddr_t *) aip->ai_addr); logger(mesh, MESHLINK_INFO, "Listening on %s", hostname); free(hostname); } - mesh->listen_socket[mesh->listen_sockets].bindto = bindto; memcpy(&mesh->listen_socket[mesh->listen_sockets].sa, aip->ai_addr, aip->ai_addrlen); memcpy(&mesh->listen_socket[mesh->listen_sockets].broadcast_sa, aip->ai_addr, aip->ai_addrlen); @@ -402,6 +503,18 @@ static bool add_listen_address(meshlink_handle_t *mesh, char *address, bool bind } freeaddrinfo(ai); + + if(!success) { + for(int i = 0; i < mesh->listen_sockets; i++) { + io_del(&mesh->loop, &mesh->listen_socket[i].tcp); + io_del(&mesh->loop, &mesh->listen_socket[i].udp); + close(mesh->listen_socket[i].tcp.fd); + close(mesh->listen_socket[i].udp.fd); + } + + mesh->listen_sockets = 0; + } + return success; } @@ -416,30 +529,28 @@ bool setup_myself(meshlink_handle_t *mesh) { /* Done */ mesh->self->nexthop = mesh->self; - mesh->self->status.reachable = true; - mesh->self->last_state_change = mesh->loop.now.tv_sec; node_add(mesh, mesh->self); - graph(mesh); - - config_scan_all(mesh, "current", "hosts", load_node, NULL); + if(!config_scan_all(mesh, "current", "hosts", load_node, NULL)) { + logger(mesh, MESHLINK_WARNING, "Could not scan all host config files"); + } /* Open sockets */ mesh->listen_sockets = 0; - if(!add_listen_address(mesh, NULL, NULL)) { + if(!add_listen_sockets(mesh)) { if(strcmp(mesh->myport, "0")) { - logger(mesh, MESHLINK_INFO, "Could not bind to port %s, asking OS to choose one for us", mesh->myport); - free(mesh->myport); - mesh->myport = strdup("0"); + logger(mesh, MESHLINK_WARNING, "Could not bind to port %s, trying to find an alternative port", mesh->myport); - if(!mesh->myport) { - return false; + if(!check_port(mesh)) { + logger(mesh, MESHLINK_WARNING, "Could not bind to any port, trying to bind to port 0"); + free(mesh->myport); + mesh->myport = xstrdup("0"); } - if(!add_listen_address(mesh, NULL, NULL)) { + if(!add_listen_sockets(mesh)) { return false; } } else { @@ -455,6 +566,7 @@ bool setup_myself(meshlink_handle_t *mesh) { /* Done. */ mesh->last_config_check = mesh->loop.now.tv_sec; + mesh->last_unreachable = mesh->loop.now.tv_sec; return true; }