Gecko [ tsuru-no-tsurugi.tokyo ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2024-09-07 23:29
rdtypes	[ DIR ]	drwxr-xr-x	2024-09-07 23:28
__init__.py	1.56 KB	-rw-r--r--	2020-10-24 23:17
_asyncbackend.py	1.53 KB	-rw-r--r--	2020-08-19 21:18
_asyncio_backend.py	4.34 KB	-rw-r--r--	2020-10-24 23:17
_curio_backend.py	3.33 KB	-rw-r--r--	2020-10-24 23:17
_immutable_attr.py	2.92 KB	-rw-r--r--	2020-10-24 23:17
_immutable_ctx.py	2.4 KB	-rw-r--r--	2020-10-24 23:17
_trio_backend.py	3.68 KB	-rw-r--r--	2020-10-24 23:17
asyncbackend.py	3 KB	-rw-r--r--	2020-10-24 23:17
asyncquery.py	17.4 KB	-rw-r--r--	2020-10-24 23:17
asyncresolver.py	8.71 KB	-rw-r--r--	2020-10-24 23:17
dnssec.py	19.19 KB	-rw-r--r--	2020-10-24 23:17
e164.py	3.6 KB	-rw-r--r--	2020-08-17 01:59
edns.py	10.72 KB	-rw-r--r--	2020-10-24 23:17
entropy.py	4.07 KB	-rw-r--r--	2020-08-17 01:59
enum.py	2.75 KB	-rw-r--r--	2020-10-24 23:17
exception.py	5.13 KB	-rw-r--r--	2020-10-24 23:17
flags.py	2.58 KB	-rw-r--r--	2020-10-24 23:17
grange.py	2.04 KB	-rw-r--r--	2020-10-24 23:17
immutable.py	1.97 KB	-rw-r--r--	2020-10-24 23:17
inet.py	4.74 KB	-rw-r--r--	2020-10-24 23:17
ipv4.py	1.93 KB	-rw-r--r--	2020-08-17 01:59
ipv6.py	5.92 KB	-rw-r--r--	2020-10-24 23:17
message.py	52.73 KB	-rw-r--r--	2021-01-05 04:29
name.py	31.54 KB	-rw-r--r--	2020-10-24 23:17
namedict.py	3.83 KB	-rw-r--r--	2020-08-17 01:59
node.py	6.42 KB	-rw-r--r--	2020-10-24 23:17
opcode.py	2.58 KB	-rw-r--r--	2021-01-05 03:22
py.typed	0 B	-rw-r--r--	2020-06-27 03:20
query.py	40.46 KB	-rw-r--r--	2020-10-24 23:17
rcode.py	3.9 KB	-rw-r--r--	2021-01-05 03:23
rdata.py	24.32 KB	-rw-r--r--	2021-01-06 03:41
rdataclass.py	2.81 KB	-rw-r--r--	2020-10-24 23:17
rdataset.py	14.54 KB	-rw-r--r--	2020-10-24 23:17
rdatatype.py	6.64 KB	-rw-r--r--	2020-10-24 23:17
renderer.py	8.59 KB	-rw-r--r--	2020-08-17 01:59
resolver.py	59.29 KB	-rw-r--r--	2020-12-29 03:07
reversename.py	3.65 KB	-rw-r--r--	2020-08-17 01:59
rrset.py	7.44 KB	-rw-r--r--	2020-10-24 23:17
serial.py	3.53 KB	-rw-r--r--	2020-08-17 01:59
set.py	7.64 KB	-rw-r--r--	2020-10-24 23:17
tokenizer.py	21.86 KB	-rw-r--r--	2020-10-24 23:17
transaction.py	16.93 KB	-rw-r--r--	2020-10-24 23:17
tsig.py	10.93 KB	-rw-r--r--	2020-10-30 22:15
tsigkeyring.py	2.43 KB	-rw-r--r--	2020-10-24 23:17
ttl.py	2.58 KB	-rw-r--r--	2020-10-24 23:17
update.py	10.55 KB	-rw-r--r--	2020-10-24 23:17
version.py	1.55 KB	-rw-r--r--	2021-01-07 18:20
versioned.py	15.91 KB	-rw-r--r--	2020-10-24 23:17
wire.py	2.48 KB	-rw-r--r--	2020-10-24 23:17
xfr.py	11.61 KB	-rw-r--r--	2020-10-24 23:17
zone.py	34.25 KB	-rw-r--r--	2020-10-24 23:17
zonefile.py	15.25 KB	-rw-r--r--	2020-10-24 23:17

Rename

# Copyright (C) Dnspython Contributors, see LICENSE for text of ISC license

"""curio async I/O library query support"""

import socket
import curio
import curio.socket  # type: ignore

import dns._asyncbackend
import dns.exception
import dns.inet

def _maybe_timeout(timeout):
    if timeout:
        return curio.ignore_after(timeout)
    else:
        return dns._asyncbackend.NullContext()

# for brevity
_lltuple = dns.inet.low_level_address_tuple

# pylint: disable=redefined-outer-name

class DatagramSocket(dns._asyncbackend.DatagramSocket):
    def __init__(self, socket):
        self.socket = socket
        self.family = socket.family

async def sendto(self, what, destination, timeout):
        async with _maybe_timeout(timeout):
            return await self.socket.sendto(what, destination)
        raise dns.exception.Timeout(timeout=timeout)  # pragma: no cover

async def recvfrom(self, size, timeout):
        async with _maybe_timeout(timeout):
            return await self.socket.recvfrom(size)
        raise dns.exception.Timeout(timeout=timeout)

async def close(self):
        await self.socket.close()

async def getpeername(self):
        return self.socket.getpeername()

async def getsockname(self):
        return self.socket.getsockname()

class StreamSocket(dns._asyncbackend.DatagramSocket):
    def __init__(self, socket):
        self.socket = socket
        self.family = socket.family

async def sendall(self, what, timeout):
        async with _maybe_timeout(timeout):
            return await self.socket.sendall(what)
        raise dns.exception.Timeout(timeout=timeout)

async def recv(self, size, timeout):
        async with _maybe_timeout(timeout):
            return await self.socket.recv(size)
        raise dns.exception.Timeout(timeout=timeout)

async def close(self):
        await self.socket.close()

async def getpeername(self):
        return self.socket.getpeername()

async def getsockname(self):
        return self.socket.getsockname()

class Backend(dns._asyncbackend.Backend):
    def name(self):
        return 'curio'

async def make_socket(self, af, socktype, proto=0,
                          source=None, destination=None, timeout=None,
                          ssl_context=None, server_hostname=None):
        if socktype == socket.SOCK_DGRAM:
            s = curio.socket.socket(af, socktype, proto)
            try:
                if source:
                    s.bind(_lltuple(source, af))
            except Exception:  # pragma: no cover
                await s.close()
                raise
            return DatagramSocket(s)
        elif socktype == socket.SOCK_STREAM:
            if source:
                source_addr = _lltuple(source, af)
            else:
                source_addr = None
            async with _maybe_timeout(timeout):
                s = await curio.open_connection(destination[0], destination[1],
                                                ssl=ssl_context,
                                                source_addr=source_addr,
                                                server_hostname=server_hostname)
            return StreamSocket(s)
        raise NotImplementedError('unsupported socket ' +
                                  f'type {socktype}')  # pragma: no cover

async def sleep(self, interval):
        await curio.sleep(interval)