134 lines
2.8 KiB
Bash
134 lines
2.8 KiB
Bash
|
#!/bin/sh
|
||
|
|
# SPDX-License-Identifier: GPL-2.0
|
||
|
|
#
|
||
|
|
# Test for mishandling of splice() on pseudofilesystems, which should catch
|
||
|
|
# bugs like 11990a5bd7e5 ("module: Correctly truncate sysfs sections output")
|
||
|
|
#
|
||
|
|
# Since splice fallback was removed as part of the set_fs() rework, many of these
|
||
|
|
# tests expect to fail now. See https://lore.kernel.org/lkml/202009181443.C2179FB@keescook/
|
||
|
|
set -e
|
||
|
|
|
||
|
|
DIR=$(dirname "$0")
|
||
|
|
|
||
|
|
ret=0
|
||
|
|
|
||
|
|
expect_success()
|
||
|
|
{
|
||
|
|
title="$1"
|
||
|
|
shift
|
||
|
|
|
||
|
|
echo "" >&2
|
||
|
|
echo "$title ..." >&2
|
||
|
|
|
||
|
|
set +e
|
||
|
|
"$@"
|
||
|
|
rc=$?
|
||
|
|
set -e
|
||
|
|
|
||
|
|
case "$rc" in
|
||
|
|
0)
|
||
|
|
echo "ok: $title succeeded" >&2
|
||
|
|
;;
|
||
|
|
1)
|
||
|
|
echo "FAIL: $title should work" >&2
|
||
|
|
ret=$(( ret + 1 ))
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
echo "FAIL: something else went wrong" >&2
|
||
|
|
ret=$(( ret + 1 ))
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
}
|
||
|
|
|
||
|
|
expect_failure()
|
||
|
|
{
|
||
|
|
title="$1"
|
||
|
|
shift
|
||
|
|
|
||
|
|
echo "" >&2
|
||
|
|
echo "$title ..." >&2
|
||
|
|
|
||
|
|
set +e
|
||
|
|
"$@"
|
||
|
|
rc=$?
|
||
|
|
set -e
|
||
|
|
|
||
|
|
case "$rc" in
|
||
|
|
0)
|
||
|
|
echo "FAIL: $title unexpectedly worked" >&2
|
||
|
|
ret=$(( ret + 1 ))
|
||
|
|
;;
|
||
|
|
1)
|
||
|
|
echo "ok: $title correctly failed" >&2
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
echo "FAIL: something else went wrong" >&2
|
||
|
|
ret=$(( ret + 1 ))
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
}
|
||
|
|
|
||
|
|
do_splice()
|
||
|
|
{
|
||
|
|
filename="$1"
|
||
|
|
bytes="$2"
|
||
|
|
expected="$3"
|
||
|
|
report="$4"
|
||
|
|
|
||
|
|
out=$("$DIR"/splice_read "$filename" "$bytes" | cat)
|
||
|
|
if [ "$out" = "$expected" ] ; then
|
||
|
|
echo " matched $report" >&2
|
||
|
|
return 0
|
||
|
|
else
|
||
|
|
echo " no match: '$out' vs $report" >&2
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
test_splice()
|
||
|
|
{
|
||
|
|
filename="$1"
|
||
|
|
|
||
|
|
echo " checking $filename ..." >&2
|
||
|
|
|
||
|
|
full=$(cat "$filename")
|
||
|
|
rc=$?
|
||
|
|
if [ $rc -ne 0 ] ; then
|
||
|
|
return 2
|
||
|
|
fi
|
||
|
|
|
||
|
|
two=$(echo "$full" | grep -m1 . | cut -c-2)
|
||
|
|
|
||
|
|
# Make sure full splice has the same contents as a standard read.
|
||
|
|
echo " splicing 4096 bytes ..." >&2
|
||
|
|
if ! do_splice "$filename" 4096 "$full" "full read" ; then
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Make sure a partial splice see the first two characters.
|
||
|
|
echo " splicing 2 bytes ..." >&2
|
||
|
|
if ! do_splice "$filename" 2 "$two" "'$two'" ; then
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
|
||
|
|
### /proc/$pid/ has no splice interface; these should all fail.
|
||
|
|
expect_failure "proc_single_open(), seq_read() splice" test_splice /proc/$$/limits
|
||
|
|
expect_failure "special open(), seq_read() splice" test_splice /proc/$$/comm
|
||
|
|
|
||
|
|
### /proc/sys/ has a splice interface; these should all succeed.
|
||
|
|
expect_success "proc_handler: proc_dointvec_minmax() splice" test_splice /proc/sys/fs/nr_open
|
||
|
|
expect_success "proc_handler: proc_dostring() splice" test_splice /proc/sys/kernel/modprobe
|
||
|
|
expect_success "proc_handler: special read splice" test_splice /proc/sys/kernel/version
|
||
|
|
|
||
|
|
### /sys/ has no splice interface; these should all fail.
|
||
|
|
if ! [ -d /sys/module/test_module/sections ] ; then
|
||
|
|
expect_success "test_module kernel module load" modprobe test_module
|
||
|
|
fi
|
||
|
|
expect_success "kernfs attr splice" test_splice /sys/module/test_module/coresize
|
||
|
|
expect_success "kernfs binattr splice" test_splice /sys/module/test_module/sections/.init.text
|
||
|
|
|
||
|
|
exit $ret
|