source: trunk/server/examples/ad-bench/utils.sh

#!/bin/bash
# AD-Bench utility functions
#
# Copyright (C) 2009  Kai Blin  <kai@samba.org>
#
# This file is part of AD-Bench, an Active Directory benchmark tool
#
# AD-Bench 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 3 of the License, or
# (at your option) any later version.
#
# AD-Bench 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 AD-Bench.  If not, see <http://www.gnu.org/licenses/>.

source `dirname $0`/settings.sh

start_timer () {
        START_TIME=$( ${DATE} ${DATE_FORMATSTR} )
        echo "$START_TIME"
}

stop_timer () {
        STOP_TIME=$( ${DATE} ${DATE_FORMATSTR} )
        echo "$STOP_TIME"
}

total_time () {
        START_TIME=$1
        END_TIME=$2
        TOTAL_TIME=$( echo "scale=9;$STOP_TIME - $START_TIME" | ${BC} )
        echo "$TOTAL_TIME"
}

iterations_per_minute () {
        START_TIME=$1
        STOP_TIME=$2
        TOTAL_RUNS=$3

        TOTAL_TIME=$( total_time $START_TIME $STOP_TIME )

        ITER_PER_MIN=$( echo "scale=2; 60 * $TOTAL_RUNS / $TOTAL_TIME" | ${BC} )
        echo "$ITER_PER_MIN"
}

get_principal () {
        PRINCIPAL=$( echo $1 | ${SED} -e "s/\(.*\)%.*/\1/" )
        echo "$PRINCIPAL"
}

get_password () {
        PASSWORD=$( echo $1 | ${SED} -e "s/.*%\(.*\)/\1/" )
        echo "$PASSWORD"
}

get_realm () {
        REALM=$( echo $1 | ${SED} -e "s/.*@\(.*\)%.*/\1/" )
        echo "$REALM"
}

get_nt_dom () {
        NT_DOM=$( echo $1 | ${SED} -e "s/.*@\([A-Z1-9-]*\)\..*/\1/" )
        echo "$NT_DOM"
}

set_krb_env () {
        OLD_KRB5CCNAME="${KRB5CCNAME}"
        KRB5CCNAME="${NEW_KRB5CCNAME}"
        export KRB5CCNAME
}

restore_krb_env () {
        KRB5CCNAME="${OLD_KRB5CCNAME}"
        export KRB5CCNAME
}

setup_kinit () {
        ${KINIT} --invalid 2>&1 | grep -q "password-file"
        if [ $? -eq 0 ]; then
                KINIT="${KINIT} ${KINIT_PARAM_OLD}"
        else
                KINIT="${KINIT} ${KINIT_PARAM_NEW}"
        fi
}

write_configfile () {
        REALM=$1
        NT_DOM=$2
        echo -e "[global]" > $CONFIG_FILE
        echo -e "\trealm = $REALM" >> $CONFIG_FILE
        echo -e "\tworkgroup = $NT_DOM" >> $CONFIG_FILE
        echo -e "\tsecurity = ADS" >> $CONFIG_FILE
}

call_kinit () {
        PRINCIPAL=$1
        PASSWORD=$2
        echo "${PASSWORD}" | ${KINIT} ${PRINCIPAL} > /dev/null
        RET=$?
        if [ $RET -ne 0 ]; then
                echo "kinit returned an error: $RET"
                exit 1
        fi
}

pad_number () {
        NUMBER=$1
        DIGITS=$2
        PADDED_NO=$(printf "%0${DIGITS}d" $NUMBER)
        echo $PADDED_NO
}