\8 @ s d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
m Z d d l m
Z
m Z d d l Z e j j d r d d l j j j j Z n e j e j Z y
e Z Wn e k
r d Z Yn Xe Z d d l m Z d d l m! Z! d d d d
g Z" d d d Z# e j$ d d
d Z% e j$ d d Z& e j$ d d Z' e j$ d d Z( Gd d d e) Z* Gd d d Z+ e j$ d d Z, d d Z- e j$ d d Z. e j$ d d Z/ d! d" Z0 d# d$ Z1 d% d
Z2 Gd& d d Z3 e4 e d' rse j5 g Z6 n g Z6 Gd( d d e3 Z7 e j8 e j9 d) d* d+ j: D Z; Gd, d d e Z< d S)- N)six)builtinsmapjava)DistutilsError) working_setAbstractSandboxDirectorySandboxSandboxViolation run_setupc C s d } t | | } | j } Wd QRXt j d d d k sv t j d d d k r t j d d d
k r | j d d } | j d d } | d k r | } t | | d
} t | | | d S)z.
Python 3 implementation of execfile.
rbN r s
s
s
exec)r
r )r r )r r
)openreadsys version_inforeplacecompiler )filenameglobalslocalsmodestreamZscriptcode r /sandbox.py _execfile# s K r c
c sV t j d d } | d k r2 | t j d d <z | VWd | t j d d <Xd S)N)r argv)replsavedr r r save_argv4 s r# c
c s7 t j d d } z | VWd | t j d d <Xd S)N)r path)r" r r r save_path? s r% c c sB t j j | d d t j } | t _ z d VWd | t _ Xd S)zL
Monkey-patch tempfile.tempdir with replacement, ensuring it exists
exist_okTN)
pkg_resourcesZ
py31compatmakedirstempfiletempdir)Z replacementr" r r r
override_tempH s r+ c c s7 t j } t j | z | VWd t j | Xd S)N)osgetcwdchdir)targetr" r r r pushdY s
r0 c @ s( e Z d Z d Z e d d Z d S)UnpickleableExceptionzP
An exception representing another Exception that could not be pickled.
c C se y t j | t j | f SWn> t k
r` d d l m } | j | | t | SYn Xd S)z
Always return a dumped (pickled) type and exc. If exc can't be pickled,
wrap it in UnpickleableException first.
r )r1 N)pickledumps Exceptionsetuptools.sandboxr1 dumprepr)typeexcclsr r r r6 h s
zUnpickleableException.dumpN)__name__
__module__ __qualname____doc__ staticmethodr6 r r r r r1 c s r1 c @ s: e Z d Z d Z d d Z d d Z d d Z d S) ExceptionSaverz^
A Context Manager that will save an exception, serialized, and restore it
later.
c C s | S)Nr )selfr r r __enter__| s zExceptionSaver.__enter__c C s, | s
d St j | | | _ | | _ d S)NT)r1 r6 _saved_tb)rA r8 r9 tbr r r __exit__ s
zExceptionSaver.__exit__c C sK d t | k r d St t j | j \ } } t j | | | j d S)z"restore and re-raise any exceptionrC N)varsr r2 loadsrC r ZreraiserD )rA r8 r9 r r r resume s zExceptionSaver.resumeN)r; r< r= r> rB rF rI r r r r r@ v s r@ c
# sk t j j t } VWd QRXt j j f d d t j D } t | | j d S)z
Context in which imported modules are saved.
Translates exceptions internal to the context into the equivalent exception
outside the context.
Nc 3 s1 | ]' } | k r | j d r | Vq d S)z
encodings.N)
startswith).0mod_name)r" r r <genexpr> s z save_modules.<locals>.<genexpr>)r modulescopyr@ update_clear_modulesrI ) saved_excZ del_modulesr )r" r save_modules s
rS c C s% x t | D] } t j | =q
Wd S)N)listr rN ) module_namesrL r r r rQ s rQ c c s* t j } z | VWd t j | Xd S)N)r' __getstate__ __setstate__)r" r r r save_pkg_resources_state s rX c , c s t j j | d } t v t e t t M t < t | ( t | t
d d VWd QRXWd QRXWd QRXWd QRXWd QRXWd QRXd S)NZtempZ
setuptools) r, r$ joinrX rS hide_setuptoolsr% r# r+ r0
__import__) setup_dirZtemp_dirr r r
setup_context s
r] c C s" t j d } t | j | S)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
z1(setuptools|pkg_resources|distutils|Cython)(\.|$))rer boolmatch)rL patternr r r
_needs_hiding s rb c C s t t t j } t | d S)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.
N)filterrb r rN rQ )rN r r r rZ s rZ c C s. t j j t j j | } t | y | g t | t j d d <t j j d | t j
t j j d d t
| t r | n | j t j } t | ' t d | d d } t | | Wd QRXWn> t k
r"} z | j r| j d r WYd d } ~ Xn XWd QRXd S)z8Run a distutils setup script, sandboxed in its directoryNr c S s
| j S)N)Zactivate)Zdistr r r <lambda> s zrun_setup.<locals>.<lambda>__file__r; __main__)r, r$ abspathdirnamer] rT r r insertr __init__Z callbacksappend
isinstancestrencodegetfilesystemencodingr dictr
SystemExitargs)Z setup_scriptrr r\ Z dunder_filensvr r r r s
c @ s e Z d Z d Z d Z d d Z d d Z d d Z d d
Z d d Z d
d Z
x9 d d d g D]( Z e e
e rp e
e e e <qp Wd d d Z e r e d e Z e d e Z xi d d d d d d d d d d d d! d" d# d$ d% d& d' d( g D]( Z e e
e re e e e <qWd) d* Z x6 d+ d, g D]( Z e e
e rTe e e e <qTWd- d. Z x6 d/ d0 g D]( Z e e
e re e e e <qWd1 d2 Z d3 d4 Z d5 d6 Z d7 d8 Z d S)9r zDWrap 'os' module and 'open()' builtin for virtualizing setup scriptsFc s& f d d t t D _ d S)Nc s5 g | ]+ } | j d r t | r | q S)_)rJ hasattr)rK name)rA r r
<listcomp> s z,AbstractSandbox.__init__.<locals>.<listcomp>)dir_os_attrs)rA r )rA r rj
s zAbstractSandbox.__init__c C s1 x* | j D] } t t | t | | q
Wd S)N)r{ setattrr, getattr)rA sourcerw r r r _copy s zAbstractSandbox._copyc C s8 | j | t r | j t _ | j t _ d | _ d S)NT)r _filer file_openr _active)rA r r r rB s
zAbstractSandbox.__enter__c C s2 d | _ t r t t _ t t _ | j t d S)NF)r r r r r r r rz )rA exc_type exc_value tracebackr r r rF s
zAbstractSandbox.__exit__c C s | | SWd QRXd S)z Run 'func' under os sandboxingNr )rA funcr r r run" s zAbstractSandbox.runc s( t t f d d } | S)Nc s= | j r* | j | | | | \ } } | | | | S)N)r _remap_pair)rA srcdstrr kw)rw originalr r wrap* s !z3AbstractSandbox._mk_dual_path_wrapper.<locals>.wrap)r} rz )rw r r )rw r r _mk_dual_path_wrapper' s z%AbstractSandbox._mk_dual_path_wrapperrenamelinksymlinkNc s. p t t f d d } | S)Nc s1 | j r! | j | | | } | | | S)N)r _remap_input)rA r$ rr r )rw r r r r 8 s z5AbstractSandbox._mk_single_path_wrapper.<locals>.wrap)r} rz )rw r r r )rw r r _mk_single_path_wrapper5 s z'AbstractSandbox._mk_single_path_wrapperr r statlistdirr. chmodchownmkdirremoveunlinkrmdirutimelchownchrootlstatZ startfilemkfifomknodpathconfaccessc s( t t f d d } | S)Nc sM | j r= | j | | | } | j | | | S | | | S)N)r r
_remap_output)rA r$ rr r )rw r r r r M s z4AbstractSandbox._mk_single_with_return.<locals>.wrap)r} rz )rw r r )rw r r _mk_single_with_returnJ s z&AbstractSandbox._mk_single_with_returnreadlinktempnamc s( t t f d d } | S)Nc s, | | } | j r( | j | S| S)N)r r )rA rr r Zretval)rw r r r r \ s z'AbstractSandbox._mk_query.<locals>.wrap)r} rz )rw r r )rw r r _mk_queryY s zAbstractSandbox._mk_queryr- tmpnamc C s | S)z=Called to remap or validate any path, whether input or outputr )rA r$ r r r _validate_pathh s z AbstractSandbox._validate_pathc O s
| j | S)zCalled for path inputs)r )rA operationr$ rr r r r r r l s z AbstractSandbox._remap_inputc C s
| j | S)zCalled for path outputs)r )rA r r$ r r r r p s z AbstractSandbox._remap_outputc O s6 | j | d | | | | j | d | | | f S)z?Called for path pairs like rename, link, and symlink operationsz-fromz-to)r )rA r r r rr r r r r r t s zAbstractSandbox._remap_pair)r; r< r= r> r rj r rB rF r r rw rv rz r r r r r r r r r r r r r r r sB
devnullc @ s e Z d Z d Z e j d d d d d d d d d
d d d
d g
Z d g Z e d d Z d d Z
e r d d d Z d d d Z d d Z
d d Z d d Z d d Z d! d" Z d# d$ d% Z d& 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 sc t j j t j j | | _ t j j | j d | _ d d | D | _ t j | d S)N c S s. g | ]$ } t j j t j j | q Sr )r, r$ normcaserealpath)rK r$ r r r rx s z-DirectorySandbox.__init__.<locals>.<listcomp>)
r, r$ r r _sandboxrY _prefix _exceptionsr rj )rA Zsandbox
exceptionsr r r rj s
!
zDirectorySandbox.__init__c O s& d d l m } | | | | d S)Nr )r
)r5 r
)rA r rr r r
r r r
_violation s zDirectorySandbox._violationrc O sH | d k r5 | j | r5 | j d | | | | t | | | | S)Nr rtr rUUr )r r r r r )_okr r )rA r$ r rr r r r r r s zDirectorySandbox._filec O sH | d k r5 | j | r5 | j d | | | | t | | | | S)Nr r r r r r )r r r r r )r r r )rA r$ r rr r r r r r s zDirectorySandbox._openc C s | j d d S)Nr )r )rA r r r r s zDirectorySandbox.tmpnamc C ss | j } zY d | _ t j j t j j | } | j | p` | | j k p` | j | j SWd | | _ Xd S)NF) r r, r$ r r _exemptedr rJ r )rA r$ Zactiver r r r r s zDirectorySandbox._okc sT f d d | j D } f d d | j D } t j | | } t | S)Nc 3 s | ] } j | Vq d S)N)rJ )rK exception)filepathr r rM s z-DirectorySandbox._exempted.<locals>.<genexpr>c 3 s! | ] } t j | Vq d S)N)r^ r` )rK ra )r r r rM s )r _exception_patterns itertoolschainany)rA r Z
start_matchesZpattern_matchesZ
candidatesr )r r r s
zDirectorySandbox._exemptedc O sE | | j k rA | j | rA | j | t j j | | | | S)zCalled for path inputs) write_opsr r r, r$ r )rA r r$ rr r r r r r s "z DirectorySandbox._remap_inputc O sC | j | s | j | r9 | j | | | | | | | f S)z?Called for path pairs like rename, link, and symlink operations)r r )rA r r r rr r r r r r s z DirectorySandbox._remap_pairi c O sO | t @r6 | j | r6 | j d | | | | | t j | | | | | S)z Called for low-level os.open()zos.open) WRITE_FLAGSr r rz r )rA r flagsr rr r r r r r s zDirectorySandbox.openN)r; r< r= r> rp fromkeysr r _EXCEPTIONSrj r r r r r r r r r r r r r r s
c C s" g | ] } t t | d q S)r )r} rz )rK ar r r rx s rx z4O_WRONLY O_RDWR O_APPEND O_CREAT O_TRUNC O_TEMPORARYc @ s7 e Z d Z d Z e j d j Z d d Z d S)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 \ } } } | j j t S)N)rr tmplformatr )rA cmdrr kwargsr r r __str__ s zSandboxViolation.__str__N) r; r< r= r> textwrapdedentlstripr r r r r r r
s
)=r, r r) operator functoolsr r^
contextlibr2 r Zsetuptools.externr Zsetuptools.extern.six.movesr r Zpkg_resources.py31compatr' platformrJ Z$org.python.modules.posix.PosixModulepythonrN posixZ PosixModulerz rw r r NameErrorr r Zdistutils.errorsr r __all__r contextmanagerr# r% r+ r0 r4 r1 r@ rS rQ rX r] rb rZ r r rv r r r reduceor_splitr r
r r r r <module> s^
wV