1 // Copyright 2017 Google LLC
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //    http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 #include "cpuinfo_aarch64.h"
16 
17 #include "filesystem_for_testing.h"
18 #include "gtest/gtest.h"
19 #include "hwcaps_for_testing.h"
20 
21 namespace cpu_features {
22 namespace {
23 
DisableHardwareCapabilities()24 void DisableHardwareCapabilities() { SetHardwareCapabilities(0, 0); }
25 
TEST(CpuinfoAarch64Test,FromHardwareCap)26 TEST(CpuinfoAarch64Test, FromHardwareCap) {
27   SetHardwareCapabilities(AARCH64_HWCAP_FP | AARCH64_HWCAP_AES, 0);
28   GetEmptyFilesystem();  // disabling /proc/cpuinfo
29   const auto info = GetAarch64Info();
30   EXPECT_TRUE(info.features.fp);
31   EXPECT_FALSE(info.features.asimd);
32   EXPECT_FALSE(info.features.evtstrm);
33   EXPECT_TRUE(info.features.aes);
34   EXPECT_FALSE(info.features.pmull);
35   EXPECT_FALSE(info.features.sha1);
36   EXPECT_FALSE(info.features.sha2);
37   EXPECT_FALSE(info.features.crc32);
38   EXPECT_FALSE(info.features.atomics);
39   EXPECT_FALSE(info.features.fphp);
40   EXPECT_FALSE(info.features.asimdhp);
41   EXPECT_FALSE(info.features.cpuid);
42   EXPECT_FALSE(info.features.asimdrdm);
43   EXPECT_FALSE(info.features.jscvt);
44   EXPECT_FALSE(info.features.fcma);
45   EXPECT_FALSE(info.features.lrcpc);
46   EXPECT_FALSE(info.features.dcpop);
47   EXPECT_FALSE(info.features.sha3);
48   EXPECT_FALSE(info.features.sm3);
49   EXPECT_FALSE(info.features.sm4);
50   EXPECT_FALSE(info.features.asimddp);
51   EXPECT_FALSE(info.features.sha512);
52   EXPECT_FALSE(info.features.sve);
53   EXPECT_FALSE(info.features.asimdfhm);
54   EXPECT_FALSE(info.features.dit);
55   EXPECT_FALSE(info.features.uscat);
56   EXPECT_FALSE(info.features.ilrcpc);
57   EXPECT_FALSE(info.features.flagm);
58   EXPECT_FALSE(info.features.ssbs);
59   EXPECT_FALSE(info.features.sb);
60   EXPECT_FALSE(info.features.paca);
61   EXPECT_FALSE(info.features.pacg);
62 }
63 
TEST(CpuinfoAarch64Test,FromHardwareCap2)64 TEST(CpuinfoAarch64Test, FromHardwareCap2) {
65   SetHardwareCapabilities(AARCH64_HWCAP_FP,
66                           AARCH64_HWCAP2_SVE2 | AARCH64_HWCAP2_BTI);
67   GetEmptyFilesystem();  // disabling /proc/cpuinfo
68   const auto info = GetAarch64Info();
69   EXPECT_TRUE(info.features.fp);
70 
71   EXPECT_TRUE(info.features.sve2);
72   EXPECT_TRUE(info.features.bti);
73 
74   EXPECT_FALSE(info.features.dcpodp);
75   EXPECT_FALSE(info.features.sveaes);
76   EXPECT_FALSE(info.features.svepmull);
77   EXPECT_FALSE(info.features.svebitperm);
78   EXPECT_FALSE(info.features.svesha3);
79   EXPECT_FALSE(info.features.svesm4);
80   EXPECT_FALSE(info.features.flagm2);
81   EXPECT_FALSE(info.features.frint);
82   EXPECT_FALSE(info.features.svei8mm);
83   EXPECT_FALSE(info.features.svef32mm);
84   EXPECT_FALSE(info.features.svef64mm);
85   EXPECT_FALSE(info.features.svebf16);
86   EXPECT_FALSE(info.features.i8mm);
87   EXPECT_FALSE(info.features.bf16);
88   EXPECT_FALSE(info.features.dgh);
89   EXPECT_FALSE(info.features.rng);
90 }
91 
TEST(CpuinfoAarch64Test,ARMCortexA53)92 TEST(CpuinfoAarch64Test, ARMCortexA53) {
93   DisableHardwareCapabilities();
94   auto& fs = GetEmptyFilesystem();
95   fs.CreateFile("/proc/cpuinfo",
96                 R"(Processor   : AArch64 Processor rev 3 (aarch64)
97 processor   : 0
98 processor   : 1
99 processor   : 2
100 processor   : 3
101 processor   : 4
102 processor   : 5
103 processor   : 6
104 processor   : 7
105 Features    : fp asimd evtstrm aes pmull sha1 sha2 crc32
106 CPU implementer : 0x41
107 CPU architecture: AArch64
108 CPU variant : 0x0
109 CPU part    : 0xd03
110 CPU revision    : 3)");
111   const auto info = GetAarch64Info();
112   EXPECT_EQ(info.implementer, 0x41);
113   EXPECT_EQ(info.variant, 0x0);
114   EXPECT_EQ(info.part, 0xd03);
115   EXPECT_EQ(info.revision, 3);
116 
117   EXPECT_TRUE(info.features.fp);
118   EXPECT_TRUE(info.features.asimd);
119   EXPECT_TRUE(info.features.evtstrm);
120   EXPECT_TRUE(info.features.aes);
121   EXPECT_TRUE(info.features.pmull);
122   EXPECT_TRUE(info.features.sha1);
123   EXPECT_TRUE(info.features.sha2);
124   EXPECT_TRUE(info.features.crc32);
125 
126   EXPECT_FALSE(info.features.atomics);
127   EXPECT_FALSE(info.features.fphp);
128   EXPECT_FALSE(info.features.asimdhp);
129   EXPECT_FALSE(info.features.cpuid);
130   EXPECT_FALSE(info.features.asimdrdm);
131   EXPECT_FALSE(info.features.jscvt);
132   EXPECT_FALSE(info.features.fcma);
133   EXPECT_FALSE(info.features.lrcpc);
134   EXPECT_FALSE(info.features.dcpop);
135   EXPECT_FALSE(info.features.sha3);
136   EXPECT_FALSE(info.features.sm3);
137   EXPECT_FALSE(info.features.sm4);
138   EXPECT_FALSE(info.features.asimddp);
139   EXPECT_FALSE(info.features.sha512);
140   EXPECT_FALSE(info.features.sve);
141   EXPECT_FALSE(info.features.asimdfhm);
142   EXPECT_FALSE(info.features.dit);
143   EXPECT_FALSE(info.features.uscat);
144   EXPECT_FALSE(info.features.ilrcpc);
145   EXPECT_FALSE(info.features.flagm);
146   EXPECT_FALSE(info.features.ssbs);
147   EXPECT_FALSE(info.features.sb);
148   EXPECT_FALSE(info.features.paca);
149   EXPECT_FALSE(info.features.pacg);
150   EXPECT_FALSE(info.features.dcpodp);
151   EXPECT_FALSE(info.features.sve2);
152   EXPECT_FALSE(info.features.sveaes);
153   EXPECT_FALSE(info.features.svepmull);
154   EXPECT_FALSE(info.features.svebitperm);
155   EXPECT_FALSE(info.features.svesha3);
156   EXPECT_FALSE(info.features.svesm4);
157   EXPECT_FALSE(info.features.flagm2);
158   EXPECT_FALSE(info.features.frint);
159   EXPECT_FALSE(info.features.svei8mm);
160   EXPECT_FALSE(info.features.svef32mm);
161   EXPECT_FALSE(info.features.svef64mm);
162   EXPECT_FALSE(info.features.svebf16);
163   EXPECT_FALSE(info.features.i8mm);
164   EXPECT_FALSE(info.features.bf16);
165   EXPECT_FALSE(info.features.dgh);
166   EXPECT_FALSE(info.features.rng);
167   EXPECT_FALSE(info.features.bti);
168 }
169 
170 }  // namespace
171 }  // namespace cpu_features
172