#!/bin/bash
tcpPort=1235
imageDir=$RISCV/buildroot/output/images
tvDir=$RISCV/linux-testvectors
rawRamFile="$tvDir/ramGDB.bin"
ramFile="$tvDir/ram.bin"
rawBootmemFile="$tvDir/bootmemGDB.bin"
bootmemFile="$tvDir/bootmem.bin"
rawUntrimmedBootmemFile="$tvDir/untrimmedBootmemFileGDB.bin"
untrimmedBootmemFile="$tvDir/untrimmedBootmemFile.bin"

read -p "Warning: running this script will overwrite the contents of:
  * $rawRamFile
  * $ramFile
  * $rawBootmemFile
  * $bootmemFile
  * $rawUntrimmedBootmemFile
  * $untrimmedBootmemFile
Would you like to proceed? (y/n) " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]
then
    # Create Output Directory
    echo "Elevating permissions to create memory files"
    sudo mkdir -p $tvDir
    sudo chown cad $tvDir
    sudo touch $rawRamFile 
    sudo touch $ramFile 
    sudo touch $rawBootmemFile
    sudo touch $bootmemFile
    sudo touch $rawUntrimmedBootmemFile
    sudo touch $untrimmedBootmemFile
    sudo chmod a+rw $rawRamFile
    sudo chmod a+rw $ramFile
    sudo chmod a+rw $rawBootmemFile
    sudo chmod a+rw $bootmemFile
    sudo chmod a+rw $rawUntrimmedBootmemFile
    sudo chmod a+rw $untrimmedBootmemFile

    # QEMU Simulation
    echo "Launching QEMU in replay mode!"
    (qemu-system-riscv64 \
    -M virt -dtb $RISCV/buildroot/output/images/wally-virt.dtb \
    -nographic \
    -bios $imageDir/fw_jump.elf -kernel $imageDir/Image -append "root=/dev/vda ro" -initrd $imageDir/rootfs.cpio \
    -gdb tcp::$tcpPort -S) \
    & riscv64-unknown-elf-gdb --quiet \
    -ex "set pagination off" \
    -ex "set logging overwrite on" \
    -ex "set logging redirect on" \
    -ex "set confirm off" \
    -ex "target extended-remote :$tcpPort" \
    -ex "maintenance packet Qqemu.PhyMemMode:1" \
    -ex "printf \"Creating $rawBootmemFile\n\"" \
    -ex "dump binary memory $rawBootmemFile 0x1000 0x1fff" \
    -ex "printf \"Creating $rawUntrimmedBootmemFile\n\"" \
    -ex "printf \"Warning - please verify that the second half of $rawUntrimmedBootmemFile is all 0s\n\"" \
    -ex "dump binary memory $rawUntrimmedBootmemFile 0x1000 0x2fff" \
    -ex "printf \"Creating $rawRamFile\n\"" \
    -ex "dump binary memory $rawRamFile 0x80000000 0xffffffff" \
    -ex "kill" \
    -ex "q"

    echo "Changing Endianness"
    make fixBinMem
    ./fixBinMem "$rawRamFile" "$ramFile"
    ./fixBinMem "$rawBootmemFile" "$bootmemFile"
    ./fixBinMem "$rawUntrimmedBootmemFile" "$untrimmedBootmemFile"

    # Cleanup
    echo "Elevating permissions to restrict write access to memory files"
    sudo chown cad $rawRamFile
    sudo chown cad $ramFile
    sudo chown cad $rawBootmemFile
    sudo chown cad $bootmemFile
    sudo chown cad $rawUntrimmedBootmemFile
    sudo chown cad $untrimmedBootmemFile
    sudo chmod o-w $rawRamFile
    sudo chmod o-w $ramFile
    sudo chmod o-w $rawBootmemFile
    sudo chmod o-w $bootmemFile
    sudo chmod o-w $rawUntrimmedBootmemFile
    sudo chmod o-w $untrimmedBootmemFile
fi