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-exception
6//
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 _WIN32
17#include <windows.h>
18#endif
19
20namespace {
21
22using namespace llvm;
23using namespace sys;
24
25TEST(ProcessTest, GetProcessIdTest) {
26 const Process::Pid pid = Process::getProcessId();
27
28#ifdef _WIN32
29 EXPECT_EQ((DWORD)pid, ::GetCurrentProcessId());
30#else
31 EXPECT_EQ(pid, ::getpid());
32#endif
33}
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_VER
43#define setenv(name, var, ignore) _putenv_s(name, var)
44#endif
45
46#if HAVE_SETENV || _MSC_VER
47TEST(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#endif
60
61#ifdef _WIN32
62
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#endif
77
78class PageSizeTest : public testing::Test {
79 Triple Host;
80
81protected:
82 PageSizeTest() : Host(Triple::normalize(Str: 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 64k
97 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 namespace
114

source code of llvm/unittests/Support/ProcessTest.cpp