diff options
Diffstat (limited to 'src/tcp_address.cpp')
-rw-r--r-- | src/tcp_address.cpp | 602 |
1 files changed, 0 insertions, 602 deletions
diff --git a/src/tcp_address.cpp b/src/tcp_address.cpp deleted file mode 100644 index 1c75812..0000000 --- a/src/tcp_address.cpp +++ /dev/null @@ -1,602 +0,0 @@ -/* - Copyright (c) 2007-2013 Contributors as noted in the AUTHORS file - - This file is part of 0MQ. - - 0MQ is free software; you can redistribute it and/or modify it under - the terms of the GNU Lesser General Public License as published by - the Free Software Foundation; either version 3 of the License, or - (at your option) any later version. - - 0MQ 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 - GNU Lesser General Public License for more details. - - You should have received a copy of the GNU Lesser General Public License - along with this program. If not, see <http://www.gnu.org/licenses/>. -*/ - -#include <string> -#include <sstream> - -#include "tcp_address.hpp" -#include "platform.hpp" -#include "stdint.hpp" -#include "err.hpp" -#include "ip.hpp" - -#ifdef ZMQ_HAVE_WINDOWS -#include "windows.hpp" -#else -#include <sys/types.h> -#include <arpa/inet.h> -#include <netinet/tcp.h> -#include <netdb.h> -#endif - -// Some platforms (notably Darwin/OSX and NetBSD) do not define all AI_ -// flags for getaddrinfo(). This can be worked around safely by defining -// these to 0. -#ifndef AI_ADDRCONFIG -#define AI_ADDRCONFIG 0 -#endif - -#if defined ZMQ_HAVE_SOLARIS - -#include <sys/sockio.h> -#include <net/if.h> -#include <unistd.h> -#include <stdlib.h> - -// On Solaris platform, network interface name can be queried by ioctl. -int zmq::tcp_address_t::resolve_nic_name (const char *nic_, bool ipv6_) -{ - // TODO: Unused parameter, IPv6 support not implemented for Solaris. - (void) ipv6_; - - // Create a socket. - int fd = open_socket (AF_INET, SOCK_DGRAM, 0); - errno_assert (fd != -1); - - // Retrieve number of interfaces. - lifnum ifn; - ifn.lifn_family = AF_INET; - ifn.lifn_flags = 0; - int rc = ioctl (fd, SIOCGLIFNUM, (char*) &ifn); - errno_assert (rc != -1); - - // Allocate memory to get interface names. - size_t ifr_size = sizeof (struct lifreq) * ifn.lifn_count; - char *ifr = (char*) malloc (ifr_size); - alloc_assert (ifr); - - // Retrieve interface names. - lifconf ifc; - ifc.lifc_family = AF_INET; - ifc.lifc_flags = 0; - ifc.lifc_len = ifr_size; - ifc.lifc_buf = ifr; - rc = ioctl (fd, SIOCGLIFCONF, (char*) &ifc); - errno_assert (rc != -1); - - // Find the interface with the specified name and AF_INET family. - bool found = false; - lifreq *ifrp = ifc.lifc_req; - for (int n = 0; n < (int) (ifc.lifc_len / sizeof (lifreq)); - n ++, ifrp ++) { - if (!strcmp (nic_, ifrp->lifr_name)) { - rc = ioctl (fd, SIOCGLIFADDR, (char*) ifrp); - errno_assert (rc != -1); - if (ifrp->lifr_addr.ss_family == AF_INET) { - address.ipv4 = *(sockaddr_in*) &ifrp->lifr_addr; - found = true; - break; - } - } - } - - // Clean-up. - free (ifr); - close (fd); - - if (!found) { - errno = ENODEV; - return -1; - } - return 0; -} - -#elif defined ZMQ_HAVE_AIX || defined ZMQ_HAVE_HPUX || defined ZMQ_HAVE_ANDROID - -#include <sys/types.h> -#include <unistd.h> -#include <sys/ioctl.h> -#include <net/if.h> - -int zmq::tcp_address_t::resolve_nic_name (const char *nic_, bool ipv6_) -{ - // TODO: Unused parameter, IPv6 support not implemented for AIX or HP/UX. - (void) ipv6_; - - // Create a socket. - int sd = open_socket (AF_INET, SOCK_DGRAM, 0); - errno_assert (sd != -1); - - struct ifreq ifr; - - // Copy interface name for ioctl get. - strncpy (ifr.ifr_name, nic_, sizeof (ifr.ifr_name)); - - // Fetch interface address. - int rc = ioctl (sd, SIOCGIFADDR, (caddr_t) &ifr, sizeof (struct ifreq)); - - // Clean up. - close (sd); - - if (rc == -1) { - errno = ENODEV; - return -1; - } - memcpy (&address.ipv4.sin_addr, &((sockaddr_in*) &ifr.ifr_addr)->sin_addr, - sizeof (in_addr)); - - return 0; -} - -#elif ((defined ZMQ_HAVE_LINUX || defined ZMQ_HAVE_FREEBSD ||\ - defined ZMQ_HAVE_OSX || defined ZMQ_HAVE_OPENBSD ||\ - defined ZMQ_HAVE_QNXNTO || defined ZMQ_HAVE_NETBSD)\ - && defined ZMQ_HAVE_IFADDRS) - -#include <ifaddrs.h> - -// On these platforms, network interface name can be queried -// using getifaddrs function. -int zmq::tcp_address_t::resolve_nic_name (const char *nic_, bool ipv6_) -{ - // Get the addresses. - ifaddrs *ifa = NULL; - int rc = getifaddrs (&ifa); - errno_assert (rc == 0); - zmq_assert (ifa != NULL); - - // Find the corresponding network interface. - bool found = false; - for (ifaddrs *ifp = ifa; ifp != NULL ;ifp = ifp->ifa_next) - { - if (ifp->ifa_addr == NULL) - continue; - - int family = ifp->ifa_addr->sa_family; - if ((family == AF_INET || (ipv6_ && family == AF_INET6)) - && !strcmp (nic_, ifp->ifa_name)) { - memcpy (&address, ifp->ifa_addr, - (family == AF_INET) ? sizeof (struct sockaddr_in) - : sizeof (struct sockaddr_in6)); - found = true; - break; - } - } - - // Clean-up; - freeifaddrs (ifa); - - if (!found) { - errno = ENODEV; - return -1; - } - return 0; -} - -#else - -// On other platforms we assume there are no sane interface names. -// This is true especially of Windows. -int zmq::tcp_address_t::resolve_nic_name (const char *nic_, bool ipv6_) -{ - // All unused parameters. - (void) nic_; - (void) ipv6_; - - errno = ENODEV; - return -1; -} - -#endif - -int zmq::tcp_address_t::resolve_interface (const char *interface_, bool ipv6_) -{ - // Initialize temporary output pointers with storage address. - sockaddr_storage ss; - sockaddr *out_addr = (sockaddr*) &ss; - size_t out_addrlen; - - // Initialise IP-format family/port and populate temporary output pointers - // with the address. - if (ipv6_) { - sockaddr_in6 ip6_addr; - memset (&ip6_addr, 0, sizeof (ip6_addr)); - ip6_addr.sin6_family = AF_INET6; - memcpy (&ip6_addr.sin6_addr, &in6addr_any, sizeof (in6addr_any)); - out_addrlen = sizeof ip6_addr; - memcpy (out_addr, &ip6_addr, out_addrlen); - } - else { - sockaddr_in ip4_addr; - memset (&ip4_addr, 0, sizeof (ip4_addr)); - ip4_addr.sin_family = AF_INET; - ip4_addr.sin_addr.s_addr = htonl (INADDR_ANY); - out_addrlen = sizeof ip4_addr; - memcpy (out_addr, &ip4_addr, out_addrlen); - } - // "*" resolves to INADDR_ANY or in6addr_any. - if (strcmp (interface_, "*") == 0) { - zmq_assert (out_addrlen <= sizeof address); - memcpy (&address, out_addr, out_addrlen); - return 0; - } - - // Try to resolve the string as a NIC name. - int rc = resolve_nic_name (interface_, ipv6_); - if (rc != 0 && errno != ENODEV) - return rc; - if (rc == 0) - return 0; - - // There's no such interface name. Assume literal address. -#if defined ZMQ_HAVE_OPENVMS && defined __ia64 - __addrinfo64 *res = NULL; - __addrinfo64 req; -#else - addrinfo *res = NULL; - addrinfo req; -#endif - memset (&req, 0, sizeof (req)); - - // Choose IPv4 or IPv6 protocol family. Note that IPv6 allows for - // IPv4-in-IPv6 addresses. - req.ai_family = ipv6_? AF_INET6: AF_INET; - - // Arbitrary, not used in the output, but avoids duplicate results. - req.ai_socktype = SOCK_STREAM; - - // Restrict hostname/service to literals to avoid any DNS lookups or - // service-name irregularity due to indeterminate socktype. - req.ai_flags = AI_PASSIVE | AI_NUMERICHOST; - -#if defined AI_V4MAPPED && !defined ZMQ_HAVE_FREEBSD - // In this API we only require IPv4-mapped addresses when - // no native IPv6 interfaces are available (~AI_ALL). - // This saves an additional DNS roundtrip for IPv4 addresses. - // Note: While the AI_V4MAPPED flag is defined on FreeBSD system, - // it is not supported here. See libzmq issue #331. - if (req.ai_family == AF_INET6) - req.ai_flags |= AI_V4MAPPED; -#endif - - // Resolve the literal address. Some of the error info is lost in case - // of error, however, there's no way to report EAI errors via errno. - rc = getaddrinfo (interface_, NULL, &req, &res); - if (rc) { - errno = ENODEV; - return -1; - } - - // Use the first result. - zmq_assert (res != NULL); - zmq_assert ((size_t) (res->ai_addrlen) <= sizeof (address)); - memcpy (&address, res->ai_addr, res->ai_addrlen); - - // Cleanup getaddrinfo after copying the possibly referenced result. - freeaddrinfo (res); - - return 0; -} - -int zmq::tcp_address_t::resolve_hostname (const char *hostname_, bool ipv6_) -{ - // Set up the query. -#if defined ZMQ_HAVE_OPENVMS && defined __ia64 && __INITIAL_POINTER_SIZE == 64 - __addrinfo64 req; -#else - addrinfo req; -#endif - memset (&req, 0, sizeof (req)); - - // Choose IPv4 or IPv6 protocol family. Note that IPv6 allows for - // IPv4-in-IPv6 addresses. - req.ai_family = ipv6_? AF_INET6: AF_INET; - - // Need to choose one to avoid duplicate results from getaddrinfo() - this - // doesn't really matter, since it's not included in the addr-output. - req.ai_socktype = SOCK_STREAM; - -#if defined AI_V4MAPPED && !defined ZMQ_HAVE_FREEBSD - // In this API we only require IPv4-mapped addresses when - // no native IPv6 interfaces are available. - // This saves an additional DNS roundtrip for IPv4 addresses. - // Note: While the AI_V4MAPPED flag is defined on FreeBSD system, - // it is not supported here. See libzmq issue #331. - if (req.ai_family == AF_INET6) - req.ai_flags |= AI_V4MAPPED; -#endif - - // Resolve host name. Some of the error info is lost in case of error, - // however, there's no way to report EAI errors via errno. -#if defined ZMQ_HAVE_OPENVMS && defined __ia64 && __INITIAL_POINTER_SIZE == 64 - __addrinfo64 *res; -#else - addrinfo *res; -#endif - int rc = getaddrinfo (hostname_, NULL, &req, &res); - if (rc) { - switch (rc) { - case EAI_MEMORY: - errno = ENOMEM; - break; - default: - errno = EINVAL; - break; - } - return -1; - } - - // Copy first result to output addr with hostname and service. - zmq_assert ((size_t) (res->ai_addrlen) <= sizeof (address)); - memcpy (&address, res->ai_addr, res->ai_addrlen); - - freeaddrinfo (res); - - return 0; -} - -zmq::tcp_address_t::tcp_address_t () -{ - memset (&address, 0, sizeof (address)); -} - -zmq::tcp_address_t::tcp_address_t (const sockaddr *sa, socklen_t sa_len) -{ - zmq_assert(sa && sa_len > 0); - - memset (&address, 0, sizeof (address)); - if (sa->sa_family == AF_INET && sa_len >= (socklen_t) sizeof (address.ipv4)) - memcpy(&address.ipv4, sa, sizeof (address.ipv4)); - else - if (sa->sa_family == AF_INET6 && sa_len >= (socklen_t) sizeof (address.ipv6)) - memcpy(&address.ipv6, sa, sizeof (address.ipv6)); -} - -zmq::tcp_address_t::~tcp_address_t () -{ -} - -int zmq::tcp_address_t::resolve (const char *name_, bool local_, bool ipv6_) -{ - // Find the ':' at end that separates address from the port number. - const char *delimiter = strrchr (name_, ':'); - if (!delimiter) { - errno = EINVAL; - return -1; - } - // Separate the address/port. - std::string addr_str (name_, delimiter - name_); - std::string port_str (delimiter + 1); - - // Remove square brackets around the address, if any. - if (addr_str.size () >= 2 && addr_str [0] == '[' && - addr_str [addr_str.size () - 1] == ']') - addr_str = addr_str.substr (1, addr_str.size () - 2); - - // Allow 0 specifically, to detect invalid port error in atoi if not - uint16_t port; - if (port_str == "*" || port_str == "0") - // Resolve wildcard to 0 to allow autoselection of port - port = 0; - else { - // Parse the port number (0 is not a valid port). - port = (uint16_t) atoi (port_str.c_str ()); - if (port == 0) { - errno = EINVAL; - return -1; - } - } - - // Resolve the IP address. - int rc; - if (local_) - rc = resolve_interface (addr_str.c_str (), ipv6_); - else - rc = resolve_hostname (addr_str.c_str (), ipv6_); - if (rc != 0) - return -1; - - // Set the port into the address structure. - if (address.generic.sa_family == AF_INET6) - address.ipv6.sin6_port = htons (port); - else - address.ipv4.sin_port = htons (port); - - return 0; -} - -int zmq::tcp_address_t::to_string (std::string &addr_) -{ - if (address.generic.sa_family != AF_INET - && address.generic.sa_family != AF_INET6) { - addr_.clear (); - return -1; - } - - // not using service resolv because of https://github.com/zeromq/libzmq/commit/1824574f9b5a8ce786853320e3ea09fe1f822bc4 - char hbuf[NI_MAXHOST]; - int rc = getnameinfo (addr (), addrlen (), hbuf, sizeof (hbuf), NULL, 0, NI_NUMERICHOST); - if (rc != 0) { - addr_.clear (); - return rc; - } - - if (address.generic.sa_family == AF_INET6) { - std::stringstream s; - s << "tcp://[" << hbuf << "]:" << ntohs (address.ipv6.sin6_port); - addr_ = s.str (); - } - else { - std::stringstream s; - s << "tcp://" << hbuf << ":" << ntohs (address.ipv4.sin_port); - addr_ = s.str (); - }; - return 0; -} - -const sockaddr *zmq::tcp_address_t::addr () const -{ - return &address.generic; -} - -socklen_t zmq::tcp_address_t::addrlen () const -{ - if (address.generic.sa_family == AF_INET6) - return (socklen_t) sizeof (address.ipv6); - else - return (socklen_t) sizeof (address.ipv4); -} - -#if defined ZMQ_HAVE_WINDOWS -unsigned short zmq::tcp_address_t::family () const -#else -sa_family_t zmq::tcp_address_t::family () const -#endif -{ - return address.generic.sa_family; -} - -zmq::tcp_address_mask_t::tcp_address_mask_t () : - tcp_address_t () -{ - address_mask = -1; -} - -int zmq::tcp_address_mask_t::mask () const -{ - return address_mask; -} - -int zmq::tcp_address_mask_t::resolve (const char *name_, bool ipv6_) -{ - // Find '/' at the end that separates address from the cidr mask number. - // Allow empty mask clause and threat it like '/32' for ipv4 or '/128' for ipv6. - std::string addr_str, mask_str; - const char *delimiter = strrchr (name_, '/'); - if (delimiter != NULL) { - addr_str.assign (name_, delimiter - name_); - mask_str.assign (delimiter + 1); - if (mask_str.empty ()) { - errno = EINVAL; - return -1; - } - } - else - addr_str.assign (name_); - - // Parse address part using standard routines. - int rc = tcp_address_t::resolve_hostname (addr_str.c_str (), ipv6_); - if (rc != 0) - return rc; - - // Parse the cidr mask number. - if (mask_str.empty ()) { - if (address.generic.sa_family == AF_INET6) - address_mask = 128; - else - address_mask = 32; - } - else - if (mask_str == "0") { - address_mask = 0; - } - else { - int mask = atoi (mask_str.c_str ()); - if ( - (mask < 1) || - (address.generic.sa_family == AF_INET6 && mask > 128) || - (address.generic.sa_family != AF_INET6 && mask > 32) - ) { - errno = EINVAL; - return -1; - } - address_mask = mask; - } - - return 0; -} - -int zmq::tcp_address_mask_t::to_string (std::string &addr_) -{ - if (address.generic.sa_family != AF_INET && address.generic.sa_family != AF_INET6) { - addr_.clear (); - return -1; - } - if (address_mask == -1) { - addr_.clear (); - return -1; - } - - char hbuf[NI_MAXHOST]; - int rc = getnameinfo (addr (), addrlen (), hbuf, sizeof (hbuf), NULL, 0, NI_NUMERICHOST); - if (rc != 0) { - addr_.clear (); - return rc; - } - - if (address.generic.sa_family == AF_INET6) { - std::stringstream s; - s << "[" << hbuf << "]/" << address_mask; - addr_ = s.str (); - } - else { - std::stringstream s; - s << hbuf << "/" << address_mask; - addr_ = s.str (); - }; - return 0; -} - -bool zmq::tcp_address_mask_t::match_address (const struct sockaddr *ss, const socklen_t ss_len) const -{ - zmq_assert (address_mask != -1 && ss != NULL && ss_len >= (socklen_t) sizeof (struct sockaddr)); - - if (ss->sa_family != address.generic.sa_family) - return false; - - if (address_mask > 0) { - int mask; - const uint8_t *our_bytes, *their_bytes; - if (ss->sa_family == AF_INET6) { - zmq_assert (ss_len == sizeof (struct sockaddr_in6)); - their_bytes = (const uint8_t *) &(((const struct sockaddr_in6 *) ss)->sin6_addr); - our_bytes = (const uint8_t *) &address.ipv6.sin6_addr; - mask = sizeof (struct in6_addr) * 8; - } - else { - zmq_assert (ss_len == sizeof (struct sockaddr_in)); - their_bytes = (const uint8_t *) &(((const struct sockaddr_in *) ss)->sin_addr); - our_bytes = (const uint8_t *) &address.ipv4.sin_addr; - mask = sizeof (struct in_addr) * 8; - } - if (address_mask < mask) mask = address_mask; - - size_t full_bytes = mask / 8; - if (memcmp(our_bytes, their_bytes, full_bytes)) - return false; - - uint8_t last_byte_bits = (0xffU << (8 - (mask % 8))) & 0xffU; - if (last_byte_bits) { - if ((their_bytes[full_bytes] & last_byte_bits) != (our_bytes[full_bytes] & last_byte_bits)) - return false; - } - } - - return true; -} |