main: add command line adapter cleanup

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>

1 files changed, 25 insertions, 0 deletions

diff --git a/main.go b/main.go
index 79dfcdfc..d7b28607 100644
--- a/main.go
+++ b/main.go
@@ -14,6 +14,7 @@ import (
 	"time"
 
 	"golang.org/x/sys/windows"
+	"golang.zx2c4.com/wireguard/tun"
 
 	"golang.zx2c4.com/wireguard/windows/elevate"
 	"golang.zx2c4.com/wireguard/windows/l18n"
@@ -49,6 +50,7 @@ func usage() {
 		"/ui CMD_READ_HANDLE CMD_WRITE_HANDLE CMD_EVENT_HANDLE LOG_MAPPING_HANDLE",
 		"/dumplog OUTPUT_PATH",
 		"/update [LOG_FILE]",
+		"/removealladapters [LOG_FILE]",
 	}
 	builder := strings.Builder{}
 	for _, flag := range flags {
@@ -263,6 +265,29 @@ func main() {
 			}
 		}
 		return
+	case "/removealladapters":
+		if len(os.Args) != 2 && len(os.Args) != 3 {
+			usage()
+		}
+		var f *os.File
+		var err error
+		if len(os.Args) == 2 {
+			f = os.Stdout
+		} else {
+			f, err = os.Create(os.Args[2])
+			if err != nil {
+				fatal(err)
+			}
+			defer f.Close()
+		}
+		log.SetOutput(f)
+		rebootRequired, err := tun.WintunPool.DeleteDriver()
+		if err != nil {
+			log.Printf("Error: %v\n", err)
+		} else if rebootRequired {
+			log.Println("A reboot may be required")
+		}
+		return
 	}
 	usage()
 }