Fix SAFT operation on Alex.

flashrom_handler.py

 #!/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.
 
 '''A module to support automated testing of ChromeOS firmware.
 
 Utilizes services provided by saft_flashrom_util.py read/write the
 flashrom chip and to parse the flash rom image.
 
@@ skipping 209 matching lines @@
 
         subsection_name = self.corrupt_section(section)
         self.fum.write_partial(self.image, (subsection_name, ))
 
     def restore_firmware(self, section):
         '''Restore the previously corrupted section in the FLASHROM!!!'''
 
         subsection_name = self.restore_section(section)
         self.fum.write_partial(self.image, (subsection_name, ))
 
+    def firmware_sections_equal(self):
+        '''Check if firmware sections A and B are equal.
+
+        This function presumes that the entire BIOS image integrity has been
+        verified, so different signature sections mean different images and
+        vice versa.
+        '''
+        sig_a = self.fum.get_section(self.image,
+                                      self.fv_sections['a'].get_sig_name())
+        sig_b = self.fum.get_section(self.image,
+                                      self.fv_sections['b'].get_sig_name())
+        return sig_a == sig_b
+
+    def copy_from_to(self, src, dst):
+        '''Copy one firmware image section to another.
+
+        This function copies both signature and body of one firmware section
+        into another. After this function runs both sections are identical.
+        '''
+        src_sect = self.fv_sections[src]
+        dst_sect = self.fv_sections[dst]
+        self.image = self.fum.put_section(
+            self.image,
+            dst_sect.get_body_name(),
+            self.fum.get_section(self.image, src_sect.get_body_name()))
+        self.image = self.fum.put_section(
+            self.image,
+            dst_sect.get_sig_name(),
+            self.fum.get_section(self.image, src_sect.get_sig_name()))
+
     def write_whole(self):
         '''Write the whole image into the flashrom.'''
 
         if not self.image:
             raise FlashromHandlerError(
                 'Attempt at using an uninitialized object')
         self.fum.write_whole(self.image)
 
     def dump_whole(self, filename):
         '''Write the whole image into a file.'''
@@ skipping 37 matching lines @@
         pad = ('%c' % 0xff) * (sig_size - os.path.getsize(sig_name))
         new_sig.write(pad)
         new_sig.close()
 
         # Inject the new signature block into the image
         new_sig = open(sig_name, 'r').read()
         self.image = self.fum.put_section(
             self.image, fv_section.get_sig_name(), new_sig)
         if write_through:
             self.fum.write_partial(self.image, (fv_section.get_sig_name(), ))