cpu_linux_arm64.go

     1  // Copyright 2018 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  package cpu
     6  
     7  import (
     8  	"strings"
     9  	"syscall"
    10  )
    11  
    12  // HWCAP/HWCAP2 bits. These are exposed by Linux.
    13  const (
    14  	hwcap_FP       = 1 << 0
    15  	hwcap_ASIMD    = 1 << 1
    16  	hwcap_EVTSTRM  = 1 << 2
    17  	hwcap_AES      = 1 << 3
    18  	hwcap_PMULL    = 1 << 4
    19  	hwcap_SHA1     = 1 << 5
    20  	hwcap_SHA2     = 1 << 6
    21  	hwcap_CRC32    = 1 << 7
    22  	hwcap_ATOMICS  = 1 << 8
    23  	hwcap_FPHP     = 1 << 9
    24  	hwcap_ASIMDHP  = 1 << 10
    25  	hwcap_CPUID    = 1 << 11
    26  	hwcap_ASIMDRDM = 1 << 12
    27  	hwcap_JSCVT    = 1 << 13
    28  	hwcap_FCMA     = 1 << 14
    29  	hwcap_LRCPC    = 1 << 15
    30  	hwcap_DCPOP    = 1 << 16
    31  	hwcap_SHA3     = 1 << 17
    32  	hwcap_SM3      = 1 << 18
    33  	hwcap_SM4      = 1 << 19
    34  	hwcap_ASIMDDP  = 1 << 20
    35  	hwcap_SHA512   = 1 << 21
    36  	hwcap_SVE      = 1 << 22
    37  	hwcap_ASIMDFHM = 1 << 23
    38  	hwcap_DIT      = 1 << 24
    39  
    40  	hwcap2_SVE2 = 1 << 1
    41  	hwcap2_I8MM = 1 << 13
    42  )
    43  
    44  // linuxKernelCanEmulateCPUID reports whether we're running
    45  // on Linux 4.11+. Ideally we'd like to ask the question about
    46  // whether the current kernel contains
    47  // https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git/commit/?id=77c97b4ee21290f5f083173d957843b615abbff2
    48  // but the version number will have to do.
    49  func linuxKernelCanEmulateCPUID() bool {
    50  	var un syscall.Utsname
    51  	syscall.Uname(&un)
    52  	var sb strings.Builder
    53  	for _, b := range un.Release[:] {
    54  		if b == 0 {
    55  			break
    56  		}
    57  		sb.WriteByte(byte(b))
    58  	}
    59  	major, minor, _, ok := parseRelease(sb.String())
    60  	return ok && (major > 4 || major == 4 && minor >= 11)
    61  }
    62  
    63  func doinit() {
    64  	if err := readHWCAP(); err != nil {
    65  		// We failed to read /proc/self/auxv. This can happen if the binary has
    66  		// been given extra capabilities(7) with /bin/setcap.
    67  		//
    68  		// When this happens, we have two options. If the Linux kernel is new
    69  		// enough (4.11+), we can read the arm64 registers directly which'll
    70  		// trap into the kernel and then return back to userspace.
    71  		//
    72  		// But on older kernels, such as Linux 4.4.180 as used on many Synology
    73  		// devices, calling readARM64Registers (specifically getisar0) will
    74  		// cause a SIGILL and we'll die. So for older kernels, parse /proc/cpuinfo
    75  		// instead.
    76  		//
    77  		// See golang/go#57336.
    78  		if linuxKernelCanEmulateCPUID() {
    79  			readARM64Registers()
    80  		} else {
    81  			readLinuxProcCPUInfo()
    82  		}
    83  		return
    84  	}
    85  
    86  	// HWCAP feature bits
    87  	ARM64.HasFP = isSet(hwCap, hwcap_FP)
    88  	ARM64.HasASIMD = isSet(hwCap, hwcap_ASIMD)
    89  	ARM64.HasEVTSTRM = isSet(hwCap, hwcap_EVTSTRM)
    90  	ARM64.HasAES = isSet(hwCap, hwcap_AES)
    91  	ARM64.HasPMULL = isSet(hwCap, hwcap_PMULL)
    92  	ARM64.HasSHA1 = isSet(hwCap, hwcap_SHA1)
    93  	ARM64.HasSHA2 = isSet(hwCap, hwcap_SHA2)
    94  	ARM64.HasCRC32 = isSet(hwCap, hwcap_CRC32)
    95  	ARM64.HasATOMICS = isSet(hwCap, hwcap_ATOMICS)
    96  	ARM64.HasFPHP = isSet(hwCap, hwcap_FPHP)
    97  	ARM64.HasASIMDHP = isSet(hwCap, hwcap_ASIMDHP)
    98  	ARM64.HasCPUID = isSet(hwCap, hwcap_CPUID)
    99  	ARM64.HasASIMDRDM = isSet(hwCap, hwcap_ASIMDRDM)
   100  	ARM64.HasJSCVT = isSet(hwCap, hwcap_JSCVT)
   101  	ARM64.HasFCMA = isSet(hwCap, hwcap_FCMA)
   102  	ARM64.HasLRCPC = isSet(hwCap, hwcap_LRCPC)
   103  	ARM64.HasDCPOP = isSet(hwCap, hwcap_DCPOP)
   104  	ARM64.HasSHA3 = isSet(hwCap, hwcap_SHA3)
   105  	ARM64.HasSM3 = isSet(hwCap, hwcap_SM3)
   106  	ARM64.HasSM4 = isSet(hwCap, hwcap_SM4)
   107  	ARM64.HasASIMDDP = isSet(hwCap, hwcap_ASIMDDP)
   108  	ARM64.HasSHA512 = isSet(hwCap, hwcap_SHA512)
   109  	ARM64.HasSVE = isSet(hwCap, hwcap_SVE)
   110  	ARM64.HasASIMDFHM = isSet(hwCap, hwcap_ASIMDFHM)
   111  	ARM64.HasDIT = isSet(hwCap, hwcap_DIT)
   112  
   113  	// HWCAP2 feature bits
   114  	ARM64.HasSVE2 = isSet(hwCap2, hwcap2_SVE2)
   115  	ARM64.HasI8MM = isSet(hwCap2, hwcap2_I8MM)
   116  }
   117  
   118  func isSet(hwc uint, value uint) bool {
   119  	return hwc&value != 0
   120  }
   121
View as plain text