jz4740-battery: Check if platform_data is supplied

Currently platform_data is dereferenced without checking whether it
is actually set, which can lead to kernel crashes.
This patch adds a check which will abort the drivers probe function
gracefully if no platform_data is supplied.

Signed-off-by: Lars-Peter Clausen <lars@metafoo.de>
Signed-off-by: Anton Vorontsov <cbouatmailru@gmail.com>

1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/power/jz4740-battery.c b/drivers/power/jz4740-battery.c
index 2bbe04ae13a7..5b4821593201 100644
--- a/drivers/power/jz4740-battery.c
+++ b/drivers/power/jz4740-battery.c
@@ -246,6 +246,11 @@ static int __devinit jz_battery_probe(struct platform_device *pdev)
         struct jz_battery *jz_battery;
         struct power_supply *battery;
 
+        if (!pdata) {
+                dev_err(&pdev->dev, "No platform_data supplied\n");
+                return -ENXIO;
+        }
+
         jz_battery = kzalloc(sizeof(*jz_battery), GFP_KERNEL);
         if (!jz_battery) {
                 dev_err(&pdev->dev, "Failed to allocate driver structure\n");