112 lines
2.5 KiB
Go
112 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"time"
|
|
|
|
"golang.org/x/crypto/ssh/terminal"
|
|
"gopkg.in/urfave/cli.v2"
|
|
|
|
"github.com/equinox-io/equinox"
|
|
"github.com/facebookgo/grace/gracenet"
|
|
)
|
|
|
|
const (
|
|
appID = "app_idCzgxYerVD"
|
|
noAutoupdateMessage = "cloudflared will not automatically update when run from the shell. To enable auto-updates, run cloudflared as a service: https://developers.cloudflare.com/argo-tunnel/reference/service/"
|
|
)
|
|
|
|
var publicKey = []byte(`
|
|
-----BEGIN ECDSA PUBLIC KEY-----
|
|
MHYwEAYHKoZIzj0CAQYFK4EEACIDYgAE4OWZocTVZ8Do/L6ScLdkV+9A0IYMHoOf
|
|
dsCmJ/QZ6aw0w9qkkwEpne1Lmo6+0pGexZzFZOH6w5amShn+RXt7qkSid9iWlzGq
|
|
EKx0BZogHSor9Wy5VztdFaAaVbsJiCbO
|
|
-----END ECDSA PUBLIC KEY-----
|
|
`)
|
|
|
|
type ReleaseInfo struct {
|
|
Updated bool
|
|
Version string
|
|
Error error
|
|
}
|
|
|
|
func checkForUpdates() ReleaseInfo {
|
|
var opts equinox.Options
|
|
if err := opts.SetPublicKeyPEM(publicKey); err != nil {
|
|
return ReleaseInfo{Error: err}
|
|
}
|
|
|
|
resp, err := equinox.Check(appID, opts)
|
|
switch {
|
|
case err == equinox.NotAvailableErr:
|
|
return ReleaseInfo{}
|
|
case err != nil:
|
|
return ReleaseInfo{Error: err}
|
|
}
|
|
|
|
err = resp.Apply()
|
|
if err != nil {
|
|
return ReleaseInfo{Error: err}
|
|
}
|
|
|
|
return ReleaseInfo{Updated: true, Version: resp.ReleaseVersion}
|
|
}
|
|
|
|
func update(_ *cli.Context) error {
|
|
if updateApplied() {
|
|
os.Exit(64)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func initUpdate(listeners *gracenet.Net) bool {
|
|
if updateApplied() {
|
|
os.Args = append(os.Args, "--is-autoupdated=true")
|
|
if _, err := listeners.StartProcess(); err != nil {
|
|
logger.WithError(err).Error("Unable to restart server automatically")
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func autoupdate(freq time.Duration, listeners *gracenet.Net, shutdownC chan struct{}) {
|
|
for {
|
|
if updateApplied() {
|
|
os.Args = append(os.Args, "--is-autoupdated=true")
|
|
if _, err := listeners.StartProcess(); err != nil {
|
|
logger.WithError(err).Error("Unable to restart server automatically")
|
|
}
|
|
close(shutdownC)
|
|
return
|
|
}
|
|
time.Sleep(freq)
|
|
}
|
|
}
|
|
|
|
func updateApplied() bool {
|
|
releaseInfo := checkForUpdates()
|
|
if releaseInfo.Updated {
|
|
logger.Infof("Updated to version %s", releaseInfo.Version)
|
|
return true
|
|
}
|
|
if releaseInfo.Error != nil {
|
|
logger.WithError(releaseInfo.Error).Error("Update check failed")
|
|
}
|
|
return false
|
|
}
|
|
|
|
func isAutoupdateEnabled(c *cli.Context) bool {
|
|
if isRunningFromTerminal() {
|
|
logger.Info(noAutoupdateMessage)
|
|
return false
|
|
}
|
|
|
|
return !c.Bool("no-autoupdate") && c.Duration("autoupdate-freq") != 0
|
|
}
|
|
|
|
func isRunningFromTerminal() bool {
|
|
return terminal.IsTerminal(int(os.Stdout.Fd()))
|
|
}
|