114 lines · cpp
1//===- unittest/Support/ProcessTest.cpp -----------------------------------===//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-exception6//7//===----------------------------------------------------------------------===//8 9#include "llvm/Support/Process.h"10#include "llvm/Support/Error.h"11#include "llvm/TargetParser/Host.h"12#include "llvm/TargetParser/Triple.h"13#include "gtest/gtest.h"14#include <optional>15 16#ifdef _WIN3217#include <windows.h>18#endif19 20namespace {21 22using namespace llvm;23using namespace sys;24 25TEST(ProcessTest, GetProcessIdTest) {26 const Process::Pid pid = Process::getProcessId();27 28#ifdef _WIN3229 EXPECT_EQ((DWORD)pid, ::GetCurrentProcessId());30#else31 EXPECT_EQ(pid, ::getpid());32#endif33}34 35TEST(ProcessTest, GetRandomNumberTest) {36 const unsigned r1 = Process::GetRandomNumber();37 const unsigned r2 = Process::GetRandomNumber();38 // It should be extremely unlikely that both r1 and r2 are 0.39 EXPECT_NE((r1 | r2), 0u);40}41 42#ifdef _MSC_VER43#define setenv(name, var, ignore) _putenv_s(name, var)44#endif45 46#if HAVE_SETENV || _MSC_VER47TEST(ProcessTest, Basic) {48 setenv("__LLVM_TEST_ENVIRON_VAR__", "abc", true);49 std::optional<std::string> val(Process::GetEnv("__LLVM_TEST_ENVIRON_VAR__"));50 EXPECT_TRUE(val.has_value());51 EXPECT_STREQ("abc", val->c_str());52}53 54TEST(ProcessTest, None) {55 std::optional<std::string> val(56 Process::GetEnv("__LLVM_TEST_ENVIRON_NO_SUCH_VAR__"));57 EXPECT_FALSE(val.has_value());58}59#endif60 61#ifdef _WIN3262 63TEST(ProcessTest, EmptyVal) {64 SetEnvironmentVariableA("__LLVM_TEST_ENVIRON_VAR__", "");65 std::optional<std::string> val(Process::GetEnv("__LLVM_TEST_ENVIRON_VAR__"));66 EXPECT_TRUE(val.has_value());67 EXPECT_STREQ("", val->c_str());68}69 70TEST(ProcessTest, Wchar) {71 SetEnvironmentVariableW(L"__LLVM_TEST_ENVIRON_VAR__", L"abcdefghijklmnopqrs");72 std::optional<std::string> val(Process::GetEnv("__LLVM_TEST_ENVIRON_VAR__"));73 EXPECT_TRUE(val.has_value());74 EXPECT_STREQ("abcdefghijklmnopqrs", val->c_str());75}76#endif77 78class PageSizeTest : public testing::Test {79 Triple Host;80 81protected:82 PageSizeTest() : Host(Triple::normalize(sys::getProcessTriple())) {}83 84 bool isSupported() const {85 // For now just on X86-64 and Aarch64. This can be expanded in the future.86 return (Host.getArch() == Triple::x86_64 ||87 Host.getArch() == Triple::aarch64) &&88 Host.getOS() == Triple::Linux;89 }90 91 bool pageSizeAsExpected(unsigned PageSize) const {92 switch (Host.getArch()) {93 case Triple::x86_64:94 return PageSize == 4096;95 case Triple::aarch64:96 // supported granule sizes are 4k, 16k and 64k97 return PageSize == 4096 || PageSize == 16384 || PageSize == 65536;98 default:99 llvm_unreachable("unexpected arch!");100 }101 }102};103 104TEST_F(PageSizeTest, PageSize) {105 if (!isSupported())106 GTEST_SKIP();107 108 llvm::Expected<unsigned> Result = llvm::sys::Process::getPageSize();109 ASSERT_FALSE(!Result);110 ASSERT_TRUE(pageSizeAsExpected(*Result));111}112 113} // end anonymous namespace114