CasperSecurity
o
쑛h� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ e�e �Z
dZ dZ dde
fdd�Zdde
fd d
�Zd d � Ze jdd
d��Zdde
fdd�Zdde
fdd�ZdS )� N)�utilz/run/cloud-init/tmpz/var/tmp/cloud-initF� needs_exec C s2 | d ur| S |r
t S t�� dkrtS tj�dd�S )Nr �TMPDIRz/tmp)�_EXE_ROOT_TMPDIR�os�getuid� _ROOT_TMPDIR�environ�get)�odirr � r �6/usr/lib/python3/dist-packages/cloudinit/temp_utils.py�get_tmp_ancestor s r c C sL t | |�}tj�|�st�|� t�|d� |r$t�|d�r$t� d|� |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� �noexeczDRequested temporal dir with exe permission `%s` is mounted as noexec)
r r �path�isdir�makedirs�chmodr �
has_mount_opt�LOG�warning)r r �tdirr r r
�_tempfile_dir_arg s
�r c sD t � | d< tjdi | ��� dd� }|� _� fdd�}t� d|� � S )N�dirc
S sH zt �| � W d S ty# } z|jtjkr|�W Y d }~d S d }~ww �N)r �unlink�OSError�errno�ENOENT)r �er r r
�_unlink_if_exists? s ���z0ExtendedTemporaryFile.<locals>._unlink_if_existsc s � � � j� d S r )r �namer ��fhr r
�
unlink_nowM s z)ExtendedTemporaryFile.<locals>.unlink_nowr$ r )r �tempfile�NamedTemporaryFiler �setattr)�kwargsr r$ r r" r
�ExtendedTemporaryFile8 s
r) c
k s< � t di |��}z
|V W tj|| d� d S tj|| d� w )N)�
ignore_errorsr )�mkdtemp�shutil�rmtree)�rmtree_ignore_errorsr( r r r r
�tempdirT s
�"r/ c K � t | |�} tjdd| i|��S �Nr r )r r% r+ �r r r( r r r
r+ ` �
r+ c K r0 r1 )r r% �mkstempr2 r r r
r4 e r3 r4 )NF)F)�
contextlibr �loggingr r, r% � cloudinitr � getLogger�__name__r r r �boolr r r) �contextmanagerr/ r+ r4 r r r r
�<module> s"