#!/bin/sh
# Copyright (C) 2005 Krzysztof Kozlowski
#
# Testowanie wydajnosci NAT-a/mostka
# Zasysanie FTP do ramdysku
# Uzycie:
#  - bez parametrow - wykonanie testow
#  - test_ftp -d - skasowanie ramdysku


MD=10
DIR="/var/tmp/test_kozlowk3"
URL="ftp://ftp.iem.pw.edu.pl/pub/koziols_rescue_cd/"
FILE="koziols_rescue_cd_20050427_iso.tgz"

#LIMIT_WGET="0.5m"
#LIMIT_WGET="1m"
#LIMIT_WGET="2m"
#LIMIT_WGET="4m"
#LIMIT_WGET="8m"

del_ramdisk() {
  umount "$DIR"
  mdconfig -d -u $MD
  rm -fr "$DIR"
  echo "Ramdysk usuniety..."
}

if [ $# -eq 1 ]; then
  if [ "$1" = "-d" ]; then
    del_ramdisk
    exit
  fi
fi
    
make_ramdisk() {
  STATUS=0
  mdconfig -a -t malloc -s 256m -o noreadonly -u $MD
  STATUS=`expr $STATUS + $?`

  newfs -U /dev/md${MD}
  STATUS=`expr $STATUS + $?`
  mkdir "$DIR"
  STATUS=`expr $STATUS + $?`
  mount /dev/md${MD} "$DIR"
  STATUS=`expr $STATUS + $?`
  chmod 1777 "$DIR"
  STATUS=`expr $STATUS + $?`

  if [ $STATUS -ne 0 ]; then
    echo "Blad podczas tworzenia ramdysku (/dev/md${MD})..."
    mdconfig -l
    exit
  fi
  echo "Ramdysk utworzony w $DIR..."
}


echo "Pobieram duzy plik FTP-em..."
for i in 1 2 3 4 5 6 7
do
  if [ "${LIMIT_WGET}" = "" ]; then
    LIMIT_RATE=""
  else
    LIMIT_RATE="--limit-rate=${LIMIT_WGET}"
  fi
  make_ramdisk
  cd "$DIR"
  wget ${LIMIT_RATE} ${URL}${FILE}
  rm -fr ${FILE}
  cd /
  del_ramdisk
done


#cd /
#del_ramdisk