U
+ �aG7 � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dl mZ ej�d�r�d dlm m m mZ n eje j ZzeZW n ek
r� dZY nX eZ ddddgZ d0d d
�Z ej d1d d ��Z ej d
d� �Z!ej dd� �Z"ej dd� �Z#G dd� de$�Z%G dd� d�Z&ej dd� �Z'dd� Z(ej dd � �Z)ej d d � �Z*d d d!d"d#hZ+d$d%� Z,d&d'� Z-d(d� Z.G d)d� d�Z/e0e d*��r�e j1gZ2ng Z2G d+d� de/�Z3e�4ej5d,d-� d.�6� D ��Z7G d/d� de
�Z8dS )2� N)�DistutilsError)� working_set�java�AbstractSandbox�DirectorySandbox�SandboxViolation� run_setupc C sJ d}t | |��}|�� }W 5 Q R X |dkr.|}t|| d�}t|||� dS )z.
Python 3 implementation of execfile.
�rbN�exec)�open�read�compiler
)�filename�globals�locals�mode�stream�script�code� r �:/tmp/pip-install-1bd_1mtk/setuptools/setuptools/sandbox.py� _execfile! s r c
c sD t jd d � }| d k r$| t jd d �< z
|V W 5 |t jd d �< X d S �N)�sys�argv)�repl�savedr r r � save_argv. s
r c
c s. t jd d � } z
| V W 5 | t jd d �< X d S r )r �path�r r r r � save_path9 s
r c c s2 t j| dd� tj}| t_z
dV W 5 |t_X dS )zL
Monkey-patch tempfile.tempdir with replacement, ensuring it exists
T)�exist_okN)�os�makedirs�tempfile�tempdir)� replacementr r r r �
override_tempB s
r' c c s. t �� }t �| � z
|V W 5 t �|� X d S r )r" �getcwd�chdir)�targetr r r r �pushdS s
r+ c @ s e Zd ZdZedd� �ZdS )�UnpickleableExceptionzP
An exception representing another Exception that could not be pickled.
c C sP zt �| �t �|�fW S tk
rJ ddlm} |�||t|��� Y S X dS )z�
Always return a dumped (pickled) type and exc. If exc can't be pickled,
wrap it in UnpickleableException first.
r )r, N)�pickle�dumps� Exception�setuptools.sandboxr, �dump�repr)�type�exc�clsr r r r1 b s
zUnpickleableException.dumpN)�__name__�
__module__� __qualname__�__doc__� staticmethodr1 r r r r r, ] s r, c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ExceptionSaverz^
A Context Manager that will save an exception, serialized, and restore it
later.
c C s | S r r ��selfr r r � __enter__v s zExceptionSaver.__enter__c C s |sd S t �||�| _|| _dS �NT)r, r1 �_saved�_tb)r= r3 r4 �tbr r r �__exit__y s
zExceptionSaver.__exit__c C s2 dt | �krdS ttj| j�\}}|�| j��dS )z"restore and re-raise any exceptionr@ N)�vars�mapr- �loadsr@ �with_tracebackrA )r= r3 r4 r r r �resume� s zExceptionSaver.resumeN)r6 r7 r8 r9 r> rC rH r r r r r; p s r; c # sV t j�� � t� � } � V W 5 Q R X t j�� � � fdd�t jD �}t|� | �� dS )z�
Context in which imported modules are saved.
Translates exceptions internal to the context into the equivalent exception
outside the context.
c 3 s$ | ] }|� kr|� d �s|V qdS )z
encodings.N��
startswith)�.0�mod_namer r r � <genexpr>� s
�z save_modules.<locals>.<genexpr>N)r �modules�copyr; �update�_clear_modulesrH )� saved_excZ del_modulesr r r � save_modules� s
�rS c C s t | �D ] }tj|= qd S r )�listr rN )Z module_namesrL r r r rQ � s rQ c c s$ t �� } z
| V W 5 t �| � X d S r )�
pkg_resources� __getstate__� __setstate__r r r r �save_pkg_resources_state� s
rX c c s� t j�| d�}t� �x t� �f t� �T t� t� �< t|��( t | �� t
d� d V W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X d S )N�temp�
setuptools) r" r �joinrX rS r �hide_setuptoolsr r' r+ �
__import__)� setup_dir�temp_dirr r r �
setup_context� s
r` rZ � distutilsrU ZCython�_distutils_hackc C s | � dd�d }|tkS )aH
>>> _needs_hiding('setuptools')
True
>>> _needs_hiding('pkg_resources')
True
>>> _needs_hiding('setuptools_plugin')
False
>>> _needs_hiding('setuptools.__init__')
True
>>> _needs_hiding('distutils')
True
>>> _needs_hiding('os')
False
>>> _needs_hiding('Cython')
True
�.� r )�split�_MODULES_TO_HIDE)rL � base_moduler r r �
_needs_hiding� s rh c C s6 t j�dd�} | dk r | �� ttt j�}t|� dS )a%
Remove references to setuptools' modules from sys.modules to allow the
invocation to import the most appropriate setuptools. This technique is
necessary to avoid issues such as #315 where setuptools upgrading itself
would fail to find a function declared in the metadata.
rb N)r rN �getZ remove_shim�filterrh rQ )rb rN r r r r\ � s
r\ c C s� t j�t j�| ��}t|��� zl| gt|� tjdd�< tj�d|� t �
� t j � dd� � t
|�� t| dd�}t| |� W 5 Q R X W n4 tk
r� } z|jr�|jd r�� W 5 d}~X Y nX W 5 Q R X dS )z8Run a distutils setup script, sandboxed in its directoryNr c S s | � � S r )�activate)�distr r r �<lambda>� � zrun_setup.<locals>.<lambda>�__main__)�__file__r6 )r" r �abspath�dirnamer` rT r r �insertr �__init__� callbacks�appendr �dictr �
SystemExit�args)Z setup_scriptry r^ �ns�vr r r r � s
c @ s e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dD ] Z e e
e �rDe
e �e� e < qDd$dd�Zerzede�Zede�ZdD ] Z e e
e �r�ee �e� e < q�dd� ZdD ] Z e e
e �r�ee �e� e < q�dd� ZdD ] Z e e
e �r�ee �e� e < q�d d � Zd d � Zd d!� Zd"d#� ZdS )%r zDWrap 'os' module and 'open()' builtin for virtualizing setup scriptsFc s � fdd�t t�D �� _d S )Nc s$ g | ] }|� d �st� |�r|�qS )�_)rJ �hasattr)rK �namer< r r �
<listcomp> s
�z,AbstractSandbox.__init__.<locals>.<listcomp>)�dir�_os�_attrsr< r r< r rt
s
�zAbstractSandbox.__init__c C s" | j D ]}tt|t||�� qd S r )r� �setattrr" �getattr)r= �sourcer~ r r r �_copy s
zAbstractSandbox._copyc C s( | � | � tr| jt_| jt_d| _d S r? )r� �_file�builtins�file�_openr �_activer< r r r r> s
zAbstractSandbox.__enter__c C s$ d| _ trtt_tt_| �t� d S �NF)r� r� r� r� r� r r� r� )r= �exc_type� exc_value� tracebackr r r rC s
zAbstractSandbox.__exit__c
C s" | � |� W 5 Q R � S Q R X dS )z Run 'func' under os sandboxingNr )r= �funcr r r �run" s zAbstractSandbox.runc s t t� ��� �fdd�}|S )Nc s2 | j r | j� ||f|�|�\}}�||f|�|�S r )r� � _remap_pair)r= �src�dstry �kw�r~ �originalr r �wrap* s z3AbstractSandbox._mk_dual_path_wrapper.<locals>.wrap�r� r� �r~ r� r r� r �_mk_dual_path_wrapper' s
z%AbstractSandbox._mk_dual_path_wrapper)�rename�link�symlinkNc s �p t t� ��� �fdd�}|S )Nc s* | j r| j� |f|�|�}�|f|�|�S r )r� � _remap_input�r= r ry r� r� r r r� 8 s z5AbstractSandbox._mk_single_path_wrapper.<locals>.wrapr� )r~ r� r� r r� r �_mk_single_path_wrapper5 s z'AbstractSandbox._mk_single_path_wrapperr� r )�stat�listdirr) r �chmod�chown�mkdir�remove�unlink�rmdir�utime�lchown�chroot�lstatZ startfile�mkfifo�mknod�pathconf�accessc s t t� ��� �fdd�}|S )Nc sB | j r2| j� |f|�|�}| �� �|f|�|��S �|f|�|�S r )r� r� �
_remap_outputr� r� r r r� M s z4AbstractSandbox._mk_single_with_return.<locals>.wrapr� r� r r� r �_mk_single_with_returnJ s
z&AbstractSandbox._mk_single_with_return)�readlink�tempnamc s t t� ��� �fdd�}|S )Nc s �||�}| j r | �� |�S |S r )r� r� )r= ry r� �retvalr� r r r� \ s
z'AbstractSandbox._mk_query.<locals>.wrapr� r� r r� r � _mk_queryY s
zAbstractSandbox._mk_query)r( �tmpnamc C s |S )z=Called to remap or validate any path, whether input or outputr )r= r r r r �_validate_pathh s z AbstractSandbox._validate_pathc O s
| � |�S �zCalled for path inputs�r� �r= � operationr ry r� r r r r� l s z AbstractSandbox._remap_inputc C s
| � |�S )zCalled for path outputsr� )r= r� r r r r r� p s z AbstractSandbox._remap_outputc O s0 | j |d |f|�|�| j |d |f|�|�fS )�?Called for path pairs like rename, link, and symlink operationsz-fromz-to)r� �r= r� r� r� ry r� r r r r� t s �zAbstractSandbox._remap_pair)N)r6 r7 r8 r9 r� rt r� r> rC r� r� r~ r} r� r r� r� r� r� r� r� r� r� r� r r r r r s<
�devnullc @ s� e Zd ZdZe�dddddddd d
d d d
dg
�ZdgZefdd�Z dd� Z
e rXd'dd�Z d(dd�Z dd� Z
dd � Zd d � Zd d � Zd!d"� Zd)d$d%�Zd&S )*r z<Restrict operations to a single subdirectory - pseudo-chrootr r� r� r� r� r� r� r� r� r� r� r� r� z.*lib2to3.*\.pickle$c C sF t j�t j�|��| _t j�| jd�| _dd� |D �| _t� | � d S )N� c S s g | ]}t j�t j�|���qS r )r" r �normcase�realpath)rK r r r r r � s �z-DirectorySandbox.__init__.<locals>.<listcomp>)
r" r r� r� �_sandboxr[ �_prefix� _exceptionsr rt )r= Zsandbox�
exceptionsr r r rt � s �zDirectorySandbox.__init__c O s ddl m} ||||��d S )Nr )r )r0 r )r= r� ry r� r r r r �
_violation� s zDirectorySandbox._violation�rc O s: |dkr(| � |�s(| jd||f|�|� t||f|�|�S )N�r� �rtr ZrU�Ur� )�_okr� r� �r= r r ry r� r r r r� � s zDirectorySandbox._filec O s: |dkr(| � |�s(| jd||f|�|� t||f|�|�S )Nr� r )r� r� r� r� r r r r� � s zDirectorySandbox._openc C s | � d� d S )Nr� )r� r< r r r r� � s zDirectorySandbox.tmpnamc C sR | j }z>d| _ tj�tj�|��}| �|�p@|| jkp@|�| j�W �S || _ X d S r� ) r� r" r r� r� � _exemptedr� rJ r� )r= r �activer� r r r r� � s
�
�zDirectorySandbox._okc s< � fdd�| j D �}� fdd�| jD �}t�||�}t|�S )Nc 3 s | ]}� � |�V qd S r rI )rK � exception��filepathr r rM � s �z-DirectorySandbox._exempted.<locals>.<genexpr>c 3 s | ]}t �|� �V qd S r )�re�match)rK �patternr� r r rM � s �)r� �_exception_patterns� itertools�chain�any)r= r� Z
start_matchesZpattern_matches�
candidatesr r� r r� � s
�
� zDirectorySandbox._exemptedc O s4 || j kr0| �|�s0| j|tj�|�f|�|� |S r� )� write_opsr� r� r" r r� r� r r r r� � s z DirectorySandbox._remap_inputc O s2 | � |�r| � |�s*| j|||f|�|� ||fS )r� )r� r� r� r r r r� � s z DirectorySandbox._remap_pair� c O s@ |t @ r*| �|�s*| jd|||f|�|� tj|||f|�|�S )z Called for low-level os.open()zos.open)� WRITE_FLAGSr� r� r� r )r= r� �flagsr ry r� r r r r � s zDirectorySandbox.openN)r� )r� )r� )r6 r7 r8 r9 rw �fromkeysr� r� � _EXCEPTIONSrt r� r� r� r� r� r� r� r� r r r r r r � s: ��
c C s g | ]}t t|d ��qS )r r� )rK �ar r r r � s r z4O_WRONLY O_RDWR O_APPEND O_CREAT O_TRUNC O_TEMPORARYc @ s& e Zd ZdZe�d��� Zdd� ZdS )r zEA setup script attempted to modify the filesystem outside the sandboxa
SandboxViolation: {cmd}{args!r} {kwargs}
The package setup script has attempted to modify files on your system
that are not within the EasyInstall build area, and has been aborted.
This package cannot be safely installed by EasyInstall, and may not
support alternate installation locations even if you run its setup
script by hand. Please inform the package's author and the EasyInstall
maintainers to find out if a fix or workaround is available.
c C s | j \}}}| jjf t� �S r )ry �tmpl�formatr )r= �cmdry �kwargsr r r �__str__� s zSandboxViolation.__str__N) r6 r7 r8 r9 �textwrap�dedent�lstripr� r� r r r r r � s )N)N)9r" r r$ �operator� functoolsr� r� �
contextlibr- r� r� rU �distutils.errorsr r �platformrJ Z$org.python.modules.posix.PosixModule�pythonrN �posixZ PosixModuler� r~ r� r� � NameErrorr r� �__all__r �contextmanagerr r r' r+ r/ r, r; rS rQ rX r` rf rh r\ r r r} r� r� r �reduce�or_re r� r r r r r �<module> s�
�
� w
V ��