#include "base/system/sys_info.h"
#include <stdint.h>
#include <optional>
#include <string_view>
#include <utility>
#include <vector>
#include "base/environment.h"
#include "base/files/file_util.h"
#include "base/functional/bind.h"
#include "base/numerics/safe_conversions.h"
#include "base/process/process_metrics.h"
#include "base/run_loop.h"
#include "base/strings/pattern.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_split.h"
#include "base/strings/string_util.h"
#include "base/strings/sys_string_conversions.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/scoped_chromeos_version_info.h"
#include "base/test/scoped_running_on_chromeos.h"
#include "base/test/task_environment.h"
#include "base/threading/platform_thread.h"
#include "base/threading/scoped_blocking_call.h"
#include "base/time/time.h"
#include "build/build_config.h"
#include "testing/gtest/include/gtest/gtest-death-test.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"
#if BUILDFLAG(IS_WIN)
#include "base/win/com_init_util.h"
#include "base/win/scoped_bstr.h"
#include "base/win/scoped_com_initializer.h"
#include "base/win/scoped_variant.h"
#include "base/win/wmi.h"
#endif
#if BUILDFLAG(IS_MAC)
#include "base/system/sys_info_internal.h"
#include "base/test/scoped_feature_list.h"
#endif
namespace base {
#if BUILDFLAG(IS_ANDROID)
static constexpr int kReservedPhysicalMemory = 300 * 1024;
#endif
SysInfoTest;
TEST_F(SysInfoTest, NumProcs) { … }
#if BUILDFLAG(IS_MAC)
TEST_F(SysInfoTest, NumProcsWithSecurityMitigationEnabled) {
SysInfo::ResetCpuSecurityMitigationsEnabledForTesting();
test::ScopedFeatureList feature_list_;
feature_list_.InitAndEnableFeature(kNumberOfCoresWithCpuSecurityMitigation);
SysInfo::SetCpuSecurityMitigationsEnabled();
EXPECT_EQ(SysInfo::NumberOfProcessors(),
internal::NumberOfPhysicalProcessors());
SysInfo::ResetCpuSecurityMitigationsEnabledForTesting();
}
#endif
TEST_F(SysInfoTest, AmountOfMem) { … }
#if BUILDFLAG(IS_LINUX) || BUILDFLAG(IS_CHROMEOS) || BUILDFLAG(IS_ANDROID)
#if BUILDFLAG(IS_LINUX) || BUILDFLAG(IS_CHROMEOS)
#define MAYBE_AmountOfAvailablePhysicalMemory …
#else
#define MAYBE_AmountOfAvailablePhysicalMemory …
#endif
TEST_F(SysInfoTest, MAYBE_AmountOfAvailablePhysicalMemory) { … }
#endif
TEST_F(SysInfoTest, AmountOfFreeDiskSpace) { … }
TEST_F(SysInfoTest, AmountOfTotalDiskSpace) { … }
#if BUILDFLAG(IS_FUCHSIA)
TEST_F(SysInfoTest, NestedVolumesAmountOfTotalDiskSpace) {
constexpr int64_t kOuterVolumeQuota = 1024;
constexpr int64_t kInnerVolumeQuota = kOuterVolumeQuota / 2;
FilePath tmp_path;
ASSERT_TRUE(GetTempDir(&tmp_path));
SysInfo::SetAmountOfTotalDiskSpace(tmp_path, kOuterVolumeQuota);
const FilePath subdirectory_path = tmp_path.Append("subdirectory");
SysInfo::SetAmountOfTotalDiskSpace(subdirectory_path, kInnerVolumeQuota);
EXPECT_EQ(SysInfo::AmountOfTotalDiskSpace(tmp_path), kOuterVolumeQuota);
EXPECT_EQ(SysInfo::AmountOfTotalDiskSpace(subdirectory_path),
kInnerVolumeQuota);
SysInfo::SetAmountOfTotalDiskSpace(subdirectory_path, -1);
EXPECT_EQ(SysInfo::AmountOfTotalDiskSpace(subdirectory_path),
kOuterVolumeQuota);
}
#endif
#if BUILDFLAG(IS_WIN) || BUILDFLAG(IS_APPLE) || BUILDFLAG(IS_LINUX) || \
BUILDFLAG(IS_CHROMEOS) || BUILDFLAG(IS_FUCHSIA)
TEST_F(SysInfoTest, OperatingSystemVersion) { … }
TEST_F(SysInfoTest, OperatingSystemVersionNumbers) { … }
#endif
#if BUILDFLAG(IS_IOS)
TEST_F(SysInfoTest, GetIOSBuildNumber) {
std::string build_number(SysInfo::GetIOSBuildNumber());
EXPECT_GT(build_number.length(), 0U);
}
#endif
TEST_F(SysInfoTest, Uptime) { … }
#if BUILDFLAG(IS_APPLE)
TEST_F(SysInfoTest, HardwareModelNameFormatMacAndiOS) {
std::string hardware_model = SysInfo::HardwareModelName();
ASSERT_FALSE(hardware_model.empty());
#if BUILDFLAG(IS_IOS) && TARGET_OS_SIMULATOR
EXPECT_TRUE(base::MatchPattern(hardware_model, "iOS Simulator (*)"))
<< hardware_model;
std::vector<std::string_view> mainPieces =
SplitStringPiece(hardware_model, "()", KEEP_WHITESPACE, SPLIT_WANT_ALL);
ASSERT_EQ(3u, mainPieces.size()) << hardware_model;
std::vector<std::string_view> modelPieces =
SplitStringPiece(mainPieces[1], ",", KEEP_WHITESPACE, SPLIT_WANT_ALL);
ASSERT_GE(modelPieces.size(), 1u) << hardware_model;
if (modelPieces.size() == 1u) {
EXPECT_TRUE(modelPieces[0] == "Unknown" || modelPieces[0] == "iPhone" ||
modelPieces[0] == "iPad")
<< hardware_model;
} else {
int value;
EXPECT_TRUE(StringToInt(modelPieces[1], &value)) << hardware_model;
}
#else
std::vector<std::string_view> pieces =
SplitStringPiece(hardware_model, ",", KEEP_WHITESPACE, SPLIT_WANT_ALL);
ASSERT_EQ(2u, pieces.size()) << hardware_model;
int value;
EXPECT_TRUE(StringToInt(pieces[1], &value)) << hardware_model;
#endif
}
#endif
TEST_F(SysInfoTest, GetHardwareInfo) { … }
#if BUILDFLAG(IS_WIN)
TEST_F(SysInfoTest, GetHardwareInfoWMIMatchRegistry) {
base::win::ScopedCOMInitializer com_initializer;
test::TaskEnvironment task_environment;
std::optional<SysInfo::HardwareInfo> hardware_info;
auto callback = base::BindOnce(
[](std::optional<SysInfo::HardwareInfo>* target_info,
SysInfo::HardwareInfo info) { *target_info = std::move(info); },
&hardware_info);
SysInfo::GetHardwareInfo(std::move(callback));
task_environment.RunUntilIdle();
ASSERT_TRUE(hardware_info.has_value());
Microsoft::WRL::ComPtr<IWbemServices> wmi_services;
EXPECT_TRUE(base::win::CreateLocalWmiConnection(true, &wmi_services));
static constexpr wchar_t query_computer_system[] =
L"SELECT Manufacturer,Model FROM Win32_ComputerSystem";
Microsoft::WRL::ComPtr<IEnumWbemClassObject> enumerator_computer_system;
HRESULT hr = wmi_services->ExecQuery(
base::win::ScopedBstr(L"WQL").Get(),
base::win::ScopedBstr(query_computer_system).Get(),
WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY, nullptr,
&enumerator_computer_system);
EXPECT_FALSE(FAILED(hr) || !enumerator_computer_system.Get());
Microsoft::WRL::ComPtr<IWbemClassObject> class_object;
ULONG items_returned = 0;
hr = enumerator_computer_system->Next(WBEM_INFINITE, 1, &class_object,
&items_returned);
EXPECT_FALSE(FAILED(hr) || !items_returned);
base::win::ScopedVariant manufacturerVar;
std::wstring manufacturer;
hr = class_object->Get(L"Manufacturer", 0, manufacturerVar.Receive(), nullptr,
nullptr);
if (SUCCEEDED(hr) && manufacturerVar.type() == VT_BSTR) {
manufacturer.assign(V_BSTR(manufacturerVar.ptr()),
::SysStringLen(V_BSTR(manufacturerVar.ptr())));
}
base::win::ScopedVariant modelVar;
std::wstring model;
hr = class_object->Get(L"Model", 0, modelVar.Receive(), nullptr, nullptr);
if (SUCCEEDED(hr) && modelVar.type() == VT_BSTR) {
model.assign(V_BSTR(modelVar.ptr()),
::SysStringLen(V_BSTR(modelVar.ptr())));
}
EXPECT_TRUE(hardware_info->manufacturer == base::SysWideToUTF8(manufacturer));
EXPECT_TRUE(hardware_info->model == base::SysWideToUTF8(model));
}
#endif
#if BUILDFLAG(IS_CHROMEOS)
TEST_F(SysInfoTest, GoogleChromeOSVersionNumbers) {
int32_t os_major_version = -1;
int32_t os_minor_version = -1;
int32_t os_bugfix_version = -1;
const char kLsbRelease[] =
"FOO=1234123.34.5\n"
"CHROMEOS_RELEASE_VERSION=1.2.3.4\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease, Time());
SysInfo::OperatingSystemVersionNumbers(&os_major_version, &os_minor_version,
&os_bugfix_version);
EXPECT_EQ(1, os_major_version);
EXPECT_EQ(2, os_minor_version);
EXPECT_EQ(3, os_bugfix_version);
}
TEST_F(SysInfoTest, GoogleChromeOSVersionNumbersFirst) {
int32_t os_major_version = -1;
int32_t os_minor_version = -1;
int32_t os_bugfix_version = -1;
const char kLsbRelease[] =
"CHROMEOS_RELEASE_VERSION=1.2.3.4\n"
"FOO=1234123.34.5\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease, Time());
SysInfo::OperatingSystemVersionNumbers(&os_major_version, &os_minor_version,
&os_bugfix_version);
EXPECT_EQ(1, os_major_version);
EXPECT_EQ(2, os_minor_version);
EXPECT_EQ(3, os_bugfix_version);
}
TEST_F(SysInfoTest, GoogleChromeOSNoVersionNumbers) {
int32_t os_major_version = -1;
int32_t os_minor_version = -1;
int32_t os_bugfix_version = -1;
const char kLsbRelease[] = "FOO=1234123.34.5\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease, Time());
SysInfo::OperatingSystemVersionNumbers(&os_major_version, &os_minor_version,
&os_bugfix_version);
EXPECT_EQ(0, os_major_version);
EXPECT_EQ(0, os_minor_version);
EXPECT_EQ(0, os_bugfix_version);
}
TEST_F(SysInfoTest, GoogleChromeOSLsbReleaseTime) {
const char kLsbRelease[] = "CHROMEOS_RELEASE_VERSION=1.2.3.4";
const Time lsb_release_time(Time::FromSecondsSinceUnixEpoch(12345.6));
test::ScopedChromeOSVersionInfo version(kLsbRelease, lsb_release_time);
Time parsed_lsb_release_time = SysInfo::GetLsbReleaseTime();
EXPECT_DOUBLE_EQ(lsb_release_time.InSecondsFSinceUnixEpoch(),
parsed_lsb_release_time.InSecondsFSinceUnixEpoch());
}
TEST_F(SysInfoTest, IsRunningOnChromeOS) {
{
const char kLsbRelease1[] =
"CHROMEOS_RELEASE_NAME=Non Chrome OS\n"
"CHROMEOS_RELEASE_VERSION=1.2.3.4\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease1, Time());
EXPECT_FALSE(SysInfo::IsRunningOnChromeOS());
}
{
const char kLsbRelease2[] =
"CHROMEOS_RELEASE_NAME=Chrome OS\n"
"CHROMEOS_RELEASE_VERSION=1.2.3.4\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease2, Time());
EXPECT_TRUE(SysInfo::IsRunningOnChromeOS());
}
{
const char kLsbRelease3[] = "CHROMEOS_RELEASE_NAME=Chromium OS\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease3, Time());
EXPECT_TRUE(SysInfo::IsRunningOnChromeOS());
}
}
TEST_F(SysInfoTest, ScopedChromeOSVersionInfoDoesNotChangeEnvironment) {
std::unique_ptr<Environment> environment = Environment::Create();
ASSERT_FALSE(environment->HasVar("LSB_RELEASE"));
{
const char kLsbRelease[] =
"CHROMEOS_RELEASE_NAME=Chrome OS\n"
"CHROMEOS_RELEASE_VERSION=1.2.3.4\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease, Time());
}
EXPECT_FALSE(environment->HasVar("LSB_RELEASE"));
}
TEST_F(SysInfoTest, CrashOnBaseImage) {
const char kLsbRelease[] =
"CHROMEOS_RELEASE_NAME=Chrome OS\n"
"CHROMEOS_RELEASE_VERSION=1.2.3.4\n"
"CHROMEOS_RELEASE_TRACK=stable-channel\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease, Time());
EXPECT_TRUE(SysInfo::IsRunningOnChromeOS());
EXPECT_DEATH_IF_SUPPORTED({ SysInfo::CrashIfChromeOSNonTestImage(); }, "");
}
TEST_F(SysInfoTest, NoCrashOnTestImage) {
const char kLsbRelease[] =
"CHROMEOS_RELEASE_NAME=Chrome OS\n"
"CHROMEOS_RELEASE_VERSION=1.2.3.4\n"
"CHROMEOS_RELEASE_TRACK=testimage-channel\n";
test::ScopedChromeOSVersionInfo version(kLsbRelease, Time());
EXPECT_TRUE(SysInfo::IsRunningOnChromeOS());
SysInfo::CrashIfChromeOSNonTestImage();
}
TEST_F(SysInfoTest, NoCrashOnLinuxBuild) {
test::ScopedChromeOSVersionInfo version("", Time());
EXPECT_FALSE(SysInfo::IsRunningOnChromeOS());
SysInfo::CrashIfChromeOSNonTestImage();
}
TEST_F(SysInfoTest, ScopedRunningOnChromeOS) {
bool was_running = SysInfo::IsRunningOnChromeOS();
{
test::ScopedRunningOnChromeOS running_on_chromeos;
EXPECT_TRUE(SysInfo::IsRunningOnChromeOS());
}
EXPECT_EQ(was_running, SysInfo::IsRunningOnChromeOS());
}
#endif
}