4ebb37a26c
Due to IF_NAMESIZE being a reused and conditionally defined macro, issues could arise from macro mismatches. In particular, in cases where opal/util/if.h is included, but net/if.h is not, IF_NAMESIZE will be 32. If net/if.h is included on Linux systems, IF_NAMESIZE will be 16. This can cause a mismatch when using the same macro on a system. Thus different parts of the code can have differring ideas on the size of a structure containing a char name[IF_NAMESIZE]. To avoid this error case, we avoid reusing the IF_NAMESIZE macro and instead define our own as OPAL_IF_NAMESIZE. Signed-off-by: William Zhang <wilzhang@amazon.com>
91 строка
1.9 KiB
C
91 строка
1.9 KiB
C
/*
|
|
* Copyright (c) 2016 Karol Mroz. All rights reserved.
|
|
* Copyright (c) 2016 Research Organization for Information Science
|
|
* and Technology (RIST). All rights reserved.
|
|
* Copyright (c) 2016 Cisco Systems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "opal_config.h"
|
|
|
|
#include <string.h>
|
|
#include <limits.h>
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
#ifdef HAVE_SYS_TYPES_H
|
|
#include <sys/types.h>
|
|
#endif
|
|
#ifdef HAVE_SYS_SOCKET_H
|
|
#include <sys/socket.h>
|
|
#endif
|
|
#ifdef HAVE_NET_IF_H
|
|
#include <net/if.h>
|
|
#endif
|
|
#ifdef HAVE_LINUX_ETHTOOL_H
|
|
#include <linux/ethtool.h>
|
|
#endif
|
|
#ifdef HAVE_SYS_IOCTL_H
|
|
#include <sys/ioctl.h>
|
|
#endif
|
|
#ifdef HAVE_LINUX_SOCKIOS_H
|
|
#include <linux/sockios.h>
|
|
#endif
|
|
|
|
#include "opal/util/ethtool.h"
|
|
#include "opal/util/if.h"
|
|
#include "opal/util/string_copy.h"
|
|
|
|
/*
|
|
* Obtain an appropriate bandwidth for the interface if_name. On Linux, we
|
|
* get this via an ioctl(). Elsewhere or in the error case, we return the
|
|
* speed as 0.
|
|
*/
|
|
unsigned int
|
|
opal_ethtool_get_speed (const char *if_name)
|
|
{
|
|
unsigned int speed = 0;
|
|
|
|
#if defined(HAVE_DECL_SIOCETHTOOL) && defined(HAVE_STRUCT_IFREQ) && defined(HAVE_STRUCT_ETHTOOL_CMD)
|
|
int sockfd;
|
|
struct ifreq ifr;
|
|
struct ethtool_cmd edata = {
|
|
.cmd = ETHTOOL_GSET,
|
|
};
|
|
|
|
sockfd = socket(PF_INET, SOCK_DGRAM, 0);
|
|
if (sockfd < 0) {
|
|
goto out;
|
|
}
|
|
|
|
memset(&ifr, 0, sizeof(struct ifreq));
|
|
opal_string_copy(ifr.ifr_name, if_name, OPAL_IF_NAMESIZE);
|
|
ifr.ifr_data = (char *)&edata;
|
|
|
|
if (ioctl(sockfd, SIOCETHTOOL, &ifr) < 0) {
|
|
goto out;
|
|
}
|
|
|
|
#if HAVE_DECL_ETHTOOL_CMD_SPEED
|
|
speed = ethtool_cmd_speed(&edata);
|
|
#elif defined(HAVE_STRUCT_ETHTOOL_CMD_SPEED_HI)
|
|
speed = (edata.speed_hi << 16) | edata.speed;
|
|
#else
|
|
speed = edata.speed;
|
|
#endif
|
|
if (UINT_MAX == speed) {
|
|
speed = 0;
|
|
}
|
|
|
|
out:
|
|
close(sockfd);
|
|
#endif
|
|
|
|
return speed;
|
|
}
|