#!/usr/bin/python
-
+#
+# $Id$
+# $URL$
+#
# Copyright (c) 2003 Intel Corporation
# All rights reserved.
#
-def sysexec( cmd, log= None ):
+def sysexec( cmd, log= None, fsck = False ):
"""
execute a system command, output the results to the logger
if log <> None
"""
if VERBOSE_MODE:
print ("sysexec >>> %s" % cmd)
- prog= popen2.Popen4( cmd, 0 )
+
+ prog = popen2.Popen4( cmd, 0 )
if prog is None:
raise BootManagerException, \
"Unable to create instance of popen2.Popen4 " \
except KeyboardInterrupt:
raise BootManagerException, "Interrupted by user"
- returncode= prog.wait()
- if returncode != 0:
- raise BootManagerException, "Running %s failed (rc=%d)" % (cmd,returncode)
-
- prog= None
+ returncode = prog.wait()
+
+ if fsck:
+ # The exit code returned by fsck is the sum of the following conditions:
+ # 0 - No errors
+ # 1 - File system errors corrected
+ # 2 - System should be rebooted
+ # 4 - File system errors left uncorrected
+ # 8 - Operational error
+ # 16 - Usage or syntax error
+ # 32 - Fsck canceled by user request
+ # 128 - Shared library error
+ if returncode != 0 and returncode != 1:
+ raise BootManagerException, "Running %s failed (rc=%d)" % (cmd,returncode)
+ else:
+ if returncode != 0:
+ raise BootManagerException, "Running %s failed (rc=%d)" % (cmd,returncode)
+ prog = None
return 1
return ret
+def check_file_hash(filename, hash_filename):
+ """Check the file's integrity with a given hash."""
+ return sha1_file(filename) == open(hash_filename).read().split()[0].strip()
+
+def sha1_file(filename):
+ """Calculate sha1 hash of file."""
+ try:
+ try:
+ import hashlib
+ m = hashlib.sha1()
+ except:
+ import sha
+ m=sha.new()
+ f = file(filename, 'rb')
+ while True:
+ # 256 KB seems ideal for speed/memory tradeoff
+ # It wont get much faster with bigger blocks, but
+ # heap peak grows
+ block = f.read(256 * 1024)
+ if len(block) == 0:
+ # end of file
+ break
+ m.update(block)
+ # Simple trick to keep total heap even lower
+ # Delete the previous block, so while next one is read
+ # we wont have two allocated blocks with same size
+ del block
+ return m.hexdigest()
+ except IOError:
+ raise BootManagerException, "Cannot calculate SHA1 hash of %s" % filename