Add sha256 to devserver clone updater output

bin/cros_image_to_target.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.
 
 """Create and copy update image to target host.
 
 auto-update and devserver change out from beneath us often enough
 that despite having to duplicate a litte code, it seems that the
@@ skipping 201 matching lines @@
 
   def GetSize(self, filename):
     return os.path.getsize(filename)
 
   def GetHash(self, filename):
     return self.cmd.RunPipe([['openssl', 'sha1', '-binary'],
                              ['openssl', 'base64']],
                             infile=filename,
                             capture=True, oneline=True)
 
+  def GetSha256(self, filename):
+    return self.cmd.RunPipe([['openssl', 'dgst', '-sha256', '-binary'],
+                             ['openssl', 'base64']],
+                            infile=filename,
+                            capture=True, oneline=True)
+
   def GetDefaultBoard(self):
     def_board_file = self.CrosUtilsPath('.default_board')
     if not os.path.exists(def_board_file):
       return None
     return self.FileOneLine(def_board_file)
 
   def SetRemote(self, remote):
     self.ssh_cmd = SSHCommand(self, remote)
 
   def ParseShVars(self, string):
@@ skipping 11 matching lines @@
   def GetRemoteRelease(self):
     lsb_release = self.ssh_cmd.Output('cat', '/etc/lsb-release')
     if not lsb_release:
       return None
     return self.ParseShVars(lsb_release)
 
   def CreateServer(self, port, update_file, stateful_file):
     """Start the devserver clone."""
 
     PingUpdateResponse.Setup(self.GetHash(update_file),
+                             self.GetSha256(update_file),
                              self.GetSize(update_file))
 
     UpdateHandler.SetupUrl('/update', PingUpdateResponse())
     UpdateHandler.SetupUrl('/%s' % UPDATE_FILENAME,
                            FileUpdateResponse(update_file,
                                               verbose=self.verbose))
     UpdateHandler.SetupUrl('/%s' % STATEFUL_FILENAME,
                            FileUpdateResponse(stateful_file,
                                               verbose=self.verbose))
 
@@ skipping 153 matching lines @@
   app_id = '87efface-864d-49a5-9bb3-4b050a7c227a'
   xmlns = 'http://www.google.com/update2/response'
   payload_success_template = """<?xml version="1.0" encoding="UTF-8"?>
     <gupdate xmlns="%s" protocol="2.0">
     <daystart elapsed_seconds="%s"/>
     <app appid="{%s}" status="ok">
       <ping status="ok"/>
       <updatecheck
       codebase="%s"
       hash="%s"
+      sha256="%s"
       needsadmin="false"
       size="%s"
       status="ok"/>
     </app>
     </gupdate>
   """
   payload_failure_template = """<?xml version="1.0" encoding="UTF-8"?>
       <gupdate xmlns="%s" protocol="2.0">
         <daystart elapsed_seconds="%s"/>
         <app appid="{%s}" status="ok">
           <ping status="ok"/>
           <updatecheck status="noupdate"/>
         </app>
       </gupdate>
     """
 
   def __init__(self):
     self.content_type = 'text/xml'
 
   @staticmethod
-  def Setup(filehash, filesize):
+  def Setup(filehash, filesha256, filesize):
     PingUpdateResponse.file_hash = filehash
+    PingUpdateResponse.file_sha256 = filesha256
     PingUpdateResponse.file_size = filesize
 
   def Reply(self, handler, send_content=True, post_data=None):
     """Return (using StringResponse) an XML reply to ForcedUpdate clients."""
 
     if not post_data:
       return UpdateResponse.Reply(self, handler)
 
     request_version = (minidom.parseString(post_data).firstChild.
                        getElementsByTagName('o:app')[0].
                        getAttribute('version'))
 
     if request_version == 'ForcedUpdate':
       host, pdict = cgi.parse_header(handler.headers.getheader('Host'))
       self.string = (self.payload_success_template %
                      (self.xmlns, self.SecondsSinceMidnight(),
                       self.app_id, 'http://%s/%s' % (host, UPDATE_FILENAME),
-                      self.file_hash, self.file_size))
+                      self.file_hash, self.file_sha256, self.file_size))
     else:
       self.string = (self.payload_failure_template %
                      (self.xmlns, self.SecondsSinceMidnight(), self.app_id))
 
     StringUpdateResponse.Reply(self, handler, send_content)
 
   def SecondsSinceMidnight(self):
     now = time.localtime()
     return now[3] * 3600 + now[4] * 60 + now[5]
 
@@ skipping 172 matching lines @@
 
   if child:
     os.kill(child, 15)
 
   cros_env.Info('Server exiting with status %d' % exit_status)
   sys.exit(exit_status)
 
 
 if __name__ == '__main__':
   main(sys.argv)