aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/manager/interfacecleanup.go
blob: 007c94d731bb1163f6029f2d3fc0fce53fd6e426 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
/* SPDX-License-Identifier: MIT
 *
 * Copyright (C) 2019-2021 WireGuard LLC. All Rights Reserved.
 */

package manager

import (
	"log"

	"golang.org/x/sys/windows"
	"golang.org/x/sys/windows/svc"
	"golang.org/x/sys/windows/svc/mgr"
	"golang.zx2c4.com/wireguard/tun"
	"golang.zx2c4.com/wireguard/tun/wintun"
	"golang.zx2c4.com/wireguard/windows/driver"
	"golang.zx2c4.com/wireguard/windows/services"
)

func cleanupStaleNetworkInterfaces() {
	m, err := mgr.Connect()
	if err != nil {
		return
	}
	defer m.Disconnect()

	driver.DefaultPool.DeleteMatchingAdapters(func(wintun *driver.Adapter) bool {
		interfaceName, err := wintun.Name()
		if err != nil {
			log.Printf("Removing network adapter because determining interface name failed: %v", err)
			return true
		}
		serviceName, err := services.ServiceNameOfTunnel(interfaceName)
		if err != nil {
			log.Printf("Removing network adapter ‘%s’ because determining tunnel service name failed: %v", interfaceName, err)
			return true
		}
		service, err := m.OpenService(serviceName)
		if err == windows.ERROR_SERVICE_DOES_NOT_EXIST {
			log.Printf("Removing network adapter ‘%s’ because no service for it exists", interfaceName)
			return true
		} else if err != nil {
			return false
		}
		defer service.Close()
		status, err := service.Query()
		if err != nil {
			return false
		}
		if status.State == svc.Stopped {
			log.Printf("Removing network adapter ‘%s’ because its service is stopped", interfaceName)
			return true
		}
		return false
	})

	tun.WintunPool.DeleteMatchingAdapters(func(adapter *wintun.Adapter) bool {
		interfaceName, err := adapter.Name()
		if err != nil {
			log.Printf("Removing network adapter because determining interface name failed: %v", err)
			return true
		}
		serviceName, err := services.ServiceNameOfTunnel(interfaceName)
		if err != nil {
			log.Printf("Removing network adapter ‘%s’ because determining tunnel service name failed: %v", interfaceName, err)
			return true
		}
		service, err := m.OpenService(serviceName)
		if err == windows.ERROR_SERVICE_DOES_NOT_EXIST {
			log.Printf("Removing network adapter ‘%s’ because no service for it exists", interfaceName)
			return true
		} else if err != nil {
			return false
		}
		defer service.Close()
		status, err := service.Query()
		if err != nil {
			return false
		}
		if status.State == svc.Stopped {
			log.Printf("Removing network adapter ‘%s’ because its service is stopped", interfaceName)
			return true
		}
		return false
	}, false)
}