Introduce a function equivalent to platform_driver_probe() for of. This is
needed to keep some SoC devices in the __init section.

Signed-off-by: Wolfram Sang <[email protected]>
---
 drivers/of/platform.c       |   30 +++++++++++++++++++++++++++---
 include/linux/of_platform.h |    2 ++
 2 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/drivers/of/platform.c b/drivers/of/platform.c
index 5b4a07f..511657b 100644
--- a/drivers/of/platform.c
+++ b/drivers/of/platform.c
@@ -72,7 +72,7 @@ static void platform_driver_shutdown_shim(struct 
platform_device *pdev)
 /**
  * of_register_platform_driver
  */
-int of_register_platform_driver(struct of_platform_driver *drv)
+int __of_register_platform_driver(struct of_platform_driver *drv, bool 
do_probe)
 {
        char *of_name;
 
@@ -90,18 +90,42 @@ int of_register_platform_driver(struct of_platform_driver 
*drv)
        sprintf(of_name, "of:%s", drv->driver.name);
        drv->platform_driver.driver.name = of_name;
 
-       if (drv->probe)
-               drv->platform_driver.probe = platform_driver_probe_shim;
        drv->platform_driver.remove = drv->remove;
        if (drv->shutdown)
                drv->platform_driver.shutdown = platform_driver_shutdown_shim;
        drv->platform_driver.suspend = drv->suspend;
        drv->platform_driver.resume = drv->resume;
 
+       if (do_probe)
+               return platform_driver_probe(&drv->platform_driver,
+                                       platform_driver_probe_shim);
+
+       if (drv->probe)
+               drv->platform_driver.probe = platform_driver_probe_shim;
+
        return platform_driver_register(&drv->platform_driver);
 }
+
+int of_register_platform_driver(struct of_platform_driver *drv)
+{
+       return __of_register_platform_driver(drv, false);
+}
+
 EXPORT_SYMBOL(of_register_platform_driver);
 
+int of_probe_platform_driver(struct of_platform_driver *drv,
+               int (*probe)(struct platform_device *, const struct 
of_device_id *match))
+{
+       int ret;
+
+       drv->probe = probe;
+       ret = __of_register_platform_driver(drv, true);
+       drv->probe = NULL;
+
+       return ret;
+}
+EXPORT_SYMBOL(of_probe_platform_driver);
+
 void of_unregister_platform_driver(struct of_platform_driver *drv)
 {
        platform_driver_unregister(&drv->platform_driver);
diff --git a/include/linux/of_platform.h b/include/linux/of_platform.h
index a68716a..c4082c7 100644
--- a/include/linux/of_platform.h
+++ b/include/linux/of_platform.h
@@ -53,6 +53,8 @@ extern void of_unregister_driver(struct of_platform_driver 
*drv);
 
 /* Platform drivers register/unregister */
 extern int of_register_platform_driver(struct of_platform_driver *drv);
+extern int of_probe_platform_driver(struct of_platform_driver *drv,
+               int (*probe)(struct platform_device *, const struct 
of_device_id *match));
 extern void of_unregister_platform_driver(struct of_platform_driver *drv);
 
 extern struct platform_device *of_device_alloc(struct device_node *np,
-- 
1.7.2.3

_______________________________________________
devicetree-discuss mailing list
[email protected]
https://lists.ozlabs.org/listinfo/devicetree-discuss

Reply via email to