1*43a90889SApple OSS Distributions #include <assert.h>
2*43a90889SApple OSS Distributions #include <errno.h>
3*43a90889SApple OSS Distributions #include <stdarg.h>
4*43a90889SApple OSS Distributions #include <stdio.h>
5*43a90889SApple OSS Distributions #include <stdlib.h>
6*43a90889SApple OSS Distributions #include <string.h>
7*43a90889SApple OSS Distributions #include <sys/sysctl.h>
8*43a90889SApple OSS Distributions
9*43a90889SApple OSS Distributions #include <mach/mach.h>
10*43a90889SApple OSS Distributions #include <mach/mach_vm.h>
11*43a90889SApple OSS Distributions
12*43a90889SApple OSS Distributions #include <sys/mman.h>
13*43a90889SApple OSS Distributions
14*43a90889SApple OSS Distributions #include "benchmark/helpers.h"
15*43a90889SApple OSS Distributions
16*43a90889SApple OSS Distributions #define K_CTIME_BUFFER_LEN 26
17*43a90889SApple OSS Distributions void
benchmark_log(bool verbose,const char * restrict fmt,...)18*43a90889SApple OSS Distributions benchmark_log(bool verbose, const char *restrict fmt, ...)
19*43a90889SApple OSS Distributions {
20*43a90889SApple OSS Distributions time_t now;
21*43a90889SApple OSS Distributions char time_buffer[K_CTIME_BUFFER_LEN];
22*43a90889SApple OSS Distributions struct tm local_time;
23*43a90889SApple OSS Distributions va_list args;
24*43a90889SApple OSS Distributions if (verbose) {
25*43a90889SApple OSS Distributions strncpy(time_buffer, "UNKNOWN", K_CTIME_BUFFER_LEN);
26*43a90889SApple OSS Distributions
27*43a90889SApple OSS Distributions now = time(NULL);
28*43a90889SApple OSS Distributions if (now != -1) {
29*43a90889SApple OSS Distributions struct tm* ret = localtime_r(&now, &local_time);
30*43a90889SApple OSS Distributions if (ret == &local_time) {
31*43a90889SApple OSS Distributions snprintf(time_buffer, K_CTIME_BUFFER_LEN,
32*43a90889SApple OSS Distributions "%.2d/%.2d/%.2d %.2d:%.2d:%.2d",
33*43a90889SApple OSS Distributions local_time.tm_mon + 1, local_time.tm_mday,
34*43a90889SApple OSS Distributions local_time.tm_year + 1900,
35*43a90889SApple OSS Distributions local_time.tm_hour, local_time.tm_min,
36*43a90889SApple OSS Distributions local_time.tm_sec);
37*43a90889SApple OSS Distributions }
38*43a90889SApple OSS Distributions }
39*43a90889SApple OSS Distributions
40*43a90889SApple OSS Distributions printf("%s: ", time_buffer);
41*43a90889SApple OSS Distributions va_start(args, fmt);
42*43a90889SApple OSS Distributions vprintf(fmt, args);
43*43a90889SApple OSS Distributions fflush(stdout);
44*43a90889SApple OSS Distributions }
45*43a90889SApple OSS Distributions }
46*43a90889SApple OSS Distributions
47*43a90889SApple OSS Distributions uint64_t
timespec_difference_us(const struct timespec * a,const struct timespec * b)48*43a90889SApple OSS Distributions timespec_difference_us(const struct timespec* a, const struct timespec* b)
49*43a90889SApple OSS Distributions {
50*43a90889SApple OSS Distributions assert(a->tv_sec >= b->tv_sec || a->tv_nsec >= b->tv_nsec);
51*43a90889SApple OSS Distributions long seconds_elapsed = a->tv_sec - b->tv_sec;
52*43a90889SApple OSS Distributions uint64_t nsec_elapsed;
53*43a90889SApple OSS Distributions if (b->tv_nsec > a->tv_nsec) {
54*43a90889SApple OSS Distributions seconds_elapsed--;
55*43a90889SApple OSS Distributions nsec_elapsed = kNumNanosecondsInSecond - (uint64_t) (b->tv_nsec - a->tv_nsec);
56*43a90889SApple OSS Distributions } else {
57*43a90889SApple OSS Distributions nsec_elapsed = (uint64_t) (a->tv_nsec - b->tv_nsec);
58*43a90889SApple OSS Distributions }
59*43a90889SApple OSS Distributions return (uint64_t) seconds_elapsed * kNumMicrosecondsInSecond + nsec_elapsed / kNumNanosecondsInMicrosecond;
60*43a90889SApple OSS Distributions }
61*43a90889SApple OSS Distributions
62*43a90889SApple OSS Distributions unsigned char *
map_buffer(size_t memsize,int flags)63*43a90889SApple OSS Distributions map_buffer(size_t memsize, int flags)
64*43a90889SApple OSS Distributions {
65*43a90889SApple OSS Distributions #if USE_MMAP
66*43a90889SApple OSS Distributions int fd = -1;
67*43a90889SApple OSS Distributions unsigned char* addr = (unsigned char *)mmap(NULL, memsize, PROT_READ | PROT_WRITE, MAP_ANON | MAP_PRIVATE,
68*43a90889SApple OSS Distributions fd, 0);
69*43a90889SApple OSS Distributions if ((void*) addr == MAP_FAILED) {
70*43a90889SApple OSS Distributions fprintf(stderr, "Unable to mmap a memory object: %s\n", strerror(errno));
71*43a90889SApple OSS Distributions exit(2);
72*43a90889SApple OSS Distributions }
73*43a90889SApple OSS Distributions return addr;
74*43a90889SApple OSS Distributions #else
75*43a90889SApple OSS Distributions vm_address_t address;
76*43a90889SApple OSS Distributions kern_return_t kr = vm_allocate(mach_task_self(), &address, memsize, flags | VM_FLAGS_ANYWHERE);
77*43a90889SApple OSS Distributions if (kr != KERN_SUCCESS) {
78*43a90889SApple OSS Distributions fprintf(stderr, "Unable to vm_allocate: %d\n", kr);
79*43a90889SApple OSS Distributions exit(2);
80*43a90889SApple OSS Distributions }
81*43a90889SApple OSS Distributions return (unsigned char*)address;
82*43a90889SApple OSS Distributions #endif
83*43a90889SApple OSS Distributions }
84*43a90889SApple OSS Distributions
85*43a90889SApple OSS Distributions unsigned int
get_ncpu(void)86*43a90889SApple OSS Distributions get_ncpu(void)
87*43a90889SApple OSS Distributions {
88*43a90889SApple OSS Distributions int ncpu;
89*43a90889SApple OSS Distributions size_t length = sizeof(ncpu);
90*43a90889SApple OSS Distributions
91*43a90889SApple OSS Distributions int ret = sysctlbyname("hw.ncpu", &ncpu, &length, NULL, 0);
92*43a90889SApple OSS Distributions if (ret == -1 || ncpu < 0) {
93*43a90889SApple OSS Distributions fprintf(stderr, "failed to query hw.ncpu");
94*43a90889SApple OSS Distributions exit(1);
95*43a90889SApple OSS Distributions }
96*43a90889SApple OSS Distributions return (unsigned int) ncpu;
97*43a90889SApple OSS Distributions }
98