From 0646bf2b90b817dfc834be5ecc9bf0be7550c1c2 Mon Sep 17 00:00:00 2001
From: bbracker <bbracker@hmc.edu>
Date: Mon, 6 Sep 2021 16:23:45 -0400
Subject: [PATCH] help in case a script is run accidentally

---
 .../testvector-generation/CreateTrace.sh      |  9 ++++++-
 .../testvector-generation/logBuildrootMem.sh  | 24 ++++++++++++-------
 2 files changed, 23 insertions(+), 10 deletions(-)

diff --git a/wally-pipelined/linux-testgen/testvector-generation/CreateTrace.sh b/wally-pipelined/linux-testgen/testvector-generation/CreateTrace.sh
index 7fcf340fe..c01fc4357 100755
--- a/wally-pipelined/linux-testgen/testvector-generation/CreateTrace.sh
+++ b/wally-pipelined/linux-testgen/testvector-generation/CreateTrace.sh
@@ -15,4 +15,11 @@ outDir="../linux-testvectors"
 
 # - Logs info needed by buildroot testbench
 
-($customQemu -M virt -nographic -bios $imageDir/fw_jump.elf -kernel $imageDir/Image -append "root=/dev/vda ro" -initrd $imageDir/rootfs.cpio -rtc clock=vm -icount shift=1 -d nochain,cpu,in_asm -serial /dev/null -singlestep -gdb tcp::1236 -S 2>&1 >/dev/null | ./parse_qemu.py | ./parseNew.py | ./remove_dup.awk > $outDir/all.txt) & riscv64-unknown-elf-gdb -x gdbinit_qemulog
+read -p "Warning: running this script will overwrite the contents of $outDir/all.txt.
+Would you like to proceed? (y/n) " -n 1 -r
+echo
+if [[ $REPLY =~ ^[Yy]$ ]]
+then
+    ($customQemu -M virt -nographic -bios $imageDir/fw_jump.elf -kernel $imageDir/Image -append "root=/dev/vda ro" -initrd $imageDir/rootfs.cpio -rtc clock=vm -icount shift=1 -d nochain,cpu,in_asm -serial /dev/null -singlestep -gdb tcp::1236 -S 2>&1 >/dev/null | ./parse_qemu.py | ./parseNew.py | ./remove_dup.awk > $outDir/all.txt) & riscv64-unknown-elf-gdb -x gdbinit_qemulog
+fi
+
diff --git a/wally-pipelined/linux-testgen/testvector-generation/logBuildrootMem.sh b/wally-pipelined/linux-testgen/testvector-generation/logBuildrootMem.sh
index 5935de93d..978311d92 100755
--- a/wally-pipelined/linux-testgen/testvector-generation/logBuildrootMem.sh
+++ b/wally-pipelined/linux-testgen/testvector-generation/logBuildrootMem.sh
@@ -8,14 +8,20 @@ customQemu="/courses/e190ax/qemu_sim/rv64_initrd/qemu_experimental/qemu/build/qe
 imageDir="../buildroot-image-output"
 testVecDir="../linux-testvectors"
 
-($customQemu -M virt -nographic -bios $imageDir/fw_jump.elf -kernel $imageDir/Image -append "root=/dev/vda ro" -initrd $imageDir/rootfs.cpio -d nochain,cpu,in_asm -serial /dev/null -singlestep -gdb tcp::1235 -S 2>/dev/null >/dev/null) &
-riscv64-unknown-elf-gdb -x gdbinit_mem
-echo "Translating Mem from GDB to Questa format"
-./fix_mem.py
-echo "Done"
+read -p "Warning: running this script will overwrite the contents of memory dumps needed for simulation.
+Would you like to proceed? (y/n) " -n 1 -r
+echo
+if [[ $REPLY =~ ^[Yy]$ ]]
+then
+    ($customQemu -M virt -nographic -bios $imageDir/fw_jump.elf -kernel $imageDir/Image -append "root=/dev/vda ro" -initrd $imageDir/rootfs.cpio -d nochain,cpu,in_asm -serial /dev/null -singlestep -gdb tcp::1235 -S 2>/dev/null >/dev/null) &
+    riscv64-unknown-elf-gdb -x gdbinit_mem
+    echo "Translating Mem from GDB to Questa format"
+    ./fix_mem.py
+    echo "Done"
 
-echo "Creating debugging objdump of linux image"
+    echo "Creating debugging objdump of linux image"
 
-riscv64-unknown-elf-objdump -D $imageDir/vmlinux > $testVecDir/vmlinux.objdump
-extractFunctionRadix.sh $testVecDir/vmlinux.objdump
-echo "Done"
+    riscv64-unknown-elf-objdump -D $imageDir/vmlinux > $testVecDir/vmlinux.objdump
+    extractFunctionRadix.sh $testVecDir/vmlinux.objdump
+    echo "Done"
+fi