OwlCyberSecurity - MANAGER

Edit File: init.py

# -*- coding: utf-8 -*-

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

"""
This package contains classes and functions to run X Ray utilities as end-user
"""
from socket import socket, AF_UNIX, SOCK_STREAM

from xray.console_utils.cmdline_parser import (
    parse_cmd_arguments
)
from xray.internal.constants import user_agent_sock
from xray.internal.exceptions import XRayError
from xray.internal.user_plugin_utils import (
    pack_request,
    unpack_response,
    check_for_root,
    error_response
)

from .runners import get_runner

def executed_as_root_check() -> None:
    """
    Check if User Manager is executed as root and throw error in case if yes
    """
    if check_for_root():
        raise SystemExit(
            error_response('X-Ray End-User utilities cannot be executed as root'))

def run(run_for: str = 'manager') -> None:
    """
    Main run function
    """
    executed_as_root_check()

try:
        runner = get_runner(run_for)
    except XRayError as e:
        raise SystemExit(str(e))

args = parse_cmd_arguments(runner.args_parser())
    validated_args = runner.validator(args.__dict__)
    validated_args.runner = runner.name
    to_send = pack_request(validated_args.__dict__)
    with socket(AF_UNIX, SOCK_STREAM) as s:
        try:
            s.connect(user_agent_sock)
        except (ConnectionError, OSError):
            raise SystemExit(error_response(
                'X-Ray User Plugin is not enabled. Please, contact your server administrator'))
        s.sendall(to_send.encode())
        data = unpack_response(s)
    print(data.decode())

OwlCyberSecurity - MANAGER

Edit File: __init__.py

Edit File: init.py