Merge pull request #977 from afbjorklund/windows-aes

Add AES detection for Windows using cpuid

Author: AkihiroSuda

go.mod

@@ -43,6 +43,7 @@ require (
 	github.com/golang/protobuf v1.5.2 // indirect
 	github.com/hashicorp/errwrap v1.1.0 // indirect
 	github.com/inconshreveable/mousetrap v1.0.0 // indirect
+	github.com/intel-go/cpuid v0.0.0-20220614022739-219e067757cb // indirect
 	github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51 // indirect
 	github.com/kr/fs v0.1.0 // indirect
 	github.com/kr/text v0.2.0 // indirect

go.sum

@@ -109,6 +109,8 @@ github.com/hinshun/vt10x v0.0.0-20220119200601-820417d04eec h1:qv2VnGeEQHchGaZ/u
 github.com/hinshun/vt10x v0.0.0-20220119200601-820417d04eec/go.mod h1:Q48J4R4DvxnHolD5P8pOtXigYlRuPLGl6moFx3ulM68=
 github.com/inconshreveable/mousetrap v1.0.0 h1:Z8tu5sraLXCXIcARxBp/8cbvlwVa7Z1NHg9XEKhtSvM=
 github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
+github.com/intel-go/cpuid v0.0.0-20220614022739-219e067757cb h1:Fg0Y/RDZ6UPwl3o7/IzPbneDq8g9+gH6DPs42KFUsy8=
+github.com/intel-go/cpuid v0.0.0-20220614022739-219e067757cb/go.mod h1:RmeVYf9XrPRbRc3XIx0gLYA8qOFvNoPOfaEZduRlEp4=
 github.com/jgautheron/goconst v0.0.0-20170703170152-9740945f5dcb/go.mod h1:82TxjOpWQiPmywlbIaB2ZkqJoSYJdLGPgAJDvM3PbKc=
 github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51 h1:Z9n2FFNUXsshfwJMBgNA0RU6/i7WVaAegv3PtuIHPMs=
 github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51/go.mod h1:CzGEWj7cYgsdH8dAjBGEr58BoE7ScuLd+fwFZ44+/x8=

pkg/sshutil/sshutil_others.go

@@ -1,5 +1,5 @@
-//go:build !darwin && !linux
-// +build !darwin,!linux
+//go:build !darwin && !linux && !windows
+// +build !darwin,!linux,!windows
 
 package sshutil
 

pkg/sshutil/sshutil_windows_amd64.go

@@ -0,0 +1,9 @@
+package sshutil
+
+import (
+	"github.com/intel-go/cpuid"
+)
+
+func detectAESAcceleration() bool {
+	return cpuid.HasFeature(cpuid.AES)
+}

pkg/sshutil/sshutil_windows_arm64.go

@@ -0,0 +1,11 @@
+package sshutil
+
+import (
+	"github.com/sirupsen/logrus"
+)
+
+func detectAESAcceleration() bool {
+	const fallback = false
+	logrus.Warnf("cannot detect whether AES accelerator is available, assuming %v", fallback)
+	return fallback
+}