Add netstat_poller

go.mod

@@ -4,5 +4,8 @@ go 1.15
 
 require (
 	github.com/heroku/slog v0.0.0-20140402171358-7fea2d3038fa
+	github.com/shirou/gopsutil v3.21.11+incompatible
 	github.com/xuyu/goredis v0.0.0-20140820021013-517ef62abb23
+	github.com/yusufpapurcu/wmi v1.2.2 // indirect
+	golang.org/x/sys v0.0.0-20220422013727-9388b58f7150 // indirect
 )

go.sum

@@ -1,4 +1,13 @@
+github.com/go-ole/go-ole v1.2.6 h1:/Fpf6oFPoeFik9ty7siob0G6Ke8QvQEuVcuChpwXzpY=
+github.com/go-ole/go-ole v1.2.6/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=
 github.com/heroku/slog v0.0.0-20140402171358-7fea2d3038fa h1:udpqQXzEVyYxKVOfRKCfx8FyneqHrvYEtzXc0LHxgRs=
 github.com/heroku/slog v0.0.0-20140402171358-7fea2d3038fa/go.mod h1:qakOkduMaAhe2ypljIuLEinhnl5nCnyz7iXMZHwdr14=
+github.com/shirou/gopsutil v3.21.11+incompatible h1:+1+c1VGhc88SSonWP6foOcLhvnKlUeu/erjjvaPEYiI=
+github.com/shirou/gopsutil v3.21.11+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMTu2PcXUg8+E1lC7eC3UO/RA=
 github.com/xuyu/goredis v0.0.0-20140820021013-517ef62abb23 h1:0+1EhmoODl7T8NbEJ7/9WgQNCW05HKI0y+/kDR9bECE=
 github.com/xuyu/goredis v0.0.0-20140820021013-517ef62abb23/go.mod h1:Ew+jSwVYyBT+GrKWGzlld2VGZqUy5cgGQATzqHpqGvg=
+github.com/yusufpapurcu/wmi v1.2.2 h1:KBNDSne4vP5mbSWnJbO+51IMOXJB67QiYCSBrubbPRg=
+github.com/yusufpapurcu/wmi v1.2.2/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=
+golang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20220422013727-9388b58f7150 h1:xHms4gcpe1YE7A3yIllJXP16CMAGuqwO2lX1mTyyRRc=
+golang.org/x/sys v0.0.0-20220422013727-9388b58f7150/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=

netstat_poller.go

@@ -0,0 +1,76 @@
+package shh
+
+import (
+	"syscall"
+	"time"
+
+	"github.com/heroku/slog"
+	"github.com/shirou/gopsutil/net"
+)
+
+type Netstat struct {
+	measurements chan<- Measurement
+}
+
+func NewNetstatPoller(measurements chan<- Measurement) Netstat {
+	return Netstat{measurements: measurements}
+}
+
+func getStats() (map[string]int, error) {
+	conns, err := net.Connections("all")
+	if err != nil {
+		return nil, err
+	}
+
+	counts := make(map[string]int)
+	counts["UDP"] = 0
+	for _, conn := range conns {
+		if conn.Type == syscall.SOCK_DGRAM {
+			counts["UDP"]++
+			continue
+		}
+		c, ok := counts[conn.Status]
+		if !ok {
+			counts[conn.Status] = 0
+		}
+		counts[conn.Status] = c + 1
+	}
+
+	fields := map[string]int{
+		"tcp_established": counts["ESTABLISHED"],
+		"tcp_syn_sent":    counts["SYN_SENT"],
+		"tcp_syn_recv":    counts["SYN_RECV"],
+		"tcp_fin_wait1":   counts["FIN_WAIT1"],
+		"tcp_fin_wait2":   counts["FIN_WAIT2"],
+		"tcp_time_wait":   counts["TIME_WAIT"],
+		"tcp_close":       counts["CLOSE"],
+		"tcp_close_wait":  counts["CLOSE_WAIT"],
+		"tcp_last_ack":    counts["LAST_ACK"],
+		"tcp_listen":      counts["LISTEN"],
+		"tcp_closing":     counts["CLOSING"],
+		"tcp_none":        counts["NONE"],
+		"udp_socket":      counts["UDP"],
+	}
+
+	return fields, nil
+}
+
+func (poller Netstat) Poll(tick time.Time) {
+	ctx := slog.Context{"poller": poller.Name(), "fn": "Poll", "tick": tick}
+
+	stats, err := getStats()
+	if err != nil {
+		LogError(ctx, err, "Error reading netconn")
+		return
+	}
+
+	for field, _ := range stats {
+		poller.measurements <- GaugeMeasurement{tick, poller.Name(), []string{field}, uint64(stats[field]), Connections}
+	}
+}
+
+func (poller Netstat) Name() string {
+	return "netstat"
+}
+
+func (poller Netstat) Exit() {}

pollers.go

@@ -40,6 +40,8 @@ func NewMultiPoller(measurements chan<- Measurement, config Config) Multi {
 			mp.RegisterPoller(NewProcessesPoller(measurements, config))
 		case "self":
 			mp.RegisterPoller(NewSelfPoller(measurements, config))
+		case "netstat":
+			mp.RegisterPoller(NewNetstatPoller(measurements))
 		case "conntrack":
 			mp.RegisterPoller(NewConntrackPoller(measurements))
 		case "syslogngstats":