o thg@s~dZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z e eZdefdd ZGd d d ZdS) z#A module for common socket helpers.N)suppress)Dict) performance)DEFAULT_RUN_DIRmessagecCstjdd}|s dS|ddkr|dddn |ddkr#td ttjtjtjB}t d t || || |d WddS1sOwYdS) z[Send a sd_notify message. :param message: sd-notify message (must be valid ascii) NOTIFY_SOCKETNr@/zUnsupported socket typezSending sd_notify(%s)ascii)osenvirongetreplaceOSErrorsocketAF_UNIX SOCK_DGRAM SOCK_CLOEXECLOGinfostrconnectsendallencode)r socket_pathsockr2/usr/lib/python3/dist-packages/cloudinit/socket.py sd_notifys   "r!c@s<eZdZdZdefddZdefddZdd Zd d Zd S) SocketSyncz>s z'SocketSync.__init__..z/shareiT)modeexist_okz/share/z.sockN)stagefirst_exceptionsystemd_exit_codeexperienced_any_errorsockets connectionsrmakedirsritemsrFileNotFoundErrorremovebindlisten)selfr#r&rrrrr __init__.s"     zSocketSync.__init__r*cCs"||jvr td|||_|S)aSet 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*)r6r*rrr __call__Os zSocketSync.__call__cCsttjrtddSd|_td|j d|j |j }t d|j | \}}|d\}}||j|j <Wdn1sHwYd|krb|dddtd t|d td |j d |S) zWait 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 protocolNrzDSTATUS=Waiting on external services to complete before starting the z stage.zWaiting to start stage sstartzReceived invalid message: []zSTATUS=Running (z stage))risattysysstdinfilenorrr,r!r*r.rTimedacceptrecvfromr/__exit__r8r)r6r connection_chunkrrr __enter__cs,  zSocketSync.__enter__cCsttjr dSd|j}|r6d|_d|_t|d|j }d}|j s*||_ t |t d||jp=t|j|_|j|j}|d|d |jd |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 ;)rr<r=r>r?r*r,r-reprtb_framer+rfatalr!boolr/rrclose)r6exc_typeexc_valexc_tbrstatusrrrr rCs,   zSocketSync.__exit__N) __name__ __module__ __qualname____doc__rr7r9rGrCrrrr r"+s ! "r")rUloggingrrr= contextlibrtypingr cloudinitrcloudinit.settingsr getLoggerrRrrr!r"rrrr s