a
i @ s d Z g dZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZmZ dd lmZ d
d Zd d
Zdd ZG dd dZG dd deZG dd dZdS )z>This module contains decorators for use with and without D-Bus)handle_exceptionsdbus_handle_exceptionsdbus_service_method N)
DBusException)
FirewallError)errors)log)FirewallDBusExceptionNotAuthorizedException)
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 yL } z$tt t| W Y d }~n"d }~0 tyd t Y n0 d S N)r r debug1 traceback
format_excerror Exception exception)argskwargsr func >/usr/lib/python3.9/site-packages/firewall/server/decorators.py_impl0 s z handle_exceptions.._impl) functoolswrapsr r r r r r , s r c s& t fdd}t |_|S )zDecorator 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 y } zjt t|}|tjtjtjtjfv rVt t| n t
t t
t| tt|W Y d }~nfd }~0 ty } z|W Y d }~n@d }~0 ty } z t tt|W Y d }~n
d }~0 0 d S r )r get_codestrr ZALREADY_ENABLEDZ NOT_ENABLEDZZONE_ALREADY_SETZ ALREADY_SETr Zwarningr
r r r r r r r )r r r codeexr r r r @ s
z%dbus_handle_exceptions.._impl)r r inspectZ signatureZ
__signature__r r r r r ; s r c O s | dd tjj| i |S )z Add sender argument for D-BusZsender_keywordsender)
setdefaultdbusZservicemethodr r r r r r ` s r c @ s& e Zd ZdZi ZdddZdd ZdS ) dbus_service_method_deprecatedzRDecorator that maintains a list of deprecated methods in dbus
interfaces.
Nc C s* || _ | j r&| j | jvr&t | j| j < d S r ) interface
deprecatedset)selfr( 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__.._impl)r( r) add__name__r r r+ r r r r r __call__q s
z'dbus_service_method_deprecated.__call__)N)r.
__module__ __qualname____doc__r) r, r0 r r r r r' e s
r' c @ s e Zd ZdZdS ) dbus_service_signal_deprecatedzRDecorator that maintains a list of deprecated signals in dbus
interfaces.
N)r. r1 r2 r3 r r r r r4 z s r4 c @ sD e Zd ZdZdZdZdZdZdZdZ dd Z
e dd Z d
d Z
dS ) dbus_polkit_require_authzDecorator 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)r+ Zpolkit_auth_requiredr r r r, s z!dbus_polkit_require_auth.__init__c C s | j | j d | _d | _d S r )_busZremove_signal_receiver_bus_signal_receiver_interface_polkit)clsnameZ old_ownerZ new_ownerr r r _polkit_name_owner_changed s z3dbus_polkit_require_auth._polkit_name_owner_changedc s$ t fdd}j|_|S )Nc s8 t jst t _t jsHt jjt jddjdt _t jsz6t t j
t jt j t j t _W n tj
y Y n0 j}|st
d|d}|r*t jrt jdd|if|i dd \}}}|s*t|d
n$tt j|}|d kr*t|d | i |S )
NZNameOwnerChangedzorg.freedesktop.DBus)Zhandler_functionZ signal_nameZdbus_interfaceZarg0z'Not Authorized: No action_id specified.r" zsystem-bus-namer; Zpolkitr uid)typer7 r$ Z SystemBusr8 Zadd_signal_receiverr< _polkit_namer9 Z InterfaceZ
get_object _polkit_path_polkit_interfacer r6 getZCheckAuthorizationr
r )r r Z action_idr" result_r? r r+ r r r sF
z0dbus_polkit_require_auth.__call__.._impl)r r r6 r/ r rG r r0 s +z!dbus_polkit_require_auth.__call__)r. r1 r2 r3 rA rB rC r7 r8 r9 r, classmethodr< r0 r r r r r5 s
r5 )r3 __all__r$ Z dbus.servicer r r! Zdbus.exceptionsr Zfirewall.errorsr Zfirewallr Zfirewall.core.loggerr Zfirewall.server.dbusr r
Zfirewall.dbus_utilsr r r r r' r4 r5 r r r r s$ %