| # -*- coding: utf-8 -*- |
| # Copyright 2019 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Library for handling Chrome OS partition.""" |
| |
| from __future__ import print_function |
| |
| import os |
| |
| from chromite.lib import constants |
| from chromite.lib import cros_build_lib |
| from chromite.lib import cros_logging as logging |
| from chromite.lib import image_lib |
| from chromite.lib import osutils |
| |
| |
| _STATEFUL_FILE = 'stateful.tgz' |
| |
| |
| def GenerateStatefulPayload(image_path, output_directory): |
| """Generates a stateful update payload given a full path to an image. |
| |
| Args: |
| image_path: Full path to the image. |
| output_directory: Path to the directory to leave the resulting output. |
| |
| Returns: |
| str: The full path to the generated file. |
| """ |
| logging.info('Generating stateful update file.') |
| |
| output_gz = os.path.join(output_directory, _STATEFUL_FILE) |
| |
| # Mount the image to pull out the important directories. |
| with osutils.TempDir() as stateful_mnt, \ |
| image_lib.LoopbackPartitions(image_path, stateful_mnt) as image: |
| rootfs_dir = image.Mount((constants.PART_STATE,))[0] |
| |
| try: |
| logging.info('Tarring up /usr/local and /var!') |
| cros_build_lib.sudo_run([ |
| 'tar', |
| '-czf', |
| output_gz, |
| '--directory=%s' % rootfs_dir, |
| '--transform=s,^dev_image,dev_image_new,', |
| '--transform=s,^var_overlay,var_new,', |
| 'dev_image', |
| 'var_overlay', |
| ]) |
| except: |
| logging.error('Failed to create stateful update file') |
| raise |
| |
| logging.info('Successfully generated %s', output_gz) |
| |
| return output_gz |