95 lines
3.6 KiB
Bash
95 lines
3.6 KiB
Bash
|
#!/bin/sh
|
||
|
|
#
|
||
|
|
# Copyright (c) 2005-2016 Intel Corporation
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#
|
||
|
|
#
|
||
|
|
|
||
|
|
# Usage:
|
||
|
|
# test_launcher.sh [-v] [-q] [-s] [-r <repeats>] [-u] [-l <library>] <executable> <arg1> <arg2> <argN>
|
||
|
|
# where: -v enables verbose output
|
||
|
|
# where: -q enables quiet mode
|
||
|
|
# where: -s runs the test in stress mode (until non-zero exit code or ctrl-c pressed)
|
||
|
|
# where: -r <repeats> specifies number of times to repeat execution
|
||
|
|
# where: -u limits stack size
|
||
|
|
# where: -l <library> specifies the library name to be assigned to LD_PRELOAD
|
||
|
|
|
||
|
|
while getopts "qvsr:ul:" flag #
|
||
|
|
do case $flag in #
|
||
|
|
s ) # Stress testing mode
|
||
|
|
run_prefix="stressed $run_prefix" ;; #
|
||
|
|
r ) # Repeats test n times
|
||
|
|
repeat=$OPTARG #
|
||
|
|
run_prefix="repeated $run_prefix" ;; #
|
||
|
|
l ) if [ `uname` = 'Linux' ] ; then #
|
||
|
|
LD_PRELOAD=$OPTARG #
|
||
|
|
elif [ `uname` = 'Darwin' ] ; then #
|
||
|
|
DYLD_INSERT_LIBRARIES=$OPTARG #
|
||
|
|
else #
|
||
|
|
echo 'skip' #
|
||
|
|
exit #
|
||
|
|
fi ;; #
|
||
|
|
u ) # Set stack limit
|
||
|
|
ulimit -s 10240 ;; #
|
||
|
|
q ) # Quiet mode, removes 'done' but prepends any other output by test name
|
||
|
|
OUTPUT='2>&1 | sed -e "s/done//;/^[[:space:]]*$/d;s!^!$1: !"' ;; #
|
||
|
|
v ) # Verbose mode
|
||
|
|
verbose=1 ;; #
|
||
|
|
esac done #
|
||
|
|
shift `expr $OPTIND - 1` #
|
||
|
|
if [ $MIC_OFFLOAD_NATIVE_PATH ] ; then #
|
||
|
|
LIB_NAME=${1/%.$TEST_EXT/_dll.$DLL} #
|
||
|
|
if [ -f "$MIC_OFFLOAD_NATIVE_PATH/$LIB_NAME" ]; then #
|
||
|
|
[ -z "$MIC_CARD" ] && MIC_CARD=mic0 #
|
||
|
|
TMPDIR_HOST=`mktemp -d /tmp/tbbtestXXXXXX` #
|
||
|
|
TMPDIR_MIC=`sudo ssh $MIC_CARD mktemp -d /tmp/tbbtestXXXXXX` #
|
||
|
|
sudo ssh $MIC_CARD "chmod +x $TMPDIR_MIC" #
|
||
|
|
# Test specific library may depend on libtbbmalloc*
|
||
|
|
cp "$MIC_OFFLOAD_NATIVE_PATH/$LIB_NAME" "$MIC_OFFLOAD_NATIVE_PATH"/libtbbmalloc* "$TMPDIR_HOST" >/dev/null 2>/dev/null #
|
||
|
|
sudo scp "$TMPDIR_HOST"/* $MIC_CARD:"$TMPDIR_MIC" >/dev/null 2>/dev/null #
|
||
|
|
|
||
|
|
LD_LIBRARY_PATH=$TMPDIR_MIC:$LD_LIBRARY_PATH #
|
||
|
|
export LD_LIBRARY_PATH #
|
||
|
|
fi #
|
||
|
|
fi #
|
||
|
|
stressed() { echo Doing stress testing. Press Ctrl-C to terminate #
|
||
|
|
while :; do $*; done;#
|
||
|
|
} #
|
||
|
|
repeated() { #
|
||
|
|
i=0; while [ "$i" -lt $repeat ]; do i=`expr $i + 1`; echo $i of $repeat:; $*; done #
|
||
|
|
} #
|
||
|
|
# DYLD_LIBRARY_PATH can be purged on OS X 10.11, set it again
|
||
|
|
if [ `uname` = 'Darwin' -a -z "$DYLD_LIBRARY_PATH" ] ; then #
|
||
|
|
DYLD_LIBRARY_PATH=. #
|
||
|
|
export DYLD_LIBRARY_PATH #
|
||
|
|
fi #
|
||
|
|
# Run the command line passed via parameters
|
||
|
|
[ $verbose ] && echo Running $run_prefix $* #
|
||
|
|
if [ -n "$LD_PRELOAD" ] ; then #
|
||
|
|
export LD_PRELOAD #
|
||
|
|
elif [ -n "$DYLD_INSERT_LIBRARIES" ] ; then #
|
||
|
|
export DYLD_INSERT_LIBRARIES #
|
||
|
|
fi #
|
||
|
|
exec 4>&1 # extracting exit code of the first command in pipeline needs duplicated stdout
|
||
|
|
# custom redirection needs eval, otherwise shell cannot parse it
|
||
|
|
err=`eval '( $run_prefix $* || echo \$? >&3; )' ${OUTPUT} 3>&1 >&4` #
|
||
|
|
[ -z "$err" ] || echo $1: exited with error $err #
|
||
|
|
if [ $MIC_OFFLOAD_NATIVE_PATH ] ; then #
|
||
|
|
sudo ssh $MIC_CARD rm -fr "$TMPDIR_MIC" >/dev/null 2>/dev/null #
|
||
|
|
rm -fr "$TMPDIR_HOST" >/dev/null 2>/dev/null #
|
||
|
|
fi #
|
||
|
|
exit $err #
|