2 files changed, 12 insertions, 0 deletions

diff --git a/arch/powerpc/kernel/vio.c b/arch/powerpc/kernel/vio.c
index 3aec492ec767..97082a4203ad 100644
--- a/arch/powerpc/kernel/vio.c
+++ b/arch/powerpc/kernel/vio.c
@@ -69,6 +69,16 @@ static int vio_bus_remove(struct device *dev)
         return 1;
 }
 
+/* convert from struct device to struct vio_dev and pass to driver. */
+static void vio_bus_shutdown(struct device *dev)
+{
+        struct vio_dev *viodev = to_vio_dev(dev);
+        struct vio_driver *viodrv = to_vio_driver(dev->driver);
+
+        if (viodrv->shutdown)
+                viodrv->shutdown(viodev);
+}
+
 /**
  * vio_register_driver: - Register a new vio driver
  * @drv:        The vio_driver structure to be registered.
@@ -82,6 +92,7 @@ int vio_register_driver(struct vio_driver *viodrv)
         viodrv->driver.bus = &vio_bus_type;
         viodrv->driver.probe = vio_bus_probe;
         viodrv->driver.remove = vio_bus_remove;
+        viodrv->driver.shutdown = vio_bus_shutdown;
 
         return driver_register(&viodrv->driver);
 }
diff --git a/include/asm-powerpc/vio.h b/include/asm-powerpc/vio.h
index 104c58501d0c..e0ccf108277c 100644
--- a/include/asm-powerpc/vio.h
+++ b/include/asm-powerpc/vio.h
@@ -58,6 +58,7 @@ struct vio_driver {
         const struct vio_device_id *id_table;
         int (*probe)(struct vio_dev *dev, const struct vio_device_id *id);
         int (*remove)(struct vio_dev *dev);
+        void (*shutdown)(struct vio_dev *dev);
         unsigned long driver_data;
         struct device_driver driver;
 };