test/app_test/run_app_test.ksh

   1 #!/usr/bin/env ksh
   2 # vim: ts=4 sw=4 noet :
   3 #==================================================================================
   4 #    Copyright (c) 2019-2020 Nokia
   5 #    Copyright (c) 2018-2020 AT&T Intellectual Property.
   6 #
   7 #   Licensed under the Apache License, Version 2.0 (the "License");
   8 #   you may not use this file except in compliance with the License.
   9 #   You may obtain a copy of the License at
  10 #
  11 #       http://www.apache.org/licenses/LICENSE-2.0
  12 #
  13 #   Unless required by applicable law or agreed to in writing, software
  14 #   distributed under the License is distributed on an "AS IS" BASIS,
  15 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 #   See the License for the specific language governing permissions and
  17 #   limitations under the License.
  18 #==================================================================================
  19 #
  20
  21 # ---------------------------------------------------------------------------------
  22 #       Mnemonic:       run_app_test.ksh
  23 #       Abstract:       This is a simple script to set up and run the basic send/receive
  24 #                               processes for some library validation on top of nng.
  25 #                               It should be possible to clone the repo, switch to this directory
  26 #                               and execute  'ksh run -B'  which will build RMr, make the sender and
  27 #                               recevier then  run the basic test.
  28 #
  29 #                               Example command line:
  30 #                                       ksh ./run_app_test.ksh          # default 20 messages at 2 msg/sec
  31 #                                       ksh ./run_app_test.ksh -d 100 -n 10000 # send 10k messages with 100ms delay between
  32 #
  33 #       Date:           22 April 2019
  34 #       Author:         E. Scott Daniels
  35 # ---------------------------------------------------------------------------------
  36
  37
  38 # The sender and receiver are run asynch. Their exit statuses are captured in a
  39 # file in order for the 'main' to pick them up easily.
  40 #
  41 function run_sender {
  42         ./sender${si} $nmsg $delay
  43         echo $? >/tmp/PID$$.src         # must communicate state back via file b/c asynch
  44 }
  45
  46 function run_rcvr {
  47         if (( mt_receiver ))
  48         then
  49                 echo "<TEST> testing with mt-receiver" >&2
  50                 ./mt_receiver${si} $nmsg
  51         else
  52                 ./receiver${si} $nmsg
  53         fi
  54         echo $? >/tmp/PID$$.rrc
  55 }
  56
  57 # snarf the first v4 IP (not the loopback) that belongs to this box/container/guest
  58 function snarf_ip {
  59         ip addr| sed -e '/inet /b c; d' -e ':c' -e '/127.0.0.1/d; s!/.*!!; s!^.* !!; q'
  60 }
  61
  62 #       Drop a contrived route table in. This table should add a reference to our
  63 #       local IP to an entry to ensure that the route table collector code in RMr
  64 #       is removing 'hairpin' loops. If RMr isn't removing the references to our
  65 #       hostname and IP address when it builds the endpoint lists, the sender will
  66 #       send messages to itself some of the time, causing the receiver to come up
  67 #       short when comparing messages received with expected count and thus failing.
  68 #
  69 function set_rt {
  70         typeset port=4560                       # port the receiver listens on by default
  71
  72         cat <<endKat >app_test.rt
  73                 newrt | start
  74                         mse | 0 |  0 | localhost:$port,$my_ip:43086
  75                         mse | 1 | 10 | localhost:$port,${my_host//.*/}:43086
  76                         mse | 2 | 20 | localhost:$port
  77                         rte | 3 | localhost:$port
  78                         mse | 3 | 100 | localhost:$port # special test to ensure that this does not affect previous entry
  79                         rte | 4 | localhost:$port
  80                         rte | 5 | localhost:$port
  81                         rte | 6 | localhost:$port
  82                         rte | 7 | localhost:$port
  83                         rte | 8 | localhost:$port
  84                         rte | 9 | localhost:$port
  85                         rte | 10 | localhost:$port
  86                         rte | 11 | localhost:$port
  87                         rte | 12 | localhost:$port
  88                         rte | 13 | localhost:$port
  89                 newrt | end
  90
  91 head -3 app_test.rt
  92
  93 endKat
  94
  95 }
  96
  97 # ---------------------------------------------------------
  98
  99 nmsg=20                                         # total number of messages to be exchanged (-n value changes)
 100                                                         # need two sent to each receiver to ensure hairpin entries were removed (will fail if they were not)
 101 delay=500000                            # microsec sleep between msg 1,000,000 == 1s
 102 wait=1
 103 rebuild=0
 104 nopull=""                                       # -b sets so that build does not pull
 105 verbose=0
 106 use_installed=0
 107 my_ip=$(snarf_ip)                       # get an ip to insert into the route table
 108 keep=0
 109 mt_receiver=0                           # -m sets in order to test with multi-threaded receive stuff
 110 force_make=0
 111 si=""                                           # -S sets to build and test SI versions
 112
 113
 114 while [[ $1 == -* ]]
 115 do
 116         case $1 in
 117                 -B)     rebuild=1;;                                             # build with pull first
 118                 -b)     rebuild=1; nopull="nopull";;    # buld without pull
 119                 -d)     delay=$2; shift;;
 120                 -k) keep=1;;
 121                 -i) use_installed=1;;
 122                 -M)     force_make=1;;
 123                 -m)     mt_receiver=1;;
 124                 -n)     nmsg=$2; shift;;
 125                 -N)     si="";;                                                 # build and test NNG versions (off si)
 126                 -S)     si="_si";;                                              # build and test SI95 versions
 127                 -v)     verbose=1;;
 128
 129                 *)      echo "unrecognised option: $1"
 130                         echo "usage: $0 [-B] [-d micor-sec-delay] [-i] [-k] [-M] [-m] [-n num-msgs] [-S]"
 131                         echo "  -B forces an RMR rebuild which will use .build"
 132                         echo "  -i causes the installd libraries (/usr/local) to be referenced; -B is ignored if supplied"
 133                         echo "  -k keeps the route table"
 134                         echo "  -M force make on test applications"
 135                         echo "  -m test with mt-receive mode"
 136                         echo "  -S build/test SI95 based binaries"
 137                         echo ""
 138                         echo "total number of messages must > 20 to correctly test hairpin loop removal"
 139                         exit 1
 140                         ;;
 141         esac
 142
 143         shift
 144 done
 145
 146 if [[ ! -f app_test.rt ]]               # we need the real host name in the local.rt; build one from mask if not there
 147 then
 148         my_host=$(hostname)
 149         set_rt
 150         if (( verbose ))
 151         then
 152                 cat app_test.rt
 153         fi
 154 fi
 155
 156 if (( verbose ))
 157 then
 158         echo "2" >.verbose
 159         export RMR_VCTL_FILE=".verbose"
 160 fi
 161
 162
 163 if (( use_installed ))                  # point at installed library rather than testing the build
 164 then
 165         export LD_LIBRARY_PATH=/usr/local/lib
 166         export LIBRARY_PATH=$LD_LIBRARY_PATH
 167 else
 168         if (( rebuild ))
 169         then
 170                 set -e
 171                 $SHELL ./rebuild.ksh $nopull | read build_path
 172                 set +e
 173         else
 174                 build_path=${BUILD_PATH:-"../../.build"}        # we prefer .build at the root level, but allow user option
 175
 176                 if [[ ! -d $build_path ]]
 177                 then
 178                         echo "[FAIL] cannot find build in: $build_path"
 179                         echo "[FAIL] either create, and then build RMr, or set BUILD_PATH as an evironment var before running this"
 180                         exit 1
 181                 fi
 182         fi
 183
 184         if [[ -d $build_path/lib64 ]]
 185         then
 186                 export LD_LIBRARY_PATH=$build_path:$build_path/lib64
 187         else
 188                 export LD_LIBRARY_PATH=$build_path:$build_path/lib
 189         fi
 190         export LIBRARY_PATH=$LD_LIBRARY_PATH
 191 fi
 192
 193 export RMR_SEED_RT=${RMR_SEED_RT:-./app_test.rt}                # allow easy testing with different rt
 194
 195 if (( force_make )) || [[ ! -f ./sender${si} || ! -f ./receiver${si} ]]
 196 then
 197         if ! make -B sender${si} receiver${si} >/tmp/PID$$.clog 2>&1
 198         then
 199                 echo "[FAIL] cannot find sender${si} and/or receiver${si}binary, and cannot make them.... humm?"
 200                 head -50 /tmp/PID$$.clog
 201                 rm -fr /tmp/PID$$.*
 202                 exit 1
 203         fi
 204 fi
 205
 206 run_rcvr &
 207 sleep 2                         # if sender starts faster than rcvr we can drop msgs, so pause a bit
 208 run_sender &
 209
 210 wait
 211 head -1 /tmp/PID$$.rrc | read rrc
 212 head -1 /tmp/PID$$.src | read src
 213
 214 if (( !! (src + rrc) ))
 215 then
 216         echo "[FAIL] sender rc=$src  receiver rc=$rrc"
 217 else
 218         echo "[PASS] sender rc=$src  receiver rc=$rrc"
 219 fi
 220
 221 if (( ! keep ))
 222 then
 223         rm app_test.rt
 224 fi
 225
 226 rm /tmp/PID$$.*
 227 rm -f .verbose
 228
 229 exit $(( !! (src + rrc) ))
 230