sched: build fallback UE TB info either for SRB0 or SRB1 but not both for a PDSCH grant

Author: herlesupreeth

lib/scheduler/ue_scheduling/ue.cpp

@@ -158,8 +158,10 @@ unsigned ue::build_dl_fallback_transport_block_info(dl_msg_tb_info& tb_info, uns
   unsigned total_subpdu_bytes = 0;
   total_subpdu_bytes += allocate_ue_con_res_id_mac_ce(tb_info, dl_lc_ch_mgr, tb_size_bytes);
   // Since SRB0 PDU cannot be segmented, skip SRB0 if remaining TB size is not enough to fit entire PDU.
-  if ((tb_size_bytes - total_subpdu_bytes) >= dl_lc_ch_mgr.pending_bytes(LCID_SRB0)) {
+  if (dl_lc_ch_mgr.has_pending_bytes(LCID_SRB0) and
+      ((tb_size_bytes - total_subpdu_bytes) >= dl_lc_ch_mgr.pending_bytes(LCID_SRB0))) {
     total_subpdu_bytes += allocate_mac_sdus(tb_info, dl_lc_ch_mgr, tb_size_bytes - total_subpdu_bytes, LCID_SRB0);
+    return total_subpdu_bytes;
   }
   total_subpdu_bytes += allocate_mac_sdus(tb_info, dl_lc_ch_mgr, tb_size_bytes - total_subpdu_bytes, LCID_SRB1);
   return total_subpdu_bytes;