#!/bin/bashTARGET="$1"function everythingelse() {#Verify that one parameter was passedif [ -z "$TARGET" ]; then echo "Syntax Error: must specify target directory" exit 1fi#Verify that it is a valid directoryif [ ! -d "$TARGET" ]; then echo "ERROR: Invalid directory ${TARGET}" exit 1fi#Verify that it is a writeable directoryif [ ! -w "$TARGET" ]; then echo "ERROR: You do not have write permissions to ${TARGET}" exit 1fi#Verify that it is a readable directoryif [ ! -r "$TARGET" ]; then echo "ERROR: You do not have read permissions to ${TARGET}" exit 1fi#Verify that is an executable directoryif [ ! -x "$TARGET" ]; then echo "ERROR: You do not have execute permissions to ${TARGET}" exit 1fiif [ -z $(which bonnie++) ]; then echo "ERROR: Could not find bonnie++ in PATH, please install bonnie++" exit 1fi#Verify that it is a valid fstab entryIN_FSTAB=$(egrep ".*[ \\t]*${TARGET}[ \\t]*.*" /etc/fstab | tail -n1 | wc -l)if [ $IN_FSTAB -eq 0 ]; then echo "ERROR: Could not find mountpoint ${TARGET}" exit 1fi#Verify that it is mountedIN_MTAB=$(egrep ".*[ \\t]*${TARGET}[ \\t]*.*" /etc/mtab | tail -n1 | wc -l)if [ $IN_MTAB -eq 0 ]; then #Attempt to mount the directory echo "${TARGET} is not mounted, attempting to mount" mount "${TARGET}" >/dev/null 2>/dev/null if [ $? -ne 0 ]; then echo "ERROR: could not mount ${TARGET}" exit 1 fifi#Grab the device information from fstabFSTAB_LINE=$(egrep ".*[ \\t]*${TARGET}[ \\t]*.*" /etc/fstab | tail -n1)DEVICE=$(echo "${FSTAB_LINE}" | awk '{print $1}')FSTYPE=$(echo "${FSTAB_LINE}" | awk '{print $3}')#Confirm the testecho -n "Start bonnie benchmark on ${DEVICE}? (y/N) "read answerif [ "$answer" != "y" ] && [ "$answer" != "Y" ]; then echo "Cancelled by user" exit 1fi#Unmount the directoryecho -n "Unmounting filesystem on ${DEVICE}........"umount ${TARGET} >/dev/null 2>/dev/nullif [ $? -ne 0 ]; then echo "ERROR: Could not unmount ${TARGET}" exit 1fiecho "OK"if [ "$FSTYPE" = "ext3" ]; then #Tune2fs the filesystem echo -n "Setting dir_index on ${DEVICE}........" tune2fs -O dir_index ${DEVICE} >/dev/null 2>/dev/null if [ $? -ne 0 ]; then echo "ERROR: Could not set dir_index on filesystem" exit 1 fi echo "OK" #Optimize directories echo -n "Optimizing directories on ${DEVICE}........" fsck.ext3 -n -D ${DEVICE} >/dev/null 2>/dev/null if [ $? -ne 0 ]; then echo "Could not optimize directories on ${DEVICE}, please run fsck on the filesystem to fix any errors" exit 1 fi echo "OK"fi#Get a temp dir name and make itMYTMPDIR=/tmp/bonnie_bench_$(date +%s%N | sha1sum | awk '{print $1}')mkdir $MYTMPDIR || (echo "ERROR: Could not make tmp directory ${MYTMPDIR}" && exit 1)#Logic to set readaheadIS_MD=$(echo ${DEVICE} | egrep -c 'md[0-9]')IS_HDD=$(echo ${DEVICE} | egrep -c '[sh]d[a-z][0-9]')RA_DEVS=""if [ $IS_MD -eq 1 ]; then RA_DEVS=$(mdadm -D ${DEVICE} | grep active | egrep -o '[sh]d[a-z]' | sort -u)elif [ $IS_HDD -eq 1 ]; then RA_DEVS=$(echo ${DEVICE} | egrep -o '[sh]d[a-z]')else echo "ERROR: Unknown device type: ${DEVICE}" exit 1fifor RAVALUE in 16384 32768 65536; do echo "========================================" echo "Beginning test for read-ahead value: ${RAVALUE}" #Set the readahead for any applicable devices for RA_DEV in ${RA_DEVS}; do blockdev --setra ${RAVALUE} /dev/${RA_DEV} >/dev/null 2>/dev/null if [ $? -ne 0 ]; then echo "ERROR: Could not set readahead on ${RA_DEV}" exit 1 fi done #Dump all caches if [ -f /proc/sys/vm/drop_caches ]; then echo -n "Dumping system cache......." (sync && echo "3">/proc/sys/vm/drop_caches) if [ $? -ne 0 ]; then echo "ERROR: Could not drop caches" exit 1 fi echo "OK" fi #Mount the filesystem mount ${TARGET} >/dev/null 2>/dev/null if [ $? -ne 0 ]; then echo "ERROR: Could not mount fs on ${DEVICE}" exit 1 fi #Actually run bonnie bonnie++ -d ${TARGET} -n 128 -u 0 -r $(free -m | grep ^Mem | awk '{print $2}') -s $(echo 2*$(free -m | grep ^Mem |awk '{print $2}')|bc) -f -b > ${MYTMPDIR}/run.${RAVALUE}.out #Unmount the filesystem umount ${TARGET} >/dev/null 2>/dev/null if [ $? -ne 0 ]; then echo "ERROR: Could not unmount fs on ${DEVICE}" exit 1 fidone#Write the data necessary to make the graphsPLOTFILE_RO="${MYTMPDIR}/plotdata_ro"PLOTFILE_BO="${MYTMPDIR}/plotdata_bo"PLOTFILE_BI="${MYTMPDIR}/plotdata_bi"echo "#x rewrite_output" > ${PLOTFILE_RO}echo "#x block_output" > ${PLOTFILE_BO}echo "#x block_input" > ${PLOTFILE_BI}X=23for RAVALUE in 16384 32768 65536; do RESULT_LINE=$(tail -n1 "${MYTMPDIR}/run.${RAVALUE}.out") BLOCK_OUTPUT=$(echo "scale=2;$(echo \"${RESULT_LINE}\"|cut -d, -f5)/1000" | bc) RW_OUTPUT=$(echo "scale=2;$(echo \"${RESULT_LINE}\"|cut -d, -f7)/1000" | bc) BLOCK_INPUT=$(echo "scale=2;$(echo \"${RESULT_LINE}\"|cut -d, -f11)/1000" | bc) echo "${X} ${BLOCK_OUTPUT}" >> "${PLOTFILE_BO}" X=$(($X + 2)) echo "${X} ${BLOCK_INPUT}" >> "${PLOTFILE_BI}" X=$(($X + 2)) echo "${X} ${RW_OUTPUT}" >> "${PLOTFILE_RO}" X=$(($X + 2)) X=$(($X + 19))done#Make the output graphPLOT_IMG="bonnie_out.png"gnuplot << EOFset terminal pngset output "${PLOT_IMG}"set title "Bonnie Output Benchmark on $(hostname):${TARGET}"set xlabel "Disk readahead in sectors"set ylabel "Speed in M/sec"set xtics ( "16384" 25, "32768" 50, "65536" 75 )set xrange [0:100]set xtics nomirrorset gridset boxwidth 1set style fillplot \"${PLOTFILE_BO}" title "Block Output" with boxes fs solid,\"${PLOTFILE_RO}" title "Rewrite Output" with boxes fs solidEOF#Make the input graphPLOT_IMG="bonnie_in.png"gnuplot << EOFset terminal pngset output "${PLOT_IMG}"set title "Bonnie Input Benchmark on $(hostname):${TARGET}"set xlabel "Disk readahead in sectors"set ylabel "Speed in M/sec"set xtics ( "16384" 25, "32768" 50, "65536" 75 )set xrange [0:100]set xtics nomirrorset boxwidth 1set style fillplot \"${PLOTFILE_BI}" title "Block Input" with boxes fs solidEOF#Write the data necessary for text column outputCOLFILE="${MYTMPDIR}/coldata"echo "" > "${COLFILE}"for RAVALUE in 16384 32768 65536; do RESULT_LINE=$(tail -n1 "${MYTMPDIR}/run.${RAVALUE}.out") BLOCK_OUTPUT=$(echo "scale=2;$(echo \"${RESULT_LINE}\"|cut -d, -f5)/1000" | bc) RW_OUTPUT=$(echo "scale=2;$(echo \"${RESULT_LINE}\"|cut -d, -f7)/1000" | bc) BLOCK_INPUT=$(echo "scale=2;$(echo \"${RESULT_LINE}\"|cut -d, -f11)/1000" | bc) echo "${RAVALUE}|${BLOCK_OUTPUT}|${RW_OUTPUT}|${BLOCK_INPUT}" >> ${COLFILE}done#Make a column text output and print it while also saving it to bonnie.out( echo "Readahead|Block Output|Rewrite Output|Block Input" echo "==|==|==|==" cat "${COLFILE}") | column -s '|' -t | tee bonnie.outrm -fr "${MYTMPDIR}"