mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-02-21 14:32:45 +00:00
perf: tool support ipv6
This commit is contained in:
@@ -1,5 +1,3 @@
|
||||
from django.conf import settings
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
from django_filters import rest_framework as drf_filters
|
||||
from rest_framework import status
|
||||
from rest_framework.decorators import action
|
||||
|
||||
@@ -113,6 +113,14 @@ def get_ip_city(ip):
|
||||
|
||||
def lookup_domain(domain):
|
||||
try:
|
||||
return socket.gethostbyname(domain), ''
|
||||
addrinfos = socket.getaddrinfo(
|
||||
domain, None, socket.AF_UNSPEC, socket.SOCK_DGRAM
|
||||
)
|
||||
family, _, _, _, sockaddr = addrinfos[0]
|
||||
if family == socket.AF_INET:
|
||||
return sockaddr[0], ''
|
||||
if family == socket.AF_INET6:
|
||||
return sockaddr[0], ''
|
||||
return domain, ''
|
||||
except Exception as e:
|
||||
return None, f'Cannot resolve {domain}: Unknown host, {e}'
|
||||
|
||||
Reference in New Issue
Block a user