Current File : //lib/python3.9/site-packages/cockpit/channels/__pycache__/filesystem.cpython-39.pyc
a
��h�c � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
mZ d dlmZmZmZ d dlmZmZ d dlmZ d dlmZ ddl m Z m Z m Z m Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, e�-e.�Z/e�0� e1d
�d d ��Z2ej3e4d
�dd�Z5ddd�dd�Z6e1dd�dd�Z7G dd� de �Z8G dd� de �Z9G d d � d �Z:G d d � d e �Z;G d d!� d!e e�Z<G d"d#� d#ej=�Z>G d$d%� d%e e�Z?dS )&� N)�Path)�Callable� Generator�Iterable)�Handle� PathWatch)�Event)�Listener� )� AsyncChannel�Channel� ChannelError�GeneratorChannel) �JsonDict� JsonDocument� JsonError�
JsonObject�get_bool�get_int�
get_object�get_str�get_str_or_int�get_strv�json_merge_and_filter_patch��returnc C s0 t �dtd��� t j�} | r,t| �d�d�p.dS )Nz^Umask:\s*0([0-7]*)$z/proc/self/status� � �? )�re�searchr � read_text�M�int�group)�match� r& �?/usr/lib/python3.9/site-packages/cockpit/channels/filesystem.py�my_umask8 s r( )�bufr c
C s. d| j � d| j� d| jd�d| j� d| j� �
S )Nz1:�-�o)�st_ino�st_mtime�st_mode�st_uid�st_gid�r) r& r& r' �
tag_from_stat> s r2 zint | str | Path�
str | None)�pathr c C s< zt t�| ��W S ty$ Y dS ty6 Y d S 0 d S )Nr* )r2 �os�stat�FileNotFoundError�OSError)r4 r& r& r' �
tag_from_pathB s r9 ��fdr c C s* zt t�| ��W S ty$ Y d S 0 d S �N)r2 r5 �fstatr8 )r; r& r& r' � tag_from_fdK s r> c @ s2 e Zd ZdZeddd�dd�Zddd�d d
�ZdS ) �
FsListChannelZfslist1zos.DirEntry[str]N)�event�entryr c C sD |� � rd}n |�� r d}n|�� r*d}nd}| j||j|d� d S )N�link�file� directoryZspecial)r@ r4 �type)�
is_symlink�is_file�is_dir� send_json�name)�selfr@ rA �moder& r& r' �
send_entryW s zFsListChannel.send_entryr ��optionsr c
C s t |d�}|�dd�}|r&tddd��zt�|�}W n� tyj } z tdt|�d�|�W Y d }~nrd }~0 ty� } z tdt|�d�|�W Y d }~n>d }~0 ty� } z td t|�d�|�W Y d }~n
d }~0 0 | � � |D ]}| �
d
|� q�|�s| � � | � � d S ) Nr4 �watchTz
not-supportedz)watching is not implemented, use fswatch1��message� not-found�
access-denied�internal-errorZpresent)
r �getr
r5 �scandirr7 �str�PermissionErrorr8 �readyrM �done�close)rK rO r4 rP Zscan_dir�errorrA r& r& r' �do_openc s$
&&(zFsListChannel.do_open)�__name__�
__module__� __qualname__�payloadrX rM r^ r& r& r& r' r? T s r? c @ s* e Zd ZdZeeedef d�dd�ZdS )�
FsReadChannelZfsread1NrN c
c s~ t |d�}t|dd�}t�d|� z�t|d���}t�|�� �}|dkrX|j|krXt d��| j
rzt� |j �rz| j
|jd� n| �
� |�tj�}|d kr�q�t�d
t|�� | j
s�|�d d �jd d
��� }|V q�W d � n1 s�0 Y dt|�iW S t�y ddi Y S t�yB } zt d�|�W Y d }~n@d }~0 t�yx } z t dt|�d�|�W Y d }~n
d }~0 0 d S )Nr4 �
max_read_sizei z Opening file "%s" for reading�rb���z too-large)Z size_hint� z ...sending %d bytes� �ignore��errors�tagr* rT rU rQ )r r �logger�debug�openr5 r6 �fileno�st_sizer
Z is_binary�S_ISREGr. rZ �read1r Z
BLOCK_SIZE�len�replace�decode�encoder2 r7 rY r8 rX )rK rO r4 rd Zfilepr) �data�excr& r& r' �
do_yield_data s2
& zFsReadChannel.do_yield_data)r_ r` ra rb r r �bytesrz r& r& r& r' rc | s rc c @ sJ e Zd ZU dZded< dZded< h d�ZdZded< edd�dd �Z dS )
�FSReplaceAttrsN�
int | None�uid�gid> rL �userr$ rL )�valuer c C s0 |� � | j }|r2tddd�|�� �t|�d�d �t|dd �| _t|dd �}t|dd �}|d urt|d u rttddd ��|d ur�|d u r�tdd
d ��t|t �r�zt
� |�j | _
W q� ty� td d |� �d �d �Y q�0 n|| _
t|t ��r&zt�|�j| _W n( t�y" td d
|� �d �d �Y n0 n|| _d S )Nzprotocol-errorz%"attrs" contains unsupported key(s): �,)rR �unsupported_attrsrL r� r$ z3"group" attribute is empty while "user" is providedrQ z3"user" attribute is empty while "group" is providedrS zuid not found for zgid not found for )�keys�supported_attrsr
�join�listr rL r �
isinstancerX �pwd�getpwnam�pw_uidr~ �KeyError�grp�getgrnam�gr_gidr )rK r� r� r� r$ r& r& r' �__init__� s6 ��
zFSReplaceAttrs.__init__)
r_ r` ra r~ �__annotations__r r� rL r r� r& r& r& r' r| � s
r| c @ sN e Zd ZdZdZeded�dd�Zedddd dd
�d d �Zeed
�dd�Z dS )�FsReplaceChannelZ
fsreplace1��attrsr3 )r4 rl r c C sT |d ur |t |�kr td��t�t�� t�|� W d � n1 sF0 Y dS )N�change-conflictr* )r9 r
�
contextlib�suppressr7 r5 �unlink)rK r4 rl r& r& r' �delete� s
(zFsReplaceChannel.delete� bytes | Noner} zFSReplaceAttrs | None)r4 rl rx �sizer� r c
� s� t j�|�\}}tj|d|� d�d�\}} dddd d�� fdd�
}
ddd d d
�� fd d �
} �z|d ur�t�d
||� |r�| �t j|d|�I d H | � � d} |d ur�| �t j
||�I d H | t |�7 } | � � I d H }q�|d u�r| |k �rt�d|| � | �t j
|| �I d H | �t j|�I d H |d u �rN| |� |
|� t �| |� d } n�|dk�rv| |� |
|� t �| |� n�t �|t j�}
z�t �|
�}|t|�k�r�td��| |t�|j�� |
||� z,t �|
d�}t �|d|� t�d||� W n2 t�y } zt�d|� W Y d }~n
d }~0 0 W t � |
� n t � |
� 0 t �| |� d } W t � |� | d u�r�t � | � n t � |� | d u�r�t � | � 0 t |�S )N�.r* )�dir�prefixr# zos.stat_result | None)r; r) r c sN � d ur0� j d ur0� jd ur0t�| � j � j� n|d urJt�| |j|j� d S r< )r~ r r5 �fchownr/ r0 )r; r) r� r& r' �chown_if_required� s z8FsReplaceChannel.set_contents.<locals>.chown_if_requiredr} )r; rL r c s> � d ur� j d ur� j }n|d u r.dt� @ }t�| |� d S )Ni� )rL r( r5 �fchmod)r; rL r� r& r' �apply_file_mode� s
z6FsReplaceChannel.set_contents.<locals>.apply_file_modezfallocate(%s.tmp, %d)r zftruncate(%s.tmp, %d)r� zsecurity.selinuxz SELinux context '%s' set on '%s'zAError getting or setting SELinux context from original file: '%s')N)N) r5 r4 �split�tempfileZmkstemprm rn Z in_thread�posix_fallocaterZ �writert �read� ftruncate� fdatasync�renamerB ro �O_RDONLYr6 r2 r
�S_IMODEr. �getxattr�setxattrr8 � exceptionr\ r� r9 )rK r4 rl rx r� r� �dirname�basenamer; Ztmpnamer� r� ZwrittenZpath_fdr) Zselinux_contextry r& r� r' � set_contents� sf
$
�
z FsReplaceChannel.set_contentsrN c
� s� t |dtd �}t|d�}t|dd �}t|dd �}zx|d urT| �||d||�I d H }nD| �� | �� I d H }|d u r�| �||�}n| �|||d |�I d H }| �� d|iW S t y� } zt
d�|�W Y d }~n�d }~0 t �y } zt
d�|�W Y d }~n�d }~0 t �y2 } zt
d�|�W Y d }~nvd }~0 t
�yh } z t
dt|�d �|�W Y d }~n@d }~0 t�y� } z t
d
t|�d �|�W Y d }~n
d }~0 0 d S ) Nr� r4 r� rl rg rS r� rT rQ rU )r r| r r r� rZ r� r� r[ r7 r
�FileExistsErrorrY �IsADirectoryErrorrX r8 )rK rO r� r4 r� rl rx ry r& r& r' �run, s0
&zFsReplaceChannel.runN)
r_ r` ra rb Z capabilitiesrX r� r� r r� r& r& r& r' r� � s � Ur� c @ s~ e Zd ZU dZdZded< dZdZee dd�dd ��Z
e e d
dd �d d
�Z dddd�dd�Z
edd�dd�Zdd�dd�ZdS )�FsWatchChannelZfswatch1Nr3 �_tagFztuple[str, str | None])�maskr c C sl t jp
t j| v r&dt j| v r dndfS t j| v sDt j| v sDt j| v rHdS t j| v rVdS t j| v rddS dS d S )N�createdrD rC )�deletedN)zattribute-changedN)z done-hintN)ZchangedN) � InotifyEvent�CREATE�MOVED_TOZISDIR�
MOVED_FROM�DELETEZ DELETE_SELFZATTRIBZ CLOSE_WRITE)r� r& r& r' �mask_to_event_and_type^ s
z%FsWatchChannel.mask_to_event_and_typer� )r� �_cookierJ r c C s� t �d| j||� | �|�\}}|rTtj�| j|�� �}t|�}| j ||||d� n4t| j�}|| j
krld S || _
| j || j| j
|d� d S )Nz%do_inotify_event(%s): mask %X name %s)r@ r4 rl rE ) rm rn �_pathr� r5 r4 r� rv r9 rI r� )rK r� r� rJ r@ �type_r4 rl r& r& r' �do_inotify_eventk s
z FsWatchChannel.do_inotify_eventr} �r; �errr c C sN t �d| jt|�|� |r"t|�nd| _| jrJ| j|r:dnd| j| jd� d S )Nz&do_identity_changed(%s): fd %s, err %sr* r� r� )r@ r4 rl )rm rn r� rX r> r� �_activerI �rK r; r� r& r& r' �do_identity_changed{ s z"FsWatchChannel.do_identity_changedrN c C s8 t |d�| _d | _d| _t| j| �| _d| _| �� d S )Nr4 FT)r r� r� r� r �_watchrZ )rK rO r& r& r' r^ � s zFsWatchChannel.do_openr c C s& | j d ur| j �� d | _ | �� d S r< )r� r\ �rK r& r& r' �do_close� s
zFsWatchChannel.do_close)r_ r` ra rb r� r� r� r� � staticmethodr� r� r# r� r� r r^ r� r& r& r& r' r� S s
r� c @ s e Zd ZdZdZdS )�FollowFTN)r_ r` ra �NO�YESr& r& r& r' r� � s r� c @ sl e Zd ZU dZeed< ded< eed< eed< eed< eed< eed < d
Zeed < d Z d
ed< d Z
ded< d Z ded< ded< e e
e dd�dd��Zdd�eed d�d d �Zdd�ded d�d d �Zd d �d!d"�Zeed#�d$d%�Zed d#�d&d'�Zed d(�d)d*�Zeed+�d,d-�Zed d+�d.d/�Zeed0d d1�d2d3�Zd
d4d d5�d6d7�Zd d �d8d9�Z e d d:�d;d<�Z d S )=�
FsInfoChannelZfsinfor4 zset[str]r� �fnmatch�targets�followrP �
current_value� �effective_fnmatchNz
Handle | Noner; zset[str] | None�pendingzPathWatch | None�
path_watchz*Callable[[int, str, Follow], JsonDocument]�getattrsz1Callable[[int, str, Follow], JsonDocument | None])r� r c s� t �� tdd�dd���t �� tdd�dd���dd �ttttd
d �d d
��tjdtjdtjdtj dtj
dtj dtj di��fdd�t
dd� dd� dd� dd� dd� �fd d��fd d�d � � t� fd d �| D ���tttd!d"���fd#d$� }|S )%Nz str | int)r~ r c S s* zt �| �jW S ty$ | Y S 0 d S r< )r� �getpwuid�pw_namer� )r~ r& r& r' �get_user� s z-FsInfoChannel.make_getattrs.<locals>.get_user)r r c S s* zt �| �jW S ty$ | Y S 0 d S r< )r� �getgrgid�gr_namer� )r r& r& r' � get_group� s z.FsInfoChannel.make_getattrs.<locals>.get_groupF��follow_symlinksz bool | None)�dir_fdrJ rL r� r c S sN z4|rt j||| |d�W S t jd| � �|dd�W S W n tyH Y d S 0 d S )N)r� r� �/proc/self/fd/Tr� )r5 �accessr8 )r� rJ rL r� r& r& r' �
get_access� s z/FsInfoChannel.make_getattrs.<locals>.get_accessZregr� Zlnk�chrZblkZfifoZsockc s � � t�| j��S r< )rV r6 �S_IFMTr. r1 )�
stat_typesr& r' �<lambda>� rg z-FsInfoChannel.make_getattrs.<locals>.<lambda>c S s t �| j�S r< )r6 r� r. r1 r& r& r' r� � rg c S s | j S r< )rq r1 r& r& r' r� � rg c S s | j S r< �r/ r1 r& r& r' r� � rg c S s | j S r< �r0 r1 r& r& r' r� � rg c S s | j S r< )r- r1 r& r& r' r� � rg c s
� | j �S r< r� r1 )r� r& r' r� � rg c s
� | j �S r< r� r1 )r� r& r' r� � rg ) rE rl rL r� r~ r �mtimer� r$ c 3 s" | ]}|� � |d d� �fV qdS )c S s d S r< r& )�_r& r& r' r� � rg z7FsInfoChannel.make_getattrs.<locals>.<genexpr>.<lambda>N)rV )�.0�key)�available_stat_gettersr& r' � <genexpr>� rg z.FsInfoChannel.make_getattrs.<locals>.<genexpr>zJsonDict | None)r; rJ r� r c s z$|rt j||j| d�nt �| �� W n2 ty8 Y d S tyV dd� �D � Y S 0 � fdd��D �}d|v r�t�� j�tjkr�t �
t��" t j || d�|d< W d � n1 s�0 Y dt j fdt j
fd t jffD ]&\}}||v r܈| |||jd
�||< q�|S ) N)r� r� c S s i | ]\}}|d �qS r< r& )r� rJ �funcr& r& r' �
<dictcomp>� rg zBFsInfoChannel.make_getattrs.<locals>.get_attrs.<locals>.<dictcomp>c s i | ]\}}||� ��qS r& r& )r� r� r� r1 r& r' r� � rg �target)r� zr-okzw-okzx-okr� )r5 r6 r� r= r7 r8 r� r. �S_IFLNKr� r� �readlink�R_OK�W_OK�X_OK)r; rJ r� �result�attrr� )r� � stat_gettersr1 r' � get_attrs� s $ 0$z.FsInfoChannel.make_getattrs.<locals>.get_attrs)� functools� lru_cacher# rX �boolr6 �S_IFREG�S_IFDIRr� �S_IFCHR�S_IFBLK�S_IFIFO�S_IFSOCKr2 �tupler� )r� r� r& )r� r� r� r� r� r� r' �
make_getattrs� s*
�
� zFsInfoChannel.make_getattrsF��reset)�updatesr r c C st |rVt | j�t |�@ rD| jt�| j�dd� ddi| _|jd d� n|�t�| j�� t| j|� |rp| �|� d S )NT)�partialr
)�setr� rI �dict�fromkeys�updater )rK r r r& r& r' � send_update� s
zFsInfoChannel.send_updatec s� � j d usJ �� fdd�|D �}|�di �}t|t�s:J �� jrH||d< � jr�i |d< }dd� |�� D �D ]F}t|t�rl|dkrl|d ks�d
|v s�� �|�sl� � � j |t
j �||< ql� j d |i|d � d S )
Nc s i | ]}|� � � j|tj��qS r& )r� r; r� r� )r� rJ r� r& r' r� � rg z0FsInfoChannel.process_update.<locals>.<dictcomp>r� �entriesr� c S s h | ]}t |t�r|�d ��qS )r� )r� r rV �r� �er& r& r' � <setcomp> rg z/FsInfoChannel.process_update.<locals>.<setcomp>r� z..�/�infor
)
r; �popr� r r� r� �valuesrX � interestingr� r� r� r )rK r r r r r� rJ r& r� r' �process_update� s z FsInfoChannel.process_updater c C s* | j d usJ �| j r | �| j � d | _ d S r< )r� r r� r& r& r' �process_pending_updates s z%FsInfoChannel.process_pending_updates)rJ r c C s |dkr dS t � || j�S d S )Nr� T)r� r� �rK rJ r& r& r' r s zFsInfoChannel.interestingc C sB | � |�sd S | jd u r2t�� �d| j� t� | _| j�|� d S )Ng�������?)r r� �asyncioZget_running_loopZ
call_laterr r �addr r& r& r' �schedule_update! s
z FsInfoChannel.schedule_update)r� r c C sd |t jkrd}n*|t jt jfv r&d}n|t jkr6d}nd}| jd|t�|�t j| d�idd� d S ) NrS rT z
not-directoryrU r] )�problemrR �errnoTr
) r# �ENOENT�EPERMZEACCES�ENOTDIRr r5 �strerror� errorcode)rK r� r" r& r& r' � report_error+ s
��zFsInfoChannel.report_errorr: c C sF | j r| j�d�sdS t�|�}t�|j�tjkrB| � t
j � dS dS )Nr FT) rP r4 �endswithr5 r= r6 r� r. r r) r# r& )rK r; r) r& r& r' �flag_onlydir_error9 s
z FsInfoChannel.flag_onlydir_errorc s~ � � |�rd S |� _dh}� jr^z$|�t�d� j� ��� � j� _W n ty\ d� _Y n0 � j� fdd�|D �dd� d S )Nr� r� c s h | ]}� � |�r|�qS r& )r r r� r& r' r V rg z5FsInfoChannel.report_initial_state.<locals>.<setcomp>Tr
) r+ r; r� r r5 �listdirr� r8 r )rK r; r r& r� r' �report_initial_stateG s
z"FsInfoChannel.report_initial_stater� )r� �cookie�rawnamer c C sX t �d|||� |pdjdd�}| �|� |rT|tjtjB tjB tjB B rT| �d� d S )Nz do_inotify_event(%r, %r, %r)rg �surrogateescaperj r� ) rm rn rv r! r� r� r� r� r� )rK r� r. r/ rJ r& r& r' r� X s
��z FsInfoChannel.do_inotify_eventr} r� c C sP t �d||� | jd ur"| j�� |d u rB|d us6J �| �|� n
| �|� d S )Nzdo_identity_changed(%r, %r))rm rn r� �clearr- r) r� r& r& r' r� c s
z!FsInfoChannel.do_identity_changedc C s$ | j d usJ �| j �� | �� d S r< )r� r\ r� r& r& r' r� p s
zFsInfoChannel.do_closerN c
C sh t |d�| _tj�| j�s$t|d��tt|d��}| �|ddh �| _t |dd|v rXdnd�| _ d|v | _
t |d d
d �| _ t |d d
d �| _
| j
r�| j s�t|d��| j
r�| j s�t|d��i | _| �� | j
�sVz&t�| j| j r�tjn
tjtjB �}W n> t�y0 } z$|j�sJ �| �|j� W Y d }~n d }~0 0 | �|� |�� | �� | �� nt| j| �| _d S )Nr4 z "path" must be an absolute pathr� r� r r� �*r� r� T)�defaultrP Fz>"watch: true" and "follow: false" are (currently) incompatiblezB`targets: "stat"` and `follow: false` are (currently) incompatible) r r4 r5 �isabsr r r r r� r� r� r r� rP r� rZ r ro �O_PATH�
O_NOFOLLOWr8 r# r) r- r\ r[ r r� )rK rO r� r; ry r& r& r' r^ v s4
& "
zFsInfoChannel.do_open) r_ r` ra rb rX r� r r r� r; r� r� r� r r r r r r r! r# r) r r+ r- r� r� r� r� r r^ r& r& r& r' r� � s6
?
r� )@r r� �enumr# r� r� r� Zloggingr5 r� r r6 r� �pathlibr �typingr r r Z cockpit._vendor.systemd_ctypesr r Z&cockpit._vendor.systemd_ctypes.inotifyr r� Z(cockpit._vendor.systemd_ctypes.pathwatchr ZPathWatchListenerZchannelr r r
r Zjsonutilr r r r r r r r r r r Z getLoggerr_ rm r r# r( � stat_resultrX r2 r9 r> r? rc r| r� r� �Enumr� r� r& r& r& r' �<module> sB 4
((( ?
Mini Shell