mirror of
https://github.com/google/cpu_features.git
synced 2025-04-28 07:23:37 +02:00
129 lines
4.4 KiB
C
129 lines
4.4 KiB
C
// Copyright 2017 Google Inc.
|
|
//
|
|
// 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.
|
|
|
|
#include "cpuinfo_aarch64.h"
|
|
|
|
#include "internal/filesystem.h"
|
|
#include "internal/hwcaps.h"
|
|
#include "internal/stack_line_reader.h"
|
|
#include "internal/string_view.h"
|
|
#include "internal/unix_features_aggregator.h"
|
|
|
|
#include <assert.h>
|
|
#include <ctype.h>
|
|
|
|
DECLARE_SETTER(Aarch64Features, fp)
|
|
DECLARE_SETTER(Aarch64Features, asimd)
|
|
DECLARE_SETTER(Aarch64Features, aes)
|
|
DECLARE_SETTER(Aarch64Features, pmull)
|
|
DECLARE_SETTER(Aarch64Features, sha1)
|
|
DECLARE_SETTER(Aarch64Features, sha2)
|
|
DECLARE_SETTER(Aarch64Features, crc32)
|
|
|
|
static const CapabilityConfig kConfigs[] = {
|
|
[AARCH64_FP] = {{AARCH64_HWCAP_FP, 0}, "fp", &set_fp}, //
|
|
[AARCH64_ASIMD] = {{AARCH64_HWCAP_ASIMD, 0}, "asimd", &set_asimd}, //
|
|
[AARCH64_AES] = {{AARCH64_HWCAP_AES, 0}, "aes", &set_aes}, //
|
|
[AARCH64_PMULL] = {{AARCH64_HWCAP_PMULL, 0}, "pmull", &set_pmull}, //
|
|
[AARCH64_SHA1] = {{AARCH64_HWCAP_SHA1, 0}, "sha1", &set_sha1}, //
|
|
[AARCH64_SHA2] = {{AARCH64_HWCAP_SHA2, 0}, "sha2", &set_sha2}, //
|
|
[AARCH64_CRC32] {{AARCH64_HWCAP_CRC32, 0}, "crc32", &set_crc32}, //
|
|
};
|
|
|
|
static const size_t kConfigsSize = sizeof(kConfigs) / sizeof(CapabilityConfig);
|
|
|
|
static bool HandleAarch64Line(const LineResult result,
|
|
Aarch64Info* const info) {
|
|
StringView line = result.line;
|
|
StringView key, value;
|
|
if (CpuFeatures_StringView_GetAttributeKeyValue(line, &key, &value)) {
|
|
if (CpuFeatures_StringView_IsEquals(key, str("Features"))) {
|
|
CpuFeatures_SetFromFlags(kConfigsSize, kConfigs, value, &info->features);
|
|
} else if (CpuFeatures_StringView_IsEquals(key, str("CPU implementer"))) {
|
|
info->implementer = CpuFeatures_StringView_ParsePositiveNumber(value);
|
|
} else if (CpuFeatures_StringView_IsEquals(key, str("CPU variant"))) {
|
|
info->variant = CpuFeatures_StringView_ParsePositiveNumber(value);
|
|
} else if (CpuFeatures_StringView_IsEquals(key, str("CPU part"))) {
|
|
info->part = CpuFeatures_StringView_ParsePositiveNumber(value);
|
|
} else if (CpuFeatures_StringView_IsEquals(key, str("CPU revision"))) {
|
|
info->revision = CpuFeatures_StringView_ParsePositiveNumber(value);
|
|
}
|
|
}
|
|
return !result.eof;
|
|
}
|
|
|
|
static void FillProcCpuInfoData(Aarch64Info* const info) {
|
|
const int fd = CpuFeatures_OpenFile("/proc/cpuinfo");
|
|
if (fd >= 0) {
|
|
StackLineReader reader;
|
|
StackLineReader_Initialize(&reader, fd);
|
|
for (;;) {
|
|
if (!HandleAarch64Line(StackLineReader_NextLine(&reader), info)) {
|
|
break;
|
|
}
|
|
}
|
|
CpuFeatures_CloseFile(fd);
|
|
}
|
|
}
|
|
|
|
static const Aarch64Info kEmptyAarch64Info;
|
|
|
|
Aarch64Info GetAarch64Info(void) {
|
|
assert(kConfigsSize == AARCH64_LAST_);
|
|
|
|
// capabilities are fetched from both getauxval and /proc/cpuinfo so we can
|
|
// have some information if the executable is sandboxed (aka no access to
|
|
// /proc/cpuinfo).
|
|
Aarch64Info info = kEmptyAarch64Info;
|
|
|
|
FillProcCpuInfoData(&info);
|
|
CpuFeatures_OverrideFromHwCaps(kConfigsSize, kConfigs,
|
|
CpuFeatures_GetHardwareCapabilities(),
|
|
&info.features);
|
|
|
|
return info;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// Introspection functions
|
|
|
|
int GetAarch64FeaturesEnumValue(const Aarch64Features* features,
|
|
Aarch64FeaturesEnum value) {
|
|
switch (value) {
|
|
case AARCH64_FP:
|
|
return features->fp;
|
|
case AARCH64_ASIMD:
|
|
return features->asimd;
|
|
case AARCH64_AES:
|
|
return features->aes;
|
|
case AARCH64_PMULL:
|
|
return features->pmull;
|
|
case AARCH64_SHA1:
|
|
return features->sha1;
|
|
case AARCH64_SHA2:
|
|
return features->sha2;
|
|
case AARCH64_CRC32:
|
|
return features->crc32;
|
|
case AARCH64_LAST_:
|
|
break;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
const char* GetAarch64FeaturesEnumName(Aarch64FeaturesEnum value) {
|
|
if(value >= kConfigsSize)
|
|
return "unknown feature";
|
|
return kConfigs[value].proc_cpuinfo_flag;
|
|
}
|