chore(cart): refactor sections of cartridge code

This commit is contained in:
Rekai Nyangadzayi Musuka 2021-11-25 03:31:49 -04:00
parent 7f872a554b
commit ef4f3d9ec6
2 changed files with 158 additions and 172 deletions

View File

@ -35,8 +35,8 @@ impl Cartridge {
self.mbc.ext_ram() self.mbc.ext_ram()
} }
pub(crate) fn write_ext_ram(&mut self, mem: Vec<u8>) { pub(crate) fn ext_ram_mut(&mut self) -> Option<&mut [u8]> {
self.mbc.write_ext_ram(mem) self.mbc.ext_ram_mut()
} }
#[inline] #[inline]
@ -57,14 +57,15 @@ impl Cartridge {
match mbc_kind { match mbc_kind {
MBCKind::None => Box::new(NoMBC), MBCKind::None => Box::new(NoMBC),
MBCKind::MBC1 => Box::new(MBC1::new(ram_size, rom_size)), MBCKind::MBC1(hw) => Box::new(MBC1::new(hw, ram_size, rom_size)),
MBCKind::MBC1WithBattery => Box::new(MBC1::with_battery(ram_size, rom_size)), MBCKind::MBC2(hw) => Box::new(MBC2::new(hw, rom_cap)),
MBCKind::MBC2 => Box::new(MBC2::new(rom_cap)), MBCKind::MBC3(hw @ MBC3Hardware::RTC) => Box::new(MBC3::new(hw, ram_cap)),
MBCKind::MBC2WithBattery => Box::new(MBC2::with_battery(rom_cap)), MBCKind::MBC3(hw @ MBC3Hardware::RTCAndBatteryRAM) => Box::new(MBC3::new(hw, ram_cap)),
MBCKind::MBC3 => Box::new(MBC3::new(ram_cap)), MBCKind::MBC5(hw @ MBC5Hardware::None) => Box::new(MBC5::new(hw, ram_cap, rom_cap)),
MBCKind::MBC3WithBattery => Box::new(MBC3::with_battery(ram_cap)), MBCKind::MBC5(hw @ MBC5Hardware::BatteryRAM) => {
MBCKind::MBC5 => Box::new(MBC5::new(ram_cap, rom_cap)), Box::new(MBC5::new(hw, ram_cap, rom_cap))
MBCKind::MBC5WithBattery => Box::new(MBC5::with_battery(ram_cap, rom_cap)), }
kind => todo!("ROMS with {:?} are currently unsupported", kind),
} }
} }
@ -92,15 +93,25 @@ impl Cartridge {
match id { match id {
0x00 => None, 0x00 => None,
0x01 | 0x02 => MBC1, 0x01 => MBC1(MBC1Hardware::None),
0x03 => MBC1WithBattery, 0x02 => MBC1(MBC1Hardware::RAM),
0x05 => MBC2, 0x03 => MBC1(MBC1Hardware::BatteryRAM),
0x06 => MBC2WithBattery, 0x05 => MBC2(MBC2Hardware::None),
0x19 | 0x1A => MBC5, 0x06 => MBC2(MBC2Hardware::BatteryRAM),
0x1B => MBC5WithBattery, 0x08 | 0x09 => unimplemented!("NoMBC + RAM and NoMBC + Battery unsupported"),
0x13 => MBC3WithBattery, 0x0B | 0x0C | 0x0D => unimplemented!("MM01 unsupported"),
0x11 | 0x12 => MBC3, 0x0F => MBC3(MBC3Hardware::RTC),
id => unimplemented!("id {:#04X} is an unsupported MBC", id), 0x10 => MBC3(MBC3Hardware::RTCAndBatteryRAM),
0x11 => MBC3(MBC3Hardware::None),
0x12 => MBC3(MBC3Hardware::RAM),
0x13 => MBC3(MBC3Hardware::BatteryRAM),
0x19 => MBC5(MBC5Hardware::None),
0x1A => MBC5(MBC5Hardware::RAM),
0x1B => MBC5(MBC5Hardware::BatteryRAM),
0x1C => MBC5(MBC5Hardware::Rumble),
0x1D => MBC5(MBC5Hardware::RumbleRAM),
0x1E => MBC5(MBC5Hardware::RumbleBatteryRAM),
id => unimplemented!("MBC with code {:#04X} is unsupported", id),
} }
} }
} }
@ -110,8 +121,8 @@ impl BusIo for Cartridge {
use MBCResult::*; use MBCResult::*;
match self.mbc.handle_read(addr) { match self.mbc.handle_read(addr) {
Address(addr) => self.mem[addr], Addr(addr) => self.mem[addr],
Value(byte) => byte, Byte(byte) => byte,
} }
} }
@ -132,11 +143,11 @@ struct MBC1 {
rom_size: RomSize, rom_size: RomSize,
mem_enabled: bool, mem_enabled: bool,
has_battery: bool, hw: MBC1Hardware,
} }
impl MBC1 { impl MBC1 {
fn new(ram_size: RamSize, rom_size: RomSize) -> Self { fn new(hw: MBC1Hardware, ram_size: RamSize, rom_size: RomSize) -> Self {
Self { Self {
rom_bank: 0x01, rom_bank: 0x01,
mem: vec![0; ram_size.capacity() as usize], mem: vec![0; ram_size.capacity() as usize],
@ -145,20 +156,7 @@ impl MBC1 {
ram_bank: Default::default(), ram_bank: Default::default(),
mode: Default::default(), mode: Default::default(),
mem_enabled: Default::default(), mem_enabled: Default::default(),
has_battery: Default::default(), hw,
}
}
fn with_battery(ram_size: RamSize, rom_size: RomSize) -> Self {
Self {
rom_bank: 0x01,
mem: vec![0; ram_size.capacity() as usize],
ram_size,
rom_size,
ram_bank: Default::default(),
mode: Default::default(),
mem_enabled: Default::default(),
has_battery: true,
} }
} }
@ -229,15 +227,16 @@ impl MBC1 {
impl Savable for MBC1 { impl Savable for MBC1 {
fn ext_ram(&self) -> Option<&[u8]> { fn ext_ram(&self) -> Option<&[u8]> {
match self.has_battery { match self.hw {
true => Some(&self.mem), MBC1Hardware::BatteryRAM => Some(&self.mem),
false => None, _ => None,
} }
} }
fn write_ext_ram(&mut self, memory: Vec<u8>) { fn ext_ram_mut(&mut self) -> Option<&mut [u8]> {
if self.has_battery { match self.hw {
self.mem.copy_from_slice(&memory); MBC1Hardware::BatteryRAM => Some(&mut self.mem),
_ => None,
} }
} }
} }
@ -248,16 +247,12 @@ impl MBCIo for MBC1 {
match addr { match addr {
0x0000..=0x3FFF if self.mode => { 0x0000..=0x3FFF if self.mode => {
Address(0x4000 * self.zero_bank() as usize + addr as usize) Addr(0x4000 * self.zero_bank() as usize + addr as usize)
} }
0x0000..=0x3FFF => Address(addr as usize), 0x0000..=0x3FFF => Addr(addr as usize),
0x4000..=0x7FFF => { 0x4000..=0x7FFF => Addr(0x4000 * self.high_bank() as usize + (addr as usize - 0x4000)),
Address(0x4000 * self.high_bank() as usize + (addr as usize - 0x4000)) 0xA000..=0xBFFF if self.mem_enabled => Byte(self.mem[self.ram_addr(addr)]),
} 0xA000..=0xBFFF => Byte(0xFF),
0xA000..=0xBFFF if self.mem_enabled && self.ram_size != RamSize::None => {
Value(self.mem[self.ram_addr(addr)])
}
0xA000..=0xBFFF => Value(0xFF),
_ => unreachable!("A read from {:#06X} should not be handled by MBC1", addr), _ => unreachable!("A read from {:#06X} should not be handled by MBC1", addr),
} }
} }
@ -272,7 +267,7 @@ impl MBCIo for MBC1 {
} }
0x4000..=0x5FFF => self.ram_bank = byte & 0x03, 0x4000..=0x5FFF => self.ram_bank = byte & 0x03,
0x6000..=0x7FFF => self.mode = (byte & 0x01) == 0x01, 0x6000..=0x7FFF => self.mode = (byte & 0x01) == 0x01,
0xA000..=0xBFFF if self.mem_enabled && self.ram_size != RamSize::None => { 0xA000..=0xBFFF if self.mem_enabled => {
let ram_addr = self.ram_addr(addr); let ram_addr = self.ram_addr(addr);
self.mem[ram_addr] = byte; self.mem[ram_addr] = byte;
} }
@ -407,34 +402,20 @@ struct MBC3 {
devs_enabled: bool, devs_enabled: bool,
mapped: Option<MBC3Device>, mapped: Option<MBC3Device>,
memory: Vec<u8>, mem: Vec<u8>,
// RTC Data Latch Previous Write // RTC Data Latch Previous Write
prev_latch_write: Option<u8>, prev_latch_write: Option<u8>,
has_battery: bool, hw: MBC3Hardware,
rtc: RtClock, rtc: RtClock,
rtc_latch: Option<RtClock>, rtc_latch: Option<RtClock>,
} }
impl MBC3 { impl MBC3 {
fn new(ram_cap: usize) -> Self { fn new(hw: MBC3Hardware, ram_cap: usize) -> Self {
Self { Self {
memory: vec![0; ram_cap], mem: vec![0; ram_cap],
rom_bank: Default::default(),
ram_bank: Default::default(),
devs_enabled: Default::default(),
mapped: Default::default(),
prev_latch_write: Default::default(),
has_battery: Default::default(),
rtc: Default::default(),
rtc_latch: Default::default(),
}
}
fn with_battery(ram_cap: usize) -> Self {
Self {
memory: vec![0; ram_cap],
rom_bank: Default::default(), rom_bank: Default::default(),
ram_bank: Default::default(), ram_bank: Default::default(),
devs_enabled: Default::default(), devs_enabled: Default::default(),
@ -442,22 +423,23 @@ impl MBC3 {
prev_latch_write: Default::default(), prev_latch_write: Default::default(),
rtc: Default::default(), rtc: Default::default(),
rtc_latch: Default::default(), rtc_latch: Default::default(),
has_battery: true, hw,
} }
} }
} }
impl Savable for MBC3 { impl Savable for MBC3 {
fn ext_ram(&self) -> Option<&[u8]> { fn ext_ram(&self) -> Option<&[u8]> {
match self.has_battery { match self.hw {
true => Some(&self.memory), MBC3Hardware::BatteryRAM | MBC3Hardware::RTCAndBatteryRAM => Some(&self.mem),
false => None, _ => None,
} }
} }
fn write_ext_ram(&mut self, memory: Vec<u8>) { fn ext_ram_mut(&mut self) -> Option<&mut [u8]> {
if self.has_battery { match self.hw {
self.memory.copy_from_slice(&memory); MBC3Hardware::BatteryRAM | MBC3Hardware::RTCAndBatteryRAM => Some(&mut self.mem),
_ => None,
} }
} }
} }
@ -468,13 +450,13 @@ impl MBCIo for MBC3 {
use RtcRegister::*; use RtcRegister::*;
let res = match addr { let res = match addr {
0x0000..=0x3FFF => Address(addr as usize), 0x0000..=0x3FFF => Addr(addr as usize),
0x4000..=0x7FFF => Address(0x4000 * self.rom_bank as usize + (addr as usize - 0x4000)), 0x4000..=0x7FFF => Addr(0x4000 * self.rom_bank as usize + (addr as usize - 0x4000)),
0xA000..=0xBFFF => match self.mapped { 0xA000..=0xBFFF => match self.mapped {
Some(MBC3Device::ExternalRam) if self.devs_enabled => { Some(MBC3Device::ExternalRam) if self.devs_enabled => {
Value(self.memory[0x2000 * self.ram_bank as usize + (addr as usize - 0xA000)]) Byte(self.mem[0x2000 * self.ram_bank as usize + (addr as usize - 0xA000)])
} }
Some(MBC3Device::Clock(reg)) if self.devs_enabled => Value( Some(MBC3Device::Clock(reg)) if self.devs_enabled => Byte(
self.rtc_latch self.rtc_latch
.as_ref() .as_ref()
.map(|rtc| match reg { .map(|rtc| match reg {
@ -486,7 +468,7 @@ impl MBCIo for MBC3 {
}) })
.unwrap_or(0xFF), .unwrap_or(0xFF),
), ),
_ => Value(0xFF), _ => Byte(0xFF),
}, },
_ => unreachable!("A read from {:#06X} should not be handled by MBC3", addr), _ => unreachable!("A read from {:#06X} should not be handled by MBC3", addr),
}; };
@ -528,7 +510,7 @@ impl MBCIo for MBC3 {
} }
0xA000..=0xBFFF => match self.mapped { 0xA000..=0xBFFF => match self.mapped {
Some(MBC3Device::ExternalRam) if self.devs_enabled => { Some(MBC3Device::ExternalRam) if self.devs_enabled => {
self.memory[0x2000 * self.ram_bank as usize + (addr as usize - 0xA000)] = byte self.mem[0x2000 * self.ram_bank as usize + (addr as usize - 0xA000)] = byte
} }
Some(MBC3Device::Clock(rtc_reg)) if self.devs_enabled => match rtc_reg { Some(MBC3Device::Clock(rtc_reg)) if self.devs_enabled => match rtc_reg {
Second => { Second => {
@ -550,8 +532,10 @@ impl MBCIo for MBC3 {
impl RtClockTick for MBC3 { impl RtClockTick for MBC3 {
fn tick(&mut self) { fn tick(&mut self) {
if let MBC3Hardware::RTCAndBatteryRAM | MBC3Hardware::RTC = self.hw {
self.rtc.tick(); self.rtc.tick();
} }
}
} }
#[derive(Debug)] #[derive(Debug)]
@ -562,32 +546,21 @@ struct MBC5 {
ram_bank: u8, ram_bank: u8,
rom_cap: usize, rom_cap: usize,
memory: Vec<u8>, mem: Vec<u8>,
mem_enabled: bool, mem_enabled: bool,
has_battery: bool, hw: MBC5Hardware,
} }
impl MBC5 { impl MBC5 {
fn new(ram_cap: usize, rom_cap: usize) -> Self { fn new(hw: MBC5Hardware, ram_cap: usize, rom_cap: usize) -> Self {
Self { Self {
rom_bank: 0x01, rom_bank: 0x01,
memory: vec![0; ram_cap], mem: vec![0; ram_cap],
rom_cap, rom_cap,
ram_bank: Default::default(), ram_bank: Default::default(),
mem_enabled: Default::default(), mem_enabled: Default::default(),
has_battery: Default::default(), hw,
}
}
fn with_battery(ram_cap: usize, rom_cap: usize) -> Self {
Self {
rom_bank: 0x01,
memory: vec![0; ram_cap],
rom_cap,
ram_bank: Default::default(),
mem_enabled: Default::default(),
has_battery: true,
} }
} }
@ -598,15 +571,16 @@ impl MBC5 {
impl Savable for MBC5 { impl Savable for MBC5 {
fn ext_ram(&self) -> Option<&[u8]> { fn ext_ram(&self) -> Option<&[u8]> {
match self.has_battery { match self.hw {
true => Some(&self.memory), MBC5Hardware::RumbleBatteryRAM | MBC5Hardware::BatteryRAM => Some(&self.mem),
false => None, _ => None,
} }
} }
fn write_ext_ram(&mut self, memory: Vec<u8>) { fn ext_ram_mut(&mut self) -> Option<&mut [u8]> {
if self.has_battery { match self.hw {
self.memory.copy_from_slice(&memory); MBC5Hardware::RumbleBatteryRAM | MBC5Hardware::BatteryRAM => Some(&mut self.mem),
_ => None,
} }
} }
} }
@ -616,12 +590,12 @@ impl MBCIo for MBC5 {
use MBCResult::*; use MBCResult::*;
match addr { match addr {
0x0000..=0x3FFF => Address(addr as usize), 0x0000..=0x3FFF => Addr(addr as usize),
0x4000..=0x7FFF => Address(self.bank_addr(addr)), 0x4000..=0x7FFF => Addr(self.bank_addr(addr)),
0xA000..=0xBFFF if self.mem_enabled => { 0xA000..=0xBFFF if self.mem_enabled => {
Value(self.memory[0x2000 * self.ram_bank as usize + (addr as usize - 0xA000)]) Byte(self.mem[0x2000 * self.ram_bank as usize + (addr as usize - 0xA000)])
} }
0xA000..=0xBFFF => Value(0xFF), 0xA000..=0xBFFF => Byte(0xFF),
_ => unreachable!("A read from {:#06X} should not be handled by MBC5", addr), _ => unreachable!("A read from {:#06X} should not be handled by MBC5", addr),
} }
} }
@ -633,7 +607,7 @@ impl MBCIo for MBC5 {
0x3000..=0x3FFF => self.rom_bank = (self.rom_bank & 0x00FF) | (byte as u16 & 0x01) << 8, 0x3000..=0x3FFF => self.rom_bank = (self.rom_bank & 0x00FF) | (byte as u16 & 0x01) << 8,
0x4000..=0x5FFF => self.ram_bank = byte & 0x0F, 0x4000..=0x5FFF => self.ram_bank = byte & 0x0F,
0xA000..=0xBFFF if self.mem_enabled => { 0xA000..=0xBFFF if self.mem_enabled => {
self.memory[0x2000 * self.ram_bank as usize + (addr as usize - 0xA000)] = byte; self.mem[0x2000 * self.ram_bank as usize + (addr as usize - 0xA000)] = byte;
} }
0xA000..=0xBFFF => {} 0xA000..=0xBFFF => {}
_ => unreachable!("A write to {:#06X} should not be handled by MBC5", addr), _ => unreachable!("A write to {:#06X} should not be handled by MBC5", addr),
@ -649,33 +623,23 @@ impl RtClockTick for MBC5 {
struct MBC2 { struct MBC2 {
/// 4-bit number /// 4-bit number
rom_bank: u8, rom_bank: u8,
memory: Box<[u8; Self::RAM_SIZE]>, mem: Box<[u8; Self::RAM_SIZE]>,
mem_enabled: bool,
mem_enabled: bool,
rom_cap: usize, rom_cap: usize,
has_battery: bool, hw: MBC2Hardware,
} }
impl MBC2 { impl MBC2 {
const RAM_SIZE: usize = 0x0200; const RAM_SIZE: usize = 0x0200;
fn new(rom_cap: usize) -> Self { fn new(hw: MBC2Hardware, rom_cap: usize) -> Self {
Self { Self {
rom_bank: 0x01, rom_bank: 0x01,
memory: Box::new([0; Self::RAM_SIZE]), mem: Box::new([0; Self::RAM_SIZE]),
rom_cap, rom_cap,
mem_enabled: Default::default(), mem_enabled: Default::default(),
has_battery: Default::default(), hw,
}
}
fn with_battery(rom_cap: usize) -> Self {
Self {
rom_bank: 0x01,
memory: Box::new([0; Self::RAM_SIZE]),
rom_cap,
mem_enabled: Default::default(),
has_battery: true,
} }
} }
@ -686,15 +650,16 @@ impl MBC2 {
impl Savable for MBC2 { impl Savable for MBC2 {
fn ext_ram(&self) -> Option<&[u8]> { fn ext_ram(&self) -> Option<&[u8]> {
match self.has_battery { match self.hw {
true => Some(self.memory.as_ref()), MBC2Hardware::BatteryRAM => Some(self.mem.as_ref()),
false => None, MBC2Hardware::None => None,
} }
} }
fn write_ext_ram(&mut self, memory: Vec<u8>) { fn ext_ram_mut(&mut self) -> Option<&mut [u8]> {
if self.has_battery { match self.hw {
self.memory.copy_from_slice(&memory); MBC2Hardware::BatteryRAM => Some(self.mem.as_mut()),
MBC2Hardware::None => None,
} }
} }
} }
@ -704,13 +669,13 @@ impl MBCIo for MBC2 {
use MBCResult::*; use MBCResult::*;
match addr { match addr {
0x0000..=0x3FFF => Address(addr as usize), 0x0000..=0x3FFF => Addr(addr as usize),
0x4000..=0x7FFF => Address(self.rom_addr(addr)), 0x4000..=0x7FFF => Addr(self.rom_addr(addr)),
0xA000..=0xBFFF if self.mem_enabled => { 0xA000..=0xBFFF if self.mem_enabled => {
let mbc2_addr = addr as usize & (Self::RAM_SIZE - 1); let mbc2_addr = addr as usize & (Self::RAM_SIZE - 1);
Value(self.memory[mbc2_addr] | 0xF0) Byte(self.mem[mbc2_addr] | 0xF0)
} }
0xA000..=0xBFFF => Value(0xFF), 0xA000..=0xBFFF => Byte(0xFF),
_ => unreachable!("A read from {:#06X} should not be handled by MBC2", addr), _ => unreachable!("A read from {:#06X} should not be handled by MBC2", addr),
} }
} }
@ -725,7 +690,7 @@ impl MBCIo for MBC2 {
0x0000..=0x3FFF => self.mem_enabled = nybble == 0x0A, 0x0000..=0x3FFF => self.mem_enabled = nybble == 0x0A,
0xA000..=0xBFFF if self.mem_enabled => { 0xA000..=0xBFFF if self.mem_enabled => {
let mbc2_addr = addr as usize & (Self::RAM_SIZE - 1); let mbc2_addr = addr as usize & (Self::RAM_SIZE - 1);
self.memory[mbc2_addr] = nybble; self.mem[mbc2_addr] = nybble;
} }
0x4000..=0x7FFF | 0xA000..=0xBFFF => {} 0x4000..=0x7FFF | 0xA000..=0xBFFF => {}
_ => unreachable!("A write to {:#06X} should not be handled by MBC2", addr), _ => unreachable!("A write to {:#06X} should not be handled by MBC2", addr),
@ -745,14 +710,14 @@ impl Savable for NoMBC {
None None
} }
fn write_ext_ram(&mut self, _memory: Vec<u8>) { fn ext_ram_mut(&mut self) -> Option<&mut [u8]> {
// Nothing Happens Here None
} }
} }
impl MBCIo for NoMBC { impl MBCIo for NoMBC {
fn handle_read(&self, addr: u16) -> MBCResult { fn handle_read(&self, addr: u16) -> MBCResult {
MBCResult::Address(addr as usize) MBCResult::Addr(addr as usize)
} }
fn handle_write(&mut self, _: u16, byte: u8) { fn handle_write(&mut self, _: u16, byte: u8) {
@ -771,21 +736,49 @@ trait MBCIo: Savable + RtClockTick {
#[derive(Debug, Clone, Copy)] #[derive(Debug, Clone, Copy)]
enum MBCResult { enum MBCResult {
Address(usize), Addr(usize),
Value(u8), Byte(u8),
} }
#[derive(Debug, Clone, Copy)] #[derive(Debug, Clone, Copy)]
enum MBCKind { enum MBCKind {
None, None,
MBC1, MBC1(MBC1Hardware),
MBC1WithBattery, MBC2(MBC2Hardware),
MBC2, MBC3(MBC3Hardware),
MBC2WithBattery, MBC5(MBC5Hardware),
MBC3, }
MBC3WithBattery,
MBC5, #[derive(Debug, Clone, Copy)]
MBC5WithBattery, enum MBC1Hardware {
None,
RAM,
BatteryRAM,
}
#[derive(Debug, Clone, Copy)]
enum MBC2Hardware {
None,
BatteryRAM,
}
#[derive(Debug, Clone, Copy)]
enum MBC3Hardware {
RTC,
RTCAndBatteryRAM,
None,
RAM,
BatteryRAM,
}
#[derive(Debug, Clone, Copy)]
enum MBC5Hardware {
None,
RAM,
BatteryRAM,
Rumble,
RumbleRAM,
RumbleBatteryRAM,
} }
#[derive(Debug, Clone, Copy, PartialEq)] #[derive(Debug, Clone, Copy, PartialEq)]
@ -813,12 +806,6 @@ impl RamSize {
} }
} }
impl Default for RamSize {
fn default() -> Self {
Self::None
}
}
impl From<u8> for RamSize { impl From<u8> for RamSize {
fn from(byte: u8) -> Self { fn from(byte: u8) -> Self {
use RamSize::*; use RamSize::*;
@ -893,15 +880,9 @@ impl std::fmt::Debug for Box<dyn MBCIo> {
} }
} }
impl Default for Box<dyn MBCIo> {
fn default() -> Self {
Box::new(NoMBC)
}
}
trait Savable { trait Savable {
fn ext_ram(&self) -> Option<&[u8]>; fn ext_ram(&self) -> Option<&[u8]>;
fn write_ext_ram(&mut self, memory: Vec<u8>); fn ext_ram_mut(&mut self) -> Option<&mut [u8]>;
} }
#[cfg(test)] #[cfg(test)]

View File

@ -70,6 +70,7 @@ pub fn write_save(cpu: &Cpu) {
Some(cart) => match write_save_to_file(cart) { Some(cart) => match write_save_to_file(cart) {
Ok(path) => tracing::info!("Wrote to save at {:?}", path), Ok(path) => tracing::info!("Wrote to save at {:?}", path),
Err(err @ SaveError::NotApplicable) => tracing::warn!("Unable to Save: {:?}", err), Err(err @ SaveError::NotApplicable) => tracing::warn!("Unable to Save: {:?}", err),
Err(SaveError::DiffSize) => unreachable!(),
Err(SaveError::Io(err)) => tracing::error!("{:?}", err), Err(SaveError::Io(err)) => tracing::error!("{:?}", err),
}, },
None => tracing::error!("No cartridge is currently present"), None => tracing::error!("No cartridge is currently present"),
@ -80,7 +81,8 @@ pub fn load_save(cpu: &mut Cpu) {
match cpu.bus.cart.as_mut() { match cpu.bus.cart.as_mut() {
Some(cart) => match read_save_from_file(cart) { Some(cart) => match read_save_from_file(cart) {
Ok(path) => tracing::info!("Loaded save from {:?}", path), Ok(path) => tracing::info!("Loaded save from {:?}", path),
Err(err @ SaveError::NotApplicable) => tracing::warn!("Unable to load save: {:?}", err), Err(err @ SaveError::NotApplicable) => tracing::warn!("Unable to load save: {}", err),
Err(err @ SaveError::DiffSize) => tracing::error!("Unable to load save: {}", err),
Err(SaveError::Io(err)) => match err.kind() { Err(SaveError::Io(err)) => match err.kind() {
std::io::ErrorKind::NotFound => tracing::warn!("Save not found"), std::io::ErrorKind::NotFound => tracing::warn!("Save not found"),
_ => tracing::error!("{:?}", err), _ => tracing::error!("{:?}", err),
@ -106,8 +108,8 @@ fn write_save_to_file(cart: &Cartridge) -> Result<PathBuf, SaveError> {
} }
fn read_save_from_file(cart: &mut Cartridge) -> Result<PathBuf, SaveError> { fn read_save_from_file(cart: &mut Cartridge) -> Result<PathBuf, SaveError> {
match cart.title.as_deref() { match cart.title.clone().zip(cart.ext_ram_mut()) {
Some(title) => { Some((title, ext_ram)) => {
let mut save_path = data_path().unwrap_or_else(|| PathBuf::from(".")); let mut save_path = data_path().unwrap_or_else(|| PathBuf::from("."));
save_path.push(title); save_path.push(title);
save_path.set_extension("sav"); save_path.set_extension("sav");
@ -116,10 +118,11 @@ fn read_save_from_file(cart: &mut Cartridge) -> Result<PathBuf, SaveError> {
let mut memory = Vec::new(); let mut memory = Vec::new();
file.read_to_end(&mut memory)?; file.read_to_end(&mut memory)?;
// FIXME: We call this whether we can write to Ext RAM or not. if ext_ram.len() != memory.len() {
// We should add a check that ensures that by this point we know whether return Err(SaveError::DiffSize);
// the cartridge has external RAM or not. }
cart.write_ext_ram(memory);
ext_ram.copy_from_slice(&memory);
Ok(save_path) Ok(save_path)
} }
None => Err(SaveError::NotApplicable), None => Err(SaveError::NotApplicable),
@ -151,4 +154,6 @@ pub enum SaveError {
NotApplicable, NotApplicable,
#[error(transparent)] #[error(transparent)]
Io(#[from] std::io::Error), Io(#[from] std::io::Error),
#[error("save file size differs from external ram")]
DiffSize,
} }