Exemplo n.º 1
0
def patch_img():
    root_img = os.path.join(magic_consts.ACTIVE_IMG_PATH, "root.squashfs")
    active_metadata_path = os.path.join(
        magic_consts.ACTIVE_IMG_PATH, "metadata.yaml")
    patch_file = os.path.join(magic_consts.CWD, "patches/fuel_agent/patch")
    path_archname_pairs = [(os.path.join(magic_consts.ACTIVE_IMG_PATH, p), p)
                           for p in ["vmlinuz", "initrd.img"]]

    with temp_util.temp_dir() as temp_dir:
        patched_img = os.path.join(temp_dir, "root.squashfs")
        patched_metadata_path = os.path.join(temp_dir, "metadata.yaml")

        _patch_squashfs(root_img, patched_img, patch_file)
        _mk_metadata(active_metadata_path, patched_metadata_path, patched_img)

        path_archname_pairs.append((patched_img, "root.squashfs"))
        path_archname_pairs.append((patched_metadata_path, "metadata.yaml"))

        with tempfile.NamedTemporaryFile() as archive_file:
            with tarfile.open(name=archive_file.name, mode="w:gz") as archive:
                for path, archname in path_archname_pairs:
                    archive.add(path, archname)

            LOG.info("Import image using fuel-bootstrap")
            subprocess.call(["fuel-bootstrap", "import", archive_file.name])
            LOG.info("Activate image using `fuel-bootstrap activate`")
Exemplo n.º 2
0
def _patch_squashfs(root_img, patched_img, *patches):
    with temp_util.temp_dir() as patch_dir:
        LOG.info("unsquash root image to temporary directory")
        subprocess.call(["unsquashfs", "-f", "-d", patch_dir, root_img])
        LOG.info("apply patch to root image")
        patch.patch_apply(patch_dir, patches)
        LOG.info("create new root.squashfs image")
        subprocess.call(["mksquashfs", patch_dir, patched_img])
Exemplo n.º 3
0
def test_temp_dir(mocker, is_exception, prefix):

    class TestException(Exception):
        pass

    kwargs = {}
    if prefix is not None:
        kwargs['prefix'] = prefix

    temp_dir_name = mock.Mock()
    mkdtemp_mock = mocker.patch("tempfile.mkdtemp", return_value=temp_dir_name)
    rm_tree_mock = mocker.patch("shutil.rmtree")
    if is_exception:
        with pytest.raises(TestException):
            with tempfile.temp_dir(**kwargs):
                raise TestException
    else:
        with tempfile.temp_dir(**kwargs):
            pass
    mkdtemp_mock.assert_called_once_with(**kwargs)
    rm_tree_mock.assert_called_once_with(temp_dir_name)
Exemplo n.º 4
0
def apply_patches(container, prefix, *patches, **kwargs):
    """Apply set of patches to a container's filesystem"""
    revert = kwargs.pop('revert', False)
    # TODO: review all logic here to apply all preprocessing steps to patches
    # beforehand
    files = [os.path.join(prefix, f)
             for f in patch.get_filenames_from_patches(prefix, *patches)]
    if not files:
        LOG.warn("Nothing to patch!")
        return
    with tempfile.temp_dir(prefix='octane_docker_patches.') as tempdir:
        get_files_from_docker(container, files, tempdir)
        patch.patch_apply(os.path.join(tempdir, prefix), patches, revert)
        put_files_to_docker(container, "/", tempdir)