1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
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()
|
