2 * Copyright 2017 Facebook, Inc.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include <folly/portability/Time.h>
18 #include <folly/Likely.h>
24 template <typename _Rep, typename _Period>
25 static void duration_to_ts(
26 std::chrono::duration<_Rep, _Period> d,
27 struct timespec* ts) {
29 time_t(std::chrono::duration_cast<std::chrono::seconds>(d).count());
30 ts->tv_nsec = long(std::chrono::duration_cast<std::chrono::nanoseconds>(
31 d % std::chrono::seconds(1))
35 #if !FOLLY_HAVE_CLOCK_GETTIME || FOLLY_FORCE_CLOCK_GETTIME_DEFINITION
38 #include <mach/mach_init.h>
39 #include <mach/mach_port.h>
40 #include <mach/mach_time.h>
41 #include <mach/mach_types.h>
42 #include <mach/task.h>
43 #include <mach/thread_act.h>
44 #include <mach/vm_map.h>
46 static std::chrono::nanoseconds time_value_to_ns(time_value_t t) {
47 return std::chrono::seconds(t.seconds) +
48 std::chrono::microseconds(t.microseconds);
51 static int clock_process_cputime(struct timespec* ts) {
52 // Get CPU usage for live threads.
53 task_thread_times_info thread_times_info;
54 mach_msg_type_number_t thread_times_info_count = TASK_THREAD_TIMES_INFO_COUNT;
55 kern_return_t kern_result = task_info(
57 TASK_THREAD_TIMES_INFO,
58 (thread_info_t)&thread_times_info,
59 &thread_times_info_count);
60 if (UNLIKELY(kern_result != KERN_SUCCESS)) {
64 // Get CPU usage for terminated threads.
65 mach_task_basic_info task_basic_info;
66 mach_msg_type_number_t task_basic_info_count = MACH_TASK_BASIC_INFO_COUNT;
67 kern_result = task_info(
70 (thread_info_t)&task_basic_info,
71 &task_basic_info_count);
72 if (UNLIKELY(kern_result != KERN_SUCCESS)) {
76 auto cputime = time_value_to_ns(thread_times_info.user_time) +
77 time_value_to_ns(thread_times_info.system_time) +
78 time_value_to_ns(task_basic_info.user_time) +
79 time_value_to_ns(task_basic_info.system_time);
80 duration_to_ts(cputime, ts);
84 static int clock_thread_cputime(struct timespec* ts) {
85 mach_msg_type_number_t count = THREAD_BASIC_INFO_COUNT;
86 thread_basic_info_data_t thread_info_data;
87 thread_act_t thread = mach_thread_self();
88 kern_return_t kern_result = thread_info(
89 thread, THREAD_BASIC_INFO, (thread_info_t)&thread_info_data, &count);
90 mach_port_deallocate(mach_task_self(), thread);
91 if (UNLIKELY(kern_result != KERN_SUCCESS)) {
94 auto cputime = time_value_to_ns(thread_info_data.system_time) +
95 time_value_to_ns(thread_info_data.user_time);
96 duration_to_ts(cputime, ts);
100 int clock_gettime(clockid_t clk_id, struct timespec* ts) {
102 case CLOCK_REALTIME: {
103 auto now = std::chrono::system_clock::now().time_since_epoch();
104 duration_to_ts(now, ts);
107 case CLOCK_MONOTONIC: {
108 auto now = std::chrono::steady_clock::now().time_since_epoch();
109 duration_to_ts(now, ts);
112 case CLOCK_PROCESS_CPUTIME_ID:
113 return clock_process_cputime(ts);
114 case CLOCK_THREAD_CPUTIME_ID:
115 return clock_thread_cputime(ts);
122 int clock_getres(clockid_t clk_id, struct timespec* ts) {
123 if (clk_id != CLOCK_MONOTONIC) {
127 static auto info = [] {
128 static mach_timebase_info_data_t info;
129 auto result = (mach_timebase_info(&info) == KERN_SUCCESS) ? &info : nullptr;
135 ts->tv_nsec = info->numer / info->denom;
139 #elif defined(_WIN32)
145 #include <folly/portability/Windows.h>
147 using unsigned_nanos = std::chrono::duration<uint64_t, std::nano>;
149 static unsigned_nanos filetimeToUnsignedNanos(FILETIME ft) {
151 i.HighPart = ft.dwHighDateTime;
152 i.LowPart = ft.dwLowDateTime;
154 // FILETIMEs are in units of 100ns.
155 return unsigned_nanos(i.QuadPart * 100);
158 static LARGE_INTEGER performanceFrequency() {
159 static auto result = [] {
161 // On Windows XP or later, this will never fail.
162 BOOL res = QueryPerformanceFrequency(&freq);
169 extern "C" int clock_getres(clockid_t clock_id, struct timespec* res) {
175 static constexpr size_t kNsPerSec = 1000000000;
177 case CLOCK_REALTIME: {
178 constexpr auto perSec = double(std::chrono::system_clock::period::num) /
179 std::chrono::system_clock::period::den;
180 res->tv_sec = time_t(perSec);
181 res->tv_nsec = time_t(perSec * kNsPerSec);
184 case CLOCK_MONOTONIC: {
185 constexpr auto perSec = double(std::chrono::steady_clock::period::num) /
186 std::chrono::steady_clock::period::den;
187 res->tv_sec = time_t(perSec);
188 res->tv_nsec = time_t(perSec * kNsPerSec);
191 case CLOCK_PROCESS_CPUTIME_ID:
192 case CLOCK_THREAD_CPUTIME_ID: {
193 DWORD adj, timeIncrement;
195 if (!GetSystemTimeAdjustment(&adj, &timeIncrement, &adjDisabled)) {
201 res->tv_nsec = long(timeIncrement * 100);
211 extern "C" int clock_gettime(clockid_t clock_id, struct timespec* tp) {
217 const auto unanosToTimespec = [](timespec* tp, unsigned_nanos t) -> int {
218 static constexpr unsigned_nanos one_sec{std::chrono::seconds(1)};
220 time_t(std::chrono::duration_cast<std::chrono::seconds>(t).count());
221 tp->tv_nsec = long((t % one_sec).count());
225 FILETIME createTime, exitTime, kernalTime, userTime;
227 case CLOCK_REALTIME: {
228 auto now = std::chrono::system_clock::now().time_since_epoch();
229 duration_to_ts(now, tp);
232 case CLOCK_MONOTONIC: {
233 auto now = std::chrono::steady_clock::now().time_since_epoch();
234 duration_to_ts(now, tp);
237 case CLOCK_PROCESS_CPUTIME_ID: {
238 if (!GetProcessTimes(
248 return unanosToTimespec(
250 filetimeToUnsignedNanos(kernalTime) +
251 filetimeToUnsignedNanos(userTime));
253 case CLOCK_THREAD_CPUTIME_ID: {
264 return unanosToTimespec(
266 filetimeToUnsignedNanos(kernalTime) +
267 filetimeToUnsignedNanos(userTime));
276 #error No clock_gettime(3) compatibility wrapper available for this platform.
284 #include <folly/portability/Windows.h>
287 char* asctime_r(const tm* tm, char* buf) {
289 if (asctime_s(tmpBuf, tm)) {
292 // Nothing we can do if the buff is to small :(
293 return strcpy(buf, tmpBuf);
296 char* ctime_r(const time_t* t, char* buf) {
298 if (ctime_s(tmpBuf, 64, t)) {
301 // Nothing we can do if the buff is to small :(
302 return strcpy(buf, tmpBuf);
305 tm* gmtime_r(const time_t* t, tm* res) {
306 if (!gmtime_s(res, t)) {
312 tm* localtime_r(const time_t* t, tm* o) {
313 if (!localtime_s(o, t)) {
319 int nanosleep(const struct timespec* request, struct timespec* remain) {
320 Sleep((DWORD)((request->tv_sec * 1000) + (request->tv_nsec / 1000000)));
321 if (remain != nullptr) {
328 char* strptime(const char* __restrict s,
329 const char* __restrict f,
330 struct tm* __restrict tm) {
331 // Isn't the C++ standard lib nice? std::get_time is defined such that its
332 // format parameters are the exact same as strptime. Of course, we have to
333 // create a string stream first, and imbue it with the current C locale, and
334 // we also have to make sure we return the right things if it fails, or
335 // if it succeeds, but this is still far simpler an implementation than any
336 // of the versions in any of the C standard libraries.
337 std::istringstream input(s);
338 input.imbue(std::locale(setlocale(LC_ALL, nullptr)));
339 input >> std::get_time(tm, f);
343 return const_cast<char*>(s + input.tellg());