# Authors: Karl MacMillan # # Copyright (C) 2007 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; version 2 only # # 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, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # import logging, sys import os import tempfile from ipapython import sysrestore from ipapython import ipautil def stop(service_name, instance_name=""): ipautil.run(["/sbin/service", service_name, "stop", instance_name]) def start(service_name, instance_name=""): ipautil.run(["/sbin/service", service_name, "start", instance_name]) def restart(service_name, instance_name=""): ipautil.run(["/sbin/service", service_name, "restart", instance_name]) def is_running(service_name, instance_name=""): ret = True try: ipautil.run(["/sbin/service", service_name, "status", instance_name]) except ipautil.CalledProcessError: ret = False return ret def chkconfig_on(service_name): ipautil.run(["/sbin/chkconfig", service_name, "on"]) def chkconfig_off(service_name): ipautil.run(["/sbin/chkconfig", service_name, "off"]) def chkconfig_add(service_name): ipautil.run(["/sbin/chkconfig", "--add", service_name]) def chkconfig_del(service_name): ipautil.run(["/sbin/chkconfig", "--del", service_name]) def is_enabled(service_name): (stdout, stderr) = ipautil.run(["/sbin/chkconfig", "--list", service_name]) runlevels = {} for runlevel in range(0, 7): runlevels[runlevel] = False for line in stdout.split("\n"): parts = line.split() if parts[0] == service_name: for s in parts[1:]: (runlevel, status) = s.split(":")[0:2] try: runlevels[int(runlevel)] = status == "on" except ValueError: pass break return (runlevels[3] and runlevels[4] and runlevels[5]) def print_msg(message, output_fd=sys.stdout): logging.debug(message) output_fd.write(message) output_fd.write("\n") class Service: def __init__(self, service_name, sstore=None, dm_password=None): self.service_name = service_name self.steps = [] self.output_fd = sys.stdout self.dm_password = dm_password if sstore: self.sstore = sstore else: self.sstore = sysrestore.StateFile('/var/lib/ipa/sysrestore') def _ldap_mod(self, ldif, sub_dict = None): assert self.dm_password is not None fd = None path = ipautil.SHARE_DIR + ldif if sub_dict is not None: txt = ipautil.template_file(path, sub_dict) fd = ipautil.write_tmp_file(txt) path = fd.name [pw_fd, pw_name] = tempfile.mkstemp() os.write(pw_fd, self.dm_password) os.close(pw_fd) args = ["/usr/bin/ldapmodify", "-h", "127.0.0.1", "-xv", "-D", "cn=Directory Manager", "-y", pw_name, "-f", path] try: try: ipautil.run(args) except ipautil.CalledProcessError, e: logging.critical("Failed to load %s: %s" % (ldif, str(e))) finally: os.remove(pw_name) if fd is not None: fd.close() def set_output(self, fd): self.output_fd = fd def stop(self, instance_name=""): stop(self.service_name, instance_name) def start(self, instance_name=""): start(self.service_name, instance_name) def restart(self, instance_name=""): restart(self.service_name, instance_name) def is_running(self): return is_running(self.service_name) def chkconfig_add(self): chkconfig_add(self.service_name) def chkconfig_del(self): chkconfig_del(self.service_name) def chkconfig_on(self): chkconfig_on(self.service_name) def chkconfig_off(self): chkconfig_off(self.service_name) def is_enabled(self): return is_enabled(self.service_name) def backup_state(self, key, value): self.sstore.backup_state(self.service_name, key, value) def restore_state(self, key): return self.sstore.restore_state(self.service_name, key) def print_msg(self, message): print_msg(message, self.output_fd) def step(self, message, method): self.steps.append((message, method)) def start_creation(self, message): self.print_msg(message) step = 0 for (message, method) in self.steps: self.print_msg(" [%d/%d]: %s" % (step+1, len(self.steps), message)) method() step += 1 self.print_msg("done configuring %s." % self.service_name) self.steps = [] class SimpleServiceInstance(Service): def create_instance(self): self.step("starting %s " % self.service_name, self.__start) self.step("configuring %s to start on boot" % self.service_name, self.__enable) self.start_creation("Configuring %s" % self.service_name) def __start(self): self.backup_state("running", self.is_running()) self.restart() def __enable(self): self.chkconfig_add() self.backup_state("enabled", self.is_enabled()) self.chkconfig_on() def uninstall(self): running = self.restore_state("running") enabled = not self.restore_state("enabled") if not running is None and not running: self.stop() if not enabled is None and not enabled: self.chkconfig_off() self.chkconfig_del()