Add a very long overall timeout for Autopilot downloads

This will ensure that the download will fail at some point, even if the remote server is artificially slow. Signed-off-by: Tom Wieczorek <[email protected]>
k0sproject · Sep 23, 2024 · 1a51cce · 1a51cce
1 parent 7f83b8f
commit 1a51cce
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/autopilot/download/downloader.go b/pkg/autopilot/download/downloader.go
@@ -23,6 +23,7 @@ import (
 	"hash"
 	"io"
 	"path/filepath"
+	"time"
 
 	internalhttp "github.com/k0sproject/k0s/internal/http"
 	"github.com/k0sproject/k0s/internal/pkg/file"
@@ -73,6 +74,12 @@ func (d *downloader) Download(ctx context.Context) (err error) {
 	defer func() { err = errors.Join(err, target.Close()) }()
 	targets = append(targets, target)
 
+	// Set a very long overall download timeout. This will ensure that the
+	// download will fail at some point, even if the remote server is
+	// artificially slow.
+	ctx, cancel := context.WithTimeout(ctx, 6*time.Hour)
+	defer cancel()
+
 	// Download from URL into targets.
 	var fileName string
 	if err = internalhttp.Download(ctx, d.config.URL, io.MultiWriter(targets...),