1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
timeout_killer_thread() {
while ! test_frozen; do
sleep 1
if test_frozen; then
return
fi
local multiplier=$(cat "$test_timeout_multiplier_file")
local begin_ts=$(cat "$test_timeout_begin_ts_file") || true
local value_string=$(cat "$test_timeout_value_file") || true
local value=$(($(date +%s -d "$value_string") - $(date +%s)))
local now_ts=$(date +%s)
if [[ -z $begin_ts || -z $value_string || -z $multiplier ]]; then
# A race with timeout_set occurred (it truncates the endTS file and then writes it)
# or a race with test_cleanup (test_timeout_end_ts_file has just been removed)
continue
fi
local end_ts=$(($begin_ts + $value * $multiplier))
if (( now_ts >= end_ts )); then
if (( multiplier != 1 )); then
test_add_failure "Test timed out (${multiplier} * ${value_string})"
else
test_add_failure "Test timed out (${value_string})"
fi
test_freeze_result
killall -9 -u $(whoami)
fi
done
}
timeout_init() {
test_timeout_begin_ts_file="$TEMP_DIR/$(unique_file)_timeout_beginTS.txt"
test_timeout_multiplier_file="$TEMP_DIR/$(unique_file)_timeout_multiplier.txt"
test_timeout_value_file="$TEMP_DIR/$(unique_file)_timeout_value.txt"
# default timeout values
timeout_set "30 seconds"
timeout_set_multiplier 1
# Parentheses below are needed to make 'wait' command work in tests.
# They make the killer thread to be a job owned by a temporary subshell, not ours
( timeout_killer_thread & )
}
timeout_set() {
echo "$*" > "$test_timeout_value_file"
date +%s > "$test_timeout_begin_ts_file"
}
timeout_set_multiplier() {
echo "$1" > "$test_timeout_multiplier_file"
}
timeout_get_multiplier() {
cat "$test_timeout_multiplier_file"
}
# takes as parameter timeout (e.g. '1 minute') and rescales it by valgrind multiplier
# prints result in seconds to standard output (e.g. '900 seconds')
timeout_rescale() {
if valgrind_enabled; then
local multiplier=$(timeout_get_multiplier)
local value=$(($(date +%s -d "$1") - $(date +%s)))
echo $((value * multiplier)) seconds
else
echo "$1"
fi
}
# takes as parameter timeout in seconds (e.g. '60') and rescales it by valgrind multiplier
# prints result to standard output (e.g. '900')
timeout_rescale_seconds() {
if valgrind_enabled; then
local multiplier=$(timeout_get_multiplier)
echo $(($1 * multiplier))
else
echo "$1"
fi
}
|
