gettime: make cpuclock calibration more tolerant
[fio.git] / gettime.c
1 /*
2  * Clock functions
3  */
4
5 #include <unistd.h>
6 #include <math.h>
7 #include <sys/time.h>
8 #include <time.h>
9
10 #include "fio.h"
11 #include "smalloc.h"
12
13 #include "hash.h"
14 #include "os/os.h"
15
16 #if defined(ARCH_HAVE_CPU_CLOCK)
17 #ifndef ARCH_CPU_CLOCK_CYCLES_PER_USEC
18 static unsigned long cycles_per_usec;
19 static unsigned long inv_cycles_per_usec;
20 static uint64_t max_cycles_for_mult;
21 #endif
22 #ifdef ARCH_CPU_CLOCK_WRAPS
23 static unsigned long long cycles_start, cycles_wrap;
24 #endif
25 #endif
26 int tsc_reliable = 0;
27
28 struct tv_valid {
29         uint64_t last_cycles;
30         int last_tv_valid;
31         int warned;
32 };
33 #ifdef ARCH_HAVE_CPU_CLOCK
34 #ifdef CONFIG_TLS_THREAD
35 static __thread struct tv_valid static_tv_valid;
36 #else
37 static pthread_key_t tv_tls_key;
38 #endif
39 #endif
40
41 enum fio_cs fio_clock_source = FIO_PREFERRED_CLOCK_SOURCE;
42 int fio_clock_source_set = 0;
43 static enum fio_cs fio_clock_source_inited = CS_INVAL;
44
45 #ifdef FIO_DEBUG_TIME
46
47 #define HASH_BITS       8
48 #define HASH_SIZE       (1 << HASH_BITS)
49
50 static struct flist_head hash[HASH_SIZE];
51 static int gtod_inited;
52
53 struct gtod_log {
54         struct flist_head list;
55         void *caller;
56         unsigned long calls;
57 };
58
59 static struct gtod_log *find_hash(void *caller)
60 {
61         unsigned long h = hash_ptr(caller, HASH_BITS);
62         struct flist_head *entry;
63
64         flist_for_each(entry, &hash[h]) {
65                 struct gtod_log *log = flist_entry(entry, struct gtod_log,
66                                                                         list);
67
68                 if (log->caller == caller)
69                         return log;
70         }
71
72         return NULL;
73 }
74
75 static void inc_caller(void *caller)
76 {
77         struct gtod_log *log = find_hash(caller);
78
79         if (!log) {
80                 unsigned long h;
81
82                 log = malloc(sizeof(*log));
83                 INIT_FLIST_HEAD(&log->list);
84                 log->caller = caller;
85                 log->calls = 0;
86
87                 h = hash_ptr(caller, HASH_BITS);
88                 flist_add_tail(&log->list, &hash[h]);
89         }
90
91         log->calls++;
92 }
93
94 static void gtod_log_caller(void *caller)
95 {
96         if (gtod_inited)
97                 inc_caller(caller);
98 }
99
100 static void fio_exit fio_dump_gtod(void)
101 {
102         unsigned long total_calls = 0;
103         int i;
104
105         for (i = 0; i < HASH_SIZE; i++) {
106                 struct flist_head *entry;
107                 struct gtod_log *log;
108
109                 flist_for_each(entry, &hash[i]) {
110                         log = flist_entry(entry, struct gtod_log, list);
111
112                         printf("function %p, calls %lu\n", log->caller,
113                                                                 log->calls);
114                         total_calls += log->calls;
115                 }
116         }
117
118         printf("Total %lu gettimeofday\n", total_calls);
119 }
120
121 static void fio_init gtod_init(void)
122 {
123         int i;
124
125         for (i = 0; i < HASH_SIZE; i++)
126                 INIT_FLIST_HEAD(&hash[i]);
127
128         gtod_inited = 1;
129 }
130
131 #endif /* FIO_DEBUG_TIME */
132
133 #ifdef CONFIG_CLOCK_GETTIME
134 static int fill_clock_gettime(struct timespec *ts)
135 {
136 #if defined(CONFIG_CLOCK_MONOTONIC_RAW)
137         return clock_gettime(CLOCK_MONOTONIC_RAW, ts);
138 #elif defined(CONFIG_CLOCK_MONOTONIC)
139         return clock_gettime(CLOCK_MONOTONIC, ts);
140 #else
141         return clock_gettime(CLOCK_REALTIME, ts);
142 #endif
143 }
144 #endif
145
146 static void __fio_gettime(struct timeval *tp)
147 {
148         switch (fio_clock_source) {
149 #ifdef CONFIG_GETTIMEOFDAY
150         case CS_GTOD:
151                 gettimeofday(tp, NULL);
152                 break;
153 #endif
154 #ifdef CONFIG_CLOCK_GETTIME
155         case CS_CGETTIME: {
156                 struct timespec ts;
157
158                 if (fill_clock_gettime(&ts) < 0) {
159                         log_err("fio: clock_gettime fails\n");
160                         assert(0);
161                 }
162
163                 tp->tv_sec = ts.tv_sec;
164                 tp->tv_usec = ts.tv_nsec / 1000;
165                 break;
166                 }
167 #endif
168 #ifdef ARCH_HAVE_CPU_CLOCK
169         case CS_CPUCLOCK: {
170                 uint64_t usecs, t;
171                 struct tv_valid *tv;
172
173 #ifdef CONFIG_TLS_THREAD
174                 tv = &static_tv_valid;
175 #else
176                 tv = pthread_getspecific(tv_tls_key);
177 #endif
178
179                 t = get_cpu_clock();
180 #ifdef ARCH_CPU_CLOCK_WRAPS
181                 if (t < cycles_start && !cycles_wrap)
182                         cycles_wrap = 1;
183                 else if (cycles_wrap && t >= cycles_start && !tv->warned) {
184                         log_err("fio: double CPU clock wrap\n");
185                         tv->warned = 1;
186                 }
187
188                 t -= cycles_start;
189 #endif
190                 tv->last_cycles = t;
191                 tv->last_tv_valid = 1;
192 #ifdef ARCH_CPU_CLOCK_CYCLES_PER_USEC
193                 usecs = t / ARCH_CPU_CLOCK_CYCLES_PER_USEC;
194 #else
195                 if (t < max_cycles_for_mult)
196                         usecs = (t * inv_cycles_per_usec) / 16777216UL;
197                 else
198                         usecs = t / cycles_per_usec;
199 #endif
200                 tp->tv_sec = usecs / 1000000;
201                 tp->tv_usec = usecs % 1000000;
202                 break;
203                 }
204 #endif
205         default:
206                 log_err("fio: invalid clock source %d\n", fio_clock_source);
207                 break;
208         }
209 }
210
211 #ifdef FIO_DEBUG_TIME
212 void fio_gettime(struct timeval *tp, void *caller)
213 #else
214 void fio_gettime(struct timeval *tp, void fio_unused *caller)
215 #endif
216 {
217 #ifdef FIO_DEBUG_TIME
218         if (!caller)
219                 caller = __builtin_return_address(0);
220
221         gtod_log_caller(caller);
222 #endif
223         if (fio_unlikely(fio_gettime_offload(tp)))
224                 return;
225
226         __fio_gettime(tp);
227 }
228
229 #if defined(ARCH_HAVE_CPU_CLOCK) && !defined(ARCH_CPU_CLOCK_CYCLES_PER_USEC)
230 static unsigned long get_cycles_per_usec(void)
231 {
232         struct timeval s, e;
233         uint64_t c_s, c_e;
234         enum fio_cs old_cs = fio_clock_source;
235         uint64_t elapsed;
236
237 #ifdef CONFIG_CLOCK_GETTIME
238         fio_clock_source = CS_CGETTIME;
239 #else
240         fio_clock_source = CS_GTOD;
241 #endif
242         __fio_gettime(&s);
243
244         c_s = get_cpu_clock();
245         do {
246                 __fio_gettime(&e);
247
248                 elapsed = utime_since(&s, &e);
249                 if (elapsed >= 1280) {
250                         c_e = get_cpu_clock();
251                         break;
252                 }
253         } while (1);
254
255         fio_clock_source = old_cs;
256         return ((c_e - c_s) * 10) / elapsed;
257 }
258
259 #define NR_TIME_ITERS   50
260
261 static int calibrate_cpu_clock(void)
262 {
263         double delta, mean, S;
264         uint64_t minc, maxc, avg, cycles[NR_TIME_ITERS];
265         int i, samples;
266
267         cycles[0] = get_cycles_per_usec();
268         S = delta = mean = 0.0;
269         for (i = 0; i < NR_TIME_ITERS; i++) {
270                 cycles[i] = get_cycles_per_usec();
271                 delta = cycles[i] - mean;
272                 if (delta) {
273                         mean += delta / (i + 1.0);
274                         S += delta * (cycles[i] - mean);
275                 }
276         }
277
278         /*
279          * The most common platform clock breakage is returning zero
280          * indefinitely. Check for that and return failure.
281          */
282         if (!cycles[0] && !cycles[NR_TIME_ITERS - 1])
283                 return 1;
284
285         S = sqrt(S / (NR_TIME_ITERS - 1.0));
286
287         minc = -1ULL;
288         maxc = samples = avg = 0;
289         for (i = 0; i < NR_TIME_ITERS; i++) {
290                 double this = cycles[i];
291
292                 minc = min(cycles[i], minc);
293                 maxc = max(cycles[i], maxc);
294
295                 if ((fmax(this, mean) - fmin(this, mean)) > S)
296                         continue;
297                 samples++;
298                 avg += this;
299         }
300
301         S /= (double) NR_TIME_ITERS;
302         mean /= 10.0;
303
304         for (i = 0; i < NR_TIME_ITERS; i++)
305                 dprint(FD_TIME, "cycles[%d]=%llu\n", i,
306                                         (unsigned long long) cycles[i] / 10);
307
308         avg /= samples;
309         avg = (avg + 5) / 10;
310         minc /= 10;
311         maxc /= 10;
312         dprint(FD_TIME, "avg: %llu\n", (unsigned long long) avg);
313         dprint(FD_TIME, "min=%llu, max=%llu, mean=%f, S=%f\n",
314                         (unsigned long long) minc,
315                         (unsigned long long) maxc, mean, S);
316
317         cycles_per_usec = avg;
318         inv_cycles_per_usec = 16777216UL / cycles_per_usec;
319         max_cycles_for_mult = ~0ULL / inv_cycles_per_usec;
320         dprint(FD_TIME, "inv_cycles_per_usec=%lu\n", inv_cycles_per_usec);
321 #ifdef ARCH_CPU_CLOCK_WRAPS
322         cycles_start = get_cpu_clock();
323         dprint(FD_TIME, "cycles_start=%llu\n", cycles_start);
324 #endif
325         return 0;
326 }
327 #else
328 static int calibrate_cpu_clock(void)
329 {
330 #ifdef ARCH_CPU_CLOCK_CYCLES_PER_USEC
331         return 0;
332 #else
333         return 1;
334 #endif
335 }
336 #endif // ARCH_HAVE_CPU_CLOCK
337
338 #ifndef CONFIG_TLS_THREAD
339 void fio_local_clock_init(int is_thread)
340 {
341         struct tv_valid *t;
342
343         t = calloc(1, sizeof(*t));
344         if (pthread_setspecific(tv_tls_key, t)) {
345                 log_err("fio: can't set TLS key\n");
346                 assert(0);
347         }
348 }
349
350 static void kill_tv_tls_key(void *data)
351 {
352         free(data);
353 }
354 #else
355 void fio_local_clock_init(int is_thread)
356 {
357 }
358 #endif
359
360 void fio_clock_init(void)
361 {
362         if (fio_clock_source == fio_clock_source_inited)
363                 return;
364
365 #ifndef CONFIG_TLS_THREAD
366         if (pthread_key_create(&tv_tls_key, kill_tv_tls_key))
367                 log_err("fio: can't create TLS key\n");
368 #endif
369
370         fio_clock_source_inited = fio_clock_source;
371
372         if (calibrate_cpu_clock())
373                 tsc_reliable = 0;
374
375         /*
376          * If the arch sets tsc_reliable != 0, then it must be good enough
377          * to use as THE clock source. For x86 CPUs, this means the TSC
378          * runs at a constant rate and is synced across CPU cores.
379          */
380         if (tsc_reliable) {
381                 if (!fio_clock_source_set && !fio_monotonic_clocktest(0))
382                         fio_clock_source = CS_CPUCLOCK;
383         } else if (fio_clock_source == CS_CPUCLOCK)
384                 log_info("fio: clocksource=cpu may not be reliable\n");
385 }
386
387 uint64_t utime_since(const struct timeval *s, const struct timeval *e)
388 {
389         long sec, usec;
390         uint64_t ret;
391
392         sec = e->tv_sec - s->tv_sec;
393         usec = e->tv_usec - s->tv_usec;
394         if (sec > 0 && usec < 0) {
395                 sec--;
396                 usec += 1000000;
397         }
398
399         /*
400          * time warp bug on some kernels?
401          */
402         if (sec < 0 || (sec == 0 && usec < 0))
403                 return 0;
404
405         ret = sec * 1000000ULL + usec;
406
407         return ret;
408 }
409
410 uint64_t utime_since_now(const struct timeval *s)
411 {
412         struct timeval t;
413
414         fio_gettime(&t, NULL);
415         return utime_since(s, &t);
416 }
417
418 uint64_t mtime_since(const struct timeval *s, const struct timeval *e)
419 {
420         long sec, usec, ret;
421
422         sec = e->tv_sec - s->tv_sec;
423         usec = e->tv_usec - s->tv_usec;
424         if (sec > 0 && usec < 0) {
425                 sec--;
426                 usec += 1000000;
427         }
428
429         if (sec < 0 || (sec == 0 && usec < 0))
430                 return 0;
431
432         sec *= 1000UL;
433         usec /= 1000UL;
434         ret = sec + usec;
435
436         return ret;
437 }
438
439 uint64_t mtime_since_now(const struct timeval *s)
440 {
441         struct timeval t;
442         void *p = __builtin_return_address(0);
443
444         fio_gettime(&t, p);
445         return mtime_since(s, &t);
446 }
447
448 uint64_t time_since_now(const struct timeval *s)
449 {
450         return mtime_since_now(s) / 1000;
451 }
452
453 #if defined(FIO_HAVE_CPU_AFFINITY) && defined(ARCH_HAVE_CPU_CLOCK)  && \
454     defined(CONFIG_SFAA)
455
456 #define CLOCK_ENTRIES_DEBUG     100000
457 #define CLOCK_ENTRIES_TEST      10000
458
459 struct clock_entry {
460         uint32_t seq;
461         uint32_t cpu;
462         uint64_t tsc;
463 };
464
465 struct clock_thread {
466         pthread_t thread;
467         int cpu;
468         int debug;
469         pthread_mutex_t lock;
470         pthread_mutex_t started;
471         unsigned long nr_entries;
472         uint32_t *seq;
473         struct clock_entry *entries;
474 };
475
476 static inline uint32_t atomic32_inc_return(uint32_t *seq)
477 {
478         return 1 + __sync_fetch_and_add(seq, 1);
479 }
480
481 static void *clock_thread_fn(void *data)
482 {
483         struct clock_thread *t = data;
484         struct clock_entry *c;
485         os_cpu_mask_t cpu_mask;
486         uint32_t last_seq;
487         unsigned long long first;
488         int i;
489
490         if (fio_cpuset_init(&cpu_mask)) {
491                 int __err = errno;
492
493                 log_err("clock cpuset init failed: %s\n", strerror(__err));
494                 goto err_out;
495         }
496
497         fio_cpu_set(&cpu_mask, t->cpu);
498
499         if (fio_setaffinity(gettid(), cpu_mask) == -1) {
500                 int __err = errno;
501
502                 log_err("clock setaffinity failed: %s\n", strerror(__err));
503                 goto err;
504         }
505
506         pthread_mutex_lock(&t->lock);
507         pthread_mutex_unlock(&t->started);
508
509         first = get_cpu_clock();
510         last_seq = 0;
511         c = &t->entries[0];
512         for (i = 0; i < t->nr_entries; i++, c++) {
513                 uint32_t seq;
514                 uint64_t tsc;
515
516                 c->cpu = t->cpu;
517                 do {
518                         seq = atomic32_inc_return(t->seq);
519                         if (seq < last_seq)
520                                 break;
521                         tsc = get_cpu_clock();
522                 } while (seq != *t->seq);
523
524                 c->seq = seq;
525                 c->tsc = tsc;
526         }
527
528         if (t->debug) {
529                 unsigned long long clocks;
530
531                 clocks = t->entries[i - 1].tsc - t->entries[0].tsc;
532                 log_info("cs: cpu%3d: %llu clocks seen, first %llu\n", t->cpu,
533                                                         clocks, first);
534         }
535
536         /*
537          * The most common platform clock breakage is returning zero
538          * indefinitely. Check for that and return failure.
539          */
540         if (!t->entries[i - 1].tsc && !t->entries[0].tsc)
541                 goto err;
542
543         fio_cpuset_exit(&cpu_mask);
544         return NULL;
545 err:
546         fio_cpuset_exit(&cpu_mask);
547 err_out:
548         return (void *) 1;
549 }
550
551 static int clock_cmp(const void *p1, const void *p2)
552 {
553         const struct clock_entry *c1 = p1;
554         const struct clock_entry *c2 = p2;
555
556         if (c1->seq == c2->seq)
557                 log_err("cs: bug in atomic sequence!\n");
558
559         return c1->seq - c2->seq;
560 }
561
562 int fio_monotonic_clocktest(int debug)
563 {
564         struct clock_thread *cthreads;
565         unsigned int nr_cpus = cpus_online();
566         struct clock_entry *entries;
567         unsigned long nr_entries, tentries, failed = 0;
568         struct clock_entry *prev, *this;
569         uint32_t seq = 0;
570         unsigned int i;
571
572         if (debug) {
573                 log_info("cs: reliable_tsc: %s\n", tsc_reliable ? "yes" : "no");
574
575 #ifdef FIO_INC_DEBUG
576                 fio_debug |= 1U << FD_TIME;
577 #endif
578                 nr_entries = CLOCK_ENTRIES_DEBUG;
579         } else
580                 nr_entries = CLOCK_ENTRIES_TEST;
581
582         calibrate_cpu_clock();
583
584         if (debug) {
585 #ifdef FIO_INC_DEBUG
586                 fio_debug &= ~(1U << FD_TIME);
587 #endif
588         }
589
590         cthreads = malloc(nr_cpus * sizeof(struct clock_thread));
591         tentries = nr_entries * nr_cpus;
592         entries = malloc(tentries * sizeof(struct clock_entry));
593
594         if (debug)
595                 log_info("cs: Testing %u CPUs\n", nr_cpus);
596
597         for (i = 0; i < nr_cpus; i++) {
598                 struct clock_thread *t = &cthreads[i];
599
600                 t->cpu = i;
601                 t->debug = debug;
602                 t->seq = &seq;
603                 t->nr_entries = nr_entries;
604                 t->entries = &entries[i * nr_entries];
605                 pthread_mutex_init(&t->lock, NULL);
606                 pthread_mutex_init(&t->started, NULL);
607                 pthread_mutex_lock(&t->lock);
608                 if (pthread_create(&t->thread, NULL, clock_thread_fn, t)) {
609                         failed++;
610                         nr_cpus = i;
611                         break;
612                 }
613         }
614
615         for (i = 0; i < nr_cpus; i++) {
616                 struct clock_thread *t = &cthreads[i];
617
618                 pthread_mutex_lock(&t->started);
619         }
620
621         for (i = 0; i < nr_cpus; i++) {
622                 struct clock_thread *t = &cthreads[i];
623
624                 pthread_mutex_unlock(&t->lock);
625         }
626
627         for (i = 0; i < nr_cpus; i++) {
628                 struct clock_thread *t = &cthreads[i];
629                 void *ret;
630
631                 pthread_join(t->thread, &ret);
632                 if (ret)
633                         failed++;
634         }
635         free(cthreads);
636
637         if (failed) {
638                 if (debug)
639                         log_err("Clocksource test: %lu threads failed\n", failed);
640                 goto err;
641         }
642
643         qsort(entries, tentries, sizeof(struct clock_entry), clock_cmp);
644
645         /* silence silly gcc */
646         prev = NULL;
647         for (failed = i = 0; i < tentries; i++) {
648                 this = &entries[i];
649
650                 if (!i) {
651                         prev = this;
652                         continue;
653                 }
654
655                 if (prev->tsc > this->tsc) {
656                         uint64_t diff = prev->tsc - this->tsc;
657
658                         if (!debug) {
659                                 failed++;
660                                 break;
661                         }
662
663                         log_info("cs: CPU clock mismatch (diff=%llu):\n",
664                                                 (unsigned long long) diff);
665                         log_info("\t CPU%3u: TSC=%llu, SEQ=%u\n", prev->cpu, (unsigned long long) prev->tsc, prev->seq);
666                         log_info("\t CPU%3u: TSC=%llu, SEQ=%u\n", this->cpu, (unsigned long long) this->tsc, this->seq);
667                         failed++;
668                 }
669
670                 prev = this;
671         }
672
673         if (debug) {
674                 if (failed)
675                         log_info("cs: Failed: %lu\n", failed);
676                 else
677                         log_info("cs: Pass!\n");
678         }
679 err:
680         free(entries);
681         return !!failed;
682 }
683
684 #else /* defined(FIO_HAVE_CPU_AFFINITY) && defined(ARCH_HAVE_CPU_CLOCK) */
685
686 int fio_monotonic_clocktest(int debug)
687 {
688         if (debug)
689                 log_info("cs: current platform does not support CPU clocks\n");
690         return 1;
691 }
692
693 #endif