Current File : //usr/lib/python3/dist-packages/cloudinit/__pycache__/socket.cpython-312.pyc
�
���f; � � � d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ej e � Z de
fd�Z G d� d� Zy) z#A module for common socket helpers.� N)�suppress)�DEFAULT_RUN_DIR�messagec � � t j j dd� }|sy|d dk( r|j ddd� n|d dk7 r t d � �t j
t
j t
j t
j z � 5 }t j d
t | � � |j |� |j | j d � � ddd� y# 1 sw Y yxY w) 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�socks �2/usr/lib/python3/dist-packages/cloudinit/socket.py� sd_notifyr s� � �
�*�*�.�.� �"�5�K� �� �Q� �3� ����C��q�)�
�Q� �3� ��/�0�0� � � �����)�)�F�,?�,?�?�
� .�
� ���(�#�g�,�7� � � �[�!� � � �W�^�^�G�,�-� .� .� .�s
�AC0�0C9c �4 � e Zd ZdZdefd�Zdefd�Zd� Zd� Zy) �
SocketSyncz<A two way synchronization protocol over Unix domain sockets.�namesc �0 � d| _ d| _ d| _ d| _ d| _ |D �ci c]F }|t j
t
j t
j t
j z � ��H c}| _ t j t � d�dd�� | j j � D ]O \ }}t � d|� d �}t t � 5 t j |� d
d
d
� |j# |� �Q y
c c}w # 1 sw Y �"xY w) 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 Fz/sharei� T)�mode�exist_ok�/share/z.sockN)�stage�remote�first_exception�systemd_exit_code�experienced_any_errorr r r r �socketsr �makedirsr �itemsr �FileNotFoundError�remove�bind)�selfr# �namer r s r �__init__zSocketSync.__init__- s� � � ��
��� � !�� �!"�� �%*��"�
�
� �
�&�-�-����� 1� 1�F�4G�4G� G��
�
�� � � � ��'�v�.�U�T�J� �,�,�,�,�.� #�J�D�$� ,� -�W�T�F�%�@�K��+�,�
'�� � �+�&�
'� �I�I�k� "� #��
�
'�
'�s �A D�D � D r( c �L � || j vrt d|� �� �|| _ | 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( )r3 r( s r �__call__zSocketSync.__call__M s/ � � �� � � $� � 3�E�7� ;�<� <���
�� � c �, � t j t j j � � rt
j
d� yd| _ t d| j � d�� t j � }| j | j }|j d� \ }| _ d|k7 r+| j ddd� t! dt# |� � d �� �t$ � d
| j � d �t# | j � k7 r+| j ddd� t! d | j � �� �t j � |z
}|d
kD rd|d�d�nd}t d| j � d�� t
j' d| j |� | 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.� s startzReceived invalid message: [�]r' z -return.sockz Unexpected path to unix socket: g{�G�z�?ztook z .3fzs to r zSTATUS=Running (z stage)z$sync(%s): synchronization %scomplete)r �isatty�sys�stdin�filenor r r+ r r( �time� monotonicr- �recvfromr) �__exit__r7 r r �debug)r3 �
start_timer �chunk�total�time_msgs r � __enter__zSocketSync.__enter__a sj � �
�9�9�S�Y�Y�%�%�'� (� �H�H�L�
�
�!"�� ��
,�,0�J�J�<�w�
@�
� �^�^�%�
� �|�|�D�J�J�'�� !�]�]�1� -���t�{� �u� � �M�M�$��d� +� � :�3�u�:�,�a� H�I� I� � ���
�
�|�<�
@�C� �K�K�E
�
�
�M�M�$��d� +� � ?�� � �}� M�N� N���� �:�-��05�� �U�5��,�e�,�"���$�T�Z�Z�L��8�9� � � �8�$�*�*�h�O�� r9 c � � d| j � �}|r`d| _ d| _ t |� � d|j � �}d}| j
s|| _ t j |� t d|� �� | j xs t | j � | _ | j | j }|j | j � |j d|� d| j � d �j � � |j � y)
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 �fatalr �boolr- r r) r r �close)r3 �exc_type�exc_val�exc_tbr �statusr s r rD zSocketSync.__exit__� s� � �@���� �M�� �%&�D� "�)-�D� &� �W�
��d�6�?�?�*;�<�F�B�
� �'�'�'-��$� �I�I�f� � ���x�(� )�%)�%?�%?� &
�4� � "� "�D
��"� �|�|�D�J�J�'�� � � �T�[�[�!�
� � ��W�I�X�d�&<�&<�%=�Q� ?� F� F� H�
�
�
�
� � r9 N) �__name__�
__module__� __qualname__�__doc__r r5 r8 rJ rD � r9 r r" r" * s) � �F� #�s� #�@�c� �((�T#r9 r" )rY �loggingr r r> rA �
contextlibr �cloudinit.settingsr � getLoggerrV r r r r" rZ r9 r �<module>r_ sG �� )� � �
�
� � � .��g����!��.�s� .�6D� Dr9
Mini Shell