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