Re: [PATCH] PCI: Add vf reset notification for pf

2024-02-04 Thread Leon Romanovsky
On Sun, Feb 04, 2024 at 02:12:57PM +0800, Emily Deng wrote:
> When a vf has been reset, the pf wants to get notification to remove the vf
> out of schedule.

It is very questionable if this is right thing to do. The idea of SR-IOV
is that VFs represent a physical device and they should be treated
separately from the PF.

In addition to that Keith said, this patch needs better justification.

Thanks

> 
> Solution:
> Add the callback function in pci_driver sriov_vf_reset_notification. When
> vf reset happens, then call this callback function.
> 
> Signed-off-by: Emily Deng 
> ---
>  drivers/pci/pci.c   | 8 
>  include/linux/pci.h | 1 +
>  2 files changed, 9 insertions(+)
> 
> diff --git a/drivers/pci/pci.c b/drivers/pci/pci.c
> index 60230da957e0..aca937b05531 100644
> --- a/drivers/pci/pci.c
> +++ b/drivers/pci/pci.c
> @@ -4780,6 +4780,14 @@ EXPORT_SYMBOL_GPL(pcie_flr);
>   */
>  int pcie_reset_flr(struct pci_dev *dev, bool probe)
>  {
> + struct pci_dev *pf_dev;
> +
> + if (dev->is_virtfn) {
> + pf_dev = dev->physfn;
> + if (pf_dev->driver->sriov_vf_reset_notification)
> + pf_dev->driver->sriov_vf_reset_notification(pf_dev, 
> dev);
> + }
> +
>   if (dev->dev_flags & PCI_DEV_FLAGS_NO_FLR_RESET)
>   return -ENOTTY;
>  
> diff --git a/include/linux/pci.h b/include/linux/pci.h
> index c69a2cc1f412..4fa31d9b0aa7 100644
> --- a/include/linux/pci.h
> +++ b/include/linux/pci.h
> @@ -926,6 +926,7 @@ struct pci_driver {
>   int  (*sriov_configure)(struct pci_dev *dev, int num_vfs); /* On PF */
>   int  (*sriov_set_msix_vec_count)(struct pci_dev *vf, int 
> msix_vec_count); /* On PF */
>   u32  (*sriov_get_vf_total_msix)(struct pci_dev *pf);
> + void  (*sriov_vf_reset_notification)(struct pci_dev *pf, struct pci_dev 
> *vf);
>   const struct pci_error_handlers *err_handler;
>   const struct attribute_group **groups;
>   const struct attribute_group **dev_groups;
> -- 
> 2.36.1
> 
> 


[PATCH] drm/amd/display: Implement bounds check for stream encoder creation in DCN301

2024-02-04 Thread Srinivasan Shanmugam
'stream_enc_regs' array is an array of dcn10_stream_enc_registers
structures. The array is initialized with four elements, corresponding
to the four calls to stream_enc_regs() in the array initializer. This
means that valid indices for this array are 0, 1, 2, and 3.

The error message 'stream_enc_regs' 4 <= 5 below, is indicating that
there is an attempt to access this array with an index of 5, which is
out of bounds. This could lead to undefined behavior

Here, eng_id is used as an index to access the stream_enc_regs array. If
eng_id is 5, this would result in an out-of-bounds access.

Fixes the below:
drivers/gpu/drm/amd/amdgpu/../display/dc/resource/dcn301/dcn301_resource.c:1011 
dcn301_stream_encoder_create() error: buffer overflow 'stream_enc_regs' 4 <= 5

Fixes: 3a83e4e64bb1 ("drm/amd/display: Add dcn3.01 support to DC (v2)")
Cc: Roman Li 
Cc: Rodrigo Siqueira 
Cc: Aurabindo Pillai 
Signed-off-by: Srinivasan Shanmugam 
---
 .../display/dc/resource/dcn301/dcn301_resource.c | 16 
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/drivers/gpu/drm/amd/display/dc/resource/dcn301/dcn301_resource.c 
b/drivers/gpu/drm/amd/display/dc/resource/dcn301/dcn301_resource.c
index 511ff6b5b985..f915d7c3980e 100644
--- a/drivers/gpu/drm/amd/display/dc/resource/dcn301/dcn301_resource.c
+++ b/drivers/gpu/drm/amd/display/dc/resource/dcn301/dcn301_resource.c
@@ -1006,10 +1006,18 @@ static struct stream_encoder 
*dcn301_stream_encoder_create(enum engine_id eng_id
return NULL;
}
 
-   dcn30_dio_stream_encoder_construct(enc1, ctx, ctx->dc_bios,
-   eng_id, vpg, afmt,
-   &stream_enc_regs[eng_id],
-   &se_shift, &se_mask);
+   if (eng_id < ARRAY_SIZE(stream_enc_regs)) {
+   dcn30_dio_stream_encoder_construct(enc1, ctx, ctx->dc_bios,
+  eng_id, vpg, afmt,
+  &stream_enc_regs[eng_id],
+  &se_shift, &se_mask);
+   } else {
+   DRM_ERROR("Invalid engine id: %d\n", eng_id);
+   kfree(enc1);
+   kfree(vpg);
+   kfree(afmt);
+   return NULL;
+   }
 
return &enc1->base;
 }
-- 
2.34.1