REDROOM

Navigation

Breadcrumbs

Create

Preview: temp_utils.py Size: 3.15 KB

//lib/python3.6/site-packages/cloudinit/temp_utils.py

# This file is part of cloud-init. See LICENSE file for license information.

import contextlib
import errno
import logging
import os
import shutil
import tempfile

from cloudinit import util

LOG = logging.getLogger(__name__)
_TMPDIR = None
_ROOT_TMPDIR = "/run/cloud-init/tmp"
_EXE_ROOT_TMPDIR = "/var/tmp/cloud-init"


def get_tmp_ancestor(odir=None, needs_exe: bool = False):
    if odir is not None:
        return odir
    if needs_exe:
        return _EXE_ROOT_TMPDIR
    if _TMPDIR:
        return _TMPDIR
    if os.getuid() == 0:
        return _ROOT_TMPDIR
    return os.environ.get("TMPDIR", "/tmp")


def _tempfile_dir_arg(odir=None, needs_exe: bool = False):
    """Return the proper 'dir' argument for tempfile functions.

    When root, cloud-init will use /run/cloud-init/tmp to avoid
    any cleaning that a distro boot might do on /tmp (such as
    systemd-tmpfiles-clean).

    If the caller of this function (mkdtemp or mkstemp) was provided
    with a 'dir' argument, then that is respected.

    @param odir: original 'dir' arg to 'mkdtemp' or other.
    @param needs_exe: Boolean specifying whether or not exe permissions are
        needed for tempdir. This is needed because /run is mounted noexec.
    """
    tdir = get_tmp_ancestor(odir, needs_exe)
    if not os.path.isdir(tdir):
        os.makedirs(tdir)
        os.chmod(tdir, 0o1777)

    if needs_exe:
        if util.has_mount_opt(tdir, "noexec"):
            LOG.warning(
                "Requested temporal dir with exe permission `%s` is"
                " mounted as noexec",
                tdir,
            )

    if odir is None and not needs_exe:
        global _TMPDIR
        _TMPDIR = tdir

    return tdir


def ExtendedTemporaryFile(**kwargs):
    kwargs["dir"] = _tempfile_dir_arg(
        kwargs.pop("dir", None), kwargs.pop("needs_exe", False)
    )
    fh = tempfile.NamedTemporaryFile(**kwargs)
    # Replace its unlink with a quiet version
    # that does not raise errors when the
    # file to unlink has been unlinked elsewhere..

    def _unlink_if_exists(path):
        try:
            os.unlink(path)
        except OSError as e:
            if e.errno != errno.ENOENT:
                raise e

    fh.unlink = _unlink_if_exists

    # Add a new method that will unlink
    # right 'now' but still lets the exit
    # method attempt to remove it (which will
    # not throw due to our del file being quiet
    # about files that are not there)
    def unlink_now():
        fh.unlink(fh.name)

    setattr(fh, "unlink_now", unlink_now)
    return fh


@contextlib.contextmanager
def tempdir(rmtree_ignore_errors=False, **kwargs):
    # This seems like it was only added in python 3.2
    # Make it since its useful...
    # See: http://bugs.python.org/file12970/tempdir.patch
    tdir = mkdtemp(**kwargs)
    try:
        yield tdir
    finally:
        shutil.rmtree(tdir, ignore_errors=rmtree_ignore_errors)


def mkdtemp(dir=None, needs_exe: bool = False, **kwargs):
    dir = _tempfile_dir_arg(dir, needs_exe)
    return tempfile.mkdtemp(dir=dir, **kwargs)


def mkstemp(dir=None, needs_exe: bool = False, **kwargs):
    dir = _tempfile_dir_arg(dir, needs_exe)
    return tempfile.mkstemp(dir=dir, **kwargs)

Name	Size	Perms	Modified	Actions
analyze DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
cmd DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
config DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
distros DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
filters DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
handlers DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
mergers DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
net DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
reporting DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
sources DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
apport.py	7.05 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
atomic_helper.py	2.45 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
cloud.py	3.22 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
dmi.py	6.77 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
event.py	2.00 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
features.py	3.38 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
gpg.py	4.28 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
helpers.py	16.41 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
importer.py	2.43 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
log.py	5.47 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
netinfo.py	22.97 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
persistence.py	2.52 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
registry.py	1022 B	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
safeyaml.py	10.28 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
settings.py	2.02 KB	lrw-r--r--	2026-01-23 08:56:46	Edit Download recursive recursive
signal_handler.py	1.74 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
simpletable.py	1.93 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
ssh_util.py	22.28 KB	lrw-r--r--	2026-01-23 08:56:47	Edit Download recursive recursive
stages.py	38.88 KB	lrw-r--r--	2026-01-23 08:56:46	Edit Download recursive recursive
subp.py	13.23 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
templater.py	5.95 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
temp_utils.py	3.15 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
type_utils.py	703 B	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
url_helper.py	27.32 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
user_data.py	14.43 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
util.py	96.43 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
version.py	566 B	lrw-r--r--	2026-01-23 08:56:50	Edit Download recursive recursive
warnings.py	3.76 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
__init__.py	0 B	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive

Directory Contents