mirror of
https://github.com/yugr/Implib.so
synced 2026-06-08 18:33:41 +00:00
57 lines
1.3 KiB
Bash
Executable File
57 lines
1.3 KiB
Bash
Executable File
#!/bin/sh
|
|
|
|
# Copyright 2025 Yury Gribov
|
|
#
|
|
# The MIT License (MIT)
|
|
#
|
|
# Use of this source code is governed by MIT license that can be
|
|
# found in the LICENSE.txt file.
|
|
|
|
# This is a stress test of thread safety via unthread library
|
|
# (https://github.com/mpdn/unthread)
|
|
|
|
set -eu
|
|
|
|
cd $(dirname $0)
|
|
|
|
CC=clang
|
|
|
|
CFLAGS="-g -O2 -fsanitize=thread"
|
|
|
|
UNTHREAD=$HOME/src/unthread
|
|
CFLAGS="-I$UNTHREAD/include $CFLAGS"
|
|
LIBS="$UNTHREAD/bin/unthread.o -ldl"
|
|
|
|
# ASLR keeps breaking Tsan mmaps
|
|
if test $(cat /proc/sys/kernel/randomize_va_space) != 0; then
|
|
echo 0 | sudo tee /proc/sys/kernel/randomize_va_space
|
|
fi
|
|
|
|
# Build shlib to test against
|
|
$CC $CFLAGS -shared -fPIC interposed.c -o libinterposed.so
|
|
|
|
# Prepare implib
|
|
${PYTHON:-} ../../implib-gen.py -q libinterposed.so
|
|
|
|
# Test without export shims
|
|
|
|
$CC $CFLAGS main.c libinterposed.so.tramp.S libinterposed.so.init.c $LIBS
|
|
|
|
for i in $(seq 1 1000); do
|
|
export UNTHREAD_SEED=$(printf '%032x' $i)
|
|
LD_LIBRARY_PATH=.:${LD_LIBRARY_PATH:-} ./a.out > a.out.log
|
|
diff test.ref a.out.log
|
|
done
|
|
|
|
# Test with export shims
|
|
|
|
$CC $CFLAGS -DIMPLIB_EXPORT_SHIMS main.c libinterposed.so.tramp.S libinterposed.so.init.c $LIBS
|
|
|
|
for i in $(seq 1 1000); do
|
|
export UNTHREAD_SEED=$(printf '%032x' $i)
|
|
LD_LIBRARY_PATH=.:${LD_LIBRARY_PATH:-} ./a.out > a.out.log
|
|
diff test.ref a.out.log
|
|
done
|
|
|
|
echo SUCCESS
|