fix(offline-download): block cloud metadata endpoints (#2487)

jyxjjj · web-flow · commit ace7482bc2c5 · 2026-06-18T23:16:16.000+08:00
diff --git a/internal/net/serve.go b/internal/net/serve.go
@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"io"
 	"mime/multipart"
+	gonet "net"
 	"net/http"
 	"strconv"
 	"strings"
@@ -292,6 +293,31 @@ func NewHttpClient() *http.Client {
 
 	return &http.Client{
 		Timeout:   time.Hour * 48,
-		Transport: transport,
+		Transport: &safeTransport{base: transport},
 	}
 }
+
+type safeTransport struct {
+	base http.RoundTripper
+}
+
+func (t *safeTransport) RoundTrip(req *http.Request) (*http.Response, error) {
+	host := req.URL.Hostname()
+	addrs, err := gonet.DefaultResolver.LookupIPAddr(req.Context(), host)
+	if err != nil || len(addrs) == 0 {
+		return nil, errors.Wrapf(err, "failed to resolve host: %s", host)
+	}
+	for _, addr := range addrs {
+		if isCloudMetadataIP(addr.IP) {
+			return nil, ErrCloudMetadataEndpoint
+		}
+	}
+	return t.base.RoundTrip(req)
+}
+
+var ErrCloudMetadataEndpoint = errors.New("access to cloud metadata endpoint is not allowed")
+
+func isCloudMetadataIP(ip gonet.IP) bool {
+	ip = ip.To4()
+	return ip != nil && ip[0] == 169 && ip[1] == 254 && ip[2] == 169 && ip[3] == 254
+}
diff --git a/internal/offline_download/115/client.go b/internal/offline_download/115/client.go
@@ -62,18 +62,16 @@ func (p *Cloud115) AddURL(args *tool.AddUrlArgs) (string, error) {
 		return "", fmt.Errorf("unsupported storage driver for offline download, only 115 Cloud is supported")
 	}
 
-	ctx := context.Background()
-
-	if err := op.MakeDir(ctx, storage, actualPath); err != nil {
+	if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {
 		return "", err
 	}
 
-	parentDir, err := op.GetUnwrap(ctx, storage, actualPath)
+	parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)
 	if err != nil {
 		return "", err
 	}
 
-	hashs, err := driver115.OfflineDownload(ctx, []string{args.Url}, parentDir)
+	hashs, err := driver115.OfflineDownload(args.Ctx, []string{args.Url}, parentDir)
 	if err != nil || len(hashs) < 1 {
 		return "", fmt.Errorf("failed to add offline download task: %w", err)
 	}
diff --git a/internal/offline_download/115_open/client.go b/internal/offline_download/115_open/client.go
@@ -58,18 +58,16 @@ func (o *Open115) AddURL(args *tool.AddUrlArgs) (string, error) {
 		return "", fmt.Errorf("unsupported storage driver for offline download, only 115 Cloud is supported")
 	}
 
-	ctx := context.Background()
-
-	if err := op.MakeDir(ctx, storage, actualPath); err != nil {
+	if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {
 		return "", err
 	}
 
-	parentDir, err := op.GetUnwrap(ctx, storage, actualPath)
+	parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)
 	if err != nil {
 		return "", err
 	}
 
-	hashs, err := driver115Open.OfflineDownload(ctx, []string{args.Url}, parentDir)
+	hashs, err := driver115Open.OfflineDownload(args.Ctx, []string{args.Url}, parentDir)
 	if err != nil || len(hashs) < 1 {
 		return "", fmt.Errorf("failed to add offline download task: %w", err)
 	}
diff --git a/internal/offline_download/123/client.go b/internal/offline_download/123/client.go
@@ -58,15 +58,14 @@ func (*Pan123) AddURL(args *tool.AddUrlArgs) (string, error) {
 	if !ok {
 		return "", fmt.Errorf("unsupported storage driver for offline download, only 123Pan is supported")
 	}
-	ctx := context.Background()
-	if err := op.MakeDir(ctx, storage, actualPath); err != nil {
+	if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {
 		return "", err
 	}
-	parentDir, err := op.GetUnwrap(ctx, storage, actualPath)
+	parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)
 	if err != nil {
 		return "", err
 	}
-	taskID, err := driver123.OfflineDownload(ctx, args.Url, parentDir)
+	taskID, err := driver123.OfflineDownload(args.Ctx, args.Url, parentDir)
 	if err != nil {
 		return "", fmt.Errorf("failed to add offline download task: %w", err)
 	}
diff --git a/internal/offline_download/123_open/client.go b/internal/offline_download/123_open/client.go
@@ -56,16 +56,15 @@ func (*Open123) AddURL(args *tool.AddUrlArgs) (string, error) {
 	if !ok {
 		return "", fmt.Errorf("unsupported storage driver for offline download, only 123 Open is supported")
 	}
-	ctx := context.Background()
-	if err := op.MakeDir(ctx, storage, actualPath); err != nil {
+	if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {
 		return "", err
 	}
-	parentDir, err := op.GetUnwrap(ctx, storage, actualPath)
+	parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)
 	if err != nil {
 		return "", err
 	}
 	cb := setting.GetStr(conf.Pan123OpenOfflineDownloadCallbackUrl)
-	taskID, err := driver123Open.OfflineDownload(ctx, args.Url, parentDir, cb)
+	taskID, err := driver123Open.OfflineDownload(args.Ctx, args.Url, parentDir, cb)
 	if err != nil {
 		return "", fmt.Errorf("failed to add offline download task: %w", err)
 	}
diff --git a/internal/offline_download/http/client.go b/internal/offline_download/http/client.go
@@ -11,13 +11,13 @@ import (
 
 	"github.com/OpenListTeam/OpenList/v4/drivers/base"
 	"github.com/OpenListTeam/OpenList/v4/internal/model"
+	"github.com/OpenListTeam/OpenList/v4/internal/net"
 	"github.com/OpenListTeam/OpenList/v4/internal/offline_download/tool"
 	"github.com/OpenListTeam/OpenList/v4/pkg/http_range"
 	"github.com/OpenListTeam/OpenList/v4/pkg/utils"
 )
 
 type SimpleHttp struct {
-	client http.Client
 }
 
 func (s SimpleHttp) Name() string {
@@ -62,7 +62,7 @@ func (s SimpleHttp) Run(task *tool.DownloadTask) error {
 	if streamPut {
 		req.Header.Set("Range", "bytes=0-")
 	}
-	resp, err := s.client.Do(req)
+	resp, err := net.HttpClient().Do(req)
 	if err != nil {
 		return err
 	}
diff --git a/internal/offline_download/pikpak/pikpak.go b/internal/offline_download/pikpak/pikpak.go
@@ -63,18 +63,16 @@ func (p *PikPak) AddURL(args *tool.AddUrlArgs) (string, error) {
 		return "", fmt.Errorf("unsupported storage driver for offline download, only Pikpak is supported")
 	}
 
-	ctx := context.Background()
-
-	if err := op.MakeDir(ctx, storage, actualPath); err != nil {
+	if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {
 		return "", err
 	}
 
-	parentDir, err := op.GetUnwrap(ctx, storage, actualPath)
+	parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)
 	if err != nil {
 		return "", err
 	}
 
-	t, err := pikpakDriver.OfflineDownload(ctx, args.Url, parentDir, "")
+	t, err := pikpakDriver.OfflineDownload(args.Ctx, args.Url, parentDir, "")
 	if err != nil {
 		return "", fmt.Errorf("failed to add offline download task: %w", err)
 	}
diff --git a/internal/offline_download/thunder/thunder.go b/internal/offline_download/thunder/thunder.go
@@ -64,18 +64,16 @@ func (t *Thunder) AddURL(args *tool.AddUrlArgs) (string, error) {
 		return "", fmt.Errorf("unsupported storage driver for offline download, only Thunder is supported")
 	}
 
-	ctx := context.Background()
-
-	if err := op.MakeDir(ctx, storage, actualPath); err != nil {
+	if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {
 		return "", err
 	}
 
-	parentDir, err := op.GetUnwrap(ctx, storage, actualPath)
+	parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)
 	if err != nil {
 		return "", err
 	}
 
-	task, err := thunderDriver.OfflineDownload(ctx, args.Url, parentDir, "")
+	task, err := thunderDriver.OfflineDownload(args.Ctx, args.Url, parentDir, "")
 	if err != nil {
 		return "", fmt.Errorf("failed to add offline download task: %w", err)
 	}
diff --git a/internal/offline_download/thunder_browser/thunder_browser.go b/internal/offline_download/thunder_browser/thunder_browser.go
@@ -63,23 +63,21 @@ func (t *ThunderBrowser) AddURL(args *tool.AddUrlArgs) (string, error) {
 		return "", err
 	}
 
-	ctx := context.Background()
-
-	if err := op.MakeDir(ctx, storage, actualPath); err != nil {
+	if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {
 		return "", err
 	}
 
-	parentDir, err := op.GetUnwrap(ctx, storage, actualPath)
+	parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)
 	if err != nil {
 		return "", err
 	}
 
 	var task *thunder_browser.OfflineTask
 	switch v := storage.(type) {
 	case *thunder_browser.ThunderBrowser:
-		task, err = v.OfflineDownload(ctx, args.Url, parentDir, "")
+		task, err = v.OfflineDownload(args.Ctx, args.Url, parentDir, "")
 	case *thunder_browser.ThunderBrowserExpert:
-		task, err = v.OfflineDownload(ctx, args.Url, parentDir, "")
+		task, err = v.OfflineDownload(args.Ctx, args.Url, parentDir, "")
 	default:
 		return "", fmt.Errorf("unsupported storage driver for offline download, only ThunderBrowser is supported")
 	}
diff --git a/internal/offline_download/thunderx/thunderx.go b/internal/offline_download/thunderx/thunderx.go
@@ -58,18 +58,16 @@ func (t *ThunderX) AddURL(args *tool.AddUrlArgs) (string, error) {
 		return "", fmt.Errorf("unsupported storage driver for offline download, only ThunderX is supported")
 	}
 
-	ctx := context.Background()
-
-	if err := op.MakeDir(ctx, storage, actualPath); err != nil {
+	if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {
 		return "", err
 	}
 
-	parentDir, err := op.GetUnwrap(ctx, storage, actualPath)
+	parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)
 	if err != nil {
 		return "", err
 	}
 
-	task, err := thunderXDriver.OfflineDownload(ctx, args.Url, parentDir, "")
+	task, err := thunderXDriver.OfflineDownload(args.Ctx, args.Url, parentDir, "")
 	if err != nil {
 		return "", fmt.Errorf("failed to add offline download task: %w", err)
 	}
diff --git a/internal/offline_download/tool/base.go b/internal/offline_download/tool/base.go
@@ -1,6 +1,8 @@
 package tool
 
 import (
+	"context"
+
 	"github.com/OpenListTeam/OpenList/v4/internal/model"
 )
 
@@ -9,6 +11,7 @@ type AddUrlArgs struct {
 	UID     string
 	TempDir string
 	Signal  chan int
+	Ctx     context.Context
 }
 
 type Status struct {
diff --git a/internal/offline_download/tool/download.go b/internal/offline_download/tool/download.go
@@ -54,6 +54,7 @@ func (t *DownloadTask) Run() error {
 		t.Signal = nil
 	}()
 	gid, err := t.tool.AddURL(&AddUrlArgs{
+		Ctx:     t.Ctx(),
 		Url:     t.Url,
 		UID:     t.ID,
 		TempDir: t.TempDir,
diff --git a/internal/offline_download/transmission/client.go b/internal/offline_download/transmission/client.go
@@ -9,9 +9,11 @@ import (
 	"net/url"
 	"strconv"
 
+	"github.com/OpenListTeam/OpenList/v4/drivers/base"
 	"github.com/OpenListTeam/OpenList/v4/internal/conf"
 	"github.com/OpenListTeam/OpenList/v4/internal/errs"
 	"github.com/OpenListTeam/OpenList/v4/internal/model"
+	"github.com/OpenListTeam/OpenList/v4/internal/net"
 	"github.com/OpenListTeam/OpenList/v4/internal/offline_download/tool"
 	"github.com/OpenListTeam/OpenList/v4/internal/setting"
 	"github.com/OpenListTeam/OpenList/v4/pkg/utils"
@@ -84,7 +86,12 @@ func (t *Transmission) AddURL(args *tool.AddUrlArgs) (string, error) {
 	}
 	// http url for .torrent file
 	if endpoint.Scheme == "http" || endpoint.Scheme == "https" {
-		resp, err := http.Get(args.Url)
+		resp, err := net.RequestHttp(
+			args.Ctx,
+			http.MethodGet,
+			http.Header{"User-Agent": []string{base.UserAgent}},
+			args.Url,
+		)
 		if err != nil {
 			return "", errors.Wrap(err, "failed to get .torrent file")
 		}
@@ -106,7 +113,7 @@ func (t *Transmission) AddURL(args *tool.AddUrlArgs) (string, error) {
 		rpcPayload.Filename = &args.Url
 	}
 
-	torrent, err := t.client.TorrentAdd(context.TODO(), rpcPayload)
+	torrent, err := t.client.TorrentAdd(args.Ctx, rpcPayload)
 	if err != nil {
 		return "", err
 	}
@@ -123,7 +130,7 @@ func (t *Transmission) Remove(task *tool.DownloadTask) error {
 	if err != nil {
 		return err
 	}
-	err = t.client.TorrentRemove(context.TODO(), transmissionrpc.TorrentRemovePayload{
+	err = t.client.TorrentRemove(task.Ctx(), transmissionrpc.TorrentRemovePayload{
 		IDs:             []int64{gid},
 		DeleteLocalData: false,
 	})
@@ -135,7 +142,7 @@ func (t *Transmission) Status(task *tool.DownloadTask) (*tool.Status, error) {
 	if err != nil {
 		return nil, err
 	}
-	infos, err := t.client.TorrentGetAllFor(context.TODO(), []int64{gid})
+	infos, err := t.client.TorrentGetAllFor(task.Ctx(), []int64{gid})
 	if err != nil {
 		return nil, err
 	}

Original file line number	Diff line number	Diff line change
`@@ -62,18 +62,16 @@ func (p Cloud115) AddURL(args tool.AddUrlArgs) (string, error) {`
`62`	`62`	`return "", fmt.Errorf("unsupported storage driver for offline download, only 115 Cloud is supported")`
`63`	`63`	`}`
`64`	`64`
`65`		`- ctx := context.Background()`
`66`		`-`
`67`		`- if err := op.MakeDir(ctx, storage, actualPath); err != nil {`
	`65`	`+ if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {`
`68`	`66`	`return "", err`
`69`	`67`	`}`
`70`	`68`
`71`		`- parentDir, err := op.GetUnwrap(ctx, storage, actualPath)`
	`69`	`+ parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)`
`72`	`70`	`if err != nil {`
`73`	`71`	`return "", err`
`74`	`72`	`}`
`75`	`73`
`76`		`- hashs, err := driver115.OfflineDownload(ctx, []string{args.Url}, parentDir)`
	`74`	`+ hashs, err := driver115.OfflineDownload(args.Ctx, []string{args.Url}, parentDir)`
`77`	`75`	`if err != nil \|\| len(hashs) < 1 {`
`78`	`76`	`return "", fmt.Errorf("failed to add offline download task: %w", err)`
`79`	`77`	`}`
Original file line number	Diff line number	Diff line change
`@@ -58,18 +58,16 @@ func (o Open115) AddURL(args tool.AddUrlArgs) (string, error) {`
`58`	`58`	`return "", fmt.Errorf("unsupported storage driver for offline download, only 115 Cloud is supported")`
`59`	`59`	`}`
`60`	`60`
`61`		`- ctx := context.Background()`
`62`		`-`
`63`		`- if err := op.MakeDir(ctx, storage, actualPath); err != nil {`
	`61`	`+ if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {`
`64`	`62`	`return "", err`
`65`	`63`	`}`
`66`	`64`
`67`		`- parentDir, err := op.GetUnwrap(ctx, storage, actualPath)`
	`65`	`+ parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)`
`68`	`66`	`if err != nil {`
`69`	`67`	`return "", err`
`70`	`68`	`}`
`71`	`69`
`72`		`- hashs, err := driver115Open.OfflineDownload(ctx, []string{args.Url}, parentDir)`
	`70`	`+ hashs, err := driver115Open.OfflineDownload(args.Ctx, []string{args.Url}, parentDir)`
`73`	`71`	`if err != nil \|\| len(hashs) < 1 {`
`74`	`72`	`return "", fmt.Errorf("failed to add offline download task: %w", err)`
`75`	`73`	`}`
Original file line number	Diff line number	Diff line change
`@@ -58,15 +58,14 @@ func (Pan123) AddURL(args tool.AddUrlArgs) (string, error) {`
`58`	`58`	`if !ok {`
`59`	`59`	`return "", fmt.Errorf("unsupported storage driver for offline download, only 123Pan is supported")`
`60`	`60`	`}`
`61`		`- ctx := context.Background()`
`62`		`- if err := op.MakeDir(ctx, storage, actualPath); err != nil {`
	`61`	`+ if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {`
`63`	`62`	`return "", err`
`64`	`63`	`}`
`65`		`- parentDir, err := op.GetUnwrap(ctx, storage, actualPath)`
	`64`	`+ parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)`
`66`	`65`	`if err != nil {`
`67`	`66`	`return "", err`
`68`	`67`	`}`
`69`		`- taskID, err := driver123.OfflineDownload(ctx, args.Url, parentDir)`
	`68`	`+ taskID, err := driver123.OfflineDownload(args.Ctx, args.Url, parentDir)`
`70`	`69`	`if err != nil {`
`71`	`70`	`return "", fmt.Errorf("failed to add offline download task: %w", err)`
`72`	`71`	`}`
Original file line number	Diff line number	Diff line change
`@@ -56,16 +56,15 @@ func (Open123) AddURL(args tool.AddUrlArgs) (string, error) {`
`56`	`56`	`if !ok {`
`57`	`57`	`return "", fmt.Errorf("unsupported storage driver for offline download, only 123 Open is supported")`
`58`	`58`	`}`
`59`		`- ctx := context.Background()`
`60`		`- if err := op.MakeDir(ctx, storage, actualPath); err != nil {`
	`59`	`+ if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {`
`61`	`60`	`return "", err`
`62`	`61`	`}`
`63`		`- parentDir, err := op.GetUnwrap(ctx, storage, actualPath)`
	`62`	`+ parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)`
`64`	`63`	`if err != nil {`
`65`	`64`	`return "", err`
`66`	`65`	`}`
`67`	`66`	`cb := setting.GetStr(conf.Pan123OpenOfflineDownloadCallbackUrl)`
`68`		`- taskID, err := driver123Open.OfflineDownload(ctx, args.Url, parentDir, cb)`
	`67`	`+ taskID, err := driver123Open.OfflineDownload(args.Ctx, args.Url, parentDir, cb)`
`69`	`68`	`if err != nil {`
`70`	`69`	`return "", fmt.Errorf("failed to add offline download task: %w", err)`
`71`	`70`	`}`
Original file line number	Diff line number	Diff line change
`@@ -11,13 +11,13 @@ import (`
`11`	`11`
`12`	`12`	`"github.com/OpenListTeam/OpenList/v4/drivers/base"`
`13`	`13`	`"github.com/OpenListTeam/OpenList/v4/internal/model"`
	`14`	`+ "github.com/OpenListTeam/OpenList/v4/internal/net"`
`14`	`15`	`"github.com/OpenListTeam/OpenList/v4/internal/offline_download/tool"`
`15`	`16`	`"github.com/OpenListTeam/OpenList/v4/pkg/http_range"`
`16`	`17`	`"github.com/OpenListTeam/OpenList/v4/pkg/utils"`
`17`	`18`	`)`
`18`	`19`
`19`	`20`	`type SimpleHttp struct {`
`20`		`- client http.Client`
`21`	`21`	`}`
`22`	`22`
`23`	`23`	`func (s SimpleHttp) Name() string {`
`@@ -62,7 +62,7 @@ func (s SimpleHttp) Run(task *tool.DownloadTask) error {`
`62`	`62`	`if streamPut {`
`63`	`63`	`req.Header.Set("Range", "bytes=0-")`
`64`	`64`	`}`
`65`		`- resp, err := s.client.Do(req)`
	`65`	`+ resp, err := net.HttpClient().Do(req)`
`66`	`66`	`if err != nil {`
`67`	`67`	`return err`
`68`	`68`	`}`
Original file line number	Diff line number	Diff line change
`@@ -63,18 +63,16 @@ func (p PikPak) AddURL(args tool.AddUrlArgs) (string, error) {`
`63`	`63`	`return "", fmt.Errorf("unsupported storage driver for offline download, only Pikpak is supported")`
`64`	`64`	`}`
`65`	`65`
`66`		`- ctx := context.Background()`
`67`		`-`
`68`		`- if err := op.MakeDir(ctx, storage, actualPath); err != nil {`
	`66`	`+ if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {`
`69`	`67`	`return "", err`
`70`	`68`	`}`
`71`	`69`
`72`		`- parentDir, err := op.GetUnwrap(ctx, storage, actualPath)`
	`70`	`+ parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)`
`73`	`71`	`if err != nil {`
`74`	`72`	`return "", err`
`75`	`73`	`}`
`76`	`74`
`77`		`- t, err := pikpakDriver.OfflineDownload(ctx, args.Url, parentDir, "")`
	`75`	`+ t, err := pikpakDriver.OfflineDownload(args.Ctx, args.Url, parentDir, "")`
`78`	`76`	`if err != nil {`
`79`	`77`	`return "", fmt.Errorf("failed to add offline download task: %w", err)`
`80`	`78`	`}`
Original file line number	Diff line number	Diff line change
`@@ -64,18 +64,16 @@ func (t Thunder) AddURL(args tool.AddUrlArgs) (string, error) {`
`64`	`64`	`return "", fmt.Errorf("unsupported storage driver for offline download, only Thunder is supported")`
`65`	`65`	`}`
`66`	`66`
`67`		`- ctx := context.Background()`
`68`		`-`
`69`		`- if err := op.MakeDir(ctx, storage, actualPath); err != nil {`
	`67`	`+ if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {`
`70`	`68`	`return "", err`
`71`	`69`	`}`
`72`	`70`
`73`		`- parentDir, err := op.GetUnwrap(ctx, storage, actualPath)`
	`71`	`+ parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)`
`74`	`72`	`if err != nil {`
`75`	`73`	`return "", err`
`76`	`74`	`}`
`77`	`75`
`78`		`- task, err := thunderDriver.OfflineDownload(ctx, args.Url, parentDir, "")`
	`76`	`+ task, err := thunderDriver.OfflineDownload(args.Ctx, args.Url, parentDir, "")`
`79`	`77`	`if err != nil {`
`80`	`78`	`return "", fmt.Errorf("failed to add offline download task: %w", err)`
`81`	`79`	`}`
Original file line number	Diff line number	Diff line change
`@@ -58,18 +58,16 @@ func (t ThunderX) AddURL(args tool.AddUrlArgs) (string, error) {`
`58`	`58`	`return "", fmt.Errorf("unsupported storage driver for offline download, only ThunderX is supported")`
`59`	`59`	`}`
`60`	`60`
`61`		`- ctx := context.Background()`
`62`		`-`
`63`		`- if err := op.MakeDir(ctx, storage, actualPath); err != nil {`
	`61`	`+ if err := op.MakeDir(args.Ctx, storage, actualPath); err != nil {`
`64`	`62`	`return "", err`
`65`	`63`	`}`
`66`	`64`
`67`		`- parentDir, err := op.GetUnwrap(ctx, storage, actualPath)`
	`65`	`+ parentDir, err := op.GetUnwrap(args.Ctx, storage, actualPath)`
`68`	`66`	`if err != nil {`
`69`	`67`	`return "", err`
`70`	`68`	`}`
`71`	`69`
`72`		`- task, err := thunderXDriver.OfflineDownload(ctx, args.Url, parentDir, "")`
	`70`	`+ task, err := thunderXDriver.OfflineDownload(args.Ctx, args.Url, parentDir, "")`
`73`	`71`	`if err != nil {`
`74`	`72`	`return "", fmt.Errorf("failed to add offline download task: %w", err)`
`75`	`73`	`}`