[[ "$VERBOSE" == "1" ]] && set -x
RUSTC=$(which rustc)
RUSTDOC=$(which rustdoc)
CARGO=$(which cargo)
SUDO=$(which sudo)
SYSTEM=$(uname -s)
TESTER="$CARGO test"
CC=$(which clang || which gcc)
CARGO_FLAGS=
PNET_TEST_IFACE=$(ifconfig | egrep 'UP| active' | \
perl -pe '/^[A-z0-9]+:([^\n]|\n\t)*status: active/' | \
grep active -B1 | head -n1 | cut -f1 -d:)
build() {
if [[ -x "$CARGO" ]]; then
$CARGO build $CARGO_FLAGS --release
else
$RUSTC src/lib.rs
fi
}
build_doc() {
if [[ -x "$CARGO" ]]; then
$CARGO doc $CARGO_FLAGS
else
$RUSTDOC src/lib.rs -o target/doc --crate-name pnet
fi
}
build_test() {
if [[ -x "$CARGO" ]]; then
$CARGO test --no-run $CARGO_FLAGS
else
$RUSTC src/lib.rs --test --out-dir ./target/ -C extra-filename=-no-cargo
fi
}
run_test() {
build_test &&
echo "Setting permissions for test suite - enter sudo password if prompted" &&
case "$SYSTEM" in
Linux)
if [[ "$(id -u)" != "0" ]]; then
$SUDO setcap cap_net_raw+ep target/pnet-*;
fi
RUST_TEST_TASKS=1 $TESTER
;;
FreeBSD|Darwin)
$SUDO PNET_TEST_IFACE=$PNET_TEST_IFACE RUST_TEST_TASKS=1 $TESTER
;;
MINGW*|MSYS*)
PNET_TEST_IFACE=$PNET_TEST_IFACE RUST_TEST_TASKS=1 $TESTER
;;
*)
echo "Unsupported testing platform"
;;
esac
}
clean() {
if [[ -x "$CARGO" ]]; then
$CARGO clean $CARGO_FLAGS
else
rm -fr target
fi
}
benchmarks() {
[[ "$SYSTEM" != "Darwin" ]] && echo warning: C benchmarks only work on OS X
$CC -W -Wall -O2 benches/c_receiver.c -o target/benches/c_receiver
$CC -W -Wall -O2 benches/c_sender.c -o target/benches/c_sender
$RUSTC -O benches/rs_receiver.rs --out-dir target/benches -L target/release
$RUSTC -O benches/rs_sender.rs --out-dir target/benches -L target/release
}
mkdir -p target/doc
mkdir -p target/benches
if [[ "$VERBOSE" == "1" ]]; then
CARGO_FLAGS="--verbose"
TESTER="$TESTER $CARGO_FLAGS"
fi
if [[ ! -x "$CARGO" ]]; then
TESTER='./target/pnet-*'
fi
case "$1" in
test)
run_test
;;
doc)
build_doc
;;
clean)
clean
;;
benchmarks)
benchmarks
;;
*)
build
;;
esac