hc
2024-05-09 b9d5c334faa47a75f1f28e72d203fc0334e8471d
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
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
#!/bin/sh
# SPDX-License-Identifier: GPL-2.0-only
 
usage() {
   echo "Ftrace boottime trace test tool"
   echo "Usage: $0 [--apply|--init] [--debug] BOOTCONFIG-FILE"
   echo "    --apply: Test actual apply to tracefs (need sudo)"
   echo "    --init:  Initialize ftrace before applying (imply --apply)"
   exit 1
}
 
[ $# -eq 0 ] && usage
 
BCONF=
DEBUG=
APPLY=
INIT=
while [ x"$1" != x ]; do
   case "$1" in
   "--debug")
       DEBUG=$1;;
   "--apply")
       APPLY=$1;;
   "--init")
       APPLY=$1
       INIT=$1;;
   *)
       [ ! -f $1 ] && usage
       BCONF=$1;;
   esac
   shift 1
done
 
if [ x"$APPLY" != x ]; then
   if [ `id -u` -ne 0 ]; then
       echo "This must be run by root user. Try sudo." 1>&2
       exec sudo $0 $DEBUG $APPLY $BCONF
   fi
fi
 
run_cmd() { # command
   echo "$*"
   if [ x"$APPLY" != x ]; then # apply command
       eval $*
   fi
}
 
if [ x"$DEBUG" != x ]; then
   set -x
fi
 
TRACEFS=`grep -m 1 -w tracefs /proc/mounts | cut -f 2 -d " "`
if [ -z "$TRACEFS" ]; then
   if ! grep -wq debugfs /proc/mounts; then
       echo "Error: No tracefs/debugfs was mounted." 1>&2
       exit 1
   fi
   TRACEFS=`grep -m 1 -w debugfs /proc/mounts | cut -f 2 -d " "`/tracing
   if [ ! -d $TRACEFS ]; then
       echo "Error: ftrace is not enabled on this kernel." 1>&2
       exit 1
   fi
fi
 
if [ x"$INIT" != x ]; then
   . `dirname $0`/ftrace.sh
   (cd $TRACEFS; initialize_ftrace)
fi
 
. `dirname $0`/xbc.sh
 
######## main #########
set -e
 
xbc_init $BCONF
 
set_value_of() { # key file
   if xbc_has_key $1; then
       val=`xbc_get_val $1 1`
       run_cmd "echo '$val' >> $2"
   fi
}
 
set_array_of() { # key file
   if xbc_has_key $1; then
       xbc_get_val $1 | while read line; do
           run_cmd "echo '$line' >> $2"
       done
   fi
}
 
compose_synth() { # event_name branch
   echo -n "$1 "
   xbc_get_val $2 | while read field; do echo -n "$field; "; done
}
 
setup_event() { # prefix group event [instance]
   branch=$1.$2.$3
   if [ "$4" ]; then
       eventdir="$TRACEFS/instances/$4/events/$2/$3"
   else
       eventdir="$TRACEFS/events/$2/$3"
   fi
   case $2 in
   kprobes)
       xbc_get_val ${branch}.probes | while read line; do
           run_cmd "echo 'p:kprobes/$3 $line' >> $TRACEFS/kprobe_events"
       done
       ;;
   synthetic)
       run_cmd "echo '`compose_synth $3 ${branch}.fields`' >> $TRACEFS/synthetic_events"
       ;;
   esac
 
   set_value_of ${branch}.filter ${eventdir}/filter
   set_array_of ${branch}.actions ${eventdir}/trigger
 
   if xbc_has_key ${branch}.enable; then
       run_cmd "echo 1 > ${eventdir}/enable"
   fi
}
 
setup_events() { # prefix("ftrace" or "ftrace.instance.INSTANCE") [instance]
   prefix="${1}.event"
   if xbc_has_branch ${1}.event; then
       for grpev in `xbc_subkeys ${1}.event 2`; do
           setup_event $prefix ${grpev%.*} ${grpev#*.} $2
       done
   fi
}
 
size2kb() { # size[KB|MB]
   case $1 in
   *KB)
       echo ${1%KB};;
   *MB)
       expr ${1%MB} \* 1024;;
   *)
       expr $1 / 1024 ;;
   esac
}
 
setup_instance() { # [instance]
   if [ "$1" ]; then
       instance="ftrace.instance.${1}"
       instancedir=$TRACEFS/instances/$1
   else
       instance="ftrace"
       instancedir=$TRACEFS
   fi
 
   set_array_of ${instance}.options ${instancedir}/trace_options
   set_value_of ${instance}.trace_clock ${instancedir}/trace_clock
   set_value_of ${instance}.cpumask ${instancedir}/tracing_cpumask
   set_value_of ${instance}.tracing_on ${instancedir}/tracing_on
   set_value_of ${instance}.tracer ${instancedir}/current_tracer
   set_array_of ${instance}.ftrace.filters \
       ${instancedir}/set_ftrace_filter
   set_array_of ${instance}.ftrace.notrace \
       ${instancedir}/set_ftrace_notrace
 
   if xbc_has_key ${instance}.alloc_snapshot; then
       run_cmd "echo 1 > ${instancedir}/snapshot"
   fi
 
   if xbc_has_key ${instance}.buffer_size; then
       size=`xbc_get_val ${instance}.buffer_size 1`
       size=`eval size2kb $size`
       run_cmd "echo $size >> ${instancedir}/buffer_size_kb"
   fi
 
   setup_events ${instance} $1
   set_array_of ${instance}.events ${instancedir}/set_event
}
 
# ftrace global configs (kernel.*)
if xbc_has_key "kernel.dump_on_oops"; then
   dump_mode=`xbc_get_val "kernel.dump_on_oops" 1`
   [ "$dump_mode" ] && dump_mode=`eval echo $dump_mode` || dump_mode=1
   run_cmd "echo \"$dump_mode\" > /proc/sys/kernel/ftrace_dump_on_oops"
fi
 
set_value_of kernel.fgraph_max_depth $TRACEFS/max_graph_depth
set_array_of kernel.fgraph_filters $TRACEFS/set_graph_function
set_array_of kernel.fgraph_notraces $TRACEFS/set_graph_notrace
 
# Per-instance/per-event configs
if ! xbc_has_branch "ftrace" ; then
   exit 0
fi
 
setup_instance # root instance
 
if xbc_has_branch "ftrace.instance"; then
   for i in `xbc_subkeys "ftrace.instance" 1`; do
       run_cmd "mkdir -p $TRACEFS/instances/$i"
       setup_instance $i
   done
fi