X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=lttctl%2Fltt-armall.sh;h=d73cdbaaf349f94bd5996ad681f5a2bae8f44a50;hb=a4dc603fbbc706fbad0b6dcf4b2b56239494e45b;hp=f0824c819eebab7a9656b2ebf69714a10b5b46c1;hpb=5e1fd42a8d1bbbe1e1283e4c652994ff738dd6a7;p=ltt-control.git

diff --git a/lttctl/ltt-armall.sh b/lttctl/ltt-armall.sh
index f0824c8..d73cdba 100755
--- a/lttctl/ltt-armall.sh
+++ b/lttctl/ltt-armall.sh
@@ -1,47 +1,76 @@
-#excluding core markers (already connected)
-#excluding locking markers (high traffic)
+# Copyright (C) 2009 Benjamin Poirier
 
-DEBUGFSROOT=$(grep ^debugfs /proc/mounts | head -1 | awk '{print $2}')
+# 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 -n 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
+	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
+	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";;
+		i) INPUT="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
 
-# Connect the interesting high-speed markers to the marker tap.
-# Markers starting with "tap_" are considered high-speed.
-#echo Connecting high-rate markers to tap
-#MARKERS=`cat /proc/ltt|grep -v %k|awk '{print $2}'|sort -u |grep ^tap_`
-#
-##Uncomment the following to also record lockdep events.
-##MARKERS=`cat /proc/ltt|grep -v %k|awk '{print $2}'|sort -u|grep -e ^tap_ -e ^lockdep`
-#
-#IFS=${N}
-#for a in $MARKERS; do
-#	echo Connecting $a
-#
-#	#redirect markers carrying state information to dedicated channels
-#	case $a in
-#	*)
-#		CHANNEL=
-#		;;
-#	esac
-#
-#	echo "connect $a ltt_tap_marker" > /proc/ltt
-#done
+if [ ! ${INPUT} ]; then
+	TESTS="${TESTS} -name input -prune -o"
+fi
+
+(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