The code is simplified

This commit is contained in:
q3aql 2024-07-07 21:59:58 +02:00
parent 62101b193f
commit 989550aea1

View File

@ -3,141 +3,92 @@
set -e set -e
# These vars will be modified automatically with sed # These vars will be modified automatically with sed
run_mode=packer
compress_flag=-z compress_flag=-z
program=
ld_so=
load() { unpack() {
script_path="$0" filename=$1
echo "#!/usr/bin/env sh" > $filename
while link_path=$(readlink "$script_path"); do echo "#set -o pipefail # bash extension" >> $filename
script_path="$link_path" echo "set -e" >> $filename
done echo "" >> $filename
echo "compress_flag=$compress_flag" >> $filename
unpack_dir=$(dirname "$script_path") echo "executable_run=$2" >> $filename
echo "tmp_parent=/tmp" >> $filename
exec "$unpack_dir/$program.res/$ld_so" \ echo 'if [ "$1" = "--packelf-extract" ] ; then' >> $filename
--library-path "$unpack_dir/$program.res" \ echo ' mkdir -p packelf-files' >> $filename
"$unpack_dir/$program.res/$program" "$@" echo " echo \"Extracting to 'packelf-files'\"" >> $filename
# unreachable echo " sed '1,/^#__END__\$/d' \"\$0\" | tar \$compress_flag -x -C packelf-files" >> $filename
} echo " exit 0" >> $filename
echo "fi" >> $filename
if [ "$run_mode" = loader ]; then echo 'mkdir -p "$tmp_parent"' >> $filename
load "$@" echo 'unpack_dir=$(mktemp -d -p "$tmp_parent" || echo "$tmp_parent")' >> $filename
fi echo "sed '1,/^#__END__\$/d' \"\$0\" | tar \$compress_flag -x -C \"\$unpack_dir\"" >> $filename
echo "chmod 777 -R \"\$unpack_dir\"/* 2> /dev/null" >> $filename
pack_help() { echo '"$unpack_dir/$executable_run" "$@"' >> $filename
echo "$0 <ELF_SRC_PATH> <DST_PATH> [ADDITIONAL_LIBS]" echo "rm -rf \$unpack_dir" >> $filename
echo "exit 0" >> $filename
echo "#__END__" >> $filename
} }
pack() { pack() {
[ $# -ge 2 ] || { folder=$1
pack_help filename=$2
exit 1 executable_run=$3
} temp_file=$(mktemp)
current_dir=$(pwd)
if [ ! -d "$folder" ] ; then
echo "Folder $folder does not exist"
exit 0
fi
#echo "Creating executable $2"
unpack $filename $executable_run
cd $folder
chmod 777 -R * 2> /dev/null
tar $compress_flag -c -f $temp_file *
cd ${current_dir}
cat $temp_file >> $filename
rm -rf $temp_file
echo "Created successfully"
}
case $1 in help() {
-h|--help) echo "$0 <ELF_SRC_PATH> <DST_PATH>"
pack_help exit 0
exit 0 }
;;
*)
;;
esac
src="$1" if [ ! -z "$1" ] ; then
shift if [ ! -f "$1" ] ; then
dst="$1" echo "$0: File $1 does not exist"
shift exit 1
else
libs="$(ldd "$src" | grep -F '/' | sed -E 's|[^/]*/([^ ]+).*?|/\1|')" libs="$(ldd "$1" | grep -F '/' | sed -E 's|[^/]*/([^ ]+).*?|/\1|')"
ld_so="$(echo "$libs" | grep -F '/ld-linux-' || echo "$libs" | grep -F '/ld-musl-' || echo "$libs" | grep -F '/ld.so')" ld_so="$(echo "$libs" | grep -F '/ld-linux-' || echo "$libs" | grep -F '/ld-musl-' || echo "$libs" | grep -F '/ld.so')"
ld_so="$(basename "$ld_so")" ld_so="$(basename "$ld_so")"
program="$(basename "$src")" program="$(basename "$1")"
if [ -z "$libs" ] ; then
cat "$0" | sed -E \ echo "$0: Not a dynamic executable"
-e 's/^run_mode=[^ ]*$/run_mode=unpacker/' \ exit 1
-e 's/^compress_flag=[^ ]*$/compress_flag='"$compress_flag"'/' \
-e 's/^program=[^ ]*$/program='"$program"'/' \
-e 's/^ld_so=[^ ]*$/ld_so='"$ld_so"'/' \
>"$dst"
echo "Creating static executable $dst from $src"
for libraries in ${libs} ; do
echo "Linking library ${libraries}"
done
echo "Creating executable linker"
echo "Building static executable in $dst"
tar $compress_flag -ch --transform 's@.*/@'"$program"'.res/@' "$src" $libs "$@" >>"$dst"
chmod +x "$dst"
echo "Created successfully"
#echo "'$src' was packed to '$dst'"
echo ""
echo "$dst" | grep -q / || dst="./$dst"
echo "Just run '$dst <ARGS>' to execute the command."
echo "Or run 'PACKELF_UNPACK_DIR=xxx $dst' to unpack it only."
}
unpack() {
if [ -n "$PACKELF_UNPACK_DIR" ]; then
[ -d "$PACKELF_UNPACK_DIR" ] || {
echo "'$PACKELF_UNPACK_DIR' is not a dir."
exit 1
}
[ -e "$PACKELF_UNPACK_DIR/$program" ] && {
echo "'$PACKELF_UNPACK_DIR/$program' already exists, please remove it first."
exit 1
}
unpack_dir="$PACKELF_UNPACK_DIR"
else
if [ -n "$PACKELF_TMP_DIR" ]; then
unpack_dir="$PACKELF_TMP_DIR"
else
tmp_parent=/tmp
mkdir -p "$tmp_parent"
unpack_dir=$(mktemp -d -p "$tmp_parent" || echo "$tmp_parent")
fi
trap 'rm -rf "$unpack_dir"' 0 1 2 3 6 10 12 13 14 15
fi fi
fi
check_path="$unpack_dir/__check_permission__"
if ! (echo > "$check_path" && chmod +x "$check_path" && [ -x "$check_path" ]); then
rm -rf "$unpack_dir"
tmp_parent="$(pwd)/packelf_tmp"
mkdir -p "$tmp_parent"
unpack_dir="$(mktemp -d -p "$tmp_parent" || echo "$tmp_parent")"
fi
sed '1,/^#__END__$/d' "$0" | tar $compress_flag -x -C "$unpack_dir"
sed -i 's@/etc/ld.so.preload@/etc/___so.preload@g' "$unpack_dir/$program.res/$ld_so"
if [ -n "$PACKELF_UNPACK_DIR" ]; then
sed '/^#__END__$/,$d' "$0" > "$unpack_dir/$program"
sed -i -E 's/^run_mode=\w*$/run_mode=loader/' "$unpack_dir/$program"
chmod +x "$unpack_dir/$program"
rm -f "$check_path"
echo "'$program' was unpacked to '$unpack_dir'."
echo "$unpack_dir" | grep -q / || unpack_dir="./$unpack_dir"
echo "You can run '$unpack_dir/$program ARGS...' to execute the command."
else
"$unpack_dir/$program.res/$ld_so" \
--library-path "$unpack_dir/$program.res" \
"$unpack_dir/$program.res/$program" "$@"
exit $?
fi
}
if [ "$run_mode" = unpacker ]; then
unpack "$@"
else else
pack "$@" help
exit 0
fi
if [ -z "$2" ] ; then
help
else
temp_dir=$(mktemp -d)
echo "Creating static binary ${2} from ${1}"
echo "Linking libraries"
cp -L ${1} ${temp_dir}/
for libraries in ${libs} ; do
cp -L ${libraries} ${temp_dir}/
done
echo "#!/usr/bin/env sh" > ${temp_dir}/AppRun
echo "" >> ${temp_dir}/AppRun
echo "\$(dirname \$0)/${ld_so} --library-path \$(dirname \$0) \$(dirname \$0)/${program} \"\$@\"" >> ${temp_dir}/AppRun
chmod 777 -R "$temp_dir"
pack "$temp_dir" "$2" AppRun
fi fi
exit 0
#__END__