restore: Close files tested with --verify

Before, the target files were opened once per blob and never closed,
this commit fixes that.
This commit is contained in:
Alexander Neumann 2018-10-14 21:00:14 +02:00
parent 4c2072d875
commit c0fca3f50a
1 changed files with 13 additions and 5 deletions

View File

@ -306,25 +306,33 @@ func (res *Restorer) VerifyFiles(ctx context.Context, dst string) (int, error) {
return errors.Errorf("Invalid file size: expected %d got %d", node.Size, stat.Size())
}
file, err := os.Open(target)
if err != nil {
return err
}
offset := int64(0)
for _, blobID := range node.Content {
rd, err := os.Open(target)
if err != nil {
return err
}
blobs, _ := res.repo.Index().Lookup(blobID, restic.DataBlob)
length := blobs[0].Length - uint(crypto.Extension)
buf := make([]byte, length) // TODO do I want to reuse the buffer somehow?
_, err = rd.ReadAt(buf, offset)
_, err = file.ReadAt(buf, offset)
if err != nil {
_ = file.Close()
return err
}
if !blobID.Equal(restic.Hash(buf)) {
_ = file.Close()
return errors.Errorf("Unexpected contents starting at offset %d", offset)
}
offset += int64(length)
}
err = file.Close()
if err != nil {
return err
}
return nil
},
leaveDir: func(node *restic.Node, target, location string) error { return nil },