3
\me @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ eje Z
da dZ dZ
dedddZdedd d
Zd d Ze jdd
dZdedddZdedddZdS ) N)utilz/run/cloud-init/tmpz/var/tmp/cloud-initF) needs_exec C s: | d k r | S |rt S tr tS tj dkr,tS tjjddS )Nr ZTMPDIRz/tmp)_EXE_ROOT_TMPDIR_TMPDIRosgetuid _ROOT_TMPDIRenvironget)odirr r /usr/lib/python3.6/temp_utils.pyget_tmp_ancestor s r c C s^ t | |}tjj|s,tj| tj|d |rHtj|drHtj d| | dkrZ| rZ|a
|S )a% 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.
i ZnoexeczDRequested temporal dir with exe permission `%s` is mounted as noexecN) r r pathisdirmakedirschmodr Z
has_mount_optLOGZwarningr )r r tdirr r r
_tempfile_dir_arg s
r c sT t | jdd | jdd| d< tjf | dd }| _ fdd}t d| S ) Ndirr Fc S sF yt j| W n2 tk
r@ } z|jtjkr0|W Y d d }~X nX d S )N)r unlinkOSErrorerrnoENOENT)r er r r
_unlink_if_existsI s
z0ExtendedTemporaryFile.._unlink_if_existsc s j j d S )N)r namer )fhr r
unlink_nowW s z)ExtendedTemporaryFile..unlink_nowr )r poptempfileZNamedTemporaryFiler setattr)kwargsr r r )r r
ExtendedTemporaryFile@ s r$ c
k s* t f |}z
|V W d tj|| d X d S )N)
ignore_errors)mkdtempshutilZrmtree)Zrmtree_ignore_errorsr# r r r r
tempdir^ s
r( c K s t | |} tjf d| i|S )Nr )r r! r& )r r r# r r r
r& j s
r& c K s t | |} tjf d| i|S )Nr )r r! mkstemp)r r r# r r r
r) o s
r) )NF)NF)F)NF)NF)
contextlibr Zloggingr r' r! Z cloudinitr Z getLogger__name__r r r r boolr r r$ contextmanagerr( r& r) r r r r
s"
"