Current File : //lib/python3.6/site-packages/firewall/__pycache__/command.cpython-36.pyc
3
@)�f�^ � @ sf d Z dgZddlZddlmZ ddlmZ ddlmZ ddl m
Z
m Z m Z m
Z
mZ G dd� de�ZdS ) z<FirewallCommand class for command line client simplification�FirewallCommand� N)�errors)�
FirewallError)�
DBusException)� checkIPnMask�
checkIP6nMask� check_mac�
check_port�check_single_addressc @ s� e Zd Zd\dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zd]dd�Z d^dd�Z
d_dd�Z d`dd�Z dadd�Z
dbdd�Zdcd d �Zddd d �Zded d!�Zdfd"d#�Zdgd$d%�Zdhd&d'�Zdid(d)�Zdjd*d+�Zdkd,d-�Zd.d/� Zdld1d2�Zdmd3d4�Zd5d6� Zd7d8� Z d9d:� Z d;d<� Z d=d>� Z d?d@� Z dg dAfdBdC�Z!dg fdDdE�Z"dg fdFdG�Z#dHdI� Z$dJdK� Z%dLdM� Z&dNdO� Z'dPdQ� Z(dRdS� Z)dTdU� Z*dVdW� Z+dXdY� Z,dZd[� Z-dS )nr Fc C s || _ || _d| _d | _d S )NT)�quiet�verbose�'_FirewallCommand__use_exception_handler�fw)�selfr r � r � /usr/lib/python3.6/command.py�__init__# s zFirewallCommand.__init__c C s
|| _ d S )N)r )r r r r r �set_fw) s zFirewallCommand.set_fwc C s
|| _ d S )N)r )r �flagr r r � set_quiet, s zFirewallCommand.set_quietc C s | j S )N)r )r r r r � get_quiet/ s zFirewallCommand.get_quietc C s
|| _ d S )N)r )r r r r r � set_verbose2 s zFirewallCommand.set_verbosec C s | j S )N)r )r r r r � get_verbose5 s zFirewallCommand.get_verboseNc C s$ |d k r | j r tjj|d � d S )N�
)r �sys�stdout�write)r �msgr r r � print_msg8 s zFirewallCommand.print_msgc C s$ |d k r | j r tjj|d � d S )Nr )r r �stderrr )r r r r r �print_error_msg< s z FirewallCommand.print_error_msgc C s, d}d}t jj� r || | }| j|� d S )Nz[91mz[00m)r r �isattyr )r r ZFAILZENDr r r �
print_warning@ s
z FirewallCommand.print_warningr c C s, |dkr| j |� n
| j|� tj|� d S )N� )r" r r �exit)r r Z exit_coder r r �print_and_exitG s
z FirewallCommand.print_and_exitc C s | j |d� d S )N� )r% )r r r r r �failR s zFirewallCommand.failc C s" |d k r | j r tjj|d � d S )Nr )r r r r )r r r r r �print_if_verboseU s z FirewallCommand.print_if_verbosec
C s� | j d k r| j j� g }
d} g } x�|D ]�}
|d k r�y ||
�}
W nx tk
r� } z\tjt|��}t|�dkrz| jd| � n| jd| |� || kr�| j |� | d7 } w&W Y d d }~X nX |
j |
� q&W �xb|
D �]X}
g }|d k r�||7 }t
|
t � o�t
|
t � �r|j |
� n||
7 }|d k �r(||7 }| j
� y ||� W n� ttfk
�r } z�t
|t��rx| j|j� � |j� }nt|�}tj|�}|tjtjtjtjgk�r�d}t|�dk�r�| jd| � n,|dk�r�| jd| � d S | jd| |� || k�r| j |� | d7 } W Y d d }~X nX | j� q�W | �s�t|�| k�sJd| k�rNd S t| �dk�rltj| d � nt| �dk�r�tjtj� d S )Nr r# z Warning: %sz Error: %s)r Z authorizeAll� Exceptionr �get_code�str�lenr" r% �append�
isinstance�list�tuple� deactivate_exception_handlerr �fail_if_not_authorized�
get_dbus_name�get_dbus_messager �ALREADY_ENABLED� NOT_ENABLED�ZONE_ALREADY_SET� ALREADY_SET�activate_exception_handlerr r$ Z
UNKNOWN_ERROR)r Zcmd_type�option�
action_method� query_method� parse_method�message�
start_args�end_args�no_exit�itemsZ_errorsZ _error_codes�itemr �code� call_itemr r r Z__cmd_sequenceY sr
z FirewallCommand.__cmd_sequencec C s | j d||||||d� d S )N�add)rA )� _FirewallCommand__cmd_sequence)r r: r; r< r= r> rA r r r � add_sequence� s z FirewallCommand.add_sequencec
C s | j d||||||g|d� d S )NrF )r? rA )rG )r �xr: r; r< r= r> rA r r r �x_add_sequence� s z FirewallCommand.x_add_sequencec C s$ | j d||||||g|g|d� d S )NrF )r? r@ rA )rG ) r �zoner: r; r< r= r> ZtimeoutrA r r r �zone_add_timeout_sequence� s z)FirewallCommand.zone_add_timeout_sequencec C s | j d||||||d� d S )N�remove)rA )rG )r r: r; r< r= r> rA r r r �remove_sequence� s z FirewallCommand.remove_sequencec
C s | j d||||||g|d� d S )NrM )r? rA )rG )r rI r: r; r< r= r> rA r r r �x_remove_sequence� s z!FirewallCommand.x_remove_sequencec
C s g }x�|D ]�}|d k r�y ||�}W n^ t k
r� } zBt|�dkrR| jd| � w
n tjt| ��}
| jd| |
� W Y d d } ~ X nX |j|� q
W �xv|D �]l}g } |d k r�| |7 } t|t � r�t|t
� r�| j|� n| |7 } | j � y || � } W n� t k
�rj } zZ| j
| j� � tj| j� �}
t|�dk�rF| jd| j� � w�n| jd| j� |
� W Y d d } ~ X n` t k
�r� } zBtjt| ��}
t|�dk�r�| jd| � n| jd| |
� W Y d d } ~ X nX | j� t|�dk�r�| jd|| d| f � q�| j| � q�W |�stjd� d S ) Nr# z Warning: %sz Error: %sz%s: %s�no�yesr )rP rQ )r) r, r" r r* r+ r% r- r. r/ r0 r1 r r2 r3 r4 r9 r �print_query_resultr r$ )
r r: r< r= r> r? rA rB rC r rD rE �resr r r Z__query_sequence� sR
" " z FirewallCommand.__query_sequencec C s | j |||||d� d S )N)rA )� _FirewallCommand__query_sequence)r r: r< r= r> rA r r r �query_sequence� s
z FirewallCommand.query_sequencec C s | j |||||g|d� d S )N)r? rA )rT )r rI r: r< r= r> rA r r r �x_query_sequence� s
z FirewallCommand.x_query_sequencec C sJ t |� rFt|� rFt|� rF|jd�o2t|�dk rFttjd| ��|S )Nzipset:� z8'%s' is no valid IPv4, IPv6 or MAC address, nor an ipset)r r r �
startswithr, r r � INVALID_ADDR)r �valuer r r � parse_source� s
z FirewallCommand.parse_source�/c
C sl y|j |�\}}W n$ tk
r6 ttjd| ��Y nX t|�sLttj|��|dkrdttjd| ��||fS )NzTbad port (most likely missing protocol), correct syntax is portid[-portid]%sprotocol�tcp�udp�sctp�dccpz''%s' not in {'tcp'|'udp'|'sctp'|'dccp'})r] r^ r_ r` )�split�
ValueErrorr r � INVALID_PORTr �INVALID_PROTOCOL)r rZ Z separator�port�protor r r �
parse_port s zFirewallCommand.parse_portc
C s� d }d }d }d }d}x�d||d � kr�||d � j dd�d }|t|�d 7 }d||d � krx||d � j dd�d } n ||d � } |t| �d 7 }|dkr�| }q|dkr�| }q|dkr�| }q|dkr�| }q|d kr�|r�qttjd
| ��qW |�s ttjd ��|�sttjd ��|�p|�s*ttjd
��t|��s@ttj|��|dk�rZttjd| ��|�rxt|� �rxttj|��|�r�td|� �r�|�s�td|� �r�ttj |��||||fS )Nr �=r# �:re rf �toport�toaddr�ifz invalid forward port arg '%s'z missing portzmissing protocolzmissing destinationr] r^ r_ r` z''%s' not in {'tcp'|'udp'|'sctp'|'dccp'}�ipv4�ipv6)r] r^ r_ r` )
ra r, r r ZINVALID_FORWARDr rc rd r
rY )
r rZ �compatre Zprotocolrj rk �i�opt�valr r r �parse_forward_port sT
z"FirewallCommand.parse_forward_portc C sF |j d�}t|�dkr"|d dfS t|�dkr2|S ttjd| ��d S )Nrh r# r � r& zinvalid ipset option '%s')ra r, r r ZINVALID_OPTION)r rZ �argsr r r �parse_ipset_optionH s
z"FirewallCommand.parse_ipset_optionc C s. ddg}||kr*t tjd|dj|�f ��|S )Nrm rn z'invalid argument: %s (choose from '%s')z', ')r r � INVALID_IPV�join)r rZ �ipvsr r r �check_destination_ipvR s z%FirewallCommand.check_destination_ipvc C sD y|j dd�\}}W n tk
r4 ttjd��Y nX | j|�|fS )Nri r# z(destination syntax is ipv:address[/mask])ra rb r r ZINVALID_DESTINATIONrz )r rZ ZipvZ destinationr r r �parse_service_destinationZ s z)FirewallCommand.parse_service_destinationc C s0 dddg}||kr,t tjd|dj|�f ��|S )Nrm rn Zebz'invalid argument: %s (choose from '%s')z', ')r r rw rx )r rZ ry r r r � check_ipvb s
zFirewallCommand.check_ipvc C s0 dddg}||kr,t tjd|dj|�f ��|S )Nrt rm rn z'invalid argument: %s (choose from '%s')z', ')r r rw rx )r rZ ry r r r �check_helper_familyj s
z#FirewallCommand.check_helper_familyc C sB |j d�sttjd| ��t|jdd��dk r>ttjd| ��|S )NZ
nf_conntrack_z('%s' does not start with 'nf_conntrack_'rt r# zModule name '%s' too short)rX r r ZINVALID_MODULEr, �replace)r rZ r r r � check_moduler s
z FirewallCommand.check_moduleTc C s� |j � }|j� }|j� }|j� } |j� }
|j� } |j� } |j� }
|j� }|j � }|j
� }|rv|j � }|j � }|j
� }n,|j� }tt|j� | ��}|j� }|j� }dd� }g }|d k r�||kr�|jd� | r�|s�|s�|r�|r�|r�|jd� |�r|ddj|� }| j|� | j�r2| jd| � | jd| � |�rJ| jd t|� � | jd
| � |�sv| jd |�rnd nd
� |�r�| jddj|� � | jddj|� � n(| jddj|� � | jddj|� � | jddjt|�� � | jddjdd� |D �� � | jddjt| �� � |�s:| jd|�r2d nd
� | jd|
�rJd nd
� | jd| �rbdnd djd d� | D �� � | jd djd d� | D �� � | jd dj|
� � | jd!|�r�dnd djt||d"�� � d S )#Nc S sf d}d}y| j |�}W n tk
r* Y n8X |t|�7 }t| ||| |d � j d� � jdd��}|S )Nr z priority=� �"rt )�indexrb r, �intr~ )Zrule�priorityZ
search_strrp r r r �rich_rule_sorted_key� s *zDFirewallCommand.print_zone_policy_info.<locals>.rich_rule_sorted_key�defaultZactivez (%s)z, z summary: z description: z priority: z
target: z icmp-block-inversion: %srQ rP z ingress-zones: r� z egress-zones: z interfaces: z sources: z services: z ports: c S s g | ]}d |d |d f �qS )z%s/%sr r# r )�.0re r r r �
<listcomp>� s z:FirewallCommand.print_zone_policy_info.<locals>.<listcomp>z
protocols: z
forward: %sz masquerade: %sz forward-ports: z
rt c S s$ g | ] \}}}}d ||||f �qS )z$port=%s:proto=%s:toport=%s:toaddr=%sr )r� re rf rj rk r r r r� � s z source-ports: c S s g | ]}d |d |d f �qS )z%s/%sr r# r )r� re r r r r� � s z icmp-blocks: z rich rules: )�key)Z getTargetZ getServices�getPorts� getProtocolsZ
getMasqueradeZgetForwardPorts�getSourcePortsZ
getIcmpBlocksZ getRichRules�getDescription�getShortZgetIngressZonesZgetEgressZonesZ getPriorityZgetIcmpBlockInversion�sorted�setZ
getInterfacesZ
getSourcesZ
getForwardr- rx r r r+ )r rK �settings� default_zone�extra_interfaces�isPolicy�targetZservices�ports� protocolsZ
masqueradeZ
forward_ports� source_portsZ icmp_blocksZrules� description�short_descriptionZ
ingress_zonesZ egress_zonesr� Zicmp_block_inversionZ
interfacesZsourcesZforwardr� Z
attributesr r r �print_zone_policy_info| sx
z&FirewallCommand.print_zone_policy_infoc C s | j ||||dd� d S )NF)r� r� r� )r� )r rK r� r� r� r r r �print_zone_info� s z FirewallCommand.print_zone_infoc C s | j ||||dd� d S )NT)r� r� r� )r� )r Zpolicyr� r� r� r r r �print_policy_info� s z!FirewallCommand.print_policy_infoc C s. |j � }|j� }|j� }|j� }|j� }|j� }|j� } |j� }
|j� } | j |� | j
rt| j d| � | j d| � | j ddj dd� |D �� � | j ddj |� � | j ddj d d� |D �� � | j d
dj |� � | j d dj d d� |j � D �� � | j d
dj t
|
�� � | j ddj t
| �� � d S )Nz summary: z description: z ports: r� c S s g | ]}d |d |d f �qS )z%s/%sr r# r )r� re r r r r� � s z6FirewallCommand.print_service_info.<locals>.<listcomp>z
protocols: z source-ports: c S s g | ]}d |d |d f �qS )z%s/%sr r# r )r� re r r r r� � s z modules: z destination: c S s g | ]\}}d ||f �qS )z%s:%sr )r� �k�vr r r r� � s z includes: z helpers: )r� r� r� Z
getModulesr� �getDestinationsr� Z getIncludesZ
getHelpersr r rx rB r� ) r Zservicer� r� r� r� �modulesr� � destinationsr� ZincludesZhelpersr r r �print_service_info� s2
z"FirewallCommand.print_service_infoc C sp |j � }|j� }|j� }t|�dkr,ddg}| j|� | jrX| jd| � | jd| � | jddj|� � d S )Nr rm rn z summary: z description: z destination: r� )r� r� r� r, r r rx )r Zicmptyper� r� r� r� r r r �print_icmptype_info� s
z#FirewallCommand.print_icmptype_infoc C s� |j � }|j� }|j� }|j� }|j� }| j|� | jrT| jd| � | jd| � | jd| � | jddjdd� |j� D �� � | jddj|� � d S ) Nz summary: z description: z type: z options: r� c S s$ g | ] \}}|r d ||f n|�qS )z%s=%sr )r� r� r� r r r r� s z4FirewallCommand.print_ipset_info.<locals>.<listcomp>z entries: ) ZgetTypeZ
getOptionsZ
getEntriesr� r� r r rx rB )r Zipsetr� Z
ipset_typeZoptions�entriesr� r� r r r �print_ipset_info� s
z FirewallCommand.print_ipset_infoc C s� |j � }|j� }|j� }|j� }|j� }| j|� | jrT| jd| � | jd| � | jd| � | jd| � | jddjdd� |D �� � d S ) Nz summary: z description: z
family: z
module: z ports: r� c S s g | ]}d |d |d f �qS )z%s/%sr r# r )r� re r r r r� s z5FirewallCommand.print_helper_info.<locals>.<listcomp>)r� Z getModuleZ getFamilyr� r� r r rx )r �helperr� r� �moduleZfamilyr� r� r r r �print_helper_info s
z!FirewallCommand.print_helper_infoc C s |r| j d� n | j dd� d S )NrQ rP r# )r% )r rZ r r r rR s z"FirewallCommand.print_query_resultc C s\ | j s� | j|� tjt|��}|tjtjtjtj gkrH| j
d| � n| j d| |� d S )Nz Warning: %sz Error: %s) r
r2 r r* r+ r r5 r6 r7 r8 r" r% )r �exception_messagerD r r r �exception_handler s
z!FirewallCommand.exception_handlerc C s d|krd}| j |tj� d S )NZNotAuthorizedExceptionz`Authorization failed.
Make sure polkit agent is running or run the application as superuser.)r% r ZNOT_AUTHORIZED)r r� r r r r r2 ' s z&FirewallCommand.fail_if_not_authorizedc C s
d| _ d S )NF)r
)r r r r r1 - s z,FirewallCommand.deactivate_exception_handlerc C s
d| _ d S )NT)r
)r r r r r9 0 s z*FirewallCommand.activate_exception_handlerc C sp g }t � }t|�}xP|D ]H}|s"P |j� }t|�dk s|d dkrDq||kr|j|� |j|� qW |j� |S )Nr# r �#�;)r� r� )r� �open�stripr, r- rF �close)r �filenamer� Z entries_set�f�liner r r �get_ipset_entries_from_file3 s
z+FirewallCommand.get_ipset_entries_from_file)FF)N)N)N)Nr )N)N)NNF)F)F)F)F)F)NF)F)F)r\ )F).�__name__�
__module__� __qualname__r r r r r r r r r" r% r' r( rG rH rJ rL rN rO rT rU rV r[ rg rs rv rz r{ r| r} r r� r� r� r� r� r� r� rR r� r2 r1 r9 r� r r r r r " sX
J
2
2
O )�__doc__�__all__r Zfirewallr Zfirewall.errorsr Zdbus.exceptionsr Zfirewall.functionsr r r r r
�objectr r r r r �<module> s
Mini Shell