1 // Copyright 2017 The Abseil Authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 // The implementation of the absl::Duration class, which is declared in
16 // //absl/time.h. This class behaves like a numeric type; it has no public
17 // methods and is used only through the operators defined here.
18 //
19 // Implementation notes:
20 //
21 // An absl::Duration is represented as
22 //
23 // rep_hi_ : (int64_t) Whole seconds
24 // rep_lo_ : (uint32_t) Fractions of a second
25 //
26 // The seconds value (rep_hi_) may be positive or negative as appropriate.
27 // The fractional seconds (rep_lo_) is always a positive offset from rep_hi_.
28 // The API for Duration guarantees at least nanosecond resolution, which
29 // means rep_lo_ could have a max value of 1B - 1 if it stored nanoseconds.
30 // However, to utilize more of the available 32 bits of space in rep_lo_,
31 // we instead store quarters of a nanosecond in rep_lo_ resulting in a max
32 // value of 4B - 1. This allows us to correctly handle calculations like
33 // 0.5 nanos + 0.5 nanos = 1 nano. The following example shows the actual
34 // Duration rep using quarters of a nanosecond.
35 //
36 // 2.5 sec = {rep_hi_=2, rep_lo_=2000000000} // lo = 4 * 500000000
37 // -2.5 sec = {rep_hi_=-3, rep_lo_=2000000000}
38 //
39 // Infinite durations are represented as Durations with the rep_lo_ field set
40 // to all 1s.
41 //
42 // +InfiniteDuration:
43 // rep_hi_ : kint64max
44 // rep_lo_ : ~0U
45 //
46 // -InfiniteDuration:
47 // rep_hi_ : kint64min
48 // rep_lo_ : ~0U
49 //
50 // Arithmetic overflows/underflows to +/- infinity and saturates.
51
52 #if defined(_MSC_VER)
53 #include <winsock2.h> // for timeval
54 #endif
55
56 #include <algorithm>
57 #include <cassert>
58 #include <cctype>
59 #include <cerrno>
60 #include <cmath>
61 #include <cstdint>
62 #include <cstdlib>
63 #include <cstring>
64 #include <ctime>
65 #include <functional>
66 #include <limits>
67 #include <string>
68
69 #include "absl/base/casts.h"
70 #include "absl/base/macros.h"
71 #include "absl/numeric/int128.h"
72 #include "absl/strings/string_view.h"
73 #include "absl/strings/strip.h"
74 #include "absl/time/time.h"
75
76 namespace absl {
77 ABSL_NAMESPACE_BEGIN
78
79 namespace {
80
81 using time_internal::kTicksPerNanosecond;
82 using time_internal::kTicksPerSecond;
83
84 constexpr int64_t kint64max = std::numeric_limits<int64_t>::max();
85 constexpr int64_t kint64min = std::numeric_limits<int64_t>::min();
86
87 // Can't use std::isinfinite() because it doesn't exist on windows.
IsFinite(double d)88 inline bool IsFinite(double d) {
89 if (std::isnan(d)) return false;
90 return d != std::numeric_limits<double>::infinity() &&
91 d != -std::numeric_limits<double>::infinity();
92 }
93
IsValidDivisor(double d)94 inline bool IsValidDivisor(double d) {
95 if (std::isnan(d)) return false;
96 return d != 0.0;
97 }
98
99 // *sec may be positive or negative. *ticks must be in the range
100 // -kTicksPerSecond < *ticks < kTicksPerSecond. If *ticks is negative it
101 // will be normalized to a positive value by adjusting *sec accordingly.
NormalizeTicks(int64_t * sec,int64_t * ticks)102 inline void NormalizeTicks(int64_t* sec, int64_t* ticks) {
103 if (*ticks < 0) {
104 --*sec;
105 *ticks += kTicksPerSecond;
106 }
107 }
108
109 // Makes a uint128 from the absolute value of the given scalar.
MakeU128(int64_t a)110 inline uint128 MakeU128(int64_t a) {
111 uint128 u128 = 0;
112 if (a < 0) {
113 ++u128;
114 ++a; // Makes it safe to negate 'a'
115 a = -a;
116 }
117 u128 += static_cast<uint64_t>(a);
118 return u128;
119 }
120
121 // Makes a uint128 count of ticks out of the absolute value of the Duration.
MakeU128Ticks(Duration d)122 inline uint128 MakeU128Ticks(Duration d) {
123 int64_t rep_hi = time_internal::GetRepHi(d);
124 uint32_t rep_lo = time_internal::GetRepLo(d);
125 if (rep_hi < 0) {
126 ++rep_hi;
127 rep_hi = -rep_hi;
128 rep_lo = kTicksPerSecond - rep_lo;
129 }
130 uint128 u128 = static_cast<uint64_t>(rep_hi);
131 u128 *= static_cast<uint64_t>(kTicksPerSecond);
132 u128 += rep_lo;
133 return u128;
134 }
135
136 // Breaks a uint128 of ticks into a Duration.
MakeDurationFromU128(uint128 u128,bool is_neg)137 inline Duration MakeDurationFromU128(uint128 u128, bool is_neg) {
138 int64_t rep_hi;
139 uint32_t rep_lo;
140 const uint64_t h64 = Uint128High64(u128);
141 const uint64_t l64 = Uint128Low64(u128);
142 if (h64 == 0) { // fastpath
143 const uint64_t hi = l64 / kTicksPerSecond;
144 rep_hi = static_cast<int64_t>(hi);
145 rep_lo = static_cast<uint32_t>(l64 - hi * kTicksPerSecond);
146 } else {
147 // kMaxRepHi64 is the high 64 bits of (2^63 * kTicksPerSecond).
148 // Any positive tick count whose high 64 bits are >= kMaxRepHi64
149 // is not representable as a Duration. A negative tick count can
150 // have its high 64 bits == kMaxRepHi64 but only when the low 64
151 // bits are all zero, otherwise it is not representable either.
152 const uint64_t kMaxRepHi64 = 0x77359400UL;
153 if (h64 >= kMaxRepHi64) {
154 if (is_neg && h64 == kMaxRepHi64 && l64 == 0) {
155 // Avoid trying to represent -kint64min below.
156 return time_internal::MakeDuration(kint64min);
157 }
158 return is_neg ? -InfiniteDuration() : InfiniteDuration();
159 }
160 const uint128 kTicksPerSecond128 = static_cast<uint64_t>(kTicksPerSecond);
161 const uint128 hi = u128 / kTicksPerSecond128;
162 rep_hi = static_cast<int64_t>(Uint128Low64(hi));
163 rep_lo =
164 static_cast<uint32_t>(Uint128Low64(u128 - hi * kTicksPerSecond128));
165 }
166 if (is_neg) {
167 rep_hi = -rep_hi;
168 if (rep_lo != 0) {
169 --rep_hi;
170 rep_lo = kTicksPerSecond - rep_lo;
171 }
172 }
173 return time_internal::MakeDuration(rep_hi, rep_lo);
174 }
175
176 // Convert between int64_t and uint64_t, preserving representation. This
177 // allows us to do arithmetic in the unsigned domain, where overflow has
178 // well-defined behavior. See operator+=() and operator-=().
179 //
180 // C99 7.20.1.1.1, as referenced by C++11 18.4.1.2, says, "The typedef
181 // name intN_t designates a signed integer type with width N, no padding
182 // bits, and a two's complement representation." So, we can convert to
183 // and from the corresponding uint64_t value using a bit cast.
EncodeTwosComp(int64_t v)184 inline uint64_t EncodeTwosComp(int64_t v) {
185 return absl::bit_cast<uint64_t>(v);
186 }
DecodeTwosComp(uint64_t v)187 inline int64_t DecodeTwosComp(uint64_t v) { return absl::bit_cast<int64_t>(v); }
188
189 // Note: The overflow detection in this function is done using greater/less *or
190 // equal* because kint64max/min is too large to be represented exactly in a
191 // double (which only has 53 bits of precision). In order to avoid assigning to
192 // rep->hi a double value that is too large for an int64_t (and therefore is
193 // undefined), we must consider computations that equal kint64max/min as a
194 // double as overflow cases.
SafeAddRepHi(double a_hi,double b_hi,Duration * d)195 inline bool SafeAddRepHi(double a_hi, double b_hi, Duration* d) {
196 double c = a_hi + b_hi;
197 if (c >= static_cast<double>(kint64max)) {
198 *d = InfiniteDuration();
199 return false;
200 }
201 if (c <= static_cast<double>(kint64min)) {
202 *d = -InfiniteDuration();
203 return false;
204 }
205 *d = time_internal::MakeDuration(c, time_internal::GetRepLo(*d));
206 return true;
207 }
208
209 // A functor that's similar to std::multiplies<T>, except this returns the max
210 // T value instead of overflowing. This is only defined for uint128.
211 template <typename Ignored>
212 struct SafeMultiply {
operator ()absl::__anonaf9cf3ed0111::SafeMultiply213 uint128 operator()(uint128 a, uint128 b) const {
214 // b hi is always zero because it originated as an int64_t.
215 assert(Uint128High64(b) == 0);
216 // Fastpath to avoid the expensive overflow check with division.
217 if (Uint128High64(a) == 0) {
218 return (((Uint128Low64(a) | Uint128Low64(b)) >> 32) == 0)
219 ? static_cast<uint128>(Uint128Low64(a) * Uint128Low64(b))
220 : a * b;
221 }
222 return b == 0 ? b : (a > kuint128max / b) ? kuint128max : a * b;
223 }
224 };
225
226 // Scales (i.e., multiplies or divides, depending on the Operation template)
227 // the Duration d by the int64_t r.
228 template <template <typename> class Operation>
ScaleFixed(Duration d,int64_t r)229 inline Duration ScaleFixed(Duration d, int64_t r) {
230 const uint128 a = MakeU128Ticks(d);
231 const uint128 b = MakeU128(r);
232 const uint128 q = Operation<uint128>()(a, b);
233 const bool is_neg = (time_internal::GetRepHi(d) < 0) != (r < 0);
234 return MakeDurationFromU128(q, is_neg);
235 }
236
237 // Scales (i.e., multiplies or divides, depending on the Operation template)
238 // the Duration d by the double r.
239 template <template <typename> class Operation>
ScaleDouble(Duration d,double r)240 inline Duration ScaleDouble(Duration d, double r) {
241 Operation<double> op;
242 double hi_doub = op(time_internal::GetRepHi(d), r);
243 double lo_doub = op(time_internal::GetRepLo(d), r);
244
245 double hi_int = 0;
246 double hi_frac = std::modf(hi_doub, &hi_int);
247
248 // Moves hi's fractional bits to lo.
249 lo_doub /= kTicksPerSecond;
250 lo_doub += hi_frac;
251
252 double lo_int = 0;
253 double lo_frac = std::modf(lo_doub, &lo_int);
254
255 // Rolls lo into hi if necessary.
256 int64_t lo64 = std::round(lo_frac * kTicksPerSecond);
257
258 Duration ans;
259 if (!SafeAddRepHi(hi_int, lo_int, &ans)) return ans;
260 int64_t hi64 = time_internal::GetRepHi(ans);
261 if (!SafeAddRepHi(hi64, lo64 / kTicksPerSecond, &ans)) return ans;
262 hi64 = time_internal::GetRepHi(ans);
263 lo64 %= kTicksPerSecond;
264 NormalizeTicks(&hi64, &lo64);
265 return time_internal::MakeDuration(hi64, lo64);
266 }
267
268 // Tries to divide num by den as fast as possible by looking for common, easy
269 // cases. If the division was done, the quotient is in *q and the remainder is
270 // in *rem and true will be returned.
IDivFastPath(const Duration num,const Duration den,int64_t * q,Duration * rem)271 inline bool IDivFastPath(const Duration num, const Duration den, int64_t* q,
272 Duration* rem) {
273 // Bail if num or den is an infinity.
274 if (time_internal::IsInfiniteDuration(num) ||
275 time_internal::IsInfiniteDuration(den))
276 return false;
277
278 int64_t num_hi = time_internal::GetRepHi(num);
279 uint32_t num_lo = time_internal::GetRepLo(num);
280 int64_t den_hi = time_internal::GetRepHi(den);
281 uint32_t den_lo = time_internal::GetRepLo(den);
282
283 if (den_hi == 0 && den_lo == kTicksPerNanosecond) {
284 // Dividing by 1ns
285 if (num_hi >= 0 && num_hi < (kint64max - kTicksPerSecond) / 1000000000) {
286 *q = num_hi * 1000000000 + num_lo / kTicksPerNanosecond;
287 *rem = time_internal::MakeDuration(0, num_lo % den_lo);
288 return true;
289 }
290 } else if (den_hi == 0 && den_lo == 100 * kTicksPerNanosecond) {
291 // Dividing by 100ns (common when converting to Universal time)
292 if (num_hi >= 0 && num_hi < (kint64max - kTicksPerSecond) / 10000000) {
293 *q = num_hi * 10000000 + num_lo / (100 * kTicksPerNanosecond);
294 *rem = time_internal::MakeDuration(0, num_lo % den_lo);
295 return true;
296 }
297 } else if (den_hi == 0 && den_lo == 1000 * kTicksPerNanosecond) {
298 // Dividing by 1us
299 if (num_hi >= 0 && num_hi < (kint64max - kTicksPerSecond) / 1000000) {
300 *q = num_hi * 1000000 + num_lo / (1000 * kTicksPerNanosecond);
301 *rem = time_internal::MakeDuration(0, num_lo % den_lo);
302 return true;
303 }
304 } else if (den_hi == 0 && den_lo == 1000000 * kTicksPerNanosecond) {
305 // Dividing by 1ms
306 if (num_hi >= 0 && num_hi < (kint64max - kTicksPerSecond) / 1000) {
307 *q = num_hi * 1000 + num_lo / (1000000 * kTicksPerNanosecond);
308 *rem = time_internal::MakeDuration(0, num_lo % den_lo);
309 return true;
310 }
311 } else if (den_hi > 0 && den_lo == 0) {
312 // Dividing by positive multiple of 1s
313 if (num_hi >= 0) {
314 if (den_hi == 1) {
315 *q = num_hi;
316 *rem = time_internal::MakeDuration(0, num_lo);
317 return true;
318 }
319 *q = num_hi / den_hi;
320 *rem = time_internal::MakeDuration(num_hi % den_hi, num_lo);
321 return true;
322 }
323 if (num_lo != 0) {
324 num_hi += 1;
325 }
326 int64_t quotient = num_hi / den_hi;
327 int64_t rem_sec = num_hi % den_hi;
328 if (rem_sec > 0) {
329 rem_sec -= den_hi;
330 quotient += 1;
331 }
332 if (num_lo != 0) {
333 rem_sec -= 1;
334 }
335 *q = quotient;
336 *rem = time_internal::MakeDuration(rem_sec, num_lo);
337 return true;
338 }
339
340 return false;
341 }
342
343 } // namespace
344
345 namespace time_internal {
346
347 // The 'satq' argument indicates whether the quotient should saturate at the
348 // bounds of int64_t. If it does saturate, the difference will spill over to
349 // the remainder. If it does not saturate, the remainder remain accurate,
350 // but the returned quotient will over/underflow int64_t and should not be used.
IDivDuration(bool satq,const Duration num,const Duration den,Duration * rem)351 int64_t IDivDuration(bool satq, const Duration num, const Duration den,
352 Duration* rem) {
353 int64_t q = 0;
354 if (IDivFastPath(num, den, &q, rem)) {
355 return q;
356 }
357
358 const bool num_neg = num < ZeroDuration();
359 const bool den_neg = den < ZeroDuration();
360 const bool quotient_neg = num_neg != den_neg;
361
362 if (time_internal::IsInfiniteDuration(num) || den == ZeroDuration()) {
363 *rem = num_neg ? -InfiniteDuration() : InfiniteDuration();
364 return quotient_neg ? kint64min : kint64max;
365 }
366 if (time_internal::IsInfiniteDuration(den)) {
367 *rem = num;
368 return 0;
369 }
370
371 const uint128 a = MakeU128Ticks(num);
372 const uint128 b = MakeU128Ticks(den);
373 uint128 quotient128 = a / b;
374
375 if (satq) {
376 // Limits the quotient to the range of int64_t.
377 if (quotient128 > uint128(static_cast<uint64_t>(kint64max))) {
378 quotient128 = quotient_neg ? uint128(static_cast<uint64_t>(kint64min))
379 : uint128(static_cast<uint64_t>(kint64max));
380 }
381 }
382
383 const uint128 remainder128 = a - quotient128 * b;
384 *rem = MakeDurationFromU128(remainder128, num_neg);
385
386 if (!quotient_neg || quotient128 == 0) {
387 return Uint128Low64(quotient128) & kint64max;
388 }
389 // The quotient needs to be negated, but we need to carefully handle
390 // quotient128s with the top bit on.
391 return -static_cast<int64_t>(Uint128Low64(quotient128 - 1) & kint64max) - 1;
392 }
393
394 } // namespace time_internal
395
396 //
397 // Additive operators.
398 //
399
operator +=(Duration rhs)400 Duration& Duration::operator+=(Duration rhs) {
401 if (time_internal::IsInfiniteDuration(*this)) return *this;
402 if (time_internal::IsInfiniteDuration(rhs)) return *this = rhs;
403 const int64_t orig_rep_hi = rep_hi_.Get();
404 rep_hi_ = DecodeTwosComp(EncodeTwosComp(rep_hi_.Get()) +
405 EncodeTwosComp(rhs.rep_hi_.Get()));
406 if (rep_lo_ >= kTicksPerSecond - rhs.rep_lo_) {
407 rep_hi_ = DecodeTwosComp(EncodeTwosComp(rep_hi_.Get()) + 1);
408 rep_lo_ -= kTicksPerSecond;
409 }
410 rep_lo_ += rhs.rep_lo_;
411 if (rhs.rep_hi_.Get() < 0 ? rep_hi_.Get() > orig_rep_hi
412 : rep_hi_.Get() < orig_rep_hi) {
413 return *this =
414 rhs.rep_hi_.Get() < 0 ? -InfiniteDuration() : InfiniteDuration();
415 }
416 return *this;
417 }
418
operator -=(Duration rhs)419 Duration& Duration::operator-=(Duration rhs) {
420 if (time_internal::IsInfiniteDuration(*this)) return *this;
421 if (time_internal::IsInfiniteDuration(rhs)) {
422 return *this = rhs.rep_hi_.Get() >= 0 ? -InfiniteDuration()
423 : InfiniteDuration();
424 }
425 const int64_t orig_rep_hi = rep_hi_.Get();
426 rep_hi_ = DecodeTwosComp(EncodeTwosComp(rep_hi_.Get()) -
427 EncodeTwosComp(rhs.rep_hi_.Get()));
428 if (rep_lo_ < rhs.rep_lo_) {
429 rep_hi_ = DecodeTwosComp(EncodeTwosComp(rep_hi_.Get()) - 1);
430 rep_lo_ += kTicksPerSecond;
431 }
432 rep_lo_ -= rhs.rep_lo_;
433 if (rhs.rep_hi_.Get() < 0 ? rep_hi_.Get() < orig_rep_hi
434 : rep_hi_.Get() > orig_rep_hi) {
435 return *this = rhs.rep_hi_.Get() >= 0 ? -InfiniteDuration()
436 : InfiniteDuration();
437 }
438 return *this;
439 }
440
441 //
442 // Multiplicative operators.
443 //
444
operator *=(int64_t r)445 Duration& Duration::operator*=(int64_t r) {
446 if (time_internal::IsInfiniteDuration(*this)) {
447 const bool is_neg = (r < 0) != (rep_hi_.Get() < 0);
448 return *this = is_neg ? -InfiniteDuration() : InfiniteDuration();
449 }
450 return *this = ScaleFixed<SafeMultiply>(*this, r);
451 }
452
operator *=(double r)453 Duration& Duration::operator*=(double r) {
454 if (time_internal::IsInfiniteDuration(*this) || !IsFinite(r)) {
455 const bool is_neg = std::signbit(r) != (rep_hi_.Get() < 0);
456 return *this = is_neg ? -InfiniteDuration() : InfiniteDuration();
457 }
458 return *this = ScaleDouble<std::multiplies>(*this, r);
459 }
460
operator /=(int64_t r)461 Duration& Duration::operator/=(int64_t r) {
462 if (time_internal::IsInfiniteDuration(*this) || r == 0) {
463 const bool is_neg = (r < 0) != (rep_hi_.Get() < 0);
464 return *this = is_neg ? -InfiniteDuration() : InfiniteDuration();
465 }
466 return *this = ScaleFixed<std::divides>(*this, r);
467 }
468
operator /=(double r)469 Duration& Duration::operator/=(double r) {
470 if (time_internal::IsInfiniteDuration(*this) || !IsValidDivisor(r)) {
471 const bool is_neg = std::signbit(r) != (rep_hi_.Get() < 0);
472 return *this = is_neg ? -InfiniteDuration() : InfiniteDuration();
473 }
474 return *this = ScaleDouble<std::divides>(*this, r);
475 }
476
operator %=(Duration rhs)477 Duration& Duration::operator%=(Duration rhs) {
478 time_internal::IDivDuration(false, *this, rhs, this);
479 return *this;
480 }
481
FDivDuration(Duration num,Duration den)482 double FDivDuration(Duration num, Duration den) {
483 // Arithmetic with infinity is sticky.
484 if (time_internal::IsInfiniteDuration(num) || den == ZeroDuration()) {
485 return (num < ZeroDuration()) == (den < ZeroDuration())
486 ? std::numeric_limits<double>::infinity()
487 : -std::numeric_limits<double>::infinity();
488 }
489 if (time_internal::IsInfiniteDuration(den)) return 0.0;
490
491 double a =
492 static_cast<double>(time_internal::GetRepHi(num)) * kTicksPerSecond +
493 time_internal::GetRepLo(num);
494 double b =
495 static_cast<double>(time_internal::GetRepHi(den)) * kTicksPerSecond +
496 time_internal::GetRepLo(den);
497 return a / b;
498 }
499
500 //
501 // Trunc/Floor/Ceil.
502 //
503
Trunc(Duration d,Duration unit)504 Duration Trunc(Duration d, Duration unit) {
505 return d - (d % unit);
506 }
507
Floor(const Duration d,const Duration unit)508 Duration Floor(const Duration d, const Duration unit) {
509 const absl::Duration td = Trunc(d, unit);
510 return td <= d ? td : td - AbsDuration(unit);
511 }
512
Ceil(const Duration d,const Duration unit)513 Duration Ceil(const Duration d, const Duration unit) {
514 const absl::Duration td = Trunc(d, unit);
515 return td >= d ? td : td + AbsDuration(unit);
516 }
517
518 //
519 // Factory functions.
520 //
521
DurationFromTimespec(timespec ts)522 Duration DurationFromTimespec(timespec ts) {
523 if (static_cast<uint64_t>(ts.tv_nsec) < 1000 * 1000 * 1000) {
524 int64_t ticks = ts.tv_nsec * kTicksPerNanosecond;
525 return time_internal::MakeDuration(ts.tv_sec, ticks);
526 }
527 return Seconds(ts.tv_sec) + Nanoseconds(ts.tv_nsec);
528 }
529
DurationFromTimeval(timeval tv)530 Duration DurationFromTimeval(timeval tv) {
531 if (static_cast<uint64_t>(tv.tv_usec) < 1000 * 1000) {
532 int64_t ticks = tv.tv_usec * 1000 * kTicksPerNanosecond;
533 return time_internal::MakeDuration(tv.tv_sec, ticks);
534 }
535 return Seconds(tv.tv_sec) + Microseconds(tv.tv_usec);
536 }
537
538 //
539 // Conversion to other duration types.
540 //
541
ToInt64Nanoseconds(Duration d)542 int64_t ToInt64Nanoseconds(Duration d) {
543 if (time_internal::GetRepHi(d) >= 0 &&
544 time_internal::GetRepHi(d) >> 33 == 0) {
545 return (time_internal::GetRepHi(d) * 1000 * 1000 * 1000) +
546 (time_internal::GetRepLo(d) / kTicksPerNanosecond);
547 }
548 return d / Nanoseconds(1);
549 }
ToInt64Microseconds(Duration d)550 int64_t ToInt64Microseconds(Duration d) {
551 if (time_internal::GetRepHi(d) >= 0 &&
552 time_internal::GetRepHi(d) >> 43 == 0) {
553 return (time_internal::GetRepHi(d) * 1000 * 1000) +
554 (time_internal::GetRepLo(d) / (kTicksPerNanosecond * 1000));
555 }
556 return d / Microseconds(1);
557 }
ToInt64Milliseconds(Duration d)558 int64_t ToInt64Milliseconds(Duration d) {
559 if (time_internal::GetRepHi(d) >= 0 &&
560 time_internal::GetRepHi(d) >> 53 == 0) {
561 return (time_internal::GetRepHi(d) * 1000) +
562 (time_internal::GetRepLo(d) / (kTicksPerNanosecond * 1000 * 1000));
563 }
564 return d / Milliseconds(1);
565 }
ToInt64Seconds(Duration d)566 int64_t ToInt64Seconds(Duration d) {
567 int64_t hi = time_internal::GetRepHi(d);
568 if (time_internal::IsInfiniteDuration(d)) return hi;
569 if (hi < 0 && time_internal::GetRepLo(d) != 0) ++hi;
570 return hi;
571 }
ToInt64Minutes(Duration d)572 int64_t ToInt64Minutes(Duration d) {
573 int64_t hi = time_internal::GetRepHi(d);
574 if (time_internal::IsInfiniteDuration(d)) return hi;
575 if (hi < 0 && time_internal::GetRepLo(d) != 0) ++hi;
576 return hi / 60;
577 }
ToInt64Hours(Duration d)578 int64_t ToInt64Hours(Duration d) {
579 int64_t hi = time_internal::GetRepHi(d);
580 if (time_internal::IsInfiniteDuration(d)) return hi;
581 if (hi < 0 && time_internal::GetRepLo(d) != 0) ++hi;
582 return hi / (60 * 60);
583 }
584
ToDoubleNanoseconds(Duration d)585 double ToDoubleNanoseconds(Duration d) {
586 return FDivDuration(d, Nanoseconds(1));
587 }
ToDoubleMicroseconds(Duration d)588 double ToDoubleMicroseconds(Duration d) {
589 return FDivDuration(d, Microseconds(1));
590 }
ToDoubleMilliseconds(Duration d)591 double ToDoubleMilliseconds(Duration d) {
592 return FDivDuration(d, Milliseconds(1));
593 }
ToDoubleSeconds(Duration d)594 double ToDoubleSeconds(Duration d) {
595 return FDivDuration(d, Seconds(1));
596 }
ToDoubleMinutes(Duration d)597 double ToDoubleMinutes(Duration d) {
598 return FDivDuration(d, Minutes(1));
599 }
ToDoubleHours(Duration d)600 double ToDoubleHours(Duration d) {
601 return FDivDuration(d, Hours(1));
602 }
603
ToTimespec(Duration d)604 timespec ToTimespec(Duration d) {
605 timespec ts;
606 if (!time_internal::IsInfiniteDuration(d)) {
607 int64_t rep_hi = time_internal::GetRepHi(d);
608 uint32_t rep_lo = time_internal::GetRepLo(d);
609 if (rep_hi < 0) {
610 // Tweak the fields so that unsigned division of rep_lo
611 // maps to truncation (towards zero) for the timespec.
612 rep_lo += kTicksPerNanosecond - 1;
613 if (rep_lo >= kTicksPerSecond) {
614 rep_hi += 1;
615 rep_lo -= kTicksPerSecond;
616 }
617 }
618 ts.tv_sec = static_cast<decltype(ts.tv_sec)>(rep_hi);
619 if (ts.tv_sec == rep_hi) { // no time_t narrowing
620 ts.tv_nsec = rep_lo / kTicksPerNanosecond;
621 return ts;
622 }
623 }
624 if (d >= ZeroDuration()) {
625 ts.tv_sec = std::numeric_limits<time_t>::max();
626 ts.tv_nsec = 1000 * 1000 * 1000 - 1;
627 } else {
628 ts.tv_sec = std::numeric_limits<time_t>::min();
629 ts.tv_nsec = 0;
630 }
631 return ts;
632 }
633
ToTimeval(Duration d)634 timeval ToTimeval(Duration d) {
635 timeval tv;
636 timespec ts = ToTimespec(d);
637 if (ts.tv_sec < 0) {
638 // Tweak the fields so that positive division of tv_nsec
639 // maps to truncation (towards zero) for the timeval.
640 ts.tv_nsec += 1000 - 1;
641 if (ts.tv_nsec >= 1000 * 1000 * 1000) {
642 ts.tv_sec += 1;
643 ts.tv_nsec -= 1000 * 1000 * 1000;
644 }
645 }
646 tv.tv_sec = static_cast<decltype(tv.tv_sec)>(ts.tv_sec);
647 if (tv.tv_sec != ts.tv_sec) { // narrowing
648 if (ts.tv_sec < 0) {
649 tv.tv_sec = std::numeric_limits<decltype(tv.tv_sec)>::min();
650 tv.tv_usec = 0;
651 } else {
652 tv.tv_sec = std::numeric_limits<decltype(tv.tv_sec)>::max();
653 tv.tv_usec = 1000 * 1000 - 1;
654 }
655 return tv;
656 }
657 tv.tv_usec = static_cast<int>(ts.tv_nsec / 1000); // suseconds_t
658 return tv;
659 }
660
ToChronoNanoseconds(Duration d)661 std::chrono::nanoseconds ToChronoNanoseconds(Duration d) {
662 return time_internal::ToChronoDuration<std::chrono::nanoseconds>(d);
663 }
ToChronoMicroseconds(Duration d)664 std::chrono::microseconds ToChronoMicroseconds(Duration d) {
665 return time_internal::ToChronoDuration<std::chrono::microseconds>(d);
666 }
ToChronoMilliseconds(Duration d)667 std::chrono::milliseconds ToChronoMilliseconds(Duration d) {
668 return time_internal::ToChronoDuration<std::chrono::milliseconds>(d);
669 }
ToChronoSeconds(Duration d)670 std::chrono::seconds ToChronoSeconds(Duration d) {
671 return time_internal::ToChronoDuration<std::chrono::seconds>(d);
672 }
ToChronoMinutes(Duration d)673 std::chrono::minutes ToChronoMinutes(Duration d) {
674 return time_internal::ToChronoDuration<std::chrono::minutes>(d);
675 }
ToChronoHours(Duration d)676 std::chrono::hours ToChronoHours(Duration d) {
677 return time_internal::ToChronoDuration<std::chrono::hours>(d);
678 }
679
680 //
681 // To/From string formatting.
682 //
683
684 namespace {
685
686 // Formats a positive 64-bit integer in the given field width. Note that
687 // it is up to the caller of Format64() to ensure that there is sufficient
688 // space before ep to hold the conversion.
Format64(char * ep,int width,int64_t v)689 char* Format64(char* ep, int width, int64_t v) {
690 do {
691 --width;
692 *--ep = static_cast<char>('0' + (v % 10)); // contiguous digits
693 } while (v /= 10);
694 while (--width >= 0) *--ep = '0'; // zero pad
695 return ep;
696 }
697
698 // Helpers for FormatDuration() that format 'n' and append it to 'out'
699 // followed by the given 'unit'. If 'n' formats to "0", nothing is
700 // appended (not even the unit).
701
702 // A type that encapsulates how to display a value of a particular unit. For
703 // values that are displayed with fractional parts, the precision indicates
704 // where to round the value. The precision varies with the display unit because
705 // a Duration can hold only quarters of a nanosecond, so displaying information
706 // beyond that is just noise.
707 //
708 // For example, a microsecond value of 42.00025xxxxx should not display beyond 5
709 // fractional digits, because it is in the noise of what a Duration can
710 // represent.
711 struct DisplayUnit {
712 absl::string_view abbr;
713 int prec;
714 double pow10;
715 };
716 ABSL_CONST_INIT const DisplayUnit kDisplayNano = {"ns", 2, 1e2};
717 ABSL_CONST_INIT const DisplayUnit kDisplayMicro = {"us", 5, 1e5};
718 ABSL_CONST_INIT const DisplayUnit kDisplayMilli = {"ms", 8, 1e8};
719 ABSL_CONST_INIT const DisplayUnit kDisplaySec = {"s", 11, 1e11};
720 ABSL_CONST_INIT const DisplayUnit kDisplayMin = {"m", -1, 0.0}; // prec ignored
721 ABSL_CONST_INIT const DisplayUnit kDisplayHour = {"h", -1,
722 0.0}; // prec ignored
723
AppendNumberUnit(std::string * out,int64_t n,DisplayUnit unit)724 void AppendNumberUnit(std::string* out, int64_t n, DisplayUnit unit) {
725 char buf[sizeof("2562047788015216")]; // hours in max duration
726 char* const ep = buf + sizeof(buf);
727 char* bp = Format64(ep, 0, n);
728 if (*bp != '0' || bp + 1 != ep) {
729 out->append(bp, static_cast<size_t>(ep - bp));
730 out->append(unit.abbr.data(), unit.abbr.size());
731 }
732 }
733
734 // Note: unit.prec is limited to double's digits10 value (typically 15) so it
735 // always fits in buf[].
AppendNumberUnit(std::string * out,double n,DisplayUnit unit)736 void AppendNumberUnit(std::string* out, double n, DisplayUnit unit) {
737 constexpr int kBufferSize = std::numeric_limits<double>::digits10;
738 const int prec = std::min(kBufferSize, unit.prec);
739 char buf[kBufferSize]; // also large enough to hold integer part
740 char* ep = buf + sizeof(buf);
741 double d = 0;
742 int64_t frac_part = std::round(std::modf(n, &d) * unit.pow10);
743 int64_t int_part = d;
744 if (int_part != 0 || frac_part != 0) {
745 char* bp = Format64(ep, 0, int_part); // always < 1000
746 out->append(bp, static_cast<size_t>(ep - bp));
747 if (frac_part != 0) {
748 out->push_back('.');
749 bp = Format64(ep, prec, frac_part);
750 while (ep[-1] == '0') --ep;
751 out->append(bp, static_cast<size_t>(ep - bp));
752 }
753 out->append(unit.abbr.data(), unit.abbr.size());
754 }
755 }
756
757 } // namespace
758
759 // From Go's doc at https://golang.org/pkg/time/#Duration.String
760 // [FormatDuration] returns a string representing the duration in the
761 // form "72h3m0.5s". Leading zero units are omitted. As a special
762 // case, durations less than one second format use a smaller unit
763 // (milli-, micro-, or nanoseconds) to ensure that the leading digit
764 // is non-zero.
765 // Unlike Go, we format the zero duration as 0, with no unit.
FormatDuration(Duration d)766 std::string FormatDuration(Duration d) {
767 constexpr Duration kMinDuration = Seconds(kint64min);
768 std::string s;
769 if (d == kMinDuration) {
770 // Avoid needing to negate kint64min by directly returning what the
771 // following code should produce in that case.
772 s = "-2562047788015215h30m8s";
773 return s;
774 }
775 if (d < ZeroDuration()) {
776 s.append("-");
777 d = -d;
778 }
779 if (d == InfiniteDuration()) {
780 s.append("inf");
781 } else if (d < Seconds(1)) {
782 // Special case for durations with a magnitude < 1 second. The duration
783 // is printed as a fraction of a single unit, e.g., "1.2ms".
784 if (d < Microseconds(1)) {
785 AppendNumberUnit(&s, FDivDuration(d, Nanoseconds(1)), kDisplayNano);
786 } else if (d < Milliseconds(1)) {
787 AppendNumberUnit(&s, FDivDuration(d, Microseconds(1)), kDisplayMicro);
788 } else {
789 AppendNumberUnit(&s, FDivDuration(d, Milliseconds(1)), kDisplayMilli);
790 }
791 } else {
792 AppendNumberUnit(&s, IDivDuration(d, Hours(1), &d), kDisplayHour);
793 AppendNumberUnit(&s, IDivDuration(d, Minutes(1), &d), kDisplayMin);
794 AppendNumberUnit(&s, FDivDuration(d, Seconds(1)), kDisplaySec);
795 }
796 if (s.empty() || s == "-") {
797 s = "0";
798 }
799 return s;
800 }
801
802 namespace {
803
804 // A helper for ParseDuration() that parses a leading number from the given
805 // string and stores the result in *int_part/*frac_part/*frac_scale. The
806 // given string pointer is modified to point to the first unconsumed char.
ConsumeDurationNumber(const char ** dpp,const char * ep,int64_t * int_part,int64_t * frac_part,int64_t * frac_scale)807 bool ConsumeDurationNumber(const char** dpp, const char* ep, int64_t* int_part,
808 int64_t* frac_part, int64_t* frac_scale) {
809 *int_part = 0;
810 *frac_part = 0;
811 *frac_scale = 1; // invariant: *frac_part < *frac_scale
812 const char* start = *dpp;
813 for (; *dpp != ep; *dpp += 1) {
814 const int d = **dpp - '0'; // contiguous digits
815 if (d < 0 || 10 <= d) break;
816
817 if (*int_part > kint64max / 10) return false;
818 *int_part *= 10;
819 if (*int_part > kint64max - d) return false;
820 *int_part += d;
821 }
822 const bool int_part_empty = (*dpp == start);
823 if (*dpp == ep || **dpp != '.') return !int_part_empty;
824
825 for (*dpp += 1; *dpp != ep; *dpp += 1) {
826 const int d = **dpp - '0'; // contiguous digits
827 if (d < 0 || 10 <= d) break;
828 if (*frac_scale <= kint64max / 10) {
829 *frac_part *= 10;
830 *frac_part += d;
831 *frac_scale *= 10;
832 }
833 }
834 return !int_part_empty || *frac_scale != 1;
835 }
836
837 // A helper for ParseDuration() that parses a leading unit designator (e.g.,
838 // ns, us, ms, s, m, h) from the given string and stores the resulting unit
839 // in "*unit". The given string pointer is modified to point to the first
840 // unconsumed char.
ConsumeDurationUnit(const char ** start,const char * end,Duration * unit)841 bool ConsumeDurationUnit(const char** start, const char* end, Duration* unit) {
842 size_t size = static_cast<size_t>(end - *start);
843 switch (size) {
844 case 0:
845 return false;
846 default:
847 switch (**start) {
848 case 'n':
849 if (*(*start + 1) == 's') {
850 *start += 2;
851 *unit = Nanoseconds(1);
852 return true;
853 }
854 break;
855 case 'u':
856 if (*(*start + 1) == 's') {
857 *start += 2;
858 *unit = Microseconds(1);
859 return true;
860 }
861 break;
862 case 'm':
863 if (*(*start + 1) == 's') {
864 *start += 2;
865 *unit = Milliseconds(1);
866 return true;
867 }
868 break;
869 default:
870 break;
871 }
872 ABSL_FALLTHROUGH_INTENDED;
873 case 1:
874 switch (**start) {
875 case 's':
876 *unit = Seconds(1);
877 *start += 1;
878 return true;
879 case 'm':
880 *unit = Minutes(1);
881 *start += 1;
882 return true;
883 case 'h':
884 *unit = Hours(1);
885 *start += 1;
886 return true;
887 default:
888 return false;
889 }
890 }
891 }
892
893 } // namespace
894
895 // From Go's doc at https://golang.org/pkg/time/#ParseDuration
896 // [ParseDuration] parses a duration string. A duration string is
897 // a possibly signed sequence of decimal numbers, each with optional
898 // fraction and a unit suffix, such as "300ms", "-1.5h" or "2h45m".
899 // Valid time units are "ns", "us" "ms", "s", "m", "h".
ParseDuration(absl::string_view dur_sv,Duration * d)900 bool ParseDuration(absl::string_view dur_sv, Duration* d) {
901 int sign = 1;
902 if (absl::ConsumePrefix(&dur_sv, "-")) {
903 sign = -1;
904 } else {
905 absl::ConsumePrefix(&dur_sv, "+");
906 }
907 if (dur_sv.empty()) return false;
908
909 // Special case for a string of "0".
910 if (dur_sv == "0") {
911 *d = ZeroDuration();
912 return true;
913 }
914
915 if (dur_sv == "inf") {
916 *d = sign * InfiniteDuration();
917 return true;
918 }
919
920 const char* start = dur_sv.data();
921 const char* end = start + dur_sv.size();
922
923 Duration dur;
924 while (start != end) {
925 int64_t int_part;
926 int64_t frac_part;
927 int64_t frac_scale;
928 Duration unit;
929 if (!ConsumeDurationNumber(&start, end, &int_part, &frac_part,
930 &frac_scale) ||
931 !ConsumeDurationUnit(&start, end, &unit)) {
932 return false;
933 }
934 if (int_part != 0) dur += sign * int_part * unit;
935 if (frac_part != 0) dur += sign * frac_part * unit / frac_scale;
936 }
937 *d = dur;
938 return true;
939 }
940
AbslParseFlag(absl::string_view text,Duration * dst,std::string *)941 bool AbslParseFlag(absl::string_view text, Duration* dst, std::string*) {
942 return ParseDuration(text, dst);
943 }
944
AbslUnparseFlag(Duration d)945 std::string AbslUnparseFlag(Duration d) { return FormatDuration(d); }
ParseFlag(const std::string & text,Duration * dst,std::string *)946 bool ParseFlag(const std::string& text, Duration* dst, std::string* ) {
947 return ParseDuration(text, dst);
948 }
949
UnparseFlag(Duration d)950 std::string UnparseFlag(Duration d) { return FormatDuration(d); }
951
952 ABSL_NAMESPACE_END
953 } // namespace absl
954