drm/display/dp_mst: Correct the kref of port.

[why & how]
We still need to refer to port while removing payload at commit_tail.
we should keep the kref till then to release.

Bug: https://gitlab.freedesktop.org/drm/amd/-/issues/2171
Signed-off-by: Wayne Lin <[email protected]>
Signed-off-by: Harry Wentland <[email protected]>
Fixes: 4d07b0b ("drm/display/dp_mst: Move all payload info into the atomic state")
Cc: [email protected] # 6.1
Acked-by: Harry Wentland <[email protected]>
Reviewed-by: Lyude Paul <[email protected]>
Tested-by: Didier Raboud <[email protected]>
Signed-off-by: Alex Deucher <[email protected]>

Author: Wayne Lin

drivers/gpu/drm/display/drm_dp_mst_topology.c

@@ -3372,6 +3372,9 @@ void drm_dp_remove_payload(struct drm_dp_mst_topology_mgr *mgr,
 
 	mgr->payload_count--;
 	mgr->next_start_slot -= payload->time_slots;
+
+	if (payload->delete)
+		drm_dp_mst_put_port_malloc(payload->port);
 }
 EXPORT_SYMBOL(drm_dp_remove_payload);
 
@@ -4327,7 +4330,6 @@ int drm_dp_atomic_release_time_slots(struct drm_atomic_state *state,
 
 	drm_dbg_atomic(mgr->dev, "[MST PORT:%p] TU %d -> 0\n", port, payload->time_slots);
 	if (!payload->delete) {
-		drm_dp_mst_put_port_malloc(port);
 		payload->pbn = 0;
 		payload->delete = true;
 		topology_state->payload_mask &= ~BIT(payload->vcpi - 1);