Set up octal SPI functions

Author: bugadani

ld/esp32s3.x

@@ -155,4 +155,10 @@ PROVIDE( Enable_QMode = 0x40000d2c );
 
 
 /* Functions */
-ets_efuse_get_spiconfig = 0x40001f74;
+ets_efuse_get_spiconfig = 0x40001f74;
+ets_efuse_flash_octal_mode = 0x40002004;
+
+/* Data (.data, .bss, .rodata) */
+PROVIDE( rom_spiflash_legacy_funcs = 0x3fceffe8 );
+PROVIDE( rom_spiflash_legacy_data = 0x3fceffe4 );
+PROVIDE( g_flash_guard_ops = 0x3fceffec );

src/flash.rs

@@ -70,6 +70,17 @@ pub fn attach() -> i32 {
 
     if config_result == 0 {
         unsafe { esp_rom_spiflash_attach(spiconfig, false) };
+
+        #[cfg(feature = "esp32s3")]
+        {
+            extern "C" {
+                fn ets_efuse_flash_octal_mode() -> bool;
+            }
+            if unsafe { ets_efuse_flash_octal_mode() } {
+                init_ospi_funcs();
+            }
+        }
+
         0
     } else {
         -1
@@ -115,3 +126,73 @@ pub fn wait_for_idle() -> i32 {
 
     0
 }
+
+#[cfg(feature = "esp32s3")]
+#[allow(non_camel_case_types)]
+fn init_ospi_funcs() {
+    #[repr(C)]
+    struct spiflash_legacy_funcs_t {
+        pp_addr_bit_len: u8,
+        se_addr_bit_len: u8,
+        be_addr_bit_len: u8,
+        rd_addr_bit_len: u8,
+        read_sub_len: u32,
+        write_sub_len: u32,
+        unlock: Option<spi_flash_op_t>,
+        erase_sector: Option<spi_flash_erase_t>,
+        erase_block: Option<spi_flash_erase_t>,
+        read: Option<spi_flash_rd_t>,
+        write: Option<spi_flash_wr_t>,
+        encrypt_write: Option<spi_flash_ewr_t>,
+        check_sus: Option<spi_flash_func_t>,
+        wren: Option<spi_flash_wren_t>,
+        wait_idle: Option<spi_flash_op_t>,
+        erase_area: Option<spi_flash_erase_area_t>,
+    }
+
+    type spi_flash_func_t = unsafe extern "C" fn();
+    type spi_flash_op_t = unsafe extern "C" fn() -> i32;
+    type spi_flash_erase_t = unsafe extern "C" fn(u32) -> i32;
+    type spi_flash_rd_t = unsafe extern "C" fn(u32, *mut (), i32) -> i32;
+    type spi_flash_wr_t = unsafe extern "C" fn(u32, *const u32, i32) -> i32;
+    type spi_flash_ewr_t = unsafe extern "C" fn(u32, *const (), u32) -> i32;
+    type spi_flash_wren_t = unsafe extern "C" fn(*mut ()) -> i32;
+    type spi_flash_erase_area_t = unsafe extern "C" fn(u32, u32) -> i32;
+
+    extern "C" {
+        static mut rom_spiflash_legacy_funcs: *const spiflash_legacy_funcs_t;
+
+        fn esp_rom_opiflash_wait_idle() -> i32;
+        fn esp_rom_opiflash_erase_block_64k(addr: u32) -> i32;
+        fn esp_rom_opiflash_erase_sector(addr: u32) -> i32;
+        fn esp_rom_opiflash_read(addr: u32, buf: *mut (), len: i32) -> i32;
+        fn esp_rom_opiflash_write(addr: u32, data: *const u32, len: i32) -> i32;
+        fn esp_rom_opiflash_wren(p: *mut ()) -> i32;
+        fn esp_rom_opiflash_erase_area(start_addr: u32, end_addr: u32) -> i32;
+    }
+
+    static FUNCS: spiflash_legacy_funcs_t = spiflash_legacy_funcs_t {
+        pp_addr_bit_len: 24,
+        se_addr_bit_len: 24,
+        be_addr_bit_len: 24,
+        rd_addr_bit_len: 24,
+        read_sub_len: 16,
+        write_sub_len: 32,
+        unlock: Some(esp_rom_opiflash_wait_idle),
+        erase_block: Some(esp_rom_opiflash_erase_block_64k),
+        erase_sector: Some(esp_rom_opiflash_erase_sector),
+        read: Some(esp_rom_opiflash_read),
+        write: Some(esp_rom_opiflash_write),
+        encrypt_write: None,
+        check_sus: None,
+        wait_idle: Some(esp_rom_opiflash_wait_idle),
+        wren: Some(esp_rom_opiflash_wren),
+        erase_area: Some(esp_rom_opiflash_erase_area),
+    };
+
+    unsafe {
+        let funcs_iram = &raw const FUNCS;
+        rom_spiflash_legacy_funcs =
+            ((funcs_iram as usize) - 0x4038_0400 + 0x3FC9_0400) as *const spiflash_legacy_funcs_t;
+    }
+}