forked from FloatTech/zbpdata
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
67 lines (63 loc) · 1.43 KB
/
main.go
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
//go:build ignore
// +build ignore
package main
import (
"crypto/md5"
"encoding/hex"
"fmt"
"io/fs"
"os"
"strings"
"unicode"
"unsafe"
"github.com/fumiama/go-registry"
"github.com/wdvxdr1123/ZeroBot/utils/helper"
)
func main() {
var files []string
fs.WalkDir(os.DirFS("./"), ".", func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if !d.IsDir() && strings.Contains(path, "/") && unicode.IsUpper([]rune(path)[0]) {
files = append(files, path)
}
return nil
})
fmt.Println(files)
md5s := make([]string, len(files))
for i, fn := range files {
data, err := os.ReadFile(fn)
if err != nil {
panic(err)
}
buf := md5.Sum(data)
*(*unsafe.Pointer)(unsafe.Pointer(&md5s[i])) = unsafe.Pointer(&buf)
*(*uintptr)(unsafe.Add(unsafe.Pointer(&md5s[i]), unsafe.Sizeof(uintptr(0)))) = uintptr(16)
}
r := registry.NewRegedit("reilia.fumiama.top:32664", "", "fumiama", os.Getenv("REILIA_SPS"))
err := r.Connect()
if err != nil {
panic(err)
}
for i, fn := range files {
for c := 0; c < 5; c++ {
err = r.Set("data/"+fn, md5s[i])
fmt.Println("set", "data/"+fn, "=", hex.EncodeToString(helper.StringToBytes(md5s[i])))
if err == nil {
break
}
if c >= 4 {
panic("ERROR:" + err.Error() + "max retry times exceeded")
} else {
fmt.Println("ERROR:", err, ", retry times:", c)
}
_ = r.Close()
err = r.Connect()
if err != nil {
panic(err)
}
}
}
_ = r.Close()
}