0x1949 Team - FAZEMRX - MANAGER
Edit File: socket.cpython-310.pyc
o ���f; � @ sn d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z e� e �Z de fdd�ZG dd � d �ZdS ) z#A module for common socket helpers.� N)�suppress)�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| _d| _dd� |D �| _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 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�remove�bind)�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( )r3 r( r r r �__call__M s zSocketSync.__call__c C s t �tj�� �rt�d� dS d| _td| j � d�� t � � }| j | j }|� d�\}| _d|krB| �ddd� tdt|�� d ���t� d | j � d �t| j�kr_| �ddd� td | j� ���t � � | }|d krpd|d�d�nd}td| j � d�� t�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__r5 r r �debug)r3 � start_timer �chunk�total�time_msgr r r � __enter__a s6 ��� � zSocketSync.__enter__c C s� d| j � �}|r,d| _d| _t|�� d|j� �}d}| js || _t�|� td|� �� | jp3t | 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 �fatalr �boolr- r r) r r �close)r3 �exc_type�exc_val�exc_tbr �statusr r r r r@ � s* � � �zSocketSync.__exit__N) �__name__� __module__� __qualname__�__doc__r r4 r6 rF r@ r r r r r * s *r )rT �loggingr r r: r= � contextlibr �cloudinit.settingsr � getLoggerrQ r r r r r r r r �<module> s