JezK

Edit File: cldetect

#!/opt/cloudlinux/venv/bin/python3 -Isbb

#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2021 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENCE.TXT
#

import getopt
import os
import sys
import textwrap

import cldetectlib as detect
import cllicenselib as cllicense
import clsetuplib as clsetup
from clcommon.lib.cledition import get_cl_edition_readable, skip_without_lve
from clcommon.utils import get_os_version, is_ea4, is_nginx_running

# Check for root
def check_root():
    if os.geteuid() != 0:
        print("Error: root privileges required. Abort.")
        sys.exit(-1)

# Show help
def print_usage() -> None:
    help_message = """
    -h | --help                         show this message
    --detect-cp                         print control panel and its version (CP_NAME, CP_VERSION)
    --detect-cp-full                    print control panel, version and panel specific data (CP_NAME, CP_VERSION, ...)
                                        Specific data: for ISP Manager5 - Master/Node
    --detect-cp-nameonly                print control panel name (CP_NAME)
    --get-admin-email                   print control panel admin email (CP_ADMIN_EMAIL)
    --cxs-installed                     check if CXS is installed. Returns 0 if installed, 1 otherwise
    --cpanel-suphp-enabled              check if suPHP is enabled in cPanel. Returns 0 if enabled, 1 otherwise
    --detect-litespeed                  check if LiteSpeed is installed. Returns 0 if installed, 1 otherwise
    --detect-postgresql                 check if PostGreSQL is installed. Returns 0 if installed, 1 otherwise
    --detect-ea4                        check if EA4 is installed. Only for cPanel
    --detect-nginx                      check if nginx is running
    --print-apache-gid                  print current apache gid
    --print-da-admin                    print DirectAdmin admin user
    --set-securelinks-gid               change sysctl conf if apache gid != 48 (default)
    --set-nagios                        do some adjustments to make nagios work correctly if it's installed
    --setup-supergids                   do some adjustments to make some software work correctly if it's installed
    --cl-setup                          check if CloudLinux is installing. Returns 0 if installing, 1 otherwise
    --update-license                    update license
    --update-new-key KEY                update license with new key ("-" reads the key from stdin to keep it out of argv)
    --check-license                     check license. Return OK if license is not older than 3 days, error message otherwise
    --check-license --quiet             check license. Exit with code 0 if license is not older than 3 days, 1 otherwise
    --no-valid-license-screen      	    return 'no valid license found' screen
    --license-out-of-date-email    	    return the 'license out of date' email
    --check-openvz                      return environment ID
    --detect-edition                    return edition of CloudLinux
    --detect-os                         print OS name and version (OS_NAME,OS_VERSION)
    --detect-os-nameonly                print OS name only (OS_NAME)
    """  # noqa: E501

# Strip newlines from the start and end
    # Keep in mind that multi-line indentation is preserved
    print(help_message.strip("\n"))

def update_new_key(key):
    cllicense.update_license_with_key(key)

def check_license(quiet):
    is_license_valid = cllicense.check_license()
    if quiet:
        if is_license_valid:
            sys.exit(0)
        else:
            sys.exit(1)
    else:
        print(cllicense.last_license_check(is_license_valid))
        sys.exit(0)

def main():
    try:
        opts, args = getopt.getopt(
            sys.argv[1:],
            "hq",
            [
                "help",
                "quiet",
                "detect-cp",
                "detect-cp-full",
                "detect-cp-nameonly",
                "cxs-installed",
                "cpanel-suphp-enabled",
                "get-admin-email",
                "print-apache-gid",
                "detect-litespeed",
                "detect-postgresql",
                "set-securelinks-gid",
                "print-da-admin",
                "cl-setup",
                "set-nagios",
                "update-license",
                "update-new-key",
                "check-license",
                "no-valid-license-screen",
                "license-out-of-date-email",
                "check-openvz",
                "detect-ea4",
                "detect-nginx",
                "setup-supergids",
                "detect-edition",
                "detect-os",
                "detect-os-nameonly",
            ],
        )
    except getopt.GetoptError:
        print("error: unknown command")
        print_usage()
        sys.exit(1)

executed = False
    quiet = False
    for option, _argument in opts:
        if option in ("--quiet", "-q"):
            quiet = True
            break

skipped_without_lve_checks = ("--set-securelinks-gid", "--set-nagios", "--setup-supergids")
    for option, _argument in opts:
        if option in skipped_without_lve_checks:
            skip_without_lve()

if option in ("--help", "-h"):
            executed = True
            check_root()
            print_usage()

elif option in ("--cxs-installed",):
            executed = True
            check_root()
            if detect.CXS_check():
                sys.exit(0)
            else:
                sys.exit(1)

elif option in ("--detect-cp",):
            executed = True
            check_root()
            detect.getCP()
            print(str(detect.CP_NAME) + "," + str(detect.CP_VERSION))

elif option in ("--detect-cp-full",):
            executed = True
            check_root()
            detect.getCP()
            cp_name = str(detect.CP_NAME)
            cp_version = str(detect.CP_VERSION)

if detect.CP_ISP_TYPE:
                # Panel additional info present
                print(f"{cp_name},{cp_version},{detect.CP_ISP_TYPE}")
            else:
                # No panel additional info present
                print(f"{cp_name},{cp_version}")

elif option in ("--detect-cp-nameonly",):
            executed = True
            check_root()
            detect.getCPName()
            print(str(detect.CP_NAME))

elif option in ("--get-admin-email",):
            executed = True
            check_root()
            print(detect.getCPAdminEmail())

elif option in ("--cpanel-suphp-enabled",):
            executed = True
            check_root()
            if detect.mod_suPHP_check():
                sys.exit(0)
            else:
                sys.exit(1)

elif option in ("--detect-litespeed",):
            executed = True
            check_root()
            if detect.detect_litespeed():
                sys.exit(0)
            else:
                sys.exit(1)

elif option in ("--detect-postgresql",):
            executed = True
            check_root()
            if detect.detect_postgresql():
                sys.exit(0)
            else:
                sys.exit(1)

elif option in ("--print-apache-gid",):
            executed = True
            check_root()
            if detect.get_apache_gid():
                print(detect.APACHE_GID)
            else:
                print("error: unknown control panel")

elif option in ("--set-securelinks-gid",):
            executed = True
            check_root()
            if detect.get_apache_gid():
                clsetup.set_securelinks_gid(detect.APACHE_GID)

# Actually, this command could be removed because it's a sub-set of
        # --setup-supergids, but I left it here only for backward compatibility
        elif option in ("--set-nagios",):
            executed = True
            check_root()
            clsetup.setup_nagios()

elif option in ("--setup-supergids",):
            executed = True
            check_root()
            clsetup.setup_supergids()

elif option in ("--print-da-admin",):
            executed = True
            check_root()
            admin = detect.detect_DA_admin()
            if admin:
                print(admin)
            else:
                print("Error: can not find admin user for DirectAdmin")

elif option in ("--cl-setup",):
            executed = True
            check_root()
            if detect.check_CL_installing():
                sys.exit(0)
            else:
                sys.exit(1)

elif option in ("--update-license",):
            executed = True
            check_root()
            cllicense.update_license_timestamp_file()

elif option in ("--update-new-key",):
            executed = True
            check_root()
            # A flag-shaped key cannot smuggle a privileged getopt sub-option -- the value is
            # consumed as args[0] (the key); if it is absent, the except below sys.exit(1)s
            # before any further dispatch.
            try:
                key = args[0].strip()
                # "-" means read the key from stdin so it is not exposed in
                # this process's argv (/proc/<pid>/cmdline). Callers that hold
                # the key (e.g. cloudlinux-license) pipe it in this way.
                if key == "-":
                    key = sys.stdin.readline().rstrip("\n")
                update_new_key(key)
            except:
                print("Error: key required.")
                sys.exit(1)

elif option in ("--check-license",):
            executed = True
            check_license(quiet)

elif option in ("--no-valid-license-screen",):
            executed = True
            check_root()
            print(cllicense.get_novalid_template())

elif option in ("--license-out-of-date-email",):
            executed = True
            check_root()
            print(cllicense.get_email_template())

elif option in ("--check-openvz",):
            executed = True
            result = detect.is_openvz()
            sys.stdout.write(str(result))

elif option in ("--detect-ea4",):
            executed = True
            if is_ea4():
                print("EA4 detected")
            else:
                print("No EA4 detected")

elif option in ("--detect-nginx",):
            executed = True
            if is_nginx_running():
                print("Nginx is running")
            else:
                print("Nginx is not running")

elif option in ("--detect-edition",):
            executed = True
            print(get_cl_edition_readable())

elif option in ("--detect-os",):
            executed = True
            os_name, os_ver = get_os_version()
            print(os_name, os_ver)

elif option in ("--detect-os-nameonly",):
            executed = True
            os_name, _ = get_os_version()
            print(os_name)

if not executed:
        print("error: argument required")
        print_usage()
        sys.exit(1)

if __name__ == "__main__":
    main()