sqlite-parquet-vtable/tests/test-failmalloc

42 lines
1010 B
Bash
Executable File

#!/bin/bash
set -euo pipefail
# A harness that runs SQLite with the parquet extension in an environment where malloc randomly
# fails. "Success" is if the logs don't have any C++ exceptions that talk about std::bad_alloc
ensure_failmalloc() {
if [ ! -d libfailmalloc ]; then
git clone https://github.com/cldellow/libfailmalloc.git
fi
if [ ! -e libfailmalloc/.libs/libfailmalloc.so ]; then
cd libfailmalloc
./configure
make
fi
}
run_under_low_memory() {
start=$(date +%s%3N)
set +e
env LD_PRELOAD="$here"/libfailmalloc/.libs/libfailmalloc.so FAILMALLOC_PROBABILITY=0.00001 ./test-random &> results.bad_alloc
set -e
now=$(date +%s%3N)
echo "Bailed after $((now-start)) ms"
! grep std::bad_alloc results.bad_alloc
}
main() {
here=$(dirname "${BASH_SOURCE[0]}")
here=$(readlink -f "$here")
cd "$here"
ensure_failmalloc
# Sometimes we'll exit due to a Python memory issue, so try a few times.
for i in {0..10}; do
run_under_low_memory
done
}
main "$@"