mirror of https://github.com/CIRCL/Circlean
88 lines
2.6 KiB
Bash
Executable File
88 lines
2.6 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
clean(){
|
|
if [ "${DEBUG}" = true ]; then
|
|
sleep 20
|
|
fi
|
|
|
|
# Write anything in memory to disk
|
|
${SYNC}
|
|
|
|
# Remove temporary files from destination key
|
|
rm -rf "${TEMP}"
|
|
}
|
|
|
|
check_not_root() {
|
|
if ! [ "${ID}" -ge "1000" ]; then
|
|
echo "GROOMER: groomer.sh cannot run as root."
|
|
exit
|
|
fi
|
|
}
|
|
|
|
check_has_partitions () {
|
|
local partitions=$1
|
|
if [ -z "${partitions}" ]; then
|
|
echo "GROOMER: ${SRC_DEV} does not have any partitions."
|
|
exit
|
|
fi
|
|
}
|
|
|
|
unmount_source_partition() {
|
|
# Unmount anything that is mounted on /media/src
|
|
if [ "$(${MOUNT} | grep -c "${SRC_MNT}")" -ne 0 ]; then
|
|
${PUMOUNT} "${SRC_MNT}"
|
|
fi
|
|
}
|
|
|
|
run_groomer() {
|
|
local dev_partitions
|
|
# Find the partition names on the device
|
|
dev_partitions=$(ls "${SRC_DEV}"* | grep "${SRC_DEV}[1-9][0-6]*" || true)
|
|
check_has_partitions dev_partitions
|
|
local partcount=1
|
|
local partition
|
|
for partition in ${dev_partitions}
|
|
do
|
|
echo "GROOMER: Processing partition ${partition}"
|
|
unmount_source_partition
|
|
# Mount the current partition in write mode
|
|
${PMOUNT} -w ${partition} "${SRC_MNT}"
|
|
# Mark any autorun.inf files as dangerous on the source device to be extra careful
|
|
ls "${SRC_MNT}" | grep -i autorun.inf | xargs -I {} mv "${SRC_MNT}"/{} "${SRC_MNT}"/DANGEROUS_{}_DANGEROUS || true
|
|
# Unmount and remount the current partition in read-only mode
|
|
${PUMOUNT} "${SRC_MNT}"
|
|
|
|
if ${PMOUNT} -r "${partition}" "${SRC_MNT}"; then
|
|
echo "GROOMER: ${partition} mounted at ${SRC_MNT}"
|
|
|
|
# Create a directory on ${DST_MNT} named PARTION_$PARTCOUNT
|
|
local target_dir="${DST_MNT}/FROM_PARTITION_${partcount}"
|
|
mkdir -p "${target_dir}"
|
|
|
|
# Run the current partition through filecheck.py
|
|
filecheck.py --source "${SRC_MNT}" --destination "${target_dir}" || true
|
|
|
|
# List destination files (recursively) for debugging
|
|
if [ "${DEBUG}" = true ]; then
|
|
ls -lR "${target_dir}"
|
|
fi
|
|
else
|
|
# Previous command (mounting current partition) failed
|
|
echo "GROOMER: Unable to mount ${partition} on ${SRC_MNT}"
|
|
fi
|
|
let partcount=$((partcount + 1))
|
|
done
|
|
}
|
|
|
|
main() {
|
|
set -eu # exit when a line returns non-0 status, treat unset variables as errors
|
|
trap clean EXIT TERM INT # run clean when the script ends or is interrupted
|
|
source ./config.sh # get config values
|
|
if [ "${DEBUG}" = true ]; then
|
|
set -x
|
|
fi
|
|
run_groomer
|
|
}
|
|
|
|
main
|