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
|
# Simple function to test that systemtap can generate instument a module
# function, install it, and get some output.
set test "kmodule"
set test2 "kprobe_module"
if {![installtest_p]} {
untested $test
return
}
set build_dir ""
set uname [exec /bin/uname -r]
proc build_and_install_module {} {
global build_dir
global srcdir subdir
# Create the build directory and populate it
if {[catch {exec mktemp -d staptestXXXXXX} build_dir]} {
verbose -log "Failed to create temporary directory: $build_dir"
return 0
}
catch {exec cp $srcdir/$subdir/stap_kmodule.c $build_dir/}
catch {exec cp -p $srcdir/$subdir/stap_kmodule.Makefile $build_dir/Makefile }
# Build the module
if {[catch {exec make -C $build_dir clean} res]} {
verbose -log "$res"
return 0
}
catch {exec make -C $build_dir} res
if {![file exists $build_dir/stap_kmodule.ko]} {
verbose -log "$res"
return 0
}
set res [as_root [list cp $build_dir/stap_kmodule.ko /lib/modules/$::uname/kernel/]]
if { $res != 0 } {
verbose -log "$res"
return 0
}
# Install the module
set res [as_root [list /sbin/insmod $build_dir/stap_kmodule.ko]]
if {$res != 0} {
verbose -log "$res"
return 0
}
return 1
}
proc cleanup_module {} {
global build_dir
as_root [list /bin/rm -f /lib/modules/$::uname/kernel/stap_kmodule.ko]
as_root [list /sbin/rmmod stap_kmodule]
if {$build_dir != ""} {
catch { exec rm -rf $build_dir }
}
}
proc kmodule_load2 {} {
global build_dir
# Unload the test module (in case it was leftover from a previous
# run).
as_root [list /sbin/rmmod stap_kmodule]
# Load the test module
set res [as_root [list /sbin/insmod $build_dir/stap_kmodule.ko]]
if { $res != 0 } {
verbose -log "$res"
return 1
}
return 0
}
set output_string "count = 1\r\n"
if {[build_and_install_module] == 0} {
verbose -log "BUILD FAILED"
fail "$test (could not build/install module)"
return
} else {
pass "$test (built and installed module)"
}
# Test 'module("foo").function("bar")
sleep 1
stap_run $test "" $output_string $srcdir/$subdir/$test.stp
# Test 'module("foo").function("bar").return
sleep 1
stap_run "$test return" "" $output_string $srcdir/$subdir/$test.stp return
# Test 'kprobe.module("foo").function("bar")
sleep 1
stap_run $test2 "" $output_string $srcdir/$subdir/$test2.stp
# Test 'kprobe.module("foo").function("bar").return
sleep 1
stap_run "$test2 return" "" $output_string $srcdir/$subdir/$test2.stp return
# Now let's test and see if the same scripts work when the test module
# is loaded *after* the systemtap script starts.
set test_suffix "(loaded after)"
if {![module_refresh_p]} {
untested "$subdir/$test.stp $test_suffix"
untested "$subdir/$test2.stp $test_suffix"
cleanup_module
return
}
as_root [list /sbin/rmmod stap_kmodule]
# Test 'module("foo").function("bar")
set test_file "$subdir/$test.stp"
sleep 1
stap_run "$test $test_suffix" kmodule_load2 \
$output_string $srcdir/$test_file
# Test 'kprobe.module("foo").function("bar")
set test_file "$subdir/$test2.stp"
sleep 1
stap_run "$test2 $test_suffix" kmodule_load2 \
$output_string $srcdir/$test_file
cleanup_module
|
