File: //proc/thread-self/root/lib/python3/dist-packages/secretstorage/__pycache__/util.cpython-38.pyc
U
h��Ws � @ s� d Z ddlZddlZddlmZmZmZmZmZm Z m
Z
m Z m Z m
Z
ddlmZmZ ddlmZmZ ddlmZmZmZ ddlmZ ddlmZ d Z e d
Z G d d � d ej �Z d d
d�Z dd� Z!dd� Z"dd� Z#dd� Z$dd� Z%d dd�Z&dd � Z'dS ) zkThis module provides some utility functions, but these shouldn't
normally be used by external applications.� N)
�DBUS_UNKNOWN_METHOD�DBUS_NO_SUCH_OBJECT�DBUS_SERVICE_UNKNOWN�
DBUS_NO_REPLY�DBUS_NOT_SUPPORTED�DBUS_EXEC_FAILED�SS_PATH� SS_PREFIX� ALGORITHM_DH�ALGORITHM_PLAIN)�Session� int_to_bytes)�ItemNotFoundException�"SecretServiceNotAvailableException)�Cipher�
algorithms�modes)�default_backend)�int_from_byteszorg.freedesktop.secrets�Servicec @ s e Zd ZdZdd� Zdd� ZdS )�InterfaceWrapperzxWraps :cls:`dbus.Interface` class and replaces some D-Bus exceptions
with :doc:`SecretStorage exceptions <exceptions>`.c s � fdd�}|S )Nc
s� z � | |�W S t jjk
r| } zN|�� tkr6td��|�� tkrNt|�� ��|�� tt fkrjt
|�� ��� W 5 d }~X Y nX d S )NzItem does not exist!) �dbus�
exceptions�
DBusException�
get_dbus_namer r r �get_dbus_messager r r )�args�kwargs�e�� function_in� �4/usr/lib/python3/dist-packages/secretstorage/util.py� function_out s �z3InterfaceWrapper.catch_errors.<locals>.function_outr! )�selfr r# r! r r" � catch_errors s z InterfaceWrapper.catch_errorsc C s$ t j�| |�}t|�r | �|�}|S �N)r � Interface� __getattr__�callabler% )r$ Z attribute�resultr! r! r" r( + s
z InterfaceWrapper.__getattr__N)�__name__�
__module__� __qualname__�__doc__r% r( r! r! r! r" r s r c
C sf |pt }z| j||dd�W S tjjk
r` } z$|�� tttfkrNt |�
� ��� W 5 d}~X Y nX dS )z�A wrapper around :meth:`SessionBus.get_object` that raises
:exc:`~secretstorage.exceptions.SecretServiceNotAvailableException`
when appropriate.F)Z
introspectN) �BUS_NAMEZ
get_objectr r r r r r r r r )�bus� object_pathZ service_name�namer r! r! r" �bus_get_object1 s
� r3 c
C s� t | t�}t�|td �}t� }z$|jtt�t |j
��dd�\}}W nN tj j k
r� } z,|�
� tkrh� |jtddd�\}}d|_W 5 d}~X Y nX tt|�d�}|�|� ||_|S )z%Returns a new Secret Service session.r Zsv�Z signature� FNZbig)r3 r r r' r r Z OpenSessionr
� ByteArrayr
Z
my_public_keyr r r r r � encryptedr � bytearrayZset_server_public_keyr1 )r0 � service_obj�
service_iface�session�outputr* r r! r! r" � open_session> s,
� �
r= c C s� t |t�s|�d�}| js4t�| jdt�|�|f�S dt|�d@ }|tt |f�| �7 }t
� d�}t �
| j�}t|t�|�t� ��� }|�|�|�� }t�| jt�|�t�t |��|f�S )zHFormats `secret` to make possible to pass it to the
Secret Service API.zutf-8r5 � � )�
isinstance�bytes�encoder7 r ZStructr1 r6 �lenr8 �os�urandomr ZAESZaes_keyr r ZCBCr � encryptor�update�finalize�Array)r; ZsecretZ content_typeZpaddingZaes_ivZaesrF Zencrypted_secretr! r! r" �
format_secretX s&
�
�rJ c sD t | |�}t�|td �}|jddd� � fdd�}|�d|� dS ) a Executes the given `prompt`, when complete calls `callback`
function with two arguments: a boolean representing whether the
operation was dismissed and a list of unlocked item paths. A main
loop should be running and registered for this function to work.�Promptr5 �sr4 c s& t |tj�rt|�}� t| �|� d S r& )r@ r rI �list�bool�Z dismissedZunlocked��callbackr! r" � new_callbackv s z!exec_prompt.<locals>.new_callbackZ CompletedN)r3 r r' r rK Zconnect_to_signal)r0 �promptrQ Z
prompt_objZ prompt_ifacerR r! rP r" � exec_promptn s
rT c sJ ddl m} |�� � g �� �fdd�}t| ||� � �� �d �d fS )znLike :func:`exec_prompt`, but synchronous (uses loop from GLib
API). Returns (*dismissed*, *unlocked*) tuple.r )�GLibc s �� | � �� |� � �� d S r& ��append�quitrO �Zloopr* r! r" rQ � s
z"exec_prompt_glib.<locals>.callback� )Z
gi.repositoryrU ZMainLooprT �run)r0 rS rU rQ r! rY r" �exec_prompt_glib| s r\ c sJ ddl m} |g �� g �� �fdd�}t| ||� � �� �d �d fS )zoLike :func:`exec_prompt`, but synchronous (uses loop from PyQt5
API). Returns (*dismissed*, *unlocked*) tuple.r )�QCoreApplicationc s �� | � �� |� � �� d S r& rV rO �Zappr* r! r" rQ � s
z exec_prompt_qt.<locals>.callbackrZ )Z PyQt5.QtCorer] rT Zexec_)r0 rS r] rQ r! r^ r" �exec_prompt_qt� s r_ c C sn t | t�}t|t�}|j|dd�\}}t|�}t|�dkr\|rLt| ||� qjt| |�d S n|rj|d|� dS )a0 Requests unlocking objects specified in `paths`. If `callback`
is specified, calls it when unlocking is complete (see
:func:`exec_prompt` description for details).
Otherwise, uses the loop from GLib API and returns a boolean
representing whether the operation was dismissed.
.. versionadded:: 2.1.2Zaor4 rZ r FN) r3 r r �
SERVICE_IFACEZUnlockrM rC rT r\ )r0 �pathsrQ r9 r: Zunlocked_pathsrS r! r! r" �unlock_objects� s
rb c C s, z
t | �W S tk
r& t| � Y S X dS )z(Converts D-Bus string to unicode string.N)Zunicode� NameError�str)�stringr! r! r" �
to_unicode� s
rf )N)N)(r. r rD Zsecretstorage.definesr r r r r r r r r
r Zsecretstorage.dhcryptor r
Zsecretstorage.exceptionsr r Z&cryptography.hazmat.primitives.ciphersr r r Z cryptography.hazmat.backendsr Zcryptography.utilsr r/ r` r' r r3 r= rJ rT r\ r_ rb rf r! r! r! r" �<module> s& 0