mirror of
https://github.com/flatcar/scripts.git
synced 2025-08-07 21:16:57 +02:00
Add the ability to specify a comma separated list of compression formats. Signed-off-by: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>
117 lines
3.9 KiB
Bash
Executable File
117 lines
3.9 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Copyright (c) 2014 The CoreOS Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
# Shell library for modifying an image built with build_image.
|
|
|
|
start_modify_image() {
|
|
# Default to the most recent image
|
|
if [[ -z "${FLAGS_from}" ]] ; then
|
|
FLAGS_from="$(${SCRIPT_ROOT}/get_latest_image.sh --board=${FLAGS_board})"
|
|
else
|
|
FLAGS_from="$(readlink -f "${FLAGS_from}")"
|
|
fi
|
|
|
|
local src_image="${FLAGS_from}/${FLATCAR_PRODUCTION_IMAGE_NAME}"
|
|
if [[ ! -f "${src_image}" ]]; then
|
|
die_notrace "Source image does not exist: ${src_image}"
|
|
fi
|
|
|
|
# Source should include version.txt, switch to its version information
|
|
if [[ ! -f "${FLAGS_from}/version.txt" ]]; then
|
|
die_notrace "Source version info does not exist: ${FLAGS_from}/version.txt"
|
|
fi
|
|
source "${FLAGS_from}/version.txt"
|
|
FLATCAR_VERSION_STRING="${FLATCAR_VERSION}"
|
|
|
|
# Load after version.txt to set the correct output paths
|
|
. "${BUILD_LIBRARY_DIR}/toolchain_util.sh"
|
|
. "${BUILD_LIBRARY_DIR}/board_options.sh"
|
|
. "${BUILD_LIBRARY_DIR}/build_image_util.sh"
|
|
|
|
# Handle existing directory.
|
|
if [[ -e "${BUILD_DIR}" ]]; then
|
|
if [[ ${FLAGS_replace} -eq ${FLAGS_TRUE} ]]; then
|
|
sudo rm -rf "${BUILD_DIR}"
|
|
else
|
|
error "Directory ${BUILD_DIR} already exists."
|
|
error "Use --build_attempt option to specify an unused attempt."
|
|
error "Or use --replace if you want to overwrite this directory."
|
|
die "Unwilling to overwrite ${BUILD_DIR}."
|
|
fi
|
|
fi
|
|
|
|
# Create the output directory and temporary mount points.
|
|
DST_IMAGE="${BUILD_DIR}/${FLATCAR_PRODUCTION_IMAGE_NAME}"
|
|
ROOT_FS_DIR="${BUILD_DIR}/rootfs"
|
|
mkdir -p "${ROOT_FS_DIR}"
|
|
|
|
info "Copying from ${FLAGS_from}"
|
|
cp "${src_image}" "${DST_IMAGE}"
|
|
|
|
# Copy all extra useful things, these do not need to be modified.
|
|
local update_prefix="${FLATCAR_PRODUCTION_IMAGE_NAME%_image.bin}_update"
|
|
local production_prefix="${FLATCAR_PRODUCTION_IMAGE_NAME%.bin}"
|
|
local container_prefix="${FLATCAR_DEVELOPER_CONTAINER_NAME%.bin}"
|
|
local pcr_data="${FLATCAR_PRODUCTION_IMAGE_NAME%.bin}_pcr_policy.zip"
|
|
EXTRA_FILES=(
|
|
"version.txt"
|
|
"${update_prefix}.bin"
|
|
"${update_prefix}.zip"
|
|
"${pcr_data}"
|
|
"${production_prefix}_contents.txt"
|
|
"${production_prefix}_packages.txt"
|
|
"${production_prefix}_kernel_config.txt"
|
|
"${FLATCAR_DEVELOPER_CONTAINER_NAME}"
|
|
"${container_prefix}_contents.txt"
|
|
"${container_prefix}_packages.txt"
|
|
)
|
|
for filename in "${EXTRA_FILES[@]}"; do
|
|
if [[ -e "${FLAGS_from}/${filename}" ]]; then
|
|
cp "${FLAGS_from}/${filename}" "${BUILD_DIR}/${filename}"
|
|
fi
|
|
done
|
|
|
|
"${BUILD_LIBRARY_DIR}/disk_util" --disk_layout="${FLAGS_disk_layout}" \
|
|
mount "${DST_IMAGE}" "${ROOT_FS_DIR}"
|
|
trap "cleanup_mounts '${ROOT_FS_DIR}'" EXIT
|
|
}
|
|
|
|
finish_modify_image() {
|
|
cleanup_mounts "${ROOT_FS_DIR}"
|
|
trap - EXIT
|
|
|
|
|
|
declare -a files_to_evaluate
|
|
declare -a compressed_images
|
|
declare -a extra_files
|
|
|
|
files_to_evaluate+=( "${DST_IMAGE}" )
|
|
compress_disk_images files_to_evaluate compressed_images extra_files
|
|
|
|
upload_image -d "${DST_IMAGE}.DIGESTS" \
|
|
"${compressed_images[@]}" \
|
|
"${extra_files[@]}"
|
|
|
|
# Upload legacy digests
|
|
upload_legacy_digests "${DST_IMAGE}.DIGESTS" compressed_images
|
|
|
|
for filename in "${EXTRA_FILES[@]}"; do
|
|
if [[ -e "${BUILD_DIR}/${filename}" ]]; then
|
|
upload_image "${BUILD_DIR}/${filename}"
|
|
fi
|
|
done
|
|
|
|
set_build_symlinks "${FLAGS_group}-latest"
|
|
|
|
info "Done. Updated image is in ${BUILD_DIR}"
|
|
cat << EOF
|
|
To convert it to a virtual machine image, use:
|
|
./image_to_vm.sh --from=${OUTSIDE_OUTPUT_DIR} --board=${BOARD}
|
|
|
|
The default type is qemu, see ./image_to_vm.sh --help for other options.
|
|
EOF
|
|
}
|