�
x[h� � � � S r SSKrSSKrSSKrSSKrSSKJr SSKJr SSK J
r
\R " \ 5 r
S\4S jr " S S 5 rg)
z#A module for common socket helpers.� N)�suppress)� performance)�DEFAULT_RUN_DIR�messagec � � [ R R SS5 nU( d gUS S:X a UR SSS5 OUS S:w a [ S 5 e[
R
" [
R [
R [
R - 5 n[ R S
[ U 5 5 UR U5 UR U R S 5 5 SSS5 g! , ( d f g= f) zSSend 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
� AC8�8
Dc �@ � \ rS rSrSrS\4S jrS\4S jrS rS r S r
g
) �
SocketSync�* z<A two way synchronization protocol over Unix domain sockets.�namesc �B � SU l SU l SU l SU l SU l U Vs0 s HH nU[
R
" [
R [
R [
R - 5 _MJ snU l [ R " [ S3SSS9 U R R 5 HP u p#[ SU S 3n[ [ 5 [ R " U5 S
S
S
5 UR# U5 MR g
s snf ! , ( d f N)= f) a6 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__�SocketSync.__init__- s� � � ��
��� � !�� �!"�� �%*��"�
�
� ��
�&�-�-����� 1� 1�F�4G�4G� G��
� �
�� � � � ��'�v�.�U�T�J� �,�,�,�,�.�J�D� ,� -�W�T�F�%�@�K��+�,�� � �+�&� -� �I�I�k� "� /��
� -�,�s �AD �D�
D r* c �L � XR ; a [ SU 35 eXl U $ )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* )r5 r* s r �__call__�SocketSync.__call__M s+ � � � � � $� � 3�E�7� ;�<� <��
�� � c � � [ R " [ R R 5 5 ( a [
R
S5 gSU l [ SU R S35 U R U R n[ R " SU R 35 UR S5 u o l SSS5 SW:w a+ U R SSS5 [! S [# U5 S
35 e[$ S U R S 3[# U R 5 :w a+ U R SSS5 [! S
U R 35 e[ SU R S35 U $ ! , ( d f N�= f)zwWait 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 �Timed�recvfromr+ �__exit__r: r r )r5 r �chunks r � __enter__�SocketSync.__enter__a sK � �
�9�9�S�Y�Y�%�%�'� (� (� �H�H�L�
�
�!"�� ��
,�,0�J�J�<�w�
@�
� �|�|�D�J�J�'��
�
�
�!8���� � E�
F�!%� � �q�!1� �E�;� G� �u� � �M�M�$��d� +� � :�3�u�:�,�a� H�I� I� � ���
�
�|�<�
@�C� �K�K�E
�
�
�M�M�$��d� +� � ?�� � �}� M�N� N��$�T�Z�Z�L��8�9�� �! G�
F�s �*E/�/
E=c �: � SU R 3nU( ad SU l SU l [ U5 SUR 3nSnU R
( d XPl [ R U5 [ SU 35 U R =( d [ U R 5 U l U R U R nUR U R 5 UR SU SU R S 3R 5 5 UR 5 g)
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)r5 �exc_type�exc_val�exc_tbr �statusr s r rG �SocketSync.__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�
�
�
�
� � r= )r. r, r+ r/ r* r- N) �__name__�
__module__� __qualname__�__firstlineno__�__doc__r r7 r; rI rG �__static_attributes__� r= r r# r# * s) � �F� #�s� #�@�c� �(%�N#r= r# )r[ �loggingr r rB �
contextlibr � cloudinitr �cloudinit.settingsr � getLoggerrW r r r! r# r] r= r �<module>rc sE �� )� � �
�
� � !� .�
����!��.�s� .�6A� Ar=