137 lines
3.3 KiB
Text
137 lines
3.3 KiB
Text
|
#!/bin/sh
|
||
|
|
|
||
|
|
#pkcallbacks.test
|
||
|
|
|
||
|
|
# if we can, isolate the network namespace to eliminate port collisions.
|
||
|
|
if [ "${AM_BWRAPPED-}" != "yes" ]; then
|
||
|
|
bwrap_path="$(command -v bwrap)"
|
||
|
|
if [ -n "$bwrap_path" ]; then
|
||
|
|
export AM_BWRAPPED=yes
|
||
|
|
exec "$bwrap_path" --unshare-net --dev-bind / / "$0" "$@"
|
||
|
|
fi
|
||
|
|
unset AM_BWRAPPED
|
||
|
|
fi
|
||
|
|
|
||
|
|
exit_code=1
|
||
|
|
counter=0
|
||
|
|
# need a unique resume port since may run the same time as testsuite
|
||
|
|
# use server port zero hack to get one
|
||
|
|
pk_port=0
|
||
|
|
#no_pid tells us process was never started if -1
|
||
|
|
no_pid=-1
|
||
|
|
#server_pid captured on startup, stores the id of the server process
|
||
|
|
server_pid=$no_pid
|
||
|
|
# let's use absolute path to a local dir (make distcheck may be in sub dir)
|
||
|
|
# also let's add some randomness by adding pid in case multiple 'make check's
|
||
|
|
# per source tree
|
||
|
|
ready_file=`pwd`/wolfssl_pk_ready$$
|
||
|
|
|
||
|
|
remove_ready_file() {
|
||
|
|
if test -e $ready_file; then
|
||
|
|
echo -e "removing existing ready file"
|
||
|
|
rm $ready_file
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
do_cleanup() {
|
||
|
|
echo "in cleanup"
|
||
|
|
|
||
|
|
if [ $server_pid != $no_pid ]
|
||
|
|
then
|
||
|
|
echo "killing server"
|
||
|
|
kill -9 $server_pid
|
||
|
|
fi
|
||
|
|
remove_ready_file
|
||
|
|
}
|
||
|
|
|
||
|
|
# trap this function so if user aborts with ^C or other kill signal we still
|
||
|
|
# get an exit that will in turn clean up the file system
|
||
|
|
abort_trap() {
|
||
|
|
echo "script aborted"
|
||
|
|
|
||
|
|
if [ $server_pid != $no_pid ]
|
||
|
|
then
|
||
|
|
echo "killing server"
|
||
|
|
kill -9 $server_pid
|
||
|
|
fi
|
||
|
|
|
||
|
|
exit_code=2 #different exit code in case of user interrupt
|
||
|
|
|
||
|
|
echo "got abort signal, exiting with $exit_code"
|
||
|
|
exit $exit_code
|
||
|
|
}
|
||
|
|
trap abort_trap INT TERM
|
||
|
|
|
||
|
|
|
||
|
|
# trap this function so that if we exit on an error the file system will still
|
||
|
|
# be restored and the other tests may still pass. Never call this function
|
||
|
|
# instead use "exit <some value>" and this function will run automatically
|
||
|
|
restore_file_system() {
|
||
|
|
remove_ready_file
|
||
|
|
}
|
||
|
|
trap restore_file_system EXIT
|
||
|
|
|
||
|
|
run_test() {
|
||
|
|
echo -e "\nStarting example server for pkcallbacks test...\n"
|
||
|
|
|
||
|
|
remove_ready_file
|
||
|
|
|
||
|
|
# starts the server on pk_port, -R generates ready file to be used as a
|
||
|
|
# mutex lock, -P does pkcallbacks. We capture the processid
|
||
|
|
# into the variable server_pid
|
||
|
|
./examples/server/server -P -R $ready_file -p $pk_port &
|
||
|
|
server_pid=$!
|
||
|
|
|
||
|
|
while [ ! -s $ready_file -a "$counter" -lt 20 ]; do
|
||
|
|
echo -e "waiting for ready file..."
|
||
|
|
sleep 0.1
|
||
|
|
counter=$((counter+ 1))
|
||
|
|
done
|
||
|
|
|
||
|
|
if test -e $ready_file; then
|
||
|
|
echo -e "found ready file, starting client..."
|
||
|
|
else
|
||
|
|
echo -e "NO ready file ending test..."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# sleep for an additional 0.1 to mitigate race on write/read of $ready_file:
|
||
|
|
sleep 0.1
|
||
|
|
|
||
|
|
# get created port 0 ephemeral port
|
||
|
|
pk_port=`cat $ready_file`
|
||
|
|
|
||
|
|
# starts client on pk_port with pkcallbacks, captures the output from client
|
||
|
|
capture_out=$(./examples/client/client -P -p $pk_port 2>&1)
|
||
|
|
client_result=$?
|
||
|
|
|
||
|
|
if [ $client_result != 0 ]
|
||
|
|
then
|
||
|
|
echo -e "client failed!"
|
||
|
|
do_cleanup
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
wait $server_pid
|
||
|
|
server_result=$?
|
||
|
|
|
||
|
|
if [ $server_result != 0 ]
|
||
|
|
then
|
||
|
|
echo -e "server failed!"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
######### begin program #########
|
||
|
|
|
||
|
|
# run the test
|
||
|
|
run_test
|
||
|
|
|
||
|
|
# If we get to this, success
|
||
|
|
echo "Success!"
|
||
|
|
exit 0
|
||
|
|
########## end program ##########
|
||
|
|
|