img_from_target_files.py revision 6701db814585431ee032a5dab26a9c9af2323c7e
1#!/usr/bin/env python 2# 3# Copyright (C) 2008 The Android Open Source Project 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16 17""" 18Given a target-files zipfile, produces an image zipfile suitable for 19use with 'fastboot update'. 20 21Usage: img_from_target_files [flags] input_target_files output_image_zip 22 23 -z (--bootable_zip) 24 Include only the bootable images (eg 'boot' and 'recovery') in 25 the output. 26 27""" 28 29import sys 30 31if sys.hexversion < 0x02070000: 32 print >> sys.stderr, "Python 2.7 or newer is required." 33 sys.exit(1) 34 35import errno 36import os 37import re 38import shutil 39import subprocess 40import tempfile 41import zipfile 42 43# missing in Python 2.4 and before 44if not hasattr(os, "SEEK_SET"): 45 os.SEEK_SET = 0 46 47import common 48 49OPTIONS = common.OPTIONS 50 51 52def CopyInfo(output_zip): 53 """Copy the android-info.txt file from the input to the output.""" 54 output_zip.write(os.path.join(OPTIONS.input_tmp, "OTA", "android-info.txt"), 55 "android-info.txt") 56 57 58def main(argv): 59 bootable_only = [False] 60 61 def option_handler(o, a): 62 if o in ("-z", "--bootable_zip"): 63 bootable_only[0] = True 64 else: 65 return False 66 return True 67 68 args = common.ParseOptions(argv, __doc__, 69 extra_opts="z", 70 extra_long_opts=["bootable_zip"], 71 extra_option_handler=option_handler) 72 73 bootable_only = bootable_only[0] 74 75 if len(args) != 2: 76 common.Usage(__doc__) 77 sys.exit(1) 78 79 OPTIONS.input_tmp, input_zip = common.UnzipTemp(args[0]) 80 output_zip = zipfile.ZipFile(args[1], "w", compression=zipfile.ZIP_DEFLATED) 81 CopyInfo(output_zip) 82 83 try: 84 done = False 85 images_path = os.path.join(OPTIONS.input_tmp, "IMAGES") 86 if os.path.exists(images_path): 87 # If this is a new target-files, it already contains the images, 88 # and all we have to do is copy them to the output zip. 89 images = os.listdir(images_path) 90 if images: 91 for image in images: 92 if bootable_only and image not in ("boot.img", "recovery.img"): 93 continue 94 if not image.endswith(".img"): 95 continue 96 common.ZipWrite( 97 output_zip, os.path.join(images_path, image), image) 98 done = True 99 100 if not done: 101 # We have an old target-files that doesn't already contain the 102 # images, so build them. 103 import add_img_to_target_files 104 105 OPTIONS.info_dict = common.LoadInfoDict(input_zip) 106 107 # If this image was originally labelled with SELinux contexts, 108 # make sure we also apply the labels in our new image. During 109 # building, the "file_contexts" is in the out/ directory tree, 110 # but for repacking from target-files.zip it's in the root 111 # directory of the ramdisk. 112 if "selinux_fc" in OPTIONS.info_dict: 113 OPTIONS.info_dict["selinux_fc"] = os.path.join( 114 OPTIONS.input_tmp, "BOOT", "RAMDISK", "file_contexts") 115 116 boot_image = common.GetBootableImage( 117 "boot.img", "boot.img", OPTIONS.input_tmp, "BOOT") 118 if boot_image: 119 boot_image.AddToZip(output_zip) 120 recovery_image = common.GetBootableImage( 121 "recovery.img", "recovery.img", OPTIONS.input_tmp, "RECOVERY") 122 if recovery_image: 123 recovery_image.AddToZip(output_zip) 124 125 def banner(s): 126 print "\n\n++++ " + s + " ++++\n\n" 127 128 if not bootable_only: 129 banner("AddSystem") 130 add_img_to_target_files.AddSystem(output_zip, prefix="") 131 try: 132 input_zip.getinfo("VENDOR/") 133 banner("AddVendor") 134 add_img_to_target_files.AddVendor(output_zip, prefix="") 135 except KeyError: 136 pass # no vendor partition for this device 137 banner("AddUserdata") 138 add_img_to_target_files.AddUserdata(output_zip, prefix="") 139 banner("AddCache") 140 add_img_to_target_files.AddCache(output_zip, prefix="") 141 142 finally: 143 print "cleaning up..." 144 # http://b/18015246 145 # See common.py for context. zipfile also refers to ZIP64_LIMIT during 146 # close() when it writes out the central directory. 147 saved_zip64_limit = zipfile.ZIP64_LIMIT 148 zipfile.ZIP64_LIMIT = (1 << 32) - 1 149 output_zip.close() 150 zipfile.ZIP64_LIMIT = saved_zip64_limit 151 shutil.rmtree(OPTIONS.input_tmp) 152 153 print "done." 154 155 156if __name__ == '__main__': 157 try: 158 common.CloseInheritedPipes() 159 main(sys.argv[1:]) 160 except common.ExternalError, e: 161 print 162 print " ERROR: %s" % (e,) 163 print 164 sys.exit(1) 165