forked from M-Labs/zynq-rs
1
0
Fork 0

add a println! for convenience

This commit is contained in:
Astro 2019-06-20 00:30:18 +02:00
parent b3b65f9b74
commit d65398205f
2 changed files with 37 additions and 14 deletions

View File

@ -6,7 +6,6 @@
#![feature(compiler_builtins_lib)] #![feature(compiler_builtins_lib)]
#![feature(never_type)] #![feature(never_type)]
use core::fmt::Write;
use core::mem::uninitialized; use core::mem::uninitialized;
use r0::zero_bss; use r0::zero_bss;
@ -16,7 +15,7 @@ mod regs;
mod cortex_a9; mod cortex_a9;
mod slcr; mod slcr;
mod uart; mod uart;
use uart::Uart; mod stdio;
mod eth; mod eth;
use crate::regs::{RegisterR, RegisterW}; use crate::regs::{RegisterR, RegisterW};
@ -71,31 +70,27 @@ fn l1_cache_init() {
dccisw(); dccisw();
} }
const UART_RATE: u32 = 115_200;
fn main() { fn main() {
let mut uart = Uart::serial(UART_RATE); println!("Main.");
writeln!(uart, "\r\nHello World!\r");
let mut eth = eth::Eth::default([0x0, 0x17, 0xde, 0xea, 0xbe, 0xef]); let mut eth = eth::Eth::default([0x0, 0x17, 0xde, 0xea, 0xbe, 0xef]);
writeln!(uart, "Eth on\r"); println!("Eth on");
match eth::phy::Phy::find(&mut eth) { match eth::phy::Phy::find(&mut eth) {
Some((addr, phy)) => { Some((addr, phy)) => {
writeln!(uart, "Found {} PHY at addr {}\r", phy.name(), addr); println!("Found {} PHY at addr {}", phy.name(), addr);
} }
None => { None => {
use eth::phy::PhyAccess; use eth::phy::PhyAccess;
for addr in 1..32 { for addr in 1..32 {
match eth::phy::id::identify_phy(&mut eth, addr) { match eth::phy::id::identify_phy(&mut eth, addr) {
Some(identifier) => { Some(identifier) => {
writeln!(uart, "phy {}: {:?}\r", addr, identifier); println!("phy {}: {:?}", addr, identifier);
} }
None => {} None => {}
} }
} }
} }
} }
while !uart.tx_fifo_empty() {}
let mut rx_buffers = [[0u8; 1536]; eth::rx::DESCS]; let mut rx_buffers = [[0u8; 1536]; eth::rx::DESCS];
let mut rx_buffer_ptrs: [&mut [u8]; eth::rx::DESCS] = unsafe { let mut rx_buffer_ptrs: [&mut [u8]; eth::rx::DESCS] = unsafe {
@ -110,7 +105,7 @@ fn main() {
match eth.recv_next() { match eth.recv_next() {
None => {} None => {}
Some(pkt) => { Some(pkt) => {
writeln!(uart, "eth: received {} bytes\r", pkt.len()); println!("eth: received {} bytes", pkt.len());
} }
} }
} }
@ -119,9 +114,7 @@ fn main() {
#[panic_handler] #[panic_handler]
fn panic(info: &core::panic::PanicInfo) -> ! { fn panic(info: &core::panic::PanicInfo) -> ! {
let mut uart = Uart::serial(UART_RATE); println!("\nPanic: {}", info);
writeln!(uart, "\r\nPanic: {}\r", info);
while !uart.tx_fifo_empty() {}
slcr::RegisterBlock::unlocked(|slcr| slcr.soft_reset()); slcr::RegisterBlock::unlocked(|slcr| slcr.soft_reset());
loop {} loop {}

30
src/stdio.rs Normal file
View File

@ -0,0 +1,30 @@
use crate::uart::Uart;
const UART_RATE: u32 = 115_200;
static mut UART: Option<Uart> = None;
// TODO: locking for SMP
#[doc(hidden)]
pub fn get_uart() -> &'static mut Uart {
unsafe {
match &mut UART {
None => {
let mut uart = Uart::serial(UART_RATE);
UART = Some(uart);
UART.as_mut().unwrap()
}
Some(uart) => uart,
}
}
}
#[macro_export]
macro_rules! println {
($($arg:tt)*) => ({
use core::fmt::Write;
let uart = crate::stdio::get_uart();
write!(uart, $($arg)*);
write!(uart, "\r\n");
while !uart.tx_fifo_empty() {}
})
}