1 files changed, 23 insertions, 16 deletions

diff --git a/cpu.c b/cpu.c
index df9c852..bba0ed5 100644
--- a/cpu.c
+++ b/cpu.c
@@ -1,4 +1,5 @@
 #include <libgen.h>
+#include <stdbool.h>
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
@@ -151,6 +152,20 @@ opcode_mem(enum addressing_mode mode)
         return val;
 }
 
+static void
+branch(uint16_t addr, bool cond)
+{
+        if (!cond)
+                return;
+        cycles++;
+
+        addr += regs.pc;
+        if ((regs.pc + 1) & 0xFF00 != addr & 0xFF00)
+                cycles++;
+
+        regs.pc = addr;
+}
+
 void
 ADC(uint16_t arg)
 {
@@ -201,22 +216,19 @@ ASL(uint16_t mem)
 void
 BCC(uint16_t arg)
 {
-        if (regs.status.carry == 0)
-                regs.pc += arg, cycles++;
+        branch(arg, regs.status.carry == 0);
 }
 
 void
 BCS(uint16_t arg)
 {
-        if (regs.status.carry == 1)
-                regs.pc += arg, cycles++;
+        branch(arg, regs.status.carry == 1);
 }
 
 void
 BEQ(uint16_t arg)
 {
-        if (regs.status.zero == 1)
-                regs.pc += arg, cycles++;
+        branch(arg, regs.status.zero == 1);
 }
 
 void
@@ -232,22 +244,19 @@ BIT(uint16_t arg)
 void
 BMI(uint16_t arg)
 {
-        if (regs.status.negative == 1)
-                regs.pc += arg, cycles++;
+        branch(arg, regs.status.negative == 1);
 }
 
 void
 BNE(uint16_t arg)
 {
-        if (regs.status.zero == 0)
-                regs.pc += arg, cycles++;
+        branch(arg, regs.status.zero == 0);
 }
 
 void
 BPL(uint16_t arg)
 {
-        if (regs.status.negative == 0)
-                regs.pc += arg, cycles++;
+        branch(arg, regs.status.negative == 0);
 }
 
 void
@@ -263,16 +272,14 @@ BVC(uint16_t arg)
 {
         //regs.status.overflow = (STATUS_TO_INT() & (1 << 6)) == 0;
         //if (regs.status.overflow == 0)
-        if ((STATUS_TO_INT() & (1 << 6)) == 0)
-                regs.pc += arg, cycles++;
+        branch(arg, (STATUS_TO_INT() & (1 << 6)) == 0);
 }
 
 void
 BVS(uint16_t arg)
 {
         regs.status.overflow = (STATUS_TO_INT() & (1 << 6)) != 0;
-        if (regs.status.overflow == 1)
-                regs.pc += arg, cycles++;
+        branch(arg, regs.status.overflow == 1);
 }
 
 void