Ejemplo n.º 1
0
def check(computer_name: str, connection: SMBConnection) -> (str, dict):
    """
    Return Health check for a Samba connection.

    :param computer_name: Remote computer name.
    :param connection: Samba connection.
    :return: A tuple with a string providing the status (pass, warn, fail) and the checks.
    Checks are based on https://inadarei.github.io/rfc-healthcheck/
    """
    try:
        response = connection.echo(b"")
        return (
            "pass",
            {
                f"{computer_name}:echo": {
                    "componentType": connection.remote_name,
                    "observedValue": response.decode(),
                    "status": "pass",
                    "time": datetime.utcnow().isoformat(),
                }
            },
        )
    except Exception as e:
        return (
            "fail",
            {
                f"{computer_name}:echo": {
                    "componentType": connection.remote_name,
                    "status": "fail",
                    "time": datetime.utcnow().isoformat(),
                    "output": str(e),
                }
            },
        )
Ejemplo n.º 2
0
 def get_connection(self):
     conn = SMBConnection(self.user_id,
                          self.password,
                          self.client_name,
                          self.server_name,
                          use_ntlm_v2=True,
                          is_direct_tcp=True)
     try:
         conn.connect(self.server_ip, 445)
         print(conn.echo('SMB CONNECTED'))
         self.conn = conn
     except Exception as e:
         print(e)
class Connection(komand.Connection):

    def __init__(self):
        self.conn = None
        super(self.__class__, self).__init__(input=ConnectionSchema())

    def connect(self, params):
        username = params["credentials"]["username"]
        password = params["credentials"]["password"]
        host = params["host"]
        port = params["port"]
        netbios_name = params["netbios_name"]
        domain = params.get("domain", None)
        use_ntvlm_v2 = params.get("use_ntlmv2", True)
        is_direct_tcp = True

        if domain:
            self.conn = SMBConnection(username, password, "InsightConnect", netbios_name, domain=domain,
                                      use_ntlm_v2=use_ntvlm_v2, is_direct_tcp=is_direct_tcp)
        else:
            self.conn = SMBConnection(username, password, "InsightConnect", netbios_name, use_ntlm_v2=use_ntvlm_v2,
                                      is_direct_tcp=is_direct_tcp)

        try:
            result = self.conn.connect(host, port, timeout=params.get(Input.TIMEOUT))
            if not result:
                error = "Failed to authenticate to SMB endpoint. Validate credentials and connection details."
                raise Exception(error)
        except socket.timeout:
            raise Exception("Timeout reached when connecting to SMB endpoint. Validate network connectivity or "
                            "extend connection timeout")
        except Exception as e:
            self.logger.error(f"Error connecting to SMB endpoint: {e}")
            raise

    def test(self):
        from komand.exceptions import ConnectionTestException

        message = "Hello World"
        try:
            echo_response = self.conn.echo(message.encode())
        except Exception as e:
            raise ConnectionTestException(cause="Connectivity test to SMB server failed.", assistance=e)

        if echo_response.decode() == message:
            self.logger.info("Connectivity test to SMB server was successful")
            return
        else:
            raise ConnectionTestException(cause="Connectivity test to SMB server failed.", assistance="echo response was not as expected")
Ejemplo n.º 4
0
        raise RuntimeError('Cannot connect to host ' + target + '; looking up NetBIOS IP failed')
    target_ip = ips[0]

if target_nb_name is None:
    print('Looking up NetBIOS name from target IP: ' + target_ip)
    nb_names = nb.queryIPForName(target_ip)
    print('Got NB names: ' + str(nb_names))
    if nb_names is None or len(nb_names) < 1:
        raise RuntimeError('Cannot connect to host ' + target + '; looking up NetBIOS name failed')
    target_nb_name = nb_names[0]

nb.close()

client_machine_name = socket.gethostbyaddr(socket.gethostname())[0]
# client_machine_name = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(15))
# print('Generated client machine name: ' + client_machine_name + '\n')

domain = input('Enter domain [none]: ')
username = input('Enter username: '******'Enter password: '******'Could not connect to host ' + target + '; establishing connection failed')

if conn.echo('blah') != 'blah':
    raise RuntimeError('Connection test (echo) failed')

conn.close()
Ejemplo n.º 5
0
    if nb_names is None or len(nb_names) < 1:
        raise RuntimeError('Cannot connect to host ' + target +
                           '; looking up NetBIOS name failed')
    target_nb_name = nb_names[0]

nb.close()

client_machine_name = socket.gethostbyaddr(socket.gethostname())[0]
# client_machine_name = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(15))
# print('Generated client machine name: ' + client_machine_name + '\n')

domain = input('Enter domain [none]: ')
username = input('Enter username: '******'Enter password: '******'Could not connect to host ' + target +
                       '; establishing connection failed')

if conn.echo('blah') != 'blah':
    raise RuntimeError('Connection test (echo) failed')

conn.close()