Add more images support for system and EC firmware.

src/platform/dev/autoupdate.py

 # Copyright (c) 2009 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.
 
 from buildutil import BuildObject
 from xml.dom import minidom
 
 import os
 import shutil
 import sys
@@ skipping 176 matching lines @@
       config = [
         {
           'qual_ids': set([1, 2, 3, "x86-generic"]),
           'factory_image': 'generic-factory.gz',
           'factory_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
           'release_image': 'generic-release.gz',
           'release_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
           'oempartitionimg_image': 'generic-oem.gz',
           'oempartitionimg_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
           'stateimg_image': 'generic-state.gz',
-          'stateimg_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM='
+          'stateimg_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
+          'systemrom_image': 'generic-systemrom.gz',
+          'systemrom_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
+          'ecrom_image': 'generic-ecrom.gz',
+          'ecrom_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
         },
         {
           'qual_ids': set([6]),
           'factory_image': '6-factory.gz',
           'factory_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
           'release_image': '6-release.gz',
           'release_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
           'oempartitionimg_image': '6-oem.gz',
           'oempartitionimg_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
           'stateimg_image': '6-state.gz',
-          'stateimg_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM='
+          'stateimg_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
+          'systemrom_image': '6-systemrom.gz',
+          'systemrom_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
+          'ecrom_image': '6-ecrom.gz',
+          'ecrom_checksum': 'AtiI8B64agHVN+yeBAyiNMX3+HM=',
         },
       ]
     The server will look for the files by name in the static files
     directory.
 
     If validate_checksums is True, validates checksums and exits. If
     a checksum mismatch is found, it's printed to the screen.
     """
     f = open(filename, 'r')
     output = {}
     exec(f.read(), output)
     self.factory_config = output['config']
     success = True
     for stanza in self.factory_config:
-      for kind in ('factory', 'oempartitionimg', 'release', 'stateimg'):
-        stanza[kind + '_size'] = \
-          os.path.getsize(self.static_dir + '/' + stanza[kind + '_image'])
-        if validate_checksums:
-          factory_checksum = self.GetHash(self.static_dir + '/' +
-                                          stanza[kind + '_image'])
-          if factory_checksum != stanza[kind + '_checksum']:
-            print 'Error: checksum mismatch for %s. Expected "%s" but file ' \
-                'has checksum "%s".' % (stanza[kind + '_image'],
-                                        stanza[kind + '_checksum'],
-                                        factory_checksum)
-            success = False
+      for key in stanza.copy().iterkeys():
+        suffix = '_image'
+        if key.endswith(suffix):
+          kind = key[:-len(suffix)]
+          stanza[kind + '_size'] = \
+            os.path.getsize(self.static_dir + '/' + stanza[kind + '_image'])
+          if validate_checksums:
+            factory_checksum = self.GetHash(self.static_dir + '/' +
+                                            stanza[kind + '_image'])
+            if factory_checksum != stanza[kind + '_checksum']:
+              print 'Error: checksum mismatch for %s. Expected "%s" but file ' \
+                  'has checksum "%s".' % (stanza[kind + '_image'],
+                                          stanza[kind + '_checksum'],
+                                          factory_checksum)
+              success = False
     if validate_checksums:
       if success is False:
         raise Exception('Checksum mismatch in conf file.')
       print 'Config file looks good.'
 
   def GetFactoryImage(self, board_id, channel):
     kind = channel.rsplit('-', 1)[0]
     for stanza in self.factory_config:
       if board_id not in stanza['qual_ids']:
         continue
@@ skipping 66 matching lines @@
       ok = self.BuildUpdateImage(latest_image_path)
       if ok != True:
         web.debug('Failed to build an update image')
         return self.GetNoUpdatePayload()
 
       hash = self.GetHash('%s/update.gz' % self.static_dir)
       size = self.GetSize('%s/update.gz' % self.static_dir)
 
       url = 'http://%s/static/update.gz' % hostname
       return self.GetUpdatePayload(hash, size, url)