1*61046927SAndroid Build Coastguard Worker /**************************************************************************
2*61046927SAndroid Build Coastguard Worker *
3*61046927SAndroid Build Coastguard Worker * Copyright 2008-2010 VMware, Inc.
4*61046927SAndroid Build Coastguard Worker * All Rights Reserved.
5*61046927SAndroid Build Coastguard Worker *
6*61046927SAndroid Build Coastguard Worker * Permission is hereby granted, free of charge, to any person obtaining a
7*61046927SAndroid Build Coastguard Worker * copy of this software and associated documentation files (the
8*61046927SAndroid Build Coastguard Worker * "Software"), to deal in the Software without restriction, including
9*61046927SAndroid Build Coastguard Worker * without limitation the rights to use, copy, modify, merge, publish,
10*61046927SAndroid Build Coastguard Worker * distribute, sub license, and/or sell copies of the Software, and to
11*61046927SAndroid Build Coastguard Worker * permit persons to whom the Software is furnished to do so, subject to
12*61046927SAndroid Build Coastguard Worker * the following conditions:
13*61046927SAndroid Build Coastguard Worker *
14*61046927SAndroid Build Coastguard Worker * The above copyright notice and this permission notice (including the
15*61046927SAndroid Build Coastguard Worker * next paragraph) shall be included in all copies or substantial portions
16*61046927SAndroid Build Coastguard Worker * of the Software.
17*61046927SAndroid Build Coastguard Worker *
18*61046927SAndroid Build Coastguard Worker * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19*61046927SAndroid Build Coastguard Worker * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20*61046927SAndroid Build Coastguard Worker * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21*61046927SAndroid Build Coastguard Worker * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
22*61046927SAndroid Build Coastguard Worker * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23*61046927SAndroid Build Coastguard Worker * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24*61046927SAndroid Build Coastguard Worker * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25*61046927SAndroid Build Coastguard Worker *
26*61046927SAndroid Build Coastguard Worker **************************************************************************/
27*61046927SAndroid Build Coastguard Worker
28*61046927SAndroid Build Coastguard Worker /**
29*61046927SAndroid Build Coastguard Worker * @file
30*61046927SAndroid Build Coastguard Worker * OS independent time-manipulation functions.
31*61046927SAndroid Build Coastguard Worker *
32*61046927SAndroid Build Coastguard Worker * @author Jose Fonseca <[email protected]>
33*61046927SAndroid Build Coastguard Worker */
34*61046927SAndroid Build Coastguard Worker
35*61046927SAndroid Build Coastguard Worker #include "os_time.h"
36*61046927SAndroid Build Coastguard Worker #include "detect_os.h"
37*61046927SAndroid Build Coastguard Worker
38*61046927SAndroid Build Coastguard Worker #include "c11/time.h"
39*61046927SAndroid Build Coastguard Worker
40*61046927SAndroid Build Coastguard Worker #include "util/u_atomic.h"
41*61046927SAndroid Build Coastguard Worker
42*61046927SAndroid Build Coastguard Worker #if DETECT_OS_POSIX_LITE
43*61046927SAndroid Build Coastguard Worker # include <unistd.h> /* usleep */
44*61046927SAndroid Build Coastguard Worker # include <time.h> /* timeval */
45*61046927SAndroid Build Coastguard Worker # include <sys/time.h> /* timeval */
46*61046927SAndroid Build Coastguard Worker # include <sched.h> /* sched_yield */
47*61046927SAndroid Build Coastguard Worker # include <errno.h>
48*61046927SAndroid Build Coastguard Worker #elif DETECT_OS_WINDOWS
49*61046927SAndroid Build Coastguard Worker # include <windows.h>
50*61046927SAndroid Build Coastguard Worker #else
51*61046927SAndroid Build Coastguard Worker # error Unsupported OS
52*61046927SAndroid Build Coastguard Worker #endif
53*61046927SAndroid Build Coastguard Worker
54*61046927SAndroid Build Coastguard Worker
55*61046927SAndroid Build Coastguard Worker int64_t
os_time_get_nano(void)56*61046927SAndroid Build Coastguard Worker os_time_get_nano(void)
57*61046927SAndroid Build Coastguard Worker {
58*61046927SAndroid Build Coastguard Worker struct timespec ts;
59*61046927SAndroid Build Coastguard Worker timespec_get(&ts, TIME_MONOTONIC);
60*61046927SAndroid Build Coastguard Worker return ts.tv_nsec + ts.tv_sec*INT64_C(1000000000);
61*61046927SAndroid Build Coastguard Worker }
62*61046927SAndroid Build Coastguard Worker
63*61046927SAndroid Build Coastguard Worker
64*61046927SAndroid Build Coastguard Worker
65*61046927SAndroid Build Coastguard Worker void
os_time_sleep(int64_t usecs)66*61046927SAndroid Build Coastguard Worker os_time_sleep(int64_t usecs)
67*61046927SAndroid Build Coastguard Worker {
68*61046927SAndroid Build Coastguard Worker #if DETECT_OS_LINUX || DETECT_OS_MANAGARM || DETECT_OS_FUCHSIA
69*61046927SAndroid Build Coastguard Worker struct timespec time;
70*61046927SAndroid Build Coastguard Worker time.tv_sec = usecs / 1000000;
71*61046927SAndroid Build Coastguard Worker time.tv_nsec = (usecs % 1000000) * 1000;
72*61046927SAndroid Build Coastguard Worker while (clock_nanosleep(CLOCK_MONOTONIC, 0, &time, &time) == EINTR);
73*61046927SAndroid Build Coastguard Worker
74*61046927SAndroid Build Coastguard Worker #elif DETECT_OS_POSIX
75*61046927SAndroid Build Coastguard Worker usleep(usecs);
76*61046927SAndroid Build Coastguard Worker
77*61046927SAndroid Build Coastguard Worker #elif DETECT_OS_WINDOWS
78*61046927SAndroid Build Coastguard Worker DWORD dwMilliseconds = (DWORD) ((usecs + 999) / 1000);
79*61046927SAndroid Build Coastguard Worker /* Avoid Sleep(O) as that would cause to sleep for an undetermined duration */
80*61046927SAndroid Build Coastguard Worker if (dwMilliseconds) {
81*61046927SAndroid Build Coastguard Worker Sleep(dwMilliseconds);
82*61046927SAndroid Build Coastguard Worker }
83*61046927SAndroid Build Coastguard Worker #else
84*61046927SAndroid Build Coastguard Worker # error Unsupported OS
85*61046927SAndroid Build Coastguard Worker #endif
86*61046927SAndroid Build Coastguard Worker }
87*61046927SAndroid Build Coastguard Worker
88*61046927SAndroid Build Coastguard Worker
89*61046927SAndroid Build Coastguard Worker
90*61046927SAndroid Build Coastguard Worker int64_t
os_time_get_absolute_timeout(uint64_t timeout)91*61046927SAndroid Build Coastguard Worker os_time_get_absolute_timeout(uint64_t timeout)
92*61046927SAndroid Build Coastguard Worker {
93*61046927SAndroid Build Coastguard Worker int64_t time, abs_timeout;
94*61046927SAndroid Build Coastguard Worker
95*61046927SAndroid Build Coastguard Worker /* Also check for the type upper bound. */
96*61046927SAndroid Build Coastguard Worker if (timeout == OS_TIMEOUT_INFINITE || timeout > INT64_MAX)
97*61046927SAndroid Build Coastguard Worker return OS_TIMEOUT_INFINITE;
98*61046927SAndroid Build Coastguard Worker
99*61046927SAndroid Build Coastguard Worker time = os_time_get_nano();
100*61046927SAndroid Build Coastguard Worker
101*61046927SAndroid Build Coastguard Worker /* Do the addition in unsigned, because signed overflow is UB, then convert to signed again. */
102*61046927SAndroid Build Coastguard Worker abs_timeout = (uint64_t)time + (uint64_t)timeout;
103*61046927SAndroid Build Coastguard Worker
104*61046927SAndroid Build Coastguard Worker /* Check for overflow. */
105*61046927SAndroid Build Coastguard Worker if (abs_timeout < time)
106*61046927SAndroid Build Coastguard Worker return OS_TIMEOUT_INFINITE;
107*61046927SAndroid Build Coastguard Worker
108*61046927SAndroid Build Coastguard Worker return abs_timeout;
109*61046927SAndroid Build Coastguard Worker }
110*61046927SAndroid Build Coastguard Worker
111*61046927SAndroid Build Coastguard Worker
112*61046927SAndroid Build Coastguard Worker bool
os_wait_until_zero(volatile int * var,uint64_t timeout)113*61046927SAndroid Build Coastguard Worker os_wait_until_zero(volatile int *var, uint64_t timeout)
114*61046927SAndroid Build Coastguard Worker {
115*61046927SAndroid Build Coastguard Worker if (!p_atomic_read(var))
116*61046927SAndroid Build Coastguard Worker return true;
117*61046927SAndroid Build Coastguard Worker
118*61046927SAndroid Build Coastguard Worker if (!timeout)
119*61046927SAndroid Build Coastguard Worker return false;
120*61046927SAndroid Build Coastguard Worker
121*61046927SAndroid Build Coastguard Worker if (timeout == OS_TIMEOUT_INFINITE) {
122*61046927SAndroid Build Coastguard Worker while (p_atomic_read(var)) {
123*61046927SAndroid Build Coastguard Worker #if DETECT_OS_POSIX_LITE
124*61046927SAndroid Build Coastguard Worker sched_yield();
125*61046927SAndroid Build Coastguard Worker #endif
126*61046927SAndroid Build Coastguard Worker }
127*61046927SAndroid Build Coastguard Worker return true;
128*61046927SAndroid Build Coastguard Worker }
129*61046927SAndroid Build Coastguard Worker else {
130*61046927SAndroid Build Coastguard Worker int64_t start_time = os_time_get_nano();
131*61046927SAndroid Build Coastguard Worker int64_t end_time = start_time + timeout;
132*61046927SAndroid Build Coastguard Worker
133*61046927SAndroid Build Coastguard Worker while (p_atomic_read(var)) {
134*61046927SAndroid Build Coastguard Worker if (os_time_timeout(start_time, end_time, os_time_get_nano()))
135*61046927SAndroid Build Coastguard Worker return false;
136*61046927SAndroid Build Coastguard Worker
137*61046927SAndroid Build Coastguard Worker #if DETECT_OS_POSIX_LITE
138*61046927SAndroid Build Coastguard Worker sched_yield();
139*61046927SAndroid Build Coastguard Worker #endif
140*61046927SAndroid Build Coastguard Worker }
141*61046927SAndroid Build Coastguard Worker return true;
142*61046927SAndroid Build Coastguard Worker }
143*61046927SAndroid Build Coastguard Worker }
144*61046927SAndroid Build Coastguard Worker
145*61046927SAndroid Build Coastguard Worker
146*61046927SAndroid Build Coastguard Worker bool
os_wait_until_zero_abs_timeout(volatile int * var,int64_t timeout)147*61046927SAndroid Build Coastguard Worker os_wait_until_zero_abs_timeout(volatile int *var, int64_t timeout)
148*61046927SAndroid Build Coastguard Worker {
149*61046927SAndroid Build Coastguard Worker if (!p_atomic_read(var))
150*61046927SAndroid Build Coastguard Worker return true;
151*61046927SAndroid Build Coastguard Worker
152*61046927SAndroid Build Coastguard Worker if (timeout == OS_TIMEOUT_INFINITE)
153*61046927SAndroid Build Coastguard Worker return os_wait_until_zero(var, OS_TIMEOUT_INFINITE);
154*61046927SAndroid Build Coastguard Worker
155*61046927SAndroid Build Coastguard Worker while (p_atomic_read(var)) {
156*61046927SAndroid Build Coastguard Worker if (os_time_get_nano() >= timeout)
157*61046927SAndroid Build Coastguard Worker return false;
158*61046927SAndroid Build Coastguard Worker
159*61046927SAndroid Build Coastguard Worker #if DETECT_OS_POSIX_LITE
160*61046927SAndroid Build Coastguard Worker sched_yield();
161*61046927SAndroid Build Coastguard Worker #endif
162*61046927SAndroid Build Coastguard Worker }
163*61046927SAndroid Build Coastguard Worker return true;
164*61046927SAndroid Build Coastguard Worker }
165