o i6 @sTdZddlZddlZddlZddlZddlZddlZzddlmZWne y1ddl mZYnwzej Z Wne yJe defddiZ YnweeZddZd d ZGd d d eZGd ddeZGdddeZGdddeZddZGddde ZddZGdddeZGdddeZGdddeZGdd d eZdS)!aPA generic persistence layer, optionally encrypted on Windows, OSX, and Linux. Should a certain encryption is unavailable, exception will be raised at run-time, rather than at import time. By successfully creating and using a certain persistence object, app developer would naturally know whether the data are protected by encryption. N)PathABC __slots__c Csx|sdStjdkrtj|dddSzt|WdSty;}z|jtjkr/tj|r/nWYd}~dSd}~ww)zCreates a directory, and any necessary parents. If the path provided is an existing file, this function raises an exception. :param path: The directory name that should be created. N)T)exist_ok) sys version_infoosmakedirsOSErrorerrnoEEXISTpathisdir)rexprrR/home/app/Keep/.python/lib/python3.10/site-packages/msal_extensions/persistence.py_mkdir_ps rcCst|dS)Nzutf-8)hashlibsha256encode hexdigest)Z input_stringrrr _auto_hash6src"eZdZdZdfdd ZZS)PersistenceErrorz#The base exception for persistence.Ncstt||||dSN)superr__init__selferr_nomessagelocation __class__rrrCszPersistenceError.__init__NNN__name__ __module__ __qualname____doc__r __classcell__rrr$rr=srcr)PersistenceNotFoundzZThis happens when attempting BasePersistence.load() on a non-existent persistence instanceNcs tt|jtj|p d|ddS)NzPersistence not foundr!r"r#)rr-rrENOENTrr$rrrIs  zPersistenceNotFound.__init__r&r'rrr$rr-Gsr-c@eZdZdZdS)PersistenceEncryptionErrorz*This could be raised by persistence.save()Nr(r)r*r+rrrrr1Or1c@r0)PersistenceDecryptionErrorz*This could be raised by persistence.load()Nr2rrrrr4Rr3r4cCsLtjdr t|Stjdrt|Stjdrt|Stdtj)zBuild a suitable encrypted persistence instance based your current OS. If you do not need encryption, then simply use ``FilePersistence`` constructor. windarwinlinuxzUnsupported platform: {})r platform startswith!FilePersistenceWithDataProtectionKeychainPersistenceLibsecretPersistence RuntimeErrorformatr#rrrbuild_encrypted_persistenceVs   r@c@sLeZdZdZdZejddZejddZejddZ ejd d Z d S) BasePersistencezDAn abstract persistence defining the common interface of this familyFcCt)&Save the content into this persistenceNotImplementedErrorr contentrrrsavekszBasePersistence.savecCrB)zuLoad content from this persistence. Could raise PersistenceNotFound if no save() was called before. rDr rrrloadqszBasePersistence.loadcCrB)zGet the last time when this persistence has been modified. Could raise PersistenceNotFound if no save() was called before. rDrIrrrtime_last_modifiedzsz"BasePersistence.time_last_modifiedcCrB)zBReturn the file path which this persistence stores (meta)data intorDrIrrr get_locationszBasePersistence.get_locationN) r(r)r*r+ is_encryptedabcabstractmethodrHrJrKrLrrrrrAfs   rAcCst|tjtjBtjBdS)Ni)r openO_RDWRO_CREATO_TRUNCr?rrr_opensrTc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)FilePersistencez8A generic persistence, storing data in a plain-text filecCs0|stdtj||_ttj|jdS)NzRequires a file path) ValueErrorr r expanduser _locationrdirname)r r#rrrrszFilePersistence.__init__cCsBtt|jd}||WddS1swYdS)rCzw+N)r fdopenrTrXwrite)r rGhandlerrrrHs "zFilePersistence.savec Csrzt|jd }|WdWS1swYWdSty8}z|jtjkr3td|jdd}~ww)z"Load content from this persistencerNGPersistence not initialized. You can recover by calling a save() first.r"r#)rPrXreadEnvironmentErrorrr/r-)r r\rrrrrJs( zFilePersistence.loadc CsFztj|jWSty"}z|jtjkrtd|jdd}~ww)Nr^r_)r rgetmtimerXrarr/r-)r rrrrrKs z"FilePersistence.time_last_modifiedcCst|jdS)zDTo touch this file-based persistence without writing content into itN)rrXtouchrIrrrrcszFilePersistence.touchcCs|jSr)rXrIrrrrLszFilePersistence.get_locationN) r(r)r*r+rrHrJrKrcrLrrrrrUs rUcs6eZdZdZdZd fdd ZddZdd ZZS) r:zcA generic persistence with data stored in a file, protected by Win32 encryption APIs on WindowsTcs,ddlm}||d|_tt||dS)z7Initialization could fail due to unsatisfied dependency)WindowsDataProtectionAgent)entropyN)windowsrf _dp_agentrr:r)r r#rgrfr$rrrs  z*FilePersistenceWithDataProtection.__init__c Csz|j|}Wnty!}z tt|ddd|dd}~wwtt|j d}| |WddS1sr rZrTrXr[)r rGdata exceptionr\rrrrHs  "z&FilePersistenceWithDataProtection.savec Cszt|jd }|}Wdn1swYWnty;}z|jtjkr1td|jdtdd}~wwz|j |WSt y^}zt t |ddd||jdd}~ww)Nrbr^r_zDPAPI error likely caused by file content not previously encrypted. App developer should migrate by calling save(plaintext) first.rjzDecryption failed: {} App developer may consider this guidance: https://github.com/AzureAD/microsoft-authentication-extensions-for-python/wiki/PersistenceDecryptionErrorr.)rPrXr`rarr/r-loggerrmriZ unprotectr r4rkr>)r r\rlrrmrrrrJs8    z&FilePersistenceWithDataProtection.load)rd) r(r)r*r+rMrrHrJr,rrr$rr:s  r:c@>eZdZdZdZdddZddZdd Zd d Zd d Z dS)r;z`A generic persistence with data stored in, and protected by native Keychain libraries on OSXTNcCsFddlm}m}t||_||_||_d}|p||_|pt||_ dS)zInitialization could fail due to unsatisfied dependency. :param signal_location: See :func:`persistence.LibsecretPersistence.__init__` re)Keychain KeychainErrorzmsal-extensionsN) ZosxrqrrrU_file_persistence _Keychain_KeychainError _service_namer _account_name)r signal_locationZ service_nameZ account_namerqrrZdefault_service_namerrrrs  zKeychainPersistence.__init__cCsH|}||j|j|Wdn1swY|jdSr)rtZset_generic_passwordrvrwrsrc)r rGlockerrrrrH s  zKeychainPersistence.savec Cs|3}z||j|jWWdS|jy6}z|j|jjkr1td|j|jddd}~ww1s:wYdS)NzService:{} Account:{}zMKeychain persistence not initialized. You can recover by call a save() first.)r#r") rtZget_generic_passwordrvrwruZ exit_statusZITEM_NOT_FOUNDr-r>)r ryexrrrrJs$ zKeychainPersistence.loadcC |jSrrsrKrIrrrrK" z&KeychainPersistence.time_last_modifiedcCr{rrsrLrIrrrrL%r}z KeychainPersistence.get_locationNN r(r)r*r+rMrrHrJrKrLrrrrr;s   r;c@rp)r<zcA generic persistence with data stored in, and protected by native libsecret libraries on LinuxTNcKsDddlm}m}|||pt||pifi||_t||_dS)aQInitialization could fail due to unsatisfied dependency. :param string signal_location: Besides saving the real payload into encrypted storage, this class will also touch this signal file. Applications may listen a FileSystemWatcher.Changed event for reload. https://docs.microsoft.com/en-us/dotnet/api/system.io.filesystemwatcher.changed?view=netframework-4.8#remarks :param string schema_name: See :func:`libsecret.LibSecretAgent.__init__` :param dict attributes: See :func:`libsecret.LibSecretAgent.__init__` re)LibSecretAgent trial_runN)Z libsecretrrr_agentrUrs)r rxZ schema_name attributeskwargsrrrrrr.s zLibsecretPersistence.__init__cCs|j|r |jdSdSr)rrHrsrcrFrrrrHAs zLibsecretPersistence.savecCs |j}|durtdd|S)NzLKeyring persistence not initialized. You can recover by call a save() first.)r")rrJr-)r rlrrrrJEs zLibsecretPersistence.loadcCr{rr|rIrrrrKOr}z'LibsecretPersistence.time_last_modifiedcCr{rr~rIrrrrLRr}z!LibsecretPersistence.get_locationrrrrrrr<)s  r<) r+rNr rrloggingr pathlibr ImportErrorZpathlib2rAttributeErrorABCMetaobject getLoggerr(rorrIOErrorrr-r1r4r@rArTrUr:r;r<rrrrs>     "561