diff options
Diffstat (limited to 'releasetools/galaxys2_img_from_target_files')
| -rwxr-xr-x | releasetools/galaxys2_img_from_target_files | 186 |
1 files changed, 26 insertions, 160 deletions
diff --git a/releasetools/galaxys2_img_from_target_files b/releasetools/galaxys2_img_from_target_files index 240de1b..e929af6 100755 --- a/releasetools/galaxys2_img_from_target_files +++ b/releasetools/galaxys2_img_from_target_files @@ -14,172 +14,38 @@ # See the License for the specific language governing permissions and # limitations under the License. -""" -Given a target-files zipfile, produces an image zipfile suitable for -use with 'fastboot update'. +import os, sys, imp -Usage: img_from_target_files [flags] input_target_files output_image_zip +LOCAL_DIR = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')) +RELEASETOOLS_DIR = os.path.abspath(os.path.join(LOCAL_DIR, '../../../build/tools/releasetools')) +TARGET_DIR = os.getenv('OUT') - -b (--board_config) <file> - Deprecated. +# Add releasetools directory to python path +sys.path.append(RELEASETOOLS_DIR) -""" +# Import the existing file so we just have to rewrite the modules we need. +# This is a nasty hack as the filename doesn't end in .py, but it works +filename = os.path.join(RELEASETOOLS_DIR, 'img_from_target_files') +f = open(filename, 'rU') +img_from_target_files = imp.load_module('img_from_target_files', f, filename, ('', 'U', 1)) +f.close() -import sys +from img_from_target_files import * -if sys.hexversion < 0x02040000: - print >> sys.stderr, "Python 2.4 or newer is required." - sys.exit(1) +__doc__ = img_from_target_files.__doc__ -import errno -import os -import re -import shutil -import subprocess -import tempfile -import zipfile - -# missing in Python 2.4 and before -if not hasattr(os, "SEEK_SET"): - os.SEEK_SET = 0 - -import galaxys2_common as common - -OPTIONS = common.OPTIONS - -def AddUserdata(output_zip): - """Create an empty userdata image and store it in output_zip.""" - - print "creating userdata.img..." - - # The name of the directory it is making an image out of matters to - # mkyaffs2image. So we create a temp dir, and within it we create an - # empty dir named "data", and build the image from that. - temp_dir = tempfile.mkdtemp() - user_dir = os.path.join(temp_dir, "data") - os.mkdir(user_dir) - img = tempfile.NamedTemporaryFile() - - build_command = [] - if OPTIONS.info_dict["fstab"]["/data"].fs_type.startswith("ext"): - build_command = ["mkuserimg.sh", - user_dir, img.name, - OPTIONS.info_dict["fstab"]["/data"].fs_type, "data"] - if "userdata_size" in OPTIONS.info_dict: - build_command.append(str(OPTIONS.info_dict["userdata_size"])) - else: - build_command = ["mkyaffs2image", "-f"] - extra = OPTIONS.info_dict.get("mkyaffs2_extra_flags", None) - if extra: - build_command.extend(extra.split()) - build_command.append(user_dir) - build_command.append(img.name) - - p = common.Run(build_command) - p.communicate() - assert p.returncode == 0, "build userdata.img image failed" - - common.CheckSize(img.name, "userdata.img", OPTIONS.info_dict) - output_zip.write(img.name, "userdata.img") - img.close() - os.rmdir(user_dir) - os.rmdir(temp_dir) - - -def AddSystem(output_zip): - """Turn the contents of SYSTEM into a system image and store it in - output_zip.""" - - print "creating system.img..." - - img = tempfile.NamedTemporaryFile() - - # The name of the directory it is making an image out of matters to - # mkyaffs2image. It wants "system" but we have a directory named - # "SYSTEM", so create a symlink. - try: - os.symlink(os.path.join(OPTIONS.input_tmp, "SYSTEM"), - os.path.join(OPTIONS.input_tmp, "system")) - except OSError, e: - if (e.errno == errno.EEXIST): - pass - - build_command = [] - if OPTIONS.info_dict["fstab"]["/system"].fs_type.startswith("ext"): - build_command = ["mkuserimg.sh", - os.path.join(OPTIONS.input_tmp, "system"), img.name, - OPTIONS.info_dict["fstab"]["/system"].fs_type, "system"] - if "system_size" in OPTIONS.info_dict: - build_command.append(str(OPTIONS.info_dict["system_size"])) - else: - build_command = ["mkyaffs2image", "-f"] - extra = OPTIONS.info_dict.get("mkyaffs2_extra_flags", None) - if extra: - build_command.extend(extra.split()) - build_command.append(os.path.join(OPTIONS.input_tmp, "system")) - build_command.append(img.name) - - p = common.Run(build_command) - p.communicate() - assert p.returncode == 0, "build system.img image failed" - - img.seek(os.SEEK_SET, 0) - data = img.read() - img.close() - - common.CheckSize(data, "system.img", OPTIONS.info_dict) - common.ZipWriteStr(output_zip, "system.img", data) - - -def CopyInfo(output_zip): - """Copy the android-info.txt file from the input to the output.""" - output_zip.write(os.path.join(OPTIONS.input_tmp, "OTA", "android-info.txt"), - "android-info.txt") - - -def main(argv): - - def option_handler(o, a): - if o in ("-b", "--board_config"): - pass # deprecated - else: - return False - return True - - args = common.ParseOptions(argv, __doc__, - extra_opts="b:", - extra_long_opts=["board_config="], - extra_option_handler=option_handler) - - if len(args) != 2: - common.Usage(__doc__) - sys.exit(1) - - OPTIONS.input_tmp = common.UnzipTemp(args[0]) - - input_zip = zipfile.ZipFile(args[0], "r") - OPTIONS.info_dict = common.LoadInfoDict(input_zip) - - output_zip = zipfile.ZipFile(args[1], "w", compression=zipfile.ZIP_DEFLATED) - - common.AddBoot(output_zip, OPTIONS.info_dict) - common.AddRecovery(output_zip, OPTIONS.info_dict) - AddSystem(output_zip) - AddUserdata(output_zip) - CopyInfo(output_zip) - - print "cleaning up..." - output_zip.close() - shutil.rmtree(OPTIONS.input_tmp) - - print "done." +def GetBootableImage(name, prebuilt_name, unpack_dir, tree_subdir): + from common import File + return File.FromLocalFile(name, os.path.join(TARGET_DIR, prebuilt_name)) +common.GetBootableImage = GetBootableImage if __name__ == '__main__': - try: - main(sys.argv[1:]) - except common.ExternalError, e: - print - print " ERROR: %s" % (e,) - print - sys.exit(1) + try: + common.CloseInheritedPipes() + main(sys.argv[1:]) + except common.ExternalError, e: + print + print " ERROR: %s" % (e,) + print + sys.exit(1) |