1//+build !noasm,!appengine 2 3package sha256 4 5import ( 6 "crypto/sha256" 7 "encoding/binary" 8 "testing" 9) 10 11func 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 58func 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 67func TestSha0(t *testing.T) { 68 if !runTestSha(Sum256) { 69 t.Errorf("FAILED") 70 } 71} 72 73func TestSha1(t *testing.T) { 74 if sha && ssse3 && sse41 && !runTestSha(sha256hash) { 75 t.Errorf("FAILED") 76 } 77} 78