package repository_test import ( "bytes" "context" "io" "testing" "github.com/restic/restic/internal/backend" "github.com/restic/restic/internal/backend/cache" "github.com/restic/restic/internal/backend/mem" "github.com/restic/restic/internal/backend/mock" "github.com/restic/restic/internal/errors" "github.com/restic/restic/internal/repository" "github.com/restic/restic/internal/restic" rtest "github.com/restic/restic/internal/test" ) const KiB = 1 << 10 const MiB = 1 << 20 func TestLoadRaw(t *testing.T) { b := mem.New() repo, err := repository.New(b, repository.Options{}) rtest.OK(t, err) for i := 0; i < 5; i++ { data := rtest.Random(23+i, 500*KiB) id := restic.Hash(data) h := backend.Handle{Name: id.String(), Type: backend.PackFile} err := b.Save(context.TODO(), h, backend.NewByteReader(data, b.Hasher())) rtest.OK(t, err) buf, err := repo.LoadRaw(context.TODO(), backend.PackFile, id) rtest.OK(t, err) if len(buf) != len(data) { t.Errorf("length of returned buffer does not match, want %d, got %d", len(data), len(buf)) continue } if !bytes.Equal(buf, data) { t.Errorf("wrong data returned") continue } } } func TestLoadRawBroken(t *testing.T) { b := mock.NewBackend() repo, err := repository.New(b, repository.Options{}) rtest.OK(t, err) data := rtest.Random(23, 10*KiB) id := restic.Hash(data) // damage buffer data[0] ^= 0xff b.OpenReaderFn = func(ctx context.Context, h backend.Handle, length int, offset int64) (io.ReadCloser, error) { return io.NopCloser(bytes.NewReader(data)), nil } // must detect but still return corrupt data buf, err := repo.LoadRaw(context.TODO(), backend.PackFile, id) rtest.Assert(t, bytes.Equal(buf, data), "wrong data returned") rtest.Assert(t, errors.Is(err, restic.ErrInvalidData), "missing expected ErrInvalidData error, got %v", err) // cause the first access to fail, but repair the data for the second access data[0] ^= 0xff loadCtr := 0 b.OpenReaderFn = func(ctx context.Context, h backend.Handle, length int, offset int64) (io.ReadCloser, error) { data[0] ^= 0xff loadCtr++ return io.NopCloser(bytes.NewReader(data)), nil } // must retry load of corrupted data buf, err = repo.LoadRaw(context.TODO(), backend.PackFile, id) rtest.OK(t, err) rtest.Assert(t, bytes.Equal(buf, data), "wrong data returned") rtest.Equals(t, 2, loadCtr, "missing retry on broken data") } func TestLoadRawBrokenWithCache(t *testing.T) { b := mock.NewBackend() c := cache.TestNewCache(t) repo, err := repository.New(b, repository.Options{}) rtest.OK(t, err) repo.UseCache(c) data := rtest.Random(23, 10*KiB) id := restic.Hash(data) loadCtr := 0 // cause the first access to fail, but repair the data for the second access b.OpenReaderFn = func(ctx context.Context, h backend.Handle, length int, offset int64) (io.ReadCloser, error) { data[0] ^= 0xff loadCtr++ return io.NopCloser(bytes.NewReader(data)), nil } // must retry load of corrupted data buf, err := repo.LoadRaw(context.TODO(), backend.SnapshotFile, id) rtest.OK(t, err) rtest.Assert(t, bytes.Equal(buf, data), "wrong data returned") rtest.Equals(t, 2, loadCtr, "missing retry on broken data") }