oshi.software.os.mac.MacInternetProtocolStats Maven / Gradle / Ivy
The newest version!
/*
* Copyright 2020-2022 The OSHI Project Contributors
* SPDX-License-Identifier: MIT
*/
package oshi.software.os.mac;
import static com.sun.jna.platform.mac.SystemB.INT_SIZE;
import static com.sun.jna.platform.mac.SystemB.PROC_ALL_PIDS;
import static oshi.jna.platform.mac.SystemB.AF_INET;
import static oshi.jna.platform.mac.SystemB.AF_INET6;
import static oshi.jna.platform.mac.SystemB.PROC_PIDFDSOCKETINFO;
import static oshi.jna.platform.mac.SystemB.PROC_PIDLISTFDS;
import static oshi.jna.platform.mac.SystemB.PROX_FDTYPE_SOCKET;
import static oshi.jna.platform.mac.SystemB.SOCKINFO_IN;
import static oshi.jna.platform.mac.SystemB.SOCKINFO_TCP;
import static oshi.software.os.InternetProtocolStats.TcpState.CLOSED;
import static oshi.software.os.InternetProtocolStats.TcpState.CLOSE_WAIT;
import static oshi.software.os.InternetProtocolStats.TcpState.CLOSING;
import static oshi.software.os.InternetProtocolStats.TcpState.ESTABLISHED;
import static oshi.software.os.InternetProtocolStats.TcpState.FIN_WAIT_1;
import static oshi.software.os.InternetProtocolStats.TcpState.FIN_WAIT_2;
import static oshi.software.os.InternetProtocolStats.TcpState.LAST_ACK;
import static oshi.software.os.InternetProtocolStats.TcpState.LISTEN;
import static oshi.software.os.InternetProtocolStats.TcpState.NONE;
import static oshi.software.os.InternetProtocolStats.TcpState.SYN_RECV;
import static oshi.software.os.InternetProtocolStats.TcpState.SYN_SENT;
import static oshi.software.os.InternetProtocolStats.TcpState.TIME_WAIT;
import static oshi.software.os.InternetProtocolStats.TcpState.UNKNOWN;
import static oshi.util.Memoizer.defaultExpiration;
import static oshi.util.Memoizer.memoize;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Supplier;
import com.sun.jna.Memory;
import oshi.annotation.concurrent.ThreadSafe;
import oshi.driver.unix.NetStat;
import oshi.jna.platform.mac.SystemB;
import oshi.jna.platform.mac.SystemB.InSockInfo;
import oshi.jna.platform.mac.SystemB.ProcFdInfo;
import oshi.jna.platform.mac.SystemB.SocketFdInfo;
import oshi.jna.platform.unix.CLibrary.BsdIp6stat;
import oshi.jna.platform.unix.CLibrary.BsdIpstat;
import oshi.jna.platform.unix.CLibrary.BsdTcpstat;
import oshi.jna.platform.unix.CLibrary.BsdUdpstat;
import oshi.software.common.AbstractInternetProtocolStats;
import oshi.util.ParseUtil;
import oshi.util.platform.mac.SysctlUtil;
import oshi.util.tuples.Pair;
/**
* Internet Protocol Stats implementation
*/
@ThreadSafe
public class MacInternetProtocolStats extends AbstractInternetProtocolStats {
private boolean isElevated;
public MacInternetProtocolStats(boolean elevated) {
this.isElevated = elevated;
}
private Supplier> establishedv4v6 = memoize(NetStat::queryTcpnetstat, defaultExpiration());
private Supplier tcpstat = memoize(MacInternetProtocolStats::queryTcpstat, defaultExpiration());
private Supplier udpstat = memoize(MacInternetProtocolStats::queryUdpstat, defaultExpiration());
// With elevated permissions use tcpstat only
// Backup estimate get ipstat and subtract off udp
private Supplier ipstat = memoize(MacInternetProtocolStats::queryIpstat, defaultExpiration());
private Supplier ip6stat = memoize(MacInternetProtocolStats::queryIp6stat, defaultExpiration());
@Override
public TcpStats getTCPv4Stats() {
BsdTcpstat tcp = tcpstat.get();
if (this.isElevated) {
return new TcpStats(establishedv4v6.get().getA(), ParseUtil.unsignedIntToLong(tcp.tcps_connattempt),
ParseUtil.unsignedIntToLong(tcp.tcps_accepts), ParseUtil.unsignedIntToLong(tcp.tcps_conndrops),
ParseUtil.unsignedIntToLong(tcp.tcps_drops), ParseUtil.unsignedIntToLong(tcp.tcps_sndpack),
ParseUtil.unsignedIntToLong(tcp.tcps_rcvpack), ParseUtil.unsignedIntToLong(tcp.tcps_sndrexmitpack),
ParseUtil.unsignedIntToLong(
tcp.tcps_rcvbadsum + tcp.tcps_rcvbadoff + tcp.tcps_rcvmemdrop + tcp.tcps_rcvshort),
0L);
}
BsdIpstat ip = ipstat.get();
BsdUdpstat udp = udpstat.get();
return new TcpStats(establishedv4v6.get().getA(), ParseUtil.unsignedIntToLong(tcp.tcps_connattempt),
ParseUtil.unsignedIntToLong(tcp.tcps_accepts), ParseUtil.unsignedIntToLong(tcp.tcps_conndrops),
ParseUtil.unsignedIntToLong(tcp.tcps_drops),
Math.max(0L, ParseUtil.unsignedIntToLong(ip.ips_delivered - udp.udps_opackets)),
Math.max(0L, ParseUtil.unsignedIntToLong(ip.ips_total - udp.udps_ipackets)),
ParseUtil.unsignedIntToLong(tcp.tcps_sndrexmitpack),
Math.max(0L, ParseUtil.unsignedIntToLong(ip.ips_badsum + ip.ips_tooshort + ip.ips_toosmall
+ ip.ips_badhlen + ip.ips_badlen - udp.udps_hdrops + udp.udps_badsum + udp.udps_badlen)),
0L);
}
@Override
public TcpStats getTCPv6Stats() {
BsdIp6stat ip6 = ip6stat.get();
BsdUdpstat udp = udpstat.get();
return new TcpStats(establishedv4v6.get().getB(), 0L, 0L, 0L, 0L,
ip6.ip6s_localout - ParseUtil.unsignedIntToLong(udp.udps_snd6_swcsum),
ip6.ip6s_total - ParseUtil.unsignedIntToLong(udp.udps_rcv6_swcsum), 0L, 0L, 0L);
}
@Override
public UdpStats getUDPv4Stats() {
BsdUdpstat stat = udpstat.get();
return new UdpStats(ParseUtil.unsignedIntToLong(stat.udps_opackets),
ParseUtil.unsignedIntToLong(stat.udps_ipackets), ParseUtil.unsignedIntToLong(stat.udps_noportmcast),
ParseUtil.unsignedIntToLong(stat.udps_hdrops + stat.udps_badsum + stat.udps_badlen));
}
@Override
public UdpStats getUDPv6Stats() {
BsdUdpstat stat = udpstat.get();
return new UdpStats(ParseUtil.unsignedIntToLong(stat.udps_snd6_swcsum),
ParseUtil.unsignedIntToLong(stat.udps_rcv6_swcsum), 0L, 0L);
}
@Override
public List getConnections() {
List conns = new ArrayList<>();
int[] pids = new int[1024];
int numberOfProcesses = SystemB.INSTANCE.proc_listpids(PROC_ALL_PIDS, 0, pids, pids.length * INT_SIZE)
/ INT_SIZE;
for (int i = 0; i < numberOfProcesses; i++) {
// Handle off-by-one bug in proc_listpids where the size returned
// is: SystemB.INT_SIZE * (pids + 1)
if (pids[i] > 0) {
for (Integer fd : queryFdList(pids[i])) {
IPConnection ipc = queryIPConnection(pids[i], fd);
if (ipc != null) {
conns.add(ipc);
}
}
}
}
return conns;
}
private static List queryFdList(int pid) {
List fdList = new ArrayList<>();
int bufferSize = SystemB.INSTANCE.proc_pidinfo(pid, PROC_PIDLISTFDS, 0, null, 0);
if (bufferSize > 0) {
ProcFdInfo fdInfo = new ProcFdInfo();
int numStructs = bufferSize / fdInfo.size();
ProcFdInfo[] fdArray = (ProcFdInfo[]) fdInfo.toArray(numStructs);
bufferSize = SystemB.INSTANCE.proc_pidinfo(pid, PROC_PIDLISTFDS, 0, fdArray[0], bufferSize);
numStructs = bufferSize / fdInfo.size();
for (int i = 0; i < numStructs; i++) {
if (fdArray[i].proc_fdtype == PROX_FDTYPE_SOCKET) {
fdList.add(fdArray[i].proc_fd);
}
}
}
return fdList;
}
private static IPConnection queryIPConnection(int pid, int fd) {
try (SocketFdInfo si = new SocketFdInfo()) {
int ret = SystemB.INSTANCE.proc_pidfdinfo(pid, fd, PROC_PIDFDSOCKETINFO, si, si.size());
if (si.size() == ret && si.psi.soi_family == AF_INET || si.psi.soi_family == AF_INET6) {
InSockInfo ini;
String type;
TcpState state;
if (si.psi.soi_kind == SOCKINFO_TCP) {
si.psi.soi_proto.setType("pri_tcp");
si.psi.soi_proto.read();
ini = si.psi.soi_proto.pri_tcp.tcpsi_ini;
state = stateLookup(si.psi.soi_proto.pri_tcp.tcpsi_state);
type = "tcp";
} else if (si.psi.soi_kind == SOCKINFO_IN) {
si.psi.soi_proto.setType("pri_in");
si.psi.soi_proto.read();
ini = si.psi.soi_proto.pri_in;
state = NONE;
type = "udp";
} else {
return null;
}
byte[] laddr;
byte[] faddr;
if (ini.insi_vflag == 1) {
laddr = ParseUtil.parseIntToIP(ini.insi_laddr[3]);
faddr = ParseUtil.parseIntToIP(ini.insi_faddr[3]);
type += "4";
} else if (ini.insi_vflag == 2) {
laddr = ParseUtil.parseIntArrayToIP(ini.insi_laddr);
faddr = ParseUtil.parseIntArrayToIP(ini.insi_faddr);
type += "6";
} else {
return null;
}
int lport = ParseUtil.bigEndian16ToLittleEndian(ini.insi_lport);
int fport = ParseUtil.bigEndian16ToLittleEndian(ini.insi_fport);
return new IPConnection(type, laddr, lport, faddr, fport, state, si.psi.soi_qlen, si.psi.soi_incqlen,
pid);
}
}
return null;
}
private static TcpState stateLookup(int state) {
switch (state) {
case 0:
return CLOSED;
case 1:
return LISTEN;
case 2:
return SYN_SENT;
case 3:
return SYN_RECV;
case 4:
return ESTABLISHED;
case 5:
return CLOSE_WAIT;
case 6:
return FIN_WAIT_1;
case 7:
return CLOSING;
case 8:
return LAST_ACK;
case 9:
return FIN_WAIT_2;
case 10:
return TIME_WAIT;
default:
return UNKNOWN;
}
}
/*
* There are multiple versions of some tcp/udp/ip stats structures in macOS. Since we only need a few of the
* hundreds of fields, we can improve performance by selectively reading the ints from the appropriate offsets,
* which are consistent across the structure.
*/
private static BsdTcpstat queryTcpstat() {
BsdTcpstat mt = new BsdTcpstat();
try (Memory m = SysctlUtil.sysctl("net.inet.tcp.stats")) {
if (m != null && m.size() >= 128) {
mt.tcps_connattempt = m.getInt(0);
mt.tcps_accepts = m.getInt(4);
mt.tcps_drops = m.getInt(12);
mt.tcps_conndrops = m.getInt(16);
mt.tcps_sndpack = m.getInt(64);
mt.tcps_sndrexmitpack = m.getInt(72);
mt.tcps_rcvpack = m.getInt(104);
mt.tcps_rcvbadsum = m.getInt(112);
mt.tcps_rcvbadoff = m.getInt(116);
mt.tcps_rcvmemdrop = m.getInt(120);
mt.tcps_rcvshort = m.getInt(124);
}
}
return mt;
}
private static BsdIpstat queryIpstat() {
BsdIpstat mi = new BsdIpstat();
try (Memory m = SysctlUtil.sysctl("net.inet.ip.stats")) {
if (m != null && m.size() >= 60) {
mi.ips_total = m.getInt(0);
mi.ips_badsum = m.getInt(4);
mi.ips_tooshort = m.getInt(8);
mi.ips_toosmall = m.getInt(12);
mi.ips_badhlen = m.getInt(16);
mi.ips_badlen = m.getInt(20);
mi.ips_delivered = m.getInt(56);
}
}
return mi;
}
private static BsdIp6stat queryIp6stat() {
BsdIp6stat mi6 = new BsdIp6stat();
try (Memory m = SysctlUtil.sysctl("net.inet6.ip6.stats")) {
if (m != null && m.size() >= 96) {
mi6.ip6s_total = m.getLong(0);
mi6.ip6s_localout = m.getLong(88);
}
}
return mi6;
}
public static BsdUdpstat queryUdpstat() {
BsdUdpstat ut = new BsdUdpstat();
try (Memory m = SysctlUtil.sysctl("net.inet.udp.stats")) {
if (m != null && m.size() >= 1644) {
ut.udps_ipackets = m.getInt(0);
ut.udps_hdrops = m.getInt(4);
ut.udps_badsum = m.getInt(8);
ut.udps_badlen = m.getInt(12);
ut.udps_opackets = m.getInt(36);
ut.udps_noportmcast = m.getInt(48);
ut.udps_rcv6_swcsum = m.getInt(64);
ut.udps_snd6_swcsum = m.getInt(80);
}
}
return ut;
}
}