mirror of
https://github.com/restic/restic.git
synced 2025-03-11 22:53:23 +00:00
99e105eeb6
Those methods now only allow modifying snapshots. Internal data types used by the repository are now read-only. The repository-internal code can bypass the restrictions by wrapping the repository in an `internalRepository` type. The restriction itself is implemented by using a new datatype WriteableFileType in the SaveUnpacked and RemoveUnpacked methods. This statically ensures that code cannot bypass the access restrictions. The test changes are somewhat noisy as some of them modify repository internals and therefore require some way to bypass the access restrictions. This works by capturing an `internalRepository` or `Backend` when creating the Repository using a test helper function.
43 lines
970 B
Go
43 lines
970 B
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/restic/restic/internal/restic"
|
|
"golang.org/x/sync/errgroup"
|
|
)
|
|
|
|
// Test saving a blob and loading it again, with varying buffer sizes.
|
|
// Also a regression test for #3783.
|
|
func FuzzSaveLoadBlob(f *testing.F) {
|
|
f.Fuzz(func(t *testing.T, blob []byte, buflen uint) {
|
|
if buflen > 64<<20 {
|
|
// Don't allocate enormous buffers. We're not testing the allocator.
|
|
t.Skip()
|
|
}
|
|
|
|
id := restic.Hash(blob)
|
|
repo, _, _ := TestRepositoryWithVersion(t, 2)
|
|
|
|
var wg errgroup.Group
|
|
repo.StartPackUploader(context.TODO(), &wg)
|
|
|
|
_, _, _, err := repo.SaveBlob(context.TODO(), restic.DataBlob, blob, id, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = repo.Flush(context.TODO())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
buf, err := repo.LoadBlob(context.TODO(), restic.DataBlob, id, make([]byte, buflen))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if restic.Hash(buf) != id {
|
|
t.Fatal("mismatch")
|
|
}
|
|
})
|
|
}
|