summaryrefslogtreecommitdiffstats
path: root/src/lib/Client/Redhat.py
blob: d68288b7967c1f2b0c5e75520aa150f55649ea9e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
# This is the bcfg2 support for redhat
# $Id$

'''This is redhat client support'''
__revision__ = '$Revision$'

from os import popen, system

from Bcfg2.Client.Toolset import Toolset

class ToolsetImpl(Toolset):
    '''This class implelements support for rpm packages and standard chkconfig services'''
    __name__ = 'Redhat'
    pkgtool = {'rpm':("rpm --oldpackage --replacepkgs --quiet -U %s", ("%s", ["url"]))}

    def __init__(self, cfg, setup):
        Toolset.__init__(self, cfg, setup)
        self.pkgwork = {'add':[], 'update':[], 'remove':[]}
        self.Refresh()
        for pkg in [cpkg for cpkg in self.cfg.findall(".//Package") if not cpkg.attrib.has_key('type')]:
            pkg.set('type', 'rpm')
        for srv in [csrv for csrv in self.cfg.findall(".//Service") if not csrv.attrib.has_key('type')]:
            srv.set('type', 'chkconfig')
        # relocation hack: we will define one pkgtool per relocation location
        for pkg in [cpkg for cpkg in self.cfg.findall('.//Package') if cpkg.attrib.has_key('reloc')]:
            ptoolname = "rpm-reloc-%s" % (pkg.get('reloc'))
            if not self.pkgtool.has_key(ptoolname):
                cmd = "rpm --relocate %s --oldpackage --replacepkgs --quiet -U %%s" % (pkg.get('reloc'))
                self.pkgtool[ptoolname] = (cmd, ("%s", ["url"]))
            pkg.set('type', ptoolname)


    def Refresh(self):
        '''Refresh memory hashes of packages'''
        self.installed = {}

        # Build list of packages
        instp = popen("rpm -qa --qf '%{NAME} %{VERSION}-%{RELEASE}\n'")
        for line in instp:
            [name, version] = line.split(' ')
            self.installed[name] = version[:-1]

    def VerifyService(self, entry):
        '''Verify Service status for entry'''
        try:
            srvdata = popen("/sbin/chkconfig --list %s"%entry.attrib['name']).readlines()[0].split()
        except IndexError:
            # Ocurrs when no lines are returned (service not installed)
            return False
        if entry.attrib['type'] == 'xinetd':
            return entry.attrib['status'] == srvdata[1]

        onlevels = [level.split(':')[0] for level in srvdata[1:] if level.split(':')[1] == 'on']

        # chkconfig/init.d service
        if entry.get('status') == 'on':
            return len(onlevels) > 0
        else:
            return len(onlevels) == 0
    
    def InstallService(self, entry):
        '''Install Service entry'''
        system("/sbin/chkconfig --add %s"%(entry.attrib['name']))
        self.CondPrint('verbose', "Installing Service %s" % (entry.get('name')))
        if not entry.get('status'):
            print "Can't install service %s, not enough data" % (entry.get('name'))
            return False
        if entry.attrib['status'] == 'off':
            if self.setup['dryrun']:
                print "Disabling server %s" % (entry.get('name'))
            else:
                cmdrc = system("/sbin/chkconfig %s %s" % (entry.attrib['name'], entry.attrib['status']))
        else:
            if self.setup['dryrun']:
                print "Enabling server %s" % (entry.get('name'))
            else:
                cmdrc = system("/sbin/chkconfig %s %s" %
                            (entry.attrib['name'], entry.attrib['status']))
        return cmdrc == 0

    def VerifyPackage(self, entry, modlist):
        '''Verify Package status for entry'''
        if not entry.get('version'):
            print "Can't install package %s, not enough data." % (entry.get('name'))
            return False
        if self.installed.has_key(entry.get('name')):
            if entry.get('version') == self.installed[entry.get('name')]:
                if (self.setup['quick'] or (entry.get('verify', 'true') == 'false')):
                    return True
            else:
                self.CondPrint('debug', "Package %s: wrong version installed. want %s have %s" %
                               (entry.get('name'), entry.get('version'), self.installed[entry.get('name')]))
                return False
        else:
            self.CondPrint('debug', "Package %s: not installed" % (entry.get('name')))
            return False

        (vstat, output) = self.saferun("rpm --verify -q %s-%s" % (entry.get('name'), entry.get('version')))
        if vstat != 0:
            if [name for name in output if name.split()[-1] not in modlist]:
                self.CondPrint('debug',
                               "Package %s content verification failed" % entry.get('name'))
                return False
        return True

    def HandleExtra(self):
        '''Deal with extra configuration detected'''
        if len(self.pkgwork) > 0:
            if self.setup['remove'] in ['all', 'packages']:
                self.CondPrint('verbose', "Removing packages: %s" % self.pkgwork['remove'])
                if not system("rpm --quiet -e %s" % " ".join(self.pkgwork['remove'])):
                    self.pkgwork['remove'] = []
                    self.Inventory()
            else:
                self.CondDisplayList('verbose', "Need to remove packages", self.pkgwork['remove'])
        if len(self.extra_services) > 0:
            if self.setup['remove'] in ['all', 'services']:
                self.CondDisplayList('verbose', 'Removing services:', self.extra_services)
                for service in self.extra_services:
                    if not system("/sbin/chkconfig %s off" % service):
                        self.extra_services.remove(service)
            else:
                self.CondDisplayList('verbose', 'Need to remove services:', self.extra_services)
        
    def Inventory(self):
        '''Do standard inventory plus debian extra service check'''
        Toolset.Inventory(self)
        allsrv = [line.split()[0] for line in popen("/sbin/chkconfig --list|grep :on").readlines()]
        self.CondDisplayList('debug', 'Found active services:', allsrv)
        csrv = self.cfg.findall(".//Service")
        [allsrv.remove(svc.get('name')) for svc in csrv if
         svc.get('status') == 'on' and svc.get('name') in allsrv]
        self.extra_services = allsrv