factory_test_tools: support write protection for ARM

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 88 matching lines @@
     VerboseMsg('MatchComponentsDatabases: Matching current system by %s' % db)
     (matched, failure) = hwcomp.match_current_system(db)
     if failure:
       DebugMsg('Unmatched for %s:%s\n' % (db, hwcomp.pformat(failure)))
     else:
       matched_list[db] = matched
       VerboseMsg('Matched: %s' % db)
   return matched_list
 
 
-def EnableWriteProtect(target):
+def EnableWriteProtect(target, image):
   """ Enables and verifies firmware write protection. """
-  return gft_wpfw.EnableWriteProtect(target)
+  return gft_wpfw.EnableWriteProtect(target, image)
 
 
 def WriteGBB(db_file):
   """ Writes a prepared GBB data to system main firwmare. """
   flashrom = flashrom_util.flashrom_util(verbose_msg=VerboseMsg,
                                          exception_type=gft_common.GFTError)
   flashrom.select_target('bios')
   image_file = gft_common.GetTemporaryFileName('wgbb')
   if not flashrom.read_whole_to_file(image_file):
     ErrorDie('WriteGBB: failed to read current firmware.')
@@ skipping 183 matching lines @@
     ErrorDie('upload_report: failed to upload.')
   return True
 
 
 @GFTLogCommand
 def wpfw():
   """ Enables and verifies firmware write protection. """
   if g_options.debug_dryrun_wpfw:
     ErrorMsg('wpfw is by-passed. This device CANNOT be qualified.')
     return True
-  if EnableWriteProtect('ec') and EnableWriteProtect('bios'):
-    return True
-  ErrorDie('wpfw: failed to enable firmware write protection.')
+  hwcomp = InitializeHardwareComponents(do_probe=False)
+  # TODO(hungte) chrome-os-partner:3023 We need a cleaner way for EC detection,
+  # for examing using crossystem. Currently we only know there is no EC on arm.
+  with os.popen("uname -m") as uname_process:
+    arch = uname_process.read().strip()
+  read_binary = gft_common.ReadBinaryFile
+  target_set = [('ec', hwcomp.load_ec_firmware),
+                ('bios', hwcomp.load_main_firmware)]
+  if arch.startswith('arm'):
+    VerboseMsg('wpfw: ARM platform detected, ignore EC.')
+    assert target_set[0][0] == 'ec'
+    target_set = target_set[1:]
+  assert len(target_set)
+  for target_name, loader_api in target_set:
+    if not EnableWriteProtect(target_name, read_binary(loader_api())):
+      ErrorDie('wpfw: failed to enable firmware write protection.')
+  return True
 
 
 @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')
@@ skipping 149 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())