Fix up all use of /dev/stderr for portability to busybox /bin/sh

[ltt-control.git] / lttctl / ltt-armall.sh

diff --git a/lttctl/ltt-armall.sh b/lttctl/ltt-armall.sh
index 1b1c3f9598f8470276c9e08e60ff2d42cf906e76..3bb245b0fc1112c3b51c558401ed54ac1592ba77 100755 (executable)
--- a/lttctl/ltt-armall.sh
+++ b/lttctl/ltt-armall.sh
@@ -17,29 +17,41 @@
 
 DEBUGFSROOT=$(awk '{if ($3 == "debugfs") print $2}' /proc/mounts | head -n 1)
 MARKERSROOT=${DEBUGFSROOT}/ltt/markers
+DEFAULTMODULES="ltt-trace-control ltt-marker-control ltt-tracer ltt-relay ltt-kprobes ltt-userspace-event ltt-statedump ipc-trace kernel-trace mm-trace net-trace fs-trace jbd2-trace syscall-trace trap-trace block-trace rcu-trace"
 
 usage () {
-       echo "Usage: $0 [OPTION]..." > /dev/stderr
-       echo "Connect lttng markers" > /dev/stderr
-       echo "" > /dev/stderr
-       echo "Options:" > /dev/stderr
-       printf "\t-l           Also activate locking markers (high traffic)\n" > /dev/stderr
-       printf "\t-n           Also activate detailed network markers (large size)\n" > /dev/stderr
-       printf "\t-i           Also activate input subsystem events (security implication: records keyboard inputs)\n" > /dev/stderr
-       echo "" > /dev/stderr
-       printf "\t-q           Quiet mode, suppress output\n" > /dev/stderr
-       printf "\t-h           Print this help\n" > /dev/stderr
-       echo "" > /dev/stderr
+       echo "Usage: $0 [OPTION]..." 1>&2
+       echo "Connect lttng markers" 1>&2
+       echo "" 1>&2
+       echo "Options:" 1>&2
+       printf "\t-l           Also activate locking markers (high traffic)\n" 1>&2
+       printf "\t-n           Also activate detailed network markers (large size)\n" 1>&2
+       printf "\t-i           Also activate input subsystem events (security implication: records keyboard inputs)\n" 1>&2
+       echo "" 1>&2
+       printf "\t-q           Quiet mode, suppress output\n" 1>&2
+       printf "\t-h           Print this help\n" 1>&2
+       echo "" 1>&2
 }
 
+if [ "$(id -u)" != "0" ]; then
+       echo "Error: This script needs to be run as root." 1>&2
+       exit 1;
+fi
+
 if [ ! "${DEBUGFSROOT}" ]; then
-       echo "Error: debugfs not mounted" > /dev/stderr
+       echo "Error: debugfs not mounted" 1>&2
        exit 1;
 fi
 
 if [ ! -d "${MARKERSROOT}" ]; then
-       echo "Error: LTT trace controller not found (did you compile and load LTTng?)" > /dev/stderr
-       exit 1;
+       #Try loading the kernel modules first
+       for i in ${DEFAULTMODULES}; do
+               modprobe $i
+       done
+       if [ ! -d "${MARKERSROOT}" ]; then
+               echo "Error: LTT trace controller not found (did you compile and load LTTng?)" 1>&2
+               exit 1;
+       fi
 fi
 
 while getopts "lnqh" options; do
@@ -59,10 +71,14 @@ shift $((${OPTIND} - 1))
 
 if [ ! ${LOCKING} ]; then
        TESTS="${TESTS} -name lockdep -prune -o -name locking -prune -o"
+else
+       modprobe lockdep-trace
 fi
 
 if [ ! ${NETWORK} ]; then
        TESTS="${TESTS} -path '*/net/*_extended' -prune -o"
+else
+       modprobe net-extended-trace
 fi
 
 if [ ! ${INPUT} ]; then