Files
kopia/cli/command_content_verify.go
Jarek Kowalski 70d4c8764a cli: improvements to content selection for list/rewrite/stats/verify (#409)
They now uniformly support 3 flags:

--prefix=P       selects contents with the specified prefix
--prefixed       selects contents with ANY prefix
--non-prefixed   selects non-prefixed contents

Also changed content manager iteration API to support ranges.

cli: add --prefix to 'blob gc' and 'blob stats'
2020-04-06 18:43:41 -07:00

100 lines
2.6 KiB
Go

package cli
import (
"context"
"sync/atomic"
"github.com/pkg/errors"
"github.com/kopia/kopia/repo"
"github.com/kopia/kopia/repo/blob"
"github.com/kopia/kopia/repo/content"
)
var (
contentVerifyCommand = contentCommands.Command("verify", "Verify that each content is backed by a valid blob")
contentVerifyParallel = contentVerifyCommand.Flag("parallel", "Parallelism").Default("16").Int()
contentVerifyFull = contentVerifyCommand.Flag("full", "Full verification (including download)").Bool()
)
func runContentVerifyCommand(ctx context.Context, rep *repo.DirectRepository) error {
blobMap := map[blob.ID]blob.Metadata{}
if !*contentVerifyFull {
printStderr("Listing blobs...\n")
if err := rep.Blobs.ListBlobs(ctx, "", func(bm blob.Metadata) error {
blobMap[bm.BlobID] = bm
if len(blobMap)%10000 == 0 {
printStderr(" %v blobs...\n", len(blobMap))
}
return nil
}); err != nil {
return errors.Wrap(err, "unable to list blobs")
}
printStderr("Listed %v blobs.\n", len(blobMap))
}
var totalCount, successCount, errorCount int32
printStderr("Verifying all contents...\n")
err := rep.Content.IterateContents(ctx, content.IterateOptions{
Range: contentIDRange(),
Parallel: *contentVerifyParallel,
}, func(ci content.Info) error {
if err := contentVerify(ctx, rep, &ci, blobMap); err != nil {
log(ctx).Errorf("error %v", err)
atomic.AddInt32(&errorCount, 1)
} else {
atomic.AddInt32(&successCount, 1)
}
if t := atomic.AddInt32(&totalCount, 1); t%100000 == 0 {
printStderr(" %v contents, %v errors...\n", t, atomic.LoadInt32(&errorCount))
}
return nil
})
if err != nil {
return errors.Wrap(err, "iterate contents")
}
printStderr("Finished verifying %v contents, found %v errors.\n", totalCount, errorCount)
if errorCount == 0 {
return nil
}
return errors.Errorf("encountered %v errors", errorCount)
}
func contentVerify(ctx context.Context, r *repo.DirectRepository, ci *content.Info, blobMap map[blob.ID]blob.Metadata) error {
if *contentVerifyFull {
if _, err := r.Content.GetContent(ctx, ci.ID); err != nil {
return errors.Wrapf(err, "content %v is invalid", ci.ID)
}
return nil
}
bi, ok := blobMap[ci.PackBlobID]
if !ok {
return errors.Errorf("content %v depends on missing blob %v", ci.ID, ci.PackBlobID)
}
if int64(ci.PackOffset+ci.Length) > bi.Length {
return errors.Errorf("content %v out of bounds of its pack blob %v", ci.ID, ci.PackBlobID)
}
return nil
}
func init() {
contentVerifyCommand.Action(directRepositoryAction(runContentVerifyCommand))
setupContentIDRangeFlags(contentVerifyCommand)
}