From 8daa83a594a2e98f39d764422bfbdbc62c9efd44 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 19 Apr 2024 19:20:00 +0200 Subject: Adding upstream version 2:4.20.0+dfsg. Signed-off-by: Daniel Baumann --- lib/socket/interfaces.c | 435 +++++++++++++++++++++++++++++++++++++++++++++++ lib/socket/interfaces.h | 47 +++++ lib/socket/wscript | 7 + lib/socket/wscript_build | 7 + 4 files changed, 496 insertions(+) create mode 100644 lib/socket/interfaces.c create mode 100644 lib/socket/interfaces.h create mode 100644 lib/socket/wscript create mode 100644 lib/socket/wscript_build (limited to 'lib/socket') diff --git a/lib/socket/interfaces.c b/lib/socket/interfaces.c new file mode 100644 index 0000000..4908b0f --- /dev/null +++ b/lib/socket/interfaces.c @@ -0,0 +1,435 @@ +/* + Unix SMB/CIFS implementation. + return a list of network interfaces + Copyright (C) Andrew Tridgell 1998 + Copyright (C) Jeremy Allison 2007 + Copyright (C) Jelmer Vernooij 2007 + + 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 3 of the License, or + (at your option) any later version. + + 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 + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . +*/ + + +#include "includes.h" +#include "system/network.h" +#include "interfaces.h" +#include "lib/util/tsort.h" +#include "librpc/gen_ndr/ioctl.h" + +#ifdef HAVE_ETHTOOL +#include "linux/sockios.h" +#include "linux/ethtool.h" +#endif + +/**************************************************************************** + Create a struct sockaddr_storage with the netmask bits set to 1. +****************************************************************************/ + +bool make_netmask(struct sockaddr_storage *pss_out, + const struct sockaddr_storage *pss_in, + unsigned long masklen) +{ + *pss_out = *pss_in; + /* Now apply masklen bits of mask. */ +#if defined(HAVE_IPV6) + if (pss_in->ss_family == AF_INET6) { + char *p = (char *)&((struct sockaddr_in6 *)pss_out)->sin6_addr; + unsigned int i; + + if (masklen > 128) { + return false; + } + for (i = 0; masklen >= 8; masklen -= 8, i++) { + *p++ = 0xff; + } + /* Deal with the partial byte. */ + *p++ &= (0xff & ~(0xff>>masklen)); + i++; + for (;i < sizeof(struct in6_addr); i++) { + *p++ = '\0'; + } + return true; + } +#endif + if (pss_in->ss_family == AF_INET) { + if (masklen > 32) { + return false; + } + ((struct sockaddr_in *)pss_out)->sin_addr.s_addr = + htonl(((0xFFFFFFFFL >> masklen) ^ 0xFFFFFFFFL)); + return true; + } + return false; +} + +/**************************************************************************** + Create a struct sockaddr_storage set to the broadcast or network address from + an incoming sockaddr_storage. +****************************************************************************/ + +static void make_bcast_or_net(struct sockaddr_storage *pss_out, + const struct sockaddr_storage *pss_in, + const struct sockaddr_storage *nmask, + bool make_bcast_p) +{ + unsigned int i = 0, len = 0; + const char *pmask = NULL; + char *p = NULL; + *pss_out = *pss_in; + + /* Set all zero netmask bits to 1. */ +#if defined(HAVE_IPV6) + if (pss_in->ss_family == AF_INET6) { + p = (char *)&((struct sockaddr_in6 *)pss_out)->sin6_addr; + pmask = (const char *)&((const struct sockaddr_in6 *)nmask)->sin6_addr; + len = 16; + } +#endif + if (pss_in->ss_family == AF_INET) { + p = (char *)&((struct sockaddr_in *)pss_out)->sin_addr; + pmask = (const char *)&((const struct sockaddr_in *)nmask)->sin_addr; + len = 4; + } + + for (i = 0; i < len; i++, p++, pmask++) { + if (make_bcast_p) { + *p = (*p & *pmask) | (*pmask ^ 0xff); + } else { + /* make_net */ + *p = (*p & *pmask); + } + } +} + +void make_bcast(struct sockaddr_storage *pss_out, + const struct sockaddr_storage *pss_in, + const struct sockaddr_storage *nmask) +{ + make_bcast_or_net(pss_out, pss_in, nmask, true); +} + +void make_net(struct sockaddr_storage *pss_out, + const struct sockaddr_storage *pss_in, + const struct sockaddr_storage *nmask) +{ + make_bcast_or_net(pss_out, pss_in, nmask, false); +} + +#ifdef HAVE_ETHTOOL +static void query_iface_speed_from_name(const char *name, uint64_t *speed) +{ + int ret = 0; + struct ethtool_cmd ecmd; + struct ethtool_value edata; + struct ifreq ifr; + int fd; + + fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP); + if (fd == -1) { + DBG_ERR("Failed to open socket.\n"); + return; + } + + if (strlen(name) >= IF_NAMESIZE) { + DBG_ERR("Interface name too long.\n"); + goto done; + } + + ZERO_STRUCT(ifr); + strlcpy(ifr.ifr_name, name, IF_NAMESIZE); + + ifr.ifr_data = (void *)&edata; + ZERO_STRUCT(edata); + edata.cmd = ETHTOOL_GLINK; + ret = ioctl(fd, SIOCETHTOOL, &ifr); + if (ret == -1) { + goto done; + } + if (edata.data == 0) { + /* no link detected */ + *speed = 0; + goto done; + } + + ifr.ifr_data = (void *)&ecmd; + ZERO_STRUCT(ecmd); + ecmd.cmd = ETHTOOL_GSET; + ret = ioctl(fd, SIOCETHTOOL, &ifr); + if (ret == -1) { + goto done; + } + *speed = ((uint64_t)ethtool_cmd_speed(&ecmd)) * 1000 * 1000; + +done: + (void)close(fd); +} + +static void query_iface_rx_queues_from_name(const char *name, + uint64_t *rx_queues) +{ + int ret = 0; + struct ethtool_rxnfc rxcmd; + struct ifreq ifr; + int fd; + + fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP); + if (fd == -1) { + DBG_ERR("Failed to open socket.\n"); + return; + } + + if (strlen(name) >= IF_NAMESIZE) { + DBG_ERR("Interface name too long.\n"); + goto done; + } + + ZERO_STRUCT(ifr); + strlcpy(ifr.ifr_name, name, IF_NAMESIZE); + + ifr.ifr_data = (void *)&rxcmd; + ZERO_STRUCT(rxcmd); + rxcmd.cmd = ETHTOOL_GRXRINGS; + ret = ioctl(fd, SIOCETHTOOL, &ifr); + if (ret == -1) { + goto done; + } + + *rx_queues = rxcmd.data; + +done: + (void)close(fd); +} +#endif + +/**************************************************************************** + Try the "standard" getifaddrs/freeifaddrs interfaces. + Also gets IPv6 interfaces. +****************************************************************************/ + +/**************************************************************************** + Get the netmask address for a local interface. +****************************************************************************/ + +static int _get_interfaces(TALLOC_CTX *mem_ctx, struct iface_struct **pifaces) +{ + struct iface_struct *ifaces; + struct ifaddrs *iflist = NULL; + struct ifaddrs *ifptr = NULL; + int count; + int total = 0; + size_t copy_size; + + if (getifaddrs(&iflist) < 0) { + return -1; + } + + count = 0; + for (ifptr = iflist; ifptr != NULL; ifptr = ifptr->ifa_next) { + if (!ifptr->ifa_addr || !ifptr->ifa_netmask) { + continue; + } + if (!(ifptr->ifa_flags & IFF_UP)) { + continue; + } + count += 1; + } + + ifaces = talloc_array(mem_ctx, struct iface_struct, count); + if (ifaces == NULL) { + errno = ENOMEM; + return -1; + } + + /* Loop through interfaces, looking for given IP address */ + for (ifptr = iflist; ifptr != NULL; ifptr = ifptr->ifa_next) { + uint64_t if_speed = 1000 * 1000 * 1000; /* 1Gbps */ + uint64_t rx_queues = 1; + + if (!ifptr->ifa_addr || !ifptr->ifa_netmask) { + continue; + } + + /* Check the interface is up. */ + if (!(ifptr->ifa_flags & IFF_UP)) { + continue; + } + + memset(&ifaces[total], '\0', sizeof(ifaces[total])); + + copy_size = sizeof(struct sockaddr_in); + + ifaces[total].flags = ifptr->ifa_flags; + +#if defined(HAVE_IPV6) + if (ifptr->ifa_addr->sa_family == AF_INET6) { + copy_size = sizeof(struct sockaddr_in6); + } +#endif + + memcpy(&ifaces[total].ip, ifptr->ifa_addr, copy_size); + memcpy(&ifaces[total].netmask, ifptr->ifa_netmask, copy_size); + + /* calculate broadcast address */ +#if defined(HAVE_IPV6) + if (ifptr->ifa_addr->sa_family == AF_INET6) { + struct sockaddr_in6 *sin6 = + (struct sockaddr_in6 *)ifptr->ifa_addr; + struct in6_addr *in6 = + (struct in6_addr *)&sin6->sin6_addr; + + if (IN6_IS_ADDR_LINKLOCAL(in6) || IN6_IS_ADDR_V4COMPAT(in6)) { + continue; + } + /* IPv6 does not have broadcast it uses multicast. */ + memset(&ifaces[total].bcast, '\0', copy_size); + } else +#endif + if (ifaces[total].flags & (IFF_BROADCAST|IFF_LOOPBACK)) { + make_bcast(&ifaces[total].bcast, + &ifaces[total].ip, + &ifaces[total].netmask); + } else if ((ifaces[total].flags & IFF_POINTOPOINT) && + ifptr->ifa_dstaddr ) { + memcpy(&ifaces[total].bcast, + ifptr->ifa_dstaddr, + copy_size); + } else { + continue; + } + + ifaces[total].if_index = if_nametoindex(ifptr->ifa_name); + if (ifaces[total].if_index == 0) { + DBG_ERR("Failed to retrieve interface index for '%s': " + "%s\n", ifptr->ifa_name, strerror(errno)); + } + +#ifdef HAVE_ETHTOOL + query_iface_speed_from_name(ifptr->ifa_name, &if_speed); + query_iface_rx_queues_from_name(ifptr->ifa_name, &rx_queues); +#endif + ifaces[total].linkspeed = if_speed; + ifaces[total].capability = FSCTL_NET_IFACE_NONE_CAPABLE; + if (rx_queues > 1) { + ifaces[total].capability |= FSCTL_NET_IFACE_RSS_CAPABLE; + } + + if (strlcpy(ifaces[total].name, ifptr->ifa_name, + sizeof(ifaces[total].name)) >= + sizeof(ifaces[total].name)) { + /* Truncation ! Ignore. */ + continue; + } + total++; + } + + freeifaddrs(iflist); + + *pifaces = ifaces; + return total; +} + +static int iface_comp(struct iface_struct *i1, struct iface_struct *i2) +{ + int r; + +#if defined(HAVE_IPV6) + /* + * If we have IPv6 - sort these interfaces lower + * than any IPv4 ones. + */ + if (i1->ip.ss_family == AF_INET6 && + i2->ip.ss_family == AF_INET) { + return -1; + } else if (i1->ip.ss_family == AF_INET && + i2->ip.ss_family == AF_INET6) { + return 1; + } + + if (i1->ip.ss_family == AF_INET6) { + struct sockaddr_in6 *s1 = (struct sockaddr_in6 *)&i1->ip; + struct sockaddr_in6 *s2 = (struct sockaddr_in6 *)&i2->ip; + + r = memcmp(&s1->sin6_addr, + &s2->sin6_addr, + sizeof(struct in6_addr)); + if (r) { + return r; + } + + s1 = (struct sockaddr_in6 *)&i1->netmask; + s2 = (struct sockaddr_in6 *)&i2->netmask; + + r = memcmp(&s1->sin6_addr, + &s2->sin6_addr, + sizeof(struct in6_addr)); + if (r) { + return r; + } + } +#endif + + /* AIX uses __ss_family instead of ss_family inside of + sockaddr_storage. Instead of trying to figure out which field to + use, we can just cast it to a sockaddr. + */ + + if (((struct sockaddr *)&i1->ip)->sa_family == AF_INET) { + struct sockaddr_in *s1 = (struct sockaddr_in *)&i1->ip; + struct sockaddr_in *s2 = (struct sockaddr_in *)&i2->ip; + + r = ntohl(s1->sin_addr.s_addr) - + ntohl(s2->sin_addr.s_addr); + if (r) { + return r; + } + + s1 = (struct sockaddr_in *)&i1->netmask; + s2 = (struct sockaddr_in *)&i2->netmask; + + return ntohl(s1->sin_addr.s_addr) - + ntohl(s2->sin_addr.s_addr); + } + return 0; +} + +/* this wrapper is used to remove duplicates from the interface list generated + above */ +int get_interfaces(TALLOC_CTX *mem_ctx, struct iface_struct **pifaces) +{ + struct iface_struct *ifaces = NULL; + int total, i, j; + + total = _get_interfaces(mem_ctx, &ifaces); + /* If we have an error, no interface or just one we can leave */ + if (total <= 1) { + *pifaces = ifaces; + return total; + } + + /* now we need to remove duplicates */ + TYPESAFE_QSORT(ifaces, total, iface_comp); + + for (i=1;i. +*/ + +#include "system/network.h" + +struct iface_struct { + char name[16]; + int flags; + struct sockaddr_storage ip; + struct sockaddr_storage netmask; + struct sockaddr_storage bcast; + uint32_t if_index; + uint64_t linkspeed; + uint32_t capability; +}; + +struct interface; + +bool make_netmask(struct sockaddr_storage *pss_out, + const struct sockaddr_storage *pss_in, + unsigned long masklen); +void make_bcast(struct sockaddr_storage *pss_out, + const struct sockaddr_storage *pss_in, + const struct sockaddr_storage *nmask); +void make_net(struct sockaddr_storage *pss_out, + const struct sockaddr_storage *pss_in, + const struct sockaddr_storage *nmask); + +int get_interfaces(TALLOC_CTX *mem_ctx, struct iface_struct **pifaces); diff --git a/lib/socket/wscript b/lib/socket/wscript new file mode 100644 index 0000000..d8c269a --- /dev/null +++ b/lib/socket/wscript @@ -0,0 +1,7 @@ +#!/usr/bin/env python + +def configure(conf): + conf.CHECK_HEADERS('linux/sockios.h linux/ethtool.h') + if (conf.CONFIG_SET('HAVE_LINUX_SOCKIOS_H') and \ + conf.CONFIG_SET('HAVE_LINUX_ETHTOOL_H')): + conf.DEFINE('HAVE_ETHTOOL', 1) diff --git a/lib/socket/wscript_build b/lib/socket/wscript_build new file mode 100644 index 0000000..92e5aef --- /dev/null +++ b/lib/socket/wscript_build @@ -0,0 +1,7 @@ +#!/usr/bin/env python + +bld.SAMBA_LIBRARY('interfaces', + source='interfaces.c', + deps='samba-util samba-debug', + private_library=True + ) -- cgit v1.2.3