File: //snap/core20/2669/usr/lib/python3/dist-packages/cloudinit/__pycache__/socket.cpython-38.pyc
U
Ӈg� � @ 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 ��0}t �
d
t | �� |� |� |�
| �d �� W 5 Q R X 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 �2/usr/lib/python3/dist-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� d
|� d �}t
t �� t� |� W 5 Q R X |�
|� 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 krtd|� ���|| _| 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 5 Q R X 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 s, � � �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 Zstatusr r r r r< � s* �
� �zSocketSync.__exit__N) �__name__�
__module__� __qualname__�__doc__r r3 r5 r> r< r r r r r * s
'r )rH Zloggingr r r9 �
contextlibr Z cloudinitr Zcloudinit.settingsr Z getLoggerrE r r r r r r r r �<module> s