From 9d42bc32284b39c7a3ff91f17e05dd5dd7eece29 Mon Sep 17 00:00:00 2001
From: MerryMage <MerryMage@users.noreply.github.com>
Date: Wed, 17 Jan 2018 00:35:01 +0000
Subject: [PATCH] tests/A64: Single random instruction: Test branch
 instructions as well

---
 tests/A64/fuzz_with_unicorn.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/A64/fuzz_with_unicorn.cpp b/tests/A64/fuzz_with_unicorn.cpp
index ae9b5980..795ab2fd 100644
--- a/tests/A64/fuzz_with_unicorn.cpp
+++ b/tests/A64/fuzz_with_unicorn.cpp
@@ -103,7 +103,7 @@ static std::vector<InstructionGenerator> instruction_generators = []{
     return result;
 }();
 
-static u32 GenRandomInst(u64 pc) {
+static u32 GenRandomInst(u64 pc, bool is_last_inst) {
     const A64::LocationDescriptor location{pc, {}};
 
 restart:
@@ -112,7 +112,7 @@ restart:
 
     IR::Block block{location};
     bool should_continue = A64::TranslateSingleInstruction(block, location, instruction);
-    if (!should_continue)
+    if (!should_continue && !is_last_inst)
         goto restart;
     for (const auto& ir_inst : block)
         if (ir_inst.IsMemoryWrite() || ir_inst.GetOpcode() == IR::Opcode::A64ExceptionRaised || ir_inst.GetOpcode() == IR::Opcode::A64CallSupervisor)
@@ -159,7 +159,7 @@ TEST_CASE("A64: Single random instruction", "[a64]") {
         std::array<u64, 31> regs;
         std::generate_n(regs.begin(), 31, []{ return RandInt<u64>(0, ~u64(0)); });
         std::vector<u32> instructions;
-        instructions.push_back(GenRandomInst(0));
+        instructions.push_back(GenRandomInst(0, true));
         u32 pstate = RandInt<u32>(0, 0xF) << 28;
 
         // printf("%08x\n", instructions[0]);