11 Commits

Author SHA1 Message Date
renovate[bot]
184ea2acf2 fix(deps): update module golang.org/x/net to v0.51.0 (#175) 2026-02-28 06:55:09 -05:00
renovate[bot]
e2a3d5ab73 chore(deps): update goreleaser/goreleaser-action action to v7 (#174)
Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
2026-02-21 12:43:22 -05:00
github-actions[bot]
7948580d1d chore: update vendorHash in flake.nix 2026-02-16 14:41:10 +00:00
bb88db92c1 Refactor codebase 2026-02-16 09:40:34 -05:00
323709b0a1 Ignore windows arm 2026-02-14 07:49:46 -05:00
renovate[bot]
65509016cb chore(deps): update dependency go to v1.26.0 (#169)
* Update dependency go to v1.26.0

* Add nix dev shell

* Go mod tidy

* Remove version in glangci-lint

---------

Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
Co-authored-by: Dave Gallant <davegallant@proton.me>
2026-02-14 07:38:10 -05:00
renovate[bot]
9892ebe864 Update actions/checkout action to v6 (#172)
* Update actions/checkout action to v6

* chore: update vendorHash in flake.nix

---------

Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2026-02-14 07:15:06 -05:00
renovate[bot]
a3e448a658 Update cachix/install-nix-action action to v31 (#173)
* Update cachix/install-nix-action action to v31

* chore: update vendorHash in flake.nix

---------

Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2026-02-14 07:14:54 -05:00
github-actions[bot]
33a1b1b1d3 chore: update vendorHash in flake.nix 2026-02-14 12:13:48 +00:00
renovate[bot]
c03d27afcb Update module github.com/olekukonko/tablewriter to v1.1.3 (#171)
Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
2026-02-14 07:11:23 -05:00
4027784d1d Update flake.nix vendor-hash in a workflow 2026-02-14 06:59:08 -05:00
17 changed files with 295 additions and 190 deletions

1
.envrc Normal file
View File

@@ -0,0 +1 @@
use flake

View File

@@ -15,11 +15,9 @@ jobs:
steps:
- uses: actions/setup-go@v6
with:
go-version: 1.25
go-version: 1.26
- uses: actions/checkout@v6
- name: golangci-lint
uses: golangci/golangci-lint-action@v9
with:
version: v2.6.2
- name: test
run: make test

View File

@@ -18,12 +18,11 @@ jobs:
name: Set up Go
uses: actions/setup-go@v6
with:
go-version: "1.25"
go-version: "1.26"
-
name: Run GoReleaser
uses: goreleaser/goreleaser-action@v6
uses: goreleaser/goreleaser-action@v7
with:
version: '~> v2'
args: release --clean
env:
GITHUB_TOKEN: ${{ secrets.PERSONAL_GITHUB_TOKEN }}

View File

@@ -0,0 +1,58 @@
name: Update vendorHash in flake.nix
on:
push:
paths:
- 'go.mod'
- 'go.sum'
workflow_dispatch:
permissions:
contents: write
jobs:
update-hash:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v6
- uses: cachix/install-nix-action@v31
with:
nix_path: nixpkgs=channel:nixpkgs-unstable
- name: Calculate new vendorHash
id: hash
run: |
# Set vendorHash to empty string to trigger hash mismatch
sed -i 's|vendorHash = .*|vendorHash = "";|' flake.nix
# Try to build and extract the expected hash from error message
BUILD_OUTPUT=$(nix build .#vpngate 2>&1 || true)
HASH=$(echo "$BUILD_OUTPUT" | grep -oP 'got:\s*\K(sha256-[a-zA-Z0-9+/]+={0,2})' | head -1)
if [ -z "$HASH" ]; then
echo "Build output:"
echo "$BUILD_OUTPUT"
echo "Failed to extract hash from build output"
exit 1
fi
echo "hash=$HASH" >> $GITHUB_OUTPUT
echo "Calculated hash: $HASH"
- name: Update flake.nix with correct hash
run: |
sed -i "s|vendorHash = \"\";|vendorHash = \"${{ steps.hash.outputs.hash }}\";|" flake.nix
- name: Commit and push if changed
run: |
git config user.name "github-actions[bot]"
git config user.email "github-actions[bot]@users.noreply.github.com"
if git diff --quiet flake.nix; then
echo "No changes to commit"
else
git add flake.nix
git commit -m "chore: update vendorHash in flake.nix"
git push
fi

4
.gitignore vendored
View File

@@ -1,3 +1,7 @@
dist
.vscode
# direnv
.direnv
.envrc.local

View File

@@ -21,6 +21,9 @@ builds:
- arm64
goarm:
- "7"
ignore:
- goos: windows
goarch: arm
ldflags:
- -s -w
mod_timestamp: "{{ .CommitTimestamp }}"

View File

@@ -31,7 +31,6 @@ func init() {
var connectCmd = &cobra.Command{
Use: "connect",
Short: "Connect to a vpn server (survey selection appears if hostname is not provided)",
Long: `Connect to a vpn from a list of relay servers`,
Args: cobra.RangeArgs(0, 1),
@@ -39,49 +38,43 @@ var connectCmd = &cobra.Command{
vpnServers, err := vpn.GetList(flagProxy, flagSocks5Proxy)
if err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
serverSelection := []string{}
serverSelected := vpn.Server{}
for _, s := range *vpnServers {
serverSelection = append(serverSelection, fmt.Sprintf("%s (%s)", s.HostName, s.CountryLong))
// Build server selection options and hostname lookup map
serverSelection := make([]string, len(*vpnServers))
serverMap := make(map[string]vpn.Server, len(*vpnServers))
for i, s := range *vpnServers {
serverSelection[i] = fmt.Sprintf("%s (%s)", s.HostName, s.CountryLong)
serverMap[s.HostName] = s
}
selection := ""
var serverSelected vpn.Server
if !flagRandom {
if len(args) > 0 {
selection = args[0]
} else {
prompt := &survey.Select{
Message: "Choose a server:",
Options: serverSelection,
}
if !flagRandom {
if len(args) > 0 {
selection = args[0]
} else {
err := survey.AskOne(prompt, &selection, survey.WithPageSize(10))
if err != nil {
log.Error().Msg("Unable to obtain hostname from survey")
os.Exit(1)
log.Fatal().Msg("Unable to obtain hostname from survey")
}
}
// Server lookup from selection could be more optimized with a hash map
for _, s := range *vpnServers {
if strings.Contains(selection, s.HostName) {
serverSelected = s
}
}
if serverSelected.HostName == "" {
// Lookup server from selection using map for O(1) lookup
hostname := extractHostname(selection)
if server, exists := serverMap[hostname]; exists {
serverSelected = server
} else {
log.Fatal().Msgf("Server '%s' was not found", selection)
os.Exit(1)
}
}
for {
if flagRandom {
// Select a random server
serverSelected = (*vpnServers)[rand.Intn(len(*vpnServers))]
@@ -90,23 +83,19 @@ var connectCmd = &cobra.Command{
decodedConfig, err := base64.StdEncoding.DecodeString(serverSelected.OpenVpnConfigData)
if err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
tmpfile, err := os.CreateTemp("", "vpngate-openvpn-config-")
if err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
if _, err := tmpfile.Write(decodedConfig); err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
if err := tmpfile.Close(); err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
log.Info().Msgf("Connecting to %s (%s) in %s", serverSelected.HostName, serverSelected.IPAddr, serverSelected.CountryLong)
@@ -114,16 +103,22 @@ var connectCmd = &cobra.Command{
err = vpn.Connect(tmpfile.Name())
if err != nil && !flagReconnect {
log.Fatal().Msg(err.Error())
os.Exit(1)
} else {
err = os.Remove(tmpfile.Name())
if err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
// VPN connection failed and reconnect is disabled
_ = os.Remove(tmpfile.Name())
log.Fatal().Msg("VPN connection failed")
}
// Always try to clean up temporary file
_ = os.Remove(tmpfile.Name())
}
},
}
// extractHostname extracts the hostname from the selection string (format: "hostname (country)")
func extractHostname(selection string) string {
parts := strings.Split(selection, " (")
if len(parts) > 0 {
return parts[0]
}
return selection
}

View File

@@ -27,7 +27,6 @@ var listCmd = &cobra.Command{
vpnServers, err := vpn.GetList(flagProxy, flagSocks5Proxy)
if err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
table := tw.NewWriter(os.Stdout)
@@ -37,13 +36,11 @@ var listCmd = &cobra.Command{
err := table.Append([]string{strconv.Itoa(i + 1), v.HostName, v.CountryLong, v.Ping, strconv.Itoa(v.Score)})
if err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
}
err = table.Render()
if err != nil {
log.Fatal().Msg(err.Error())
os.Exit(1)
}
},
}

6
flake.lock generated
View File

@@ -20,11 +20,11 @@
},
"nixpkgs": {
"locked": {
"lastModified": 1763934636,
"narHash": "sha256-9glbI7f1uU+yzQCq5LwLgdZqx6svOhZWkd4JRY265fc=",
"lastModified": 1770843696,
"narHash": "sha256-LovWTGDwXhkfCOmbgLVA10bvsi/P8eDDpRudgk68HA8=",
"owner": "NixOS",
"repo": "nixpkgs",
"rev": "ee09932cedcef15aaf476f9343d1dea2cb77e261",
"rev": "2343bbb58f99267223bc2aac4fc9ea301a155a16",
"type": "github"
},
"original": {

View File

@@ -1,4 +1,6 @@
{
description = "vpngate - VPN server connector";
inputs = {
nixpkgs.url = "github:NixOS/nixpkgs/nixpkgs-unstable";
flake-utils.url = "github:numtide/flake-utils";
@@ -16,7 +18,7 @@
pkgs.buildGo125Module rec {
name = "vpngate";
src = ./.;
vendorHash = "sha256-tVNffrT+r3pA+0pvBaNKsq9K4wkB7WepkuSa1nCWloc=";
vendorHash = "sha256-ZPLGGyg056/wGOE90KaHjGc9eypHrGJBZMDg5KpBWqw=";
nativeBuildInputs = pkgs.lib.optionals pkgs.stdenv.isLinux [ pkgs.makeWrapper ];
env.CGO_ENABLED = 0;
doCheck = false;
@@ -33,12 +35,19 @@
default = vg;
vpngate = vg;
};
devShell = pkgs.mkShell {
devShells.default = pkgs.mkShell {
name = "vpngate-dev";
description = "Development environment for vpngate";
packages = with pkgs; [
go_1_26
gopls
gotools
go_1_25
golangci-lint
];
shellHook = ''
echo "Welcome to the vpngate dev environment"
go version
'';
};
};
in

13
go.mod
View File

@@ -1,23 +1,22 @@
module github.com/davegallant/vpngate
go 1.24.0
go 1.25.0
toolchain go1.25.6
toolchain go1.26.0
require (
github.com/AlecAivazis/survey/v2 v2.3.7
github.com/jszwec/csvutil v1.10.0
github.com/juju/errors v1.0.0
github.com/olekukonko/tablewriter v1.1.2
github.com/olekukonko/tablewriter v1.1.3
github.com/rs/zerolog v1.34.0
github.com/spf13/afero v1.15.0
github.com/spf13/cobra v1.10.2
github.com/stretchr/testify v1.11.1
golang.org/x/net v0.50.0
golang.org/x/net v0.51.0
)
require (
github.com/clipperhouse/displaywidth v0.6.0 // indirect
github.com/clipperhouse/displaywidth v0.6.2 // indirect
github.com/clipperhouse/stringish v0.1.1 // indirect
github.com/clipperhouse/uax29/v2 v2.3.0 // indirect
github.com/davecgh/go-spew v1.1.1 // indirect
@@ -31,7 +30,7 @@ require (
github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d // indirect
github.com/olekukonko/cat v0.0.0-20250911104152-50322a0618f6 // indirect
github.com/olekukonko/errors v1.1.0 // indirect
github.com/olekukonko/ll v0.1.3 // indirect
github.com/olekukonko/ll v0.1.4-0.20260115111900-9e59c2286df0 // indirect
github.com/pmezard/go-difflib v1.0.0 // indirect
github.com/rogpeppe/go-internal v1.14.1 // indirect
github.com/spf13/pflag v1.0.10 // indirect

16
go.sum
View File

@@ -2,8 +2,8 @@ github.com/AlecAivazis/survey/v2 v2.3.7 h1:6I/u8FvytdGsgonrYsVn2t8t4QiRnh6QSTqkk
github.com/AlecAivazis/survey/v2 v2.3.7/go.mod h1:xUTIdE4KCOIjsBAE1JYsUPoCqYdZ1reCfTwbto0Fduo=
github.com/Netflix/go-expect v0.0.0-20220104043353-73e0943537d2 h1:+vx7roKuyA63nhn5WAunQHLTznkw5W8b1Xc0dNjp83s=
github.com/Netflix/go-expect v0.0.0-20220104043353-73e0943537d2/go.mod h1:HBCaDeC1lPdgDeDbhX8XFpy1jqjK0IBG8W5K+xYqA0w=
github.com/clipperhouse/displaywidth v0.6.0 h1:k32vueaksef9WIKCNcoqRNyKbyvkvkysNYnAWz2fN4s=
github.com/clipperhouse/displaywidth v0.6.0/go.mod h1:R+kHuzaYWFkTm7xoMmK1lFydbci4X2CicfbGstSGg0o=
github.com/clipperhouse/displaywidth v0.6.2 h1:ZDpTkFfpHOKte4RG5O/BOyf3ysnvFswpyYrV7z2uAKo=
github.com/clipperhouse/displaywidth v0.6.2/go.mod h1:R+kHuzaYWFkTm7xoMmK1lFydbci4X2CicfbGstSGg0o=
github.com/clipperhouse/stringish v0.1.1 h1:+NSqMOr3GR6k1FdRhhnXrLfztGzuG+VuFDfatpWHKCs=
github.com/clipperhouse/stringish v0.1.1/go.mod h1:v/WhFtE1q0ovMta2+m+UbpZ+2/HEXNWYXQgCt4hdOzA=
github.com/clipperhouse/uax29/v2 v2.3.0 h1:SNdx9DVUqMoBuBoW3iLOj4FQv3dN5mDtuqwuhIGpJy4=
@@ -51,10 +51,10 @@ github.com/olekukonko/cat v0.0.0-20250911104152-50322a0618f6 h1:zrbMGy9YXpIeTnGj
github.com/olekukonko/cat v0.0.0-20250911104152-50322a0618f6/go.mod h1:rEKTHC9roVVicUIfZK7DYrdIoM0EOr8mK1Hj5s3JjH0=
github.com/olekukonko/errors v1.1.0 h1:RNuGIh15QdDenh+hNvKrJkmxxjV4hcS50Db478Ou5sM=
github.com/olekukonko/errors v1.1.0/go.mod h1:ppzxA5jBKcO1vIpCXQ9ZqgDh8iwODz6OXIGKU8r5m4Y=
github.com/olekukonko/ll v0.1.3 h1:sV2jrhQGq5B3W0nENUISCR6azIPf7UBUpVq0x/y70Fg=
github.com/olekukonko/ll v0.1.3/go.mod h1:b52bVQRRPObe+yyBl0TxNfhesL0nedD4Cht0/zx55Ew=
github.com/olekukonko/tablewriter v1.1.2 h1:L2kI1Y5tZBct/O/TyZK1zIE9GlBj/TVs+AY5tZDCDSc=
github.com/olekukonko/tablewriter v1.1.2/go.mod h1:z7SYPugVqGVavWoA2sGsFIoOVNmEHxUAAMrhXONtfkg=
github.com/olekukonko/ll v0.1.4-0.20260115111900-9e59c2286df0 h1:jrYnow5+hy3WRDCBypUFvVKNSPPCdqgSXIE9eJDD8LM=
github.com/olekukonko/ll v0.1.4-0.20260115111900-9e59c2286df0/go.mod h1:b52bVQRRPObe+yyBl0TxNfhesL0nedD4Cht0/zx55Ew=
github.com/olekukonko/tablewriter v1.1.3 h1:VSHhghXxrP0JHl+0NnKid7WoEmd9/urKRJLysb70nnA=
github.com/olekukonko/tablewriter v1.1.3/go.mod h1:9VU0knjhmMkXjnMKrZ3+L2JhhtsQ/L38BbL3CRNE8tM=
github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
@@ -66,8 +66,6 @@ github.com/rs/xid v1.6.0/go.mod h1:7XoLgs4eV+QndskICGsho+ADou8ySMSjJKDIan90Nz0=
github.com/rs/zerolog v1.34.0 h1:k43nTLIwcTVQAncfCw4KZ2VY6ukYoZaBPNOE8txlOeY=
github.com/rs/zerolog v1.34.0/go.mod h1:bJsvje4Z08ROH4Nhs5iH600c3IkWhwp44iRc54W6wYQ=
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
github.com/spf13/afero v1.15.0 h1:b/YBCLWAJdFWJTN9cLhiXXcD7mzKn9Dm86dNnfyQw1I=
github.com/spf13/afero v1.15.0/go.mod h1:NC2ByUVxtQs4b3sIUphxK0NioZnmxgyCrfzeuq8lxMg=
github.com/spf13/cobra v1.10.2 h1:DMTTonx5m65Ic0GOoRY2c16WCbHxOOw6xxezuLaBpcU=
github.com/spf13/cobra v1.10.2/go.mod h1:7C1pvHqHw5A4vrJfjNwvOdzYu0Gml16OCs2GRiTUUS4=
github.com/spf13/pflag v1.0.9/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
@@ -87,6 +85,8 @@ golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
golang.org/x/net v0.50.0 h1:ucWh9eiCGyDR3vtzso0WMQinm2Dnt8cFMuQa9K33J60=
golang.org/x/net v0.50.0/go.mod h1:UgoSli3F/pBgdJBHCTc+tp3gmrU4XswgGRgtnwWTfyM=
golang.org/x/net v0.51.0 h1:94R/GTO7mt3/4wIKpcR5gkGmRLOuE/2hNGeWq/GBIFo=
golang.org/x/net v0.51.0/go.mod h1:aamm+2QF5ogm02fjy5Bb7CQ0WMt1/WVM7FtyaTLlA9Y=
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=

View File

@@ -2,46 +2,58 @@ package exec
import (
"bufio"
"os"
"io"
"os/exec"
"strings"
"github.com/rs/zerolog/log"
)
// Run executes a command in workDir and returns stdout and error.
// The spawned process will exit upon termination of this application
// to ensure a clean exit
// Run executes a command in workDir and logs its output.
// If the command fails to start or setup fails, an error is logged and returned.
// If the command exits with a non-zero status, the error is returned without logging
// (this allows the caller to decide how to handle it).
func Run(path string, workDir string, args ...string) error {
_, err := exec.LookPath(path)
if err != nil {
log.Error().Msgf("%s is required, please install it", path)
os.Exit(1)
return err
}
cmd := exec.Command(path, args...)
cmd.Dir = workDir
log.Debug().Msg("Executing " + strings.Join(cmd.Args, " "))
log.Debug().Strs("command", cmd.Args).Msg("Executing command")
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Fatal().Msgf("Failed to get stdout pipe: %v", err)
log.Error().Msgf("Failed to get stdout pipe: %v", err)
return err
}
stderr, err := cmd.StderrPipe()
if err != nil {
log.Error().Msgf("Failed to get stderr pipe: %v", err)
return err
}
if err := cmd.Start(); err != nil {
log.Fatal().Msgf("Failed to start command: %v", err)
log.Error().Msgf("Failed to start command: %v", err)
return err
}
scanner := bufio.NewScanner(stdout)
// Combine stdout and stderr into a single reader
combined := io.MultiReader(stdout, stderr)
scanner := bufio.NewScanner(combined)
for scanner.Scan() {
log.Debug().Msg(scanner.Text())
}
if err := scanner.Err(); err != nil {
log.Fatal().Msgf("Error reading stdout: %v", err)
}
if err := cmd.Wait(); err != nil {
log.Fatal().Msgf("Command finished with error: %v", err)
log.Error().Msgf("Error reading output: %v", err)
return err
}
return nil
// cmd.Wait() returns an error if the command exits with non-zero status
// We return this without logging since it's expected behavior for some commands
return cmd.Wait()
}

View File

@@ -2,16 +2,17 @@ package util
import (
"time"
"github.com/rs/zerolog/log"
)
func Retry(attempts int, delay time.Duration,fn func() error) error {
func Retry(attempts int, delay time.Duration, fn func() error) error {
var err error
for i := 0; i < attempts; i++ {
if err = fn(); err == nil {
return nil
}
log.Error().Msgf("Retrying after %d seconds. An error occured: %s", delay, err)
log.Error().Msgf("Retrying after %v. An error occurred: %s", delay, err)
time.Sleep(delay)
}
return err

View File

@@ -4,37 +4,42 @@ import (
"encoding/json"
"io"
"os"
"path"
"path/filepath"
"time"
"github.com/rs/zerolog/log"
"github.com/spf13/afero"
)
const serverCachefile = "servers.json"
func getCacheDir() string {
func getCacheDir() (string, error) {
homeDir, err := os.UserHomeDir()
if err != nil {
log.Error().Msgf("Failed to get user's home directory: %s ", err)
return ""
return "", err
}
cacheDir := path.Join(homeDir, ".vpngate", "cache")
return cacheDir
cacheDir := filepath.Join(homeDir, ".vpngate", "cache")
return cacheDir, nil
}
func createCacheDir() error {
cacheDir := getCacheDir()
AppFs := afero.NewOsFs()
return AppFs.MkdirAll(cacheDir, 0o700)
cacheDir, err := getCacheDir()
if err != nil {
return err
}
return os.MkdirAll(cacheDir, 0o700)
}
func getVpnListCache() (*[]Server, error) {
cacheFile := path.Join(getCacheDir(), serverCachefile)
cacheDir, err := getCacheDir()
if err != nil {
return nil, err
}
cacheFile := filepath.Join(cacheDir, serverCachefile)
serversFile, err := os.Open(cacheFile)
if err != nil {
return nil, err
}
defer func() {
_ = serversFile.Close()
}()
byteValue, err := io.ReadAll(serversFile)
if err != nil {
@@ -44,7 +49,6 @@ func getVpnListCache() (*[]Server, error) {
var servers []Server
err = json.Unmarshal(byteValue, &servers)
if err != nil {
return nil, err
}
@@ -53,8 +57,7 @@ func getVpnListCache() (*[]Server, error) {
}
func writeVpnListToCache(servers []Server) error {
err := createCacheDir()
if err != nil {
if err := createCacheDir(); err != nil {
return err
}
@@ -63,20 +66,26 @@ func writeVpnListToCache(servers []Server) error {
return err
}
cacheFile := path.Join(getCacheDir(), serverCachefile)
err = os.WriteFile(cacheFile, f, 0o644)
cacheDir, err := getCacheDir()
if err != nil {
return err
}
cacheFile := filepath.Join(cacheDir, serverCachefile)
return os.WriteFile(cacheFile, f, 0o644)
}
func vpnListCacheIsExpired() bool {
file, err := os.Stat(path.Join(getCacheDir(), serverCachefile))
cacheDir, err := getCacheDir()
if err != nil {
return true
}
file, err := os.Stat(filepath.Join(cacheDir, serverCachefile))
if err != nil {
return true
}
lastModified := file.ModTime()
return (time.Since(lastModified)) > time.Duration(24*time.Hour)
return time.Since(lastModified) > 24*time.Hour
}

View File

@@ -1,28 +1,17 @@
package vpn
import (
"os"
"runtime"
"github.com/davegallant/vpngate/pkg/exec"
"github.com/juju/errors"
)
// Connect to a specified OpenVPN configuration
func Connect(configPath string) error {
tmpLogFile, err := os.CreateTemp("", "vpngate-openvpn-log-")
if err != nil {
return errors.Annotate(err, "Unable to create a temporary log file")
}
defer func() {
_ = os.Remove(tmpLogFile.Name())
}()
executable := "openvpn"
if runtime.GOOS == "windows" {
executable = "C:\\Program Files\\OpenVPN\\bin\\openvpn.exe"
}
err = exec.Run(executable, ".", "--verb", "4", "--config", configPath, "--data-ciphers", "AES-128-CBC")
return err
return exec.Run(executable, ".", "--verb", "4", "--config", configPath, "--data-ciphers", "AES-128-CBC")
}

View File

@@ -2,10 +2,12 @@ package vpn
import (
"bytes"
"context"
"io"
"net"
"net/http"
"net/url"
"os"
"time"
"github.com/jszwec/csvutil"
"github.com/rs/zerolog/log"
@@ -17,9 +19,11 @@ import (
const (
vpnList = "https://www.vpngate.net/api/iphone/"
httpClientTimeout = 30 * time.Second
dialTimeout = 10 * time.Second
)
// Server holds in formation about a vpn relay server
// Server holds information about a vpn relay server
type Server struct {
HostName string `csv:"#HostName"`
CountryLong string `csv:"CountryLong"`
@@ -30,20 +34,14 @@ type Server struct {
Ping string `csv:"Ping"`
}
func streamToBytes(stream io.Reader) []byte {
buf := new(bytes.Buffer)
_, err := buf.ReadFrom(stream)
if err != nil {
log.Error().Msg("Unable to stream bytes")
}
return buf.Bytes()
}
// parse csv
// parseVpnList parses the VPN server list from CSV format
func parseVpnList(r io.Reader) (*[]Server, error) {
var servers []Server
serverList := streamToBytes(r)
serverList, err := io.ReadAll(r)
if err != nil {
return nil, errors.Annotate(err, "Unable to read stream")
}
// Trim known invalid rows
serverList = bytes.TrimPrefix(serverList, []byte("*vpn_servers\r\n"))
@@ -57,86 +55,119 @@ func parseVpnList(r io.Reader) (*[]Server, error) {
return &servers, nil
}
// createHTTPClient creates an HTTP client with optional proxy configuration
func createHTTPClient(httpProxy string, socks5Proxy string) (*http.Client, error) {
if httpProxy != "" {
proxyURL, err := url.Parse(httpProxy)
if err != nil {
return nil, errors.Annotatef(err, "Error parsing HTTP proxy: %s", httpProxy)
}
transport := &http.Transport{
Proxy: http.ProxyURL(proxyURL),
}
return &http.Client{
Transport: transport,
Timeout: httpClientTimeout,
}, nil
}
if socks5Proxy != "" {
dialer, err := proxy.SOCKS5("tcp", socks5Proxy, nil, proxy.Direct)
if err != nil {
return nil, errors.Annotatef(err, "Error creating SOCKS5 dialer: %v", err)
}
// Create a DialContext function from the SOCKS5 dialer
dialContext := func(ctx context.Context, network, addr string) (net.Conn, error) {
// Check if context is already done
select {
case <-ctx.Done():
return nil, ctx.Err()
default:
}
// Use the dialer with a timeout
conn, err := dialer.Dial(network, addr)
if err != nil {
return nil, err
}
// Respect context cancellation after connection
go func() {
<-ctx.Done()
_ = conn.Close()
}()
return conn, nil
}
httpTransport := &http.Transport{
DialContext: dialContext,
}
return &http.Client{
Transport: httpTransport,
Timeout: httpClientTimeout,
}, nil
}
return &http.Client{
Timeout: httpClientTimeout,
Transport: &http.Transport{
DialContext: (&net.Dialer{
Timeout: dialTimeout,
}).DialContext,
},
}, nil
}
// GetList returns a list of vpn servers
func GetList(httpProxy string, socks5Proxy string) (*[]Server, error) {
cacheExpired := vpnListCacheIsExpired()
var servers *[]Server
var client *http.Client
// Try to use cached list if not expired
if !cacheExpired {
servers, err := getVpnListCache()
if err != nil {
log.Info().Msg("Unable to retrieve vpn list from cache")
} else {
if err == nil {
return servers, nil
}
log.Info().Msg("Unable to retrieve vpn list from cache")
} else {
log.Info().Msg("The vpn server list cache has expired")
}
log.Info().Msg("Fetching the latest server list")
if httpProxy != "" {
proxyURL, err := url.Parse(httpProxy)
client, err := createHTTPClient(httpProxy, socks5Proxy)
if err != nil {
log.Error().Msgf("Error parsing proxy: %s", err)
os.Exit(1)
}
transport := &http.Transport{
Proxy: http.ProxyURL(proxyURL),
return nil, err
}
client = &http.Client{
Transport: transport,
}
var servers *[]Server
} else if socks5Proxy != "" {
dialer, err := proxy.SOCKS5("tcp", socks5Proxy, nil, proxy.Direct)
if err != nil {
log.Error().Msgf("Error creating SOCKS5 dialer: %v", err)
os.Exit(1)
}
httpTransport := &http.Transport{
Dial: dialer.Dial,
}
client = &http.Client{
Transport: httpTransport,
}
} else {
client = &http.Client{}
}
var r *http.Response
err := util.Retry(5, 1, func() error {
var err error
r, err = client.Get(vpnList)
err = util.Retry(5, 1, func() error {
resp, err := client.Get(vpnList)
if err != nil {
return err
}
defer func() {
_ = r.Body.Close()
_ = resp.Body.Close()
}()
if r.StatusCode != 200 {
return errors.Annotatef(err, "Unexpected status code when retrieving vpn list: %d", r.StatusCode)
if resp.StatusCode != http.StatusOK {
return errors.Annotatef(err, "Unexpected status code when retrieving vpn list: %d", resp.StatusCode)
}
servers, err = parseVpnList(r.Body)
parsedServers, err := parseVpnList(resp.Body)
if err != nil {
return err
}
err = writeVpnListToCache(*servers)
servers = parsedServers
if err != nil {
log.Warn().Msgf("Unable to write servers to cache: %s", err)
// Cache the servers for future use
cacheErr := writeVpnListToCache(*servers)
if cacheErr != nil {
log.Warn().Msgf("Unable to write servers to cache: %s", cacheErr)
}
return nil
})