metainfo/hash.go

   1 package metainfo
   2
   3 import (
   4         "crypto/sha1"
   5         "encoding"
   6         "encoding/hex"
   7         "fmt"
   8 )
   9
  10 const HashSize = 20
  11
  12 // 20-byte SHA1 hash used for info and pieces.
  13 type Hash [HashSize]byte
  14
  15 var (
  16         _ fmt.Formatter            = (*Hash)(nil)
  17         _ encoding.TextUnmarshaler = (*Hash)(nil)
  18 )
  19
  20 func (h Hash) Format(f fmt.State, c rune) {
  21         // TODO: I can't figure out a nice way to just override the 'x' rune, since it's meaningless
  22         // with the "default" 'v', or .String() already returning the hex.
  23         f.Write([]byte(h.HexString()))
  24 }
  25
  26 func (h Hash) Bytes() []byte {
  27         return h[:]
  28 }
  29
  30 func (h Hash) AsString() string {
  31         return string(h[:])
  32 }
  33
  34 func (h Hash) String() string {
  35         return h.HexString()
  36 }
  37
  38 func (h Hash) HexString() string {
  39         return fmt.Sprintf("%x", h[:])
  40 }
  41
  42 func (h *Hash) FromHexString(s string) (err error) {
  43         if len(s) != 2*HashSize {
  44                 err = fmt.Errorf("hash hex string has bad length: %d", len(s))
  45                 return
  46         }
  47         n, err := hex.Decode(h[:], []byte(s))
  48         if err != nil {
  49                 return
  50         }
  51         if n != HashSize {
  52                 panic(n)
  53         }
  54         return
  55 }
  56
  57 func (h *Hash) UnmarshalText(b []byte) error {
  58         return h.FromHexString(string(b))
  59 }
  60
  61 func NewHashFromHex(s string) (h Hash) {
  62         err := h.FromHexString(s)
  63         if err != nil {
  64                 panic(err)
  65         }
  66         return
  67 }
  68
  69 func HashBytes(b []byte) (ret Hash) {
  70         hasher := sha1.New()
  71         hasher.Write(b)
  72         copy(ret[:], hasher.Sum(nil))
  73         return
  74 }