Fix clock_gettime() for Apple OSX
[proxmark3-svn] / client / util_posix.c
1 //-----------------------------------------------------------------------------
2 // Copyright (C) 2010 iZsh <izsh at fail0verflow.com>
3 //
4 // This code is licensed to you under the terms of the GNU GPL, version 2 or,
5 // at your option, any later version. See the LICENSE.txt file for the text of
6 // the license.
7 //-----------------------------------------------------------------------------
8 // utilities requiring Posix library functions
9 //-----------------------------------------------------------------------------
10
11 #if !defined(_WIN32)
12 #define _POSIX_C_SOURCE 199309L // need nanosleep()
13 #else
14 #include <windows.h>
15 #endif
16
17 #include "util_posix.h"
18 #include <stdint.h>
19 #include <time.h>
20
21
22 // Timer functions
23 #if !defined (_WIN32)
24 #include <errno.h>
25
26 static void nsleep(uint64_t n) {
27 struct timespec timeout;
28 timeout.tv_sec = n/1000000000;
29 timeout.tv_nsec = n%1000000000;
30 while (nanosleep(&timeout, &timeout) && errno == EINTR);
31 }
32
33 void msleep(uint32_t n) {
34 nsleep(1000000 * (uint64_t)n);
35 }
36 #endif // _WIN32
37
38 #ifdef __APPLE__
39
40 #define CLOCK_MONOTONIC (1)
41 #define CLOCK_REALTIME (2)
42
43 #include <sys/time.h>
44 #include <mach/clock.h>
45 #include <mach/mach.h>
46 #include <mach/mach_time.h>
47
48 /* clock_gettime is not implemented on OSX prior to 10.12 */
49 int _civet_clock_gettime(int clk_id, struct timespec *t);
50
51 int _civet_clock_gettime(int clk_id, struct timespec *t)
52 {
53 memset(t, 0, sizeof(*t));
54 if (clk_id == CLOCK_REALTIME) {
55 struct timeval now;
56 int rv = gettimeofday(&now, NULL);
57 if (rv) {
58 return rv;
59 }
60 t->tv_sec = now.tv_sec;
61 t->tv_nsec = now.tv_usec * 1000;
62 return 0;
63
64 } else if (clk_id == CLOCK_MONOTONIC) {
65 static uint64_t clock_start_time = 0;
66 static mach_timebase_info_data_t timebase_info = {0, 0};
67
68 uint64_t now = mach_absolute_time();
69
70 if (clock_start_time == 0) {
71 mach_timebase_info(&timebase_info);
72 clock_start_time = now;
73 }
74
75 now = (uint64_t)((double)(now - clock_start_time)
76 * (double)timebase_info.numer
77 / (double)timebase_info.denom);
78
79 t->tv_sec = now / 1000000000;
80 t->tv_nsec = now % 1000000000;
81 return 0;
82 }
83 return -1; // EINVAL - Clock ID is unknown
84 }
85
86 /* if clock_gettime is declared, then __CLOCK_AVAILABILITY will be defined */
87 #ifdef __CLOCK_AVAILABILITY
88 /* If we compiled with Mac OSX 10.12 or later, then clock_gettime will be declared
89 * but it may be NULL at runtime. So we need to check before using it. */
90 int _civet_safe_clock_gettime(int clk_id, struct timespec *t);
91
92 int _civet_safe_clock_gettime(int clk_id, struct timespec *t) {
93 if( clock_gettime ) {
94 return clock_gettime(clk_id, t);
95 }
96 return _civet_clock_gettime(clk_id, t);
97 }
98 #define clock_gettime _civet_safe_clock_gettime
99 #else
100 #define clock_gettime _civet_clock_gettime
101 #endif
102
103 #endif
104
105
106 // a milliseconds timer for performance measurement
107 uint64_t msclock() {
108 #if defined(_WIN32)
109 #include <sys/types.h>
110
111 // WORKAROUND FOR MinGW (some versions - use if normal code does not compile)
112 // It has no _ftime_s and needs explicit inclusion of timeb.h
113 #include <sys/timeb.h>
114 struct _timeb t;
115 _ftime(&t);
116 return 1000 * t.time + t.millitm;
117
118 // NORMAL CODE (use _ftime_s)
119 //struct _timeb t;
120 //if (_ftime_s(&t)) {
121 // return 0;
122 //} else {
123 // return 1000 * t.time + t.millitm;
124 //}
125 #else
126 struct timespec t;
127 clock_gettime(CLOCK_MONOTONIC, &t);
128 return (t.tv_sec * 1000 + t.tv_nsec / 1000000);
129 #endif
130 }
131
Impressum, Datenschutz