移除aiobrute向aiodns_query_a()函数传semaphore参数

2026-08-26 04:47:48 +08:00 · 2019-08-05 23:38:53 +08:00
parent e3f9d5e149
commit 31666fdd75
2 changed files with 14 additions and 13 deletions
@@ -48,11 +48,14 @@ async def aiodns_query_a(hostname, semaphore=None):
    :return: 主机名或查询结果或查询异常
    """
    if semaphore is None:
-        semaphore = utils.get_semaphore()
-    async with semaphore:
        resolver = aiodns_resolver()
        answers = await resolver.query(hostname, 'A')
        return hostname, answers
+    else:
+        async with semaphore:
+            resolver = aiodns_resolver()
+            answers = await resolver.query(hostname, 'A')
+            return hostname, answers


 def resolve_callback(future, index, datas):