*/
 enum dmub_status dmub_srv_sync_inboxes(struct dmub_srv *dmub);
 
+/**
+ * dmub_srv_wait_for_inbox_free() - Waits for space in the DMUB inbox to free up
+ * @dmub: the dmub service
+ * @timeout_us: the maximum number of microseconds to wait
+ * @num_free_required: number of free entries required
+ *
+ * Waits until the DMUB buffer is freed to the specified number.
+ *  The maximum wait time is given in microseconds to prevent spinning
+ * forever.
+ *
+ * Return:
+ *   DMUB_STATUS_OK - success
+ *   DMUB_STATUS_TIMEOUT - wait for buffer to flush timed out
+ *   DMUB_STATUS_INVALID - unspecified error
+ */
+enum dmub_status dmub_srv_wait_for_inbox_free(struct dmub_srv *dmub,
+               uint32_t timeout_us,
+               uint32_t num_free_required);
+
 #endif /* _DMUB_SRV_H_ */
 
        return DMUB_STATUS_TIMEOUT;
 }
 
+static enum dmub_status dmub_srv_update_inbox_status(struct dmub_srv *dmub)
+{
+       uint32_t rptr;
+
+       /* update inbox1 state */
+               rptr = dmub->hw_funcs.get_inbox1_rptr(dmub);
+
+       if (rptr > dmub->inbox1.rb.capacity)
+               return DMUB_STATUS_HW_FAILURE;
+
+       if (dmub->inbox1.rb.rptr > rptr) {
+               /* rb wrapped */
+               dmub->inbox1.num_reported += (rptr + dmub->inbox1.rb.capacity - dmub->inbox1.rb.rptr) / DMUB_RB_CMD_SIZE;
+       } else {
+               dmub->inbox1.num_reported += (rptr - dmub->inbox1.rb.rptr) / DMUB_RB_CMD_SIZE;
+       }
+       dmub->inbox1.rb.rptr = rptr;
+
+       /* update reg_inbox0 */
+       dmub_srv_update_reg_inbox0_status(dmub);
+
+       return DMUB_STATUS_OK;
+}
+
 enum dmub_status dmub_srv_wait_for_idle(struct dmub_srv *dmub,
                                        uint32_t timeout_us)
 {
-       uint32_t i, rptr;
+       enum dmub_status status;
+       uint32_t i;
        const uint32_t polling_interval_us = 1;
 
        if (!dmub->hw_init)
                return DMUB_STATUS_INVALID;
 
        for (i = 0; i < timeout_us; i += polling_interval_us) {
-               /* update inbox1 state */
-                       rptr = dmub->hw_funcs.get_inbox1_rptr(dmub);
+               status = dmub_srv_update_inbox_status(dmub);
 
-               if (rptr > dmub->inbox1.rb.capacity)
-                       return DMUB_STATUS_HW_FAILURE;
-
-               if (dmub->inbox1.rb.rptr > rptr) {
-                       /* rb wrapped */
-                       dmub->inbox1.num_reported += (rptr + dmub->inbox1.rb.capacity - dmub->inbox1.rb.rptr) / DMUB_RB_CMD_SIZE;
-               } else {
-                       dmub->inbox1.num_reported += (rptr - dmub->inbox1.rb.rptr) / DMUB_RB_CMD_SIZE;
-               }
-               dmub->inbox1.rb.rptr = rptr;
-
-               /* update reg_inbox0 */
-               dmub_srv_update_reg_inbox0_status(dmub);
+               if (status != DMUB_STATUS_OK)
+                       return status;
 
                /* check for idle */
                if (dmub_rb_empty(&dmub->inbox1.rb) && !dmub->reg_inbox0.is_pending)
 
        return DMUB_STATUS_OK;
 }
+
+enum dmub_status dmub_srv_wait_for_inbox_free(struct dmub_srv *dmub,
+               uint32_t timeout_us,
+               uint32_t num_free_required)
+{
+       enum dmub_status status;
+       uint32_t i;
+       const uint32_t polling_interval_us = 1;
+
+       if (!dmub->hw_init)
+               return DMUB_STATUS_INVALID;
+
+       for (i = 0; i < timeout_us; i += polling_interval_us) {
+               status = dmub_srv_update_inbox_status(dmub);
+
+               if (status != DMUB_STATUS_OK)
+                       return status;
+
+               /* check for space in inbox1 */
+               if (dmub_rb_num_free(&dmub->inbox1.rb) >= num_free_required)
+                       return DMUB_STATUS_OK;
+
+               udelay(polling_interval_us);
+       }
+
+       return DMUB_STATUS_TIMEOUT;
+}