mender-setup-image.inc

IMAGE_FSTYPES += "${@bb.utils.contains('DISTRO_FEATURES', 'mender-image', ' mender mender.bmap', '', d)}"
IMAGE_FSTYPES += "${@bb.utils.contains('DISTRO_FEATURES', 'mender-image-sd', ' sdimg sdimg.bmap', '', d)}"
IMAGE_FSTYPES += "${@bb.utils.contains('DISTRO_FEATURES', 'mender-image-ubi', ' ubimg mtdimg ubimg.bmap', '', d)}"
IMAGE_FSTYPES += "${@bb.utils.contains('DISTRO_FEATURES', 'mender-image-uefi', ' uefiimg uefiimg.bmap', '', d)}"
IMAGE_FSTYPES += "${@bb.utils.contains('DISTRO_FEATURES', 'mender-image-bios', ' biosimg biosimg.bmap', '', d)}"
IMAGE_FSTYPES += "${@bb.utils.contains('DISTRO_FEATURES', 'mender-image-gpt', ' gptimg gptimg.bmap', '', d)}"

MACHINE_ESSENTIAL_EXTRA_RDEPENDS_append_mender-image_x86 =     " kernel-image"
MACHINE_ESSENTIAL_EXTRA_RDEPENDS_append_mender-image_x86-64 =  " kernel-image"
MACHINE_ESSENTIAL_EXTRA_RDEPENDS_append_mender-image_arm =     " kernel-image kernel-devicetree"
MACHINE_ESSENTIAL_EXTRA_RDEPENDS_append_mender-image_aarch64 = " kernel-image kernel-devicetree"

python() {
    features = d.getVar('DISTRO_FEATURES').split()
    for dep in features:
        if dep.startswith("mender-image-") and 'mender-image' not in features:
            bb.fatal("If enabling %s in MENDER_FEATURES_ENABLE or DISTRO_FEATURES, mender-image must also be enabled"
                     % dep)
}

MENDER_BOOT_PART_MOUNT_LOCATION = "/boot/efi"
MENDER_BOOT_PART_MOUNT_LOCATION_mender-uboot = "/uboot"
MENDER_BOOT_PART_MOUNT_LOCATION_mender-grub_mender-bios = "/boot/grub"

# Update fstab for Mender
ROOTFS_POSTPROCESS_COMMAND_append = " mender_update_fstab_file;"
mender_update_fstab_file() {
    local tmpBootPart="${MENDER_BOOT_PART}"
    local tmpDataPart="${MENDER_DATA_PART}"
    if ${@bb.utils.contains('DISTRO_FEATURES', 'mender-partuuid', 'true', 'false', d)}; then
        tmpBootPart="PARTUUID=${@mender_get_partuuid_from_device(d, '${MENDER_BOOT_PART}')}"
        tmpDataPart="PARTUUID=${@mender_get_partuuid_from_device(d, '${MENDER_DATA_PART}')}"
    fi

    if ! ${@bb.utils.contains('DISTRO_FEATURES', 'mender-image-ubi', 'true', 'false', d)}; then
        if [ "${MENDER_BOOT_PART_SIZE_MB}" != "0" ] && [ -n "${MENDER_BOOT_PART}" ]; then
            echo "# Where the U-Boot environment resides; for devices with SD card support ONLY!" >> ${IMAGE_ROOTFS}${sysconfdir}/fstab
            mkdir -p ${IMAGE_ROOTFS}/${MENDER_BOOT_PART_MOUNT_LOCATION}
            printf "%-20s %-20s %-10s %-21s %-2s %s\n" ${tmpBootPart} ${MENDER_BOOT_PART_MOUNT_LOCATION} ${MENDER_BOOT_PART_FSTYPE} defaults,sync 0 0 >> ${IMAGE_ROOTFS}${sysconfdir}/fstab
        else
            bbdebug 2 "MENDER_BOOT_PART not applicable. Not adding to fstab..."
        fi
    fi

    mkdir -p ${IMAGE_ROOTFS}/data
    printf "%-20s %-20s %-10s %-21s %-2s %s\n" ${tmpDataPart} /data ${MENDER_DATA_PART_FSTYPE} ${MENDER_DATA_PART_FSTAB_OPTS} 0 0 >> ${IMAGE_ROOTFS}${sysconfdir}/fstab
}

def mender_default_state_scripts_version(d):
    pref_mender = d.getVar('PREFERRED_VERSION_pn-mender')
    if pref_mender and pref_mender.startswith("1."):
        return "2"
    else:
        return "3"

# Setup state script version file.
MENDER_STATE_SCRIPTS_VERSION = "${@mender_default_state_scripts_version(d)}"
ROOTFS_POSTPROCESS_COMMAND_append = "${@bb.utils.contains('DISTRO_FEATURES', 'mender-image', ' mender_create_scripts_version_file;', '', d)}"

mender_create_scripts_version_file() {
    install -d -m 755 ${IMAGE_ROOTFS}${sysconfdir}/mender/scripts/
    echo -n "${MENDER_STATE_SCRIPTS_VERSION}" > ${IMAGE_ROOTFS}${sysconfdir}/mender/scripts/version
}

IMAGE_ROOTFS_EXCLUDE_PATH_append_mender-image = " data/ ${@d.getVar('MENDER_BOOT_PART_MOUNT_LOCATION')[1:]}/"


################################################################################
# Implementation of IMAGE_ROOTFS_EXCLUDE_PATH
################################################################################

python prepare_excluded_directories() {
    exclude_var = d.getVar('IMAGE_ROOTFS_EXCLUDE_PATH')
    if not exclude_var:
        return

    taskname = d.getVar("BB_CURRENTTASK")

    if d.getVarFlag('do_%s' % taskname, 'respect_exclude_path') == '0':
        bb.debug(1, "'IMAGE_ROOTFS_EXCLUDE_PATH' is set but 'respect_exclude_path' variable flag is 0 for this image type, so ignoring it")
        return

    import shutil
    from oe.path import copyhardlinktree

    exclude_list = exclude_var.split()

    rootfs_orig = d.getVar('IMAGE_ROOTFS')
    # We need a new rootfs directory we can delete files from. Copy to
    # workdir.
    new_rootfs = os.path.realpath(os.path.join(d.getVar("WORKDIR"), "rootfs.%s" % taskname))

    if os.path.lexists(new_rootfs):
        shutil.rmtree(os.path.join(new_rootfs))

    copyhardlinktree(rootfs_orig, new_rootfs)

    for orig_path in exclude_list:
        path = orig_path
        if os.path.isabs(path):
            bb.fatal("IMAGE_ROOTFS_EXCLUDE_PATH: Must be relative: %s" % orig_path)

        full_path = os.path.realpath(os.path.join(new_rootfs, path))

        # Disallow climbing outside of parent directory using '..',
        # because doing so could be quite disastrous (we will delete the
        # directory).
        if not full_path.startswith(new_rootfs):
            bb.fatal("'%s' points to a path outside the rootfs" % orig_path)

        if not os.path.lexists(full_path):
            # Nothing to do if it doesn't exist.
            continue

        if path.endswith(os.sep):
            # Delete content only.
            for entry in os.listdir(full_path):
                full_entry = os.path.join(full_path, entry)
                if os.path.isdir(full_entry) and not os.path.islink(full_entry):
                    shutil.rmtree(full_entry)
                else:
                    os.remove(full_entry)
        else:
            # Delete whole directory.
            shutil.rmtree(full_path)

    # Save old value for cleanup later.
    d.setVar('IMAGE_ROOTFS_ORIG', rootfs_orig)
    d.setVar('IMAGE_ROOTFS', new_rootfs)
}

python cleanup_excluded_directories() {
    exclude_var = d.getVar('IMAGE_ROOTFS_EXCLUDE_PATH')
    if not exclude_var:
        return

    taskname = d.getVar("BB_CURRENTTASK")

    if d.getVarFlag('do_%s' % taskname, 'respect_exclude_path') == '0':
        return

    import shutil

    rootfs_dirs_excluded = d.getVar('IMAGE_ROOTFS')
    rootfs_orig = d.getVar('IMAGE_ROOTFS_ORIG')
    # This should never happen, since we should have set it to a different
    # directory in the prepare function.
    assert rootfs_dirs_excluded != rootfs_orig

    shutil.rmtree(rootfs_dirs_excluded)
    d.setVar('IMAGE_ROOTFS', rootfs_orig)
}

python() {
    if not bb.data.inherits_class("image", d):
        return

    d.appendVarFlag("do_rootfs", "vardeps", " IMAGE_ROOTFS_EXCLUDE_PATH")

    for image_type in set(d.getVar('IMAGE_FSTYPES').split()):
        task = "do_image_%s" % image_type
        d.appendVarFlag(task, "prefuncs", " prepare_excluded_directories")
        d.appendVarFlag(task, "postfuncs", " cleanup_excluded_directories")
}