projects / ust.git / blame
| Commit | Line | Data |
|---|---|---|
| eb6adb25 PMF |
1 | #!/bin/bash |
| 2 | ||
| 4a7c7161 PMF |
3 | # usttrace by Pierre-Marc Fournier 2009 |
| 4 | # Distributed under the GPLv2. | |
| 5 | ||
| eb6adb25 PMF |
6 | USTD="./ustd/ustd" |
| 7 | LIBINTERFORK="./libinterfork/.libs/libinterfork.so" | |
| 8 | ||
| 9 | BASE_TRACE_DIR="$HOME/.usttraces" | |
| 10 | ||
| 11 | function usage () { | |
| 12 | echo "usage: $0 COMMAND" 2>/dev/stderr | |
| 13 | } | |
| 14 | ||
| 15 | function error() { | |
| 16 | echo "$0: error: $1" 2>/dev/stderr | |
| 17 | } | |
| 18 | ||
| 19 | # Prepare vars | |
| 20 | CMD=$1 | |
| 21 | ||
| 22 | # Validate input | |
| 23 | if [ -z "$HOME" ]; | |
| 24 | then | |
| 25 | error "no home specified" | |
| 26 | fi | |
| 27 | ||
| 28 | if [ -z "$CMD" ]; | |
| 29 | then | |
| 30 | error "no command specified" | |
| 31 | usage; | |
| 32 | exit 1 | |
| 33 | fi | |
| 34 | ||
| 35 | # Create directory for trace output | |
| 36 | DATESTRING="$(hostname)-$(date +%Y%m%d%H%M%S)" | |
| 37 | OUTDIR="$BASE_TRACE_DIR/$DATESTRING" | |
| 38 | mkdir -p "$OUTDIR" | |
| 39 | ||
| 40 | # Choose socket path | |
| 41 | SOCKPATH="/tmp/ust-sock-$$" | |
| 42 | ||
| 43 | # Start daemon | |
| 44 | $USTD -s "$SOCKPATH" -o "$OUTDIR" >"$OUTDIR/ustd.log" 2>&1 & | |
| 45 | USTDPID=$! | |
| 46 | ||
| 47 | # Establish the environment for the command | |
| 48 | export UST_TRACE=1 | |
| 49 | export UST_AUTOPROBE=1 | |
| 50 | export UST_DAEMON_SOCKET="$SOCKPATH" | |
| 51 | ||
| 52 | # Execute the command | |
| 53 | bash -c "$CMD" | |
| 54 | ||
| 55 | ## Because of the keepalive mechanism, we're sure that by the time | |
| 56 | ## we get here, the daemon is connected to all the buffers that still exist. | |
| 57 | ## Therefore we can politely ask it to die when it's done. | |
| 58 | ||
| 59 | kill -SIGTERM "$USTDPID" | |
| 60 | ||
| 61 | # Tell the daemon to die | |
| 62 | echo "Waiting for ustd to shutdown..." | |
| 63 | wait "$USTDPID" |