Adding upstream version 1.34.4.

Signed-off-by: Daniel Baumann <daniel@debian.org>

plugins/inputs/ipvs/ipvs.go

@@ -0,0 +1,153 @@
+//go:generate ../../../tools/readme_config_includer/generator
+//go:build linux
+
+package ipvs
+
+import (
+	_ "embed"
+	"fmt"
+	"math/bits"
+	"strconv"
+	"syscall"
+
+	"github.com/moby/ipvs"
+
+	"github.com/influxdata/telegraf"
+	"github.com/influxdata/telegraf/plugins/common/logrus"
+	"github.com/influxdata/telegraf/plugins/inputs"
+)
+
+//go:embed sample.conf
+var sampleConfig string
+
+type IPVS struct {
+	Log    telegraf.Logger `toml:"-"`
+	handle *ipvs.Handle
+}
+
+func (*IPVS) SampleConfig() string {
+	return sampleConfig
+}
+
+func (i *IPVS) Gather(acc telegraf.Accumulator) error {
+	if i.handle == nil {
+		h, err := ipvs.New("") // TODO: make the namespace configurable
+		if err != nil {
+			return fmt.Errorf("unable to open IPVS handle: %w", err)
+		}
+		i.handle = h
+	}
+
+	services, err := i.handle.GetServices()
+	if err != nil {
+		i.handle.Close()
+		i.handle = nil // trigger a reopen on next call to gather
+		return fmt.Errorf("failed to list IPVS services: %w", err)
+	}
+	for _, s := range services {
+		fields := map[string]interface{}{
+			"connections": s.Stats.Connections,
+			"pkts_in":     s.Stats.PacketsIn,
+			"pkts_out":    s.Stats.PacketsOut,
+			"bytes_in":    s.Stats.BytesIn,
+			"bytes_out":   s.Stats.BytesOut,
+			"pps_in":      s.Stats.PPSIn,
+			"pps_out":     s.Stats.PPSOut,
+			"cps":         s.Stats.CPS,
+		}
+		acc.AddGauge("ipvs_virtual_server", fields, serviceTags(s))
+
+		destinations, err := i.handle.GetDestinations(s)
+		if err != nil {
+			i.Log.Errorf("Failed to list destinations for a virtual server: %v", err)
+			continue // move on to the next virtual server
+		}
+
+		for _, d := range destinations {
+			fields := map[string]interface{}{
+				"active_connections":   d.ActiveConnections,
+				"inactive_connections": d.InactiveConnections,
+				"connections":          d.Stats.Connections,
+				"pkts_in":              d.Stats.PacketsIn,
+				"pkts_out":             d.Stats.PacketsOut,
+				"bytes_in":             d.Stats.BytesIn,
+				"bytes_out":            d.Stats.BytesOut,
+				"pps_in":               d.Stats.PPSIn,
+				"pps_out":              d.Stats.PPSOut,
+				"cps":                  d.Stats.CPS,
+			}
+			destTags := destinationTags(d)
+			if s.FWMark > 0 {
+				destTags["virtual_fwmark"] = strconv.Itoa(int(s.FWMark))
+			} else {
+				destTags["virtual_protocol"] = protocolToString(s.Protocol)
+				destTags["virtual_address"] = s.Address.String()
+				destTags["virtual_port"] = strconv.Itoa(int(s.Port))
+			}
+			acc.AddGauge("ipvs_real_server", fields, destTags)
+		}
+	}
+
+	return nil
+}
+
+// helper: given a Service, return tags that identify it
+func serviceTags(s *ipvs.Service) map[string]string {
+	ret := map[string]string{
+		"sched":          s.SchedName,
+		"netmask":        strconv.Itoa(bits.OnesCount32(s.Netmask)),
+		"address_family": addressFamilyToString(s.AddressFamily),
+	}
+	// Per the ipvsadm man page, a virtual service is defined "based on
+	// protocol/addr/port or firewall mark"
+	if s.FWMark > 0 {
+		ret["fwmark"] = strconv.Itoa(int(s.FWMark))
+	} else {
+		ret["protocol"] = protocolToString(s.Protocol)
+		ret["address"] = s.Address.String()
+		ret["port"] = strconv.Itoa(int(s.Port))
+	}
+	return ret
+}
+
+// helper: given a Destination, return tags that identify it
+func destinationTags(d *ipvs.Destination) map[string]string {
+	return map[string]string{
+		"address":        d.Address.String(),
+		"port":           strconv.Itoa(int(d.Port)),
+		"address_family": addressFamilyToString(d.AddressFamily),
+	}
+}
+
+// helper: convert protocol uint16 to human-readable string (if possible)
+func protocolToString(p uint16) string {
+	switch p {
+	case syscall.IPPROTO_TCP:
+		return "tcp"
+	case syscall.IPPROTO_UDP:
+		return "udp"
+	case syscall.IPPROTO_SCTP:
+		return "sctp"
+	default:
+		return strconv.FormatUint(uint64(p), 10)
+	}
+}
+
+// helper: convert addressFamily to a human-readable string
+func addressFamilyToString(af uint16) string {
+	switch af {
+	case syscall.AF_INET:
+		return "inet"
+	case syscall.AF_INET6:
+		return "inet6"
+	default:
+		return strconv.FormatUint(uint64(af), 10)
+	}
+}
+
+func init() {
+	inputs.Add("ipvs", func() telegraf.Input {
+		logrus.InstallHook()
+		return &IPVS{}
+	})
+}