mirror of
https://salsa.debian.org/freeipa-team/freeipa.git
synced 2025-01-12 09:11:55 -06:00
205 lines
8.5 KiB
Python
205 lines
8.5 KiB
Python
|
# Author: Alexander Bokovoy <abokovoy@redhat.com>
|
||
|
|
#
|
||
|
|
# Copyright (C) 2011 Red Hat
|
||
|
|
# see file 'COPYING' for use and warranty information
|
||
|
|
#
|
||
|
|
# This program is free software; you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU General Public License as published by
|
||
|
|
# the Free Software Foundation, either version 3 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
#
|
||
|
|
# This program is distributed in the hope that it will be useful,
|
||
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
# GNU General Public License for more details.
|
||
|
|
#
|
||
|
|
# You should have received a copy of the GNU General Public License
|
||
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
#
|
||
|
|
|
||
|
|
from ipapython import ipautil
|
||
|
|
from ipapython.platform import base
|
||
|
|
import sys, os, shutil
|
||
|
|
|
||
|
|
class SystemdService(base.PlatformService):
|
||
|
|
SYSTEMD_ETC_PATH = "/etc/systemd/system/"
|
||
|
|
SYSTEMD_LIB_PATH = "/lib/systemd/system/"
|
||
|
|
SYSTEMD_SRV_TARGET = "%s.target.wants"
|
||
|
|
|
||
|
|
def __init__(self, service_name):
|
||
|
|
super(SystemdService, self).__init__(service_name)
|
||
|
|
self.lib_path = os.path.join(self.SYSTEMD_LIB_PATH, self.service_name)
|
||
|
|
self.lib_path_exists = None
|
||
|
|
|
||
|
|
def service_instance(self, instance_name):
|
||
|
|
if self.lib_path_exists is None:
|
||
|
|
self.lib_path_exists = os.path.exists(self.lib_path)
|
||
|
|
|
||
|
|
elements = self.service_name.split("@")
|
||
|
|
|
||
|
|
# Short-cut: if there is already exact service name, return it
|
||
|
|
if self.lib_path_exists and len(instance_name) == 0:
|
||
|
|
if len(elements) == 1:
|
||
|
|
# service name is like pki-cad.target or krb5kdc.service
|
||
|
|
return self.service_name
|
||
|
|
if len(elements) > 1 and elements[1][0] != '.':
|
||
|
|
# Service name is like pki-cad@pki-ca.service and that file exists
|
||
|
|
return self.service_name
|
||
|
|
|
||
|
|
if len(elements) > 1:
|
||
|
|
# We have dynamic service
|
||
|
|
if len(instance_name) > 0:
|
||
|
|
# Instanciate dynamic service
|
||
|
|
return "%s@%s.service" % (elements[0], instance_name)
|
||
|
|
else:
|
||
|
|
# No instance name, try with target
|
||
|
|
tgt_name = "%s.target" % (elements[0])
|
||
|
|
srv_lib = os.path.join(self.SYSTEMD_LIB_PATH, tgt_name)
|
||
|
|
if os.path.exists(srv_lib):
|
||
|
|
return tgt_name
|
||
|
|
|
||
|
|
return self.service_name
|
||
|
|
|
||
|
|
def parse_variables(self, text, separator=None):
|
||
|
|
"""
|
||
|
|
Parses 'systemctl show' output and returns a dict[variable]=value
|
||
|
|
Arguments: text -- 'systemctl show' output as string
|
||
|
|
separator -- optional (defaults to None), what separates the key/value pairs in the text
|
||
|
|
"""
|
||
|
|
def splitter(x, separator=None):
|
||
|
|
if len(x) > 1:
|
||
|
|
y = x.split(separator)
|
||
|
|
return (y[0], y[-1])
|
||
|
|
return (None,None)
|
||
|
|
return dict(map(lambda x: splitter(x, separator=separator), text.split("\n")))
|
||
|
|
|
||
|
|
def stop(self, instance_name="", capture_output=True):
|
||
|
|
ipautil.run(["/bin/systemctl", "stop", self.service_instance(instance_name)], capture_output=capture_output)
|
||
|
|
|
||
|
|
def start(self, instance_name="", capture_output=True):
|
||
|
|
ipautil.run(["/bin/systemctl", "start", self.service_instance(instance_name)], capture_output=capture_output)
|
||
|
|
|
||
|
|
def restart(self, instance_name="", capture_output=True):
|
||
|
|
# Restart command is broken before systemd-36-3.fc16
|
||
|
|
# If you have older systemd version, restart of dependent services will hang systemd indefinetly
|
||
|
|
ipautil.run(["/bin/systemctl", "restart", self.service_instance(instance_name)], capture_output=capture_output)
|
||
|
|
|
||
|
|
def is_running(self, instance_name=""):
|
||
|
|
ret = True
|
||
|
|
try:
|
||
|
|
(sout, serr, rcode) = ipautil.run(["/bin/systemctl", "is-active", self.service_instance(instance_name)],capture_output=True)
|
||
|
|
if rcode != 0:
|
||
|
|
ret = False
|
||
|
|
except ipautil.CalledProcessError:
|
||
|
|
ret = False
|
||
|
|
return ret
|
||
|
|
|
||
|
|
def is_installed(self):
|
||
|
|
installed = True
|
||
|
|
try:
|
||
|
|
(sout,serr,rcode) = ipautil.run(["/bin/systemctl", "list-unit-files", "--full"])
|
||
|
|
if rcode != 0:
|
||
|
|
installed = False
|
||
|
|
else:
|
||
|
|
svar = self.parse_variables(sout)
|
||
|
|
if not self.service_instance("") in svar:
|
||
|
|
# systemd doesn't show the service
|
||
|
|
installed = False
|
||
|
|
except ipautil.CalledProcessError, e:
|
||
|
|
installed = False
|
||
|
|
return installed
|
||
|
|
|
||
|
|
def is_enabled(self, instance_name=""):
|
||
|
|
enabled = True
|
||
|
|
try:
|
||
|
|
(sout,serr,rcode) = ipautil.run(["/bin/systemctl", "is-enabled", self.service_instance(instance_name)])
|
||
|
|
if rcode != 0:
|
||
|
|
enabled = False
|
||
|
|
except ipautil.CalledProcessError, e:
|
||
|
|
enabled = False
|
||
|
|
return enabled
|
||
|
|
|
||
|
|
def enable(self, instance_name=""):
|
||
|
|
if self.lib_path_exists is None:
|
||
|
|
self.lib_path_exists = os.path.exists(self.lib_path)
|
||
|
|
elements = self.service_name.split("@")
|
||
|
|
l = len(elements)
|
||
|
|
|
||
|
|
if self.lib_path_exists and (l > 1 and elements[1][0] != '.'):
|
||
|
|
# There is explicit service unit supporting this instance, follow normal systemd enabler
|
||
|
|
self.__enable(instance_name)
|
||
|
|
return
|
||
|
|
|
||
|
|
if self.lib_path_exists and (l == 1):
|
||
|
|
# There is explicit service unit which does not support the instances, ignore instance
|
||
|
|
self.__enable()
|
||
|
|
return
|
||
|
|
|
||
|
|
if len(instance_name) > 0 and l > 1:
|
||
|
|
# New instance, we need to do following:
|
||
|
|
# 1. Copy <service>@.service to /etc/systemd/system/ if it is not there
|
||
|
|
# 2. Make /etc/systemd/system/<service>.target.wants/ if it is not there
|
||
|
|
# 3. Link /etc/systemd/system/<service>.target.wants/<service>@<instance_name>.service to
|
||
|
|
# /etc/systemd/system/<service>@.service
|
||
|
|
srv_etc = os.path.join(self.SYSTEMD_ETC_PATH, self.service_name)
|
||
|
|
srv_tgt = os.path.join(self.SYSTEMD_ETC_PATH, self.SYSTEMD_SRV_TARGET % (elements[0]))
|
||
|
|
srv_lnk = os.path.join(srv_tgt, self.service_instance(instance_name))
|
||
|
|
try:
|
||
|
|
if not ipautil.file_exists(srv_etc):
|
||
|
|
shutil.copy(self.lib_path, srv_etc)
|
||
|
|
if not ipautil.dir_exists(srv_tgt):
|
||
|
|
os.mkdir(srv_tgt)
|
||
|
|
if os.path.exists(srv_lnk):
|
||
|
|
# Remove old link
|
||
|
|
os.unlink(srv_lnk)
|
||
|
|
if not os.path.exists(srv_lnk):
|
||
|
|
# object does not exist _or_ is a broken link
|
||
|
|
if not os.path.islink(srv_lnk):
|
||
|
|
# if it truly does not exist, make a link
|
||
|
|
os.symlink(srv_etc, srv_lnk)
|
||
|
|
else:
|
||
|
|
# Link exists and it is broken, make new one
|
||
|
|
os.unlink(srv_lnk)
|
||
|
|
os.symlink(srv_etc, srv_lnk)
|
||
|
|
ipautil.run(["/bin/systemctl", "--system", "daemon-reload"])
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
self.__enable(instance_name)
|
||
|
|
|
||
|
|
def disable(self, instance_name=""):
|
||
|
|
elements = self.service_name.split("@")
|
||
|
|
if instance_name != "" and len(elements) > 1:
|
||
|
|
# Remove instance, we need to do following:
|
||
|
|
# Remove link from /etc/systemd/system/<service>.target.wants/<service>@<instance_name>.service
|
||
|
|
# to /etc/systemd/system/<service>@.service
|
||
|
|
srv_tgt = os.path.join(self.SYSTEMD_ETC_PATH, self.SYSTEMD_SRV_TARGET % (elements[0]))
|
||
|
|
srv_lnk = os.path.join(srv_tgt, self.service_instance(instance_name))
|
||
|
|
try:
|
||
|
|
if ipautil.dir_exists(srv_tgt):
|
||
|
|
if os.path.islink(srv_lnk):
|
||
|
|
os.unlink(srv_lnk)
|
||
|
|
ipautil.run(["/bin/systemctl", "--system", "daemon-reload"])
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
self.__disable(instance_name)
|
||
|
|
|
||
|
|
def __enable(self, instance_name=""):
|
||
|
|
try:
|
||
|
|
ipautil.run(["/bin/systemctl", "enable", self.service_instance(instance_name)])
|
||
|
|
except ipautil.CalledProcessError, e:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def __disable(self, instance_name=""):
|
||
|
|
try:
|
||
|
|
ipautil.run(["/bin/systemctl", "disable", self.service_instance(instance_name)])
|
||
|
|
except ipautil.CalledProcessError, e:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def install(self):
|
||
|
|
self.enable()
|
||
|
|
|
||
|
|
def remove(self):
|
||
|
|
self.disable()
|