Last active
September 17, 2018 10:29
-
-
Save c0mix/085a350df0b78290d3b3fb175a5c5555 to your computer and use it in GitHub Desktop.
a simple fix for firmadyne makeImage.sh script
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/bash | |
set -e | |
set -u | |
if [ -e ./firmadyne.config ]; then | |
source ./firmadyne.config | |
elif [ -e ../firmadyne.config ]; then | |
source ../firmadyne.config | |
else | |
echo "Error: Could not find 'firmadyne.config'!" | |
exit 1 | |
fi | |
if check_number $1; then | |
echo "Usage: makeImage.sh <image ID> [<architecture]" | |
exit 1 | |
fi | |
IID=${1} | |
if check_root; then | |
echo "Error: This script requires root privileges!" | |
exit 1 | |
fi | |
if [ $# -gt 1 ]; then | |
if check_arch "${2}"; then | |
echo "Error: Invalid architecture!" | |
exit 1 | |
fi | |
ARCH=${2} | |
else | |
echo -n "Querying database for architecture... " | |
ARCH=$(psql -d firmware -U firmadyne -h 127.0.0.1 -t -q -c "SELECT arch from image WHERE id=${1};") | |
ARCH="${ARCH#"${ARCH%%[![:space:]]*}"}" | |
echo "${ARCH}" | |
if [ -z "${ARCH}" ]; then | |
echo "Error: Unable to lookup architecture. Please specify {armel,mipseb,mipsel} as the second argument!" | |
exit 1 | |
fi | |
fi | |
echo "----Running----" | |
WORK_DIR=`get_scratch ${IID}` | |
IMAGE=`get_fs ${IID}` | |
IMAGE_DIR=`get_fs_mount ${IID}` | |
CONSOLE=`get_console ${ARCH}` | |
LIBNVRAM=`get_nvram ${ARCH}` | |
DEVICE=`get_device` | |
echo "----Copying Filesystem Tarball----" | |
mkdir -p "${WORK_DIR}" | |
chmod a+rwx "${WORK_DIR}" | |
chown -R "${USER}" "${WORK_DIR}" | |
chgrp -R "${USER}" "${WORK_DIR}" | |
if [ ! -e "${WORK_DIR}/${IID}.tar.gz" ]; then | |
if [ ! -e "${TARBALL_DIR}/${IID}.tar.gz" ]; then | |
echo "Error: Cannot find tarball of root filesystem for ${IID}!" | |
exit 1 | |
else | |
cp "${TARBALL_DIR}/${IID}.tar.gz" "${WORK_DIR}/${IID}.tar.gz" | |
fi | |
fi | |
echo "----Creating QEMU Image----" | |
qemu-img create -f raw "${IMAGE}" 1G | |
chmod a+rw "${IMAGE}" | |
echo "----Creating Partition Table----" | |
echo -e "o\nn\np\n1\n\n\nw" | /sbin/fdisk "${IMAGE}" | |
echo "----Mounting QEMU Image----" | |
output=$(kpartx -a -s -v "${IMAGE}") | |
foo=$(echo "$output" | cut -d ' ' -f 3) | |
DEVICE="/dev/mapper/$foo" | |
#echo $DEVICE # for debug purpose (can be removed) | |
sleep 1 | |
echo "----Creating Filesystem----" | |
mkfs.ext2 "${DEVICE}" | |
sync | |
echo "----Making QEMU Image Mountpoint----" | |
if [ ! -e "${IMAGE_DIR}" ]; then | |
mkdir "${IMAGE_DIR}" | |
chown "${USER}" "${IMAGE_DIR}" | |
fi | |
echo "----Mounting QEMU Image Partition 1----" | |
mount "${DEVICE}" "${IMAGE_DIR}" | |
echo "----Extracting Filesystem Tarball----" | |
tar -xf "${WORK_DIR}/$IID.tar.gz" -C "${IMAGE_DIR}" | |
rm "${WORK_DIR}/${IID}.tar.gz" | |
echo "----Creating FIRMADYNE Directories----" | |
mkdir "${IMAGE_DIR}/firmadyne/" | |
mkdir "${IMAGE_DIR}/firmadyne/libnvram/" | |
mkdir "${IMAGE_DIR}/firmadyne/libnvram.override/" | |
echo "----Patching Filesystem (chroot)----" | |
cp $(which busybox) "${IMAGE_DIR}" | |
cp "${SCRIPT_DIR}/fixImage.sh" "${IMAGE_DIR}" | |
chroot "${IMAGE_DIR}" /busybox ash /fixImage.sh | |
rm "${IMAGE_DIR}/fixImage.sh" | |
rm "${IMAGE_DIR}/busybox" | |
echo "----Setting up FIRMADYNE----" | |
cp "${CONSOLE}" "${IMAGE_DIR}/firmadyne/console" | |
chmod a+x "${IMAGE_DIR}/firmadyne/console" | |
mknod -m 666 "${IMAGE_DIR}/firmadyne/ttyS1" c 4 65 | |
cp "${LIBNVRAM}" "${IMAGE_DIR}/firmadyne/libnvram.so" | |
chmod a+x "${IMAGE_DIR}/firmadyne/libnvram.so" | |
cp "${SCRIPT_DIR}/preInit.sh" "${IMAGE_DIR}/firmadyne/preInit.sh" | |
chmod a+x "${IMAGE_DIR}/firmadyne/preInit.sh" | |
echo "----Unmounting QEMU Image----" | |
sync | |
umount "${DEVICE}" | |
kpartx -d "${IMAGE}" | |
losetup -d "${DEVICE}" &>/dev/null | |
dmsetup remove $(basename "$DEVICE") &>/dev/null |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment