From 34e3a000c561fd2c8ef65bdb5a296f0b03095a99 Mon Sep 17 00:00:00 2001
From: "Jason A. Donenfeld" <Jason@zx2c4.com>
Date: Sun, 5 Apr 2020 00:28:47 -0600
Subject: updater: allow updating from the command line

The administrator user may run `wireguard.exe /update`, which will check
for updates and install it if available. A log file may be written using
`wireguard.exe /update path\to\log\file.txt`.

Requested-by: Elliot Saba <staticfloat@gmail.com>
Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
---
 main.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

(limited to 'main.go')

diff --git a/main.go b/main.go
index 132753cd..79dfcdfc 100644
--- a/main.go
+++ b/main.go
@@ -7,6 +7,7 @@ package main
 
 import (
 	"fmt"
+	"log"
 	"os"
 	"strconv"
 	"strings"
@@ -20,6 +21,7 @@ import (
 	"golang.zx2c4.com/wireguard/windows/ringlogger"
 	"golang.zx2c4.com/wireguard/windows/tunnel"
 	"golang.zx2c4.com/wireguard/windows/ui"
+	"golang.zx2c4.com/wireguard/windows/updater"
 )
 
 func fatal(v ...interface{}) {
@@ -46,6 +48,7 @@ func usage() {
 		"/tunnelservice CONFIG_PATH",
 		"/ui CMD_READ_HANDLE CMD_WRITE_HANDLE CMD_EVENT_HANDLE LOG_MAPPING_HANDLE",
 		"/dumplog OUTPUT_PATH",
+		"/update [LOG_FILE]",
 	}
 	builder := strings.Builder{}
 	for _, flag := range flags {
@@ -224,6 +227,42 @@ func main() {
 			fatal(err)
 		}
 		return
+	case "/update":
+		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()
+		}
+		l := log.New(f, "", log.LstdFlags)
+		for progress := range updater.DownloadVerifyAndExecute(0) {
+			if len(progress.Activity) > 0 {
+				if progress.BytesTotal > 0 || progress.BytesDownloaded > 0 {
+					var percent float64
+					if progress.BytesTotal > 0 {
+						percent = float64(progress.BytesDownloaded) / float64(progress.BytesTotal) * 100.0
+					}
+					l.Printf("%s: %d/%d (%.2f%%)\n", progress.Activity, progress.BytesDownloaded, progress.BytesTotal, percent)
+				} else {
+					l.Println(progress.Activity)
+				}
+			}
+			if progress.Error != nil {
+				l.Printf("Error: %v\n", progress.Error)
+			}
+			if progress.Complete || progress.Error != nil {
+				return
+			}
+		}
+		return
 	}
 	usage()
 }
-- 
cgit v1.2.3-59-g8ed1b