171 lines
3.8 KiB
Bash
Executable File
171 lines
3.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
cd "$(dirname "$0")"
|
|
|
|
usage() {
|
|
cat <<EOF
|
|
Usage: do.sh <action> <project> <action specific arguments>
|
|
Supported actions:
|
|
build
|
|
build-lib
|
|
clean
|
|
clippy
|
|
doc
|
|
dump
|
|
fmt
|
|
test
|
|
update
|
|
Supported projects:
|
|
all
|
|
any directory containing a Cargo.toml file
|
|
EOF
|
|
}
|
|
|
|
sdkParentDir=bin
|
|
sdkDir="$sdkParentDir"/bpf-sdk
|
|
profile=bpfel-unknown-unknown/release
|
|
|
|
perform_action() {
|
|
set -e
|
|
projectDir="$PWD"/$2
|
|
targetDir="$projectDir"/target
|
|
case "$1" in
|
|
build)
|
|
if [[ -f "$projectDir"/Xargo.toml ]]; then
|
|
"$sdkDir"/rust/build.sh "$projectDir"
|
|
|
|
so_path="$targetDir/$profile"
|
|
so_name="spl_${2//\-/_}"
|
|
cp "$so_path/${so_name}.so" "$so_path/${so_name}_debug.so"
|
|
"$sdkDir"/dependencies/llvm-native/bin/llvm-objcopy --strip-all "$so_path/${so_name}.so" "$so_path/$so_name.so"
|
|
else
|
|
echo "$projectDir does not contain a program, skipping"
|
|
fi
|
|
;;
|
|
build-lib)
|
|
(
|
|
cd "$projectDir"
|
|
echo "build $projectDir"
|
|
export RUSTFLAGS="${@:3}"
|
|
cargo build
|
|
)
|
|
;;
|
|
clean)
|
|
"$sdkDir"/rust/clean.sh "$projectDir"
|
|
;;
|
|
clippy)
|
|
(
|
|
cd "$projectDir"
|
|
echo "clippy $projectDir"
|
|
cargo +nightly clippy --features=program ${@:3}
|
|
)
|
|
;;
|
|
doc)
|
|
(
|
|
cd "$projectDir"
|
|
echo "generating docs $projectDir"
|
|
cargo doc ${@:3}
|
|
)
|
|
;;
|
|
dump)
|
|
# Dump depends on tools that are not installed by default and must be installed manually
|
|
# - greadelf
|
|
# - rustfilt
|
|
(
|
|
pwd
|
|
"$0" build "$2"
|
|
|
|
so_path="$targetDir/$profile"
|
|
so_name="spl_${2//\-/_}"
|
|
so="$so_path/${so_name}_debug.so"
|
|
dump="$so_path/${so_name}_dump"
|
|
|
|
echo $so_path
|
|
echo $so_name
|
|
echo $so
|
|
echo $dump
|
|
|
|
if [ -f "$so" ]; then
|
|
ls \
|
|
-la \
|
|
"$so" \
|
|
>"${dump}_mangled.txt"
|
|
greadelf \
|
|
-aW \
|
|
"$so" \
|
|
>>"${dump}_mangled.txt"
|
|
"$sdkDir/dependencies/llvm-native/bin/llvm-objdump" \
|
|
-print-imm-hex \
|
|
--source \
|
|
--disassemble \
|
|
"$so" \
|
|
>>"${dump}_mangled.txt"
|
|
sed \
|
|
s/://g \
|
|
<"${dump}_mangled.txt" |
|
|
rustfilt \
|
|
>"${dump}.txt"
|
|
else
|
|
echo "Warning: No dump created, cannot find: $so"
|
|
fi
|
|
)
|
|
;;
|
|
fmt)
|
|
(
|
|
cd "$projectDir"
|
|
echo "formatting $projectDir"
|
|
cargo fmt ${@:3}
|
|
)
|
|
;;
|
|
help)
|
|
usage
|
|
exit
|
|
;;
|
|
test)
|
|
(
|
|
cd "$projectDir"
|
|
echo "test $projectDir"
|
|
cargo test --features=program ${@:3}
|
|
)
|
|
;;
|
|
update)
|
|
mkdir -p $sdkParentDir
|
|
./bpf-sdk-install.sh $sdkParentDir
|
|
./do.sh clean all
|
|
;;
|
|
*)
|
|
echo "Error: Unknown command"
|
|
usage
|
|
exit
|
|
;;
|
|
esac
|
|
}
|
|
|
|
set -e
|
|
if [[ $1 == "update" ]]; then
|
|
perform_action "$1"
|
|
exit
|
|
else
|
|
if [[ "$#" -lt 2 ]]; then
|
|
usage
|
|
exit
|
|
fi
|
|
if [[ ! -d "$sdkDir" ]]; then
|
|
./do.sh update
|
|
fi
|
|
fi
|
|
|
|
if [[ $2 == "all" ]]; then
|
|
# Perform operation on all projects
|
|
for project in */; do
|
|
if [[ -f "$project"Cargo.toml ]]; then
|
|
perform_action "$1" "${project%/}" ${@:3}
|
|
else
|
|
continue
|
|
fi
|
|
done
|
|
else
|
|
# Perform operation on requested project
|
|
perform_action "$1" "$2" "${@:3}"
|
|
fi
|