BurritOS/src/print.rs
2022-11-10 00:47:32 +01:00

369 lines
15 KiB
Rust

#![allow(dead_code)]
#![allow(unused_variables)]
use crate::decode::Instruction;
const RISCV_LUI: u8 = 0x37;
const RISCV_AUIPC: u8 = 0x17;
const RISCV_JAL: u8 = 0x6f;
const RISCV_JALR: u8 = 0x67;
const RISCV_BR: u8 = 0x63;
const RISCV_LD: u8 = 0x3;
const RISCV_ST: u8 = 0x23;
const RISCV_OPI: u8 = 0x13;
const RISCV_OP: u8 = 0x33;
const RISCV_OPIW: u8 = 0x1b;
const RISCV_OPW: u8 = 0x3b;
const RISCV_BR_BEQ: u8 = 0x0;
const RISCV_BR_BNE: u8 = 0x1;
const RISCV_BR_BLT: u8 = 0x4;
const RISCV_BR_BGE: u8 = 0x5;
const RISCV_BR_BLTU: u8 = 0x6;
const RISCV_BR_BGEU: u8 = 0x7;
const RISCV_LD_LB: u8 = 0x0;
const RISCV_LD_LH: u8 = 0x1;
const RISCV_LD_LW: u8 = 0x2;
const RISCV_LD_LD: u8 = 0x3;
const RISCV_LD_LBU: u8 = 0x4;
const RISCV_LD_LHU: u8 = 0x5;
const RISCV_LD_LWU: u8 = 0x6;
const RISCV_ST_STB: u8 = 0x0;
const RISCV_ST_STH: u8 = 0x1;
const RISCV_ST_STW: u8 = 0x2;
const RISCV_ST_STD: u8 = 0x3;
const RISCV_OPI_ADDI: u8 = 0x0;
const RISCV_OPI_SLTI: u8 = 0x2;
const RISCV_OPI_SLTIU: u8 = 0x3;
const RISCV_OPI_XORI: u8 = 0x4;
const RISCV_OPI_ORI: u8 = 0x6;
const RISCV_OPI_ANDI: u8 = 0x7;
const RISCV_OPI_SLLI: u8 = 0x1;
const RISCV_OPI_SRI: u8 = 0x5;
const RISCV_OPI_SRI_SRAI: u8 = 0x20;
const RISCV_OPI_SRI_SRLI: u8 = 0x0;
const RISCV_OP_ADD: u8 = 0x0;
const RISCV_OP_SLL: u8 = 0x1;
const RISCV_OP_SLT: u8 = 0x2;
const RISCV_OP_SLTU: u8 = 0x3;
const RISCV_OP_XOR: u8 = 0x4;
const RISCV_OP_SR: u8 = 0x5;
const RISCV_OP_OR: u8 = 0x6;
const RISCV_OP_AND: u8 = 0x7;
const RISCV_OP_ADD_ADD: u8 = 0x0;
const RISCV_OP_ADD_SUB: u8 = 0x20;
const RISCV_OP_SR_SRL: u8 = 0x0;
const RISCV_OP_SR_SRA: u8 = 0x20;
const RISCV_SYSTEM: u8 = 0x73;
const RISCV_OPIW_ADDIW: u8 = 0x0;
const RISCV_OPIW_SLLIW: u8 = 0x1;
const RISCV_OPIW_SRW: u8 = 0x5;
const RISCV_OPIW_SRW_SRLIW: u8 = 0x0;
const RISCV_OPIW_SRW_SRAIW: u8 = 0x20;
const RISCV_OPW_ADDSUBW: u8 = 0x0;
const RISCV_OPW_SLLW: u8 = 0x1;
const RISCV_OPW_SRW: u8 = 0x5;
const RISCV_OPW_ADDSUBW_ADDW: u8 = 0x0;
const RISCV_OPW_ADDSUBW_SUBW: u8 = 0x20;
const RISCV_OPW_SRW_SRLW: u8 = 0x0;
const RISCV_OPW_SRW_SRAW: u8 = 0x20;
const RISCV_SYSTEM_ENV: u8 = 0x0;
const RISCV_SYSTEM_ENV_ECALL: u8 = 0x0;
const RISCV_SYSTEM_ENV_EBREAK: u8 = 0x1;
const RISCV_SYSTEM_CSRRS: u8 = 0x2;
const RISCV_SYSTEM_CSRRW: u8 = 0x1;
const RISCV_SYSTEM_CSRRC: u8 = 0x3;
const RISCV_SYSTEM_CSRRWI: u8 = 0x5;
const RISCV_SYSTEM_CSRRSI: u8 = 0x6;
const RISCV_SYSTEM_CSRRCI: u8 = 0x7;
const RISCV_FLW: u8 = 0x07;
const RISCV_FSW: u8 = 0x27;
const RISCV_FMADD: u8 = 0x43;
const RISCV_FMSUB: u8 = 0x47;
const RISCV_FNMSUB: u8 = 0x4b;
const RISCV_FNMADD: u8 = 0x4f;
const RISCV_FP: u8 = 0x53;
const RISCV_FP_ADD: u8 = 0x0;
const RISCV_FP_SUB: u8 = 0x4;
const RISCV_FP_MUL: u8 = 0x8;
const RISCV_FP_DIV: u8 = 0xc;
const RISCV_FP_SQRT: u8 = 0x2c;
const RISCV_FP_FSGN: u8 = 0x10;
const RISCV_FP_MINMAX: u8 = 0x14;
const RISCV_FP_FCVTW: u8 = 0x60;
const RISCV_FP_FMVXFCLASS: u8 = 0x70;
const RISCV_FP_FCMP: u8 = 0x50;
const RISCV_FP_FEQS: u8 = 0x53;
const RISCV_FP_FCVTS: u8 = 0x68;
const RISCV_FP_FCVTDS: u8 = 0x21;
const RISCV_FP_FSGN_J: u8 = 0x0;
const RISCV_FP_FSGN_JN: u8 = 0x1;
const RISCV_FP_FSGN_JX: u8 = 0x2;
const RISCV_FP_MINMAX_MIN: u8 = 0x0;
const RISCV_FP_MINMAX_MAX: u8 = 0x1;
const RISCV_FP_FCVTW_W: u8 = 0x0;
const RISCV_FP_FCVTW_WU: u8 = 0x1;
const RISCV_FP_FCVTS_W: u8 = 0x0;
const RISCV_FP_FCVTS_WU: u8 = 0x1;
const RISCV_FP_FMVXFCLASS_FMVX: u8 = 0x0;
const RISCV_FP_FMVXFCLASS_FCLASS: u8 = 0x1;
const RISCV_FP_FCMP_FEQ: u8 = 2;
const RISCV_FP_FCMP_FLT: u8 = 1;
const RISCV_FP_FCMP_FLE: u8 = 0;
const RISCV_FP_FMVW: u8 = 0x78;
const names_op: [&str; 8] = ["add", "sll", "slt", "sltu", "xor", "sr", "or", "and"];
const names_opi: [&str; 8] = ["addi", "slli", "slti", "sltiu", "xori", "slri", "ori", "andi"];
const names_mul: [&str; 8] = ["mpylo", "mpyhi", "mpyhi", "mpyhi", "divhi", "divhi", "divlo", "divlo"];
const names_br: [&str; 8] = ["beq", "bne", "", "", "blt", "bge", "bltu", "bgeu"];
const names_st: [&str; 4] = ["sb", "sh", "sw", "sd"];
const names_ld: [&str; 7] = ["lb", "lh", "lw", "ld", "lbu", "lhu", "lwu"];
const names_opw: [&str; 8] = ["addw", "sllw", "", "", "", "srw", "", ""];
const names_opiw: [&str; 8] = ["addwi", "sllwi", "", "", "", "sri", "", ""];
pub fn print(ins: Instruction, pc: i32) -> String { //TODO pc should be u64
match ins.opcode {
RISCV_OP => {
let name: &str;
if ins.funct7 == 1 { // Use mul array
name = names_mul[ins.funct3 as usize]
} else {
if ins.funct3 == RISCV_OP_ADD {
// Add or Sub
if ins.funct7 == RISCV_OP_ADD_ADD {
name = "add";
} else {
name = "sub";
}
} else if ins.funct3 == RISCV_OP_SR {
// Srl or Sra
if ins.funct7 == RISCV_OP_SR_SRL {
name = "srl";
} else {
name = "sra";
}
} else {
name = names_op[ins.funct3 as usize];
}
}
format!("{} r{}, r{}, r{}", name.to_string(), &ins.rd.to_string(), &ins.rs1.to_string(), &ins.rs2.to_string())
},
RISCV_OPI => {
// SHAMT OR IMM
if ins.funct3 == RISCV_OPI_SRI {
if ins.funct7 == RISCV_OPI_SRI_SRLI {
format!("slrii x{}, x{}, {}", ins.rd.to_string(), ins.rs1.to_string(), ins.shamt.to_string())
} else {
format!("srai x{}, x{}, {}", ins.rd.to_string(), ins.rs1.to_string(), ins.shamt.to_string())
}
} else if ins.funct3 == RISCV_OPI_SLLI {
format!("{} x{}, x{}, {}", names_opi[ins.funct3 as usize], ins.rd.to_string(), ins.rs1.to_string(), ins.shamt.to_string())
} else {
format!("{} x{}, x{}, {}", names_opi[ins.funct3 as usize], ins.rd.to_string(), ins.rs1.to_string(), ins.imm12_I_signed.to_string())
}
},
RISCV_LUI => {
format!("lui x{}, 0x{:X}", ins.rd.to_string(), ins.imm31_12)
},
RISCV_AUIPC => {
format!("auipc x{}, {:X}", ins.rd.to_string(), ins.imm31_12)
},
RISCV_JAL => {
if ins.rd == 0 {
format!("j {}", ins.imm31_12.to_string())
} else {
format!("jal x{}, {:X}", ins.rd.to_string(), (pc - 4 + ins.imm21_1_signed))
}
},
RISCV_JALR => {
if ins.rd == 0 {
if ins.rs1 == 1 {
"ret".to_string()
} else {
format!("jr {:X}", ins.imm31_12)
}
} else {
format!("jalr {}, (r{})", ins.imm12_I_signed.to_string(), ins.rs1.to_string())
}
},
RISCV_BR => {
format!("{} r{} x{} {}", names_br[ins.funct3 as usize].to_string(), ins.rs1.to_string(), ins.rs2.to_string(), ins.imm13_signed.to_string())
},
RISCV_LD => {
format!("{} x{}, {}(x{})", names_ld[ins.funct3 as usize].to_string(), ins.rd.to_string(), ins.imm12_I_signed.to_string(), ins.rs1.to_string())
},
RISCV_ST => {
format!("{} x{}, {}(x{})", names_st[ins.funct3 as usize].to_string(), ins.rs2.to_string(), ins.imm12_S_signed.to_string(), ins.rs1.to_string())
},
RISCV_OPIW => {
if ins.funct3 == RISCV_OPIW_SRW {
if ins.funct7 == RISCV_OPIW_SRW_SRLIW {
format!("srlwi x{}, x{}, x{}", ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
} else {
format!("srawi x{}, x{}, x{}", ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
}
} else if ins.funct3 == RISCV_OPIW_SLLIW {
format!("{} x{}, x{}, x{}", names_opi[ins.funct3 as usize], ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
} else {
format!("{} x{}, x{}, x{}", names_opiw[ins.funct3 as usize], ins.rd.to_string(), ins.rs1.to_string(), ins.imm12_I_signed.to_string())
}
},
RISCV_OPW => {
if ins.funct7 == 1 {
format!("{}w x{}, x{}, x{}", names_mul[ins.funct3 as usize].to_string(), ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
} else {
if ins.funct3 == RISCV_OP_ADD {
if ins.funct7 == RISCV_OPW_ADDSUBW_ADDW {
format!("addw x{}, x{}, x{}", ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
} else {
format!("subw x{}, x{}, x{}", ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
}
} else if ins.funct3 == RISCV_OPW_SRW {
if ins.funct7 == RISCV_OPW_SRW_SRLW {
format!("srlw x{}, x{}, x{}", ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
} else {
format!("sraw x{}, x{}, x{}", ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
}
} else {
format!("{} x{}, x{}, x{}", names_opw[ins.funct3 as usize], ins.rd.to_string(), ins.rs1.to_string(), ins.rs2.to_string())
}
}
},
RISCV_SYSTEM => {
"ecall".to_string()
},
_ => "Unknown".to_string() // Error
}
}
#[cfg(test)]
mod test {
use crate::{print, decode};
#[test]
fn test_op() {
let sub = decode::decode(0b0100000_10000_10001_000_11100_0110011);
let add = decode::decode(0b0000000_10000_10001_000_11100_0110011);
let xor = decode::decode(0b0000000_10000_10001_100_11100_0110011);
let slr = decode::decode(0b0000000_10000_10001_101_11100_0110011);
let sra = decode::decode(0b0100000_10000_10001_101_11100_0110011);
assert_eq!("sub r28, r17, r16", print::print(sub, 0));
assert_eq!("xor r28, r17, r16", print::print(xor, 0));
assert_eq!("srl r28, r17, r16", print::print(slr, 0));
assert_eq!("sra r28, r17, r16", print::print(sra, 0));
assert_eq!("add r28, r17, r16", print::print(add, 0));
}
#[test]
fn test_opi() {
let addi = decode::decode(0b0000000000_10001_000_11100_0010011);
let slli = decode::decode(0b0000000000_10001_001_11100_0010011);
let slti = decode::decode(0b0000000000_10001_010_11100_0010011);
let sltiu = decode::decode(0b0000000000_10001_011_11100_0010011);
let xori = decode::decode(0b_0000000000010001_100_11100_0010011);
let ori = decode::decode(0b00000000000_10001_110_11100_0010011);
let andi = decode::decode(0b000000000000_10001_111_11100_0010011);
assert_eq!("andi x28, x17, 0", print::print(andi, 0));
assert_eq!("addi x28, x17, 0", print::print(addi, 0));
assert_eq!("slli x28, x17, 0", print::print(slli, 0));
assert_eq!("slti x28, x17, 0", print::print(slti, 0));
assert_eq!("sltiu x28, x17, 0", print::print(sltiu, 0));
assert_eq!("xori x28, x17, 0", print::print(xori, 0));
assert_eq!("ori x28, x17, 0", print::print(ori, 0));
}
#[test]
fn test_lui() {
let lui = decode::decode(0b01110001000011111000_11100_0110111);
let lui_negatif = decode::decode(0b11110001000011111000_11100_0110111);
assert_eq!("lui x28, 0x710F8000", print::print(lui, 0));
assert_eq!("lui x28, 0xF10F8000", print::print(lui_negatif, 0));
}
#[test]
fn test_ld() {
// imm rs1 f3 rd opcode
let lb = decode::decode(0b010111110000_10001_000_11100_0000011);
let lh = decode::decode(0b010111110000_10001_001_11100_0000011);
let lw = decode::decode(0b010111110000_10001_010_11100_0000011);
let lbu = decode::decode(0b010111110000_10001_100_11100_0000011);
let lhu = decode::decode(0b010111110000_10001_101_11100_0000011);
let ld = decode::decode(0b010111110000_10001_011_11100_0000011);
let lwu = decode::decode(0b010111110000_10001_110_11100_0000011);
// TODO: imm négatif produit une erreur
assert_eq!("lb x28, 1520(x17)", print::print(lb, 0));
assert_eq!("lh x28, 1520(x17)", print::print(lh, 0));
assert_eq!("lw x28, 1520(x17)", print::print(lw, 0));
assert_eq!("lbu x28, 1520(x17)", print::print(lbu, 0));
assert_eq!("lhu x28, 1520(x17)", print::print(lhu, 0));
assert_eq!("ld x28, 1520(x17)", print::print(ld, 0));
assert_eq!("lwu x28, 1520(x17)", print::print(lwu, 0));
}
#[test]
fn test_opw() {
let addw: decode::Instruction = decode::decode(0b0000000_10000_10001_000_11100_0111011);
let sllw: decode::Instruction = decode::decode(0b0000000_10000_10001_001_11100_0111011);
let srlw: decode::Instruction = decode::decode(0b0000000_10000_10001_101_11100_0111011);
let sraw: decode::Instruction = decode::decode(0b0100000_10000_10001_101_11100_0111011);
assert_eq!("addw x28, x17, x16", print::print(addw, 0));
assert_eq!("sllw x28, x17, x16", print::print(sllw, 0));
assert_eq!("srlw x28, x17, x16", print::print(srlw, 0));
assert_eq!("sraw x28, x17, x16", print::print(sraw, 0));
}
#[test]
fn test_opwi() {
let addwi: decode::Instruction =decode::decode(0b000000000000_10001_000_11100_0011011);
let sllwi: decode::Instruction = decode::decode(0b0000000_10000_10001_001_11100_0011011);
let srai: decode::Instruction = decode::decode(0b010000010001_10001_101_11100_0010011);
assert_eq!("addwi x28, x17, x0", print::print(addwi, 0));
assert_eq!("slli x28, x17, x16", print::print(sllwi, 0));
assert_eq!("srai x28, x17, 17", print::print(srai, 0));
}
#[test]
fn test_br() {
let beq: decode::Instruction = decode::decode(0b0000000_10000_10001_000_00000_1100011);
let bne: decode::Instruction = decode::decode(0b0000000_10000_10001_001_00000_1100011);
let blt: decode::Instruction = decode::decode(0b0000000_10000_10001_100_00000_1100011);
let bge: decode::Instruction = decode::decode(0b0000000_10000_10001_101_00000_1100011);
let bltu: decode::Instruction = decode::decode(0b0000000_10000_10001_110_00000_1100011);
let bgeu: decode::Instruction = decode::decode(0b0000000_10000_10001_111_00000_1100011);
assert_eq!("blt r17 x16 0", print::print(blt, 0));
assert_eq!("bge r17 x16 0", print::print(bge, 0));
assert_eq!("bltu r17 x16 0", print::print(bltu, 0));
assert_eq!("bgeu r17 x16 0", print::print(bgeu, 0));
assert_eq!("bne r17 x16 0", print::print(bne, 0));
assert_eq!("beq r17 x16 0", print::print(beq, 0));
}
}