File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/win_functions.cpython-310.pyc
Back
o ;jb= � @ sT d Z ddlZddlZddlZddlmZ z"ddlZddlZddlZddl Z ddl m Z m Z m Z ddlZdZW n1 eyg z ddlZddlmZmZmZm Z ddl m Z m Z m Z W n eyd dZY nw Y nw dd � Zd d � Zd d � Zdd� Zd'dd�Zdd� Zd(dd�Zdd� Zdd� Zd(dd�Zd d � Z d)d d �Z d!d"� Z d#d$� Z d%d&� Z dS )*zq Various functions to be used by windows during start up and to monkey patch missing functions in other modules. � N)�CommandExecutionError)�HWND_BROADCAST�SMTO_ABORTIFHUNG�WM_SETTINGCHANGET)� pywintypes�win32api�win32net� win32securityFc C s t sdS dS )z4 Only load if Win32 Libraries are installed )Fz This utility requires pywin32Z win_functions)� HAS_WIN32� r r �L/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/win_functions.py� __virtual__! s r c C s t �� �� S )z� This is a monkey patch for os.getppid. Used in: - salt.utils.parsers Returns: int: The parent process id )�psutilZProcessZppidr r r r �get_parent_pid+ s r c C s& t | d�}|D ] }|dv r dS qdS )z� Is the passed user a member of the Administrators group Args: name (str): The name to check Returns: bool: True if user is a member of the Administrators group, False otherwise T)z S-1-5-32-544�S-1-5-18F)�get_user_groups)�name�groups�groupr r r �is_admin6 s �r c C sB zddl } | jjjdddd�}|�d�dkW S ty Y d S w ) a_ Detect whether the current machine is a Windows Domain Controller. Reads ``HKLM\SYSTEM\CurrentControlSet\Control\ProductOptions\ProductType`` directly. A value of ``LanmanNT`` indicates a Domain Controller. Returns: bool: ``True`` if the machine is a Domain Controller, ``False`` otherwise (including on error). r NZHKLMz/SYSTEM\CurrentControlSet\Control\ProductOptionsZ ProductType)Zhive�keyZvnameZvdataZLanmanNTF)Zsalt.utils.win_reg�utilsZwin_regZ read_value�get� Exception)�salt�resultr r r �is_domain_controllerJ s � �r c C s g }| � � dkr dg}nkzt�d| �}W nb tjtjfyv } zR|jdv r-t�d| d�}n?zt�d| �}W n6 tjyT } z|jdv rJt�d| d�}W Y d}~n d}~w tjyk |jdv rht�d| d�}n� Y nw W Y d}~nd}~ww |s{|S g }|D ] }|�t|�� q|S )a Get the groups to which a user belongs Args: name (str): The user name to query sid (bool): True will return a list of SIDs, False will return a list of group names Returns: list: A list of group names or sids �SYSTEMN)� i� i� i� iK r � ) �upperr ZNetUserGetLocalGroups�errorr �winerrorZNetUserGetGroups�append�get_sid_from_name)r �sidr �excZ ret_groupsr r r r r e s8 � ����r c C sZ | du rd} z t �d| �d }W n tjy' } z td| � d|� ���d}~ww t �|�S )z� This is a tool for getting a sid from a name. The name can be any object. Usually a user or a group Args: name (str): The name of the user or group for which to get the sid Returns: str: The corresponding SID NzNULL SIDr zUser z not found: )r �LookupAccountNamer r! r �ConvertSidToStringSid)r r% r& r r r r$ � s �� r$ c C s� z(t �t j�}|d dkr!t �� }|dkrd}nt|�dkr d}n| s't �� }W n tjy< } ztd|� ���d}~ww |sAdS |S )z� Gets the user executing the process Args: with_domain (bool): ``True`` will prepend the user name with the machine name or domain separated by a backslash Returns: str: The user name ����$r r z Failed to get current user: NF)r Z GetUserNameExZNameSamCompatibleZ GetUserNamer$ r r! r )Z with_domainZ user_nameZ test_userr& r r r �get_current_user� s& ����r+ c C s� dg}z t �d| �d }W n tjy& d�t�� dd� �� | g� Y S w t �|�}t � d|�\} }}||v r;| S d�|| g�S )a� Gets the SAM name for a user. It basically prefixes a username without a backslash with the computer name. If the user does not exist, a SAM compatible name will be returned using the local hostname as the domain. i.e. salt.utils.get_same_name('Administrator') would return 'DOMAIN.COM\Administrator' .. note:: Long computer names are truncated to 15 characters zS-1-1-0Nr �\� ) r r' r r! �join�platform�noder r( ZLookupAccountSid)�usernameZspecial_id_groupsZsid_objr% �domain�_r r r � get_sam_name� s "� r4 c s$ t rd� t�� fdd�d� dS dS )z0 Set the control handler on the console r c s | � krdS dS )NTFr )�event�Zctrl_logoff_eventr r �<lambda>� s z,enable_ctrl_logoff_handler.<locals>.<lambda>r N)r r ZSetConsoleCtrlHandlerr r r6 r �enable_ctrl_logoff_handler� s ��r8 c C s4 | rt �d| �rd| �dd� d } |s| S t| �S )a� Escape the argument for the cmd.exe shell. See http://blogs.msdn.com/b/twistylittlepassagesallalike/archive/2011/04/23/everyone-quotes-arguments-the-wrong-way.aspx First we escape the quote chars to produce a argument suitable for CommandLineToArgvW. We don't need to do this for simple arguments. Args: arg (str): a single command line argument to escape for the cmd.exe shell Kwargs: escape (bool): True will call the escape_for_cmd_exe() function which escapes the characters '()%!^"<>&|'. False will not call the function and only quotes the cmd Returns: str: an escaped string suitable to be passed as a program argument to the cmd.exe shell z(["\s])�"�\")�re�search�replace�escape_for_cmd_exe)�arg�escaper r r �escape_argument� s rA c sP d}t �dd�dd� t|�D �� d �}dd� |D �� � fd d �}|�|| �S ) ag Escape an argument string to be suitable to be passed to cmd.exe on Windows This method takes an argument that is expected to already be properly escaped for the receiving program to be properly parsed. This argument will be further escaped to pass the interpolation performed by cmd.exe unchanged. Any meta-characters will be escaped, removing the ability to e.g. use redirects or variables. Args: arg (str): a single command line argument to escape for cmd.exe Returns: str: an escaped string suitable to be passed as a program argument to cmd.exe z ()%!^"<>&|�(�|c s s � | ]}t �|�V qd S )N)r; r@ ��.0�charr r r � <genexpr>+ s � z%escape_for_cmd_exe.<locals>.<genexpr>�)c S s i | ]}|d |� ��qS )�^r rD r r r � <dictcomp>- s z&escape_for_cmd_exe.<locals>.<dictcomp>c s | � d�}� | S )Nr )r )�mrF �Zmeta_mapr r �escape_meta_chars/ s z-escape_for_cmd_exe.<locals>.escape_meta_chars)r; �compiler. �list�sub)r? Z meta_charsZmeta_rerM r rL r r> s � r> � Environmentc C s6 t �| �}t jddd�}|�ttd|tdd�}|dkS )a Send a WM_SETTINGCHANGE Broadcast to all Windows Args: message (str): A string value representing the portion of the system that has been updated and needs to be refreshed. Default is ``Environment``. These are some common values: - "Environment" : to effect a change in the environment variables - "intl" : to effect a change in locale settings - "Policy" : to effect a change in Group Policy Settings - a leaf node in the registry - the name of a section in the ``Win.ini`` file See lParam within msdn docs for `WM_SETTINGCHANGE <https://msdn.microsoft.com/en-us/library/ms725497%28VS.85%29.aspx>`_ for more information on Broadcasting Messages. See GWL_WNDPROC within msdn docs for `SetWindowLong <https://msdn.microsoft.com/en-us/library/windows/desktop/ms633591(v=vs.85).aspx>`_ for information on how to retrieve those messages. .. note:: This will only affect new processes that aren't launched by services. To apply changes to the path or registry to services, the host must be restarted. The ``salt-minion``, if running as a service, will not see changes to the environment until the system is restarted. Services inherit their environment from ``services.exe`` which does not respond to messaging events. See `MSDN Documentation <https://support.microsoft.com/en-us/help/821761/changes-that-you-make-to-environment-variables-do-not-affect-services>`_ for more information. CLI Example: .. code-block:: python import salt.utils.win_functions salt.utils.win_functions.broadcast_setting_change('Environment') �user32T)�use_last_errorr i� r )�ctypes�create_unicode_bufferZWinDLLZSendMessageTimeoutWr r r )�messageZbroadcast_messagerR r r r r �broadcast_setting_change6 s =� rW c C sL t �d�}|�| �}d}|dur$tdd�D ]}||�|�ddd� 7 }q|S )a� Converts a GUID to a compressed guid (SQUID) Each Guid has 5 parts separated by '-'. For the first three each one will be totally reversed, and for the remaining two each one will be reversed by every other character. Then the final compressed Guid will be constructed by concatenating all the reversed parts without '-'. .. Example:: Input: 2BE0FA87-5B36-43CF-95C8-C68D6673FB94 Reversed: 78AF0EB2-63B5-FC34-598C-6CD86637BF49 Final Compressed Guid: 78AF0EB263B5FC34598C6CD86637BF49 Args: guid (str): A valid GUID Returns: str: A valid compressed GUID (SQUID) zO^\{(\w{8})-(\w{4})-(\w{4})-(\w\w)(\w\w)-(\w\w)(\w\w)(\w\w)(\w\w)(\w\w)(\w\w)\}$� Nr � r) )r; rN �match�ranger )�guidZ guid_patternZ guid_match�squid�indexr r r � guid_to_squid� s � r_ c C s� t �d�}|�| �}d}|durdd|�d�ddd� d |�d�ddd� d |�d �ddd� d |�d �ddd� |�d �ddd� d }td d �D ]}||�|�ddd� 7 }qQ|d7 }|S )z� Converts a compressed GUID (SQUID) back into a GUID Args: squid (str): A valid compressed GUID Returns: str: A valid GUID zG^(\w{8})(\w{4})(\w{4})(\w\w)(\w\w)(\w\w)(\w\w)(\w\w)(\w\w)(\w\w)(\w\w)$rX N�{r r) �-� � � r � rY �})r; rN rZ r r[ )r] Z squid_patternZ squid_matchr\ r^ r r r � squid_to_guid� s< � �������� �� rg c C s� d}g }d}t �|| �D ]O\}}}}}} } |rn=|r |d }n6| s#|r6|dur,|�|� |r3|�|� d}q | r<td��|rS|�dd��dd�}td krR|�d d�}n|}|pXd | }q |dure|�|� |S ) z� Windows version of shlex.split() Based on winshlex: https://github.com/jdjebi/winshlex/blob/master/winshlex/lex.py z["((?:""|\\["\\]|[^"])*)"?()|(\\\\(?=\\*")|\\")|(&&?|\|\|?|\d?>|[<])|([^\s"&|<>]+)|(\s+)|(.)Nr z"invalid or incomplete shell stringr: r9 z\\r, r z""rX )r; �findallr# � ValueErrorr= r/ ) �stringZ re_cmd_lex�argsZaccu�qsZqss�esc�pipe�wordZwhiteZfailr r r � shlex_split� s6 � rp )F)T)rQ )!�__doc__rT r/ r; Zsalt.exceptionsr r r r r Zwin32conr r r r r � ImportError�win32r r r r r r$ r+ r4 r8 rA r> rW r_ rg rp r r r r �<module> sL ��� 4 " K! #
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings