Re: [PATCH] PCI/AER: increments pci bus reference count in aer-inject process

[Xiongfeng Wang]

Kindly Ping ...

Thanks,
Xiongfeng

On 2020/1/11 18:34, Xiongfeng Wang wrote:
> When I test 'aer-inject' with the following procedures:
> 1. inject a fatal error into a PCI device
> 2. remove the parent device by sysfs
> 3. execute command 'rmmod aer-inject'
> 
> I came across the following use-after-free.
> 
> [  297.581524] ==================================================================
> [  297.581543] BUG: KASAN: use-after-free in pci_bus_set_ops+0xb4/0xb8
> [  297.581545] Read of size 8 at addr ffff802edbde80e0 by task rmmod/21839
> 
> [  297.581552] CPU: 119 PID: 21839 Comm: rmmod Kdump: loaded Not tainted 4.19.36 #1
> [  297.581554] Hardware name: Huawei TaiShan 2280 V2/BC82AMDD, BIOS 1.05 09/18/2019
> [  297.581556] Call trace:
> [  297.581561]  dump_backtrace+0x0/0x360
> [  297.581563]  show_stack+0x24/0x30
> [  297.581569]  dump_stack+0xd8/0x104
> [  297.581576]  print_address_description+0x68/0x278
> [  297.581578]  kasan_report+0x204/0x330
> [  297.581580]  __asan_report_load8_noabort+0x30/0x40
> [  297.581582]  pci_bus_set_ops+0xb4/0xb8
> [  297.581591]  aer_inject_exit+0x198/0x334 [aer_inject]
> [  297.581595]  __arm64_sys_delete_module+0x310/0x490
> [  297.581601]  el0_svc_common+0xfc/0x278
> [  297.581603]  el0_svc_handler+0x50/0xc0
> [  297.581605]  el0_svc+0x8/0xc
> 
> [  297.581608] Allocated by task 1:
> [  297.581611]  kasan_kmalloc+0xe0/0x190
> [  297.581614]  kmem_cache_alloc_trace+0x104/0x218
> [  297.581616]  pci_alloc_bus+0x50/0x2e0
> [  297.581618]  pci_add_new_bus+0xa8/0xe08
> [  297.581620]  pci_scan_bridge_extend+0x884/0xb28
> [  297.581623]  pci_scan_child_bus_extend+0x350/0x628
> [  297.581625]  pci_scan_child_bus+0x24/0x30
> [  297.581627]  pci_scan_bridge_extend+0x3b8/0xb28
> [  297.581629]  pci_scan_child_bus_extend+0x350/0x628
> [  297.581631]  pci_scan_child_bus+0x24/0x30
> [  297.581635]  acpi_pci_root_create+0x558/0x888
> [  297.581640]  pci_acpi_scan_root+0x198/0x330
> [  297.581641]  acpi_pci_root_add+0x7bc/0xbb0
> [  297.581646]  acpi_bus_attach+0x2f4/0x728
> [  297.581647]  acpi_bus_attach+0x1b0/0x728
> [  297.581649]  acpi_bus_attach+0x1b0/0x728
> [  297.581651]  acpi_bus_scan+0xa0/0x110
> [  297.581657]  acpi_scan_init+0x20c/0x500
> [  297.581659]  acpi_init+0x54c/0x5d4
> [  297.581661]  do_one_initcall+0xbc/0x480
> [  297.581665]  kernel_init_freeable+0x5fc/0x6ac
> [  297.581670]  kernel_init+0x18/0x128
> [  297.581671]  ret_from_fork+0x10/0x18
> 
> [  297.581673] Freed by task 19270:
> [  297.581675]  __kasan_slab_free+0x120/0x228
> [  297.581677]  kasan_slab_free+0x10/0x18
> [  297.581678]  kfree+0x80/0x1f8
> [  297.581680]  release_pcibus_dev+0x54/0x68
> [  297.581686]  device_release+0xd4/0x1c0
> [  297.581689]  kobject_put+0x12c/0x400
> [  297.581691]  device_unregister+0x30/0xc0
> [  297.581693]  pci_remove_bus+0xe8/0x1c0
> [  297.581695]  pci_remove_bus_device+0xd0/0x2f0
> [  297.581697]  pci_stop_and_remove_bus_device_locked+0x2c/0x40
> [  297.581701]  remove_store+0x1b8/0x1d0
> [  297.581703]  dev_attr_store+0x60/0x80
> [  297.581708]  sysfs_kf_write+0x104/0x170
> [  297.581710]  kernfs_fop_write+0x23c/0x430
> [  297.581713]  __vfs_write+0xec/0x4e0
> [  297.581714]  vfs_write+0x12c/0x3d0
> [  297.581715]  ksys_write+0xd0/0x190
> [  297.581716]  __arm64_sys_write+0x70/0xa0
> [  297.581718]  el0_svc_common+0xfc/0x278
> [  297.581720]  el0_svc_handler+0x50/0xc0
> [  297.581721]  el0_svc+0x8/0xc
> 
> [  297.581724] The buggy address belongs to the object at ffff802edbde8000
>                 which belongs to the cache kmalloc-2048 of size 2048
> [  297.581726] The buggy address is located 224 bytes inside of
>                 2048-byte region [ffff802edbde8000, ffff802edbde8800)
> [  297.581727] The buggy address belongs to the page:
> [  297.581730] page:ffff7e00bb6f7a00 count:1 mapcount:0 mapping:ffff8026de810780 index:0x0 compound_mapcount: 0
> [  297.591520] flags: 0x2ffffe0000008100(slab|head)
> [  297.596121] raw: 2ffffe0000008100 ffff7e00bb6f5008 ffff7e00bb6ff608 ffff8026de810780
> [  297.596123] raw: 0000000000000000 00000000000f000f 00000001ffffffff 0000000000000000
> [  297.596124] page dumped because: kasan: bad access detected
> 
> [  297.596126] Memory state around the buggy address:
> [  297.596128]  ffff802edbde7f80: fc fc fc fc fc fc fc fc fc fc fc fc fc fc fc fc
> [  297.596129]  ffff802edbde8000: fb fb fb fb fb fb fb fb fb fb fb fb fb fb fb fb
> [  297.596131] >ffff802edbde8080: fb fb fb fb fb fb fb fb fb fb fb fb fb fb fb fb
> [  297.596132]                                                        ^
> [  297.596133]  ffff802edbde8100: fb fb fb fb fb fb fb fb fb fb fb fb fb fb fb fb
> [  297.596135]  ffff802edbde8180: fb fb fb fb fb fb fb fb fb fb fb fb fb fb fb fb
> [  297.596135] ==================================================================
> 
> It is because when we unload the module and restore the member 'pci_ops'
> of 'pci_bus', the 'pci_bus' has been freed. This patch increments the
> reference count of 'pci_bus' when we modify its member 'pci_ops' and
> decrements the reference count after we have restored its member.
> 
> This patch export pci_bus_get() and pci_bus_put() again and reverts the
> following two commits.
> fae6b93b19b4 ("PCI: Unexport pci_bus_get() and pci_bus_put()")
> ecd29c1a38af ("PCI: Make pci_bus_get(), pci_bus_put() private")
> 
> Signed-off-by: Xiongfeng Wang <wangxiongfeng2@xxxxxxxxxx>
> ---
>  drivers/pci/bus.c             | 2 ++
>  drivers/pci/pci.h             | 2 --
>  drivers/pci/pcie/aer_inject.c | 8 ++++++++
>  include/linux/pci.h           | 2 ++
>  4 files changed, 12 insertions(+), 2 deletions(-)
> 
> diff --git a/drivers/pci/bus.c b/drivers/pci/bus.c
> index 8e40b3e..495059d 100644
> --- a/drivers/pci/bus.c
> +++ b/drivers/pci/bus.c
> @@ -417,9 +417,11 @@ struct pci_bus *pci_bus_get(struct pci_bus *bus)
>  		get_device(&bus->dev);
>  	return bus;
>  }
> +EXPORT_SYMBOL(pci_bus_get);
>  
>  void pci_bus_put(struct pci_bus *bus)
>  {
>  	if (bus)
>  		put_device(&bus->dev);
>  }
> +EXPORT_SYMBOL(pci_bus_put);
> diff --git a/drivers/pci/pci.h b/drivers/pci/pci.h
> index a0a53bd..cf849b7 100644
> --- a/drivers/pci/pci.h
> +++ b/drivers/pci/pci.h
> @@ -286,8 +286,6 @@ void __pci_bus_assign_resources(const struct pci_bus *bus,
>  
>  void pci_reassigndev_resource_alignment(struct pci_dev *dev);
>  void pci_disable_bridge_window(struct pci_dev *dev);
> -struct pci_bus *pci_bus_get(struct pci_bus *bus);
> -void pci_bus_put(struct pci_bus *bus);
>  
>  /* PCIe link information */
>  #define PCIE_SPEED2STR(speed) \
> diff --git a/drivers/pci/pcie/aer_inject.c b/drivers/pci/pcie/aer_inject.c
> index 6988fe7..0b6b3d0 100644
> --- a/drivers/pci/pcie/aer_inject.c
> +++ b/drivers/pci/pcie/aer_inject.c
> @@ -307,6 +307,13 @@ static int pci_bus_set_aer_ops(struct pci_bus *bus)
>  	spin_lock_irqsave(&inject_lock, flags);
>  	if (ops == &aer_inj_pci_ops)
>  		goto out;
> +	/*
> +	 * increments the reference count of the pci bus. Otherwise, when we
> +	 * restore the 'pci_ops' in 'aer_inject_exit', the 'pci_bus' may have
> +	 * been freed.
> +	 */
> +	pci_bus_get(bus);
> +
>  	pci_bus_ops_init(bus_ops, bus, ops);
>  	list_add(&bus_ops->list, &pci_bus_ops_list);
>  	bus_ops = NULL;
> @@ -529,6 +536,7 @@ static void __exit aer_inject_exit(void)
>  
>  	while ((bus_ops = pci_bus_ops_pop())) {
>  		pci_bus_set_ops(bus_ops->bus, bus_ops->ops);
> +		pci_bus_put(bus_ops->bus);
>  		kfree(bus_ops);
>  	}
>  
> diff --git a/include/linux/pci.h b/include/linux/pci.h
> index c393dff..adef0da 100644
> --- a/include/linux/pci.h
> +++ b/include/linux/pci.h
> @@ -1288,6 +1288,8 @@ int pci_add_ext_cap_save_buffer(struct pci_dev *dev,
>  void pci_release_selected_regions(struct pci_dev *, int);
>  
>  /* drivers/pci/bus.c */
> +struct pci_bus *pci_bus_get(struct pci_bus *bus);
> +void pci_bus_put(struct pci_bus *bus);
>  void pci_add_resource(struct list_head *resources, struct resource *res);
>  void pci_add_resource_offset(struct list_head *resources, struct resource *res,
>  			     resource_size_t offset);
>