Add metrics server

.github/workflows/main.yml

@@ -0,0 +1,36 @@
+name: main
+on:
+  pull_request:
+  push:
+    branches:
+    - 'master'
+env:
+  go-version: 1.20.1
+jobs:
+  lint:
+    name: Lint
+    runs-on: ubuntu-20.04
+    steps:
+    - uses: actions/checkout@v3
+    - uses: actions/setup-go@v3
+      with:
+        go-version: ${{ env.go-version }}
+    - uses: golangci/golangci-lint-action@v3
+  build:
+    name: Build
+    runs-on: ubuntu-20.04
+    steps:
+    - uses: actions/checkout@v3
+    - uses: actions/setup-go@v3
+      with:
+        go-version: ${{ env.go-version }}
+    - run: go build -trimpath ./...
+  test:
+    name: Test
+    runs-on: ubuntu-20.04
+    steps:
+    - uses: actions/checkout@v3
+    - uses: actions/setup-go@v3
+      with:
+        go-version: ${{ env.go-version }}
+    - run: NO_PROXY= go test -race -v -coverprofile cover.out ./...

.gitignore

@@ -28,6 +28,4 @@ _testmain.go
 *.exe
 *.test
 *.prof
-
-# Ignore go.sum
-/go.sum
+*.out

.golangci.yml

@@ -3,6 +3,9 @@ linters:
         - dupl
         - goconst
         - gofmt
-        - golint
+        - revive
         - typecheck
         - unparam
+linters-settings:
+    staticcheck:
+        checks: ["all", "-SA6002"]

README.md

@@ -61,7 +61,7 @@ Features
 Install
 -------
 
-Use Go 1.7 or better.
+Use a recent version of Go.
 
 ```
 go get -u github.com/cybozu-go/usocksd/...

address_group.go

@@ -70,7 +70,7 @@ func (a *AddressGroup) detectInvalid() {
 		}
 		a.lock.Lock()
 		if len(invalids) > 0 && len(a.invalids) != len(invalids) {
-			log.Warn("detect black-listed IP", map[string]interface{}{
+			_ = log.Warn("detect black-listed IP", map[string]interface{}{
 				"_bad_ips": toStringList(invalids),
 			})
 		}

cmd/usocksd/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"fmt"
 	"net"
 	"os"
 
@@ -18,13 +19,24 @@ var (
 	optFile = flag.String("f", "", "configuration file name")
 )
 
+func serveMetrics(c *usocksd.Config) error {
+	metricsServer := usocksd.NewMetricsServer(c)
+	mln, err := usocksd.MetricsListener(c)
+	if err != nil {
+		return fmt.Errorf("could not initialize metrics server: %w", err)
+	}
+	return metricsServer.Serve(mln)
+}
+
 func serve(lns []net.Listener, c *usocksd.Config) {
 	socksServer := usocksd.NewServer(c)
 	for _, ln := range lns {
 		socksServer.Serve(ln)
 	}
-	err := well.Wait()
-	if err != nil && !well.IsSignaled(err) {
+	if err := serveMetrics(c); err != nil {
+		log.ErrorExit(err)
+	}
+	if err := well.Wait(); err != nil && !well.IsSignaled(err) {
 		log.ErrorExit(err)
 	}
 }

config.go

@@ -10,12 +10,14 @@ import (
 )
 
 const (
-	defaultPort = 1080
+	defaultPort        = 1080
+	defaultMetricsPort = 1081
 )
 
 // IncomingConfig is a set of configurations to accept clients.
 type IncomingConfig struct {
 	Port         int
+	MetricsPort  int `toml:"metrics_port"`
 	Addresses    []net.IP
 	AllowFrom    []string `toml:"allow_from"`
 	allowSubnets []*net.IPNet
@@ -42,6 +44,7 @@ type Config struct {
 func NewConfig() *Config {
 	c := new(Config)
 	c.Incoming.Port = defaultPort
+	c.Incoming.MetricsPort = defaultMetricsPort
 	return c
 }
 

config_test.go

@@ -30,6 +30,9 @@ func TestConfig(t *testing.T) {
 	if c.Incoming.Port != 1080 {
 		t.Error("incoming.port != 1080")
 	}
+	if c.Incoming.MetricsPort != 8081 {
+		t.Error("incoming.port != 1080")
+	}
 	if len(c.Incoming.Addresses) != 1 {
 		t.Error("empty incoming.addresses")
 	} else {

go.mod

@@ -1,10 +1,43 @@
 module github.com/cybozu-go/usocksd
 
 require (
-	github.com/BurntSushi/toml v0.3.1
-	github.com/cybozu-go/log v1.6.0
-	github.com/cybozu-go/netutil v1.2.0
-	github.com/cybozu-go/well v1.10.0
+	github.com/BurntSushi/toml v1.1.0
+	github.com/cybozu-go/log v1.6.1
+	github.com/cybozu-go/netutil v1.4.2
+	github.com/cybozu-go/well v1.11.0
+	github.com/prometheus/client_golang v1.12.2
 )
 
-go 1.13
+require (
+	github.com/beorn7/perks v1.0.1 // indirect
+	github.com/cespare/xxhash/v2 v2.1.2 // indirect
+	github.com/fsnotify/fsnotify v1.5.4 // indirect
+	github.com/golang/protobuf v1.5.2 // indirect
+	github.com/hashicorp/hcl v1.0.0 // indirect
+	github.com/magiconair/properties v1.8.6 // indirect
+	github.com/matttproud/golang_protobuf_extensions v1.0.1 // indirect
+	github.com/mitchellh/mapstructure v1.5.0 // indirect
+	github.com/onsi/gomega v1.19.0 // indirect
+	github.com/pelletier/go-toml v1.9.5 // indirect
+	github.com/pelletier/go-toml/v2 v2.0.2 // indirect
+	github.com/prometheus/client_model v0.2.0 // indirect
+	github.com/prometheus/common v0.32.1 // indirect
+	github.com/prometheus/procfs v0.7.3 // indirect
+	github.com/spf13/afero v1.8.2 // indirect
+	github.com/spf13/cast v1.5.0 // indirect
+	github.com/spf13/jwalterweatherman v1.1.0 // indirect
+	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/spf13/viper v1.12.0 // indirect
+	github.com/subosito/gotenv v1.4.0 // indirect
+	github.com/vishvananda/netlink v1.1.0 // indirect
+	github.com/vishvananda/netns v0.0.0-20211101163701-50045581ed74 // indirect
+	golang.org/x/net v0.0.0-20220706163947-c90051bbdb60 // indirect
+	golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e // indirect
+	golang.org/x/text v0.3.7 // indirect
+	google.golang.org/protobuf v1.28.0 // indirect
+	gopkg.in/ini.v1 v1.66.6 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)
+
+go 1.19