3
\�me� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ eje �Z
da dZ dZ
ded�dd�Zded�d d
�Zd d � Ze jdd
d��Zded�dd�Zded�dd�ZdS )� 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 tjjdd�S )Nr ZTMPDIRz/tmp)�_EXE_ROOT_TMPDIR�_TMPDIR�os�getuid� _ROOT_TMPDIR�environ�get)�odirr � r � /usr/lib/python3.6/temp_utils.py�get_tmp_ancestor s r c C s^ t | |�}tjj|�s,tj|� tj|d� |rHtj|d�rHtj 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 �path�isdir�makedirs�chmodr Z
has_mount_opt�LOGZwarningr )r r �tdirr r r
�_tempfile_dir_arg s
r c sT t | jdd �| jdd��| d< tjf | �� dd� }|� _� fdd�}t� d|� � S ) N�dirr Fc S sF yt j| � W n2 tk
r@ } z|jtjkr0|�W Y d d }~X nX d S )N)r �unlink�OSError�errno�ENOENT)r �er r r
�_unlink_if_existsI s
z0ExtendedTemporaryFile.<locals>._unlink_if_existsc s � j � j� d S )N)r �namer )�fhr r
�
unlink_nowW s z)ExtendedTemporaryFile.<locals>.unlink_nowr )r �pop�tempfileZNamedTemporaryFiler �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)�mkdtemp�shutilZrmtree)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
�<module> s"
"