vendor/github.com/minio/sha256-simd/sha256blockSha_amd64_test.go

   1 //+build !noasm
   2
   3 package sha256
   4
   5 import (
   6         "crypto/sha256"
   7         "encoding/binary"
   8         "testing"
   9 )
  10
  11 func sha256hash(m []byte) (r [32]byte) {
  12         var h [8]uint32
  13
  14         h[0] = 0x6a09e667
  15         h[1] = 0xbb67ae85
  16         h[2] = 0x3c6ef372
  17         h[3] = 0xa54ff53a
  18         h[4] = 0x510e527f
  19         h[5] = 0x9b05688c
  20         h[6] = 0x1f83d9ab
  21         h[7] = 0x5be0cd19
  22
  23         blockSha(&h, m)
  24         l0 := len(m)
  25         l := l0 & (BlockSize - 1)
  26         m = m[l0-l:]
  27
  28         var k [64]byte
  29         copy(k[:], m)
  30
  31         k[l] = 0x80
  32
  33         if l >= 56 {
  34                 blockSha(&h, k[:])
  35                 binary.LittleEndian.PutUint64(k[0:8], 0)
  36                 binary.LittleEndian.PutUint64(k[8:16], 0)
  37                 binary.LittleEndian.PutUint64(k[16:24], 0)
  38                 binary.LittleEndian.PutUint64(k[24:32], 0)
  39                 binary.LittleEndian.PutUint64(k[32:40], 0)
  40                 binary.LittleEndian.PutUint64(k[40:48], 0)
  41                 binary.LittleEndian.PutUint64(k[48:56], 0)
  42         }
  43         binary.BigEndian.PutUint64(k[56:64], uint64(l0)<<3)
  44         blockSha(&h, k[:])
  45
  46         binary.BigEndian.PutUint32(r[0:4], h[0])
  47         binary.BigEndian.PutUint32(r[4:8], h[1])
  48         binary.BigEndian.PutUint32(r[8:12], h[2])
  49         binary.BigEndian.PutUint32(r[12:16], h[3])
  50         binary.BigEndian.PutUint32(r[16:20], h[4])
  51         binary.BigEndian.PutUint32(r[20:24], h[5])
  52         binary.BigEndian.PutUint32(r[24:28], h[6])
  53         binary.BigEndian.PutUint32(r[28:32], h[7])
  54
  55         return
  56 }
  57
  58 func runTestSha(hashfunc func([]byte) [32]byte) bool {
  59         var m = []byte("This is a message. This is a message. This is a message. This is a message.")
  60
  61         ar := hashfunc(m)
  62         br := sha256.Sum256(m)
  63
  64         return ar == br
  65 }
  66
  67 func TestSha0(t *testing.T) {
  68         if !runTestSha(Sum256) {
  69                 t.Errorf("FAILED")
  70         }
  71 }
  72
  73 func TestSha1(t *testing.T) {
  74         if sha && ssse3 && sse41 && !runTestSha(sha256hash) {
  75                 t.Errorf("FAILED")
  76         }
  77 }