"""This module contains various utility functions around scanning."""
import json
import socket
from contextlib import closing
from typing import List
import requests
from requests.exceptions import ConnectTimeout, ReadTimeout
from teatime.plugins import PluginException
[docs]def check_port(host: str, port: int, timeout: int = 2) -> bool:
"""Check whether a given port is available on the target host.
This helper function will attempt to connect to a given port on the
target host.
:param timeout: Number of seconds to time out after
:param host: The target host to connect to
:param port: The target port to connect to
:return: A boolean indicating whether the connection was successful
"""
with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
sock.settimeout(timeout)
return sock.connect_ex((host, port)) == 0
[docs]def reverse_dns(address: str) -> str:
"""Attempt to resolve an IP address to its DNS name.
:param address: The IP address to resolve
:return: The IP's DNS name as a string
"""
try:
result = socket.gethostbyaddr(address)[0]
except socket.herror:
result = address
return result