Improve guest program build scripts

paritytech · Nov 23, 2023 · ea6e91a · ea6e91a
1 parent a520c57
commit ea6e91a
Show file tree

Hide file tree

Showing 2 changed files with 74 additions and 14 deletions.
diff --git a/guest-programs/build-benchmarks.sh b/guest-programs/build-benchmarks.sh
@@ -1,28 +1,80 @@
 #!/bin/bash
 
 set -euo pipefail
+cd -- "$(dirname -- "${BASH_SOURCE[0]}")"
+
+source ../ci/jobs/detect-or-install-riscv-toolchain.sh
+
+if [ "${RV32E_TOOLCHAIN:-}" == "" ]; then
+    echo "WARN: rv32e toolchain is missing; PolkaVM binaries won't be built!"
+fi
+
+TOOLCHAIN_VERSION="1.72.1"
+
+BUILD_WASM=0
+BUILD_CKBVM=0
+BUILD_NATIVE_X86_64=0
+BUILD_NATIVE_X86=0
+
+if [[ "$(rustup toolchain list)" =~ "$TOOLCHAIN_VERSION" ]]; then
+    if [[ "$(rustup run $TOOLCHAIN_VERSION rustup target list --installed)" =~ "wasm32-unknown-unknown" ]]; then
+        BUILD_WASM=1
+    else
+        echo "WARN: the wasm32-unknown-unknown target is not installed; WASM binaries won't be built!"
+        echo "      You can add it with: rustup run $TOOLCHAIN_VERSION rustup target add wasm32-unknown-unknown"
+    fi
+
+    if [[ "$(rustup run $TOOLCHAIN_VERSION rustup target list --installed)" =~ "riscv64imac-unknown-none-elf" ]]; then
+        BUILD_CKBVM=1
+    else
+        echo "WARN: the riscv64imac-unknown-none-elf target is not installed; CKBVM binaries won't be built!"
+        echo "      You can add it with: rustup run $TOOLCHAIN_VERSION rustup target add riscv64imac-unknown-none-elf"
+    fi
+
+    if [[ "$(rustup run $TOOLCHAIN_VERSION rustc --print cfg)" =~ "target_os=\"linux\"" ]]; then
+        if [[ "$(rustup run $TOOLCHAIN_VERSION rustc --print cfg)" =~ "target_arch=\"x86_64\"" ]]; then
+            BUILD_NATIVE_X86_64=1
+            if [[ "$(rustup run $TOOLCHAIN_VERSION rustup target list --installed)" =~ "i686-unknown-linux-gnu" ]]; then
+                BUILD_NATIVE_X86=1
+            fi
+        fi
+    fi
+else
+    echo "WARN: Rust $TOOLCHAIN_VERSION is not installed; non-PolkaVM binaries won't be built!"
+    echo "      You can add it with: rustup toolchain install $TOOLCHAIN_VERSION"
+fi
 
 function build_benchmark() {
     current_dir=$(pwd)
     extra_flags="${extra_flags:-}"
 
-    echo "> Building: '$1' (polkavm)"
-    RUSTFLAGS="-C relocation-model=pie -C link-arg=--emit-relocs -C link-arg=-T.cargo/memory.ld $extra_flags" cargo build -q --release --bin $1 -p $1
-    cd ..
-    cargo run -q -p polkatool link guest-programs/target/riscv32em-unknown-none-elf/release/$1 -o guest-programs/target/riscv32em-unknown-none-elf/release/$1.polkavm
-    cd $current_dir
+    if [ "${RV32E_TOOLCHAIN:-}" != "" ]; then
+        echo "> Building: '$1' (polkavm)"
+        RUSTFLAGS="-C relocation-model=pie -C link-arg=--emit-relocs -C link-arg=-T.cargo/memory.ld $extra_flags" rustup run $RV32E_TOOLCHAIN cargo build -q --release --bin $1 -p $1
+        cd ..
+        cargo run -q -p polkatool link --run-only-if-newer guest-programs/target/riscv32em-unknown-none-elf/release/$1 -o guest-programs/target/riscv32em-unknown-none-elf/release/$1.polkavm
+        cd $current_dir
+    fi
 
-    echo "> Building: '$1' (wasm)"
-    RUSTFLAGS="-C target-cpu=mvp -C target-feature=-sign-ext $extra_flags" rustup run 1.72.1 cargo build -q --target=wasm32-unknown-unknown --release --bin $1 -p $1
+    if [ "${BUILD_WASM}" == "1" ]; then
+        echo "> Building: '$1' (wasm)"
+        RUSTFLAGS="-C target-cpu=mvp -C target-feature=-sign-ext $extra_flags" rustup run $TOOLCHAIN_VERSION cargo build -q --target=wasm32-unknown-unknown --release --bin $1 -p $1
+    fi
 
-    echo "> Building: '$1' (native, x86_64)"
-    RUSTFLAGS="$extra_flags" rustup run 1.72.1 cargo build -q --target=x86_64-unknown-linux-gnu --release --lib -p $1
+    if [ "${BUILD_NATIVE_X86_64}" == "1" ]; then
+        echo "> Building: '$1' (native, x86_64)"
+        RUSTFLAGS="$extra_flags" rustup run $TOOLCHAIN_VERSION cargo build -q --target=x86_64-unknown-linux-gnu --release --lib -p $1
+    fi
 
-    echo "> Building: '$1' (native, i686)"
-    RUSTFLAGS="$extra_flags" rustup run 1.72.1 cargo build -q --target=i686-unknown-linux-gnu --release --lib -p $1
+    if [ "${BUILD_NATIVE_X86}" == "1" ]; then
+        echo "> Building: '$1' (native, i686)"
+        RUSTFLAGS="$extra_flags" rustup run $TOOLCHAIN_VERSION cargo build -q --target=i686-unknown-linux-gnu --release --lib -p $1
+    fi
 
-    echo "> Building: '$1' (CKB VM)"
-    RUSTFLAGS="$extra_flags -C link-arg=-s --cfg=target_ckb_vm" rustup run 1.72.1 cargo build -q --target=riscv64imac-unknown-none-elf --release --bin $1 -p $1
+    if [ "${BUILD_CKBVM}" == "1" ]; then
+        echo "> Building: '$1' (CKB VM)"
+        RUSTFLAGS="$extra_flags -C link-arg=-s --cfg=target_ckb_vm" rustup run $TOOLCHAIN_VERSION cargo build -q --target=riscv64imac-unknown-none-elf --release --bin $1 -p $1
+    fi
 }
 
 build_benchmark "bench-pinky"

diff --git a/guest-programs/build-examples.sh b/guest-programs/build-examples.sh
@@ -1,6 +1,14 @@
 #!/bin/bash
 
 set -euo pipefail
+cd -- "$(dirname -- "${BASH_SOURCE[0]}")"
+
+source ../ci/jobs/detect-or-install-riscv-toolchain.sh
+
+if [ "${RV32E_TOOLCHAIN:-}" == "" ]; then
+    echo "ERROR: rv32e toolchain is missing; PolkaVM binaries can't be built!"
+    exit 1
+fi
 
 function build_example () {
     output_path="output/$1.polkavm"
@@ -9,7 +17,7 @@ function build_example () {
     echo "> Building: '$1' (-> $output_path)"
     RUSTFLAGS="-C relocation-model=pie -C link-arg=--emit-relocs -C link-arg=-T.cargo/memory.ld --remap-path-prefix=$(pwd)= --remap-path-prefix=$HOME=~" cargo build -q --release --bin $1 -p $1
     cd ..
-    cargo run -q -p polkatool link -s guest-programs/target/riscv32em-unknown-none-elf/release/$1 -o guest-programs/$output_path
+    cargo run -q -p polkatool link --run-only-if-newer -s guest-programs/target/riscv32em-unknown-none-elf/release/$1 -o guest-programs/$output_path
     cd $current_dir
 }