Added enable/disable verity support (#76)

Author: MohammadAG

adb/adb_commands.py

@@ -218,6 +218,14 @@ def Root(self):
     """Restart adbd as root on the device."""
     return self.protocol_handler.Command(self.handle, service=b'root')
 
+  def EnableVerity(self):
+    """Re-enable dm-verity checking on userdebug builds"""
+    return self.protocol_handler.Command(self.handle, service=b'enable-verity')
+
+  def DisableVerity(self):
+    """Disable dm-verity checking on userdebug builds"""
+    return self.protocol_handler.Command(self.handle, service=b'disable-verity')
+
   def Shell(self, command, timeout_ms=None):
     """Run command on the device, returning the output."""
     return self.protocol_handler.Command(

adb/adb_debug.py

@@ -141,6 +141,8 @@ def main():
   common_cli.MakeSubparser(
       subparsers, parents, adb_commands.AdbCommands.Remount)
   common_cli.MakeSubparser(subparsers, parents, adb_commands.AdbCommands.Root)
+  common_cli.MakeSubparser(subparsers, parents, adb_commands.AdbCommands.EnableVerity)
+  common_cli.MakeSubparser(subparsers, parents, adb_commands.AdbCommands.DisableVerity)
   common_cli.MakeSubparser(subparsers, parents, Shell)
 
   if len(sys.argv) == 1:

test/adb_test.py

@@ -151,6 +151,15 @@ def testRoot(self):
     adb_commands = self._Connect(usb)
     adb_commands.Root()
 
+  def testEnableVerity(self):
+    usb = self._ExpectCommand(b'enable-verity', b'', b'')
+    adb_commands = self._Connect(usb)
+    adb_commands.EnableVerity()
+
+  def testDisableVerity(self):
+    usb = self._ExpectCommand(b'disable-verity', b'', b'')
+    adb_commands = self._Connect(usb)
+    adb_commands.DisableVerity()
 
 class FilesyncAdbTest(BaseAdbTest):