tools/tools/fetchbench/fetchbench

   1 #!/bin/sh
   2 #-
   3 # Copyright (c) 2017 Edward Tomasz Napierala <trasz@FreeBSD.org>
   4 #
   5 # This software was developed by SRI International and the University of
   6 # Cambridge Computer Laboratory under DARPA/AFRL contract (FA8750-10-C-0237)
   7 # ("CTSRD"), as part of the DARPA CRASH research programme.
   8 #
   9 # Redistribution and use in source and binary forms, with or without
  10 # modification, are permitted provided that the following conditions
  11 # are met:
  12 # 1. Redistributions of source code must retain the above copyright
  13 #    notice, this list of conditions and the following disclaimer.
  14 # 2. Redistributions in binary form must reproduce the above copyright
  15 #    notice, this list of conditions and the following disclaimer in the
  16 #    documentation and/or other materials provided with the distribution.
  17 #
  18 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  19 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21 # ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  22 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  27 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  28 # SUCH DAMAGE.
  29 #
  30 # $FreeBSD$
  31
  32 # This is a simple HTTP benchmark.  It works by running a number of fetch(1)
  33 # instances in parallel, 10 by default, each performing a number of fetches,
  34 # 100 by default, and then printing out the time it took.
  35 #
  36 # Example usage: ./fetchbench -i 2 -n 5 http://freebsd.org
  37
  38 usage()
  39 {
  40         cat <<EOF 1>&2
  41 usage: fetchbench [-i number-of-instances] [-n fetches-per-instance] url
  42 EOF
  43         exit 1
  44 }
  45
  46 NPROC=10
  47 NFETCH=100
  48
  49 while getopts "i:n:X" opt; do
  50         case "$opt" in
  51         i)      NPROC="${OPTARG}" ;;
  52         n)      NFETCH="${OPTARG}" ;;
  53         X)      MEASURE_PLEASE=1 ;;
  54         *)      usage ;;
  55         esac
  56 done
  57 shift $(($OPTIND - 1))
  58
  59 if [ $# -ne 1 ]; then
  60         usage
  61 fi
  62
  63 URL="${1}"
  64
  65 if [ -n "${MEASURE_PLEASE}" ]; then
  66         for p in `/usr/bin/jot ${NPROC}`; do
  67                 (
  68                 for f in `/usr/bin/jot ${NFETCH}`; do echo "${URL}"; done | /usr/bin/xargs /usr/bin/fetch -qo /dev/null
  69                 ) &
  70         done
  71         wait
  72         echo -n "${0}: $((${NFETCH} * ${NPROC})) requests for ${URL}, spread among ${NPROC} parallel processes, in "
  73 else
  74         ( /usr/bin/time -h "${0}" -i "${NPROC}" -n "${NFETCH}" -X "${URL}" ) 2>&1 | sed -E 's/  //;s/   +/, /g'
  75 fi