-
-
Notifications
You must be signed in to change notification settings - Fork 25
Expand file tree
/
Copy pathbackup.go
More file actions
133 lines (111 loc) · 2.25 KB
/
Copy pathbackup.go
File metadata and controls
133 lines (111 loc) · 2.25 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
package keyring
import (
"archive/tar"
"bytes"
"compress/gzip"
"io"
"io/ioutil"
"os"
"time"
"github.com/pkg/errors"
)
// Backup into {path}.tgz.
func Backup(path string, kr Keyring, now time.Time) error {
tmpPath := path + ".tmp"
defer func() { _ = os.Remove(tmpPath) }()
file, err := os.Create(tmpPath)
if err != nil {
return err
}
if err := backup(file, kr, now); err != nil {
_ = file.Close()
return err
}
if err := file.Close(); err != nil {
return err
}
if err := os.Rename(tmpPath, path); err != nil {
return err
}
return nil
}
func backup(file *os.File, kr Keyring, now time.Time) error {
gz := gzip.NewWriter(file)
tw := tar.NewWriter(gz)
items, err := kr.Items("")
if err != nil {
_ = tw.Close()
_ = gz.Close()
return err
}
for _, item := range items {
id := item.ID
b := item.Data
header := new(tar.Header)
header.Name = id
header.Size = int64(len(b))
header.Mode = 0600
header.ModTime = now
if err := tw.WriteHeader(header); err != nil {
_ = tw.Close()
_ = gz.Close()
return err
}
if _, err := io.Copy(tw, bytes.NewReader(item.Data)); err != nil {
_ = tw.Close()
_ = gz.Close()
return err
}
}
if err := tw.Close(); err != nil {
return err
}
if err := gz.Close(); err != nil {
return err
}
return nil
}
// Restore from path.tgz.
func Restore(path string, kr Keyring) error {
file, err := os.OpenFile(path, os.O_RDONLY, 0) // #nosec
if err != nil {
return errors.Wrapf(err, "failed to open backup")
}
if err := restore(file, kr); err != nil {
_ = file.Close()
return err
}
return file.Close()
}
func restore(file *os.File, kr Keyring) error {
gz, err := gzip.NewReader(file)
if err != nil {
return errors.Wrapf(err, "failed to open gzip")
}
tr := tar.NewReader(gz)
for {
header, err := tr.Next()
if err == io.EOF {
break
}
if err != nil {
return errors.Wrapf(err, "failed to read next tar")
}
switch header.Typeflag {
case tar.TypeDir:
continue
case tar.TypeReg:
b, err := ioutil.ReadAll(tr)
if err != nil {
return errors.Wrapf(err, "failed to read tar")
}
path := header.Name
if err := kr.Set(path, b); err != nil {
return err
}
default:
return errors.Errorf("invalid tar flag")
}
}
return nil
}