Skip to content

Commit

Permalink
selftests/bpf: Indent ima_setup.sh with tabs.
Browse files Browse the repository at this point in the history
The file was formatted with spaces instead of tabs and went unnoticed
as checkpatch.pl did not complain (probably because this is a shell
script). Re-indent it with tabs to be consistent with other scripts.

Signed-off-by: KP Singh <[email protected]>
Signed-off-by: Andrii Nakryiko <[email protected]>
Link: https://lore.kernel.org/bpf/[email protected]
  • Loading branch information
sinkap authored and anakryiko committed Dec 3, 2020
1 parent d932e04 commit ffebecd
Showing 1 changed file with 54 additions and 54 deletions.
108 changes: 54 additions & 54 deletions tools/testing/selftests/bpf/ima_setup.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,90 +10,90 @@ TEST_BINARY="/bin/true"

usage()
{
echo "Usage: $0 <setup|cleanup|run> <existing_tmp_dir>"
exit 1
echo "Usage: $0 <setup|cleanup|run> <existing_tmp_dir>"
exit 1
}

ensure_mount_securityfs()
{
local securityfs_dir=$(grep "securityfs" /proc/mounts | awk '{print $2}')
local securityfs_dir=$(grep "securityfs" /proc/mounts | awk '{print $2}')

if [ -z "${securityfs_dir}" ]; then
securityfs_dir=/sys/kernel/security
mount -t securityfs security "${securityfs_dir}"
fi
if [ -z "${securityfs_dir}" ]; then
securityfs_dir=/sys/kernel/security
mount -t securityfs security "${securityfs_dir}"
fi

if [ ! -d "${securityfs_dir}" ]; then
echo "${securityfs_dir}: securityfs is not mounted" && exit 1
fi
if [ ! -d "${securityfs_dir}" ]; then
echo "${securityfs_dir}: securityfs is not mounted" && exit 1
fi
}

setup()
{
local tmp_dir="$1"
local mount_img="${tmp_dir}/test.img"
local mount_dir="${tmp_dir}/mnt"
local copied_bin_path="${mount_dir}/$(basename ${TEST_BINARY})"
mkdir -p ${mount_dir}
local tmp_dir="$1"
local mount_img="${tmp_dir}/test.img"
local mount_dir="${tmp_dir}/mnt"
local copied_bin_path="${mount_dir}/$(basename ${TEST_BINARY})"
mkdir -p ${mount_dir}

dd if=/dev/zero of="${mount_img}" bs=1M count=10
dd if=/dev/zero of="${mount_img}" bs=1M count=10

losetup -f "${mount_img}"
local loop_device=$(losetup -a | grep ${mount_img:?} | cut -d ":" -f1)
losetup -f "${mount_img}"
local loop_device=$(losetup -a | grep ${mount_img:?} | cut -d ":" -f1)

mkfs.ext2 "${loop_device:?}"
mount "${loop_device}" "${mount_dir}"
mkfs.ext2 "${loop_device:?}"
mount "${loop_device}" "${mount_dir}"

cp "${TEST_BINARY}" "${mount_dir}"
local mount_uuid="$(blkid ${loop_device} | sed 's/.*UUID="\([^"]*\)".*/\1/')"
cp "${TEST_BINARY}" "${mount_dir}"
local mount_uuid="$(blkid ${loop_device} | sed 's/.*UUID="\([^"]*\)".*/\1/')"

ensure_mount_securityfs
echo "measure func=BPRM_CHECK fsuuid=${mount_uuid}" > ${IMA_POLICY_FILE}
ensure_mount_securityfs
echo "measure func=BPRM_CHECK fsuuid=${mount_uuid}" > ${IMA_POLICY_FILE}
}

cleanup() {
local tmp_dir="$1"
local mount_img="${tmp_dir}/test.img"
local mount_dir="${tmp_dir}/mnt"
local tmp_dir="$1"
local mount_img="${tmp_dir}/test.img"
local mount_dir="${tmp_dir}/mnt"

local loop_devices=$(losetup -a | grep ${mount_img:?} | cut -d ":" -f1)
local loop_devices=$(losetup -a | grep ${mount_img:?} | cut -d ":" -f1)

for loop_dev in "${loop_devices}"; do
losetup -d $loop_dev
done
for loop_dev in "${loop_devices}"; do
losetup -d $loop_dev
done

umount ${mount_dir}
rm -rf ${tmp_dir}
umount ${mount_dir}
rm -rf ${tmp_dir}
}

run()
{
local tmp_dir="$1"
local mount_dir="${tmp_dir}/mnt"
local copied_bin_path="${mount_dir}/$(basename ${TEST_BINARY})"
local tmp_dir="$1"
local mount_dir="${tmp_dir}/mnt"
local copied_bin_path="${mount_dir}/$(basename ${TEST_BINARY})"

exec "${copied_bin_path}"
exec "${copied_bin_path}"
}

main()
{
[[ $# -ne 2 ]] && usage

local action="$1"
local tmp_dir="$2"

[[ ! -d "${tmp_dir}" ]] && echo "Directory ${tmp_dir} doesn't exist" && exit 1

if [[ "${action}" == "setup" ]]; then
setup "${tmp_dir}"
elif [[ "${action}" == "cleanup" ]]; then
cleanup "${tmp_dir}"
elif [[ "${action}" == "run" ]]; then
run "${tmp_dir}"
else
echo "Unknown action: ${action}"
exit 1
fi
[[ $# -ne 2 ]] && usage

local action="$1"
local tmp_dir="$2"

[[ ! -d "${tmp_dir}" ]] && echo "Directory ${tmp_dir} doesn't exist" && exit 1

if [[ "${action}" == "setup" ]]; then
setup "${tmp_dir}"
elif [[ "${action}" == "cleanup" ]]; then
cleanup "${tmp_dir}"
elif [[ "${action}" == "run" ]]; then
run "${tmp_dir}"
else
echo "Unknown action: ${action}"
exit 1
fi
}

main "$@"

0 comments on commit ffebecd

Please sign in to comment.