A quick glance through this module reveals that most of it "just works"
on Python 3 with the source transformer active. There are a few
places where we need to ensure we're using str.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| Path | Packages | |||
|---|---|---|---|---|
| M | hgext/zeroconf/Zeroconf.py (13 lines) |
| __author__ = "Paul Scott-Murphy" | __author__ = "Paul Scott-Murphy" | ||||
| __email__ = "paul at scott dash murphy dot com" | __email__ = "paul at scott dash murphy dot com" | ||||
| __version__ = "0.12" | __version__ = "0.12" | ||||
| import errno | import errno | ||||
| import itertools | import itertools | ||||
| import select | import select | ||||
| import socket | import socket | ||||
| import string | |||||
| import struct | import struct | ||||
| import threading | import threading | ||||
| import time | import time | ||||
| import traceback | import traceback | ||||
| __all__ = ["Zeroconf", "ServiceInfo", "ServiceBrowser"] | __all__ = ["Zeroconf", "ServiceInfo", "ServiceBrowser"] | ||||
| # hook for threads | # hook for threads | ||||
| globals()['_GLOBAL_DONE'] = 0 | globals()['_GLOBAL_DONE'] = 0 | ||||
| # Some timing constants | # Some timing constants | ||||
| _UNREGISTER_TIME = 125 | _UNREGISTER_TIME = 125 | ||||
| _CHECK_TIME = 175 | _CHECK_TIME = 175 | ||||
| _REGISTER_TIME = 225 | _REGISTER_TIME = 225 | ||||
| _LISTENER_TIME = 200 | _LISTENER_TIME = 200 | ||||
| _BROWSER_TIME = 500 | _BROWSER_TIME = 500 | ||||
| # Some DNS constants | # Some DNS constants | ||||
| _MDNS_ADDR = '224.0.0.251' | _MDNS_ADDR = r'224.0.0.251' | ||||
| _MDNS_PORT = 5353 | _MDNS_PORT = 5353 | ||||
| _DNS_PORT = 53 | _DNS_PORT = 53 | ||||
| _DNS_TTL = 60 * 60 # one hour default TTL | _DNS_TTL = 60 * 60 # one hour default TTL | ||||
| _MAX_MSG_TYPICAL = 1460 # unused | _MAX_MSG_TYPICAL = 1460 # unused | ||||
| _MAX_MSG_ABSOLUTE = 8972 | _MAX_MSG_ABSOLUTE = 8972 | ||||
| _FLAGS_QR_MASK = 0x8000 # query response mask | _FLAGS_QR_MASK = 0x8000 # query response mask | ||||
| pass | pass | ||||
| # implementation classes | # implementation classes | ||||
| class DNSEntry(object): | class DNSEntry(object): | ||||
| """A DNS entry""" | """A DNS entry""" | ||||
| def __init__(self, name, type, clazz): | def __init__(self, name, type, clazz): | ||||
| self.key = string.lower(name) | self.key = name.lower() | ||||
| self.name = name | self.name = name | ||||
| self.type = type | self.type = type | ||||
| self.clazz = clazz & _CLASS_MASK | self.clazz = clazz & _CLASS_MASK | ||||
| self.unique = (clazz & _CLASS_UNIQUE) != 0 | self.unique = (clazz & _CLASS_UNIQUE) != 0 | ||||
| def __eq__(self, other): | def __eq__(self, other): | ||||
| """Equality test on name, type, and class""" | """Equality test on name, type, and class""" | ||||
| if isinstance(other, DNSEntry): | if isinstance(other, DNSEntry): | ||||
| def readName(self): | def readName(self): | ||||
| """Reads a domain name from the packet""" | """Reads a domain name from the packet""" | ||||
| result = '' | result = '' | ||||
| off = self.offset | off = self.offset | ||||
| next = -1 | next = -1 | ||||
| first = off | first = off | ||||
| while True: | while True: | ||||
| len = ord(self.data[off]) | len = ord(self.data[off:off + 1]) | ||||
| off += 1 | off += 1 | ||||
| if len == 0: | if len == 0: | ||||
| break | break | ||||
| t = len & 0xC0 | t = len & 0xC0 | ||||
| if t == 0x00: | if t == 0x00: | ||||
| result = ''.join((result, self.readUTF(off, len) + '.')) | result = ''.join((result, self.readUTF(off, len) + '.')) | ||||
| off += len | off += len | ||||
| elif t == 0xC0: | elif t == 0xC0: | ||||
| if next < 0: | if next < 0: | ||||
| next = off + 1 | next = off + 1 | ||||
| off = ((len & 0x3F) << 8) | ord(self.data[off]) | off = ((len & 0x3F) << 8) | ord(self.data[off:off + 1]) | ||||
| if off >= first: | if off >= first: | ||||
| raise BadDomainNameCircular(off) | raise BadDomainNameCircular(off) | ||||
| first = off | first = off | ||||
| else: | else: | ||||
| raise BadDomainName(off) | raise BadDomainName(off) | ||||
| if next >= 0: | if next >= 0: | ||||
| self.offset = next | self.offset = next | ||||
| self.socket.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_LOOP, "\x01") | self.socket.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_LOOP, "\x01") | ||||
| try: | try: | ||||
| self.socket.bind(self.group) | self.socket.bind(self.group) | ||||
| except Exception: | except Exception: | ||||
| # Some versions of linux raise an exception even though | # Some versions of linux raise an exception even though | ||||
| # SO_REUSEADDR and SO_REUSEPORT have been set, so ignore it | # SO_REUSEADDR and SO_REUSEPORT have been set, so ignore it | ||||
| pass | pass | ||||
| self.socket.setsockopt(socket.SOL_IP, socket.IP_ADD_MEMBERSHIP, | self.socket.setsockopt(socket.SOL_IP, socket.IP_ADD_MEMBERSHIP, | ||||
| socket.inet_aton(_MDNS_ADDR) + socket.inet_aton('0.0.0.0')) | socket.inet_aton(_MDNS_ADDR) + socket.inet_aton(r'0.0.0.0')) | ||||
| self.listeners = [] | self.listeners = [] | ||||
| self.browsers = [] | self.browsers = [] | ||||
| self.services = {} | self.services = {} | ||||
| self.servicetypes = {} | self.servicetypes = {} | ||||
| self.cache = DNSCache() | self.cache = DNSCache() | ||||
| """Ends the background threads, and prevent this instance from | """Ends the background threads, and prevent this instance from | ||||
| servicing further queries.""" | servicing further queries.""" | ||||
| if globals()['_GLOBAL_DONE'] == 0: | if globals()['_GLOBAL_DONE'] == 0: | ||||
| globals()['_GLOBAL_DONE'] = 1 | globals()['_GLOBAL_DONE'] = 1 | ||||
| self.notifyAll() | self.notifyAll() | ||||
| self.engine.notify() | self.engine.notify() | ||||
| self.unregisterAllServices() | self.unregisterAllServices() | ||||
| self.socket.setsockopt(socket.SOL_IP, socket.IP_DROP_MEMBERSHIP, | self.socket.setsockopt(socket.SOL_IP, socket.IP_DROP_MEMBERSHIP, | ||||
| socket.inet_aton(_MDNS_ADDR) + socket.inet_aton('0.0.0.0')) | socket.inet_aton(_MDNS_ADDR) + socket.inet_aton(r'0.0.0.0')) | ||||
| self.socket.close() | self.socket.close() | ||||
| # Test a few module features, including service registration, service | # Test a few module features, including service registration, service | ||||
| # query (for Zoe), and service unregistration. | # query (for Zoe), and service unregistration. | ||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| print("Multicast DNS Service Discovery for Python, version", __version__) | print("Multicast DNS Service Discovery for Python, version", __version__) | ||||
| r = Zeroconf() | r = Zeroconf() | ||||