Generate public key from update engine private key.

ctest/ctest.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.
 
 """Wrapper for tests that are run on builders."""
 
 import fileinput
 import optparse
@@ skipping 189 matching lines @@
     fh.close()
 
   version = zip_url.split('/')[-2]
   if not _GreaterVersion(version, _NEW_STYLE_VERSION) == version:
     # If the version isn't ready for new style, touch file to use old style.
     old_style_touch_path = os.path.join(download_folder, '.use_e1000')
     fh = open(old_style_touch_path, 'w+')
     fh.close()
 
 
+def GeneratePublicKey(private_key_path):
+  """Returns the path to a newly generated public key from given private key."""
+  # Just output to local directory.
+  public_key_path = 'public_key.pem'
+  cros_lib.Info('Generating public key from private key.')
+  cros_lib.RunCommand(['/usr/bin/openssl',
+                       'rsa',
+                       '-in', private_key_path,
+                       '-pubout',
+                       '-out', public_key_path,
+                      ], print_cmd=False)
+  return public_key_path
+
+
+
 def RunAUTestHarness(board, channel, zip_server_base,
                      no_graphics, type, remote, clean, test_results_root):
   """Runs the auto update test harness.
 
   The auto update test harness encapsulates testing the auto-update mechanism
   for the latest image against the latest official image from the channel.  This
   also tests images with suite_Smoke (built-in as part of its verification
   process).
 
   Args:
@@ skipping 14 matching lines @@
   # Tests go here.
   return_object = cros_lib.RunCommand(
       ['./get_latest_image.sh', '--board=%s' % board], cwd=crosutils_root,
       redirect_stdout=True, print_cmd=True)
 
   latest_image = return_object.output.strip()
 
   update_engine_path = os.path.join(crosutils_root, '..', 'platform',
                                     'update_engine')
 
+  private_key_path = os.path.join(update_engine_path, 'unittest_key.pem')
+  public_key_path = GeneratePublicKey(private_key_path)
+
   cmd = ['bin/cros_au_test_harness',
          '--base_image=%s' % os.path.join(download_folder,
                                           _IMAGE_TO_EXTRACT),
          '--target_image=%s' % os.path.join(latest_image,
                                             _IMAGE_TO_EXTRACT),
          '--board=%s' % board,
          '--type=%s' % type,
          '--remote=%s' % remote,
-         '--private_key=%s' % os.path.join(update_engine_path,
-                                           'unittest_key.pem'),
-         '--public_key=%s' % os.path.join(update_engine_path,
-                                          'unittest_key.pub.pem'),
+         '--private_key=%s' % private_key_path,
+         '--public_key=%s' % public_key_path,
          ]
   if test_results_root: cmd.append('--test_results_root=%s' % test_results_root)
   if no_graphics: cmd.append('--no_graphics')
   if clean: cmd.append('--clean')
 
   cros_lib.RunCommand(cmd, cwd=crosutils_root)
 
 
 def main():
   parser = optparse.OptionParser()
@@ skipping 25 matching lines @@
   if not options.zipbase: parser.error('Need zip url base to get images.')
 
   RunAUTestHarness(options.board, options.channel, options.zipbase,
                    options.no_graphics, options.type, options.remote,
                    not options.cache, options.test_results_root)
 
 
 if __name__ == '__main__':
   main()