srenblad/artiq
1
0
Fork 0
forked from M-Labs/artiq
Code Pull Requests Activity

netboot: support slave FPGA loading

Browse Source
This commit is contained in:
Sebastien Bourdeauducq 2019-11-05 16:28:49 +08:00
parent 307f39e900
commit b25a17fa37
3 changed files with 111 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

97
artiq/firmware/bootloader/main.rs
View File

@ -140,15 +140,14 @@ fn load_slave_fpga() {
println!(" ...Error during loading: {}", e);
return
}
if let Err(e) = slave_fpga::complete() {
println!(" ...Error during completion: {}", e);
if let Err(e) = slave_fpga::startup() {
println!(" ...Error during startup: {}", e);
return
}
println!(" ...done");
}
fn flash_boot() {
const FIRMWARE: *mut u8 = board_mem::FLASH_BOOT_ADDRESS as *mut u8;
const MAIN_RAM: *mut u8 = board_mem::MAIN_RAM_BASE as *mut u8;
@ -193,8 +192,16 @@ enum NetConnState {
WaitCommand,
FirmwareLength(usize, u8),
FirmwareDownload(usize, usize),
WaitO,
WaitK
FirmwareWaitO,
FirmwareWaitK,
#[cfg(has_slave_fpga_cfg)]
GatewareLength(usize, u8),
#[cfg(has_slave_fpga_cfg)]
GatewareDownload(usize, usize),
#[cfg(has_slave_fpga_cfg)]
GatewareWaitO,
#[cfg(has_slave_fpga_cfg)]
GatewareWaitK
}
#[cfg(has_ethmac)]
@ -228,6 +235,12 @@ impl NetConn {
self.state = NetConnState::FirmwareLength(0, 0);
Ok(1)
},
#[cfg(has_slave_fpga_cfg)]
b'G' => {
println!("Received gateware load command");
self.state = NetConnState::GatewareLength(0, 0);
Ok(1)
}
b'B' => {
if self.firmware_downloaded {
println!("Received boot command");
@ -245,6 +258,7 @@ impl NetConn {
}
}
},
NetConnState::FirmwareLength(firmware_length, recv_bytes) => {
let firmware_length = (firmware_length << 8) | (buf[0] as usize);
let recv_bytes = recv_bytes + 1;
@ -269,23 +283,23 @@ impl NetConn {
let recv_bytes = recv_bytes + length;
if recv_bytes == firmware_length {
self.state = NetConnState::WaitO;
self.state = NetConnState::FirmwareWaitO;
Ok(length)
} else {
self.state = NetConnState::FirmwareDownload(firmware_length, recv_bytes);
Ok(length)
}
},
NetConnState::WaitO => {
NetConnState::FirmwareWaitO => {
if buf[0] == b'O' {
self.state = NetConnState::WaitK;
self.state = NetConnState::FirmwareWaitK;
Ok(1)
} else {
println!("End-of-firmware confirmation failed");
Err(())
}
},
NetConnState::WaitK => {
NetConnState::FirmwareWaitK => {
if buf[0] == b'K' {
println!("Firmware successfully downloaded");
self.state = NetConnState::WaitCommand;
@ -296,6 +310,71 @@ impl NetConn {
Err(())
}
}
#[cfg(has_slave_fpga_cfg)]
NetConnState::GatewareLength(gateware_length, recv_bytes) => {
let gateware_length = (gateware_length << 8) | (buf[0] as usize);
let recv_bytes = recv_bytes + 1;
if recv_bytes == 4 {
if let Err(e) = slave_fpga::prepare() {
println!(" Error during slave FPGA preparation: {}", e);
return Err(())
}
self.state = NetConnState::GatewareDownload(gateware_length, 0);
} else {
self.state = NetConnState::GatewareLength(gateware_length, recv_bytes);
}
Ok(1)
},
#[cfg(has_slave_fpga_cfg)]
NetConnState::GatewareDownload(gateware_length, recv_bytes) => {
let max_length = gateware_length - recv_bytes;
let buf = if buf.len() > max_length {
&buf[..max_length]
} else {
&buf[..]
};
let length = buf.len();
if let Err(e) = slave_fpga::input(buf) {
println!("Error during slave FPGA loading: {}", e);
return Err(())
}
let recv_bytes = recv_bytes + length;
if recv_bytes == gateware_length {
self.state = NetConnState::GatewareWaitO;
Ok(length)
} else {
self.state = NetConnState::GatewareDownload(gateware_length, recv_bytes);
Ok(length)
}
},
#[cfg(has_slave_fpga_cfg)]
NetConnState::GatewareWaitO => {
if buf[0] == b'O' {
self.state = NetConnState::GatewareWaitK;
Ok(1)
} else {
println!("End-of-gateware confirmation failed");
Err(())
}
},
#[cfg(has_slave_fpga_cfg)]
NetConnState::GatewareWaitK => {
if buf[0] == b'K' {
if let Err(e) = slave_fpga::startup() {
println!("Error during slave FPGA startup: {}", e);
return Err(())
}
println!("Gateware successfully downloaded");
self.state = NetConnState::WaitCommand;
Ok(1)
} else {
println!("End-of-gateware confirmation failed");
Err(())
}
}
}
}

2
artiq/firmware/libboard_misoc/slave_fpga.rs
View File

@ -59,7 +59,7 @@ pub fn input(data: &[u8]) -> Result<(), &'static str> {
Ok(())
}
pub fn complete() -> Result<(), &'static str> {
pub fn startup() -> Result<(), &'static str> {
unsafe {
let t = clock::get_ms();
while csr::slave_fpga_cfg::in_read() & DONE_BIT == 0 {

41
artiq/frontend/artiq_netboot.py
View File

@ -5,21 +5,9 @@ import socket
import struct
import os
def main():
parser = argparse.ArgumentParser(description="ARTIQ netboot tool")
parser.add_argument("hostname", metavar="HOSTNAME",
help="hostname of the target board")
parser.add_argument("-f", "--firmware", nargs=1,
help="firmware to load")
parser.add_argument("-b", "--boot", action="store_true",
help="boot the device")
args = parser.parse_args()
sock = socket.create_connection((args.hostname, 4269))
try:
if args.firmware is not None:
with open(args.firmware[0], "rb") as input_file:
sock.sendall(b"F")
def send_file(sock, filename):
with open(filename, "rb") as input_file:
sock.sendall(struct.pack(">I", os.fstat(input_file.fileno()).st_size))
while True:
data = input_file.read(4096)
@ -27,6 +15,31 @@ def main():
break
sock.sendall(data)
sock.sendall(b"OK")
def main():
parser = argparse.ArgumentParser(description="ARTIQ netboot tool")
parser.add_argument("hostname", metavar="HOSTNAME",
help="hostname of the target board")
parser.add_argument("-f", "--firmware", nargs=1,
help="firmware to load")
# Note that on softcore systems, the main gateware cannot be replaced
# with -g. This option is used for loading the RTM FPGA from the AMC
# on Sayma, and the PL on Zynq.
parser.add_argument("-g", "--gateware", nargs=1,
help="gateware to load")
parser.add_argument("-b", "--boot", action="store_true",
help="boot the device")
args = parser.parse_args()
sock = socket.create_connection((args.hostname, 4269))
try:
if args.firmware is not None:
sock.sendall(b"F")
send_file(sock, args.firmware[0])
if args.gateware is not None:
sock.sendall(b"G")
send_file(sock, args.gateware[0])
if args.boot:
sock.sendall(b"B")
finally: