#include <linux/mutex.h>
 #include <linux/device.h>
 #include <linux/jiffies.h>
+#include <linux/thermal.h>
+#include <linux/of.h>
 
 #define        DRIVER_NAME "tmp102"
 
 
 struct tmp102 {
        struct device *hwmon_dev;
+       struct thermal_zone_device *tz;
        struct mutex lock;
        u16 config_orig;
        unsigned long last_update;
        return tmp102;
 }
 
+static int tmp102_read_temp(void *dev, long *temp)
+{
+       struct tmp102 *tmp102 = tmp102_update_device(to_i2c_client(dev));
+
+       *temp = tmp102->temp[0];
+
+       return 0;
+}
+
 static ssize_t tmp102_show_temp(struct device *dev,
                                struct device_attribute *attr,
                                char *buf)
                goto fail_remove_sysfs;
        }
 
+       tmp102->tz = thermal_zone_of_sensor_register(&client->dev, 0,
+                                                    &client->dev,
+                                                    tmp102_read_temp, NULL);
+       if (IS_ERR(tmp102->tz))
+               tmp102->tz = NULL;
+
        dev_info(&client->dev, "initialized\n");
 
        return 0;
 {
        struct tmp102 *tmp102 = i2c_get_clientdata(client);
 
+       thermal_zone_of_sensor_unregister(&client->dev, tmp102->tz);
        hwmon_device_unregister(tmp102->hwmon_dev);
        sysfs_remove_group(&client->dev.kobj, &tmp102_attr_group);