move atomics into core.rs

1 files changed, 15 insertions, 85 deletions

diff --git a/src/main.rs b/src/main.rs
index 965826e..48ce998 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -128,15 +128,15 @@ fn handle_instruction(state: &mut CpuState) {
             match f3 {
                 0 => {
                     sb(state, rs1, rs2, imm);
-                    release_reservations(state, &[addr]);
+                    state.release_reservations(&[addr]);
                 }
                 1 => {
                     sh(state, rs1, rs2, imm);
-                    release_reservations(state, &[addr, addr + 1]);
+                    state.release_reservations(&[addr, addr + 1]);
                 }
                 2 => {
                     sw(state, rs1, rs2, imm);
-                    release_reservations(state, &[addr, addr + 1, addr + 2, addr + 3]);
+                    state.release_reservations(&[addr, addr + 1, addr + 2, addr + 3]);
                 }
                 _ => panic!("Illegal instruction: {:x}", word),
             };
@@ -195,7 +195,7 @@ fn handle_instruction(state: &mut CpuState) {
         0b0010111 => {
             let imm = word >> 12;
             let rd = (word >> 7) & REGISTER_NAME_WIDTH;
-            auipc(state, *pc, rd as usize, imm);
+            auipc(state, rd as usize, imm);
         }
         0b1110011 => {
             // ecall
@@ -217,76 +217,17 @@ fn handle_instruction(state: &mut CpuState) {
             }
 
             match f5 {
-                2 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = number;
-                    acquire_reservations(reservations, &[addr, addr + 1, addr + 2, addr + 3]);
-                } // lr.w
-                3 => {
-                    let addr = state[rs1] as usize;
-                    if (*reservations).contains(&addr) {
-                        memory[addr..addr + 4].copy_from_slice(&state[rs2].to_le_bytes());
-                        state[rd] = 0;
-                    } else {
-                        state[rd] = 1;
-                    }
-                } // sc.w
-                1 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = number;
-                    store_word(addr, state[rs2]);
-                } // amoswap.w
-                0 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = (number as i32 + state[rs2] as i32) as u32;
-                    store_word(addr, state[rd]);
-                } // amoadd.w
-                12 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = number & state[rs2];
-                    store_word(addr, state[rd]);
-                } // amoand.w
-                8 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = number | state[rs2];
-                    store_word(addr, state[rd]);
-                } // amoor.w
-                4 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = number ^ state[rs2];
-                    store_word(addr, state[rd]);
-                } // amoxor.w
-                20 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = std::cmp::max(number as i32, state[rs2] as i32) as u32;
-                    store_word(addr, state[rd]);
-                } // amomax.w
-                16 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = std::cmp::min(number as i32, state[rs2] as i32) as u32;
-                    store_word(addr, state[rd]);
-                } // amomin.w
-                28 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = std::cmp::max(number, state[rs2]);
-                    store_word(addr, state[rd]);
-                } // amomaxu.w
-                24 => {
-                    let addr = state[rs1] as usize;
-                    let number: u32 = load_word(&addr);
-                    state[rd] = std::cmp::min(number, state[rs2]);
-                    store_word(addr, state[rd]);
-                } // amomin.w
-
+                2 => lr_w(state, rd, rs1),
+                3 => sc_w(state, rd, rs1, rs2),
+                1 => amoswap_w(state, rd, rs1, rs2),
+                0 => amoadd_w(state, rd, rs1, rs2),
+                12 => amoand_w(state, rd, rs1, rs2),
+                8 => amoor_w(state, rd, rs1, rs2),
+                4 => amoxor_w(state, rd, rs1, rs2),
+                20 => amomax_w(state, rd, rs1, rs2),
+                16 => amomin_w(state, rd, rs1, rs2),
+                28 => amomaxu_w(state, rd, rs1, rs2),
+                24 => amominu_w(state, rd, rs1, rs2),
                 _ => panic!("Illegal instruction: {:x}", word),
             }
         }
@@ -380,15 +321,4 @@ fn handle_instruction(state: &mut CpuState) {
     }
 
     state.pc += 4;
-}
-
-fn release_reservations(state: &mut CpuState, indices: &[usize]) {
-    for index in indices {
-        state.reservations.remove(&index);
-    }
-}
-fn acquire_reservations(state: &mut CpuState, indices: &[usize]) {
-    for index in indices {
-        state.reservations.insert(*index);
-    }
 }
\ No newline at end of file