xref: /freebsd/contrib/xz/src/xz/mytime.c (revision e0c4386e)
1 ///////////////////////////////////////////////////////////////////////////////
2 //
3 /// \file       mytime.c
4 /// \brief      Time handling functions
5 //
6 //  Author:     Lasse Collin
7 //
8 //  This file has been put into the public domain.
9 //  You can do whatever you want with this file.
10 //
11 ///////////////////////////////////////////////////////////////////////////////
12 
13 #include "private.h"
14 
15 #if defined(HAVE_CLOCK_GETTIME) && defined(HAVE_CLOCK_MONOTONIC)
16 #	include <time.h>
17 #else
18 #	include <sys/time.h>
19 #endif
20 
21 uint64_t opt_flush_timeout = 0;
22 
23 static uint64_t start_time;
24 static uint64_t next_flush;
25 
26 
27 /// \brief      Get the current time as milliseconds
28 ///
29 /// It's relative to some point but not necessarily to the UNIX Epoch.
30 static uint64_t
31 mytime_now(void)
32 {
33 #if defined(HAVE_CLOCK_GETTIME) && defined(HAVE_CLOCK_MONOTONIC)
34 	// If CLOCK_MONOTONIC was available at compile time but for some
35 	// reason isn't at runtime, fallback to CLOCK_REALTIME which
36 	// according to POSIX is mandatory for all implementations.
37 	static clockid_t clk_id = CLOCK_MONOTONIC;
38 	struct timespec tv;
39 	while (clock_gettime(clk_id, &tv))
40 		clk_id = CLOCK_REALTIME;
41 
42 	return (uint64_t)tv.tv_sec * 1000 + (uint64_t)(tv.tv_nsec / 1000000);
43 #else
44 	struct timeval tv;
45 	gettimeofday(&tv, NULL);
46 	return (uint64_t)tv.tv_sec * 1000 + (uint64_t)(tv.tv_usec / 1000);
47 #endif
48 }
49 
50 
51 extern void
52 mytime_set_start_time(void)
53 {
54 	start_time = mytime_now();
55 	return;
56 }
57 
58 
59 extern uint64_t
60 mytime_get_elapsed(void)
61 {
62 	return mytime_now() - start_time;
63 }
64 
65 
66 extern void
67 mytime_set_flush_time(void)
68 {
69 	next_flush = mytime_now() + opt_flush_timeout;
70 	return;
71 }
72 
73 
74 extern int
75 mytime_get_flush_timeout(void)
76 {
77 	if (opt_flush_timeout == 0 || opt_mode != MODE_COMPRESS)
78 		return -1;
79 
80 	const uint64_t now = mytime_now();
81 	if (now >= next_flush)
82 		return 0;
83 
84 	const uint64_t remaining = next_flush - now;
85 	return remaining > INT_MAX ? INT_MAX : (int)remaining;
86 }
87