File: //lib/python3/dist-packages/firewall/server/__pycache__/decorators.cpython-310.pyc
o
bhAb� � @ s� d Z g d�ZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZmZ dd lmZ d
d � Zd d
� Zdd� ZG dd� d�ZG dd� de�ZG dd� d�ZdS )z>This module contains decorators for use with and without D-Bus)�handle_exceptions�dbus_handle_exceptions�dbus_service_method� N)�
DBusException)�
FirewallError)�errors)�log)�FirewallDBusException�NotAuthorizedException)�
uid_of_senderc s t �� �� fdd��}|S )zTDecorator to handle exceptions and log them. Used if not conneced
to D-Bus.
c
sj z� | i |��W S t y' } zt�t�� � t�|� W Y d }~d S d }~w ty4 t�� Y d S w �N)r r �debug1� traceback�
format_exc�error� Exception� exception)�args�kwargsr ��func� �</usr/lib/python3/dist-packages/firewall/server/decorators.py�_impl0 s � �z handle_exceptions.<locals>._impl)� functools�wraps�r r r r r r , s r c s& t �� �� fdd��}t�� �|_|S )z�Decorator to handle exceptions, log and report them into D-Bus
:Raises DBusException: on a firewall error code problems.
c
s� z� | i |��W S t yC } z/t �t|��}|tjtjtjtjfv r+t� t|�� nt�
t � � � t�
t|�� tt|���d }~w tyP } z|�d }~w tye } z
t�� tt|���d }~ww r )r �get_code�strr �ALREADY_ENABLED� NOT_ENABLED�ZONE_ALREADY_SET� ALREADY_SETr �warningr
r r r r r r r )r r r �code�exr r r r @ s(
� �� ��z%dbus_handle_exceptions.<locals>._impl)r r �inspect� signature�
__signature__r r r r r ; s r c O s |� dd� tjj| i |��S )z Add sender argument for D-Bus�sender_keyword�sender)�
setdefault�dbus�service�method�r r r r r r ` s r c @ s& e Zd ZdZi Zddd�Zdd� ZdS )� dbus_service_method_deprecatedzRDecorator that maintains a list of deprecated methods in dbus
interfaces.
Nc C s2 || _ | j r| j | jvrt� | j| j < d S d S d S r )� interface�
deprecated�set)�selfr1 r r r �__init__k s �z'dbus_service_method_deprecated.__init__c s4 | j r
| j| j �� j� t�� �� fdd��}|S )Nc s � | i |��S r r r/ r r r r u s z6dbus_service_method_deprecated.__call__.<locals>._impl)r1 r2 �add�__name__r r �r4 r r r r r �__call__q s
z'dbus_service_method_deprecated.__call__r )r7 �
__module__� __qualname__�__doc__r2 r5 r9 r r r r r0 e s
r0 c @ s e Zd ZdZdS )� dbus_service_signal_deprecatedzRDecorator that maintains a list of deprecated signals in dbus
interfaces.
N)r7 r: r; r< r r r r r= z s r= c @ sD e Zd ZdZdZdZdZdZdZdZ dd� Z
e dd � �Z d
d � Z
dS ) �dbus_polkit_require_authz�Decorator factory that checks if the interface/method can be used by the
sender/user. Assumes wrapped function is a method inside a class derived
from DbusServiceObject.
zorg.freedesktop.PolicyKit1z%/org/freedesktop/PolicyKit1/Authorityz$org.freedesktop.PolicyKit1.AuthorityNc C s
|| _ d S r )�_polkit_auth_required)r4 �polkit_auth_requiredr r r r5 � s
z!dbus_polkit_require_auth.__init__c C s | j �| j� d | _d | _d S r )�_bus�remove_signal_receiver�_bus_signal_receiver�_interface_polkit)�cls�name� old_owner� new_ownerr r r �_polkit_name_owner_changed� s z3dbus_polkit_require_auth._polkit_name_owner_changedc s$ t �� �� �fdd��}�j|_|S )Nc s0 t ��js t�� t ��_t ��js$t ��jjt ��jdd�jd�t ��_t ��jsOzt� t ��j�
t ��jt ��j �t ��j �t ��_W n
tj
yN Y nw �j}|sYt�
d��|�d�}|r�t ��jr�t ��j�dd|if|i dd �\}}}|st|d
��ntt ��j|�}|d kr�t|d ��� | i |��S )
N�NameOwnerChangedzorg.freedesktop.DBus)�handler_function� signal_name�dbus_interface�arg0z'Not Authorized: No action_id specified.r* zsystem-bus-namerF � � �polkitr �uid)�typerA r, � SystemBusrC �add_signal_receiverrI � _polkit_namerD � Interface�
get_object� _polkit_path�_polkit_interfacer r? �get�CheckAuthorizationr
r )r r � action_idr* �result�_rR �r r4 r r r � sJ
�
���
�
�
z0dbus_polkit_require_auth.__call__.<locals>._impl)r r r? r8 r r` r r9 � s +z!dbus_polkit_require_auth.__call__)r7 r: r; r< rV rY rZ rA rC rD r5 � classmethodrI r9 r r r r r> � s
r> )r< �__all__r, � dbus.servicer r r& �dbus.exceptionsr �firewall.errorsr �firewallr �firewall.core.loggerr �firewall.server.dbusr r
�firewall.dbus_utilsr r r r r0 r= r> r r r r �<module> s& %