This commit is contained in:
K Rin 2025-08-10 11:40:05 +07:00 committed by GitHub
commit 0129dc44b8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -17,7 +17,6 @@
package collector
import (
"errors"
"fmt"
"log/slog"
"unsafe"
@ -33,21 +32,68 @@ import (
#include <netinet/tcp.h>
#include <netinet/tcp_var.h>
#include <netinet/udp.h>
#include <netinet/udp_var.h>
*/
import "C"
var (
// TCP metrics
bsdNetstatTcpSendPacketsTotal = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "tcp_transmit_packets_total"),
"TCP packets sent",
nil, nil,
)
bsdNetstatTcpRecvPacketsTotal = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "tcp_receive_packets_total"),
"TCP packets received",
nil, nil,
)
bsdNetstatTcpConnectionAttempts = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "tcp_connection_attempts_total"),
"Number of times TCP connections have been initiated",
nil, nil,
)
bsdNetstatTcpConnectionAccepts = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "tcp_connection_accepts_total"),
"Number of times TCP connections have made it to established state",
nil, nil,
)
bsdNetstatTcpConnectionDrops = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "tcp_connection_drops_total"),
"Number of dropped TCP connections",
nil, nil,
)
bsdNetstatTcpRetransmitPackets = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "tcp_retransmit_packets_total"),
"Number of TCP data packets retransmitted",
nil, nil,
)
// UDP metrics
bsdNetstatUdpSendPacketsTotal = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "udp_transmit_packets_total"),
"UDP packets sent",
nil, nil,
)
bsdNetstatUdpRecvPacketsTotal = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "udp_receive_packets_total"),
"UDP packets received",
nil, nil,
)
bsdNetstatUdpHeaderDrops = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "udp_header_drops_total"),
"Number of UDP packets dropped due to invalid header",
nil, nil,
)
bsdNetstatUdpBadChecksum = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "udp_bad_checksum_total"),
"Number of UDP packets dropped due to bad checksum",
nil, nil,
)
bsdNetstatUdpNoPort = prometheus.NewDesc(
prometheus.BuildFQName(namespace, "netstat", "udp_no_port_total"),
"Number of UDP packets to port with no listener",
nil, nil,
)
)
type netStatCollector struct {
@ -76,18 +122,16 @@ func getData(queryString string) ([]byte, error) {
fmt.Println("Error:", err)
return nil, err
}
if len(data) < int(unsafe.Sizeof(C.struct_tcpstat{})) {
return nil, errors.New("Data Size mismatch")
}
return data, nil
}
func (c *netStatCollector) Update(ch chan<- prometheus.Metric) error {
tcpData, err := getData("net.inet.tcp.stats")
if err != nil {
return err
return fmt.Errorf("failed to get TCP stats: %w", err)
}
if len(tcpData) < int(unsafe.Sizeof(C.struct_tcpstat{})) {
return fmt.Errorf("TCP data size mismatch: got %d, want >= %d", len(tcpData), unsafe.Sizeof(C.struct_tcpstat{}))
}
tcpStats := *(*C.struct_tcpstat)(unsafe.Pointer(&tcpData[0]))
@ -97,12 +141,67 @@ func (c *netStatCollector) Update(ch chan<- prometheus.Metric) error {
prometheus.CounterValue,
float64(tcpStats.tcps_sndtotal),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatTcpRecvPacketsTotal,
prometheus.CounterValue,
float64(tcpStats.tcps_rcvtotal),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatTcpConnectionAttempts,
prometheus.CounterValue,
float64(tcpStats.tcps_connattempt),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatTcpConnectionAccepts,
prometheus.CounterValue,
float64(tcpStats.tcps_accepts),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatTcpConnectionDrops,
prometheus.CounterValue,
float64(tcpStats.tcps_drops),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatTcpRetransmitPackets,
prometheus.CounterValue,
float64(tcpStats.tcps_sndrexmitpack),
)
udpData, err := getData("net.inet.udp.stats")
if err != nil {
return fmt.Errorf("failed to get UDP stats: %w", err)
}
if len(udpData) < int(unsafe.Sizeof(C.struct_udpstat{})) {
return fmt.Errorf("UDP data size mismatch: got %d, want >= %d", len(udpData), unsafe.Sizeof(C.struct_udpstat{}))
}
udpStats := *(*C.struct_udpstat)(unsafe.Pointer(&udpData[0]))
ch <- prometheus.MustNewConstMetric(
bsdNetstatUdpSendPacketsTotal,
prometheus.CounterValue,
float64(udpStats.udps_opackets),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatUdpRecvPacketsTotal,
prometheus.CounterValue,
float64(udpStats.udps_ipackets),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatUdpHeaderDrops,
prometheus.CounterValue,
float64(udpStats.udps_hdrops),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatUdpBadChecksum,
prometheus.CounterValue,
float64(udpStats.udps_badsum),
)
ch <- prometheus.MustNewConstMetric(
bsdNetstatUdpNoPort,
prometheus.CounterValue,
float64(udpStats.udps_noport),
)
return nil
}