관리-도구
편집 파일: socket.cpython-39.opt-1.pyc
a '�Dg� � @ sr d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m Z e� e �Z ed�dd�ZG d d � d �ZdS ) z#A module for common socket helpers.� N)�suppress)� performance)�DEFAULT_RUN_DIR)�messagec C s� t j�dd�}|sdS |d dkr2|�ddd� n|d dkrFtd ��t�tjtjtjB ��:}t � d t | �� |� |� |� | �d �� W d � n1 s�0 Y dS ) z[Send a sd_notify message. :param message: sd-notify message (must be valid ascii) Z NOTIFY_SOCKET� Nr �@� � �/zUnsupported socket typezSending sd_notify(%s)�ascii)�os�environ�get�replace�OSError�socket�AF_UNIX� SOCK_DGRAM� SOCK_CLOEXEC�LOG�info�str�connect�sendall�encode)r � socket_path�sock� r �4/usr/lib/python3.9/site-packages/cloudinit/socket.py� sd_notify s � r c @ s< e Zd ZdZed�dd�Zed�dd�Zdd � Zd d � Zd S ) � SocketSyncz<A two way synchronization protocol over Unix domain sockets.)�namesc G s� d| _ d| _d| _d| _d| _dd� |D �| _tjt� d�ddd � | j� � D ]T\}}t� d |� d �}t t �� t� |� W d � n1 s�0 Y |� |� qNd S ) an Initialize a synchronization context. 1) Ensure that the socket directory exists. 2) Bind a socket for each stage. Binding the sockets on initialization allows receipt of stage "start" notifications prior to the cloud-init stage being ready to start. :param names: stage names, used as a unique identifiers r r Fc S s$ i | ] }|t � t jt jt jB ��qS r )r r r r )�.0�namer r r � <dictcomp>> s ��z'SocketSync.__init__.<locals>.<dictcomp>z/sharei� T)�mode�exist_ok�/share/z.sockN)�stage�remote�first_exception�systemd_exit_code�experienced_any_error�socketsr �makedirsr �itemsr �FileNotFoundError�removeZbind)�selfr! r# r r r r r �__init__- s � (zSocketSync.__init__)r( c C s" || j vrtd|� ���|| _| S )a� Set the stage before entering context. This enables the context manager to be initialized separately from each stage synchronization. :param stage: the name of a stage to synchronize Example: sync = SocketSync("stage 1", "stage 2"): with sync("stage 1"): pass with sync("stage 2"): pass zInvalid stage name: )r- � ValueErrorr( )r2 r( r r r �__call__M s zSocketSync.__call__c C s t �tj�� �r t�d� dS d| _td| j � d�� | j | j }t � d| j � ��� |� d�\}| _W d � n1 sz0 Y d|kr�| �ddd� td t|�� d ���n:t� d | j � d �t| j�kr�| �ddd� td | j� ���td| j � d�� | S )z�Wait until a message has been received on this stage's socket. Once the message has been received, enter the context. z:Stdin is a tty, so skipping stage synchronization protocolNr zDSTATUS=Waiting on external services to complete before starting the z stage.zWaiting to start stage � s startzReceived invalid message: [�]r' z -return.sockz Unexpected path to unix socket: zSTATUS=Running (z stage))r �isatty�sys�stdin�filenor r r+ r r( r- r ZTimedZrecvfromr) �__exit__r4 r r )r2 r �chunkr r r � __enter__a s0 ��� .�zSocketSync.__enter__c C s� d| j � �}|rXd| _d| _t|�� d|j� �}d}| js@|| _t�|� td|� �� | jpft | j�| _| j | j }|� | j � |� d|� d| j� d ��� � |�� dS ) z.Notify the socket that this stage is complete.z,Completed socket interaction for boot stage r Tz in zkfatal error, run "systemctl status cloud-init-main.service" and "cloud-init status --long" for more detailszSTATUS=zecho 'z'; exit �;)r( r+ r, �repr�tb_framer* r Zfatalr �boolr- r r) r r �close)r2 �exc_typeZexc_valZexc_tbr �statusr r r r r< � s* � � �zSocketSync.__exit__N) �__name__� __module__� __qualname__�__doc__r r3 r5 r> r< r r r r r * s 'r )rI Zloggingr r r9 � contextlibr Z cloudinitr Zcloudinit.settingsr Z getLoggerrF r r r r r r r r �<module> s