backend/webdav/tus.go - master - githubmirror/rclone

backend/webdav/tus.go
112 строк3 KB
Nick Craig-Wood
webdav: tus: fix potential nil pointer crash GHSA-3x6r-wxxg-53vv
31 июл 2026, 15:21
31 июл 2026, 15:215871d98
package webdav

/*
   Chunked upload based on the tus protocol for ownCloud Infinite Scale
   See https://tus.io/protocols/resumable-upload
*/

import (
	"context"
	"fmt"
	"io"
	"net/http"
	"path/filepath"
	"strconv"

	"github.com/rclone/rclone/fs"
	"github.com/rclone/rclone/lib/rest"
)

func (o *Object) updateViaTus(ctx context.Context, in io.Reader, contentType string, src fs.ObjectInfo, options ...fs.OpenOption) (err error) {

	fn := filepath.Base(src.Remote())
	metadata := map[string]string{
		"filename": fn,
		"mtime":    strconv.FormatInt(src.ModTime(ctx).Unix(), 10),
		"filetype": contentType,
	}

	// Fingerprint is used to identify the upload when resuming. That is not yet implemented
	fingerprint := ""

	// create an upload from a file.
	upload := NewUpload(in, src.Size(), metadata, fingerprint)

	// create the uploader.
	uploader, err := o.CreateUploader(ctx, upload, options...)
	if err == nil {
		// start the uploading process.
		err = uploader.Upload(ctx, options...)
	}

	return err
}

func (f *Fs) getTusLocationOrRetry(ctx context.Context, resp *http.Response, err error) (bool, string, error) {

	// resp is nil if the HTTP transaction failed before a response
	// was received, eg on connection refused or reset
	if resp != nil {
		switch resp.StatusCode {
		case 201:
			location := resp.Header.Get("Location")
			return false, location, nil
		case 412:
			return false, "", ErrVersionMismatch
		case 413:
			return false, "", ErrLargeUpload
		}
	}

	retry, err := f.shouldRetry(ctx, resp, err)
	return retry, "", err
}

// CreateUploader creates a new upload to the server.
func (o *Object) CreateUploader(ctx context.Context, u *Upload, options ...fs.OpenOption) (*Uploader, error) {
	if u == nil {
		return nil, ErrNilUpload
	}

	// if c.Config.Resume && len(u.Fingerprint) == 0 {
	//		return nil, ErrFingerprintNotSet
	//	}

	l := int64(0)
	p := o.filePath()
	// cut the filename off
	dir, _ := filepath.Split(p)
	if dir == "" {
		dir = "/"
	}

	opts := rest.Opts{
		Method:        "POST",
		Path:          dir,
		NoResponse:    true,
		RootURL:       o.fs.endpointURL,
		ContentLength: &l,
		ExtraHeaders:  o.extraHeaders(ctx, o),
		Options:       options,
	}
	opts.ExtraHeaders["Upload-Length"] = strconv.FormatInt(u.size, 10)
	opts.ExtraHeaders["Upload-Metadata"] = u.EncodedMetadata()
	opts.ExtraHeaders["Tus-Resumable"] = "1.0.0"
	// opts.ExtraHeaders["mtime"] = strconv.FormatInt(src.ModTime(ctx).Unix(), 10)

	var tusLocation string
	// rclone http call
	err := o.fs.pacer.CallNoRetry(func() (bool, error) {
		var retry bool
		res, err := o.fs.srv.Call(ctx, &opts)
		retry, tusLocation, err = o.fs.getTusLocationOrRetry(ctx, res, err)
		return retry, err
	})
	if err != nil {
		return nil, fmt.Errorf("making upload directory failed: %w", err)
	}

	uploader := NewUploader(o.fs, tusLocation, u, 0)

	return uploader, nil
}