feeab84204
Currently, the cmd/restic package contains a significant amount of code that modifies repository internals. This code should in the mid-term move into the repository package.
108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/restic/restic/internal/backend"
|
|
"github.com/restic/restic/internal/errors"
|
|
"github.com/restic/restic/internal/repository"
|
|
"github.com/restic/restic/internal/restic"
|
|
"github.com/restic/restic/internal/ui/termstatus"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var cmdRepairPacks = &cobra.Command{
|
|
Use: "packs [packIDs...]",
|
|
Short: "Salvage damaged pack files",
|
|
Long: `
|
|
WARNING: The CLI for this command is experimental and will likely change in the future!
|
|
|
|
The "repair packs" command extracts intact blobs from the specified pack files, rebuilds
|
|
the index to remove the damaged pack files and removes the pack files from the repository.
|
|
|
|
EXIT STATUS
|
|
===========
|
|
|
|
Exit status is 0 if the command was successful, and non-zero if there was any error.
|
|
`,
|
|
DisableAutoGenTag: true,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
term, cancel := setupTermstatus()
|
|
defer cancel()
|
|
return runRepairPacks(cmd.Context(), globalOptions, term, args)
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
cmdRepair.AddCommand(cmdRepairPacks)
|
|
}
|
|
|
|
func runRepairPacks(ctx context.Context, gopts GlobalOptions, term *termstatus.Terminal, args []string) error {
|
|
// FIXME discuss and add proper feature flag mechanism
|
|
flag, _ := os.LookupEnv("RESTIC_FEATURES")
|
|
if flag != "repair-packs-v1" {
|
|
return errors.Fatal("This command is experimental and may change/be removed without notice between restic versions. " +
|
|
"Set the environment variable 'RESTIC_FEATURES=repair-packs-v1' to enable it.")
|
|
}
|
|
|
|
ids := restic.NewIDSet()
|
|
for _, arg := range args {
|
|
id, err := restic.ParseID(arg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ids.Insert(id)
|
|
}
|
|
if len(ids) == 0 {
|
|
return errors.Fatal("no ids specified")
|
|
}
|
|
|
|
repo, err := OpenRepository(ctx, gopts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
lock, ctx, err := lockRepoExclusive(ctx, repo, gopts.RetryLock, gopts.JSON)
|
|
defer unlockRepo(lock)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
bar := newIndexProgress(gopts.Quiet, gopts.JSON)
|
|
err = repo.LoadIndex(ctx, bar)
|
|
if err != nil {
|
|
return errors.Fatalf("%s", err)
|
|
}
|
|
|
|
printer := newTerminalProgressPrinter(gopts.verbosity, term)
|
|
|
|
printer.P("saving backup copies of pack files to current folder")
|
|
for id := range ids {
|
|
f, err := os.OpenFile("pack-"+id.String(), os.O_WRONLY|os.O_CREATE|os.O_EXCL, 0o666)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = repo.Backend().Load(ctx, backend.Handle{Type: restic.PackFile, Name: id.String()}, 0, 0, func(rd io.Reader) error {
|
|
_, err := f.Seek(0, 0)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = io.Copy(f, rd)
|
|
return err
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
err = repository.RepairPacks(ctx, repo, ids, printer)
|
|
if err != nil {
|
|
return errors.Fatalf("%s", err)
|
|
}
|
|
|
|
Warnf("\nUse `restic repair snapshots --forget` to remove the corrupted data blobs from all snapshots\n")
|
|
return nil
|
|
}
|