Whamcloud - gitweb
LU-14129 kernel: kernel update RHEL7.9 [3.10.0-1160.6.1.el7]
[fs/lustre-release.git] / contrib / lbuild / exit_traps.sh
1 #!/bin/bash
2
3 # a framework for stacking EXIT traps
4 # this could, should be further enhanced to allow stacks of traps for various
5 # exits.  but right now it's hard-coded for EXIT traps
6
7 exit_actions=()
8
9 push_exit_trap() {
10     local action="$1"
11     local trap_handle="$2"
12
13     local var="exit_trap_handle_$trap_handle"
14
15     if [ -n "${!var}" ]; then
16         echo "fail!  trap handle $trap_handle is already in use"
17         return 1
18     fi
19
20     local num_items=${#exit_actions[@]}
21     exit_actions[$num_items]="$action"
22     eval $var="$num_items"
23
24     return 0
25
26 }
27
28 delete_exit_trap() {
29     local trap_handles="$@"
30
31     local handle
32     for handle in $trap_handles; do
33         local var="exit_trap_handle_$handle"
34         local trap_num=${!var}
35         exit_actions[$trap_num]=""
36         eval unset $var
37     done
38 }
39
40 print_exit_traps() {
41
42     local i num_items=${#exit_actions[@]}
43     for i in $(seq 0 $((num_items-1))); do
44         if [ -z "${exit_actions[$i]}" ]; then
45             continue
46         fi
47         echo "${exit_actions[$i]}"
48     done
49
50 }
51
52 run_exit_traps() {
53
54     local i num_items=${#exit_actions[@]}
55     for i in $(seq $((num_items-1)) -1 0); do
56         if [ -z "${exit_actions[$i]}" ]; then
57             continue
58         fi
59         eval ${exit_actions[$i]}
60     done
61
62 }
63
64 trap run_exit_traps EXIT
65
66 if [ "$1" = "unit_test" ]; then
67     if ! push_exit_trap "echo \"this is the first trap\"" "a"; then
68         echo "failed to install trap 1"
69         exit 1
70     fi
71     if ! push_exit_trap "echo \"this is the second trap\"" "b"; then
72         echo "failed to install trap 2"
73         exit 2
74     fi
75     delete_exit_trap "b"
76     if ! push_exit_trap "echo \"this is the third trap\"" "b"; then
77         echo "failed to install trap 3"
78         exit 3
79     fi
80     
81     # to see the traps
82     print_exit_traps
83     echo "------------"
84
85     delete_exit_trap "a" "b"
86     print_exit_traps
87     echo "------------"
88    
89     if ! push_exit_trap "echo \"this is the first trap\"" "a"; then
90         echo "failed to install trap 1"
91         exit 1
92     fi
93     if ! push_exit_trap "echo \"this is the second trap\"" "b"; then
94         echo "failed to install trap 2"
95         exit 2
96     fi
97     if ! push_exit_trap "echo \"this is the third trap\"" "c"; then
98         echo "failed to install trap 3"
99         exit 3
100     fi
101     delete_exit_trap "a" "c"
102
103     print_exit_traps
104     echo "------------"
105 fi