.dockerignore

@@ -1,6 +0,0 @@
-*
-!cmd/
-!*.go
-
-!go.mod
-!go.sum

.github/workflows/release.yml

@@ -1,31 +0,0 @@
-name: goreleaser
-
-on:
-  push:
-    tags:
-      - 'v*' # Push events to matching v*, i.e. v1.0, v20.15.10
-
-permissions:
-  contents: write
-
-jobs:
-  goreleaser:
-    runs-on: ubuntu-latest
-    steps:
-      -
-        name: Checkout
-        uses: actions/checkout@v4
-        with:
-          fetch-depth: 0
-      -
-        name: Set up Go
-        uses: actions/setup-go@v5
-      -
-        name: Run GoReleaser
-        uses: goreleaser/goreleaser-action@v6
-        with:
-          distribution: goreleaser
-          version: '~> v2'
-          args: release --clean
-        env:
-          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/update-go-modules.yml

@@ -1,30 +0,0 @@
-name: Auto-Update Go Modules
-
-on:
-  schedule:
-    - cron: "0 0 * * 1" # Runs every Monday at midnight
-
-jobs:
-  update-go-modules:
-    runs-on: ubuntu-latest
-    permissions:
-      contents: write
-
-    steps:
-      - name: Checkout Code
-        uses: actions/checkout@v4
-
-      - name: Set up Go
-        uses: actions/setup-go@v5
-        with:
-          go-version: stable
-
-      - name: Update Dependencies
-        run: |
-          go get -u ./...
-          go mod tidy
-          git config user.name "github-actions[bot]"
-          git config user.email "github-actions[bot]@users.noreply.github.com"
-          git add go.mod go.sum
-          git commit -m "chore: auto-update Go modules"
-          git push

.gitignore

@@ -1,5 +1,3 @@
-# Auto-generated .gitignore by gignore: github.com/onyx-and-iris/gignore
-### Go ###
 # If you prefer the allow list template instead of the deny list, see community template:
 # https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
 #
@@ -9,7 +7,6 @@
 *.dll
 *.so
 *.dylib
-bin/
 
 # Test binary, built with `go test -c`
 *.test
@@ -22,7 +19,7 @@ bin/
 
 # Go workspace file
 go.work
-# End of gignore: github.com/onyx-and-iris/gignore
 
-# Added by goreleaser init:
-dist/
+# testing
+run.sh
+server.yaml

.goreleaser.yaml

@@ -1,55 +0,0 @@
-# This is an example .goreleaser.yml file with some sensible defaults.
-# Make sure to check the documentation at https://goreleaser.com
-
-# The lines below are called `modelines`. See `:help modeline`
-# Feel free to remove those if you don't want/need to use them.
-# yaml-language-server: $schema=https://goreleaser.com/static/schema.json
-# vim: set ts=2 sw=2 tw=0 fo=cnqoj
-
-version: 2
-
-before:
-  hooks:
-    # You may remove this if you don't use go modules.
-    - go mod tidy
-    # you may remove this if you don't need go generate
-    - go generate ./...
-
-builds:
-  - main: ./cmd/q3rcon-proxy/
-    env:
-      - CGO_ENABLED=0
-    goos:
-      - linux
-      - windows
-    goarch:
-      - amd64
-
-archives:
-  - formats: ['tar.gz']
-    # this name template makes the OS and Arch compatible with the results of `uname`.
-    name_template: >-
-      {{ .ProjectName }}_
-      {{- title .Os }}_
-      {{- if eq .Arch "amd64" }}x86_64
-      {{- else if eq .Arch "386" }}i386
-      {{- else }}{{ .Arch }}{{ end }}
-      {{- if .Arm }}v{{ .Arm }}{{ end }}
-    # use zip for windows archives
-    format_overrides:
-      - goos: windows
-        formats: ['zip']
-
-changelog:
-  sort: asc
-  filters:
-    exclude:
-      - '^docs:'
-      - '^test:'
-
-release:
-  footer: >-
-
-    ---
-
-    Released by [GoReleaser](https://github.com/goreleaser/goreleaser).

.vscode/launch.json

@@ -9,9 +9,9 @@
             "type": "go",
             "request": "launch",
             "mode": "auto",
-            "program": "${workspaceFolder}/cmd/q3rcon-proxy/",
+            "program": "${workspaceFolder}/cmd/q3rcon-proxy/main.go",
             "env": {
-                "Q3RCON_PORTS_MAPPING": "28961:28960",
+                "Q3RCON_PROXY": "28961:28960",
             }
         }
     ]

CHANGELOG.md

@@ -11,50 +11,6 @@ Before any major/minor/patch bump all unit tests will be run to verify they pass
 
 -   [x]
 
-## [1.7.0] - 2025-06-05
-
-### Added
-
--   Taskfile added for running and building project.
--   The binary may be passed CLI flags as well as environment variables.
-
-### Changed
-
--   CLI component rewritten with urfave/cli.
--   env var `Q3RCON_TARGET_PORTS` renamed to `Q3RCON_PORTS_MAPPING`
-
-## [1.4.0] - 2024-11-29
-
-### Added
-
--   new environment variable `Q3RCON_TARGET_HOST` for setting the host the gameserver is on.
-
-### Changed
-
--   environment variable `Q3RCON_HOST` renamed to `Q3RCON_PROXY_HOST`
--   environment variable `Q3RCON_PROXY` renamed to `Q3RCON_TARGET_PORTS`.
--   default session timeout changed from 5 to 20 minutes.
-
-## [1.3.0] - 2024-10-23
-
-### Added
-
--   Add sessionCache for tracking sessions.
--   Functional option `WithStaleTimeout` renamed to `WithSessionTimeout`
-
-## [1.2.0] - 2024-10-19
-
-### Added
-
--   optional function `WithStaleTimeout`, use it to configure the session timeout value.
-    -   it defaults to 5 minutes.
-
-## [1.1.0] - 2024-09-28
-
-### Added
-
--   connection (challenge) requests are now logged.
-
 ## [0.6.0] - 2024-03-21
 
 ### Added

Dockerfile

@@ -1,4 +1,4 @@
-FROM golang:1.24 AS build_image
+FROM golang:1.21 AS build_image
 
 WORKDIR /usr/src/app
 

Makefile

@@ -1,40 +1,2 @@
-PROGRAM = q3rcon-proxy
-
-GO = @go
-BIN_DIR := bin
-
-WINDOWS=$(BIN_DIR)/$(PROGRAM)_windows_amd64.exe
-LINUX=$(BIN_DIR)/$(PROGRAM)_linux_amd64
-VERSION=$(shell git log -n 1 --format=%h)
-
-.DEFAULT_GOAL := build
-
-.PHONY: fmt vet build windows linux test clean
-fmt:        
-	$(GO) fmt ./...
-
-vet: fmt        
-	$(GO) vet ./...
-
-build: vet windows linux | $(BIN_DIR)
-	@echo version: $(VERSION)
-
-windows: $(WINDOWS)
-
-linux: $(LINUX)
-
-
-$(WINDOWS):
-	env GOOS=windows GOARCH=amd64 go build -v -o $(WINDOWS) -ldflags="-s -w -X main.version=$(VERSION)"  ./cmd/$(PROGRAM)/
-
-$(LINUX):
-	env GOOS=linux GOARCH=amd64 go build -v -o $(LINUX) -ldflags="-s -w -X main.version=$(VERSION)"  ./cmd/$(PROGRAM)/
-
-test:
-	$(GO) test ./...
-
-$(BIN_DIR):
-	@mkdir -p $@
-
-clean:
-	@rm -rv $(BIN_DIR)
+go-build:
+	go build ./cmd/q3rcon-proxy/

README.md

@@ -8,52 +8,23 @@ Unfortunately the Q3Rcon engine ties the rcon port to the game servers public po
 
 ### Use
 
-#### Flags
+Run one or multiple rcon proxies by setting an environment variable `Q3RCON_PROXY`
+
+for example:
 
 ```bash
-#!/usr/bin/env bash
-
-/usr/local/bin/q3rcon-proxy \
-    --proxy-host=0.0.0.0 \
-    --target-host=localhost \
-    --ports-mapping=28961:28960 \
-    --session-timeout=20 \
-    --loglevel=debug
+export Q3RCON_PROXY="20000:28960;20001:28961;20002:28962"
 ```
 
-#### Environment Variables
+This would configure q3rcon-proxy to run 3 proxy servers listening on ports `20000`, `20001` and `20002` that redirect rcon requests to game servers on ports `28960`, `28961` and `28962` respectively.
 
-Each of the flags has a corresponding environment variable:
-
--   `Q3RCON_PROXY_HOST`: The host the proxy server sits on.
--   `Q3RCON_TARGET_HOST`: The host the game servers sit on.
--   `Q3RCON_PORTS_MAPPING`: A mapping as a string with `source:target` pairs delimited by `;`.
--   `Q3RCON_SESSION_TIMEOUT`: Timeout in seconds for each udp session.
--   `Q3RCON_LOGLEVEL`: The application's logging level (see [Logging][logging]).
-
-Multiple rcon proxies may be configured by setting *--ports-mapping/Q3RCON_PORTS_MAPPING* like so:
-
-```console
-export Q3RCON_PORTS_MAPPING="20000:28960;20001:28961;20002:28962"
-```
-
-This would configure q3rcon-proxy to run 3 proxy servers listening on ports 20000, 20001 and 20002 that redirect rcon requests to game servers on ports 28960, 28961 and 28962 respectively.
+Then just run the binary which you can compile yourself, download from `Releases` or use the included Dockerfile.
 
 ### Logging
 
-Set the log level with environment variable `Q3RCON_LOGLEVEL`.
+Set the log level with environment variable `Q3RCON_LOGLEVEL`:
 
-Acceptable values are:
-
-- `trace`
-- `debug`
-- `info`
-- `warn`
-- `error`
-- `fatal`
-- `panic`
-
-If not set it will default to `info`.
+`0 = Panic, 1 = Fatal, 2 = Error, 3 = Warning, 4 = Info, 5 = Debug, 6 = Trace`
 
 ### Special Thanks
 
@@ -61,10 +32,3 @@ If not set it will default to `info`.
 
 [lilproxy_url]: https://github.com/dgparker/lilproxy
 [user_link]: https://github.com/dgparker
-
-### Further Notes
-
-For a compatible rcon client also written in Go consider checking out the [Q3 Rcon][q3rcon] package.
-
-[q3rcon]: https://github.com/onyx-and-iris/q3rcon
-[logging]: https://github.com/onyx-and-iris/q3rcon-proxy/tree/dev?tab=readme-ov-file#logging

Taskfile.yml

@@ -1,71 +0,0 @@
-version: '3'
-
-includes:
-  docker: ./docker/Taskfile.docker.yml
-
-vars:
-  PROGRAM: q3rcon-proxy
-  SHELL: '{{if eq .OS "Windows_NT"}}powershell{{end}}'
-  BIN_DIR: bin
-
-  WINDOWS: '{{.BIN_DIR}}/{{.PROGRAM}}_windows_amd64.exe'
-  LINUX: '{{.BIN_DIR}}/{{.PROGRAM}}_linux_amd64'
-  GIT_COMMIT:
-    sh: git log -n 1 --format=%h
-
-tasks:
-  default:
-    desc: Build the q3rcon-proxy project
-    cmds:
-      - task: build
-
-  build:
-    desc: Build the q3rcon-proxy project
-    deps: [vet]
-    cmds:
-      - task: build-windows
-      - task: build-linux
-
-  vet:
-    desc: Vet the code
-    deps: [fmt]
-    cmds:
-      - go vet ./...
-
-  fmt:
-    desc: Fmt the code
-    cmds:
-      - go fmt ./...
-
-  build-windows:
-    desc: Build the q3rcon-proxy project for Windows
-    cmds:
-      - GOOS=windows GOARCH=amd64 go build -o {{.WINDOWS}} -ldflags="-X main.Version={{.GIT_COMMIT}}" ./cmd/{{.PROGRAM}}/
-    internal: true
-
-  build-linux:
-    desc: Build the q3rcon-proxy project for Linux
-    cmds:
-      - GOOS=linux GOARCH=amd64 go build -o {{.LINUX}} -ldflags="-X main.Version={{.GIT_COMMIT}}" ./cmd/{{.PROGRAM}}/
-    internal: true
-
-  test:
-    desc: Run tests
-    cmds:
-      - go test ./...
-
-  clean:
-    desc: Clean the build artifacts
-    cmds:
-      - '{{.SHELL}} rm -r {{.BIN_DIR}}'
-
-  run:
-    desc: Run the q3rcon-proxy project
-    cmds:
-      - |
-        go run ./cmd/{{.PROGRAM}} \
-        --proxy-host=0.0.0.0 \
-        --target-host=localhost \
-        --ports-mapping=28961:28960 \
-        --session-timeout=20 \
-        --loglevel=debug

cmd/q3rcon-proxy/main.go

@@ -1,132 +1,63 @@
 package main
 
 import (
-	"context"
 	"fmt"
 	"os"
-	"strconv"
+	"slices"
 	"strings"
 	"time"
 
-	udpproxy "github.com/onyx-and-iris/q3rcon-proxy"
 	log "github.com/sirupsen/logrus"
-	"github.com/urfave/cli/v3"
+
+	"github.com/onyx-and-iris/q3rcon-proxy/pkg/udpproxy"
 )
 
-// proxyConfig holds the configuration for a single UDP proxy server.
-type proxyConfig struct {
-	proxyHost      string
-	targetHost     string
-	portsMapping   []string
-	sessionTimeout int
-}
-
 func main() {
-	cmd := &cli.Command{
-		Name:  "q3rcon-proxy",
-		Usage: "A Quake 3 RCON proxy server",
-		Flags: []cli.Flag{
-			&cli.StringFlag{
-				Name:    "proxy-host",
-				Value:   "0.0.0.0",
-				Usage:   "Proxy host address",
-				Sources: cli.EnvVars("Q3RCON_PROXY_HOST"),
-			},
-			&cli.StringFlag{
-				Name:    "target-host",
-				Value:   "localhost",
-				Usage:   "Target host address",
-				Sources: cli.EnvVars("Q3RCON_TARGET_HOST"),
-			},
-			&cli.StringFlag{
-				Name:     "ports-mapping",
-				Usage:    "Proxy and target ports (proxy:target)",
-				Sources:  cli.EnvVars("Q3RCON_PORTS_MAPPING"),
-				Required: true,
-				Action: func(ctx context.Context, cmd *cli.Command, v string) error {
-					// Validate the ports mapping
-					for mapping := range strings.SplitSeq(v, ";") {
-						ports := strings.Split(mapping, ":")
-						if len(ports) != 2 {
-							return fmt.Errorf("invalid ports mapping: %s", mapping)
-						}
-						proxyPort, err := strconv.Atoi(ports[0])
-						if err != nil || proxyPort < 1 || proxyPort > 65535 {
-							return fmt.Errorf("invalid proxy port: %s", ports[0])
-						}
-						targetPort, err := strconv.Atoi(ports[1])
-						if err != nil || targetPort < 1 || targetPort > 65535 {
-							return fmt.Errorf("invalid target port: %s", ports[1])
-						}
-						if proxyPort == targetPort {
-							return fmt.Errorf("proxy and target ports cannot be the same: %s", mapping)
-						}
-					}
-					return nil
-				},
-			},
-			&cli.IntFlag{
-				Name:    "session-timeout",
-				Value:   20,
-				Usage:   "Session timeout in minutes",
-				Sources: cli.EnvVars("Q3RCON_SESSION_TIMEOUT"),
-			},
-			&cli.StringFlag{
-				Name:    "loglevel",
-				Value:   "info",
-				Usage:   "Log level (trace, debug, info, warn, error, fatal, panic)",
-				Sources: cli.EnvVars("Q3RCON_LOGLEVEL"),
-			},
-		},
-		Before: func(ctx context.Context, cmd *cli.Command) (context.Context, error) {
-			logLevel, err := log.ParseLevel(cmd.String("loglevel"))
-			if err != nil {
-				return ctx, fmt.Errorf("invalid log level: %w", err)
-			}
-			log.SetLevel(logLevel)
-			return ctx, nil
-		},
-		Action: func(_ context.Context, cmd *cli.Command) error {
-			errChan := make(chan error)
-
-			for mapping := range strings.SplitSeq(cmd.String("ports-mapping"), ";") {
-				cfg := proxyConfig{
-					proxyHost:      cmd.String("proxy-host"),
-					targetHost:     cmd.String("target-host"),
-					portsMapping:   strings.Split(mapping, ":"),
-					sessionTimeout: cmd.Int("session-timeout"),
-				}
-
-				go launchProxy(cfg, errChan)
-			}
-
-			// Under normal circumstances, the main goroutine will block here.
-			// If we receive an error we will log it and exit
-			return <-errChan
-		},
-	}
-
-	log.Fatal(cmd.Run(context.Background(), os.Args))
-}
-
-// launchProxy initializes the UDP proxy server with the given configuration.
-// It listens on the specified proxy host and port, and forwards traffic to the target host and port.
-// server.ListenAndServe blocks until the server is stopped or an error occurs.
-func launchProxy(cfg proxyConfig, errChan chan<- error) {
-	proxyPort, targetPort := cfg.portsMapping[0], cfg.portsMapping[1]
-
-	hostAddr := fmt.Sprintf("%s:%s", cfg.proxyHost, proxyPort)
-	proxyAddr := fmt.Sprintf("%s:%s", cfg.targetHost, targetPort)
-
-	server, err := udpproxy.New(
-		hostAddr, proxyAddr,
-		udpproxy.WithSessionTimeout(time.Duration(cfg.sessionTimeout)*time.Minute))
+	logLevel, err := getEnvInt("Q3RCON_LOGLEVEL")
 	if err != nil {
-		errChan <- fmt.Errorf("failed to create proxy: %w", err)
-		return
+		log.Fatalf("unable to parse Q3RCON_LEVEL: %s", err.Error())
+	}
+	if slices.Contains(log.AllLevels, log.Level(logLevel)) {
+		log.SetLevel(log.Level(logLevel))
 	}
 
-	log.Printf("q3rcon-proxy initialized: [proxy] (%s) [target] (%s)", hostAddr, proxyAddr)
+	proxies := os.Getenv("Q3RCON_PROXY")
+	if proxies == "" {
+		log.Fatal("env Q3RCON_PROXY required")
+	}
 
-	errChan <- server.ListenAndServe()
+	host := os.Getenv("Q3RCON_HOST")
+	if host == "" {
+		host = "0.0.0.0"
+	}
+
+	sessionTimeout, err := getEnvInt("Q3RCON_SESSION_TIMEOUT")
+	if err != nil {
+		log.Fatalf("unable to parse Q3RCON_SESSION_TIMEOUT: %s", err.Error())
+	}
+
+	for _, proxy := range strings.Split(proxies, ";") {
+		go start(host, proxy, sessionTimeout)
+	}
+
+	<-make(chan int)
+}
+
+func start(host, proxy string, sessionTimeout int) {
+	port, target := func() (string, string) {
+		x := strings.Split(proxy, ":")
+		return x[0], x[1]
+	}()
+
+	c, err := udpproxy.New(
+		fmt.Sprintf("%s:%s", host, port),
+		fmt.Sprintf("127.0.0.1:%s", target),
+		udpproxy.WithSessionTimeout(time.Duration(sessionTimeout)*time.Minute))
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	log.Printf("q3rcon-proxy initialized: [proxy] (%s:%s) [target] (127.0.0.1:%s)", host, port, target)
+
+	log.Fatal(c.ListenAndServe())
 }

cmd/q3rcon-proxy/util.go

@@ -0,0 +1,18 @@
+package main
+
+import (
+	"os"
+	"strconv"
+)
+
+func getEnvInt(key string) (int, error) {
+	s := os.Getenv(key)
+	if s == "" {
+		return 0, nil
+	}
+	v, err := strconv.Atoi(s)
+	if err != nil {
+		return 0, err
+	}
+	return v, nil
+}

debian/q3rcon-proxy.service

@@ -6,9 +6,9 @@ After=network.target
 [Service]
 Type=simple
 User=gameservers
-Environment="Q3RCON_PORTS_MAPPING=20000:28960;20001:28961;20002:28962"
+Environment="Q3RCON_PROXY=20000:28960;20001:28961;20002:28962"
 Environment="Q3RCON_HOST=0.0.0.0"
-Environment="Q3RCON_LOGLEVEL=info"
+Environment="Q3RCON_DEBUG=0"
 
 ExecStart=/usr/local/bin/q3rcon-proxy
 Restart=always

docker/Taskfile.docker.yml

@@ -1,26 +0,0 @@
-version: '3'
-
-vars:
-  IMAGE: q3rcon-proxy
-
-tasks:
-  build:
-    desc: Build the Docker image
-    cmds:
-      - docker build -t {{.IMAGE}} -f docker/Dockerfile .
-    dir: .
-
-  login:
-    desc: Login to Github Container Registry
-    cmds:
-      - docker login ghcr.io -u {{.GHCR_USER}} --password-stdin <<< {{.GHCR_TOKEN}}
-    internal: true
-
-  push:
-    desc: Push the Docker image to Github Container Registry
-    deps:
-      - task: build
-      - task: login
-    cmds:
-      - docker tag {{.IMAGE}} ghcr.io/{{.GHCR_USER}}/{{.IMAGE}}:latest
-      - docker push ghcr.io/{{.GHCR_USER}}/{{.IMAGE}}:latest

go.mod

@@ -1,12 +1,7 @@
 module github.com/onyx-and-iris/q3rcon-proxy
 
-go 1.24.0
+go 1.18
 
-toolchain go1.24.1
+require github.com/sirupsen/logrus v1.9.3
 
-require (
-	github.com/sirupsen/logrus v1.9.3
-	github.com/urfave/cli/v3 v3.3.8
-)
-
-require golang.org/x/sys v0.33.0 // indirect
+require golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 // indirect

go.sum

@@ -6,15 +6,10 @@ github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZN
 github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ=
 github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
-github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
-github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
-github.com/urfave/cli/v3 v3.3.8 h1:BzolUExliMdet9NlJ/u4m5vHSotJ3PzEqSAZ1oPMa/E=
-github.com/urfave/cli/v3 v3.3.8/go.mod h1:FJSKtM/9AiiTOJL4fJ6TbMUkxBXn7GO9guZqoZtpYpo=
+golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 h1:0A+M6Uqn+Eje4kHMK80dtF3JCXC4ykBgQG4Fe06QRhQ=
 golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=
-golang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
-gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
-gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

option.go

@@ -1,22 +0,0 @@
-package udpproxy
-
-import (
-	"time"
-
-	log "github.com/sirupsen/logrus"
-)
-
-// Option is a functional option type that allows us to configure the Client.
-type Option func(*Client)
-
-// WithSessionTimeout is a functional option to set the session timeout
-func WithSessionTimeout(timeout time.Duration) Option {
-	return func(c *Client) {
-		if timeout < time.Minute {
-			log.Warnf("cannot set stale session timeout to less than 1 minute.. defaulting to 20 minutes")
-			return
-		}
-
-		c.sessionTimeout = timeout
-	}
-}

pkg/udpproxy/sessioncache.go

@@ -15,8 +15,8 @@ func newSessionCache() sessionCache {
 	}
 }
 
-// read returns the associated session for an addr
-func (sc *sessionCache) read(addr string) (*session, bool) {
+// Read returns the associated session for an addr
+func (sc *sessionCache) Read(addr string) (*session, bool) {
 	sc.mu.RLock()
 	defer sc.mu.RUnlock()
 
@@ -24,16 +24,16 @@ func (sc *sessionCache) read(addr string) (*session, bool) {
 	return v, ok
 }
 
-// insert adds a session for a given addr.
-func (sc *sessionCache) insert(addr string, session *session) {
+// Upsert overrides the session for a given addr.
+func (sc *sessionCache) Upsert(addr string, session *session) {
 	sc.mu.Lock()
 	defer sc.mu.Unlock()
 
 	sc.data[addr] = session
 }
 
-// delete removes the session for the given addr.
-func (sc *sessionCache) delete(addr string) {
+// Delete removes the session for the given addr.
+func (sc *sessionCache) Delete(addr string) {
 	sc.mu.Lock()
 	defer sc.mu.Unlock()
 

pkg/udpproxy/udpproxy.go

@@ -7,6 +7,21 @@ import (
 	log "github.com/sirupsen/logrus"
 )
 
+// Option is a functional option type that allows us to configure the Client.
+type Option func(*Client)
+
+// WithSessionTimeout is a functional option to set the session timeout
+func WithSessionTimeout(timeout time.Duration) Option {
+	return func(c *Client) {
+		if timeout < time.Minute {
+			log.Warnf("cannot set stale session timeout to less than 1 minute.. defaulting to 5 minutes")
+			return
+		}
+
+		c.sessionTimeout = timeout
+	}
+}
+
 type Client struct {
 	laddr *net.UDPAddr
 	raddr *net.UDPAddr
@@ -17,8 +32,8 @@ type Client struct {
 	sessionTimeout time.Duration
 }
 
-func New(proxy, target string, options ...Option) (*Client, error) {
-	laddr, err := net.ResolveUDPAddr("udp", proxy)
+func New(port, target string, options ...Option) (*Client, error) {
+	laddr, err := net.ResolveUDPAddr("udp", port)
 	if err != nil {
 		return nil, err
 	}
@@ -32,7 +47,7 @@ func New(proxy, target string, options ...Option) (*Client, error) {
 		laddr:          laddr,
 		raddr:          raddr,
 		sessionCache:   newSessionCache(),
-		sessionTimeout: 20 * time.Minute,
+		sessionTimeout: 5 * time.Minute,
 	}
 
 	for _, o := range options {
@@ -58,7 +73,7 @@ func (c *Client) ListenAndServe() error {
 			log.Error(err)
 		}
 
-		session, ok := c.sessionCache.read(caddr.String())
+		session, ok := c.sessionCache.Read(caddr.String())
 		if !ok {
 			session, err = newSession(caddr, c.raddr, c.proxyConn)
 			if err != nil {
@@ -66,7 +81,7 @@ func (c *Client) ListenAndServe() error {
 				continue
 			}
 
-			c.sessionCache.insert(caddr.String(), session)
+			c.sessionCache.Upsert(caddr.String(), session)
 		}
 
 		go session.proxyTo(buf[:n])
@@ -79,7 +94,7 @@ func (c *Client) pruneSessions() {
 	for range ticker.C {
 		for _, session := range c.sessionCache.data {
 			if time.Since(session.updateTime) > c.sessionTimeout {
-				c.sessionCache.delete(session.caddr.String())
+				c.sessionCache.Delete(session.caddr.String())
 				log.Tracef("session for %s deleted", session.caddr)
 			}
 		}