1 files changed, 8 insertions, 4 deletions
diff --git a/device.h b/device.h
index d761d2a..c11f52e 100644
--- a/device.h
+++ b/device.h
@@ -64,10 +64,11 @@ class Device : public VoldWatcher {
         // APPLY_ADB_SIDELOAD was 4.
         WIPE_DATA = 5,
         WIPE_CACHE = 6,
-        REBOOT_BOOTLOADER = 7,
-        SHUTDOWN = 8,
-        VIEW_RECOVERY_LOGS = 9,
-        MOUNT_SYSTEM = 10,
+        WIPE_MEDIA = 7,
+        REBOOT_BOOTLOADER = 8,
+        SHUTDOWN = 9,
+        VIEW_RECOVERY_LOGS = 10,
+        MOUNT_SYSTEM = 11,
     };
 
     // Return the list of menu items (an array of strings,
@@ -104,6 +105,9 @@ class Device : public VoldWatcher {
     virtual bool PreWipeData() { return true; }
     virtual bool PostWipeData() { return true; }
 
+    virtual bool PreWipeMedia() { return true; }
+    virtual bool PostWipeMedia() { return true; }
+
     // Called before reboot
     virtual char const* GetRebootReason() { return ""; }