forked from TrueCloudLab/restic
113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/restic/restic/backend"
|
|
"github.com/restic/restic/checker"
|
|
)
|
|
|
|
type CmdCheck struct {
|
|
ReadData bool `long:"read-data" description:"Read data blobs" default:"false"`
|
|
RemoveOrphaned bool `long:"remove" description:"Remove data that isn't used" default:"false"`
|
|
|
|
global *GlobalOptions
|
|
}
|
|
|
|
func init() {
|
|
_, err := parser.AddCommand("check",
|
|
"check the repository",
|
|
"The check command check the integrity and consistency of the repository",
|
|
&CmdCheck{global: &globalOpts})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (cmd CmdCheck) Usage() string {
|
|
return "[check-options]"
|
|
}
|
|
|
|
func (cmd CmdCheck) Execute(args []string) error {
|
|
if len(args) != 0 {
|
|
return errors.New("check has no arguments")
|
|
}
|
|
|
|
repo, err := cmd.global.OpenRepository()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.global.Verbosef("Create exclusive lock for repository\n")
|
|
lock, err := lockRepoExclusive(repo)
|
|
defer unlockRepo(lock)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
chkr := checker.New(repo)
|
|
|
|
cmd.global.Verbosef("Load indexes\n")
|
|
if err = chkr.LoadIndex(); err != nil {
|
|
return err
|
|
}
|
|
|
|
done := make(chan struct{})
|
|
defer close(done)
|
|
|
|
errorsFound := false
|
|
errChan := make(chan error)
|
|
|
|
cmd.global.Verbosef("Check all packs\n")
|
|
go chkr.Packs(errChan, done)
|
|
|
|
foundOrphanedPacks := false
|
|
for err := range errChan {
|
|
errorsFound = true
|
|
fmt.Fprintf(os.Stderr, "%v\n", err)
|
|
|
|
if e, ok := err.(checker.PackError); ok && e.Orphaned {
|
|
foundOrphanedPacks = true
|
|
}
|
|
}
|
|
|
|
cmd.global.Verbosef("Check snapshots, trees and blobs\n")
|
|
errChan = make(chan error)
|
|
go chkr.Structure(errChan, done)
|
|
|
|
for err := range errChan {
|
|
errorsFound = true
|
|
if e, ok := err.(checker.TreeError); ok {
|
|
fmt.Fprintf(os.Stderr, "error for tree %v:\n", e.ID.Str())
|
|
for _, treeErr := range e.Errors {
|
|
fmt.Fprintf(os.Stderr, " %v\n", treeErr)
|
|
}
|
|
} else {
|
|
fmt.Fprintf(os.Stderr, "error: %v\n", err)
|
|
}
|
|
}
|
|
|
|
for _, id := range chkr.UnusedBlobs() {
|
|
cmd.global.Verbosef("unused blob %v\n", id.Str())
|
|
}
|
|
|
|
if foundOrphanedPacks && cmd.RemoveOrphaned {
|
|
IDs := chkr.OrphanedPacks()
|
|
cmd.global.Verbosef("Remove %d orphaned packs... ", len(IDs))
|
|
|
|
for _, id := range IDs {
|
|
if err := repo.Backend().Remove(backend.Data, id.String()); err != nil {
|
|
fmt.Fprintf(os.Stderr, "%v\n", err)
|
|
}
|
|
}
|
|
|
|
cmd.global.Verbosef("done\n")
|
|
}
|
|
|
|
if errorsFound {
|
|
return errors.New("repository contains errors")
|
|
}
|
|
return nil
|
|
}
|