xref: /freebsd/contrib/llvm-project/compiler-rt/lib/scudo/standalone/checksum.cpp (revision 85868e8a1daeaae7a0e48effb2ea2310ae3b02c6)
1 //===-- checksum.cpp --------------------------------------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "checksum.h"
10 #include "atomic_helpers.h"
11 
12 #if defined(__x86_64__) || defined(__i386__)
13 #include <cpuid.h>
14 #elif defined(__arm__) || defined(__aarch64__)
15 #if SCUDO_FUCHSIA
16 #include <zircon/features.h>
17 #include <zircon/syscalls.h>
18 #else
19 #include <sys/auxv.h>
20 #endif
21 #endif
22 
23 namespace scudo {
24 
25 Checksum HashAlgorithm = {Checksum::BSD};
26 
27 #if defined(__x86_64__) || defined(__i386__)
28 // i386 and x86_64 specific code to detect CRC32 hardware support via CPUID.
29 // CRC32 requires the SSE 4.2 instruction set.
30 #ifndef bit_SSE4_2
31 #define bit_SSE4_2 bit_SSE42 // clang and gcc have different defines.
32 #endif
33 
34 bool hasHardwareCRC32() {
35   u32 Eax, Ebx = 0, Ecx = 0, Edx = 0;
36   __get_cpuid(0, &Eax, &Ebx, &Ecx, &Edx);
37   const bool IsIntel = (Ebx == signature_INTEL_ebx) &&
38                        (Edx == signature_INTEL_edx) &&
39                        (Ecx == signature_INTEL_ecx);
40   const bool IsAMD = (Ebx == signature_AMD_ebx) && (Edx == signature_AMD_edx) &&
41                      (Ecx == signature_AMD_ecx);
42   if (!IsIntel && !IsAMD)
43     return false;
44   __get_cpuid(1, &Eax, &Ebx, &Ecx, &Edx);
45   return !!(Ecx & bit_SSE4_2);
46 }
47 
48 #elif defined(__arm__) || defined(__aarch64__)
49 #ifndef AT_HWCAP
50 #define AT_HWCAP 16
51 #endif
52 #ifndef HWCAP_CRC32
53 #define HWCAP_CRC32 (1U << 7) // HWCAP_CRC32 is missing on older platforms.
54 #endif
55 
56 bool hasHardwareCRC32() {
57 #if SCUDO_FUCHSIA
58   u32 HWCap;
59   const zx_status_t Status =
60       zx_system_get_features(ZX_FEATURE_KIND_CPU, &HWCap);
61   if (Status != ZX_OK)
62     return false;
63   return !!(HWCap & ZX_ARM64_FEATURE_ISA_CRC32);
64 #else
65   return !!(getauxval(AT_HWCAP) & HWCAP_CRC32);
66 #endif // SCUDO_FUCHSIA
67 }
68 #endif // defined(__x86_64__) || defined(__i386__)
69 
70 } // namespace scudo
71