3
U'íc9 ã @ s U d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
ddl mZ d d
lmZ d d lmZ d d lmZ d d
lmZ ejdejdejf dZdZe Z ej e e Z ej e ddhZ ddhZ!ddhZ"ej#e$ddddd d d d gfej%e$dd d d!d gfej&e$d"d#d$d%d&d gfe e$d"d'dd&d(d d)d d*g ffZ'ej(ej(ej)ej*e f df 'ejej+e d+d,d-Z,e-e.d.d/d0Z/eed1d2d3Z0eje e1d4d5d6Z2eje e1d4d7d8Z3G d9d: d:eZ4G d;d< d<e4Z5G d=d> d>e Z6G d?d@ d@e6e
Z7dS )AzA sandbox layer that ensures unsafe operations cannot be performed.
Useful when the template itself comes from an untrusted source.
é N)Úformatter_field_name_split)Úabc)Údeque)Ú Formatter)ÚEscapeFormatter)ÚMarkupé )Ú Environment)Ú
SecurityError)ÚContext)Ú UndefinedÚF.)Zboundi Úgi_frameÚgi_codeÚcr_frameÚcr_codeÚag_codeÚag_frameÚaddÚclearÚdifference_updateÚdiscardÚpopÚremoveÚsymmetric_difference_updateÚupdateÚpopitemÚ
setdefaultÚappendÚreverseÚinsertÚsortÚextendÚ
appendleftÚ
extendleftÚpopleftÚrotate)ÚcallableÚreturnc C s: t | tjtjf s | jdkr"d S | j}t |tr6|S d S )NÚformatÚ
format_map)r) r* )Ú
isinstanceÚtypesÚ
MethodTypeÚBuiltinMethodTypeÚ__name__Ú__self__Ústr)r' Úobj© r3 ú0/tmp/pip-build-gk9425m9/Jinja2/jinja2/sandbox.pyÚinspect_format_methodS s
r5 )Úargsr( c G s( t | }t|tkr$tdt d|S )zWA range that can't generate ranges with a length of more than
MAX_RANGE items.
z@Range too big. The sandbox blocks ranges larger than MAX_RANGE (z).)ÚrangeÚlenÚ MAX_RANGEÚ
OverflowError)r6 Úrngr3 r3 r4 Ú
safe_rangea s
r< )Úfr( c C s
d| _ | S )zMarks a function or method as unsafe.
.. code-block: python
@unsafe
def delete(self):
pass
T)Úunsafe_callable)r= r3 r3 r4 Úunsafep s r? )r2 Úattrr( c C sØ t | tjr|tkrÎdS n´t | tjr<|tks6|tkrÎdS nt | trT|dkrÎdS nzt | tjtjtj frndS t | tj
r|t krÎdS nFt tdr¬t | tj
r¬|tkrÎdS n"t tdrÎt | tjrÎ|tkrÎdS |jdS )a´ Test if the attribute given is an internal python attribute. For
example this function returns `True` for the `func_code` attribute of
python objects. This is useful if the environment method
:meth:`~SandboxedEnvironment.is_safe_attribute` is overridden.
>>> from jinja2.sandbox import is_internal_attribute
>>> is_internal_attribute(str, "mro")
True
>>> is_internal_attribute(str, "upper")
False
TÚmroÚ
CoroutineTypeÚAsyncGeneratorTypeÚ__)r+ r, Ú FunctionTypeÚUNSAFE_FUNCTION_ATTRIBUTESr- ÚUNSAFE_METHOD_ATTRIBUTESÚtypeÚCodeTypeÚ
TracebackTypeÚ FrameTypeÚ
GeneratorTypeÚUNSAFE_GENERATOR_ATTRIBUTESÚhasattrrB ÚUNSAFE_COROUTINE_ATTRIBUTESrC Ú!UNSAFE_ASYNC_GENERATOR_ATTRIBUTESÚ
startswith)r2 r@ r3 r3 r4 Úis_internal_attribute} s,
rR c C s( x"t D ]\}}t| |r||kS qW dS )að This function checks if an attribute on a builtin mutable object
(list, dict, set or deque) or the corresponding ABCs would modify it
if called.
>>> modifies_known_mutable({}, "clear")
True
>>> modifies_known_mutable({}, "keys")
False
>>> modifies_known_mutable([], "append")
True
>>> modifies_known_mutable([], "index")
False
If called with an unsupported object, ``False`` is returned.
>>> modifies_known_mutable("foo", "upper")
False
F)Ú
_mutable_specr+ )r2 r@ Ztypespecr? r3 r3 r4 Úmodifies_known_mutable¢ s
rT c sØ e Zd ZU dZdZejejejej ej
ej ej dZ
ejeejejejgejf f
ejejdZejeejejgejf f e Zeje e Zeje ejejdd fdd Zejeejed d
d Z ejed d
dZ e eejejejdddZ e eejejdddZ ejej!eejf ej!eje"f dddZ#ejeej!eje"f dddZ$ejee"ddd Z%d$eej&ejd f ejeejf ej'ej ed d d Z(e ejejejejd!d"d#Z) Z*S )%ÚSandboxedEnvironmentaé The sandboxed environment. It works like the regular environment but
tells the compiler to generate sandboxed code. Additionally subclasses of
this environment may override the methods that tell the runtime what
attributes or functions are safe to access.
If the template tries to access insecure code a :exc:`SecurityError` is
raised. However also other exceptions may occur during the rendering so
the caller has to ensure that all exceptions are caught.
T)ú+ú-Ú*ú/z//z**ú%)rV rW N)r6 Úkwargsr( c s4 t j|| t| jd< | jj | _| jj | _d S )Nr7 ) ÚsuperÚ__init__r< ÚglobalsÚdefault_binop_tableÚcopyÚ binop_tableÚdefault_unop_tableÚ
unop_table)Úselfr6 r[ )Ú __class__r3 r4 r] ü s
z SandboxedEnvironment.__init__)r2 r@ Úvaluer( c C s |j dpt|| S )aY The sandboxed environment will call this method to check if the
attribute of an object is safe to access. Per default all attributes
starting with an underscore are considered private as well as the
special attributes of internal python objects as returned by the
:func:`is_internal_attribute` function.
Ú_)rQ rR )rd r2 r@ rf r3 r3 r4 Úis_safe_attribute s z&SandboxedEnvironment.is_safe_attribute)r2 r( c C s t |ddpt |dd S )zêCheck if an object is safely callable. By default callables
are considered safe unless decorated with :func:`unsafe`.
This also recognizes the Django convention of setting
``func.alters_data = True``.
r> FZ alters_data)Úgetattr)rd r2 r3 r3 r4 Úis_safe_callable s z%SandboxedEnvironment.is_safe_callable)ÚcontextÚoperatorÚleftÚrightr( c C s | j | ||S )z÷For intercepted binary operator calls (:meth:`intercepted_binops`)
this function is executed instead of the builtin operator. This can
be used to fine tune the behavior of certain operators.
.. versionadded:: 2.6
)ra )rd rk rl rm rn r3 r3 r4 Ú
call_binop s z SandboxedEnvironment.call_binop)rk rl Úargr( c C s | j | |S )zõFor intercepted unary operator calls (:meth:`intercepted_unops`)
this function is executed instead of the builtin operator. This can
be used to fine tune the behavior of certain operators.
.. versionadded:: 2.6
)rc )rd rk rl rp r3 r3 r4 Ú call_unop! s z SandboxedEnvironment.call_unop)r2 Úargumentr( c C s y|| S t tfk
r t|try t|}W n tk
rF Y nDX yt||}W n tk
rj Y n X | j|||r~|S | j||S Y nX | j ||dS )z(Subscribe an object from sandboxed code.)r2 Úname)
Ú TypeErrorÚ LookupErrorr+ r1 Ú Exceptionri ÚAttributeErrorrh Úunsafe_undefinedÚ undefined)rd r2 rr r@ rf r3 r3 r4 Úgetitem* s
z SandboxedEnvironment.getitem)r2 Ú attributer( c C sr yt ||}W n6 tk
rD y|| S ttfk
r> Y nX Y n X | j|||rX|S | j||S | j||dS )zSubscribe an object from sandboxed code and prefer the
attribute. The attribute passed *must* be a bytestring.
)r2 rs )ri rw rt ru rh rx ry )rd r2 r{ rf r3 r3 r4 ri A s z SandboxedEnvironment.getattrc C s& | j d|dt|jd||tdS )z1Return an undefined object for unsafe attributes.zaccess to attribute z of z object is unsafe.)rs r2 Úexc)ry rH r/ r
)rd r2 r{ r3 r3 r4 rx R s
z%SandboxedEnvironment.unsafe_undefined.)Úsr6 r[ Ú format_funcr( c C s t |trt| |jd}nt| }|dk rl|jdkrlt|dksD|r`tdt||dk d|d }f }|j|||}t ||S )z
If a format call is detected, then this is routed through this
method so that our safety sandbox can be used for it.
)ÚescapeNr* r z(format_map() takes exactly one argument z givenr )
r+ r ÚSandboxedEscapeFormatterr ÚSandboxedFormatterr/ r8 rt ÚvformatrH )rd r} r6 r[ r~ Ú formatterÚrvr3 r3 r4 Ú
format_string\ s
z"SandboxedEnvironment.format_string)Ú _SandboxedEnvironment__contextÚ_SandboxedEnvironment__objr6 r[ r( c O sJ t |}|dk r | j||||S | j|s8t|d|j|f||S )z#Call an object from sandboxed code.Nz is not safely callable)r5 r
rj r
Úcall)Z_SandboxedEnvironment__selfr r r6 r[ Úfmtr3 r3 r4 r y s
zSandboxedEnvironment.call)N)+r/ Ú
__module__Ú __qualname__Ú__doc__Z sandboxedrl r ÚsubÚmulÚtruedivÚfloordivÚpowÚmodr_ ÚtÚDictr1 ÚCallableÚAnyÚposÚnegrb Ú frozensetZintercepted_binopsÚ FrozenSetZintercepted_unopsr] Úboolrh rj r ro rq ÚUnionr rz ri rx ÚTupleÚOptionalr
r Ú
__classcell__r3 r3 )re r4 rU » s<
,(
*rU c s0 e Zd ZdZejeejed fdd Z Z S )Ú ImmutableSandboxedEnvironmentzÓWorks exactly like the regular `SandboxedEnvironment` but does not
permit modifications on the builtin mutable objects `list`, `set`, and
`dict` by using the :func:`modifies_known_mutable` function.
)r2 r@ rf r( c s t j|||sdS t|| S )NF)r\ rh rT )rd r2 r@ rf )re r3 r4 rh s z/ImmutableSandboxedEnvironment.is_safe_attribute)
r/ r r r r r r1 r rh r r3 r3 )re r4 r s r c s\ e Zd Zeejdd fdd Zeejej ej eejf ej
ejef dddZ Z S )r N)Úenvr[ r( c s || _ t jf | d S )N)Ú_envr\ r] )rd r¡ r[ )re r3 r4 r] s zSandboxedFormatter.__init__)Ú
field_namer6 r[ r( c C sV t |\}}| j|||}x2|D ]*\}}|r<| jj||}q | jj||}q W ||fS )N)r Ú get_valuer¢ ri rz ) rd r£ r6 r[ ÚfirstÚrestr2 Úis_attrÚir3 r3 r4 Ú get_field s z SandboxedFormatter.get_field)
r/ r r r r r r] r1 ÚSequenceÚMappingr r© r r3 r3 )re r4 r s r c @ s e Zd ZdS )r N)r/ r r r3 r3 r3 r4 r « s r )8r rl r, Útypingr Ú_stringr Ú collectionsr r Ústringr Z
markupsafer r Ú environmentr Ú
exceptionsr
Zruntimer r ZTypeVarr r r
r9 ÚsetrF ÚSetr1 rG rM rO rP Ú
MutableSetr ÚMutableMappingÚMutableSequencerS r ÚTyper r r5 Úintr7 r< r? r rR rT rU r r r r3 r3 r3 r4 Ú<module> sn ,
% R