Check return value and conduct null pointer handling to avoid null pointer 
dereference.

Signed-off-by: Bob Zhou <[email protected]>
---
 .../drm/amd/pm/powerplay/hwmgr/vega10_hwmgr.c | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/drivers/gpu/drm/amd/pm/powerplay/hwmgr/vega10_hwmgr.c 
b/drivers/gpu/drm/amd/pm/powerplay/hwmgr/vega10_hwmgr.c
index 6524d99e5cab..68b93a0c16e4 100644
--- a/drivers/gpu/drm/amd/pm/powerplay/hwmgr/vega10_hwmgr.c
+++ b/drivers/gpu/drm/amd/pm/powerplay/hwmgr/vega10_hwmgr.c
@@ -3438,6 +3438,8 @@ static int 
vega10_find_dpm_states_clocks_in_dpm_table(struct pp_hwmgr *hwmgr, co
                        (const struct phm_set_power_state_input *)input;
        const struct vega10_power_state *vega10_ps =
                        cast_const_phw_vega10_power_state(states->pnew_state);
+       if (vega10_ps == NULL)
+               return -EINVAL;
        struct vega10_single_dpm_table *sclk_table = 
&(data->dpm_table.gfx_table);
        uint32_t sclk = vega10_ps->performance_levels
                        [vega10_ps->performance_level_count - 1].gfx_clock;
@@ -3752,6 +3754,9 @@ static int vega10_generate_dpm_level_enable_mask(
                        cast_const_phw_vega10_power_state(states->pnew_state);
        int i;
 
+       if (vega10_ps == NULL)
+               return -EINVAL;
+
        PP_ASSERT_WITH_CODE(!vega10_trim_dpm_states(hwmgr, vega10_ps),
                        "Attempt to Trim DPM States Failed!",
                        return -1);
@@ -5035,6 +5040,8 @@ static int vega10_check_states_equal(struct pp_hwmgr 
*hwmgr,
 
        vega10_psa = cast_const_phw_vega10_power_state(pstate1);
        vega10_psb = cast_const_phw_vega10_power_state(pstate2);
+       if (vega10_psa == NULL || vega10_psb == NULL)
+               return -EINVAL;
 
        /* If the two states don't even have the same number of performance 
levels
         * they cannot be the same state.
@@ -5168,6 +5175,8 @@ static int vega10_set_sclk_od(struct pp_hwmgr *hwmgr, 
uint32_t value)
                return -EINVAL;
 
        vega10_ps = cast_phw_vega10_power_state(&ps->hardware);
+       if (vega10_ps == NULL)
+               return -EINVAL;
 
        vega10_ps->performance_levels
        [vega10_ps->performance_level_count - 1].gfx_clock =
@@ -5219,6 +5228,8 @@ static int vega10_set_mclk_od(struct pp_hwmgr *hwmgr, 
uint32_t value)
                return -EINVAL;
 
        vega10_ps = cast_phw_vega10_power_state(&ps->hardware);
+       if (vega10_ps == NULL)
+               return -EINVAL;
 
        vega10_ps->performance_levels
        [vega10_ps->performance_level_count - 1].mem_clock =
@@ -5460,6 +5471,9 @@ static void vega10_odn_update_power_state(struct pp_hwmgr 
*hwmgr)
                return;
 
        vega10_ps = cast_phw_vega10_power_state(&ps->hardware);
+       if (vega10_ps == NULL)
+               return;
+
        max_level = vega10_ps->performance_level_count - 1;
 
        if (vega10_ps->performance_levels[max_level].gfx_clock !=
@@ -5482,6 +5496,9 @@ static void vega10_odn_update_power_state(struct pp_hwmgr 
*hwmgr)
 
        ps = (struct pp_power_state *)((unsigned long)(hwmgr->ps) + 
hwmgr->ps_size * (hwmgr->num_ps - 1));
        vega10_ps = cast_phw_vega10_power_state(&ps->hardware);
+       if (vega10_ps == NULL)
+               return;
+
        max_level = vega10_ps->performance_level_count - 1;
 
        if (vega10_ps->performance_levels[max_level].gfx_clock !=
@@ -5672,6 +5689,8 @@ static int vega10_get_performance_level(struct pp_hwmgr 
*hwmgr, const struct pp_
                return -EINVAL;
 
        vega10_ps = cast_const_phw_vega10_power_state(state);
+       if (vega10_ps == NULL)
+               return -EINVAL;
 
        i = index > vega10_ps->performance_level_count - 1 ?
                        vega10_ps->performance_level_count - 1 : index;
-- 
2.34.1

Reply via email to