blob: d5a19c8d16055feed8ce8e96ac02ade68b491e0c (
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
|
/* SPDX-License-Identifier: MIT
*
* Copyright (C) 2019-2022 WireGuard LLC. All Rights Reserved.
*/
package manager
import (
"log"
"time"
_ "unsafe"
"golang.zx2c4.com/wireguard/windows/services"
"golang.zx2c4.com/wireguard/windows/updater"
"golang.zx2c4.com/wireguard/windows/version"
)
//go:linkname fastrandn runtime.fastrandn
func fastrandn(n uint32) uint32
type UpdateState uint32
const (
UpdateStateUnknown UpdateState = iota
UpdateStateFoundUpdate
UpdateStateUpdatesDisabledUnofficialBuild
)
var updateState = UpdateStateUnknown
func jitterSleep(min, max time.Duration) {
time.Sleep(min + time.Millisecond*time.Duration(fastrandn(uint32((max-min+1)/time.Millisecond))))
}
func checkForUpdates() {
if !version.IsRunningOfficialVersion() {
log.Println("Build is not official, so updates are disabled")
updateState = UpdateStateUpdatesDisabledUnofficialBuild
IPCServerNotifyUpdateFound(updateState)
return
}
if services.StartedAtBoot() {
jitterSleep(time.Minute*2, time.Minute*5)
}
noError, didNotify := true, false
for {
update, err := updater.CheckForUpdate()
if err == nil && update != nil && !didNotify {
log.Println("An update is available")
updateState = UpdateStateFoundUpdate
IPCServerNotifyUpdateFound(updateState)
didNotify = true
} else if err != nil && !didNotify {
log.Printf("Update checker: %v", err)
if noError {
jitterSleep(time.Minute*4, time.Minute*6)
noError = false
} else {
jitterSleep(time.Minute*25, time.Minute*30)
}
} else {
jitterSleep(time.Hour-time.Minute*3, time.Hour+time.Minute*3)
}
}
}
|