U hWs@sdZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z m Z ddlmZmZddlmZmZddlmZmZmZddlmZddlmZd Ze d ZGd d d ejZdd dZ ddZ!ddZ"ddZ#ddZ$ddZ%dddZ&ddZ'dS)zkThis module provides some utility functions, but these shouldn't normally be used by external applications.N) DBUS_UNKNOWN_METHODDBUS_NO_SUCH_OBJECTDBUS_SERVICE_UNKNOWN DBUS_NO_REPLYDBUS_NOT_SUPPORTEDDBUS_EXEC_FAILEDSS_PATH SS_PREFIX ALGORITHM_DHALGORITHM_PLAIN)Session int_to_bytes)ItemNotFoundException"SecretServiceNotAvailableException)Cipher algorithmsmodes)default_backend)int_from_byteszorg.freedesktop.secretsServicec@s eZdZdZddZddZdS)InterfaceWrapperzxWraps :cls:`dbus.Interface` class and replaces some D-Bus exceptions with :doc:`SecretStorage exceptions `.csfdd}|S)Nc sz ||WStjjk r|}zN|tkr6td|tkrNt||tt fkrjt |W5d}~XYnXdS)NzItem does not exist!) dbus exceptions DBusException get_dbus_namerrrget_dbus_messagerrr)argskwargse function_in4/usr/lib/python3/dist-packages/secretstorage/util.py function_outs    z3InterfaceWrapper.catch_errors..function_outr!)selfr r#r!rr" catch_errorss zInterfaceWrapper.catch_errorscCs$tj||}t|r ||}|SN)r Interface __getattr__callabler%)r$Z attributeresultr!r!r"r(+s zInterfaceWrapper.__getattr__N)__name__ __module__ __qualname____doc__r%r(r!r!r!r"rsrc Csf|pt}z|j||ddWStjjk r`}z$|tttfkrNt | W5d}~XYnXdS)zA wrapper around :meth:`SessionBus.get_object` that raises :exc:`~secretstorage.exceptions.SecretServiceNotAvailableException` when appropriate.F)Z introspectN) BUS_NAMEZ get_objectrrrrrrrrr)bus object_pathZ service_namenamerr!r!r"bus_get_object1s  r3c Cst|t}t|td}t}z$|jttt |j dd\}}WnNtj j k r}z,| tkrh|jtddd\}}d|_W5d}~XYnXtt|d}||||_|S)z%Returns a new Secret Service session.rZsvZ signatureFNZbig)r3rrr'r r Z OpenSessionr ByteArrayr Z my_public_keyrrrrr encryptedr bytearrayZset_server_public_keyr1)r0 service_obj service_ifacesessionoutputr*rr!r!r" open_session>s,    r=cCst|ts|d}|js4t|jdt||fSdt|d@}|tt |f|7}t d}t |j}t|t|t}|||}t|jt|tt ||fS)zHFormats `secret` to make possible to pass it to the Secret Service API.zutf-8r5) isinstancebytesencoder7rZStructr1r6lenr8osurandomrZAESZaes_keyrrZCBCr encryptorupdatefinalizeArray)r;ZsecretZ content_typeZpaddingZaes_ivZaesrFZencrypted_secretr!r!r" format_secretXs&      rJcsDt||}t|td}|jdddfdd}|d|dS) aExecutes 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.Promptr5sr4cs&t|tjrt|}t||dSr&)r@rrIlistboolZ dismissedZunlockedcallbackr!r" new_callbackvs z!exec_prompt..new_callbackZ CompletedN)r3rr'r rKZconnect_to_signal)r0promptrQZ prompt_objZ prompt_ifacerRr!rPr" exec_promptns   rTcsJddlm}|gfdd}t|||ddfS)znLike :func:`exec_prompt`, but synchronous (uses loop from GLib API). Returns (*dismissed*, *unlocked*) tuple.r)GLibcs ||dSr&appendquitrOZloopr*r!r"rQs  z"exec_prompt_glib..callback)Z gi.repositoryrUZMainLooprTrun)r0rSrUrQr!rYr"exec_prompt_glib|s  r\csJddlm}|ggfdd}t|||ddfS)zoLike :func:`exec_prompt`, but synchronous (uses loop from PyQt5 API). Returns (*dismissed*, *unlocked*) tuple.r)QCoreApplicationcs ||dSr&rVrOZappr*r!r"rQs  z exec_prompt_qt..callbackrZ)Z PyQt5.QtCorer]rTZexec_)r0rSr]rQr!r^r"exec_prompt_qts  r_cCsnt|t}t|t}|j|dd\}}t|}t|dkr\|rLt|||qjt||dSn|rj|d|dS)a0Requests 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.2Zaor4rZrFN) r3rr SERVICE_IFACEZUnlockrMrCrTr\)r0pathsrQr9r:Zunlocked_pathsrSr!r!r"unlock_objectss   rbcCs,z t|WStk r&t|YSXdS)z(Converts D-Bus string to unicode string.N)Zunicode NameErrorstr)stringr!r!r" to_unicodes rf)N)N)(r.rrDZsecretstorage.definesrrrrrrrr r r Zsecretstorage.dhcryptor r Zsecretstorage.exceptionsrrZ&cryptography.hazmat.primitives.ciphersrrrZcryptography.hazmat.backendsrZcryptography.utilsrr/r`r'rr3r=rJrTr\r_rbrfr!r!r!r"s&0