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
29from __future__ import print_function
30
31import sys
32
33if sys.hexversion < 0x02070000:
34  print("Python 2.7 or newer is required.", file=sys.stderr)
35  sys.exit(1)
36
37import os
38import shutil
39import zipfile
40
41import common
42
43OPTIONS = common.OPTIONS
44
45
46def CopyInfo(output_zip):
47  """Copy the android-info.txt file from the input to the output."""
48  common.ZipWrite(
49      output_zip, os.path.join(OPTIONS.input_tmp, "OTA", "android-info.txt"),
50      "android-info.txt")
51
52
53def main(argv):
54  bootable_only = [False]
55
56  def option_handler(o, _):
57    if o in ("-z", "--bootable_zip"):
58      bootable_only[0] = True
59    else:
60      return False
61    return True
62
63  args = common.ParseOptions(argv, __doc__,
64                             extra_opts="z",
65                             extra_long_opts=["bootable_zip"],
66                             extra_option_handler=option_handler)
67
68  bootable_only = bootable_only[0]
69
70  if len(args) != 2:
71    common.Usage(__doc__)
72    sys.exit(1)
73
74  OPTIONS.input_tmp = common.UnzipTemp(args[0], ["IMAGES/*", "OTA/*"])
75  output_zip = zipfile.ZipFile(args[1], "w", compression=zipfile.ZIP_DEFLATED)
76  CopyInfo(output_zip)
77
78  try:
79    images_path = os.path.join(OPTIONS.input_tmp, "IMAGES")
80    # A target-files zip must contain the images since Lollipop.
81    assert os.path.exists(images_path)
82    for image in sorted(os.listdir(images_path)):
83      if bootable_only and image not in ("boot.img", "recovery.img"):
84        continue
85      if not image.endswith(".img"):
86        continue
87      if image == "recovery-two-step.img":
88        continue
89      common.ZipWrite(output_zip, os.path.join(images_path, image), image)
90
91  finally:
92    print("cleaning up...")
93    common.ZipClose(output_zip)
94    shutil.rmtree(OPTIONS.input_tmp)
95
96  print("done.")
97
98
99if __name__ == '__main__':
100  try:
101    common.CloseInheritedPipes()
102    main(sys.argv[1:])
103  except common.ExternalError as e:
104    print("\n   ERROR: %s\n" % (e,))
105    sys.exit(1)
106