#import "config.h"
#import "CPUTime.h"
#import "MachSendRight.h"
#import <mach/mach.h>
#import <mach/mach_time.h>
#import <mach/task.h>
#import <mach/task_info.h>
#import <mach/thread_info.h>
#import <sys/time.h>
namespace WTF {
static const int64_t microsecondsPerSecond = 1000000;
static int64_t timeValueToMicroseconds(const time_value_t& value)
{
int64_t result = value.seconds;
result *= microsecondsPerSecond;
result += value.microseconds;
return result;
}
std::optional<CPUTime> CPUTime::get()
{
task_thread_times_info threadInfoData;
mach_msg_type_number_t threadInfoCount = TASK_THREAD_TIMES_INFO_COUNT;
kern_return_t result = task_info(mach_task_self(), TASK_THREAD_TIMES_INFO, reinterpret_cast<task_info_t>(&threadInfoData), &threadInfoCount);
if (result != KERN_SUCCESS)
return std::nullopt;
int64_t userTime = timeValueToMicroseconds(threadInfoData.user_time);
int64_t systemTime = timeValueToMicroseconds(threadInfoData.system_time);
task_basic_info taskInfoData;
mach_msg_type_number_t taskInfoCount = TASK_BASIC_INFO_COUNT;
result = task_info(mach_task_self(), TASK_BASIC_INFO, reinterpret_cast<task_info_t>(&taskInfoData), &taskInfoCount);
if (result != KERN_SUCCESS)
return std::nullopt;
userTime += timeValueToMicroseconds(taskInfoData.user_time);
systemTime += timeValueToMicroseconds(taskInfoData.system_time);
return CPUTime { MonotonicTime::now(), Seconds::fromMicroseconds(userTime), Seconds::fromMicroseconds(systemTime) };
}
Seconds CPUTime::forCurrentThread()
{
mach_msg_type_number_t infoCount = THREAD_BASIC_INFO_COUNT;
thread_basic_info_data_t info;
auto threadPort = MachSendRight::adopt(mach_thread_self());
auto ret = thread_info(threadPort.sendRight(), THREAD_BASIC_INFO, reinterpret_cast<thread_info_t>(&info), &infoCount);
RELEASE_ASSERT(ret == KERN_SUCCESS);
return Seconds(info.user_time.seconds + info.system_time.seconds) + Seconds::fromMicroseconds(info.user_time.microseconds + info.system_time.microseconds);
}
}