blob: 37c1502c065106d4d9fce6b95abd6fcd5f745b26 (
plain)
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
# Test kernel functions for kprobe safety.
#
# Copyright (C) 2006 IBM Corp.
# Copyright (C) 2008 Red Hat Inc.
#
# This file is part of systemtap, and is free software. You can
# redistribute it and/or modify it under the terms of the GNU General
# Public License (GPL); either version 2, or (at your option) any
# later version.
#
# Based on software written by Gui Jian <guijian@cn.ibm.com>
#
load_lib "whitelist_lib.exp"
set MAX_RUNNING_LEVEL 3
set GENKPROBES_RUNNING "/genkprobes_running"
set PROBES_RESULT "probe.out"
set PROBES_ALL "probes.all"
set PROBES_PENDING "probes.pending"
set PROBES_CURRENT "probes.current"
set PROBES_PASSED "probes.passed"
set PROBES_FAILED "probes.failed"
set PROBES_UNTRIGGERED "probes.untriggered"
set PROBES_UNREGISTERED "probes.unregistered"
proc proper_current_size {level inputfile} {
set totalsize [get_linesize $inputfile]
switch $level {
{1} {set currentsize [expr $totalsize/9]
# Use a max size of 1000
if {$currentsize > 1000} {
set currentsize 1000
}
# Use a min size of 50
if {$currentsize < 50} {
set currentsize 50
}
}
{2} {set currentsize [expr $totalsize/49]
# Use a max size of 400
if {$currentsize > 400} {
set currentsize 400
}
# Use a min size of 10
if {$currentsize < 10} {
set currentsize 10
}
}
{3} {set currentsize 100}
{4} {set currentsize 10}
default {
puts "Reached unexpected iteration level: $level"
set currentsize $totalsize
}
}
if {$currentsize <= 0} {
set currentsize 5
}
return $currentsize
}
#
# Main routine of the whole test
#
if {[get_running_level] == 0} {
# Not started yet, start the whole test from the scratch
# Append the startup code to /etc/init.d/rc.local if not yet
register_service
# Check whether probes.all is empty or not given
if {[get_linesize $PROBES_ALL] == 0} {
init_probes_all
}
# Set current_running_level as 1 to indicate a new test started
init_running_level
# Initialize intermediate files based on probe.all
exec rm -f $PROBES_PENDING $PROBES_CURRENT
exec rm -f $PROBES_PASSED $PROBES_UNTRIGGERED $PROBES_FAILED $PROBES_UNREGISTERED
file copy $PROBES_ALL $PROBES_PENDING
exec touch $PROBES_PASSED $PROBES_UNTRIGGERED $PROBES_FAILED $PROBES_UNREGISTERED
sync_discs
puts "Start a fresh stp_genwhitelist test."
} else {
# Maybe started already, so do some cleaning if necessary
garbage_collect $ALREADY_CRASHED
puts "Recovered from last maybe crashed probe test."
}
set current_size_const [proper_current_size [get_running_level] $PROBES_ALL]
puts "current_size_const is initialized as $current_size_const"
while {1} {
puts "Current size of probes.pending is [get_linesize $PROBES_PENDING]"
if {[get_linesize $PROBES_PENDING] == 0} {
# Check whether we need the next iteration or not
global MAX_RUNNING_LEVEL
# incr running_level for the start of a new iteration
set old_running_level [get_running_level]
incr_running_level
puts "Running level increased to [get_running_level]"
if {[get_running_level] > $MAX_RUNNING_LEVEL} {
puts "Reached max iteration limit: [get_running_level]"
break
} else {
exec cp $PROBES_PASSED $PROBES_PASSED-pass$old_running_level
exec cp $PROBES_FAILED $PROBES_FAILED-pass$old_running_level
exec cp $PROBES_UNTRIGGERED $PROBES_UNTRIGGERED-pass$old_running_level
puts "Current running level is [get_running_level]"
exec rm -f $PROBES_PENDING
if {[get_linesize $PROBES_FAILED] > 0} {
# Append probes.failed to probes.pending
exec cat $PROBES_FAILED >> $PROBES_PENDING
file delete $PROBES_FAILED
exec touch $PROBES_FAILED
puts "Append $PROBES_FAILED to $PROBES_PENDING, now size=[get_linesize $PROBES_PENDING]"
}
if {[get_linesize $PROBES_UNTRIGGERED] > 0} {
# Append probes.untriggered to probes.pending
exec cat $PROBES_UNTRIGGERED >> $PROBES_PENDING
file delete $PROBES_UNTRIGGERED
exec touch $PROBES_UNTRIGGERED
puts "Append $PROBES_UNTRIGGERED to $PROBES_PENDING, now size=[get_linesize $PROBES_PENDING]"
}
if {[get_linesize $PROBES_PENDING] == 0} {
# No more pending probe points
puts "No more iterations needed. Stopped."
break
}
# set new value of current_size_const for new iteration level
set current_size_const [proper_current_size [get_running_level] $PROBES_ALL]
puts "current_size_const is set as $current_size_const now"
continue
}
}
# Now, non-empty probes.pending should be ready
# Generate probes.current
exec rm -f $PROBES_CURRENT
exec head -n $current_size_const $PROBES_PENDING > $PROBES_CURRENT
exec tail -n+[expr $current_size_const+1] $PROBES_PENDING > /tmp/whitelist_tmpfile
exec mv /tmp/whitelist_tmpfile $PROBES_PENDING
sync_discs
set d [exec date]
puts "Ready to do current probe test at $d"
# Do actual probe test
do_current_test
set d [exec date]
puts "Completed one probe test successfully at $d"
# No crash fortunately, so do some cleaning to prepare for next test
garbage_collect $NO_CRASH
}
# Congratulations for arriving here
# Remove all temporary files and unregister myself
puts "Now removing all temporary files, unregistering the service and exit."
exec rm -f $PROBES_PENDING $PROBES_CURRENT $GENKPROBES_RUNNING
exec rm -f $PROBES_RESULT /tmp/whitelist_tmpfile
unregister_service
exit 0
|
