diff options
author | Kai Blin <kai@samba.org> | 2009-10-07 15:45:17 +0200 |
---|---|---|
committer | Kai Blin <kai@samba.org> | 2009-10-07 15:45:46 +0200 |
commit | 37038905965631cbc8f10a2bfe719df0d2ad067e (patch) | |
tree | e76ef4fdde7a17dbae8689e7fd06283dcda0f3d5 /examples/ad-bench/utils.sh | |
parent | 3c3725a340ffe20ab679cf2f9d41ccd0b51b4b3a (diff) | |
download | samba-37038905965631cbc8f10a2bfe719df0d2ad067e.tar.gz samba-37038905965631cbc8f10a2bfe719df0d2ad067e.tar.bz2 samba-37038905965631cbc8f10a2bfe719df0d2ad067e.zip |
AD-Bench: A first go at an Active Directory benchmark.
Diffstat (limited to 'examples/ad-bench/utils.sh')
-rw-r--r-- | examples/ad-bench/utils.sh | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/examples/ad-bench/utils.sh b/examples/ad-bench/utils.sh new file mode 100644 index 0000000000..89da09cb1f --- /dev/null +++ b/examples/ad-bench/utils.sh @@ -0,0 +1,116 @@ +#!/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 +} |