import io import socket from typing import NoReturn, Optional import asyncio import re import ipinfo import ipwhois import pydig import aiohttp from ipinfo.exceptions import RequestQuotaExceededError from ipwhois import Net from ipwhois.asn import IPASN from aiocache import cached, Cache from aiocache.serializers import PickleSerializer from tuxbot.cogs.Network.functions.exceptions import ( VersionNotFound, RFC18, InvalidIp, InvalidQueryType, InvalidAsn, ) def _(x): return x @cached( ttl=24 * 3600, serializer=PickleSerializer(), cache=Cache.MEMORY, namespace="network", ) async def get_ip(loop, ip: str, inet: Optional[dict]) -> str: _inet: socket.AddressFamily | int = 0 # pylint: disable=no-member if inet: if inet["inet"] == "6": _inet = socket.AF_INET6 elif inet["inet"] == "4": _inet = socket.AF_INET def _get_ip(_ip: str): try: return socket.getaddrinfo(_ip, None, _inet)[1][4][0] except socket.gaierror as e: raise VersionNotFound( _( "Unable to collect information on this in the given " "version", ) ) from e return await loop.run_in_executor(None, _get_ip, str(ip)) @cached( ttl=24 * 3600, serializer=PickleSerializer(), cache=Cache.MEMORY, namespace="network", ) async def get_hostname(loop, ip: str) -> str: def _get_hostname(_ip: str): try: return socket.gethostbyaddr(ip)[0] except socket.herror: return "N/A" try: return await asyncio.wait_for( loop.run_in_executor(None, _get_hostname, str(ip)), timeout=0.200, ) # assuming that if the hostname isn't retrieved in first .3sec, # it doesn't exists except asyncio.exceptions.TimeoutError: return "N/A" @cached( ttl=24 * 3600, serializer=PickleSerializer(), cache=Cache.MEMORY, namespace="network", ) async def get_ipwhois_result(loop, ip: str) -> NoReturn | dict: def _get_ipwhois_result(_ip: str) -> NoReturn | dict: try: net = Net(ip) obj = IPASN(net) return obj.lookup() except ipwhois.exceptions.ASNRegistryError: return {} except ipwhois.exceptions.IPDefinedError as e: raise RFC18( _( "IP address {ip_address} is already defined as Private-Use" " Networks via RFC 1918." ) ) from e try: return await asyncio.wait_for( loop.run_in_executor(None, _get_ipwhois_result, str(ip)), timeout=0.200, ) except asyncio.exceptions.TimeoutError: return {} @cached( ttl=24 * 3600, serializer=PickleSerializer(), cache=Cache.MEMORY, namespace="network", ) async def get_ipinfo_result(loop, apikey: str, ip: str) -> dict: def _get_ipinfo_result(_ip: str) -> NoReturn | dict: """ Q. Why no getHandlerAsync ? A. Use of this return "Unclosed client session" and "Unclosed connector" """ try: handler = ipinfo.getHandler(apikey, request_options={"timeout": 7}) return (handler.getDetails(ip)).all except RequestQuotaExceededError: return {} try: return await asyncio.wait_for( loop.run_in_executor(None, _get_ipinfo_result, str(ip)), timeout=8, ) except asyncio.exceptions.TimeoutError: return {} @cached( ttl=24 * 3600, serializer=PickleSerializer(), cache=Cache.MEMORY, namespace="network", ) async def get_crimeflare_result(ip: str) -> Optional[str]: try: async with aiohttp.ClientSession() as cs: async with cs.post( "http://www.crimeflare.org:82/cgi-bin/cfsearch.cgi", data=f"cfS={ip}", timeout=aiohttp.ClientTimeout(total=21), ) as s: result = re.search(r"(\d*\.\d*\.\d*\.\d*)", await s.text()) if result: return result.group() except (aiohttp.ClientError, asyncio.exceptions.TimeoutError): pass return None def merge_ipinfo_ipwhois(ipinfo_result: dict, ipwhois_result: dict) -> dict: output = { "belongs": "N/A", "rir": "N/A", "region": "N/A", "flag": "N/A", "map": "N/A", } if ipinfo_result: org = ipinfo_result.get("org", "N/A") asn = org.split()[0] if len(org.split()) > 1 else "N/A" output["belongs"] = f"[{org}](https://bgp.he.net/{asn})" output["rir"] = f"```{ipwhois_result.get('asn_registry', 'N/A')}```" output["region"] = ( f"```{ipinfo_result.get('city', 'N/A')} - " f"{ipinfo_result.get('region', 'N/A')} " f"({ipinfo_result.get('country', 'N/A')})```" ) output[ "flag" ] = f"https://flagcdn.com/144x108/{ipinfo_result['country'].lower()}.png" output["map"] = ipinfo_result["loc"] elif ipwhois_result: org = ipwhois_result.get("asn_description", "N/A") asn = ipwhois_result.get("asn", "N/A") asn_country = ipwhois_result.get("asn_country_code", "N/A") output["belongs"] = f"{org} ([AS{asn}](https://bgp.he.net/{asn}))" output["rir"] = f"```{ipwhois_result['asn_registry']}```" output["region"] = f"```{asn_country}```" output[ "flag" ] = f"https://flagcdn.com/144x108/{asn_country.lower()}.png" return output @cached( ttl=24 * 3600, serializer=PickleSerializer(), cache=Cache.MEMORY, namespace="network", ) async def get_map_bytes(apikey: str, latlon: str) -> Optional[io.BytesIO]: if latlon == "N/A": return None url = ( "https://maps.geoapify.com/v1/staticmap" "?style=osm-carto" "&width=333" "&height=250" "¢er=lonlat:{lonlat}" "&zoom=12" "&marker=lonlat:{lonlat};color:%23ff0000;size:small" "&apiKey={apikey}" ) lonlat = ",".join(latlon.split(",")[::-1]) url = url.format(lonlat=lonlat, apikey=apikey) try: async with aiohttp.ClientSession( timeout=aiohttp.ClientTimeout(total=5) ) as cs: async with cs.get(url) as s: if s.status != 200: return None return io.BytesIO(await s.read()) except asyncio.exceptions.TimeoutError: from ..images.load_fail import value return io.BytesIO(value) @cached( ttl=24 * 3600, serializer=PickleSerializer(), cache=Cache.MEMORY, namespace="network", ) async def get_pydig_result( loop, domain: str, query_type: str, dnssec: str | bool ) -> list: additional_args = [] if dnssec is False else ["+dnssec"] def _get_pydig_result(_domain: str) -> NoReturn | dict: resolver = pydig.Resolver( nameservers=[ "80.67.169.40", "80.67.169.12", ], additional_args=additional_args, ) return resolver.query(_domain, query_type) try: return await asyncio.wait_for( loop.run_in_executor(None, _get_pydig_result, str(domain)), timeout=0.500, ) except asyncio.exceptions.TimeoutError: return [] def check_ip_version_or_raise(version: Optional[dict]) -> bool | NoReturn: if version is None or version["inet"] in ("4", "6", ""): return True raise InvalidIp(_("Invalid ip version")) def check_query_type_or_raise(query_type: str) -> bool | NoReturn: query_types = ( "a", "aaaa", "cname", "ns", "ds", "dnskey", "soa", "txt", "ptr", "mx", ) if query_type in query_types: return True raise InvalidQueryType( _( "Supported queries : A, AAAA, CNAME, NS, DS, DNSKEY, SOA, TXT, PTR, MX" ) ) def check_asn_or_raise(asn: str) -> bool | NoReturn: if asn.isdigit() and int(asn) < 4_294_967_295: return True raise InvalidAsn(_("Invalid ASN provided"))