powerpc/pmac: Fix dangling pointers

[net-next-2.6.git] / drivers / macintosh / therm_adt746x.c

diff --git a/drivers/macintosh/therm_adt746x.c b/drivers/macintosh/therm_adt746x.c
index c42eeb43042daa92d9f04e22b2205ce6d1d2a33d..16d82f17ae82b3a2e0f5a5e37a0a73e18e51e7b5 100644 (file)
--- a/drivers/macintosh/therm_adt746x.c
+++ b/drivers/macintosh/therm_adt746x.c
@@ -182,6 +182,7 @@ remove_thermostat(struct i2c_client *client)
 
        thermostat = NULL;
 
+       i2c_set_clientdata(client, NULL);
        kfree(th);
 
        return 0;
@@ -399,6 +400,7 @@ static int probe_thermostat(struct i2c_client *client,
        rc = read_reg(th, CONFIG_REG);
        if (rc < 0) {
                dev_err(&client->dev, "Thermostat failed to read config!\n");
+               i2c_set_clientdata(client, NULL);
                kfree(th);
                return -ENODEV;
        }