blake2b: test all three hashing implementations This adds use* flags for all arches so a common test can observe what implementations are supported and test all supported implementations. Change-Id: I0497b91294ed99309074ac4614ac996e006630c5 Reviewed-on: https://go-review.googlesource.com/31714 Run-TryBot: Austin Clements <austin@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Minux Ma <minux@golang.org> Reviewed-by: Andreas Auernhammer <aead@mail.de>

diff --git a/blake2b/blake2b_amd64.go b/blake2b/blake2b_amd64.go
index 374b5f7..f11dcbc 100644
--- a/blake2b/blake2b_amd64.go
+++ b/blake2b/blake2b_amd64.go

@@ -6,6 +6,7 @@
 
 package blake2b
 
+var useAVX2 = false
 var useSSE4 = supportSSE4()
 
 //go:noescape

diff --git a/blake2b/blake2b_ref.go b/blake2b/blake2b_ref.go
index da156a1..2c3c68b 100644
--- a/blake2b/blake2b_ref.go
+++ b/blake2b/blake2b_ref.go

@@ -6,6 +6,9 @@
 
 package blake2b
 
+var useAVX2 = false
+var useSSE4 = false
+
 func hashBlocks(h *[8]uint64, c *[2]uint64, flag uint64, blocks []byte) {
 	hashBlocksGeneric(h, c, flag, blocks)
 }

diff --git a/blake2b/blake2b_test.go b/blake2b/blake2b_test.go
index 54778d1..4607fda 100644
--- a/blake2b/blake2b_test.go
+++ b/blake2b/blake2b_test.go

@@ -21,6 +21,25 @@
 }
 
 func TestHashes(t *testing.T) {
+	defer func(sse4, avx2 bool) {
+		useSSE4, useAVX2 = sse4, avx2
+	}(useSSE4, useAVX2)
+
+	if useAVX2 {
+		t.Log("AVX2 version")
+		testHashes(t)
+		useAVX2 = false
+	}
+	if useSSE4 {
+		t.Log("SSE4 version")
+		testHashes(t)
+		useSSE4 = false
+	}
+	t.Log("generic version")
+	testHashes(t)
+}
+
+func testHashes(t *testing.T) {
 	key, _ := hex.DecodeString("000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f")
 
 	input := make([]byte, 255)