#!/bin/sh

# Copyright 2018 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 wpa_supplicant


WPA_LEVEL_LIST="excessive msgdump debug info warning error"

usage() {
  echo "
Usage: wpa_debug [level]|[--reset]|[--help][--list_valid_levels]

  wpa_debug sets the debug level of wpa_supplicant.
  Current debug level is displayed if no parameters are provided

  level: The level is the level we want to set the debugging level to. The valid
         levels can be viewed by using the --list_valid_levels flag

         eg: wpa_debug msgdump
           Sets the wpa_supplicant logging level to msgdump

  --reset : Resets the level to 'info'

  --help : Displays this output

  --list_valid_levels: Displays the valid levels wpa_supplicant can be set to
"
}

CMD_FLAG="<<cmd>>"

WPA_CMD="dbus-send --system --dest=fi.w1.wpa_supplicant1 --print-reply /fi/w1/wpa_supplicant1 org.freedesktop.DBus.Properties.${CMD_FLAG} string:fi.w1.wpa_supplicant1 string:DebugLevel"

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

get_wpa_logging() {
  set_cmd="$(echo "${WPA_CMD}" | sed "s/${CMD_FLAG}/Get/")"
  ${set_cmd} | sed -e '/string/!d; s/[[:space:]]\+/ /g' | cut -d "\"" -f 2
}

set_wpa_logging() {
  if ! is_valid_tag "${WPA_LEVEL_LIST}" "$1"; then
    return 1
  fi

  if [ "$1" = "$(get_wpa_logging)" ]; then
    return 1
  fi

  set_cmd="$(echo "${WPA_CMD}" | sed "s/${CMD_FLAG}/Set/") variant:string:$1"
  ${set_cmd}
}


if [ $# -gt 0 ]; then
  for param in "$@"; do
    case $param in
    --reset)
      set_wpa_logging "info"
      ;;
    --list*)
      echo "Valid levels are: $(echo "${WPA_LEVEL_LIST}" | sed 's/ /, /g')"
      exit 0
      ;;
    --help|--*)
      usage
      exit 0
      ;;
    *)
      old_level="$(get_wpa_logging)"
      set_wpa_logging "$param"
      echo "Old wpa level: ${old_level}"
      ;;
    esac
  done
fi

echo "Current wpa level: $(get_wpa_logging)"