Gecko [ tanishkshah.com ]

Name	Size	Permission	Date
async_dns.py	874 B	-rw-r--r--	2024-02-18 18:35
ddns.py	1.17 KB	-rw-r--r--	2024-02-18 18:35
ddr.py	950 B	-rw-r--r--	2024-02-18 18:35
doh-json.py	2.93 KB	-rw-r--r--	2024-02-18 18:35
doh.py	534 B	-rw-r--r--	2024-02-18 18:35
doq.py	3.07 KB	-rw-r--r--	2024-02-18 18:35
e164.py	115 B	-rw-r--r--	2024-02-18 18:35
ecs.py	450 B	-rw-r--r--	2024-02-18 18:35
edns.py	1.7 KB	-rw-r--r--	2024-02-18 18:35
edns_resolver.py	1.5 KB	-rw-r--r--	2024-02-18 18:35
mx.py	189 B	-rw-r--r--	2024-02-18 18:35
name.py	356 B	-rw-r--r--	2024-02-18 18:35
query_specific.py	1.62 KB	-rw-r--r--	2024-02-18 18:35
receive_notify.py	1022 B	-rw-r--r--	2024-02-18 18:35
reverse.py	1.36 KB	-rw-r--r--	2024-02-18 18:35
reverse_name.py	140 B	-rw-r--r--	2024-02-18 18:35
send_notify.py	2.4 KB	-rw-r--r--	2024-02-18 18:35
wire_read_tcp.py	1.62 KB	-rw-r--r--	2024-02-18 18:35
xfr.py	340 B	-rw-r--r--	2024-02-18 18:35
zonediff.py	11.89 KB	-rw-r--r--	2024-02-18 18:35

Rename

import asyncio
import threading

import dns.asyncbackend
import dns.asyncquery
import dns.message
import dns.query
import dns.quic
import dns.rdatatype

try:
    import trio

have_trio = True
except ImportError:
    have_trio = False

# This demo assumes you have the aioquic example doq_server.py running on localhost
# on port 4784 on localhost.
peer_address = "127.0.0.1"
peer_port = 4784
query_name = "www.dnspython.org"
tls_verify_mode = False

def squery(rdtype="A", connection=None):
    q = dns.message.make_query(query_name, rdtype)
    r = dns.query.quic(
        q, peer_address, port=peer_port, connection=connection, verify=tls_verify_mode
    )
    print(r)

def srun():
    squery()

def smultirun():
    with dns.quic.SyncQuicManager(verify_mode=tls_verify_mode) as manager:
        connection = manager.connect(peer_address, peer_port)
        t1 = threading.Thread(target=squery, args=["A", connection])
        t1.start()
        t2 = threading.Thread(target=squery, args=["AAAA", connection])
        t2.start()
        t1.join()
        t2.join()

async def aquery(rdtype="A", connection=None):
    q = dns.message.make_query(query_name, rdtype)
    r = await dns.asyncquery.quic(
        q, peer_address, port=peer_port, connection=connection, verify=tls_verify_mode
    )
    print(r)

def arun():
    asyncio.run(aquery())

async def amulti():
    async with dns.quic.AsyncioQuicManager(verify_mode=tls_verify_mode) as manager:
        connection = manager.connect(peer_address, peer_port)
        t1 = asyncio.Task(aquery("A", connection))
        t2 = asyncio.Task(aquery("AAAA", connection))
        await t1
        await t2

def amultirun():
    asyncio.run(amulti())

if have_trio:

def trun():
        trio.run(aquery)

async def tmulti():
        async with trio.open_nursery() as nursery:
            async with dns.quic.TrioQuicManager(
                nursery, verify_mode=tls_verify_mode
            ) as manager:
                async with trio.open_nursery() as query_nursery:
                    # We run queries in a separate nursery so we can demonstrate
                    # waiting for them all to exit without waiting for the manager to
                    # exit as well.
                    connection = manager.connect(peer_address, peer_port)
                    query_nursery.start_soon(aquery, "A", connection)
                    query_nursery.start_soon(aquery, "AAAA", connection)

def tmultirun():
        trio.run(tmulti)

def main():
    print("*** Single Queries ***")
    print("--- Sync ---")
    srun()
    print("--- Asyncio ---")
    dns.asyncbackend.set_default_backend("asyncio")
    arun()
    if have_trio:
        print("--- Trio ---")
        dns.asyncbackend.set_default_backend("trio")
        trun()
    print("*** Multi-connection Queries ***")
    print("--- Sync ---")
    smultirun()
    print("--- Asyncio ---")
    dns.asyncbackend.set_default_backend("asyncio")
    amultirun()
    if have_trio:
        print("--- Trio ---")
        dns.asyncbackend.set_default_backend("trio")
        tmultirun()

if __name__ == "__main__":
    main()