aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/manager/interfacecleanup.go
blob: 32976b7986ded53c264a7232c56a35a571d5d8a7 (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
87
88
89
/* 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/conf"
	"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()

	if conf.AdminBool("ExperimentalKernelDriver") {
		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)
}