kcare-scripts/kc-compat.py at master · cloudlinux/kcare-scripts · GitHub

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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
from __future__ import print_function
import struct
import sys
import os

try:
    from urllib.request import urlopen
    from urllib.error import HTTPError, URLError
except ImportError:
    from urllib2 import urlopen, HTTPError, URLError

__author__ = 'Igor Seletskiy'
__copyright__ = "Copyright (c) Cloud Linux GmbH & Cloud Linux Software, Inc"
__credits__ = 'Igor Seletskiy'
__license__ = 'Apache License v2.0'
__maintainer__ = 'Igor Seletskiy'
__email__ = '[email protected]'
__status__ = 'Production'
__version__ = '1.0'


SUPPORTED_DISTROS = {
    "almalinux",
    "amzn",
    "centos",
    "cloudlinux",
    "debian",
    "ol",
    "raspbian",
    "rhel",
    "rocky",
    "ubuntu",
    "proxmox",
}


def get_kernel_hash():
    try:
        # noinspection PyCompatibility
        from hashlib import sha1
    except ImportError:
        from sha import sha as sha1
    f = open('/proc/version', 'rb')
    try:
        return sha1(f.read()).hexdigest()
    finally:
        f.close()


def inside_vz_container():
    """
    determines if we are inside Virtuozzo container
    :return: True if inside container, false otherwise
    """
    return os.path.exists('/proc/vz/veinfo') and not os.path.exists('/proc/vz/version')


def inside_lxc_container():
    return '/lxc/' in open('/proc/1/cgroup').read()


def get_distro_info():
    """
    Get current distribution name and version
    :return: distro name or None if detection fails
    """

    def parse_value(line):
        return line.split('=', 1)[1].strip().strip('"\'')

    os_release_path = '/etc/os-release'
    if not os.path.exists(os_release_path):
        return None

    try:
        with open(os_release_path, 'r') as f:
            for line in f:
                line = line.strip()
                if line.startswith('ID='):
                    return parse_value(line)
    except (IOError, OSError):
        return None


def is_distro_supported(distro_name):
    """
    Check if the given distro name is supported
    """
    return distro_name in SUPPORTED_DISTROS


def _check_url(url):
    try:
        urlopen(url)
        return True
    except HTTPError as e:
        if e.code == 404:
            return False
        raise
    except URLError:
        raise


def is_compat():
    base = 'http://patches.kernelcare.com/' + get_kernel_hash()
    for endpoint in ('/version', '/latest.v2', '/latest.v3'):
        if _check_url(base + endpoint):
            return True
    return False


def myprint(silent, message):
    if not silent:
        print(message)


def main():
    """
    if --silent or -q argument provided, don't print anything, just use exit code
    otherwise print results (COMPATIBLE or support contact messages)
    else exit with 0 if COMPATIBLE, 1 or more otherwise
    """
    silent = len(sys.argv) > 1 and (sys.argv[1] == '--silent' or sys.argv[1] == '-q')
    if inside_vz_container() or inside_lxc_container():
        myprint(silent, "UNSUPPORTED; INSIDE CONTAINER")
        return 2

    try:
        if is_compat():
            myprint(silent, "COMPATIBLE")
            return 0
        else:
            # Handle 404 case - check if distro is supported
            distro_name = get_distro_info()
            if distro_name and is_distro_supported(distro_name):
                myprint(silent, "NEEDS REVIEW")
                myprint(silent, "We support your distribution, but we're having trouble detecting your precise kernel configuration. Please, contact CloudLinux Inc. support by email at [email protected] or by request form at https://www.cloudlinux.com/index.php/support")
                return 1
            else:
                myprint(silent, "NEEDS REVIEW")
                myprint(silent, "Please contact CloudLinux Inc. support by email at [email protected] or by request form at https://www.cloudlinux.com/index.php/support")
                return 1
    except HTTPError as e:
        myprint(silent, "CONNECTION ERROR; HTTP %d" % e.code)
        return 3
    except URLError as e:
        myprint(silent, "CONNECTION ERROR; %s" % str(e.reason))
        return 3
    except (IOError, OSError) as e:
        myprint(silent, "SYSTEM ERROR; %s" % str(e))
        return 4
    except Exception as e:
        myprint(silent, "UNEXPECTED ERROR; %s" % str(e))
        return 5


if __name__ == "__main__":
    exit(main())