#!/bin/sh
#
# ngIRCd Test Suite
# Copyright (c)2002-2004 by Alexander Barton (alex@barton.de)
#
# This program 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 2 of the License, or
# (at your option) any later version.
# Please read the file COPYING, README and AUTHORS for more information.
#
# $Id: stress-server.sh,v 1.10 2004/09/04 14:23:09 alex Exp $
#

# detect source directory
[ -z "$srcdir" ] && srcdir=`dirname $0`

# parse command line
[ "$1" -gt 0 ] 2> /dev/null && CLIENTS="$1" || CLIENTS=5
[ "$2" -gt 0 ] 2> /dev/null && LOOPS="$2" || LOOPS=1

# get our name
name=`basename $0`

# create directories
mkdir -p logs tests

# test for required external tools
type expect > /dev/null 2>&1
if [ $? -ne 0 ]; then
  echo "      ${name}: \"expect\" not found.";  exit 77
fi
type telnet > /dev/null 2>&1
if [ $? -ne 0 ]; then
  echo "      ${name}: \"telnet\" not found.";  exit 77
fi

# hello world! :-)
[ $LOOPS -gt 1 ] \
  && echo "      stressing server with $CLIENTS clients in $LOOPS loops (be patient!):" \
  || echo "      stressing server with $CLIENTS clients (be patient!):"

# create scripts for expect(1)
no=0
while [ ${no} -lt $CLIENTS ]; do
  cat ${srcdir}/stress-A.e > tests/${no}.e
  echo "send \"nick test${no}\\r\"" >> tests/${no}.e
  cat ${srcdir}/stress-B.e >> tests/${no}.e
  no=`expr ${no} + 1`
done

# main loop ...
loop=0
while [ ${loop} -lt $LOOPS ]; do
  no=0
  loop=`expr ${loop} + 1`
  while [ ${no} -lt $CLIENTS ]; do
    expect tests/${no}.e > logs/stress-${no}.log 2> /dev/null &
    no=`expr ${no} + 1`
  done
  if [ $LOOPS -gt 1 ]; then
    echo "      loop $loop/$LOOPS: started $no clients."
    echo -n "      loop $loop/$LOOPS: waiting for clients to complete: ."
  else
    echo "      started $no clients."
    echo -n "      waiting for clients to complete: ."
  fi

  res=3
  touch logs/check-idle.log
  while true; do
    expect ${srcdir}/check-idle.e >> logs/check-idle.log; res=$?
    echo "====================" >> logs/check-idle.log
    [ $res -ne 99 ] && break

    # there are still clients connected. Wait ...
    sleep 3
    echo -n "."
  done

  if [ $res -ne 0 ]; then
    echo " ERROR!"
    break
  fi
  echo " done."
done

exit $res

# -eof-