mirror
/
micropython-lib
kopia lustrzana https://github.com/micropython/micropython-lib
1
0
Forkuj 0
Kod Zgłoszenia Projekty Wydania Wiki Aktywność
micropython-lib/micropython/udnspkt/udnspkt.py

79 wiersze
1.6 KiB
Python
Czysty Zwykły widok Historia

udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
import uio
def write_fqdn(buf, name):
parts = name.split(".")
for p in parts:
buf.writebin("B", len(p))
buf.write(p)
buf.writebin("B", 0)
def skip_fqdn(buf):
while True:
sz = buf.readbin("B")
if not sz:
break
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
if sz >= 0xC0:
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
buf.readbin("B")
break
buf.read(sz)
def make_req(buf, fqdn, is_ipv6):
typ = 1 # A
if is_ipv6:
typ = 28 # AAAA
buf.writebin(">H", 0)
buf.writebin(">H", 0x100)
# q count
buf.writebin(">H", 1)
buf.writebin(">H", 0)
# squashed together
buf.writebin(">I", 0)
write_fqdn(buf, fqdn)
buf.writebin(">H", typ)
buf.writebin(">H", 1) # Class
def parse_resp(buf, is_ipv6):
typ = 1 # A
if is_ipv6:
typ = 28 # AAAA
id = buf.readbin(">H")
flags = buf.readbin(">H")
assert flags & 0x8000
qcnt = buf.readbin(">H")
acnt = buf.readbin(">H")
nscnt = buf.readbin(">H")
addcnt = buf.readbin(">H")
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print(qcnt, acnt, nscnt, addcnt)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
skip_fqdn(buf)
v = buf.readbin(">H")
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print(v)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
v = buf.readbin(">H")
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print(v)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
for i in range(acnt):
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print("Resp #%d" % i)
# v = read_fqdn(buf)
# print(v)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
skip_fqdn(buf)
t = buf.readbin(">H")
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print("Type", t)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
v = buf.readbin(">H")
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print("Class", v)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
v = buf.readbin(">I")
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print("TTL", v)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
rlen = buf.readbin(">H")
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print("rlen", rlen)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
rval = buf.read(rlen)
all: Run black over all code. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
2021-05-27 05:50:04 +00:00
# print(rval)
udnspkt: A module to create/parse DNS packets to resolve hostnames. This module implements "Sans I/O" approach, where packets are created/ parsed using BytesIO objects, and all networking happens outside the module. The module implements enough functionality to resolve a domain name into IP address, for both IPv4 and IPv6. Other DNS functionality is outside the scope of this module, that's why it's called *u*dnspkt. The API is experimental and subject to change. This module requires .readbin()/.writebin() methods on a stream.
2017-12-17 14:02:35 +00:00
if t == typ:
return rval