Пример #1
0
def get_os_type(ip, ssh_key):
    node = NodeUtil(ip, ssh_key)
    log.debug("Checking OS Type")
    output = node.ssh_run_command(["uname"])
    if "Linux" in output:
        log.debug("Found a Linux machine")
        return "linux"
    else:
        log.debug("Found a Windows machine")
        return "windows"
    return "None"
Пример #2
0
 def __init__(self, ip, ssh_key):
     self.ip = ip
     self.ssh_key = ssh_key
     self.node = NodeUtil(ip, ssh_key)
     self.check_user_load()
Пример #3
0
class Linux(Generic):
    def __init__(self, ip, ssh_key):
        self.ip = ip
        self.ssh_key = ssh_key
        self.node = NodeUtil(ip, ssh_key)
        self.check_user_load()

    def add_user(self, username, password):
        try:
            output = self.node.ssh_run_command(["adduser", username])
            output = self.change_user_password(username, password)
            log.debug('THE PASSWORD WAS SET')
            log.debug("Added %s" % username)
            return True, ""
        except HostNotConnectableError:
            return False, ""


    def change_user_password(self, username, password):
        try:
            output = ssh_node.ssh_run_command(["passwd", "--stdin", username, "<<<", password])
            return output
        except HostNotConnectableError:
            return False


    def log_user_off(self, username):
        try:
            output = self.node.ssh_run_command(["pkill", "-KILL", "-u", str(username)])
            log.debug("Logged %s off" % username)
            return output
        except HostNotConnectableError:
            return False


    def add_administrator(self, username):
        try:
            self.node.ssh_run_command(["chmod", "+w", "/etc/sudoers",
                                        "|", "echo", "'%s ALL=(ALL) ALL'" % username, ">>", "/etc/sudoers",
                                        "|", "chmod", "-w", "/etc/sudoers"])
            return True
        except HostNotConnectableError:
            return False


    def check_user_load(self):
        self.sessions = []
        try:
            output = self.node.ssh_run_command(["w"])
            output = output.split('WHAT')[1]
            for line in output.split('\n'):
                fields = line.split()
                if (len(fields) > 1):
                    user = dict()
                    for i,value in enumerate(fields):
                        if(i == 0):
                            user["username"] = string.rstrip(value)
                        if(i == 1):
                            user["sessionid"] = string.rstrip(value)
                        if(i == 4):
                            user["idletime"] = string.rstrip(value)
                        if(i == 3):
                            user["logontime"] = string.rstrip(value)
                    if (len(user) == 4):
                        self.sessions.append(user)
            return self.sessions
        except HostNotConnectableError:
            return False


    def user_cleanup(self, timeout):
        """
        Checks idle time for all sessions on the node.  If any session is disconnected then the user   
        is logged off.  If the idle time exceeds the timeout parameter, the user is logged off. Timeout
        is in MINUTES.
        """
        log.debug(self.sessions)
        for session in self.sessions:
            log.debug("Reported idletime: " + session["idletime"])
            if 's' in session["idletime"]:
                seconds = session["idletime"].split('s')[0]
                idletime = float(seconds) / 60
            elif 'm' in session["idletime"]:
                idletime = session["idletime"].split('m')[0].split(':')
                minutes = idletime[1]
                hours = idletime[0]
                idletime = int(hours) * 60 + int(minutes)
            elif 'days'in session["idletime"]:
                days = session["idletime"].split('days')[0]
                idletime = float(days) * 24 * 60
            else:
                idletime = session["idletime"].split(':')
                minutes = idletime[0]
                seconds = idletime[1]
                idletime = int(minutes) + (int(seconds) / 60)
            log.debug("Calculated idle time: "+str(idletime))
            if(idletime > timeout):
                self.log_user_off(session["username"])
Пример #4
0
class Windows(Generic):
    def __init__(self, ip, ssh_key):
        self.ip = ip
        self.ssh_key = ssh_key
        self.node = NodeUtil(ip, ssh_key)
        self.check_user_load()

    def add_user(self, username, password):
        try:
            output = self.node.ssh_run_command(["NET", "USER", username, password, "/ADD"])
            if output.find("The command completed successfully.") > -1:
                log.debug("User %s has been created" % username)
            elif output.find("The account already exists.") > -1:
                log.debug("User %s already exists, going to try to set the password" % username)
                output = self.change_user_password(username, password)
                if output.find("The command completed successfully.") > -1:
                    log.debug("THE PASSWORD WAS RESET")
                else:
                    error_string = (
                        "An unknown error occured while trying to set the password for user %s on machine %s.  The error from the machine was %s"
                        % (username, self.ip, output)
                    )
                    log.error(error_string)
                    return False, error_string
            else:
                error_string = (
                    "An unknown error occured while trying to create user %s on machine %s.  The error from the machine was %s"
                    % (username, self.ip, output)
                )
                log.error(error_string)
                return False, error_string
            return True, ""
        except HostNotConnectableError:
            return False, ""

    def change_user_password(self, username, password):
        try:
            output = self.node.ssh_run_command(["NET", "USER", username, password])
            return output
        except HostNotConnectableError:
            return False

    def enable_rdp_for_user(self, username):
        try:
            self.node.ssh_run_command(["NET", "localgroup", '"Remote Desktop Users"', "/add", username])
            return True
        except HostNotConnectableError:
            return False

    def add_administrator(self, username):
        try:
            self.node.ssh_run_command(["NET", "localgroup", '"Administrators"', "/add", username])
            return True
        except HostNotConnectableError:
            return False

    def log_user_off(self, username):
        try:
            output = self.node.ssh_run_command(["c:\logoff.exe", str(username)])
            return output
        except HostNotConnectableError:
            return False

    def check_user_load(self):
        try:
            self.sessions = []
            output = self.node.ssh_run_command(["Quser"])
            for line in output.split("\n"):
                fields = split(
                    "(\S+) +(\d+) +(Disc) +([none]*[\d+\+]*[\. ]*[\d*\:\d* ]*[\d ]*) (\d*/\d*/\d*) +(\d*\:\d* +[AM]*[PM]*)",
                    line,
                )
                if len(fields) > 1:
                    user = dict()
                    for i, value in enumerate(fields):
                        if i == 1:
                            user["username"] = string.rstrip(value)
                        if i == 2:
                            user["sessionid"] = string.rstrip(value)
                        if i == 3:
                            user["state"] = string.rstrip(value)
                        if i == 4:
                            user["idletime"] = string.rstrip(value)
                        if i == 5:
                            user["logondate"] = string.rstrip(value)
                        if i == 6:
                            user["logontime"] = string.rstrip(value)
                    if len(user) == 6:
                        self.sessions.append(user)

                else:
                    fields = split(
                        "(\S+) +(\S+) +(\d+) +(Active) +([none]*[\d+\+]*[\. ]*[\d*\:\d* ]*[\d ]*) (\d*/\d*/\d*) +(\d*\:\d* +[AM]*[PM]*)",
                        line,
                    )
                    user = dict()
                    for i, value in enumerate(fields):
                        if i == 1:
                            user["username"] = string.rstrip(value)
                        if i == 2:
                            user["sessionname"] = string.rstrip(value)
                        if i == 3:
                            user["sessionid"] = string.rstrip(value)
                        if i == 4:
                            user["state"] = string.rstrip(value)
                        if i == 5:
                            user["idletime"] = string.rstrip(value)
                        if i == 6:
                            user["logondate"] = string.rstrip(value)
                        if i == 7:
                            user["logontime"] = string.rstrip(value)
                    if len(user) == 7:
                        self.sessions.append(user)
            return self.sessions
        except HostNotConnectableError:
            return False

    def user_cleanup(self, timeout):
        """
        Checks idle time for all sessions on the node.  If any session is disconnected then the user   
        is logged off.  If the idle time exceeds the timeout parameter, the user is logged off. Timeout
        is in MINUTES.
        """
        log.debug(self.sessions)
        for session in self.sessions:
            log.debug("Reported idletime: " + session["idletime"])
            if session["state"] == "Disc":
                self.log_user_off(session["sessionid"])
            elif (session["idletime"] == ".") or (session["idletime"] == "none"):
                continue
            else:
                days = session["idletime"].split("+")
                if len(days) > 1:
                    idletime = 24 * 60 * int(days[0])
                    for i, digit in enumerate(days[1].split(":")):
                        if i == 0:
                            idletime += 60 * int(digit)
                        else:
                            idletime += int(digit)
                else:
                    idletime = 0
                    hoursandmins = days[0].split(":")
                    if len(hoursandmins) > 1:
                        idletime += 60 * int(hoursandmins[0])
                        idletime += int(hoursandmins[1])
                    else:
                        idletime += int(hoursandmins[0])
                log.debug("Calculated idle time: " + str(idletime))
                if idletime > timeout:
                    self.log_user_off(session["sessionid"])