summaryrefslogtreecommitdiffstats
path: root/compass-tasks/actions/health_check/check_tftp.py
blob: 7ca6405d91d5a0e21725b4c68ce7b98b7ec73225 (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
# Copyright 2014 Huawei Technologies Co. Ltd
#
# 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.

"""Health Check module for TFTP service."""
import os
import socket
import xmlrpclib

from compass.actions.health_check import base
from compass.actions.health_check import utils as health_check_utils


class TftpCheck(base.BaseCheck):
    """tftp health check class."""
    NAME = "TFTP Check"

    def run(self):
        """do health check."""
        method_name = "self.check_" + self.os_installer['name'] + "_tftp()"
        return eval(method_name)

    def check_cobbler_tftp(self):
        """Checks if Cobbler manages TFTP service.

        :note: we assume TFTP service is running at the
        same machine where this health check runs at
        """

        try:
            remote = xmlrpclib.Server(
                self.os_installer['cobbler_url'],
                allow_none=True)
            credentials = self.os_installer['credentials']
            remote.login(
                credentials['username'], credentials['password'])
        except Exception:
            self._set_status(
                0,
                "[%s]Error: Cannot login to Cobbler with the tokens "
                " provided in the config file" % self.NAME)
            return (self.code, self.messages)

        cobbler_settings = remote.get_settings()
        if cobbler_settings['manage_tftp'] == 0:
            self.messages.append(
                '[TFTP]Info: tftp service is not managed by Compass')
            return (0, self.messages)
        self.check_tftp_dir()
        print "[Done]"
        self.check_tftp_service()
        print "[Done]"
        if self.code == 1:
            self.messages.append(
                "[%s]Info: tftp service health check has completed. "
                "No problems found, all systems go." % self.NAME)

        return (self.code, self.messages)

    def check_tftp_dir(self):
        """Validates TFTP directories and configurations."""
        print "Checking TFTP directories......",
        if not os.path.exists('/var/lib/tftpboot/'):
            self._set_status(
                0,
                "[%s]Error: No tftp-boot libraries found, "
                "please check if tftp server is properly "
                "installed/managed" % self.NAME)

        return True

    def check_tftp_service(self):
        """Checks if TFTP is running on port 69."""
        print "Checking TFTP services......",
        serv_err_msg = health_check_utils.check_service_running(self.NAME,
                                                                'xinetd')
        if not serv_err_msg == "":
            self._set_status(0, serv_err_msg)

        if 'tftp' != socket.getservbyport(69):
            self._set_status(
                0,
                "[%s]Error: tftp doesn't seem to be listening "
                "on Port 60." % self.NAME)

        return True