157 lines
4.8 KiB
Python
157 lines
4.8 KiB
Python
"""
|
||
doh函数的实现。
|
||
author: https://github.com/C5H12O5/syno-videoinfo-plugin
|
||
"""
|
||
import base64
|
||
import concurrent
|
||
import concurrent.futures
|
||
import json
|
||
import socket
|
||
import struct
|
||
import urllib
|
||
import urllib.request
|
||
from typing import Dict, Optional
|
||
|
||
from app.core.config import settings
|
||
from app.log import logger
|
||
|
||
# 定义一个全局集合来存储注册的主机
|
||
_registered_hosts = {
|
||
'api.themoviedb.org',
|
||
'api.tmdb.org',
|
||
'webservice.fanart.tv',
|
||
'api.github.com',
|
||
'github.com',
|
||
'raw.githubusercontent.com',
|
||
'api.telegram.org'
|
||
}
|
||
|
||
# 定义一个全局线程池执行器
|
||
_executor = concurrent.futures.ThreadPoolExecutor()
|
||
|
||
# 定义默认的DoH配置
|
||
_doh_timeout = 5
|
||
_doh_cache: Dict[str, str] = {}
|
||
_doh_resolvers = [
|
||
# https://developers.cloudflare.com/1.1.1.1/encryption/dns-over-https
|
||
"1.0.0.1",
|
||
"1.1.1.1",
|
||
# https://support.quad9.net/hc/en-us
|
||
"9.9.9.9",
|
||
"149.112.112.112"
|
||
]
|
||
|
||
|
||
def _patched_getaddrinfo(host, *args, **kwargs):
|
||
"""
|
||
socket.getaddrinfo的补丁版本。
|
||
"""
|
||
if host not in _registered_hosts:
|
||
return _orig_getaddrinfo(host, *args, **kwargs)
|
||
|
||
# 检查主机是否已解析
|
||
if host in _doh_cache:
|
||
ip = _doh_cache[host]
|
||
logger.info("已解析 [%s] 为 [%s] (缓存)", host, ip)
|
||
return _orig_getaddrinfo(ip, *args, **kwargs)
|
||
|
||
# 使用DoH解析主机
|
||
futures = []
|
||
for resolver in _doh_resolvers:
|
||
futures.append(_executor.submit(_doh_query, resolver, host))
|
||
|
||
for future in concurrent.futures.as_completed(futures):
|
||
ip = future.result()
|
||
if ip is not None:
|
||
logger.info("已解析 [%s] 为 [%s]", host, ip)
|
||
_doh_cache[host] = ip
|
||
host = ip
|
||
break
|
||
|
||
return _orig_getaddrinfo(host, *args, **kwargs)
|
||
|
||
|
||
# 对 socket.getaddrinfo 进行补丁
|
||
if settings.DOH_ENABLE:
|
||
_orig_getaddrinfo = socket.getaddrinfo
|
||
socket.getaddrinfo = _patched_getaddrinfo
|
||
|
||
|
||
def _doh_query(resolver: str, host: str) -> Optional[str]:
|
||
"""
|
||
使用给定的DoH解析器查询给定主机的IP地址。
|
||
"""
|
||
|
||
# 构造DNS查询消息（RFC 1035）
|
||
header = b"".join(
|
||
[
|
||
b"\x00\x00", # ID: 0
|
||
b"\x01\x00", # FLAGS: 标准递归查询
|
||
b"\x00\x01", # QDCOUNT: 1
|
||
b"\x00\x00", # ANCOUNT: 0
|
||
b"\x00\x00", # NSCOUNT: 0
|
||
b"\x00\x00", # ARCOUNT: 0
|
||
]
|
||
)
|
||
question = b"".join(
|
||
[
|
||
b"".join(
|
||
[
|
||
struct.pack("B", len(item)) + item.encode("utf-8")
|
||
for item in host.split(".")
|
||
]
|
||
)
|
||
+ b"\x00", # QNAME: 域名序列
|
||
b"\x00\x01", # QTYPE: A
|
||
b"\x00\x01", # QCLASS: IN
|
||
]
|
||
)
|
||
message = header + question
|
||
|
||
try:
|
||
# 发送GET请求到DoH解析器（RFC 8484）
|
||
b64message = base64.b64encode(message).decode("utf-8").rstrip("=")
|
||
url = f"https://{resolver}/dns-query?dns={b64message}"
|
||
headers = {"Content-Type": "application/dns-message"}
|
||
logger.debug("DoH请求: %s", url)
|
||
|
||
request = urllib.request.Request(url, headers=headers, method="GET")
|
||
with urllib.request.urlopen(request, timeout=_doh_timeout) as response:
|
||
logger.debug("解析器(%s)响应: %s", resolver, response.status)
|
||
if response.status != 200:
|
||
return None
|
||
resp_body = response.read()
|
||
|
||
# 解析DNS响应消息（RFC 1035）
|
||
# name（压缩）:2 + type:2 + class:2 + ttl:4 + rdlength:2 = 12字节
|
||
first_rdata_start = len(header) + len(question) + 12
|
||
# rdata（A记录）= 4字节
|
||
first_rdata_end = first_rdata_start + 4
|
||
# 将rdata转换为IP地址
|
||
return socket.inet_ntoa(resp_body[first_rdata_start:first_rdata_end])
|
||
except Exception as e:
|
||
logger.error("解析器(%s)请求错误: %s", resolver, e)
|
||
return None
|
||
|
||
|
||
def doh_query_json(resolver: str, host: str) -> Optional[str]:
|
||
"""
|
||
使用给定的DoH解析器查询给定主机的IP地址。
|
||
"""
|
||
url = f"https://{resolver}/dns-query?name={host}&type=A"
|
||
headers = {"Accept": "application/dns-json"}
|
||
logger.debug("DoH请求: %s", url)
|
||
try:
|
||
request = urllib.request.Request(url, headers=headers, method="GET")
|
||
with urllib.request.urlopen(request, timeout=_doh_timeout) as response:
|
||
logger.debug("解析器(%s)响应: %s", resolver, response.status)
|
||
if response.status != 200:
|
||
return None
|
||
response_body = response.read().decode("utf-8")
|
||
logger.debug("<== body: %s", response_body)
|
||
answer = json.loads(response_body)["Answer"]
|
||
return answer[0]["data"]
|
||
except Exception as e:
|
||
logger.error("解析器(%s)请求错误: %s", resolver, e)
|
||
return None
|