# Copyright (c) 2012 NTT DOCOMO, INC. 
# Copyright 2012 Hewlett-Packard Development Company, L.P.
#
# All Rights Reserved.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

function get_kernel_parameter() {
    local name=$1
    for i in `cat /proc/cmdline`; do
        case "$i" in
        ${name}=*)
            echo "${i#${name}=}"
            return 0
            ;;
        ${name})
            echo ""
            return 0
            ;;
        *)
            ;;
        esac
    done
    echo ""
    return 1
}

function string_contains() {
  local string=$1
  local word=$2
  if [ "$string" != "${string/$word/}" ]; then
    return 0
  else
    return 1
  fi
}

function load_modules_by_udev() {
    udevadm trigger --action=add
    udevadm settle
}

function strip_dev() {
    echo "$1" | sed -e 's:^/dev/::'
}

function prepend_dev() {
    case "$1" in
    /dev/*)
        echo "$1"
        ;;
    *)
        echo "/dev/$1"
        ;;
    esac
}

function whole_disk_name() {
    local dev=`prepend_dev "$1"`
    case "$dev" in
    /dev/sd*|/dev/hd*|/dev/vd*)
        echo "$dev" | sed -e 's/[0-9]*$//'
        ;;
    /dev/cciss/*)
        echo "$dev" | sed -e 's/p[0-9]*$//'
        ;;
    *)
        echo ""
        return 1
        ;;
    esac
}

function partition_name() {
    local dev=`prepend_dev "$1"`
    local part=$2
    case "$dev" in
    /dev/sd*|/dev/hd*|/dev/vd*)
        echo "${dev}${part}"
        ;;
    /dev/cciss/*)
        echo "${dev}p${part}"
        ;;
    *)
        echo ""
        return 1
        ;;
    esac
}

function find_interface() {
    local mac=$1
    eth=`ifconfig -a | grep -i "$mac" | awk {' print $1 '} | head -n 1`
    if [ -n "$eth" ]; then
        echo "$eth"
        return 0
    else
        return 1
    fi
}

function set_mac() {
    local dev=$1
    local mac=$2
    ip link set "$dev" address "$mac"
}

function swap_ifname() {
    local dev=$1
    local dev2=$2
    if [ "$dev" = "$dev2" ]; then
        return
    fi
    if ip link show "$dev2" >/dev/null; then
        # dev2 exists
        # swap device name
        ip link set "$dev" name "_$dev"
        ip link set "$dev2" name "$dev"
        ip link set "_$dev" name "$dev2"
    else
        ip link set "$dev" name "$dev2"
    fi
}

function partition_exists() {
    local dev=$1
    dev=`strip_dev "$dev"`
    if tail -n +3 /proc/partitions | grep "$dev" >/dev/null; then
        return 0
    else
        return 1
    fi
}

function find_disk() {
    local disks=$1
    local dev

    # find device
    local OLD_IFS=$IFS
    IFS=,
    for i in $disks; do
        dev=`whole_disk_name "$i"`
        if partition_exists "$dev"; then
            dev=`prepend_dev "$dev"`
            break
        fi
    dev=""
    done
    IFS=$OLD_IFS

    if [ -z "$dev" ]; then
        return 1
    fi
    echo "$dev"
    return 0
}

function check_tgtd_socket() {
    echo -n "waiting for tgtd socket..."
    if [ -e /var/run/tgtd.ipc_abstract_namespace.0 ]; then
        echo "found"
        return 0
    else
        echo "not found"
        return 1
    fi
}

wait_for(){
    LOOPS=$1
    SLEEPTIME=$2
    shift ; shift
    i=0
    while [ $i -lt $LOOPS ] ; do
        i=$((i + 1))
        eval "$@" && return 0 || true
        sleep $SLEEPTIME
    done
    return 1
}

function start_iscsi_target() {
    local iqn=$1
    local dev=$2
    local cli=$3

    # used by tgtd
    mkdir -p /var/run

    tgtd
    wait_for 10 0.5 check_tgtd_socket

    tgtadm --lld iscsi --mode target --op new --tid 1 --targetname "$iqn"
    tgtadm --lld iscsi --mode logicalunit --op new --tid 1 --lun 1 --backing-store "$dev"
    tgtadm --lld iscsi --mode target --op bind --tid 1 --initiator-address "$cli"
}

function stop_iscsi_target() {
    tgtadm --lld iscsi --mode logicalunit --op delete --tid 1 --lun 1
    tgtadm --lld iscsi --mode target --op delete --tid 1
    killall tgtd
}

function troubleshoot() {
    if [ "$TROUBLESHOOT" != 1 ]; then
        _DO_TROUBLESHOOT=""
        _t=0
        echo -n "Troubleshooting required, press t to launch shell."
        while [ $_t -lt 10 ]; do
            read -n 1 -t 1 _DO_TROUBLESHOOT
            _t=$(($_t + 1))
            if [ "$_DO_TROUBLESHOOT" == "t" ]; then
                export TROUBLESHOOT=1
                break
            fi
            echo -n "."
        done
        echo ""
    fi
    if [ "$TROUBLESHOOT" == 1 ]; then
        echo "Starting troubleshooting shell."
        bash
    fi
}

function safe_url_encode() {
    local str=$1
    local out=""

    for (( i=0; i<${#str}; i++ )); do
        c=${str:$i:1}
        case $c in
        [a-zA-Z0-9.-_] )
            out+="$c"
            ;;
        ' ' )
            out+="+"
            ;;
        *)
            #skip it
            ;;
        esac
    done
    echo "$out"

}

function err_msg() {
    message=$1
    if [ -z "$FIRST_ERR_MSG" ]; then
        FIRST_ERR_MSG=$(safe_url_encode "$message")
    fi
    echo "$message"
}

