Current File : //usr/lib/python3.9/site-packages/cockpit/__pycache__/polkit.cpython-39.pyc
a
��j�! � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z d dl
m Z d dl m
Z
mZ dZdZdZe�e�Ze eeee
f f ZG dd � d ej�ZG d
d � d �ZdS ) � N)�Dict�List�Sequence�Tuple)�AskpassHandler)�Variant�busz /run/polkit/agent-helper.socketz'/usr/lib/polkit-1/polkit-agent-helper-1z /PolkitAgentc s� e Zd Zed�� fdd� Zej�dg d��eeee eef ee
e dd�dd ��Z eedd
�d d �Z
eedd
�d
d�Zejejdd�dd�Z� ZS )�.org_freedesktop_PolicyKit1_AuthenticationAgent�� responderc s t � �� || _d S �N)�super�__init__r ��selfr �� __class__� �2/usr/lib/python3.9/site-packages/cockpit/polkit.pyr & s
z7org_freedesktop_PolicyKit1_AuthenticationAgent.__init__� )�sr r za{ss}r z a(sa{sv})N)� action_id�message� icon_name�details�cookie�
identities�returnc � s� t �d||||||� t�� }|D ]8\}} |dkr"d| v r"| d j|kr"t �d| |� qlq"t �d|� d S t�|�j}
z| � |
|�I d H W n. t
y� t �d� | � |
|�I d H Y n0 d S )Nz[BeginAuthentication: action %s, message "%s", icon %s, details %s, cookie %s, identities %rz unix-user�uidz,Authentication subject %s matches our uid %dz*Not supporting authentication as any of %sz:Socket helper not available, falling back to legacy helper) �logger�debug�os�geteuid�value�warning�pwd�getpwuid�pw_name�_authenticate_socket�OSError�_authenticate_suid_helper) r r r r r r r Zmy_uidZ auth_type�subject� user_namer r r �begin_authentication+ s �
zCorg_freedesktop_PolicyKit1_AuthenticationAgent.begin_authentication)r, r r c � s t �dt� tjt||tjjtjjd�I dH }|jdus<J �|jdusJJ �zxz| � |j|j�I dH W n tj
y� t �d� Y n0 W z |� � W n t y� Y n0 |�
� I dH }t �d|� n<z |� � W n t y� Y n0 |�
� I dH }t �d|� 0 dS )z.Authenticate using legacy setuid helper binaryz!Trying legacy polkit helper at %s)�stdin�stdoutN�Cancelled authenticationzhelper exited with code %i)r r � HELPER_PATH�asyncioZcreate_subprocess_exec�
subprocess�PIPEr. r/ � _communicate�CancelledErrorZ terminate�ProcessLookupError�wait)r r, r Zprocess�resr r r r* E s.
� � zHorg_freedesktop_PolicyKit1_AuthenticationAgent._authenticate_suid_helperc � s� t �dt� t�t�I dH \}}z�z>|�|� d|� d��� � |�� I dH | �||�I dH W n tj y~ t �d� Y n0 W |�
� |� � I dH t �d� n"|�
� |� � I dH t �d� 0 dS )z:Authenticate using socket-activated helper (polkit >= 127)z+Trying socket-activated polkit helper at %sN�
r0 zsocket connection closed) r r �
HELPER_SOCKETr2 Zopen_unix_connection�write�encode�drainr5 r6 �closeZ wait_closed)r r, r �reader�writerr r r r( Z s �zCorg_freedesktop_PolicyKit1_AuthenticationAgent._authenticate_socket)r. r/ r c � s: g }|2 �z(3 d H W }t �d|� |�� �� �d�\}}}|�d�r�|�d�rRd}| j�d�|�|d�I d H }|� � |d u r�t �d|� t
� d��t �d |� |� |�
� � |� d
� ~|�� I d H t �d � q|d v r�|�|� q|d
k�rt �d� �q6q|dk�r t �d� �q6qt �d|� �q6q6 d S )NzRead line from helper: %s� Z
PAM_PROMPTZPasswordr r: z/got PAM_PROMPT %s, but do_askpass returned Nonezno password givenz1got PAM_PROMPT %s, do_askpass returned a password�
z(got PAM_PROMPT, wrote password to helper)Z
PAM_TEXT_INFOZ PAM_ERROR�SUCCESSzAuthentication succeeded�FAILUREzAuthentication failedz&Unknown line from helper, aborting: %s)r r �strip�decode� partition�
startswithr Z
do_askpass�join�clearr2 r6 r< r= r> �appendr$ ) r r. r/ �messages�lineZcommand�_r# �passwdr r r r5 k s8
z;org_freedesktop_PolicyKit1_AuthenticationAgent._communicate)�__name__�
__module__� __qualname__r r r Z InterfaceZMethod�strr r �Identityr- r* r( r2 � StreamWriter� StreamReaderr5 �
__classcell__r r r r r % s �r c @ s. e Zd ZdZed�dd�Zdd� Zdd� Zd S )
� PolkitAgentz}Register polkit agent when required
Use this as a context manager to ensure that the agent gets unregistered again.
r
c C s || _ d | _d S r )r �
agent_slotr r r r r � s zPolkitAgent.__init__c
� s� zt j�� | _W n4 tyD } z t�d|� | W Y d }~S d }~0 0 z ddttj d d�if| _
W n t y� t� d� | Y S 0 t
| j�}| j�t|�| _t�tjd �}| j�ddd d
d | j
|t�I d H t� d | j
|� | S )
Nz>cannot connect to system bus, not registering polkit agent: %sz unix-sessionz
session-idZXDG_SESSION_IDr z4XDG_SESSION_ID not set, not registering polkit agent�org.freedesktop.PolicyKit1�%/org/freedesktop/PolicyKit1/Authority�$org.freedesktop.PolicyKit1.AuthorityZRegisterAuthenticationAgentz
(sa{sv})ssz%Registered agent for %r and locale %s)r ZBusZdefault_system�
system_busr) r r$ r r! �environr+ �KeyErrorr r r Z
add_object�AGENT_DBUS_PATHrZ �locale� setlocale� LC_MESSAGES�call_method_async)r �eZ agent_objectZ locale_namer r r �
__aenter__� s.
�
zPolkitAgent.__aenter__c � sB | j r>| j�ddddd| jt�I d H | j �� t�d| j� d S )Nr[ r\ r] Z UnregisterAuthenticationAgentz (sa{sv})szUnregistered agent for %r)rZ r^ re r+ ra �cancelr r )r Z _exc_typeZ
_exc_valueZ
_tracebackr r r � __aexit__� s �
zPolkitAgent.__aexit__N)rQ rR rS �__doc__r r rg ri r r r r rY � s rY )r2 rb Zloggingr! r% �typingr r r r Zcockpit._vendor.fernyr Z cockpit._vendor.systemd_ctypesr r r; r1 ra Z getLoggerrQ r rT rU ZObjectr rY r r r r �<module> s
m
Mini Shell