Graphics (#13)

Virtual Terminals

ahcid/Cargo.toml

@@ -4,6 +4,7 @@ version = "0.1.0"
 
 [dependencies]
 bitflags = "*"
+dma = { path = "../dma/" }
 io = { path = "../io/" }
 spin = "*"
 syscall = { path = "../../syscall/" }

ahcid/src/ahci/disk.rs

 use std::ptr;
 
+use dma::Dma;
 use syscall::error::Result;
 
-use super::dma::Dma;
 use super::hba::{HbaPort, HbaCmdTable, HbaCmdHeader};
 
 pub struct Disk {

ahcid/src/ahci/hba.rs

-use io::{Io, Mmio};
-
 use std::mem::size_of;
 use std::ops::DerefMut;
 use std::{ptr, u32};
 
+use dma::Dma;
+use io::{Io, Mmio};
 use syscall::error::{Error, Result, EIO};
 
-use super::dma::Dma;
 use super::fis::{FisType, FisRegH2D};
 
 const ATA_CMD_READ_DMA_EXT: u8 = 0x25;

ahcid/src/ahci/mod.rs

@@ -4,7 +4,6 @@ use self::disk::Disk;
 use self::hba::{HbaMem, HbaPortType};
 
 pub mod disk;
-pub mod dma;
 pub mod fis;
 pub mod hba;
 

ahcid/src/main.rs

@@ -3,6 +3,7 @@
 
 #[macro_use]
 extern crate bitflags;
+extern crate dma;
 extern crate io;
 extern crate spin;
 extern crate syscall;

dma/Cargo.toml

+[package]
+name = "dma"
+version = "0.1.0"
+
+[dependencies]
+syscall = { path = "../../syscall/" }

ahcid/src/ahci/dma.rs → dma/src/lib.rs

+#![feature(question_mark)]
+
+extern crate syscall;
+
 use std::{mem, ptr};
 use std::ops::{Deref, DerefMut};
 
-use syscall::{self, Result};
+use syscall::Result;
 
 struct PhysBox {
     address: usize,

e1000d/Cargo.toml

+[package]
+name = "rtl8139d"
+version = "0.1.0"
+
+[dependencies]
+bitflags = "*"
+dma = { path = "../dma/" }
+io = { path = "../io/" }
+spin = "*"
+syscall = { path = "../../syscall/" }

e1000d/src/main.rs

+#![feature(asm)]
+
+extern crate dma;
+extern crate syscall;
+
+use std::{env, thread};
+use std::fs::File;
+use std::io::{Read, Write};
+
+use syscall::{iopl, physmap, physunmap, Packet, MAP_WRITE};
+
+fn main() {
+    let mut args = env::args().skip(1);
+
+    let bar_str = args.next().expect("e1000d: no address provided");
+    let bar = usize::from_str_radix(&bar_str, 16).expect("e1000d: failed to parse address");
+
+    let irq_str = args.next().expect("e1000d: no irq provided");
+    let irq = irq_str.parse::<u8>().expect("e1000d: failed to parse irq");
+
+    thread::spawn(move || {
+        unsafe {
+            iopl(3).expect("e1000d: failed to get I/O permission");
+            asm!("cli" :::: "intel", "volatile");
+        }
+
+        let address = unsafe { physmap(bar, 4096, MAP_WRITE).expect("e1000d: failed to map address") };
+        {
+            println!("e1000d {:X}", bar);
+            let mut socket = File::create(":network").expect("e1000d: failed to create network scheme");
+            //let scheme = DiskScheme::new(ahci::disks(address, irq));
+            loop {
+                let mut packet = Packet::default();
+                socket.read(&mut packet).expect("e1000d: failed to read network scheme");
+                //scheme.handle(&mut packet);
+                socket.write(&mut packet).expect("e1000d: failed to read network scheme");
+            }
+        }
+        unsafe { let _ = physunmap(address); }
+    });
+}

ps2d/src/keyboard.rs

@@ -36,6 +36,15 @@ pub fn keyboard() {
                 rshift = pressed;
             } else if pressed {
                 match scancode {
+                    f @ 0x3B ... 0x44 => { // F1 through F10
+                        input.write(&[(f - 0x3B) + 0xF4]).unwrap();
+                    },
+                    0x57 => { // F11
+                        input.write(&[0xFE]).unwrap();
+                    },
+                    0x58 => { // F12
+                        input.write(&[0xFF]).unwrap();
+                    },
                     0x47 => { // Home
                         input.write(b"\x1B[H").unwrap();
                     },
@@ -69,7 +78,7 @@ pub fn keyboard() {
                     _ => {
                         let c = if ctrl {
                             match keymap::get_char(scancode, false) {
-                                c @ 'a' ... 'z' => (c as u8 - b'a' + b'\x01') as char,
+                                c @ 'a' ... 'z' => ((c as u8 - b'a') + b'\x01') as char,
                                 c => c
                             }
                         } else {

vesad/Cargo.toml

@@ -4,5 +4,8 @@ version = "0.1.0"
 
 [dependencies]
 ransid = { git = "https://github.com/redox-os/ransid.git", branch = "new_api" }
-rusttype = { git = "https://github.com/dylanede/rusttype.git" }
+rusttype = { git = "https://github.com/dylanede/rusttype.git", optional = true }
 syscall = { path = "../../syscall/" }
+
+[features]
+default = []

vesad/src/console.rs

+pub struct Text {
+
+}

vesad/src/display.rs

 #[cfg(feature="rusttype")]
 extern crate rusttype;
 
-use std::cmp;
+use alloc::heap;
+use std::{cmp, slice};
 
 use primitive::{fast_set32, fast_set64, fast_copy, fast_copy64};
 
@@ -11,6 +12,15 @@ use self::rusttype::{Font, FontCollection, Scale, point};
 #[cfg(not(feature="rusttype"))]
 static FONT: &'static [u8] = include_bytes!("../../../res/fonts/unifont.font");
 
+#[cfg(feature="rusttype")]
+static FONT: &'static [u8] = include_bytes!("../../../res/fonts/DejaVuSansMono.ttf");
+#[cfg(feature="rusttype")]
+static FONT_BOLD: &'static [u8] = include_bytes!("../../../res/fonts/DejaVuSansMono-Bold.ttf");
+#[cfg(feature="rusttype")]
+static FONT_BOLD_ITALIC: &'static [u8] = include_bytes!("../../../res/fonts/DejaVuSansMono-BoldOblique.ttf");
+#[cfg(feature="rusttype")]
+static FONT_ITALIC: &'static [u8] = include_bytes!("../../../res/fonts/DejaVuSansMono-Oblique.ttf");
+
 /// A display
 #[cfg(not(feature="rusttype"))]
 pub struct Display {
@@ -39,26 +49,32 @@ pub struct Display {
 
 impl Display {
     #[cfg(not(feature="rusttype"))]
-    pub fn new(width: usize, height: usize, onscreen: &'static mut [u32], offscreen: &'static mut [u32]) -> Display {
+    pub fn new(width: usize, height: usize, onscreen: usize) -> Display {
+        let size = width * height;
+        let offscreen = unsafe { heap::allocate(size * 4, 4096) };
+        unsafe { fast_set64(offscreen as *mut u64, 0, size/2) };
         Display {
             width: width,
             height: height,
-            onscreen: onscreen,
-            offscreen: offscreen
+            onscreen: unsafe { slice::from_raw_parts_mut(onscreen as *mut u32, size) },
+            offscreen: unsafe { slice::from_raw_parts_mut(offscreen as *mut u32, size) }
         }
     }
 
     #[cfg(feature="rusttype")]
-    pub fn new(width: usize, height: usize, onscreen: &'static mut [u32], offscreen: &'static mut [u32]) -> Display {
+    pub fn new(width: usize, height: usize, onscreen: usize) -> Display {
+        let size = width * height;
+        let offscreen = unsafe { heap::allocate(size * 4, 4096) };
+        unsafe { fast_set64(offscreen as *mut u64, 0, size/2) };
         Display {
             width: width,
             height: height,
-            onscreen: onscreen,
-            offscreen: offscreen,
-            font: FontCollection::from_bytes(include_bytes!("../../../res/fonts/DejaVuSansMono.ttf")).into_font().unwrap(),
-            font_bold: FontCollection::from_bytes(include_bytes!("../../../res/fonts/DejaVuSansMono-Bold.ttf")).into_font().unwrap(),
-            font_bold_italic: FontCollection::from_bytes(include_bytes!("../../../res/fonts/DejaVuSansMono-BoldOblique.ttf")).into_font().unwrap(),
-            font_italic: FontCollection::from_bytes(include_bytes!("../../../res/fonts/DejaVuSansMono-Oblique.ttf")).into_font().unwrap()
+            onscreen: unsafe { slice::from_raw_parts_mut(onscreen as *mut u32, size) },
+            offscreen: unsafe { slice::from_raw_parts_mut(offscreen as *mut u32, size) },
+            font: FontCollection::from_bytes(FONT).into_font().unwrap(),
+            font_bold: FontCollection::from_bytes(FONT_BOLD).into_font().unwrap(),
+            font_bold_italic: FontCollection::from_bytes(FONT_BOLD_ITALIC).into_font().unwrap(),
+            font_italic: FontCollection::from_bytes(FONT_ITALIC).into_font().unwrap()
         }
     }
 

vesad/src/scheme.rs

+use std::cell::{Cell, RefCell};
+use std::collections::BTreeMap;
+use std::str;
+
+use syscall::{Result, Error, EBADF, ENOENT, Scheme};
+
+use display::Display;
+use screen::TextScreen;
+
+pub struct DisplayScheme {
+    width: usize,
+    height: usize,
+    onscreen: usize,
+    active: Cell<usize>,
+    screens: RefCell<BTreeMap<usize, TextScreen>>
+}
+
+impl DisplayScheme {
+    pub fn new(width: usize, height: usize, onscreen: usize) -> DisplayScheme {
+        let mut screens = BTreeMap::new();
+        for i in 1..7 {
+            screens.insert(i, TextScreen::new(Display::new(width, height, onscreen)));
+        }
+
+        DisplayScheme {
+            width: width,
+            height: height,
+            onscreen: onscreen,
+            active: Cell::new(1),
+            screens: RefCell::new(screens)
+        }
+    }
+
+    pub fn will_block(&self, id: usize) -> bool {
+        let screens = self.screens.borrow();
+        if let Some(screen) = screens.get(&id) {
+            screen.will_block()
+        } else {
+            false
+        }
+    }
+}
+
+impl Scheme for DisplayScheme {
+    fn open(&self, path: &[u8], _flags: usize, _uid: u32, _gid: u32) -> Result<usize> {
+        if path == b"input" {
+            Ok(0)
+        } else {
+            let path_str = str::from_utf8(path).unwrap_or("");
+            let id = path_str.parse::<usize>().unwrap_or(0);
+            if self.screens.borrow().contains_key(&id) {
+                Ok(id)
+            } else {
+                Err(Error::new(ENOENT))
+            }
+        }
+    }
+
+    fn dup(&self, id: usize) -> Result<usize> {
+        Ok(id)
+    }
+
+    fn fevent(&self, id: usize, flags: usize) -> Result<usize> {
+        let mut screens = self.screens.borrow_mut();
+        if let Some(mut screen) = screens.get_mut(&id) {
+            println!("fevent {:X}", flags);
+            screen.requested = flags;
+            Ok(0)
+        } else {
+            Err(Error::new(EBADF))
+        }
+    }
+
+    fn fpath(&self, id: usize, buf: &mut [u8]) -> Result<usize> {
+        let screens = self.screens.borrow();
+        let path_str = if id == 0 {
+            format!("display:input")
+        } else if let Some(screen) = screens.get(&id) {
+            format!("display:{}/{}/{}", id, screen.console.w, screen.console.h)
+        } else {
+            return Err(Error::new(EBADF));
+        };
+
+        let path = path_str.as_bytes();
+
+        let mut i = 0;
+        while i < buf.len() && i < path.len() {
+            buf[i] = path[i];
+            i += 1;
+        }
+
+        Ok(i)
+    }
+
+    fn fsync(&self, id: usize) -> Result<usize> {
+        let mut screens = self.screens.borrow_mut();
+        if let Some(mut screen) = screens.get_mut(&id) {
+            if id == self.active.get() {
+                screen.sync();
+            }
+            Ok(0)
+        } else {
+            Err(Error::new(EBADF))
+        }
+    }
+
+    fn read(&self, id: usize, buf: &mut [u8]) -> Result<usize> {
+        let mut screens = self.screens.borrow_mut();
+        if let Some(mut screen) = screens.get_mut(&id) {
+            screen.read(buf)
+        } else {
+            Err(Error::new(EBADF))
+        }
+    }
+
+    fn write(&self, id: usize, buf: &[u8]) -> Result<usize> {
+        let mut screens = self.screens.borrow_mut();
+        if id == 0 {
+            if buf.len() == 1 && buf[0] >= 0xF4 {
+                let new_active = (buf[0] - 0xF4) as usize + 1;
+                if let Some(mut screen) = screens.get_mut(&new_active) {
+                    self.active.set(new_active);
+                    screen.redraw();
+                }
+                Ok(1)
+            } else {
+                if let Some(mut screen) = screens.get_mut(&self.active.get()) {
+                    screen.input(buf)
+                } else {
+                    Err(Error::new(EBADF))
+                }
+            }
+        } else if let Some(mut screen) = screens.get_mut(&id) {
+            screen.write(buf, id == self.active.get())
+        } else {
+            Err(Error::new(EBADF))
+        }
+    }
+
+    fn close(&self, _id: usize) -> Result<usize> {
+        Ok(0)
+    }
+}

vesad/src/screen/graphic.rs

+pub struct GraphicScreen;

vesad/src/screen/mod.rs

+pub use self::graphic::GraphicScreen;
+pub use self::text::TextScreen;
+
+mod graphic;
+mod text;
+
+pub enum Screen {
+    Graphic(GraphicScreen),
+    Text(TextScreen)
+}

vesad/src/screen/text.rs

+extern crate ransid;
+
+use std::collections::{BTreeSet, VecDeque};
+
+use self::ransid::{Console, Event};
+use syscall::Result;
+
+use display::Display;
+
+pub struct TextScreen {
+    pub console: Console,
+    pub display: Display,
+    pub changed: BTreeSet<usize>,
+    pub input: VecDeque<u8>,
+    pub end_of_input: bool,
+    pub cooked: VecDeque<u8>,
+    pub requested: usize
+}
+
+impl TextScreen {
+    pub fn new(display: Display) -> TextScreen {
+        TextScreen {
+            console: Console::new(display.width/8, display.height/16),
+            display: display,
+            changed: BTreeSet::new(),
+            input: VecDeque::new(),
+            end_of_input: false,
+            cooked: VecDeque::new(),
+            requested: 0
+        }
+    }
+
+    pub fn input(&mut self, buf: &[u8]) -> Result<usize> {
+        if self.console.raw_mode {
+            for &b in buf.iter() {
+                self.input.push_back(b);
+            }
+        } else {
+            for &b in buf.iter() {
+                match b {
+                    b'\x03' => {
+                        self.end_of_input = true;
+                        self.write(b"^C\n", true)?;
+                    },
+                    b'\x08' | b'\x7F' => {
+                        if let Some(_c) = self.cooked.pop_back() {
+                            self.write(b"\x08", true)?;
+                        }
+                    },
+                    b'\n' | b'\r' => {
+                        self.cooked.push_back(b);
+                        while let Some(c) = self.cooked.pop_front() {
+                            self.input.push_back(c);
+                        }
+                        self.write(b"\n", true)?;
+                    },
+                    _ => {
+                        self.cooked.push_back(b);
+                        self.write(&[b], true)?;
+                    }
+                }
+            }
+        }
+        Ok(buf.len())
+    }
+
+    pub fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
+        let mut i = 0;
+
+        while i < buf.len() && ! self.input.is_empty() {
+            buf[i] = self.input.pop_front().unwrap();
+            i += 1;
+        }
+
+        if i == 0 {
+            self.end_of_input = false;
+        }
+
+        Ok(i)
+    }
+
+    pub fn will_block(&self) -> bool {
+        self.input.is_empty() && ! self.end_of_input
+    }
+
+    pub fn write(&mut self, buf: &[u8], sync: bool) -> Result<usize> {
+        if self.console.cursor && self.console.x < self.console.w && self.console.y < self.console.h {
+            let x = self.console.x;
+            let y = self.console.y;
+            let color = self.console.background;
+            self.display.rect(x * 8, y * 16, 8, 16, color.data);
+            self.changed.insert(y);
+        }
+
+        {
+            let display = &mut self.display;
+            let changed = &mut self.changed;
+            self.console.write(buf, |event| {
+                match event {
+                    Event::Char { x, y, c, color, bold, .. } => {
+                        display.char(x * 8, y * 16, c, color.data, bold, false);
+                        changed.insert(y);
+                    },
+                    Event::Rect { x, y, w, h, color } => {
+                        display.rect(x * 8, y * 16, w * 8, h * 16, color.data);
+                        for y2 in y..y + h {
+                            changed.insert(y2);
+                        }
+                    },
+                    Event::Scroll { rows, color } => {
+                        display.scroll(rows * 16, color.data);
+                        for y in 0..display.height/16 {
+                            changed.insert(y);
+                        }
+                    }
+                }
+            });
+        }
+
+        if self.console.cursor && self.console.x < self.console.w && self.console.y < self.console.h {
+            let x = self.console.x;
+            let y = self.console.y;
+            let color = self.console.foreground;
+            self.display.rect(x * 8, y * 16, 8, 16, color.data);
+            self.changed.insert(y);
+        }
+
+        if ! self.console.raw_mode && sync {
+            self.sync();
+        }
+
+        Ok(buf.len())
+    }
+
+    pub fn sync(&mut self) {
+        let width = self.display.width;
+        for change in self.changed.iter() {
+            self.display.sync(0, change * 16, width, 16);
+        }
+        self.changed.clear();
+    }
+
+    pub fn redraw(&mut self) {
+        let width = self.display.width;
+        let height = self.display.height;
+        self.display.sync(0, 0, width, height);
+        self.changed.clear();
+    }
+}