Handle CORS requests (#6289)
Tamal Saha
6 years ago
committed by
techknowlogick
parent
6fb58a8cdc
commit
34d06f4c6b
@ -0,0 +1,22 @@ |
||||
// Copyright 2019 The Gitea Authors. All rights reserved.
|
||||
// Use of this source code is governed by a MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package integrations |
||||
|
||||
import ( |
||||
"net/http" |
||||
"testing" |
||||
|
||||
"github.com/stretchr/testify/assert" |
||||
) |
||||
|
||||
func TestCORSNotSet(t *testing.T) { |
||||
prepareTestEnv(t) |
||||
req := NewRequestf(t, "GET", "/api/v1/version") |
||||
session := loginUser(t, "user2") |
||||
resp := session.MakeRequest(t, req, http.StatusOK) |
||||
assert.Equal(t, resp.Code, http.StatusOK) |
||||
corsHeader := resp.Header().Get("Access-Control-Allow-Origin") |
||||
assert.Equal(t, corsHeader, "", "Access-Control-Allow-Origin: generated header should match") // header not set
|
||||
} |
||||
@ -0,0 +1,41 @@ |
||||
// Copyright 2019 The Gitea Authors. All rights reserved.
|
||||
// Use of this source code is governed by a MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package setting |
||||
|
||||
import ( |
||||
"time" |
||||
|
||||
"code.gitea.io/gitea/modules/log" |
||||
|
||||
"github.com/go-macaron/cors" |
||||
) |
||||
|
||||
var ( |
||||
// CORSConfig defines CORS settings
|
||||
CORSConfig cors.Options |
||||
// EnableCORS defines whether CORS settings is enabled or not
|
||||
EnableCORS bool |
||||
) |
||||
|
||||
func newCORSService() { |
||||
sec := Cfg.Section("cors") |
||||
// Check cors setting.
|
||||
EnableCORS = sec.Key("ENABLED").MustBool(false) |
||||
|
||||
maxAge := sec.Key("MAX_AGE").MustDuration(10 * time.Minute) |
||||
|
||||
CORSConfig = cors.Options{ |
||||
Scheme: sec.Key("SCHEME").String(), |
||||
AllowDomain: sec.Key("ALLOW_DOMAIN").String(), |
||||
AllowSubdomain: sec.Key("ALLOW_SUBDOMAIN").MustBool(), |
||||
Methods: sec.Key("METHODS").Strings(","), |
||||
MaxAgeSeconds: int(maxAge.Seconds()), |
||||
AllowCredentials: sec.Key("ALLOW_CREDENTIALS").MustBool(), |
||||
} |
||||
|
||||
if EnableCORS { |
||||
log.Info("CORS Service Enabled") |
||||
} |
||||
} |
||||
@ -0,0 +1,8 @@ |
||||
module github.com/Unknwon/com |
||||
|
||||
require ( |
||||
github.com/gopherjs/gopherjs v0.0.0-20181103185306-d547d1d9531e // indirect |
||||
github.com/jtolds/gls v4.2.1+incompatible // indirect |
||||
github.com/smartystreets/assertions v0.0.0-20190116191733-b6c0e53d7304 // indirect |
||||
github.com/smartystreets/goconvey v0.0.0-20181108003508-044398e4856c |
||||
) |
||||
@ -0,0 +1,8 @@ |
||||
github.com/gopherjs/gopherjs v0.0.0-20181103185306-d547d1d9531e h1:JKmoR8x90Iww1ks85zJ1lfDGgIiMDuIptTOhJq+zKyg= |
||||
github.com/gopherjs/gopherjs v0.0.0-20181103185306-d547d1d9531e/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= |
||||
github.com/jtolds/gls v4.2.1+incompatible h1:fSuqC+Gmlu6l/ZYAoZzx2pyucC8Xza35fpRVWLVmUEE= |
||||
github.com/jtolds/gls v4.2.1+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= |
||||
github.com/smartystreets/assertions v0.0.0-20190116191733-b6c0e53d7304 h1:Jpy1PXuP99tXNrhbq2BaPz9B+jNAvH1JPQQpG/9GCXY= |
||||
github.com/smartystreets/assertions v0.0.0-20190116191733-b6c0e53d7304/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc= |
||||
github.com/smartystreets/goconvey v0.0.0-20181108003508-044398e4856c h1:Ho+uVpkel/udgjbwB5Lktg9BtvJSh2DT0Hi6LPSyI2w= |
||||
github.com/smartystreets/goconvey v0.0.0-20181108003508-044398e4856c/go.mod h1:XDJAKZRPZ1CvBcN2aX5YOUTYGHki24fSF0Iv48Ibg0s= |
||||
@ -0,0 +1,12 @@ |
||||
# Binaries for programs and plugins |
||||
*.exe |
||||
*.exe~ |
||||
*.dll |
||||
*.so |
||||
*.dylib |
||||
|
||||
# Test binary, build with `go test -c` |
||||
*.test |
||||
|
||||
# Output of the go coverage tool, specifically when used with LiteIDE |
||||
*.out |
||||
@ -0,0 +1,201 @@ |
||||
Apache License |
||||
Version 2.0, January 2004 |
||||
http://www.apache.org/licenses/ |
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION |
||||
|
||||
1. Definitions. |
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction, |
||||
and distribution as defined by Sections 1 through 9 of this document. |
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by |
||||
the copyright owner that is granting the License. |
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all |
||||
other entities that control, are controlled by, or are under common |
||||
control with that entity. For the purposes of this definition, |
||||
"control" means (i) the power, direct or indirect, to cause the |
||||
direction or management of such entity, whether by contract or |
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the |
||||
outstanding shares, or (iii) beneficial ownership of such entity. |
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity |
||||
exercising permissions granted by this License. |
||||
|
||||
"Source" form shall mean the preferred form for making modifications, |
||||
including but not limited to software source code, documentation |
||||
source, and configuration files. |
||||
|
||||
"Object" form shall mean any form resulting from mechanical |
||||
transformation or translation of a Source form, including but |
||||
not limited to compiled object code, generated documentation, |
||||
and conversions to other media types. |
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or |
||||
Object form, made available under the License, as indicated by a |
||||
copyright notice that is included in or attached to the work |
||||
(an example is provided in the Appendix below). |
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object |
||||
form, that is based on (or derived from) the Work and for which the |
||||
editorial revisions, annotations, elaborations, or other modifications |
||||
represent, as a whole, an original work of authorship. For the purposes |
||||
of this License, Derivative Works shall not include works that remain |
||||
separable from, or merely link (or bind by name) to the interfaces of, |
||||
the Work and Derivative Works thereof. |
||||
|
||||
"Contribution" shall mean any work of authorship, including |
||||
the original version of the Work and any modifications or additions |
||||
to that Work or Derivative Works thereof, that is intentionally |
||||
submitted to Licensor for inclusion in the Work by the copyright owner |
||||
or by an individual or Legal Entity authorized to submit on behalf of |
||||
the copyright owner. For the purposes of this definition, "submitted" |
||||
means any form of electronic, verbal, or written communication sent |
||||
to the Licensor or its representatives, including but not limited to |
||||
communication on electronic mailing lists, source code control systems, |
||||
and issue tracking systems that are managed by, or on behalf of, the |
||||
Licensor for the purpose of discussing and improving the Work, but |
||||
excluding communication that is conspicuously marked or otherwise |
||||
designated in writing by the copyright owner as "Not a Contribution." |
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity |
||||
on behalf of whom a Contribution has been received by Licensor and |
||||
subsequently incorporated within the Work. |
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of |
||||
this License, each Contributor hereby grants to You a perpetual, |
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable |
||||
copyright license to reproduce, prepare Derivative Works of, |
||||
publicly display, publicly perform, sublicense, and distribute the |
||||
Work and such Derivative Works in Source or Object form. |
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of |
||||
this License, each Contributor hereby grants to You a perpetual, |
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable |
||||
(except as stated in this section) patent license to make, have made, |
||||
use, offer to sell, sell, import, and otherwise transfer the Work, |
||||
where such license applies only to those patent claims licensable |
||||
by such Contributor that are necessarily infringed by their |
||||
Contribution(s) alone or by combination of their Contribution(s) |
||||
with the Work to which such Contribution(s) was submitted. If You |
||||
institute patent litigation against any entity (including a |
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work |
||||
or a Contribution incorporated within the Work constitutes direct |
||||
or contributory patent infringement, then any patent licenses |
||||
granted to You under this License for that Work shall terminate |
||||
as of the date such litigation is filed. |
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the |
||||
Work or Derivative Works thereof in any medium, with or without |
||||
modifications, and in Source or Object form, provided that You |
||||
meet the following conditions: |
||||
|
||||
(a) You must give any other recipients of the Work or |
||||
Derivative Works a copy of this License; and |
||||
|
||||
(b) You must cause any modified files to carry prominent notices |
||||
stating that You changed the files; and |
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works |
||||
that You distribute, all copyright, patent, trademark, and |
||||
attribution notices from the Source form of the Work, |
||||
excluding those notices that do not pertain to any part of |
||||
the Derivative Works; and |
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its |
||||
distribution, then any Derivative Works that You distribute must |
||||
include a readable copy of the attribution notices contained |
||||
within such NOTICE file, excluding those notices that do not |
||||
pertain to any part of the Derivative Works, in at least one |
||||
of the following places: within a NOTICE text file distributed |
||||
as part of the Derivative Works; within the Source form or |
||||
documentation, if provided along with the Derivative Works; or, |
||||
within a display generated by the Derivative Works, if and |
||||
wherever such third-party notices normally appear. The contents |
||||
of the NOTICE file are for informational purposes only and |
||||
do not modify the License. You may add Your own attribution |
||||
notices within Derivative Works that You distribute, alongside |
||||
or as an addendum to the NOTICE text from the Work, provided |
||||
that such additional attribution notices cannot be construed |
||||
as modifying the License. |
||||
|
||||
You may add Your own copyright statement to Your modifications and |
||||
may provide additional or different license terms and conditions |
||||
for use, reproduction, or distribution of Your modifications, or |
||||
for any such Derivative Works as a whole, provided Your use, |
||||
reproduction, and distribution of the Work otherwise complies with |
||||
the conditions stated in this License. |
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise, |
||||
any Contribution intentionally submitted for inclusion in the Work |
||||
by You to the Licensor shall be under the terms and conditions of |
||||
this License, without any additional terms or conditions. |
||||
Notwithstanding the above, nothing herein shall supersede or modify |
||||
the terms of any separate license agreement you may have executed |
||||
with Licensor regarding such Contributions. |
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade |
||||
names, trademarks, service marks, or product names of the Licensor, |
||||
except as required for reasonable and customary use in describing the |
||||
origin of the Work and reproducing the content of the NOTICE file. |
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or |
||||
agreed to in writing, Licensor provides the Work (and each |
||||
Contributor provides its Contributions) on an "AS IS" BASIS, |
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or |
||||
implied, including, without limitation, any warranties or conditions |
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A |
||||
PARTICULAR PURPOSE. You are solely responsible for determining the |
||||
appropriateness of using or redistributing the Work and assume any |
||||
risks associated with Your exercise of permissions under this License. |
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory, |
||||
whether in tort (including negligence), contract, or otherwise, |
||||
unless required by applicable law (such as deliberate and grossly |
||||
negligent acts) or agreed to in writing, shall any Contributor be |
||||
liable to You for damages, including any direct, indirect, special, |
||||
incidental, or consequential damages of any character arising as a |
||||
result of this License or out of the use or inability to use the |
||||
Work (including but not limited to damages for loss of goodwill, |
||||
work stoppage, computer failure or malfunction, or any and all |
||||
other commercial damages or losses), even if such Contributor |
||||
has been advised of the possibility of such damages. |
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing |
||||
the Work or Derivative Works thereof, You may choose to offer, |
||||
and charge a fee for, acceptance of support, warranty, indemnity, |
||||
or other liability obligations and/or rights consistent with this |
||||
License. However, in accepting such obligations, You may act only |
||||
on Your own behalf and on Your sole responsibility, not on behalf |
||||
of any other Contributor, and only if You agree to indemnify, |
||||
defend, and hold each Contributor harmless for any liability |
||||
incurred by, or claims asserted against, such Contributor by reason |
||||
of your accepting any such warranty or additional liability. |
||||
|
||||
END OF TERMS AND CONDITIONS |
||||
|
||||
APPENDIX: How to apply the Apache License to your work. |
||||
|
||||
To apply the Apache License to your work, attach the following |
||||
boilerplate notice, with the fields enclosed by brackets "[]" |
||||
replaced with your own identifying information. (Don't include |
||||
the brackets!) The text should be enclosed in the appropriate |
||||
comment syntax for the file format. We also recommend that a |
||||
file or class name and description of purpose be included on the |
||||
same "printed page" as the copyright notice for easier |
||||
identification within third-party archives. |
||||
|
||||
Copyright [yyyy] [name of copyright owner] |
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License"); |
||||
you may not use this file except in compliance with the License. |
||||
You may obtain a copy of the License at |
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0 |
||||
|
||||
Unless required by applicable law or agreed to in writing, software |
||||
distributed under the License is distributed on an "AS IS" BASIS, |
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
See the License for the specific language governing permissions and |
||||
limitations under the License. |
||||
@ -0,0 +1,2 @@ |
||||
# cors |
||||
Package cors is a middleware that handles CORS requests & headers for Macaron. |
||||
@ -0,0 +1,139 @@ |
||||
package cors |
||||
|
||||
import ( |
||||
"fmt" |
||||
"log" |
||||
"net/http" |
||||
"net/url" |
||||
"strconv" |
||||
"strings" |
||||
|
||||
macaron "gopkg.in/macaron.v1" |
||||
) |
||||
|
||||
const _VERSION = "0.1.0" |
||||
|
||||
func Version() string { |
||||
return _VERSION |
||||
} |
||||
|
||||
// Options represents a struct for specifying configuration options for the CORS middleware.
|
||||
type Options struct { |
||||
Section string |
||||
Scheme string |
||||
AllowDomain string |
||||
AllowSubdomain bool |
||||
Methods []string |
||||
MaxAgeSeconds int |
||||
AllowCredentials bool |
||||
} |
||||
|
||||
func prepareOptions(options []Options) Options { |
||||
var opt Options |
||||
if len(options) > 0 { |
||||
opt = options[0] |
||||
} |
||||
|
||||
if len(opt.Section) == 0 { |
||||
opt.Section = "cors" |
||||
} |
||||
sec := macaron.Config().Section(opt.Section) |
||||
|
||||
if len(opt.Scheme) == 0 { |
||||
opt.Scheme = sec.Key("SCHEME").MustString("http") |
||||
} |
||||
if len(opt.AllowDomain) == 0 { |
||||
opt.AllowDomain = sec.Key("ALLOW_DOMAIN").MustString("*") |
||||
} |
||||
if !opt.AllowSubdomain { |
||||
opt.AllowSubdomain = sec.Key("ALLOW_SUBDOMAIN").MustBool(false) |
||||
} |
||||
if len(opt.Methods) == 0 { |
||||
opt.Methods = sec.Key("METHODS").Strings(",") |
||||
if len(opt.Methods) == 0 { |
||||
opt.Methods = []string{ |
||||
http.MethodGet, |
||||
http.MethodHead, |
||||
http.MethodPost, |
||||
http.MethodPut, |
||||
http.MethodPatch, |
||||
http.MethodDelete, |
||||
http.MethodOptions, |
||||
} |
||||
} |
||||
} |
||||
if opt.MaxAgeSeconds <= 0 { |
||||
// cache options response for 600 secs
|
||||
// ref: https://stackoverflow.com/questions/54300997/is-it-possible-to-cache-http-options-response?noredirect=1#comment95790277_54300997
|
||||
// ref: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Access-Control-Max-Age
|
||||
opt.MaxAgeSeconds = sec.Key("MAX_AGE_SECONDS").MustInt(600) |
||||
} |
||||
if !opt.AllowCredentials { |
||||
opt.AllowCredentials = sec.Key("ALLOW_CREDENTIALS").MustBool(true) |
||||
} |
||||
|
||||
return opt |
||||
} |
||||
|
||||
// https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Access-Control-Allow-Origin
|
||||
// https://fetch.spec.whatwg.org/#cors-protocol-and-credentials
|
||||
// For requests without credentials, the server may specify "*" as a wildcard, thereby allowing any origin to access the resource.
|
||||
func CORS(options ...Options) macaron.Handler { |
||||
opt := prepareOptions(options) |
||||
return func(ctx *macaron.Context, log *log.Logger) { |
||||
reqOptions := ctx.Req.Method == http.MethodOptions |
||||
|
||||
headers := map[string]string{ |
||||
"access-control-allow-methods": strings.Join(opt.Methods, ","), |
||||
"access-control-allow-headers": ctx.Req.Header.Get("access-control-request-headers"), |
||||
"access-control-max-age": strconv.Itoa(opt.MaxAgeSeconds), |
||||
} |
||||
if opt.AllowDomain == "*" { |
||||
headers["access-control-allow-origin"] = "*" |
||||
} else if opt.AllowDomain != "" { |
||||
origin := ctx.Req.Header.Get("Origin") |
||||
if reqOptions && origin == "" { |
||||
respErrorf(ctx, log, http.StatusBadRequest, "missing origin header in CORS request") |
||||
return |
||||
} |
||||
|
||||
u, err := url.Parse(origin) |
||||
if err != nil { |
||||
respErrorf(ctx, log, http.StatusBadRequest, "Failed to parse CORS origin header. Reason: %v", err) |
||||
return |
||||
} |
||||
|
||||
ok := u.Hostname() == opt.AllowDomain || |
||||
(opt.AllowSubdomain && strings.HasSuffix(u.Hostname(), "."+opt.AllowDomain)) |
||||
if ok { |
||||
u.Scheme = opt.Scheme |
||||
headers["access-control-allow-origin"] = u.String() |
||||
headers["access-control-allow-credentials"] = strconv.FormatBool(opt.AllowCredentials) |
||||
headers["vary"] = "Origin" |
||||
} |
||||
if reqOptions && !ok { |
||||
respErrorf(ctx, log, http.StatusBadRequest, "CORS request from prohibited domain %v", origin) |
||||
return |
||||
} |
||||
} |
||||
ctx.Resp.Before(func(w macaron.ResponseWriter) { |
||||
for k, v := range headers { |
||||
w.Header().Set(k, v) |
||||
} |
||||
}) |
||||
if reqOptions { |
||||
ctx.Status(200) // return response
|
||||
} |
||||
} |
||||
} |
||||
|
||||
func respErrorf(ctx *macaron.Context, log *log.Logger, statusCode int, format string, a ...interface{}) { |
||||
msg := fmt.Sprintf(format, a...) |
||||
log.Println(msg) |
||||
ctx.WriteHeader(statusCode) |
||||
_, err := ctx.Write([]byte(msg)) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
return |
||||
} |
||||
@ -0,0 +1,11 @@ |
||||
module github.com/go-macaron/cors |
||||
|
||||
go 1.12 |
||||
|
||||
require ( |
||||
github.com/Unknwon/com v0.0.0-20190321035513-0fed4efef755 // indirect |
||||
github.com/go-macaron/inject v0.0.0-20160627170012-d8a0b8677191 // indirect |
||||
golang.org/x/crypto v0.0.0-20190418165655-df01cb2cc480 // indirect |
||||
gopkg.in/ini.v1 v1.42.0 // indirect |
||||
gopkg.in/macaron.v1 v1.3.2 |
||||
) |
||||
@ -0,0 +1,19 @@ |
||||
github.com/Unknwon/com v0.0.0-20190321035513-0fed4efef755 h1:1B7wb36fHLSwZfHg6ngZhhtIEHQjiC5H4p7qQGBEffg= |
||||
github.com/Unknwon/com v0.0.0-20190321035513-0fed4efef755/go.mod h1:voKvFVpXBJxdIPeqjoJuLK+UVcRlo/JLjeToGxPYu68= |
||||
github.com/go-macaron/inject v0.0.0-20160627170012-d8a0b8677191 h1:NjHlg70DuOkcAMqgt0+XA+NHwtu66MkTVVgR4fFWbcI= |
||||
github.com/go-macaron/inject v0.0.0-20160627170012-d8a0b8677191/go.mod h1:VFI2o2q9kYsC4o7VP1HrEVosiZZTd+MVT3YZx4gqvJw= |
||||
github.com/gopherjs/gopherjs v0.0.0-20181103185306-d547d1d9531e h1:JKmoR8x90Iww1ks85zJ1lfDGgIiMDuIptTOhJq+zKyg= |
||||
github.com/gopherjs/gopherjs v0.0.0-20181103185306-d547d1d9531e/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= |
||||
github.com/jtolds/gls v4.2.1+incompatible h1:fSuqC+Gmlu6l/ZYAoZzx2pyucC8Xza35fpRVWLVmUEE= |
||||
github.com/jtolds/gls v4.2.1+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= |
||||
github.com/smartystreets/assertions v0.0.0-20190116191733-b6c0e53d7304 h1:Jpy1PXuP99tXNrhbq2BaPz9B+jNAvH1JPQQpG/9GCXY= |
||||
github.com/smartystreets/assertions v0.0.0-20190116191733-b6c0e53d7304/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc= |
||||
github.com/smartystreets/goconvey v0.0.0-20181108003508-044398e4856c h1:Ho+uVpkel/udgjbwB5Lktg9BtvJSh2DT0Hi6LPSyI2w= |
||||
github.com/smartystreets/goconvey v0.0.0-20181108003508-044398e4856c/go.mod h1:XDJAKZRPZ1CvBcN2aX5YOUTYGHki24fSF0Iv48Ibg0s= |
||||
golang.org/x/crypto v0.0.0-20190418165655-df01cb2cc480 h1:O5YqonU5IWby+w98jVUG9h7zlCWCcH4RHyPVReBmhzk= |
||||
golang.org/x/crypto v0.0.0-20190418165655-df01cb2cc480/go.mod h1:WFFai1msRO1wXaEeE5yQxYXgSfI8pQAWXbQop6sCtWE= |
||||
golang.org/x/sys v0.0.0-20190403152447-81d4e9dc473e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= |
||||
gopkg.in/ini.v1 v1.42.0 h1:7N3gPTt50s8GuLortA00n8AqRTk75qOP98+mTPpgzRk= |
||||
gopkg.in/ini.v1 v1.42.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k= |
||||
gopkg.in/macaron.v1 v1.3.2 h1:AvWIaPmwBUA87/OWzePkoxeaw6YJWDfBt1pDFPBnLf8= |
||||
gopkg.in/macaron.v1 v1.3.2/go.mod h1:PrsiawTWAGZs6wFbT5hlr7SQ2Ns9h7cUVtcUu4lQOVo= |
||||
@ -0,0 +1,308 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved. |
||||
// Use of this source code is governed by a BSD-style |
||||
// license that can be found in the LICENSE file. |
||||
|
||||
// +build go1.11 |
||||
// +build !gccgo,!appengine |
||||
|
||||
#include "textflag.h" |
||||
|
||||
#define NUM_ROUNDS 10 |
||||
|
||||
// func xorKeyStreamVX(dst, src []byte, key *[8]uint32, nonce *[3]uint32, counter *uint32) |
||||
TEXT ·xorKeyStreamVX(SB), NOSPLIT, $0 |
||||
MOVD dst+0(FP), R1 |
||||
MOVD src+24(FP), R2 |
||||
MOVD src_len+32(FP), R3 |
||||
MOVD key+48(FP), R4 |
||||
MOVD nonce+56(FP), R6 |
||||
MOVD counter+64(FP), R7 |
||||
|
||||
MOVD $·constants(SB), R10 |
||||
MOVD $·incRotMatrix(SB), R11 |
||||
|
||||
MOVW (R7), R20 |
||||
|
||||
AND $~255, R3, R13 |
||||
ADD R2, R13, R12 // R12 for block end |
||||
AND $255, R3, R13 |
||||
loop: |
||||
MOVD $NUM_ROUNDS, R21 |
||||
VLD1 (R11), [V30.S4, V31.S4] |
||||
|
||||
// load contants |
||||
// VLD4R (R10), [V0.S4, V1.S4, V2.S4, V3.S4] |
||||
WORD $0x4D60E940 |
||||
|
||||
// load keys |
||||
// VLD4R 16(R4), [V4.S4, V5.S4, V6.S4, V7.S4] |
||||
WORD $0x4DFFE884 |
||||
// VLD4R 16(R4), [V8.S4, V9.S4, V10.S4, V11.S4] |
||||
WORD $0x4DFFE888 |
||||
SUB $32, R4 |
||||
|
||||
// load counter + nonce |
||||
// VLD1R (R7), [V12.S4] |
||||
WORD $0x4D40C8EC |
||||
|
||||
// VLD3R (R6), [V13.S4, V14.S4, V15.S4] |
||||
WORD $0x4D40E8CD |
||||
|
||||
// update counter |
||||
VADD V30.S4, V12.S4, V12.S4 |
||||
|
||||
chacha: |
||||
// V0..V3 += V4..V7 |
||||
// V12..V15 <<<= ((V12..V15 XOR V0..V3), 16) |
||||
VADD V0.S4, V4.S4, V0.S4 |
||||
VADD V1.S4, V5.S4, V1.S4 |
||||
VADD V2.S4, V6.S4, V2.S4 |
||||
VADD V3.S4, V7.S4, V3.S4 |
||||
VEOR V12.B16, V0.B16, V12.B16 |
||||
VEOR V13.B16, V1.B16, V13.B16 |
||||
VEOR V14.B16, V2.B16, V14.B16 |
||||
VEOR V15.B16, V3.B16, V15.B16 |
||||
VREV32 V12.H8, V12.H8 |
||||
VREV32 V13.H8, V13.H8 |
||||
VREV32 V14.H8, V14.H8 |
||||
VREV32 V15.H8, V15.H8 |
||||
// V8..V11 += V12..V15 |
||||
// V4..V7 <<<= ((V4..V7 XOR V8..V11), 12) |
||||
VADD V8.S4, V12.S4, V8.S4 |
||||
VADD V9.S4, V13.S4, V9.S4 |
||||
VADD V10.S4, V14.S4, V10.S4 |
||||
VADD V11.S4, V15.S4, V11.S4 |
||||
VEOR V8.B16, V4.B16, V16.B16 |
||||
VEOR V9.B16, V5.B16, V17.B16 |
||||
VEOR V10.B16, V6.B16, V18.B16 |
||||
VEOR V11.B16, V7.B16, V19.B16 |
||||
VSHL $12, V16.S4, V4.S4 |
||||
VSHL $12, V17.S4, V5.S4 |
||||
VSHL $12, V18.S4, V6.S4 |
||||
VSHL $12, V19.S4, V7.S4 |
||||
VSRI $20, V16.S4, V4.S4 |
||||
VSRI $20, V17.S4, V5.S4 |
||||
VSRI $20, V18.S4, V6.S4 |
||||
VSRI $20, V19.S4, V7.S4 |
||||
|
||||
// V0..V3 += V4..V7 |
||||
// V12..V15 <<<= ((V12..V15 XOR V0..V3), 8) |
||||
VADD V0.S4, V4.S4, V0.S4 |
||||
VADD V1.S4, V5.S4, V1.S4 |
||||
VADD V2.S4, V6.S4, V2.S4 |
||||
VADD V3.S4, V7.S4, V3.S4 |
||||
VEOR V12.B16, V0.B16, V12.B16 |
||||
VEOR V13.B16, V1.B16, V13.B16 |
||||
VEOR V14.B16, V2.B16, V14.B16 |
||||
VEOR V15.B16, V3.B16, V15.B16 |
||||
VTBL V31.B16, [V12.B16], V12.B16 |
||||
VTBL V31.B16, [V13.B16], V13.B16 |
||||
VTBL V31.B16, [V14.B16], V14.B16 |
||||
VTBL V31.B16, [V15.B16], V15.B16 |
||||
|
||||
// V8..V11 += V12..V15 |
||||
// V4..V7 <<<= ((V4..V7 XOR V8..V11), 7) |
||||
VADD V12.S4, V8.S4, V8.S4 |
||||
VADD V13.S4, V9.S4, V9.S4 |
||||
VADD V14.S4, V10.S4, V10.S4 |
||||
VADD V15.S4, V11.S4, V11.S4 |
||||
VEOR V8.B16, V4.B16, V16.B16 |
||||
VEOR V9.B16, V5.B16, V17.B16 |
||||
VEOR V10.B16, V6.B16, V18.B16 |
||||
VEOR V11.B16, V7.B16, V19.B16 |
||||
VSHL $7, V16.S4, V4.S4 |
||||
VSHL $7, V17.S4, V5.S4 |
||||
VSHL $7, V18.S4, V6.S4 |
||||
VSHL $7, V19.S4, V7.S4 |
||||
VSRI $25, V16.S4, V4.S4 |
||||
VSRI $25, V17.S4, V5.S4 |
||||
VSRI $25, V18.S4, V6.S4 |
||||
VSRI $25, V19.S4, V7.S4 |
||||
|
||||
// V0..V3 += V5..V7, V4 |
||||
// V15,V12-V14 <<<= ((V15,V12-V14 XOR V0..V3), 16) |
||||
VADD V0.S4, V5.S4, V0.S4 |
||||
VADD V1.S4, V6.S4, V1.S4 |
||||
VADD V2.S4, V7.S4, V2.S4 |
||||
VADD V3.S4, V4.S4, V3.S4 |
||||
VEOR V15.B16, V0.B16, V15.B16 |
||||
VEOR V12.B16, V1.B16, V12.B16 |
||||
VEOR V13.B16, V2.B16, V13.B16 |
||||
VEOR V14.B16, V3.B16, V14.B16 |
||||
VREV32 V12.H8, V12.H8 |
||||
VREV32 V13.H8, V13.H8 |
||||
VREV32 V14.H8, V14.H8 |
||||
VREV32 V15.H8, V15.H8 |
||||
|
||||
// V10 += V15; V5 <<<= ((V10 XOR V5), 12)
|
||||
// ... |
||||
VADD V15.S4, V10.S4, V10.S4 |
||||
VADD V12.S4, V11.S4, V11.S4 |
||||
VADD V13.S4, V8.S4, V8.S4 |
||||
VADD V14.S4, V9.S4, V9.S4 |
||||
VEOR V10.B16, V5.B16, V16.B16 |
||||
VEOR V11.B16, V6.B16, V17.B16 |
||||
VEOR V8.B16, V7.B16, V18.B16 |
||||
VEOR V9.B16, V4.B16, V19.B16 |
||||
VSHL $12, V16.S4, V5.S4 |
||||
VSHL $12, V17.S4, V6.S4 |
||||
VSHL $12, V18.S4, V7.S4 |
||||
VSHL $12, V19.S4, V4.S4 |
||||
VSRI $20, V16.S4, V5.S4 |
||||
VSRI $20, V17.S4, V6.S4 |
||||
VSRI $20, V18.S4, V7.S4 |
||||
VSRI $20, V19.S4, V4.S4 |
||||
|
||||
// V0 += V5; V15 <<<= ((V0 XOR V15), 8)
|
||||
// ... |
||||
VADD V5.S4, V0.S4, V0.S4 |
||||
VADD V6.S4, V1.S4, V1.S4 |
||||
VADD V7.S4, V2.S4, V2.S4 |
||||
VADD V4.S4, V3.S4, V3.S4 |
||||
VEOR V0.B16, V15.B16, V15.B16 |
||||
VEOR V1.B16, V12.B16, V12.B16 |
||||
VEOR V2.B16, V13.B16, V13.B16 |
||||
VEOR V3.B16, V14.B16, V14.B16 |
||||
VTBL V31.B16, [V12.B16], V12.B16 |
||||
VTBL V31.B16, [V13.B16], V13.B16 |
||||
VTBL V31.B16, [V14.B16], V14.B16 |
||||
VTBL V31.B16, [V15.B16], V15.B16 |
||||
|
||||
// V10 += V15; V5 <<<= ((V10 XOR V5), 7)
|
||||
// ... |
||||
VADD V15.S4, V10.S4, V10.S4 |
||||
VADD V12.S4, V11.S4, V11.S4 |
||||
VADD V13.S4, V8.S4, V8.S4 |
||||
VADD V14.S4, V9.S4, V9.S4 |
||||
VEOR V10.B16, V5.B16, V16.B16 |
||||
VEOR V11.B16, V6.B16, V17.B16 |
||||
VEOR V8.B16, V7.B16, V18.B16 |
||||
VEOR V9.B16, V4.B16, V19.B16 |
||||
VSHL $7, V16.S4, V5.S4 |
||||
VSHL $7, V17.S4, V6.S4 |
||||
VSHL $7, V18.S4, V7.S4 |
||||
VSHL $7, V19.S4, V4.S4 |
||||
VSRI $25, V16.S4, V5.S4 |
||||
VSRI $25, V17.S4, V6.S4 |
||||
VSRI $25, V18.S4, V7.S4 |
||||
VSRI $25, V19.S4, V4.S4 |
||||
|
||||
SUB $1, R21 |
||||
CBNZ R21, chacha |
||||
|
||||
// VLD4R (R10), [V16.S4, V17.S4, V18.S4, V19.S4] |
||||
WORD $0x4D60E950 |
||||
|
||||
// VLD4R 16(R4), [V20.S4, V21.S4, V22.S4, V23.S4] |
||||
WORD $0x4DFFE894 |
||||
VADD V30.S4, V12.S4, V12.S4 |
||||
VADD V16.S4, V0.S4, V0.S4 |
||||
VADD V17.S4, V1.S4, V1.S4 |
||||
VADD V18.S4, V2.S4, V2.S4 |
||||
VADD V19.S4, V3.S4, V3.S4 |
||||
// VLD4R 16(R4), [V24.S4, V25.S4, V26.S4, V27.S4] |
||||
WORD $0x4DFFE898 |
||||
// restore R4 |
||||
SUB $32, R4 |
||||
|
||||
// load counter + nonce |
||||
// VLD1R (R7), [V28.S4] |
||||
WORD $0x4D40C8FC |
||||
// VLD3R (R6), [V29.S4, V30.S4, V31.S4] |
||||
WORD $0x4D40E8DD |
||||
|
||||
VADD V20.S4, V4.S4, V4.S4 |
||||
VADD V21.S4, V5.S4, V5.S4 |
||||
VADD V22.S4, V6.S4, V6.S4 |
||||
VADD V23.S4, V7.S4, V7.S4 |
||||
VADD V24.S4, V8.S4, V8.S4 |
||||
VADD V25.S4, V9.S4, V9.S4 |
||||
VADD V26.S4, V10.S4, V10.S4 |
||||
VADD V27.S4, V11.S4, V11.S4 |
||||
VADD V28.S4, V12.S4, V12.S4 |
||||
VADD V29.S4, V13.S4, V13.S4 |
||||
VADD V30.S4, V14.S4, V14.S4 |
||||
VADD V31.S4, V15.S4, V15.S4 |
||||
|
||||
VZIP1 V1.S4, V0.S4, V16.S4 |
||||
VZIP2 V1.S4, V0.S4, V17.S4 |
||||
VZIP1 V3.S4, V2.S4, V18.S4 |
||||
VZIP2 V3.S4, V2.S4, V19.S4 |
||||
VZIP1 V5.S4, V4.S4, V20.S4 |
||||
VZIP2 V5.S4, V4.S4, V21.S4 |
||||
VZIP1 V7.S4, V6.S4, V22.S4 |
||||
VZIP2 V7.S4, V6.S4, V23.S4 |
||||
VZIP1 V9.S4, V8.S4, V24.S4 |
||||
VZIP2 V9.S4, V8.S4, V25.S4 |
||||
VZIP1 V11.S4, V10.S4, V26.S4 |
||||
VZIP2 V11.S4, V10.S4, V27.S4 |
||||
VZIP1 V13.S4, V12.S4, V28.S4 |
||||
VZIP2 V13.S4, V12.S4, V29.S4 |
||||
VZIP1 V15.S4, V14.S4, V30.S4 |
||||
VZIP2 V15.S4, V14.S4, V31.S4 |
||||
VZIP1 V18.D2, V16.D2, V0.D2 |
||||
VZIP2 V18.D2, V16.D2, V4.D2 |
||||
VZIP1 V19.D2, V17.D2, V8.D2 |
||||
VZIP2 V19.D2, V17.D2, V12.D2 |
||||
VLD1.P 64(R2), [V16.B16, V17.B16, V18.B16, V19.B16] |
||||
|
||||
VZIP1 V22.D2, V20.D2, V1.D2 |
||||
VZIP2 V22.D2, V20.D2, V5.D2 |
||||
VZIP1 V23.D2, V21.D2, V9.D2 |
||||
VZIP2 V23.D2, V21.D2, V13.D2 |
||||
VLD1.P 64(R2), [V20.B16, V21.B16, V22.B16, V23.B16] |
||||
VZIP1 V26.D2, V24.D2, V2.D2 |
||||
VZIP2 V26.D2, V24.D2, V6.D2 |
||||
VZIP1 V27.D2, V25.D2, V10.D2 |
||||
VZIP2 V27.D2, V25.D2, V14.D2 |
||||
VLD1.P 64(R2), [V24.B16, V25.B16, V26.B16, V27.B16] |
||||
VZIP1 V30.D2, V28.D2, V3.D2 |
||||
VZIP2 V30.D2, V28.D2, V7.D2 |
||||
VZIP1 V31.D2, V29.D2, V11.D2 |
||||
VZIP2 V31.D2, V29.D2, V15.D2 |
||||
VLD1.P 64(R2), [V28.B16, V29.B16, V30.B16, V31.B16] |
||||
VEOR V0.B16, V16.B16, V16.B16 |
||||
VEOR V1.B16, V17.B16, V17.B16 |
||||
VEOR V2.B16, V18.B16, V18.B16 |
||||
VEOR V3.B16, V19.B16, V19.B16 |
||||
VST1.P [V16.B16, V17.B16, V18.B16, V19.B16], 64(R1) |
||||
VEOR V4.B16, V20.B16, V20.B16 |
||||
VEOR V5.B16, V21.B16, V21.B16 |
||||
VEOR V6.B16, V22.B16, V22.B16 |
||||
VEOR V7.B16, V23.B16, V23.B16 |
||||
VST1.P [V20.B16, V21.B16, V22.B16, V23.B16], 64(R1) |
||||
VEOR V8.B16, V24.B16, V24.B16 |
||||
VEOR V9.B16, V25.B16, V25.B16 |
||||
VEOR V10.B16, V26.B16, V26.B16 |
||||
VEOR V11.B16, V27.B16, V27.B16 |
||||
VST1.P [V24.B16, V25.B16, V26.B16, V27.B16], 64(R1) |
||||
VEOR V12.B16, V28.B16, V28.B16 |
||||
VEOR V13.B16, V29.B16, V29.B16 |
||||
VEOR V14.B16, V30.B16, V30.B16 |
||||
VEOR V15.B16, V31.B16, V31.B16 |
||||
VST1.P [V28.B16, V29.B16, V30.B16, V31.B16], 64(R1) |
||||
|
||||
ADD $4, R20 |
||||
MOVW R20, (R7) // update counter |
||||
|
||||
CMP R2, R12 |
||||
BGT loop |
||||
|
||||
RET |
||||
|
||||
|
||||
DATA ·constants+0x00(SB)/4, $0x61707865 |
||||
DATA ·constants+0x04(SB)/4, $0x3320646e |
||||
DATA ·constants+0x08(SB)/4, $0x79622d32 |
||||
DATA ·constants+0x0c(SB)/4, $0x6b206574 |
||||
GLOBL ·constants(SB), NOPTR|RODATA, $32 |
||||
|
||||
DATA ·incRotMatrix+0x00(SB)/4, $0x00000000 |
||||
DATA ·incRotMatrix+0x04(SB)/4, $0x00000001 |
||||
DATA ·incRotMatrix+0x08(SB)/4, $0x00000002 |
||||
DATA ·incRotMatrix+0x0c(SB)/4, $0x00000003 |
||||
DATA ·incRotMatrix+0x10(SB)/4, $0x02010003 |
||||
DATA ·incRotMatrix+0x14(SB)/4, $0x06050407 |
||||
DATA ·incRotMatrix+0x18(SB)/4, $0x0A09080B |
||||
DATA ·incRotMatrix+0x1c(SB)/4, $0x0E0D0C0F |
||||
GLOBL ·incRotMatrix(SB), NOPTR|RODATA, $32 |
||||
@ -0,0 +1,31 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build go1.11
|
||||
// +build !gccgo
|
||||
|
||||
package chacha20 |
||||
|
||||
const ( |
||||
haveAsm = true |
||||
bufSize = 256 |
||||
) |
||||
|
||||
//go:noescape
|
||||
func xorKeyStreamVX(dst, src []byte, key *[8]uint32, nonce *[3]uint32, counter *uint32) |
||||
|
||||
func (c *Cipher) xorKeyStreamAsm(dst, src []byte) { |
||||
|
||||
if len(src) >= bufSize { |
||||
xorKeyStreamVX(dst, src, &c.key, &c.nonce, &c.counter) |
||||
} |
||||
|
||||
if len(src)%bufSize != 0 { |
||||
i := len(src) - len(src)%bufSize |
||||
c.buf = [bufSize]byte{} |
||||
copy(c.buf[:], src[i:]) |
||||
xorKeyStreamVX(c.buf[:], c.buf[:], &c.key, &c.nonce, &c.counter) |
||||
c.len = bufSize - copy(dst[i:], c.buf[:len(src)%bufSize]) |
||||
} |
||||
} |
||||
@ -0,0 +1,11 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build !amd64 gccgo appengine
|
||||
|
||||
package poly1305 |
||||
|
||||
type mac struct{ macGeneric } |
||||
|
||||
func newMAC(key *[32]byte) mac { return mac{newMACGeneric(key)} } |
||||
@ -0,0 +1,30 @@ |
||||
// Copyright 2019 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package cpu |
||||
|
||||
import ( |
||||
"encoding/binary" |
||||
"runtime" |
||||
) |
||||
|
||||
// hostByteOrder returns binary.LittleEndian on little-endian machines and
|
||||
// binary.BigEndian on big-endian machines.
|
||||
func hostByteOrder() binary.ByteOrder { |
||||
switch runtime.GOARCH { |
||||
case "386", "amd64", "amd64p32", |
||||
"arm", "arm64", |
||||
"mipsle", "mips64le", "mips64p32le", |
||||
"ppc64le", |
||||
"riscv", "riscv64": |
||||
return binary.LittleEndian |
||||
case "armbe", "arm64be", |
||||
"mips", "mips64", "mips64p32", |
||||
"ppc", "ppc64", |
||||
"s390", "s390x", |
||||
"sparc", "sparc64": |
||||
return binary.BigEndian |
||||
} |
||||
panic("unknown architecture") |
||||
} |
||||
@ -0,0 +1,126 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// Package cpu implements processor feature detection for
|
||||
// various CPU architectures.
|
||||
package cpu |
||||
|
||||
// Initialized reports whether the CPU features were initialized.
|
||||
//
|
||||
// For some GOOS/GOARCH combinations initialization of the CPU features depends
|
||||
// on reading an operating specific file, e.g. /proc/self/auxv on linux/arm
|
||||
// Initialized will report false if reading the file fails.
|
||||
var Initialized bool |
||||
|
||||
// CacheLinePad is used to pad structs to avoid false sharing.
|
||||
type CacheLinePad struct{ _ [cacheLineSize]byte } |
||||
|
||||
// X86 contains the supported CPU features of the
|
||||
// current X86/AMD64 platform. If the current platform
|
||||
// is not X86/AMD64 then all feature flags are false.
|
||||
//
|
||||
// X86 is padded to avoid false sharing. Further the HasAVX
|
||||
// and HasAVX2 are only set if the OS supports XMM and YMM
|
||||
// registers in addition to the CPUID feature bit being set.
|
||||
var X86 struct { |
||||
_ CacheLinePad |
||||
HasAES bool // AES hardware implementation (AES NI)
|
||||
HasADX bool // Multi-precision add-carry instruction extensions
|
||||
HasAVX bool // Advanced vector extension
|
||||
HasAVX2 bool // Advanced vector extension 2
|
||||
HasBMI1 bool // Bit manipulation instruction set 1
|
||||
HasBMI2 bool // Bit manipulation instruction set 2
|
||||
HasERMS bool // Enhanced REP for MOVSB and STOSB
|
||||
HasFMA bool // Fused-multiply-add instructions
|
||||
HasOSXSAVE bool // OS supports XSAVE/XRESTOR for saving/restoring XMM registers.
|
||||
HasPCLMULQDQ bool // PCLMULQDQ instruction - most often used for AES-GCM
|
||||
HasPOPCNT bool // Hamming weight instruction POPCNT.
|
||||
HasRDRAND bool // RDRAND instruction (on-chip random number generator)
|
||||
HasRDSEED bool // RDSEED instruction (on-chip random number generator)
|
||||
HasSSE2 bool // Streaming SIMD extension 2 (always available on amd64)
|
||||
HasSSE3 bool // Streaming SIMD extension 3
|
||||
HasSSSE3 bool // Supplemental streaming SIMD extension 3
|
||||
HasSSE41 bool // Streaming SIMD extension 4 and 4.1
|
||||
HasSSE42 bool // Streaming SIMD extension 4 and 4.2
|
||||
_ CacheLinePad |
||||
} |
||||
|
||||
// ARM64 contains the supported CPU features of the
|
||||
// current ARMv8(aarch64) platform. If the current platform
|
||||
// is not arm64 then all feature flags are false.
|
||||
var ARM64 struct { |
||||
_ CacheLinePad |
||||
HasFP bool // Floating-point instruction set (always available)
|
||||
HasASIMD bool // Advanced SIMD (always available)
|
||||
HasEVTSTRM bool // Event stream support
|
||||
HasAES bool // AES hardware implementation
|
||||
HasPMULL bool // Polynomial multiplication instruction set
|
||||
HasSHA1 bool // SHA1 hardware implementation
|
||||
HasSHA2 bool // SHA2 hardware implementation
|
||||
HasCRC32 bool // CRC32 hardware implementation
|
||||
HasATOMICS bool // Atomic memory operation instruction set
|
||||
HasFPHP bool // Half precision floating-point instruction set
|
||||
HasASIMDHP bool // Advanced SIMD half precision instruction set
|
||||
HasCPUID bool // CPUID identification scheme registers
|
||||
HasASIMDRDM bool // Rounding double multiply add/subtract instruction set
|
||||
HasJSCVT bool // Javascript conversion from floating-point to integer
|
||||
HasFCMA bool // Floating-point multiplication and addition of complex numbers
|
||||
HasLRCPC bool // Release Consistent processor consistent support
|
||||
HasDCPOP bool // Persistent memory support
|
||||
HasSHA3 bool // SHA3 hardware implementation
|
||||
HasSM3 bool // SM3 hardware implementation
|
||||
HasSM4 bool // SM4 hardware implementation
|
||||
HasASIMDDP bool // Advanced SIMD double precision instruction set
|
||||
HasSHA512 bool // SHA512 hardware implementation
|
||||
HasSVE bool // Scalable Vector Extensions
|
||||
HasASIMDFHM bool // Advanced SIMD multiplication FP16 to FP32
|
||||
_ CacheLinePad |
||||
} |
||||
|
||||
// PPC64 contains the supported CPU features of the current ppc64/ppc64le platforms.
|
||||
// If the current platform is not ppc64/ppc64le then all feature flags are false.
|
||||
//
|
||||
// For ppc64/ppc64le, it is safe to check only for ISA level starting on ISA v3.00,
|
||||
// since there are no optional categories. There are some exceptions that also
|
||||
// require kernel support to work (DARN, SCV), so there are feature bits for
|
||||
// those as well. The minimum processor requirement is POWER8 (ISA 2.07).
|
||||
// The struct is padded to avoid false sharing.
|
||||
var PPC64 struct { |
||||
_ CacheLinePad |
||||
HasDARN bool // Hardware random number generator (requires kernel enablement)
|
||||
HasSCV bool // Syscall vectored (requires kernel enablement)
|
||||
IsPOWER8 bool // ISA v2.07 (POWER8)
|
||||
IsPOWER9 bool // ISA v3.00 (POWER9)
|
||||
_ CacheLinePad |
||||
} |
||||
|
||||
// S390X contains the supported CPU features of the current IBM Z
|
||||
// (s390x) platform. If the current platform is not IBM Z then all
|
||||
// feature flags are false.
|
||||
//
|
||||
// S390X is padded to avoid false sharing. Further HasVX is only set
|
||||
// if the OS supports vector registers in addition to the STFLE
|
||||
// feature bit being set.
|
||||
var S390X struct { |
||||
_ CacheLinePad |
||||
HasZARCH bool // z/Architecture mode is active [mandatory]
|
||||
HasSTFLE bool // store facility list extended
|
||||
HasLDISP bool // long (20-bit) displacements
|
||||
HasEIMM bool // 32-bit immediates
|
||||
HasDFP bool // decimal floating point
|
||||
HasETF3EH bool // ETF-3 enhanced
|
||||
HasMSA bool // message security assist (CPACF)
|
||||
HasAES bool // KM-AES{128,192,256} functions
|
||||
HasAESCBC bool // KMC-AES{128,192,256} functions
|
||||
HasAESCTR bool // KMCTR-AES{128,192,256} functions
|
||||
HasAESGCM bool // KMA-GCM-AES{128,192,256} functions
|
||||
HasGHASH bool // KIMD-GHASH function
|
||||
HasSHA1 bool // K{I,L}MD-SHA-1 functions
|
||||
HasSHA256 bool // K{I,L}MD-SHA-256 functions
|
||||
HasSHA512 bool // K{I,L}MD-SHA-512 functions
|
||||
HasSHA3 bool // K{I,L}MD-SHA3-{224,256,384,512} and K{I,L}MD-SHAKE-{128,256} functions
|
||||
HasVX bool // vector facility
|
||||
HasVXE bool // vector-enhancements facility 1
|
||||
_ CacheLinePad |
||||
} |
||||
@ -0,0 +1,30 @@ |
||||
// Copyright 2019 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build aix,ppc64
|
||||
|
||||
package cpu |
||||
|
||||
import "golang.org/x/sys/unix" |
||||
|
||||
const cacheLineSize = 128 |
||||
|
||||
const ( |
||||
// getsystemcfg constants
|
||||
_SC_IMPL = 2 |
||||
_IMPL_POWER8 = 0x10000 |
||||
_IMPL_POWER9 = 0x20000 |
||||
) |
||||
|
||||
func init() { |
||||
impl := unix.Getsystemcfg(_SC_IMPL) |
||||
if impl&_IMPL_POWER8 != 0 { |
||||
PPC64.IsPOWER8 = true |
||||
} |
||||
if impl&_IMPL_POWER9 != 0 { |
||||
PPC64.IsPOWER9 = true |
||||
} |
||||
|
||||
Initialized = true |
||||
} |
||||
@ -0,0 +1,9 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package cpu |
||||
|
||||
const cacheLineSize = 32 |
||||
|
||||
func doinit() {} |
||||
@ -0,0 +1,21 @@ |
||||
// Copyright 2019 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build !gccgo
|
||||
|
||||
package cpu |
||||
|
||||
// haveAsmFunctions reports whether the other functions in this file can
|
||||
// be safely called.
|
||||
func haveAsmFunctions() bool { return true } |
||||
|
||||
// The following feature detection functions are defined in cpu_s390x.s.
|
||||
// They are likely to be expensive to call so the results should be cached.
|
||||
func stfle() facilityList |
||||
func kmQuery() queryResult |
||||
func kmcQuery() queryResult |
||||
func kmctrQuery() queryResult |
||||
func kmaQuery() queryResult |
||||
func kimdQuery() queryResult |
||||
func klmdQuery() queryResult |
||||
@ -0,0 +1,16 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build 386 amd64 amd64p32
|
||||
// +build !gccgo
|
||||
|
||||
package cpu |
||||
|
||||
// cpuid is implemented in cpu_x86.s for gc compiler
|
||||
// and in cpu_gccgo.c for gccgo.
|
||||
func cpuid(eaxArg, ecxArg uint32) (eax, ebx, ecx, edx uint32) |
||||
|
||||
// xgetbv with ecx = 0 is implemented in cpu_x86.s for gc compiler
|
||||
// and in cpu_gccgo.c for gccgo.
|
||||
func xgetbv() (eax, edx uint32) |
||||
@ -0,0 +1,43 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build 386 amd64 amd64p32
|
||||
// +build gccgo
|
||||
|
||||
#include <cpuid.h> |
||||
#include <stdint.h> |
||||
|
||||
// Need to wrap __get_cpuid_count because it's declared as static.
|
||||
int |
||||
gccgoGetCpuidCount(uint32_t leaf, uint32_t subleaf, |
||||
uint32_t *eax, uint32_t *ebx, |
||||
uint32_t *ecx, uint32_t *edx) |
||||
{ |
||||
return __get_cpuid_count(leaf, subleaf, eax, ebx, ecx, edx); |
||||
} |
||||
|
||||
// xgetbv reads the contents of an XCR (Extended Control Register)
|
||||
// specified in the ECX register into registers EDX:EAX.
|
||||
// Currently, the only supported value for XCR is 0.
|
||||
//
|
||||
// TODO: Replace with a better alternative:
|
||||
//
|
||||
// #include <xsaveintrin.h>
|
||||
//
|
||||
// #pragma GCC target("xsave")
|
||||
//
|
||||
// void gccgoXgetbv(uint32_t *eax, uint32_t *edx) {
|
||||
// unsigned long long x = _xgetbv(0);
|
||||
// *eax = x & 0xffffffff;
|
||||
// *edx = (x >> 32) & 0xffffffff;
|
||||
// }
|
||||
//
|
||||
// Note that _xgetbv is defined starting with GCC 8.
|
||||
void |
||||
gccgoXgetbv(uint32_t *eax, uint32_t *edx) |
||||
{ |
||||
__asm(" xorl %%ecx, %%ecx\n" |
||||
" xgetbv" |
||||
: "=a"(*eax), "=d"(*edx)); |
||||
} |
||||
@ -0,0 +1,26 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build 386 amd64 amd64p32
|
||||
// +build gccgo
|
||||
|
||||
package cpu |
||||
|
||||
//extern gccgoGetCpuidCount
|
||||
func gccgoGetCpuidCount(eaxArg, ecxArg uint32, eax, ebx, ecx, edx *uint32) |
||||
|
||||
func cpuid(eaxArg, ecxArg uint32) (eax, ebx, ecx, edx uint32) { |
||||
var a, b, c, d uint32 |
||||
gccgoGetCpuidCount(eaxArg, ecxArg, &a, &b, &c, &d) |
||||
return a, b, c, d |
||||
} |
||||
|
||||
//extern gccgoXgetbv
|
||||
func gccgoXgetbv(eax, edx *uint32) |
||||
|
||||
func xgetbv() (eax, edx uint32) { |
||||
var a, d uint32 |
||||
gccgoXgetbv(&a, &d) |
||||
return a, d |
||||
} |
||||
@ -0,0 +1,22 @@ |
||||
// Copyright 2019 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build gccgo
|
||||
|
||||
package cpu |
||||
|
||||
// haveAsmFunctions reports whether the other functions in this file can
|
||||
// be safely called.
|
||||
func haveAsmFunctions() bool { return false } |
||||
|
||||
// TODO(mundaym): the following feature detection functions are currently
|
||||
// stubs. See https://golang.org/cl/162887 for how to fix this.
|
||||
// They are likely to be expensive to call so the results should be cached.
|
||||
func stfle() facilityList { panic("not implemented for gccgo") } |
||||
func kmQuery() queryResult { panic("not implemented for gccgo") } |
||||
func kmcQuery() queryResult { panic("not implemented for gccgo") } |
||||
func kmctrQuery() queryResult { panic("not implemented for gccgo") } |
||||
func kmaQuery() queryResult { panic("not implemented for gccgo") } |
||||
func kimdQuery() queryResult { panic("not implemented for gccgo") } |
||||
func klmdQuery() queryResult { panic("not implemented for gccgo") } |
||||
@ -0,0 +1,59 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
//+build !amd64,!amd64p32,!386
|
||||
|
||||
package cpu |
||||
|
||||
import ( |
||||
"io/ioutil" |
||||
) |
||||
|
||||
const ( |
||||
_AT_HWCAP = 16 |
||||
_AT_HWCAP2 = 26 |
||||
|
||||
procAuxv = "/proc/self/auxv" |
||||
|
||||
uintSize = int(32 << (^uint(0) >> 63)) |
||||
) |
||||
|
||||
// For those platforms don't have a 'cpuid' equivalent we use HWCAP/HWCAP2
|
||||
// These are initialized in cpu_$GOARCH.go
|
||||
// and should not be changed after they are initialized.
|
||||
var hwCap uint |
||||
var hwCap2 uint |
||||
|
||||
func init() { |
||||
buf, err := ioutil.ReadFile(procAuxv) |
||||
if err != nil { |
||||
// e.g. on android /proc/self/auxv is not accessible, so silently
|
||||
// ignore the error and leave Initialized = false
|
||||
return |
||||
} |
||||
|
||||
bo := hostByteOrder() |
||||
for len(buf) >= 2*(uintSize/8) { |
||||
var tag, val uint |
||||
switch uintSize { |
||||
case 32: |
||||
tag = uint(bo.Uint32(buf[0:])) |
||||
val = uint(bo.Uint32(buf[4:])) |
||||
buf = buf[8:] |
||||
case 64: |
||||
tag = uint(bo.Uint64(buf[0:])) |
||||
val = uint(bo.Uint64(buf[8:])) |
||||
buf = buf[16:] |
||||
} |
||||
switch tag { |
||||
case _AT_HWCAP: |
||||
hwCap = val |
||||
case _AT_HWCAP2: |
||||
hwCap2 = val |
||||
} |
||||
} |
||||
doinit() |
||||
|
||||
Initialized = true |
||||
} |
||||
@ -0,0 +1,67 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package cpu |
||||
|
||||
const cacheLineSize = 64 |
||||
|
||||
// HWCAP/HWCAP2 bits. These are exposed by Linux.
|
||||
const ( |
||||
hwcap_FP = 1 << 0 |
||||
hwcap_ASIMD = 1 << 1 |
||||
hwcap_EVTSTRM = 1 << 2 |
||||
hwcap_AES = 1 << 3 |
||||
hwcap_PMULL = 1 << 4 |
||||
hwcap_SHA1 = 1 << 5 |
||||
hwcap_SHA2 = 1 << 6 |
||||
hwcap_CRC32 = 1 << 7 |
||||
hwcap_ATOMICS = 1 << 8 |
||||
hwcap_FPHP = 1 << 9 |
||||
hwcap_ASIMDHP = 1 << 10 |
||||
hwcap_CPUID = 1 << 11 |
||||
hwcap_ASIMDRDM = 1 << 12 |
||||
hwcap_JSCVT = 1 << 13 |
||||
hwcap_FCMA = 1 << 14 |
||||
hwcap_LRCPC = 1 << 15 |
||||
hwcap_DCPOP = 1 << 16 |
||||
hwcap_SHA3 = 1 << 17 |
||||
hwcap_SM3 = 1 << 18 |
||||
hwcap_SM4 = 1 << 19 |
||||
hwcap_ASIMDDP = 1 << 20 |
||||
hwcap_SHA512 = 1 << 21 |
||||
hwcap_SVE = 1 << 22 |
||||
hwcap_ASIMDFHM = 1 << 23 |
||||
) |
||||
|
||||
func doinit() { |
||||
// HWCAP feature bits
|
||||
ARM64.HasFP = isSet(hwCap, hwcap_FP) |
||||
ARM64.HasASIMD = isSet(hwCap, hwcap_ASIMD) |
||||
ARM64.HasEVTSTRM = isSet(hwCap, hwcap_EVTSTRM) |
||||
ARM64.HasAES = isSet(hwCap, hwcap_AES) |
||||
ARM64.HasPMULL = isSet(hwCap, hwcap_PMULL) |
||||
ARM64.HasSHA1 = isSet(hwCap, hwcap_SHA1) |
||||
ARM64.HasSHA2 = isSet(hwCap, hwcap_SHA2) |
||||
ARM64.HasCRC32 = isSet(hwCap, hwcap_CRC32) |
||||
ARM64.HasATOMICS = isSet(hwCap, hwcap_ATOMICS) |
||||
ARM64.HasFPHP = isSet(hwCap, hwcap_FPHP) |
||||
ARM64.HasASIMDHP = isSet(hwCap, hwcap_ASIMDHP) |
||||
ARM64.HasCPUID = isSet(hwCap, hwcap_CPUID) |
||||
ARM64.HasASIMDRDM = isSet(hwCap, hwcap_ASIMDRDM) |
||||
ARM64.HasJSCVT = isSet(hwCap, hwcap_JSCVT) |
||||
ARM64.HasFCMA = isSet(hwCap, hwcap_FCMA) |
||||
ARM64.HasLRCPC = isSet(hwCap, hwcap_LRCPC) |
||||
ARM64.HasDCPOP = isSet(hwCap, hwcap_DCPOP) |
||||
ARM64.HasSHA3 = isSet(hwCap, hwcap_SHA3) |
||||
ARM64.HasSM3 = isSet(hwCap, hwcap_SM3) |
||||
ARM64.HasSM4 = isSet(hwCap, hwcap_SM4) |
||||
ARM64.HasASIMDDP = isSet(hwCap, hwcap_ASIMDDP) |
||||
ARM64.HasSHA512 = isSet(hwCap, hwcap_SHA512) |
||||
ARM64.HasSVE = isSet(hwCap, hwcap_SVE) |
||||
ARM64.HasASIMDFHM = isSet(hwCap, hwcap_ASIMDFHM) |
||||
} |
||||
|
||||
func isSet(hwc uint, value uint) bool { |
||||
return hwc&value != 0 |
||||
} |
||||
@ -0,0 +1,33 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build linux
|
||||
// +build ppc64 ppc64le
|
||||
|
||||
package cpu |
||||
|
||||
const cacheLineSize = 128 |
||||
|
||||
// HWCAP/HWCAP2 bits. These are exposed by the kernel.
|
||||
const ( |
||||
// ISA Level
|
||||
_PPC_FEATURE2_ARCH_2_07 = 0x80000000 |
||||
_PPC_FEATURE2_ARCH_3_00 = 0x00800000 |
||||
|
||||
// CPU features
|
||||
_PPC_FEATURE2_DARN = 0x00200000 |
||||
_PPC_FEATURE2_SCV = 0x00100000 |
||||
) |
||||
|
||||
func doinit() { |
||||
// HWCAP2 feature bits
|
||||
PPC64.IsPOWER8 = isSet(hwCap2, _PPC_FEATURE2_ARCH_2_07) |
||||
PPC64.IsPOWER9 = isSet(hwCap2, _PPC_FEATURE2_ARCH_3_00) |
||||
PPC64.HasDARN = isSet(hwCap2, _PPC_FEATURE2_DARN) |
||||
PPC64.HasSCV = isSet(hwCap2, _PPC_FEATURE2_SCV) |
||||
} |
||||
|
||||
func isSet(hwc uint, value uint) bool { |
||||
return hwc&value != 0 |
||||
} |
||||
@ -0,0 +1,161 @@ |
||||
// Copyright 2019 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package cpu |
||||
|
||||
const cacheLineSize = 256 |
||||
|
||||
const ( |
||||
// bit mask values from /usr/include/bits/hwcap.h
|
||||
hwcap_ZARCH = 2 |
||||
hwcap_STFLE = 4 |
||||
hwcap_MSA = 8 |
||||
hwcap_LDISP = 16 |
||||
hwcap_EIMM = 32 |
||||
hwcap_DFP = 64 |
||||
hwcap_ETF3EH = 256 |
||||
hwcap_VX = 2048 |
||||
hwcap_VXE = 8192 |
||||
) |
||||
|
||||
// bitIsSet reports whether the bit at index is set. The bit index
|
||||
// is in big endian order, so bit index 0 is the leftmost bit.
|
||||
func bitIsSet(bits []uint64, index uint) bool { |
||||
return bits[index/64]&((1<<63)>>(index%64)) != 0 |
||||
} |
||||
|
||||
// function is the code for the named cryptographic function.
|
||||
type function uint8 |
||||
|
||||
const ( |
||||
// KM{,A,C,CTR} function codes
|
||||
aes128 function = 18 // AES-128
|
||||
aes192 function = 19 // AES-192
|
||||
aes256 function = 20 // AES-256
|
||||
|
||||
// K{I,L}MD function codes
|
||||
sha1 function = 1 // SHA-1
|
||||
sha256 function = 2 // SHA-256
|
||||
sha512 function = 3 // SHA-512
|
||||
sha3_224 function = 32 // SHA3-224
|
||||
sha3_256 function = 33 // SHA3-256
|
||||
sha3_384 function = 34 // SHA3-384
|
||||
sha3_512 function = 35 // SHA3-512
|
||||
shake128 function = 36 // SHAKE-128
|
||||
shake256 function = 37 // SHAKE-256
|
||||
|
||||
// KLMD function codes
|
||||
ghash function = 65 // GHASH
|
||||
) |
||||
|
||||
// queryResult contains the result of a Query function
|
||||
// call. Bits are numbered in big endian order so the
|
||||
// leftmost bit (the MSB) is at index 0.
|
||||
type queryResult struct { |
||||
bits [2]uint64 |
||||
} |
||||
|
||||
// Has reports whether the given functions are present.
|
||||
func (q *queryResult) Has(fns ...function) bool { |
||||
if len(fns) == 0 { |
||||
panic("no function codes provided") |
||||
} |
||||
for _, f := range fns { |
||||
if !bitIsSet(q.bits[:], uint(f)) { |
||||
return false |
||||
} |
||||
} |
||||
return true |
||||
} |
||||
|
||||
// facility is a bit index for the named facility.
|
||||
type facility uint8 |
||||
|
||||
const ( |
||||
// cryptography facilities
|
||||
msa4 facility = 77 // message-security-assist extension 4
|
||||
msa8 facility = 146 // message-security-assist extension 8
|
||||
) |
||||
|
||||
// facilityList contains the result of an STFLE call.
|
||||
// Bits are numbered in big endian order so the
|
||||
// leftmost bit (the MSB) is at index 0.
|
||||
type facilityList struct { |
||||
bits [4]uint64 |
||||
} |
||||
|
||||
// Has reports whether the given facilities are present.
|
||||
func (s *facilityList) Has(fs ...facility) bool { |
||||
if len(fs) == 0 { |
||||
panic("no facility bits provided") |
||||
} |
||||
for _, f := range fs { |
||||
if !bitIsSet(s.bits[:], uint(f)) { |
||||
return false |
||||
} |
||||
} |
||||
return true |
||||
} |
||||
|
||||
func doinit() { |
||||
// test HWCAP bit vector
|
||||
has := func(featureMask uint) bool { |
||||
return hwCap&featureMask == featureMask |
||||
} |
||||
|
||||
// mandatory
|
||||
S390X.HasZARCH = has(hwcap_ZARCH) |
||||
|
||||
// optional
|
||||
S390X.HasSTFLE = has(hwcap_STFLE) |
||||
S390X.HasLDISP = has(hwcap_LDISP) |
||||
S390X.HasEIMM = has(hwcap_EIMM) |
||||
S390X.HasETF3EH = has(hwcap_ETF3EH) |
||||
S390X.HasDFP = has(hwcap_DFP) |
||||
S390X.HasMSA = has(hwcap_MSA) |
||||
S390X.HasVX = has(hwcap_VX) |
||||
if S390X.HasVX { |
||||
S390X.HasVXE = has(hwcap_VXE) |
||||
} |
||||
|
||||
// We need implementations of stfle, km and so on
|
||||
// to detect cryptographic features.
|
||||
if !haveAsmFunctions() { |
||||
return |
||||
} |
||||
|
||||
// optional cryptographic functions
|
||||
if S390X.HasMSA { |
||||
aes := []function{aes128, aes192, aes256} |
||||
|
||||
// cipher message
|
||||
km, kmc := kmQuery(), kmcQuery() |
||||
S390X.HasAES = km.Has(aes...) |
||||
S390X.HasAESCBC = kmc.Has(aes...) |
||||
if S390X.HasSTFLE { |
||||
facilities := stfle() |
||||
if facilities.Has(msa4) { |
||||
kmctr := kmctrQuery() |
||||
S390X.HasAESCTR = kmctr.Has(aes...) |
||||
} |
||||
if facilities.Has(msa8) { |
||||
kma := kmaQuery() |
||||
S390X.HasAESGCM = kma.Has(aes...) |
||||
} |
||||
} |
||||
|
||||
// compute message digest
|
||||
kimd := kimdQuery() // intermediate (no padding)
|
||||
klmd := klmdQuery() // last (padding)
|
||||
S390X.HasSHA1 = kimd.Has(sha1) && klmd.Has(sha1) |
||||
S390X.HasSHA256 = kimd.Has(sha256) && klmd.Has(sha256) |
||||
S390X.HasSHA512 = kimd.Has(sha512) && klmd.Has(sha512) |
||||
S390X.HasGHASH = kimd.Has(ghash) // KLMD-GHASH does not exist
|
||||
sha3 := []function{ |
||||
sha3_224, sha3_256, sha3_384, sha3_512, |
||||
shake128, shake256, |
||||
} |
||||
S390X.HasSHA3 = kimd.Has(sha3...) && klmd.Has(sha3...) |
||||
} |
||||
} |
||||
@ -0,0 +1,11 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build mips64 mips64le
|
||||
|
||||
package cpu |
||||
|
||||
const cacheLineSize = 32 |
||||
|
||||
func doinit() {} |
||||
@ -0,0 +1,11 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build mips mipsle
|
||||
|
||||
package cpu |
||||
|
||||
const cacheLineSize = 32 |
||||
|
||||
func doinit() {} |
||||
@ -0,0 +1,11 @@ |
||||
// Copyright 2019 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build !linux,arm64
|
||||
|
||||
package cpu |
||||
|
||||
const cacheLineSize = 64 |
||||
|
||||
func doinit() {} |
||||
@ -0,0 +1,57 @@ |
||||
// Copyright 2019 The Go Authors. All rights reserved. |
||||
// Use of this source code is governed by a BSD-style |
||||
// license that can be found in the LICENSE file. |
||||
|
||||
// +build !gccgo |
||||
|
||||
#include "textflag.h" |
||||
|
||||
// func stfle() facilityList |
||||
TEXT ·stfle(SB), NOSPLIT|NOFRAME, $0-32 |
||||
MOVD $ret+0(FP), R1 |
||||
MOVD $3, R0 // last doubleword index to store |
||||
XC $32, (R1), (R1) // clear 4 doublewords (32 bytes) |
||||
WORD $0xb2b01000 // store facility list extended (STFLE) |
||||
RET |
||||
|
||||
// func kmQuery() queryResult |
||||
TEXT ·kmQuery(SB), NOSPLIT|NOFRAME, $0-16 |
||||
MOVD $0, R0 // set function code to 0 (KM-Query) |
||||
MOVD $ret+0(FP), R1 // address of 16-byte return value |
||||
WORD $0xB92E0024 // cipher message (KM) |
||||
RET |
||||
|
||||
// func kmcQuery() queryResult |
||||
TEXT ·kmcQuery(SB), NOSPLIT|NOFRAME, $0-16 |
||||
MOVD $0, R0 // set function code to 0 (KMC-Query) |
||||
MOVD $ret+0(FP), R1 // address of 16-byte return value |
||||
WORD $0xB92F0024 // cipher message with chaining (KMC) |
||||
RET |
||||
|
||||
// func kmctrQuery() queryResult |
||||
TEXT ·kmctrQuery(SB), NOSPLIT|NOFRAME, $0-16 |
||||
MOVD $0, R0 // set function code to 0 (KMCTR-Query) |
||||
MOVD $ret+0(FP), R1 // address of 16-byte return value |
||||
WORD $0xB92D4024 // cipher message with counter (KMCTR) |
||||
RET |
||||
|
||||
// func kmaQuery() queryResult |
||||
TEXT ·kmaQuery(SB), NOSPLIT|NOFRAME, $0-16 |
||||
MOVD $0, R0 // set function code to 0 (KMA-Query) |
||||
MOVD $ret+0(FP), R1 // address of 16-byte return value |
||||
WORD $0xb9296024 // cipher message with authentication (KMA) |
||||
RET |
||||
|
||||
// func kimdQuery() queryResult |
||||
TEXT ·kimdQuery(SB), NOSPLIT|NOFRAME, $0-16 |
||||
MOVD $0, R0 // set function code to 0 (KIMD-Query) |
||||
MOVD $ret+0(FP), R1 // address of 16-byte return value |
||||
WORD $0xB93E0024 // compute intermediate message digest (KIMD) |
||||
RET |
||||
|
||||
// func klmdQuery() queryResult |
||||
TEXT ·klmdQuery(SB), NOSPLIT|NOFRAME, $0-16 |
||||
MOVD $0, R0 // set function code to 0 (KLMD-Query) |
||||
MOVD $ret+0(FP), R1 // address of 16-byte return value |
||||
WORD $0xB93F0024 // compute last message digest (KLMD) |
||||
RET |
||||
@ -0,0 +1,15 @@ |
||||
// Copyright 2019 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build wasm
|
||||
|
||||
package cpu |
||||
|
||||
// We're compiling the cpu package for an unknown (software-abstracted) CPU.
|
||||
// Make CacheLinePad an empty struct and hope that the usual struct alignment
|
||||
// rules are good enough.
|
||||
|
||||
const cacheLineSize = 0 |
||||
|
||||
func doinit() {} |
||||
@ -0,0 +1,59 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
// +build 386 amd64 amd64p32
|
||||
|
||||
package cpu |
||||
|
||||
const cacheLineSize = 64 |
||||
|
||||
func init() { |
||||
Initialized = true |
||||
|
||||
maxID, _, _, _ := cpuid(0, 0) |
||||
|
||||
if maxID < 1 { |
||||
return |
||||
} |
||||
|
||||
_, _, ecx1, edx1 := cpuid(1, 0) |
||||
X86.HasSSE2 = isSet(26, edx1) |
||||
|
||||
X86.HasSSE3 = isSet(0, ecx1) |
||||
X86.HasPCLMULQDQ = isSet(1, ecx1) |
||||
X86.HasSSSE3 = isSet(9, ecx1) |
||||
X86.HasFMA = isSet(12, ecx1) |
||||
X86.HasSSE41 = isSet(19, ecx1) |
||||
X86.HasSSE42 = isSet(20, ecx1) |
||||
X86.HasPOPCNT = isSet(23, ecx1) |
||||
X86.HasAES = isSet(25, ecx1) |
||||
X86.HasOSXSAVE = isSet(27, ecx1) |
||||
X86.HasRDRAND = isSet(30, ecx1) |
||||
|
||||
osSupportsAVX := false |
||||
// For XGETBV, OSXSAVE bit is required and sufficient.
|
||||
if X86.HasOSXSAVE { |
||||
eax, _ := xgetbv() |
||||
// Check if XMM and YMM registers have OS support.
|
||||
osSupportsAVX = isSet(1, eax) && isSet(2, eax) |
||||
} |
||||
|
||||
X86.HasAVX = isSet(28, ecx1) && osSupportsAVX |
||||
|
||||
if maxID < 7 { |
||||
return |
||||
} |
||||
|
||||
_, ebx7, _, _ := cpuid(7, 0) |
||||
X86.HasBMI1 = isSet(3, ebx7) |
||||
X86.HasAVX2 = isSet(5, ebx7) && osSupportsAVX |
||||
X86.HasBMI2 = isSet(8, ebx7) |
||||
X86.HasERMS = isSet(9, ebx7) |
||||
X86.HasRDSEED = isSet(18, ebx7) |
||||
X86.HasADX = isSet(19, ebx7) |
||||
} |
||||
|
||||
func isSet(bitpos uint, value uint32) bool { |
||||
return value&(1<<bitpos) != 0 |
||||
} |
||||
@ -0,0 +1,27 @@ |
||||
// Copyright 2018 The Go Authors. All rights reserved. |
||||
// Use of this source code is governed by a BSD-style |
||||
// license that can be found in the LICENSE file. |
||||
|
||||
// +build 386 amd64 amd64p32 |
||||
// +build !gccgo |
||||
|
||||
#include "textflag.h" |
||||
|
||||
// func cpuid(eaxArg, ecxArg uint32) (eax, ebx, ecx, edx uint32) |
||||
TEXT ·cpuid(SB), NOSPLIT, $0-24 |
||||
MOVL eaxArg+0(FP), AX |
||||
MOVL ecxArg+4(FP), CX |
||||
CPUID |
||||
MOVL AX, eax+8(FP) |
||||
MOVL BX, ebx+12(FP) |
||||
MOVL CX, ecx+16(FP) |
||||
MOVL DX, edx+20(FP) |
||||
RET |
||||
|
||||
// func xgetbv() (eax, edx uint32) |
||||
TEXT ·xgetbv(SB),NOSPLIT,$0-8 |
||||
MOVL $0, CX |
||||
XGETBV |
||||
MOVL AX, eax+0(FP) |
||||
MOVL DX, edx+4(FP) |
||||
RET |
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue