mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
513b582ba3
This is part of the effort to make windows parameters conform to other modules. Usually parameters don't include the unit in the parameter name. See also #20160
175 lines
6.6 KiB
Python
175 lines
6.6 KiB
Python
# (c) 2016, Matt Davis <mdavis@ansible.com>
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import socket
|
|
import time
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
from ansible.plugins.action import ActionBase
|
|
|
|
try:
|
|
from __main__ import display
|
|
except ImportError:
|
|
from ansible.utils.display import Display
|
|
display = Display()
|
|
|
|
|
|
class TimedOutException(Exception):
|
|
pass
|
|
|
|
|
|
class ActionModule(ActionBase):
|
|
TRANSFERS_FILES = False
|
|
|
|
DEFAULT_SHUTDOWN_TIMEOUT = 600
|
|
DEFAULT_REBOOT_TIMEOUT = 600
|
|
DEFAULT_CONNECT_TIMEOUT = 5
|
|
DEFAULT_PRE_REBOOT_DELAY = 2
|
|
DEFAULT_POST_REBOOT_DELAY = 0
|
|
DEFAULT_TEST_COMMAND = 'whoami'
|
|
DEFAULT_REBOOT_MESSAGE = 'Reboot initiated by Ansible.'
|
|
|
|
def do_until_success_or_timeout(self, what, timeout, what_desc, fail_sleep=1):
|
|
max_end_time = datetime.utcnow() + timedelta(seconds=timeout)
|
|
|
|
e = None
|
|
while datetime.utcnow() < max_end_time:
|
|
try:
|
|
what()
|
|
if what_desc:
|
|
display.debug("win_reboot: %s success" % what_desc)
|
|
return
|
|
except Exception as e:
|
|
if what_desc:
|
|
display.debug("win_reboot: %s fail (expected), retrying in %d seconds..." % (what_desc, fail_sleep))
|
|
time.sleep(fail_sleep)
|
|
|
|
raise TimedOutException("timed out waiting for %s: %s" % (what_desc, e))
|
|
|
|
def run(self, tmp=None, task_vars=None):
|
|
|
|
self._supports_check_mode = True
|
|
self._supports_async = True
|
|
|
|
if self._play_context.check_mode:
|
|
return dict(changed=True, elapsed=0, rebooted=True)
|
|
|
|
if task_vars is None:
|
|
task_vars = dict()
|
|
|
|
result = super(ActionModule, self).run(tmp, task_vars)
|
|
|
|
if result.get('skipped', False) or result.get('failed', False):
|
|
return result
|
|
|
|
winrm_host = self._connection._winrm_host
|
|
winrm_port = self._connection._winrm_port
|
|
|
|
# Handle timeout parameters and its alias
|
|
if self._task.args.get('shutdown_timeout') is not None:
|
|
shutdown_timeout = int(self._task.args.get('shutdown_timeout', self.DEFAULT_SHUTDOWN_TIMEOUT))
|
|
else:
|
|
shutdown_timeout = int(self._task.args.get('shutdown_timeout_sec', self.DEFAULT_SHUTDOWN_TIMEOUT))
|
|
|
|
if self._task.args.get('reboot_timeout') is not None:
|
|
reboot_timeout = int(self._task.args.get('reboot_timeout', self.DEFAULT_REBOOT_TIMEOUT))
|
|
else:
|
|
reboot_timeout = int(self._task.args.get('reboot_timeout_sec', self.DEFAULT_REBOOT_TIMEOUT))
|
|
|
|
if self._task.args.get('connect_timeout') is not None:
|
|
connect_timeout = int(self._task.args.get('connect_timeout', self.DEFAULT_CONNECT_TIMEOUT))
|
|
else:
|
|
connect_timeout = int(self._task.args.get('connect_timeout_sec', self.DEFAULT_CONNECT_TIMEOUT))
|
|
|
|
if self._task.args.get('pre_reboot_delay') is not None:
|
|
pre_reboot_delay = int(self._task.args.get('pre_reboot_delay', self.DEFAULT_PRE_REBOOT_DELAY))
|
|
else:
|
|
pre_reboot_delay = int(self._task.args.get('pre_reboot_delay_sec', self.DEFAULT_PRE_REBOOT_DELAY))
|
|
|
|
if self._task.args.get('post_reboot_delay') is not None:
|
|
post_reboot_delay = int(self._task.args.get('post_reboot_delay', self.DEFAULT_POST_REBOOT_DELAY))
|
|
else:
|
|
post_reboot_delay = int(self._task.args.get('post_reboot_delay_sec', self.DEFAULT_POST_REBOOT_DELAY))
|
|
|
|
test_command = str(self._task.args.get('test_command', self.DEFAULT_TEST_COMMAND))
|
|
msg = str(self._task.args.get('msg', self.DEFAULT_REBOOT_MESSAGE))
|
|
|
|
# Initiate reboot
|
|
(rc, stdout, stderr) = self._connection.exec_command('shutdown /r /t %d /c "%s"' % (pre_reboot_delay, msg))
|
|
|
|
# Test for "A system shutdown has already been scheduled. (1190)" and handle it gracefully
|
|
if rc == 1190:
|
|
display.warning('A scheduled reboot was pre-empted by Ansible.')
|
|
|
|
# Try to abort (this may fail if it was already aborted)
|
|
(rc, stdout1, stderr1) = self._connection.exec_command('shutdown /a')
|
|
|
|
# Initiate reboot again
|
|
(rc, stdout2, stderr2) = self._connection.exec_command('shutdown /r /t %d' % pre_reboot_delay)
|
|
stdout += stdout1 + stdout2
|
|
stderr += stderr1 + stderr2
|
|
|
|
if rc != 0:
|
|
result['failed'] = True
|
|
result['rebooted'] = False
|
|
result['msg'] = "Shutdown command failed, error text was %s" % stderr
|
|
return result
|
|
|
|
def raise_if_port_open():
|
|
try:
|
|
sock = socket.create_connection((winrm_host, winrm_port), connect_timeout)
|
|
sock.close()
|
|
except:
|
|
return False
|
|
|
|
raise Exception("port is open")
|
|
|
|
start = datetime.now()
|
|
|
|
try:
|
|
self.do_until_success_or_timeout(raise_if_port_open, shutdown_timeout, what_desc="winrm port down")
|
|
|
|
def connect_winrm_port():
|
|
sock = socket.create_connection((winrm_host, winrm_port), connect_timeout)
|
|
sock.close()
|
|
|
|
self.do_until_success_or_timeout(connect_winrm_port, reboot_timeout, what_desc="winrm port up")
|
|
|
|
def run_test_command():
|
|
display.vvv("attempting post-reboot test command '%s'" % test_command)
|
|
# call connection reset between runs if it's there
|
|
try:
|
|
self._connection._reset()
|
|
except AttributeError:
|
|
pass
|
|
|
|
(rc, stdout, stderr) = self._connection.exec_command(test_command)
|
|
|
|
if rc != 0:
|
|
raise Exception('test command failed')
|
|
|
|
# FUTURE: ensure that a reboot has actually occurred by watching for change in last boot time fact
|
|
# FUTURE: add a stability check (system must remain up for N seconds) to deal with self-multi-reboot updates
|
|
|
|
self.do_until_success_or_timeout(run_test_command, reboot_timeout, what_desc="post-reboot test command success")
|
|
|
|
result['rebooted'] = True
|
|
result['changed'] = True
|
|
|
|
except TimedOutException as toex:
|
|
result['failed'] = True
|
|
result['rebooted'] = True
|
|
result['msg'] = toex.message
|
|
|
|
if post_reboot_delay != 0:
|
|
display.vvv("win_reboot: waiting an additional %d seconds" % post_reboot_delay)
|
|
time.sleep(post_reboot_delay)
|
|
|
|
elapsed = datetime.now() - start
|
|
result['elapsed'] = elapsed.seconds
|
|
|
|
return result
|