OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	10/02/2024 07:52:55 PM	rwxr-xr-x
📄 __init__.py	1.4 KB	03/19/2021 02:37:22 PM	rw-r--r--
📁 __pycache__	-	10/02/2024 07:52:55 PM	rwxr-xr-x
📄 cc_apk_configure.py	9.35 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_apt_configure.py	39.92 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_apt_pipelining.py	2.44 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_bootcmd.py	3.53 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_byobu.py	3.13 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_ca_certs.py	6.37 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_chef.py	23.6 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_debug.py	3.08 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_disable_ec2_metadata.py	1.59 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_disk_setup.py	33.08 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_emit_upstart.py	2 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_fan.py	2.85 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_final_message.py	2.37 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_foo.py	2.07 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_growpart.py	11.63 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_grub_dpkg.py	4.94 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_keys_to_console.py	2.79 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_landscape.py	3.96 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_locale.py	2.22 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_lxd.py	10.51 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_mcollective.py	5.08 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_migrator.py	3.07 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_mounts.py	17.75 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_ntp.py	21.59 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_package_update_upgrade_install.py	4.14 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_phone_home.py	4.03 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_power_state_change.py	7.48 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_puppet.py	8.86 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_refresh_rmc_and_interface.py	5.07 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_reset_rmc.py	4.15 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_resizefs.py	9.73 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_resolv_conf.py	3.45 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_rh_subscription.py	15.68 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_rightscale_userdata.py	3.75 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_rsyslog.py	14.12 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_runcmd.py	3.11 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_salt_minion.py	4.75 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_scripts_per_boot.py	1.2 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_scripts_per_instance.py	1.38 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_scripts_per_once.py	1.31 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_scripts_user.py	1.42 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_scripts_vendor.py	1.41 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_seed_random.py	4.57 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_set_hostname.py	3.39 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_set_passwords.py	8.53 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_snap.py	8.68 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_spacewalk.py	2.89 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_ssh.py	12.76 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_ssh_authkey_fingerprints.py	3.44 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_ssh_import_id.py	2.91 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_timezone.py	1.15 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_ubuntu_advantage.py	6.08 KB	04/25/2023 09:28:10 PM	rw-r--r--
📄 cc_ubuntu_drivers.py	5.69 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_update_etc_hosts.py	3.33 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_update_hostname.py	1.58 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_users_groups.py	7.44 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_write_files.py	8.83 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_yum_add_repo.py	4.21 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 cc_zypper_add_repo.py	7.58 KB	03/19/2021 02:37:22 PM	rw-r--r--
📄 schema.py	19.05 KB	03/19/2021 02:37:22 PM	rw-r--r--

Editing: cc_power_state_change.py

# Copyright (C) 2011 Canonical Ltd.
#
# Author: Scott Moser <scott.moser@canonical.com>
#
# This file is part of cloud-init. See LICENSE file for license information.

"""
Power State Change
------------------
**Summary:** change power state

This module handles shutdown/reboot after all config modules have been run. By
default it will take no action, and the system will keep running unless a
package installation/upgrade requires a system reboot (e.g. installing a new
kernel) and ``package_reboot_if_required`` is true. The ``power_state`` config
key accepts a dict of options. If ``mode`` is any value other than
``poweroff``, ``halt``, or ``reboot``, then no action will be taken.

The system
can be shutdown before cloud-init has finished using the ``timeout`` option.
The ``delay`` key specifies a duration to be added onto any shutdown command
used. Therefore, if a 5 minute delay and a 120 second shutdown are specified,
the maximum amount of time between cloud-init starting and the system shutting
down is 7 minutes, and the minimum amount of time is 5 minutes. The ``delay``
key must have an argument in either the form ``'+5'`` for 5 minutes or ``now``
for immediate shutdown.

Optionally, a command can be run to determine whether or not
the system should shut down. The command to be run should be specified in the
``condition`` key. For command formatting, see the documentation for
``cc_runcmd``. The specified shutdown behavior will only take place if the
``condition`` key is omitted or the command specified by the ``condition``
key returns 0.

.. note::
    With Alpine Linux any message value specified is ignored as Alpine's halt,
    poweroff, and reboot commands do not support broadcasting a message.

**Internal name:** ``cc_power_state_change``

**Module frequency:** per instance

**Supported distros:** all

**Config keys**::

power_state:
        delay: <now/'+minutes'>
        mode: <poweroff/halt/reboot>
        message: <shutdown message>
        timeout: <seconds>
        condition: <true/false/command>
"""

import errno
import os
import re
import subprocess
import time

from cloudinit.settings import PER_INSTANCE
from cloudinit import subp
from cloudinit import util

frequency = PER_INSTANCE

EXIT_FAIL = 254

def givecmdline(pid):
    # Returns the cmdline for the given process id. In Linux we can use procfs
    # for this but on BSD there is /usr/bin/procstat.
    try:
        # Example output from procstat -c 1
        #   PID COMM             ARGS
        #     1 init             /bin/init --
        if util.is_FreeBSD():
            (output, _err) = subp.subp(['procstat', '-c', str(pid)])
            line = output.splitlines()[1]
            m = re.search(r'\d+ (\w|\.|-)+\s+(/\w.+)', line)
            return m.group(2)
        else:
            return util.load_file("/proc/%s/cmdline" % pid)
    except IOError:
        return None

def check_condition(cond, log=None):
    if isinstance(cond, bool):
        if log:
            log.debug("Static Condition: %s" % cond)
        return cond

pre = "check_condition command (%s): " % cond
    try:
        proc = subprocess.Popen(cond, shell=not isinstance(cond, list))
        proc.communicate()
        ret = proc.returncode
        if ret == 0:
            if log:
                log.debug(pre + "exited 0. condition met.")
            return True
        elif ret == 1:
            if log:
                log.debug(pre + "exited 1. condition not met.")
            return False
        else:
            if log:
                log.warning(pre + "unexpected exit %s. " % ret +
                            "do not apply change.")
            return False
    except Exception as e:
        if log:
            log.warning(pre + "Unexpected error: %s" % e)
        return False

def handle(_name, cfg, cloud, log, _args):
    try:
        (args, timeout, condition) = load_power_state(cfg, cloud.distro)
        if args is None:
            log.debug("no power_state provided. doing nothing")
            return
    except Exception as e:
        log.warning("%s Not performing power state change!" % str(e))
        return

if condition is False:
        log.debug("Condition was false. Will not perform state change.")
        return

mypid = os.getpid()

cmdline = givecmdline(mypid)
    if not cmdline:
        log.warning("power_state: failed to get cmdline of current process")
        return

devnull_fp = open(os.devnull, "w")

log.debug("After pid %s ends, will execute: %s" % (mypid, ' '.join(args)))

util.fork_cb(run_after_pid_gone, mypid, cmdline, timeout, log,
                 condition, execmd, [args, devnull_fp])

def load_power_state(cfg, distro):
    # returns a tuple of shutdown_command, timeout
    # shutdown_command is None if no config found
    pstate = cfg.get('power_state')

if pstate is None:
        return (None, None, None)

if not isinstance(pstate, dict):
        raise TypeError("power_state is not a dict.")

modes_ok = ['halt', 'poweroff', 'reboot']
    mode = pstate.get("mode")
    if mode not in distro.shutdown_options_map:
        raise TypeError(
            "power_state[mode] required, must be one of: %s. found: '%s'." %
            (','.join(modes_ok), mode))

args = distro.shutdown_command(mode=mode,
                                   delay=pstate.get("delay", "now"),
                                   message=pstate.get("message"))

try:
        timeout = float(pstate.get('timeout', 30.0))
    except ValueError as e:
        raise ValueError(
            "failed to convert timeout '%s' to float." % pstate['timeout']
        ) from e

condition = pstate.get("condition", True)
    if not isinstance(condition, (str, list, bool)):
        raise TypeError("condition type %s invalid. must be list, bool, str")
    return (args, timeout, condition)

def doexit(sysexit):
    os._exit(sysexit)

def execmd(exe_args, output=None, data_in=None):
    ret = 1
    try:
        proc = subprocess.Popen(exe_args, stdin=subprocess.PIPE,
                                stdout=output, stderr=subprocess.STDOUT)
        proc.communicate(data_in)
        ret = proc.returncode
    except Exception:
        doexit(EXIT_FAIL)
    doexit(ret)

def run_after_pid_gone(pid, pidcmdline, timeout, log, condition, func, args):
    # wait until pid, with /proc/pid/cmdline contents of pidcmdline
    # is no longer alive.  After it is gone, or timeout has passed
    # execute func(args)
    msg = None
    end_time = time.time() + timeout

def fatal(msg):
        if log:
            log.warning(msg)
        doexit(EXIT_FAIL)

known_errnos = (errno.ENOENT, errno.ESRCH)

while True:
        if time.time() > end_time:
            msg = "timeout reached before %s ended" % pid
            break

try:
            cmdline = givecmdline(pid)
            if cmdline != pidcmdline:
                msg = "cmdline changed for %s [now: %s]" % (pid, cmdline)
                break

except IOError as ioerr:
            if ioerr.errno in known_errnos:
                msg = "pidfile gone [%d]" % ioerr.errno
            else:
                fatal("IOError during wait: %s" % ioerr)
            break

except Exception as e:
            fatal("Unexpected Exception: %s" % e)

time.sleep(.25)

if not msg:
        fatal("Unexpected error in run_after_pid_gone")

if log:
        log.debug(msg)

try:
        if not check_condition(condition, log):
            return
    except Exception as e:
        fatal("Unexpected Exception when checking condition: %s" % e)

func(*args)

# vi: ts=4 expandtab