From c20ac9cdf8eaa414ab3114c8caea49746f7dc6e5 Mon Sep 17 00:00:00 2001
From: Michel Heily <michelheily@gmail.com>
Date: Fri, 17 Jan 2020 16:08:23 +0200
Subject: [PATCH] core: cartridge: Try to detect and print the backup type

In preparation for backup impl


Former-commit-id: 75efbbaea82ba30899787b46b8eaca834622493c
---
 Cargo.toml            |  1 +
 src/core/cartridge.rs | 39 +++++++++++++++++++++++++++++----------
 src/lib.rs            |  2 ++
 3 files changed, 32 insertions(+), 10 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 8da6e0c..86df067 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -29,6 +29,7 @@ bit-set = "0.5.1"
 ringbuf = "0.2.1"
 debug_stub_derive = "0.3.0"
 bytesize = "1.0.0"
+memmem = "0.1.1"
 
 rustyline = {version = "5.0.0", optional = true}
 nom = {version = "5.0.0", optional = true}
diff --git a/src/core/cartridge.rs b/src/core/cartridge.rs
index 3ddb333..18da6cc 100644
--- a/src/core/cartridge.rs
+++ b/src/core/cartridge.rs
@@ -1,12 +1,11 @@
-use std::ffi::OsStr;
 use std::fs::File;
 use std::io::prelude::*;
 use std::path::Path;
 use std::str::from_utf8;
 
 use serde::{Deserialize, Serialize};
-
-use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};
+use memmem::{Searcher, TwoWaySearcher};
+use num::FromPrimitive;
 use zip::ZipArchive;
 
 use super::super::util::read_bin_file;
@@ -71,6 +70,15 @@ impl CartridgeHeader {
     }
 }
 
+#[derive(Debug, Primitive, Serialize, Deserialize, Clone)]
+pub enum BackupType {
+    Eeprom = 0,
+    Sram = 1,
+    Flash = 2,
+    Flash512 = 3,
+    Flash1M = 4,
+}
+
 #[derive(Serialize, Deserialize, Clone, Debug)]
 pub struct Cartridge {
     pub header: CartridgeHeader,
@@ -109,16 +117,13 @@ fn load_rom(path: &Path) -> GBAResult<Vec<u8>> {
 impl Cartridge {
     pub fn from_path(rom_path: &Path) -> GBAResult<Cartridge> {
         let rom_bin = load_rom(rom_path)?;
-        let size = rom_bin.len();
-        let header = CartridgeHeader::parse(&rom_bin);
-        Ok(Cartridge {
-            header: header,
-            bytes: rom_bin.into_boxed_slice(),
-            size: size,
-        })
+        Ok(Cartridge::from_bytes(&rom_bin))
     }
 
     pub fn from_bytes(bytes: &[u8]) -> Cartridge {
+        let backup = Cartridge::detect_backup_type(&bytes);
+        println!("Backup detected: {:?}", backup);
+
         let size = bytes.len();
         let header = CartridgeHeader::parse(&bytes);
 
@@ -128,6 +133,20 @@ impl Cartridge {
             size: size,
         }
     }
+
+    fn detect_backup_type(bin: &[u8]) -> Option<BackupType> {
+        const ID_STRINGS: &'static [&'static str] =
+            &["EEPROM_V", "SRAM_V", "FLASH_V", "FLASH512_V", "FLASH1M_V"];
+
+        for i in 0..5 {
+            let search = TwoWaySearcher::new(ID_STRINGS[i].as_bytes());
+            match search.search_in(bin) {
+                Some(_) => return Some(BackupType::from_u8(i as u8).unwrap()),
+                _ => {}
+            }
+        }
+        return None;
+    }
 }
 
 impl Bus for Cartridge {
diff --git a/src/lib.rs b/src/lib.rs
index 0a8d82d..11dc30d 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -34,6 +34,8 @@ extern crate colored; // not needed in Rust 2018
 
 extern crate zip;
 
+extern crate memmem;
+
 #[macro_use]
 pub mod util;
 pub mod core;