From 28dbc45637364b6ac68d2d091b6c5fa2420f3546 Mon Sep 17 00:00:00 2001
From: Jeremy Soller <jackpot51@gmail.com>
Date: Sat, 18 Feb 2017 14:05:04 -0700
Subject: [PATCH] WIP: NVME driver

---
 nvmed/Cargo.toml  |  8 +++++
 nvmed/src/main.rs | 83 +++++++++++++++++++++++++++++++++++++++++++++++
 nvmed/src/nvme.rs | 31 ++++++++++++++++++
 3 files changed, 122 insertions(+)
 create mode 100644 nvmed/Cargo.toml
 create mode 100644 nvmed/src/main.rs
 create mode 100644 nvmed/src/nvme.rs

diff --git a/nvmed/Cargo.toml b/nvmed/Cargo.toml
new file mode 100644
index 0000000000..fbdc36977d
--- /dev/null
+++ b/nvmed/Cargo.toml
@@ -0,0 +1,8 @@
+[package]
+name = "nvmed"
+version = "0.1.0"
+
+[dependencies]
+bitflags = "0.7"
+spin = "0.4"
+redox_syscall = "0.1"
diff --git a/nvmed/src/main.rs b/nvmed/src/main.rs
new file mode 100644
index 0000000000..ebf646b496
--- /dev/null
+++ b/nvmed/src/main.rs
@@ -0,0 +1,83 @@
+//#![deny(warnings)]
+#![feature(asm)]
+
+extern crate bitflags;
+extern crate spin;
+extern crate syscall;
+
+use std::{env, usize};
+use std::fs::File;
+use std::io::{Read, Write};
+use std::os::unix::io::{AsRawFd, FromRawFd, RawFd};
+use syscall::{EVENT_READ, MAP_WRITE, Event, Packet, Result, Scheme};
+
+mod nvme;
+
+fn create_scheme_fallback<'a>(name: &'a str, fallback: &'a str) -> Result<(&'a str, RawFd)> {
+    if let Ok(fd) = syscall::open(&format!(":{}", name), syscall::O_RDWR | syscall::O_CREAT | syscall::O_NONBLOCK) {
+        Ok((name, fd))
+    } else {
+        syscall::open(&format!(":{}", fallback), syscall::O_RDWR | syscall::O_CREAT | syscall::O_NONBLOCK)
+                .map(|fd| (fallback, fd))
+    }
+}
+
+fn main() {
+    let mut args = env::args().skip(1);
+
+    let mut name = args.next().expect("nvmed: no name provided");
+    name.push_str("_nvme");
+
+    let bar_str = args.next().expect("nvmed: no address provided");
+    let bar = usize::from_str_radix(&bar_str, 16).expect("nvmed: failed to parse address");
+
+    let irq_str = args.next().expect("nvmed: no irq provided");
+    let irq = irq_str.parse::<u8>().expect("nvmed: failed to parse irq");
+
+    print!("{}", format!(" + NVME {} on: {:X} IRQ: {}\n", name, bar, irq));
+
+    // Daemonize
+    if unsafe { syscall::clone(0).unwrap() } == 0 {
+        let address = unsafe { syscall::physmap(bar, 4096, MAP_WRITE).expect("nvmed: failed to map address") };
+        {
+            /*
+            let (_scheme_name, socket_fd) = create_scheme_fallback("disk", &name).expect("nvmed: failed to create disk scheme");
+            let mut socket = unsafe { File::from_raw_fd(socket_fd) };
+            syscall::fevent(socket_fd, EVENT_READ).expect("nvmed: failed to fevent disk scheme");
+
+            let mut irq_file = File::open(&format!("irq:{}", irq)).expect("nvmed: failed to open irq file");
+            let irq_fd = irq_file.as_raw_fd();
+            syscall::fevent(irq_fd, EVENT_READ).expect("nvmed: failed to fevent irq file");
+
+            let mut event_file = File::open("event:").expect("nvmed: failed to open event file");
+
+            let scheme = DiskScheme::new(nvme::disks(address, &name));
+            loop {
+                let mut event = Event::default();
+                if event_file.read(&mut event).expect("nvmed: failed to read event file") == 0 {
+                    break;
+                }
+                if event.id == socket_fd {
+                    loop {
+                        let mut packet = Packet::default();
+                        if socket.read(&mut packet).expect("nvmed: failed to read disk scheme") == 0 {
+                            break;
+                        }
+                        scheme.handle(&mut packet);
+                        socket.write(&mut packet).expect("nvmed: failed to write disk scheme");
+                    }
+                } else if event.id == irq_fd {
+                    let mut irq = [0; 8];
+                    if irq_file.read(&mut irq).expect("nvmed: failed to read irq file") >= irq.len() {
+                        //TODO : Test for IRQ
+                        //irq_file.write(&irq).expect("nvmed: failed to write irq file");
+                    }
+                } else {
+                    println!("Unknown event {}", event.id);
+                }
+            }
+            */
+        }
+        unsafe { let _ = syscall::physunmap(address); }
+    }
+}
diff --git a/nvmed/src/nvme.rs b/nvmed/src/nvme.rs
new file mode 100644
index 0000000000..c6e022ac37
--- /dev/null
+++ b/nvmed/src/nvme.rs
@@ -0,0 +1,31 @@
+use syscall::io::{Io, Mmio};
+
+#[repr(packed)]
+pub struct NvmeRegs {
+    /// Controller Capabilities
+    cap: Mmio<u64>,
+    /// Version
+    vs: Mmio<u32>,
+    /// Interrupt mask set
+    intms: Mmio<u32>,
+    /// Interrupt mask clear
+    intmc: Mmio<u32>,
+    /// Controller configuration
+    cc: Mmio<u32>,
+    /// Reserved
+    _rsvd: Mmio<u32>,
+    /// Controller status
+    csts: Mmio<u32>,
+    /// NVM subsystem reset
+    nssr: Mmio<u32>,
+    /// Admin queue attributes
+    aqa: Mmio<u32>,
+    /// Admin submission queue base address
+    asq: Mmio<u64>,
+    /// Admin completion queue base address
+    acq: Mmio<u64>,
+    /// Controller memory buffer location
+    cmbloc: Mmio<u32>,
+    /// Controller memory buffer size
+    cmbsz: Mmio<u32>,    
+}
\ No newline at end of file