2 GoCheese -- Python private package repository and caching proxy
3 Copyright (C) 2019-2023 Sergey Matveev <stargrave@stargrave.org>
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, version 3 of the License.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>.
36 func checker(jobs chan string, isGood *bool, workers *sync.WaitGroup) {
37 hasherBLAKE2b256 := blake2b256New()
38 hasherSHA256 := sha256.New()
39 digestBuf := make([]byte, 32)
45 for fn := range jobs {
48 fmt.Println("ERR", fn, err)
53 digest, err = os.ReadFile(fn + "." + HashAlgoBLAKE2b256)
55 if !errors.Is(err, fs.ErrNotExist) {
56 fmt.Println("ERR", fn+"."+HashAlgoBLAKE2b256, err)
61 hasherBLAKE2b256.Reset()
63 _, err = io.Copy(hasherBLAKE2b256, &br)
64 fd.Seek(0, io.SeekStart)
66 fmt.Println("ERR", fn+"."+HashAlgoBLAKE2b256, err)
70 if bytes.Equal(hasherBLAKE2b256.Sum(digestBuf[:0]), digest) {
71 fmt.Println("GOOD", fn+"."+HashAlgoBLAKE2b256)
73 fmt.Println("BAD", fn+"."+HashAlgoBLAKE2b256)
78 digest, err = os.ReadFile(fn + "." + HashAlgoSHA256)
80 if !errors.Is(err, fs.ErrNotExist) {
81 fmt.Println("ERR", fn+"."+HashAlgoSHA256, err)
89 _, err = io.Copy(hasherSHA256, &br)
92 fmt.Println("ERR", fn+"."+HashAlgoBLAKE2b256, err)
96 if bytes.Equal(hasherSHA256.Sum(digestBuf[:0]), digest) {
97 fmt.Println("GOOD", fn+"."+HashAlgoSHA256)
99 fmt.Println("BAD", fn+"."+HashAlgoSHA256)
105 func goodIntegrity() bool {
107 jobs := make(chan string, 1<<10)
108 var workers sync.WaitGroup
109 workers.Add(runtime.NumCPU())
110 for i := 0; i < runtime.NumCPU(); i++ {
111 go checker(jobs, &isGood, &workers)
113 fd, err := os.Open(Root)
117 dirs, err := fd.ReadDir(0)
122 for _, dir := range dirs {
123 fd, err = os.Open(filepath.Join(Root, dir.Name()))
127 files, err := fd.ReadDir(0)
132 for _, file := range files {
133 for _, ext := range KnownExts {
134 if strings.HasSuffix(file.Name(), ext) {
135 jobs <- filepath.Join(Root, dir.Name(), file.Name())