ubi: implement support for LED activity

Implement support for LED activity. If the feature is enabled,
make the defined ACTIVITY LED to signal ubi write operation.

Signed-off-by: Christian Marangi <ansuelsmth@gmail.com>
Reviewed-by: Simon Glass <sjg@chromium.org>

diff --git a/cmd/ubi.c b/cmd/ubi.c
index 0e62e449327523facec6081d93e724f5beca62e0..56d7da826292d4eb2f13619e18c16590455cb820 100644 (file)
--- a/cmd/ubi.c
+++ b/cmd/ubi.c
@@ -14,6 +14,7 @@
 #include <command.h>
 #include <env.h>
 #include <exports.h>
+#include <led.h>
 #include <malloc.h>
 #include <memalign.h>
 #include <mtd.h>
@@ -488,10 +489,18 @@ exit:
 
 int ubi_volume_write(char *volume, void *buf, loff_t offset, size_t size)
 {
+       int ret;
+
+       led_activity_blink();
+
        if (!offset)
-               return ubi_volume_begin_write(volume, buf, size, size);
+               ret = ubi_volume_begin_write(volume, buf, size, size);
+       else
+               ret = ubi_volume_offset_write(volume, buf, offset, size);
 
-       return ubi_volume_offset_write(volume, buf, offset, size);
+       led_activity_off();
+
+       return ret;
 }
 
 int ubi_volume_read(char *volume, char *buf, loff_t offset, size_t size)