updater: immediately store executable

for some reason I kept the response body and downloaded file in memory,
which led to timeouts and failed updates.
pull/97/head
Harvey Tindall 4 years ago
parent d51a6abb02
commit 2687af31ca
No known key found for this signature in database
GPG Key ID: BBC65952848FB1A2

@ -7,6 +7,7 @@ import (
"errors"
"fmt"
"io"
"io/fs"
"net/http"
"os"
"path/filepath"
@ -390,11 +391,13 @@ func (ud *Updater) pullInternal(url string) (applyUpdate ApplyUpdate, status int
if err != nil || resp.StatusCode != 200 {
return
}
defer resp.Body.Close()
gz, err := gzip.NewReader(resp.Body)
if err != nil {
status = -1
return
}
defer gz.Close()
tarReader := tar.NewReader(gz)
var header *tar.Header
for {
@ -410,31 +413,33 @@ func (ud *Updater) pullInternal(url string) (applyUpdate ApplyUpdate, status int
case tar.TypeReg:
// Search only for file named ud.binary
if header.Name == ud.binary {
applyUpdate = func() error {
defer gz.Close()
defer resp.Body.Close()
file, err := os.Executable()
var file string
file, err = os.Executable()
if err != nil {
return err
return
}
path, err := filepath.EvalSymlinks(file)
var path string
path, err = filepath.EvalSymlinks(file)
if err != nil {
return err
return
}
info, err := os.Stat(path)
var info fs.FileInfo
info, err = os.Stat(path)
if err != nil {
return err
return
}
mode := info.Mode()
f, err := os.OpenFile(path+"_", os.O_RDWR|os.O_CREATE|os.O_TRUNC, mode)
var f *os.File
f, err = os.OpenFile(path+"_", os.O_RDWR|os.O_CREATE|os.O_TRUNC, mode)
if err != nil {
return err
return
}
defer f.Close()
_, err = io.Copy(f, tarReader)
if err != nil {
return err
return
}
applyUpdate = func() error {
return os.Rename(path+"_", path)
}
return

Loading…
Cancel
Save