Release IP address on session stop only for Windows

Signed-off-by: Anton Litvinov <[email protected]>
mysteriumnetwork · Apr 30, 2024 · ab5ba2e · ab5ba2e
1 parent 537003d
commit ab5ba2e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/services/wireguard/connection/connection_test.go b/services/wireguard/connection/connection_test.go
@@ -158,6 +158,7 @@ func (mce *mockConnectionEndpoint) ConfigureRoutes(_ net.IP) error       { retur
 func (mce *mockConnectionEndpoint) PeerStats() (wgcfg.Stats, error) {
 	return wgcfg.Stats{LastHandshake: time.Now(), BytesSent: 10, BytesReceived: 11}, nil
 }
+func (mce *mockConnectionEndpoint) ReleaseIP(ip net.IPNet) {}
 
 type mockHandshakeWaiter struct {
 	err error

diff --git a/services/wireguard/service/service.go b/services/wireguard/service/service.go
@@ -21,6 +21,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"net"
+	"runtime"
 	"sync"
 	"time"
 
@@ -189,7 +190,9 @@ func (m *Manager) ProvideConfig(sessionID string, sessionConfig json.RawMessage,
 		if err := conn.Stop(); err != nil {
 			log.Error().Err(err).Msg("Failed to stop connection endpoint")
 		}
-		conn.ReleaseIP(config.Consumer.IPAddress)
+		if runtime.GOOS == "windows" {
+			conn.ReleaseIP(config.Consumer.IPAddress)
+		}
 
 		if err := m.resourcesAllocator.ReleaseIPNet(providerConfig.Subnet); err != nil {
 			log.Error().Err(err).Msg("Failed to release IP network")

diff --git a/services/wireguard/service/service_test.go b/services/wireguard/service/service_test.go
@@ -153,6 +153,7 @@ func (mce *mockConnectionEndpoint) ConfigureRoutes(_ net.IP) error       { retur
 func (mce *mockConnectionEndpoint) PeerStats() (wgcfg.Stats, error) {
 	return wgcfg.Stats{LastHandshake: time.Now()}, nil
 }
+func (mce *mockConnectionEndpoint) ReleaseIP(ip net.IPNet) {}
 
 func newManagerStub(pub, out, country string) *Manager {
 	dnsHandler, _ := dns.ResolveViaSystem()