File: linaro-android-media-create

package info (click to toggle)
linaro-image-tools 2016.05-1.1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye, buster, sid
  • size: 1,180 kB
  • sloc: python: 15,832; sh: 393; makefile: 6
file content (195 lines) | stat: -rwxr-xr-x 7,104 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
#!/usr/bin/env python
# Copyright (C) 2011 Linaro
#
# Author: Jeremy Chang <jeremy.chang@linaro.org>
#
# This file is part of Linaro Image Tools.
#
# Linaro Image Tools is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Linaro Image Tools is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Linaro Image Tools.  If not, see <http://www.gnu.org/licenses/>.

import atexit
import os
import sys
import tempfile

from linaro_image_tools import cmd_runner

from linaro_image_tools.media_create.android_boards import (
    get_board_config,
    )
from linaro_image_tools.media_create.check_device import (
    confirm_device_selection_and_ensure_it_is_ready)
from linaro_image_tools.media_create.partitions import (
    Media,
    setup_android_partitions,
    partition_mounted,
    )
from linaro_image_tools.media_create.rootfs import populate_partition
from linaro_image_tools.media_create.unpack_binary_tarball import (
    unpack_android_binary_tarball
    )
from linaro_image_tools.media_create import get_android_args_parser
from linaro_image_tools.utils import (
    additional_android_option_checks,
    android_hwpack_in_boot_tarball,
    ensure_command,
    get_logger,
    disable_automount,
    enable_automount,
    )


# Just define the global variables
TMP_DIR = None
BOOT_DISK = None
CACHE_DISK = None
SDCARD_DISK = None


# Registered as the first atexit handler as we want this to be the last
# handler to execute.
@atexit.register
def cleanup_tempdir():
    """Remove TEMP_DIR with all its contents.

    Before doing so, make sure DISKs are not mounted.
    """
    devnull = open('/dev/null', 'w')
    # ignore non-zero return codes
    for disk in BOOT_DISK, CACHE_DISK, \
                SDCARD_DISK:
        if disk is not None:
            try:
                cmd_runner.run(['umount', disk],
                      stdout=devnull, stderr=devnull, as_root=True).wait()
            except cmd_runner.SubcommandNonZeroReturnValue:
                pass
    # Remove TMP_DIR as root because some files written there are
    # owned by root.
    if TMP_DIR is not None:
        cmd_runner.run(['rm', '-rf', TMP_DIR], as_root=True).wait()


def ensure_required_commands(args):
    """Ensure we have the commands that we know are going to be used."""
    required_commands = [
        'mkfs.vfat', 'sfdisk', 'mkimage', 'parted']
    for command in required_commands:
        ensure_command(command)


if __name__ == '__main__':
    parser = get_android_args_parser()
    args = parser.parse_args()

    logger = get_logger(debug=args.debug)

    additional_android_option_checks(args)

    # If --help was specified this won't execute.
    # Create temp dir and initialize rest of path vars.
    TMP_DIR = tempfile.mkdtemp()
    BOOT_DIR = os.path.join(TMP_DIR, 'boot')
    SYSTEM_DIR = os.path.join(TMP_DIR, 'system')
    DATA_DIR = os.path.join(TMP_DIR, 'data')

    BOOT_DISK = os.path.join(TMP_DIR, 'boot-disc')
    CACHE_DISK = os.path.join(TMP_DIR, 'cache-disc')
    SDCARD_DISK = os.path.join(TMP_DIR, 'sdcard-disc')

    if args.dev == 'iMX53':
        # XXX: remove this and the corresponding entry in android_board_configs
        logger.warning("DEPRECATION WARNING: iMX53 is deprecated, please "
                       "use mx53loco.")

    ensure_required_commands(args)

    # Do this by default, disable automount options and re-enable them at exit.
    disable_automount()
    atexit.register(enable_automount)

    media = Media(args.device)
    if media.is_block_device:
        if not confirm_device_selection_and_ensure_it_is_ready(args.device):
            sys.exit(1)
    elif not args.should_create_partitions:
        logger.error("Do not use --no-part in conjunction with --image_file.")
        sys.exit(1)

    cmd_runner.run(['mkdir', '-p', BOOT_DIR]).wait()
    cmd_runner.run(['mkdir', '-p', SYSTEM_DIR]).wait()
    cmd_runner.run(['mkdir', '-p', DATA_DIR]).wait()

    unpack_android_binary_tarball(args.boot, BOOT_DIR)

    board_config = get_board_config(args.dev)

    hwpack_exists, config_file = android_hwpack_in_boot_tarball(BOOT_DIR)
    if not args.hwpack and not hwpack_exists:
        # No hwpack in the boot tarball nor provided on the command line.
        logger.warning("No hwpack found in the boot tarball nor passed on "
                       "the command line. Default values will be used.")
    elif not args.hwpack and hwpack_exists:
        board_config.from_file(config_file)
    elif args.hwpack:
        logger.warning("Values from the hwpack provided on the command line "
                       "will be used.")
        board_config.from_file(args.hwpack)

    board_config.add_boot_args(args.extra_boot_args)
    board_config.add_boot_args_from_file(args.extra_boot_args_file)

    # Create partitions
    boot_partition, system_partition, cache_partition, \
        data_partition, sdcard_partition = setup_android_partitions( \
        board_config, media, args.image_size, args.boot_label,
        args.should_create_partitions, args.should_align_boot_part)

    board_config.populate_raw_partition(args.device, BOOT_DIR)
    populate_partition(BOOT_DIR + "/boot", BOOT_DISK, boot_partition)
    board_config.populate_boot_script(boot_partition, BOOT_DISK, args.consoles)
    with partition_mounted(boot_partition, BOOT_DISK):
        board_config.install_boot_loader(args.device, BOOT_DISK)

    if args.system:
        with partition_mounted(system_partition, SYSTEM_DIR):
            unpack_android_binary_tarball(args.system, TMP_DIR)
    elif args.systemimage :
        cmd_runner.run( [ 'e2label', args.systemimage, "system"],
                        stderr=open('/dev/null', 'w'),
                        as_root=True).wait()
        cmd_runner.run( [ 'dd', 'if=%s' % args.systemimage,
                          'of=%s' % system_partition],
                        stderr=open('/dev/null', 'w'),
                        as_root=True).wait()
    else:
        #should not reach here
        pass

    if args.userdata:
        with partition_mounted(data_partition, DATA_DIR):
            unpack_android_binary_tarball(args.userdata, TMP_DIR)
    elif args.userdataimage:
        cmd_runner.run( [ 'e2label', args.userdataimage, "userdata"],
                        stderr=open('/dev/null', 'w'),
                        as_root=True).wait()
        cmd_runner.run( [ 'dd', 'if=%s' % args.userdataimage,
                          'of=%s' % data_partition],
                        stderr=open('/dev/null', 'w'),
                        as_root=True).wait()
    else:
        #should not reach here
        pass

    print "Done creating Linaro Android image on %s" % args.device