Editing: rexec.pyo
� z fc @ s# d Z d d l m Z e d d d �[ d d l Z d d l Z d d l Z d d l Z d d l Z d g Z d f d � � YZ d e f d � � YZ d Z d e f d � � YZ d e j f d � � YZ e j Z e j Z d e j f d � � YZ d � Z e d k r e j e � � n d S( s9 Restricted execution facilities. The class RExec exports methods r_exec(), r_eval(), r_execfile(), and r_import(), which correspond roughly to the built-in operations exec, eval(), execfile() and import, but executing the code in an environment that only exposes those built-in operations that are deemed safe. To this end, a modest collection of 'fake' modules is created which mimics the standard modules by the same names. It is a policy decision which built-in modules and operations are made available; this module provides a reasonable default, but derived classes can change the policies e.g. by overriding or extending class variables like ok_builtin_modules or methods like make_sys(). XXX To do: - r_open should allow writing tmp dir - r_exec etc. with explicit globals/locals? (Use rexec("exec ... in ...")?) i����( t warnpy3ks/ the rexec module has been removed in Python 3.0t stackleveli Nt RExect FileBasec B s e Z d Z RS( t filenot flusht isattyt readt readlinet readlinest seekt tellt writet writelinest xreadlinest __iter__( R R R R R R R R R R R R ( t __name__t __module__t ok_file_methods( ( ( s /usr/lib64/python2.7/rexec.pyR s t FileWrapperc B s e Z d � Z d � Z RS( c C sS xL | j D]A } t | | � r t | | � r t | | t | | � � q q Wd S( N( R t hasattrt setattrt getattr( t selft ft m( ( s /usr/lib64/python2.7/rexec.pyt __init__+ s c C s | j � d S( N( R ( R ( ( s /usr/lib64/python2.7/rexec.pyt close0 s ( R R R R ( ( ( s /usr/lib64/python2.7/rexec.pyR ' s sL def %s(self, *args): return getattr(self.mod, self.name).%s(*args) t FileDelegatec B s; e Z d � Z x' e j d D] Z e e e f d Uq WRS( c C s | | _ | | _ d S( N( t modt name( R R R ( ( s /usr/lib64/python2.7/rexec.pyR ; s R N( R ( R R R R R R t TEMPLATE( ( ( s /usr/lib64/python2.7/rexec.pyR 9 s t RHooksc B s} e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c G s� d } d } | rE t | d � t d � k rE | d } | d } n | ru t | d d � ru | d } | d } n | r� t d � n t j j | | � | | _ d S( Ni i����t __class__i s too many arguments( t Nonet typeR t TypeErrort ihookst HooksR t rexec( R t argst verboseR' ( ( s /usr/lib64/python2.7/rexec.pyR E s " c C s | | _ d S( N( R' ( R R' ( ( s /usr/lib64/python2.7/rexec.pyt set_rexecV s c C s | j j � S( N( R' t get_suffixes( R ( ( s /usr/lib64/python2.7/rexec.pyR+ Z s c C s | j j | � S( N( R' t is_builtin( R R ( ( s /usr/lib64/python2.7/rexec.pyR, ] s c C s t | � } | j j | d � S( N( ( t __import__R' t copy_except( R R R ( ( s /usr/lib64/python2.7/rexec.pyt init_builtin` s c C s t d � d S( Ns don't use this( t SystemError( R R ( ( s /usr/lib64/python2.7/rexec.pyt init_frozend t c G s t d � d S( Ns don't use this( R0 ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt load_sourcee R2 c G s t d � d S( Ns don't use this( R0 ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt load_compiledf R2 c G s t d � d S( Ns don't use this( R0 ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt load_packageg R2 c C s | j j | | | � S( N( R' t load_dynamic( R R t filenamet file( ( s /usr/lib64/python2.7/rexec.pyR6 i s c C s | j j | � S( N( R' t add_module( R R ( ( s /usr/lib64/python2.7/rexec.pyR9 l s c C s | j j S( N( R' t modules( R ( ( s /usr/lib64/python2.7/rexec.pyt modules_dicto s c C s | j j d j S( Nt sys( R' R: t path( R ( ( s /usr/lib64/python2.7/rexec.pyt default_pathr s ( R R R R* R+ R, R/ R1 R3 R4 R5 R6 R9 R; R> ( ( ( s /usr/lib64/python2.7/rexec.pyR C s c B s� e Z d Z e e j � Z dX Z dY Z dZ Z d[ Z e j e j f Z d\ d3 d4 � Z d5 � Z d6 � Z d7 � Z d8 � Z d9 � Z d: � Z d; � Z d< � Z d= � Z d> � Z d? � Z d@ � Z dA � Z dB � Z dC � Z dD � Z i i g dE � Z! dF � Z" dG � Z# dH � Z$ dI � Z% dJ � Z& dK � Z' dL � Z( d] i dM � Z) dN � Z* dO � Z+ dP � Z, dQ � Z- dR � Z. dS � Z/ dT dU dV � Z0 dW � Z1 RS(^ s� Basic restricted execution framework. Code executed in this restricted environment will only have access to modules and functions that are deemed safe; you can subclass RExec to add or remove capabilities as desired. The RExec class can prevent code from performing unsafe operations like reading or writing disk files, or using TCP/IP sockets. However, it does not protect against code using extremely large amounts of memory or processor time. t audioopt arrayt binasciit cmatht errnot imageopt marshalt matht md5t operatort parsert selectt shat _sret stropt structt timet _weakreft errort fstatt listdirt lstatt readlinkt statt timest unamet getpidt getppidt getcwdt getuidt getgidt geteuidt getegidt byteordert copyrightt exitt getdefaultencodingt getrefcountt hexversiont maxintt maxunicodet platformt ps1t ps2t versiont version_infot openR8 t reloadR- i c C s� t d � t j j | | � | p+ t | � | _ | j j | � i | _ | j | _ g } x0 | j D]% } | t j k rf | j | � qf qf Wt | � | _ | j � | j � | j � | j � t | j | � | _ t | j | � | _ d S( s� Returns an instance of the RExec class. The hooks parameter is an instance of the RHooks class or a subclass of it. If it is omitted or None, the default RHooks class is instantiated. Whenever the RExec module searches for a module (even a built-in one) or reads a module's code, it doesn't actually go out to the file system itself. Rather, it calls methods of an RHooks instance that was passed to or created by its constructor. (Actually, the RExec object doesn't make these calls --- they are made by a module loader object that's part of the RExec object. This allows another level of flexibility, which can be useful when changing the mechanics of import within the restricted environment.) By providing an alternate RHooks object, we can control the file system accesses made to import a module, without changing the actual algorithm that controls the order in which those accesses are made. For instance, we could substitute an RHooks object that passes all filesystem requests to a file server elsewhere, via some RPC mechanism such as ILU. Grail's applet loader uses this to support importing applets from a URL for a directory. If the verbose parameter is true, additional debugging output may be sent to standard output. s/ This code is not secure in Python 2.2 and laterN( t RuntimeErrorR% t _VerboseR R t hooksR* R: t ok_builtin_modulest ok_dynamic_modulesR< t builtin_module_namest appendt tuplet set_trusted_patht make_builtint make_initial_modulest make_syst RModuleLoadert loadert RModuleImportert importer( R Rq R) t listt mname( ( s /usr/lib64/python2.7/rexec.pyR � s" c C s t t j j t j � | _ d S( N( t filtert osR= t isabsR< t trusted_path( R ( ( s /usr/lib64/python2.7/rexec.pyRw � s c C si | | j k r t d | � n | t j k r> t j | } n t j | | | � } | j | g � } | S( Ns untrusted dynamic module: %s( Rs t ImportErrorR< R: t impR6 R. ( R R R7 R8 t srct dst( ( s /usr/lib64/python2.7/rexec.pyR6 � s c C s | j � | j � d S( N( t make_maint make_osname( R ( ( s /usr/lib64/python2.7/rexec.pyRy � s c C s0 g t j � D] } | d | j k r | ^ q S( Ni ( R� R+ t ok_file_types( R t item( ( s /usr/lib64/python2.7/rexec.pyR+ � s c C s | | j k S( N( Rr ( R R� ( ( s /usr/lib64/python2.7/rexec.pyR, � s c C sD | j t | j � } | j | _ | j | _ | j | _ | _ d S( N( R. t __builtin__t nok_builtin_namest r_importR- t r_reloadRn t r_openRm R8 ( R R ( ( s /usr/lib64/python2.7/rexec.pyRx � s c C s | j d � d S( Nt __main__( R9 ( R ( ( s /usr/lib64/python2.7/rexec.pyR� � s c C se t j } t | � } | j | | j � } i | _ } x' t j j � D] \ } } | | | <qG Wd S( N( R� R R- t copy_onlyt ok_posix_namest environt items( R t osnameR� R� t et keyt value( ( s /usr/lib64/python2.7/rexec.pyR� � s c C s� | j t | j � } | j | _ d g | _ t d | j � | _ | j | _ | j d } | j j � t | j � } | j � t | � | _ d S( Nt RESTRICTEDR< ( R� R< t ok_sys_namesR: t argvt mapR" t ok_pathR= t r_exc_infot exc_infot keysR Rr t sortRv Rt ( R R t l( ( s /usr/lib64/python2.7/rexec.pyRz s c C sy | j | � } x- t | � D] } t | | t | | � � q Wx3 | D]+ } y t | | � WqF t k rp qF XqF W| S( N( t copy_nonet dirR R t delattrt AttributeError( R R� t exceptionsR� R ( ( s /usr/lib64/python2.7/rexec.pyR. s c C s^ | j | � } xH | D]@ } y t | | � } Wn t k rE q n Xt | | | � q W| S( N( R� R R� R ( R R� t namesR� R R� ( ( s /usr/lib64/python2.7/rexec.pyR� s c C s" | j | j � } | j | _ | S( N( R9 R t __doc__( R R� R ( ( s /usr/lib64/python2.7/rexec.pyR� # s c C sR | j j | � } | d k r> | j j | � | j | <} n | j d | _ | S( NR� ( R: t getR" Rq t new_modulet __builtins__( R R� R ( ( s /usr/lib64/python2.7/rexec.pyR9 * s c B s | j d � } | | j Ud S( s Execute code within a restricted environment. The code parameter must either be a string containing one or more lines of Python code, or a compiled code object, which will be executed in the restricted environment's __main__ module. R� N( R9 t __dict__( R t codeR ( ( s /usr/lib64/python2.7/rexec.pyt r_exec3 s c C s | j d � } t | | j � S( sF Evaluate code within a restricted environment. The code parameter must either be a string containing a Python expression, or a compiled code object, which will be evaluated in the restricted environment's __main__ module. The value of the expression or code object will be returned. R� ( R9 t evalR� ( R R� R ( ( s /usr/lib64/python2.7/rexec.pyt r_eval> s c C s# | j d � } t | | j � d S( sf Execute the Python code in the file in the restricted environment's __main__ module. R� N( R9 t execfileR� ( R R8 R ( ( s /usr/lib64/python2.7/rexec.pyt r_execfileJ s c C s | j j | | | | � S( s= Import a module, raising an ImportError exception if the module is considered unsafe. This method is implicitly called by code executing in the restricted environment. Overriding this method in a subclass is used to change the policies enforced by a restricted environment. ( R~ t import_module( R R� t globalst localst fromlist( ( s /usr/lib64/python2.7/rexec.pyR� R s c C s | j j | � S( s Reload the module object, re-parsing and re-initializing it. This method is implicitly called by code executing in the restricted environment. Overriding this method in a subclass is used to change the policies enforced by a restricted environment. ( R~ Rn ( R R ( ( s /usr/lib64/python2.7/rexec.pyR� ] s c C s | j j | � S( s@ Unload the module. Removes it from the restricted environment's sys.modules dictionary. This method is implicitly called by code executing in the restricted environment. Overriding this method in a subclass is used to change the policies enforced by a restricted environment. ( R~ t unload( R R ( ( s /usr/lib64/python2.7/rexec.pyt r_unloadg s c C s} | j d } t | d � | _ t | d � | _ t | d � | _ t t j � | _ t t j � | _ t t j � | _ d S( NR< t stdint stdoutt stderr( R: R t delegate_stdint delegate_stdoutt delegate_stderrR R< R� t restricted_stdinR� t restricted_stdoutR� t restricted_stderr( R t s( ( s /usr/lib64/python2.7/rexec.pyt make_delegate_filesu s c C s� t | d � s | j � n t | d � s8 | j � n | j d } | j | _ | j | _ | j | _ | j t _ | j t _ | j t _ d S( Nt save_stdinR� R< ( R t save_filesR� R: R� R� R� R� R� R� R� R< R� R� ( R R� ( ( s /usr/lib64/python2.7/rexec.pyt set_files~ s c C s? | j � | j d } | j | _ | j | _ | j | _ d S( NR< ( t restore_filesR: R� R� R� R� R� R� ( R R� ( ( s /usr/lib64/python2.7/rexec.pyt reset_files� s c C s( t j | _ t j | _ t j | _ d S( N( R< R� R� R� t save_stdoutR� t save_stderr( R ( ( s /usr/lib64/python2.7/rexec.pyR� � s c C s( | j t _ | j t _ | j t _ d S( N( R� R< R� R� R� R� R� ( R ( ( s /usr/lib64/python2.7/rexec.pyR� � s c C s9 | j � z | j � | | | � } Wd | j � X| S( N( R� R� R� ( R t funcR( t kwt r( ( s /usr/lib64/python2.7/rexec.pyt s_apply� s c G s | j | j | � S( s� Execute code within a restricted environment. Similar to the r_exec() method, but the code will be granted access to restricted versions of the standard I/O streams sys.stdin, sys.stderr, and sys.stdout. The code parameter must either be a string containing one or more lines of Python code, or a compiled code object, which will be executed in the restricted environment's __main__ module. ( R� R� ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt s_exec� s c G s | j | j | � S( s� Evaluate code within a restricted environment. Similar to the r_eval() method, but the code will be granted access to restricted versions of the standard I/O streams sys.stdin, sys.stderr, and sys.stdout. The code parameter must either be a string containing a Python expression, or a compiled code object, which will be evaluated in the restricted environment's __main__ module. The value of the expression or code object will be returned. ( R� R� ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt s_eval� s c G s | j | j | � S( s! Execute the Python code in the file in the restricted environment's __main__ module. Similar to the r_execfile() method, but the code will be granted access to restricted versions of the standard I/O streams sys.stdin, sys.stderr, and sys.stdout. ( R� R� ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt s_execfile� s c G s | j | j | � S( s� Import a module, raising an ImportError exception if the module is considered unsafe. This method is implicitly called by code executing in the restricted environment. Overriding this method in a subclass is used to change the policies enforced by a restricted environment. Similar to the r_import() method, but has access to restricted versions of the standard I/O streams sys.stdin, sys.stderr, and sys.stdout. ( R� R� ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt s_import� s c G s | j | j | � S( s� Reload the module object, re-parsing and re-initializing it. This method is implicitly called by code executing in the restricted environment. Overriding this method in a subclass is used to change the policies enforced by a restricted environment. Similar to the r_reload() method, but has access to restricted versions of the standard I/O streams sys.stdin, sys.stderr, and sys.stdout. ( R� R� ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt s_reload� s c G s | j | j | � S( s� Unload the module. Removes it from the restricted environment's sys.modules dictionary. This method is implicitly called by code executing in the restricted environment. Overriding this method in a subclass is used to change the policies enforced by a restricted environment. Similar to the r_unload() method, but has access to restricted versions of the standard I/O streams sys.stdin, sys.stderr, and sys.stdout. ( R� R� ( R R( ( ( s /usr/lib64/python2.7/rexec.pyt s_unload� s R� i����c C s4 t | � } | d k r$ t d � n t | | | � S( sG Method called when open() is called in the restricted environment. The arguments are identical to those of the open() function, and a file object (or a class instance compatible with file objects) should be returned. RExec's default behaviour is allow opening any file for reading, but forbidding any attempt to write a file. This method is implicitly called by code executing in the restricted environment. Overriding this method in a subclass is used to change the policies enforced by a restricted environment. R� t rbs/ can't open files for writing in restricted mode( R� R� ( t strt IOErrorRm ( R R8 t modet buf( ( s /usr/lib64/python2.7/rexec.pyR� � s c C s( t j � \ } } } d } | | | f S( N( R< R� R" ( R t tyt vat tr( ( s /usr/lib64/python2.7/rexec.pyR� s ( R? R@ RA RB RC RD RE RF RG RH RI RJ RK RL RM RN RO RP ( RQ RR RS RT RU RV RW RX RY RZ R[ R\ R] R^ R_ ( R` Ra Rb Rc Rd Re Rf Rg Rh Ri Rj Rk Rl ( Rm R8 Rn R- N( (2 R R R� Rv R< R= R� Rr R� R� R� R� t C_EXTENSIONt PY_SOURCER� R" R Rw R6 Ry R+ R, Rx R� R� Rz R. R� R� R9 R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� ( ( ( s /usr/lib64/python2.7/rexec.pyR { sd 3 c s d d l } d d l } | j t j d d � \ } } d } g } xI | D]A \ } } | d k ro | d } n | d k rJ | j | � qJ qJ Wt d | � � | r� � j t | � � _ n | r | � j d _ � j d j j d t j j | d � � n � j d j j d d � t j } | r�| d d k r�y t | d � } Wq�t k r}} d t j d | d f GHd SXn | j � ry d d l } Wn t k r�n Xd d l � d � j f � � f d � � Y} y | � j d j � j � Wq{t k r} | SXnd | j � } | j � t | | j d � } y � j | � Wn% t k ri} | S| j � d SXd S( Ni����i s vt:i s -vs -tR) R< R2 t -s %s: can't open file %rt RestrictedConsolec s e Z � � f d � Z RS( c s4 � j d | j d <� j � j j | | f � d S( NR� R� ( R: R� R� t InteractiveConsolet runcode( R t co( R� R� ( s /usr/lib64/python2.7/rexec.pyR� 7 s ( R R R� ( ( R� R� ( s /usr/lib64/python2.7/rexec.pyR� 6 s R� t exec( t getoptt tracebackR< R� Ru R Rr Rv R: R= t insertR� t dirnameR� Rm R� R R R� R� R� R� t interactt SystemExitR R t compileR R� t print_exc( R� R� t optsR( R) t trustedt ot at fpt msgR R� t nt textt c( ( R� R� s /usr/lib64/python2.7/rexec.pyt test sX - " R� ( R� t warningsR R< R� R� R% R� t __all__R R R R R& R t FancyModuleLoaderR{ t ModuleImporterR} Rp R R� R Rb ( ( ( s /usr/lib64/python2.7/rexec.pyt <module> s* 4 � � 4
Save
Back