#include <linux/slab.h>
 #include <linux/usb.h>
 #include <linux/input/mt.h>
+#include <linux/string.h>
 
 
 MODULE_AUTHOR("Stephane Chatty <chatty@enac.fr>");
        { }
 };
 
+static void mt_free_input_name(struct hid_input *hi)
+{
+       struct hid_device *hdev = hi->report->device;
+
+       if (hi->input->name != hdev->name)
+               kfree(hi->input->name);
+}
+
 static ssize_t mt_show_quirks(struct device *dev,
                           struct device_attribute *attr,
                           char *buf)
 static void mt_pen_input_configured(struct hid_device *hdev,
                                        struct hid_input *hi)
 {
+       char *name = kzalloc(strlen(hi->input->name) + 5, GFP_KERNEL);
+       if (name) {
+               sprintf(name, "%s Pen", hi->input->name);
+               mt_free_input_name(hi);
+               hi->input->name = name;
+       }
 }
 
 static int mt_touch_input_mapping(struct hid_device *hdev, struct hid_input *hi,
 static void mt_input_configured(struct hid_device *hdev, struct hid_input *hi)
 {
        struct mt_device *td = hid_get_drvdata(hdev);
+       char *name = kstrdup(hdev->name, GFP_KERNEL);
+
+       if (name)
+               hi->input->name = name;
 
        if (hi->report->id == td->mt_report_id)
                mt_touch_input_configured(hdev, hi);
        int ret, i;
        struct mt_device *td;
        struct mt_class *mtclass = mt_classes; /* MT_CLS_DEFAULT */
+       struct hid_input *hi;
 
        for (i = 0; mt_classes[i].name ; i++) {
                if (id->driver_data == mt_classes[i].name) {
 
        ret = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
        if (ret)
-               goto fail;
+               goto hid_fail;
 
        ret = sysfs_create_group(&hdev->dev.kobj, &mt_attribute_group);
 
 
        return 0;
 
+hid_fail:
+       list_for_each_entry(hi, &hdev->inputs, list)
+               mt_free_input_name(hi);
 fail:
        kfree(td->fields);
        kfree(td);
 static void mt_remove(struct hid_device *hdev)
 {
        struct mt_device *td = hid_get_drvdata(hdev);
+       struct hid_input *hi;
+
        sysfs_remove_group(&hdev->dev.kobj, &mt_attribute_group);
        hid_hw_stop(hdev);
+
+       list_for_each_entry(hi, &hdev->inputs, list)
+               mt_free_input_name(hi);
+
        kfree(td);
        hid_set_drvdata(hdev, NULL);
 }