drm/amd/display: add null check

[ Upstream commit 158b9201c17fc93ed4253c2f03b77fd2671669a1 ]

[WHY]
Prevents null pointer dereferences to enhance function robustness

[HOW]
Adds early null check and return false if invalid.

Reviewed-by: Cruise Hung <cruise.hung@amd.com>
Signed-off-by: Peichen Huang <PeiChen.Huang@amd.com>
Signed-off-by: Ray Wu <ray.wu@amd.com>
Tested-by: Daniel Wheeler <daniel.wheeler@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>
Signed-off-by: Sasha Levin <sashal@kernel.org>

diff --git a/drivers/gpu/drm/amd/display/dc/core/dc.c b/drivers/gpu/drm/amd/display/dc/core/dc.c
index 3dd7e2b6d5300c43c8d56e3c3927dcec18102dfa..943ebd5c79e3e36a34d9d2b22f99c885216fded0 100644 (file)
--- a/drivers/gpu/drm/amd/display/dc/core/dc.c
+++ b/drivers/gpu/drm/amd/display/dc/core/dc.c
@@ -6394,11 +6394,13 @@ unsigned int dc_get_det_buffer_size_from_state(const struct dc_state *context)
  */
 bool dc_get_host_router_index(const struct dc_link *link, unsigned int *host_router_index)
 {
-       struct dc *dc = link->ctx->dc;
+       struct dc *dc;
 
-       if (link->ep_type != DISPLAY_ENDPOINT_USB4_DPIA)
+       if (!link || !host_router_index || link->ep_type != DISPLAY_ENDPOINT_USB4_DPIA)
                return false;
 
+       dc = link->ctx->dc;
+
        if (link->link_index < dc->lowest_dpia_link_index)
                return false;