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
|
package loose
import (
"errors"
"fmt"
"io/fs"
"os"
"path/filepath"
)
// Promote publishes all quarantined loose objects into the parent loose store
// and invalidates the receiver.
func (quarantine *objectQuarantine) Promote() error {
closeErr := quarantine.Close()
promoteErr := promoteLooseQuarantine(quarantine.parent, quarantine.tempName, quarantine.tempRoot)
tempRootErr := quarantine.tempRoot.Close()
removeErr := quarantine.parent.root.RemoveAll(quarantine.tempName)
if closeErr != nil {
return closeErr
}
if promoteErr != nil {
return promoteErr
}
if tempRootErr != nil {
return tempRootErr
}
return removeErr
}
func promoteLooseQuarantine(parent *Store, tempName string, tempRoot *os.Root) error {
entries, err := fs.ReadDir(tempRoot.FS(), ".")
if err != nil && !errors.Is(err, fs.ErrNotExist) {
return err
}
for _, entry := range entries {
if !entry.IsDir() {
return fmt.Errorf("objectstore/loose: quarantine contains unexpected file %q", entry.Name())
}
if len(entry.Name()) != 2 || !isHexString(entry.Name()) {
return fmt.Errorf("objectstore/loose: quarantine contains invalid shard %q", entry.Name())
}
err := promoteLooseQuarantineShard(parent, tempName, tempRoot, entry.Name())
if err != nil {
return err
}
}
return nil
}
func promoteLooseQuarantineShard(parent *Store, tempName string, tempRoot *os.Root, shard string) error {
entries, err := fs.ReadDir(tempRoot.FS(), shard)
if err != nil {
return err
}
err = parent.root.MkdirAll(shard, 0o755)
if err != nil {
return err
}
wantNameLen := parent.algo.HexLen() - 2
for _, entry := range entries {
if entry.IsDir() {
return fmt.Errorf("objectstore/loose: quarantine shard %q contains unexpected directory %q", shard, entry.Name())
}
if len(entry.Name()) != wantNameLen || !isHexString(entry.Name()) {
return fmt.Errorf("objectstore/loose: quarantine shard %q contains invalid object path %q", shard, entry.Name())
}
err := promoteLooseQuarantineObject(parent.root, filepath.Join(tempName, shard, entry.Name()), filepath.Join(shard, entry.Name()))
if err != nil {
return err
}
}
return nil
}
func promoteLooseQuarantineObject(root *os.Root, src, dst string) error {
err := root.Link(src, dst)
if err == nil {
_ = root.Remove(src)
return nil
}
if errors.Is(err, fs.ErrExist) {
_ = root.Remove(src)
return nil
}
return fmt.Errorf("objectstore/loose: promote quarantine %q -> %q: %w", src, dst, err)
}
func isHexString(s string) bool {
for _, ch := range s {
if ('0' <= ch && ch <= '9') || ('a' <= ch && ch <= 'f') || ('A' <= ch && ch <= 'F') {
continue
}
return false
}
return true
}
|
