sha256block_amd64.mx raw

   1  // Copyright 2017 The Go Authors. All rights reserved.
   2  // Use of this source code is governed by a BSD-style
   3  // license that can be found in the LICENSE file.
   4  
   5  //go:build !purego
   6  
   7  package sha256
   8  
   9  import (
  10  	"crypto/internal/fips140deps/cpu"
  11  	"crypto/internal/impl"
  12  )
  13  
  14  var useAVX2 = cpu.X86HasAVX && cpu.X86HasAVX2 && cpu.X86HasBMI2
  15  var useSHANI = cpu.X86HasAVX && cpu.X86HasSHA && cpu.X86HasSSE41 && cpu.X86HasSSSE3
  16  
  17  func init() {
  18  	impl.Register("sha256", "AVX2", &useAVX2)
  19  	impl.Register("sha256", "SHA-NI", &useSHANI)
  20  }
  21  
  22  //go:noescape
  23  func blockAVX2(dig *Digest, p []byte)
  24  
  25  //go:noescape
  26  func blockSHANI(dig *Digest, p []byte)
  27  
  28  func block(dig *Digest, p []byte) {
  29  	if useSHANI {
  30  		blockSHANI(dig, p)
  31  	} else if useAVX2 {
  32  		blockAVX2(dig, p)
  33  	} else {
  34  		blockGeneric(dig, p)
  35  	}
  36  }
  37