/Users/brunogarcia/projects/bitcoin-core-dev/src/common/system.cpp
Line | Count | Source |
1 | | // Copyright (c) 2009-2010 Satoshi Nakamoto |
2 | | // Copyright (c) 2009-present The Bitcoin Core developers |
3 | | // Distributed under the MIT software license, see the accompanying |
4 | | // file COPYING or http://www.opensource.org/licenses/mit-license.php. |
5 | | |
6 | | #include <bitcoin-build-config.h> // IWYU pragma: keep |
7 | | |
8 | | #include <common/system.h> |
9 | | |
10 | | #include <logging.h> |
11 | | #include <util/string.h> |
12 | | #include <util/time.h> |
13 | | |
14 | | #ifdef WIN32 |
15 | | #include <cassert> |
16 | | #include <codecvt> |
17 | | #include <compat/compat.h> |
18 | | #include <windows.h> |
19 | | #else |
20 | | #include <sys/stat.h> |
21 | | #include <unistd.h> |
22 | | #endif |
23 | | |
24 | | #ifdef HAVE_MALLOPT_ARENA_MAX |
25 | | #include <malloc.h> |
26 | | #endif |
27 | | |
28 | | #include <algorithm> |
29 | | #include <cstddef> |
30 | | #include <cstdint> |
31 | | #include <cstdlib> |
32 | | #include <locale> |
33 | | #include <optional> |
34 | | #include <stdexcept> |
35 | | #include <string> |
36 | | #include <thread> |
37 | | |
38 | | using util::ReplaceAll; |
39 | | |
40 | | // Application startup time (used for uptime calculation) |
41 | | const int64_t nStartupTime = GetTime(); |
42 | | |
43 | | #ifndef WIN32 |
44 | | std::string ShellEscape(const std::string& arg) |
45 | 0 | { |
46 | 0 | std::string escaped = arg; |
47 | 0 | ReplaceAll(escaped, "'", "'\"'\"'"); |
48 | 0 | return "'" + escaped + "'"; |
49 | 0 | } |
50 | | #endif |
51 | | |
52 | | #if HAVE_SYSTEM |
53 | | void runCommand(const std::string& strCommand) |
54 | 0 | { |
55 | 0 | if (strCommand.empty()) return; |
56 | 0 | #ifndef WIN32 |
57 | 0 | int nErr = ::system(strCommand.c_str()); |
58 | | #else |
59 | | int nErr = ::_wsystem(std::wstring_convert<std::codecvt_utf8_utf16<wchar_t>,wchar_t>().from_bytes(strCommand).c_str()); |
60 | | #endif |
61 | 0 | if (nErr) |
62 | 0 | LogPrintf("runCommand error: system(%s) returned %d\n", strCommand, nErr);Line | Count | Source | 373 | 0 | #define LogPrintf(...) LogInfo(__VA_ARGS__) Line | Count | Source | 368 | 0 | #define LogInfo(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Info, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 362 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(std::source_location::current(), category, level, should_ratelimit, __VA_ARGS__) |
|
|
|
63 | 0 | } |
64 | | #endif |
65 | | |
66 | | void SetupEnvironment() |
67 | 0 | { |
68 | | #ifdef HAVE_MALLOPT_ARENA_MAX |
69 | | // glibc-specific: On 32-bit systems set the number of arenas to 1. |
70 | | // By default, since glibc 2.10, the C library will create up to two heap |
71 | | // arenas per core. This is known to cause excessive virtual address space |
72 | | // usage in our usage. Work around it by setting the maximum number of |
73 | | // arenas to 1. |
74 | | if (sizeof(void*) == 4) { |
75 | | mallopt(M_ARENA_MAX, 1); |
76 | | } |
77 | | #endif |
78 | | // On most POSIX systems (e.g. Linux, but not BSD) the environment's locale |
79 | | // may be invalid, in which case the "C.UTF-8" locale is used as fallback. |
80 | | #if !defined(WIN32) && !defined(__APPLE__) && !defined(__FreeBSD__) && !defined(__OpenBSD__) && !defined(__NetBSD__) |
81 | | try { |
82 | | std::locale(""); // Raises a runtime error if current locale is invalid |
83 | | } catch (const std::runtime_error&) { |
84 | | setenv("LC_ALL", "C.UTF-8", 1); |
85 | | } |
86 | | #elif defined(WIN32) |
87 | | assert(GetACP() == CP_UTF8); |
88 | | // Set the default input/output charset is utf-8 |
89 | | SetConsoleCP(CP_UTF8); |
90 | | SetConsoleOutputCP(CP_UTF8); |
91 | | #endif |
92 | |
|
93 | 0 | #ifndef WIN32 |
94 | 0 | constexpr mode_t private_umask = 0077; |
95 | 0 | umask(private_umask); |
96 | 0 | #endif |
97 | 0 | } |
98 | | |
99 | | bool SetupNetworking() |
100 | 0 | { |
101 | | #ifdef WIN32 |
102 | | // Initialize Windows Sockets |
103 | | WSADATA wsadata; |
104 | | int ret = WSAStartup(MAKEWORD(2,2), &wsadata); |
105 | | if (ret != NO_ERROR || LOBYTE(wsadata.wVersion ) != 2 || HIBYTE(wsadata.wVersion) != 2) |
106 | | return false; |
107 | | #endif |
108 | 0 | return true; |
109 | 0 | } |
110 | | |
111 | | int GetNumCores() |
112 | 0 | { |
113 | 0 | return std::thread::hardware_concurrency(); |
114 | 0 | } |
115 | | |
116 | | std::optional<size_t> GetTotalRAM() |
117 | 0 | { |
118 | 0 | [[maybe_unused]] auto clamp{[](uint64_t v) { return size_t(std::min(v, uint64_t{std::numeric_limits<size_t>::max()})); }}; |
119 | | #ifdef WIN32 |
120 | | if (MEMORYSTATUSEX m{}; (m.dwLength = sizeof(m), GlobalMemoryStatusEx(&m))) return clamp(m.ullTotalPhys); |
121 | | #elif defined(__APPLE__) || \ |
122 | | defined(__FreeBSD__) || \ |
123 | | defined(__NetBSD__) || \ |
124 | | defined(__OpenBSD__) || \ |
125 | | defined(__illumos__) || \ |
126 | | defined(__linux__) |
127 | 0 | if (long p{sysconf(_SC_PHYS_PAGES)}, s{sysconf(_SC_PAGESIZE)}; p > 0 && s > 0) return clamp(1ULL * p * s); |
128 | 0 | #endif |
129 | 0 | return std::nullopt; |
130 | 0 | } |
131 | | |
132 | | // Obtain the application startup time (used for uptime calculation) |
133 | | int64_t GetStartupTime() |
134 | 0 | { |
135 | 0 | return nStartupTime; |
136 | 0 | } |