ACIL FM

Nama	Ukuran	Aksi
analyze	-	Open
cmd	-	Open
config	-	Open
distros	-	Open
filters	-	Open
handlers	-	Open
log	-	Open
mergers	-	Open
net	-	Open
reporting	-	Open
sources	-	Open
__pycache__	-	Open
apport.py	8.27 MB	View DL Edit
atomic_helper.py	2.79 MB	View DL Edit
cloud.py	3.71 MB	View DL Edit
dmi.py	7.86 MB	View DL Edit
event.py	2 MB	View DL Edit
features.py	4.87 MB	View DL Edit
gpg.py	7.99 MB	View DL Edit
helpers.py	16.16 MB	View DL Edit
importer.py	2.43 MB	View DL Edit
lifecycle.py	7.78 MB	View DL Edit
netinfo.py	24.02 MB	View DL Edit
performance.py	3.1 MB	View DL Edit
persistence.py	2.52 MB	View DL Edit
registry.py	1022 B	View DL Edit
safeyaml.py	10.11 MB	View DL Edit
settings.py	2.15 MB	View DL Edit
signal_handler.py	1.75 MB	View DL Edit
simpletable.py	1.93 MB	View DL Edit
socket.py	5.93 MB	View DL Edit
ssh_util.py	22.22 MB	View DL Edit
stages.py	41.53 MB	View DL Edit
subp.py	12.36 MB	View DL Edit
templater.py	7.8 MB	View DL Edit
temp_utils.py	2.94 MB	View DL Edit
type_utils.py	703 B	View DL Edit
url_helper.py	34.7 MB	View DL Edit
user_data.py	14.44 MB	View DL Edit
util.py	91.22 MB	View DL Edit
version.py	564 B	View DL Edit
warnings.py	3.76 MB	View DL Edit
__init__.py	0 B	View DL Edit

Edit file: /lib/python3.9/site-packages/cloudinit/socket.py

# This file is part of cloud-init. See LICENSE file for license information.
"""A module for common socket helpers."""
import logging
import os
import socket
import sys
from contextlib import suppress

from cloudinit import performance
from cloudinit.settings import DEFAULT_RUN_DIR

LOG = logging.getLogger(__name__)

def sd_notify(message: str):
    """Send a sd_notify message.

:param message: sd-notify message (must be valid ascii)
    """
    socket_path = os.environ.get("NOTIFY_SOCKET", "")

if not socket_path:
        # not running under systemd, no-op
        return

elif socket_path[0] == "@":
        # abstract
        socket_path.replace("@", "\0", 1)

# unix domain
    elif socket_path[0] != "/":
        raise OSError("Unsupported socket type")

with socket.socket(
        socket.AF_UNIX, socket.SOCK_DGRAM | socket.SOCK_CLOEXEC
    ) as sock:
        LOG.info("Sending sd_notify(%s)", str(message))
        sock.connect(socket_path)
        sock.sendall(message.encode("ascii"))

class SocketSync:
    """A two way synchronization protocol over Unix domain sockets."""

def __init__(self, *names: str):
        """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
        """
        self.stage = ""
        self.remote = ""
        self.first_exception = ""
        self.systemd_exit_code = 0
        self.experienced_any_error = False
        self.sockets = {
            name: socket.socket(
                socket.AF_UNIX, socket.SOCK_DGRAM | socket.SOCK_CLOEXEC
            )
            for name in names
        }
        # ensure the directory exists
        os.makedirs(f"{DEFAULT_RUN_DIR}/share", mode=0o700, exist_ok=True)
        # removing stale sockets and bind
        for name, sock in self.sockets.items():
            socket_path = f"{DEFAULT_RUN_DIR}/share/{name}.sock"
            with suppress(FileNotFoundError):
                os.remove(socket_path)
            sock.bind(socket_path)

def __call__(self, stage: str):
        """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
        """
        if stage not in self.sockets:
            raise ValueError(f"Invalid stage name: {stage}")
        self.stage = stage
        return self

def __enter__(self):
        """Wait until a message has been received on this stage's socket.

Once the message has been received, enter the context.
        """
        if os.isatty(sys.stdin.fileno()):
            LOG.info(
                "Stdin is a tty, so skipping stage synchronization protocol"
            )
            return
        self.systemd_exit_code = 0
        sd_notify(
            "STATUS=Waiting on external services to "
            f"complete before starting the {self.stage} stage."
        )
        # block until init system sends us data
        # the first value returned contains a message from the init system
        #     (should be "start")
        # the second value contains the path to a unix socket on which to
        #     reply, which is expected to be /path/to/{self.stage}-return.sock
        sock = self.sockets[self.stage]
        with performance.Timed(f"Waiting to start stage {self.stage}"):
            chunk, self.remote = sock.recvfrom(5)

if b"start" != chunk:
            # The protocol expects to receive a command "start"
            self.__exit__(None, None, None)
            raise ValueError(f"Received invalid message: [{str(chunk)}]")
        elif f"{DEFAULT_RUN_DIR}/share/{self.stage}-return.sock" != str(
            self.remote
        ):
            # assert that the return path is in a directory with appropriate
            # permissions
            self.__exit__(None, None, None)
            raise ValueError(f"Unexpected path to unix socket: {self.remote}")

sd_notify(f"STATUS=Running ({self.stage} stage)")
        return self

def __exit__(self, exc_type, exc_val, exc_tb):
        """Notify the socket that this stage is complete."""
        message = f"Completed socket interaction for boot stage {self.stage}"
        if exc_type:
            # handle exception thrown in context
            self.systemd_exit_code = 1
            self.experienced_any_error = True
            status = f"{repr(exc_val)} in {exc_tb.tb_frame}"
            message = (
                'fatal error, run "systemctl status cloud-init-main.service" '
                'and "cloud-init status --long" for more details'
            )
            if not self.first_exception:
                self.first_exception = status
            LOG.fatal(status)
            sd_notify(f"STATUS={status}")

self.experienced_any_error = self.experienced_any_error or bool(
            self.systemd_exit_code
        )
        sock = self.sockets[self.stage]
        sock.connect(self.remote)

# the returned message will be executed in a subshell
        # hardcode this message rather than sending a more informative message
        # to avoid having to sanitize inputs (to prevent escaping the shell)
        sock.sendall(
            f"echo '{message}'; exit {self.systemd_exit_code};".encode()
        )
        sock.close()

# suppress exception - the exception was logged and the init system
        # notified of stage completion (and the exception received as a status
        # message). Raising an exception would block the rest of boot, so carry
        # on in a degraded state.
        return True

Batal