a
«°iÆ$ ã @ s d gZ ddlZddlmZ ddlmZ ddlmZm Z m
Z
ddl m Z ddl
mZ ddlmZmZ ddlZd gg d
¢g d ¢d œZi Zi Zi Ze ¡ D ]pZg ee< eƒ ee< ee D ]PZee d
e ¡ ee deef ¡ ee de ¡ ee de ¡ q²q”G dd „ d e ƒZ dS )Úebtablesé N)ÚrunProg)Úlog)ÚtempFileÚreadfileÚ splitArgs)ÚCOMMANDS)Ú ipXtables©Ú
FirewallErrorÚ INVALID_IPVZBROUTING)Z
PREROUTINGZ POSTROUTINGÚOUTPUT)ZINPUTr
ZFORWARD)ZbrouteZnatÚfilterz -N %s_directz-I %s 1 -j %s_directz-I %s_direct 1 -j RETURNz %s_directc @ sÄ e Zd ZdZd ZdZdd„ Zdd„ Zdd„ Zd d
„ Z d d „ Z
d
d„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd d „ Zd/d d!„Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd0d+d,„Zd-d.„ Zd S )1r ZebFc C sB t | j | _t d| j | _| ¡ | _| ¡ | _| ¡ g | _ d S )Nz
%s-restore)
r ÚipvÚ_commandÚ_restore_commandÚ _detect_restore_noflush_optionZrestore_noflush_optionÚ_detect_concurrent_optionÚconcurrent_optionÚ fill_existsÚavailable_tables©Úself© r ú:/usr/lib/python3.9/site-packages/firewall/core/ebtables.pyÚ__init__9 s
zebtables.__init__c C s$ t j | j¡| _t j | j¡| _d S ©N)ÚosÚpathÚexistsr Zcommand_existsr Zrestore_command_existsr r r r r A s zebtables.fill_existsc C s( d}t | jddgƒ}|d dkr$d}|S )NÚ z --concurrentú-Lr )r r )r r Úretr r r r E s
z"ebtables._detect_concurrent_optionc C s. g }z| |d¡ W n ty( Y dS 0 dS )NÚoffFT)Ú set_rulesÚ
ValueError)r Úrulesr r r r O s z'ebtables._detect_restore_noflush_optionc C s‚ g }| j r | j |vr | | j ¡ |dd„ |D ƒ7 }t d| j| jd |¡¡ t| j|ƒ\}}|dkr~td| jd |¡|f ƒ‚|S )Nc S s g | ] }d | ‘qS )z%sr )Ú.0Úitemr r r Ú
^ ó z"ebtables.__run..ú %s: %s %sú r ú'%s %s' failed: %s) r Úappendr Údebug2Ú __class__r Újoinr r% )r ÚargsÚ_argsÚstatusr" r r r Z__runY s
ÿzebtables.__runc C s$ dD ]}||v rt td| ƒ‚qd S )N)z
%%REJECT%%z%%ICMP%%z %%LOGTYPE%%z'%s' invalid for ebtablesr
)r ÚruleÚstrr r r Ú_rule_validatef s
ÿzebtables._rule_validatec C s |t v o|t | v S r )ÚBUILT_IN_CHAINS)r r ÚtableÚchainr r r Úis_chain_builtinl s
ÿzebtables.is_chain_builtinc C sJ g }|r4| d|d|g¡ | d|d|dddg¡ n| d|d|g¡ |S )Nú-tz-Nú-IÚ1z-jZRETURNú-X)r. )r Úaddr9 r: r& r r r Úbuild_chain_rulesp s zebtables.build_chain_rulesc C s8 d|g}|r |d|t |ƒg7 }n |d|g7 }||7 }|S )Nr< r= z-D)r6 )r r@ r9 r: Úindexr2 r5 r r r Ú
build_rule{ s zebtables.build_rulec C s
t |¡S r )r Zcommon_reverse_rule©r r2 r r r Ú reverse_rule„ s zebtables.reverse_rulec C s t |¡ d S r )r Zcommon_check_passthroughrD r r r Úcheck_passthrough‡ s zebtables.check_passthroughc C s
t |¡S r )r Zcommon_reverse_passthroughrD r r r Úreverse_passthroughŠ s z ebtables.reverse_passthroughc
C s t ƒ }d}i }|D ]Â}|d d … }| |¡ dD ]J}z| |¡} W n tyT Y q00 t|ƒ| d kr0| | ¡ | | ¡}q0t|ƒD ]<\} }
tjD ],} | |
v r’|
d¡r²|
d¡s’d|
|| < q’q„| |g ¡ |¡ q|D ]4}|
d| ¡ || D ]}|
d |¡d ¡ qôqÚ| ¡ t |j¡} t d | j| jd
|j| jf ¡ g }
|
d ¡ t| j|
|jd \}}t ¡ d
krÔt|jƒ}|d urÔd} |D ]@}tjd| |f ddd |
d¡sÈtjddd | d7 } q’t |j¡ |dkrtd| jd |
¡|f ƒ‚d S )Nr )r< z--tableé ú"z"%s"z*%s
r, Ú
r+ z%s: %dz --noflush)Ústdiné z%8d: %sr )ÚnofmtÚnlr )rM r- ) r r7 rB r% ÚlenÚpopÚ enumerateÚstringZ
whitespaceÚ
startswithÚendswithÚ
setdefaultr. Úwriter1 Úcloser ÚstatÚnamer r/ r0 r Úst_sizer ZgetDebugLogLevelr Zdebug3Úunlink)r r& Ú
log_deniedZ temp_filer9 Z table_rulesZ_ruler5 ÚoptÚiÚelementÚcrX r2 r4 r" ÚlinesÚliner r r r$ sb
ÿ ÿ
ÿ
ÿzebtables.set_rulesc C s | |¡ | |¡S r )r7 Ú_ebtables__run)r r5 r\ r r r Úset_ruleÍ s
zebtables.set_ruleNc C s† g }|r|gnt ¡ }|D ]f}|| jv r4| |¡ qz*| d|dg¡ | j |¡ | |¡ W q ty~ t d| ¡ Y q0 q|S )Nr< r! z#ebtables table '%s' does not exist.)r8 Úkeysr r. rc r% r Zdebug1)r r9 r" Ztablesr r r Úget_available_tablesÑ s
z ebtables.get_available_tablesc C s i S r r )r r9 r r r Úget_zone_table_chainsá s z ebtables.get_zone_table_chainsc C s> g }t ¡ D ],}|| ¡ vr q dD ]}| d||g¡ q"q |S )N)z-Fr? z-Zr< ©r8 re rf r. )r r& r9 Úflagr r r Úbuild_flush_rulesä s zebtables.build_flush_rulesc C sV g }|dkrdn|}t ¡ D ]4}|| ¡ vr.q t | D ]}| d|d||g¡ q6q |S )NZPANICZDROPr< z-Prh )r ZpolicyZpolicy_detailsr& Z_policyr9 r: r r r Úbuild_set_policy_rulesð s z ebtables.build_set_policy_rulesc C s g S r r r r r r Úbuild_default_tablesú s z ebtables.build_default_tablesr# c C sŽ g }t D ]€}|| ¡ vrqt | d d … }|dkrH|tv rH| t| ¡ d|g}|D ]2}t|ƒtkrt| || ¡ qT| |t|ƒ ¡ qTq|S )Nr# r< )Ú
DEFAULT_RULESrf Ú LOG_RULESÚextendÚtypeÚlistr. r )r r\ Z
default_rulesr9 Z_default_rulesÚprefixr5 r r r Úbuild_default_rulesþ s z ebtables.build_default_rulesc C s
|| j kS r )r )r r r r r Úis_ipv_supported s zebtables.is_ipv_supported)N)r# )Ú__name__Ú
__module__Ú __qualname__r rY Zpolicies_supportedr r r r rc r7 r; rA rC rE rF rG r$ rd rf rg rj rk rl rs rt r r r r r 4 s0
@
) Ú__all__Zos.pathr Zfirewall.core.progr Zfirewall.core.loggerr Zfirewall.functionsr r r Zfirewall.configr Z
firewall.corer Zfirewall.errorsr r rR r8 rm rn Z
OUR_CHAINSre r9 Úsetr: r. r@ Úobjectr r r r r Ú s0 ý