From d7dfab863ada652e95a4bd57c53b6038cb6b7648 Mon Sep 17 00:00:00 2001
From: David Mak <csdavidmak@ust.hk>
Date: Fri, 13 Oct 2023 14:36:03 +0800
Subject: [PATCH] standalone: Fix run_demo script

- Link main and module*.bc together if using multiple threads
- Fix temporary files not being deleted
---
 nac3standalone/demo/run_demo.sh | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/nac3standalone/demo/run_demo.sh b/nac3standalone/demo/run_demo.sh
index a4f5b08..b93dfdb 100755
--- a/nac3standalone/demo/run_demo.sh
+++ b/nac3standalone/demo/run_demo.sh
@@ -31,9 +31,8 @@ else
     nac3standalone=../../target/x86_64-unknown-linux-gnu/release/nac3standalone
 fi
 
+rm -f ./*.o ./*.bc demo
 if [ -z "$use_lli" ]; then
-  rm -f "*.o" demo
-
   $nac3standalone "${nac3args[@]}"
 
   clang -c -std=gnu11 -Wall -Wextra -O3 -o demo.o demo.c
@@ -45,15 +44,19 @@ if [ -z "$use_lli" ]; then
     ./demo > "$outfile"
   fi
 else
-  rm -f "*.o" "*.bc" demo
-
   $nac3standalone --emit-llvm "${nac3args[@]}"
 
   clang -c -std=gnu11 -Wall -Wextra -O3 -emit-llvm -o demo.bc demo.c
 
-  if [ -z "$outfile" ]; then
-    lli --extra-module demo.bc --extra-module irrt.bc main.bc
+  if ls module*.bc >/dev/null 2>&1; then
+    llvm-link -o module.bc module*.bc main.bc
   else
-    lli --extra-module demo.bc --extra-module irrt.bc main.bc > "$outfile"
+    llvm-link -o module.bc main.bc
+  fi
+
+  if [ -z "$outfile" ]; then
+    lli --extra-module demo.bc --extra-module irrt.bc module.bc
+  else
+    lli --extra-module demo.bc --extra-module irrt.bc module.bc > "$outfile"
   fi
 fi