1 files changed, 22 insertions, 0 deletions

diff --git a/bootloader/x1000/utils.c b/bootloader/x1000/utils.c
index 629a2c6280..4d3fb69e57 100644
--- a/bootloader/x1000/utils.c
+++ b/bootloader/x1000/utils.c
@@ -29,6 +29,7 @@
 #include "rb-loader.h"
 #include "loader_strerror.h"
 #include "linuxboot.h"
+#include "screendump.h"
 #include "nand-x1000.h"
 
 /* Set to true if a SYS_USB_CONNECTED event is seen
@@ -37,6 +38,8 @@
  * TODO: this is an ugly kludge */
 bool is_usb_connected = false;
 
+static bool screenshot_enabled = false;
+
 /* this is both incorrect and incredibly racy... */
 int check_disk(bool wait)
 {
@@ -75,6 +78,25 @@ void usb_mode(void)
     splashf(3*HZ, "USB disconnected");
 }
 
+void screenshot(void)
+{
+#ifdef HAVE_SCREENDUMP
+    if(!screenshot_enabled || check_disk(false) != DISK_PRESENT)
+        return;
+
+    screen_dump();
+#endif
+}
+
+void screenshot_enable(void)
+{
+#ifdef HAVE_SCREENDUMP
+    splashf(3*HZ, "Screenshots enabled\nPress %s for screenshot",
+            BL_SCREENSHOT_NAME);
+    screenshot_enabled = true;
+#endif
+}
+
 int load_rockbox(const char* filename, size_t* sizep)
 {
     if(check_disk(true) != DISK_PRESENT)