Xray-core/transport/internet/xdrive/local.go

package xdrive

import (
	"context"
	"os"
	"path"
	"path/filepath"
	"strings"

	"github.com/xtls/xray-core/common/errors"
)

const tempPrefix = ".xdrive-tmp-"

type localStorage struct {
	root string
}

func newLocalStorage(root string) (*localStorage, error) {
	if root == "" {
		return nil, errors.New(`empty "remoteFolder"`)
	}
	if err := os.MkdirAll(root, 0o700); err != nil {
		return nil, errors.New("failed to create remote folder").Base(err)
	}
	return &localStorage{root: root}, nil
}

func (s *localStorage) resolve(name string) (string, error) {
	clean := path.Clean("/" + name)
	if clean == "/" {
		return "", errors.New("invalid object name: ", name)
	}
	if strings.HasPrefix(path.Base(clean), tempPrefix) {
		return "", errors.New("reserved object name: ", name)
	}
	return filepath.Join(s.root, filepath.FromSlash(clean[1:])), nil
}

func (s *localStorage) Put(ctx context.Context, name string, data []byte) error {
	full, err := s.resolve(name)
	if err != nil {
		return err
	}
	dir := filepath.Dir(full)
	if err := os.MkdirAll(dir, 0o700); err != nil {
		return errors.New("failed to create folder ", dir).Base(err)
	}

	tmp, err := os.CreateTemp(dir, tempPrefix+"*")
	if err != nil {
		return errors.New("failed to create temp file in ", dir).Base(err)
	}
	tmpName := tmp.Name()
	defer os.Remove(tmpName)

	if _, err := tmp.Write(data); err != nil {
		tmp.Close()
		return errors.New("failed to write ", name).Base(err)
	}
	if err := tmp.Close(); err != nil {
		return errors.New("failed to close ", name).Base(err)
	}
	if err := os.Rename(tmpName, full); err != nil {
		return errors.New("failed to commit ", name).Base(err)
	}
	return nil
}

func (s *localStorage) Get(ctx context.Context, name string) ([]byte, error) {
	full, err := s.resolve(name)
	if err != nil {
		return nil, err
	}
	data, err := os.ReadFile(full)
	if err != nil {
		if os.IsNotExist(err) {
			return nil, errNotFound
		}
		return nil, errors.New("failed to read ", name).Base(err)
	}
	return data, nil
}

func (s *localStorage) Delete(ctx context.Context, name string) error {
	full, err := s.resolve(name)
	if err != nil {
		return err
	}
	if err := os.RemoveAll(full); err != nil {
		return errors.New("failed to delete ", name).Base(err)
	}
	return nil
}

func (s *localStorage) List(ctx context.Context, prefix string) ([]Entry, error) {
	full, err := s.resolve(prefix)
	if err != nil {
		return nil, err
	}
	entries, err := os.ReadDir(full)
	if err != nil {
		if os.IsNotExist(err) {
			return nil, nil
		}
		return nil, errors.New("failed to list ", prefix).Base(err)
	}
	found := make([]Entry, 0, len(entries))
	for _, entry := range entries {
		if strings.HasPrefix(entry.Name(), tempPrefix) {
			continue
		}
		found = append(found, Entry{Name: entry.Name()})
	}
	return found, nil
}

func (s *localStorage) Close() error {
	return nil
}