Unify ltt-armall* scripts

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

diff --git a/lttctl/ltt-armall.sh b/lttctl/ltt-armall.sh
index 4c117de1f771dd81e0769bfff66c43594feb41e6..7771505f6d3d494d2cf401191f0060ad3630d1a5 100755 (executable)
--- a/lttctl/ltt-armall.sh
+++ b/lttctl/ltt-armall.sh
@@ -1,21 +1,71 @@
+#!/bin/bash
+# Copyright (C) 2009 Benjamin Poirier
+
+# 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
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License along
+# 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}')
 MARKERSROOT=${DEBUGFSROOT}/ltt/markers
 
-echo Connecting all markers
-
-for c in ${MARKERSROOT}/*; do
-       case ${c} in
-       ${MARKERSROOT}/metadata)
-               ;;
-       ${MARKERSROOT}/locking)
-               ;;
-       ${MARKERSROOT}/lockdep)
-               ;;
-       *)
-               for m in ${c}/*; do
-                       echo Connecting ${m}
-                       echo 1 > ${m}/enable
-               done
-               ;;
+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
+       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
+       exit 1;
+fi
+
+if [ ! -d "$MARKERSROOT" ]; then
+       echo "Error: LTT trace controller not found (did you compile and load LTTng?)" > /dev/stderr
+       exit 1;
+fi
+
+while getopts "lnqh" options; do
+       case $options in
+               l) LOCKING="0";;
+               n) NETWORK="0";;
+               q) QUIET="0";;
+               h) usage;
+                       exit 0;;
+               \?) usage;
+                       exit 1;;
        esac
 done
+shift $(($OPTIND - 1))
+
+
+if [ ! $LOCKING ]; then
+       TESTS="${TESTS} -name lockdep -prune -o -name locking -prune -o"
+fi
+
+if [ ! $NETWORK ]; then
+       TESTS="${TESTS} -path '*/net/*_extended' -prune -o"
+fi
+
+while read -r -d $'\0' 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")