lttngtrace update todo

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

diff --git a/lttctl/ltt-armall.sh b/lttctl/ltt-armall.sh
index 7771505f6d3d494d2cf401191f0060ad3630d1a5..d2548b86a610ce193656bc419dac0ef87dd6978b 100755 (executable)
--- a/lttctl/ltt-armall.sh
+++ b/lttctl/ltt-armall.sh
@@ -1,5 +1,5 @@
-#!/bin/bash
 # Copyright (C) 2009 Benjamin Poirier
+# Copyright (C) 2010 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
 
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -15,8 +15,9 @@
 # with this program; if not, write to the Free Software Foundation, Inc.,
 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 
-DEBUGFSROOT=$(grep ^debugfs /proc/mounts | head -1 | awk '{print $2}')
+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
@@ -25,47 +26,68 @@ usage () {
        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
 }
 
-if [ ! "$DEBUGFSROOT" ]; then
-       echo "Error: debugfs not mounted" > /dev/stderr
+if [ "$(id -u)" != "0" ]; then
+       echo "Error: This script needs to be run as root." > /dev/stderr
        exit 1;
 fi
 
-if [ ! -d "$MARKERSROOT" ]; then
-       echo "Error: LTT trace controller not found (did you compile and load LTTng?)" > /dev/stderr
+if [ ! "${DEBUGFSROOT}" ]; then
+       echo "Error: debugfs not mounted" > /dev/stderr
        exit 1;
 fi
 
+if [ ! -d "${MARKERSROOT}" ]; then
+       #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?)" > /dev/stderr
+               exit 1;
+       fi
+fi
+
 while getopts "lnqh" options; do
-       case $options in
+       case ${options} in
                l) LOCKING="0";;
                n) NETWORK="0";;
                q) QUIET="0";;
+               i) INPUT="0";;
                h) usage;
                        exit 0;;
                \?) usage;
                        exit 1;;
        esac
 done
-shift $(($OPTIND - 1))
+shift $((${OPTIND} - 1))
 
 
-if [ ! $LOCKING ]; then
+if [ ! ${LOCKING} ]; then
        TESTS="${TESTS} -name lockdep -prune -o -name locking -prune -o"
+else
+       modprobe lockdep-trace
 fi
 
-if [ ! $NETWORK ]; then
+if [ ! ${NETWORK} ]; then
        TESTS="${TESTS} -path '*/net/*_extended' -prune -o"
+else
+       modprobe net-extended-trace
+fi
+
+if [ ! ${INPUT} ]; then
+       TESTS="${TESTS} -name input -prune -o"
 fi
 
-while read -r -d $'\0' marker; do
-       if [ ! $QUIET ]; then
+(eval "find '${MARKERSROOT}' ${TESTS} -name metadata -prune -o -name enable -print") | while read -r marker; do
+       if [ ! ${QUIET} ]; then
                echo "Connecting ${marker%/enable}"
        fi
-       echo 1 > $marker
-done < <(eval "find '$MARKERSROOT' $TESTS -name metadata -prune -o -name enable -print0")
+       echo 1 > ${marker}
+done