2 files changed, 29 insertions, 7 deletions

diff --git a/drivers/video/fbsysfs.c b/drivers/video/fbsysfs.c
index c151dcf6878..d3a50417ed9 100644
--- a/drivers/video/fbsysfs.c
+++ b/drivers/video/fbsysfs.c
@@ -20,6 +20,8 @@
 #include <linux/console.h>
 #include <linux/module.h>
 
+#define FB_SYSFS_FLAG_ATTR 1
+
 /**
  * framebuffer_alloc - creates a new frame buffer info structure
  *
@@ -483,12 +485,27 @@ static struct class_device_attribute class_device_attrs[] = {
 
 int fb_init_class_device(struct fb_info *fb_info)
 {
-        unsigned int i;
+        int i, error = 0;
+
         class_set_devdata(fb_info->class_device, fb_info);
 
-        for (i = 0; i < ARRAY_SIZE(class_device_attrs); i++)
-                class_device_create_file(fb_info->class_device,
-                                         &class_device_attrs[i]);
+        fb_info->class_flag |= FB_SYSFS_FLAG_ATTR;
+
+        for (i = 0; i < ARRAY_SIZE(class_device_attrs); i++) {
+                error = class_device_create_file(fb_info->class_device,
+                                                 &class_device_attrs[i]);
+
+                if (error)
+                        break;
+        }
+
+        if (error) {
+                while (--i >= 0)
+                        class_device_remove_file(fb_info->class_device,
+                                                 &class_device_attrs[i]);
+                fb_info->class_flag &= ~FB_SYSFS_FLAG_ATTR;
+        }
+
         return 0;
 }
 
@@ -496,9 +513,13 @@ void fb_cleanup_class_device(struct fb_info *fb_info)
 {
         unsigned int i;
 
-        for (i = 0; i < ARRAY_SIZE(class_device_attrs); i++)
-                class_device_remove_file(fb_info->class_device,
-                                         &class_device_attrs[i]);
+        if (fb_info->class_flag & FB_SYSFS_FLAG_ATTR) {
+                for (i = 0; i < ARRAY_SIZE(class_device_attrs); i++)
+                        class_device_remove_file(fb_info->class_device,
+                                                 &class_device_attrs[i]);
+
+                fb_info->class_flag &= ~FB_SYSFS_FLAG_ATTR;
+        }
 }
 
 #ifdef CONFIG_FB_BACKLIGHT
diff --git a/include/linux/fb.h b/include/linux/fb.h
index 37aab314f0c..3e69241e6a8 100644
--- a/include/linux/fb.h
+++ b/include/linux/fb.h
@@ -776,6 +776,7 @@ struct fb_info {
         struct fb_ops *fbops;
         struct device *device;
         struct class_device *class_device; /* sysfs per device attrs */
+        int class_flag;                    /* private sysfs flags */
 #ifdef CONFIG_FB_TILEBLITTING
         struct fb_tile_ops *tileops;    /* Tile Blitting */
 #endif