gooftool: add --wipe_method to support secure/fast wiping methods

gooftool

 #!/usr/bin/python
 #
 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """
 gooftool: Google Factory Tool, providing all Google Required Test
           functionality.
 """
@@ skipping 323 matching lines @@
 @GFTLogCommand
 def prepare_wipe():
   """ Prepares system to reboot for transitioning to release state. """
   if g_options.debug_dryrun_prepare_wipe:
     ErrorMsg('prepare_wipe is by-passed. This device CANNOT be qualified.')
     return True
   wipe_script = os.path.join(os.path.split(sys.argv[0])[0],
                              'gft_prepare_wipe.sh')
   if not os.path.exists(wipe_script):
     ErrorDie('prepare_wipe: cannot find script to prepare for wiping.')
-  return_code = os.system('FACTORY_WIPE_TAGS=fast %s' % wipe_script)
+  wipe_method_map = {
+      'fast': 'fast',
+      'secure': '',
+  }
+  method = g_options.wipe_method
+  if method not in wipe_method_map:
+    ErrorDie('prepare_wipe: unknown wipe method: %s' % method)
+  tag = wipe_method_map[method]
+  command = 'FACTORY_WIPE_TAGS=%s %s' % (tag, wipe_script)
+  DebugMsg('prepare_wipe: execute command: %s' % command)
+  return_code = os.system(command)
   if return_code != 0:
     ErrorDie('prepare_wipe: failed(%s) in calling preparation script'
               % return_code)
   return True
 
 
 @GFTLogCommand
 def verify():
   """ Verifies if whole factory process is ready for finalization. """
   if (verify_switch_dev() and
@@ skipping 75 matching lines @@
                     help='upload the data generated by --create_report')
   parser.add_option('--upload_method', metavar='METHOD:PARAM',
                     help='assign the upload method (see gft_upload).')
 
   # Write Protection Wiping, and Finalization
   parser.add_option('--wpfw', action='store_true',
                     help='enable and verify firmware write protection')
   parser.add_option('--prepare_wipe', action='store_true',
                     help='prepare for wiping factory tests and '
                         'transit to release mode in next reboot')
+  parser.add_option('--wipe_method', metavar='METHOD', default='secure',
+                    help='assign the wipe method (secure/fast).')
   parser.add_option('--verify', action='store_true',
                     help='runs: --verify_switch_wp, --verify_switch_dev, '
                          '--verify_hwid, --verify_keys')
   parser.add_option('--finalize', action='store_true',
                     help='runs: --verify, --wpfw, '
                          '--create_report, --upload_report, --prepare_wipe')
   return parser
 
 
 ########################################################################
@@ skipping 41 matching lines @@
       return_value = 1
     executed_commands = executed_commands + 1
 
   if executed_commands == 0:
     parser.print_help()
   return return_value
 
 
 if __name__ == '__main__':
   sys.exit(_main())