This commit is contained in:
Quentin McGaw
2020-06-04 22:13:49 +00:00
parent 900fa261d8
commit f6a2aac475

View File

@@ -9,22 +9,22 @@ import (
"github.com/qdm12/golibs/files" "github.com/qdm12/golibs/files"
"github.com/qdm12/golibs/network" "github.com/qdm12/golibs/network"
"github.com/qdm12/private-internet-access-docker/internal/params" "github.com/qdm12/private-internet-access-docker/internal/constants"
) )
func HealthCheck() error { func HealthCheck() error {
paramsReader := params.NewReader(nil) // Get all VPN ip addresses from openvpn configuration file
ipStatusFilepath, err := paramsReader.GetIPStatusFilepath()
if err != nil {
return err
}
// Get VPN ip address written to file
fileManager := files.NewFileManager() fileManager := files.NewFileManager()
b, err := fileManager.ReadFile(string(ipStatusFilepath)) b, err := fileManager.ReadFile(string(constants.OpenVPNConf))
if err != nil { if err != nil {
return err return err
} }
vpnIP := string(b) var vpnIPs []string
for _, line := range strings.Split(string(b), "\n") {
if strings.HasPrefix(line, "remote ") {
vpnIPs = append(vpnIPs, strings.TrimPrefix(line, "remote "))
}
}
// Get public IP address from one of the following urls // Get public IP address from one of the following urls
urls := []string{ urls := []string{
@@ -46,8 +46,15 @@ func HealthCheck() error {
return fmt.Errorf("Received unexpected status code %d from %s", status, url) return fmt.Errorf("Received unexpected status code %d from %s", status, url)
} }
publicIP := strings.ReplaceAll(string(content), "\n", "") publicIP := strings.ReplaceAll(string(content), "\n", "")
if publicIP != vpnIP { match := false
return fmt.Errorf("Public IP address %s does not match VPN ip address %s on file", publicIP, vpnIP) for _, vpnIP := range vpnIPs {
if publicIP == vpnIP {
match = true
break
}
}
if !match {
return fmt.Errorf("Public IP address %s does not match any of the VPN ip addresses %s", publicIP, strings.Join(vpnIPs, ", "))
} }
return nil return nil
} }