update

Author: ved-rivos

riscv/insns/mret.h

@@ -15,14 +15,10 @@ if (ZICFILP_xLPE(prev_virt, prev_prv)) {
 s = set_field(s, MSTATUS_MPELP, elp_t::NO_LP_EXPECTED);
 
 s = set_field(s, MSTATUS_MDT, 0);
-if (prev_prv == PRV_U || prev_virt) {
+if (prev_prv == PRV_U || prev_virt)
   s = set_field(s, MSTATUS_SDT, 0);
-}
-if (prev_virt && prev_prv == PRV_U) {
-  reg_t vs = STATE.vsstatus->read();
-  vs = set_field(vs, SSTATUS_SDT, 0);
-  STATE.vsstatus->write(vs);
-}
+if (prev_virt && prev_prv == PRV_U)
+  STATE.vsstatus->write(STATE.vsstatus->read() & ~SSTATUS_SDT);
 
 STATE.mstatus->write(s);
 if (STATE.mstatush) STATE.mstatush->write(s >> 32); // log mstatush change

riscv/insns/sret.h

@@ -28,17 +28,11 @@ if (ZICFILP_xLPE(prev_virt, prev_prv)) {
 }
 
 if (STATE.prv == PRV_M) {
-  reg_t m = STATE.mstatus->read();
-  m = set_field(m, MSTATUS_MDT, 0);
-  if (prev_prv == PRV_U || prev_virt) {
-    s = set_field(s, MSTATUS_SDT, 0);
-  }
-  STATE.mstatus->write(m);
-  if (prev_virt && prev_prv == PRV_U) {
-    reg_t vs = STATE.vsstatus->read();
-    vs = set_field(vs, SSTATUS_SDT, 0);
-    STATE.vsstatus->write(vs);
-  }
+  STATE.mstatus->write(STATE.mstatus->read() & ~MSTATUS_MDT);
+  if (prev_prv == PRV_U || prev_virt)
+    STATE.mstatus->write(STATE.mstatus->read() & ~MSTATUS_SDT);
+  if (prev_virt && prev_prv == PRV_U)
+    STATE.vsstatus->write(STATE.vsstatus->read() & ~SSTATUS_SDT);
 }
 
 s = set_field(s, SSTATUS_SPELP, elp_t::NO_LP_EXPECTED);