2 meta4a -- Metalink 4.0 creator
3 Copyright (C) 2021-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/>.
18 // Metalink 4.0 creator
36 "go.cypherpunks.ru/gogost/v5/gost34112012256"
37 "go.cypherpunks.ru/gogost/v5/gost34112012512"
38 "go.stargrave.org/meta4ra"
39 "golang.org/x/crypto/sha3"
42 type MultiHasher struct {
45 streebog256h hash.Hash
46 streebog512h hash.Hash
47 shake128h sha3.ShakeHash
48 shake256h sha3.ShakeHash
52 func NewMultiHasher() *MultiHasher {
54 sha256h: sha256.New(),
55 sha512h: sha512.New(),
56 streebog256h: gost34112012256.New(),
57 streebog512h: gost34112012512.New(),
58 shake128h: sha3.NewShake128(),
59 shake256h: sha3.NewShake256(),
63 func (h *MultiHasher) Write(p []byte) (n int, err error) {
66 if _, err := h.sha256h.Write(p); err != nil {
72 if _, err := h.sha512h.Write(p); err != nil {
78 if _, err := h.streebog256h.Write(p); err != nil {
84 if _, err := h.streebog512h.Write(p); err != nil {
90 if _, err := h.shake128h.Write(p); err != nil {
96 if _, err := h.shake256h.Write(p); err != nil {
106 fn := flag.String("fn", "", "Filename")
107 mtime := flag.String("mtime", "", "Take that file's mtime as a Published date")
108 desc := flag.String("desc", "", "Description")
109 sig := flag.String("sig", "", "Path to signature file")
110 torrent := flag.String("torrent", "", "Torrent URL")
111 log.SetFlags(log.Lshortfile)
114 log.Fatalln("empty -fn")
116 urls := make([]meta4ra.URL, 0, len(flag.Args()))
117 for _, u := range flag.Args() {
118 urls = append(urls, meta4ra.URL{URL: u})
120 br := bufio.NewReaderSize(os.Stdin, 1<<20)
121 buf := make([]byte, 1<<20)
122 h := NewMultiHasher()
123 size, err := io.CopyBuffer(h, br, buf)
127 shake128 := make([]byte, 32)
128 if _, err = io.ReadFull(h.shake128h, shake128); err != nil {
131 shake256 := make([]byte, 64)
132 if _, err = io.ReadFull(h.shake256h, shake256); err != nil {
136 Name: path.Base(*fn),
140 Hashes: []meta4ra.Hash{
141 {Type: meta4ra.HashSHA256, Hash: hex.EncodeToString(h.sha256h.Sum(nil))},
142 {Type: meta4ra.HashSHA512, Hash: hex.EncodeToString(h.sha512h.Sum(nil))},
143 {Type: meta4ra.HashStreebog256, Hash: hex.EncodeToString(h.streebog256h.Sum(nil))},
144 {Type: meta4ra.HashStreebog512, Hash: hex.EncodeToString(h.streebog512h.Sum(nil))},
145 {Type: meta4ra.HashSHAKE128, Hash: hex.EncodeToString(shake128)},
146 {Type: meta4ra.HashSHAKE256, Hash: hex.EncodeToString(shake256)},
150 sigData, err := os.ReadFile(*sig)
154 f.Signature = &meta4ra.Signature{
155 MediaType: meta4ra.GPGSigMediaType,
156 Signature: "\n" + string(sigData),
160 f.MetaURLs = []meta4ra.MetaURL{{MediaType: "torrent", URL: *torrent}}
162 var published time.Time
164 published = time.Now()
166 fi, err := os.Stat(*mtime)
170 published = fi.ModTime()
172 published = published.UTC().Truncate(time.Second)
173 m := meta4ra.Metalink{
174 Files: []meta4ra.File{f},
175 Generator: meta4ra.Generator,
176 Published: published,
178 out, err := xml.MarshalIndent(&m, "", " ")
182 os.Stdout.Write([]byte(xml.Header))
184 os.Stdout.Write([]byte("\n"))