bin/ff_debug

#!/bin/bash
#
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
#
# Script that turns on useful logging for flimflam

FLIMFLAM="org.chromium.flimflam"
MANAGER="${FLIMFLAM}.Manager"
SEND_FF_CMD="dbus-send --system --dest=${FLIMFLAM} --print-reply / $MANAGER"

FF_TAGLIST=`${SEND_FF_CMD}.ListDebugTags | awk '/string/ {print substr($2, 2, length($2)-2);}' | tr "+" " "`

usage(){
  echo "
Usage: ff_debug [<tag_expression>]|[--reset]|[--help]|[--list_valid_tags]|[--level <level>]

  ff_debug adds and removes debug tags for flimflam.
  Current debug settings are displayed if no parameters are provided

  <tag_expression> is defined in boolean notation using <debug_tag> separated
    by an operator [+-], where + and - imply adding and removing of the tag immediately
    following the operator. An expression beginning with either operators [+-]
    takes the existing tags and modifies them appropriately.  Otherwise, the existing tags
    are replaced by those specified in the command.

    <debug_tag> can be listed using the --list_valid_tags

    eg: ff_debug network+wifi
      Sets debug tags to network and wifi
    eg: ff_debug +network-service
      Adds network and removes service tags from the existing debug settings

  --list_valid_tags : Displays all valid tags

  --level: Displays or sets current debug level for logging

  --reset : Removes all tagging

  --help : Displays this output
"
}

get_ff_debug_tags() {
  ${SEND_FF_CMD}.GetDebugTags 2>/dev/null | awk '/string/ {print substr($2,2,length($2)-2);}'
}

set_ff_debug_tags() {
  ${SEND_FF_CMD}.SetDebugTags string:"$1" >& /dev/null
}

get_ff_debug_level() {
  ${SEND_FF_CMD}.GetDebugLevel 2>/dev/null | awk '/int32/ {print $2}'
}

set_ff_debug_level() {
  ${SEND_FF_CMD}.SetDebugLevel int32:"$1" >& /dev/null
}

is_different(){
  list1=`echo $1 | tr "[ +-]" "\n" | sort | tr "\n" " "`
  list2=`echo $2 | tr "[ +-]" "\n" | sort | tr "\n" " "`

  [[$list == "$list2"]]
}

starting_ff_tags=`get_ff_debug_tags`
# Global strings
ff_tags_to_set=$starting_ff_tags

# Returns whether ot not $2 exists in $1 where $1 is a space
# separated list of tags
is_valid_tag(){
  expr match " $1 " ".* $2 .*" > /dev/null
}

# Takes a boolean expression and changes
# flimflam string appropriately
modify_flimflam_tag_string(){
  if [ -n "$1" ]; then
    for tag in `echo "$1" | sed 's/[+-]/ &/g'`; do

      # Skip the tag if it's not in the list of valid tags
      if ! is_valid_tag "$FF_TAGLIST" "`echo $tag | sed 's/^[+-]//'`"; then
        continue
      fi

      case $tag in
        -*)
          tag=`echo $tag | sed 's/-//'`
          if [ -n "$tag" ]; then
            # First check/remove instances where it lies between +'s
            # Then check/remove instances where it lies at the ends
            ff_tags_to_set=`echo $ff_tags_to_set \
                            | sed 's/+'$tag'+/+/g' \
                            | sed 's/\(^\|+\)'$tag'\(+\|$\)//g'`
          fi
          ;;
        +*)
          tag=`echo $tag | sed 's/+//'`
          if [ -z "$tag" ]; then
            continue
          # Check if exact tag is between + symbols, or at the ends of expression
          elif [ -n "`echo "$ff_tags_to_set" | egrep "\+$tag\+|\+$tag$|^$tag\+"`" ]; then
            continue
          elif [ -n "$ff_tags_to_set" ]; then
            ff_tags_to_set="${ff_tags_to_set}+${tag}"
          else
            ff_tags_to_set="$tag"
          fi
          ;;
        *)
          ff_tags_to_set="$tag"
          ;;
      esac
    done
    ff_tags_to_set=`echo "$ff_tags_to_set" | sed 's/^+//'`
  else
    ff_tags_to_set=""
  fi
}

get_or_set_debug_level() {
  local ff_debug_level=`get_ff_debug_level`
  if [ -z "$ff_debug_level" ]; then
    # flimflam does not implement GetDebugLevel / SetDebugLevel, simply return
    return
  fi

  if [ $# -gt 0 ]; then
    echo "Old flimflam debug level: $ff_debug_level"
    set_ff_debug_level "$1"
    ff_debug_level=`get_ff_debug_level`
  fi
  echo "Current flimflam debug level: $ff_debug_level"
}

if [ $# -gt 0 ]; then
  while [ $# -gt 0 ]; do
    case "$1" in
    --reset)
      ff_tags_to_set=""
      break
      ;;
    --level)
      shift  # move forward to the <level> argument if specified
      get_or_set_debug_level "$@"
      exit 0
      ;;
    --list*)
      echo "Valid Tags: [`echo $FF_TAGLIST | sed 's/ /, /g'`]"
      exit 0
      ;;
    --help|--*)
      usage
      exit 0
      ;;
    *)
      modify_flimflam_tag_string "$1"
      ;;
    esac
    shift
  done

  # set tags only when the starting and ending are different
  if [ "$ff_tags_to_set" != "$starting_ff_tags" ]; then
    set_ff_debug_tags "$ff_tags_to_set"
  fi
  echo "Old flimflam tags: [$starting_ff_tags]"
fi
echo "Current flimflam tags: [`get_ff_debug_tags`]"