File: //usr/lib/python3/dist-packages/cloudinit/__pycache__/socket.cpython-310.pyc
o
�t�hg � @ s~ d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddl m Z e�
e�Zdefdd �ZG d
d � d �ZdS ) z#A module for common socket helpers.� N)�suppress)�Dict)� performance)�DEFAULT_RUN_DIR�messagec C s� t j�dd�}|s dS |d dkr|�ddd� n
|d dkr#td ��t�tjtjtjB �� }t �
d
t | �� |� |� |�
| �d �� W d � dS 1 sOw Y dS ) z[Send a sd_notify message.
:param message: sd-notify message (must be valid ascii)
�
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 ZdZdefdd�Zdefdd�Zdd � Zd
d � Zd S )
�
SocketSyncz<A two way synchronization protocol over Unix domain sockets.�namesc G s� d| _ d| _d| _d| _dd� |D �| _i | _tjt� d�ddd � | j� � D ].\}}t� d
|� d �}t
t ��
t� |� W d � n1 sGw Y |�
|� |�� q'd 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 � SOCK_STREAMr )�.0�namer r r �
<dictcomp>> s ���z'SocketSync.__init__.<locals>.<dictcomp>z/sharei� T)�mode�exist_okz/share/z.sockN)�stage�first_exception�systemd_exit_code�experienced_any_error�sockets� connectionsr �makedirsr �itemsr �FileNotFoundError�remove�bind�listen)�selfr# r& r r r r r �__init__. s" �
�
�zSocketSync.__init__r* c C s" || j vr td|� ���|| _| 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* )r6 r* r r r �__call__O s
zSocketSync.__call__c C s� t �tj�� �rt�d� dS d| _td| j � d�� | j
| j }t � d| j � ��� |�
� \}}|�d�\}}|| j| j < W d � n1 sHw Y d|krb| �ddd� td t|�� d
���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: [�]zSTATUS=Running (z stage))r �isatty�sys�stdin�filenor r r, r! r* r. r �Timed�accept�recvfromr/ �__exit__r8 r )r6 r �
connection�_�chunkr r r � __enter__c s, ��� �zSocketSync.__enter__c C s� t �tj�� �r
dS d| j� �}|r6d| _d| _t|�� d|j � �}d}| j
s*|| _
t � |� t
d|� �� | jp=t| j�| _| j| j }|�d|� d | j� d
��� � |�� dS ) z.Notify the socket that this stage is complete.Nz,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= r> r? r* r, r- �repr�tb_framer+ r �fatalr! �boolr/ r r �close)r6 �exc_type�exc_val�exc_tbr �statusr r r r rC � s, �
� �zSocketSync.__exit__N) �__name__�
__module__� __qualname__�__doc__r r7 r9 rG rC r r r r r" + s ! "r" )rU �loggingr r r= �
contextlibr �typingr � cloudinitr �cloudinit.settingsr � getLoggerrR r r r! r" r r r r �<module> s