Switch command to subprocess in build_image.py
Bug: 36012162 Test: image builds successfully for bullhead and sailfish Change-Id: Iaa83034b39d392ff8a2154c7b32b21bf33ef552f
This commit is contained in:
parent
c87b38f547
commit
e3ad41bb2f
|
@ -25,7 +25,6 @@ import os.path
|
|||
import re
|
||||
import subprocess
|
||||
import sys
|
||||
import commands
|
||||
import common
|
||||
import shlex
|
||||
import shutil
|
||||
|
@ -52,29 +51,24 @@ def RunCommand(cmd):
|
|||
return (output, p.returncode)
|
||||
|
||||
def GetVerityFECSize(partition_size):
|
||||
cmd = "fec -s %d" % partition_size
|
||||
status, output = commands.getstatusoutput(cmd)
|
||||
if status:
|
||||
print output
|
||||
cmd = ["fec", "-s", str(partition_size)]
|
||||
output, exit_code = RunCommand(cmd)
|
||||
if exit_code != 0:
|
||||
return False, 0
|
||||
return True, int(output)
|
||||
|
||||
def GetVerityTreeSize(partition_size):
|
||||
cmd = "build_verity_tree -s %d"
|
||||
cmd %= partition_size
|
||||
status, output = commands.getstatusoutput(cmd)
|
||||
if status:
|
||||
print output
|
||||
cmd = ["build_verity_tree", "-s", str(partition_size)]
|
||||
output, exit_code = RunCommand(cmd)
|
||||
if exit_code != 0:
|
||||
return False, 0
|
||||
return True, int(output)
|
||||
|
||||
def GetVerityMetadataSize(partition_size):
|
||||
cmd = "system/extras/verity/build_verity_metadata.py size %d"
|
||||
cmd %= partition_size
|
||||
|
||||
status, output = commands.getstatusoutput(cmd)
|
||||
if status:
|
||||
print output
|
||||
cmd = ["system/extras/verity/build_verity_metadata.py", "size",
|
||||
str(partition_size)]
|
||||
output, exit_code = RunCommand(cmd)
|
||||
if exit_code != 0:
|
||||
return False, 0
|
||||
return True, int(output)
|
||||
|
||||
|
@ -191,21 +185,19 @@ AdjustPartitionSizeForVerity.results = {}
|
|||
|
||||
def BuildVerityFEC(sparse_image_path, verity_path, verity_fec_path,
|
||||
padding_size):
|
||||
cmd = "fec -e -p %d %s %s %s" % (padding_size, sparse_image_path,
|
||||
verity_path, verity_fec_path)
|
||||
print cmd
|
||||
status, output = commands.getstatusoutput(cmd)
|
||||
if status:
|
||||
cmd = ["fec", "-e", "-p", str(padding_size), sparse_image_path,
|
||||
verity_path, verity_fec_path]
|
||||
output, exit_code = RunCommand(cmd)
|
||||
if exit_code != 0:
|
||||
print "Could not build FEC data! Error: %s" % output
|
||||
return False
|
||||
return True
|
||||
|
||||
def BuildVerityTree(sparse_image_path, verity_image_path, prop_dict):
|
||||
cmd = "build_verity_tree -A %s %s %s" % (
|
||||
FIXED_SALT, sparse_image_path, verity_image_path)
|
||||
print cmd
|
||||
status, output = commands.getstatusoutput(cmd)
|
||||
if status:
|
||||
cmd = ["build_verity_tree", "-A", FIXED_SALT, sparse_image_path,
|
||||
verity_image_path]
|
||||
output, exit_code = RunCommand(cmd)
|
||||
if exit_code != 0:
|
||||
print "Could not build verity tree! Error: %s" % output
|
||||
return False
|
||||
root, salt = output.split()
|
||||
|
@ -215,16 +207,13 @@ def BuildVerityTree(sparse_image_path, verity_image_path, prop_dict):
|
|||
|
||||
def BuildVerityMetadata(image_size, verity_metadata_path, root_hash, salt,
|
||||
block_device, signer_path, key, signer_args):
|
||||
cmd_template = (
|
||||
"system/extras/verity/build_verity_metadata.py build " +
|
||||
"%s %s %s %s %s %s %s")
|
||||
cmd = cmd_template % (image_size, verity_metadata_path, root_hash, salt,
|
||||
block_device, signer_path, key)
|
||||
cmd = ["system/extras/verity/build_verity_metadata.py", "build",
|
||||
str(image_size), verity_metadata_path, root_hash, salt, block_device,
|
||||
signer_path, key]
|
||||
if signer_args:
|
||||
cmd += " --signer_args=\"%s\"" % (' '.join(signer_args),)
|
||||
print cmd
|
||||
status, output = commands.getstatusoutput(cmd)
|
||||
if status:
|
||||
cmd.append("--signer_args=\"%s\"" % (' '.join(signer_args),))
|
||||
output, exit_code = RunCommand(cmd)
|
||||
if exit_code != 0:
|
||||
print "Could not build verity metadata! Error: %s" % output
|
||||
return False
|
||||
return True
|
||||
|
@ -238,22 +227,19 @@ def Append2Simg(sparse_image_path, unsparse_image_path, error_message):
|
|||
Returns:
|
||||
True on success, False on failure.
|
||||
"""
|
||||
cmd = "append2simg %s %s"
|
||||
cmd %= (sparse_image_path, unsparse_image_path)
|
||||
print cmd
|
||||
status, output = commands.getstatusoutput(cmd)
|
||||
if status:
|
||||
cmd = ["append2simg", sparse_image_path, unsparse_image_path]
|
||||
output, exit_code = RunCommand(cmd)
|
||||
if exit_code != 0:
|
||||
print "%s: %s" % (error_message, output)
|
||||
return False
|
||||
return True
|
||||
|
||||
def Append(target, file_to_append, error_message):
|
||||
cmd = 'cat %s >> %s' % (file_to_append, target)
|
||||
print cmd
|
||||
status, output = commands.getstatusoutput(cmd)
|
||||
if status:
|
||||
print "%s: %s" % (error_message, output)
|
||||
return False
|
||||
print "appending %s to %s" % (file_to_append, target)
|
||||
with open(target, "a") as out_file:
|
||||
with open(file_to_append, "r") as input_file:
|
||||
for line in input_file:
|
||||
out_file.write(line)
|
||||
return True
|
||||
|
||||
def BuildVerifiedImage(data_image_path, verity_image_path,
|
||||
|
|
Loading…
Reference in New Issue