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