um: virtio_uml: Fix use-after-free after put_device in probe

When register_virtio_device() fails in virtio_uml_probe(),
the code sets vu_dev->registered = 1 even though
the device was not successfully registered.
This can lead to use-after-free or other issues.

Fixes: 04e5b1fb0183 ("um: virtio: Remove device on disconnect")
Signed-off-by: Miaoqian Lin <linmq006@gmail.com>
Signed-off-by: Johannes Berg <johannes.berg@intel.com>

diff --git a/arch/um/drivers/virtio_uml.c b/arch/um/drivers/virtio_uml.c
index ad8d78fb1d9aafd135fc47bb9920f3bc77b6b29f..de7867ae220d0c59dc93b2df1d820a5f77a98603 100644 (file)
--- a/arch/um/drivers/virtio_uml.c
+++ b/arch/um/drivers/virtio_uml.c
@@ -1250,10 +1250,12 @@ static int virtio_uml_probe(struct platform_device *pdev)
        device_set_wakeup_capable(&vu_dev->vdev.dev, true);
 
        rc = register_virtio_device(&vu_dev->vdev);
-       if (rc)
+       if (rc) {
                put_device(&vu_dev->vdev.dev);
+               return rc;
+       }
        vu_dev->registered = 1;
-       return rc;
+       return 0;
 
 error_init:
        os_close_file(vu_dev->sock);