| Server IP : 52.223.31.75 / Your IP : 172.31.6.220 [ Web Server : Apache/2.4.66 () OpenSSL/1.0.2k-fips PHP/7.4.33 System : Linux ip-172-31-14-81.eu-central-1.compute.internal 4.14.281-212.502.amzn2.x86_64 #1 SMP Thu May 26 09:52:17 UTC 2022 x86_64 User : apache ( 48) PHP Version : 7.4.33 Disable Function : NONE Domains : 4 Domains MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : OFF Directory : /lib/python3.7/site-packages/cfnbootstrap/ |
Upload File : |
#==============================================================================
# Copyright 2011 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#==============================================================================
from cfnbootstrap import util
from cfnbootstrap.construction_errors import ToolError
from cfnbootstrap.util import ProcessHelper
import collections
import logging
import os.path
log = logging.getLogger("cfn.init")
_windows_supported = True
try:
import win32service
import win32serviceutil
except ImportError:
_windows_supported = False
class ServiceTool(object):
def _detect_required_restart(self, serviceProperties, changes):
if self._list_type_change_occurred(serviceProperties, changes, 'files'):
return True
if self._list_type_change_occurred(serviceProperties, changes, 'sources'):
return True
if self._list_type_change_occurred(serviceProperties, changes, 'groups'):
return True
if self._list_type_change_occurred(serviceProperties, changes, 'users'):
return True
if self._list_type_change_occurred(serviceProperties, changes, 'commands'):
return True
if "packages" in serviceProperties and "packages" in changes:
for manager, pkg_list in changes["packages"].items():
if manager in serviceProperties["packages"]:
if frozenset(
serviceProperties["packages"][manager]
) & frozenset(pkg_list):
return True
return False
def _list_type_change_occurred(self, serviceProperties, changes, key):
if key in serviceProperties and key in changes:
if frozenset(serviceProperties[key]) & frozenset(changes[key]):
return True
return False
class WindowsServiceTool(ServiceTool):
"""
Manages Windows services
"""
def apply(self, action, changes = collections.defaultdict(list)):
"""
Takes a dict of service name to dict.
Keys we look for are:
- "enabled" (setting a service to "Automatic")
- "ensureRunning" (actually start the service)
"""
if not action.keys():
log.debug("No Windows services specified")
return
if not _windows_supported:
raise ToolError("Cannot modify windows services without pywin32")
manager = win32service.OpenSCManager(None, None, win32service.SC_MANAGER_ALL_ACCESS)
try:
for service, serviceProperties in action.items():
handle = win32service.OpenService(manager, service, win32service.SERVICE_ALL_ACCESS)
try:
if "enabled" in serviceProperties:
start_type = win32service.SERVICE_AUTO_START if util.interpret_boolean(serviceProperties["enabled"]) else win32service.SERVICE_DEMAND_START
self._set_service_startup_type(handle, start_type)
else:
log.debug("Not modifying enabled state of service %s", service)
if self._detect_required_restart(serviceProperties, changes):
log.debug("Restarting %s due to change detected in dependency", service)
win32serviceutil.RestartService(service)
elif "ensureRunning" in serviceProperties:
ensureRunning = util.interpret_boolean(serviceProperties["ensureRunning"])
status = win32service.QueryServiceStatus(handle)[1]
isRunning = status & win32service.SERVICE_RUNNING or status & win32service.SERVICE_START_PENDING
if ensureRunning and not isRunning:
log.debug("Starting service %s as it is not running", service)
win32service.StartService(handle, None)
elif not ensureRunning and isRunning:
log.debug("Stopping service %s as it is running", service)
win32service.ControlService(handle, win32service.SERVICE_CONTROL_STOP)
else:
log.debug("No need to modify running state of service %s", service)
else:
log.debug("Not modifying running state of service %s", service)
finally:
win32service.CloseServiceHandle(handle)
finally:
win32service.CloseServiceHandle(manager)
def _set_service_startup_type(self, service, start_type):
win32service.ChangeServiceConfig(service,
win32service.SERVICE_NO_CHANGE,
start_type,
win32service.SERVICE_NO_CHANGE,
None,
None,
0,
None,
None,
None,
None)
class SysVInitTool(ServiceTool):
"""
Manages SysV Init services
"""
def apply(self, action, changes = collections.defaultdict(list)):
"""
Takes a dict of service name to dict.
Keys we look for are:
- "enabled" (equivalent to /sbin/chkconfig service on)
- "ensureRunning" (equivalent to /sbin/service service start)
"""
if not action.keys():
log.debug("No System V init scripts specified")
return
for service, serviceProperties in action.items():
force_restart = self._detect_required_restart(serviceProperties, changes)
if "enabled" in serviceProperties:
self._set_service_enabled(service, util.interpret_boolean(serviceProperties["enabled"]))
else:
log.debug("Not modifying enabled state of service %s", service)
if force_restart:
log.debug("Restarting %s due to change detected in dependency", service)
self._restart_service(service)
elif "ensureRunning" in serviceProperties:
ensureRunning = util.interpret_boolean(serviceProperties["ensureRunning"])
isRunning = self._is_service_running(service)
if ensureRunning and not isRunning:
log.debug("Starting service %s as it is not running", service)
self._start_service(service)
elif not ensureRunning and isRunning:
log.debug("Stopping service %s as it is running", service)
self._stop_service(service)
else:
log.debug("No need to modify running state of service %s", service)
else:
log.debug("Not modifying running state of service %s", service)
def _restart_service(self, service):
cmd = self._get_service_executable(service)
cmd.append("restart")
result = ProcessHelper(cmd).call()
if result.returncode:
log.error("Could not restart service %s; return code was %s", service, result.returncode)
log.debug("Service output: %s", result.stdout)
raise ToolError("Could not restart %s" % service)
else:
log.info("Restarted %s successfully", service)
def _start_service(self, service):
cmd = self._get_service_executable(service)
cmd.append("start")
result = ProcessHelper(cmd).call()
if result.returncode:
log.error("Could not start service %s; return code was %s", service, result.returncode)
log.debug("Service output: %s", result.stdout)
raise ToolError("Could not start %s" % service)
else:
log.info("Started %s successfully", service)
def _stop_service(self, service):
cmd = self._get_service_executable(service)
cmd.append("stop")
result = ProcessHelper(cmd).call()
if result.returncode:
log.error("Could not stop service %s; return code was %s", service, result.returncode)
log.debug("Service output: %s", result.stdout)
raise ToolError("Could not stop %s" % service)
else:
log.info("Stopped %s successfully", service)
def _is_service_running(self, service):
cmd = self._get_service_executable(service)
cmd.append("status")
result = ProcessHelper(cmd).call()
if result.returncode:
return False
return True
def _set_service_enabled(self, service, enabled=True):
modifier = self._get_service_modifier()
if not modifier:
log.error("Could not enable %s, as chkconfig and update-rc.d were not available", service)
return
log.debug("Setting service %s to %s", service, "enabled" if enabled else "disabled")
modifier.set_service_enabled(service, enabled);
def _get_service_modifier(self):
if hasattr(self, '_cached_modifier'):
return self._cached_modifier
if Chkconfig.installed():
self._cached_modifier = Chkconfig()
elif UpdateRcD.installed():
self._cached_modifier = UpdateRcD()
else:
self._cached_modifier = None
log.debug("Using service modifier: %s", self._cached_modifier)
return self._cached_modifier
def _get_service_executable(self, service):
"""
Returns the service executable as an array
Right now either ["/sbin/service", service], ["/usr/sbin/service", service], or ["/etc/init.d/<service>"]
"""
if not hasattr(self, '_service_runner'):
if os.path.exists("/sbin/service"):
self._service_runner = "/sbin/service"
elif os.path.exists("/usr/sbin/service"):
self._service_runner = "/usr/sbin/service"
else:
self._service_runner = None
if self._service_runner:
log.debug("Using service runner: %s", self._service_runner)
else:
log.debug("Running init scripts directly")
if self._service_runner:
return [self._service_runner, service]
else:
return ["/etc/init.d/%s" % service]
class Chkconfig(object):
_executable = "/sbin/chkconfig"
@classmethod
def installed(cls):
return os.path.exists(cls._executable)
def __init__(self):
#Leaving this open for multiple locations of chkconfig
self._executable = Chkconfig._executable
def __str__(self, *args, **kwargs):
return self._executable
def set_service_enabled(self, service, enabled=True):
if not os.path.exists(self._executable):
raise ToolError("Cannot find chkconfig")
result = ProcessHelper([self._executable, service, 'on' if enabled else 'off']).call()
if result.returncode:
log.error("chkconfig failed with error %s. Output: %s", result.returncode, result.stdout)
raise ToolError("Could not %s service %s" % ("enable" if enabled else "disable", service), result.returncode)
else:
log.info("%s service %s", "enabled" if enabled else "disabled", service)
class UpdateRcD(object):
_executable = "/usr/sbin/update-rc.d"
@classmethod
def installed(cls):
return os.path.exists(cls._executable)
def __init__(self):
#Leaving this open for multiple locations of update-rc.d
self._executable = UpdateRcD._executable
def __str__(self, *args, **kwargs):
return self._executable
def set_service_enabled(self, service, enabled=True):
if not os.path.exists(self._executable):
raise ToolError("Cannot find update-rc.d")
result = ProcessHelper([self._executable, service, 'enable' if enabled else 'disable']).call()
if result.returncode:
log.error("update-rc.d failed with error %s. Output: %s", result.returncode, result.stdout)
raise ToolError("Could not %s service %s" % ("enable" if enabled else "disable", service), result.returncode)
else:
log.info("%s service %s", "enabled" if enabled else "disabled", service)