a
BiP@ @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z G d d
d
Z
d d Z d
d Z dd Z
dd Zdd ZG dd dZG dd dZG dd dZG dd d ZdS ) z7
Classes for representing and manipulating interfaces.
N )access) refpolicy) objectmodel)matching)_c @ sH e Zd ZdZdd Zdd Zdd ZeeeZedd d
Z d d Z
d
S )Paramz;
Object representing a parameter for an interface.
c C s" d| _ tj| _t | _d| _d S )N T) _Param__namer SRC_TYPEtypeIdSet obj_classesrequiredself r 7/usr/lib/python3.9/site-packages/sepolgen/interfaces.py__init__& s
zParam.__init__c C s t |std| || _d S )NzName [%s] is not a param)r
is_idparam
ValueErrorr
)r namer r r set_name, s
zParam.set_namec C s | j S N)r
r r r r get_name1 s zParam.get_namec C s t | jdd S Nr )intr r r r r 6 zParam.)fgetc C s d| j tj| j d| jf S )Nz0 )r r field_to_strr joinr r r r r __repr__8 s zParam.__repr__N) __name__
__module__ __qualname____doc__r r r propertyr numr# r r r r r " s
r c C s d}| |v r|| }||j kr"d S |tjks6|tjkr|j tjksN|j tjkrd}|r`|jg}ng }t|j|D ]}|tj v rrd} qqrtj|_ qd}n t
}| |_ ||_ |||j < |r|j |j |S Nr r )
r r r TGT_TYPE obj_class itertoolschainr r Zimplicitly_typed_objectsr r add)r r avparamsretpZavobjsobjr r r __param_insert> s6
r5 c C s~ d}d}t | jr.t| jtj| |dkr.d}t | jrTt| jtj| |dkrTd}t | jrzt| jtj | |dkrzd}|S )aj Extract the parameters from an access vector.
Extract the parameters (in the form $N) from an access
vector, storing them as Param objects in a dictionary.
Some attempt is made at resolving conflicts with other
entries in the dict, but if an unresolvable conflict is
found it is reported to the caller.
The goal here is to figure out how interface parameters are
actually used in the interface - e.g., that $1 is a domain used as
a SRC_TYPE. In general an interface will look like this:
interface(`foo', `
allow $1 foo : file read;
')
This is simple to figure out - $1 is a SRC_TYPE. A few interfaces
are more complex, for example:
interface(`foo_trans',`
domain_auto_trans($1,fingerd_exec_t,fingerd_t)
allow $1 fingerd_t:fd use;
allow fingerd_t $1:fd use;
allow fingerd_t $1:fifo_file rw_file_perms;
allow fingerd_t $1:process sigchld;
')
Here the usage seems ambiguous, but it is not. $1 is still domain
and therefore should be returned as a SRC_TYPE.
Returns:
0 - success
1 - conflict found
r Fr )
r r src_typer5 r r tgt_typer+ r, OBJ_CLASS)r0 r1 r2 Z found_srcr r r av_extract_paramsj s $ r9 c C s" t | jr t| jtjd |S d S r )r r roler5 r ZROLE)r: r1 r r r role_extract_params s r; c sl fdd}d}|| j tjr"d}|| jtjr4d}|| jtjrFd}t| j rht
| j tj d rhd}|S )Nc s. d}| D ] }t |rt||d rd}q|S r* )r r r5 )setr r2 xr1 r r extract_from_set s
z2type_rule_extract_params..extract_from_setr r ) src_typesr r tgt_typesr+ r r8 r r Z dest_typer5 Z DEST_TYPE)ruler1 r? r2 r r> r type_rule_extract_params s rC c C s2 d}| j D ]"}t|r
t|tjd |r
d}q
|S r* )argsr r r5 r r )ifcallr1 r2 argr r r ifcall_extract_params s
rG c @ s e Zd Zdd Zdd ZdS )AttributeVectorc C s d| _ t | _d S )Nr )r r AccessVectorSetr r r r r s zAttributeVector.__init__c C s | j | d S r )r add_avr r0 r r r rJ s zAttributeVector.add_avN)r$ r% r& r rJ r r r r rH s rH c @ s$ e Zd Zdd Zdd Zdd ZdS ) AttributeSetc C s
i | _ d S r )
attributesr r r r r s zAttributeSet.__init__c C s || j |j< d S r )rM r )r attrr r r add_attr s zAttributeSet.add_attrc C sz dd }d }|D ]V}|d d }|d dkrD|r:| | ||}q|r|d}t|}|| q|rv| | d S )Nc S sH | dd }t|dks(|d dkr4td| t }|d |_|S )Nr r Z Attributez#Syntax error Attribute statement %s)splitlen SyntaxErrorrH r )linefieldsar r r
parse_attr s
z*AttributeSet.from_file..parse_attrrP r [,)rO rR r AccessVectorrJ )r fdrX rW rU lr0 r r r from_file s
zAttributeSet.from_fileN)r$ r% r& r rO r^ r r r r rL s rL c @ sF e Zd Zdi fddZi fddZdd Zdd Zd
d Zd d
ZdS )InterfaceVectorNc C s6 d| _ d| _t | _i | _|r,| || d| _d S )NTr F)enabledr r rI r1 from_interfaceexpanded)r interfacerM r r r r s
zInterfaceVector.__init__c
C s" |j | _ | D ]>}|jtjjkr$qd|j v r0qt|}|D ]}| | q>q|r| D ]n}|j
D ]b}||j
vrvqf|j
| }|jD ]@} t | }|j |j kr|j
|_ |j|j kr|j
|_| | qqfq\| D ]}
t|
| jrq| D ]} t| | jrq| D ]} t| | jrqd S )NZ dontaudit)r ZavrulesZ rule_typer ZAVRuleZALLOWr Zavrule_to_access_vectorsrJ ZtypeattributesrM copyr6 r r7 Zrolesr; r1 Z typerulesrC interface_callsrG )
r rc rM ZavruleZavsr0 Z
typeattributerN Zattr_vecrW r: rB rE r r r ra s>
z InterfaceVector.from_interfacec C s t || jdkr| j| d S r )r9 r1 r rJ rK r r r rJ 3 s zInterfaceVector.add_avc C s8 g }| d| j | jD ]}| t| qd|S )Nz[InterfaceVector %s]
)appendr r strr" )r sr0 r r r to_string9 s
zInterfaceVector.to_stringc C s | S r )r# r r r r __str__@ s zInterfaceVector.__str__c C s d| j | jf S )Nz)r r` r r r r r# C s zInterfaceVector.__repr__) r$ r% r& r ra rJ rj rk r# r r r r r_ s 4r_ c @ sx e Zd ZdddZdd Zdd Zdd Zd
d Zd d
Zi fddZ di fddZ
dd Z dd Z dd Z
dd ZdS ) InterfaceSetNc C s i | _ i | _g | _|| _d S r )
interfaces tgt_type_map tgt_type_alloutput)r rp r r r r H s zInterfaceSet.__init__c C s | j r| j |d d S )Nrf )rp write)r rh r r r oN s zInterfaceSet.oc C s t | j dd dD ]}|d|j t |j dd dD ] }|d|jtj|j f q@|d t |j
}|D ] }|d | |d q~qd S )
Nc S s | j S r r r= r r r r S r z&InterfaceSet.to_file..)keyz[InterfaceVector %s c S s | j S r rs rt r r r r U r z%s:%s z]
rZ rf ) sortedrm valuesrq r r1 r r! r r Zto_listr" )r r\ ZivparamZavlr0 r r r to_fileR s
zInterfaceSet.to_filec C s dd }d }|D ]V}|d d }|d dkrD|r:| | ||}q|r|d}t|}|| q|rv| | | d S )Nc S s | dd }t|dk s(|d dkr4td| t }|d |_t|dkrTd S |dd D ]R}| d}t|dkrtd| t }|d |_tj|d |_||j |j< q`|S ) Nr rP rQ r r_ z)Syntax error InterfaceVector statement %s:z-Invalid param in InterfaceVector statement %s)
rR rS rT r_ r r r Z str_to_fieldr r1 )rU rV ifvZfieldr3 rx r r r parse_ifv^ s
z)InterfaceSet.from_file..parse_ifvrP r rY rZ )add_ifvrR r r[ rJ index)r r\ r| r{ rU r] r0 r r r r^ ] s
zInterfaceSet.from_filec C s || j |j< d S r )rm r )r r{ r r r r} s zInterfaceSet.add_ifvc C sv | j D ]f}t }|jD ]2}t|jr@| j| t } qN||j q|D ] }| j
|g }|| qRq
d S r ) rm rw r< r r r7 ro rg r/ rn
setdefault)r r{ rA r0 r r] r r r r~ s
zInterfaceSet.indexc C s t ||}| | d S r )r_ r} )r rc rM r{ r r r r/ s
zInterfaceSet.addc C s< t | | D ]}| || q| | | d S r )r- r. rm templatesr/ expand_ifcallsr~ )r headersrp rM ir r r add_headers s
zInterfaceSet.add_headersc C sZ t |rPt|dd }|t|jkr,d S |j|d }t|trH|S |gS n|gS d S r )r r r rS rD
isinstancelist)r idrE r) rF r r r map_param s
zInterfaceSet.map_paramc
C s | |j|}|d u rd S | |j|}|d u r4d S | |j|}|d u rNd S t }|jD ]&}| ||} | d u rxq\q\|| q\t|dkrd S |D ]*}
|D ] } |D ]} |j
|
| | | qqqd S )Nr ) r r6 r7 r, r r
ZpermsupdaterS r r/ )
r r{ r0 rE r@ rA r Z new_permsZpermr3 r6 r7 r, r r r
map_add_av s*
zInterfaceSet.map_add_avc C s |d fg}| j |j }d|_t|dkr|d\}}| j |j }||krl|jD ]}| ||| qP|jrlq | D ]l} | j|jkr| t
d d S z|| j }
W n* t y | t
d| j Y qtY n0 | |
| f qtq d S )NTr rP z Found circular interface classz#Missing interface definition for %s)
rm r rb rS popr r re Zifnamerr r KeyErrorrg ) r rc
if_by_namestackr{ ZcurZ
cur_ifcallZcur_ifvr0 rE Znewifr r r do_expand_ifcalls s*
z InterfaceSet.do_expand_ifcallsc C sR i }t | | D ]}|||j< qt | | D ]}| || q s" ,4 Z