| 1 | #!/bin/bash
 | 
|---|
| 2 | # AD-Bench user add/remove benchmark
 | 
|---|
| 3 | #
 | 
|---|
| 4 | # Copyright (C) 2009  Kai Blin  <kai@samba.org>
 | 
|---|
| 5 | #
 | 
|---|
| 6 | # This file is part of AD-Bench, an Active Directory benchmark tool
 | 
|---|
| 7 | #
 | 
|---|
| 8 | # AD-Bench is free software: you can redistribute it and/or modify
 | 
|---|
| 9 | # it under the terms of the GNU General Public License as published by
 | 
|---|
| 10 | # the Free Software Foundation, either version 3 of the License, or
 | 
|---|
| 11 | # (at your option) any later version.
 | 
|---|
| 12 | #
 | 
|---|
| 13 | # AD-Bench is distributed in the hope that it will be useful,
 | 
|---|
| 14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
|---|
| 15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
|---|
| 16 | # GNU General Public License for more details.
 | 
|---|
| 17 | #
 | 
|---|
| 18 | # You should have received a copy of the GNU General Public License
 | 
|---|
| 19 | # along with AD-Bench.  If not, see <http://www.gnu.org/licenses/>.
 | 
|---|
| 20 | 
 | 
|---|
| 21 | ITERATIONS=100
 | 
|---|
| 22 | 
 | 
|---|
| 23 | source `dirname $0`/utils.sh
 | 
|---|
| 24 | 
 | 
|---|
| 25 | PRINCIPAL=$(get_principal $1)
 | 
|---|
| 26 | PASSWORD=$(get_password $1)
 | 
|---|
| 27 | REALM=$(get_realm $1)
 | 
|---|
| 28 | NT_DOM=$(get_nt_dom $1)
 | 
|---|
| 29 | SERVER=$2
 | 
|---|
| 30 | 
 | 
|---|
| 31 | add_user () {
 | 
|---|
| 32 |         USER=$1
 | 
|---|
| 33 |         ${NET} ads user add "${USER}" 'Sup3rS3cr3T!' -k -s $CONFIG_FILE -S ${SERVER} > /dev/null
 | 
|---|
| 34 |         RET=$?
 | 
|---|
| 35 |         if [ $RET -ne 0 ]; then
 | 
|---|
| 36 |                 echo "${NET} ads user add returned error: $RET"
 | 
|---|
| 37 |                 exit 1
 | 
|---|
| 38 |         fi
 | 
|---|
| 39 | }
 | 
|---|
| 40 | 
 | 
|---|
| 41 | del_user () {
 | 
|---|
| 42 |         USER=$1
 | 
|---|
| 43 |         ${NET} ads user delete "${USER}" -k -s $CONFIG_FILE -S ${SERVER} > /dev/null
 | 
|---|
| 44 |         RET=$?
 | 
|---|
| 45 |         if [ $RET -ne 0 ]; then
 | 
|---|
| 46 |                 echo "${NET} returned error: $RET"
 | 
|---|
| 47 |                 exit 1
 | 
|---|
| 48 |         fi
 | 
|---|
| 49 | }
 | 
|---|
| 50 | 
 | 
|---|
| 51 | enum_user () {
 | 
|---|
| 52 |         ${NET} ads user -k -s $CONFIG_FILE -S $SERVER > /dev/null
 | 
|---|
| 53 |         RET=$?
 | 
|---|
| 54 |         if [ $RET -ne 0 ]; then
 | 
|---|
| 55 |                 echo "${NET} returned error: $RET"
 | 
|---|
| 56 |                 exit 1
 | 
|---|
| 57 |         fi
 | 
|---|
| 58 | }
 | 
|---|
| 59 | 
 | 
|---|
| 60 | info_user () {
 | 
|---|
| 61 |         USER=$1
 | 
|---|
| 62 |         ${NET} ads user info "${USER}" -k -s $CONFIG_FILE -S ${SERVER} > /dev/null
 | 
|---|
| 63 |         RET=$?
 | 
|---|
| 64 |         if [ $RET -ne 0 ]; then
 | 
|---|
| 65 |                 echo "${NET} returned error: $RET"
 | 
|---|
| 66 |                 exit 1
 | 
|---|
| 67 |         fi
 | 
|---|
| 68 | }
 | 
|---|
| 69 | 
 | 
|---|
| 70 | set_up () {
 | 
|---|
| 71 |         set_krb_env
 | 
|---|
| 72 |         setup_kinit
 | 
|---|
| 73 |         call_kinit "${PRINCIPAL}" "${PASSWORD}"
 | 
|---|
| 74 |         write_configfile "${REALM}" "${NT_DOM}"
 | 
|---|
| 75 | }
 | 
|---|
| 76 | 
 | 
|---|
| 77 | tear_down () {
 | 
|---|
| 78 |         ${KDESTROY}
 | 
|---|
| 79 |         restore_krb_env
 | 
|---|
| 80 | }
 | 
|---|
| 81 | 
 | 
|---|
| 82 | set_up
 | 
|---|
| 83 | 
 | 
|---|
| 84 | echo -e "\tUSER $SERVER"
 | 
|---|
| 85 | 
 | 
|---|
| 86 | START_TIME=$(start_timer)
 | 
|---|
| 87 | 
 | 
|---|
| 88 | echo -en "\t"
 | 
|---|
| 89 | for i in $( ${SEQ} 1 $ITERATIONS ); do
 | 
|---|
| 90 |         USER=$( echo "ad_test_$(pad_number $i 3)" )
 | 
|---|
| 91 |         add_user $USER
 | 
|---|
| 92 |         echo -n "."
 | 
|---|
| 93 | done
 | 
|---|
| 94 | echo "done"
 | 
|---|
| 95 | 
 | 
|---|
| 96 | enum_user
 | 
|---|
| 97 | 
 | 
|---|
| 98 | # Requires winbind, which requires root perms to start. Skip this for now
 | 
|---|
| 99 | #echo -en "\t"
 | 
|---|
| 100 | #for i in $( ${SEQ} 1 $ITERATIONS ); do
 | 
|---|
| 101 | #       USER=$( echo "ad_test_$(pad_number $i 3)" )
 | 
|---|
| 102 | #       info_user $USER
 | 
|---|
| 103 | #       echo -n "."
 | 
|---|
| 104 | #done
 | 
|---|
| 105 | #echo "done"
 | 
|---|
| 106 | 
 | 
|---|
| 107 | echo -en "\t"
 | 
|---|
| 108 | for i in $( ${SEQ} 1 $ITERATIONS ); do
 | 
|---|
| 109 |         USER=$( echo "ad_test_$(pad_number $i 3)" )
 | 
|---|
| 110 |         del_user $USER
 | 
|---|
| 111 |         echo -n "."
 | 
|---|
| 112 | done
 | 
|---|
| 113 | echo "done"
 | 
|---|
| 114 | 
 | 
|---|
| 115 | STOP_TIME=$(stop_timer)
 | 
|---|
| 116 | 
 | 
|---|
| 117 | TOTAL_TIME=$( total_time $START_TIME $STOP_TIME )
 | 
|---|
| 118 | 
 | 
|---|
| 119 | echo -e "\t\ttotal time:\t\t${TOTAL_TIME}s"
 | 
|---|
| 120 | 
 | 
|---|
| 121 | LOGINS_PER_MINUTE=$(iterations_per_minute $START_TIME $STOP_TIME $ITERATIONS)
 | 
|---|
| 122 | 
 | 
|---|
| 123 | echo -e "\t\titerations/min:\t\t$LOGINS_PER_MINUTE"
 | 
|---|
| 124 | 
 | 
|---|
| 125 | tear_down
 | 
|---|