U ÓZj9_ã@s`dZddlmZddlZddlZddlZz     þ     r)cCs:trt ||¡Stj}ztt_t ||¡W¢S|t_XdS)a3 Rewrite of original os.makedirs function, that will ensure on Windows that given mode is correctly applied. :param str file_path: The file path to open :param int mode: POSIX mode to apply on leaf directory when created, Python defaults will be applied if ``None`` N)rrÚmakedirsÚmkdir)r r Z orig_mkdir_fnr r rrAÅs  rAc Cs¦trt ||¡St ¡}|j}tƒ}t||ƒ}| |d¡|  d|d¡zt   ||¡WnHt j k r }z(|jtjkrŒttj|j||jƒ‚|‚W5d}~XYnXdS)a+ Rewrite of original os.mkdir function, that will ensure on Windows that given mode is correctly applied. :param str file_path: The file path to open :param int mode: POSIX mode to apply on directory when created, Python defaults will be applied if ``None`` Fr(rN)rrrBrr-r.r"r/r0r1r3ZCreateDirectoryr4r5r6ZERROR_ALREADY_EXISTSr7r8r9r:)r r r=r#r$r>r@r r rrBÛs     rBcCs,ttdƒrttdƒ||ƒn t ||¡dS)z´ Rename a file to a destination path and handles situations where the destination exists. :param str src: The current file path. :param str dst: The new file path. ÚreplaceN)ÚhasattrrÚgetattrÚrename)rrr r rrCús rCcCs |}tr2tj |¡}tj |¡r.td |¡ƒ‚|Sg}tj |¡r”|}t |¡}tj |¡srtj  tj  |¡|¡}||krˆtd |¡ƒ‚|  |¡q6tj  |¡S)z² Find the real path for the given path. This method resolves symlinks, including recursive symlinks, and is protected against symlinks that creates an infinite loop. zError, link {0} is a loop!) rrÚpathÚrealpathÚislinkÚ RuntimeErrorÚformatÚreadlinkÚisabsÚjoinÚdirnameÚappendÚabspath)r Z original_pathrGZinspected_pathsÚ link_pathr r rrH s       rHcCs&trtj |¡ot |tj¡St|ƒS)zˆ Is path an executable file? :param str path: path to test :return: True if path is an executable file :rtype: bool )rrrGÚisfileÚaccessÚX_OKÚ_win_is_executable)rGr r rÚ is_executable/srWcCsVtr tt t |¡j¡tj@ƒSt |tj ¡}|  ¡}t|  tj tj t d¡dœ¡ƒS)zÐ Check if everybody/world has any right (read/write/execute) on a file given its path :param str path: path to test :return: True if everybody/world has any right to the file :rtype: bool úS-1-1-0©Z TrusteeFormZ TrusteeTypeZ Identifier)rÚboolrrrrÚS_IRWXOrrÚDACL_SECURITY_INFORMATIONÚGetSecurityDescriptorDaclÚGetEffectiveRightsFromAclÚTRUSTEE_IS_SIDÚTRUSTEE_IS_USERÚConvertStringSidToSid)rGr#r>r r rÚhas_world_permissions=sýrbcCs:tr6t t |¡j¡tjtjBtjBtjB@}||BS|S)a Calculate the POSIX mode to apply to a private key given the previous private key :param str old_key: path to the previous private key :param int base_mode: the minimum modes to apply to a private key :return: the POSIX mode to apply :rtype: int ) rrrrrÚS_IRGRPÚS_IWGRPÚS_IXGRPÚS_IROTH)Úold_keyZ base_modeZold_moder r rÚcompute_private_key_modeRs ÿrhcCsdtr0t |¡}t |¡}|j|jf|j|jfkSt |tj¡}| ¡}t |tj¡}| ¡}||kS)ar Return True if the ownership of two files given their respective path is the same. On Windows, ownership is checked against owner only, since files do not have a group owner. :param str path1: path to the first file :param str path2: path to the second file :return: True if both files have the same ownership, False otherwise :rtype: bool ) rrrrrrrr r!)Zpath1Zpath2Zstats1Zstats2Z security1Zuser1Z security2Zuser2r r rÚhas_same_ownershipgs   ric Csªtrt |¡j}|||BkSt|ƒ}t |tjtjB¡}|  ¡}|  ¡}t ||ƒ}t |  ¡ƒD]F}| |¡}|d} |d}| tjtj|dœ¡} | | | Bkr^dSq^dS)a“ Check if a file given its path has at least the permissions defined by the given minimal mode. On Windows, group permissions are ignored since files do not have a group owner. :param str path: path to the file to check :param int min_mode: the minimal permissions expected :return: True if the file matches the minimal permissions expectations, False otherwise :rtype: bool r(érYFT)rrrrrHrrr r\r!r]r/ÚrangeÚ GetAceCountÚGetAcer^r_r`) rGZmin_moderr#r$r>Zmin_daclÚindexZmin_aceÚmaskZeffective_maskr r rÚhas_min_permissions€s.    ÿ  ý rpcCsNtj |¡sdSt |tj¡}| ¡}| tjtj t ƒdœ¡}|t j @t j kS)NFrY) rrGrSrrr\r]r^r_r`r"Ú ntsecurityconÚFILE_GENERIC_EXECUTE)rGr#r>r r r rrV­s ýrVcCsJt|ƒ}t |tj¡}| ¡}t||ƒ}| d|d¡t |tj|¡dS)zà This function converts the given POSIX mode into a Windows ACL list, and applies it to the file given its path. If the given path is a symbolic link, it will resolved to apply the mode on the targeted file. r(rN) rHrrr r!r/r1ÚSetFileSecurityr\)r r r#r$r>r r rr ½s  r c Cs®t|ƒ}t d¡}t d¡}t d¡}t ¡}|||fkrZt|dƒ}|rZ| tj||¡t|dƒ}|rz| tj||¡tddddœƒ} | tj| |¡| tj| |¡|S)NzS-1-5-18z S-1-5-32-544rXr$ÚallT©ÚreadÚwriteÚexecute)Ú _analyze_moderraZACLÚ_generate_windows_flagsZAddAccessAllowedAceZ ACL_REVISION) Zuser_sidr ZanalysisÚsystemZadminsZeveryoner>Z user_flagsZeverybody_flagsZfull_permissionsr r rr/Ðs       r/cCs>|tj@|tj@|tj@dœ|tj@|tj@|tj@dœdœS)Nru)r$rt)rÚS_IRUSRÚS_IWUSRÚS_IXUSRrfÚS_IWOTHÚS_IXOTH)r r r rryòsýýúrycCsDt |tj¡}| ¡}t |tj¡}| |d¡t |tj|¡dS©NF)rrr r!r0rs)rrZ security_srcZuser_srcZ security_dstr r rrs  rcCsJd}|dr|tjB}|dr4|tjtjAtjAB}|drF|tjB}|S)Nrrvrwrx)rqZFILE_GENERIC_READZFILE_ALL_ACCESSrr)Z rights_descÚflagr r rrz s ÿþ rzcCsHt|ƒ}t |tjtjB¡}| ¡}| ¡}|s4dSt||ƒ}t||ƒSr) rHrrr r\r]r!r/Ú_compare_dacls)r r r#r>r$Zref_daclr r rr,s ÿ rcs4‡fdd„tˆ ¡ƒDƒ‡fdd„tˆ ¡ƒDƒkS)z¥ This method compare the two given DACLs to check if they are identical. Identical means here that they contains the same set of ACEs in the same order. csg|]}ˆ |¡‘qSr ©rm©Ú.0rn)Údacl1r rÚ Gsz"_compare_dacls..csg|]}ˆ |¡‘qSr r„r…)Údacl2r rrˆHs)rkrl©r‡r‰r rŠrrƒBsÿrƒcCst tj¡}t d|¡dS)z= Return the pySID corresponding to the current user. Nr)Úwin32apiZ GetUserNameExZNameSamCompatiblerZLookupAccountName)Z account_namer r rr"Ks r")r')r')r')+Ú__doc__Ú __future__rr8rrrqrr,r‹r3r4r6Ú ImportErrorrZacme.magic_typingrrrrrrr%r&r)rArBrCrHrWrbrhrirprVr r/ryrrzrrƒr"r r r rÚsP    E  %-"