From 944cfe99f03034c37b2d963f7048c41c56f21b9f Mon Sep 17 00:00:00 2001 From: jet2tlf Date: Mon, 3 Jun 2024 14:54:41 -0300 Subject: codecrafters submit [skip ci] --- cmd/mybittorrent/meta.go | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 cmd/mybittorrent/meta.go (limited to 'cmd/mybittorrent/meta.go') diff --git a/cmd/mybittorrent/meta.go b/cmd/mybittorrent/meta.go new file mode 100644 index 0000000..34369c4 --- /dev/null +++ b/cmd/mybittorrent/meta.go @@ -0,0 +1,37 @@ +package main + +import ( + "crypto/sha1" + + bencode "github.com/jackpal/bencode-go" +) + +type Meta struct { + Announce string `bencode:"announce"` + Info FileInfo `bencode:"info"` +} + +type FileInfo struct { + Length int `bencode:"length"` + Name string `bencode:"name"` + PieceLength int `bencode:"piece length"` + Pieces string `bencode:"pieces"` +} + +func (m Meta) InfoHash() ([]byte, error) { + sha := sha1.New() + if err := bencode.Marshal(sha, m.Info); err != nil { + return nil, err + } + + return sha.Sum(nil), nil +} + +func (m Meta) PieceHashes() []string { + hashes := make([]string, 0, len(m.Info.Pieces)/20) + for i := 0; i < len(m.Info.Pieces); i += 20 { + hashes = append(hashes, m.Info.Pieces[i:i+20]) + } + + return hashes +} -- cgit v1.2.3