diff --git a/blacktip.go b/blacktip.go index 7cb136e..5c7946a 100644 --- a/blacktip.go +++ b/blacktip.go @@ -38,6 +38,13 @@ type Config struct { DataDir string Docker bool // report local Docker containers for discovery + // Version is the release this binary was built from, reported to Hammerhead. AutoUpdate says + // whether the program may update itself; only then does it act on an update request from + // Hammerhead, by calling UpdateNow (which normally restarts the process into the new binary). + Version string + AutoUpdate bool + UpdateNow func(ctx context.Context) (newVersion string, err error) + // Traefik, when its Path is set, serves http/https resources instead of the built-in proxy. Traefik TraefikConfig @@ -106,6 +113,9 @@ type Blacktip struct { logq chan model.LogEntry reportMu sync.Mutex + updating atomic.Bool + updateMsg string // outcome of the update Hammerhead asked for, reported back + updateID string // the request that updateMsg answers dockerMu sync.Mutex dockerAt time.Time dockerList []model.Container @@ -252,6 +262,9 @@ func (b *Blacktip) apply(cfg *model.EdgeConfig) { if b.dashboard == "" { b.dashboard = b.cfg.Hammerhead } + if cfg.UpdateID != "" { + b.requestUpdate(cfg.UpdateID) + } b.local = map[string]bool{} for _, id := range cfg.Local { b.local[id] = true @@ -517,8 +530,10 @@ func (b *Blacktip) report() { IPs: localIPs(), PublicKey: b.wgPub, Status: map[string]model.MappingStatus{}, Stats: map[string]model.Stat{}, Containers: b.containers(), MailChecks: map[string]model.MailCheck{}, + Version: b.cfg.Version, AutoUpdate: b.cfg.AutoUpdate, } b.mu.Lock() + rep.UpdateMsg, rep.UpdateID = b.updateMsg, b.updateID for id, chk := range b.mailChecks { rep.MailChecks[id] = chk } @@ -736,3 +751,38 @@ func (ms *mapState) dialOff(proto string, prefer, off int) (net.Conn, bool, erro } return nil, false, last } + +// requestUpdate acts on "update now" from Hammerhead (id identifies the request). It only ever installs +// what this machine's own auto-update would (the newest release from its configured server), and only +// when auto-update is on. Caller holds b.mu. +func (b *Blacktip) requestUpdate(id string) { + if id == b.updateID { + return // already handled, or in progress + } + b.updateID, b.updateMsg = id, "" + if !b.cfg.AutoUpdate || b.cfg.UpdateNow == nil { + b.updateMsg = "auto-update is off on this machine (set BLACKTIP_AUTO_UPDATE=1 and restart it)" + go b.report() + return + } + if !b.updating.CompareAndSwap(false, true) { + b.updateMsg = "another update is already running" + go b.report() + return + } + go func() { + defer b.updating.Store(false) + tag, err := b.cfg.UpdateNow(b.ctx) // on success this restarts the process and does not return + b.mu.Lock() + switch { + case err != nil: + b.updateMsg = "update failed: " + err.Error() + case tag == "": + b.updateMsg = "already on the newest release" + default: + b.updateMsg = "updated to " + tag + ", restarting" + } + b.mu.Unlock() + b.report() + }() +} diff --git a/cmd/blacktip/main.go b/cmd/blacktip/main.go index 34859b2..ab67a39 100644 --- a/cmd/blacktip/main.go +++ b/cmd/blacktip/main.go @@ -50,9 +50,12 @@ func main() { flag.Usage() os.Exit(2) } + upd := selfupdate.Config{Server: "https://git.veillax.com", Repo: "Veillax/blacktip", App: "blacktip", Current: version, Logf: log.Printf} b, err := blacktip.New(blacktip.Config{ Hammerhead: strings.TrimRight(*hh, "/"), Token: *token, TunnelAddr: *tun, DataDir: *data, Docker: *dockerOn, Traefik: blacktip.TraefikConfig{Path: *traefikPath, ACMEDNS: *acmeDNS, NoACME: *noACME}, + Version: version, AutoUpdate: *autoUpdate, + UpdateNow: func(ctx context.Context) (string, error) { return selfupdate.CheckOnce(ctx, upd) }, }) if err != nil { log.Fatal(err) @@ -61,7 +64,7 @@ func main() { ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt) defer stop() if *autoUpdate { - go selfupdate.Run(ctx, selfupdate.Config{Server: "https://git.veillax.com", Repo: "Veillax/blacktip", App: "blacktip", Current: version, Logf: log.Printf}) + go selfupdate.Run(ctx, upd) } if err := b.Run(ctx); err != nil { log.Fatal(err) diff --git a/go.mod b/go.mod index c2d9851..c337d61 100644 --- a/go.mod +++ b/go.mod @@ -3,7 +3,7 @@ module git.veillax.com/Veillax/blacktip go 1.27.3 require ( - git.veillax.com/Veillax/reef v0.1.3 + git.veillax.com/Veillax/reef v0.1.6 github.com/emersion/go-msgauth v0.7.0 golang.org/x/crypto v0.58.0 ) diff --git a/go.sum b/go.sum index f233d8f..9526c6e 100644 --- a/go.sum +++ b/go.sum @@ -2,6 +2,10 @@ git.veillax.com/Veillax/reef v0.1.0 h1:O0OhEKFx2CsvACJ20sfzJNjGsngomtfDdKdxu4ROp git.veillax.com/Veillax/reef v0.1.0/go.mod h1:ZBUd/BARO5kkpUNTP81TTY1aQXLHFR6bKDhCxckjHso= git.veillax.com/Veillax/reef v0.1.3 h1:RxSq2xwaup1+MieAkhlvOTZnYwPf19thFHEAe/7VxsI= git.veillax.com/Veillax/reef v0.1.3/go.mod h1:ZBUd/BARO5kkpUNTP81TTY1aQXLHFR6bKDhCxckjHso= +git.veillax.com/Veillax/reef v0.1.5 h1:BydjLJsfGgU4zEIqUsePJRjPi9NgbFdmvq0M8dvJCgM= +git.veillax.com/Veillax/reef v0.1.5/go.mod h1:ZBUd/BARO5kkpUNTP81TTY1aQXLHFR6bKDhCxckjHso= +git.veillax.com/Veillax/reef v0.1.6 h1:NgZ4u2Z3hJ+BYD/zX9lov8fox4s2i8txtYNOpIsikVo= +git.veillax.com/Veillax/reef v0.1.6/go.mod h1:ZBUd/BARO5kkpUNTP81TTY1aQXLHFR6bKDhCxckjHso= github.com/Microsoft/go-winio v0.6.4 h1:kJF2pamtYCWXf0xAT8sAkwKC0pMl2HYBrRPHAqyUc+g= github.com/Microsoft/go-winio v0.6.4/go.mod h1:1xl5JXG4ze/m74GH6Dc4Mwxfc3VL5siRVQjIAlHH07U= github.com/emersion/go-msgauth v0.7.0 h1:vj2hMn6KhFtW41kshIBTXvp6KgYSqpA/ZN9Pv4g1INc=