seaweedfs/weed/replication/sink/localsink/local_sink_dir_test.go

package localsink

import (
	"os"
	"path/filepath"
	"testing"

	"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
	"github.com/seaweedfs/seaweedfs/weed/replication/source"
)

func TestLocalSinkStaysInsideDir(t *testing.T) {
	root := t.TempDir()
	dir := filepath.Join(root, "backup")
	sink := &LocalSink{}
	sink.initialize(dir, false)
	sink.SetSourceFiler(&source.FilerSource{})

	entry := func() *filer_pb.Entry {
		return &filer_pb.Entry{
			Name:       "marker.txt",
			Content:    []byte("marker"),
			Attributes: &filer_pb.FuseAttributes{FileMode: 0644, FileSize: 6},
		}
	}

	inside := filepath.Join(dir, "safe", "marker.txt")
	if err := sink.CreateEntry(inside, entry(), nil); err != nil {
		t.Fatalf("create inside: %v", err)
	}
	if _, err := os.Stat(inside); err != nil {
		t.Fatalf("inside file missing: %v", err)
	}

	outside := filepath.Join(dir, "..", "escape", "marker.txt")
	if err := sink.CreateEntry(outside, entry(), nil); err != nil {
		t.Fatalf("create outside: %v", err)
	}
	if _, err := sink.UpdateEntry(outside, nil, "", entry(), false, nil); err != nil {
		t.Fatalf("update outside: %v", err)
	}
	if _, err := os.Stat(filepath.Join(root, "escape")); !os.IsNotExist(err) {
		t.Fatalf("sink wrote outside %s: %v", dir, err)
	}

	victim := filepath.Join(root, "victim.txt")
	if err := os.WriteFile(victim, []byte("keep"), 0644); err != nil {
		t.Fatal(err)
	}
	if err := sink.DeleteEntry(filepath.Join(dir, "..", "victim.txt"), false, false, nil); err != nil {
		t.Fatalf("delete outside: %v", err)
	}
	if _, err := os.Stat(victim); err != nil {
		t.Fatalf("sink deleted a file outside %s: %v", dir, err)
	}
}

func TestLocalSinkDoesNotFollowSymlinksOutOfDir(t *testing.T) {
	root := t.TempDir()
	dir := filepath.Join(root, "backup")
	outside := filepath.Join(root, "outside")
	for _, d := range []string{dir, outside} {
		if err := os.MkdirAll(d, 0755); err != nil {
			t.Fatal(err)
		}
	}
	if err := os.Symlink(outside, filepath.Join(dir, "link")); err != nil {
		t.Skipf("symlink: %v", err)
	}
	sink := &LocalSink{}
	sink.initialize(dir, false)
	sink.SetSourceFiler(&source.FilerSource{})

	entry := &filer_pb.Entry{
		Name:       "marker.txt",
		Content:    []byte("marker"),
		Attributes: &filer_pb.FuseAttributes{FileMode: 0644, FileSize: 6},
	}
	if err := sink.CreateEntry(filepath.Join(dir, "link", "marker.txt"), entry, nil); err == nil {
		t.Fatalf("create through a symlink out of %s succeeded", dir)
	}
	if _, err := os.Stat(filepath.Join(outside, "marker.txt")); !os.IsNotExist(err) {
		t.Fatalf("sink wrote through a symlink out of %s: %v", dir, err)
	}

	victim := filepath.Join(outside, "victim.txt")
	if err := os.WriteFile(victim, []byte("keep"), 0644); err != nil {
		t.Fatal(err)
	}
	if err := sink.DeleteEntry(filepath.Join(dir, "link", "victim.txt"), false, false, nil); err != nil {
		t.Fatalf("delete through link: %v", err)
	}
	if _, err := os.Stat(victim); err != nil {
		t.Fatalf("sink deleted through a symlink out of %s: %v", dir, err)
	}

	if err := os.Mkdir(filepath.Join(dir, "real"), 0755); err != nil {
		t.Fatal(err)
	}
	if err := os.Symlink("real", filepath.Join(dir, "inner")); err != nil {
		t.Fatal(err)
	}
	if err := sink.CreateEntry(filepath.Join(dir, "inner", "marker.txt"), entry, nil); err != nil {
		t.Fatalf("create through a symlink inside %s: %v", dir, err)
	}
	if _, err := os.Stat(filepath.Join(dir, "real", "marker.txt")); err != nil {
		t.Fatalf("inside symlink target missing: %v", err)
	}
}

func TestLocalSinkFollowsAbsoluteSymlinksInsideDir(t *testing.T) {
	dir := filepath.Join(t.TempDir(), "backup")
	target := filepath.Join(dir, "real")
	if err := os.MkdirAll(target, 0755); err != nil {
		t.Fatal(err)
	}
	if err := os.Symlink(target, filepath.Join(dir, "abs")); err != nil {
		t.Skipf("symlink: %v", err)
	}
	sink := &LocalSink{}
	sink.initialize(dir, false)
	sink.SetSourceFiler(&source.FilerSource{})

	entry := &filer_pb.Entry{
		Name:       "marker.txt",
		Content:    []byte("marker"),
		Attributes: &filer_pb.FuseAttributes{FileMode: 0644, FileSize: 6},
	}
	for _, name := range []string{"marker.txt", filepath.Join("sub", "marker.txt")} {
		if err := sink.CreateEntry(filepath.Join(dir, "abs", name), entry, nil); err != nil {
			t.Fatalf("create %s through an absolute symlink inside %s: %v", name, dir, err)
		}
		if _, err := os.Stat(filepath.Join(target, name)); err != nil {
			t.Fatalf("absolute symlink target %s missing: %v", name, err)
		}
		if err := sink.DeleteEntry(filepath.Join(dir, "abs", name), false, false, nil); err != nil {
			t.Fatalf("delete %s through an absolute symlink: %v", name, err)
		}
		if _, err := os.Stat(filepath.Join(target, name)); !os.IsNotExist(err) {
			t.Fatalf("delete through an absolute symlink left %s: %v", name, err)
		}
	}
	if _, err := os.Lstat(filepath.Join(dir, "abs")); err != nil {
		t.Fatalf("symlink itself removed: %v", err)
	}
}