File: //usr/lib/python3.9/site-packages/cockpit/channels/__pycache__/filesystem.cpython-39.opt-1.pyc
a
�/�h�V � @ 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* e�+e,�Z-e�.� e/d
�d d ��Z0e1d
�d
d�Z2dd
�dd�Z3dd
�dd�Z4G dd� de �Z5G dd� de �Z6G dd� de �Z7G dd� de e�Z8G d d � d ej9�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_str�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_umask6 s r& 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��bufr$ r$ r% �
tag_from_stat< s r0 �
str | Nonec C s< zt t�| ��W S ty$ Y dS ty6 Y d S 0 d S )Nr' )r0 �os�stat�FileNotFoundError�OSError)�pathr$ r$ r% �
tag_from_path@ s r7 c C s* zt t�| ��W S ty$ Y d S 0 d S �N)r0 r2 �fstatr5 )�fdr$ r$ r% � tag_from_fdI s r; c @ s e Zd ZdZdd� Zdd� ZdS )�
FsListChannelZfslist1c C sD |� � rd}n |�� r d}n|�� r*d}nd}| j||j|d� d S )N�link�file� directoryZspecial)�eventr6 �type)�
is_symlink�is_file�is_dir� send_json�name)�selfr@ �entry�moder$ r$ r% �
send_entryU s zFsListChannel.send_entryc
C s |� 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 ) Nr6 �watchTz
not-supportedz)watching is not implemented, use fswatch1��message� not-found�
access-denied�internal-errorZpresent) �getr
r2 �scandirr4 �str�PermissionErrorr5 �readyrJ �done�close)rG �optionsr6 rK Zscan_dir�errorrH r$ r$ r% �do_opena s$
&&(zFsListChannel.do_openN)�__name__�
__module__� __qualname__�payloadrJ rZ r$ r$ r$ r% r< R s r< c @ s* e Zd ZdZeeedef d�dd�ZdS )�
FsReadChannelZfsread1N�rX r c
c s~ t |d�}t|dd �}t�d|� z�t|d���}t�|�� �}|d urX|j|krXt d��| j
rzt� |j �rz| j
|jd� n| �
� |�tj�}|dkr�q�t�dt|�� | j
s�|�d d�jd
d ��� }|V q�W d � n1 s�0 Y d t|�iW S t�y d d
i 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 )Nr6 �
max_read_sizez Opening file "%s" for reading�rbz too-large)Z size_hint� z ...sending %d bytes� �ignore��errors�tagr' rO rP rL )r r �logger�debug�openr2 r3 �fileno�st_sizer
Z is_binary�S_ISREGr+ rU �read1r Z
BLOCK_SIZE�len�replace�decode�encoder0 r4 rT r5 rS )rG rX r6 ra Zfilepr/ �data�excr$ r$ r% �
do_yield_data} s2
& zFsReadChannel.do_yield_data)r[ r\ r] r^ r r �bytesrv r$ r$ r$ r% r_ z s r_ c @ sH e Zd ZdZeded�dd�Zeddddd�d d
�Zeed �d d
�ZdS )�FsReplaceChannelZ
fsreplace1r1 )r6 rh r c C sT |d ur |t |�kr td��t�t�� t�|� W d � n1 sF0 Y dS )N�change-conflictr' )r7 r
�
contextlib�suppressr4 r2 �unlink)rG r6 rh r$ r$ r% �delete� s
(zFsReplaceChannel.delete� bytes | None�
int | None)r6 rh rt �sizer c
� sv t j�|�\}}tj|d|� d�d�\}}�z |d urht�d||� |r`| �t j|d|�I d H | � � d} |d ur�| �t j
||�I d H | t |�7 } | � � I d H }ql|d ur�| |k r�t�d|| � | �t j
|| �I d H | �t j|�I d H |d u �r t �|dt� @ � t �||� d }�n |dk�rLt �|dt� @ � t �||� n�t �|t j�}
z�t �|
�} |t| �k�r|td��t �|t�| j�� t �|| j| 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�rnt �"|� n t �!|� |d u�rlt �"|� 0 t#|�S ) N�.r' )�dir�prefixzfallocate(%s.tmp, %d)r zftruncate(%s.tmp, %d)i� ry zsecurity.selinuxz SELinux context '%s' set on '%s'zAError getting or setting SELinux context from original file: '%s')$r2 r6 �split�tempfileZmkstempri rj Z in_thread�posix_fallocaterU �writerp �read� ftruncate� fdatasync�fchmodr& �renamer= rk �O_RDONLYr3 r0 r
�S_IMODEr+ �fchownr, r- �getxattr�setxattrr5 � exceptionrW r| r7 )rG r6 rh rt r� �dirname�basenamer: ZtmpnameZwrittenZpath_fdr/ Zselinux_contextru r$ r$ r% � set_contents� s^
$
�
z FsReplaceChannel.set_contentsr` c
� s� t |d�}t|dd �}t |dd �}zt|d urD| �||d|�I d H }nB| �� | �� I d H }|d u rp| �||�}n| �|||d �I d H }| �� d|iW S ty� } ztd�|�W Y d }~n�d }~0 t y� } ztd�|�W Y d }~n�d }~0 t
�y } ztd�|�W Y d }~nvd }~0 t �yT } z tdt |�d�|�W Y d }~n@d }~0 t
�y� } z td t |�d�|�W Y d }~n
d }~0 0 d S )
Nr6 r� rh rc rN ry rO rL rP )r r r� rU r� r} rV r4 r
�FileExistsErrorrT �IsADirectoryErrorrS r5 )rG rX r6 r� rh rt ru r$ r$ r% �run� s.
&zFsReplaceChannel.runN) r[ r\ r] r^ rS r} r� r r� r$ r$ r$ r% rx � s � Crx 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 )�FsWatchChannelZfswatch1Nr1 �_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�createdr? r> )�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� �_cookierF 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@ r6 rh rA ) ri rj �_pathr� r2 r6 �joinrr r7 rE r� )rG r� r� rF r@ �type_r6 rh r$ r$ r% �do_inotify_event- 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@ r6 rh )ri rj r� rS r; r� �_activerE �rG r: r� r$ r$ r% �do_identity_changed= s z"FsWatchChannel.do_identity_changedr` c C s8 t |d�| _d | _d| _t| j| �| _d| _| �� d S )Nr6 FT)r r� r� r� r �_watchrU )rG rX r$ r$ r% rZ C s zFsWatchChannel.do_openr c C s& | j d ur| j �� d | _ | �� d S r8 )r� rW �rG r$ r$ r% �do_closeM s
zFsWatchChannel.do_close)r[ r\ r] r^ r� �__annotations__r� r� � staticmethodr� r� r! r� r� r rZ r� r$ r$ r$ r% r� s
r� c @ s e Zd ZdZdZdS )�FollowFTN)r[ r\ r] �NO�YESr$ r$ r$ r% r� T 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 )=�
FsInfoChannelZfsinfor6 zset[str]�attrs�fnmatch�targets�followrK �
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���tjdtjd tjd
tjd tjd tj d
tj
di��fdd�t dd� dd� dd� dd� dd� �fdd��fdd�d� � t � fdd�| D ���tt
tdd ��fd d � }|S ) Nz str | int)�uidr c S s* zt �| �jW S ty$ | Y S 0 d S r8 )�pwd�getpwuid�pw_name�KeyError)r� r$ r$ r% �get_usero s z-FsInfoChannel.make_getattrs.<locals>.get_user)�gidr c S s* zt �| �jW S ty$ | Y S 0 d S r8 )�grp�getgrgid�gr_namer� )r� r$ r$ r% � get_groupv s z.FsInfoChannel.make_getattrs.<locals>.get_groupZregr� Zlnk�chrZblkZfifoZsockc s � � t�| j��S r8 )rQ r3 �S_IFMTr+ r. )�
stat_typesr$ r% �<lambda>� rc z-FsInfoChannel.make_getattrs.<locals>.<lambda>c S s t �| j�S r8 )r3 r� r+ r. r$ r$ r% r� � rc c S s | j S r8 )rm r. r$ r$ r% r� � rc c S s | j S r8 �r, r. r$ r$ r% r� � rc c S s | j S r8 �r- r. r$ r$ r% r� � rc c S s | j S r8 )r* r. r$ r$ r% r� � rc c s
� | j �S r8 r� r. )r� r$ r% r� � rc c s
� | j �S r8 r� r. )r� r$ r% r� � rc ) rA rh rI r� r� r� �mtime�userr" c 3 s" | ]}|� � |d d� �fV qdS )c S s d S r8 r$ )�_r$ r$ r% r� � rc z7FsInfoChannel.make_getattrs.<locals>.<genexpr>.<lambda>N)rQ )�.0�key)�available_stat_gettersr$ r% � <genexpr>� rc z.FsInfoChannel.make_getattrs.<locals>.<genexpr>zJsonDict | None)r: rF 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 |S )N)�follow_symlinks�dir_fdc S s i | ]\}}|d �qS r8 r$ )r� rF �funcr$ r$ r% �
<dictcomp>� rc zBFsInfoChannel.make_getattrs.<locals>.get_attrs.<locals>.<dictcomp>c s i | ]\}}||� ��qS r$ r$ )r� r� r� r. r$ r% r� � rc �target)r� ) r2 r3 �valuer9 r4 r5 r� r+ �S_IFLNKrz r{ �readlink)r: rF r� �result)� stat_gettersr. r% � get_attrs� s $ 0z.FsInfoChannel.make_getattrs.<locals>.get_attrs)� functools� lru_cacher! r3 �S_IFREG�S_IFDIRr� �S_IFCHR�S_IFBLK�S_IFIFO�S_IFSOCKr0 �tuplerS r� )r� r� r$ )r� r� r� r� r� r% �
make_getattrsl 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� rE �dict�fromkeys�updater )rG r� r� r$ r$ r% � send_update� s
zFsInfoChannel.send_updatec s� � fdd�|D �}|� di �}� jr,||d< � jr�i |d< }dd� |�� D �D ]F}t|t�rP|dkrP|d ks�d
|v s�� �|�sP� �� j|t j
�||< qP� j d |i|d � d S )
Nc s i | ]}|� � � j|tj��qS r$ )r� r: r� r� )r� rF r� r$ r% r� � rc z0FsInfoChannel.process_update.<locals>.<dictcomp>r� �entriesr� c S s h | ]}t |t�r|�d ��qS )r� )�
isinstancer� rQ �r� �er$ r$ r% � <setcomp>� rc z/FsInfoChannel.process_update.<locals>.<setcomp>r� z..�/�infor� ) �popr� r� �valuesr� rS � interestingr� r: r� r� r� )rG r� r� r� r� r� rF r$ r� r% �process_update� s z FsInfoChannel.process_updater c C s | j r| �| j � d | _ d S r8 )r� r r� r$ r$ r% �process_pending_updates� s z%FsInfoChannel.process_pending_updates)rF r c C s |dkr dS t � || j�S d S )Nr� T)r� r� �rG rF 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 ) NrN rO z
not-directoryrP rY )�problemrM �errnoTr� ) r �ENOENT�EPERMZEACCES�ENOTDIRr� r2 �strerror� errorcode)rG r� r r$ r$ r% � report_error� s
��zFsInfoChannel.report_error)r: r c C sF | j r| j�d�sdS t�|�}t�|j�tjkrB| � t
j � dS dS )Nr� FT) rK r6 �endswithr2 r9 r3 r� r+ r� r r r )rG r: r/ r$ r$ r% �flag_onlydir_error� 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� z/proc/self/fd/c s h | ]}� � |�r|�qS r$ )r r� r� r$ r% r�
rc z5FsInfoChannel.report_initial_state.<locals>.<setcomp>Tr� ) r r: r� r� r2 �listdirr� r5 r )rG r: r� r$ r� r% �report_initial_state� 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)rc �surrogateescaperf r� ) ri rj rr r r� r� r� r� r� )rG r� r r rF r$ r$ r% r� s
��z FsInfoChannel.do_inotify_eventr r� c C sD t �d||� | jd ur"| j�� |d u r6| �|� n
| �|� d S )Nzdo_identity_changed(%r, %r))ri rj r� �clearr r r� r$ r$ r% r� s
z!FsInfoChannel.do_identity_changedc C s | j �� | �� d S r8 )r� rW r� r$ r$ r% r� $ s
zFsInfoChannel.do_closer` c
C s\ 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
�sJz&t�| j| j r�tjn
tjtjB �}W n2 t�y$ } z| �|j� W Y d }~n d }~0 0 | �|� |�� | �� | �� nt| j| �| _d S )Nr6 z "path" must be an absolute pathr� r� r� r� �*r� r� T)�defaultrK Fz>"watch: true" and "follow: false" are (currently) incompatiblezB`targets: "stat"` and `follow: false` are (currently) incompatible) r r6 r2 �isabsr r� r r� r� r� r� r r� rK r� rU r rk �O_PATH�
O_NOFOLLOWr5 r r r rW rV r r� )rG rX r� r: ru r$ r$ r% rZ * s2
&"
zFsInfoChannel.do_open) r[ r\ r] r^ rS r� �boolr r� r: r� r� r� r r� r� r r r r r! r r r r r� r� r� r� r rZ r$ r$ r$ r% r� Y s6
1
r� )<r rz �enumr r� r� r� Zloggingr2 r� r r3 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 Z getLoggerr[ ri r� r! r& rS r0 r7 r; r< r_ rx r� �Enumr� r� r$ r$ r$ r% �<module> s> ,
((s?