You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
160 lines
3.4 KiB
160 lines
3.4 KiB
#!/bin/sh -e
|
|
|
|
# NEEDED: lz4 readelf tar cat awk cut sed sh sort
|
|
|
|
WORKDIR="$(basename "$1")"
|
|
if [ -z "$LIBPATH" ]; then
|
|
LIBPATH='/lib'
|
|
fi
|
|
PAYLOAD_SCRIPT='payload'
|
|
LIBLIST="$WORKDIR"/liblist
|
|
SUMFILE="$WORKDIR"/checksum
|
|
tmpfile="$WORKDIR"/tt
|
|
|
|
die ()
|
|
{
|
|
echo "$1"
|
|
exit
|
|
}
|
|
|
|
getlibs ()
|
|
{
|
|
tmplist="$WORKDIR"/tmplist
|
|
|
|
touch "$LIBLIST"
|
|
|
|
# Take the executable and get the first level dependencies
|
|
readelf -d "$1" |
|
|
grep NEEDED |
|
|
awk '{print $5}' |
|
|
sed s/\\[/\ / | sed s/\\]/\ / > "$tmplist"
|
|
|
|
# While there are more dependencies
|
|
while [ "$(cat "$tmplist")" ]; do
|
|
# Add them to the master file
|
|
cat "$LIBLIST" "$tmplist" > "$tmpfile"
|
|
mv -f "$tmpfile" "$LIBLIST"
|
|
|
|
# Copy them to lib/
|
|
while read -r l; do
|
|
( IFS=:
|
|
for p in $LIBPATH; do
|
|
find -L "$p" -maxdepth 2 -name "$l" -exec cp -n {} "$WORKDIR"/lib/ \;
|
|
done
|
|
)
|
|
done < "$tmplist"
|
|
|
|
# Extract their dependencies
|
|
touch "$tmpfile"
|
|
for f in "$WORKDIR"/lib/*; do
|
|
if grep -Fq "$(basename "$f")" "$tmplist"; then
|
|
readelf -d "$f" |
|
|
grep NEEDED |
|
|
awk '{print $5}' |
|
|
sed s/\\[/\ / | sed s/\\]/\ / >> "$tmpfile"
|
|
fi
|
|
done
|
|
sort -u "$tmpfile" > "$tmplist"
|
|
done
|
|
sort -u "$LIBLIST" > "$tmpfile"
|
|
mv -f "$tmpfile" "$LIBLIST"
|
|
rm -f "$tmplist"
|
|
}
|
|
|
|
if ! [ "$WORKDIR" ]; then
|
|
die 'Not enough arguments'
|
|
fi
|
|
|
|
if ! [ -d "$WORKDIR" ]; then
|
|
die 'Argument is not a directory'
|
|
fi
|
|
|
|
if [ -e "$WORKDIR"/bin ]; then
|
|
if ! [ -d "$WORKDIR"/bin ]; then
|
|
die 'bin/ is not a directory'
|
|
fi
|
|
else
|
|
die 'Directory does not contain bin/'
|
|
fi
|
|
|
|
if ! [ -e "$WORKDIR"/id ]; then
|
|
die 'id file not present'
|
|
fi
|
|
|
|
if ls -1 "$WORKDIR"/bin; then
|
|
for f in "$WORKDIR"/bin/*; do
|
|
if ! [ -x "$f" ]; then
|
|
die "$f is not executable"
|
|
fi
|
|
done
|
|
else
|
|
die 'bin/ is empty, there has to be at least one executable'
|
|
fi
|
|
|
|
# Create necessary directories and files
|
|
mkdir -p "$WORKDIR"/lib
|
|
mkdir -p "$WORKDIR"/ext
|
|
mkdir -p "$WORKDIR"/man
|
|
touch "$WORKDIR"/env
|
|
|
|
# TODO: add usage and error checking
|
|
DIRNAME="$WORKDIR"/"$(head -1 "$WORKDIR"/id | awk '{print $1}')"
|
|
|
|
echo "Fetching binary dependencies..."
|
|
if [ -s "$WORKDIR"/deps ]; then
|
|
while read -r d; do
|
|
( IFS=:
|
|
for p in $PATH; do
|
|
find -L "$p" -maxdepth 2 -name "$d" -exec cp -n {} "$WORKDIR"/bin/ \;
|
|
done
|
|
)
|
|
done < "$tmplist"
|
|
fi
|
|
|
|
echo "Fetching dependencies and stripping binaries..."
|
|
# TODO: get al binary dependencies from deps file
|
|
# Get all dependencies and strip them
|
|
for b in "$WORKDIR"/bin/*; do
|
|
getlibs "$b"
|
|
strip "$b"
|
|
done
|
|
echo "Stripping libraries..."
|
|
for l in "$WORKDIR"/lib/*; do
|
|
strip "$l"
|
|
done
|
|
|
|
echo "Compressing destination directory..."
|
|
if [ -d "$DIRNAME" ]; then
|
|
rm -rf "$DIRNAME"
|
|
fi
|
|
mkdir -p "$DIRNAME"
|
|
|
|
cp -r "$WORKDIR"/bin "$DIRNAME"/
|
|
cp -r "$WORKDIR"/lib "$DIRNAME"/
|
|
cp -r "$WORKDIR"/man "$DIRNAME"/
|
|
cp -r "$WORKDIR"/ext "$DIRNAME"/
|
|
cp "$WORKDIR"/env "$DIRNAME"/
|
|
cp "$WORKDIR"/id "$DIRNAME"/
|
|
|
|
tar -c -f "$DIRNAME".tar -C "$WORKDIR" "$(basename "$DIRNAME")"
|
|
md5sum -b "$DIRNAME".tar | cut -d " " -f1 > "$SUMFILE"
|
|
lz4 --rm -9 -c "$DIRNAME.tar" > "$DIRNAME".tar.lz4
|
|
|
|
echo "Injecting payload..."
|
|
sed s/SUM/"$(cat "$SUMFILE")"/ "$PAYLOAD_SCRIPT" |
|
|
sed s/ID/"$(basename "$DIRNAME")"/ |
|
|
awk '!/^ *#/ && NF' > tpp
|
|
|
|
size="$(wc -c tpp | cut -d " " -f1)"
|
|
strsize="$(echo "BYTES" | wc -c)"
|
|
numsize="$(echo "$size" | wc -c)"
|
|
normsize="$((size - (strsize - numsize) + 1))"
|
|
|
|
sed s/BYTES/"$normsize"/ tpp > tpr
|
|
cat tpr "$DIRNAME".tar.lz4 > "$DIRNAME".ti
|
|
|
|
echo "Cleaning up..."
|
|
rm -f tpr tpp
|
|
chmod +x "$DIRNAME".ti
|
|
|
|
rm -rf "$DIRNAME" "$DIRNAME".tar*
|
|
|