1/*************************************************************************************** 2* Copyright (c) 2020-2021 Institute of Computing Technology, Chinese Academy of Sciences 3* Copyright (c) 2020-2021 Peng Cheng Laboratory 4* 5* XiangShan is licensed under Mulan PSL v2. 6* You can use this software according to the terms and conditions of the Mulan PSL v2. 7* You may obtain a copy of Mulan PSL v2 at: 8* http://license.coscl.org.cn/MulanPSL2 9* 10* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, 11* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, 12* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE. 13* 14* See the Mulan PSL v2 for more details. 15***************************************************************************************/ 16 17package xiangshan.frontend 18 19import chipsalliance.rocketchip.config.Parameters 20import freechips.rocketchip.rocket.{RVCDecoder, ExpandedInstruction} 21import chisel3.{util, _} 22import chisel3.util._ 23import utils._ 24import xiangshan._ 25import xiangshan.frontend.icache._ 26import xiangshan.backend.decode.isa.predecode.PreDecodeInst 27 28trait HasPdConst extends HasXSParameter with HasICacheParameters with HasIFUConst{ 29 def isRVC(inst: UInt) = (inst(1,0) =/= 3.U) 30 def isLink(reg:UInt) = reg === 1.U || reg === 5.U 31 def brInfo(instr: UInt) = { 32 val brType::Nil = ListLookup(instr, List(BrType.notCFI), PreDecodeInst.brTable) 33 val rd = Mux(isRVC(instr), instr(12), instr(11,7)) 34 val rs = Mux(isRVC(instr), Mux(brType === BrType.jal, 0.U, instr(11, 7)), instr(19, 15)) 35 val isCall = (brType === BrType.jal && !isRVC(instr) || brType === BrType.jalr) && isLink(rd) // Only for RV64 36 val isRet = brType === BrType.jalr && isLink(rs) && !isCall 37 List(brType, isCall, isRet) 38 } 39 def jal_offset(inst: UInt, rvc: Bool): UInt = { 40 val rvc_offset = Cat(inst(12), inst(8), inst(10, 9), inst(6), inst(7), inst(2), inst(11), inst(5, 3), 0.U(1.W)) 41 val rvi_offset = Cat(inst(31), inst(19, 12), inst(20), inst(30, 21), 0.U(1.W)) 42 val max_width = rvi_offset.getWidth 43 SignExt(Mux(rvc, SignExt(rvc_offset, max_width), SignExt(rvi_offset, max_width)), XLEN) 44 } 45 def br_offset(inst: UInt, rvc: Bool): UInt = { 46 val rvc_offset = Cat(inst(12), inst(6, 5), inst(2), inst(11, 10), inst(4, 3), 0.U(1.W)) 47 val rvi_offset = Cat(inst(31), inst(7), inst(30, 25), inst(11, 8), 0.U(1.W)) 48 val max_width = rvi_offset.getWidth 49 SignExt(Mux(rvc, SignExt(rvc_offset, max_width), SignExt(rvi_offset, max_width)), XLEN) 50 } 51 52 def NOP = "h4501".U(16.W) 53} 54 55object BrType { 56 def notCFI = "b00".U 57 def branch = "b01".U 58 def jal = "b10".U 59 def jalr = "b11".U 60 def apply() = UInt(2.W) 61} 62 63object ExcType { //TODO:add exctype 64 def notExc = "b000".U 65 def apply() = UInt(3.W) 66} 67 68class PreDecodeInfo extends Bundle { // 8 bit 69 val valid = Bool() 70 val isRVC = Bool() 71 val brType = UInt(2.W) 72 val isCall = Bool() 73 val isRet = Bool() 74 //val excType = UInt(3.W) 75 def isBr = brType === BrType.branch 76 def isJal = brType === BrType.jal 77 def isJalr = brType === BrType.jalr 78 def notCFI = brType === BrType.notCFI 79} 80 81class PreDecodeResp(implicit p: Parameters) extends XSBundle with HasPdConst { 82 val pd = Vec(PredictWidth, new PreDecodeInfo) 83 val hasHalfValid = Vec(PredictWidth, Bool()) 84 //val expInstr = Vec(PredictWidth, UInt(32.W)) 85 val instr = Vec(PredictWidth, UInt(32.W)) 86 val jumpOffset = Vec(PredictWidth, UInt(XLEN.W)) 87// val hasLastHalf = Bool() 88 val triggered = Vec(PredictWidth, new TriggerCf) 89} 90 91class PreDecode(implicit p: Parameters) extends XSModule with HasPdConst{ 92 val io = IO(new Bundle() { 93 val in = Input(new IfuToPreDecode) 94 val out = Output(new PreDecodeResp) 95 }) 96 97 val data = io.in.data 98// val lastHalfMatch = io.in.lastHalfMatch 99 val validStart, validEnd = Wire(Vec(PredictWidth, Bool())) 100 val h_validStart, h_validEnd = Wire(Vec(PredictWidth, Bool())) 101 102 val rawInsts = if (HasCExtension) VecInit((0 until PredictWidth).map(i => Cat(data(i+1), data(i)))) 103 else VecInit((0 until PredictWidth).map(i => data(i))) 104 105 for (i <- 0 until PredictWidth) { 106 val inst =WireInit(rawInsts(i)) 107 //val expander = Module(new RVCExpander) 108 val currentIsRVC = isRVC(inst) 109 val currentPC = io.in.pc(i) 110 //expander.io.in := inst 111 112 val brType::isCall::isRet::Nil = brInfo(inst) 113 val jalOffset = jal_offset(inst, currentIsRVC) 114 val brOffset = br_offset(inst, currentIsRVC) 115 116 //val lastIsValidEnd = if (i == 0) { !lastHalfMatch } else { validEnd(i-1) || !HasCExtension.B } 117 val lastIsValidEnd = if (i == 0) { true.B } else { validEnd(i-1) || !HasCExtension.B } 118 validStart(i) := (lastIsValidEnd || !HasCExtension.B) 119 validEnd(i) := validStart(i) && currentIsRVC || !validStart(i) || !HasCExtension.B 120 121 //prepared for last half match 122 //TODO if HasCExtension 123 val h_lastIsValidEnd = if (i == 0) { false.B } else { h_validEnd(i-1) || !HasCExtension.B } 124 h_validStart(i) := (h_lastIsValidEnd || !HasCExtension.B) 125 h_validEnd(i) := h_validStart(i) && currentIsRVC || !h_validStart(i) || !HasCExtension.B 126 127 io.out.hasHalfValid(i) := h_validStart(i) 128 129 io.out.triggered(i) := DontCare//VecInit(Seq.fill(10)(false.B)) 130 131 132 io.out.pd(i).valid := validStart(i) 133 io.out.pd(i).isRVC := currentIsRVC 134 io.out.pd(i).brType := brType 135 io.out.pd(i).isCall := isCall 136 io.out.pd(i).isRet := isRet 137 138 //io.out.expInstr(i) := expander.io.out.bits 139 io.out.instr(i) :=inst 140 io.out.jumpOffset(i) := Mux(io.out.pd(i).isBr, brOffset, jalOffset) 141 } 142 143// io.out.hasLastHalf := !io.out.pd(PredictWidth - 1).isRVC && io.out.pd(PredictWidth - 1).valid 144 145 for (i <- 0 until PredictWidth) { 146 XSDebug(true.B, 147 p"instr ${Hexadecimal(io.out.instr(i))}, " + 148 p"validStart ${Binary(validStart(i))}, " + 149 p"validEnd ${Binary(validEnd(i))}, " + 150 p"isRVC ${Binary(io.out.pd(i).isRVC)}, " + 151 p"brType ${Binary(io.out.pd(i).brType)}, " + 152 p"isRet ${Binary(io.out.pd(i).isRet)}, " + 153 p"isCall ${Binary(io.out.pd(i).isCall)}\n" 154 ) 155 } 156} 157 158class RVCExpander(implicit p: Parameters) extends XSModule { 159 val io = IO(new Bundle { 160 val in = Input(UInt(32.W)) 161 val out = Output(new ExpandedInstruction) 162 }) 163 164 if (HasCExtension) { 165 io.out := new RVCDecoder(io.in, XLEN).decode 166 } else { 167 io.out := new RVCDecoder(io.in, XLEN).passthrough 168 } 169} 170 171/* --------------------------------------------------------------------- 172 * Predict result check 173 * 174 * --------------------------------------------------------------------- 175 */ 176 177object FaultType { 178 def noFault = "b000".U 179 def jalFault = "b001".U //not CFI taken or invalid instruction taken 180 def retFault = "b010".U //not CFI taken or invalid instruction taken 181 def targetFault = "b011".U 182 def notCFIFault = "b100".U //not CFI taken or invalid instruction taken 183 def invalidTaken = "b101".U 184 def apply() = UInt(3.W) 185} 186 187class CheckInfo extends Bundle { // 8 bit 188 val value = UInt(3.W) 189 def isjalFault = value === FaultType.jalFault 190 def isRetFault = value === FaultType.retFault 191 def istargetFault = value === FaultType.targetFault 192 def invalidTakenFault = value === FaultType.invalidTaken 193 def notCFIFault = value === FaultType.notCFIFault 194} 195 196class PredCheckerResp(implicit p: Parameters) extends XSBundle with HasPdConst { 197 //to Ibuffer write port (stage 1) 198 val stage1Out = new Bundle{ 199 val fixedRange = Vec(PredictWidth, Bool()) 200 val fixedTaken = Vec(PredictWidth, Bool()) 201 } 202 //to Ftq write back port (stage 2) 203 val stage2Out = new Bundle{ 204 val fixedTarget = Vec(PredictWidth, UInt(VAddrBits.W)) 205 val fixedMissPred = Vec(PredictWidth, Bool()) 206 val faultType = Vec(PredictWidth, new CheckInfo) 207 } 208} 209 210 211class PredChecker(implicit p: Parameters) extends XSModule with HasPdConst { 212 val io = IO( new Bundle{ 213 val in = Input(new IfuToPredChecker) 214 val out = Output(new PredCheckerResp) 215 }) 216 217 val (takenIdx, predTaken) = (io.in.ftqOffset.bits, io.in.ftqOffset.valid) 218 val predTarget = (io.in.target) 219 val (instrRange, instrValid) = (io.in.instrRange, io.in.instrValid) 220 val (pds, pc, jumpOffset) = (io.in.pds, io.in.pc, io.in.jumpOffset) 221 222 val jalFaultVec, retFaultVec, targetFault, notCFITaken, invalidTaken = Wire(Vec(PredictWidth, Bool())) 223 224 /** remask fault may appear together with other faults, but other faults are exclusive 225 * so other f ault mast use fixed mask to keep only one fault would be found and redirect to Ftq 226 * we first detecct remask fault and then use fixedRange to do second check 227 **/ 228 229 //Stage 1: detect remask fault 230 /** first check: remask Fault */ 231 jalFaultVec := VecInit(pds.zipWithIndex.map{case(pd, i) => pd.isJal && instrRange(i) && instrValid(i) && (takenIdx > i.U && predTaken || !predTaken) }) 232 retFaultVec := VecInit(pds.zipWithIndex.map{case(pd, i) => pd.isRet && instrRange(i) && instrValid(i) && (takenIdx > i.U && predTaken || !predTaken) }) 233 val remaskFault = VecInit((0 until PredictWidth).map(i => jalFaultVec(i) || retFaultVec(i))) 234 val remaskIdx = ParallelPriorityEncoder(remaskFault.asUInt) 235 val needRemask = ParallelOR(remaskFault) 236 val fixedRange = instrRange.asUInt & (Fill(PredictWidth, !needRemask) | Fill(PredictWidth, 1.U(1.W)) >> ~remaskIdx) 237 238 io.out.stage1Out.fixedRange := fixedRange.asTypeOf((Vec(PredictWidth, Bool()))) 239 240 io.out.stage1Out.fixedTaken := VecInit(pds.zipWithIndex.map{case(pd, i) => instrValid (i) && fixedRange(i) && (pd.isRet || pd.isJal || takenIdx === i.U && predTaken && !pd.notCFI) }) 241 242 /** second check: faulse prediction fault and target fault */ 243 notCFITaken := VecInit(pds.zipWithIndex.map{case(pd, i) => fixedRange(i) && instrValid(i) && i.U === takenIdx && pd.notCFI && predTaken }) 244 invalidTaken := VecInit(pds.zipWithIndex.map{case(pd, i) => fixedRange(i) && !instrValid(i) && i.U === takenIdx && predTaken }) 245 246 val jumpTargets = VecInit(pds.zipWithIndex.map{case(pd,i) => pc(i) + jumpOffset(i)}) 247 val seqTargets = VecInit((0 until PredictWidth).map(i => pc(i) + Mux(pds(i).isRVC || !instrValid(i), 2.U, 4.U ) )) 248 249 //Stage 2: detect target fault 250 /** target calculation: in the next stage */ 251 val fixedRangeNext = RegNext(fixedRange) 252 val instrValidNext = RegNext(instrValid) 253 val takenIdxNext = RegNext(takenIdx) 254 val predTakenNext = RegNext(predTaken) 255 val predTargetNext = RegNext(predTarget) 256 val jumpTargetsNext = RegNext(jumpTargets) 257 val seqTargetsNext = RegNext(seqTargets) 258 val pdsNext = RegNext(pds) 259 val jalFaultVecNext = RegNext(jalFaultVec) 260 val retFaultVecNext = RegNext(retFaultVec) 261 val notCFITakenNext = RegNext(notCFITaken) 262 val invalidTakenNext = RegNext(invalidTaken) 263 264 targetFault := VecInit(pdsNext.zipWithIndex.map{case(pd,i) => fixedRangeNext(i) && instrValidNext(i) && (pd.isJal || pd.isBr) && takenIdxNext === i.U && predTakenNext && (predTargetNext =/= jumpTargetsNext(i))}) 265 266 267 io.out.stage2Out.faultType.zipWithIndex.map{case(faultType, i) => faultType.value := Mux(jalFaultVecNext(i) , FaultType.jalFault , 268 Mux(retFaultVecNext(i), FaultType.retFault , 269 Mux(targetFault(i), FaultType.targetFault , 270 Mux(notCFITakenNext(i) , FaultType.notCFIFault, 271 Mux(invalidTakenNext(i), FaultType.invalidTaken, FaultType.noFault)))))} 272 273 io.out.stage2Out.fixedMissPred.zipWithIndex.map{case(missPred, i ) => missPred := jalFaultVecNext(i) || retFaultVecNext(i) || notCFITakenNext(i) || invalidTakenNext(i) || targetFault(i)} 274 io.out.stage2Out.fixedTarget.zipWithIndex.map{case(target, i) => target := Mux(jalFaultVecNext(i) || targetFault(i), jumpTargetsNext(i), seqTargetsNext(i) )} 275 276} 277 278class FrontendTrigger(implicit p: Parameters) extends XSModule { 279 val io = IO(new Bundle(){ 280 val frontendTrigger = Input(new FrontendTdataDistributeIO) 281 val csrTriggerEnable = Input(Vec(4, Bool())) 282 val triggered = Output(Vec(PredictWidth, new TriggerCf)) 283 284 val pds = Input(Vec(PredictWidth, new PreDecodeInfo)) 285 val pc = Input(Vec(PredictWidth, UInt(VAddrBits.W))) 286 val data = if(HasCExtension) Input(Vec(PredictWidth + 1, UInt(16.W))) 287 else Input(Vec(PredictWidth, UInt(32.W))) 288 }) 289 290 val data = io.data 291 292 val rawInsts = if (HasCExtension) VecInit((0 until PredictWidth).map(i => Cat(data(i+1), data(i)))) 293 else VecInit((0 until PredictWidth).map(i => data(i))) 294 295 val tdata = RegInit(VecInit(Seq.fill(4)(0.U.asTypeOf(new MatchTriggerIO)))) 296 when(io.frontendTrigger.t.valid) { 297 tdata(io.frontendTrigger.t.bits.addr) := io.frontendTrigger.t.bits.tdata 298 } 299 io.triggered.map{i => i := 0.U.asTypeOf(new TriggerCf)} 300 val triggerEnable = RegInit(VecInit(Seq.fill(4)(false.B))) // From CSR, controlled by priv mode, etc. 301 triggerEnable := io.csrTriggerEnable 302 XSDebug(triggerEnable.asUInt.orR, "Debug Mode: At least one frontend trigger is enabled\n") 303 304 for (i <- 0 until 4) {PrintTriggerInfo(triggerEnable(i), tdata(i))} 305 306 for (i <- 0 until PredictWidth) { 307 val currentPC = io.pc(i) 308 val currentIsRVC = io.pds(i).isRVC 309 val inst = WireInit(rawInsts(i)) 310 val triggerHitVec = Wire(Vec(4, Bool())) 311 312 for (j <- 0 until 4) { 313 triggerHitVec(j) := Mux(tdata(j).select, TriggerCmp(Mux(currentIsRVC, inst(15, 0), inst), tdata(j).tdata2, tdata(j).matchType, triggerEnable(j)), 314 TriggerCmp(currentPC, tdata(j).tdata2, tdata(j).matchType, triggerEnable(j))) 315 } 316 317 // fix chains this could be moved further into the pipeline 318 io.triggered(i).frontendHit := triggerHitVec 319 val enableChain = tdata(0).chain 320 when(enableChain){ 321 io.triggered(i).frontendHit(0) := triggerHitVec(0) && triggerHitVec(1) && (tdata(0).timing === tdata(1).timing) 322 io.triggered(i).frontendHit(1) := triggerHitVec(0) && triggerHitVec(1) && (tdata(0).timing === tdata(1).timing) 323 } 324 for(j <- 0 until 2) { 325 io.triggered(i).backendEn(j) := Mux(tdata(j+2).chain, triggerHitVec(j+2), true.B) 326 io.triggered(i).frontendHit(j+2) := !tdata(j+2).chain && triggerHitVec(j+2) // temporary workaround 327 } 328 XSDebug(io.triggered(i).getHitFrontend, p"Debug Mode: Predecode Inst No. ${i} has trigger hit vec ${io.triggered(i).frontendHit}" + 329 p"and backend en ${io.triggered(i).backendEn}\n") 330 } 331} 332